changing status field in host table now results in only the target host's

[moira.git] / regtape / employee.dc

/* $Header$
 */

#include <stdio.h>
#include <strings.h>
#include <ctype.h>
#include <sys/time.h>
#include <moira.h>
#include <moira_site.h>
EXEC SQL INCLUDE sqlca;


#define WHO 11859		/* root */
#define PROG "emp-tape"

#define MAX_ID_VALUE	31999
#define MIN_ID_VALUE	101

/* File format is:

0-8	id number
9-38	name
39-62	office address
63-74	phone1
75-86	phone2
87-106	dept
107-156	title
157-186	username
187-241	host

*/

#define LOC_ID 0
#define LOC_NAME 9
#define LOC_OFFICE 39
#define LOC_PHONE 63
#define LOC_PHONE2 75
#define LOC_DEPT 87
#define LOC_TITLE 107
#define LOC_USERNAME 157
#define LOC_HOST 187

#define LEN_ID 9
#define LEN_NAME 30
#define LEN_OFFICE 24
#define LEN_PHONE 12
#define LEN_PHONE2 12
#define LEN_DEPT 20
#define LEN_TITLE 50
#define LEN_USERNAME 30
#define LEN_HOST 55


struct entry {
    char *name;
    char *last;
    char *first;
    char *middle;
    char *title;
    char *class;
    char *id;
    char *eid;
    char *dept;
    char *address;
    char *phone;
    char *phone2;
    char *email;
};


char *whoami;
int newfinger = 0;

#define sqlfail() (sqlca.sqlcode && sqlca.sqlcode != 100)
#define SQL_DUPLICATE -40100
#define SQL_DEADLOCK -49900


main(argc, argv)
int argc;
char **argv;
{
    FILE *in;
    struct entry *e, *get_next_entry();
    int i, wait = 0;
    char buf[BUFSIZ], *file = NULL;

    whoami = rindex(argv[0], '/');
    if (whoami)
      whoami++;
    else
      whoami = argv[0];

    for (i = 1; i < argc; i++) {
	if (!strcmp(argv[i], "-w"))
	  wait++;
	else if (!strcmp(argv[i], "-D"))
	  setenv("ING_SET", "set printqry");
	else if (!strcmp(argv[i], "-n"))
	  newfinger++;
	else if (file != NULL)
	  fprintf(stderr, "Usage: %s [-w] [-D] [-n] inputfile\n", whoami);
	else
	  file = argv[i];
    }

    in = fopen(file, "r");
    if (in == NULL) {
	fprintf(stderr, "Unable to open %s for input\n", file);
	exit(1);
    }

    setlinebuf(stdout);
    setlinebuf(stderr);

    EXEC SQL CONNECT moira;
    if (sqlca.sqlcode != 0) {
	com_err(whoami, 0, "ingres error %d opening database", sqlca.sqlcode);
	exit(1);
    }

    while (e = get_next_entry(in)) {
    again:
	process_entry(e);
	EXEC SQL COMMIT WORK;
	if (sqlca.sqlcode != 0) {
            if (sqlca.sqlcode == SQL_DEADLOCK) {
                com_err(whoami, MR_DEADLOCK, "commiting work");
                goto again;
            } else {
                com_err(whoami, 0, "ingres error %d committing work", sqlca.sqlcode);
                exit(1);
            }
	}
	if (wait) {
	    printf("Next");
	    fflush(stdout);
	    gets(buf);
	}
    }

    exit(0);
}


char *substr(buf, key)
char *buf;
char *key;
{
    int l;

    for (l = strlen(key); *buf; buf++)
      if (!strncmp(buf, key, l))
	return(buf);
    return(NULL);
}


