Add cdb_init_at and cdb_init_fromfd, rename cdb_reader

Signed-off-by: Laurent Bercot <ska@appnovation.com>

5 files changed, 47 insertions, 25 deletions

diff --git a/src/libstddjb/cdb_find.c b/src/libstddjb/cdb_find.c
index a9c66e3..af642cc 100644
--- a/src/libstddjb/cdb_find.c
+++ b/src/libstddjb/cdb_find.c
@@ -7,7 +7,7 @@
 #include <skalibs/cdb.h>
 #include "cdb-internal.h"
 
-int cdb_find (cdb const *c, cdb_reader *d, cdb_data *out, char const *key, uint32_t len)
+int cdb_find (cdb const *c, cdb_data *out, char const *key, uint32_t len, cdb_find_state *d)
 {
   if (!d->loop)
   {
diff --git a/src/libstddjb/cdb_init.c b/src/libstddjb/cdb_init.c
index 7a6c2f6..fad6a2c 100644
--- a/src/libstddjb/cdb_init.c
+++ b/src/libstddjb/cdb_init.c
@@ -1,31 +1,17 @@
 /* ISC license. */
 
-#include <sys/stat.h>
-#include <sys/mman.h>
-#include <stdint.h>
-#include <errno.h>
-
 #include <skalibs/djbunix.h>
 #include <skalibs/cdb.h>
 
 int cdb_init (cdb *c, char const *file)
 {
-  char *map ;
-  struct stat st ;
   int fd = openc_read(file) ;
   if (fd < 0) return 0 ;
-  if (fstat(fd, &st) < 0) goto err ;
-  if (st.st_size > UINT32_MAX) goto eoverf ;
-  map = mmap(0, st.st_size, PROT_READ, MAP_SHARED, fd, 0) ;
-  if (map == MAP_FAILED) goto err ;
-  c->map = map ;
-  c->size = st.st_size ;
+  if (!cdb_init_fromfd(c, fd))
+  {
+    fd_close(fd) ;
+    return 0 ;
+  }
   fd_close(fd) ;
   return 1 ;
-
- eoverf:
-  errno = EOVERFLOW ;
- err:
-  fd_close(fd) ;
-  return 0 ;
 }
diff --git a/src/libstddjb/cdb_init_at.c b/src/libstddjb/cdb_init_at.c
new file mode 100644
index 0000000..faea790
--- /dev/null
+++ b/src/libstddjb/cdb_init_at.c
@@ -0,0 +1,18 @@
+/* ISC license. */
+
+#include <skalibs/djbunix.h>
+#include <skalibs/cdb.h>
+#include <skalibs/unix-transactional.h>
+
+int cdb_init_at (cdb *c, int dirfd, char const *file)
+{
+  int fd = open_readat(dirfd, file) ;
+  if (fd < 0) return 0 ;
+  if (!cdb_init_fromfd(c, fd))
+  {
+    fd_close(fd) ;
+    return 0 ;
+  }
+  fd_close(fd) ;
+  return 1 ;
+}
diff --git a/src/libstddjb/cdb_init_fromfd.c b/src/libstddjb/cdb_init_fromfd.c
new file mode 100644
index 0000000..7446f4a
--- /dev/null
+++ b/src/libstddjb/cdb_init_fromfd.c
@@ -0,0 +1,23 @@
+/* ISC license. */
+
+#include <skalibs/bsdsnowflake.h>
+
+#include <sys/stat.h>
+#include <sys/mman.h>
+#include <stdint.h>
+#include <errno.h>
+
+#include <skalibs/cdb.h>
+
+int cdb_init_fromfd (cdb *c, int fd)
+{
+  char *map ;
+  struct stat st ;
+  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->map = map ;
+  c->size = st.st_size ;
+  return 1 ;
+}
diff --git a/src/libstddjb/cdb_reader_zero.c b/src/libstddjb/cdb_reader_zero.c
deleted file mode 100644
index 1dbb275..0000000
--- a/src/libstddjb/cdb_reader_zero.c
+++ /dev/null
@@ -1,5 +0,0 @@
-/* ISC license. */
-
-#include <skalibs/cdb.h>
-
-cdb_reader const cdb_reader_zero = CDB_READER_ZERO ;