[moira.git] / clients / addusr / addusr.c

/* $Id$
 *
 * Program to add users en masse to the moira database
 *
 * by Mark Rosenstein, July 1992.
 *
 * Copyright (C) 1992-1998 by the Massachusetts Institute of Technology.
 * For copying and distribution information, please see the file
 * <mit-copyright.h>.
 */

#include <mit-copyright.h>
#include <moira.h>
#include <moira_site.h>
#include <mrclient.h>

#include <ctype.h>
#include <errno.h>
#include <stdio.h>
#include <string.h>

RCSID("$Header$");

#ifdef ATHENA
#define DEFAULT_SHELL "/bin/athena/tcsh"
#else
#define DEFAULT_SHELL "/bin/csh"
#endif

#define DEFAULT_WINCONSOLESHELL "cmd"

/* flags from command line */
char *class, *comment, *status_str, *shell, *winconsoleshell, *filename;
int reg_only, reg, verbose, nodupcheck, securereg;

/* argument parsing macro */
#define argis(a, b) (!strcmp(*arg + 1, a) || !strcmp(*arg + 1, b))

char *whoami;
int duplicate, errors;

void usage(char **argv);
int usercheck(int argc, char **argv, void *qargv);
int get_uid(int argc, char **argv, void *qargv);

int main(int argc, char **argv)
{
  int status, lineno;
  char **arg = argv, *qargv[U_END];
  char *p, *first, *middle, *last, *id, *login, *server;
  char buf[BUFSIZ];
  FILE *input;

  /* clear all flags & lists */
  reg_only = reg = verbose = lineno = nodupcheck = errors = securereg = 0;
  server = NULL;
  filename = "-";
  shell = DEFAULT_SHELL;
  winconsoleshell = DEFAULT_WINCONSOLESHELL;
  class = "TEMP";
  comment = "";
  status_str = "0";

  whoami = argv[0];

  /* parse args */
  while (++arg - argv < argc)
    {
      if  (**arg == '-')
	{
	  if (argis("c", "class"))
	    {
	      if (arg - argv < argc - 1)
		{
		  ++arg;
		  class = *arg;
		}
	      else
		usage(argv);
	    }
	  else if (argis("C", "comment"))
	    {
	      if (arg - argv < argc - 1)
		{
		  ++arg;
		  comment = *arg;
		}
	      else
		usage(argv);
	    }
	  else if (argis("s", "status"))
	    {
	      if (arg - argv < argc - 1)
		{
		  ++arg;
		  status_str = *arg;
		}
	      else
		usage(argv);
	    }
	  else if (argis("h", "shell"))
	    {
	      if (arg - argv < argc - 1)
		{
		  ++arg;
		  shell = *arg;
		}
	      else
		usage(argv);
	    }
	  else if (argis("w", "winshell"))
	    {
	      if (arg - argv < argc - 1)
		{
		  ++arg;
		  winconsoleshell = *arg;
		}
	      else 
		usage(argv);
	    }
	  else if (argis("6", "secure"))
	    securereg++;
	  else if (argis("r", "reg_only"))
	    reg_only++;
	  else if (argis("R", "register"))
	    reg++;
	  else if (argis("f", "file"))
	    {
	      if (arg - argv < argc - 1)
		{
		  ++arg;
		  filename = *arg;
		}
	      else
		usage(argv);
	    }
	  else if (argis("v", "verbose"))
	    verbose++;
	  else if (argis("d", "nodupcheck"))
	    nodupcheck++;
	  else if (argis("S", "server") || argis("db", "database"))
	    {
	      if (arg - argv < argc - 1)
		{
		  ++arg;
		  server = *arg;
		}
	      else
		usage(argv);
	    }
	  else
	    usage(argv);
	}
      else
	usage(argv);
    }

  if (!strcmp(filename, "-"))
    input = stdin;
  else
    {
      input = fopen(filename, "r");
      if (!input)
	{
	  com_err(whoami, errno, "opening input file %s", filename);
	  exit(2);
	}
    }

  /* fire up Moira */
  if (mrcl_connect(server, "addusr", 3, 1) != MRCL_SUCCESS)
    exit(2);

  qargv[U_NAME] = UNIQUE_LOGIN;
  qargv[U_UID] = UNIQUE_UID;
  qargv[U_SHELL] = shell;
  qargv[U_WINCONSOLESHELL] = winconsoleshell;
  qargv[U_STATE] = status_str;
  qargv[U_CLASS] = class;
  qargv[U_COMMENT] = comment;
  qargv[U_SIGNATURE] = "";
  qargv[U_SECURE] = securereg ? "1" : "0";
  while (fgets(buf, BUFSIZ, input))
    {
      /* throw away terminating newline */
      p = &buf[strlen(buf) - 1];
      if (*p == '\n')
	*p = '\0';
      lineno++;
      if (strlen(buf) == 0)
	continue;
      /* Last name is first thing on line */
      last = buf;
      /* First name follows a comma */
      p = strchr(last, ',');
      if (!p)
	{
	  com_err(whoami, MR_BAD_CHAR, "Missing comma on line %d", lineno);
	  errors++;
	  continue;
	}
      *p++ = '\0';
      first = strtrim(p);
      while (*p)		/* find end-of-line */
	p++;
      if (reg_only || reg)
	{
	  while (!isspace(*p))
	    p--;
	  if (p <= first)
	    {
	      com_err(whoami, 0, "Missing login on line %d", lineno);
	      errors++;
	      continue;
	    }
	  login = strtrim(&p[1]);
	  *p-- = '\0';
	}
      else
	{
	  /* ID is last thing on line */
	  p--;
	}
      while (isspace(*p))
	p--;
      while (!isspace(*p))
	p--;
      if (p <= first)
	{
	  com_err(whoami, 0, "Missing ID on line %d", lineno);
	  errors++;
	  continue;
	}
      id = &p[1];
      *p-- = '\0';
      /* If something between first name & ID, it's middle name */
      while (isspace(*p))
	p--;
      while (!isspace(*p))
	p--;
      if (p <= first)
	middle = "";
      else
	{
	  middle = &p[1];
	  *p = '\0';
	}
      qargv[U_FIRST] = strtrim(first);
      qargv[U_MIDDLE] = strtrim(middle);
      qargv[U_LAST] = strtrim(last);
      qargv[U_MITID] = strtrim(id);
      FixCase(qargv[U_FIRST]);
      FixCase(qargv[U_MIDDLE]);
      FixCase(qargv[U_LAST]);
      RemoveHyphens(qargv[U_MITID]);
      if (!reg_only)
	{
	  if (!nodupcheck)
	    {
	      char *dargv[2];

	      dargv[0] = qargv[U_FIRST];
	      dargv[1] = qargv[U_LAST];
	      duplicate = 0;
	      status = mr_query("get_user_account_by_name", 2, dargv,
				usercheck, qargv);
	      if (status && status != MR_NO_MATCH)
		{
		  com_err(whoami, status,
			  "checking to see if user %s %s already exists",
			  qargv[U_FIRST], qargv[U_LAST]);
		  com_err(whoami, 0, "NOT ADDING USER");
		  errors++;
		  continue;
		}
	      if (duplicate > 0)
		{
		  com_err(whoami, MR_EXISTS, "user %s %s already exists",
			  qargv[U_FIRST], qargv[U_LAST]);
		  com_err(whoami, 0, "NOT ADDING USER");
		  errors++;
		  continue;
		}
	      else if (duplicate < 0)
		{
		  com_err(whoami, MR_EXISTS,
			  "user %s %s already exists with different ID number",
			  qargv[U_FIRST], qargv[U_LAST]);
		  com_err(whoami, 0, "ADDING user anyway");
		}
	    }
	  status = mr_query("add_user_account", U_SECURE + 1, qargv,
			    NULL, NULL);
	  if (status)
	    {
	      com_err(whoami, status, "adding user %s %s", first, last);
	      errors++;
	    }
	  else if (verbose)
	    {
	      printf("Added user %s %s %s (%s)\n", qargv[U_FIRST],
		     qargv[U_MIDDLE], qargv[U_LAST], qargv[U_MITID]);
	    }
	}
      if (reg || reg_only)
	{
	  char *gargv[2], *rargv[3], uid[10];

	  uid[0] = '\0';
	  gargv[0] = qargv[U_FIRST];
	  gargv[1] = qargv[U_LAST];
	  status = mr_query("get_user_account_by_name", 2, gargv,
			    get_uid, &uid);
	  if (status)
	    {
	      com_err(whoami, status, "while looking up uid for %s %s",
		      qargv[U_FIRST], qargv[U_LAST]);
	      errors++;
	      continue;
	    }

	  rargv[0] = uid;
	  rargv[1] = login;
	  rargv[2] = "IMAP";

	  status = mr_query("register_user", 3, rargv, NULL, NULL);
	  if (status)
	    {
	      com_err(whoami, status, "while registering (login) %s %s",
		      qargv[U_FIRST], qargv[U_LAST]);
	      errors++;
	      continue;
	    }
	  else if (verbose)
	    {
	      printf("Registered user %s %s as %s\n", qargv[U_FIRST],
		     qargv[U_LAST], login);
	    }
	}
    }

  exit(errors);
}