struct entry *get_next_entry(in)
FILE *in;
{
    static struct entry e;
    static char buf[BUFSIZ], mid[16], eid[16], email[256];
    static char name[LEN_NAME+1], sname[LEN_NAME+1], id[LEN_ID+1];
    static char office[LEN_OFFICE+1], phone[LEN_PHONE+1], phone2[LEN_PHONE2+1];
    static char dept[LEN_DEPT+1], title[LEN_TITLE+1], username[LEN_USERNAME+1];
    static char host[LEN_HOST+1];
    int ends_sr, ends_jr, ends_iii, ends_iv, ends_ii, ends_v;
    char *p;

    if (fgets(buf, sizeof(buf), in) == NULL)
      return((struct entry *)NULL);

    strncpy(id, &buf[LOC_ID], LEN_ID); id[LEN_ID] = 0;
    strncpy(name, &buf[LOC_NAME], LEN_NAME); name[LEN_NAME] = 0;
    strncpy(office, &buf[LOC_OFFICE], LEN_OFFICE); office[LEN_OFFICE] = 0;
    strncpy(phone, &buf[LOC_PHONE], LEN_PHONE); phone[LEN_PHONE] = 0;
    strncpy(phone2, &buf[LOC_PHONE2], LEN_PHONE2); phone2[LEN_PHONE2] = 0;
    strncpy(dept, &buf[LOC_DEPT], LEN_DEPT); dept[LEN_DEPT] = 0;
    strncpy(title, &buf[LOC_TITLE], LEN_TITLE); title[LEN_TITLE] = 0;
    strncpy(username, &buf[LOC_USERNAME], LEN_USERNAME); username[LEN_USERNAME] = 0;
    strncpy(host, &buf[LOC_HOST], LEN_HOST); host[LEN_HOST] = 0;

    strcpy(sname, name);
    e.name = strtrim(sname);
    p = index(name, ',');
    if (p)
      *p = 0;
    e.last = strtrim(name);
    if (p) {
	p++;
	while (isspace(*p))
	  p++;
	e.first = p;
	if (p = index(e.first, ' ')) {
	    *p = 0;
	    e.first = strtrim(e.first);
	    e.middle = strtrim(p + 1);
	} else {
	    e.first = strtrim(e.first);
	    e.middle = "";
	}
    } else {
	e.first = "";
	e.middle = "";
    }
    ends_sr = ends_jr = ends_iii = ends_iv = ends_ii = ends_v = 0;
    LookForSt(e.last);
    LookForO(e.last);
    LookForJrAndIII(e.last, &ends_sr, &ends_jr, &ends_iii, &ends_iv,
		    &ends_ii, &ends_v);
    LookForJrAndIII(e.first, &ends_sr, &ends_jr, &ends_iii, &ends_iv,
		    &ends_ii, &ends_v);
    FixCase(e.last);
    FixCase(e.first);
    FixCase(e.middle);

    e.id = id;
    e.eid = eid;
    EncryptID(e.eid, e.id, e.first, e.last);

    e.address = strtrim(office);
    e.phone = strtrim(phone);
    e.phone2 = strtrim(phone2);
    e.dept = strtrim(dept);
    e.title = strtrim(title);

    e.class = "MITS";
    if (substr(e.title, "PROF") || substr(e.title, "LECTURE"))
      e.class = "FACULTY";

    strcpy(email, strtrim(username));
    if (host[0] == '@')
      strncat(email, strtrim(host));
    e.email = email;

    return(&e);
}


process_entry(e)
struct entry *e;
{
    int changed, nochange, encrypted;
    char buf[BUFSIZ], *from, *to;
    EXEC SQL BEGIN DECLARE SECTION;
    char *first, *last, *middle, *eid, *sid, *name, *title, *phone2, *rdept, *rtitle;
    char *raddr, *rhphone, *rophone, *prog;
    char class[9], oaddr[25], ophone[17], dept[128], dfirst[17], dlast[17], dmiddle[17];
    int id, status, who;
    EXEC SQL END DECLARE SECTION;

    /* Don't process Lincoln Labs */
    if (!strncmp(e->address, "LL", 2))
      return;

    who = WHO;
    prog = PROG;
    first = e->first;
    if (strlen(first) > 16)
      first[16] = 0;
    last = e->last;
    if (strlen(last) > 16)
      last[16] = 0;
    middle = e->middle;
    eid = e->eid;
    sid = e->id;
    id = 0;
    encrypted = 0;

