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;
  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 sms;
  50 #endsql
  51 #ifsql INFORMIX
  52     EXEC SQL DATABASE sms;
  53 #endsql
  54
  55     if (argc == 2) {
  56         if (stat(argv[1], &sb) == 0) {
  57             if (ModDiff (&flag1, "users", sb.st_mtime) == 0 &&
  58                 ModDiff (&flag2, "filesys", sb.st_mtime) == 0 &&
  59                 flag1 < 0 && flag2 < 0) {
  60                 fprintf(stderr, "File %s does not need to be rebuilt.\n",
  61                         argv[1]);
  62                 exit(MR_NO_CHANGE);
  63             }
  64         }
  65         outf = argv[1];
  66         sprintf(outft, "%s~", outf);
  67         if ((out = fopen(outft, "w")) == NULL) {
  68             fprintf(stderr, "unable to open %s for output\n", outf);
  69             exit(MR_OCONFIG);
  70         }
  71     } else if (argc != 1) {
  72         fprintf(stderr, "usage: %s [outfile]\n", argv[0]);
  73         exit(MR_ARGS);
  74     } else {
  75         outf = NULL;
  76     }
  77
  78     EXEC SQL WHENEVER SQLERROR GOTO sqlerr;
  79     sq = sq_create();
  80
  81     EXEC SQL DECLARE y CURSOR FOR
  82       SELECT p.nfsphys_id, p.dir, m.name
  83       FROM nfsphys p, machine m
  84         WHERE m.mach_id = p.mach_id;
  85     EXEC SQL OPEN y;
  86     while (1) {
  87         EXEC SQL FETCH y INTO :id, :dir, :name;
  88         if (sqlca.sqlcode != 0) break;
  89         if (id == 0) continue;
  90         strtrim(dir);
  91         strtrim(name);
  92         prec = (struct prec *) malloc(sizeof(struct prec));
  93         prec->phys_id = id;
  94         strcpy(prec->mach, strtrim(name));
  95         strcpy(prec->dir, strtrim(dir));
  96         sq_save_data(sq, prec);
  97     }
  98     EXEC SQL CLOSE y;
  99
 100     while (sq_get_data(sq, &prec)) {
 101         id = prec->phys_id;
 102 #ifdef DEBUG
 103         printf("got ID %d\n", id);
 104         printf("mach %s dir %s\n", prec->mach, prec->dir);
 105 #endif
 106         EXEC SQL DECLARE x CURSOR FOR
 107           SELECT u.login, u.fullname
 108           FROM users u, filesys f
 109             WHERE f.label = u.login and u.status = 1 and f.phys_id = :id;
 110         EXEC SQL OPEN x;
 111         first = 1;
 112         while (1) {
 113             EXEC SQL FETCH x INTO :login, :name;
 114             if (sqlca.sqlcode != 0) break;
 115             if (first) {
 116                 first = 0;
 117                 fprintf(out, "*%s:%s\n", prec->mach, prec->dir);
 118             }
 119             strtrim(login);
 120             strtrim(name);
 121             fprintf(out, "%s,%s\n", login, name);
 122         }
 123         EXEC SQL CLOSE x;
 124     }
 125
 126     EXEC SQL DECLARE z CURSOR FOR
 127       SELECT DISTINCT u.login, u.fullname, f.name
 128       FROM users u, filesys f
 129         WHERE f.label = u.login and u.status = 1 and f.type = "AFS"
 130         ORDER BY name;
 131     EXEC SQL OPEN z;
 132     while (1) {
 133         EXEC SQL FETCH z INTO :login, :name, :dir;
 134         if (sqlca.sqlcode != 0) break;
 135         p = rindex(dir, '/');
 136         if (p) *p = 0;
 137         if (strcmp(last, dir)) {
 138             strcpy(last, dir);
 139             fprintf(out, "*%s:%s\n", "AFS", dir);
 140         }
 141         strtrim(login);
 142         strtrim(name);
 143         fprintf(out, "%s,%s\n", login, name);
 144     }
 145     EXEC SQL CLOSE z;
 146
 147 #ifsql INGRES
 148     EXEC SQL DISCONNECT;
 149 #endsql
 150 #ifsql INFORMIX
 151     EXEC SQL CLOSE DATABASE;
 152 #endsql
 153
 154     if (fclose(out)) {
 155         perror("close failed");
 156         exit(MR_CCONFIG);
 157     }
 158     if (outf)
 159       fix_file(outf);
 160     exit(MR_SUCCESS);
 161
 162  sqlerr:
 163     com_err(whoami, MR_INGRES_ERR, " code %d\n", sqlca.sqlcode);
 164 #ifsql INGRES
 165     if (sqlca.sqlcode == 17700)
 166       exit(MR_DEADLOCK);
 167 #endsql
 168     critical_alert("DCM", "Central America build encountered INGRES ERROR %d",
 169                    sqlca.sqlcode);
 170     exit(MR_INGRES_ERR);
 171 }