Make a half-hearted attempt at returning a useful exit status, instead

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

/* $Header$
 *
 * Command line oriented Moira List tool.
 *
 * by Mark Rosenstein, September 1988.
 *
 * Copyright 1989 by the Massachusetts Institute of Technology.
 *
 * (c) Copyright 1988 by the Massachusetts Institute of Technology.
 * For copying and distribution information, please see the file
 * <mit-copyright.h>.
 */

/* ### Aren't there a lot of sq abstraction barrier violations here?
   Do we need to improve the support for queue operations? */

#include <mit-copyright.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <string.h>
#include <moira.h>
#include <moira_site.h>

#ifndef LINT
static char blanche_rcsid[] = "$Header$";
#endif


struct member {
    int type;
    char *name;
};

/* It is important to membercmp that M_USER < M_LIST < M_STRING */
#define M_ANY		0
#define M_USER		1
#define M_LIST		2
#define M_STRING	3
#define M_KERBEROS	4

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

/* flags from command line */
int infoflg, verbose, syncflg, memberflg, recursflg, debugflg, noauth;
int showusers, showstrings, showkerberos, showlists;

/* various member lists */
struct save_queue *addlist, *dellist, *memberlist, *synclist;

char *listname, *whoami;

#ifndef POSIX
extern int errno;
#endif

int show_list_info(), show_list_count(), get_list_members(), scream();
int show_list_members(), membercmp();
struct member *parse_member();



