1 /*
2 * Copyright (c) 2008-2021 Apple Inc. All rights reserved.
3 *
4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5 *
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. The rights granted to you under the License
10 * may not be used to create, or enable the creation or redistribution of,
11 * unlawful or unlicensed copies of an Apple operating system, or to
12 * circumvent, violate, or enable the circumvention or violation of, any
13 * terms of an Apple operating system software license agreement.
14 *
15 * Please obtain a copy of the License at
16 * http://www.opensource.apple.com/apsl/ and read it before using this file.
17 *
18 * The Original Code and all software distributed under the License are
19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23 * Please see the License for the specific language governing rights and
24 * limitations under the License.
25 *
26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27 */
28
29 #include <sys/types.h>
30 #include <kern/locks.h>
31 #include <kern/zalloc.h>
32 #include <sys/errno.h>
33 #include <sys/sysctl.h>
34 #include <sys/malloc.h>
35 #include <sys/socket.h>
36 #include <libkern/OSAtomic.h>
37 #include <libkern/libkern.h>
38 #include <net/if.h>
39 #include <net/if_mib.h>
40 #include <string.h>
41
42 #include "net/net_str_id.h"
43
44 #define NET_ID_STR_MAX_LEN 2048
45 #define NET_ID_STR_ENTRY_SIZE(__str) \
46 (__builtin_offsetof(struct net_str_id_entry, nsi_string[0]) + \
47 strlen(__str) + 1)
48
49 #define FIRST_NET_STR_ID 1000
50 static SLIST_HEAD(, net_str_id_entry) net_str_id_list = {NULL};
51 static LCK_GRP_DECLARE(net_str_id_grp, "mbuf_tag_allocate_id");
52 static LCK_MTX_DECLARE(net_str_id_lock, &net_str_id_grp);
53
54 static u_int32_t nsi_kind_next[NSI_MAX_KIND] = { FIRST_NET_STR_ID, FIRST_NET_STR_ID, FIRST_NET_STR_ID };
55 static u_int32_t nsi_next_id = FIRST_NET_STR_ID;
56
57 extern int sysctl_if_family_ids SYSCTL_HANDLER_ARGS;
58
59 SYSCTL_DECL(_net_link_generic_system);
60
61 SYSCTL_PROC(_net_link_generic_system, OID_AUTO, if_family_ids, CTLTYPE_STRUCT | CTLFLAG_RD | CTLFLAG_LOCKED,
62 0, 0, sysctl_if_family_ids, "S, if_family_id", "Interface Family ID table");
63
64 __private_extern__ void
net_str_id_first_last(u_int32_t * first,u_int32_t * last,u_int32_t kind)65 net_str_id_first_last(u_int32_t *first, u_int32_t *last, u_int32_t kind)
66 {
67 *first = FIRST_NET_STR_ID;
68
69 switch (kind) {
70 case NSI_MBUF_TAG:
71 case NSI_VENDOR_CODE:
72 case NSI_IF_FAM_ID:
73 *last = nsi_kind_next[kind] - 1;
74 break;
75 default:
76 *last = FIRST_NET_STR_ID - 1;
77 break;
78 }
79 }
80
81 __private_extern__ errno_t
net_str_id_find_internal(const char * string,u_int32_t * out_id,u_int32_t kind,int create)82 net_str_id_find_internal(const char *string, u_int32_t *out_id,
83 u_int32_t kind, int create)
84 {
85 struct net_str_id_entry *entry = NULL;
86
87
88 if (string == NULL || out_id == NULL || kind >= NSI_MAX_KIND) {
89 return EINVAL;
90 }
91 if (strlen(string) > NET_ID_STR_MAX_LEN) {
92 return EINVAL;
93 }
94
95 *out_id = 0;
96
97 /* Look for an existing entry */
98 lck_mtx_lock(&net_str_id_lock);
99 SLIST_FOREACH(entry, &net_str_id_list, nsi_next) {
100 if (strcmp(string, entry->nsi_string) == 0) {
101 break;
102 }
103 }
104
105 if (entry == NULL) {
106 if (create == 0) {
107 lck_mtx_unlock(&net_str_id_lock);
108 return ENOENT;
109 }
110
111 entry = zalloc_permanent(NET_ID_STR_ENTRY_SIZE(string),
112 ZALIGN_PTR);
113 if (entry == NULL) {
114 lck_mtx_unlock(&net_str_id_lock);
115 return ENOMEM;
116 }
117
118 strlcpy(entry->nsi_string, string, strlen(string) + 1);
119 entry->nsi_flags = (1 << kind);
120 entry->nsi_id = nsi_next_id++;
121 nsi_kind_next[kind] = nsi_next_id;
122 SLIST_INSERT_HEAD(&net_str_id_list, entry, nsi_next);
123 } else if ((entry->nsi_flags & (1 << kind)) == 0) {
124 if (create == 0) {
125 lck_mtx_unlock(&net_str_id_lock);
126 return ENOENT;
127 }
128 entry->nsi_flags |= (1 << kind);
129 if (entry->nsi_id >= nsi_kind_next[kind]) {
130 nsi_kind_next[kind] = entry->nsi_id + 1;
131 }
132 }
133 lck_mtx_unlock(&net_str_id_lock);
134
135 *out_id = entry->nsi_id;
136
137 return 0;
138 }
139
140
141 #define ROUNDUP32(a) \
142 ((a) > 0 ? (1 + (((a) - 1) | (sizeof(uint32_t) - 1))) : sizeof(uint32_t))
143
144 int
145 sysctl_if_family_ids SYSCTL_HANDLER_ARGS /* XXX bad syntax! */
146 {
147 #pragma unused(oidp)
148 #pragma unused(arg1)
149 #pragma unused(arg2)
150 errno_t error = 0;
151 struct net_str_id_entry *entry = NULL;
152 struct if_family_id *iffmid = NULL;
153 size_t iffmid_allocated_size = 0;
154 size_t max_size = 0;
155
156 lck_mtx_lock(&net_str_id_lock);
157 SLIST_FOREACH(entry, &net_str_id_list, nsi_next) {
158 size_t str_size;
159 size_t iffmid_size;
160
161 if ((entry->nsi_flags & (1 << NSI_IF_FAM_ID)) == 0) {
162 continue;
163 }
164
165 str_size = strlen(entry->nsi_string);
166 if (str_size > NET_ID_STR_MAX_LEN) {
167 str_size = NET_ID_STR_MAX_LEN;
168 }
169 str_size += 1; // make room for end-of-string
170 iffmid_size = ROUNDUP32(offsetof(struct net_str_id_entry, nsi_string) + str_size);
171
172 if (iffmid_size > max_size) {
173 if (iffmid) {
174 kfree_data(iffmid, iffmid->iffmid_len);
175 }
176 iffmid = (struct if_family_id *)kalloc_data(iffmid_size,
177 Z_WAITOK | Z_ZERO);
178 if (iffmid == NULL) {
179 lck_mtx_unlock(&net_str_id_lock);
180 error = ENOMEM;
181 goto done;
182 }
183 iffmid_allocated_size = iffmid_size;
184 max_size = iffmid_size;
185 }
186
187 iffmid->iffmid_len = (uint32_t)iffmid_size;
188 iffmid->iffmid_id = entry->nsi_id;
189 strlcpy(iffmid->iffmid_str, entry->nsi_string, str_size);
190 error = SYSCTL_OUT(req, iffmid, iffmid_size);
191 if (error) {
192 lck_mtx_unlock(&net_str_id_lock);
193 goto done;
194 }
195 }
196 lck_mtx_unlock(&net_str_id_lock);
197
198 done:
199 if (iffmid) {
200 kfree_data(iffmid, iffmid_allocated_size);
201 }
202 return error;
203 }
204