1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
|
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <systime.h>
#include <iobuf/iobuf.h>
#include <msg/msg.h>
#include <str/str.h>
#include "mailfront.h"
#include "qmtp.h"
unsigned long maxdatabytes;
static const response* resp;
static char buf[8192];
static str line;
static str tmp;
static void get_body(ibuf* in)
{
unsigned long bodylen;
char nl;
switch (get_netstring_len(in, &bodylen)) {
case -1: exit(0);
case 0: die1(111, "Invalid message body netstring");
}
if (bodylen == 0) die1(111, "Zero length body");
--bodylen;
if (!ibuf_getc(in, &nl)) die1(111, "EOF while reading body NL");
if (nl != LF) die1(111, "Cannot handle non-LF line terminator");
if (response_ok(resp)) resp = handle_data_start(0, "QMTP");
while (bodylen > 0) {
unsigned long len = sizeof buf;
if (len > bodylen) len = bodylen;
if (!ibuf_read(in, buf, len) && in->count == 0)
die1(111, "EOF while reading body");
if (response_ok(resp)) handle_data_bytes(buf, in->count);
bodylen -= in->count;
}
if (!ibuf_getc(in, &nl)) die1(111, "EOF while reading comma");
if (nl != ',') die1(111, "Invalid netstring terminator");
}
static void get_sender(ibuf* in)
{
switch (get_netstring(in, &line)) {
case -1: die1(111, "EOF while reading sender address");
case 0: die1(111, "Invalid sender netstring");
}
msg3("sender <", line.s, ">");
if (response_ok(resp))
resp = handle_sender(&line);
}
static void get_recips(ibuf* in)
{
unsigned long i;
switch (get_netstring(in, &line)) {
case -1: die1(111, "EOF while reading recipient list");
case 0: die1(111, "Invalid recipient netstring");
}
for (i = 0; response_ok(resp) && i < line.len; i++) {
unsigned long j;
unsigned long len;
for (len = 0, j = i; j < line.len; ++j) {
if (line.s[j] == ':') break;
if (line.s[j] < '0' || line.s[j] > '9')
die1(111, "Invalid netstring length");
len = len * 10 + line.s[j] - '0';
}
++j;
if (j + len > line.len) die1(111, "Netstring length too long");
if (line.s[j+len] != ',') die1(111, "Netstring missing comma");
str_copyb(&tmp, line.s+j, len);
msg3("recipient <", tmp.s, ">");
if (response_ok(resp))
resp = handle_recipient(&tmp);
i = j + len;
}
}
static void get_package(ibuf* in)
{
resp = handle_reset();
get_body(in);
get_sender(in);
get_recips(in);
if (response_ok(resp)) resp = handle_data_end();
if (!resp) resp = &resp_accepted;
if (!respond_resp(resp, 1)) die1(111, "EOF while sending response");
}
int qmtp_mainloop(void)
{
const response* r;
if ((r = handle_init()) != 0) {
respond_resp(r, 1);
return 1;
}
alarm(3600);
for (;;) get_package(&inbuf);
}
|