main(argc, argv)
int argc;
char **argv;
{
    int status, success;
    char **arg = argv;
    char *membervec[3], *motd;
    struct member *memberstruct;
    char *server = NULL, *p;

    /* clear all flags & lists */
    infoflg = verbose = syncflg = memberflg = debugflg = recursflg = 0;
    noauth = showusers = showstrings = showkerberos = showlists = 0;
    listname = NULL;
    addlist = sq_create();
    dellist = sq_create();
    memberlist = sq_create();
    synclist = sq_create();
    whoami = argv[0];

    success = 1;

    /* parse args, building addlist, dellist, & synclist */
    while (++arg - argv < argc) {
	if  (**arg == '-')
	{
	    if (argis("m", "members"))
		memberflg++;
	    else if (argis("u", "users"))
		showusers++;
	    else if (argis("s", "strings"))
		showstrings++;
	    else if (argis("l", "lists"))
		showlists++;
	    else if (argis("k", "kerberos"))
	      	showkerberos++;
	    else if (argis("D", "debug"))
		debugflg++;
	    else if (argis("i","info"))
	      infoflg++;
	    else if (argis("n","noauth"))
	      noauth++;
	    else if (argis("v","verbose"))
	      verbose++;
	    else if (argis("r","recursive"))
	      recursflg++;
	    else if (argis("S","server") || argis("db","database"))
		if (arg - argv < argc - 1) {
		    ++arg;
		    server = *arg;
		} else
		    usage(argv);
	    else if (argis("a","add"))
		if (arg - argv < argc - 1) {
		    ++arg;
		    if (memberstruct = parse_member(*arg))
			sq_save_data(addlist, memberstruct);
		} else
		    usage(argv);
	    else if (argis("al","addlist"))
		if (arg - argv < argc - 1) {
		    ++arg;
		    get_members_from_file(*arg, addlist);
		} else
		  usage(argv);
	    else if (argis("d","delete"))
		if (arg - argv < argc - 1) {
		    ++arg;
		    if (memberstruct = parse_member(*arg))
			sq_save_data(dellist, memberstruct);
		} else
		    usage(argv);
	    else if (argis("dl","deletelist"))
		if (arg - argv < argc - 1) {
		    ++arg;
		    get_members_from_file(*arg, dellist);
		} else
		  usage(argv);
	    else if (argis("f","file"))
		if (arg - argv < argc - 1) {
		    syncflg++;
		    ++arg;
		    get_members_from_file(*arg, synclist);
		} else
		  usage(argv);
	    else
		usage(argv);
	}
	else if (listname == NULL)
	  listname = *arg;
	else
	  usage(argv);
    }
    if (listname == NULL)
      usage(argv);

    /* if no other options specified, turn on list members flag */
    if (!(infoflg || syncflg ||
	  addlist->q_next != addlist || dellist->q_next != dellist))
      memberflg++;

    /* If none of {users,strings,lists,kerberos} specified, turn them all on */
    if (!(showusers || showstrings || showlists || showkerberos))
      showusers = showstrings = showlists = showkerberos = 1;

    /* fire up Moira */
    if (status = mr_connect(server)) {
	com_err(whoami, status, "unable to connect to the Moira server");
	exit(2);
    }
    if ( status = mr_motd(&motd) ) {
        com_err(whoami, status, "unable to check server status");
	exit(2);
    }
    if (motd) {
	fprintf(stderr, "The Moira server is currently unavailable:\n%s\n", motd);
	mr_disconnect();
	exit(2);
    }

    if (!noauth && (status = mr_auth("blanche"))) {
	if (status == MR_USER_AUTH)
	  com_err(whoami, status, "");
	else {
	    com_err(whoami, status, "unable to authenticate to Moira");
	    com_err(whoami, 0,
		    " Try the -noauth flag if you don't need authentication");
	    exit(2);
	}
    }

    /* display list info if requested to */
    if (infoflg) {
	status = mr_query("get_list_info", 1, &listname, show_list_info,NULL);
	if (status)
	  com_err(whoami, status, "while getting list information");
	if (verbose && !memberflg) {
	    status = mr_query("count_members_of_list", 1, &listname,
			       show_list_count, NULL);
	    if (status)
	      com_err(whoami, status, "while getting list count");
	}
    }

    /* if we're synchronizing to a file, we need to:
     *  get the current members of the list
     *	for each member of the sync file
     *	   if they are on the list, remove them from the in-memory copy
     *	   if they're not on the list, add them to add-list
     *	if anyone is left on the in-memory copy, put them on the delete-list
     * lastly, reset memberlist so we can use it again later
     */
    if (syncflg) {
	status = mr_query("get_members_of_list", 1, &listname,
			   get_list_members, (char *)memberlist);
	if (status) {
	    com_err(whoami, status, "getting members of list %s", listname);
	    exit (2);
	}
	while (sq_get_data(synclist, &memberstruct)) {
	    struct save_queue *q;
	    int removed = 0;

	    for (q = memberlist->q_next; q != memberlist; q = q->q_next) {
		if (membercmp(q->q_data, memberstruct) == 0) {
		    q->q_prev->q_next = q->q_next;
		    q->q_next->q_prev = q->q_prev;
		    removed++;
		    break;
		}
	    }
	    if (!removed)
	      sq_save_data(addlist, memberstruct);
	}
	while (sq_get_data(memberlist, &memberstruct))
	  sq_save_data(dellist, memberstruct);
	sq_destroy(memberlist);
	memberlist = sq_create();
    }

    /* Process the add list */
    while (sq_get_data(addlist, &memberstruct)) {
	/* canonicalize string if necessary */
	if (memberstruct->type == M_STRING &&
	    (p = strchr(memberstruct->name, '@'))) {
	    char *host = canonicalize_hostname(strsave(++p));
	    static char **mailhubs = NULL;
	    char *argv[4];
	    int i, collect();

	    if (!mailhubs) {
		argv[0] = "mailhub";
		argv[1] = "TYPE";
		argv[2] = "*";
		mailhubs = (char **)malloc(sizeof(char *));
		mailhubs[0] = NULL;
		status = mr_query("get_alias", 3, argv, collect,
				   (char *)&mailhubs);
		if (status != MR_SUCCESS && status != MR_NO_MATCH) {
		    com_err(whoami, status,
			    " while reading list of MAILHUB servers");
		    mailhubs[0] = NULL;
		}
	    }
	    for (i = 0; p = mailhubs[i]; i++) {
		if (!strcasecmp(p, host)) {
		    host = strsave(memberstruct->name);
		    *(strchr(memberstruct->name, '@')) = 0;
		    memberstruct->type = M_ANY;
		    fprintf(stderr, "Warning: \"STRING:%s\" converted to \"%s\" because it is a local name.\n",
			    host, memberstruct->name);
		    break;
		}
	    }
	    free(host);
	}
	/* now continue adding member */
	membervec[0] = listname;
	membervec[2] = memberstruct->name;
	if (verbose) {
	    printf("Adding member ");
	    show_list_member(memberstruct);
	}
	switch (memberstruct->type) {
	case M_ANY:
	case M_USER:
	    membervec[1] = "USER";
	    status = mr_query("add_member_to_list", 3, membervec, scream, NULL);
	    if (status == MR_SUCCESS)
	      break;
	    else if (status != MR_USER || memberstruct->type != M_ANY) {
		com_err(whoami, status, "while adding member %s to %s",
			memberstruct->name, listname);
		success = 0;
		break;
	    }
	case M_LIST:
	    membervec[1] = "LIST";
	    status = mr_query("add_member_to_list", 3, membervec,
			       scream, NULL);
	    if (status == MR_SUCCESS) {
	        if (!strcmp(membervec[0], getenv("USER"))) {
		    fprintf(stderr, "\nWARNING: \"LIST:%s\" was just added to list \"%s\".\n",
			    membervec[2], membervec[0]);
		    fprintf(stderr, "If you meant to add yourself to the list \"%s\", type:\n", membervec[2]);
		    fprintf(stderr, "\tblanche %s -d %s\t(to undo this)\n",
			    membervec[0], membervec[2]);
		    fprintf(stderr, "\tblanche %s -a %s\t(to add yourself to that list)\n",
			    membervec[2], membervec[0]);
		}
	        break;
	    } else if (status != MR_LIST || memberstruct->type != M_ANY) {
		com_err(whoami, status, "while adding member %s to %s",
			memberstruct->name, listname);
		success = 0;
		break;
	    }
	case M_STRING:
	    membervec[1] = "STRING";
	    status = mr_query("add_member_to_list", 3, membervec,
			       scream, NULL);
	    if (status != MR_SUCCESS) {
	      com_err(whoami, status, "while adding member %s to %s",
		      memberstruct->name, listname);
	      success = 0;
	    }
	    else if (!strchr(memberstruct->name, '@') &&
		     !strchr(memberstruct->name, '!') &&
		     !strchr(memberstruct->name, '%')) {
		fprintf(stderr, "\nWARNING: \"STRING:%s\" was just added to list \"%s\".\n",
			memberstruct->name, listname);
		fprintf(stderr, "\tIf %s is a mailing list, this may cause it to stop working.\n", listname);
		fprintf(stderr, "\tYou should consider removing \"STRING:%s\" from the list.\n", memberstruct->name);
  	    }
	    break;
	case M_KERBEROS:
	    membervec[1] = "KERBEROS";
	    status = mr_query("add_member_to_list", 3, membervec,
			       scream, NULL);
	    if (status != MR_SUCCESS) {
	      com_err(whoami, status, "while adding member %s to %s",
		      memberstruct->name, listname);
	      success = 0;
	    }
	}
    }

    /* Process the delete list */
    while (sq_get_data(dellist, &memberstruct)) {
	membervec[0] = listname;
	membervec[2] = memberstruct->name;
	if (verbose) {
	    printf("Deleting member ");
	    show_list_member(memberstruct);
	}
	switch (memberstruct->type) {
	case M_ANY:
	case M_USER:
	    membervec[1] = "USER";
	    status = mr_query("delete_member_from_list", 3, membervec,
			       scream, NULL);
	    if (status == MR_SUCCESS)
	      break;
	    else if ((status != MR_USER && status != MR_NO_MATCH) ||
		     memberstruct->type != M_ANY) {
		com_err(whoami, status, "while deleting member %s from %s",
			memberstruct->name, listname);
		success = 0;
		break;
	    }
	case M_LIST:
	    membervec[1] = "LIST";
	    status = mr_query("delete_member_from_list", 3, membervec,
			       scream, NULL);
	    if (status == MR_SUCCESS)
	      break;
	    else if ((status != MR_LIST && status != MR_NO_MATCH) ||
		     memberstruct->type != M_ANY) {
	        if (status == MR_PERM && memberstruct->type == M_ANY)  {
		  /* M_ANY means we've fallen through from the user case
		   * The fact that we didn't get MR_PERM there indicates
		   * that we had permission to remove the specified member 
		   * from the list if it is a user, but not a list.  This is 
		   * if we are the member in question.  Since we exist as a user
		   * we must have gotten the MR_NO_MATCH error, so we will
		   * return that, since it will be less confusing.  However,
		   * This will generate the wrongerror if the user was trying
		   * to remove the list with his/her username from a list they
		   * don't administrate, without explicitly specifying "list:".
		   */
		  status = MR_NO_MATCH;
		}
		com_err(whoami, status, "while deleting member %s from %s",
			memberstruct->name, listname);
		success = 0;
		break;
	    }
	case M_STRING:
	    membervec[1] = "STRING";
	    status = mr_query("delete_member_from_list", 3, membervec,
			       scream, NULL);
	    if (status == MR_STRING && memberstruct->type == M_ANY) {
	      com_err(whoami, 0, " Unable to find member %s to delete from %s",
		      memberstruct->name, listname);
	      success = 0;
	      if (!strcmp(membervec[0], getenv("USER"))) {
		fprintf(stderr, "(If you were trying to remove yourself from the list \"%s\",\n", membervec[2]);
		fprintf(stderr, "the correct command is \"blanche %s -d %s\".)\n",
			membervec[2], membervec[0]);
	      }
	    } else if (status != MR_SUCCESS) {
	      com_err(whoami, status, "while deleting member %s from %s",
		      memberstruct->name, listname);
	      success = 0;
	    }
	    break;
	case M_KERBEROS:
	    membervec[1] = "KERBEROS";
	    status = mr_query("delete_member_from_list", 3, membervec,
			       scream, NULL);
	    if (status != MR_SUCCESS) {
	      com_err(whoami, status, "while deleting member %s from %s",
		      memberstruct->name, listname);
	      success = 0;
	    }
	}
    }

    /* Display the members of the list now, if requested */
    if (memberflg) {
	if (recursflg)
	  recursive_display_list_members();
	else {
	    status = mr_query("get_members_of_list", 1, &listname,
			       get_list_members, (char *)memberlist);
	    if (status)
	      com_err(whoami, status, "while getting members of list %s",
		      listname);
	    while (sq_get_data(memberlist, &memberstruct))
	      show_list_member(memberstruct);
	}
    }

    /* We're done! */
    mr_disconnect();
    exit(success?0:1);
}

