?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
network - Fix socket & mbuf leak
[dragonfly.git] / lib / libc / nameser / ns_name.c
blob9d409f3d95368b5d5fe88ed0c577606ab17f3f7a
1 /*
2 * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
3 * Copyright (c) 1996,1999 by Internet Software Consortium.
4 *
5 * Permission to use, copy, modify, and distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
8 *
9 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR
12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
15 * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16 */
17
18 #ifndef lint
19 static const char rcsid[] = "$Id: ns_name.c,v 1.10 2005/04/27 04:56:40 sra Exp $";
20 #endif
21
22 #include "port_before.h"
23
24 #include <sys/types.h>
25
26 #include <netinet/in.h>
27 #include <arpa/nameser.h>
28
29 #include <errno.h>
30 #include <resolv.h>
31 #include <string.h>
32 #include <ctype.h>
33 #include <stdlib.h>
34 #include <limits.h>
35
36 #include "port_after.h"
37
38 #ifdef SPRINTF_CHAR
39 # define SPRINTF(x) strlen(sprintf/**/x)
40 #else
41 # define SPRINTF(x) ((size_t)sprintf x)
42 #endif
43
44 #define NS_TYPE_ELT 0x40 /*%< EDNS0 extended label type */
45 #define DNS_LABELTYPE_BITSTRING 0x41
46
47 /* Data. */
48
49 static const char digits[] = "0123456789";
50
51 static const char digitvalue[256] = {
52 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /*16*/
53 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /*32*/
54 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /*48*/
55 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, -1, -1, -1, -1, -1, -1, /*64*/
56 -1, 10, 11, 12, 13, 14, 15, -1, -1, -1, -1, -1, -1, -1, -1, -1, /*80*/
57 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /*96*/
58 -1, 10, 11, 12, 13, 14, 15, -1, -1, -1, -1, -1, -1, -1, -1, -1, /*112*/
59 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /*128*/
60 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
61 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
62 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
63 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
64 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
65 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
66 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
67 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /*256*/
68 };
69
70 /* Forward. */
71
72 static int special(int);
73 static int printable(int);
74 static int dn_find(const u_char *, const u_char *,
75 const u_char * const *,
76 const u_char * const *);
77 static int encode_bitsring(const char **, const char *,
78 unsigned char **, unsigned char **,
79 unsigned const char *);
80 static int labellen(const u_char *);
81 static int decode_bitstring(const unsigned char **,
82 char *, const char *);
83
84 /* Public. */
85
86 /*%
87 * Convert an encoded domain name to printable ascii as per RFC1035.
88
89 * return:
90 *\li Number of bytes written to buffer, or -1 (with errno set)
91 *
92 * notes:
93 *\li The root is returned as "."
94 *\li All other domains are returned in non absolute form
95 */
96 int
97 ns_name_ntop(const u_char *src, char *dst, size_t dstsiz)
98 {
99 const u_char *cp;
100 char *dn, *eom;
101 u_char c;
102 u_int n;
103 int l;
104
105 cp = src;
106 dn = dst;
107 eom = dst + dstsiz;
108
109 while ((n = *cp++) != 0) {
110 if ((n & NS_CMPRSFLGS) == NS_CMPRSFLGS) {
111 /* Some kind of compression pointer. */
112 errno = EMSGSIZE;
113 return (-1);
114 }
115 if (dn != dst) {
116 if (dn >= eom) {
117 errno = EMSGSIZE;
118 return (-1);
119 }
120 *dn++ = '.';
121 }
122 if ((l = labellen(cp - 1)) < 0) {
123 errno = EMSGSIZE; /*%< XXX */
124 return(-1);
125 }
126 if (dn + l >= eom) {
127 errno = EMSGSIZE;
128 return (-1);
129 }
130 if ((n & NS_CMPRSFLGS) == NS_TYPE_ELT) {
131 int m;
132
133 if (n != DNS_LABELTYPE_BITSTRING) {
134 /* XXX: labellen should reject this case */
135 errno = EINVAL;
136 return(-1);
137 }
138 if ((m = decode_bitstring(&cp, dn, eom)) < 0)
139 {
140 errno = EMSGSIZE;
141 return(-1);
142 }
143 dn += m;
144 continue;
145 }
146 for ((void)NULL; l > 0; l--) {
147 c = *cp++;
148 if (special(c)) {
149 if (dn + 1 >= eom) {
150 errno = EMSGSIZE;
151 return (-1);
152 }
153 *dn++ = '\\';
154 *dn++ = (char)c;
155 } else if (!printable(c)) {
156 if (dn + 3 >= eom) {
157 errno = EMSGSIZE;
158 return (-1);
159 }
160 *dn++ = '\\';
161 *dn++ = digits[c / 100];
162 *dn++ = digits[(c % 100) / 10];
163 *dn++ = digits[c % 10];
164 } else {
165 if (dn >= eom) {
166 errno = EMSGSIZE;
167 return (-1);
168 }
169 *dn++ = (char)c;
170 }
171 }
172 }
173 if (dn == dst) {
174 if (dn >= eom) {
175 errno = EMSGSIZE;
176 return (-1);
177 }
178 *dn++ = '.';
179 }
180 if (dn >= eom) {
181 errno = EMSGSIZE;
182 return (-1);
183 }
184 *dn++ = '\0';
185 return (dn - dst);
186 }
187
188 /*%
189 * Convert a ascii string into an encoded domain name as per RFC1035.
190 *
191 * return:
192 *
193 *\li -1 if it fails
194 *\li 1 if string was fully qualified
195 *\li 0 is string was not fully qualified
196 *
197 * notes:
198 *\li Enforces label and domain length limits.
199 */
200
201 int
202 ns_name_pton(const char *src, u_char *dst, size_t dstsiz)
203 {
204 u_char *label, *bp, *eom;
205 int c, n, escaped, e = 0;
206 char *cp;
207
208 escaped = 0;
209 bp = dst;
210 eom = dst + dstsiz;
211 label = bp++;
212
213 while ((c = *src++) != 0) {
214 if (escaped) {
215 if (c == '[') { /*%< start a bit string label */
216 if ((cp = strchr(src, ']')) == NULL) {
217 errno = EINVAL; /*%< ??? */
218 return(-1);
219 }
220 if ((e = encode_bitsring(&src, cp + 2,
221 &label, &bp, eom))
222 != 0) {
223 errno = e;
224 return(-1);
225 }
226 escaped = 0;
227 label = bp++;
228 if ((c = *src++) == 0)
229 goto done;
230 else if (c != '.') {
231 errno = EINVAL;
232 return(-1);
233 }
234 continue;
235 }
236 else if ((cp = strchr(digits, c)) != NULL) {
237 n = (cp - digits) * 100;
238 if ((c = *src++) == 0 ||
239 (cp = strchr(digits, c)) == NULL) {
240 errno = EMSGSIZE;
241 return (-1);
242 }
243 n += (cp - digits) * 10;
244 if ((c = *src++) == 0 ||
245 (cp = strchr(digits, c)) == NULL) {
246 errno = EMSGSIZE;
247 return (-1);
248 }
249 n += (cp - digits);
250 if (n > 255) {
251 errno = EMSGSIZE;
252 return (-1);
253 }
254 c = n;
255 }
256 escaped = 0;
257 } else if (c == '\\') {
258 escaped = 1;
259 continue;
260 } else if (c == '.') {
261 c = (bp - label - 1);
262 if ((c & NS_CMPRSFLGS) != 0) { /*%< Label too big. */
263 errno = EMSGSIZE;
264 return (-1);
265 }
266 if (label >= eom) {
267 errno = EMSGSIZE;
268 return (-1);
269 }
270 *label = c;
271 /* Fully qualified ? */
272 if (*src == '\0') {
273 if (c != 0) {
274 if (bp >= eom) {
275 errno = EMSGSIZE;
276 return (-1);
277 }
278 *bp++ = '\0';
279 }
280 if ((bp - dst) > MAXCDNAME) {
281 errno = EMSGSIZE;
282 return (-1);
283 }
284 return (1);
285 }
286 if (c == 0 || *src == '.') {
287 errno = EMSGSIZE;
288 return (-1);
289 }
290 label = bp++;
291 continue;
292 }
293 if (bp >= eom) {
294 errno = EMSGSIZE;
295 return (-1);
296 }
297 *bp++ = (u_char)c;
298 }
299 c = (bp - label - 1);
300 if ((c & NS_CMPRSFLGS) != 0) { /*%< Label too big. */
301 errno = EMSGSIZE;
302 return (-1);
303 }
304 done:
305 if (label >= eom) {
306 errno = EMSGSIZE;
307 return (-1);
308 }
309 *label = c;
310 if (c != 0) {
311 if (bp >= eom) {
312 errno = EMSGSIZE;
313 return (-1);
314 }
315 *bp++ = 0;
316 }
317 if ((bp - dst) > MAXCDNAME) { /*%< src too big */
318 errno = EMSGSIZE;
319 return (-1);
320 }
321 return (0);
322 }
323
324 /*%
325 * Convert a network strings labels into all lowercase.
326 *
327 * return:
328 *\li Number of bytes written to buffer, or -1 (with errno set)
329 *
330 * notes:
331 *\li Enforces label and domain length limits.
332 */
333
334 int
335 ns_name_ntol(const u_char *src, u_char *dst, size_t dstsiz)
336 {
337 const u_char *cp;
338 u_char *dn, *eom;
339 u_char c;
340 u_int n;
341 int l;
342
343 cp = src;
344 dn = dst;
345 eom = dst + dstsiz;
346
347 if (dn >= eom) {
348 errno = EMSGSIZE;
349 return (-1);
350 }
351 while ((n = *cp++) != 0) {
352 if ((n & NS_CMPRSFLGS) == NS_CMPRSFLGS) {
353 /* Some kind of compression pointer. */
354 errno = EMSGSIZE;
355 return (-1);
356 }
357 *dn++ = n;
358 if ((l = labellen(cp - 1)) < 0) {
359 errno = EMSGSIZE;
360 return (-1);
361 }
362 if (dn + l >= eom) {
363 errno = EMSGSIZE;
364 return (-1);
365 }
366 for ((void)NULL; l > 0; l--) {
367 c = *cp++;
368 if (isupper(c))
369 *dn++ = tolower(c);
370 else
371 *dn++ = c;
372 }
373 }
374 *dn++ = '\0';
375 return (dn - dst);
376 }
377
378 /*%
379 * Unpack a domain name from a message, source may be compressed.
380 *
381 * return:
382 *\li -1 if it fails, or consumed octets if it succeeds.
383 */
384 int
385 ns_name_unpack(const u_char *msg, const u_char *eom, const u_char *src,
386 u_char *dst, size_t dstsiz)
387 {
388 const u_char *srcp, *dstlim;
389 u_char *dstp;
390 int n, len, checked, l;
391
392 len = -1;
393 checked = 0;
394 dstp = dst;
395 srcp = src;
396 dstlim = dst + dstsiz;
397 if (srcp < msg || srcp >= eom) {
398 errno = EMSGSIZE;
399 return (-1);
400 }
401 /* Fetch next label in domain name. */
402 while ((n = *srcp++) != 0) {
403 /* Check for indirection. */
404 switch (n & NS_CMPRSFLGS) {
405 case 0:
406 case NS_TYPE_ELT:
407 /* Limit checks. */
408 if ((l = labellen(srcp - 1)) < 0) {
409 errno = EMSGSIZE;
410 return(-1);
411 }
412 if (dstp + l + 1 >= dstlim || srcp + l >= eom) {
413 errno = EMSGSIZE;
414 return (-1);
415 }
416 checked += l + 1;
417 *dstp++ = n;
418 memcpy(dstp, srcp, l);
419 dstp += l;
420 srcp += l;
421 break;
422
423 case NS_CMPRSFLGS:
424 if (srcp >= eom) {
425 errno = EMSGSIZE;
426 return (-1);
427 }
428 if (len < 0)
429 len = srcp - src + 1;
430 srcp = msg + (((n & 0x3f) << 8) | (*srcp & 0xff));
431 if (srcp < msg || srcp >= eom) { /*%< Out of range. */
432 errno = EMSGSIZE;
433 return (-1);
434 }
435 checked += 2;
436 /*
437 * Check for loops in the compressed name;
438 * if we've looked at the whole message,
439 * there must be a loop.
440 */
441 if (checked >= eom - msg) {
442 errno = EMSGSIZE;
443 return (-1);
444 }
445 break;
446
447 default:
448 errno = EMSGSIZE;
449 return (-1); /*%< flag error */
450 }
451 }
452 *dstp = '\0';
453 if (len < 0)
454 len = srcp - src;
455 return (len);
456 }
457
458 /*%
459 * Pack domain name 'domain' into 'comp_dn'.
460 *
461 * return:
462 *\li Size of the compressed name, or -1.
463 *
464 * notes:
465 *\li 'dnptrs' is an array of pointers to previous compressed names.
466 *\li dnptrs[0] is a pointer to the beginning of the message. The array
467 * ends with NULL.
468 *\li 'lastdnptr' is a pointer to the end of the array pointed to
469 * by 'dnptrs'.
470 *
471 * Side effects:
472 *\li The list of pointers in dnptrs is updated for labels inserted into
473 * the message as we compress the name. If 'dnptr' is NULL, we don't
474 * try to compress names. If 'lastdnptr' is NULL, we don't update the
475 * list.
476 */
477 int
478 ns_name_pack(const u_char *src, u_char *dst, int dstsiz,
479 const u_char **dnptrs, const u_char **lastdnptr)
480 {
481 u_char *dstp;
482 const u_char **cpp, **lpp, *eob, *msg;
483 const u_char *srcp;
484 int n, l, first = 1;
485
486 srcp = src;
487 dstp = dst;
488 eob = dstp + dstsiz;
489 lpp = cpp = NULL;
490 if (dnptrs != NULL) {
491 if ((msg = *dnptrs++) != NULL) {
492 for (cpp = dnptrs; *cpp != NULL; cpp++)
493 (void)NULL;
494 lpp = cpp; /*%< end of list to search */
495 }
496 } else
497 msg = NULL;
498
499 /* make sure the domain we are about to add is legal */
500 l = 0;
501 do {
502 int l0;
503
504 n = *srcp;
505 if ((n & NS_CMPRSFLGS) == NS_CMPRSFLGS) {
506 errno = EMSGSIZE;
507 return (-1);
508 }
509 if ((l0 = labellen(srcp)) < 0) {
510 errno = EINVAL;
511 return(-1);
512 }
513 l += l0 + 1;
514 if (l > MAXCDNAME) {
515 errno = EMSGSIZE;
516 return (-1);
517 }
518 srcp += l0 + 1;
519 } while (n != 0);
520
521 /* from here on we need to reset compression pointer array on error */
522 srcp = src;
523 do {
524 /* Look to see if we can use pointers. */
525 n = *srcp;
526 if (n != 0 && msg != NULL) {
527 l = dn_find(srcp, msg, (const u_char * const *)dnptrs,
528 (const u_char * const *)lpp);
529 if (l >= 0) {
530 if (dstp + 1 >= eob) {
531 goto cleanup;
532 }
533 *dstp++ = (l >> 8) | NS_CMPRSFLGS;
534 *dstp++ = l % 256;
535 return (dstp - dst);
536 }
537 /* Not found, save it. */
538 if (lastdnptr != NULL && cpp < lastdnptr - 1 &&
539 (dstp - msg) < 0x4000 && first) {
540 *cpp++ = dstp;
541 *cpp = NULL;
542 first = 0;
543 }
544 }
545 /* copy label to buffer */
546 if ((n & NS_CMPRSFLGS) == NS_CMPRSFLGS) {
547 /* Should not happen. */
548 goto cleanup;
549 }
550 n = labellen(srcp);
551 if (dstp + 1 + n >= eob) {
552 goto cleanup;
553 }
554 memcpy(dstp, srcp, n + 1);
555 srcp += n + 1;
556 dstp += n + 1;
557 } while (n != 0);
558
559 if (dstp > eob) {
560 cleanup:
561 if (msg != NULL)
562 *lpp = NULL;
563 errno = EMSGSIZE;
564 return (-1);
565 }
566 return (dstp - dst);
567 }
568
569 /*%
570 * Expand compressed domain name to presentation format.
571 *
572 * return:
573 *\li Number of bytes read out of `src', or -1 (with errno set).
574 *
575 * note:
576 *\li Root domain returns as "." not "".
577 */
578 int
579 ns_name_uncompress(const u_char *msg, const u_char *eom, const u_char *src,
580 char *dst, size_t dstsiz)
581 {
582 u_char tmp[NS_MAXCDNAME];
583 int n;
584
585 if ((n = ns_name_unpack(msg, eom, src, tmp, sizeof tmp)) == -1)
586 return (-1);
587 if (ns_name_ntop(tmp, dst, dstsiz) == -1)
588 return (-1);
589 return (n);
590 }
591
592 /*%
593 * Compress a domain name into wire format, using compression pointers.
594 *
595 * return:
596 *\li Number of bytes consumed in `dst' or -1 (with errno set).
597 *
598 * notes:
599 *\li 'dnptrs' is an array of pointers to previous compressed names.
600 *\li dnptrs[0] is a pointer to the beginning of the message.
601 *\li The list ends with NULL. 'lastdnptr' is a pointer to the end of the
602 * array pointed to by 'dnptrs'. Side effect is to update the list of
603 * pointers for labels inserted into the message as we compress the name.
604 *\li If 'dnptr' is NULL, we don't try to compress names. If 'lastdnptr'
605 * is NULL, we don't update the list.
606 */
607 int
608 ns_name_compress(const char *src, u_char *dst, size_t dstsiz,
609 const u_char **dnptrs, const u_char **lastdnptr)
610 {
611 u_char tmp[NS_MAXCDNAME];
612
613 if (ns_name_pton(src, tmp, sizeof tmp) == -1)
614 return (-1);
615 return (ns_name_pack(tmp, dst, dstsiz, dnptrs, lastdnptr));
616 }
617
618 /*%
619 * Reset dnptrs so that there are no active references to pointers at or
620 * after src.
621 */
622 void
623 ns_name_rollback(const u_char *src, const u_char **dnptrs,
624 const u_char **lastdnptr)
625 {
626 while (dnptrs < lastdnptr && *dnptrs != NULL) {
627 if (*dnptrs >= src) {
628 *dnptrs = NULL;
629 break;
630 }
631 dnptrs++;
632 }
633 }
634
635 /*%
636 * Advance *ptrptr to skip over the compressed name it points at.
637 *
638 * return:
639 *\li 0 on success, -1 (with errno set) on failure.
640 */
641 int
642 ns_name_skip(const u_char **ptrptr, const u_char *eom)
643 {
644 const u_char *cp;
645 u_int n;
646 int l;
647
648 cp = *ptrptr;
649 while (cp < eom && (n = *cp++) != 0) {
650 /* Check for indirection. */
651 switch (n & NS_CMPRSFLGS) {
652 case 0: /*%< normal case, n == len */
653 cp += n;
654 continue;
655 case NS_TYPE_ELT: /*%< EDNS0 extended label */
656 if ((l = labellen(cp - 1)) < 0) {
657 errno = EMSGSIZE; /*%< XXX */
658 return(-1);
659 }
660 cp += l;
661 continue;
662 case NS_CMPRSFLGS: /*%< indirection */
663 cp++;
664 break;
665 default: /*%< illegal type */
666 errno = EMSGSIZE;
667 return (-1);
668 }
669 break;
670 }
671 if (cp > eom) {
672 errno = EMSGSIZE;
673 return (-1);
674 }
675 *ptrptr = cp;
676 return (0);
677 }
678
679 /* Private. */
680
681 /*%
682 * Thinking in noninternationalized USASCII (per the DNS spec),
683 * is this characted special ("in need of quoting") ?
684 *
685 * return:
686 *\li boolean.
687 */
688 static int
689 special(int ch) {
690 switch (ch) {
691 case 0x22: /*%< '"' */
692 case 0x2E: /*%< '.' */
693 case 0x3B: /*%< ';' */
694 case 0x5C: /*%< '\\' */
695 case 0x28: /*%< '(' */
696 case 0x29: /*%< ')' */
697 /* Special modifiers in zone files. */
698 case 0x40: /*%< '@' */
699 case 0x24: /*%< '$' */
700 return (1);
701 default:
702 return (0);
703 }
704 }
705
706 /*%
707 * Thinking in noninternationalized USASCII (per the DNS spec),
708 * is this character visible and not a space when printed ?
709 *
710 * return:
711 *\li boolean.
712 */
713 static int
714 printable(int ch) {
715 return (ch > 0x20 && ch < 0x7f);
716 }
717
718 /*%
719 * Thinking in noninternationalized USASCII (per the DNS spec),
720 * convert this character to lower case if it's upper case.
721 */
722 static int
723 mklower(int ch) {
724 if (ch >= 0x41 && ch <= 0x5A)
725 return (ch + 0x20);
726 return (ch);
727 }
728
729 /*%
730 * Search for the counted-label name in an array of compressed names.
731 *
732 * return:
733 *\li offset from msg if found, or -1.
734 *
735 * notes:
736 *\li dnptrs is the pointer to the first name on the list,
737 *\li not the pointer to the start of the message.
738 */
739 static int
740 dn_find(const u_char *domain, const u_char *msg,
741 const u_char * const *dnptrs,
742 const u_char * const *lastdnptr)
743 {
744 const u_char *dn, *cp, *sp;
745 const u_char * const *cpp;
746 u_int n;
747
748 for (cpp = dnptrs; cpp < lastdnptr; cpp++) {
749 sp = *cpp;
750 /*
751 * terminate search on:
752 * root label
753 * compression pointer
754 * unusable offset
755 */
756 while (*sp != 0 && (*sp & NS_CMPRSFLGS) == 0 &&
757 (sp - msg) < 0x4000) {
758 dn = domain;
759 cp = sp;
760 while ((n = *cp++) != 0) {
761 /*
762 * check for indirection
763 */
764 switch (n & NS_CMPRSFLGS) {
765 case 0: /*%< normal case, n == len */
766 n = labellen(cp - 1); /*%< XXX */
767 if (n != *dn++)
768 goto next;
769
770 for ((void)NULL; n > 0; n--)
771 if (mklower(*dn++) !=
772 mklower(*cp++))
773 goto next;
774 /* Is next root for both ? */
775 if (*dn == '\0' && *cp == '\0')
776 return (sp - msg);
777 if (*dn)
778 continue;
779 goto next;
780 case NS_CMPRSFLGS: /*%< indirection */
781 cp = msg + (((n & 0x3f) << 8) | *cp);
782 break;
783
784 default: /*%< illegal type */
785 errno = EMSGSIZE;
786 return (-1);
787 }
788 }
789 next: ;
790 sp += *sp + 1;
791 }
792 }
793 errno = ENOENT;
794 return (-1);
795 }
796
797 static int
798 decode_bitstring(const unsigned char **cpp, char *dn, const char *eom)
799 {
800 const unsigned char *cp = *cpp;
801 char *beg = dn, tc;
802 int b, blen, plen, i;
803
804 if ((blen = (*cp & 0xff)) == 0)
805 blen = 256;
806 plen = (blen + 3) / 4;
807 plen += sizeof("\\[x/]") + (blen > 99 ? 3 : (blen > 9) ? 2 : 1);
808 if (dn + plen >= eom)
809 return(-1);
810
811 cp++;
812 i = SPRINTF((dn, "\\[x"));
813 if (i < 0)
814 return (-1);
815 dn += i;
816 for (b = blen; b > 7; b -= 8, cp++) {
817 i = SPRINTF((dn, "%02x", *cp & 0xff));
818 if (i < 0)
819 return (-1);
820 dn += i;
821 }
822 if (b > 4) {
823 tc = *cp++;
824 i = SPRINTF((dn, "%02x", tc & (0xff << (8 - b))));
825 if (i < 0)
826 return (-1);
827 dn += i;
828 } else if (b > 0) {
829 tc = *cp++;
830 i = SPRINTF((dn, "%1x",
831 ((tc >> 4) & 0x0f) & (0x0f << (4 - b))));
832 if (i < 0)
833 return (-1);
834 dn += i;
835 }
836 i = SPRINTF((dn, "/%d]", blen));
837 if (i < 0)
838 return (-1);
839 dn += i;
840
841 *cpp = cp;
842 return(dn - beg);
843 }
844
845 static int
846 encode_bitsring(const char **bp, const char *end, unsigned char **labelp,
847 unsigned char ** dst, unsigned const char *eom)
848 {
849 int afterslash = 0;
850 const char *cp = *bp;
851 unsigned char *tp;
852 char c;
853 const char *beg_blen;
854 char *end_blen = NULL;
855 int value = 0, count = 0, tbcount = 0, blen = 0;
856
857 beg_blen = end_blen = NULL;
858
859 /* a bitstring must contain at least 2 characters */
860 if (end - cp < 2)
861 return(EINVAL);
862
863 /* XXX: currently, only hex strings are supported */
864 if (*cp++ != 'x')
865 return(EINVAL);
866 if (!isxdigit((*cp) & 0xff)) /*%< reject '\[x/BLEN]' */
867 return(EINVAL);
868
869 for (tp = *dst + 1; cp < end && tp < eom; cp++) {
870 switch((c = *cp)) {
871 case ']': /*%< end of the bitstring */
872 if (afterslash) {
873 if (beg_blen == NULL)
874 return(EINVAL);
875 blen = (int)strtol(beg_blen, &end_blen, 10);
876 if (*end_blen != ']')
877 return(EINVAL);
878 }
879 if (count)
880 *tp++ = ((value << 4) & 0xff);
881 cp++; /*%< skip ']' */
882 goto done;
883 case '/':
884 afterslash = 1;
885 break;
886 default:
887 if (afterslash) {
888 if (!isdigit(c&0xff))
889 return(EINVAL);
890 if (beg_blen == NULL) {
891
892 if (c == '0') {
893 /* blen never begings with 0 */
894 return(EINVAL);
895 }
896 beg_blen = cp;
897 }
898 } else {
899 if (!isxdigit(c&0xff))
900 return(EINVAL);
901 value <<= 4;
902 value += digitvalue[(int)c];
903 count += 4;
904 tbcount += 4;
905 if (tbcount > 256)
906 return(EINVAL);
907 if (count == 8) {
908 *tp++ = value;
909 count = 0;
910 }
911 }
912 break;
913 }
914 }
915 done:
916 if (cp >= end || tp >= eom)
917 return(EMSGSIZE);
918
919 /*
920 * bit length validation:
921 * If a <length> is present, the number of digits in the <bit-data>
922 * MUST be just sufficient to contain the number of bits specified
923 * by the <length>. If there are insignificant bits in a final
924 * hexadecimal or octal digit, they MUST be zero.
925 * RFC2673, Section 3.2.
926 */
927 if (blen > 0) {
928 int traillen;
929
930 if (((blen + 3) & ~3) != tbcount)
931 return(EINVAL);
932 traillen = tbcount - blen; /*%< between 0 and 3 */
933 if (((value << (8 - traillen)) & 0xff) != 0)
934 return(EINVAL);
935 }
936 else
937 blen = tbcount;
938 if (blen == 256)
939 blen = 0;
940
941 /* encode the type and the significant bit fields */
942 **labelp = DNS_LABELTYPE_BITSTRING;
943 **dst = blen;
944
945 *bp = cp;
946 *dst = tp;
947
948 return(0);
949 }
950
951 static int
952 labellen(const u_char *lp)
953 {
954 int bitlen;
955 u_char l = *lp;
956
957 if ((l & NS_CMPRSFLGS) == NS_CMPRSFLGS) {
958 /* should be avoided by the caller */
959 return(-1);
960 }
961
962 if ((l & NS_CMPRSFLGS) == NS_TYPE_ELT) {
963 if (l == DNS_LABELTYPE_BITSTRING) {
964 if ((bitlen = *(lp + 1)) == 0)
965 bitlen = 256;
966 return((bitlen + 7 ) / 8 + 1);
967 }
968 return(-1); /*%< unknwon ELT */
969 }
970 return(l);
971 }
972
973 /*! \file */
DragonFly BSD