void aesctr_packet_put(struct buf *b) {
long long pos;
crypto_uint8 paddinglen;
struct buf *sendbuf = &packet.sendbuf;
pos = sendbuf->len; /* get position */
buf_putnum32(sendbuf, 0); /* packet id, for MAC */
buf_putnum32(sendbuf, 0); /* packet length */
buf_putnum8(sendbuf, 0); /* padding length */
buf_put(sendbuf, b->buf, b->len); /* add data */
/* add padding */
paddinglen = 2 * BB - ((sendbuf->len - pos - 4) % BB);
paddinglen += randommod(2) * BB;
buf_putpadding(sendbuf, paddinglen);
/* add space for mac */
buf_putzerobytes(sendbuf, AB);
/* add packet ID */
uint32_pack_big(sendbuf->buf + pos, packet.sendpacketid++);
/* add packet length */
uint32_pack_big(sendbuf->buf + pos + 4, sendbuf->len - pos - AB - 4 - 4);
/* add padding length*/
sendbuf->buf[pos + 8] = paddinglen;
/* get mac */
if (sshcrypto_auth(
sendbuf->buf + sendbuf->len - AB, /* a */
sendbuf->buf + pos, /* m */
sendbuf->len - pos - AB, /* mlen */
packet.servermackey /* k */
) != 0) bug_proto();
/* encrypt */
if (sshcrypto_stream_xor(
sendbuf->buf + pos + 4, /* c */
sendbuf->buf + pos + 4, /* m */
sendbuf->len - pos - 4 - AB, /* mlen */
packet.servernonce, /* n */
packet.serverkey /* k */
) != 0) bug_proto();
/* remove packet id */
byte_copy(sendbuf->buf + pos, sendbuf->len - pos - 4, sendbuf->buf + pos + 4);
sendbuf->len -= 4;
/* cleanup */
purge(sendbuf->buf + sendbuf->len, 4);
}
static void server_mainloop() {
int64 sock;
/* inlined livesync_init() */
memset( g_peerbuffer_start, 0, sizeof( g_peerbuffer_start ) );
g_peerbuffer_pos = g_peerbuffer_start;
memcpy( g_peerbuffer_pos, &g_tracker_id, sizeof( g_tracker_id ) );
uint32_pack_big( (char*)g_peerbuffer_pos + sizeof( g_tracker_id ), OT_SYNC_PEER);
g_peerbuffer_pos += sizeof( g_tracker_id ) + sizeof( uint32_t);
g_next_packet_time = time(NULL) + LIVESYNC_MAXDELAY;
while(1) {
/* See, if we need to connect to anyone */
if( time(NULL) > g_connection_reconn )
handle_reconnects( );
/* Wait for io events until next approx reconn check time */
io_waituntil2( 30*1000 );
/* Loop over readable sockets */
while( ( sock = io_canread( ) ) != -1 ) {
const void *cookie = io_getcookie( sock );
if( (uintptr_t)cookie == FLAG_SERVERSOCKET )
handle_accept( sock );
else
handle_read( sock );
}
/* Loop over writable sockets */
while( ( sock = io_canwrite( ) ) != -1 )
handle_write( sock );
livesync_ticker( );
}
}
static void _tobytes(unsigned char *out, fe in) {
long long i;
fe_reducesmall(in, m, 0);
for (i = 0; i < 8; ++i) uint32_pack_big(out + 28 - 4 * i, in[i]);
}
static int init_smb_response(struct smb_response* sr,unsigned char* in_response_to,size_t size) {
if (size<200) size=200;
sr->buf=malloc(sr->allocated=size);
if (!sr->buf) return -1;
sr->used=netbiosheadersize+smbheadersize;
uint32_pack_big(sr->buf,32); // size field in NMB header
byte_copy(sr->buf+netbiosheadersize,smbheadersize-8,
"\xffSMB" // magic
"x" // smb command, filled in later; ofs 4
"\x00\x00\x00\x00" // STATUS_SUCCESS
"\x80" // Flags: response+case sensitive
"\x41\xc0" // Flags2: unicode+long names allowed
"\x00\x00" // Process ID High: 0
"\x00\x00\x00\x00\x00\x00\x00\x00" // Signature
"\x00\x00" // Reserved
); // TID, PID, UID, MID; ofs 24
sr->buf[netbiosheadersize+4]=in_response_to[4];
uint16_pack(sr->buf+netbiosheadersize+24,uint16_read((char*)in_response_to+24));
uint16_pack(sr->buf+netbiosheadersize+26,uint16_read((char*)in_response_to+26));
uint16_pack(sr->buf+netbiosheadersize+28,0);
uint16_pack(sr->buf+netbiosheadersize+30,uint16_read((char*)in_response_to+30));
sr->andxtypeofs=netbiosheadersize+4;
return 0;
}
int aesctr_packet_get(struct buf *b) {
long long len;
struct buf *recvbuf = &packet.recvbuf;
unsigned char *pp;
long long l;
/* we need at least one block */
if (recvbuf->len - ZB < BB) { packet.packet_length = 0; return 1; }
/* decrypt first block */
if (packet.packet_length == 0) {
if (sshcrypto_stream_xor(
recvbuf->buf + ZB, /* c */
recvbuf->buf + ZB, /* m */
BB, /* mlen */
packet.clientnonce, /* n */
packet.clientkey /* k */
) != 0) bug_proto();
packet.packet_length = uint32_unpack_big(recvbuf->buf + ZB);
}
if (packet.packet_length > PACKET_LIMIT) bug_proto();
if (packet.packet_length + 4 + AB > recvbuf->len - ZB) return 1;
/* decrypt and check MAC */
uint32_pack_big(recvbuf->buf + ZB - 4, packet.receivepacketid++);
if (sshcrypto_stream_xor(
recvbuf->buf + ZB + BB, /* c */
recvbuf->buf + ZB + BB, /* m */
packet.packet_length + 4 - BB, /* mlen */
packet.clientnonce, /* n */
packet.clientkey /* k */
) != 0) bug_proto();
if (sshcrypto_auth(
recvbuf->buf, /* a */
recvbuf->buf + ZB - 4, /* m */
packet.packet_length + 8, /* mlen */
packet.clientmackey /* k */
) != 0) bug_proto();
if (crypto_verify_32(recvbuf->buf, recvbuf->buf + ZB + packet.packet_length + 4) != 0) bug_proto();
len = packet.packet_length;
len -= recvbuf->buf[ZB + 4] + 1;
if (len <= 0) bug_proto();
buf_put(b, recvbuf->buf + ZB + 5, len);
pp = recvbuf->buf + ZB;
l = recvbuf->len - ZB;
byte_copy(pp, l - packet.packet_length + AB + 4, pp + packet.packet_length + AB + 4);
purge(pp + l - packet.packet_length + AB + 4, packet.packet_length + AB + 4);
recvbuf->len -= packet.packet_length + AB + 4;
packet.packet_length = 0;
purge(recvbuf->buf, ZB);
return 1;
}
void defaultsoa_init(int fd)
{
struct stat st;
if (fstat(fd,&st) == -1)
strerr_die2sys(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 response_rstart(const char *d,const char type[2],uint32 ttl)
{
char ttlstr[4];
if (!response_addname(d)) return 0;
if (!response_addbytes(type,2)) return 0;
if (!response_addbytes(DNS_C_IN,2)) return 0;
if (flaghidettl) ttl = 0;
uint32_pack_big(ttlstr,ttl);
if (!response_addbytes(ttlstr,4)) return 0;
if (!response_addbytes("\0\0",2)) return 0;
dpos = response_len;
return 1;
}
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 livesync_init( ) {
if( g_socket_in == -1 )
exerr( "No socket address for live sync specified." );
/* Prepare outgoing peers buffer */
memcpy( g_outbuf, &g_tracker_id, sizeof( g_tracker_id ) );
uint32_pack_big( g_outbuf + sizeof( g_tracker_id ), OT_SYNC_PEER);
g_outbuf_data = sizeof( g_tracker_id ) + sizeof( uint32_t );
g_next_packet_time = g_now_seconds + LIVESYNC_MAXDELAY;
pthread_create( &thread_id, NULL, livesync_worker, NULL );
}
void rr_start(const char type[2],unsigned long ttl,const char ttd[8],uint32 loc)
{
char buf[4];
char locp[4];
if (!stralloc_copyb(&result,type,2)) nomem();
//if (byte_equal(loc,2,"\0\0")) char[2]->>uint32loc
if (loc == 0U)
rr_add("=",1);
else {
rr_add(">",1);
uint32_pack(locp,loc);
rr_add(locp,4);
}
uint32_pack_big(buf,ttl);
rr_add(buf,4);
rr_add(ttd,8);
}
int skaserver2_sync (bufalloc *b, char const *before, unsigned int beforelen, char const *after, unsigned int afterlen, struct taia const *deadline, struct taia *stamp)
{
char const *x = env_get("SKACLIENT_ADDITIONAL_FDS") ;
if (x)
{
unsigned int fd ;
if (!uint0_scan(x, &fd)) return (errno = EINVAL, 0) ;
if (!bufalloc_put(bufalloc_1, before, beforelen)) return 0 ;
if (!timed_bufalloc_flush(bufalloc_1, deadline, stamp)) return 0 ;
if (ndelay_on(fd) < 0) return 0 ;
bufalloc_init(b, &fd_write, (int)fd) ;
if (!bufalloc_put(b, after, afterlen)) return 0 ;
if (!timed_bufalloc_flush(b, deadline, stamp)) return 0 ;
return 1 ;
}
else
{
int p[2] ;
if (pipe(p) < 0) return 0 ;
if (ndelay_on(p[1]) < 0) goto fail ;
if (!bufalloc_put(bufalloc_1, before, beforelen)) goto fail ;
{
char pack[4] ;
uint32_pack_big(pack, 1) ;
if (!bufalloc_put(bufalloc_1, pack, 4)) goto fail ;
}
if (!timed_bufalloc_flush(bufalloc_1, deadline, stamp)) goto fail ;
if (!timed_ancil_fd_send(1, p[0], deadline, stamp)) goto fail ;
if (!bufalloc_put(bufalloc_1, after, afterlen)) goto fail ;
if (!timed_bufalloc_flush(bufalloc_1, deadline, stamp)) goto fail ;
#ifndef HASANCILAUTOCLOSE
fd_close(p[0]) ;
#endif
bufalloc_init(b, &fd_write, p[1]) ;
return 1 ;
fail:
{
register int e = errno ;
fd_close(p[1]) ;
fd_close(p[0]) ;
errno = e ;
}
return 0 ;
}
}
static int add_smb_response(struct smb_response* sr,const char* buf,size_t size,unsigned char type) {
if (sr->allocated+size<size) return -1; // check int overflow
if (sr->used+size>sr->allocated) {
size_t n=sr->allocated+size;
void* x;
n=((n-1)|0xfff)+1; // round up to multiple of 0x1000
if (!n) return -1; // check int overflow
x=realloc(sr->buf,n);
if (!x) return -1;
sr->buf=x;
sr->allocated=n;
}
sr->buf[sr->andxtypeofs]=type;
if (sr->andxtypeofs!=netbiosheadersize+4)
uint16_pack(sr->buf+sr->andxtypeofs+2,sr->used-netbiosheadersize);
byte_copy(sr->buf+sr->used,size,buf);
sr->andxtypeofs=sr->used+1;
sr->used+=size;
if (sr->used%2)
sr->buf[++sr->used]=0;
uint32_pack_big(sr->buf,sr->used-4); // update netbios size field
return 0;
}
int main()
{
int fddata;
int i;
int j;
int k;
char ch;
unsigned long ttl;
char ttd[8];
char loc[2];
unsigned long u;
char ip[4];
char ip6[16];
char type[2];
char soa[20];
char buf[4];
umask(022);
fddata = open_read("data");
if (fddata == -1)
strerr_die2sys(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)
strerr_die2sys(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;
j = 1;
for (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 '%':
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':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) nomem();
if (!stralloc_0(&f[3])) nomem();
if (!scan_ulong(f[3].s,&u)) uint32_unpack_big(defaultsoa,&u);
uint32_pack_big(soa,u);
if (!stralloc_0(&f[4])) nomem();
if (!scan_ulong(f[4].s,&u)) uint32_unpack_big(defaultsoa + 4,&u);
uint32_pack_big(soa + 4,u);
if (!stralloc_0(&f[5])) nomem();
if (!scan_ulong(f[5].s,&u)) uint32_unpack_big(defaultsoa + 8,&u);
uint32_pack_big(soa + 8,u);
if (!stralloc_0(&f[6])) nomem();
if (!scan_ulong(f[6].s,&u)) uint32_unpack_big(defaultsoa + 12,&u);
uint32_pack_big(soa + 12,u);
if (!stralloc_0(&f[7])) nomem();
if (!scan_ulong(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)) nomem();
rr_addname(d2);
if (!dns_domain_fromdot(&d2,f[2].s,f[2].len)) nomem();
rr_addname(d2);
rr_add(soa,20);
rr_finish(d1);
break;
case '.': case '&':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) 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)) 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 (ip4_scan(f[1].s,ip)) {
rr_start(DNS_T_A,ttl,ttd,loc);
rr_add(ip,4);
rr_finish(d2);
}
break;
case '+': case '=':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) 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,ip)) {
rr_start(DNS_T_A,ttl,ttd,loc);
rr_add(ip,4);
rr_finish(d1);
if (line.s[0] == '=') {
dns_name4_domain(dptr,ip);
rr_start(DNS_T_PTR,ttl,ttd,loc);
rr_addname(d1);
rr_finish(dptr);
}
}
break;
case '6': case '3':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) 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_flat(f[1].s,ip6)) {
rr_start(DNS_T_AAAA,ttl,ttd,loc);
rr_add(ip6,16);
rr_finish(d1);
if (line.s[0] == '6') { /* emit both .ip6.arpa and .ip6.int */
dns_name6_domain(d6ptr,ip6,DNS_IP6_ARPA);
rr_start(DNS_T_PTR,ttl,ttd,loc);
rr_addname(d1);
rr_finish(d6ptr);
dns_name6_domain(d6ptr,ip6,DNS_IP6_INT);
rr_start(DNS_T_PTR,ttl,ttd,loc);
rr_addname(d1);
rr_finish(d6ptr);
}
}
break;
case '@':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) 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)) nomem();
if (!stralloc_0(&f[3])) nomem();
if (!scan_ulong(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 (ip4_scan(f[1].s,ip)) {
rr_start(DNS_T_A,ttl,ttd,loc);
rr_add(ip,4);
rr_finish(d2);
}
break;
case '^': case 'C':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) nomem();
if (!dns_domain_fromdot(&d2,f[1].s,f[1].len)) 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 '\'':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) 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);
txtparse(&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 ':':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) 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_ulong(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)
strerr_die2sys(111,FATAL,"unable to move data.tmp to data.cdb: ");
_exit(0);
}
int main(int argc,char* argv[]) {
static size_t x;
x=23;
atomic_add(&x,3);
printf("%u\n",x);
printf("%u\n",atomic_add_return(&x,-3));
printf("%u\n",compare_and_swap(&x,26,17));
printf("%u\n",compare_and_swap(&x,23,17));
#if 0
atomic_add(&x,3); printf("%u\n",x);
x=23;
atomic_add(&x,3); assert(x==26);
atomic_or(&x,1); assert(x==27);
atomic_and(&x,-2); assert(x==26);
#endif
#if 0
iarray a;
char* c;
iarray_init(&a,sizeof(io_entry));
printf("15 -> %p\n",c=iarray_allocate(&a,15));
printf("23 -> %p\n",c=iarray_allocate(&a,23));
printf("1234567 -> %p\n",c=iarray_allocate(&a,1234567));
printf("23 -> %p\n",iarray_get(&a,23));
#endif
#if 0
io_batch* b=iob_new(1234);
int64 fd=open("t.c",0);
iob_addbuf(b,"fnord",5);
iob_addfile_close(b,fd,0,7365);
iob_write(1,b,writecb);
#endif
#if 0
char dest[1024];
unsigned long len;
scan_urlencoded2("libstdc++.tar.gz",dest,&len);
buffer_putmflush(buffer_1,dest,"\n");
#endif
#if 0
static stralloc sa;
stralloc_copym(&sa,"foo ","bar ","baz.\n");
write(1,sa.s,sa.len);
#endif
#if 0
buffer_putmflush(buffer_1,"foo ","bar ","baz.\n");
#endif
#if 0
char* c="fnord";
int fd=open_read(c);
errmsg_iam(argv[0]);
carp("could not open file `",c,"'");
diesys(23,"could not open file `",c,"'");
#endif
#if 0
errmsg_warn("could not open file `",c,"'",0);
errmsg_warnsys("could not open file `",c,"'",0);
#endif
#if 0
char buf[100]="/usr/bin/sh";
int len=str_len(buf);
assert(byte_rchr(buf,len,'/')==8);
assert(byte_rchr(buf,len,'@')==len);
assert(byte_rchr(buf,len,'h')==len-1);
printf("%d\n",byte_rchr("x",1,'x'));
#endif
#if 0
char buf[IP6_FMT+100];
int i;
char ip[16];
uint32 scope_id;
char* s="fec0::1:220:e0ff:fe69:ad92%eth0/64";
char blubip[16]="\0\0\0\0\0\0\0\0\0\0\xff\xff\x7f\0\0\001";
i=scan_ip6if(s,ip,&scope_id);
assert(s[i]=='/');
buffer_put(buffer_1,buf,fmt_ip6if(buf,ip,scope_id));
buffer_putnlflush(buffer_1);
buffer_put(buffer_1,buf,fmt_ip6ifc(buf,blubip,scope_id));
buffer_putnlflush(buffer_1);
scan_ip6("2001:7d0:0:f015:0:0:0:1",ip);
buffer_put(buffer_1,buf,fmt_ip6(buf,ip));
buffer_putnlflush(buffer_1);
#endif
#if 0
char buf[100];
int i;
printf("%d\n",i=fmt_pad(buf,"fnord",5,7,10));
buf[i]=0;
puts(buf);
#endif
#if 0
char ip[16];
char buf[32];
printf("%d (expect 2)\n",scan_ip6("::",ip));
printf("%d (expect 3)\n",scan_ip6("::1",ip));
printf("%d (expect 16)\n",scan_ip6("fec0:0:0:ffff::1/0",ip));
printf("%.*s\n",fmt_ip6(buf,ip),buf);
#endif
#if 0
static stralloc s,t;
stralloc_copys(&s,"fnord");
stralloc_copys(&t,"abc"); printf("%d\n",stralloc_diff(&s,&t));
stralloc_copys(&t,"fnor"); printf("%d\n",stralloc_diff(&s,&t));
stralloc_copys(&t,"fnord"); printf("%d\n",stralloc_diff(&s,&t));
stralloc_copys(&t,"fnordh"); printf("%d\n",stralloc_diff(&s,&t));
stralloc_copys(&t,"hausen"); printf("%d\n",stralloc_diff(&s,&t));
#endif
#if 0
static stralloc s;
stralloc_copys(&s,"fnord");
printf("%d\n",stralloc_diffs(&s,"abc"));
printf("%d\n",stralloc_diffs(&s,"fnor"));
printf("%d\n",stralloc_diffs(&s,"fnord"));
printf("%d\n",stralloc_diffs(&s,"fnordh"));
printf("%d\n",stralloc_diffs(&s,"hausen"));
#endif
#if 0
printf("%d\n",case_starts("fnordhausen","FnOrD"));
printf("%d\n",case_starts("fnordhausen","blah"));
#endif
#if 0
char buf[]="FnOrD";
case_lowers(buf);
puts(buf);
#endif
#if 0
char buf[100]="foo bar baz";
printf("%d (expect 7)\n",byte_rchr(buf,11,' '));
#endif
#if 0
unsigned long size;
char* buf=mmap_read(argv[1],&size);
if (buf) {
unsigned int x=fmt_yenc(0,buf,size);
unsigned int y;
char* tmp=malloc(x+1);
y=fmt_yenc(tmp,buf,size);
write(1,tmp,x);
}
#endif
#if 0
char buf[100];
char buf2[100];
unsigned int len,len2;
buf[fmt_yenc(buf,"http://localhost/~fefe",22)]=0;
buffer_puts(buffer_1,buf);
buffer_putsflush(buffer_1,"\n");
if ((buf[len2=scan_yenc(buf,buf2,&len)])!='\n') {
buffer_putsflush(buffer_2,"parse error!\n");
return 1;
}
buffer_put(buffer_1,buf2,len2);
buffer_putsflush(buffer_1,"\n");
return 0;
#endif
#if 0
char buf[100];
char buf2[100];
unsigned int len,len2;
buf[fmt_base64(buf,"foo:bar",7)]=0;
buffer_puts(buffer_1,buf);
buffer_putsflush(buffer_1,"\n");
if ((buf[len2=scan_base64(buf,buf2,&len)])!=0) {
buffer_putsflush(buffer_2,"parse error!\n");
return 1;
}
buffer_put(buffer_1,buf2,len2);
buffer_putsflush(buffer_1,"\n");
return 0;
#endif
#if 0
unsigned long size;
char* buf=mmap_read(argv[1],&size);
if (buf) {
unsigned int x=fmt_uuencoded(0,buf,size);
unsigned int y;
char* tmp=malloc(x+1);
y=fmt_uuencoded(tmp,buf,size);
write(1,tmp,x);
}
#endif
#if 0
char buf[]="00000000000000000000000000000001";
char ip[16];
if (scan_ip6_flat(buf,ip) != str_len(buf))
buffer_putsflush(buffer_2,"parse error!\n");
#endif
#if 0
int fd=open_read("t.c");
buffer b;
char buf[1024];
char line[20];
int i;
buffer_init(&b,read,fd,buf,1024);
i=buffer_getline(&b,line,19);
buffer_puts(buffer_1,"getline returned ");
buffer_putulong(buffer_1,i);
buffer_puts(buffer_1,"\n");
buffer_puts(buffer_1,line);
buffer_flush(buffer_1);
#endif
#if 0
buffer_putulong(buffer_1,23);
// buffer_putspace(buffer_1);
buffer_putsflush(buffer_1,"\n");
// buffer_flush(buffer_1);
#endif
#if 0
long a,b,c;
char buf[4096];
char buf2[4096];
memcpy(buf,buf2,4096);
byte_copy(buf,4096,buf2);
rdtscl(a);
memcpy(buf,buf2,4096);
rdtscl(b);
byte_copy(buf,4096,buf2);
rdtscl(c);
printf("memcpy: %d - byte_copy: %d\n",b-a,c-b);
#endif
#if 0
char ip[16];
int i;
if ((i=scan_ip6(argv[1],ip))) {
char buf[128];
buf[fmt_ip6(buf,ip)]=0;
puts(buf);
}
#endif
#if 0
char buf[100];
strcpy(buf,"foobarbaz");
buf[fmt_fill(buf,3,5,100)]=0;
printf("\"%s\"\n",buf);
#endif
#if 0
unsigned long len;
char *c=mmap_read("/etc/passwd",&len);
printf("got map %p of len %lu\n",c,len);
#endif
#if 0
char c;
printf("%d\n",buffer_getc(buffer_0,&c));
printf("%c\n",c);
#endif
#if 0
char buf[100]="01234567890123456789012345678901234567890123456789";
long a,b,c;
#endif
#if 0
buf[ip4_fmt(buf,ip4loopback)]=0;
buffer_puts(buffer_1small,buf);
buffer_flush(buffer_1small);
#endif
#if 0
buf[0]=0;
buf[fmt_8long(buf,0)]=0;
puts(buf);
rdtscl(a);
c=str_len(buf);
rdtscl(b);
/*byte_zero_djb(buf,j); */
// printf("\n%lu %d\n",b-a,c);
#endif
#if 0
buffer_puts(buffer_1small,"hello, world\n");
buffer_flush(buffer_1small);
#endif
#if 0
int s=socket_tcp4();
char ip[4]={127,0,0,1};
int t=socket_connect4(s,ip,80);
#endif
#if 0
char buf[100]="foo bar baz fnord ";
char buf2[100]="foo braz fnord";
long a,b,c;
long i=0,j=0,k=0;
double d;
uint32 l,m,n;
stralloc sa={0};
stralloc_copys(&sa,"fnord");
stralloc_catlong0(&sa,-23,5);
stralloc_append(&sa,"\n");
printf("%d %d\n",str_equal("fnord","fnord1"),str_equal("fnord1","fnord"));
write(1,sa.s,sa.len);
printf("%d %d\n",stralloc_starts(&sa,"fnord"),stralloc_starts(&sa,"fnord\na"));
l=0xdeadbeef;
uint32_pack_big((char*)&m,l);
uint32_unpack_big((char*)&m,&n);
printf("%x %x %x\n",l,m,n);
rdtscl(a);
/* i=scan_double("3.1415",&d); */
rdtscl(b);
/*byte_zero_djb(buf,j); */
rdtscl(c);
printf("%lu %lu\n",b-a,c-b);
#endif
#if 0
size_t size;
char* buf=mmap_read(argv[1],&size);
if (buf) {
unsigned int x=fmt_urlencoded2(0,buf,size,"x");
unsigned int y;
char* tmp=malloc(x+1);
y=fmt_urlencoded2(tmp,buf,size,"x");
write(1,tmp,x);
}
#endif
#if 0
printf("%d %d\n",strcmp("foo","bar"),str_diff("foo","bar"));
printf("%d %d\n",strcmp("foo","üar"),str_diff("foo","üar"));
#endif
#if 0
{
int16 a;
int32 b;
int64 c;
assert(imult16(4,10000,&a)==0);
assert(imult16(-4,10000,&a)==0);
assert(imult16(5,10,&a)==1 && a==50);
assert(imult16(-3,10000,&a)==1 && a==-30000);
assert(imult32(0x40000000,2,&b)==0);
assert(imult32(0x3fffffff,2,&b)==1 && b==0x7ffffffe);
assert(imult64(0x4000000000000000ll,2,&c)==0);
assert(imult64(0x3fffffffffffffffll,2,&c)==1 && c==0x7ffffffffffffffell);
}
#endif
#if 0
stralloc a;
printf("%d\n",stralloc_copym(&a,"fnord",", ","foo"));
#endif
return 0;
}
int s6_fdholder_setdump (s6_fdholder_t *a, s6_fdholder_fd_t const *list, unsigned int ntot, tain_t const *deadline, tain_t *stamp)
{
uint32 trips ;
if (!ntot) return 1 ;
unsigned int i = 0 ;
for (; i < ntot ; i++)
{
unsigned int zpos = byte_chr(list[i].id, S6_FDHOLDER_ID_SIZE + 1, 0) ;
if (!zpos || zpos >= S6_FDHOLDER_ID_SIZE + 1) return (errno = EINVAL, 0) ;
}
{
char pack[5] = "!" ;
unixmessage_t m = { .s = pack, .len = 5, .fds = 0, .nfds = 0 } ;
uint32_pack_big(pack+1, ntot) ;
if (!unixmessage_put(&a->connection.out, &m)) return 0 ;
if (!unixmessage_sender_timed_flush(&a->connection.out, deadline, stamp)) return 0 ;
if (sanitize_read(unixmessage_timed_receive(&a->connection.in, &m, deadline, stamp)) < 0) return 0 ;
if (!m.len || m.nfds) { unixmessage_drop(&m) ; return (errno = EPROTO, 0) ; }
if (m.s[0]) return (errno = m.s[0], 0) ;
if (m.len != 5) return (errno = EPROTO, 0) ;
uint32_unpack_big(m.s + 1, &trips) ;
if (trips != 1 + (ntot-1) / UNIXMESSAGE_MAXFDS) return (errno = EPROTO, 0) ;
}
for (i = 0 ; i < trips ; i++, ntot -= UNIXMESSAGE_MAXFDS)
{
{
unsigned int n = ntot > UNIXMESSAGE_MAXFDS ? UNIXMESSAGE_MAXFDS : ntot ;
unsigned int j = 0 ;
siovec_t v[1 + (n<<1)] ;
int fds[n] ;
unixmessage_v_t m = { .v = v, .vlen = 1 + (n<<1), .fds = fds, .nfds = n } ;
char pack[n * (TAIN_PACK+1)] ;
v[0].s = "." ; v[0].len = 1 ;
for (; j < n ; j++, list++, ntot--)
{
unsigned int len = str_len(list->id) ;
v[1 + (j<<1)].s = pack + j * (TAIN_PACK+1) ;
v[1 + (j<<1)].len = TAIN_PACK + 1 ;
tain_pack(pack + j * (TAIN_PACK+1), &list->limit) ;
pack[j * (TAIN_PACK+1) + TAIN_PACK] = (unsigned char)len ;
v[2 + (j<<1)].s = (char *)list->id ;
v[2 + (j<<1)].len = len + 1 ;
fds[j] = list->fd ;
}
if (!unixmessage_putv(&a->connection.out, &m)) return 0 ;
}
if (!unixmessage_sender_timed_flush(&a->connection.out, deadline, stamp)) return 0 ;
{
unixmessage_t m ;
if (sanitize_read(unixmessage_timed_receive(&a->connection.in, &m, deadline, stamp)) < 0) return 0 ;
if (m.len != 1 || m.nfds)
{
unixmessage_drop(&m) ;
return (errno = EPROTO, 0) ;
}
if (!error_isagain(m.s[0]) && i < trips-1)
return errno = m.s[0] ? m.s[0] : EPROTO, 0 ;
if (i == trips - 1 && m.s[0])
return errno = error_isagain(m.s[0]) ? EPROTO : m.s[0], 0 ;
}
}
return 1 ;
}
void tain_pack (char *s, struct taia const *t)
{
tai_pack(s, &t->sec) ;
uint32_pack_big(s+8, (uint32)t->nano) ;
}
int main(int argc, char **argv)
{
int fddata;
uint32 loc;
//int v;
char ipa[4];
//char ipb[4];
//ip4_cidr_t *subnets = (ip4_cidr_t *)NULL;
//unsigned int slen = 0;
uint32 ipinta = 0U;
//unsigned int ipintb = 0;
unsigned short mask;
uint32 uid = 0;
uint32 mid = 0;
int i;
int j;
int k;
char ch;
unsigned long ttl;
char ttd[8];
//uint32 loc;
unsigned char keyloc[15];
unsigned long u;
char ip[4];
char locp[4];
//char map[9];
char type[2];
char soa[20];
char buf[4];
//int namlen = 0;
umask(022);
fddata = STDIN_FILENO;
if (argc != 3) {
fddata = open_read("data");
if (fddata == -1) strerr_die2sys(111,FATAL,"unable to open data: ");
if ( (fdcdb = open_trunc("data.tmp")) == -1) die_datatmp();
} else {
fddata = STDIN_FILENO;
if ( (fdcdb = open_trunc(argv[1])) == -1) die_datatmp();
}
defaultsoa_init(fddata);
buffer_init(&b,buffer_unixread,fddata,bspace,sizeof bspace);
if (cdb_make_start(&cdb,fdcdb) == -1) die_datatmp();
while (match) {
++linenum;
if (getln(&b,&line,&match,'\n') == -1)
strerr_die2sys(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;
//write(1,line.s,line.len);
//write(1,"\n",1);
j = 1;
for (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]) {
#ifdef USE_LOCMAPS
case '%':
//locparse(&f[0],loc);
loc = gethash(f[0].s,f[0].len);
uint32_pack(locp,loc);
if (!stralloc_0(&f[1])) nomem();
if (!stralloc_0(&f[2])) nomem();
ip4_scan(f[1].s,ipa);
//ip4_scan(f[2].s,ipb);
scan_ushort(f[2].s,&mask);
ip4_num(&ipinta,ipa);
//ip4_num(&ipintb,ipb);
//if (ipintb<ipinta) nomem();
//if (ip4_deaggregate(ipinta,ipintb,&subnets,&slen) <= 0) nomem();
mid = gethash(f[3].s,f[3].len);
uid = gethash(f[4].s,f[4].len);
//for (v = 0; v < slen; v++) {
ipdb_key4build(keyloc,ipinta,(unsigned char )mask&0xff,mid,uid);
//ipdb_key_from_uint(keyloc,ipinta,(unsigned char )mask&0xff,0,mid,uid);
cdb_make_add(&cdb,keyloc,15,locp,4);
//}
uid = 0;
//alloc_free(subnets);
//subnets = NULL;
//slen = 0;
byte_zero(ipa,4);ipinta=0; //byte_zero(ipb,4);ipinta=0;ipintb=0;
break;
#endif
case 'Z':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) nomem();
if (!stralloc_0(&f[3])) nomem();
if (!scan_ulong(f[3].s,&u)) uint32_unpack_big(defaultsoa,&u);
uint32_pack_big(soa,u);
if (!stralloc_0(&f[4])) nomem();
if (!scan_ulong(f[4].s,&u)) uint32_unpack_big(defaultsoa + 4,&u);
uint32_pack_big(soa + 4,u);
if (!stralloc_0(&f[5])) nomem();
if (!scan_ulong(f[5].s,&u)) uint32_unpack_big(defaultsoa + 8,&u);
uint32_pack_big(soa + 8,u);
if (!stralloc_0(&f[6])) nomem();
if (!scan_ulong(f[6].s,&u)) uint32_unpack_big(defaultsoa + 12,&u);
uint32_pack_big(soa + 12,u);
if (!stralloc_0(&f[7])) nomem();
if (!scan_ulong(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);
loc = gethash(f[10].s,f[10].len);
mid = gethash(f[11].s,f[11].len);
uid = gethash(f[12].s,f[12].len);
rr_addloq(DNS_T_SOA,d1,loc,mid,uid);
rr_start(DNS_T_SOA,ttl,ttd,loc);
if (!dns_domain_fromdot(&d2,f[1].s,f[1].len)) nomem();
rr_addname(d2);
if (!dns_domain_fromdot(&d2,f[2].s,f[2].len)) nomem();
rr_addname(d2);
rr_add(soa,20);
rr_finish(d1);
break;
case '.': case '&':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) nomem();
if (!stralloc_0(&f[3])) nomem();
if (!scan_ulong(f[3].s,&ttl)) ttl = TTL_NS;
ttdparse(&f[4],ttd);
loc = gethash(f[5].s,f[5].len);
mid = gethash(f[6].s,f[6].len);
uid = gethash(f[7].s,f[7].len);
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)) nomem();
if (line.s[0] == '.') {
rr_start(DNS_T_SOA,ttl ? TTL_NEGATIVE : 0,ttd,loc);
rr_addloq(DNS_T_SOA,d1,loc,mid,uid);
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_addloq(DNS_T_NS,d2,loc,mid,uid);
rr_finish(d1);
if (ip4_scan(f[1].s,ip)) {
rr_start(DNS_T_A,ttl,ttd,loc);
rr_addloq(DNS_T_A,d2,loc,mid,uid);
rr_add(ip,4);
rr_finish(d2);
}
break;
case '+': case '=':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) nomem();
if (!stralloc_0(&f[2])) nomem();
if (!scan_ulong(f[2].s,&ttl)) ttl = TTL_POSITIVE;
ttdparse(&f[3],ttd);
loc = gethash(f[4].s,f[4].len);
mid = gethash(f[5].s,f[5].len);
uid = gethash(f[6].s,f[6].len);
if (!stralloc_0(&f[1])) nomem();
if (ip4_scan(f[1].s,ip)) {
rr_addloq(DNS_T_A,d1,loc,mid,uid);
rr_start(DNS_T_A,ttl,ttd,loc);
//dbger("+addloq:d=%s,loc=%u,uid=%u,mid=%u",d1,loc,uid,mid);
rr_add(ip,4);
rr_finish(d1);
if (line.s[0] == '=') {
rr_addloq(DNS_T_PTR,d1,loc,mid,uid); //??????????????????/ TODO: checkthis
dns_name4_domain(dptr,ip);
rr_start(DNS_T_PTR,ttl,ttd,loc);
rr_addname(d1);
rr_finish(dptr);
}
}
break;
case '@':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) nomem();
if (!stralloc_0(&f[4])) nomem();
if (!scan_ulong(f[4].s,&ttl)) ttl = TTL_POSITIVE;
ttdparse(&f[5],ttd);
loc = gethash(f[6].s,f[6].len);
mid = gethash(f[7].s,f[7].len);
uid = gethash(f[8].s,f[8].len);
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)) nomem();
if (!stralloc_0(&f[3])) nomem();
if (!scan_ulong(f[3].s,&u)) u = 0;
rr_addloq(DNS_T_MX,d1,loc,mid,uid);
rr_start(DNS_T_MX,ttl,ttd,loc);
uint16_pack_big(buf,u);
rr_add(buf,2);
rr_addname(d2);
rr_finish(d1);
if (ip4_scan(f[1].s,ip)) {
rr_addloq(DNS_T_A,d2,loc,mid,uid);
rr_start(DNS_T_A,ttl,ttd,loc);
rr_add(ip,4);
rr_finish(d2);
}
break;
case '^': case 'C':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) nomem();
if (!dns_domain_fromdot(&d2,f[1].s,f[1].len)) nomem();
if (!stralloc_0(&f[2])) nomem();
if (!scan_ulong(f[2].s,&ttl)) ttl = TTL_POSITIVE;
ttdparse(&f[3],ttd);
loc = gethash(f[4].s,f[4].len);
mid = gethash(f[5].s,f[5].len);
uid = gethash(f[6].s,f[6].len);
if (line.s[0] == 'C') {
//rr_addloq(DNS_T_CNAME,d1,loc,mid,uid);
//dbger("add loq for cname!");
rr_addloq(DNS_T_A,d1,loc,mid,uid);
rr_addloq(DNS_T_CNAME,d1,loc,mid,uid);
rr_start(DNS_T_CNAME,ttl,ttd,loc);
} else {
rr_addloq(DNS_T_PTR,d1,loc,mid,uid);
rr_start(DNS_T_PTR,ttl,ttd,loc);
}
rr_addname(d2);
rr_finish(d1);
break;
case '\'':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) nomem();
if (!stralloc_0(&f[2])) nomem();
if (!scan_ulong(f[2].s,&ttl)) ttl = TTL_POSITIVE;
ttdparse(&f[3],ttd);
loc = gethash(f[4].s,f[4].len);
mid = gethash(f[5].s,f[5].len);
uid = gethash(f[6].s,f[6].len);
rr_addloq(DNS_T_TXT,d1,loc,mid,uid);
rr_start(DNS_T_TXT,ttl,ttd,loc);
txtparse(&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 ':':
if (!dns_domain_fromdot(&d1,f[0].s,f[0].len)) nomem();
if (!stralloc_0(&f[3])) nomem();
if (!scan_ulong(f[3].s,&ttl)) ttl = TTL_POSITIVE;
ttdparse(&f[4],ttd);
loc = gethash(f[5].s,f[5].len);
mid = gethash(f[6].s,f[6].len);
uid = gethash(f[7].s,f[7].len);
if (!stralloc_0(&f[1])) nomem();
scan_ulong(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_addloq(type,d1,loc,mid,uid);
rr_start(type,ttl,ttd,loc);
rr_add(f[2].s,f[2].len);
rr_finish(d1);
break;
default:
dienow("error here: %s\n", line.s);
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 (argc == 3) {
if (rename(argv[1],argv[2]) == -1) strerr_die2sys(111,FATAL,"unable to move data.tmp to data.cdb: ");
} else {
if (rename("data.tmp","data.cdb") == -1) strerr_die2sys(111,FATAL,"unable to move data.tmp to data.cdb: ");
}
_exit(0);
}
int
main(int argc, char *argv[])
{
int fddata = 0;
unsigned int i = 0, j = 0, k = 0;
unsigned long u = 0;
unsigned long ttl = 0;
char ch = 0;
char *x = NULL;
char ttd[8], loc[2];
char ip[4], type[2];
char soa[20], buf[4];
prog = strdup ((x = strrchr (argv[0], '/')) != NULL ? x + 1 : argv[0]);
i = check_option (argc, argv);
argv += i;
argc -= i;
umask(022);
if ((fddata = open_read ("data")) == -1)
err (-1, "could not open file `data'");
defaultsoa_init (fddata);
buffer_init (&b, buffer_unixread, fddata, bspace, sizeof bspace);
if ((fdcdb = open_trunc ("data.tmp")) == -1)
err (-1, "could not create file `data.tmp'");
if (cdb_make_start (&cdb, fdcdb) == -1)
err (-1, "could not create file `data.tmp'");
while (match)
{
linenum++;
if (getln (&b, &line, &match, '\n') == -1)
err (-1, "could not read line: %ld", linenum);
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;
j = 1;
for (i = 0; i < NUMFIELDS; i++)
{
if (j >= line.len)
{
if (!stralloc_copys (&f[i], ""))
err (-1, "could not allocate enough memory");
}
else
{
k = byte_chr (line.s + j, line.len - j, ':');
if (!stralloc_copyb (&f[i], line.s + j, k))
err (-1, "could not allocate enough memory");
j += k + 1;
}
}
switch (line.s[0])
{
case '%':
locparse (&f[0], loc);
if (!stralloc_copyb (&key, "\0%", 2))
err (-1, "could not allocate enough memory");
if (!stralloc_0 (&f[1]))
err (-1, "could not allocate enough memory");
ipprefix_cat (&key, f[1].s);
if (cdb_make_add(&cdb,key.s,key.len,loc,2) == -1)
err (-1, "could not create file `data.tmp'");
break;
case 'Z':
if (!dns_domain_fromdot (&d1, f[0].s, f[0].len))
err (-1, "could not allocate enough memory");
if (!stralloc_0 (&f[3]))
err (-1, "could not allocate enough memory");
if (!scan_ulong (f[3].s, &u))
uint32_unpack_big (defaultsoa, (uint32 *)&u);
uint32_pack_big (soa, u);
if (!stralloc_0 (&f[4]))
err (-1, "could not allocate enough memory");
if (!scan_ulong (f[4].s, &u))
uint32_unpack_big (defaultsoa + 4, (uint32 *)&u);
uint32_pack_big (soa + 4, u);
if (!stralloc_0 (&f[5]))
err (-1, "could not allocate enough memory");
if (!scan_ulong (f[5].s, &u))
uint32_unpack_big (defaultsoa + 8, (uint32 *)&u);
uint32_pack_big (soa + 8, u);
if (!stralloc_0 (&f[6]))
err (-1, "could not allocate enough memory");
if (!scan_ulong (f[6].s, &u))
uint32_unpack_big (defaultsoa + 12, (uint32 *)&u);
uint32_pack_big (soa + 12, u);
if (!stralloc_0 (&f[7]))
err (-1, "could not allocate enough memory");
if (!scan_ulong (f[7].s, &u))
uint32_unpack_big (defaultsoa + 16, (uint32 *)&u);
uint32_pack_big (soa + 16, u);
if (!stralloc_0 (&f[8]))
err (-1, "could not allocate enough memory");
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))
err (-1, "could not allocate enough memory");
rr_addname (d2);
if (!dns_domain_fromdot (&d2, f[2].s, f[2].len))
err (-1, "could not allocate enough memory");
rr_addname (d2);
rr_add (soa, 20);
rr_finish (d1);
break;
case '.':
case '&':
if (!dns_domain_fromdot (&d1, f[0].s, f[0].len))
err (-1, "could not allocate enough memory");
if (!stralloc_0 (&f[3]))
err (-1, "could not allocate enough memory");
if (!scan_ulong (f[3].s, &ttl))
ttl = TTL_NS;
ttdparse (&f[4], ttd);
locparse (&f[5], loc);
if (!stralloc_0 (&f[1]))
err (-1, "could not allocate enough memory");
if (byte_chr (f[2].s, f[2].len, '.') >= f[2].len)
{
if (!stralloc_cats (&f[2], ".ns."))
err (-1, "could not allocate enough memory");
if (!stralloc_catb (&f[2], f[0].s, f[0].len))
err (-1, "could not allocate enough memory");
}
if (!dns_domain_fromdot (&d2, f[2].s, f[2].len))
err (-1, "could not allocate enough memory");
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 (ip4_scan (f[1].s, ip))
{
rr_start (DNS_T_A, ttl, ttd, loc);
rr_add (ip, 4);
rr_finish (d2);
}
break;
case '+':
case '=':
if (!dns_domain_fromdot (&d1, f[0].s, f[0].len))
err (-1, "could not allocate enough memory");
if (!stralloc_0 (&f[2]))
err (-1, "could not allocate enough memory");
if (!scan_ulong (f[2].s, &ttl))
ttl = TTL_POSITIVE;
ttdparse (&f[3], ttd);
locparse (&f[4], loc);
if (!stralloc_0 (&f[1]))
err (-1, "could not allocate enough memory");
if (ip4_scan (f[1].s, ip))
{
rr_start (DNS_T_A, ttl, ttd, loc);
rr_add (ip, 4);
rr_finish (d1);
if (line.s[0] == '=')
{
dns_name4_domain (dptr,ip);
rr_start (DNS_T_PTR, ttl, ttd, loc);
rr_addname (d1);
rr_finish (dptr);
}
}
break;
case '@':
if (!dns_domain_fromdot (&d1, f[0].s, f[0].len))
err (-1, "could not allocate enough memory");
if (!stralloc_0 (&f[4]))
err (-1, "could not allocate enough memory");
if (!scan_ulong (f[4].s, &ttl))
ttl = TTL_POSITIVE;
ttdparse (&f[5], ttd);
locparse (&f[6], loc);
if (!stralloc_0 (&f[1]))
err (-1, "could not allocate enough memory");
if (byte_chr (f[2].s, f[2].len, '.') >= f[2].len)
{
if (!stralloc_cats (&f[2], ".mx."))
err (-1, "could not allocate enough memory");
if (!stralloc_catb (&f[2], f[0].s, f[0].len))
err (-1, "could not allocate enough memory");
}
if (!dns_domain_fromdot (&d2, f[2].s, f[2].len))
err (-1, "could not allocate enough memory");
if (!stralloc_0 (&f[3]))
err (-1, "could not allocate enough memory");
if (!scan_ulong (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 (ip4_scan (f[1].s, ip))
{
rr_start (DNS_T_A, ttl, ttd, loc);
rr_add (ip, 4);
rr_finish (d2);
}
break;
case '^':
case 'C':
if (!dns_domain_fromdot (&d1, f[0].s, f[0].len))
err (-1, "could not allocate enough memory");
if (!dns_domain_fromdot (&d2, f[1].s, f[1].len))
err (-1, "could not allocate enough memory");
if (!stralloc_0 (&f[2]))
err (-1, "could not allocate enough memory");
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 '\'':
if (!dns_domain_fromdot (&d1, f[0].s, f[0].len))
err (-1, "could not allocate enough memory");
if (!stralloc_0 (&f[2]))
err (-1, "could not allocate enough memory");
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);
txtparse (&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 ':':
if (!dns_domain_fromdot (&d1, f[0].s, f[0].len))
err (-1, "could not allocate enough memory");
if (!stralloc_0 (&f[3]))
err (-1, "could not allocate enough memory");
if (!scan_ulong (f[3].s, &ttl))
ttl = TTL_POSITIVE;
ttdparse (&f[4], ttd);
locparse (&f[5], loc);
if (!stralloc_0 (&f[1]))
err (-1, "could not allocate enough memory");
scan_ulong (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)
err (-1, "could not create file `data.tmp'");
if (fsync (fdcdb) == -1)
err (-1, "could not create file `data.tmp'");
if (close (fdcdb) == -1)
err (-1, "could not create file `data.tmp'"); /* NFS stupidity */
if (rename ("data.tmp", "data.cdb") == -1)
err (-1, "could not move `data.tmp' to `data.cdb'");
return 0;
}
void bu_pack_big (char *s, uint32 const *a, unsigned int n)
{
register unsigned int i = 0 ;
for (; i < n ; i++) uint32_pack_big(s + (i<<2), a[n-1-i]) ;
}
int
build (stralloc *sa, char *q, int flagsoa, char id[2])
{
char ttl[4];
char ttd[8];
char type[2];
char misc[20];
char recordloc[2];
struct tai cutoff;
unsigned int rdatapos = 0;
dpos = 0;
copy (type, 2);
if (flagsoa)
if (byte_diff (type, 2, DNS_T_SOA))
return 0;
if (!flagsoa)
if (byte_equal (type, 2, DNS_T_SOA))
return 0;
if (!stralloc_copyb (sa, id, 2))
err (-1, "could not allocate enough memory");
if (!stralloc_catb (sa, "\204\000\0\0\0\1\0\0\0\0", 10))
err (-1, "could not allocate enough memory");
copy (misc, 1);
if ((misc[0] == '=' + 1) || (misc[0] == '*' + 1))
{
--misc[0];
copy (recordloc, 2);
if (byte_diff (recordloc, 2, clientloc))
return 0;
}
if (misc[0] == '*')
{
if (flagsoa)
return 0;
if (!stralloc_catb (sa, "\1*", 2))
err (-1, "could not allocate enough memory");
}
if (!stralloc_catb (sa, q, dns_domain_length (q)))
err (-1, "could not allocate enough memory");
if (!stralloc_catb (sa, type, 2))
err (-1, "could not allocate enough memory");
copy (ttl, 4);
copy (ttd, 8);
if (byte_diff (ttd, 8, "\0\0\0\0\0\0\0\0"))
{
tai_unpack (ttd, &cutoff);
if (byte_equal (ttl, 4, "\0\0\0\0"))
{
if (tai_less (&cutoff, &now))
return 0;
uint32_pack_big (ttl, 2);
}
else
if (!tai_less (&cutoff, &now))
return 0;
}
if (!stralloc_catb (sa, DNS_C_IN, 2))
err (-1, "could not allocate enough memory");
if (!stralloc_catb (sa, ttl, 4))
err (-1, "could not allocate enough memory");
if (!stralloc_catb(sa,"\0\0",2))
err (-1, "could not allocate enough memory");
rdatapos = sa->len;
if (byte_equal (type, 2, DNS_T_SOA))
{
doname (sa);
doname (sa);
copy (misc, 20);
if (!stralloc_catb (sa, misc, 20))
err (-1, "could not allocate enough memory");
}
else if (byte_equal (type, 2, DNS_T_NS)
|| byte_equal (type, 2, DNS_T_PTR)
|| byte_equal (type, 2, DNS_T_CNAME))
{
doname (sa);
}
else if (byte_equal (type, 2, DNS_T_MX))
{
copy (misc, 2);
if (!stralloc_catb (sa, misc, 2))
err (-1, "could not allocate enough memory");
doname (sa);
}
else
{
if (!stralloc_catb (sa, data + dpos, dlen - dpos))
err (-1, "could not allocate enough memory");
}
if (sa->len > 65535)
errx (-1, "could not read from file `data.cdb': format error");
uint16_pack_big (sa->s + rdatapos - 2, sa->len - rdatapos);
return 1;
}