    /* Get user info */
    EXEC SQL SELECT users_id, first, last, middle, type, office_addr, office_phone, department, status
      INTO :id, :dfirst, :dlast, :dmiddle, :class, :oaddr, :ophone, :dept, :status
      FROM users
      WHERE clearid = :sid;
    if (sqlfail()) {
	if (sqlca.sqlcode == SQL_DUPLICATE) {
	    com_err(whoami, 0, "duplicate ID number %s on user %s %s", sid, first, last);
	    return;
        } else if (sqlca.sqlcode == SQL_DEADLOCK) {
            com_err(whoami, MR_DEADLOCK, "looking up user %s", sid);
            EXEC SQL ROLLBACK;
            return process_entry(e);
	} else
	  sqlexit();
    }
    if (id == 0) {
	EXEC SQL SELECT users_id, first, last, middle, type, office_addr, office_phone, department, status
	  INTO :id, :dfirst, :dlast, :dmiddle, :class, :oaddr, :ophone, :dept, :status
	  FROM users
	  WHERE last = :last and first = :first and clearid = :eid;
        if (sqlfail()) {
            if (sqlca.sqlcode == SQL_DEADLOCK) {
                com_err(whoami, MR_DEADLOCK, "looking up user %s", sid);
                EXEC SQL ROLLBACK;
                return process_entry(e);
            } else if (sqlca.sqlcode != SQL_DUPLICATE)
              sqlexit();
        }
	encrypted++;
	if (id == 0) {
	    newuser(e);
	    return;
	}
    }

    /* Update class/state if necessary */
    if (strcmp(e->class, strtrim(class)) &&
	strcmp(class, "STAFF") && strcmp(class, "SIPB")) {
	com_err(whoami, 0, "updating class for %s %s from %s to %s",
		first, last, class, e->class);
	if (status == US_NOT_ALLOWED && !strcmp(e->class, "FACULTY"))
	  status = US_NO_LOGIN_YET;
	if (status == US_ENROLL_NOT_ALLOWED && !strcmp(e->class, "FACULTY"))
	  status = US_ENROLLED;
	strcpy(class, e->class);
	EXEC SQL UPDATE users
	  SET type = :class, status = :status, modtime = 'now',
		modby = :who, modwith = :prog
	  WHERE users_id = :id;
	if (sqlca.sqlcode != 0) {
            if (sqlca.sqlcode == SQL_DEADLOCK) {
                com_err(whoami, MR_DEADLOCK, "updating user");
		return;
            } else {
                com_err(whoami, 0, "ingres error %d updating user", sqlca.sqlcode);
                exit(1);
	    }
	}
    }

    /* Update name if necessary */
    if (strcmp(first, strtrim(dfirst)) ||
	strcmp(last, strtrim(dlast)) ||
	strcmp(middle, strtrim(dmiddle))) {
	com_err(whoami, 0, "updating real name for %s %s", first, last);
	EXEC SQL UPDATE users
	  SET first = :first, last = :last, middle = :middle,
	  	modby = :who, modwith = :prog, modtime = 'now'
	  WHERE users_id = :id;
	if (sqlca.sqlcode != 0) {
            if (sqlca.sqlcode == SQL_DEADLOCK) {
                com_err(whoami, MR_DEADLOCK, "updating name");
		return;
            } else {
                com_err(whoami, 0, "ingres error %d updating name", sqlca.sqlcode);
                exit(1);
            }
	}
    }

