chiark - git - mdw - qmail/blob - qmail-pw2u.c

   1 #include <sys/types.h>
   2 #include <sys/stat.h>
   3 #include "substdio.h"
   4 #include "readwrite.h"
   5 #include "subfd.h"
   6 #include "sgetopt.h"
   7 #include "control.h"
   8 #include "constmap.h"
   9 #include "stralloc.h"
  10 #include "fmt.h"
  11 #include "str.h"
  12 #include "scan.h"
  13 #include "open.h"
  14 #include "error.h"
  15 #include "getln.h"
  16 #include "auto_break.h"
  17 #include "auto_qmail.h"
  18 #include "auto_usera.h"
  19
  20 void die_chdir()
  21 {
  22   substdio_putsflush(subfderr,"qmail-pw2u: fatal: unable to chdir\n");
  23   _exit(111);
  24 }
  25 void die_nomem()
  26 {
  27   substdio_putsflush(subfderr,"qmail-pw2u: fatal: out of memory\n");
  28   _exit(111);
  29 }
  30 void die_read()
  31 {
  32   substdio_putsflush(subfderr,"qmail-pw2u: fatal: unable to read input\n");
  33   _exit(111);
  34 }
  35 void die_write()
  36 {
  37   substdio_putsflush(subfderr,"qmail-pw2u: fatal: unable to write output\n");
  38   _exit(111);
  39 }
  40 void die_control()
  41 {
  42   substdio_putsflush(subfderr,"qmail-pw2u: fatal: unable to read controls\n");
  43   _exit(111);
  44 }
  45 void die_alias()
  46 {
  47   substdio_puts(subfderr,"qmail-pw2u: fatal: unable to find ");
  48   substdio_puts(subfderr,auto_usera);
  49   substdio_puts(subfderr," user\n");
  50   substdio_flush(subfderr);
  51   _exit(111);
  52 }
  53 void die_home(fn) char *fn;
  54 {
  55   substdio_puts(subfderr,"qmail-pw2u: fatal: unable to stat ");
  56   substdio_puts(subfderr,fn);
  57   substdio_puts(subfderr,"\n");
  58   substdio_flush(subfderr);
  59   _exit(111);
  60 }
  61 void die_user(s,len) char *s; unsigned int len;
  62 {
  63   substdio_puts(subfderr,"qmail-pw2u: fatal: unable to find ");
  64   substdio_put(subfderr,s,len);
  65   substdio_puts(subfderr," user for subuser\n");
  66   substdio_flush(subfderr);
  67   _exit(111);
  68 }
  69
  70 char *dashcolon = "-:";
  71 int flagalias = 0;
  72 int flagnoupper = 1;
  73 int homestrategy = 2;
  74 /* 2: skip if home does not exist; skip if home is not owned by user */
  75 /* 1: stop if home does not exist; skip if home is not owned by user */
  76 /* 0: don't worry about home */
  77
  78 int okincl; stralloc incl = {0}; struct constmap mapincl;
  79 int okexcl; stralloc excl = {0}; struct constmap mapexcl;
  80 int okmana; stralloc mana = {0}; struct constmap mapmana;
  81
  82 stralloc allusers = {0}; struct constmap mapuser;
  83
  84 stralloc uugh = {0};
  85 stralloc user = {0};
  86 stralloc uidstr = {0};
  87 stralloc gidstr = {0};
  88 stralloc home = {0};
  89 unsigned long uid;
  90
  91 stralloc line = {0};
  92
  93 void doaccount()
  94 {
  95   struct stat st;
  96   int i;
  97   char *mailnames;
  98   char *x;
  99   unsigned int xlen;
 100
 101   if (byte_chr(line.s,line.len,'\0') < line.len) return;
 102
 103   x = line.s; xlen = line.len; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 104   if (!stralloc_copyb(&user,x,i)) die_nomem();
 105   if (!stralloc_0(&user)) die_nomem();
 106   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 107   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 108   if (!stralloc_copyb(&uidstr,x,i)) die_nomem();
 109   if (!stralloc_0(&uidstr)) die_nomem();
 110   scan_ulong(uidstr.s,&uid);
 111   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 112   if (!stralloc_copyb(&gidstr,x,i)) die_nomem();
 113   if (!stralloc_0(&gidstr)) die_nomem();
 114   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 115   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 116   if (!stralloc_copyb(&home,x,i)) die_nomem();
 117   if (!stralloc_0(&home)) die_nomem();
 118
 119   if (!uid) return;
 120   if (flagnoupper)
 121     for (i = 0;i < user.len;++i)
 122       if ((user.s[i] >= 'A') && (user.s[i] <= 'Z'))
 123         return;
 124   if (okincl)
 125     if (!constmap(&mapincl,user.s,user.len - 1))
 126       return;
 127   if (okexcl)
 128     if (constmap(&mapexcl,user.s,user.len - 1))
 129       return;
 130   if (homestrategy) {
 131     if (stat(home.s,&st) == -1) {
 132       if (errno != error_noent) die_home(home.s);
 133       if (homestrategy == 1) die_home(home.s);
 134       return;
 135     }
 136     if (st.st_uid != uid) return;
 137   }
 138
 139   if (!stralloc_copys(&uugh,":")) die_nomem();
 140   if (!stralloc_cats(&uugh,user.s)) die_nomem();
 141   if (!stralloc_cats(&uugh,":")) die_nomem();
 142   if (!stralloc_cats(&uugh,uidstr.s)) die_nomem();
 143   if (!stralloc_cats(&uugh,":")) die_nomem();
 144   if (!stralloc_cats(&uugh,gidstr.s)) die_nomem();
 145   if (!stralloc_cats(&uugh,":")) die_nomem();
 146   if (!stralloc_cats(&uugh,home.s)) die_nomem();
 147   if (!stralloc_cats(&uugh,":")) die_nomem();
 148
 149   /* XXX: avoid recording in allusers unless sub actually needs it */
 150   if (!stralloc_cats(&allusers,user.s)) die_nomem();
 151   if (!stralloc_cats(&allusers,":")) die_nomem();
 152   if (!stralloc_catb(&allusers,uugh.s,uugh.len)) die_nomem();
 153   if (!stralloc_0(&allusers)) die_nomem();
 154
 155   if (str_equal(user.s,auto_usera)) {
 156     if (substdio_puts(subfdout,"+") == -1) die_write();
 157     if (substdio_put(subfdout,uugh.s,uugh.len) == -1) die_write();
 158     if (substdio_puts(subfdout,dashcolon) == -1) die_write();
 159     if (substdio_puts(subfdout,":\n") == -1) die_write();
 160     flagalias = 1;
 161   }
 162
 163   mailnames = 0;
 164   if (okmana)
 165     mailnames = constmap(&mapmana,user.s,user.len - 1);
 166   if (!mailnames)
 167     mailnames = user.s;
 168
 169   for (;;) {
 170     while (*mailnames == ':') ++mailnames;
 171     if (!*mailnames) break;
 172
 173     i = str_chr(mailnames,':');
 174
 175     if (substdio_puts(subfdout,"=") == -1) die_write();
 176     if (substdio_put(subfdout,mailnames,i) == -1) die_write();
 177     if (substdio_put(subfdout,uugh.s,uugh.len) == -1) die_write();
 178     if (substdio_puts(subfdout,"::\n") == -1) die_write();
 179
 180     if (*auto_break) {
 181       if (substdio_puts(subfdout,"+") == -1) die_write();
 182       if (substdio_put(subfdout,mailnames,i) == -1) die_write();
 183       if (substdio_put(subfdout,auto_break,1) == -1) die_write();
 184       if (substdio_put(subfdout,uugh.s,uugh.len) == -1) die_write();
 185       if (substdio_puts(subfdout,dashcolon) == -1) die_write();
 186       if (substdio_puts(subfdout,":\n") == -1) die_write();
 187     }
 188
 189     mailnames += i;
 190   }
 191 }
 192
 193 stralloc sub = {0};
 194
 195 void dosubuser()
 196 {
 197   int i;
 198   char *x;
 199   unsigned int xlen;
 200   char *uugh;
 201
 202   x = line.s; xlen = line.len; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 203   if (!stralloc_copyb(&sub,x,i)) die_nomem();
 204   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 205   uugh = constmap(&mapuser,x,i);
 206   if (!uugh) die_user(x,i);
 207   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 208
 209   if (substdio_puts(subfdout,"=") == -1) die_write();
 210   if (substdio_put(subfdout,sub.s,sub.len) == -1) die_write();
 211   if (substdio_puts(subfdout,uugh) == -1) die_write();
 212   if (substdio_puts(subfdout,dashcolon) == -1) die_write();
 213   if (substdio_put(subfdout,x,i) == -1) die_write();
 214   if (substdio_puts(subfdout,":\n") == -1) die_write();
 215
 216   if (*auto_break) {
 217     if (substdio_puts(subfdout,"+") == -1) die_write();
 218     if (substdio_put(subfdout,sub.s,sub.len) == -1) die_write();
 219     if (substdio_put(subfdout,auto_break,1) == -1) die_write();
 220     if (substdio_puts(subfdout,uugh) == -1) die_write();
 221     if (substdio_puts(subfdout,dashcolon) == -1) die_write();
 222     if (substdio_put(subfdout,x,i) == -1) die_write();
 223     if (substdio_puts(subfdout,"-:\n") == -1) die_write();
 224   }
 225 }
 226
 227 int fd;
 228 substdio ss;
 229 char ssbuf[SUBSTDIO_INSIZE];
 230
 231 void main(argc,argv)
 232 int argc;
 233 char **argv;
 234 {
 235   int opt;
 236   int match;
 237
 238   while ((opt = getopt(argc,argv,"/ohHuUc:C")) != opteof)
 239     switch(opt) {
 240       case '/': dashcolon = "-/:"; break;
 241       case 'o': homestrategy = 2; break;
 242       case 'h': homestrategy = 1; break;
 243       case 'H': homestrategy = 0; break;
 244       case 'u': flagnoupper = 0; break;
 245       case 'U': flagnoupper = 1; break;
 246       case 'c': *auto_break = *optarg; break;
 247       case 'C': *auto_break = 0; break;
 248       case '?':
 249       default:
 250         _exit(100);
 251     }
 252
 253   if (chdir(auto_qmail) == -1) die_chdir();
 254
 255   /* no need for control_init() */
 256
 257   okincl = control_readfile(&incl,"users/include",0);
 258   if (okincl == -1) die_control();
 259   if (okincl) if (!constmap_init(&mapincl,incl.s,incl.len,0)) die_nomem();
 260
 261   okexcl = control_readfile(&excl,"users/exclude",0);
 262   if (okexcl == -1) die_control();
 263   if (okexcl) if (!constmap_init(&mapexcl,excl.s,excl.len,0)) die_nomem();
 264
 265   okmana = control_readfile(&mana,"users/mailnames",0);
 266   if (okmana == -1) die_control();
 267   if (okmana) if (!constmap_init(&mapmana,mana.s,mana.len,1)) die_nomem();
 268
 269   if (!stralloc_copys(&allusers,"")) die_nomem();
 270
 271   for (;;) {
 272     if (getln(subfdin,&line,&match,'\n') == -1) die_read();
 273     doaccount();
 274     if (!match) break;
 275   }
 276   if (!flagalias) die_alias();
 277
 278   fd = open_read("users/subusers");
 279   if (fd == -1) {
 280     if (errno != error_noent) die_control();
 281   }
 282   else {
 283     substdio_fdbuf(&ss,read,fd,ssbuf,sizeof(ssbuf));
 284
 285     if (!constmap_init(&mapuser,allusers.s,allusers.len,1)) die_nomem();
 286
 287     for (;;) {
 288       if (getln(&ss,&line,&match,'\n') == -1) die_read();
 289       dosubuser();
 290       if (!match) break;
 291     }
 292
 293     close(fd);
 294   }
 295
 296   fd = open_read("users/append");
 297   if (fd == -1) {
 298     if (errno != error_noent) die_control();
 299   }
 300   else {
 301     substdio_fdbuf(&ss,read,fd,ssbuf,sizeof(ssbuf));
 302     for (;;) {
 303       if (getln(&ss,&line,&match,'\n') == -1) die_read();
 304       if (substdio_put(subfdout,line.s,line.len) == -1) die_write();
 305       if (!match) break;
 306     }
 307   }
 308
 309   if (substdio_puts(subfdout,".\n") == -1) die_write();
 310   if (substdio_flush(subfdout) == -1) die_write();
 311   _exit(0);
 312 }