gen/prlogger.dc

   1 /* $Header$
   2  *
   3  * This generates a list of active users for the prlogger
   4  *
   5  *  (c) Copyright 1990 by the Massachusetts Institute of Technology.
   6  *  For copying and distribution information, please see the file
   7  *  <mit-copyright.h>.
   8  */
   9
  10 #include <mit-copyright.h>
  11 #include <stdio.h>
  12 #include <moira.h>
  13 #include <moira_site.h>
  14 #include <sys/types.h>
  15 #include <sys/stat.h>
  16 #include <sys/time.h>
  17 EXEC SQL INCLUDE sqlca;
  18
  19 extern int errno;
  20 char *whoami = "prlogger.gen";
  21
  22
  23 main(argc, argv)
  24 int argc;
  25 char **argv;
  26 {
  27     FILE *out = stdout;
  28     char *outf = NULL, outft[64];
  29     struct stat sb;
  30     int flag;
  31     EXEC SQL BEGIN DECLARE SECTION;
  32     char login[9], class[9];
  33     EXEC SQL END DECLARE SECTION;
  34
  35 #ifsql INGRES
  36     EXEC SQL CONNECT moira;
  37     EXEC SQL SET LOCKMODE SESSION WHERE LEVEL=TABLE, READLOCK=SHARED;
  38 #endsql
  39 #ifsql INFORMIX
  40     EXEC SQL DATABASE moira;
  41 #endsql
  42
  43     if (argc == 2) {
  44         if (stat(argv[1], &sb) == 0) {
  45             if (ModDiff (&flag, "users", sb.st_mtime))
  46               exit(MR_DATE);
  47             if (flag < 0) {
  48                 fprintf(stderr, "File %s does not need to be rebuilt.\n",
  49                         argv[1]);
  50                 exit(MR_NO_CHANGE);
  51             }
  52         }
  53         outf = argv[1];
  54         sprintf(outft, "%s~", outf);
  55         if ((out = fopen(outft, "w")) == NULL) {
  56             fprintf(stderr, "unable to open %s for output\n", outf);
  57             exit(MR_OCONFIG);
  58         }
  59     } else if (argc != 1) {
  60         fprintf(stderr, "usage: %s [outfile]\n", argv[0]);
  61         exit(MR_ARGS);
  62     } else {
  63         outf = NULL;
  64     }
  65
  66     EXEC SQL WHENEVER SQLERROR GOTO sqlerr;
  67
  68     EXEC SQL DECLARE x CURSOR FOR
  69       SELECT login, type FROM users WHERE status = 1;
  70     EXEC SQL OPEN x;
  71     while (1) {
  72         EXEC SQL FETCH x INTO :login, :class;
  73         if (sqlca.sqlcode != 0) break;
  74         strtrim(login);
  75         strtrim(class);
  76         fprintf(out, "%s %s\n", login, class);
  77     }
  78     EXEC SQL CLOSE x;
  79
  80 #ifsql INGRES
  81     EXEC SQL DISCONNECT;
  82 #endsql
  83 #ifsql INFORMIX
  84     EXEC SQL CLOSE DATABASE;
  85 #endsql
  86
  87     if (fclose(out)) {
  88         perror("close failed");
  89         exit(MR_CCONFIG);
  90     }
  91     if (outf)
  92       fix_file(outf);
  93     exit(MR_SUCCESS);
  94
  95  sqlerr:
  96     com_err(whoami, MR_INGRES_ERR, " code %d\n", sqlca.sqlcode);
  97 #ifsql INGRES
  98     if (sqlca.sqlcode == 17700 || sqlca.sqlcode == -37000)
  99       exit(MR_DEADLOCK);
 100 #endsql
 101     critical_alert("DCM", "PrLogger build encountered INGRES ERROR %d",
 102                    sqlca.sqlcode);
 103     exit(MR_INGRES_ERR);
 104 }