    changed = nochange = 0;
    if (encrypted) changed++;
    strcpy(buf, e->address);
    while (to = index(buf, ','))
      *to = ';';
    while (to = index(buf, ':'))
      *to = ';';
    if (newfinger) {
	if (oaddr[0] == ' ' && buf[0]) {
	    strncpy(oaddr, buf, 16);
	    oaddr[16] = 0;
	    changed++;
	} else if (strncmp(strtrim(oaddr), buf, 15))
	  nochange++;
    } else {
	if (strncmp(strtrim(oaddr), buf, 15))
	  changed++;
	strncpy(oaddr, buf, 16);
	oaddr[16] = 0;
    }
    from = e->phone;
    to = buf;
    while (*from) {
	if (isdigit(*from))
	  *to++ = *from;
	from++;
    }
    *to = 0;
    if (newfinger) {
	if (ophone[0] == ' ') {
	    strncpy(ophone, buf, 16);
	    ophone[16] = 0;
	} else if (strncmp(strtrim(ophone), buf, 11))
	  nochange++;
    } else {
	if (strncmp(strtrim(ophone), buf, 11))
	  changed++;
	strncpy(ophone, buf, 16);
	ophone[16] = 0;
    }
    FixCase(e->dept);
    FixCase(e->title);
    if (newfinger) {
	if (dept[0] == ' ') {
	    strncpy(dept, e->dept, 12);
	    dept[12] = 0;
	} else if (strncmp(strtrim(dept), e->dept, 11))
	  nochange++;
    } else {
	if (strncmp(strtrim(dept), e->dept, 11))
	  changed++;
	strncpy(dept, e->dept, 12);
	dept[12] = 0;
    }
    sid = e->id;
    name = e->name;
    rdept = e->dept;
    rtitle = e->title;
    raddr = e->address;
    rhphone = e->phone;
    rophone = e->phone2;
    if (changed) {
	com_err(whoami, 0, "updating finger for %s %s", first, last);
	EXEC SQL REPEATED UPDATE users
	  SET office_addr = :oaddr, office_phone = :ophone, department = :dept,
	    fmodtime = 'now', fmodby = :who, fmodwith = :prog,
	    xname = :name, xdept = :rdept, xtitle = :rtitle,
	    xaddress = :raddr, xphone1 = :rhphone, xphone2 = :rophone,
	    xmodtime = 'now', clearid = :sid
	  WHERE users_id = :id;
	if (sqlca.sqlcode != 0) {
            if (sqlca.sqlcode == SQL_DEADLOCK) {
                com_err(whoami, MR_DEADLOCK, "updating user %s", sid);
                EXEC SQL ROLLBACK;
                return process_entry(e);
            } else {
                com_err(whoami, 0, "ingres error %d", sqlca.sqlcode);
                exit(1);
            }
	}
    } else {
	EXEC SQL REPEATED UPDATE users
	  SET xname = :name, xdept = :rdept, xtitle = :rtitle,
	    xaddress = :raddr, xphone1 = :rhphone, xphone2 = :rophone,
	    xmodtime = 'now', clearid = :sid
	  WHERE users_id = :id;
	if (sqlca.sqlcode != 0) {
            if (sqlca.sqlcode == SQL_DEADLOCK) {
                com_err(whoami, MR_DEADLOCK, "updating user %s", sid);
                EXEC SQL ROLLBACK;
                return process_entry(e);
            } else {
                com_err(whoami, 0, "ingres error %d", sqlca.sqlcode);
                exit(1);
            }
	}
    }
}


newuser(e)
struct entry *e;
{
    char *from, *to;
    EXEC SQL BEGIN DECLARE SECTION;
    int id, uid, st, who;
    char *last, *first, *class, *middle, login[9], *sid, fullname[65], *prog;
    char oaddr[81], ophone[17], dept[128], *name, *title, phone2[17];
    char *rdept, *rhphone, *rophone;
    EXEC SQL END DECLARE SECTION;

    who = WHO;
    prog = PROG;
    strncpy(oaddr, e->address, 16);
    oaddr[16] = 0;
    while (to = index(oaddr, ','))
      *to = ';';
    while (to = index(oaddr, ':'))
      *to = ';';
    from = e->phone;
    to = ophone;
    while (*from) {
	if (isdigit(*from))
	  *to++ = *from;
	from++;
    }
    *to = 0;
    FixCase(e->dept);
    strncpy(dept, e->dept, 12);
    dept[12] = 0;
    
    id = set_next_users_id(0);
    uid = set_next_uid(1);
    sprintf(login, "#%d", uid);
    last = e->last;
    first = e->first;
    middle = e->middle;
    class = e->class;
    if (*middle)
      sprintf(fullname, "%s %s %s", first, middle, last);
    else
      sprintf(fullname, "%s %s", first, last);
    st = US_NOT_ALLOWED;
    if (!strcmp(e->class, "FACULTY") || !strcmp(e->class, "STAFF"))
      st = US_NO_LOGIN_YET;

    sid = e->id;
    name = e->name;
    rdept = e->dept;
    title = e->title;
    rhphone = e->phone;
    rophone = e->phone2;

