]> git.saurik.com Git - apple/xnu.git/blame - bsd/kern/kpi_mbuf.c
xnu-792.tar.gz
[apple/xnu.git] / bsd / kern / kpi_mbuf.c
CommitLineData
91447636
A
1/*
2 * Copyright (c) 2004 Apple Computer, Inc. All rights reserved.
3 *
4 * @APPLE_LICENSE_HEADER_START@
5 *
6 * The contents of this file constitute Original Code as defined in and
7 * are subject to the Apple Public Source License Version 1.1 (the
8 * "License"). You may not use this file except in compliance with the
9 * License. Please obtain a copy of the License at
10 * http://www.apple.com/publicsource and read it before using this file.
11 *
12 * This Original Code and all software distributed under the License are
13 * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
14 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
15 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT. Please see the
17 * License for the specific language governing rights and limitations
18 * under the License.
19 *
20 * @APPLE_LICENSE_HEADER_END@
21 */
22
23#define __KPI__
24//#include <sys/kpi_interface.h>
25
26#include <sys/param.h>
27#include <sys/mbuf.h>
28#include <sys/socket.h>
29#include <kern/debug.h>
30#include <libkern/OSAtomic.h>
31#include <kern/kalloc.h>
32#include <string.h>
33
34void mbuf_tag_id_first_last(u_long *first, u_long *last);
35errno_t mbuf_tag_id_find_internal(const char *string, u_long *out_id, int create);
36
37static const mbuf_flags_t mbuf_flags_mask = MBUF_EXT | MBUF_PKTHDR | MBUF_EOR |
38 MBUF_BCAST | MBUF_MCAST | MBUF_FRAG | MBUF_FIRSTFRAG |
39 MBUF_LASTFRAG | MBUF_PROMISC;
40
41void* mbuf_data(mbuf_t mbuf)
42{
43 return m_mtod(mbuf);
44}
45
46void* mbuf_datastart(mbuf_t mbuf)
47{
48 if (mbuf->m_flags & M_EXT)
49 return mbuf->m_ext.ext_buf;
50 if (mbuf->m_flags & M_PKTHDR)
51 return mbuf->m_pktdat;
52 return mbuf->m_dat;
53}
54
55errno_t mbuf_setdata(mbuf_t mbuf, void* data, size_t len)
56{
57 size_t start = (size_t)((char*)mbuf_datastart(mbuf));
58 size_t maxlen = mbuf_maxlen(mbuf);
59
60 if ((size_t)data < start || ((size_t)data) + len > start + maxlen)
61 return EINVAL;
62 mbuf->m_data = data;
63 mbuf->m_len = len;
64
65 return 0;
66}
67
68errno_t mbuf_align_32(mbuf_t mbuf, size_t len)
69{
70 if ((mbuf->m_flags & M_EXT) != 0 && m_mclhasreference(mbuf))
71 return ENOTSUP;
72 mbuf->m_data = mbuf_datastart(mbuf);
73 mbuf->m_data += ((mbuf_trailingspace(mbuf) - len) &~ (sizeof(u_int32_t) - 1));
74
75 return 0;
76}
77
78addr64_t mbuf_data_to_physical(void* ptr)
79{
80 return (addr64_t)mcl_to_paddr(ptr);
81}
82
83errno_t mbuf_get(mbuf_how_t how, mbuf_type_t type, mbuf_t *mbuf)
84{
85 /* Must set *mbuf to NULL in failure case */
86 *mbuf = m_get(how, type);
87
88 return (*mbuf == NULL) ? ENOMEM : 0;
89}
90
91errno_t mbuf_gethdr(mbuf_how_t how, mbuf_type_t type, mbuf_t *mbuf)
92{
93 /* Must set *mbuf to NULL in failure case */
94 *mbuf = m_gethdr(how, type);
95
96 return (*mbuf == NULL) ? ENOMEM : 0;
97}
98
99extern struct mbuf * m_mbigget(struct mbuf *m, int nowait);
100
101errno_t mbuf_getcluster(mbuf_how_t how, mbuf_type_t type, size_t size, mbuf_t* mbuf)
102{
103 /* Must set *mbuf to NULL in failure case */
104 errno_t error = 0;
105 int created = 0;
106
107 if (mbuf == NULL)
108 return EINVAL;
109 if (*mbuf == NULL) {
110 *mbuf = m_get(how, type);
111 if (*mbuf == NULL)
112 return ENOMEM;
113 created = 1;
114 }
115 /*
116 * At the time this code was written, m_mclget and m_mbigget would always
117 * return the same value that was passed in to it.
118 */
119 if (size == MCLBYTES) {
120 *mbuf = m_mclget(*mbuf, how);
121 } else if (size == NBPG) {
122 *mbuf = m_mbigget(*mbuf, how);
123 } else {
124 error = EINVAL;
125 goto out;
126 }
127 if (*mbuf == NULL || ((*mbuf)->m_flags & M_EXT) == 0)
128 error = ENOMEM;
129out:
130 if (created && error != 0) {
131 error = ENOMEM;
132 mbuf_free(*mbuf);
133 *mbuf = NULL;
134 }
135 return error;
136}
137
138errno_t mbuf_mclget(mbuf_how_t how, mbuf_type_t type, mbuf_t *mbuf)
139{
140 /* Must set *mbuf to NULL in failure case */
141 errno_t error = 0;
142 int created = 0;
143 if (mbuf == NULL) return EINVAL;
144 if (*mbuf == NULL) {
145 error = mbuf_get(how, type, mbuf);
146 if (error)
147 return error;
148 created = 1;
149 }
150
151 /*
152 * At the time this code was written, m_mclget would always
153 * return the same value that was passed in to it.
154 */
155 *mbuf = m_mclget(*mbuf, how);
156
157 if (created && ((*mbuf)->m_flags & M_EXT) == 0) {
158 mbuf_free(*mbuf);
159 *mbuf = NULL;
160 }
161 if (*mbuf == NULL || ((*mbuf)->m_flags & M_EXT) == 0)
162 error = ENOMEM;
163 return error;
164}
165
166
167errno_t mbuf_getpacket(mbuf_how_t how, mbuf_t *mbuf)
168{
169 /* Must set *mbuf to NULL in failure case */
170 errno_t error = 0;
171
172 *mbuf = m_getpacket_how(how);
173
174 if (*mbuf == NULL) {
175 if (how == MBUF_WAITOK)
176 error = ENOMEM;
177 else
178 error = EWOULDBLOCK;
179 }
180
181 return error;
182}
183
184mbuf_t mbuf_free(mbuf_t mbuf)
185{
186 return m_free(mbuf);
187}
188
189void mbuf_freem(mbuf_t mbuf)
190{
191 m_freem(mbuf);
192}
193
194int mbuf_freem_list(mbuf_t mbuf)
195{
196 return m_freem_list(mbuf);
197}
198
199size_t mbuf_leadingspace(mbuf_t mbuf)
200{
201 return m_leadingspace(mbuf);
202}
203
204size_t mbuf_trailingspace(mbuf_t mbuf)
205{
206 return m_trailingspace(mbuf);
207}
208
209/* Manipulation */
210errno_t mbuf_copym(mbuf_t src, size_t offset, size_t len,
211 mbuf_how_t how, mbuf_t *new_mbuf)
212{
213 /* Must set *mbuf to NULL in failure case */
214 *new_mbuf = m_copym(src, offset, len, how);
215
216 return (*new_mbuf == NULL) ? ENOMEM : 0;
217}
218
219errno_t mbuf_dup(mbuf_t src, mbuf_how_t how, mbuf_t *new_mbuf)
220{
221 /* Must set *new_mbuf to NULL in failure case */
222 *new_mbuf = m_dup(src, how);
223
224 return (*new_mbuf == NULL) ? ENOMEM : 0;
225}
226
227errno_t mbuf_prepend(mbuf_t *orig, size_t len, mbuf_how_t how)
228{
229 /* Must set *orig to NULL in failure case */
230 *orig = m_prepend_2(*orig, len, how);
231
232 return (*orig == NULL) ? ENOMEM : 0;
233}
234
235errno_t mbuf_split(mbuf_t src, size_t offset,
236 mbuf_how_t how, mbuf_t *new_mbuf)
237{
238 /* Must set *new_mbuf to NULL in failure case */
239 *new_mbuf = m_split(src, offset, how);
240
241 return (*new_mbuf == NULL) ? ENOMEM : 0;
242}
243
244errno_t mbuf_pullup(mbuf_t *mbuf, size_t len)
245{
246 /* Must set *mbuf to NULL in failure case */
247 *mbuf = m_pullup(*mbuf, len);
248
249 return (*mbuf == NULL) ? ENOMEM : 0;
250}
251
252errno_t mbuf_pulldown(mbuf_t src, size_t *offset, size_t len, mbuf_t *location)
253{
254 /* Must set *location to NULL in failure case */
255 int new_offset;
256 *location = m_pulldown(src, *offset, len, &new_offset);
257 *offset = new_offset;
258
259 return (*location == NULL) ? ENOMEM : 0;
260}
261
262void mbuf_adj(mbuf_t mbuf, int len)
263{
264 m_adj(mbuf, len);
265}
266
267errno_t mbuf_copydata(mbuf_t m, size_t off, size_t len, void* out_data)
268{
269 /* Copied m_copydata, added error handling (don't just panic) */
270 int count;
271
272 while (off > 0) {
273 if (m == 0)
274 return EINVAL;
275 if (off < (size_t)m->m_len)
276 break;
277 off -= m->m_len;
278 m = m->m_next;
279 }
280 while (len > 0) {
281 if (m == 0)
282 return EINVAL;
283 count = m->m_len - off > len ? len : m->m_len - off;
284 bcopy(mtod(m, caddr_t) + off, out_data, count);
285 len -= count;
286 out_data = ((char*)out_data) + count;
287 off = 0;
288 m = m->m_next;
289 }
290
291 return 0;
292}
293
294int mbuf_mclref(mbuf_t mbuf)
295{
296 return m_mclref(mbuf);
297}
298
299int mbuf_mclunref(mbuf_t mbuf)
300{
301 return m_mclunref(mbuf);
302}
303
304int mbuf_mclhasreference(mbuf_t mbuf)
305{
306 if ((mbuf->m_flags & M_EXT))
307 return m_mclhasreference(mbuf);
308 else
309 return 0;
310}
311
312
313/* mbuf header */
314mbuf_t mbuf_next(mbuf_t mbuf)
315{
316 return mbuf->m_next;
317}
318
319errno_t mbuf_setnext(mbuf_t mbuf, mbuf_t next)
320{
321 if (next && ((next)->m_nextpkt != NULL ||
322 (next)->m_type == MT_FREE)) return EINVAL;
323 mbuf->m_next = next;
324
325 return 0;
326}
327
328mbuf_t mbuf_nextpkt(mbuf_t mbuf)
329{
330 return mbuf->m_nextpkt;
331}
332
333void mbuf_setnextpkt(mbuf_t mbuf, mbuf_t nextpkt)
334{
335 mbuf->m_nextpkt = nextpkt;
336}
337
338size_t mbuf_len(mbuf_t mbuf)
339{
340 return mbuf->m_len;
341}
342
343void mbuf_setlen(mbuf_t mbuf, size_t len)
344{
345 mbuf->m_len = len;
346}
347
348size_t mbuf_maxlen(mbuf_t mbuf)
349{
350 if (mbuf->m_flags & M_EXT)
351 return mbuf->m_ext.ext_size;
352 return &mbuf->m_dat[MLEN] - ((char*)mbuf_datastart(mbuf));
353}
354
355mbuf_type_t mbuf_type(mbuf_t mbuf)
356{
357 return mbuf->m_type;
358}
359
360errno_t mbuf_settype(mbuf_t mbuf, mbuf_type_t new_type)
361{
362 if (new_type == MBUF_TYPE_FREE) return EINVAL;
363
364 m_mchtype(mbuf, new_type);
365
366 return 0;
367}
368
369mbuf_flags_t mbuf_flags(mbuf_t mbuf)
370{
371 return mbuf->m_flags & mbuf_flags_mask;
372}
373
374errno_t mbuf_setflags(mbuf_t mbuf, mbuf_flags_t flags)
375{
376 if ((flags & ~mbuf_flags_mask) != 0) return EINVAL;
377 mbuf->m_flags = flags |
378 (mbuf->m_flags & ~mbuf_flags_mask);
379
380 return 0;
381}
382
383errno_t mbuf_setflags_mask(mbuf_t mbuf, mbuf_flags_t flags, mbuf_flags_t mask)
384{
385 if (((flags | mask) & ~mbuf_flags_mask) != 0) return EINVAL;
386
387 mbuf->m_flags = (flags & mask) | (mbuf->m_flags & ~mask);
388
389 return 0;
390}
391
392errno_t mbuf_copy_pkthdr(mbuf_t dest, mbuf_t src)
393{
394 if (((src)->m_flags & M_PKTHDR) == 0)
395 return EINVAL;
396
397 m_copy_pkthdr(dest, src);
398
399 return 0;
400}
401
402size_t mbuf_pkthdr_len(mbuf_t mbuf)
403{
404 return mbuf->m_pkthdr.len;
405}
406
407void mbuf_pkthdr_setlen(mbuf_t mbuf, size_t len)
408{
409 mbuf->m_pkthdr.len = len;
410}
411
412ifnet_t mbuf_pkthdr_rcvif(mbuf_t mbuf)
413{
414 // If we reference count ifnets, we should take a reference here before returning
415 return mbuf->m_pkthdr.rcvif;
416}
417
418errno_t mbuf_pkthdr_setrcvif(mbuf_t mbuf, ifnet_t ifnet)
419{
420 /* May want to walk ifnet list to determine if interface is valid */
421 mbuf->m_pkthdr.rcvif = (struct ifnet*)ifnet;
422 return 0;
423}
424
425void* mbuf_pkthdr_header(mbuf_t mbuf)
426{
427 return mbuf->m_pkthdr.header;
428}
429
430void mbuf_pkthdr_setheader(mbuf_t mbuf, void *header)
431{
432 mbuf->m_pkthdr.header = (void*)header;
433}
434
435/* mbuf aux data */
436errno_t mbuf_aux_add(mbuf_t mbuf, int family, mbuf_type_t type, mbuf_t *aux_mbuf)
437{
438 *aux_mbuf = m_aux_add(mbuf, family, type);
439 return (*aux_mbuf == NULL) ? ENOMEM : 0;
440}
441
442mbuf_t mbuf_aux_find(mbuf_t mbuf, int family, mbuf_type_t type)
443{
444 return m_aux_find(mbuf, family, type);
445}
446
447void mbuf_aux_delete(mbuf_t mbuf, mbuf_t aux)
448{
449 m_aux_delete(mbuf, aux);
450}
451
452void
453mbuf_inbound_modified(mbuf_t mbuf)
454{
455 /* Invalidate hardware generated checksum flags */
456 mbuf->m_pkthdr.csum_flags = 0;
457}
458
459extern void in_cksum_offset(struct mbuf* m, size_t ip_offset);
460extern void in_delayed_cksum_offset(struct mbuf *m, int ip_offset);
461
462void
463mbuf_outbound_finalize(mbuf_t mbuf, u_long protocol_family, size_t protocol_offset)
464{
465 if ((mbuf->m_pkthdr.csum_flags & (CSUM_DELAY_DATA | CSUM_DELAY_IP)) == 0)
466 return;
467
468 /* Generate the packet in software, client needs it */
469 switch (protocol_family) {
470 case PF_INET:
471 if (mbuf->m_pkthdr.csum_flags & CSUM_DELAY_DATA) {
472 in_delayed_cksum_offset(mbuf, protocol_offset);
473 }
474
475 if (mbuf->m_pkthdr.csum_flags & CSUM_DELAY_IP) {
476 in_cksum_offset(mbuf, protocol_offset);
477 }
478
479 mbuf->m_pkthdr.csum_flags &= ~(CSUM_DELAY_DATA | CSUM_DELAY_IP);
480 break;
481
482 default:
483 /*
484 * Not sure what to do here if anything.
485 * Hardware checksum code looked pretty IPv4 specific.
486 */
487 if ((mbuf->m_pkthdr.csum_flags & (CSUM_DELAY_DATA | CSUM_DELAY_IP)) != 0)
488 panic("mbuf_outbound_finalize - CSUM flags set for non-IPv4 packet (%d)!\n", protocol_family);
489 }
490}
491
492errno_t
493mbuf_set_vlan_tag(
494 mbuf_t mbuf,
495 u_int16_t vlan)
496{
497 mbuf->m_pkthdr.csum_flags |= CSUM_VLAN_TAG_VALID;
498 mbuf->m_pkthdr.vlan_tag = vlan;
499
500 return 0;
501}
502
503errno_t
504mbuf_get_vlan_tag(
505 mbuf_t mbuf,
506 u_int16_t *vlan)
507{
508 if ((mbuf->m_pkthdr.csum_flags & CSUM_VLAN_TAG_VALID) == 0)
509 return ENXIO; // No vlan tag set
510
511 *vlan = mbuf->m_pkthdr.vlan_tag;
512
513 return 0;
514}
515
516errno_t
517mbuf_clear_vlan_tag(
518 mbuf_t mbuf)
519{
520 mbuf->m_pkthdr.csum_flags &= ~CSUM_VLAN_TAG_VALID;
521 mbuf->m_pkthdr.vlan_tag = 0;
522
523 return 0;
524}
525
526static const mbuf_csum_request_flags_t mbuf_valid_csum_request_flags =
527 MBUF_CSUM_REQ_IP | MBUF_CSUM_REQ_TCP | MBUF_CSUM_REQ_UDP | MBUF_CSUM_REQ_SUM16;
528
529errno_t
530mbuf_set_csum_requested(
531 mbuf_t mbuf,
532 mbuf_csum_request_flags_t request,
533 u_int32_t value)
534{
535 request &= mbuf_valid_csum_request_flags;
536 mbuf->m_pkthdr.csum_flags = (mbuf->m_pkthdr.csum_flags & 0xffff0000) | request;
537 mbuf->m_pkthdr.csum_data = value;
538
539 return 0;
540}
541
542errno_t
543mbuf_get_csum_requested(
544 mbuf_t mbuf,
545 mbuf_csum_request_flags_t *request,
546 u_int32_t *value)
547{
548 *request = mbuf->m_pkthdr.csum_flags;
549 *request &= mbuf_valid_csum_request_flags;
550 if (value != NULL) {
551 *value = mbuf->m_pkthdr.csum_data;
552 }
553
554 return 0;
555}
556
557errno_t
558mbuf_clear_csum_requested(
559 mbuf_t mbuf)
560{
561 mbuf->m_pkthdr.csum_flags &= 0xffff0000;
562 mbuf->m_pkthdr.csum_data = 0;
563
564 return 0;
565}
566
567static const mbuf_csum_performed_flags_t mbuf_valid_csum_performed_flags =
568 MBUF_CSUM_DID_IP | MBUF_CSUM_IP_GOOD | MBUF_CSUM_DID_DATA |
569 MBUF_CSUM_PSEUDO_HDR | MBUF_CSUM_TCP_SUM16;
570
571errno_t
572mbuf_set_csum_performed(
573 mbuf_t mbuf,
574 mbuf_csum_performed_flags_t performed,
575 u_int32_t value)
576{
577 performed &= mbuf_valid_csum_performed_flags;
578 mbuf->m_pkthdr.csum_flags = (mbuf->m_pkthdr.csum_flags & 0xffff0000) | performed;
579 mbuf->m_pkthdr.csum_data = value;
580
581 return 0;
582}
583
584errno_t
585mbuf_get_csum_performed(
586 mbuf_t mbuf,
587 mbuf_csum_performed_flags_t *performed,
588 u_int32_t *value)
589{
590 *performed = mbuf->m_pkthdr.csum_flags & mbuf_valid_csum_performed_flags;
591 *value = mbuf->m_pkthdr.csum_data;
592
593 return 0;
594}
595
596errno_t
597mbuf_clear_csum_performed(
598 mbuf_t mbuf)
599{
600 mbuf->m_pkthdr.csum_flags &= 0xffff0000;
601 mbuf->m_pkthdr.csum_data = 0;
602
603 return 0;
604}
605
606/*
607 * Mbuf tag KPIs
608 */
609
610struct mbuf_tag_id_entry {
611 SLIST_ENTRY(mbuf_tag_id_entry) next;
612 mbuf_tag_id_t id;
613 char string[];
614};
615
616#define MBUF_TAG_ID_ENTRY_SIZE(__str) \
617 ((size_t)&(((struct mbuf_tag_id_entry*)0)->string[0]) + \
618 strlen(__str) + 1)
619
620#define MTAG_FIRST_ID 1000
621static u_long mtag_id_next = MTAG_FIRST_ID;
622static SLIST_HEAD(,mbuf_tag_id_entry) mtag_id_list = {NULL};
623static lck_mtx_t *mtag_id_lock = NULL;
624
625__private_extern__ void
626mbuf_tag_id_first_last(
627 u_long *first,
628 u_long *last)
629{
630 *first = MTAG_FIRST_ID;
631 *last = mtag_id_next - 1;
632}
633
634__private_extern__ errno_t
635mbuf_tag_id_find_internal(
636 const char *string,
637 u_long *out_id,
638 int create)
639{
640 struct mbuf_tag_id_entry *entry = NULL;
641
642
643 *out_id = 0;
644
645 if (string == NULL || out_id == NULL) {
646 return EINVAL;
647 }
648
649 /* Don't bother allocating the lock if we're only doing a lookup */
650 if (create == 0 && mtag_id_lock == NULL)
651 return ENOENT;
652
653 /* Allocate lock if necessary */
654 if (mtag_id_lock == NULL) {
655 lck_grp_attr_t *grp_attrib = NULL;
656 lck_attr_t *lck_attrb = NULL;
657 lck_grp_t *lck_group = NULL;
658 lck_mtx_t *new_lock = NULL;
659
660 grp_attrib = lck_grp_attr_alloc_init();
661 lck_grp_attr_setdefault(grp_attrib);
662 lck_group = lck_grp_alloc_init("mbuf_tag_allocate_id", grp_attrib);
663 lck_grp_attr_free(grp_attrib);
664 lck_attrb = lck_attr_alloc_init();
665 lck_attr_setdefault(lck_attrb);
666 lck_attr_setdebug(lck_attrb);
667 new_lock = lck_mtx_alloc_init(lck_group, lck_attrb);
668 if (!OSCompareAndSwap((UInt32)0, (UInt32)new_lock, (UInt32*)&mtag_id_lock)) {
669 /*
670 * If the atomic swap fails, someone else has already
671 * done this work. We can free the stuff we allocated.
672 */
673 lck_mtx_free(new_lock, lck_group);
674 lck_grp_free(lck_group);
675 }
676 lck_attr_free(lck_attrb);
677 }
678
679 /* Look for an existing entry */
680 lck_mtx_lock(mtag_id_lock);
681 SLIST_FOREACH(entry, &mtag_id_list, next) {
682 if (strcmp(string, entry->string) == 0) {
683 break;
684 }
685 }
686
687 if (entry == NULL) {
688 if (create == 0) {
689 lck_mtx_unlock(mtag_id_lock);
690 return ENOENT;
691 }
692
693 entry = kalloc(MBUF_TAG_ID_ENTRY_SIZE(string));
694 if (entry == NULL) {
695 lck_mtx_unlock(mtag_id_lock);
696 return ENOMEM;
697 }
698
699 strcpy(entry->string, string);
700 entry->id = mtag_id_next;
701 mtag_id_next++;
702 SLIST_INSERT_HEAD(&mtag_id_list, entry, next);
703 }
704 lck_mtx_unlock(mtag_id_lock);
705
706 *out_id = entry->id;
707
708 return 0;
709}
710
711errno_t
712mbuf_tag_id_find(
713 const char *string,
714 mbuf_tag_id_t *out_id)
715{
716 return mbuf_tag_id_find_internal(string, (u_long*)out_id, 1);
717}
718
719errno_t
720mbuf_tag_allocate(
721 mbuf_t mbuf,
722 mbuf_tag_id_t id,
723 mbuf_tag_type_t type,
724 size_t length,
725 mbuf_how_t how,
726 void** data_p)
727{
728 struct m_tag *tag;
729
730 if (data_p != NULL)
731 *data_p = NULL;
732
733 /* Sanity check parameters */
734 if (mbuf == NULL || (mbuf->m_flags & M_PKTHDR) == 0 || id < MTAG_FIRST_ID ||
735 id >= mtag_id_next || length < 1 || (length & 0xffff0000) != 0 ||
736 data_p == NULL) {
737 return EINVAL;
738 }
739
740 /* Make sure this mtag hasn't already been allocated */
741 tag = m_tag_locate(mbuf, id, type, NULL);
742 if (tag != NULL) {
743 return EEXIST;
744 }
745
746 /* Allocate an mtag */
747 tag = m_tag_alloc(id, type, length, how);
748 if (tag == NULL) {
749 return how == M_WAITOK ? ENOMEM : EWOULDBLOCK;
750 }
751
752 /* Attach the mtag and set *data_p */
753 m_tag_prepend(mbuf, tag);
754 *data_p = tag + 1;
755
756 return 0;
757}
758
759errno_t
760mbuf_tag_find(
761 mbuf_t mbuf,
762 mbuf_tag_id_t id,
763 mbuf_tag_type_t type,
764 size_t* length,
765 void** data_p)
766{
767 struct m_tag *tag;
768
769 if (length != NULL)
770 *length = 0;
771 if (data_p != NULL)
772 *data_p = NULL;
773
774 /* Sanity check parameters */
775 if (mbuf == NULL || (mbuf->m_flags & M_PKTHDR) == 0 || id < MTAG_FIRST_ID ||
776 id >= mtag_id_next || length == NULL || data_p == NULL) {
777 return EINVAL;
778 }
779
780 /* Locate an mtag */
781 tag = m_tag_locate(mbuf, id, type, NULL);
782 if (tag == NULL) {
783 return ENOENT;
784 }
785
786 /* Copy out the pointer to the data and the lenght value */
787 *length = tag->m_tag_len;
788 *data_p = tag + 1;
789
790 return 0;
791}
792
793void
794mbuf_tag_free(
795 mbuf_t mbuf,
796 mbuf_tag_id_t id,
797 mbuf_tag_type_t type)
798{
799 struct m_tag *tag;
800
801 if (mbuf == NULL || (mbuf->m_flags & M_PKTHDR) == 0 || id < MTAG_FIRST_ID ||
802 id >= mtag_id_next)
803 return;
804
805 tag = m_tag_locate(mbuf, id, type, NULL);
806 if (tag == NULL) {
807 return;
808 }
809
810 m_tag_delete(mbuf, tag);
811 return;
812}
813
814/* mbuf stats */
815void mbuf_stats(struct mbuf_stat *stats)
816{
817 stats->mbufs = mbstat.m_mbufs;
818 stats->clusters = mbstat.m_clusters;
819 stats->clfree = mbstat.m_clfree;
820 stats->drops = mbstat.m_drops;
821 stats->wait = mbstat.m_wait;
822 stats->drain = mbstat.m_drain;
823 __builtin_memcpy(stats->mtypes, mbstat.m_mtypes, sizeof(stats->mtypes));
824 stats->mcfail = mbstat.m_mcfail;
825 stats->mpfail = mbstat.m_mpfail;
826 stats->msize = mbstat.m_msize;
827 stats->mclbytes = mbstat.m_mclbytes;
828 stats->minclsize = mbstat.m_minclsize;
829 stats->mlen = mbstat.m_mlen;
830 stats->mhlen = mbstat.m_mhlen;
831 stats->bigclusters = mbstat.m_bigclusters;
832 stats->bigclfree = mbstat.m_bigclfree;
833 stats->bigmclbytes = mbstat.m_bigmclbytes;
834}
835
836errno_t
837mbuf_allocpacket(mbuf_how_t how, size_t packetlen, unsigned int *maxchunks, mbuf_t *mbuf)
838{
839 errno_t error;
840 struct mbuf *m;
841 unsigned int numpkts = 1;
842 unsigned int numchunks = maxchunks ? *maxchunks : 0;
843
844 if (packetlen == 0) {
845 error = EINVAL;
846 goto out;
847 }
848 m = m_allocpacket_internal(&numpkts, packetlen, maxchunks ? &numchunks : NULL, how, 1, 0);
849 if (m == 0) {
850 if (maxchunks && *maxchunks && numchunks > *maxchunks)
851 error = ENOBUFS;
852 else
853 error = ENOMEM;
854 } else {
855 error = 0;
856 *mbuf = m;
857 }
858out:
859 return error;
860}
861
862
863/*
864 * mbuf_copyback differs from m_copyback in a few ways:
865 * 1) mbuf_copyback will allocate clusters for new mbufs we append
866 * 2) mbuf_copyback will grow the last mbuf in the chain if possible
867 * 3) mbuf_copyback reports whether or not the operation succeeded
868 * 4) mbuf_copyback allows the caller to specify M_WAITOK or M_NOWAIT
869 */
870errno_t
871mbuf_copyback(
872 mbuf_t m,
873 size_t off,
874 size_t len,
875 const void *data,
876 mbuf_how_t how)
877{
878 size_t mlen;
879 mbuf_t m_start = m;
880 mbuf_t n;
881 int totlen = 0;
882 errno_t result = 0;
883 const char *cp = data;
884
885 if (m == NULL || len == 0 || data == NULL)
886 return EINVAL;
887
888 while (off > (mlen = m->m_len)) {
889 off -= mlen;
890 totlen += mlen;
891 if (m->m_next == 0) {
892 n = m_getclr(how, m->m_type);
893 if (n == 0) {
894 result = ENOBUFS;
895 goto out;
896 }
897 n->m_len = MIN(MLEN, len + off);
898 m->m_next = n;
899 }
900 m = m->m_next;
901 }
902
903 while (len > 0) {
904 mlen = MIN(m->m_len - off, len);
905 if (mlen < len && m->m_next == NULL && mbuf_trailingspace(m) > 0) {
906 size_t grow = MIN(mbuf_trailingspace(m), len - mlen);
907 mlen += grow;
908 m->m_len += grow;
909 }
910 bcopy(cp, off + (char*)mbuf_data(m), (unsigned)mlen);
911 cp += mlen;
912 len -= mlen;
913 mlen += off;
914 off = 0;
915 totlen += mlen;
916 if (len == 0)
917 break;
918 if (m->m_next == 0) {
919 n = m_get(how, m->m_type);
920 if (n == NULL) {
921 result = ENOBUFS;
922 goto out;
923 }
924 if (len > MINCLSIZE) {
925 /* cluter allocation failure is okay, we can grow chain */
926 mbuf_mclget(how, m->m_type, &n);
927 }
928 n->m_len = MIN(mbuf_maxlen(n), len);
929 m->m_next = n;
930 }
931 m = m->m_next;
932 }
933
934out:
935 if ((m_start->m_flags & M_PKTHDR) && (m_start->m_pkthdr.len < totlen))
936 m_start->m_pkthdr.len = totlen;
937
938 return result;
939}