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