void usage(char **argv)
{
  fprintf(stderr, "Usage: %s [options]\n", argv[0]);
  fprintf(stderr, "Options are\n");
  fprintf(stderr, "   -f | -file filename (default STDIN)\n");
  fprintf(stderr, "   -c | -class class (default TEMP)\n");
  fprintf(stderr, "   -C | -comment \"comment\" (default \"\")\n");
  fprintf(stderr, "   -s | -status status (default 0)\n");
  fprintf(stderr, "   -h | -shell shell (default %s)\n", DEFAULT_SHELL);
  fprintf(stderr, "   -w | -winshell windows console shell (default %s)\n",
	  DEFAULT_WINCONSOLESHELL);
  fprintf(stderr, "   -r | -reg_only\n");
  fprintf(stderr, "   -R | -register (and add to database)\n");
  fprintf(stderr, "   -v | -verbose\n");
  fprintf(stderr, "   -d | -nodupcheck (don't check for duplicates)\n");
  fprintf(stderr, "   -db | -database host:port\n");
  exit(1);
}


/* query callback routine to check for duplicate users */

int usercheck(int argc, char **argv, void *qargv)
{
  if (!strcmp(argv[U_MITID], ((char **)qargv)[U_MITID]))
    duplicate++;
  else
    duplicate--;

  return MR_CONT;
}

