From e0bc74ff013a3628b17911be624d17067259c07b Mon Sep 17 00:00:00 2001 From: Laurent Bercot Date: Tue, 29 Jun 2021 12:23:00 +0000 Subject: cdb always mmaps Signed-off-by: Laurent Bercot --- src/include/skalibs/cdb.h | 9 ++++----- src/libstddjb/cdb_init.c | 23 +++++++++++++++++++++++ src/libstddjb/cdb_init_map.c | 26 -------------------------- src/libstddjb/cdb_read.c | 14 ++------------ 4 files changed, 29 insertions(+), 43 deletions(-) create mode 100644 src/libstddjb/cdb_init.c delete mode 100644 src/libstddjb/cdb_init_map.c (limited to 'src') diff --git a/src/include/skalibs/cdb.h b/src/include/skalibs/cdb.h index 52e72ee..20d3903 100644 --- a/src/include/skalibs/cdb.h +++ b/src/include/skalibs/cdb.h @@ -13,8 +13,7 @@ extern uint32_t cdb_hash (char const *, unsigned int) gccattr_pure ; typedef struct cdb cdb_t, *cdb_t_ref ; struct cdb { - char *map ; /* 0 if no map */ - int fd ; /* -1 if uninitted, negative if mapped, nonnegative if nomapped */ + char *map ; uint32_t size ; /* initialized if map is nonzero */ uint32_t loop ; /* number of hash slots searched under this key */ uint32_t khash ; /* initialized if loop is nonzero */ @@ -25,13 +24,13 @@ struct cdb uint32_t dlen ; /* initialized if cdb_findnext() returns 1 */ } ; -#define CDB_ZERO { .map = 0, .fd = -1, .size = 0, .loop = 0, .khash = 0, .kpos = 0, .hpos = 0, .hslots = 0, .dpos = 0, .dlen = 0 } +#define CDB_ZERO { .map = 0, .size = 0, .loop = 0, .khash = 0, .kpos = 0, .hpos = 0, .hslots = 0, .dpos = 0, .dlen = 0 } extern struct cdb const cdb_zero ; extern void cdb_free (struct cdb *) ; -#define cdb_init(c, fd) (cdb_init_map(c, (fd), 1) ? 0 : -1) -extern int cdb_init_map (struct cdb *, int fd, int) ; +#define cdb_init_map(c, fd, domap) (!cdb_init(c, fd)) +extern int cdb_init (struct cdb *, int fd) ; extern int cdb_mapfile (struct cdb *, char const *) ; extern int cdb_read (struct cdb *, char *, unsigned int, uint32_t) ; #define cdb_findstart(c) ((c)->loop = 0) diff --git a/src/libstddjb/cdb_init.c b/src/libstddjb/cdb_init.c new file mode 100644 index 0000000..2a1ffb3 --- /dev/null +++ b/src/libstddjb/cdb_init.c @@ -0,0 +1,23 @@ +/* ISC license. */ + +#include + +#include +#include +#include +#include + +#include + +int cdb_init (struct cdb *c, int fd) +{ + struct stat st ; + char *map ; + if (fstat(fd, &st) < 0) return -1 ; + if (st.st_size > UINT32_MAX) return (errno = EOVERFLOW, -1) ; + map = mmap(0, st.st_size, PROT_READ, MAP_SHARED, fd, -1) ; + if (map == MAP_FAILED) return -1 ; + c->map = map ; + c->size = st.st_size ; + return 0 ; +} diff --git a/src/libstddjb/cdb_init_map.c b/src/libstddjb/cdb_init_map.c deleted file mode 100644 index 55fb02a..0000000 --- a/src/libstddjb/cdb_init_map.c +++ /dev/null @@ -1,26 +0,0 @@ -/* ISC license. */ - -#include -#include -#include -#include -#include -#include - -int cdb_init_map (struct cdb *c, int fd, int domap) -{ - if (domap) - { - struct stat st ; - char *map ; - if (fstat(fd, &st) < 0) return 0 ; - if (st.st_size > UINT32_MAX) return (errno = EOVERFLOW, 0) ; - map = mmap(0, st.st_size, PROT_READ, MAP_SHARED, fd, 0) ; - if (map == MAP_FAILED) return 0 ; - c->fd = -fd-2 ; - c->map = map ; - c->size = st.st_size ; - } - else c->fd = fd ; - return 1 ; -} diff --git a/src/libstddjb/cdb_read.c b/src/libstddjb/cdb_read.c index 1d2f491..1fb15da 100644 --- a/src/libstddjb/cdb_read.c +++ b/src/libstddjb/cdb_read.c @@ -1,24 +1,14 @@ /* ISC license. */ -#include #include #include -#include #include #include int cdb_read (struct cdb *c, char *buf, unsigned int len, uint32_t pos) { - if (c->map) - { - if ((pos > c->size) || (c->size - pos < len)) return (errno = EPROTO, -1) ; - memcpy(buf, c->map + pos, len) ; - } - else - { - if (lseek(c->fd, pos, SEEK_SET) < 0) return -1 ; - if (allread(c->fd, buf, len) < len) return -1 ; - } + if ((pos > c->size) || (c->size - pos < len)) return (errno = EPROTO, -1) ; + memcpy(buf, c->map + pos, len) ; return 0 ; } -- cgit v1.2.3