1 | |
2 | |
3 | |
4 | |
5 | |
6 | |
7 | |
8 | |
9 | |
10 | |
11 | |
12 | |
13 | |
14 | |
15 | |
16 | |
17 | |
18 | |
19 | |
20 | |
21 | |
22 | |
23 | |
24 | |
25 | #include <stdlib.h> |
26 | #include <string.h> |
27 | #include <stdio.h> |
28 | #include <glib.h> |
29 | #include "slirp.h" |
30 | |
31 | static const uint8_t RFC3397_OPT_DOMAIN_SEARCH = 119; |
32 | static const uint8_t MAX_OPT_LEN = 255; |
33 | static const uint8_t OPT_HEADER_LEN = 2; |
34 | static const uint8_t REFERENCE_LEN = 2; |
35 | |
36 | struct compact_domain; |
37 | |
38 | typedef struct compact_domain { |
39 | struct compact_domain *self; |
40 | struct compact_domain *refdom; |
41 | uint8_t *labels; |
42 | size_t len; |
43 | size_t common_octets; |
44 | } CompactDomain; |
45 | |
46 | static size_t |
47 | domain_suffix_diffoff(const CompactDomain *a, const CompactDomain *b) |
48 | { |
49 | size_t la = a->len, lb = b->len; |
50 | uint8_t *da = a->labels + la, *db = b->labels + lb; |
51 | size_t i, lm = (la < lb) ? la : lb; |
52 | |
53 | for (i = 0; i < lm; i++) { |
54 | da--; db--; |
55 | if (*da != *db) { |
56 | break; |
57 | } |
58 | } |
59 | return i; |
60 | } |
61 | |
62 | static int domain_suffix_ord(const void *cva, const void *cvb) |
63 | { |
64 | const CompactDomain *a = cva, *b = cvb; |
65 | size_t la = a->len, lb = b->len; |
66 | size_t doff = domain_suffix_diffoff(a, b); |
67 | uint8_t ca = a->labels[la - doff]; |
68 | uint8_t cb = b->labels[lb - doff]; |
69 | |
70 | if (ca < cb) { |
71 | return -1; |
72 | } |
73 | if (ca > cb) { |
74 | return 1; |
75 | } |
76 | if (la < lb) { |
77 | return -1; |
78 | } |
79 | if (la > lb) { |
80 | return 1; |
81 | } |
82 | return 0; |
83 | } |
84 | |
85 | static size_t domain_common_label(CompactDomain *a, CompactDomain *b) |
86 | { |
87 | size_t res, doff = domain_suffix_diffoff(a, b); |
88 | uint8_t *first_eq_pos = a->labels + (a->len - doff); |
89 | uint8_t *label = a->labels; |
90 | |
91 | while (*label && label < first_eq_pos) { |
92 | label += *label + 1; |
93 | } |
94 | res = a->len - (label - a->labels); |
95 | |
96 | return (res > REFERENCE_LEN) ? res : 0; |
97 | } |
98 | |
99 | static void domain_fixup_order(CompactDomain *cd, size_t n) |
100 | { |
101 | size_t i; |
102 | |
103 | for (i = 0; i < n; i++) { |
104 | CompactDomain *cur = cd + i, *next = cd[i].self; |
105 | |
106 | while (!cur->common_octets) { |
107 | CompactDomain *tmp = next->self; |
108 | |
109 | next->self = cur; |
110 | cur->common_octets++; |
111 | |
112 | cur = next; |
113 | next = tmp; |
114 | } |
115 | } |
116 | } |
117 | |
118 | static void domain_mklabels(CompactDomain *cd, const char *input) |
119 | { |
120 | uint8_t *len_marker = cd->labels; |
121 | uint8_t *output = len_marker; |
122 | const char *in = input; |
123 | char cur_chr; |
124 | size_t len = 0; |
125 | |
126 | if (cd->len == 0) { |
127 | goto fail; |
128 | } |
129 | cd->len++; |
130 | |
131 | do { |
132 | cur_chr = *in++; |
133 | if (cur_chr == '.' || cur_chr == '\0') { |
134 | len = output - len_marker; |
135 | if ((len == 0 && cur_chr == '.') || len >= 64) { |
136 | goto fail; |
137 | } |
138 | *len_marker = len; |
139 | |
140 | output++; |
141 | len_marker = output; |
142 | } else { |
143 | output++; |
144 | *output = cur_chr; |
145 | } |
146 | } while (cur_chr != '\0'); |
147 | |
148 | |
149 | if (len != 0) { |
150 | *len_marker = 0; |
151 | cd->len++; |
152 | } |
153 | return; |
154 | |
155 | fail: |
156 | g_warning("failed to parse domain name '%s'\n", input)g_log (((gchar*) 0), G_LOG_LEVEL_WARNING, "failed to parse domain name '%s'\n" , input); |
157 | cd->len = 0; |
158 | } |
159 | |
160 | static void |
161 | domain_mkxrefs(CompactDomain *doms, CompactDomain *last, size_t depth) |
162 | { |
163 | CompactDomain *i = doms, *target = doms; |
164 | |
165 | do { |
166 | if (i->labels < target->labels) { |
167 | target = i; |
168 | } |
169 | } while (i++ != last); |
170 | |
171 | for (i = doms; i != last; i++) { |
172 | CompactDomain *group_last; |
173 | size_t next_depth; |
174 | |
175 | if (i->common_octets == depth) { |
176 | continue; |
177 | } |
178 | |
179 | next_depth = -1; |
180 | for (group_last = i; group_last != last; group_last++) { |
181 | size_t co = group_last->common_octets; |
182 | if (co <= depth) { |
183 | break; |
184 | } |
185 | if (co < next_depth) { |
186 | next_depth = co; |
187 | } |
188 | } |
189 | domain_mkxrefs(i, group_last, next_depth); |
190 | |
191 | i = group_last; |
192 | if (i == last) { |
193 | break; |
194 | } |
195 | } |
196 | |
197 | if (depth == 0) { |
198 | return; |
199 | } |
200 | |
201 | i = doms; |
202 | do { |
203 | if (i != target && i->refdom == NULL((void*)0)) { |
204 | i->refdom = target; |
205 | i->common_octets = depth; |
206 | } |
207 | } while (i++ != last); |
208 | } |
209 | |
210 | static size_t domain_compactify(CompactDomain *domains, size_t n) |
211 | { |
212 | uint8_t *start = domains->self->labels, *outptr = start; |
213 | size_t i; |
214 | |
215 | for (i = 0; i < n; i++) { |
216 | CompactDomain *cd = domains[i].self; |
217 | CompactDomain *rd = cd->refdom; |
218 | |
219 | if (rd != NULL((void*)0)) { |
220 | size_t moff = (rd->labels - start) |
221 | + (rd->len - cd->common_octets); |
222 | if (moff < 0x3FFFu) { |
223 | cd->len -= cd->common_octets - 2; |
224 | cd->labels[cd->len - 1] = moff & 0xFFu; |
225 | cd->labels[cd->len - 2] = 0xC0u | (moff >> 8); |
226 | } |
227 | } |
228 | |
229 | if (cd->labels != outptr) { |
230 | memmove(outptr, cd->labels, cd->len); |
231 | cd->labels = outptr; |
232 | } |
233 | outptr += cd->len; |
234 | } |
235 | return outptr - start; |
236 | } |
237 | |
238 | int translate_dnssearch(Slirp *s, const char **names) |
239 | { |
240 | size_t blocks, bsrc_start, bsrc_end, bdst_start; |
241 | size_t i, num_domains, memreq = 0; |
242 | uint8_t *result = NULL((void*)0), *outptr; |
243 | CompactDomain *domains = NULL((void*)0); |
244 | const char **nameptr = names; |
245 | |
246 | while (*nameptr != NULL((void*)0)) { |
| 1 | Loop condition is true. Entering loop body | |
|
| 2 | | Loop condition is false. Execution continues on line 250 | |
|
247 | nameptr++; |
248 | } |
249 | |
250 | num_domains = nameptr - names; |
251 | if (num_domains == 0) { |
| 3 | | Assuming 'num_domains' is not equal to 0 | |
|
| |
252 | return -2; |
253 | } |
254 | |
255 | domains = g_malloc(num_domains * sizeof(*domains)); |
256 | |
257 | for (i = 0; i < num_domains; i++) { |
| 5 | | Loop condition is true. Entering loop body | |
|
| 6 | | Assuming 'i' is < 'num_domains' | |
|
| 7 | | Loop condition is true. Entering loop body | |
|
258 | size_t nlen = strlen(names[i]); |
| 8 | | Null pointer passed as an argument to a 'nonnull' parameter |
|
259 | memreq += nlen + 2; |
260 | domains[i].self = domains + i; |
261 | domains[i].len = nlen; |
262 | domains[i].common_octets = 0; |
263 | domains[i].refdom = NULL((void*)0); |
264 | } |
265 | |
266 | |
267 | memreq += ((memreq + MAX_OPT_LEN - 1) / MAX_OPT_LEN) * OPT_HEADER_LEN; |
268 | result = g_malloc(memreq * sizeof(*result)); |
269 | |
270 | outptr = result; |
271 | for (i = 0; i < num_domains; i++) { |
272 | domains[i].labels = outptr; |
273 | domain_mklabels(domains + i, names[i]); |
274 | outptr += domains[i].len; |
275 | } |
276 | |
277 | if (outptr == result) { |
278 | g_free(domains); |
279 | g_free(result); |
280 | return -1; |
281 | } |
282 | |
283 | qsort(domains, num_domains, sizeof(*domains), domain_suffix_ord); |
284 | domain_fixup_order(domains, num_domains); |
285 | |
286 | for (i = 1; i < num_domains; i++) { |
287 | size_t cl = domain_common_label(domains + i - 1, domains + i); |
288 | domains[i - 1].common_octets = cl; |
289 | } |
290 | |
291 | domain_mkxrefs(domains, domains + num_domains - 1, 0); |
292 | memreq = domain_compactify(domains, num_domains); |
293 | |
294 | blocks = (memreq + MAX_OPT_LEN - 1) / MAX_OPT_LEN; |
295 | bsrc_end = memreq; |
296 | bsrc_start = (blocks - 1) * MAX_OPT_LEN; |
297 | bdst_start = bsrc_start + blocks * OPT_HEADER_LEN; |
298 | memreq += blocks * OPT_HEADER_LEN; |
299 | |
300 | while (blocks--) { |
301 | size_t len = bsrc_end - bsrc_start; |
302 | memmove(result + bdst_start, result + bsrc_start, len); |
303 | result[bdst_start - 2] = RFC3397_OPT_DOMAIN_SEARCH; |
304 | result[bdst_start - 1] = len; |
305 | bsrc_end = bsrc_start; |
306 | bsrc_start -= MAX_OPT_LEN; |
307 | bdst_start -= MAX_OPT_LEN + OPT_HEADER_LEN; |
308 | } |
309 | |
310 | g_free(domains); |
311 | s->vdnssearch = result; |
312 | s->vdnssearch_len = memreq; |
313 | return 0; |
314 | } |