/* query callback to get uid of a just-added account */
int get_uid(int argc, char **argv, void *uidv)
{
  char *uid = uidv;

  if (uid[0] == '\0')
    strcpy(uid, argv[U_UID]);
  else
    {
      if (!strcmp(argv[U_MODWITH], "addusr"))
	strcpy(uid, argv[U_UID]);
    }

  return MR_CONT;
}
Commit	Line	Data
c441a31a	1	/* $Id$
93d0b3bf	2	*
7ac48069	3	* Program to add users en masse to the moira database
93d0b3bf	4	*
	5	* by Mark Rosenstein, July 1992.
	6	*
7ac48069	7	* Copyright (C) 1992-1998 by the Massachusetts Institute of Technology.
93d0b3bf	8	* For copying and distribution information, please see the file
	9	* <mit-copyright.h>.
	10	*/
	11
	12	#include <mit-copyright.h>
93d0b3bf	13	#include <moira.h>
93d0b3bf	14	#include <moira_site.h>
2a12a5ec	15	#include <mrclient.h>
93d0b3bf	16
7ac48069	17	#include <ctype.h>
	18	#include <errno.h>
	19	#include <stdio.h>
	20	#include <string.h>
	21
	22	RCSID("$Header$");
93d0b3bf	23
17490e42	24	#ifdef ATHENA
	25	#define DEFAULT_SHELL "/bin/athena/tcsh"
	26	#else
	27	#define DEFAULT_SHELL "/bin/csh"
	28	#endif
	29
0978529a	30	#define DEFAULT_WINCONSOLESHELL "cmd"
0978529a	31
93d0b3bf	32	/* flags from command line */
0978529a	33	char class, comment, status_str, shell, winconsoleshell, filename;
c0c29294	34	int reg_only, reg, verbose, nodupcheck, securereg;
93d0b3bf	35
93d0b3bf	36	/* argument parsing macro */
5eaef520	37	#define argis(a, b) (!strcmp(arg + 1, a) \|\| !strcmp(arg + 1, b))
93d0b3bf	38
	39	char *whoami;
	40	int duplicate, errors;
	41
