#include #include #include #include #include #include "alloc.h" #include "buffer.h" #include "byte.h" #include "case.h" #include "constmap.h" #include "direntry.h" #include "error.h" #include "exit.h" #include "fmt.h" #include "genalloc.h" #include "getln.h" #include "lock.h" #include "logmsg.h" #include "ndelay.h" #include "open.h" #include "scan.h" #include "seek.h" #include "select.h" #include "sig.h" #include "str.h" #include "stralloc.h" #include "control.h" #include "fmtqfn.h" #include "newfield.h" #include "now.h" #include "prioq.h" #include "qmail.h" #include "qsutil.h" #include "quote.h" #include "readsubdir.h" #include "sendtodo.h" #include "trigger.h" #ifdef USE_CONFIG #include "fehsqm-config.h" #else #include "auto_qmail.h" #endif long lifetime = 604800; long bouncemaxbytes = 0; stralloc percenthack = {0}; struct constmap mappercenthack; stralloc locals = {0}; struct constmap maplocals; stralloc vdoms = {0}; struct constmap mapvdoms; stralloc envnoathost = {0}; stralloc bouncefrom = {0}; stralloc bouncehost = {0}; stralloc doublebounceto = {0}; stralloc doublebouncehost = {0}; char strnum2[FMT_ULONG]; char strnum3[FMT_ULONG]; #define CHANNELS 2 char *chanaddr[CHANNELS] = {"local/", "remote/"}; char *chanstatusmsg[CHANNELS] = {" local ", " remote "}; char *tochan[CHANNELS] = {" to local ", " to remote "}; int chanfdout[CHANNELS] = {1, 3}; int chanfdin[CHANNELS] = {2, 4}; int chanskip[CHANNELS] = {10, 20}; int flagexitasap = 0; static void sigterm() { flagexitasap = 1; } int flagrunasap = 0; static void sigalrm() { flagrunasap = 1; } int flagreadasap = 0; static void sighup() { flagreadasap = 1; } static void cleandied() { log1s("alert: lost connection to qmail-clean ... exiting\n"); flagexitasap = 1; } int flagspawnalive[CHANNELS]; static void spawndied(int c) { log1s("alert: oh no! lost spawn connection! dying...\n"); flagspawnalive[c] = 0; flagexitasap = 1; } #define REPORTMAX 10000 datetime_sec recent; /* this file is too long ---------------------------------------- FILE CREATE */ stralloc fn = {0}; stralloc fn2 = {0}; char fnmake_strnum[FMT_ULONG]; static void nomem() { log1s("alert: out of memory, sleeping...\n"); sleep(10); } static void fnmake_init() { while (!stralloc_ready(&fn, FMTQFN)) nomem(); while (!stralloc_ready(&fn2, FMTQFN)) nomem(); } static void fnmake_info(unsigned long id) { fn.len = fmtqfn(fn.s, "info/", id, 1); } static void fnmake_todo(unsigned long id) { fn.len = fmtqfn(fn.s, "todo/", id, 1); } static void fnmake_mess(unsigned long id) { fn.len = fmtqfn(fn.s, "mess/", id, 1); } static void fnmake_foop(unsigned long id) { fn.len = fmtqfn(fn.s, "foop/", id, 0); } static void fnmake_split(unsigned long id) { fn.len = fmtqfn(fn.s, "", id, 1); } static void fnmake2_bounce(unsigned long id) { fn2.len = fmtqfn(fn2.s, "bounce/", id, 0); } static void fnmake_chanaddr(unsigned long id, int c) { fn.len = fmtqfn(fn.s, chanaddr[c], id, 1); } /* this file is too long ----------------------------------------- REWRITING */ static void senderadd(stralloc *sa, char *sender, char *recip) { int i; int j; int k; i = str_len(sender); if (i >= 4) { if (str_equal(sender + i - 4, "-@[]")) { j = byte_rchr(sender, i - 4, '@'); k = str_rchr(recip, '@'); if (recip[k] && (j + 5 <= i)) { /* owner-@host-@[] -> owner-recipbox=reciphost@host */ while (!stralloc_catb(sa, sender, j)) nomem(); while (!stralloc_catb(sa, recip, k)) nomem(); while (!stralloc_cats(sa, "=")) nomem(); while (!stralloc_cats(sa, recip + k + 1)) nomem(); while (!stralloc_cats(sa, "@")) nomem(); while (!stralloc_catb(sa, sender + j + 1, i - 5 - j)) nomem(); return; } } } while (!stralloc_cats(sa, sender)) nomem(); } /* this file is too long ---------------------------------------------- INFO */ static int getinfo(stralloc *sa, datetime_sec *dt, unsigned long id) { int fdnumber; struct stat st; static stralloc line = {0}; int match; char buf[128]; buffer b; fnmake_info(id); fdnumber = open_read(fn.s); if (fdnumber == -1) return 0; if (fstat(fdnumber, &st) == -1) { close(fdnumber); return 0; } buffer_init(&b, read, fdnumber, buf, sizeof(buf)); if (getln(&b, &line, &match, '\0') == -1) { close(fdnumber); return 0; } close(fdnumber); if (!match) return 0; if (line.s[0] != 'F') return 0; *dt = st.st_mtime; while (!stralloc_copys(sa, line.s + 1)) nomem(); while (!stralloc_0(sa)) nomem(); return 1; } /* this file is too long ------------------------------------- COMMUNICATION */ buffer toqc; char toqcbuf[1024]; buffer fromqc; char fromqcbuf[1024]; stralloc comm_buf[CHANNELS] = {{0}, {0}}; int comm_pos[CHANNELS]; static void comm_init() { int c; buffer_init(&toqc, write, 5, toqcbuf, sizeof(toqcbuf)); buffer_init(&fromqc, read, 6, fromqcbuf, sizeof(fromqcbuf)); for (c = 0; c < CHANNELS; ++c) if (ndelay_on(chanfdout[c]) == -1) /* this is so stupid: NDELAY semantics should be default on write */ spawndied(c); /* drastic, but better than risking deadlock */ } static int comm_canwrite(int c) { /* XXX: could allow a bigger buffer; say 10 recipients */ if (comm_buf[c].s && comm_buf[c].len) return 0; return 1; } static void comm_write(int c, int delnum, unsigned long id, char *sender, char *recip) { char ch; if (comm_buf[c].s && comm_buf[c].len) return; while (!stralloc_copys(&comm_buf[c], "")) nomem(); ch = delnum; while (!stralloc_append(&comm_buf[c], &ch)) nomem(); fnmake_split(id); while (!stralloc_cats(&comm_buf[c], fn.s)) nomem(); while (!stralloc_0(&comm_buf[c])) nomem(); senderadd(&comm_buf[c], sender, recip); while (!stralloc_0(&comm_buf[c])) nomem(); while (!stralloc_cats(&comm_buf[c], recip)) nomem(); while (!stralloc_0(&comm_buf[c])) nomem(); comm_pos[c] = 0; } static void comm_selprep(int *nfds, fd_set *wfds) { int c; for (c = 0; c < CHANNELS; ++c) { if (flagspawnalive[c]) { if (comm_buf[c].s && comm_buf[c].len) { FD_SET(chanfdout[c], wfds); if (*nfds <= chanfdout[c]) *nfds = chanfdout[c] + 1; } } } } static void comm_do(fd_set *wfds) { int c; for (c = 0; c < CHANNELS; ++c) { if (flagspawnalive[c]) { if (comm_buf[c].s && comm_buf[c].len) { if (FD_ISSET(chanfdout[c], wfds)) { int w; int len; len = comm_buf[c].len; w = write(chanfdout[c], comm_buf[c].s + comm_pos[c], len - comm_pos[c]); if (w <= 0) { if ((w == -1) && (errno == EPIPE)) spawndied(c); else continue; /* kernel select() bug; can't avoid busy-looping */ } else { comm_pos[c] += w; if (comm_pos[c] == len) comm_buf[c].len = 0; } } } } } } /* this file is too long ------------------------------------------ CLEANUPS */ int flagcleanup; /* if 1, cleanupdir is initialized and ready */ readsubdir cleanupdir; datetime_sec cleanuptime; static void cleanup_init() { flagcleanup = 0; cleanuptime = now(); } static void cleanup_selprep(datetime_sec *wakeup) { if (flagcleanup) *wakeup = 0; if (*wakeup > cleanuptime) *wakeup = cleanuptime; } static void cleanup_do() { char ch; struct stat st; unsigned long id; if (!flagcleanup) { if (recent < cleanuptime) return; readsubdir_init(&cleanupdir, "mess", pausedir); flagcleanup = 1; } switch (readsubdir_next(&cleanupdir, &id)) { case 1: break; case 0: flagcleanup = 0; cleanuptime = recent + SLEEP_CLEANUP; default: return; } fnmake_mess(id); if (stat(fn.s, &st) == -1) return; /* probably qmail-queue deleted it */ if (recent <= st.st_atime + OSSIFIED) return; fnmake_info(id); if (stat(fn.s, &st) == 0) return; if (errno != ENOENT) return; fnmake_todo(id); if (stat(fn.s, &st) == 0) return; if (errno != ENOENT) return; fnmake_foop(id); if (buffer_putflush(&toqc, fn.s, fn.len) == -1) { cleandied(); return; } if (buffer_get(&fromqc, &ch, 1) != 1) { cleandied(); return; } if (ch != '+') log3s("warning: qmail-clean unable to clean up ", fn.s, "\n"); } /* this file is too long ----------------------------------- PRIORITY QUEUES */ prioq pqdone = {0}; /* -todo +info; HOPEFULLY -local -remote */ prioq pqchan[CHANNELS] = {{0}, {0}}; /* pqchan 0: -todo +info +local ?remote */ /* pqchan 1: -todo +info ?local +remote */ prioq pqfail = {0}; /* stat() failure; has to be pqadded again */ static void pqadd(unsigned long id) { struct prioq_elt pe; struct prioq_elt pechan[CHANNELS]; int flagchan[CHANNELS]; struct stat st; int c; #define CHECKSTAT \ if (errno != ENOENT) goto FAIL; fnmake_info(id); if (stat(fn.s, &st) == -1) { CHECKSTAT return; /* someone yanking our chain */ } fnmake_todo(id); if (stat(fn.s, &st) != -1) return; /* look, ma, dad crashed writing info! */ CHECKSTAT for (c = 0; c < CHANNELS; ++c) { fnmake_chanaddr(id, c); if (stat(fn.s, &st) == -1) { flagchan[c] = 0; CHECKSTAT } else { flagchan[c] = 1; pechan[c].id = id; pechan[c].dt = st.st_mtime; } } for (c = 0; c < CHANNELS; ++c) if (flagchan[c]) while (!prioq_insert(&pqchan[c], &pechan[c])) nomem(); for (c = 0; c < CHANNELS; ++c) if (flagchan[c]) break; if (c == CHANNELS) { pe.id = id; pe.dt = now(); while (!prioq_insert(&pqdone, &pe)) nomem(); } return; FAIL: log3s("warning: unable to stat ", fn.s, "; will try again later\n"); pe.id = id; pe.dt = now() + SLEEP_SYSFAIL; while (!prioq_insert(&pqfail, &pe)) nomem(); } static void pqstart() { readsubdir rs; int x; unsigned long id; readsubdir_init(&rs, "info", pausedir); while ((x = readsubdir_next(&rs, &id))) if (x > 0) pqadd(id); } static void pqfinish() { int c; struct prioq_elt pe; //time_t ut[2]; /* XXX: more portable than utimbuf, but still worrisome */ struct utimbuf ut; for (c = 0; c < CHANNELS; ++c) { while (prioq_min(&pqchan[c], &pe)) { prioq_delmin(&pqchan[c]); fnmake_chanaddr(pe.id, c); ut.actime = ut.modtime = pe.dt; if (utime(fn.s, &ut) == -1) log3s("warning: unable to utime ", fn.s, "; message will be retried too soon\n"); } } } static void pqrun() { int c; int i; for (c = 0; c < CHANNELS; ++c) if (pqchan[c].p) if (pqchan[c].len) for (i = 0; i < pqchan[c].len; ++i) pqchan[c].p[i].dt = recent; } /* this file is too long ---------------------------------------------- JOBS */ struct job { int refs; /* if 0, this struct is unused */ unsigned long id; int channel; datetime_sec retry; stralloc sender; int numtodo; int flaghiteof; int flagdying; }; int numjobs; struct job *jo; static void job_init() { int j; while (!(jo = (struct job *)alloc(numjobs * sizeof(struct job)))) nomem(); for (j = 0; j < numjobs; ++j) { jo[j].refs = 0; jo[j].sender.s = 0; } } static int job_avail() { int j; for (j = 0; j < numjobs; ++j) if (!jo[j].refs) return 1; return 0; } static int job_open(unsigned long id, int channel) { int j; for (j = 0; j < numjobs; ++j) if (!jo[j].refs) break; if (j == numjobs) return -1; jo[j].refs = 1; jo[j].id = id; jo[j].channel = channel; jo[j].numtodo = 0; jo[j].flaghiteof = 0; return j; } static void job_close(int j) { struct prioq_elt pe; struct stat st; int c; if (0 < --jo[j].refs) return; pe.id = jo[j].id; pe.dt = jo[j].retry; if (jo[j].flaghiteof && !jo[j].numtodo) { fnmake_chanaddr(jo[j].id, jo[j].channel); if (unlink(fn.s) == -1) { log3s("warning: unable to unlink ", fn.s, "; will try again later\n"); pe.dt = now() + SLEEP_SYSFAIL; } else { for (c = 0; c < CHANNELS; ++c) if (c != jo[j].channel) { fnmake_chanaddr(jo[j].id, c); if (stat(fn.s, &st) == 0) return; /* more channels going */ if (errno != ENOENT) { log3s("warning: unable to stat ", fn.s, "\n"); break; /* this is the only reason for HOPEFULLY */ } } pe.dt = now(); while (!prioq_insert(&pqdone, &pe)) nomem(); return; } } while (!prioq_insert(&pqchan[jo[j].channel], &pe)) nomem(); } /* this file is too long ------------------------------------------- BOUNCES */ static char *stripvdomprepend(char *recip) { int i; char *domain; int domainlen; char *prepend; i = str_rchr(recip, '@'); if (!recip[i]) return recip; domain = recip + i + 1; domainlen = str_len(domain); for (i = 0; i <= domainlen; ++i) { if ((i == 0) || (i == domainlen) || (domain[i] == '.')) { if ((prepend = constmap(&mapvdoms, domain + i, domainlen - i))) { if (!*prepend) break; i = str_len(prepend); if (str_diffn(recip, prepend, i)) break; if (recip[i] != '-') break; return recip + i + 1; } } } return recip; } stralloc bouncetext = {0}; static void addbounce(unsigned long id, char *recip, char *report) { int fd; int pos; int w; while (!stralloc_copys(&bouncetext, "<")) nomem(); while (!stralloc_cats(&bouncetext, stripvdomprepend(recip))) nomem(); for (pos = 0; pos < bouncetext.len; ++pos) if (bouncetext.s[pos] == '\n') bouncetext.s[pos] = '_'; while (!stralloc_cats(&bouncetext, ">:\n")) nomem(); while (!stralloc_cats(&bouncetext, report)) nomem(); if (report[0]) if (report[str_len(report) - 1] != '\n') while (!stralloc_cats(&bouncetext, "\n")) nomem(); for (pos = bouncetext.len - 2; pos > 0; --pos) if (bouncetext.s[pos] == '\n') if (bouncetext.s[pos - 1] == '\n') bouncetext.s[pos] = '/'; while (!stralloc_cats(&bouncetext, "\n")) nomem(); fnmake2_bounce(id); for (;;) { fd = open_append(fn2.s); if (fd != -1) break; log1s("alert: unable to append to bounce message; HELP! sleeping...\n"); sleep(10); } pos = 0; while (pos < bouncetext.len) { w = write(fd, bouncetext.s + pos, bouncetext.len - pos); if (w <= 0) { log1s("alert: unable to append to bounce message; HELP! sleeping...\n"); sleep(10); } else { pos += w; } } close(fd); } static int injectbounce(unsigned long id) { struct qmail qqt; struct stat st; char *bouncesender; char *bouncerecip; int r; int fd; buffer bi; char buf[128]; char inbuf[128]; static stralloc sender = {0}; static stralloc quoted = {0}; datetime_sec birth; unsigned long qp; int bytestogo; int bytestoget; if (!getinfo(&sender, &birth, id)) return 0; /* XXX: print warning */ /* owner-@host-@[] -> owner-@host */ if (sender.len >= 5 && str_equal(sender.s + sender.len - 5, "-@[]")) { sender.len -= 4; sender.s[sender.len - 1] = 0; } fnmake2_bounce(id); fnmake_mess(id); if (stat(fn2.s, &st) == -1) { if (errno == ENOENT) return 1; log3s("warning: unable to stat ", fn2.s, "\n"); return 0; } if (str_equal(sender.s, "#@[]")) { log3s("triple bounce: discarding ", fn2.s, "\n"); } else if (!*sender.s && *doublebounceto.s == '@') { log3s("double bounce: discarding ", fn2.s, "\n"); } else { if (qmail_open(&qqt) == -1) { log1s("warning: unable to start qmail-queue, will try later\n"); return 0; } qp = qmail_qp(&qqt); if (*sender.s) { bouncesender = ""; bouncerecip = sender.s; } else { bouncesender = "#@[]"; bouncerecip = doublebounceto.s; } while (!newfield_datemake(now())) nomem(); qmail_put(&qqt, newfield_date.s, newfield_date.len); qmail_puts(&qqt, "From: "); while (!quote("ed, &bouncefrom)) nomem(); qmail_put(&qqt, quoted.s, quoted.len); qmail_puts(&qqt, "@"); qmail_put(&qqt, bouncehost.s, bouncehost.len); qmail_puts(&qqt, "\nTo: "); while (!quote2("ed, bouncerecip)) nomem(); qmail_put(&qqt, quoted.s, quoted.len); qmail_puts( &qqt, "\n\ Subject: failure notice\n\ \n\ Hi. This is the qmail-send program at "); qmail_put(&qqt, bouncehost.s, bouncehost.len); qmail_puts( &qqt, *sender.s ? ".\n\ I'm afraid I wasn't able to deliver your message to the following addresses.\n\ This is a permanent error; I've given up. Sorry it didn't work out.\n\ \n\ " : ".\n\ I tried to deliver a bounce message to this address, but the bounce bounced!\n\ \n\ "); fd = open_read(fn2.s); if (fd == -1) { qmail_fail(&qqt); } else { buffer_init(&bi, read, fd, inbuf, sizeof(inbuf)); while ((r = buffer_get(&bi, buf, sizeof(buf))) > 0) qmail_put(&qqt, buf, r); close(fd); if (r == -1) qmail_fail(&qqt); } qmail_puts( &qqt, *sender.s ? "--- Below this line is a copy of the message.\n\n" : "--- Below this line is the original bounce.\n\n"); qmail_puts(&qqt, "Return-Path: <"); while (!quote2("ed, sender.s)) nomem(); qmail_put(&qqt, quoted.s, quoted.len); qmail_puts(&qqt, ">\n"); fd = open_read(fn.s); if (fd == -1) { qmail_fail(&qqt); } else { if (bouncemaxbytes) { bytestogo = bouncemaxbytes; bytestoget = (bytestogo < sizeof(buf)) ? bytestogo : sizeof(buf); buffer_init(&bi, read, fd, inbuf, sizeof(inbuf)); while (bytestoget > 0 && (r = buffer_get(&bi, buf, bytestoget)) > 0) { qmail_put(&qqt, buf, r); bytestogo -= bytestoget; bytestoget = (bytestogo < sizeof(buf)) ? bytestogo : sizeof(buf); } if (r > 0) qmail_puts(&qqt, "\n\n--- Rest of message truncated.\n"); } else { /* preserve default behavior */ buffer_init(&bi, read, fd, inbuf, sizeof(inbuf)); while ((r = buffer_get(&bi, buf, sizeof(buf))) > 0) qmail_put(&qqt, buf, r); } close(fd); if (r == -1) qmail_fail(&qqt); } qmail_from(&qqt, bouncesender); qmail_to(&qqt, bouncerecip); if (*qmail_close(&qqt)) { log1s("warning: trouble injecting bounce message, will try later\n"); return 0; } strnum2[fmt_ulong(strnum2, id)] = 0; log2s("bounce msg ", strnum2); strnum2[fmt_ulong(strnum2, qp)] = 0; log3s(" qp ", strnum2, "\n"); } if (unlink(fn2.s) != 0) { log3s("warning: unable to unlink ", fn2.s, "\n"); return 0; } return 1; } /* this file is too long ---------------------------------------- DELIVERIES */ struct del { int used; int j; unsigned long delid; seek_pos mpos; stralloc recip; }; unsigned long masterdelid = 1; unsigned long concurrency[CHANNELS] = {10, 20}; unsigned long concurrencyused[CHANNELS] = {0, 0}; struct del *d[CHANNELS]; stralloc dline[CHANNELS]; char delbuf[2048]; static void del_status() { int c; log1s("status:"); for (c = 0; c < CHANNELS; ++c) { strnum2[fmt_ulong(strnum2, (unsigned long)concurrencyused[c])] = 0; strnum3[fmt_ulong(strnum3, (unsigned long)concurrency[c])] = 0; log2s(chanstatusmsg[c], strnum2); log2s("/", strnum3); } if (flagexitasap) log1s(" exitasap"); log1s("\n"); } static void del_init() { int c; int i; for (c = 0; c < CHANNELS; ++c) { flagspawnalive[c] = 1; while (!(d[c] = (struct del *)alloc(concurrency[c] * sizeof(struct del)))) nomem(); for (i = 0; i < concurrency[c]; ++i) { d[c][i].used = 0; d[c][i].recip.s = 0; } dline[c].s = 0; while (!stralloc_copys(&dline[c], "")) nomem(); } del_status(); } static int del_canexit() { int c; for (c = 0; c < CHANNELS; ++c) if (flagspawnalive[c]) /* if dead, nothing we can do about its jobs */ if (concurrencyused[c]) return 0; return 1; } static int del_avail(int c) { return flagspawnalive[c] && comm_canwrite(c) && (concurrencyused[c] < concurrency[c]); } static void del_start(int j, seek_pos mpos, char *recip) { int i; int c; c = jo[j].channel; if (!flagspawnalive[c]) return; if (!comm_canwrite(c)) return; for (i = 0; i < concurrency[c]; ++i) if (!d[c][i].used) break; if (i == concurrency[c]) return; if (!stralloc_copys(&d[c][i].recip, recip)) { nomem(); return; } if (!stralloc_0(&d[c][i].recip)) { nomem(); return; } d[c][i].j = j; ++jo[j].refs; d[c][i].delid = masterdelid++; d[c][i].mpos = mpos; d[c][i].used = 1; ++concurrencyused[c]; comm_write(c, i, jo[j].id, jo[j].sender.s, recip); strnum2[fmt_ulong(strnum2, d[c][i].delid)] = 0; strnum3[fmt_ulong(strnum3, jo[j].id)] = 0; log2s("starting delivery ", strnum2); log3s(": msg ", strnum3, tochan[c]); logsafe(recip); log1s("\n"); del_status(); } static void markdone(int c, unsigned long id, seek_pos pos) { struct stat st; int fd; fnmake_chanaddr(id, c); for (;;) { fd = open_write(fn.s); if (fd == -1) break; if (fstat(fd, &st) == -1) { close(fd); break; } if (seek_set(fd, pos) == -1) { close(fd); break; } if (write(fd, "D", 1) != 1) { close(fd); break; } /* further errors -> double delivery without us knowing about it, oh well */ close(fd); return; } log3s("warning: trouble marking ", fn.s, "; message will be delivered twice!\n"); } static void del_dochan(int c) { int r; char ch; int i; int delnum; r = read(chanfdin[c], delbuf, sizeof(delbuf)); if (r == -1) return; if (r == 0) { spawndied(c); return; } for (i = 0; i < r; ++i) { ch = delbuf[i]; while (!stralloc_append(&dline[c], &ch)) nomem(); if (dline[c].len > REPORTMAX) dline[c].len = REPORTMAX; /* qmail-lspawn and qmail-rspawn are responsible for keeping it short */ /* but from a security point of view, we don't trust rspawn */ if (!ch && (dline[c].len > 1)) { delnum = (unsigned int)(unsigned char)dline[c].s[0]; if ((delnum < 0) || (delnum >= concurrency[c]) || !d[c][delnum].used) { log1s("warning: internal error: delivery report out of range\n"); } else { strnum3[fmt_ulong(strnum3, d[c][delnum].delid)] = 0; if (dline[c].s[1] == 'Z') if (jo[d[c][delnum].j].flagdying) { dline[c].s[1] = 'D'; --dline[c].len; while (!stralloc_cats( &dline[c], "I'm not going to try again; this message has been in the queue too long.\n")) nomem(); while (!stralloc_0(&dline[c])) nomem(); } switch (dline[c].s[1]) { case 'K': log3s("delivery ", strnum3, ": success: "); logsafe(dline[c].s + 2); log1s("\n"); markdone(c, jo[d[c][delnum].j].id, d[c][delnum].mpos); --jo[d[c][delnum].j].numtodo; break; case 'Z': log3s("delivery ", strnum3, ": deferral: "); logsafe(dline[c].s + 2); log1s("\n"); break; case 'D': log3s("delivery ", strnum3, ": failure: "); logsafe(dline[c].s + 2); log1s("\n"); addbounce(jo[d[c][delnum].j].id, d[c][delnum].recip.s, dline[c].s + 2); markdone(c, jo[d[c][delnum].j].id, d[c][delnum].mpos); --jo[d[c][delnum].j].numtodo; break; default: log3s("delivery ", strnum3, ": report mangled, will defer\n"); } job_close(d[c][delnum].j); d[c][delnum].used = 0; --concurrencyused[c]; del_status(); } dline[c].len = 0; } } } static void del_selprep(int *nfds, fd_set *rfds) { int c; for (c = 0; c < CHANNELS; ++c) if (flagspawnalive[c]) { FD_SET(chanfdin[c], rfds); if (*nfds <= chanfdin[c]) *nfds = chanfdin[c] + 1; } } static void del_do(fd_set *rfds) { int c; for (c = 0; c < CHANNELS; ++c) if (flagspawnalive[c]) if (FD_ISSET(chanfdin[c], rfds)) del_dochan(c); } /* this file is too long -------------------------------------------- PASSES */ struct { unsigned long id; /* if 0, need a new pass */ int j; /* defined if id; job number */ int fd; /* defined if id; reading from {local,remote} */ seek_pos mpos; /* defined if id; mark position */ buffer b; char buf[128]; } pass[CHANNELS]; static void pass_init() { int c; for (c = 0; c < CHANNELS; ++c) pass[c].id = 0; } static void pass_selprep(datetime_sec *wakeup) { int c; struct prioq_elt pe; if (flagexitasap) return; for (c = 0; c < CHANNELS; ++c) if (pass[c].id) if (del_avail(c)) { *wakeup = 0; return; } if (job_avail()) for (c = 0; c < CHANNELS; ++c) if (!pass[c].id) if (prioq_min(&pqchan[c], &pe)) if (*wakeup > pe.dt) *wakeup = pe.dt; if (prioq_min(&pqfail, &pe)) if (*wakeup > pe.dt) *wakeup = pe.dt; if (prioq_min(&pqdone, &pe)) if (*wakeup > pe.dt) *wakeup = pe.dt; } static datetime_sec squareroot(datetime_sec x) /* result^2 <= x < (result + 1)^2 ; assuming: >= 0 */ { datetime_sec y; datetime_sec yy; datetime_sec y21; int j; y = 0; yy = 0; for (j = 15; j >= 0; --j) { y21 = (y << (j + 1)) + (1 << (j + j)); if (y21 <= x - yy) { y += (1 << j); yy += y21; } } return y; } static datetime_sec nextretry(datetime_sec birth, int c) { int n; if (birth > recent) n = 0; else n = squareroot(recent - birth); /* no need to add fuzz to recent */ n += chanskip[c]; return birth + n * n; } static void pass_dochan(int c) { datetime_sec birth; struct prioq_elt pe; static stralloc line = {0}; int match; if (flagexitasap) return; if (!pass[c].id) { if (!job_avail()) return; if (!prioq_min(&pqchan[c], &pe)) return; if (pe.dt > recent) return; fnmake_chanaddr(pe.id, c); prioq_delmin(&pqchan[c]); pass[c].mpos = 0; pass[c].fd = open_read(fn.s); if (pass[c].fd == -1) goto trouble; if (!getinfo(&line, &birth, pe.id)) { close(pass[c].fd); goto trouble; } pass[c].id = pe.id; buffer_init(&pass[c].b, read, pass[c].fd, pass[c].buf, sizeof(pass[c].buf)); pass[c].j = job_open(pe.id, c); jo[pass[c].j].retry = nextretry(birth, c); jo[pass[c].j].flagdying = (recent > birth + lifetime); while (!stralloc_copy(&jo[pass[c].j].sender, &line)) nomem(); } if (!del_avail(c)) return; if (getln(&pass[c].b, &line, &match, '\0') == -1) { fnmake_chanaddr(pass[c].id, c); log3s("warning: trouble reading ", fn.s, "; will try again later\n"); close(pass[c].fd); job_close(pass[c].j); pass[c].id = 0; return; } if (!match) { close(pass[c].fd); jo[pass[c].j].flaghiteof = 1; job_close(pass[c].j); pass[c].id = 0; return; } switch (line.s[0]) { case 'D': break; case 'T': ++jo[pass[c].j].numtodo; del_start(pass[c].j, pass[c].mpos, line.s + 1); break; default: fnmake_chanaddr(pass[c].id, c); log3s("warning: unknown record type in ", fn.s, "!\n"); close(pass[c].fd); job_close(pass[c].j); pass[c].id = 0; return; } pass[c].mpos += line.len; return; trouble: log3s("warning: trouble opening ", fn.s, "; will try again later\n"); pe.dt = recent + SLEEP_SYSFAIL; while (!prioq_insert(&pqchan[c], &pe)) nomem(); } static void messdone(unsigned long id) { char ch; int c; struct prioq_elt pe; struct stat st; for (c = 0; c < CHANNELS; ++c) { fnmake_chanaddr(id, c); if (stat(fn.s, &st) == 0) return; /* false alarm; consequence of HOPEFULLY */ if (errno != ENOENT) { log3s("warning: unable to stat ", fn.s, "; will try again later\n"); goto FAIL; } } fnmake_todo(id); if (stat(fn.s, &st) == 0) return; if (errno != ENOENT) { log3s("warning: unable to stat ", fn.s, "; will try again later\n"); goto FAIL; } fnmake_info(id); if (stat(fn.s, &st) == -1) { if (errno == ENOENT) return; log3s("warning: unable to stat ", fn.s, "; will try again later\n"); goto FAIL; } /* -todo +info -local -remote ?bounce */ if (!injectbounce(id)) goto FAIL; /* injectbounce() produced error message */ strnum3[fmt_ulong(strnum3, id)] = 0; log3s("end msg ", strnum3, "\n"); /* -todo +info -local -remote -bounce */ fnmake_info(id); if (unlink(fn.s) == -1) { log3s("warning: unable to unlink ", fn.s, "; will try again later\n"); goto FAIL; } /* -todo -info -local -remote -bounce; we can relax */ fnmake_foop(id); if (buffer_putflush(&toqc, fn.s, fn.len) == -1) { cleandied(); return; } if (buffer_get(&fromqc, &ch, 1) != 1) { cleandied(); return; } if (ch != '+') log3s("warning: qmail-clean unable to clean up ", fn.s, "\n"); return; FAIL: pe.id = id; pe.dt = now() + SLEEP_SYSFAIL; while (!prioq_insert(&pqdone, &pe)) nomem(); } static void pass_do() { int c; struct prioq_elt pe; for (c = 0; c < CHANNELS; ++c) pass_dochan(c); if (prioq_min(&pqfail, &pe)) if (pe.dt <= recent) { prioq_delmin(&pqfail); pqadd(pe.id); } if (prioq_min(&pqdone, &pe)) if (pe.dt <= recent) { prioq_delmin(&pqdone); messdone(pe.id); } } /* this file is too long ------------------------------------- EXTERNAL TODO */ stralloc todoline = {0}; char todobuf[2048]; int todofdin; int todofdout; int flagtodoalive; static void tododied() { log1s("alert: lost connection to qmail-todo ... exiting\n"); flagexitasap = 1; flagtodoalive = 0; } static void todo_init() { todofdout = 7; todofdin = 8; flagtodoalive = 1; /* sync with external todo */ if (write(todofdout, "S", 1) != 1) tododied(); return; } static void todo_selprep(int *nfds, fd_set *rfds, datetime_sec *wakeup) { if (flagexitasap) { if (flagtodoalive) { write(todofdout, "X", 1); } } if (flagtodoalive) { FD_SET(todofdin, rfds); if (*nfds <= todofdin) *nfds = todofdin + 1; } } static void todo_del(char *s) { int flagchan[CHANNELS]; struct prioq_elt pe; unsigned long id; unsigned int len; int c; for (c = 0; c < CHANNELS; ++c) flagchan[c] = 0; switch (*s++) { case 'L': flagchan[0] = 1; break; case 'R': flagchan[1] = 1; break; case 'B': flagchan[0] = 1; flagchan[1] = 1; break; case 'X': break; default: log1s("warning: qmail-send unable to understand qmail-todo\n"); return; } len = scan_ulong(s, &id); if (!len || s[len]) { log1s("warning: qmail-send unable to understand qmail-todo\n"); return; } pe.id = id; pe.dt = now(); for (c = 0; c < CHANNELS; ++c) if (flagchan[c]) while (!prioq_insert(&pqchan[c], &pe)) nomem(); for (c = 0; c < CHANNELS; ++c) if (flagchan[c]) break; if (c == CHANNELS) while (!prioq_insert(&pqdone, &pe)) nomem(); return; } static void todo_do(fd_set *rfds) { int r; char ch; int i; if (!flagtodoalive) return; if (!FD_ISSET(todofdin, rfds)) return; r = read(todofdin, todobuf, sizeof(todobuf)); if (r == -1) return; if (r == 0) { if (flagexitasap) flagtodoalive = 0; else tododied(); return; } for (i = 0; i < r; ++i) { ch = todobuf[i]; while (!stralloc_append(&todoline, &ch)) nomem(); if (todoline.len > REPORTMAX) todoline.len = REPORTMAX; /* qmail-todo is responsible for keeping it short */ if (!ch && (todoline.len > 1)) { switch (todoline.s[0]) { case 'D': if (flagexitasap) break; todo_del(todoline.s + 1); break; case 'L': log1s(todoline.s + 1); break; case 'X': if (flagexitasap) flagtodoalive = 0; else tododied(); break; default: log1s("warning: qmail-send unable to understand qmail-todo: report mangled\n"); break; } todoline.len = 0; } } } /* this file is too long ---------------------------------------------- MAIN */ static int getcontrols() { if (control_init() == -1) return 0; if (control_readint(&lifetime, "control/queuelifetime") == -1) return 0; if (control_readint(&concurrency[0], "control/concurrencylocal") == -1) return 0; if (control_readint(&concurrency[1], "control/concurrencyremote") == -1) return 0; if (control_rldef(&envnoathost, "control/envnoathost", 1, "envnoathost") != 1) return 0; if (control_rldef(&bouncefrom, "control/bouncefrom", 0, "MAILER-DAEMON") != 1) return 0; if (control_rldef(&bouncehost, "control/bouncehost", 1, "bouncehost") != 1) return 0; if (control_readint(&bouncemaxbytes, "control/bouncemaxbytes") == -1) return 0; if (control_rldef(&doublebouncehost, "control/doublebouncehost", 1, "doublebouncehost") != 1) return 0; if (control_rldef(&doublebounceto, "control/doublebounceto", 0, "postmaster") != 1) return 0; if (!stralloc_cats(&doublebounceto, "@")) return 0; if (!stralloc_cat(&doublebounceto, &doublebouncehost)) return 0; if (!stralloc_0(&doublebounceto)) return 0; if (control_readfile(&locals, "control/locals", 1) != 1) return 0; if (!constmap_init(&maplocals, locals.s, locals.len, 0)) return 0; switch (control_readfile(&percenthack, "control/percenthack", 0)) { case -1: return 0; case 0: if (!constmap_init(&mappercenthack, "", 0, 0)) return 0; break; case 1: if (!constmap_init(&mappercenthack, percenthack.s, percenthack.len, 0)) return 0; break; } switch (control_readfile(&vdoms, "control/virtualdomains", 0)) { case -1: return 0; case 0: if (!constmap_init(&mapvdoms, "", 0, 1)) return 0; break; case 1: if (!constmap_init(&mapvdoms, vdoms.s, vdoms.len, 1)) return 0; break; } return 1; } stralloc newlocals = {0}; stralloc newvdoms = {0}; static void regetcontrols() { int r; if (control_readfile(&newlocals, "control/locals", 1) != 1) { log1s("alert: unable to reread control/locals\n"); return; } if (control_readint(&concurrency[0], "control/concurrencylocal") == -1) { log1s("alert: unable to reread control/concurrencylocal\n"); return; } if (control_readint(&concurrency[1], "control/concurrencyremote") == -1) { log1s("alert: unable to reread control/concurrencyremote\n"); return; } if (control_readint(&lifetime, "control/queuelifetime") == -1) { log1s("alert: unable to reread control/queuelifetime\n"); return; } r = control_readfile(&newvdoms, "control/virtualdomains", 0); if (r == -1) { log1s("alert: unable to reread control/virtualdomains\n"); return; } constmap_free(&maplocals); constmap_free(&mapvdoms); while (!stralloc_copy(&locals, &newlocals)) nomem(); while (!constmap_init(&maplocals, locals.s, locals.len, 0)) nomem(); if (r) { while (!stralloc_copy(&vdoms, &newvdoms)) nomem(); while (!constmap_init(&mapvdoms, vdoms.s, vdoms.len, 1)) nomem(); } else while (!constmap_init(&mapvdoms, "", 0, 1)) nomem(); } static void reread() { if (chdir(auto_qmail) == -1) { log1s("alert: unable to reread controls: unable to switch to home directory\n"); return; } write(todofdout, "H", 1); regetcontrols(); while (chdir("queue") == -1) { log1s("alert: unable to switch back to queue directory; HELP! sleeping...\n"); sleep(10); } } int main() { int fd; datetime_sec wakeup; fd_set rfds; fd_set wfds; int nfds; struct timeval tv; int c; int u; int r; char ch; if (chdir(auto_qmail) == -1) { log1s("alert: cannot start: unable to switch to home directory\n"); _exit(110); } if (!getcontrols()) { log1s("alert: cannot start: unable to read controls\n"); _exit(111); } if (chdir("queue") == -1) { log1s("alert: cannot start: unable to switch to queue directory\n"); _exit(110); } sig_pipeignore(); sig_termcatch(sigterm); sig_alarmcatch(sigalrm); sig_hangupcatch(sighup); sig_childdefault(); umask(077); fd = open_write("lock/sendmutex"); if (fd == -1) { log1s("alert: cannot start: unable to open mutex\n"); _exit(111); } if (lock_exnb(fd) == -1) { log1s("alert: cannot start: qmail-send is already running\n"); _exit(111); } numjobs = 0; for (c = 0; c < CHANNELS; ++c) { do r = read(chanfdin[c], &ch, 1); while ((r == -1) && (errno == EINTR)); if (r < 1) { log1s("alert: cannot start: hath the daemon spawn no fire?\n"); _exit(111); } u = (unsigned int)(unsigned char)ch; if (concurrency[c] > u) concurrency[c] = u; numjobs += concurrency[c]; } fnmake_init(); comm_init(); pqstart(); job_init(); del_init(); pass_init(); todo_init(); cleanup_init(); while (!flagexitasap || !del_canexit() || flagtodoalive) { recent = now(); if (flagrunasap) { flagrunasap = 0; pqrun(); } if (flagreadasap) { flagreadasap = 0; reread(); } wakeup = recent + SLEEP_FOREVER; FD_ZERO(&rfds); FD_ZERO(&wfds); nfds = 1; comm_selprep(&nfds, &wfds); del_selprep(&nfds, &rfds); pass_selprep(&wakeup); todo_selprep(&nfds, &rfds, &wakeup); cleanup_selprep(&wakeup); if (wakeup <= recent) tv.tv_sec = 0; else tv.tv_sec = wakeup - recent + SLEEP_FUZZ; tv.tv_usec = 0; if (select(nfds, &rfds, &wfds, (fd_set *)0, &tv) == -1) if (errno == EINTR) ; else log1s("warning: trouble in select\n"); else { recent = now(); comm_do(&wfds); del_do(&rfds); todo_do(&rfds); pass_do(); cleanup_do(); } } pqfinish(); log1s("status: exiting\n"); _exit(0); }