tinydns-data.c

Go to the documentation of this file.

#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include "uint_t.h"
#include "str.h"
#include "byte.h"
#include "fmt.h"
#include "ip.h"
#include "exit.h"
#include "case.h"
#include "scan.h"
#include "buffer.h"
#include "logmsg.h"
#include "getln.h"
#include "cdbmake.h"
#include "stralloc.h"
#include "open.h"
#include "dns.h"
#include <stdio.h>
 
#define TTL_NS 259200
#define TTL_POSITIVE 86400
#define TTL_NEGATIVE 2560
 
#define WHO "tinydns-data"
 
int rename(const char *,const char *);  // keep compiler silent
 
void die_datatmp(void)
{
  logmsg(WHO,111,FATAL,"unable to create data.tmp");
}
 
void nomem(void)
{
  logmsg(WHO,111,FATAL,"nomem");
}
 
void ttdparse(stralloc *sa,char ttd[8])
{
  unsigned int i;
  char ch;
 
  byte_zero(ttd,8);
  for (i = 0; (i < 16) && (i < sa->len); ++i) {
    ch = sa->s[i];
    if ((ch >= '0') && (ch <= '9'))
      ch -= '0';
    else if ((ch >= 'a') && (ch <= 'f'))
      ch -= 'a' - 10;
    else
      ch = 0;
    if (!(i & 1)) ch <<= 4;
    ttd[i >> 1] |= ch;
  }
}
 
void locparse(stralloc *sa,char loc[2])
{
  loc[0] = (sa->len > 0) ? sa->s[0] : 0;
  loc[1] = (sa->len > 1) ? sa->s[1] : 0;
}
 
void ipprefix_cat(stralloc *out,char *s)
{
  uint32_t u;
  char ch;
  uint32_t j;
 
  if (*s == 's') {
    ++s;
    if (!stralloc_catb(out,"s",1) || !stralloc_cats(out,s)) nomem();
  } else {
    if (*s == 'f') ++s;
    if (!stralloc_catb(out,"f",1)) nomem();
    for (;;)
      if (*s == '.') ++s;
      else {
        j = scan_ulong(s,&u);
        if (!j) return;
        s += j;
        ch = u;
        if (!stralloc_catb(out,&ch,1)) nomem();
      }
    }
}
 
void hexparse(stralloc *sa)
{
  char ch, lo;
  unsigned int i;
  unsigned int j;
 
  j = 0;
  i = 0;
  while (i < sa->len) {
    ch = sa->s[i++];
    if ((ch >= '0') && (ch <= '9')) ch -= 48; 
    if ((ch >= 'a') && (ch <= 'f')) ch -= 87;
    ch <<= 4;
    lo = sa->s[i++];
    if ((lo >= '0') && (lo <= '9')) lo -= 48;
    if ((lo >= 'a') && (lo <= 'f')) lo -= 87;
    ch += lo;
    sa->s[j++] = ch;
  }
  sa->len = j;
}
 
void txtparse(stralloc *sa)
{
  char ch;
  unsigned int i;
  unsigned int j;
 
  j = 0;
  i = 0;
  while (i < sa->len) {
    ch = sa->s[i++];
    if (ch == '\\') {
      if (i >= sa->len) break;
      ch = sa->s[i++];
      if ((ch >= '0') && (ch <= '7')) {
        ch -= '0';
        if ((i < sa->len) && (sa->s[i] >= '0') && (sa->s[i] <= '7')) {
          ch <<= 3;
          ch += sa->s[i++] - '0';
          if ((i < sa->len) && (sa->s[i] >= '0') && (sa->s[i] <= '7')) {
            ch <<= 3;
            ch += sa->s[i++] - '0';
          }
        }
      }
    }
    sa->s[j++] = ch;
  }
  sa->len = j;
}
 
void textparse(stralloc *sa)
{
  char ch;
  unsigned int i;
  unsigned int j;
 
  j = 0;
  i = 0;
  while (i < sa->len) {
    ch = sa->s[i++];
    if (ch >= ' ' && ch <= '~') 
      { sa->s[j] = ch; j++; }
  }
  sa -> s[j] = '\0';
  sa->len = j ;
}
 
char defaultsoa[20];
 
void defaultsoa_init(int fd)
{
  struct stat st;
  if (fstat(fd,&st) == -1)
    logmsg(WHO,111,FATAL,"unable to stat data");
  uint32_pack_big(defaultsoa,st.st_mtime);
  if (byte_equal(defaultsoa,4,"\0\0\0\0"))
    defaultsoa[3] = 1;
  byte_copy(defaultsoa + 4,16,"\0\0\100\000\0\0\010\000\0\020\000\000\0\0\012\000");
}
 