    EXEC SQL REPEATED INSERT INTO users
      (login, users_id, uid, shell, last, first, middle, status,
       clearid, type, modtime, modby, modwith, fullname, office_addr,
       office_phone, department, fmodtime, fmodby, fmodwith,
       potype, xname, xdept, xtitle, xaddress, xphone1, xphone2, xmodtime)
      VALUES (:login, :id, :uid, '/bin/csh', :last, :first, :middle, :st,
	      :sid, :class, 'now', :who, :prog, :fullname, :oaddr, :ophone,
	      :dept, 'now', :who, :prog, 'NONE', :name, :rdept,
	      :title, :oaddr, :rhphone, :rophone, 'now');
    if (sqlca.sqlcode != 0) {
	if (sqlca.sqlcode == SQL_DEADLOCK) {
	    com_err(whoami, MR_DEADLOCK, "adding user");
	} else {
	    com_err(whoami, 0, "ingres error %d adding user", sqlca.sqlcode);
	    exit(1);
	}
    } else
      com_err(whoami, 0, "adding user %s %s", e->first, e->last);
}


set_next_users_id(limit)
    int limit;
{
    EXEC SQL BEGIN DECLARE SECTION;
    int rowcount, flag, value, retval;
    EXEC SQL END DECLARE SECTION;

    EXEC SQL REPEATED SELECT value INTO :value FROM numvalues 
      WHERE name = 'users_id';
    if (sqlfail()) sqlexit();
    if (sqlca.sqlerrd[2] != 1) {
	EXEC SQL ROLLBACK;
	com_err(whoami, MR_INTERNAL, "values table inconsistancy");
	exit(1);
    }

    flag = 0;
    EXEC SQL REPEATED SELECT users_id INTO :flag FROM users
      WHERE users_id = :value;
    if (sqlfail()) sqlexit();
    if (sqlca.sqlerrd[2] == 0)
      flag = 0;
    while (flag) {
	value++;
	if (limit && value > MAX_ID_VALUE)
	    value = MIN_ID_VALUE;
	flag = 0;
	EXEC SQL REPEATED SELECT users_id INTO :flag FROM users 
	  WHERE users_id = :value;
	if (sqlfail()) sqlexit();
	if (sqlca.sqlerrd[2] == 0)
	  flag = 0;
    }

    retval = value++;
    if (limit && value > MAX_ID_VALUE)
      value = MIN_ID_VALUE;
    EXEC SQL REPEATED UPDATE numvalues SET value = :value
      WHERE name = 'users_id';
    if (sqlca.sqlcode != 0) {
	com_err(whoami, 0, "ingres error %d assigning ID", sqlca.sqlcode);
	exit(1);
    }
    return(retval);
}

set_next_uid(limit)
    int limit;
{
    EXEC SQL BEGIN DECLARE SECTION;
    int rowcount, flag, value, retval;
    EXEC SQL END DECLARE SECTION;

    EXEC SQL REPEATED SELECT value INTO :value FROM numvalues 
      WHERE name = 'uid';
    if (sqlfail()) sqlexit();
    if (sqlca.sqlerrd[2] != 1) {
	EXEC SQL ROLLBACK;
	com_err(whoami, MR_INTERNAL, "values table inconsistancy");
	exit(1);
    }

    flag = 0;
    EXEC SQL REPEATED SELECT uid INTO :flag FROM users WHERE uid = :value;
    if (sqlfail()) sqlexit();
    if (sqlca.sqlerrd[2] == 0)
      flag = 0;
    while (flag) {
	value++;
	if (limit && value > MAX_ID_VALUE)
	    value = MIN_ID_VALUE;
	flag = 0;
	EXEC SQL REPEATED SELECT uid INTO :flag FROM users WHERE uid = :value;
	if (sqlfail()) sqlexit();
	if (sqlca.sqlerrd[2] == 0)
	  flag = 0;
    }

    retval = value++;
    if (limit && value > MAX_ID_VALUE)
      value = MIN_ID_VALUE;
    EXEC SQL REPEATED UPDATE numvalues SET value = :value WHERE name = 'uid';
    if (sqlca.sqlcode != 0) {
	com_err(whoami, 0, "ingres error %d assigning ID", sqlca.sqlcode);
	exit(1);
    }
    return(retval);
}


sqlexit()
{
    if (sqlca.sqlcode == SQL_DEADLOCK)
      com_err(whoami, MR_DEADLOCK, "unrecoverable ingres error %d",
              sqlca.sqlcode);
    else
      com_err(whoami, 0, "ingres error %d", sqlca.sqlcode);
    EXEC SQL ROLLBACK WORK;
    exit(1);
}