utils.c 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276
  1. /*
  2. * Copyright (c) 2010 Broadcom Corporation
  3. *
  4. * Permission to use, copy, modify, and/or distribute this software for any
  5. * purpose with or without fee is hereby granted, provided that the above
  6. * copyright notice and this permission notice appear in all copies.
  7. *
  8. * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9. * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10. * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
  11. * SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12. * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
  13. * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
  14. * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15. */
  16. #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  17. #include <linux/netdevice.h>
  18. #include <linux/module.h>
  19. #include <brcmu_utils.h>
  20. MODULE_AUTHOR("Broadcom Corporation");
  21. MODULE_DESCRIPTION("Broadcom 802.11n wireless LAN driver utilities.");
  22. MODULE_SUPPORTED_DEVICE("Broadcom 802.11n WLAN cards");
  23. MODULE_LICENSE("Dual BSD/GPL");
  24. struct sk_buff *brcmu_pkt_buf_get_skb(uint len)
  25. {
  26. struct sk_buff *skb;
  27. skb = dev_alloc_skb(len);
  28. if (skb) {
  29. skb_put(skb, len);
  30. skb->priority = 0;
  31. }
  32. return skb;
  33. }
  34. EXPORT_SYMBOL(brcmu_pkt_buf_get_skb);
  35. /* Free the driver packet. Free the tag if present */
  36. void brcmu_pkt_buf_free_skb(struct sk_buff *skb)
  37. {
  38. WARN_ON(skb->next);
  39. if (skb->destructor)
  40. /* cannot kfree_skb() on hard IRQ (net/core/skbuff.c) if
  41. * destructor exists
  42. */
  43. dev_kfree_skb_any(skb);
  44. else
  45. /* can free immediately (even in_irq()) if destructor
  46. * does not exist
  47. */
  48. dev_kfree_skb(skb);
  49. }
  50. EXPORT_SYMBOL(brcmu_pkt_buf_free_skb);
  51. /*
  52. * osl multiple-precedence packet queue
  53. * hi_prec is always >= the number of the highest non-empty precedence
  54. */
  55. struct sk_buff *brcmu_pktq_penq(struct pktq *pq, int prec,
  56. struct sk_buff *p)
  57. {
  58. struct sk_buff_head *q;
  59. if (pktq_full(pq) || pktq_pfull(pq, prec))
  60. return NULL;
  61. q = &pq->q[prec].skblist;
  62. skb_queue_tail(q, p);
  63. pq->len++;
  64. if (pq->hi_prec < prec)
  65. pq->hi_prec = (u8) prec;
  66. return p;
  67. }
  68. EXPORT_SYMBOL(brcmu_pktq_penq);
  69. struct sk_buff *brcmu_pktq_penq_head(struct pktq *pq, int prec,
  70. struct sk_buff *p)
  71. {
  72. struct sk_buff_head *q;
  73. if (pktq_full(pq) || pktq_pfull(pq, prec))
  74. return NULL;
  75. q = &pq->q[prec].skblist;
  76. skb_queue_head(q, p);
  77. pq->len++;
  78. if (pq->hi_prec < prec)
  79. pq->hi_prec = (u8) prec;
  80. return p;
  81. }
  82. EXPORT_SYMBOL(brcmu_pktq_penq_head);
  83. struct sk_buff *brcmu_pktq_pdeq(struct pktq *pq, int prec)
  84. {
  85. struct sk_buff_head *q;
  86. struct sk_buff *p;
  87. q = &pq->q[prec].skblist;
  88. p = skb_dequeue(q);
  89. if (p == NULL)
  90. return NULL;
  91. pq->len--;
  92. return p;
  93. }
  94. EXPORT_SYMBOL(brcmu_pktq_pdeq);
  95. struct sk_buff *brcmu_pktq_pdeq_tail(struct pktq *pq, int prec)
  96. {
  97. struct sk_buff_head *q;
  98. struct sk_buff *p;
  99. q = &pq->q[prec].skblist;
  100. p = skb_dequeue_tail(q);
  101. if (p == NULL)
  102. return NULL;
  103. pq->len--;
  104. return p;
  105. }
  106. EXPORT_SYMBOL(brcmu_pktq_pdeq_tail);
  107. void
  108. brcmu_pktq_pflush(struct pktq *pq, int prec, bool dir,
  109. bool (*fn)(struct sk_buff *, void *), void *arg)
  110. {
  111. struct sk_buff_head *q;
  112. struct sk_buff *p, *next;
  113. q = &pq->q[prec].skblist;
  114. skb_queue_walk_safe(q, p, next) {
  115. if (fn == NULL || (*fn) (p, arg)) {
  116. skb_unlink(p, q);
  117. brcmu_pkt_buf_free_skb(p);
  118. pq->len--;
  119. }
  120. }
  121. }
  122. EXPORT_SYMBOL(brcmu_pktq_pflush);
  123. void brcmu_pktq_flush(struct pktq *pq, bool dir,
  124. bool (*fn)(struct sk_buff *, void *), void *arg)
  125. {
  126. int prec;
  127. for (prec = 0; prec < pq->num_prec; prec++)
  128. brcmu_pktq_pflush(pq, prec, dir, fn, arg);
  129. }
  130. EXPORT_SYMBOL(brcmu_pktq_flush);
  131. void brcmu_pktq_init(struct pktq *pq, int num_prec, int max_len)
  132. {
  133. int prec;
  134. /* pq is variable size; only zero out what's requested */
  135. memset(pq, 0,
  136. offsetof(struct pktq, q) + (sizeof(struct pktq_prec) * num_prec));
  137. pq->num_prec = (u16) num_prec;
  138. pq->max = (u16) max_len;
  139. for (prec = 0; prec < num_prec; prec++) {
  140. pq->q[prec].max = pq->max;
  141. skb_queue_head_init(&pq->q[prec].skblist);
  142. }
  143. }
  144. EXPORT_SYMBOL(brcmu_pktq_init);
  145. struct sk_buff *brcmu_pktq_peek_tail(struct pktq *pq, int *prec_out)
  146. {
  147. int prec;
  148. if (pq->len == 0)
  149. return NULL;
  150. for (prec = 0; prec < pq->hi_prec; prec++)
  151. if (!skb_queue_empty(&pq->q[prec].skblist))
  152. break;
  153. if (prec_out)
  154. *prec_out = prec;
  155. return skb_peek_tail(&pq->q[prec].skblist);
  156. }
  157. EXPORT_SYMBOL(brcmu_pktq_peek_tail);
  158. /* Return sum of lengths of a specific set of precedences */
  159. int brcmu_pktq_mlen(struct pktq *pq, uint prec_bmp)
  160. {
  161. int prec, len;
  162. len = 0;
  163. for (prec = 0; prec <= pq->hi_prec; prec++)
  164. if (prec_bmp & (1 << prec))
  165. len += pq->q[prec].skblist.qlen;
  166. return len;
  167. }
  168. EXPORT_SYMBOL(brcmu_pktq_mlen);
  169. /* Priority dequeue from a specific set of precedences */
  170. struct sk_buff *brcmu_pktq_mdeq(struct pktq *pq, uint prec_bmp,
  171. int *prec_out)
  172. {
  173. struct sk_buff_head *q;
  174. struct sk_buff *p;
  175. int prec;
  176. if (pq->len == 0)
  177. return NULL;
  178. while ((prec = pq->hi_prec) > 0 &&
  179. skb_queue_empty(&pq->q[prec].skblist))
  180. pq->hi_prec--;
  181. while ((prec_bmp & (1 << prec)) == 0 ||
  182. skb_queue_empty(&pq->q[prec].skblist))
  183. if (prec-- == 0)
  184. return NULL;
  185. q = &pq->q[prec].skblist;
  186. p = skb_dequeue(q);
  187. if (p == NULL)
  188. return NULL;
  189. pq->len--;
  190. if (prec_out)
  191. *prec_out = prec;
  192. return p;
  193. }
  194. EXPORT_SYMBOL(brcmu_pktq_mdeq);
  195. #if defined(DEBUG)
  196. /* pretty hex print a pkt buffer chain */
  197. void brcmu_prpkt(const char *msg, struct sk_buff *p0)
  198. {
  199. struct sk_buff *p;
  200. if (msg && (msg[0] != '\0'))
  201. pr_debug("%s:\n", msg);
  202. for (p = p0; p; p = p->next)
  203. print_hex_dump_bytes("", DUMP_PREFIX_OFFSET, p->data, p->len);
  204. }
  205. EXPORT_SYMBOL(brcmu_prpkt);
  206. void brcmu_dbg_hex_dump(const void *data, size_t size, const char *fmt, ...)
  207. {
  208. struct va_format vaf;
  209. va_list args;
  210. va_start(args, fmt);
  211. vaf.fmt = fmt;
  212. vaf.va = &args;
  213. pr_debug("%pV", &vaf);
  214. va_end(args);
  215. print_hex_dump_bytes("", DUMP_PREFIX_OFFSET, data, size);
  216. }
  217. EXPORT_SYMBOL(brcmu_dbg_hex_dump);
  218. #endif /* defined(DEBUG) */