7ac48069	42	void usage(char **argv);
7ac48069	43	int usercheck(int argc, char *argv, void qargv);
aa7d79ff	44	int get_uid(int argc, char *argv, void qargv);
93d0b3bf	45
5eaef520	46	int main(int argc, char **argv)
93d0b3bf	47	{
5eaef520	48	int status, lineno;
5eaef520	49	char *arg = argv, qargv[U_END];
2a12a5ec	50	char p, first, middle, last, id, login, *server;
533bacb3	51	char buf[BUFSIZ];
5eaef520	52	FILE *input;
93d0b3bf	53
5eaef520	54	/* clear all flags & lists */
c0c29294	55	reg_only = reg = verbose = lineno = nodupcheck = errors = securereg = 0;
5eaef520	56	server = NULL;
	57	filename = "-";
	58	shell = DEFAULT_SHELL;
0978529a	59	winconsoleshell = DEFAULT_WINCONSOLESHELL;
5eaef520	60	class = "TEMP";
	61	comment = "";
	62	status_str = "0";
93d0b3bf	63
5eaef520	64	whoami = argv[0];
93d0b3bf	65
5eaef520	66	/* parse args */
	67	while (++arg - argv < argc)
	68	{
	69	if (**arg == '-')
93d0b3bf	70	{
5eaef520	71	if (argis("c", "class"))
	72	{
	73	if (arg - argv < argc - 1)
	74	{
	75	++arg;
	76	class = *arg;
	77	}
	78	else
	79	usage(argv);
	80	}
	81	else if (argis("C", "comment"))
	82	{
	83	if (arg - argv < argc - 1)
	84	{
	85	++arg;
	86	comment = *arg;
	87	}
	88	else
	89	usage(argv);
	90	}
	91	else if (argis("s", "status"))
	92	{
	93	if (arg - argv < argc - 1)
	94	{
	95	++arg;
	96	status_str = *arg;
	97	}
	98	else
	99	usage(argv);
	100	}
	101	else if (argis("h", "shell"))
	102	{
	103	if (arg - argv < argc - 1)
	104	{
	105	++arg;
	106	shell = *arg;
	107	}
	108	else
	109	usage(argv);
	110	}
0978529a	111	else if (argis("w", "winshell"))
	112	{
	113	if (arg - argv < argc - 1)
	114	{
	115	++arg;
	116	winconsoleshell = *arg;
	117	}
	118	else
	119	usage(argv);
	120	}
c0c29294	121	else if (argis("6", "secure"))
c0c29294	122	securereg++;
5eaef520	123	else if (argis("r", "reg_only"))
	124	reg_only++;
	125	else if (argis("R", "register"))
	126	reg++;
	127	else if (argis("f", "file"))
	128	{
	129	if (arg - argv < argc - 1)
	130	{
	131	++arg;
	132	filename = *arg;
	133	}
	134	else
	135	usage(argv);
	136	}
	137	else if (argis("v", "verbose"))
	138	verbose++;
	139	else if (argis("d", "nodupcheck"))
	140	nodupcheck++;
	141	else if (argis("S", "server") \|\| argis("db", "database"))
	142	{
	143	if (arg - argv < argc - 1)
	144	{
	145	++arg;
	146	server = *arg;
	147	}
	148	else
93d0b3bf	149	usage(argv);
5eaef520	150	}
	151	else
	152	usage(argv);
	153	}
	154	else
	155	usage(argv);
93d0b3bf	156	}
93d0b3bf	157
5eaef520	158	if (!strcmp(filename, "-"))
	159	input = stdin;
	160	else
	161	{
	162	input = fopen(filename, "r");
	163	if (!input)
	164	{
	165	com_err(whoami, errno, "opening input file %s", filename);
	166	exit(2);
93d0b3bf	167	}
	168	}
	169
aa7d79ff	170	/* fire up Moira */
0978529a	171	if (mrcl_connect(server, "addusr", 3, 1) != MRCL_SUCCESS)
2a12a5ec	172	exit(2);
93d0b3bf	173
5eaef520	174	qargv[U_NAME] = UNIQUE_LOGIN;
	175	qargv[U_UID] = UNIQUE_UID;
	176	qargv[U_SHELL] = shell;
0978529a	177	qargv[U_WINCONSOLESHELL] = winconsoleshell;
5eaef520	178	qargv[U_STATE] = status_str;
	179	qargv[U_CLASS] = class;
	180	qargv[U_COMMENT] = comment;
	181	qargv[U_SIGNATURE] = "";
c0c29294	182	qargv[U_SECURE] = securereg ? "1" : "0";
5eaef520	183	while (fgets(buf, BUFSIZ, input))
	184	{
	185	/* throw away terminating newline */
	186	p = &buf[strlen(buf) - 1];
	187	if (*p == '\n')
	188	*p = '\0';
	189	lineno++;
	190	if (strlen(buf) == 0)
	191	continue;
	192	/* Last name is first thing on line */
	193	last = buf;
	194	/* First name follows a comma */
	195	p = strchr(last, ',');
	196	if (!p)
	197	{
	198	com_err(whoami, MR_BAD_CHAR, "Missing comma on line %d", lineno);
	199	errors++;
93d0b3bf	200	continue;
93d0b3bf	201	}
5eaef520	202	*p++ = '\0';
2afc856a	203	first = strtrim(p);
5eaef520	204	while (p) / find end-of-line */
	205	p++;
	206	if (reg_only \|\| reg)
	207	{
5eaef520	208	while (!isspace(*p))
93d0b3bf	209	p--;
5eaef520	210	if (p <= first)
	211	{
	212	com_err(whoami, 0, "Missing login on line %d", lineno);
	213	errors++;
	214	continue;
	215	}
	216	login = strtrim(&p[1]);
	217	*p-- = '\0';
93d0b3bf	218	}
5eaef520	219	else
	220	{
	221	/* ID is last thing on line */
	222	p--;
93d0b3bf	223	}
5eaef520	224	while (isspace(*p))
	225	p--;
	226	while (!isspace(*p))
	227	p--;
	228	if (p <= first)
	229	{
	230	com_err(whoami, 0, "Missing ID on line %d", lineno);
	231	errors++;
	232	continue;
93d0b3bf	233	}
5eaef520	234	id = &p[1];
	235	*p-- = '\0';
	236	/* If something between first name & ID, it's middle name */
	237	while (isspace(*p))
	238	p--;
	239	while (!isspace(*p))
	240	p--;
	241	if (p <= first)
	242	middle = "";
	243	else
	244	{
	245	middle = &p[1];
	246	*p = '\0';
	247	}
	248	qargv[U_FIRST] = strtrim(first);
	249	qargv[U_MIDDLE] = strtrim(middle);
	250	qargv[U_LAST] = strtrim(last);
	251	qargv[U_MITID] = strtrim(id);
	252	FixCase(qargv[U_FIRST]);
	253	FixCase(qargv[U_MIDDLE]);
	254	FixCase(qargv[U_LAST]);
	255	RemoveHyphens(qargv[U_MITID]);
	256	if (!reg_only)
	257	{
	258	if (!nodupcheck)
	259	{
	260	char *dargv[2];
93d0b3bf	261
5eaef520	262	dargv[0] = qargv[U_FIRST];
	263	dargv[1] = qargv[U_LAST];
	264	duplicate = 0;
	265	status = mr_query("get_user_account_by_name", 2, dargv,
7ac48069	266	usercheck, qargv);
5eaef520	267	if (status && status != MR_NO_MATCH)
	268	{
	269	com_err(whoami, status,
	270	"checking to see if user %s %s already exists",
9e1ff037	271	qargv[U_FIRST], qargv[U_LAST]);
5eaef520	272	com_err(whoami, 0, "NOT ADDING USER");
	273	errors++;
	274	continue;
	275	}
	276	if (duplicate > 0)
	277	{
	278	com_err(whoami, MR_EXISTS, "user %s %s already exists",
9e1ff037	279	qargv[U_FIRST], qargv[U_LAST]);
5eaef520	280	com_err(whoami, 0, "NOT ADDING USER");
	281	errors++;
	282	continue;
93d0b3bf	283	}
5eaef520	284	else if (duplicate < 0)
	285	{
	286	com_err(whoami, MR_EXISTS,
	287	"user %s %s already exists with different ID number",
	288	qargv[U_FIRST], qargv[U_LAST]);
	289	com_err(whoami, 0, "ADDING user anyway");
93d0b3bf	290	}
93d0b3bf	291	}
5eaef520	292	status = mr_query("add_user_account", U_SECURE + 1, qargv,
7ac48069	293	NULL, NULL);
5eaef520	294	if (status)
	295	{
	296	com_err(whoami, status, "adding user %s %s", first, last);
	297	errors++;
	298	}
	299	else if (verbose)
	300	{
93d0b3bf	301	printf("Added user %s %s %s (%s)\n", qargv[U_FIRST],
93d0b3bf	302	qargv[U_MIDDLE], qargv[U_LAST], qargv[U_MITID]);
5eaef520	303	}
93d0b3bf	304	}
5eaef520	305	if (reg \|\| reg_only)
5eaef520	306	{
aa7d79ff	307	char gargv[2], rargv[3], uid[10];
	308
	309	uid[0] = '\0';
	310	gargv[0] = qargv[U_FIRST];
	311	gargv[1] = qargv[U_LAST];
	312	status = mr_query("get_user_account_by_name", 2, gargv,
	313	get_uid, &uid);
	314	if (status)
	315	{
	316	com_err(whoami, status, "while looking up uid for %s %s",
	317	qargv[U_FIRST], qargv[U_LAST]);
	318	errors++;
	319	continue;
	320	}
8c0ce358	321
aa7d79ff	322	rargv[0] = uid;
8c0ce358	323	rargv[1] = login;
17540151	324	rargv[2] = "IMAP";
8c0ce358	325
aa7d79ff	326	status = mr_query("register_user", 3, rargv, NULL, NULL);
5eaef520	327	if (status)
	328	{
	329	com_err(whoami, status, "while registering (login) %s %s",
	330	qargv[U_FIRST], qargv[U_LAST]);
	331	errors++;
	332	continue;
	333	}
	334	else if (verbose)
	335	{
93d0b3bf	336	printf("Registered user %s %s as %s\n", qargv[U_FIRST],
93d0b3bf	337	qargv[U_LAST], login);
5eaef520	338	}
93d0b3bf	339	}
	340	}
	341
5eaef520	342	exit(errors);
93d0b3bf	343	}
	344
	345
