[qmail] / qmail-pw2u.c

#include <sys/types.h>
#include <sys/stat.h>
#include "substdio.h"
#include "readwrite.h"
#include "subfd.h"
#include "sgetopt.h"
#include "control.h"
#include "constmap.h"
#include "stralloc.h"
#include "fmt.h"
#include "str.h"
#include "scan.h"
#include "open.h"
#include "error.h"
#include "getln.h"
#include "auto_break.h"
#include "auto_qmail.h"
#include "auto_usera.h"

void die_chdir()
{
  substdio_putsflush(subfderr,"qmail-pw2u: fatal: unable to chdir\n");
  _exit(111);
}
void die_nomem()
{
  substdio_putsflush(subfderr,"qmail-pw2u: fatal: out of memory\n");
  _exit(111);
}
void die_read()
{
  substdio_putsflush(subfderr,"qmail-pw2u: fatal: unable to read input\n");
  _exit(111);
}
void die_write()
{
  substdio_putsflush(subfderr,"qmail-pw2u: fatal: unable to write output\n");
  _exit(111);
}
void die_control()
{
  substdio_putsflush(subfderr,"qmail-pw2u: fatal: unable to read controls\n");
  _exit(111);
}
void die_alias()
{
  substdio_puts(subfderr,"qmail-pw2u: fatal: unable to find ");
  substdio_puts(subfderr,auto_usera);
  substdio_puts(subfderr," user\n");
  substdio_flush(subfderr);
  _exit(111);
}
void die_home(fn) char *fn;
{
  substdio_puts(subfderr,"qmail-pw2u: fatal: unable to stat ");
  substdio_puts(subfderr,fn);
  substdio_puts(subfderr,"\n");
  substdio_flush(subfderr);
  _exit(111);
}
void die_user(s,len) char *s; unsigned int len;
{
  substdio_puts(subfderr,"qmail-pw2u: fatal: unable to find ");
  substdio_put(subfderr,s,len);
  substdio_puts(subfderr," user for subuser\n");
  substdio_flush(subfderr);
  _exit(111);
}

char *dashcolon = "-:";
int flagalias = 0;
int flagnoupper = 1;
int homestrategy = 2;
/* 2: skip if home does not exist; skip if home is not owned by user */
/* 1: stop if home does not exist; skip if home is not owned by user */
/* 0: don't worry about home */

int okincl; stralloc incl = {0}; struct constmap mapincl;
int okexcl; stralloc excl = {0}; struct constmap mapexcl;
int okmana; stralloc mana = {0}; struct constmap mapmana;

stralloc allusers = {0}; struct constmap mapuser;

stralloc uugh = {0};
stralloc user = {0};
stralloc uidstr = {0};
stralloc gidstr = {0};
stralloc home = {0};
unsigned long uid;

stralloc line = {0};

