dbck/dbck.dc

   1 /* $Header$
   2  *
   3  * Moira database consistency checker
   4  *
   5  *  (c) Copyright 1988 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 <strings.h>
  13 #include <signal.h>
  14 #include "dbck.h"
  15
  16
  17 static char dbck_qc_rcsid[] = "$Header$";
  18
  19
  20 int debug = 0;
  21 int mode = MODE_ASK;
  22 int fast = 0;
  23 int warn = 1;
  24 int abort_p = 0;
  25 struct hash *users, *machines, *clusters, *lists, *filesys, *nfsphys;
  26 struct hash *strings, *members, *subnets;
  27 EXEC SQL BEGIN DECLARE SECTION;
  28 int dcmenable;
  29 EXEC SQL END DECLARE SECTION;
  30 struct save_queue *modtables, *sq_create();
  31 int interrupt();
  32
  33
  34 main(argc, argv)
  35 int argc;
  36 char **argv;
  37 {
  38     char **arg = argv;
  39 EXEC SQL BEGIN DECLARE SECTION;
  40     char *database;
  41 EXEC SQL END DECLARE SECTION;
  42     int ingerr();
  43     int countonly = 0;
  44
  45     database = "moira";
  46
  47     while (++arg - argv < argc) {
  48         if  (**arg == '-')
  49           switch ((*arg)[1]) {
  50           case 'd':
  51               debug = atoi((*arg)[2] ? *arg+2 : *++arg);
  52               break;
  53           case 'n':
  54               mode = MODE_NO;
  55               break;
  56           case 'y':
  57               mode = MODE_YES;
  58               break;
  59           case 'p':
  60               mode = MODE_PREEN;
  61               break;
  62           case 'a':
  63               mode = MODE_ASK;
  64               break;
  65           case 'c':
  66               countonly++;
  67               break;
  68           case 'f':
  69               fast++;
  70               break;
  71           case 'w':
  72               warn = 0;
  73               break;
  74           default:
  75               printf("Usage: %s [-d level] [-n] [-y] [-p] [-a] [-c] [-f] [-w] [database]\n",
  76                      argv[0]);
  77               exit(1);
  78           }
  79         else
  80           database = *arg;
  81     }
  82     if (countonly)
  83       printf("Only doing counts\n");
  84     else if (fast)
  85       printf("Doing fast version (skipping some checks)\n");
  86     if (mode == MODE_NO)
  87       printf("Will NOT modify the database\n");
  88     else if (mode == MODE_PREEN)
  89       printf("Will fix simple things without asking\n");
  90     else if (mode == MODE_YES)
  91       printf("Will fix everything without asking\n");
  92     if (debug)
  93       printf("Debug level is %d\n", debug);
  94
  95     setlinebuf(stdout);
  96
  97     signal(SIGHUP, interrupt);
  98     signal(SIGQUIT, interrupt);
  99     signal(SIGINT, interrupt);
 100     modtables = sq_create();
 101
 102     IIseterr(ingerr);
 103     printf("Opening database %s...", database);
 104     fflush(stdout);
 105     EXEC SQL CONNECT :database;
 106     printf("done\n");
 107     EXEC SQL SELECT value INTO :dcmenable FROM numvalues
 108         WHERE name='dcm_enable';
 109     dprintf("DCM disabled (was %d)\n", dcmenable);
 110     EXEC SQL UPDATE numvalues SET value=0 WHERE name='dcm_enable';
 111
 112     /* Begin transaction here. */
 113
 114     if (!countonly) {
 115         phase1();
 116         EXEC SQL COMMIT WORK;
 117         phase2();
 118         EXEC SQL COMMIT WORK;
 119         phase3();
 120         EXEC SQL COMMIT WORK;
 121     } else {
 122         count_only_setup();
 123         EXEC SQL COMMIT WORK;
 124     }
 125     phase4();
 126
 127     EXEC SQL COMMIT WORK;
 128
 129     cleanup();
 130     printf("Done.\n");
 131     exit(0);
 132 }
 133
 134 ingerr(num)
 135 int     *num;
 136 {
 137     if (*num == 100) return;
 138     printf("An ingres error occuurred, code %d\n", *num);
 139     printf("Aborting...\n");
 140     if (!abort_p) {
 141         abort_p++;
 142         EXEC SQL ROLLBACK WORK;
 143     }
 144     exit(1);
 145 }
 146
 147
 148 int interrupt()
 149 {
 150     printf("Signal caught\n");
 151     if (prompt("Save database changes")) {
 152         /* break out of a retrieve loop */
 153         IIbreak();
 154
 155         EXEC SQL COMMIT WORK;
 156         cleanup();
 157         exit(0);
 158     }
 159     printf("Aborting transaction\n");
 160     if (!abort_p) {
 161         abort_p++;
 162         /* break out of a retrieve loop */
 163         IIbreak();
 164         EXEC SQL ROLLBACK WORK;
 165     }
 166
 167     EXEC SQL UPDATE numvalues SET value=:dcmenable
 168         WHERE name='dcm_enable';
 169
 170     exit(0);
 171 }
 172
 173
 174 modified(table)
 175 char *table;
 176 {
 177     sq_save_unique_string(modtables, table);
 178 }
 179
 180 cleanup()
 181 {
 182     EXEC SQL BEGIN DECLARE SECTION;
 183     char *tab;
 184     EXEC SQL END DECLARE SECTION;
 185
 186     while (sq_get_data(modtables, &tab)) {
 187         EXEC SQL REPEATED UPDATE tblstats SET modtime='now'
 188             WHERE table_name = :tab;
 189     }
 190     EXEC SQL UPDATE numvalues SET value = :dcmenable
 191         WHERE name='dcm_enable';
 192 }
 193
 194
 195 out_of_mem(msg)
 196 char *msg;
 197 {
 198     fprintf(stderr, "Out of memory while %s\n", msg);
 199     if (prompt("Save database changes")) {
 200         /* break out of a retrieve loop */
 201         IIbreak();
 202
 203         EXEC SQL COMMIT WORK;
 204         cleanup();
 205         exit(1);
 206     }
 207     printf("Aborting transaction\n");
 208     IIbreak();
 209     EXEC SQL ROLLBACK WORK;
 210     exit(1);
 211 }