usage(argv)
char **argv;
{
    fprintf(stderr, "Usage: %s listname [options]\n",argv[0]);
    fprintf(stderr, "Options are\n");
    fprintf(stderr, "   -v | -verbose\n");
    fprintf(stderr, "   -m | -members\n");
    fprintf(stderr, "   -u | -users\n");
    fprintf(stderr, "   -l | -lists\n");
    fprintf(stderr, "   -s | -strings\n");
    fprintf(stderr, "   -k | -kerberos\n");
    fprintf(stderr, "   -i | -info\n");
    fprintf(stderr, "   -r | -recursive\n");
    fprintf(stderr, "   -a | -add member\n");
    fprintf(stderr, "   -d | -delete member\n");
    fprintf(stderr, "   -al | -addlist filename\n");
    fprintf(stderr, "   -dl | -deletelist filename\n");
    fprintf(stderr, "   -f | -file filename\n");
    fprintf(stderr, "   -n | -noauth\n");
    fprintf(stderr, "   -db | -database host[:port]\n");
    fprintf(stderr, "   -D | -debug\n");
    exit(1);
}


/* Display the members stored in the queue */

show_list_member(memberstruct)
struct member *memberstruct;
{
    char *s = "";

    switch (memberstruct->type) {
    case M_USER:
	if (!showusers)
	  return;
	s = "USER";
	break;
    case M_LIST:
	if (!showlists)
	  return;
	s = "LIST";
	break;
    case M_STRING:
	if (!showstrings)
	  return;
	s = "STRING";
	break;
    case M_KERBEROS:
	if (!showkerberos)
	  return;
	s = "KERBEROS";
	break;
    case M_ANY:
	printf("%s\n", memberstruct->name);
	return;
    }

