]> git.pond.sub.org Git - empserver/blobdiff - src/lib/common/xundump.c
Clean up xundump's test whether table may be truncated
[empserver] / src / lib / common / xundump.c
index 490a9e74428bab7fa786d0bf3982931481b48efe..515f2863e37d14051509eb4128f1ce33e881a8e5 100644 (file)
 
 static char *fname;            /* Name of file being read */
 static int lineno;             /* Current line number */
-static int human;              /* Reading human-readable syntax? */
-static int ellipsis;           /* Header ended with ...? */
-static int is_partial;         /* Is input split into parts? */
+
 static int cur_type;           /* Current table's file type */
 static void *cur_obj;          /* The object being read into */
 static int cur_id;             /* and its index in the table */
 static int cur_obj_is_blank;
+
+static int human;              /* Reading human-readable syntax? */
+static int ellipsis;           /* Header ended with ...? */
+static int is_partial;         /* Is input split into parts? */
 static int nflds;              /* #fields in input records */
 static struct castr **fldca;   /* Map field number to selector */
 static int *fldidx;            /* Map field number to index */
 static int *caflds;            /* Map selector number to #fields seen */
 static int *cafldspp;          /* ditto, in previous parts */
+static int may_trunc;          /* Okay to truncate? */
 
 static int gripe(char *, ...) ATTRIBUTE((format (printf, 1, 2)));
 static int deffld(int, char *, int);
@@ -110,6 +113,95 @@ gripe(char *fmt, ...)
     return -1;
 }
 
