dbck/dbck.pc

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