-/* $Id: bfile.c,v 1.45 2005-09-19 09:37:31 adam Exp $
- Copyright (C) 1995-2005
+/* $Id: bfile.c,v 1.54 2007-01-15 15:10:13 adam Exp $
+ Copyright (C) 1995-2007
Index Data ApS
This file is part of the Zebra server.
for more details.
You should have received a copy of the GNU General Public License
-along with Zebra; see the file LICENSE.zebra. If not, write to the
-Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
-02111-1307, USA.
+along with this program; if not, write to the Free Software
+Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
+
*/
#include <stdio.h>
struct BFiles_struct {
MFile_area commit_area;
- MFile_area_struct *register_area;
+ MFile_area register_area;
char *base;
char *cache_fname;
};
BFiles bfs_create (const char *spec, const char *base)
{
- BFiles bfs = (BFiles) xmalloc (sizeof(*bfs));
- bfs->commit_area = NULL;
+ BFiles bfs = (BFiles) xmalloc(sizeof(*bfs));
+ bfs->commit_area = 0;
bfs->base = 0;
bfs->cache_fname = 0;
if (base)
- bfs->base = xstrdup (base);
- bfs->register_area = mf_init("register", spec, base);
+ bfs->base = xstrdup(base);
+ bfs->register_area = mf_init("register", spec, base, 0);
if (!bfs->register_area)
{
bfs_destroy(bfs);
return bfs;
}
-void bfs_destroy (BFiles bfs)
+void bfs_destroy(BFiles bfs)
{
if (!bfs)
return;
- xfree (bfs->cache_fname);
- xfree (bfs->base);
- mf_destroy (bfs->commit_area);
- mf_destroy (bfs->register_area);
- xfree (bfs);
+ xfree(bfs->cache_fname);
+ xfree(bfs->base);
+ mf_destroy(bfs->commit_area);
+ mf_destroy(bfs->register_area);
+ xfree(bfs);
}
-static FILE *open_cache (BFiles bfs, const char *flags)
+static FILE *open_cache(BFiles bfs, const char *flags)
{
FILE *file;
- file = fopen (bfs->cache_fname, flags);
+ file = fopen(bfs->cache_fname, flags);
return file;
}
-static void unlink_cache (BFiles bfs)
+static void unlink_cache(BFiles bfs)
{
- unlink (bfs->cache_fname);
+ if (bfs->cache_fname)
+ unlink(bfs->cache_fname);
}
-ZEBRA_RES bf_cache (BFiles bfs, const char *spec)
+ZEBRA_RES bf_cache(BFiles bfs, const char *spec)
{
if (spec)
{
- yaz_log (YLOG_LOG, "enabling shadow spec=%s", spec);
+ yaz_log(YLOG_LOG, "enabling shadow spec=%s", spec);
if (!bfs->commit_area)
- bfs->commit_area = mf_init ("shadow", spec, bfs->base);
+ bfs->commit_area = mf_init("shadow", spec, bfs->base, 1);
if (bfs->commit_area)
{
- bfs->cache_fname = xmalloc (strlen(bfs->commit_area->dirs->name)+
+ bfs->cache_fname = xmalloc(strlen(bfs->commit_area->dirs->name)+
8);
- strcpy (bfs->cache_fname, bfs->commit_area->dirs->name);
- strcat (bfs->cache_fname, "/cache");
- yaz_log (YLOG_LOG, "cache_fname = %s", bfs->cache_fname);
+ strcpy(bfs->cache_fname, bfs->commit_area->dirs->name);
+ strcat(bfs->cache_fname, "/cache");
+ yaz_log(YLOG_LOG, "cache_fname = %s", bfs->cache_fname);
}
else
{
}
}
else
- bfs->commit_area = NULL;
+ bfs->commit_area = 0;
return ZEBRA_OK;
}
-int bf_close (BFile bf)
+int bf_close2(BFile bf)
{
- zebra_lock_rdwr_destroy (&bf->rdwr_lock);
+ int ret = 0;
+ zebra_lock_rdwr_destroy(&bf->rdwr_lock);
if (bf->cf)
- cf_close (bf->cf);
- mf_close (bf->mf);
+ {
+ if (cf_close(bf->cf))
+ ret = -1;
+ }
+ if (bf->mf)
+ {
+ if (mf_close(bf->mf))
+ ret = -1;
+ }
xfree(bf->alloc_buf);
xfree(bf->magic);
xfree(bf);
- return 0;
+ return ret;
}
+void bf_close(BFile bf)
+{
+ if (bf_close2(bf))
+ {
+ zebra_exit("bf_close");
+ }
+}
+
+
#define HEADER_SIZE 256
BFile bf_xopen(BFiles bfs, const char *name, int block_size, int wrflag,
return bf;
}
-int bf_xclose (BFile bf, int version, const char *more_info)
+int bf_xclose(BFile bf, int version, const char *more_info)
{
if (bf->header_dirty)
{
break;
}
}
- return bf_close(bf);
+ return bf_close2(bf);
}
-BFile bf_open (BFiles bfs, const char *name, int block_size, int wflag)
+BFile bf_open(BFiles bfs, const char *name, int block_size, int wflag)
{
- BFile bf = (BFile) xmalloc(sizeof(struct BFile_struct));
+ BFile bf = (BFile) xmalloc(sizeof(*bf));
bf->alloc_buf = 0;
bf->magic = 0;
bf->block_size = block_size;
bf->header_dirty = 0;
+ bf->cf = 0;
+ bf->mf = 0;
+ zebra_lock_rdwr_init(&bf->rdwr_lock);
+
if (bfs->commit_area)
{
int first_time;
- bf->mf = mf_open (bfs->register_area, name, block_size, 0);
- bf->cf = cf_open (bf->mf, bfs->commit_area, name, block_size,
- wflag, &first_time);
+ bf->mf = mf_open(bfs->register_area, name, block_size, 0);
+ bf->cf = cf_open(bf->mf, bfs->commit_area, name, block_size,
+ wflag, &first_time);
+ if (!bf->cf)
+ {
+ yaz_log(YLOG_FATAL, "cf_open failed for %s", name);
+ bf_close(bf);
+ return 0;
+ }
if (first_time)
{
FILE *outf;
if (!outf)
{
yaz_log(YLOG_FATAL|YLOG_ERRNO, "open %s", bfs->cache_fname);
- exit(1);
+ bf_close(bf);
+ return 0;
}
fprintf(outf, "%s %d\n", name, block_size);
- fclose(outf);
+ if (fclose(outf))
+ {
+ yaz_log(YLOG_FATAL|YLOG_ERRNO, "fclose %s", bfs->cache_fname);
+ bf_close(bf);
+ return 0;
+ }
}
}
else
{
bf->mf = mf_open(bfs->register_area, name, block_size, wflag);
- bf->cf = NULL;
}
if (!bf->mf)
{
yaz_log(YLOG_FATAL, "mf_open failed for %s", name);
- xfree(bf);
+ bf_close(bf);
return 0;
}
- zebra_lock_rdwr_init(&bf->rdwr_lock);
return bf;
}
-int bf_read (BFile bf, zint no, int offset, int nbytes, void *buf)
+int bf_read(BFile bf, zint no, int offset, int nbytes, void *buf)
{
- int r;
+ int ret = bf_read2(bf, no, offset, nbytes, buf);
+
+ if (ret == -1)
+ {
+ zebra_exit("bf_read");
+ }
+ return ret;
+}
+
+int bf_read2(BFile bf, zint no, int offset, int nbytes, void *buf)
+{
+ int ret;
- zebra_lock_rdwr_rlock (&bf->rdwr_lock);
+ zebra_lock_rdwr_rlock(&bf->rdwr_lock);
if (bf->cf)
{
- if ((r = cf_read (bf->cf, no, offset, nbytes, buf)) == -1)
- r = mf_read (bf->mf, no, offset, nbytes, buf);
+ if ((ret = cf_read(bf->cf, no, offset, nbytes, buf)) == 0)
+ ret = mf_read(bf->mf, no, offset, nbytes, buf);
}
else
- r = mf_read (bf->mf, no, offset, nbytes, buf);
- zebra_lock_rdwr_runlock (&bf->rdwr_lock);
- return r;
+ ret = mf_read(bf->mf, no, offset, nbytes, buf);
+ zebra_lock_rdwr_runlock(&bf->rdwr_lock);
+ return ret;
}
-int bf_write (BFile bf, zint no, int offset, int nbytes, const void *buf)
+int bf_write(BFile bf, zint no, int offset, int nbytes, const void *buf)
+{
+ int ret = bf_write2(bf, no, offset, nbytes, buf);
+
+ if (ret == -1)
+ {
+ zebra_exit("bf_write");
+ }
+ return ret;
+}
+
+int bf_write2(BFile bf, zint no, int offset, int nbytes, const void *buf)
{
int r;
- zebra_lock_rdwr_wlock (&bf->rdwr_lock);
+ zebra_lock_rdwr_wlock(&bf->rdwr_lock);
if (bf->cf)
- r = cf_write (bf->cf, no, offset, nbytes, buf);
+ r = cf_write(bf->cf, no, offset, nbytes, buf);
else
- r = mf_write (bf->mf, no, offset, nbytes, buf);
- zebra_lock_rdwr_wunlock (&bf->rdwr_lock);
+ r = mf_write(bf->mf, no, offset, nbytes, buf);
+ zebra_lock_rdwr_wunlock(&bf->rdwr_lock);
return r;
}
-int bf_commitExists (BFiles bfs)
+int bf_commitExists(BFiles bfs)
{
FILE *inf;
- inf = open_cache (bfs, "rb");
+ inf = open_cache(bfs, "rb");
if (inf)
{
- fclose (inf);
+ fclose(inf);
return 1;
}
return 0;
}
-void bf_reset (BFiles bfs)
+void bf_reset(BFiles bfs)
{
if (!bfs)
return;
- mf_reset (bfs->commit_area);
- mf_reset (bfs->register_area);
+ mf_reset(bfs->commit_area, 1);
+ mf_reset(bfs->register_area, 1);
+ unlink_cache(bfs);
}
-void bf_commitExec (BFiles bfs)
+int bf_commitExec(BFiles bfs)
{
FILE *inf;
int block_size;
MFile mf;
CFile cf;
int first_time;
+ int r = 0;
- assert (bfs->commit_area);
- if (!(inf = open_cache (bfs, "rb")))
+ assert(bfs->commit_area);
+ if (!(inf = open_cache(bfs, "rb")))
{
- yaz_log (YLOG_LOG, "No commit file");
- return ;
+ yaz_log(YLOG_LOG, "No commit file");
+ return -1;
}
- while (fscanf (inf, "%s %d", path, &block_size) == 2)
+ while (fscanf(inf, "%s %d", path, &block_size) == 2)
{
- mf = mf_open (bfs->register_area, path, block_size, 1);
- cf = cf_open (mf, bfs->commit_area, path, block_size, 0, &first_time);
+ mf = mf_open(bfs->register_area, path, block_size, 1);
+ if (!mf)
+ {
+ r = -1;
+ break;
+ }
+ cf = cf_open(mf, bfs->commit_area, path, block_size, 0, &first_time);
+ if (!cf)
+ {
+ mf_close(mf);
+ r = -1;
+ break;
+ }
+
+ r = cf_commit(cf);
- cf_commit (cf);
+ cf_close(cf);
+ mf_close(mf);
- cf_close (cf);
- mf_close (mf);
+ if (r)
+ break;
}
- fclose (inf);
+ fclose(inf);
+ return r;
}
-void bf_commitClean (BFiles bfs, const char *spec)
+void bf_commitClean(BFiles bfs, const char *spec)
{
- FILE *inf;
- int block_size;
- char path[256];
- MFile mf;
- CFile cf;
int mustDisable = 0;
- int firstTime;
if (!bfs->commit_area)
{
- bf_cache (bfs, spec);
+ bf_cache(bfs, spec);
mustDisable = 1;
}
- if (!(inf = open_cache (bfs, "rb")))
- return ;
- while (fscanf (inf, "%s %d", path, &block_size) == 2)
- {
- mf = mf_open (bfs->register_area, path, block_size, 0);
- cf = cf_open (mf, bfs->commit_area, path, block_size, 1, &firstTime);
- cf_unlink (cf);
- cf_close (cf);
- mf_close (mf);
- }
- fclose (inf);
- unlink_cache (bfs);
+ mf_reset(bfs->commit_area, 1);
+
+ unlink_cache(bfs);
if (mustDisable)
- bf_cache (bfs, 0);
+ bf_cache(bfs, 0);
}
int bf_alloc(BFile bf, int no, zint *blocks)
memset(buf, '\0', sizeof(buf));
blocks[i] = bf->free_list;
- if (!bf_read(bf, bf->free_list, 0, sizeof(buf), buf))
+ if (bf_read(bf, bf->free_list, 0, sizeof(buf), buf) != 1)
{
yaz_log(YLOG_WARN, "Bad freelist entry " ZINT_FORMAT,
bf->free_list);
- exit(1);
+ return -1;
}
zebra_zint_decode(&cp, &bf->free_list);
}
}
return 0;
}
+
+int bfs_register_directory_stat(BFiles bfs, int no, const char **directory,
+ double *used_bytes, double *max_bytes)
+{
+ return mf_area_directory_stat(bfs->register_area, no, directory,
+ used_bytes, max_bytes);
+}
+
+
+int bfs_shadow_directory_stat(BFiles bfs, int no, const char **directory,
+ double *used_bytes, double *max_bytes)
+{
+ if (!bfs->commit_area)
+ return 0;
+ return mf_area_directory_stat(bfs->commit_area, no, directory,
+ used_bytes, max_bytes);
+}
+/*
+ * Local variables:
+ * c-basic-offset: 4
+ * indent-tabs-mode: nil
+ * End:
+ * vim: shiftwidth=4 tabstop=8 expandtab
+ */
+