static int read_cdb_pair(ibuf* in, str* key, str* data)
{
unsigned char buf[8];
uint32 keylen;
uint32 datalen;
if (!ibuf_read(in, (char*)buf, sizeof buf))
return 0;
keylen = uint32_get_lsb(buf);
datalen = uint32_get_lsb(buf+4);
if (!str_ready(key, keylen)
|| !str_ready(data, datalen)
|| !ibuf_read(in, key->s, keylen)
|| !ibuf_read(in, data->s, datalen))
return 0;
key->s[key->len = keylen] = 0;
data->s[data->len = datalen] = 0;
return 1;
}
static void init_slots(void)
{
int slot;
for (slot = 0; slot < SLOT_MAX; ++slot) {
slots[slot].pid = 0;
slots[slot].tmpfd = -1;
wrap_str(str_ready(&slots[slot].id, 16));
}
}
int cdb_getnext(struct cdb* cdb, const str* key, str* data)
{
if (cdb == 0 || key == 0 || data == 0) return -1;
if (!cdb_findnext(cdb, key->s, key->len)) return 0;
if (!str_ready(data, cdb->dlen)) return -1;
if (cdb_read(cdb, (unsigned char*)data->s, cdb->dlen, cdb->dpos) == -1)
return -1;
data->len = cdb->dlen;
data->s[data->len] = 0;
return 1;
}
/* This parser is rather liberal in what it accepts:
The IMAP standard mandates seperate character sets for tags, commands,
unquoted strings. Since they all must be seperated by spaces, this
parser allows all non-whitespace characters for each of the above. */
static int parse_line(void)
{
#define RESPOND(T,S) do{ respond(T,S); return 0; }while(0)
unsigned i;
unsigned len;
const char* ptr;
str* arg;
/* Parse out the command tag */
str_truncate(&tag, 0);
for (i = 0, ptr = line.s; i < line.len && isspace(*ptr); ++i, ++ptr) ;
for (; i < line.len && !isspace(*ptr); ++i, ++ptr)
str_catc(&tag, line.s[i]);
if (!tag.len) RESPOND(NOTAG, "BAD Syntax error");
if (i >= line.len) RESPOND(0, "BAD Syntax error");
/* Parse out the command itself */
str_truncate(&cmd, 0);
for (; i < line.len && isspace(*ptr); ++i, ++ptr) ;
for (; i < line.len && !isspace(*ptr); ++i, ++ptr)
str_catc(&cmd, line.s[i]);
if (!cmd.len) RESPOND(0, "BAD Syntax error");
/* Parse out the command-line args */
for (line_argc = 0; line_argc < MAX_ARGC; ++line_argc) {
arg = &line_args[line_argc];
str_truncate(arg, 0);
for (; i < line.len && isspace(*ptr); ++i, ++ptr) ;
if (i >= line.len) break;
switch (*ptr) {
case LBRACE:
/* Handle a string literal */
++i, ++ptr;
if (!isdigit(*ptr)) RESPOND(0, "BAD Syntax error: missing integer");
for (len = 0; i < line.len && *ptr != RBRACE; ++i, ++ptr) {
if (!isdigit(*ptr))
RESPOND(0, "BAD Syntax error: invalid integer");
len = len * 10 + *ptr - '0';
}
++i, ++ptr;
if (*ptr != 0) RESPOND(0, "BAD Syntax error: missing LF after integer");
str_ready(arg, len);
respond(CONT, "OK");
if (len > 0)
ibuf_read(&inbuf, arg->s, len);
arg->s[arg->len = len] = 0;
ibuf_getstr_crlf(&inbuf, &line);
i = 0;
ptr = line.s;
break;
case QUOTE:
/* Handle a quoted string */
for (++i, ++ptr; i < line.len && *ptr != QUOTE; ++i, ++ptr) {
if (*ptr == ESCAPE) {
if (++i >= line.len) break;
++ptr;
}
str_catc(arg, *ptr);
}
if (i >= line.len || *ptr != QUOTE)
RESPOND(0, "BAD Syntax error: unterminated quoted string");
++i, ++ptr;
break;
default:
/* Normal case is very simple */
for (; i < line.len && !isspace(*ptr); ++i, ++ptr)
str_catc(arg, *ptr);
}
}
for (; i < line.len && isspace(*ptr); ++i, ++ptr) ;
if (i < line.len) RESPOND(0, "BAD Too many command arguments");
return 1;
}
int cli_main(int argc, char* argv[])
{
const char* server_name = 0;
const char* env;
msg_debug_init();
encr_start();
prep_sender();
service = argv[0];
if (argc > 1
&& argv[1][0] != '-'
&& argv[1][0] != '+') {
server_name = argv[1];
++argv;
--argc;
}
if (argc > 1)
load_patterns(argv + 1);
if (server_name == 0
&& (server_name = getenv("SERVER")) == 0)
die1(1, "Server address not named on command line nor in $SERVER");
if (!resolve_ipv4name(server_name, &ip))
die3(1, "Could not resolve '", server_name, "'");
brandom_init();
if ((env = getenv("KEYDIR")) != 0)
keydir = env;
load_keys(server_name);
if ((env = getenv("PORT")) != 0)
port = strtoul(env, 0, 10);
if (port == 0)
port = 11014;
if ((sock = socket_udp()) == -1)
die1sys(1, "Could not create UDP socket");
if (!socket_connect4(sock, &ip, port))
die1sys(1, "Could not bind socket");
if (!str_ready(&packet, 65535)
|| !str_ready(&rpacket, 4+4+8+256*5))
die1(1, "Out of memory");
getenvu("ACK_TIMEOUT", &ack_timeout);
getenvu("CID_TIMEOUT", &cid_timeout);
getenvu("RETRANSMITS", &retransmits);
getenvu("READWAIT", &readwait);
getenvu("STARTLINES", &startlines);
if (getenv("EXITONEOF") != 0)
exitoneof = 1;
if (getenv("NOFILES") == 0 && getenv("NOFILE") == 0)
buffer = buffer_file_init();
else
buffer = buffer_nofile_init();
sig_all_catch(sigfn);
exitasap = 0;
mainloop();
return 0;
}
