utils.c 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337
  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. #include <linux/netdevice.h>
  17. #include <linux/module.h>
  18. #include <brcmu_utils.h>
  19. MODULE_AUTHOR("Broadcom Corporation");
  20. MODULE_DESCRIPTION("Broadcom 802.11n wireless LAN driver utilities.");
  21. MODULE_SUPPORTED_DEVICE("Broadcom 802.11n WLAN cards");
  22. MODULE_LICENSE("Dual BSD/GPL");
  23. struct sk_buff *brcmu_pkt_buf_get_skb(uint len)
  24. {
  25. struct sk_buff *skb;
  26. skb = dev_alloc_skb(len);
  27. if (skb) {
  28. skb_put(skb, len);
  29. skb->priority = 0;
  30. }
  31. return skb;
  32. }
  33. EXPORT_SYMBOL(brcmu_pkt_buf_get_skb);
  34. /* Free the driver packet. Free the tag if present */
  35. void brcmu_pkt_buf_free_skb(struct sk_buff *skb)
  36. {
  37. struct sk_buff *nskb;
  38. int nest = 0;
  39. /* perversion: we use skb->next to chain multi-skb packets */
  40. while (skb) {
  41. nskb = skb->next;
  42. skb->next = NULL;
  43. if (skb->destructor)
  44. /* cannot kfree_skb() on hard IRQ (net/core/skbuff.c) if
  45. * destructor exists
  46. */
  47. dev_kfree_skb_any(skb);
  48. else
  49. /* can free immediately (even in_irq()) if destructor
  50. * does not exist
  51. */
  52. dev_kfree_skb(skb);
  53. nest++;
  54. skb = nskb;
  55. }
  56. }
  57. EXPORT_SYMBOL(brcmu_pkt_buf_free_skb);
  58. /* return total length of buffer chain */
  59. uint brcmu_pkttotlen(struct sk_buff *p)
  60. {
  61. uint total;
  62. total = 0;
  63. for (; p; p = p->next)
  64. total += p->len;
  65. return total;
  66. }
  67. EXPORT_SYMBOL(brcmu_pkttotlen);
  68. /*
  69. * osl multiple-precedence packet queue
  70. * hi_prec is always >= the number of the highest non-empty precedence
  71. */
  72. struct sk_buff *brcmu_pktq_penq(struct pktq *pq, int prec,
  73. struct sk_buff *p)
  74. {
  75. struct pktq_prec *q;
  76. if (pktq_full(pq) || pktq_pfull(pq, prec))
  77. return NULL;
  78. q = &pq->q[prec];
  79. if (q->head)
  80. q->tail->prev = p;
  81. else
  82. q->head = p;
  83. q->tail = p;
  84. q->len++;
  85. pq->len++;
  86. if (pq->hi_prec < prec)
  87. pq->hi_prec = (u8) prec;
  88. return p;
  89. }
  90. EXPORT_SYMBOL(brcmu_pktq_penq);
  91. struct sk_buff *brcmu_pktq_penq_head(struct pktq *pq, int prec,
  92. struct sk_buff *p)
  93. {
  94. struct pktq_prec *q;
  95. if (pktq_full(pq) || pktq_pfull(pq, prec))
  96. return NULL;
  97. q = &pq->q[prec];
  98. if (q->head == NULL)
  99. q->tail = p;
  100. p->prev = q->head;
  101. q->head = p;
  102. q->len++;
  103. pq->len++;
  104. if (pq->hi_prec < prec)
  105. pq->hi_prec = (u8) prec;
  106. return p;
  107. }
  108. EXPORT_SYMBOL(brcmu_pktq_penq_head);
  109. struct sk_buff *brcmu_pktq_pdeq(struct pktq *pq, int prec)
  110. {
  111. struct pktq_prec *q;
  112. struct sk_buff *p;
  113. q = &pq->q[prec];
  114. p = q->head;
  115. if (p == NULL)
  116. return NULL;
  117. q->head = p->prev;
  118. if (q->head == NULL)
  119. q->tail = NULL;
  120. q->len--;
  121. pq->len--;
  122. p->prev = NULL;
  123. return p;
  124. }
  125. EXPORT_SYMBOL(brcmu_pktq_pdeq);
  126. struct sk_buff *brcmu_pktq_pdeq_tail(struct pktq *pq, int prec)
  127. {
  128. struct pktq_prec *q;
  129. struct sk_buff *p, *prev;
  130. q = &pq->q[prec];
  131. p = q->head;
  132. if (p == NULL)
  133. return NULL;
  134. for (prev = NULL; p != q->tail; p = p->prev)
  135. prev = p;
  136. if (prev)
  137. prev->prev = NULL;
  138. else
  139. q->head = NULL;
  140. q->tail = prev;
  141. q->len--;
  142. pq->len--;
  143. return p;
  144. }
  145. EXPORT_SYMBOL(brcmu_pktq_pdeq_tail);
  146. void
  147. brcmu_pktq_pflush(struct pktq *pq, int prec, bool dir,
  148. bool (*fn)(struct sk_buff *, void *), void *arg)
  149. {
  150. struct pktq_prec *q;
  151. struct sk_buff *p, *prev = NULL;
  152. q = &pq->q[prec];
  153. p = q->head;
  154. while (p) {
  155. if (fn == NULL || (*fn) (p, arg)) {
  156. bool head = (p == q->head);
  157. if (head)
  158. q->head = p->prev;
  159. else
  160. prev->prev = p->prev;
  161. p->prev = NULL;
  162. brcmu_pkt_buf_free_skb(p);
  163. q->len--;
  164. pq->len--;
  165. p = (head ? q->head : prev->prev);
  166. } else {
  167. prev = p;
  168. p = p->prev;
  169. }
  170. }
  171. if (q->head == NULL)
  172. q->tail = NULL;
  173. }
  174. EXPORT_SYMBOL(brcmu_pktq_pflush);
  175. void brcmu_pktq_flush(struct pktq *pq, bool dir,
  176. bool (*fn)(struct sk_buff *, void *), void *arg)
  177. {
  178. int prec;
  179. for (prec = 0; prec < pq->num_prec; prec++)
  180. brcmu_pktq_pflush(pq, prec, dir, fn, arg);
  181. }
  182. EXPORT_SYMBOL(brcmu_pktq_flush);
  183. void brcmu_pktq_init(struct pktq *pq, int num_prec, int max_len)
  184. {
  185. int prec;
  186. /* pq is variable size; only zero out what's requested */
  187. memset(pq, 0,
  188. offsetof(struct pktq, q) + (sizeof(struct pktq_prec) * num_prec));
  189. pq->num_prec = (u16) num_prec;
  190. pq->max = (u16) max_len;
  191. for (prec = 0; prec < num_prec; prec++)
  192. pq->q[prec].max = pq->max;
  193. }
  194. EXPORT_SYMBOL(brcmu_pktq_init);
  195. struct sk_buff *brcmu_pktq_peek_tail(struct pktq *pq, int *prec_out)
  196. {
  197. int prec;
  198. if (pq->len == 0)
  199. return NULL;
  200. for (prec = 0; prec < pq->hi_prec; prec++)
  201. if (pq->q[prec].head)
  202. break;
  203. if (prec_out)
  204. *prec_out = prec;
  205. return pq->q[prec].tail;
  206. }
  207. EXPORT_SYMBOL(brcmu_pktq_peek_tail);
  208. /* Return sum of lengths of a specific set of precedences */
  209. int brcmu_pktq_mlen(struct pktq *pq, uint prec_bmp)
  210. {
  211. int prec, len;
  212. len = 0;
  213. for (prec = 0; prec <= pq->hi_prec; prec++)
  214. if (prec_bmp & (1 << prec))
  215. len += pq->q[prec].len;
  216. return len;
  217. }
  218. EXPORT_SYMBOL(brcmu_pktq_mlen);
  219. /* Priority dequeue from a specific set of precedences */
  220. struct sk_buff *brcmu_pktq_mdeq(struct pktq *pq, uint prec_bmp,
  221. int *prec_out)
  222. {
  223. struct pktq_prec *q;
  224. struct sk_buff *p;
  225. int prec;
  226. if (pq->len == 0)
  227. return NULL;
  228. while ((prec = pq->hi_prec) > 0 && pq->q[prec].head == NULL)
  229. pq->hi_prec--;
  230. while ((prec_bmp & (1 << prec)) == 0 || pq->q[prec].head == NULL)
  231. if (prec-- == 0)
  232. return NULL;
  233. q = &pq->q[prec];
  234. p = q->head;
  235. if (p == NULL)
  236. return NULL;
  237. q->head = p->prev;
  238. if (q->head == NULL)
  239. q->tail = NULL;
  240. q->len--;
  241. if (prec_out)
  242. *prec_out = prec;
  243. pq->len--;
  244. p->prev = NULL;
  245. return p;
  246. }
  247. EXPORT_SYMBOL(brcmu_pktq_mdeq);
  248. #if defined(BCMDBG)
  249. /* pretty hex print a pkt buffer chain */
  250. void brcmu_prpkt(const char *msg, struct sk_buff *p0)
  251. {
  252. struct sk_buff *p;
  253. if (msg && (msg[0] != '\0'))
  254. printk(KERN_DEBUG "%s:\n", msg);
  255. for (p = p0; p; p = p->next)
  256. print_hex_dump_bytes("", DUMP_PREFIX_OFFSET, p->data, p->len);
  257. }
  258. EXPORT_SYMBOL(brcmu_prpkt);
  259. #endif /* defined(BCMDBG) */