source: mainline/uspace/srv/net/dnsres/dns_msg.c@ dc95342

lfn serial ticket/834-toolchain-update topic/msim-upgrade topic/simplify-dev-export
Last change on this file since dc95342 was dc95342, checked in by Jiri Svoboda <jiri@…>, 13 years ago

Memory freeing, error paths.
  • Property mode set to 100644
File size: 13.6 KB
Line 
1/*
2 * Copyright (c) 2013 Jiri Svoboda
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 *
9 * - Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * - Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 * - The name of the author may not be used to endorse or promote products
15 * derived from this software without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29/** @addtogroup dnsres
30 * @{
31 */
32/**
33 * @file
34 */
35
36#include <bitops.h>
37#include <byteorder.h>
38#include <errno.h>
39#include <macros.h>
40#include <stdint.h>
41#include <stdlib.h>
42#include <str.h>
43
44#include "dns_msg.h"
45#include "dns_std.h"
46
47#define NAME "dnsres"
48
49static uint16_t dns_uint16_t_decode(uint8_t *, size_t);
50
51static int dns_dstr_ext(char **dstr, const char *suff)
52{
53 size_t s1, s2;
54 size_t nsize;
55 char *nstr;
56
57 if (*dstr == NULL) {
58 *dstr = str_dup(suff);
59 if (*dstr == NULL)
60 return ENOMEM;
61 return EOK;
62 }
63
64 s1 = str_size(*dstr);
65 s2 = str_size(suff);
66 nsize = s1 + s2 + 1;
67
68 nstr = realloc(*dstr, nsize);
69 if (nstr == NULL)
70 return ENOMEM;
71
72 str_cpy((*dstr) + s1, nsize - s1, suff);
73
74 *dstr = nstr;
75 return EOK;
76}
77
78#include <stdio.h>
79static int dns_name_encode(char *name, uint8_t *buf, size_t buf_size,
80 size_t *act_size)
81{
82 size_t off;
83 wchar_t c;
84 size_t lsize;
85 size_t pi, di;
86
87 pi = 0;
88 di = 1;
89 off = 0;
90
91 printf("dns_name_encode(name='%s', buf=%p, buf_size=%zu, act_size=%p\n",
92 name, buf, buf_size, act_size);
93 lsize = 0;
94 while (true) {
95 printf("off=%zu\n", off);
96 c = str_decode(name, &off, STR_NO_LIMIT);
97 printf("c=%d\n", (int)c);
98 if (c >= 127) {
99 /* Non-ASCII character */
100 printf("non-ascii character\n");
101 return EINVAL;
102 }
103
104 if (c == '.' || c == '\0') {
105 /* Empty string, starting with period or two consecutive periods. */
106 if (lsize == 0) {
107 printf("empty token\n");
108 return EINVAL;
109 }
110
111 if (lsize > DNS_LABEL_MAX_SIZE) {
112 /* Label too long */
113 printf("label too long\n");
114 return EINVAL;
115 }
116
117 if (buf != NULL && pi < buf_size)
118 buf[pi] = (uint8_t)lsize;
119
120 lsize = 0;
121 pi = di;
122 ++di;
123
124 if (c == '\0')
125 break;
126 } else {
127 if (buf != NULL && di < buf_size)
128 buf[di] = c;
129 ++di;
130 ++lsize;
131 }
132 }
133
134 if (buf != NULL && pi < buf_size)
135 buf[pi] = 0;
136
137 *act_size = di;
138 return EOK;
139}
140
141static int dns_name_decode(uint8_t *buf, size_t size, size_t boff, char **rname,
142 size_t *eoff)
143{
144 uint8_t *bp;
145 size_t bsize;
146 size_t lsize;
147 size_t i;
148 size_t ptr;
149 size_t eptr;
150 char *name;
151 char dbuf[2];
152 int rc;
153 bool first;
154
155 name = NULL;
156
157 if (boff > size)
158 return EINVAL;
159
160 bp = buf + boff;
161 bsize = min(size - boff, DNS_NAME_MAX_SIZE);
162 first = true;
163 *eoff = 0;
164
165 while (true) {
166 if (bsize == 0) {
167 rc = EINVAL;
168 goto error;
169 }
170
171 lsize = *bp;
172 ++bp;
173 --bsize;
174
175 if (lsize == 0)
176 break;
177
178 if (!first) {
179 printf(".");
180 rc = dns_dstr_ext(&name, ".");
181 if (rc != EOK) {
182 rc = ENOMEM;
183 goto error;
184 }
185 }
186
187 if ((lsize & 0xc0) == 0xc0) {
188 printf("Pointer\n");
189 /* Pointer */
190 if (bsize < 1) {
191 printf("Pointer- bsize < 1\n");
192 rc = EINVAL;
193 goto error;
194 }
195
196 ptr = dns_uint16_t_decode(bp - 1, bsize) & 0x3fff;
197 ++bp;
198 --bsize;
199
200 if (ptr >= (size_t)(bp - buf)) {
201 printf("Pointer- forward ref %u, pos=%u\n",
202 ptr, bp - buf);
203 /* Forward reference */
204 rc = EINVAL;
205 goto error;
206 }
207
208 /*
209 * Make sure we will not decode any byte twice.
210 * XXX Is assumption correct?
211 */
212 eptr = bp - buf;
213 /*
214 * This is where encoded name ends in terms where
215 * the message continues
216 */
217 *eoff = eptr;
218
219 printf("ptr=%u, eptr=%u\n", ptr, eptr);
220 bp = buf + ptr;
221 bsize = eptr - ptr;
222 continue;
223 }
224
225 if (lsize > bsize) {
226 rc = EINVAL;
227 goto error;
228 }
229
230 for (i = 0; i < lsize; i++) {
231 printf("%c", *bp);
232
233 if (*bp < 32 || *bp >= 127) {
234 rc = EINVAL;
235 goto error;
236 }
237
238 dbuf[0] = *bp;
239 dbuf[1] = '\0';
240
241 rc = dns_dstr_ext(&name, dbuf);
242 if (rc != EOK) {
243 rc = ENOMEM;
244 goto error;
245 }
246 ++bp;
247 --bsize;
248 }
249
250 first = false;
251 }
252
253 printf("\n");
254
255 *rname = name;
256 if (*eoff == 0)
257 *eoff = bp - buf;
258 return EOK;
259error:
260 free(name);
261 return rc;
262}
263
264/** Decode unaligned big-endian 16-bit integer */
265static uint16_t dns_uint16_t_decode(uint8_t *buf, size_t buf_size)
266{
267 assert(buf_size >= 2);
268
269 return ((uint16_t)buf[0] << 8) + buf[1];
270}
271
272/** Encode unaligned big-endian 16-bit integer */
273static void dns_uint16_t_encode(uint16_t w, uint8_t *buf, size_t buf_size)
274{
275 if (buf != NULL && buf_size >= 1)
276 buf[0] = w >> 8;
277
278 if (buf != NULL && buf_size >= 2)
279 buf[1] = w & 0xff;
280}
281
282/** Decode unaligned big-endian 32-bit integer */
283uint32_t dns_uint32_t_decode(uint8_t *buf, size_t buf_size)
284{
285 uint32_t w;
286 assert(buf_size >= 4);
287
288 w = ((uint32_t)buf[0] << 24) +
289 ((uint32_t)buf[1] << 16) +
290 ((uint32_t)buf[2] << 8) +
291 buf[3];
292
293 printf("dns_uint32_t_decode: %x, %x, %x, %x -> %x\n",
294 buf[0], buf[1], buf[2], buf[3], w);
295 return w;
296}
297
298static int dns_question_encode(dns_question_t *question, uint8_t *buf,
299 size_t buf_size, size_t *act_size)
300{
301 size_t name_size;
302 size_t di;
303 int rc;
304
305 rc = dns_name_encode(question->qname, buf, buf_size, &name_size);
306 if (rc != EOK)
307 return rc;
308
309 printf("name_size=%zu\n", name_size);
310
311 *act_size = name_size + sizeof(uint16_t) + sizeof(uint16_t);
312 printf("act_size=%zu\n", *act_size);
313 if (buf == NULL)
314 return EOK;
315
316 di = name_size;
317
318 dns_uint16_t_encode(question->qtype, buf + di, buf_size - di);
319 di += sizeof(uint16_t);
320
321 dns_uint16_t_encode(question->qclass, buf + di, buf_size - di);
322 di += sizeof(uint16_t);
323
324 return EOK;
325}
326
327static int dns_question_decode(uint8_t *buf, size_t buf_size, size_t boff,
328 dns_question_t **rquestion, size_t *eoff)
329{
330 dns_question_t *question;
331 size_t name_eoff;
332 int rc;
333
334 question = calloc(1, sizeof (dns_question_t));
335 if (question == NULL)
336 return ENOMEM;
337
338 printf("decode name..\n");
339 rc = dns_name_decode(buf, buf_size, boff, &question->qname, &name_eoff);
340 if (rc != EOK) {
341 printf("error decoding name..\n");
342 free(question);
343 return ENOMEM;
344 }
345
346 printf("ok decoding name..\n");
347 if (name_eoff + 2 * sizeof(uint16_t) > buf_size) {
348 printf("name_eoff + 2 * 2 = %d > buf_size = %d\n",
349 name_eoff + 2 * sizeof(uint16_t), buf_size);
350 free(question);
351 return EINVAL;
352 }
353
354 question->qtype = dns_uint16_t_decode(buf + name_eoff, buf_size - name_eoff);
355 question->qclass = dns_uint16_t_decode(buf + sizeof(uint16_t) + name_eoff,
356 buf_size - sizeof(uint16_t) - name_eoff);
357 *eoff = name_eoff + 2 * sizeof(uint16_t);
358
359 *rquestion = question;
360 return EOK;
361}
362
363static int dns_rr_decode(uint8_t *buf, size_t buf_size, size_t boff,
364 dns_rr_t **retrr, size_t *eoff)
365{
366 dns_rr_t *rr;
367 size_t name_eoff;
368 uint8_t *bp;
369 size_t bsz;
370 size_t rdlength;
371 int rc;
372
373 rr = calloc(1, sizeof (dns_rr_t));
374 if (rr == NULL)
375 return ENOMEM;
376
377 printf("decode name..\n");
378 rc = dns_name_decode(buf, buf_size, boff, &rr->name, &name_eoff);
379 if (rc != EOK) {
380 printf("error decoding name..\n");
381 free(rr);
382 return ENOMEM;
383 }
384
385 printf("ok decoding name.. '%s'\n", rr->name);
386 if (name_eoff + 2 * sizeof(uint16_t) > buf_size) {
387 printf("name_eoff + 2 * 2 = %d > buf_size = %d\n",
388 name_eoff + 2 * sizeof(uint16_t), buf_size);
389 free(rr->name);
390 free(rr);
391 return EINVAL;
392 }
393
394 bp = buf + name_eoff;
395 bsz = buf_size - name_eoff;
396
397 if (bsz < 3 * sizeof(uint16_t) + sizeof(uint32_t)) {
398 free(rr->name);
399 free(rr);
400 return EINVAL;
401 }
402
403 rr->rtype = dns_uint16_t_decode(bp, bsz);
404 bp += sizeof(uint16_t); bsz -= sizeof(uint16_t);
405 printf("rtype=%u\n", rr->rtype);
406
407 rr->rclass = dns_uint16_t_decode(bp, bsz);
408 bp += sizeof(uint16_t); bsz -= sizeof(uint16_t);
409 printf("rclass=%u\n", rr->rclass);
410
411 rr->ttl = dns_uint32_t_decode(bp, bsz);
412 bp += sizeof(uint32_t); bsz -= sizeof(uint32_t);
413 printf("ttl=%u\n", rr->ttl);
414
415 rdlength = dns_uint16_t_decode(bp, bsz);
416 bp += sizeof(uint16_t); bsz -= sizeof(uint16_t);
417 printf("rdlength=%u\n", rdlength);
418
419 if (rdlength > bsz) {
420 free(rr->name);
421 free(rr);
422 return EINVAL;
423 }
424
425 rr->rdata_size = rdlength;
426 rr->rdata = calloc(1, sizeof(rdlength));
427 if (rr->rdata == NULL) {
428 free(rr->name);
429 free(rr);
430 return ENOMEM;
431 }
432
433 memcpy(rr->rdata, bp, rdlength);
434 bp += rdlength;
435 bsz -= rdlength;
436
437 *eoff = bp - buf;
438 *retrr = rr;
439 return EOK;
440}
441
442int dns_message_encode(dns_message_t *msg, void **rdata, size_t *rsize)
443{
444 uint8_t *data;
445 size_t size;
446 dns_header_t hdr;
447 size_t q_size;
448 size_t di;
449 int rc;
450
451 hdr.id = host2uint16_t_be(msg->id);
452
453 hdr.opbits = host2uint16_t_be(
454 (msg->qr << OPB_QR) |
455 (msg->opcode << OPB_OPCODE_l) |
456 (msg->aa ? BIT_V(uint16_t, OPB_AA) : 0) |
457 (msg->tc ? BIT_V(uint16_t, OPB_TC) : 0) |
458 (msg->rd ? BIT_V(uint16_t, OPB_RD) : 0) |
459 (msg->ra ? BIT_V(uint16_t, OPB_RA) : 0) |
460 msg->rcode
461 );
462
463 hdr.qd_count = host2uint16_t_be(list_count(&msg->question));
464 hdr.an_count = 0;
465 hdr.ns_count = 0;
466 hdr.ar_count = 0;
467
468 size = sizeof(dns_header_t);
469 printf("dns header size=%zu\n", size);
470
471 list_foreach(msg->question, link) {
472 dns_question_t *q = list_get_instance(link, dns_question_t, msg);
473 rc = dns_question_encode(q, NULL, 0, &q_size);
474 if (rc != EOK)
475 return rc;
476
477 printf("q_size=%zu\n", q_size);
478 size += q_size;
479 }
480
481 data = calloc(1, size);
482 if (data == NULL)
483 return ENOMEM;
484
485 memcpy(data, &hdr, sizeof(dns_header_t));
486 di = sizeof(dns_header_t);
487
488 list_foreach(msg->question, link) {
489 dns_question_t *q = list_get_instance(link, dns_question_t, msg);
490 rc = dns_question_encode(q, data + di, size - di, &q_size);
491 if (rc != EOK) {
492 assert(rc == ENOMEM || rc == EINVAL);
493 free(data);
494 return rc;
495 }
496
497 di += q_size;
498 }
499
500 printf("-> size=%zu, di=%zu\n", size, di);
501 *rdata = data;
502 *rsize = size;
503 return EOK;
504}
505
506int dns_message_decode(void *data, size_t size, dns_message_t **rmsg)
507{
508 dns_message_t *msg;
509 dns_header_t *hdr;
510 size_t doff;
511 size_t field_eoff;
512 dns_question_t *question;
513 dns_rr_t *rr;
514 size_t qd_count;
515 size_t an_count;
516 size_t i;
517 int rc;
518
519 msg = calloc(1, sizeof(dns_message_t));
520 if (msg == NULL)
521 return ENOMEM;
522
523 if (size < sizeof(dns_header_t))
524 return EINVAL;
525
526 hdr = data;
527
528 msg->id = uint16_t_be2host(hdr->id);
529 msg->qr = BIT_RANGE_EXTRACT(uint16_t, OPB_QR, OPB_QR, hdr->opbits);
530 msg->opcode = BIT_RANGE_EXTRACT(uint16_t, OPB_OPCODE_h, OPB_OPCODE_l,
531 hdr->opbits);
532 msg->aa = BIT_RANGE_EXTRACT(uint16_t, OPB_AA, OPB_AA, hdr->opbits);
533 msg->tc = BIT_RANGE_EXTRACT(uint16_t, OPB_TC, OPB_TC, hdr->opbits);
534 msg->rd = BIT_RANGE_EXTRACT(uint16_t, OPB_RD, OPB_RD, hdr->opbits);
535 msg->ra = BIT_RANGE_EXTRACT(uint16_t, OPB_RA, OPB_RA, hdr->opbits);
536 msg->rcode = BIT_RANGE_EXTRACT(uint16_t, OPB_RCODE_h, OPB_RCODE_l,
537 hdr->opbits);
538
539 list_initialize(&msg->question);
540 list_initialize(&msg->answer);
541 list_initialize(&msg->authority);
542 list_initialize(&msg->additional);
543
544 doff = sizeof(dns_header_t);
545
546 qd_count = uint16_t_be2host(hdr->qd_count);
547 printf("qd_count = %d\n", (int)qd_count);
548
549 for (i = 0; i < qd_count; i++) {
550 printf("decode question..\n");
551 rc = dns_question_decode(data, size, doff, &question, &field_eoff);
552 if (rc != EOK) {
553 printf("error decoding question\n");
554 goto error;
555 }
556 printf("ok decoding question\n");
557
558 list_append(&question->msg, &msg->question);
559 doff = field_eoff;
560 }
561
562 an_count = uint16_t_be2host(hdr->an_count);
563 printf("an_count = %d\n", an_count);
564
565 for (i = 0; i < an_count; i++) {
566 printf("decode answer..\n");
567 rc = dns_rr_decode(data, size, doff, &rr, &field_eoff);
568 if (rc != EOK) {
569 printf("error decoding answer\n");
570 goto error;
571 }
572 printf("ok decoding answer\n");
573
574 list_append(&rr->msg, &msg->answer);
575 doff = field_eoff;
576 }
577
578 printf("ns_count = %d\n", uint16_t_be2host(hdr->ns_count));
579 printf("ar_count = %d\n", uint16_t_be2host(hdr->ar_count));
580
581 *rmsg = msg;
582 return EOK;
583error:
584 dns_message_destroy(msg);
585 return rc;
586}
587
588static void dns_question_destroy(dns_question_t *question)
589{
590 free(question->qname);
591 free(question);
592}
593
594static void dns_rr_destroy(dns_rr_t *rr)
595{
596 free(rr->name);
597 free(rr->rdata);
598 free(rr);
599}
600
601void dns_message_destroy(dns_message_t *msg)
602{
603 link_t *link;
604 dns_question_t *question;
605 dns_rr_t *rr;
606
607 while (!list_empty(&msg->question)) {
608 link = list_first(&msg->question);
609 question = list_get_instance(link, dns_question_t, msg);
610 list_remove(&question->msg);
611 dns_question_destroy(question);
612 }
613
614 while (!list_empty(&msg->answer)) {
615 link = list_first(&msg->answer);
616 rr = list_get_instance(link, dns_rr_t, msg);
617 list_remove(&rr->msg);
618 dns_rr_destroy(rr);
619 }
620
621 while (!list_empty(&msg->authority)) {
622 link = list_first(&msg->authority);
623 rr = list_get_instance(link, dns_rr_t, msg);
624 list_remove(&rr->msg);
625 dns_rr_destroy(rr);
626 }
627
628 while (!list_empty(&msg->additional)) {
629 link = list_first(&msg->additional);
630 rr = list_get_instance(link, dns_rr_t, msg);
631 list_remove(&rr->msg);
632 dns_rr_destroy(rr);
633 }
634
635 free(msg);
636}
637
638/** @}
639 */
Note: See TracBrowser for help on using the repository browser.