summaryrefslogtreecommitdiff
path: root/src/dnsstub/dns_dfd.c
blob: 772dbc32f388cb2275eaf2d148b900345543f874 (plain)
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#include "alloc.h"
#include "byte.h"
#include "dnsresolv.h"
#include "error.h"

/**
	@file dns_dfd.c
	@author djb
	@ref ucspi-tcp
	@brief domain name qualification (domain from dot)
*/

int dns_domain_fromdot(char **out, const char *buf, unsigned int n)
{
  char label[63];
  unsigned int labellen = 0; /* <= sizeof label */
  char name[255];
  unsigned int namelen = 0; /* <= sizeof name */
  char ch;
  char *x;

  errno = EPROTO;

  for (;;) {
    if (!n) break;
    ch = *buf++;
    --n;
    if (ch == '.') {
      if (labellen) {
        if (namelen + labellen + 1 > sizeof(name)) return 0;
        name[namelen++] = labellen;
        byte_copy(name + namelen, labellen, label);
        namelen += labellen;
        labellen = 0;
      }
      continue;
    }
    if (ch == '\\') {  // octal -> decimal
      if (!n) break;
      ch = *buf++;
      --n;
      if ((ch >= '0') && (ch <= '7')) {
        ch -= '0';
        if (n && (*buf >= '0') && (*buf <= '7')) {
          ch <<= 3;
          ch += *buf - '0';
          ++buf;
          --n;
          if (n && (*buf >= '0') && (*buf <= '7')) {
            ch <<= 3;
            ch += *buf - '0';
            ++buf;
            --n;
          }
        }
      }
    }
    if (labellen >= sizeof(label)) return 0;
    label[labellen++] = ch;
  }

  if (labellen) {
    if (namelen + labellen + 1 > sizeof(name)) return 0;
    name[namelen++] = labellen;
    byte_copy(name + namelen, labellen, label);
    namelen += labellen;
    labellen = 0;
  }

  if (namelen + 1 > sizeof(name)) return 0;
  name[namelen++] = 0;

  x = alloc(namelen);
  if (!x) return DNS_MEM;
  byte_copy(x, namelen, name);

  if (*out) alloc_free(*out);
  *out = x;
  return 1;
}