mdw@git.distorted.org.uk Git - 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 int flagalias = 0;
  71 int flagnoupper = 1;
  72 int homestrategy = 2;
  73 /* 2: skip if home does not exist; skip if home is not owned by user */
  74 /* 1: stop if home does not exist; skip if home is not owned by user */
  75 /* 0: don't worry about home */
  76
  77 int okincl; stralloc incl = {0}; struct constmap mapincl;
  78 int okexcl; stralloc excl = {0}; struct constmap mapexcl;
  79 int okmana; stralloc mana = {0}; struct constmap mapmana;
  80
  81 stralloc allusers = {0}; struct constmap mapuser;
  82
  83 stralloc uugh = {0};
  84 stralloc user = {0};
  85 stralloc uidstr = {0};
  86 stralloc gidstr = {0};
  87 stralloc home = {0};
  88 unsigned long uid;
  89
  90 stralloc line = {0};
  91
  92 void doaccount()
  93 {
  94   struct stat st;
  95   int i;
  96   char *mailnames;
  97   char *x;
  98   unsigned int xlen;
  99
 100   if (byte_chr(line.s,line.len,'\0') < line.len) return;
 101
 102   x = line.s; xlen = line.len; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 103   if (!stralloc_copyb(&user,x,i)) die_nomem();
 104   if (!stralloc_0(&user)) die_nomem();
 105   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 106   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 107   if (!stralloc_copyb(&uidstr,x,i)) die_nomem();
 108   if (!stralloc_0(&uidstr)) die_nomem();
 109   scan_ulong(uidstr.s,&uid);
 110   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 111   if (!stralloc_copyb(&gidstr,x,i)) die_nomem();
 112   if (!stralloc_0(&gidstr)) die_nomem();
 113   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 114   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 115   if (!stralloc_copyb(&home,x,i)) die_nomem();
 116   if (!stralloc_0(&home)) die_nomem();
 117
 118   if (!uid) return;
 119   if (flagnoupper)
 120     for (i = 0;i < user.len;++i)
 121       if ((user.s[i] >= 'A') && (user.s[i] <= 'Z'))
 122         return;
 123   if (okincl)
 124     if (!constmap(&mapincl,user.s,user.len - 1))
 125       return;
 126   if (okexcl)
 127     if (constmap(&mapexcl,user.s,user.len - 1))
 128       return;
 129   if (homestrategy) {
 130     if (stat(home.s,&st) == -1) {
 131       if (errno != error_noent) die_home(home.s);
 132       if (homestrategy == 1) die_home(home.s);
 133       return;
 134     }
 135     if (st.st_uid != uid) return;
 136   }
 137
 138   if (!stralloc_copys(&uugh,":")) die_nomem();
 139   if (!stralloc_cats(&uugh,user.s)) die_nomem();
 140   if (!stralloc_cats(&uugh,":")) die_nomem();
 141   if (!stralloc_cats(&uugh,uidstr.s)) die_nomem();
 142   if (!stralloc_cats(&uugh,":")) die_nomem();
 143   if (!stralloc_cats(&uugh,gidstr.s)) die_nomem();
 144   if (!stralloc_cats(&uugh,":")) die_nomem();
 145   if (!stralloc_cats(&uugh,home.s)) die_nomem();
 146   if (!stralloc_cats(&uugh,":")) die_nomem();
 147
 148   /* XXX: avoid recording in allusers unless sub actually needs it */
 149   if (!stralloc_cats(&allusers,user.s)) die_nomem();
 150   if (!stralloc_cats(&allusers,":")) die_nomem();
 151   if (!stralloc_catb(&allusers,uugh.s,uugh.len)) die_nomem();
 152   if (!stralloc_0(&allusers)) die_nomem();
 153
 154   if (str_equal(user.s,auto_usera)) {
 155     if (substdio_puts(subfdout,"+") == -1) die_write();
 156     if (substdio_put(subfdout,uugh.s,uugh.len) == -1) die_write();
 157     if (substdio_puts(subfdout,"-::\n") == -1) die_write();
 158     flagalias = 1;
 159   }
 160
 161   mailnames = 0;
 162   if (okmana)
 163     mailnames = constmap(&mapmana,user.s,user.len - 1);
 164   if (!mailnames)
 165     mailnames = user.s;
 166
 167   for (;;) {
 168     while (*mailnames == ':') ++mailnames;
 169     if (!*mailnames) break;
 170
 171     i = str_chr(mailnames,':');
 172
 173     if (substdio_puts(subfdout,"=") == -1) die_write();
 174     if (substdio_put(subfdout,mailnames,i) == -1) die_write();
 175     if (substdio_put(subfdout,uugh.s,uugh.len) == -1) die_write();
 176     if (substdio_puts(subfdout,"::\n") == -1) die_write();
 177
 178     if (*auto_break) {
 179       if (substdio_puts(subfdout,"+") == -1) die_write();
 180       if (substdio_put(subfdout,mailnames,i) == -1) die_write();
 181       if (substdio_put(subfdout,auto_break,1) == -1) die_write();
 182       if (substdio_put(subfdout,uugh.s,uugh.len) == -1) die_write();
 183       if (substdio_puts(subfdout,"-::\n") == -1) die_write();
 184     }
 185
 186     mailnames += i;
 187   }
 188 }
 189
 190 stralloc sub = {0};
 191
 192 void dosubuser()
 193 {
 194   int i;
 195   char *x;
 196   unsigned int xlen;
 197   char *uugh;
 198
 199   x = line.s; xlen = line.len; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 200   if (!stralloc_copyb(&sub,x,i)) die_nomem();
 201   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 202   uugh = constmap(&mapuser,x,i);
 203   if (!uugh) die_user(x,i);
 204   ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
 205
 206   if (substdio_puts(subfdout,"=") == -1) die_write();
 207   if (substdio_put(subfdout,sub.s,sub.len) == -1) die_write();
 208   if (substdio_puts(subfdout,uugh) == -1) die_write();
 209   if (substdio_puts(subfdout,"-:") == -1) die_write();
 210   if (substdio_put(subfdout,x,i) == -1) die_write();
 211   if (substdio_puts(subfdout,":\n") == -1) die_write();
 212
 213   if (*auto_break) {
 214     if (substdio_puts(subfdout,"+") == -1) die_write();
 215     if (substdio_put(subfdout,sub.s,sub.len) == -1) die_write();
 216     if (substdio_put(subfdout,auto_break,1) == -1) die_write();
 217     if (substdio_puts(subfdout,uugh) == -1) die_write();
 218     if (substdio_puts(subfdout,"-:") == -1) die_write();
 219     if (substdio_put(subfdout,x,i) == -1) die_write();
 220     if (substdio_puts(subfdout,"-:\n") == -1) die_write();
 221   }
 222 }
 223
 224 int fd;
 225 substdio ss;
 226 char ssbuf[SUBSTDIO_INSIZE];
 227
 228 void main(argc,argv)
 229 int argc;
 230 char **argv;
 231 {
 232   int opt;
 233   int match;
 234
 235   while ((opt = getopt(argc,argv,"ohHuUc:C")) != opteof)
 236     switch(opt) {
 237       case 'o': homestrategy = 2; break;
 238       case 'h': homestrategy = 1; break;
 239       case 'H': homestrategy = 0; break;
 240       case 'u': flagnoupper = 0; break;
 241       case 'U': flagnoupper = 1; break;
 242       case 'c': *auto_break = *optarg; break;
 243       case 'C': *auto_break = 0; break;
 244       case '?':
 245       default:
 246         _exit(100);
 247     }
 248
 249   if (chdir(auto_qmail) == -1) die_chdir();
 250
 251   /* no need for control_init() */
 252
 253   okincl = control_readfile(&incl,"users/include",0);
 254   if (okincl == -1) die_control();
 255   if (okincl) if (!constmap_init(&mapincl,incl.s,incl.len,0)) die_nomem();
 256
 257   okexcl = control_readfile(&excl,"users/exclude",0);
 258   if (okexcl == -1) die_control();
 259   if (okexcl) if (!constmap_init(&mapexcl,excl.s,excl.len,0)) die_nomem();
 260
 261   okmana = control_readfile(&mana,"users/mailnames",0);
 262   if (okmana == -1) die_control();
 263   if (okmana) if (!constmap_init(&mapmana,mana.s,mana.len,1)) die_nomem();
 264
 265   if (!stralloc_copys(&allusers,"")) die_nomem();
 266
 267   for (;;) {
 268     if (getln(subfdin,&line,&match,'\n') == -1) die_read();
 269     doaccount();
 270     if (!match) break;
 271   }
 272   if (!flagalias) die_alias();
 273
 274   fd = open_read("users/subusers");
 275   if (fd == -1) {
 276     if (errno != error_noent) die_control();
 277   }
 278   else {
 279     substdio_fdbuf(&ss,read,fd,ssbuf,sizeof(ssbuf));
 280
 281     if (!constmap_init(&mapuser,allusers.s,allusers.len,1)) die_nomem();
 282
 283     for (;;) {
 284       if (getln(&ss,&line,&match,'\n') == -1) die_read();
 285       dosubuser();
 286       if (!match) break;
 287     }
 288
 289     close(fd);
 290   }
 291
 292   fd = open_read("users/append");
 293   if (fd == -1) {
 294     if (errno != error_noent) die_control();
 295   }
 296   else {
 297     substdio_fdbuf(&ss,read,fd,ssbuf,sizeof(ssbuf));
 298     for (;;) {
 299       if (getln(&ss,&line,&match,'\n') == -1) die_read();
 300       if (substdio_put(subfdout,line.s,line.len) == -1) die_write();
 301       if (!match) break;
 302     }
 303   }
 304
 305   if (substdio_puts(subfdout,".\n") == -1) die_write();
 306   if (substdio_flush(subfdout) == -1) die_write();
 307   _exit(0);
 308 }