    if (verbose)
      printf("%s:%s\n", s, memberstruct->name);
    else {
	if (memberstruct->type == M_LIST)
	  printf("LIST:%s\n", memberstruct->name);
	else if (memberstruct->type == M_KERBEROS)
	  printf("KERBEROS:%s\n", memberstruct->name);
	else if (memberstruct->type == M_STRING &&
		 !strchr(memberstruct->name, '@'))
	  printf("STRING:%s\n", memberstruct->name);
	else
	  printf("%s\n", memberstruct->name);
    }
}


/* Show the retrieved information about a list */

show_list_info(argc, argv, hint)
int argc;
char **argv;
int hint;
{
    printf("List: %s\n", argv[0]);
    printf("Description: %s\n", argv[9]);
    printf("Flags: %s, %s, and %s\n",
	   atoi(argv[1]) ? "active" : "inactive",
	   atoi(argv[2]) ? "public" : "private",
	   atoi(argv[3]) ? "hidden" : "visible");
    printf("%s is %sa maillist and is %sa group", argv[0],
	   atoi(argv[4]) ? "" : "not ",
	   atoi(argv[5]) ? "" : "not ");
    if (atoi(argv[5]))
      printf(" with GID %d\n", atoi(argv[6]));
    else
      printf("\n");
    printf("Owner: %s %s\n", argv[7], argv[8]);
    printf("Last modified by %s with %s on %s\n", argv[11], argv[12], argv[10]);
    return(MR_CONT);
}


