utils.c 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278
  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. if (!skb)
  39. return;
  40. WARN_ON(skb->next);
  41. if (skb->destructor)
  42. /* cannot kfree_skb() on hard IRQ (net/core/skbuff.c) if
  43. * destructor exists
  44. */
  45. dev_kfree_skb_any(skb);
  46. else
  47. /* can free immediately (even in_irq()) if destructor
  48. * does not exist
  49. */
  50. dev_kfree_skb(skb);
  51. }
  52. EXPORT_SYMBOL(brcmu_pkt_buf_free_skb);
  53. /*
  54. * osl multiple-precedence packet queue
  55. * hi_prec is always >= the number of the highest non-empty precedence
  56. */
  57. struct sk_buff *brcmu_pktq_penq(struct pktq *pq, int prec,
  58. struct sk_buff *p)
  59. {
  60. struct sk_buff_head *q;
  61. if (pktq_full(pq) || pktq_pfull(pq, prec))
  62. return NULL;
  63. q = &pq->q[prec].skblist;
  64. skb_queue_tail(q, p);
  65. pq->len++;
  66. if (pq->hi_prec < prec)
  67. pq->hi_prec = (u8) prec;
  68. return p;
  69. }
  70. EXPORT_SYMBOL(brcmu_pktq_penq);
  71. struct sk_buff *brcmu_pktq_penq_head(struct pktq *pq, int prec,
  72. struct sk_buff *p)
  73. {
  74. struct sk_buff_head *q;
  75. if (pktq_full(pq) || pktq_pfull(pq, prec))
  76. return NULL;
  77. q = &pq->q[prec].skblist;
  78. skb_queue_head(q, p);
  79. pq->len++;
  80. if (pq->hi_prec < prec)
  81. pq->hi_prec = (u8) prec;
  82. return p;
  83. }
  84. EXPORT_SYMBOL(brcmu_pktq_penq_head);
  85. struct sk_buff *brcmu_pktq_pdeq(struct pktq *pq, int prec)
  86. {
  87. struct sk_buff_head *q;
  88. struct sk_buff *p;
  89. q = &pq->q[prec].skblist;
  90. p = skb_dequeue(q);
  91. if (p == NULL)
  92. return NULL;
  93. pq->len--;
  94. return p;
  95. }
  96. EXPORT_SYMBOL(brcmu_pktq_pdeq);
  97. struct sk_buff *brcmu_pktq_pdeq_tail(struct pktq *pq, int prec)
  98. {
  99. struct sk_buff_head *q;
  100. struct sk_buff *p;
  101. q = &pq->q[prec].skblist;
  102. p = skb_dequeue_tail(q);
  103. if (p == NULL)
  104. return NULL;
  105. pq->len--;
  106. return p;
  107. }
  108. EXPORT_SYMBOL(brcmu_pktq_pdeq_tail);
  109. void
  110. brcmu_pktq_pflush(struct pktq *pq, int prec, bool dir,
  111. bool (*fn)(struct sk_buff *, void *), void *arg)
  112. {
  113. struct sk_buff_head *q;
  114. struct sk_buff *p, *next;
  115. q = &pq->q[prec].skblist;
  116. skb_queue_walk_safe(q, p, next) {
  117. if (fn == NULL || (*fn) (p, arg)) {
  118. skb_unlink(p, q);
  119. brcmu_pkt_buf_free_skb(p);
  120. pq->len--;
  121. }
  122. }
  123. }
  124. EXPORT_SYMBOL(brcmu_pktq_pflush);
  125. void brcmu_pktq_flush(struct pktq *pq, bool dir,
  126. bool (*fn)(struct sk_buff *, void *), void *arg)
  127. {
  128. int prec;
  129. for (prec = 0; prec < pq->num_prec; prec++)
  130. brcmu_pktq_pflush(pq, prec, dir, fn, arg);
  131. }
  132. EXPORT_SYMBOL(brcmu_pktq_flush);
  133. void brcmu_pktq_init(struct pktq *pq, int num_prec, int max_len)
  134. {
  135. int prec;
  136. /* pq is variable size; only zero out what's requested */
  137. memset(pq, 0,
  138. offsetof(struct pktq, q) + (sizeof(struct pktq_prec) * num_prec));
  139. pq->num_prec = (u16) num_prec;
  140. pq->max = (u16) max_len;
  141. for (prec = 0; prec < num_prec; prec++) {
  142. pq->q[prec].max = pq->max;
  143. skb_queue_head_init(&pq->q[prec].skblist);
  144. }
  145. }
  146. EXPORT_SYMBOL(brcmu_pktq_init);
  147. struct sk_buff *brcmu_pktq_peek_tail(struct pktq *pq, int *prec_out)
  148. {
  149. int prec;
  150. if (pq->len == 0)
  151. return NULL;
  152. for (prec = 0; prec < pq->hi_prec; prec++)
  153. if (!skb_queue_empty(&pq->q[prec].skblist))
  154. break;
  155. if (prec_out)
  156. *prec_out = prec;
  157. return skb_peek_tail(&pq->q[prec].skblist);
  158. }
  159. EXPORT_SYMBOL(brcmu_pktq_peek_tail);
  160. /* Return sum of lengths of a specific set of precedences */
  161. int brcmu_pktq_mlen(struct pktq *pq, uint prec_bmp)
  162. {
  163. int prec, len;
  164. len = 0;
  165. for (prec = 0; prec <= pq->hi_prec; prec++)
  166. if (prec_bmp & (1 << prec))
  167. len += pq->q[prec].skblist.qlen;
  168. return len;
  169. }
  170. EXPORT_SYMBOL(brcmu_pktq_mlen);
  171. /* Priority dequeue from a specific set of precedences */
  172. struct sk_buff *brcmu_pktq_mdeq(struct pktq *pq, uint prec_bmp,
  173. int *prec_out)
  174. {
  175. struct sk_buff_head *q;
  176. struct sk_buff *p;
  177. int prec;
  178. if (pq->len == 0)
  179. return NULL;
  180. while ((prec = pq->hi_prec) > 0 &&
  181. skb_queue_empty(&pq->q[prec].skblist))
  182. pq->hi_prec--;
  183. while ((prec_bmp & (1 << prec)) == 0 ||
  184. skb_queue_empty(&pq->q[prec].skblist))
  185. if (prec-- == 0)
  186. return NULL;
  187. q = &pq->q[prec].skblist;
  188. p = skb_dequeue(q);
  189. if (p == NULL)
  190. return NULL;
  191. pq->len--;
  192. if (prec_out)
  193. *prec_out = prec;
  194. return p;
  195. }
  196. EXPORT_SYMBOL(brcmu_pktq_mdeq);
  197. #if defined(DEBUG)
  198. /* pretty hex print a pkt buffer chain */
  199. void brcmu_prpkt(const char *msg, struct sk_buff *p0)
  200. {
  201. struct sk_buff *p;
  202. if (msg && (msg[0] != '\0'))
  203. pr_debug("%s:\n", msg);
  204. for (p = p0; p; p = p->next)
  205. print_hex_dump_bytes("", DUMP_PREFIX_OFFSET, p->data, p->len);
  206. }
  207. EXPORT_SYMBOL(brcmu_prpkt);
  208. void brcmu_dbg_hex_dump(const void *data, size_t size, const char *fmt, ...)
  209. {
  210. struct va_format vaf;
  211. va_list args;
  212. va_start(args, fmt);
  213. vaf.fmt = fmt;
  214. vaf.va = &args;
  215. pr_debug("%pV", &vaf);
  216. va_end(args);
  217. print_hex_dump_bytes("", DUMP_PREFIX_OFFSET, data, size);
  218. }
  219. EXPORT_SYMBOL(brcmu_dbg_hex_dump);
  220. #endif /* defined(DEBUG) */