randomized continuation lines

[moira.git] / dbck / phase4.qc

diff --git a/dbck/phase4.qc b/dbck/phase4.qc
index 367bb3f415a1a5c16b1abc414c2a797a8524805e..535a22a3be1cb2f1b8bad4a6340120539c053c10 100644 (file)
--- a/dbck/phase4.qc
+++ b/dbck/phase4.qc
@@ -6,6 +6,7 @@
  */
 
 #include <mit-copyright.h>
+#include <moira.h>
 #include "dbck.h"
 
 static char phase4_qc_rcsid[] = "$Header$";
@@ -121,6 +122,8 @@ count_only_setup()
 ##           id2 = u.pop_id, id3 = u.box_id) 
 ##    where u.potype = "POP" {
        u = (struct user *) malloc(sizeof(struct user));
+       if (u == NULL)
+         out_of_mem("storing users");
        strcpy(u->login, strtrim(name));
        u->potype = buf[0];
        sprintf(buf, "%s, %s", strtrim(last), strtrim(first));
@@ -145,6 +148,8 @@ count_only_setup()
 ##  range of m is machine
 ##  retrieve (id = m.mach_id, name = m.#name) {
        m = (struct machine *) malloc(sizeof(struct machine));
+       if (m == NULL)
+         out_of_mem("storing machines");
        strcpy(m->name, strtrim(name));
        m->mach_id = id;
        hash_store(machines, id, m);
@@ -155,6 +160,8 @@ count_only_setup()
 ##  retrieve (id = nfsphys.nfsphys_id, name = nfsphys.dir,
 ##           id2 = nfsphys.mach_id, id3 = nfsphys.allocated) {
        n = (struct nfsphys *) malloc(sizeof(struct nfsphys));
+       if (n == NULL)
+         out_of_mem("storing nfsphys");
        strcpy(n->dir, strtrim(name));
        n->mach_id = id2;
        n->nfsphys_id = id;
@@ -164,7 +171,7 @@ count_only_setup()
 ##  }
 
     dprintf("Counting quotas...\n");
-##  retrieve (id = nfsquota.phys_id, id2 = nfsquota.quota) {
+##  retrieve (id = quota.phys_id, id2 = quota.quota) {
         if (n = (struct nfsphys  *) hash_lookup(nfsphys, id)) {
            n->count += id2;
        }