mdw@git.distorted.org.uk Git - qmail/blob - qmail-popup.c

   1 #include "commands.h"
   2 #include "fd.h"
   3 #include "sig.h"
   4 #include "stralloc.h"
   5 #include "substdio.h"
   6 #include "alloc.h"
   7 #include "wait.h"
   8 #include "str.h"
   9 #include "byte.h"
  10 #include "now.h"
  11 #include "fmt.h"
  12 #include "exit.h"
  13 #include "readwrite.h"
  14 #include "timeoutread.h"
  15 #include "timeoutwrite.h"
  16
  17 void die() { _exit(1); }
  18
  19 int saferead(fd,buf,len) int fd; char *buf; int len;
  20 {
  21   int r;
  22   r = timeoutread(1200,fd,buf,len);
  23   if (r <= 0) die();
  24   return r;
  25 }
  26
  27 int safewrite(fd,buf,len) int fd; char *buf; int len;
  28 {
  29   int r;
  30   r = timeoutwrite(1200,fd,buf,len);
  31   if (r <= 0) die();
  32   return r;
  33 }
  34
  35 char ssoutbuf[128];
  36 substdio ssout = SUBSTDIO_FDBUF(safewrite,1,ssoutbuf,sizeof ssoutbuf);
  37
  38 char ssinbuf[128];
  39 substdio ssin = SUBSTDIO_FDBUF(saferead,0,ssinbuf,sizeof ssinbuf);
  40
  41 void puts(s) char *s;
  42 {
  43   substdio_puts(&ssout,s);
  44 }
  45 void flush()
  46 {
  47   substdio_flush(&ssout);
  48 }
  49 void err(s) char *s;
  50 {
  51   puts("-ERR ");
  52   puts(s);
  53   puts("\r\n");
  54   flush();
  55 }
  56
  57 void die_usage() { err("usage: popup hostname subprogram"); die(); }
  58 void die_nomem() { err("out of memory"); die(); }
  59 void die_pipe() { err("unable to open pipe"); die(); }
  60 void die_write() { err("unable to write pipe"); die(); }
  61 void die_fork() { err("unable to fork"); die(); }
  62 void die_childcrashed() { err("aack, child crashed"); }
  63 void die_badauth() { err("authorization failed"); }
  64
  65 void err_syntax() { err("syntax error"); }
  66 void err_wantuser() { err("USER first"); }
  67 void err_authoriz() { err("authorization first"); }
  68
  69 void okay() { puts("+OK \r\n"); flush(); }
  70 void pop3_quit() { okay(); die(); }
  71
  72
  73 char unique[FMT_ULONG + FMT_ULONG + 3];
  74 char *hostname;
  75 stralloc username = {0};
  76 int seenuser = 0;
  77 char **childargs;
  78 substdio ssup;
  79 char upbuf[128];
  80
  81
  82 void doanddie(user,userlen,pass)
  83 char *user;
  84 unsigned int userlen; /* including 0 byte */
  85 char *pass;
  86 {
  87   int child;
  88   int wstat;
  89   int pi[2];
  90
  91   if (fd_copy(2,1) == -1) die_pipe();
  92   close(3);
  93   if (pipe(pi) == -1) die_pipe();
  94   if (pi[0] != 3) die_pipe();
  95   switch(child = fork()) {
  96     case -1:
  97       die_fork();
  98     case 0:
  99       close(pi[1]);
 100       sig_pipedefault();
 101       execvp(*childargs,childargs);
 102       _exit(1);
 103   }
 104   close(pi[0]);
 105   substdio_fdbuf(&ssup,write,pi[1],upbuf,sizeof upbuf);
 106   if (substdio_put(&ssup,user,userlen) == -1) die_write();
 107   if (substdio_put(&ssup,pass,str_len(pass) + 1) == -1) die_write();
 108   if (substdio_puts(&ssup,"<") == -1) die_write();
 109   if (substdio_puts(&ssup,unique) == -1) die_write();
 110   if (substdio_puts(&ssup,hostname) == -1) die_write();
 111   if (substdio_put(&ssup,">",2) == -1) die_write();
 112   if (substdio_flush(&ssup) == -1) die_write();
 113   close(pi[1]);
 114   byte_zero(pass,str_len(pass));
 115   byte_zero(upbuf,sizeof upbuf);
 116   if (wait_pid(&wstat,child) == -1) die();
 117   if (wait_crashed(wstat)) die_childcrashed();
 118   if (wait_exitcode(wstat)) die_badauth();
 119   die();
 120 }
 121 void pop3_greet()
 122 {
 123   char *s;
 124   s = unique;
 125   s += fmt_uint(s,getpid());
 126   *s++ = '.';
 127   s += fmt_ulong(s,(unsigned long) now());
 128   *s++ = '@';
 129   *s++ = 0;
 130   puts("+OK <");
 131   puts(unique);
 132   puts(hostname);
 133   puts(">\r\n");
 134   flush();
 135 }
 136 void pop3_user(arg) char *arg;
 137 {
 138   if (!*arg) { err_syntax(); return; }
 139   okay();
 140   seenuser = 1;
 141   if (!stralloc_copys(&username,arg)) die_nomem();
 142   if (!stralloc_0(&username)) die_nomem();
 143 }
 144 void pop3_pass(arg) char *arg;
 145 {
 146   if (!seenuser) { err_wantuser(); return; }
 147   if (!*arg) { err_syntax(); return; }
 148   doanddie(username.s,username.len,arg);
 149 }
 150 void pop3_apop(arg) char *arg;
 151 {
 152   char *space;
 153   space = arg + str_chr(arg,' ');
 154   if (!*space) { err_syntax(); return; }
 155   *space++ = 0;
 156   doanddie(arg,space - arg,space);
 157 }
 158
 159 struct commands pop3commands[] = {
 160   { "user", pop3_user, 0 }
 161 , { "pass", pop3_pass, 0 }
 162 , { "apop", pop3_apop, 0 }
 163 , { "quit", pop3_quit, 0 }
 164 , { "noop", okay, 0 }
 165 , { 0, err_authoriz, 0 }
 166 } ;
 167
 168 void main(argc,argv)
 169 int argc;
 170 char **argv;
 171 {
 172   sig_alarmcatch(die);
 173   sig_pipeignore();
 174
 175   hostname = argv[1];
 176   if (!hostname) die_usage();
 177   childargs = argv + 2;
 178   if (!*childargs) die_usage();
 179
 180   pop3_greet();
 181   commands(&ssin,pop3commands);
 182   die();
 183 }