/* Show the retrieve list member count */

show_list_count(argc, argv, hint)
int argc;
char **argv;
int hint;
{
    printf("Members: %s\n", argv[0]);
}


/* Recursively find all of the members of listname, and then display them */

recursive_display_list_members()
{
    int status, count, savecount;
    struct save_queue *lists, *members;
    struct member *m, *m1, *data;

    lists = sq_create();
    members = sq_create();
    m = (struct member *) malloc(sizeof(struct member));
    m->type = M_LIST;
    m->name = listname;
    sq_save_data(lists, m);

    while (sq_get_data(lists, &m)) {
	sq_destroy(memberlist);
	memberlist = sq_create();
	if (debugflg)
	  fprintf(stderr, "Fetching members of %s\n", m->name);
	status = mr_query("get_members_of_list", 1, &(m->name),
			   get_list_members, (char *)memberlist);
	if (status)
	  com_err(whoami, status, "while getting members of list %s", m->name);
	while (sq_get_data(memberlist, &m1)) {
	    if (m1->type == M_LIST)
	      unique_add_member(lists, m1);
	    else
	      unique_add_member(members, m1);
	}
    }
    savecount = count = sq_count_elts(members);
    data = (struct member *) malloc(count * sizeof(struct member));
    count = 0;
    while (sq_get_data(members, &m))
      memcpy(&data[count++], m, sizeof(struct member));
    qsort(data, count, sizeof(struct member), membercmp);
    for (count = 0; count < savecount; count++) {
	show_list_member(&data[count]);
    }
}


/* add a struct member to a queue if that member isn't already there. */

unique_add_member(q, m)
struct save_queue  *q;
struct member *m;
{
    struct save_queue *qp;

    for (qp = q->q_next; qp != q; qp = qp->q_next) {
	if (!membercmp(qp->q_data, m))
	  return;
    }
    sq_save_data(q, m);
}


