static int parse_json(struct cdb_make *cdbm, char *filename)
{
FILE *fp = fopen(filename, "r");
fseek(fp, 0, SEEK_END);
uint32_t size = ftell(fp);
char *str = malloc(size + 1);
fseek(fp, 0, SEEK_SET);
fread(str, 1, size, fp);
str[size] = 0;
int i;
int r;
jsmn_parser p;
jsmntok_t t[128];
jsmn_init(&p);
r = jsmn_parse(&p, str, size, t, sizeof(t)/sizeof(t[0]));
if (r < 0) {
printf("Failed to parse JSON: %d\n", r);
return 1;
}
for(i = 1; i < r; i+=2) {
if(!strncmp((str + t[i].start), "0x", 2)) {
char filename[256];
memset(filename, 0, 256);
char *tmp;
strncpy(filename,
&str[t[i + 1].start],
t[i + 1].end - t[i + 1].start);
FILE *data = fopen(filename, "rb+");
if(data == NULL) {
printf("Couldn't open file %s\n", filename);
}
fseek(data, 0, SEEK_END);
size_t size = ftell(data);
tmp = malloc(size * sizeof(char));
fseek(data, 0, SEEK_SET);
fread(tmp, sizeof(char), size, data);
cdb_make_add(cdbm,
&str[t[i].start + 2],
(t[i].end - t[i].start) - 2,
tmp, size);
fclose(data);
free(tmp);
} else {
cdb_make_add(cdbm,
&str[t[i].start],
t[i].end - t[i].start,
&str[t[i + 1].start],
t[i + 1].end - t[i + 1].start);
}
}
free(str);
fclose(fp);
return 0;
}
static int cdbb_handle_ops(struct cdbb *a)
{
int i;
int num;
struct op *t;
num = 0;
for (i = 0; i < a->tnum; i++) {
t = &a->ops[i];
if (t->kf) { /* trigger was fired */
switch (t->t) {
case O_ADD:
case O_ADD_MOD:
/* fall-trought */
case O_MOD:
cdb_make_add(a->w,
(unsigned char *)t->n.p,
array_bytes(&t->n),
(unsigned char *)t->v.p,
array_bytes(&t->v));
num++;
break;
case O_DEL:
/* do nothing, fall-trough */
case O_NONE:
default:
break;
}
} else {
switch (t->t) {
case O_ADD_MOD:
/* key not found so add, fall-trough */
case O_ADD:
cdb_make_add(a->w,
(unsigned char *)t->n.p,
array_bytes(&t->n),
(unsigned char *)t->v.p,
array_bytes(&t->v));
num++;
break;
case O_DEL:
/* key not found error */
break;
case O_MOD:
/* key not found error */
break;
case O_NONE:
/* fall-trought */
default:
break;
}
}
}
return num;
}
static int __cdbb_copy(struct cdbb *a, size_t ks, size_t vs)
{
unsigned char k[ks];
unsigned char v[vs];
uint32 kp, dp;
int found, filter;
found = 0;
kp = cdb_keypos(a->r);
cdb_read(a->r, k, ks, kp);
dp = cdb_datapos(a->r);
/* in some cases we do not need to read e.g. when deleting
* or filtering but for simplicity we just do it anyways */
cdb_read(a->r, v, vs, dp);
found = cdbb_check_ops(a, k, ks, v, vs);
filter = cdbb_check_filter(a, k, ks, v, vs);
/* if an operation applied or a filter, don't copy */
if (!found && !filter && kp) {
cdb_make_add(a->w, k, ks, v, vs);
return 1;
}
return 0;
}
SDB_API int sdb_disk_insert(Sdb* s, const char *key, const char *val) {
struct cdb_make *c = &s->m;
if (!key || !val) {
return 0;
}
//if (!*val) return 0; //undefine variable if no value
return cdb_make_add (c, key, strlen (key), val, strlen (val));
}
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();
}
int vmailmgr_autoconvert(void)
{
int writefd = -1;
ibuf reader;
struct cdb_make writer;
int error = 0;
int readall = 0;
int writerr = 0;
if ((writefd = path_mktemp(pwfile, &tmppwfile)) != -1) {
if (cdb_make_start(&writer, writefd) != 0)
error = CVME_IO | CVME_FATAL;
else {
if (ibuf_open(&reader, pwfile, 0)) {
uint32 end;
struct stat st;
if (fstat(reader.io.fd, &st) == 0
&& fchmod(writefd, st.st_mode) == 0
&& fchown(writefd, st.st_uid, st.st_gid) == 0
&& read_start(&reader, &end)) {
while (ibuf_tell(&reader) < end) {
if (!read_cdb_pair(&reader, &key, &data))
break;
if (str_diff(&key, &virtuser) == 0)
if (!convert_data()) {
writerr = 1;
break;
}
if (cdb_make_add(&writer, key.s, key.len, data.s, data.len) != 0) {
writerr = 1;
break;
}
}
readall = ibuf_tell(&reader) == end;
}
ibuf_close(&reader);
}
if (cdb_make_finish(&writer) != 0)
error |= CVME_FATAL;
else
if (readall && !writerr)
rename(tmppwfile.s, pwfile);
}
close(writefd);
unlink(tmppwfile.s);
}
return error;
}
static PyObject *
CdbMake_addmany(cdbmakeobject *self, PyObject *args) {
PyObject *list;
if (!PyArg_ParseTuple(args,"O!:addmany",&PyList_Type, &list))
return NULL;
if (self->finished) {
CDBMAKEfinished;
return NULL;
}
Py_ssize_t size = PyList_Size(list);
Py_ssize_t i;
for (i=0; i<size; i++)
{
PyObject *tuple = PyList_GetItem(list, i);
PyObject *key_item;
PyObject *data_item;
if (!PyTuple_Check(tuple)) {
PyErr_SetString(PyExc_TypeError, "list of tuples expected");
return NULL;
}
if (!(key_item = PyTuple_GetItem(tuple,0)))
return NULL;
if (!(data_item = PyTuple_GetItem(tuple,1)))
return NULL;
char *key, *dat;
Py_ssize_t klen, dlen;
if (PyString_AsStringAndSize(key_item, &key, &klen) < 0)
return NULL;
if (PyString_AsStringAndSize(data_item, &dat, &dlen) < 0)
return NULL;
if (cdb_make_add(&self->cm, key, klen, dat, dlen) == -1)
return CDBMAKEerr;
}
return Py_BuildValue("");
}
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)))
err (-1, "could not allocate enough memory");
case_lowerb (key.s, key.len);
if (cdb_make_add (&cdb, key.s, key.len, result.s, result.len) == -1)
errx (-1, "could not create file `data.tmp'");
}
int main(int argc,char **argv)
{
int fd;
unsigned long loop;
if (!*argv) _exit(0);
if (!*++argv) _exit(0);
scan_ulong(*argv,&loop);
if (cdb_make_start(&c,1) == -1) die_write();
while (loop) {
uint32_pack(key,--loop);
if (cdb_make_add(&c,key,4,data,sizeof data) == -1) die_write();
}
if (cdb_make_finish(&c) == -1) die_write();
_exit(0);
}
int main()
{
umask(033);
if (chdir(auto_qmail) == -1)
strerr_die4sys(111,FATAL,"unable to chdir to ",auto_qmail,": ");
fd = open_read("control/morercpthosts");
if (fd == -1) die_read();
substdio_fdbuf(&ssin,subread,fd,inbuf,sizeof inbuf);
fdtemp = open_trunc("control/morercpthosts.tmp");
if (fdtemp == -1) die_write();
if (cdb_make_start(&cdbm,fdtemp) == -1) die_write();
for (;;) {
if (getln(&ssin,&line,&match,'\n') != 0) die_read();
case_lowerb(line.s,line.len);
while (line.len) {
if (line.s[line.len - 1] == ' ') { --line.len; continue; }
if (line.s[line.len - 1] == '\n') { --line.len; continue; }
if (line.s[line.len - 1] == '\t') { --line.len; continue; }
if (line.s[0] != '#')
if (cdb_make_add(&cdbm,line.s,line.len,"",0) == -1)
die_write();
break;
}
if (!match) break;
}
if (cdb_make_finish(&cdbm) == -1) die_write();
if (fsync(fdtemp) == -1) die_write();
if (close(fdtemp) == -1) die_write(); /* NFS stupidity */
if (rename("control/morercpthosts.tmp","control/morercpthosts.cdb") == -1)
strerr_die2sys(111,FATAL,"unable to move control/morercpthosts.tmp to control/morercpthosts.cdb");
return 0;
}
static ERL_NIF_TERM cdbwriter_add( ErlNifEnv* env, int argc,
const ERL_NIF_TERM argv[])
{
cdbwriter_handle* handle = get_handle(env, argv[0]);
if(!handle || (handle->fd < 0))
return enif_make_atom(env, "invalid_handle");
int key_len;
char * key = get_str(env, argv[1], &key_len);
int val_len;
if(!key)
return enif_make_atom(env, "invalid_key");
char * val = get_str(env, argv[2], &val_len);
if(!val)
return enif_make_atom(env, "invalid_val");
int fres = cdb_make_add(&(handle->cdbm), key, key_len, val, val_len);
free(key);
free(val);
if(fres!=0)
return enif_make_atom(env, "failed_add");
return enif_make_atom(env, "ok");
}
static void rr_addloq(const char type[2],const char *d, uint32 loc, uint32 mid,uint32 uid)
{
#ifdef USE_LOCMAPS
char map[8];
dbger("-addloq: %s,%u (%u),%u,%u",d,loc,NOMATCH_HASH,uid,mid);
if (loc != NOMATCH_HASH) return;
dbger("NOMATCH hash found, adding loq records for %s",d);
/*
if (byte_equal(type,2,DNS_T_A)) {
dbger("lookup LOQ for A record %s",fqdn_read(out,d));
}
dbger("+addloq %s,%u,%u",fqdn_read(out,d),mid,uid);
*/
if (!stralloc_copyb(&key,"\0?",2)) nomem();
if (!stralloc_catb(&key,type,2)) nomem();
if (!stralloc_catb(&key,d,dns_domain_length(d))) nomem();
uint32_pack(map,mid);
uint32_pack(map+4,uid);
cdb_make_add(&cdb,key.s,key.len,map,8);
#endif
return;
}
int main(int argc, char **argv)
{
umask(033);
if (argc != 3)
strerr_die1sys(111,"qmail-cdb: usage: qmail-cdb rules.cdb rules.tmp");
substdio_fdbuf(&ssin,subread,0,inbuf,sizeof inbuf);
fdtemp = open_trunc(argv[2]);
if (fdtemp == -1) die_write(argv[2]);
if (cdb_make_start(&cdbm,fdtemp) == -1) die_write(argv[2]);
for (;;) {
if (getln(&ssin,&line,&match,'\n') != 0) die_read();
case_lowerb(line.s,line.len);
while (line.len) {
if (line.s[line.len - 1] == ' ') { --line.len; continue; }
if (line.s[line.len - 1] == '\n') { --line.len; continue; }
if (line.s[line.len - 1] == '\t') { --line.len; continue; }
if (line.s[0] != '#')
if (cdb_make_add(&cdbm,line.s,line.len,"",0) == -1)
die_write(argv[2]);
break;
}
if (!match) break;
}
if (cdb_make_finish(&cdbm) == -1) die_write(argv[2]);
if (fsync(fdtemp) == -1) die_write(argv[2]);
if (close(fdtemp) == -1) die_write(argv[2]); /* NFS stupidity */
if (rename(argv[2],argv[1]) == -1)
strerr_die5sys(111, FATAL, "unable to move ", argv[2], " to ", argv[1]);
return 0;
}
static int cdb_write_attrs(const char *filename, const attrs_type&attributes)
{
FILE *cdbfile;
t_hlist *curr;
t_attr *attr;
struct cdb_make cdbm;
if ((cdbfile = fopen(filename, "w+b")) == NULL) {
eventlog(eventlog_level_error, __FUNCTION__, "unable to open file \"%s\" for writing ",filename);
return -1;
}
cdb_make_start(&cdbm, cdbfile);
BEGIN(_attr,attributes)
{
attr=*_attr;
// hlist_for_each(curr,attributes) {
// attr = hlist_entry(curr, t_attr, link);
if (attr_get_key(attr) && attr_get_val(attr)) {
if (strncmp("BNET\\CharacterDefault\\", attr_get_key(attr), 20) == 0) {
eventlog(eventlog_level_debug, __FUNCTION__, "skipping attribute key=\"%s\"",attr_get_key(attr));
} else {
eventlog(eventlog_level_debug, __FUNCTION__, "saving attribute key=\"%s\" val=\"%s\"",attr_get_key(attr),attr_get_val(attr));
if (cdb_make_add(&cdbm, attr_get_key(attr), strlen(attr_get_key(attr)), attr_get_val(attr), strlen(attr_get_val(attr)))<0)
{
eventlog(eventlog_level_error, __FUNCTION__, "got error on cdb_make_add ('%s' = '%s')", attr_get_key(attr), attr_get_val(attr));
cdb_make_finish(&cdbm); /* try to bail out nicely */
fclose(cdbfile);
return -1;
}
}
} else eventlog(eventlog_level_error, __FUNCTION__,"could not save attribute key=\"%s\"",attr_get_key(attr));
attr_clear_dirty(attr);
}
static int dict_cdbm_update(DICT *dict, const char *name, const char *value)
{
DICT_CDBM *dict_cdbm = (DICT_CDBM *) dict;
unsigned ksize, vsize;
int r;
dict->error = 0;
/*
* Optionally fold the key.
*/
if (dict->flags & DICT_FLAG_FOLD_FIX) {
if (dict->fold_buf == 0)
dict->fold_buf = vstring_alloc(10);
vstring_strcpy(dict->fold_buf, name);
name = lowercase(vstring_str(dict->fold_buf));
}
ksize = strlen(name);
vsize = strlen(value);
/*
* Optionally append a null byte to key and value.
*/
if (dict->flags & DICT_FLAG_TRY1NULL) {
ksize++;
vsize++;
}
/*
* Do the add operation. No locking is done.
*/
#ifdef TINYCDB_VERSION
#ifndef CDB_PUT_ADD
#error please upgrate tinycdb to at least 0.5 version
#endif
if (dict->flags & DICT_FLAG_DUP_IGNORE)
r = CDB_PUT_ADD;
else if (dict->flags & DICT_FLAG_DUP_REPLACE)
r = CDB_PUT_REPLACE;
else
r = CDB_PUT_INSERT;
r = cdb_make_put(&dict_cdbm->cdbm, name, ksize, value, vsize, r);
if (r < 0)
msg_fatal("error writing %s: %m", dict_cdbm->tmp_path);
else if (r > 0) {
if (dict->flags & (DICT_FLAG_DUP_IGNORE | DICT_FLAG_DUP_REPLACE))
/* void */ ;
else if (dict->flags & DICT_FLAG_DUP_WARN)
msg_warn("%s: duplicate entry: \"%s\"",
dict_cdbm->dict.name, name);
else
msg_fatal("%s: duplicate entry: \"%s\"",
dict_cdbm->dict.name, name);
}
return (r);
#else
if (cdb_make_add(&dict_cdbm->cdbm, name, ksize, value, vsize) < 0)
msg_fatal("error writing %s: %m", dict_cdbm->tmp_path);
return (0);
#endif
}
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) {
int c, fd, fdtemp;
size_t klen, len = 0;
char *dbfile, *dbnewfile, *dbtemp, *dirname = NULL;
char *key;
unsigned cpos;
struct cdb cdbm;
struct cdb_make cdbn;
struct record rec;
progname = basename(argv[0]);
struct option long_options[] = {
{ "help", no_argument, 0, 'h' },
{ "dbpath", required_argument, 0, 'D' },
{ 0, 0, 0, 0 }
};
if (argc == 1)
print_help(EXIT_SUCCESS);
while ((c = getopt_long (argc, argv, "hD:", long_options, NULL)) != -1) {
switch (c) {
case 'D':
db_path = optarg;
break;
default:
case 'h':
print_help(EXIT_SUCCESS);
break;
}
}
if (db_path == NULL)
db_path = def_db_path;
if (optind == argc)
print_help(EXIT_FAILURE);
dbfile = argv[optind];
// Open old database
if ((fd = open(dbfile, O_RDONLY | O_NOFOLLOW | O_NOCTTY)) == -1)
osec_fatal(EXIT_FAILURE, errno, "%s: open", dbfile);
if (compat_db_version(fd)) {
printf("%s: Database already in new format\n", dbfile);
return EXIT_SUCCESS;
}
if (cdb_init(&cdbm, fd) < 0)
osec_fatal(EXIT_FAILURE, errno, "cdb_init(cdbm)");
// Generate new state database
len = strlen(dbfile) + 11;
dbtemp = (char *) xmalloc(sizeof(char) * len);
sprintf(dbtemp, "%s.XXXXXXXXX", dbfile);
// Open new database
if ((fdtemp = mkstemp(dbtemp)) == -1)
osec_fatal(EXIT_FAILURE, errno, "%s: mkstemp", dbtemp);
if (cdb_make_start(&cdbn, fdtemp) < 0)
osec_fatal(EXIT_FAILURE, errno, "cdb_make_start");
// Allocate buffer for reading files.
read_bufsize = (size_t) (sysconf(_SC_PAGE_SIZE) - 1);
read_buf = xmalloc(read_bufsize);
/*
* Set default data buffer. This value will increase in the process of
* creating a database.
*/
rec.len = 1024;
rec.data = xmalloc(rec.len);
cdb_seqinit(&cpos, &cdbm);
while(cdb_seqnext(&cpos, &cdbm) > 0) {
char *type;
klen = cdb_keylen(&cdbm);
key = (char *) xmalloc((size_t) (klen + 1));
if (cdb_read(&cdbm, key, (unsigned) klen, cdb_keypos(&cdbm)) < 0)
osec_fatal(EXIT_FAILURE, errno, "cdb_read");
key[klen] = '\0';
if ((type = strchr(key, '\0')) == (key + klen))
osec_fatal(EXIT_FAILURE, errno, "strchr: Cant find type\n");
klen = strlen(key);
type += 1;
if (strcmp(type, "stat") == 0) {
struct stat st;
rec.offset = 0;
if (cdb_read(&cdbm, &st, (unsigned) sizeof(st), cdb_datapos(&cdbm)) < 0)
osec_fatal(EXIT_FAILURE, errno, "cdb_read");
osec_empty_digest(&rec);
osec_empty_symlink(&rec);
osec_state(&rec, &st);
if (cdb_make_add(&cdbn, key, (unsigned) klen+1, rec.data, (unsigned) rec.offset) != 0)
osec_fatal(EXIT_FAILURE, errno, "%s: cdb_make_add", key);
}
xfree(key);
}
write_db_version(&cdbn);
xfree(rec.data);
if (cdb_make_finish(&cdbn) < 0)
osec_fatal(EXIT_FAILURE, errno, "cdb_make_finish");
if (close(fdtemp) == -1)
osec_fatal(EXIT_FAILURE, errno, "%s: close", dbtemp);
if (close(fd) == -1)
osec_fatal(EXIT_FAILURE, errno, "%s: close", dbfile);
dirname = decode_dirname(dbfile);
gen_db_name(dirname, &dbnewfile);
rename(dbtemp, dbnewfile);
remove(dbfile);
xfree(dbtemp);
xfree(dirname);
xfree(dbnewfile);
return EXIT_SUCCESS;
}
int main()
{
char ip[4];
unsigned long u;
unsigned int j;
unsigned int k;
char ch;
umask(022);
fd = open_read("data");
if (fd == -1) strerr_die2sys(111,FATAL,"unable to open data: ");
buffer_init(&b,buffer_unixread,fd,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;
switch(line.s[0]) {
default:
syntaxerror(": unrecognized leading character");
case '#':
break;
case ':':
j = byte_chr(line.s + 1,line.len - 1,':');
if (j >= line.len - 1) syntaxerror(": missing colon");
if (ip4_scan(line.s + 1,ip) != j) syntaxerror(": malformed IP address");
if (!stralloc_copyb(&tmp,ip,4)) nomem();
if (!stralloc_catb(&tmp,line.s + j + 2,line.len - j - 2)) nomem();
if (cdb_make_add(&cdb,"",0,tmp.s,tmp.len) == -1)
die_datatmp();
break;
case '0':
case '1':
case '2':
case '3':
case '4':
case '5':
case '6':
case '7':
case '8':
case '9':
if (!stralloc_0(&line)) nomem();
j = 0;
if (!stralloc_copys(&tmp,"")) nomem();
for (;;) {
k = scan_ulong(line.s + j,&u);
if (!k) break;
ch = u;
if (!stralloc_catb(&tmp,&ch,1)) nomem();
j += k;
if (line.s[j] != '.') break;
++j;
}
if (!stralloc_catb(&tmp,"\0\0\0\0",4)) nomem();
tmp.len = 4;
if (line.s[j] == '/')
scan_ulong(line.s + j + 1,&u);
else
u = 32;
if (u > 32) u = 32;
ch = u;
if (!stralloc_catb(&tmp,&ch,1)) nomem();
if (cdb_make_add(&cdb,tmp.s,tmp.len,"",0) == -1)
die_datatmp();
break;
}
}
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: ");
return 0;
}
void Lists_OP_MakeCDB(char *txt_filename, char *cdb_filename, int force)
{
/*
struct stat cdb_stat;
struct stat txt_stat;
*/
struct cdb_make cdbm;
FILE *tmp_fd;
FILE *txt_fd;
char *tmp_str;
char *key, *val;
char str[OS_MAXSTR+1];
str[OS_MAXSTR]= '\0';
char tmp_filename[OS_MAXSTR];
tmp_filename[OS_MAXSTR - 2] = '\0';
snprintf(tmp_filename, OS_MAXSTR - 2, "%s.tmp", txt_filename);
/*
if((stat(txt_filename, &txt_stat)) == -1)
debug1("%s: stat of file %s failed", ARGV0, txt_filename);
if((stat(cdb_filename, &cdb_stat)) == -1)
debug1("%s: stat of file %s failed", ARGV0, cdb_filename);
*/
if(File_DateofChange(txt_filename) > File_DateofChange(cdb_filename) ||
force)
{
printf(" * File %s need to be updated\n", cdb_filename);
tmp_fd = fopen(tmp_filename, "w+");
cdb_make_start(&cdbm, tmp_fd);
if(!(txt_fd = fopen(txt_filename, "r")))
{
merror(FOPEN_ERROR, ARGV0, txt_filename);
return;
}
while((fgets(str, OS_MAXSTR-1,txt_fd)) != NULL)
{
/* Removing new lines or carriage returns. */
tmp_str = strchr(str, '\r');
if(tmp_str)
*tmp_str = '\0';
tmp_str = strchr(str, '\n');
if(tmp_str)
*tmp_str = '\0';
if((val = strchr(str, ':')))
{
*val = '\0';
val++;
}
else
{
continue;
}
key = str;
cdb_make_add(&cdbm, key, strlen(key), val, strlen(val));
if(force) print_out(" * adding - key: %s value: %s",key,val);
}
cdb_make_finish(&cdbm);
rename(tmp_filename,cdb_filename);
}
else
{
printf(" * File %s does not need to be compiled\n", cdb_filename);
}
}
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;
}
main()
{
struct address t;
int i;
int j;
int k;
char ch;
umask(022);
if (!address_alloc_readyplus(&x,0)) nomem();
fd = open_read("data");
if (fd == -1) strerr_die2sys(111,FATAL,"unable to open data: ");
buffer_init(&b,read,fd,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;
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]) {
default:
syntaxerror(": unrecognized leading character");
case '#':
break;
case '-':
break;
case '+':
byte_zero(&t,sizeof t);
if (!dns_domain_fromdot(&t.name,f[0].s,f[0].len)) nomem();
t.namelen = dns_domain_length(t.name);
case_lowerb(t.name,t.namelen);
if (!stralloc_0(&f[1])) nomem();
if (!ip4_scan(f[1].s,t.ip)) syntaxerror(": malformed IP address");
if (!stralloc_0(&f[2])) nomem();
if (!stralloc_0(&f[2])) nomem();
byte_copy(t.location,2,f[2].s);
if (!address_alloc_append(&x,&t)) nomem();
break;
case '%':
if (!stralloc_0(&f[0])) nomem();
if (!stralloc_0(&f[0])) nomem();
if (!stralloc_copyb(&result,f[0].s,2)) nomem();
if (!stralloc_0(&f[1])) nomem();
if (!stralloc_copys(&key,"%")) nomem();
ipprefix_cat(&key,f[1].s);
if (cdb_make_add(&cdb,key.s,key.len,result.s,result.len) == -1)
die_datatmp();
break;
}
}
close(fd);
address_sort(x.s,x.len);
i = 0;
while (i < x.len) {
for (j = i + 1; j < x.len; ++j)
if (address_diff(x.s + i,x.s + j))
break;
if (!stralloc_copys(&key,"+")) nomem();
if (!stralloc_catb(&key,x.s[i].location,2)) nomem();
if (!stralloc_catb(&key,x.s[i].name,x.s[i].namelen)) nomem();
if (!stralloc_copys(&result,"")) nomem();
while (i < j)
if (!stralloc_catb(&result,x.s[i++].ip,4)) nomem();
if (cdb_make_add(&cdb,key.s,key.len,result.s,result.len) == -1)
die_datatmp();
}
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);
}
