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
  81     /* Get locks */
  82     EXEC SQL SELECT modtime INTO :name FROM filesys WHERE filsys_id = 0;
  83     EXEC SQL SELECT modtime INTO :name FROM users WHERE users_id = 0;
  84     EXEC SQL SELECT modtime INTO :name FROM machine WHERE mach_id = 0;
  85     EXEC SQL SELECT modtime INTO :name FROM nfsphys WHERE mach_id = 0;
  86     sq = sq_create();
  87
  88     cnt = 0;
  89     EXEC SQL DECLARE y CURSOR FOR
  90       SELECT p.nfsphys_id, p.dir, m.name
  91       FROM nfsphys p, machine m
  92       WHERE m.mach_id = p.mach_id
  93       ORDER BY p.nfsphys_id;
  94     EXEC SQL OPEN y;
  95     while (1) {
  96         EXEC SQL FETCH y INTO :id, :dir, :name;
  97         if (sqlca.sqlcode != 0) break;
  98         if (id == 0) continue;
  99         strtrim(dir);
 100         strtrim(name);
 101         prec = (struct prec *) malloc(sizeof(struct prec));
 102         prec->phys_id = id;
 103         strcpy(prec->mach, strtrim(name));
 104         strcpy(prec->dir, strtrim(dir));
 105         sq_save_data(sq, prec);
 106         cnt++;
 107     }
 108     EXEC SQL CLOSE y;
 109     fprintf(stderr, "Loaded %d NFS partitions\n", cnt);
 110
 111     while (sq_get_data(sq, &prec)) {
 112         id = prec->phys_id;
 113 #ifdef DEBUG
 114         printf("got ID %d\n", id);
 115         printf("mach %s dir %s\n", prec->mach, prec->dir);
 116 #endif
 117         EXEC SQL DECLARE x CURSOR FOR
 118           SELECT u.login, u.fullname
 119           FROM users u, filesys f
 120           WHERE f.label = u.login and u.status = 1 and f.phys_id = :id
 121           ORDER BY u.login;
 122         EXEC SQL OPEN x;
 123         first = 1;
 124         while (1) {
 125             EXEC SQL FETCH x INTO :login, :name;
 126             if (sqlca.sqlcode != 0) break;
 127             if (first) {
 128                 first = 0;
 129                 fprintf(out, "*%s:%s\n", prec->mach, prec->dir);
 130             }
 131             strtrim(login);
 132             strtrim(name);
 133             fprintf(out, "%s,%s\n", login, name);
 134         }
 135         EXEC SQL CLOSE x;
 136     }
 137     fprintf(stderr, "Processed NFS partitions\n");
 138
 139     cnt = 0;
 140     EXEC SQL DECLARE z CURSOR FOR
 141       SELECT DISTINCT u.login, u.fullname, f.name
 142       FROM users u, filesys f
 143       WHERE f.label = u.login and u.status = 1 and f.type = 'AFS'
 144       ORDER BY u.login;
 145     EXEC SQL OPEN z;
 146     while (1) {
 147         EXEC SQL FETCH z INTO :login, :name, :dir;
 148         if (sqlca.sqlcode != 0) break;
 149         p = rindex(dir, '/');
 150         if (p) *p = 0;
 151         if (strcmp(last, dir)) {
 152             strcpy(last, dir);
 153             fprintf(out, "*%s:%s\n", "AFS", dir);
 154         }
 155         strtrim(login);
 156         strtrim(name);
 157         fprintf(out, "%s,%s\n", login, name);
 158         cnt++;
 159     }
 160     EXEC SQL CLOSE z;
 161     fprintf(stderr, "Processed %d AFS users\n", cnt);
 162
 163 #ifsql INGRES
 164     EXEC SQL DISCONNECT;
 165 #endsql
 166 #ifsql INFORMIX
 167     EXEC SQL CLOSE DATABASE;
 168 #endsql
 169
 170     if (fclose(out)) {
 171         perror("close failed");
 172         exit(MR_CCONFIG);
 173     }
 174     if (outf)
 175       fix_file(outf);
 176     exit(MR_SUCCESS);
 177
 178  sqlerr:
 179     com_err(whoami, MR_INGRES_ERR, " code %d\n", sqlca.sqlcode);
 180 #ifsql INGRES
 181     if (sqlca.sqlcode == 17700 || sqlca.sqlcode == -37000)
 182       exit(MR_DEADLOCK);
 183 #endsql
 184     critical_alert("DCM", "Central America build encountered INGRES ERROR %d",
 185                    sqlca.sqlcode);
 186     exit(MR_INGRES_ERR);
 187 }
 188
 189