+/* Make TYPE the current table.  */
+static void
+tbl_start(int type)
+{
+    cur_type = type;
+    cur_id = -1;
+    cur_obj = NULL;
+}
+
+/* End the current table.  */
+static void
+tbl_end(void)
+{
+    tbl_start(EF_BAD);
+}
+
+/*
+ * Seek to current table's ID-th record.
+ * ID must be acceptable.
+ * Store it in cur_obj, and set cur_id and cur_obj_is_blank accordingly.
+ * Return 0 on success, -1 on failure.
+ */
+static int
+tbl_seek(int id)
+{
+    struct empfile *ep = &empfile[cur_type];
+
+    cur_obj_is_blank = id >= ep->fids;
+
+    if (id >= ef_nelem(cur_type)) {
+       if (!ef_ensure_space(cur_type, id, 1))
+           return gripe("Can't put ID %d into table %s", id, ep->name);
+    }
+
+    cur_obj = ef_ptr(cur_type, id);
+    if (CANT_HAPPEN(!cur_obj))
+       return -1;
+    cur_id = id;
+    return 0;
+}
+
+/*
+ * Get the next object.
+ * Must not have a record index.
+ * Store it in cur_obj, and set cur_id and cur_obj_is_blank accordingly.
+ * Return 0 on success, -1 on failure.
+ */
+static int
+tbl_next_obj(void)
+{
+    int max_id = ef_id_limit(cur_type);
+
+    if (cur_id >= max_id)
+       return gripe("Too many rows");
+    return tbl_seek(cur_id + 1);
+}
+
+/*
+ * Get the next object, it has record index ID.
+ * Store it in cur_obj, and set cur_id and cur_obj_is_blank accordingly.
+ * Return 0 on success, -1 on failure.
+ */
+static int
+tbl_skip_to_obj(int id)
+{
+    int max_id;
+
+    if (id < 0)
+       return gripe("Field %d must be >= 0", 1);
+    max_id = ef_id_limit(cur_type);
+    if (id > max_id)
+       return gripe("Field %d must be <= %d", 1, max_id);
+
+    if (tbl_seek(id) < 0)
+       return -1;
+    return 0;
+}
+
+/*
+ * Finish table part.
+ * If the table has variable length, truncate it.
+ */
+static void
+tbl_part_done(void)
+{
+    cur_id = -1;
+    cur_obj = NULL;
+}
+
 /*
  * Read and ignore field separators from FP.
  * Return first character that is not a field separator.
@@ -478,36 +570,6 @@ fldval_must_match(int fldno)
        || fldidx[fldno] < cafldspp[i];
 }
 
-/*
- * Get the current object.
- * Store it in cur_obj, and set cur_obj_is_blank accordingly.
- * Return cur_obj, which is null on error.
- */
-static void *
-getobj(void)
-{
-    struct empfile *ep = &empfile[cur_type];
-    int max_id;
-
-    if (!cur_obj) {
-       cur_obj_is_blank = cur_id >= ep->fids;
-       if (cur_obj_is_blank) {
-           max_id = ef_id_limit(cur_type);
-           if (cur_id > max_id)
-               gripe("Can't put ID %d into table %s, it holds only 0..%d",
-                     cur_id, ep->name, max_id);
-           else if (!ef_ensure_space(cur_type, cur_id, 1))
-               gripe("Can't put ID %d into table %s",
-                     cur_id, ep->name);
-           else
-               cur_obj = ef_ptr(cur_type, cur_id);
-       } else
-           cur_obj = ef_ptr(cur_type, cur_id);
-    }
-
-    return cur_obj;
-}
-
 /*
  * Set value of field FLDNO in current object to DBL.
  * Return 1 on success, -1 on error.
@@ -516,7 +578,7 @@ static int
 setnum(int fldno, double dbl)
 {
     struct castr *ca;
-    int idx;
+    int next_id, idx;
     char *memb_ptr;
     double old, new;
 
@@ -524,18 +586,20 @@ setnum(int fldno, double dbl)
     if (!ca)
        return -1;
 
-    /*
-     * If this is the record index, put it into cur_id.
-     */
-    if (fldno == 0 && ca->ca_table == cur_type) {
-       if (dbl < 0 || (int)dbl != dbl)
-           return gripe("Field %d can't hold this value", fldno + 1);
-       cur_id = (int)dbl;
+    if (fldno == 0) {
+       if (ca->ca_table == cur_type) {
+           /* Got record index */
+           next_id = (int)dbl;
+           if (next_id != dbl)
+               return gripe("Field %d can't hold this value", fldno + 1);
+           if (tbl_skip_to_obj(next_id) < 0)
+               return -1;
+       } else {
+           if (tbl_next_obj() < 0)
+               return -1;
+       }
     }
-
-    memb_ptr = getobj();
-    if (!memb_ptr)
-       return -1;
+    memb_ptr = cur_obj;
     memb_ptr += ca->ca_off;
 
     switch (ca->ca_type) {
@@ -624,9 +688,11 @@ setstr(int fldno, char *str)
     if (!ca)
        return -1;
 
-    memb_ptr = getobj();
-    if (!memb_ptr)
-       return -1;
+    if (fldno == 0) {
+       if (tbl_next_obj() < 0)
+           return -1;
+    }
+    memb_ptr = cur_obj;
     memb_ptr += ca->ca_off;
     must_match = fldval_must_match(fldno);
 
@@ -883,6 +949,7 @@ xufooter(FILE *fp, struct castr ca[], int recs)
     }
     if (skipfs(fp) != '\n')
        return gripe("Junk after table footer");
+    tbl_part_done();
     lineno++;
 
     for (i = 0; ca[i].ca_name; i++) {
@@ -919,19 +986,24 @@ xundump(FILE *fp, char *file, int *plno, int expected_table)
        return -1;
 
     nca = nf = 0;
+    may_trunc = empfile[type].nent < 0;
     for (i = 0; ca[i].ca_name; i++) {
        nca++;
-       if (!(ca[i].ca_flags & NSC_EXTRA))
+       if (!(ca[i].ca_flags & NSC_EXTRA)) {
            nf += MAX(1, ca[i].ca_type != NSC_STRINGY ? ca[i].ca_len : 0);
+           if (ca[i].ca_flags & NSC_CONST)
+               may_trunc = 0;
+       }
     }
     fldca = malloc(nf * sizeof(*fldca));
     fldidx = malloc(nf * sizeof(*fldidx));
     caflds = malloc(nca * sizeof(*caflds));
     cafldspp = calloc(nca, sizeof(*cafldspp));
-    cur_type = type;
 
+    tbl_start(type);
     if (xutail(fp, ca) < 0)
        type = EF_BAD;
+    tbl_end();
 
     free(cafldspp);
     free(caflds);
@@ -989,8 +1061,6 @@ xubody(FILE *fp)
        if (ch == '/')
            break;
        ungetc(ch, fp);
-       cur_obj = NULL;
-       cur_id = i;
        if (xuflds(fp, xufld) < 0)
            return -1;
        maxid = MAX(maxid, cur_id + 1);
@@ -999,9 +1069,7 @@ xubody(FILE *fp)
     if (CANT_HAPPEN(maxid > ep->fids))
        maxid = ep->fids;
     if (maxid < ep->fids) {
-       if (ep->nent < 0
-           && (EF_IS_GAME_STATE(cur_type)
-               || (cur_type >= EF_SHIP_CHR && cur_type <= EF_NUKE_CHR))) {
+       if (may_trunc) {
            if (!ef_truncate(cur_type, maxid))
                return -1;
        } else