]> git.pond.sub.org Git - empserver/blobdiff - src/lib/common/file.c
(ef_ptr, ef_read): Oops on bad ID. Some callers rely on these
[empserver] / src / lib / common / file.c
index 224be2ba28324a61259cbd5e4754735c321a7b9f..fd6e8c0f2433ba9e324af1062d54f2b369df9a8a 100644 (file)
  *
  *  ---
  *
- *  file.c: Misc. operations on files
+ *  file.c: Operations on Empire tables (`files' for historical reasons)
  * 
  *  Known contributors to this file:
  *     Dave Pare, 1989
  *     Steve McClure, 2000
+ *     Markus Armbruster, 2005
  */
 
 #include <errno.h>
 #if !defined(_WIN32)
 #include <unistd.h>
 #endif
-#include "misc.h"
-#include "nsc.h"
-#include "file.h"
 #include "common.h"
+#include "file.h"
 #include "gen.h"
-
+#include "match.h"
+#include "misc.h"
+#include "nsc.h"
+#include "optlist.h"
 
 static int fillcache(struct empfile *, int);
 static int do_write(struct empfile *, void *, int, int);
 
-
 /*
- * Open the binary file for table TYPE (EF_SECTOR, ...).
- * HOW are EFF_OPEN flags to control operation.
+ * Open the file-backed table TYPE (EF_SECTOR, ...).
+ * HOW are flags to control operation.  Naturally, immutable flags are
+ * not permitted.
  * Return non-zero on success, zero on failure.
  * You must call ef_close() before the next ef_open().
  */
@@ -65,8 +67,10 @@ ef_open(int type, int how)
 
     if (ef_check(type) < 0)
        return 0;
-    if (CANT_HAPPEN(how & ~EFF_OPEN))
-       how &= EFF_OPEN;
+    if (CANT_HAPPEN(how & EFF_IMMUTABLE))
+       how &= ~EFF_IMMUTABLE;
+
+    /* open file */
     ep = &empfile[type];
     if (CANT_HAPPEN(ep->fd >= 0))
        return 0;
@@ -82,6 +86,8 @@ ef_open(int type, int how)
        logerror("Can't open %s (%s)", ep->file, strerror(errno));
        return 0;
     }
+
+    /* get file size */
     fsiz = fsize(fd);
     if (fsiz % ep->size) {
        logerror("Can't open %s (file size not a multiple of record size %d)",
@@ -90,39 +96,59 @@ ef_open(int type, int how)
        return 0;
     }
     ep->fids = fsiz / ep->size;
-    if (how & EFF_MEM)
-       ep->csize = ep->fids;
-    else
-       ep->csize = max(1, blksize(fd) / ep->size);
-    size = ep->csize * ep->size;
-    if (size) {
+
+    /* allocate cache */
+    if (ep->flags & EFF_STATIC) {
+       /* ep->cache already points to space for e->csize elements */
+       if (how & EFF_MEM) {
+           if (ep->fids > ep->csize) {
+               logerror("Can't open %s: file larger than %d bytes",
+                        ep->file, ep->fids * ep->size);
+               close(fd);
+               return 0;
+           }
+       }
+    } else {
+       if (how & EFF_MEM)
+           ep->csize = ep->fids;
+       else
+           ep->csize = max(1, blksize(fd) / ep->size);
+       size = ep->csize * ep->size;
+       if (CANT_HAPPEN(ep->cache))
+           free(ep->cache);
        ep->cache = malloc(size);
-       if (ep->cache == NULL) {
+       if (ep->cache == NULL && size) {
            logerror("Can't open %s: out of memory", ep->file);
            close(fd);
            return 0;
        }
-    } else {
-       ep->cache = NULL;
     }
     ep->baseid = 0;
     ep->cids = 0;
-    ep->flags = (ep->flags & ~EFF_OPEN) | (how & ~EFF_CREATE);
+    ep->flags = (ep->flags & EFF_IMMUTABLE) | (how & ~EFF_CREATE);
     ep->fd = fd;
+
+    /* map file into cache */
     if ((how & EFF_MEM) && ep->fids) {
        if (fillcache(ep, 0) != ep->fids) {
            ep->cids = 0;       /* prevent cache flush */
-           ep->flags &= ~EFF_OPEN; /* maintain invariant */
+           ep->flags &= EFF_IMMUTABLE; /* maintain invariant */
            ef_close(type);
            return 0;
        }
     }
+
+    /*
+     * Could close fd if both EFF_RDONLY and EFF_MEM, but that doesn't
+     * happen, so don't bother.
+     */
+
     return 1;
 }
 
 /*
- * Close the file containing objects of the type 'type', flushing the cache
- * if applicable.
+ * Close the file-backed table TYPE (EF_SECTOR, ...).
+ * Return non-zero on success, zero on failure.
  */
 int
 ef_close(int type)
@@ -132,9 +158,11 @@ ef_close(int type)
 
     retval = ef_flush(type);
     ep = &empfile[type];
-    ep->flags &= ~EFF_OPEN;
-    free(ep->cache);
-    ep->cache = NULL;
+    ep->flags &= EFF_IMMUTABLE;
+    if (!(ep->flags & EFF_STATIC)) {
+       free(ep->cache);
+       ep->cache = NULL;
+    }
     if (close(ep->fd) < 0) {
        logerror("Error closing %s (%s)", ep->name, strerror(errno));
        retval = 0;
@@ -144,7 +172,8 @@ ef_close(int type)
 }
 
 /*
- * Flush the cache of the file containing objects of type 'type' to disk.
+ * Flush file-backed table TYPE (EF_SECTOR, ...) to disk.
+ * Return non-zero on success, zero on failure.
  */
 int
 ef_flush(int type)
@@ -169,7 +198,9 @@ ef_flush(int type)
 }
 
 /*
- * Return a pointer the id 'id' of object of type 'type' in the cache.
+ * Return pointer to element ID in table TYPE if it exists, else NULL.
+ * The table must be fully cached, i.e. flags & EFF_MEM.
+ * The caller is responsible for flushing changes he makes.
  */
 void *
 ef_ptr(int type, int id)
@@ -178,18 +209,20 @@ ef_ptr(int type, int id)
 
     if (ef_check(type) < 0)
        return NULL;
-
     ep = &empfile[type];
-    if (CANT_HAPPEN(!(ep->flags & EFF_MEM)))
+    if (CANT_HAPPEN(!(ep->flags & EFF_MEM) || !ep->cache))
+       return NULL;
+    if (id < 0 || id >= ep->fids) {
+       CANT_HAPPEN(id != ep->fids);
        return NULL;
-    if (id < 0 || id >= ep->fids)
-       return NULL;            /* FIXME can this happen? */
+    }
     return ep->cache + ep->size * id;
 }
 
 /*
- * buffered read.  Tries to read a large number of items.
- * This system won't work if item size is > sizeof buffer area.
+ * Read element ID from table TYPE into buffer INTO.
+ * FIXME pass buffer size!
+ * Return non-zero on success, zero on failure.
  */
 int
 ef_read(int type, int id, void *into)
@@ -200,21 +233,20 @@ ef_read(int type, int id, void *into)
     if (ef_check(type) < 0)
        return 0;
     ep = &empfile[type];
-    if (id < 0)
+    if (CANT_HAPPEN(!ep->cache))
+       return 0;
+    if (id < 0 || id >= ep->fids) {
+       CANT_HAPPEN(id != ep->fids);
        return 0;
+    }
+
     if (ep->flags & EFF_MEM) {
-       if (id >= ep->fids)
-           return 0;
-       from = ep->cache + (id * ep->size);
+       from = ep->cache + id * ep->size;
     } else {
-       if (id >= ep->fids) {
-           ep->fids = fsize(ep->fd) / ep->size;
-           if (id >= ep->fids)
-               return 0;
-       }
-       if (ep->baseid + ep->cids <= id || ep->baseid > id)
+       if (ep->baseid + ep->cids <= id || ep->baseid > id) {
            if (fillcache(ep, id) < 1)
                return 0;
+       }
        from = ep->cache + (id - ep->baseid) * ep->size;
     }
     memcpy(into, from, ep->size);
@@ -306,8 +338,12 @@ do_write(struct empfile *ep, void *buf, int id, int count)
 }
 
 /*
- * buffered write.  Modifies read cache (if applicable)
- * and writes through to disk.
+ * Write element ID into file-backed table TYPE from buffer FROM.
+ * FIXME pass buffer size!
+ * Write through cache straight to disk.
+ * Cannot write beyond the end of fully cached table (flags & EFF_MEM).
+ * Can write at the end of partially cached table.
+ * Return non-zero on success, zero on failure.
  */
 int
 ef_write(int type, int id, void *from)
@@ -318,11 +354,6 @@ ef_write(int type, int id, void *from)
     if (ef_check(type) < 0)
        return 0;
     ep = &empfile[type];
-    if (id > 65536) {
-       /* largest unit id; this may bite us in large games */
-       logerror("ef_write: type %d id %d is too large!\n", type, id);
-       return 0;
-    }
     if (ep->prewrite)
        ep->prewrite(id, from);
     if (CANT_HAPPEN((ep->flags & EFF_MEM) ? id >= ep->fids : id > ep->fids))
@@ -342,7 +373,8 @@ ef_write(int type, int id, void *from)
 }
 
 /*
- * Grow the file containing objects of the type 'type' by 'count' objects.
+ * Extend the file-backed table TYPE by COUNT elements.
+ * Return non-zero on success, zero on failure.
  */
 int
 ef_extend(int type, int count)
@@ -371,7 +403,7 @@ ef_extend(int type, int count)
        /* FIXME lazy bastards...  do this right */
        /* XXX this will cause problems if there are ef_ptrs (to the
         * old allocated structure) active when we do the re-open */
-       how = ep->flags & EFF_OPEN;
+       how = ep->flags & ~EFF_IMMUTABLE;
        ef_close(type);
        ef_open(type, how);
     } else {
@@ -381,19 +413,6 @@ ef_extend(int type, int count)
     return i == count;
 }
 
-/*
- * Mark the cache for the file containing objects of type 'type' as unused.
- */
-void
-ef_zapcache(int type)
-{
-    struct empfile *ep = &empfile[type];
-    if ((ep->flags & EFF_MEM) == 0) {
-       ep->cids = 0;
-       ep->baseid = -1;
-    }
-}
-
 struct castr *
 ef_cadef(int type)
 {
@@ -421,49 +440,134 @@ ef_mtime(int type)
 }
 
 /*
- * Search empfile[0..EF_MAX-1] for element named NAME.
- * Return its index in empfile[] if found, else -1.
+ * Search for a table matching NAME, return its table type.
+ * Return M_NOTFOUND if there are no matches, M_NOTUNIQUE if there are
+ * several.
  */
 int
 ef_byname(char *name)
 {
-    struct empfile *ef;
-    int i;
-    int len;
+    return stmtch(name, empfile, offsetof(struct empfile, name),
+                 sizeof(empfile[0]));
+}
 
-    len = strlen(name);
-    for (i = 0; i < EF_MAX; i++) {
-       ef = &empfile[i];
-       if (strncmp(ef->name, name, min(len, strlen(ef->name))) == 0)
-           return i;
+/*
+ * Search CHOICES[] for a table type matching NAME, return it.
+ * Return M_NOTFOUND if there are no matches, M_NOTUNIQUE if there are
+ * several.
+ * CHOICES[] must be terminated with a negative value.
+ */
+int
+ef_byname_from(char *name, int choices[])
+{
+    int res;
+    int *p;
+
+    res = M_NOTFOUND;
+    for (p = choices; *p >= 0; p++) {
+       if (ef_check(*p) < 0)
+           continue;
+       switch (mineq(name, empfile[*p].name)) {
+       case ME_MISMATCH:
+           break;
+       case ME_PARTIAL:
+           if (res >= 0)
+               return M_NOTUNIQUE;
+           res = *p;
+           break;
+       case ME_EXACT:
+           return *p;
+       }
     }
-    return -1;
+    return res;
 }
 
 char *
 ef_nameof(int type)
 {
-    if (type < 0 || type >= EF_MAX)
-       return "bad item type";
+    if (ef_check(type) < 0)
+       return "bad ef_type";
     return empfile[type].name;
 }
 
 int
 ef_check(int type)
 {
-    if (type < 0 || type >= EF_MAX) {
-       logerror("ef_ptr: bad EF_type %d\n", type);
+    if (CANT_HAPPEN((unsigned)type >= EF_MAX))
        return -1;
-    }
     return 0;
 }
 
+/*
+ * Ensure file-backed table contains ID.
+ * If necessary, extend it in steps of COUNT elements.
+ * Return non-zero on success, zero on failure.
+ */
 int
 ef_ensure_space(int type, int id, int count)
 {
+    if (ef_check(type) < 0)
+       return 0;
+    CANT_HAPPEN(id < 0);
+
     while (id >= empfile[type].fids) {
        if (!ef_extend(type, count))
            return 0;
     }
     return 1;
 }
+
+static void
+ef_fix_size(struct empfile *ep, int n)
+{
+    ep->cids = ep->fids = n;
+    ep->csize = n + 1;
+}
+
+static void
+ef_init_chr(int type, size_t size, ptrdiff_t name_offs)
+{
+    struct empfile *ep = &empfile[type];
+    char *p;
+
+    for (p = ep->cache;
+        *((char **)(p + name_offs)) && **((char **)(p + name_offs));
+        p += size) ;
+    ep->cids = ep->fids = (p - ep->cache) / size;
+}
+
+/*
+ * Initialize Empire tables.
+ * Must be called once, before using anything else from this module.
+ */
+void
+ef_init(void)
+{
+    struct castr *ca;
+    struct empfile *ep;
+    struct symbol *lup;
+    int i;
+
+    empfile[EF_MAP].size = empfile[EF_BMAP].size = (WORLD_X * WORLD_Y) / 2;
+
+    ef_init_chr(EF_SHIP_CHR,
+               sizeof(struct mchrstr), offsetof(struct mchrstr, m_name));
+    ef_init_chr(EF_PLANE_CHR,
+               sizeof(struct plchrstr), offsetof(struct plchrstr, pl_name));
+    ef_init_chr(EF_LAND_CHR,
+               sizeof(struct lchrstr), offsetof(struct lchrstr, l_name));
+    ef_init_chr(EF_NUKE_CHR,
+               sizeof(struct nchrstr), offsetof(struct nchrstr, n_name));
+
+    ca = (struct castr *)empfile[EF_META].cache;
+    for (i = 0; ca[i].ca_name; i++) ;
+    ef_fix_size(&empfile[EF_META], i);
+
+    for (ep = empfile; ep->uid >= 0; ep++) {
+       if (ep->cadef == symbol_ca) {
+           lup = (struct symbol *)ep->cache;
+           for (i = 0; lup[i].name; i++) ;
+           ef_fix_size(ep, i);
+       }
+    }
+}