Changeset 2717 for box/trunk/bin/bbackupd/BackupClientInodeToIDMap.cpp

Timestamp:

28/08/2010 10:01:16 (21 months ago)

Author:

chris

Message:

Remove QDBM, and switch BackupClientInodeToIDMap to use the bundled TDB
instead.

File:

• box/trunk/bin/bbackupd/BackupClientInodeToIDMap.cpp (modified) (13 diffs)

box/trunk/bin/bbackupd/BackupClientInodeToIDMap.cpp

@@ -11 +11 @@
 
 #include <stdlib.h>
-#include <depot.h>
+
+#define _PUBLIC_
+#include "tdb.h"
 
 #define BACKIPCLIENTINODETOIDMAP_IMPLEMENTATION
@@ -27 +29 @@
 } IDBRecord;
 
-#define BOX_DBM_MESSAGE(stuff) stuff << " (qdbm): " << dperrmsg(dpecode)
+#define BOX_DBM_MESSAGE(stuff) stuff << " (tdb): " << tdb_error(mpContext)
 
 #define BOX_LOG_DBM_ERROR(stuff) \
@@ -37 +39 @@
                 BOX_DBM_MESSAGE(message << ": " << filename));
 
-#define ASSERT_DBM_OK(operation, message, filename, exception, subtype) \
-        if(!(operation)) \
+#define ASSERT_DBM(success, message, exception, subtype) \
+        if(!(success)) \
         { \
-                THROW_DBM_ERROR(message, filename, exception, subtype); \
+                THROW_DBM_ERROR(message, mFilename, exception, subtype); \
         }
 
 #define ASSERT_DBM_OPEN() \
-        if(mpDepot == 0) \
+        if(mpContext == 0) \
         { \
                 THROW_EXCEPTION_MESSAGE(BackupStoreException, InodeMapNotOpen, \
@@ -51 +53 @@
 
 #define ASSERT_DBM_CLOSED() \
-        if(mpDepot != 0) \
+        if(mpContext != 0) \
         { \
                 THROW_EXCEPTION_MESSAGE(CommonException, Internal, \
@@ -68 +70 @@
         : mReadOnly(true),
           mEmpty(false),
-          mpDepot(0)
+          mpContext(NULL)
 {
 }
@@ -82 +84 @@
 BackupClientInodeToIDMap::~BackupClientInodeToIDMap()
 {
-        if(mpDepot != 0)
+        if(mpContext != NULL)
         {
                 Close();
@@ -109 +111 @@
         
         // Open the database file
-        int mode = ReadOnly ? DP_OREADER : DP_OWRITER;
+        int mode = ReadOnly ? O_RDONLY : O_RDWR;
         if(CreateNew)
         {
-                mode |= DP_OCREAT;
-        }
-        
-        mpDepot = dpopen(Filename, mode, 0);
-        
-        ASSERT_DBM_OK(mpDepot, "Failed to open inode database", mFilename,
+                mode |= O_CREAT;
+        }
+        
+        mpContext = tdb_open(Filename, 0, 0, mode, 0700);
+        
+        ASSERT_DBM(mpContext != NULL, "Failed to open inode database",
                 BackupStoreException, BerkelyDBFailure);
         
@@ -138 +140 @@
 {
         ASSERT_DBM_CLOSED();
-        ASSERT(mpDepot == 0);
+        ASSERT(mpContext == NULL);
         mEmpty = true;
         mReadOnly = true;
@@ -154 +156 @@
 {
         ASSERT_DBM_OPEN();
-        ASSERT_DBM_OK(dpclose(mpDepot), "Failed to close inode database",
-                mFilename, BackupStoreException, BerkelyDBFailure);
-        mpDepot = 0;
-}
+        ASSERT_DBM(tdb_close(mpContext) == 0, "Failed to close inode database",
+                BackupStoreException, BerkelyDBFailure);
+        mpContext = NULL;
+}
+
+static TDB_DATA GetDatum(void* dptr, size_t dsize)
+{
+        TDB_DATA datum;
+        datum.dptr = (unsigned char *)dptr;
+        datum.dsize = dsize;
+        return datum;
+}
+
+#define GET_STRUCT_DATUM(structure) \
+        GetDatum(&structure, sizeof(structure))
 
 // --------------------------------------------------------------------------
@@ -177 +190 @@
         }
 
-        if(mpDepot == 0)
+        if(mpContext == 0)
         {
                 THROW_EXCEPTION(BackupStoreException, InodeMapNotOpen);
@@ -189 +202 @@
         rec.mInDirectory = InDirectory;
 
-        ASSERT_DBM_OK(dpput(mpDepot, (const char *)&InodeRef, sizeof(InodeRef),
-                (const char *)&rec, sizeof(rec), DP_DOVER),
-                "Failed to add record to inode database", mFilename,
+        ASSERT_DBM(tdb_store(mpContext, GET_STRUCT_DATUM(InodeRef),
+                GET_STRUCT_DATUM(rec), 0) == 0,
+                "Failed to add record to inode database",
                 BackupStoreException, BerkelyDBFailure);
 }
@@ -215 +228 @@
         }
 
-        if(mpDepot == 0)
+        if(mpContext == 0)
         {
                 THROW_EXCEPTION(BackupStoreException, InodeMapNotOpen);
@@ -222 +235 @@
         ASSERT_DBM_OPEN();
 
-        IDBRecord rec;
-        
-        if(dpgetwb(mpDepot, (const char *)&InodeRef, sizeof(InodeRef),
-                0, sizeof(IDBRecord), (char *)&rec) == -1)
+        TDB_DATA datum = tdb_fetch(mpContext, GET_STRUCT_DATUM(InodeRef));
+        if(datum.dptr == NULL)
         {
                 // key not in file
                 return false;
         }
+
+        IDBRecord rec;
+        if(datum.dsize != sizeof(rec))
+        {
+                THROW_EXCEPTION_MESSAGE(CommonException, Internal,
+                        "Failed to get inode database entry: "
+                        "record has wrong size: expected " <<
+                        sizeof(rec) << " but was " << datum.dsize <<
+                        " in " << mFilename);
+        }
                 
+        rec = *(IDBRecord *)datum.dptr;
+        free(datum.dptr);
+        
         // Return data
         rObjectIDOut = rec.mObjectID;