gen/ca.dc

   1 /* $Header$
   2  *
   3  * This generates a list of users and their home directories for the
   4  * Centra America daemon.
   5  *
   6  *  (c) Copyright 1991 by the Massachusetts Institute of Technology.
   7  *  For copying and distribution information, please see the file
   8  *  <mit-copyright.h>.
   9  */
  10
  11 #include <mit-copyright.h>
  12 #include <stdio.h>
  13 #include <moira.h>
  14 #include <moira_site.h>
  15 #include <sys/types.h>
  16 #include <sys/stat.h>
  17 #include <sys/time.h>
  18 #include <strings.h>
  19 EXEC SQL INCLUDE sqlca;
  20
  21 extern int errno;
  22 char *whoami = "ca.gen";
  23
  24 struct prec {
  25     int phys_id;
  26     char mach[64];
  27     char dir[128];
  28 };
  29
  30
  31 main(argc, argv)
  32 int argc;
  33 char **argv;
  34 {
  35     FILE *out = stdout;
  36     char *outf = NULL, outft[64], *p, last[256];
  37     struct stat sb;
  38     struct save_queue *sq;
  39     struct prec *prec;
  40     int flag1, flag2, first, cnt;
  41     EXEC SQL BEGIN DECLARE SECTION;
  42     char login[257], name[257], dir[257];
  43     int id;
  44     EXEC SQL END DECLARE SECTION;
  45
  46     initialize_sms_error_table ();
  47
  48 #ifsql INGRES
  49     EXEC SQL CONNECT moira;
  50     EXEC SQL SET LOCKMODE SESSION WHERE LEVEL=TABLE, READLOCK=SHARED;
  51 #endsql
  52 #ifsql INFORMIX
  53     EXEC SQL DATABASE moira;
  54 #endsql
  55
  56     if (argc == 2) {
  57         if (stat(argv[1], &sb) == 0) {
  58             if (ModDiff (&flag1, "users", sb.st_mtime) == 0 &&
  59                 ModDiff (&flag2, "filesys", sb.st_mtime) == 0 &&
  60                 flag1 < 0 && flag2 < 0) {
  61                 fprintf(stderr, "File %s does not need to be rebuilt.\n",
  62                         argv[1]);
  63                 exit(MR_NO_CHANGE);
  64             }
  65         }
  66         outf = argv[1];
  67         sprintf(outft, "%s~", outf);
  68         if ((out = fopen(outft, "w")) == NULL) {
  69             fprintf(stderr, "unable to open %s for output\n", outf);
  70             exit(MR_OCONFIG);
  71         }
  72     } else if (argc != 1) {
  73         fprintf(stderr, "usage: %s [outfile]\n", argv[0]);
  74         exit(MR_ARGS);
  75     } else {
  76         outf = NULL;
  77     }
  78
  79     EXEC SQL WHENEVER SQLERROR GOTO sqlerr;
  80     sq = sq_create();
  81
  82     cnt = 0;
  83     EXEC SQL DECLARE y CURSOR FOR
  84       SELECT p.nfsphys_id, p.dir, m.name
  85       FROM nfsphys p, machine m
  86       WHERE m.mach_id = p.mach_id
  87       ORDER BY p.nfsphys_id;
  88     EXEC SQL OPEN y;
  89     while (1) {
  90         EXEC SQL FETCH y INTO :id, :dir, :name;
  91         if (sqlca.sqlcode != 0) break;
  92         if (id == 0) continue;
  93         strtrim(dir);
  94         strtrim(name);
  95         prec = (struct prec *) malloc(sizeof(struct prec));
  96         prec->phys_id = id;
  97         strcpy(prec->mach, strtrim(name));
  98         strcpy(prec->dir, strtrim(dir));
  99         sq_save_data(sq, prec);
 100         cnt++;
 101     }
 102     EXEC SQL CLOSE y;
 103     fprintf(stderr, "Loaded %d NFS partitions\n", cnt);
 104
 105     while (sq_get_data(sq, &prec)) {
 106         id = prec->phys_id;
 107 #ifdef DEBUG
 108         printf("got ID %d\n", id);
 109         printf("mach %s dir %s\n", prec->mach, prec->dir);
 110 #endif
 111         EXEC SQL DECLARE x CURSOR FOR
 112           SELECT u.login, u.fullname
 113           FROM users u, filesys f
 114           WHERE f.label = u.login and u.status = 1 and f.phys_id = :id
 115           ORDER BY u.login;
 116         EXEC SQL OPEN x;
 117         first = 1;
 118         while (1) {
 119             EXEC SQL FETCH x INTO :login, :name;
 120             if (sqlca.sqlcode != 0) break;
 121             if (first) {
 122                 first = 0;
 123                 fprintf(out, "*%s:%s\n", prec->mach, prec->dir);
 124             }
 125             strtrim(login);
 126             strtrim(name);
 127             fprintf(out, "%s,%s\n", login, name);
 128         }
 129         EXEC SQL CLOSE x;
 130     }
 131     fprintf(stderr, "Processed NFS partitions\n");
 132
 133     cnt = 0;
 134     EXEC SQL DECLARE z CURSOR FOR
 135       SELECT DISTINCT u.login, u.fullname, f.name
 136       FROM users u, filesys f
 137       WHERE f.label = u.login and u.status = 1 and f.type = 'AFS'
 138       ORDER BY u.login;
 139     EXEC SQL OPEN z;
 140     while (1) {
 141         EXEC SQL FETCH z INTO :login, :name, :dir;
 142         if (sqlca.sqlcode != 0) break;
 143         p = rindex(dir, '/');
 144         if (p) *p = 0;
 145         if (strcmp(last, dir)) {
 146             strcpy(last, dir);
 147             fprintf(out, "*%s:%s\n", "AFS", dir);
 148         }
 149         strtrim(login);
 150         strtrim(name);
 151         fprintf(out, "%s,%s\n", login, name);
 152         cnt++;
 153     }
 154     EXEC SQL CLOSE z;
 155     fprintf(stderr, "Processed %d AFS users\n", cnt);
 156
 157 #ifsql INGRES
 158     EXEC SQL DISCONNECT;
 159 #endsql
 160 #ifsql INFORMIX
 161     EXEC SQL CLOSE DATABASE;
 162 #endsql
 163
 164     if (fclose(out)) {
 165         perror("close failed");
 166         exit(MR_CCONFIG);
 167     }
 168     if (outf)
 169       fix_file(outf);
 170     exit(MR_SUCCESS);
 171
 172  sqlerr:
 173     com_err(whoami, MR_INGRES_ERR, " code %d\n", sqlca.sqlcode);
 174 #ifsql INGRES
 175     if (sqlca.sqlcode == 17700 || sqlca.sqlcode == -37000)
 176       exit(MR_DEADLOCK);
 177 #endsql
 178     critical_alert("DCM", "Central America build encountered INGRES ERROR %d",
 179                    sqlca.sqlcode);
 180     exit(MR_INGRES_ERR);
 181 }