414 lines
9.1 KiB
C
414 lines
9.1 KiB
C
/*
|
|
* Empire - A multi-player, client/server Internet based war game.
|
|
* Copyright (C) 1986-2005, Dave Pare, Jeff Bailey, Thomas Ruschak,
|
|
* Ken Stevens, Steve McClure
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*
|
|
* ---
|
|
*
|
|
* See the "LEGAL", "LICENSE", "CREDITS" and "README" files for all the
|
|
* related information and legal notices. It is expected that any future
|
|
* projects/authors will amend these files as needed.
|
|
*
|
|
* ---
|
|
*
|
|
* file.c: Misc. operations on files
|
|
*
|
|
* Known contributors to this file:
|
|
* Dave Pare, 1989
|
|
* Steve McClure, 2000
|
|
*/
|
|
|
|
#include <string.h>
|
|
#include <fcntl.h>
|
|
#include <signal.h>
|
|
#include <stdlib.h>
|
|
#if !defined(_WIN32)
|
|
#include <unistd.h>
|
|
#endif
|
|
#include "misc.h"
|
|
#include "nsc.h"
|
|
#include "file.h"
|
|
#include "common.h"
|
|
#include "gen.h"
|
|
|
|
|
|
static void fillcache(struct empfile *ep, int start);
|
|
|
|
/*
|
|
* Open the binary file for table TYPE (EF_SECTOR, ...).
|
|
* MODE is passed to open().
|
|
* HOW are EFF_OPEN flags to control operation.
|
|
* Return non-zero on success, zero on failure.
|
|
* You must call ef_close() before the next ef_open().
|
|
*/
|
|
int
|
|
ef_open(int type, int mode, int how)
|
|
{
|
|
struct empfile *ep;
|
|
int size;
|
|
|
|
#if defined(_WIN32)
|
|
mode |= O_BINARY;
|
|
#endif
|
|
if (ef_check(type) < 0)
|
|
return 0;
|
|
if (CANT_HAPPEN(how & ~EFF_OPEN))
|
|
how &= EFF_OPEN;
|
|
ep = &empfile[type];
|
|
if (CANT_HAPPEN(ep->fd >= 0))
|
|
return 0;
|
|
if ((ep->fd = open(ep->file, mode, 0660)) < 0) {
|
|
logerror("%s: open failed", ep->file);
|
|
return 0;
|
|
}
|
|
ep->baseid = 0;
|
|
ep->cids = 0;
|
|
ep->mode = mode;
|
|
ep->flags |= how;
|
|
ep->fids = fsize(ep->fd) / ep->size;
|
|
if (ep->flags & EFF_MEM)
|
|
ep->csize = ep->fids;
|
|
else
|
|
ep->csize = max(1, blksize(ep->fd) / ep->size);
|
|
size = ep->csize * ep->size;
|
|
ep->cache = malloc(size);
|
|
if (ep->cache == NULL) {
|
|
logerror("ef_open: %s malloc(%d) failed\n", ep->file, size);
|
|
return 0;
|
|
}
|
|
if (ep->flags & EFF_MEM) {
|
|
if (read(ep->fd, ep->cache, size) != size) {
|
|
logerror("ef_open: read(%s) failed\n", ep->file);
|
|
return 0;
|
|
}
|
|
ep->cids = size / ep->size;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
/*
|
|
* Close the file containing objects of the type 'type', flushing the cache
|
|
* if applicable.
|
|
*/
|
|
int
|
|
ef_close(int type)
|
|
{
|
|
struct empfile *ep;
|
|
int r;
|
|
|
|
if (ef_check(type) < 0)
|
|
return 0;
|
|
ep = &empfile[type];
|
|
if (ep->cache == NULL) {
|
|
/* no cache implies never opened */
|
|
return 0;
|
|
}
|
|
ef_flush(type);
|
|
ep->flags &= ~EFF_MEM;
|
|
free(ep->cache);
|
|
ep->cache = NULL;
|
|
if ((r = close(ep->fd)) < 0) {
|
|
logerror("ef_close: %s close(%d) -> %d", ep->name, ep->fd, r);
|
|
}
|
|
ep->fd = -1;
|
|
return 1;
|
|
}
|
|
|
|
/*
|
|
* Flush the cache of the file containing objects of type 'type' to disk.
|
|
*/
|
|
int
|
|
ef_flush(int type)
|
|
{
|
|
struct empfile *ep;
|
|
int size;
|
|
int r;
|
|
|
|
if (ef_check(type) < 0)
|
|
return 0;
|
|
ep = &empfile[type];
|
|
if (ep->cache == NULL) {
|
|
/* no cache implies never opened */
|
|
return 0;
|
|
}
|
|
size = ep->csize * ep->size;
|
|
if (ep->mode > 0 && (ep->flags & EFF_MEM)) {
|
|
if ((r = lseek(ep->fd, 0L, SEEK_SET)) < 0) {
|
|
logerror("ef_flush: %s cache lseek(%d, 0L, SEEK_SET) -> %d",
|
|
ep->name, ep->fd, r);
|
|
return 0;
|
|
}
|
|
if (write(ep->fd, ep->cache, size) != size) {
|
|
logerror("ef_flush: %s cache write(%d, %p, %d) -> %d",
|
|
ep->name, ep->fd, ep->cache, ep->size, r);
|
|
return 0;
|
|
}
|
|
}
|
|
/*ef_zapcache(type); */
|
|
return 1;
|
|
}
|
|
|
|
/*
|
|
* Return a pointer the id 'id' of object of type 'type' in the cache.
|
|
*/
|
|
char *
|
|
ef_ptr(int type, int id)
|
|
{
|
|
struct empfile *ep;
|
|
|
|
if (ef_check(type) < 0)
|
|
return NULL;
|
|
ep = &empfile[type];
|
|
if (id < 0 || id >= ep->fids)
|
|
return NULL;
|
|
if ((ep->flags & EFF_MEM) == 0) {
|
|
logerror("ef_ptr: (%s) only valid for EFF_MEM entries", ep->file);
|
|
return NULL;
|
|
}
|
|
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.
|
|
*/
|
|
int
|
|
ef_read(int type, int id, void *into)
|
|
{
|
|
struct empfile *ep;
|
|
void *from;
|
|
|
|
if (ef_check(type) < 0)
|
|
return 0;
|
|
ep = &empfile[type];
|
|
if (id < 0)
|
|
return 0;
|
|
if (ep->flags & EFF_MEM) {
|
|
if (id >= ep->fids)
|
|
return 0;
|
|
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)
|
|
fillcache(ep, id);
|
|
from = ep->cache + (id - ep->baseid) * ep->size;
|
|
}
|
|
memcpy(into, from, ep->size);
|
|
|
|
if (ep->postread)
|
|
ep->postread(id, into);
|
|
return 1;
|
|
}
|
|
|
|
static void
|
|
fillcache(struct empfile *ep, int start)
|
|
{
|
|
int n;
|
|
|
|
ep->baseid = start;
|
|
lseek(ep->fd, start * ep->size, SEEK_SET);
|
|
n = read(ep->fd, ep->cache, ep->csize * ep->size);
|
|
ep->cids = n / ep->size;
|
|
}
|
|
|
|
/*
|
|
* buffered write. Modifies read cache (if applicable)
|
|
* and writes through to disk.
|
|
*/
|
|
int
|
|
ef_write(int type, int id, void *from)
|
|
{
|
|
int r;
|
|
struct empfile *ep;
|
|
char *to;
|
|
|
|
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 ((r = lseek(ep->fd, id * ep->size, SEEK_SET)) < 0) {
|
|
logerror("ef_write: %s #%d lseek(%d, %d, SEEK_SET) -> %d",
|
|
ep->name, id, ep->fd, id * ep->size, r);
|
|
return 0;
|
|
}
|
|
if (ep->prewrite)
|
|
ep->prewrite(id, from);
|
|
if ((r = write(ep->fd, from, ep->size)) != ep->size) {
|
|
logerror("ef_write: %s #%d write(%d, %p, %d) -> %d",
|
|
ep->name, id, ep->fd, from, ep->size, r);
|
|
return 0;
|
|
}
|
|
if (id >= ep->baseid && id < ep->baseid + ep->cids) {
|
|
/* update the cache if necessary */
|
|
to = ep->cache + (id - ep->baseid) * ep->size;
|
|
memcpy(to, from, ep->size);
|
|
}
|
|
CANT_HAPPEN(id > ep->fids);
|
|
if (id >= ep->fids) {
|
|
if (ep->flags & EFF_MEM) {
|
|
logerror("file %s went beyond %d items; won't be able toread item w/o restart",
|
|
ep->name, ep->fids);
|
|
} else {
|
|
/* write expanded file; ep->fids = last id + 1 */
|
|
ep->fids = id + 1;
|
|
}
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
/*
|
|
* Grow the file containing objects of the type 'type' by 'count' objects.
|
|
*/
|
|
int
|
|
ef_extend(int type, int count)
|
|
{
|
|
struct empfile *ep;
|
|
char *tmpobj;
|
|
int cur, max;
|
|
int mode, how;
|
|
int r;
|
|
|
|
if (ef_check(type) < 0)
|
|
return 0;
|
|
ep = &empfile[type];
|
|
max = ep->fids + count;
|
|
cur = ep->fids;
|
|
tmpobj = calloc(1, ep->size);
|
|
if ((r = lseek(ep->fd, ep->fids * ep->size, SEEK_SET)) < 0) {
|
|
logerror("ef_extend: %s +#%d lseek(%d, %d, SEEK_SET) -> %d",
|
|
ep->name, count, ep->fd, ep->fids * ep->size, r);
|
|
free(tmpobj);
|
|
return 0;
|
|
}
|
|
for (cur = ep->fids; cur < max; cur++) {
|
|
if (ep->init)
|
|
ep->init(cur, tmpobj);
|
|
if ((r = write(ep->fd, tmpobj, ep->size)) != ep->size) {
|
|
logerror("ef_extend: %s +#%d write(%d, %p, %d) -> %d",
|
|
ep->name, count, ep->fd, tmpobj, ep->size, r);
|
|
free(tmpobj);
|
|
return 0;
|
|
}
|
|
}
|
|
free(tmpobj);
|
|
if (ep->flags & EFF_MEM) {
|
|
/* XXX this will cause problems if there are ef_ptrs (to the
|
|
* old allocated structure) active when we do the re-open */
|
|
mode = ep->mode;
|
|
how = ep->flags;
|
|
ef_close(type);
|
|
ef_open(type, mode, how);
|
|
} else {
|
|
ep->fids += count;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
/*
|
|
* 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)
|
|
{
|
|
return empfile[type].cadef;
|
|
}
|
|
|
|
int
|
|
ef_nelem(int type)
|
|
{
|
|
return empfile[type].fids;
|
|
}
|
|
|
|
int
|
|
ef_flags(int type)
|
|
{
|
|
return empfile[type].flags;
|
|
}
|
|
|
|
time_t
|
|
ef_mtime(int type)
|
|
{
|
|
if (empfile[type].fd <= 0)
|
|
return 0;
|
|
return fdate(empfile[type].fd);
|
|
}
|
|
|
|
/*
|
|
* Search empfile[0..EF_MAX-1] for element named NAME.
|
|
* Return its index in empfile[] if found, else -1.
|
|
*/
|
|
int
|
|
ef_byname(char *name)
|
|
{
|
|
struct empfile *ef;
|
|
int i;
|
|
int len;
|
|
|
|
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;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
char *
|
|
ef_nameof(int type)
|
|
{
|
|
if (type < 0 || type >= EF_MAX)
|
|
return "bad item 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);
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
ef_ensure_space(int type, int id, int count)
|
|
{
|
|
while (id >= empfile[type].fids) {
|
|
if (!ef_extend(type, count))
|
|
return 0;
|
|
}
|
|
return 1;
|
|
}
|