int fdcdb;
struct cdb_make cdb;
static stralloc key;
static stralloc result;
 
void rr_add(const char *buf,unsigned int len)
{
  if (!stralloc_catb(&result,buf,len)) nomem();
}
 
void rr_addname(const char *d)
{
  rr_add(d,dns_domain_length(d));
}
 
void rr_start(const char type[2],unsigned long ttl,const char ttd[8],const char loc[2])
{
  char buf[4];
  if (!stralloc_copyb(&result,type,2)) nomem();
  if (byte_equal(loc,2,"\0\0"))
    rr_add("=",1);
  else {
    rr_add(">",1);
    rr_add(loc,2);
  }
  uint32_pack_big(buf,ttl);
  rr_add(buf,4);
  rr_add(ttd,8);
}
 
void rr_finish(const char *owner)
{
  if (byte_equal(owner,2,"\1*")) {
    owner += 2;
    result.s[2] -= 19;
  }
  if (!stralloc_copyb(&key,owner,dns_domain_length(owner))) nomem();
  case_lowerb(key.s,key.len);
  if (cdb_make_add(&cdb,key.s,key.len,result.s,result.len) == -1)
    die_datatmp();
}
 
buffer b;
char bspace[1024];
 
static stralloc line;
int match = 1;
unsigned long linenum = 0;
 
#define NUMFIELDS 15
static stralloc f[NUMFIELDS];
 
static char *d1;
static char *d2;
char d4ptr[DNS_NAME4_DOMAIN];
char d6ptr[DNS_NAME6_DOMAIN];
 
char strnum[FMT_ULONG];
 
void syntaxerror(const char *why)
{
  strnum[fmt_ulong(strnum,linenum)] = 0;
  logmsg(WHO,111,FATAL,B("unable to parse data line: ",strnum,why));
}
 
static unsigned int scan_u32(const char *s,uint32 *u) 
{
  unsigned long l;
  unsigned int r = scan_ulong(s,&l);
  if ((uint32)l != l) return 0;
  if (r) *u=l;
  return r;
}
 