void doaccount()
{
  struct stat st;
  int i;
  char *mailnames;
  char *x;
  unsigned int xlen;

  if (byte_chr(line.s,line.len,'\0') < line.len) return;

  x = line.s; xlen = line.len; i = byte_chr(x,xlen,':'); if (i == xlen) return;
  if (!stralloc_copyb(&user,x,i)) die_nomem();
  if (!stralloc_0(&user)) die_nomem();
  ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
  ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
  if (!stralloc_copyb(&uidstr,x,i)) die_nomem();
  if (!stralloc_0(&uidstr)) die_nomem();
  scan_ulong(uidstr.s,&uid);
  ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
  if (!stralloc_copyb(&gidstr,x,i)) die_nomem();
  if (!stralloc_0(&gidstr)) die_nomem();
  ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
  ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
  if (!stralloc_copyb(&home,x,i)) die_nomem();
  if (!stralloc_0(&home)) die_nomem();

  if (!uid) return;
  if (flagnoupper)
    for (i = 0;i < user.len;++i)
      if ((user.s[i] >= 'A') && (user.s[i] <= 'Z'))
	return;
  if (okincl)
    if (!constmap(&mapincl,user.s,user.len - 1))
      return;
  if (okexcl)
    if (constmap(&mapexcl,user.s,user.len - 1))
      return;
  if (homestrategy) {
    if (stat(home.s,&st) == -1) {
      if (errno != error_noent) die_home(home.s);
      if (homestrategy == 1) die_home(home.s);
      return;
    }
    if (st.st_uid != uid) return;
  }

  if (!stralloc_copys(&uugh,":")) die_nomem();
  if (!stralloc_cats(&uugh,user.s)) die_nomem();
  if (!stralloc_cats(&uugh,":")) die_nomem();
  if (!stralloc_cats(&uugh,uidstr.s)) die_nomem();
  if (!stralloc_cats(&uugh,":")) die_nomem();
  if (!stralloc_cats(&uugh,gidstr.s)) die_nomem();
  if (!stralloc_cats(&uugh,":")) die_nomem();
  if (!stralloc_cats(&uugh,home.s)) die_nomem();
  if (!stralloc_cats(&uugh,":")) die_nomem();

  /* XXX: avoid recording in allusers unless sub actually needs it */
  if (!stralloc_cats(&allusers,user.s)) die_nomem();
  if (!stralloc_cats(&allusers,":")) die_nomem();
  if (!stralloc_catb(&allusers,uugh.s,uugh.len)) die_nomem();
  if (!stralloc_0(&allusers)) die_nomem();

  if (str_equal(user.s,auto_usera)) {
    if (substdio_puts(subfdout,"+") == -1) die_write();
    if (substdio_put(subfdout,uugh.s,uugh.len) == -1) die_write();
    if (substdio_puts(subfdout,dashcolon) == -1) die_write();
    if (substdio_puts(subfdout,":\n") == -1) die_write();
    flagalias = 1;
  }

  mailnames = 0;
  if (okmana)
    mailnames = constmap(&mapmana,user.s,user.len - 1);
  if (!mailnames)
    mailnames = user.s;

  for (;;) {
    while (*mailnames == ':') ++mailnames;
    if (!*mailnames) break;

    i = str_chr(mailnames,':');

    if (substdio_puts(subfdout,"=") == -1) die_write();
    if (substdio_put(subfdout,mailnames,i) == -1) die_write();
    if (substdio_put(subfdout,uugh.s,uugh.len) == -1) die_write();
    if (substdio_puts(subfdout,"::\n") == -1) die_write();
  
    if (*auto_break) {
      if (substdio_puts(subfdout,"+") == -1) die_write();
      if (substdio_put(subfdout,mailnames,i) == -1) die_write();
      if (substdio_put(subfdout,auto_break,1) == -1) die_write();
      if (substdio_put(subfdout,uugh.s,uugh.len) == -1) die_write();
      if (substdio_puts(subfdout,dashcolon) == -1) die_write();
      if (substdio_puts(subfdout,":\n") == -1) die_write();
    }

    mailnames += i;
  }
}

stralloc sub = {0};

void dosubuser()
{
  int i;
  char *x;
  unsigned int xlen;
  char *uugh;

  x = line.s; xlen = line.len; i = byte_chr(x,xlen,':'); if (i == xlen) return;
  if (!stralloc_copyb(&sub,x,i)) die_nomem();
  ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;
  uugh = constmap(&mapuser,x,i);
  if (!uugh) die_user(x,i);
  ++i; x += i; xlen -= i; i = byte_chr(x,xlen,':'); if (i == xlen) return;

  if (substdio_puts(subfdout,"=") == -1) die_write();
  if (substdio_put(subfdout,sub.s,sub.len) == -1) die_write();
  if (substdio_puts(subfdout,uugh) == -1) die_write();
  if (substdio_puts(subfdout,dashcolon) == -1) die_write();
  if (substdio_put(subfdout,x,i) == -1) die_write();
  if (substdio_puts(subfdout,":\n") == -1) die_write();

  if (*auto_break) {
    if (substdio_puts(subfdout,"+") == -1) die_write();
    if (substdio_put(subfdout,sub.s,sub.len) == -1) die_write();
    if (substdio_put(subfdout,auto_break,1) == -1) die_write();
    if (substdio_puts(subfdout,uugh) == -1) die_write();
    if (substdio_puts(subfdout,dashcolon) == -1) die_write();
    if (substdio_put(subfdout,x,i) == -1) die_write();
    if (substdio_puts(subfdout,"-:\n") == -1) die_write();
  }
}