/* Collect the retrieved members of the list */

get_list_members(argc, argv, q)
int argc;
char **argv;
struct save_queue *q;
{
    struct member *m;

    m = (struct member *) malloc(sizeof(struct member));
    switch (argv[0][0]) {
    case 'U':
	m->type = M_USER;
	break;
    case 'L':
	m->type = M_LIST;
	break;
    case 'S':
	m->type = M_STRING;
	break;
    case 'K':
	m->type = M_KERBEROS;
	break;
    }
    m->name = strsave(argv[1]);
    sq_save_data(q, m);
    return(MR_CONT);
}


/* Called only if a query returns a value that we weren't expecting */

scream()
{
    fprintf(stderr, "Programmer botch\n");
    exit(3);
}


/* Open file, parse members from file, and put them on the specified queue */
get_members_from_file(filename, queue)
char *filename;
struct save_queue *queue;
{
    FILE *in;
    char buf[BUFSIZ];
    struct member *memberstruct;

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

    while (fgets(buf, BUFSIZ, in))
      if (memberstruct = parse_member(buf))
	sq_save_data(queue, memberstruct);
    if (!feof(in)) {
      com_err(whoami, errno, "while reading from %s", filename);
      exit(2);
    }
}


/* Collect the possible expansions of the alias MAILHUB */

int collect(argc, argv, list)
int argc;
char **argv;
char ***list;
{
    int i;

    for (i = 0; (*list)[i]; i++);
    *list = (char **)realloc(*list, (i + 2) * sizeof(char *));
    (*list)[i] = strsave(argv[2]);
    (*list)[i+1] = NULL;
    return(MR_CONT);
}


/* Parse a line of input, fetching a member.  NULL is returned if a member
 * is not found.  ';' is a comment character.
 */

struct member *parse_member(s)
register char *s;
{
    register struct member *m;
    char *p, *lastchar;

    while (*s && isspace(*s))
      s++;
    lastchar = p = s;
    while (*p && *p != '\n' && *p != ';')
      if (isprint(*p) && !isspace(*p))
	lastchar = p++;
      else
	p++;
    lastchar++;
    *lastchar = 0;
    if (p == s || strlen(s) == 0)
      return(NULL);

    if ((m = (struct member *) malloc(sizeof(struct member))) == NULL)
      return(NULL);

    if (p = strchr(s, ':')) {
	*p = 0;
	m->name = ++p;
	if (!strcasecmp("user", s))
	  m->type = M_USER;
	else if (!strcasecmp("list", s))
	  m->type = M_LIST;
	else if (!strcasecmp("string", s))
	  m->type = M_STRING;
	else if (!strcasecmp("kerberos", s))
	  m->type = M_KERBEROS;
	else {
	    m->type = M_STRING;
	    *(--p) = ':';
	    m->name = s;
	}
	m->name = strsave(m->name);
    } else {
	m->name = strsave(s);
	if (strchr(s, '@') || strchr(s, '!') || strchr(s, '%') || strchr(s, ' '))
	  m->type = M_STRING;
	else
	  m->type = M_ANY;
    }
    return(m);
}


  /* 
   * This routine two compares members by the following rules:
   * 1.  A USER is less than a LIST
   * 2.  A LIST is less than a STRING
   * 3.  If two members are of the same type, the one alphabetically first
   *     is less than the other
   * It returs < 0 if the first member is less, 0 if they are identical, and
   * > 0 if the second member is less (the first member is greater).
   */

int membercmp(m1, m2)
  struct member *m1, *m2;
{
    if (m1->type == M_ANY || m2->type  == M_ANY || (m1->type == m2->type))
	return(strcmp(m1->name, m2->name));
    else
	return(m1->type - m2->type);
}


sq_count_elts(q)
struct save_queue *q;
{
    char  *foo;
    int count;

    count = 0;
    while (sq_get_data(q, &foo))
      count++;
    return(count);
}