int main()
{
  int fddata;
  int i;
  int j;
  int k;
  char ch;
  char ttd[8];
  char loc[2];
  char ip4[4];
  char ip6[16];
  char type[2];
  char soa[20];
  char buf[4];
  uint32_t ttl, u;
  unsigned char us, sl, ty;
  umask(022);
 
  fddata = open_read("data");
  if (fddata == -1)
    logmsg(WHO,111,FATAL,"unable to open data"); 
  
  defaultsoa_init(fddata); 
  buffer_init(&b,buffer_unixread,fddata,bspace,sizeof(bspace));
 
  fdcdb = open_trunc("data.tmp");
  if (fdcdb == -1) die_datatmp();
  if (cdb_make_start(&cdb,fdcdb) == -1) die_datatmp();
 
  while (match) {
    ++linenum;
    if (getln(&b,&line,&match,'\n') == -1)
      logmsg(WHO,111,FATAL,"unable to read line"); 
 
    while (line.len) {
      ch = line.s[line.len - 1];
      if ((ch != ' ') && (ch != '\t') && (ch != '\n')) break;
      --line.len;
    }
    if (!line.len) continue;
    if (line.s[0] == '#') continue;
    if (line.s[0] == '-') continue;
 
    /* Tokenization of input fields; delimitor: | */
 
//    j = 1;
    for (j = 1, i = 0; i < NUMFIELDS; ++i) {
      if (j >= line.len) {
        if (!stralloc_copys(&f[i],"")) nomem();
      }
      else {
        k = byte_chr(line.s + j,line.len - j,'|');
        if (!stralloc_copyb(&f[i],line.s + j,k)) nomem();
        j += k + 1;
      }
    }
 
    switch (line.s[0]) {
 
      case '%':                 /* local extension */
              locparse(&f[0],loc);
              if (!stralloc_copyb(&key,"\0%",2)) nomem();
              if (!stralloc_0(&f[1])) nomem();
              ipprefix_cat(&key,f[1].s);
        if (cdb_make_add(&cdb,key.s,key.len,loc,2) == -1)
          die_datatmp();
        break;
 
      case 'Z':                 /* SOA records */
        if (dns_domain_fromdot(&d1,f[0].s,f[0].len) <= 0) nomem();
 
        if (!stralloc_0(&f[3])) nomem();
        if (!scan_u32(f[3].s,&u)) uint32_unpack_big(defaultsoa,&u);
        uint32_pack_big(soa,u);
        if (!stralloc_0(&f[4])) nomem();
        if (!scan_u32(f[4].s,&u)) uint32_unpack_big(defaultsoa + 4,&u);
        uint32_pack_big(soa + 4,u);
        if (!stralloc_0(&f[5])) nomem();
        if (!scan_u32(f[5].s,&u)) uint32_unpack_big(defaultsoa + 8,&u);
        uint32_pack_big(soa + 8,u);
        if (!stralloc_0(&f[6])) nomem();
        if (!scan_u32(f[6].s,&u)) uint32_unpack_big(defaultsoa + 12,&u);
        uint32_pack_big(soa + 12,u);
        if (!stralloc_0(&f[7])) nomem();
        if (!scan_u32(f[7].s,&u)) uint32_unpack_big(defaultsoa + 16,&u);
        uint32_pack_big(soa + 16,u);
 
        if (!stralloc_0(&f[8])) nomem();
        if (!scan_ulong(f[8].s,&ttl)) ttl = TTL_NEGATIVE;
        ttdparse(&f[9],ttd);
        locparse(&f[10],loc);
 
        rr_start(DNS_T_SOA,ttl,ttd,loc);
        if (dns_domain_fromdot(&d2,f[1].s,f[1].len) <= 0) nomem();
        rr_addname(d2);
        if (dns_domain_fromdot(&d2,f[2].s,f[2].len) <= 0) nomem();
        rr_addname(d2);
        rr_add(soa,20);
        rr_finish(d1);
        break;
 
      case '.': case '&':       /* NS records */
        if (dns_domain_fromdot(&d1,f[0].s,f[0].len) <= 0) nomem();
        if (!stralloc_0(&f[3])) nomem();
        if (!scan_ulong(f[3].s,&ttl)) ttl = TTL_NS;
        ttdparse(&f[4],ttd);
        locparse(&f[5],loc);
 
        if (!stralloc_0(&f[1])) nomem();
 
        if (byte_chr(f[2].s,f[2].len,'.') >= f[2].len) {
          if (!stralloc_cats(&f[2],".ns.")) nomem();
          if (!stralloc_catb(&f[2],f[0].s,f[0].len)) nomem();
        }
        if (dns_domain_fromdot(&d2,f[2].s,f[2].len) <= 0) nomem();
 
        if (line.s[0] == '.') {
          rr_start(DNS_T_SOA,ttl ? TTL_NEGATIVE : 0,ttd,loc);
          rr_addname(d2);
          rr_add("\12hostmaster",11);
          rr_addname(d1);
          rr_add(defaultsoa,20);
          rr_finish(d1);
        }
 
        rr_start(DNS_T_NS,ttl,ttd,loc);
        rr_addname(d2);
        rr_finish(d1);
 
        if (byte_chr(f[1].s,f[1].len,':') < f[1].len) {
          if (ip6_scan(f[1].s,ip6)) {
            rr_start(DNS_T_AAAA,ttl,ttd,loc);
            rr_add(ip6,16);
            rr_finish(d2);
          }
        } else {
          if (ip4_scan(f[1].s,ip4)) {
            rr_start(DNS_T_A,ttl,ttd,loc);
            rr_add(ip4,4);
            rr_finish(d2);
          }
        }
        break;
 
      case '+': case '=':       /* A records */
        if (dns_domain_fromdot(&d1,f[0].s,f[0].len) <= 0) nomem();
        if (!stralloc_0(&f[2])) nomem();
        if (!scan_ulong(f[2].s,&ttl)) ttl = TTL_POSITIVE;
        ttdparse(&f[3],ttd);
        locparse(&f[4],loc);
 
        if (!stralloc_0(&f[1])) nomem();
 
        if (ip4_scan(f[1].s,ip4)) {
          rr_start(DNS_T_A,ttl,ttd,loc);
          rr_add(ip4,4);
          rr_finish(d1);
 
          if (line.s[0] == '=') {
            dns_name4_domain(d4ptr,ip4);
            rr_start(DNS_T_PTR,ttl,ttd,loc);
            rr_addname(d1);
            rr_finish(d4ptr);
          }
        }
        break;
 
      case ':': case '~':       /* AAAA records */
        if (dns_domain_fromdot(&d1,f[0].s,f[0].len) <= 0) nomem();
        if (!stralloc_0(&f[2])) nomem();
        if (!scan_ulong(f[2].s,&ttl)) ttl = TTL_POSITIVE;
        ttdparse(&f[3],ttd);
        locparse(&f[4],loc);
 
        if (!stralloc_0(&f[1])) nomem();
 
        if (ip6_scan(f[1].s,ip6)) {
          rr_start(DNS_T_AAAA,ttl,ttd,loc);
          rr_add(ip6,16);
          rr_finish(d1);
 
          if (line.s[0] == ':') {
            dns_name6_domain(d6ptr,ip6);
            rr_start(DNS_T_PTR,ttl,ttd,loc);
            rr_addname(d1);
            rr_finish(d6ptr);
          }
        }
        break;
 
      case '@':                 /* MX Records */
        if (dns_domain_fromdot(&d1,f[0].s,f[0].len) <= 0) nomem();
        if (!stralloc_0(&f[4])) nomem();
        if (!scan_ulong(f[4].s,&ttl)) ttl = TTL_POSITIVE;
        ttdparse(&f[5],ttd);
        locparse(&f[6],loc);
 
        if (!stralloc_0(&f[1])) nomem();
 
        if (byte_chr(f[2].s,f[2].len,'.') >= f[2].len) {
          if (!stralloc_cats(&f[2],".mx.")) nomem();
          if (!stralloc_catb(&f[2],f[0].s,f[0].len)) nomem();
        }
        if (dns_domain_fromdot(&d2,f[2].s,f[2].len) <= 0) nomem();
 
        if (!stralloc_0(&f[3])) nomem();
        if (!scan_u32(f[3].s,&u)) u = 0;
 
        rr_start(DNS_T_MX,ttl,ttd,loc);
        uint16_pack_big(buf,u);
        rr_add(buf,2);
        rr_addname(d2);
        rr_finish(d1);
 
        if (byte_chr(f[1].s,f[1].len,':') < f[1].len) {
            if (ip6_scan(f[1].s,ip6)) 
            if (!ip6_isv4mapped(ip6)) {
              rr_start(DNS_T_AAAA,ttl,ttd,loc);
              rr_add(ip6,16);
              rr_finish(d2);
            }
        } else 
          if (ip4_scan(f[1].s,ip4)) {
            rr_start(DNS_T_A,ttl,ttd,loc);
            rr_add(ip4,4);
            rr_finish(d2);
          }
        break;
 
      case '^': case 'C':       /* Pointer + CName Records */
        if (dns_domain_fromdot(&d1,f[0].s,f[0].len) <= 0) nomem();
        if (dns_domain_fromdot(&d2,f[1].s,f[1].len) <= 0) nomem();
        if (!stralloc_0(&f[2])) nomem();
        if (!scan_ulong(f[2].s,&ttl)) ttl = TTL_POSITIVE;
        ttdparse(&f[3],ttd);
        locparse(&f[4],loc);
 
        if (line.s[0] == 'C')
          rr_start(DNS_T_CNAME,ttl,ttd,loc);
        else
          rr_start(DNS_T_PTR,ttl,ttd,loc);
        rr_addname(d2);
        rr_finish(d1);
        break;
 
      case '\'':        /* TXT Records*/
        if (dns_domain_fromdot(&d1,f[0].s,f[0].len) <= 0) nomem();
        if (!stralloc_0(&f[2])) nomem();
        if (!scan_ulong(f[2].s,&ttl)) ttl = TTL_POSITIVE;
        ttdparse(&f[3],ttd);
        locparse(&f[4],loc);
       
        rr_start(DNS_T_TXT,ttl,ttd,loc);
 
        textparse(&f[1]);
        i = 0;
        while (i < f[1].len) {
          k = f[1].len - i;
          if (k > 127) k = 127;
          ch = k;
          rr_add(&ch,1);
          rr_add(f[1].s + i,k);
          i += k;
        }
        rr_finish(d1);
        break;
 
      case '_':         /* TLSA  Records*/
        /* _fqdn|u|s|fingerprint|x|port|proto|ttl|timestamp|lo */
        /*   0   1 2     3       4  5     6    7    8        9 */
 
        if (dns_domain_fromdot(&d1,f[0].s,f[0].len) <= 0) nomem();  // d1
        if (!stralloc_0(&f[7])) nomem();
        if (!scan_ulong(f[7].s,&ttl)) ttl = TTL_POSITIVE;
        ttdparse(&f[8],ttd);
        locparse(&f[9],loc);
 
        if (!stralloc_0(&f[1])) nomem(); // usage
        if (!scan_uint(f[1].s,&us)) us = 3;
        us = 003;
        if (!stralloc_0(&f[2])) nomem(); // selector
        if (!scan_uint(f[2].s,&sl)) sl = 0;
        f[2].len = 0;
        sl = 000;
        ty = 0;                         // type
        if (f[3].len == 64) ty = 001;
        if (f[3].len == 128) ty = 002;
       
        if (f[4].len == 0 && f[5].len == 0 && f[6].len == 0) { // _25._tcp.mx.fqdn
          if (!stralloc_copys(&f[2],"_25._tcp.mail.")) nomem();
          if (!stralloc_catb(&f[2],f[0].s,f[0].len)) nomem();
        } else if (f[4].s[0] != '_') {                         // synthesize base domain
          if (!stralloc_copys(&f[2],"_")) nomem();
          if (f[5].len > 0) {
            if (!stralloc_catb(&f[2],f[5].s,f[5].len)) nomem();
          } else
            if (!stralloc_cats(&f[2],"25")) nomem();
          if (!stralloc_cats(&f[2],"._")) nomem();
          if (f[6].len > 0) {
            if (!stralloc_catb(&f[2],f[6].s,f[6].len)) nomem();
          } else 
            if (!stralloc_cats(&f[2],"tcp")) nomem();
          if (f[4].s[0] != '.') 
            if (!stralloc_cats(&f[2],".")) nomem();
          if (!stralloc_catb(&f[2],f[4].s,f[4].len)) nomem();
          if (!stralloc_cats(&f[2],".")) nomem();
          if (!stralloc_catb(&f[2],f[0].s,f[0].len)) nomem();
        } else
          if (!stralloc_copy(&f[2],&f[4])) nomem();
       
        if (dns_domain_fromdot(&d2,f[2].s,f[2].len) <= 0) nomem();  // d2 - new record
 
        rr_start(DNS_T_TLSA,ttl,ttd,loc);
 
        buf[0] = us;
        rr_add(buf,1);
        buf[0] = sl;
        rr_add(buf,1);
        buf[0] = ty;
        rr_add(buf,1);
 
        case_lowerb(f[3].s,f[3].len);
        hexparse(&f[3]);
        rr_add(f[3].s,f[3].len);
        rr_finish(d2);
        break;
 
      case 'O':                 /* Any other Records with Octal representation */
        if (dns_domain_fromdot(&d1,f[0].s,f[0].len) <= 0) nomem();
        if (!stralloc_0(&f[3])) nomem();
        if (!scan_ulong(f[3].s,&ttl)) ttl = TTL_POSITIVE;
        ttdparse(&f[4],ttd);
        locparse(&f[5],loc);
 
        if (!stralloc_0(&f[1])) nomem();
        scan_u32(f[1].s,&u);
        uint16_pack_big(type,u);
        if (byte_equal(type,2,DNS_T_AXFR))
          syntaxerror(": type AXFR prohibited");
        if (byte_equal(type,2,"\0\0"))
          syntaxerror(": type 0 prohibited");
        if (byte_equal(type,2,DNS_T_SOA))
          syntaxerror(": type SOA prohibited");
        if (byte_equal(type,2,DNS_T_NS))
          syntaxerror(": type NS prohibited");
        if (byte_equal(type,2,DNS_T_CNAME))
          syntaxerror(": type CNAME prohibited");
        if (byte_equal(type,2,DNS_T_PTR))
          syntaxerror(": type PTR prohibited");
        if (byte_equal(type,2,DNS_T_MX))
          syntaxerror(": type MX prohibited");
 
        txtparse(&f[2]);
 
        rr_start(type,ttl,ttd,loc);
        rr_add(f[2].s,f[2].len);
        rr_finish(d1);
        break;
 
      default:
        syntaxerror(": unrecognized leading character");
    }
  }
 
  if (cdb_make_finish(&cdb) == -1) die_datatmp();
  if (fsync(fdcdb) == -1) die_datatmp();
  if (close(fdcdb) == -1) die_datatmp(); /* NFS stupidity */
  if (rename("data.tmp","data.cdb") == -1)
    logmsg(WHO,111,FATAL,"unable to move data.tmp to data.cdb");
 
  _exit(0);
}