7ac48069	346	void usage(char **argv)
93d0b3bf	347	{
5eaef520	348	fprintf(stderr, "Usage: %s [options]\n", argv[0]);
	349	fprintf(stderr, "Options are\n");
	350	fprintf(stderr, " -f \| -file filename (default STDIN)\n");
	351	fprintf(stderr, " -c \| -class class (default TEMP)\n");
	352	fprintf(stderr, " -C \| -comment \"comment\" (default \"\")\n");
	353	fprintf(stderr, " -s \| -status status (default 0)\n");
	354	fprintf(stderr, " -h \| -shell shell (default %s)\n", DEFAULT_SHELL);
0978529a	355	fprintf(stderr, " -w \| -winshell windows console shell (default %s)\n",
0978529a	356	DEFAULT_WINCONSOLESHELL);
5eaef520	357	fprintf(stderr, " -r \| -reg_only\n");
	358	fprintf(stderr, " -R \| -register (and add to database)\n");
	359	fprintf(stderr, " -v \| -verbose\n");
	360	fprintf(stderr, " -d \| -nodupcheck (don't check for duplicates)\n");
	361	fprintf(stderr, " -db \| -database host:port\n");
	362	exit(1);
93d0b3bf	363	}
	364
	365
93d0b3bf	366	/* query callback routine to check for duplicate users */
93d0b3bf	367
7ac48069	368	int usercheck(int argc, char *argv, void qargv)
93d0b3bf	369	{
7ac48069	370	if (!strcmp(argv[U_MITID], ((char **)qargv)[U_MITID]))
5eaef520	371	duplicate++;
	372	else
	373	duplicate--;
7ac48069	374
7ac48069	375	return MR_CONT;
93d0b3bf	376	}
aa7d79ff	377
	378	/* query callback to get uid of a just-added account */
	379	int get_uid(int argc, char *argv, void uidv)
	380	{
	381	char *uid = uidv;
	382
	383	if (uid[0] == '\0')
	384	strcpy(uid, argv[U_UID]);
	385	else
	386	{
	387	if (!strcmp(argv[U_MODWITH], "addusr"))
	388	strcpy(uid, argv[U_UID]);
	389	}
	390
	391	return MR_CONT;
	392	}