int fd;
substdio ss;
char ssbuf[SUBSTDIO_INSIZE];

void main(argc,argv)
int argc;
char **argv;
{
  int opt;
  int match;

  while ((opt = getopt(argc,argv,"/ohHuUc:C")) != opteof)
    switch(opt) {
      case '/': dashcolon = "-/:"; break;
      case 'o': homestrategy = 2; break;
      case 'h': homestrategy = 1; break;
      case 'H': homestrategy = 0; break;
      case 'u': flagnoupper = 0; break;
      case 'U': flagnoupper = 1; break;
      case 'c': *auto_break = *optarg; break;
      case 'C': *auto_break = 0; break;
      case '?':
      default:
	_exit(100);
    }

  if (chdir(auto_qmail) == -1) die_chdir();

  /* no need for control_init() */

  okincl = control_readfile(&incl,"users/include",0);
  if (okincl == -1) die_control();
  if (okincl) if (!constmap_init(&mapincl,incl.s,incl.len,0)) die_nomem();

  okexcl = control_readfile(&excl,"users/exclude",0);
  if (okexcl == -1) die_control();
  if (okexcl) if (!constmap_init(&mapexcl,excl.s,excl.len,0)) die_nomem();

  okmana = control_readfile(&mana,"users/mailnames",0);
  if (okmana == -1) die_control();
  if (okmana) if (!constmap_init(&mapmana,mana.s,mana.len,1)) die_nomem();

  if (!stralloc_copys(&allusers,"")) die_nomem();

  for (;;) {
    if (getln(subfdin,&line,&match,'\n') == -1) die_read();
    doaccount();
    if (!match) break;
  }
  if (!flagalias) die_alias();

  fd = open_read("users/subusers");
  if (fd == -1) {
    if (errno != error_noent) die_control();
  }
  else {
    substdio_fdbuf(&ss,read,fd,ssbuf,sizeof(ssbuf));

    if (!constmap_init(&mapuser,allusers.s,allusers.len,1)) die_nomem();

    for (;;) {
      if (getln(&ss,&line,&match,'\n') == -1) die_read();
      dosubuser();
      if (!match) break;
    }

    close(fd);
  }

  fd = open_read("users/append");
  if (fd == -1) {
    if (errno != error_noent) die_control();
  }
  else {
    substdio_fdbuf(&ss,read,fd,ssbuf,sizeof(ssbuf));
    for (;;) {
      if (getln(&ss,&line,&match,'\n') == -1) die_read();
      if (substdio_put(subfdout,line.s,line.len) == -1) die_write();
      if (!match) break;
    }
  }

  if (substdio_puts(subfdout,".\n") == -1) die_write();
  if (substdio_flush(subfdout) == -1) die_write();
  _exit(0);
}
Commit	Line	Data
2117e02e MW	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
212b6f5d	70	char *dashcolon = "-:";
2117e02e MW	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();
212b6f5d MW	158	if (substdio_puts(subfdout,dashcolon) == -1) die_write();
212b6f5d MW	159	if (substdio_puts(subfdout,":\n") == -1) die_write();
2117e02e MW	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();
212b6f5d MW	185	if (substdio_puts(subfdout,dashcolon) == -1) die_write();
212b6f5d MW	186	if (substdio_puts(subfdout,":\n") == -1) die_write();
2117e02e MW	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();
212b6f5d	212	if (substdio_puts(subfdout,dashcolon) == -1) die_write();
2117e02e MW	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();
212b6f5d	221	if (substdio_puts(subfdout,dashcolon) == -1) die_write();
2117e02e MW	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
212b6f5d	238	while ((opt = getopt(argc,argv,"/ohHuUc:C")) != opteof)
2117e02e	239	switch(opt) {
212b6f5d	240	case '/': dashcolon = "-/:"; break;
2117e02e MW	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	}