Added support for optional debugging malloc. No code changes if you are

[moira.git] / dbck / dbck.dc

diff --git a/dbck/dbck.dc b/dbck/dbck.dc
index 12a214b33a66b7fdaced6192a919f2a6e8ad9cd6..525d567a181e81ecd875e41c5bb13bdd7fb5ba52 100644 (file)
--- a/dbck/dbck.dc
+++ b/dbck/dbck.dc
@@ -23,7 +23,7 @@ int fast = 0;
 int warn = 1;
 int abort_p = 0;
 struct hash *users, *machines, *clusters, *lists, *filesys, *nfsphys;
-struct hash *strings, *members;
+struct hash *strings, *members, *subnets;
 EXEC SQL BEGIN DECLARE SECTION; 
 int dcmenable;
 EXEC SQL END DECLARE SECTION; 
@@ -102,24 +102,25 @@ EXEC SQL END DECLARE SECTION;
     IIseterr(ingerr);
     printf("Opening database %s...", database);
     fflush(stdout);
-/*  ingres database */
     EXEC SQL CONNECT :database;
     printf("done\n");
-/*  retrieve (dcmenable = values.value) where values.name = "dcm_enable" */
     EXEC SQL SELECT value INTO :dcmenable FROM numvalues 
        WHERE name='dcm_enable'; 
     dprintf("DCM disabled (was %d)\n", dcmenable);
-/*  replace values (value = 0) where values.name = "dcm_enable" */
     EXEC SQL UPDATE numvalues SET value=0 WHERE name='dcm_enable';
 
     /* Begin transaction here. */
 
     if (!countonly) {
        phase1();
+       EXEC SQL COMMIT WORK;
        phase2();
+       EXEC SQL COMMIT WORK;
        phase3();
+       EXEC SQL COMMIT WORK;
     } else {
        count_only_setup();
+       EXEC SQL COMMIT WORK;
     }
     phase4();
 
@@ -133,6 +134,7 @@ EXEC SQL END DECLARE SECTION;
 ingerr(num)
 int    *num;
 {
+    if (*num == 100) return;
     printf("An ingres error occuurred, code %d\n", *num);
     printf("Aborting...\n");
     if (!abort_p) {
@@ -162,12 +164,9 @@ int interrupt()
        EXEC SQL ROLLBACK WORK;
     }
 
-/*  replace values (value = dcmenable) where values.name = "dcm_enable" */
     EXEC SQL UPDATE numvalues SET value=:dcmenable
        WHERE name='dcm_enable';
 
-/*  exit */
-    /* No equivalent (?) */
     exit(0);
 }
 
@@ -185,15 +184,11 @@ cleanup()
     EXEC SQL END DECLARE SECTION; 
 
     while (sq_get_data(modtables, &tab)) {
-/*     replace tblstats (modtime = "now") where tblstats.table = tab */
        EXEC SQL REPEATED UPDATE tblstats SET modtime='now'
            WHERE table_name = :tab;
     }
-/*  replace values (value = dcmenable) where values.name = "dcm_enable" */
     EXEC SQL UPDATE numvalues SET value = :dcmenable
        WHERE name='dcm_enable';
-/*  exit */
-    /* No equivalent (?) */
 }
 
 
@@ -201,8 +196,16 @@ out_of_mem(msg)
 char *msg;
 {
     fprintf(stderr, "Out of memory while %s\n", msg);
-/*  end transaction    */
-    EXEC SQL COMMIT WORK;
-    cleanup();
+    if (prompt("Save database changes")) {
+       /* break out of a retrieve loop */
+       IIbreak();
+
+       EXEC SQL COMMIT WORK;
+       cleanup();
+       exit(1);
+    }
+    printf("Aborting transaction\n");
+    IIbreak();
+    EXEC SQL ROLLBACK WORK;
     exit(1);
 }