]> git.sur5r.net Git - openldap/blobdiff - libraries/libmdb/mdb.c
Parameter cleanup
[openldap] / libraries / libmdb / mdb.c
index f3756490345ef339839434e46c23d33e98c688dc..3e2d70e36c214ad414548b8f1782762a1e2e71f6 100644 (file)
@@ -254,10 +254,10 @@ typedef struct MDB_node {
        unsigned int    mn_flags:4;
        unsigned int    mn_ksize:12;                    /* key size */
 #define F_BIGDATA       0x01                   /* data put on overflow page */
+#define F_SUBDATA       0x02                   /* data is a sub-database */
        char            mn_data[1];
 } MDB_node;
 
-
 typedef struct MDB_dbx {
        MDB_val         md_name;
        MDB_cmp_func    *md_cmp;                /* user compare function */
@@ -354,6 +354,8 @@ static int  mdb_add_node(MDB_txn *txn, MDB_dbi dbi, MDB_page *mp,
 static void mdb_del_node(MDB_page *mp, indx_t indx);
 static int mdb_del0(MDB_txn *txn, MDB_dbi dbi, unsigned int ki,
     MDB_pageparent *mpp, MDB_node *leaf);
+static int mdb_put0(MDB_txn *txn, MDB_dbi dbi,
+    MDB_val *key, MDB_val *data, unsigned int flags);
 static int  mdb_read_data(MDB_txn *txn, MDB_node *leaf, MDB_val *data);
 
 static int              mdb_rebalance(MDB_txn *txn, MDB_dbi dbi, MDB_pageparent *mp);
@@ -452,8 +454,11 @@ mdb_alloc_page(MDB_txn *txn, MDB_page *parent, unsigned int parent_idx, int num)
 {
        MDB_dpage *dp;
        pgno_t pgno = P_INVALID;
-       ULONG oldest = txn->mt_txnid - 2;
+       ULONG oldest;
+
+       if (txn->mt_txnid > 2) {
 
+       oldest = txn->mt_txnid - 2;
        if (!txn->mt_env->me_pghead && txn->mt_dbs[FREE_DBI].md_root != P_INVALID) {
                /* See if there's anything in the free DB */
                MDB_pageparent mpp;
@@ -534,6 +539,7 @@ mdb_alloc_page(MDB_txn *txn, MDB_page *parent, unsigned int parent_idx, int num)
                        }
                }
        }
+       }
 
        if ((dp = malloc(txn->mt_env->me_psize * num + sizeof(MDB_dhead))) == NULL)
                return NULL;
@@ -770,7 +776,7 @@ mdb_txn_commit(MDB_txn *txn)
                key.mv_data = (char *)&mop->mo_txnid;
                data.mv_size = MDB_IDL_SIZEOF(mop->mo_pages);
                data.mv_data = mop->mo_pages;
-               mdb_put(txn, FREE_DBI, &key, &data, 0);
+               mdb_put0(txn, FREE_DBI, &key, &data, 0);
                free(env->me_pghead);
                env->me_pghead = NULL;
        }
@@ -802,7 +808,7 @@ mdb_txn_commit(MDB_txn *txn)
                key.mv_data = (char *)&txn->mt_txnid;
                data.mv_size = MDB_IDL_SIZEOF(txn->mt_free_pgs);
                data.mv_data = txn->mt_free_pgs;
-               mdb_put(txn, FREE_DBI, &key, &data, 0);
+               mdb_put0(txn, FREE_DBI, &key, &data, 0);
        }
 
        /* Update DB root pointers. Their pages have already been
@@ -815,7 +821,7 @@ mdb_txn_commit(MDB_txn *txn)
                for (i = 2; i < txn->mt_numdbs; i++) {
                        if (txn->mt_dbxs[i].md_dirty) {
                                data.mv_data = &txn->mt_dbs[i];
-                               mdb_put(txn, i, &txn->mt_dbxs[i].md_name, &data, 0);
+                               mdb_put0(txn, i, &txn->mt_dbxs[i].md_name, &data, 0);
                        }
                }
        }
@@ -1101,8 +1107,6 @@ mdbenv_set_mapsize(MDB_env *env, size_t size)
 int
 mdbenv_set_maxdbs(MDB_env *env, int dbs)
 {
-       if (env->me_map)
-               return EINVAL;
        env->me_maxdbs = dbs;
        return MDB_SUCCESS;
 }
@@ -1297,6 +1301,8 @@ fail:
 
 }
 
+#define LOCKNAME       "/lock.mdb"
+#define DATANAME       "/data.mdb"
 int
 mdbenv_open(MDB_env *env, const char *path, unsigned int flags, mode_t mode)
 {
@@ -1304,12 +1310,12 @@ mdbenv_open(MDB_env *env, const char *path, unsigned int flags, mode_t mode)
        char *lpath, *dpath;
 
        len = strlen(path);
-       lpath = malloc(len + sizeof("/lock.mdb") + len + sizeof("/data.db"));
+       lpath = malloc(len + sizeof(LOCKNAME) + len + sizeof(DATANAME));
        if (!lpath)
                return ENOMEM;
-       dpath = lpath + len + sizeof("/lock.mdb");
-       sprintf(lpath, "%s/lock.mdb", path);
-       sprintf(dpath, "%s/data.mdb", path);
+       dpath = lpath + len + sizeof(LOCKNAME);
+       sprintf(lpath, "%s" LOCKNAME, path);
+       sprintf(dpath, "%s" DATANAME, path);
 
        rc = mdbenv_setup_locks(env, lpath, mode, &excl);
        if (rc)
@@ -1349,6 +1355,8 @@ mdbenv_close(MDB_env *env)
        if (env == NULL)
                return;
 
+       free(env->me_dbs[1]);
+       free(env->me_dbs[0]);
        free(env->me_dbxs);
        free(env->me_path);
 
@@ -1598,7 +1606,7 @@ mdb_search_page(MDB_txn *txn, MDB_dbi dbi, MDB_val *key,
                /* For sub-databases, update main root first */
                if (dbi > MAIN_DBI && !txn->mt_dbxs[dbi].md_dirty) {
                        MDB_pageparent mp2;
-                       rc = mdb_search_page(txn, 0, &txn->mt_dbxs[dbi].md_name,
+                       rc = mdb_search_page(txn, MAIN_DBI, &txn->mt_dbxs[dbi].md_name,
                                NULL, 1, &mp2);
                        if (rc)
                                return rc;
@@ -1653,6 +1661,9 @@ mdb_get(MDB_txn *txn, MDB_dbi dbi,
        assert(data);
        DPRINTF("===> get key [%.*s]", (int)key->mv_size, (char *)key->mv_data);
 
+       if (txn == NULL || dbi >= txn->mt_numdbs)
+               return EINVAL;
+
        if (key->mv_size == 0 || key->mv_size > MAXKEYSIZE) {
                return EINVAL;
        }
@@ -1668,7 +1679,7 @@ mdb_get(MDB_txn *txn, MDB_dbi dbi,
 
                        mdb_xcursor_init0(txn, dbi, &mx);
                        mdb_xcursor_init1(txn, dbi, &mx, NODEDATA(leaf));
-                       rc = mdb_search_page(&mx.mx_txn, mx.mx_txn.mt_numdbs-1, NULL, NULL, 0, &mpp);
+                       rc = mdb_search_page(&mx.mx_txn, mx.mx_cursor.mc_dbi, NULL, NULL, 0, &mpp);
                        if (rc != MDB_SUCCESS)
                                return rc;
                        leaf = NODEPTR(mpp.mp_page, 0);
@@ -2302,7 +2313,7 @@ mdb_xcursor_fini(MDB_txn *txn, MDB_dbi dbi, MDB_xcursor *mx)
        txn->mt_dbxs[1].md_dirty = mx->mx_dbxs[1].md_dirty;
        if (dbi > 1) {
                txn->mt_dbs[dbi] = mx->mx_dbs[2];
-               txn->mt_dbxs[2].md_dirty = mx->mx_dbxs[2].md_dirty;
+               txn->mt_dbxs[dbi].md_dirty = mx->mx_dbxs[2].md_dirty;
        }
 }
 
@@ -2685,9 +2696,6 @@ mdb_del(MDB_txn *txn, MDB_dbi dbi,
                return MDB_NOTFOUND;
        }
 
-       if (data && (rc = mdb_read_data(txn, leaf, data)) != MDB_SUCCESS)
-               return rc;
-
        if (F_ISSET(txn->mt_dbs[dbi].md_flags, MDB_DUPSORT)) {
                MDB_xcursor mx;
                MDB_pageparent mp2;
@@ -2696,13 +2704,16 @@ mdb_del(MDB_txn *txn, MDB_dbi dbi,
                mdb_xcursor_init1(txn, dbi, &mx, NODEDATA(leaf));
                if (flags == MDB_DEL_DUP) {
                        rc = mdb_del(&mx.mx_txn, mx.mx_cursor.mc_dbi, data, NULL, 0);
+                       mdb_xcursor_fini(txn, dbi, &mx);
                        if (rc != MDB_SUCCESS)
                                return rc;
-                       mdb_xcursor_fini(txn, dbi, &mx);
                        /* If sub-DB still has entries, we're done */
-                       if (mx.mx_txn.mt_dbs[mx.mx_cursor.mc_dbi].md_root != P_INVALID)
+                       if (mx.mx_txn.mt_dbs[mx.mx_cursor.mc_dbi].md_root != P_INVALID) {
+                               memcpy(NODEDATA(leaf), &mx.mx_txn.mt_dbs[mx.mx_cursor.mc_dbi],
+                                       sizeof(MDB_db));
                                return rc;
-                       /* otherwise fall thru and delete the sub-db */
+                       }
+                       /* otherwise fall thru and delete the sub-DB */
                } else {
                        /* add all the child DB's pages to the free list */
                        rc = mdb_search_page(&mx.mx_txn, mx.mx_cursor.mc_dbi,
@@ -2739,6 +2750,9 @@ mdb_del(MDB_txn *txn, MDB_dbi dbi,
                }
        }
 
+       if (data && (rc = mdb_read_data(txn, leaf, data)) != MDB_SUCCESS)
+               return rc;
+
        return mdb_del0(txn, dbi, ki, &mpp, leaf);
 }
 
@@ -2899,8 +2913,8 @@ mdb_split(MDB_txn *txn, MDB_dbi dbi, MDB_page **mpp, unsigned int *newindxp,
        return rc;
 }
 
-int
-mdb_put(MDB_txn *txn, MDB_dbi dbi,
+static int
+mdb_put0(MDB_txn *txn, MDB_dbi dbi,
     MDB_val *key, MDB_val *data, unsigned int flags)
 {
        int              rc = MDB_SUCCESS, exact;
@@ -2910,20 +2924,6 @@ mdb_put(MDB_txn *txn, MDB_dbi dbi,
        MDB_val xdata, *rdata;
        MDB_db dummy;
 
-       assert(key != NULL);
-       assert(data != NULL);
-
-       if (txn == NULL)
-               return EINVAL;
-
-       if (F_ISSET(txn->mt_flags, MDB_TXN_RDONLY)) {
-               return EINVAL;
-       }
-
-       if (key->mv_size == 0 || key->mv_size > MAXKEYSIZE) {
-               return EINVAL;
-       }
-
        DPRINTF("==> put key %.*s, size %zu, data size %zu",
                (int)key->mv_size, (char *)key->mv_data, key->mv_size, data->mv_size);
 
@@ -2987,6 +2987,7 @@ mdb_put(MDB_txn *txn, MDB_dbi dbi,
 
        if (SIZELEFT(mpp.mp_page) < mdb_leaf_size(txn->mt_env, key, data)) {
                rc = mdb_split(txn, dbi, &mpp.mp_page, &ki, key, data, P_INVALID);
+               leaf = NODEPTR(mpp.mp_page, ki);
        } else {
                /* There is room already in this leaf page. */
                rc = mdb_add_node(txn, dbi, mpp.mp_page, ki, key, data, 0, 0);
@@ -2996,6 +2997,11 @@ mdb_put(MDB_txn *txn, MDB_dbi dbi,
                txn->mt_flags |= MDB_TXN_ERROR;
        else {
                txn->mt_dbs[dbi].md_entries++;
+
+               /* Remember if we just added a subdatabase */
+               if (flags & F_SUBDATA)
+                       leaf->mn_flags |= F_SUBDATA;
+
                /* Now store the actual data in the child DB. Note that we're
                 * storing the user data in the keys field, so there are strict
                 * size limits on dupdata. The actual data fields of the child
@@ -3012,8 +3018,10 @@ put_sub:
                        xdata.mv_data = "";
                        if (flags == MDB_NODUPDATA)
                                flags = MDB_NOOVERWRITE;
-                       rc = mdb_put(&mx.mx_txn, mx.mx_txn.mt_numdbs-1, data, &xdata, flags);
+                       rc = mdb_put0(&mx.mx_txn, mx.mx_cursor.mc_dbi, data, &xdata, flags);
                        mdb_xcursor_fini(txn, dbi, &mx);
+                       memcpy(NODEDATA(leaf), &mx.mx_txn.mt_dbs[mx.mx_cursor.mc_dbi],
+                               sizeof(MDB_db));
                }
        }
 
@@ -3021,6 +3029,30 @@ done:
        return rc;
 }
 
+int
+mdb_put(MDB_txn *txn, MDB_dbi dbi,
+    MDB_val *key, MDB_val *data, unsigned int flags)
+{
+       assert(key != NULL);
+       assert(data != NULL);
+
+       if (txn == NULL || dbi >= txn->mt_numdbs)
+               return EINVAL;
+
+       if (F_ISSET(txn->mt_flags, MDB_TXN_RDONLY)) {
+               return EINVAL;
+       }
+
+       if (key->mv_size == 0 || key->mv_size > MAXKEYSIZE) {
+               return EINVAL;
+       }
+
+       if ((flags & (MDB_NOOVERWRITE|MDB_NODUPDATA)) != flags)
+               return EINVAL;
+
+       return mdb_put0(txn, dbi, key, data, flags);
+}
+
 int
 mdbenv_get_flags(MDB_env *env, unsigned int *arg)
 {
@@ -3061,7 +3093,7 @@ int mdb_open(MDB_txn *txn, const char *name, unsigned int flags, MDB_dbi *dbi)
 {
        MDB_val key, data;
        MDB_dbi i;
-       int rc;
+       int rc, dirty = 0;
        size_t len;
 
        /* main DB? */
@@ -3096,7 +3128,8 @@ int mdb_open(MDB_txn *txn, const char *name, unsigned int flags, MDB_dbi *dbi)
                memset(&dummy, 0, sizeof(dummy));
                dummy.md_root = P_INVALID;
                dummy.md_flags = flags & 0xffff;
-               rc = mdb_put(txn, 0, &key, &data, 0);
+               rc = mdb_put0(txn, MAIN_DBI, &key, &data, F_SUBDATA);
+               dirty = 1;
        }
 
        /* OK, got info, add to table */
@@ -3107,7 +3140,7 @@ int mdb_open(MDB_txn *txn, const char *name, unsigned int flags, MDB_dbi *dbi)
                txn->mt_dbxs[txn->mt_numdbs].md_dcmp = NULL;
                txn->mt_dbxs[txn->mt_numdbs].md_rel = NULL;
                txn->mt_dbxs[txn->mt_numdbs].md_parent = MAIN_DBI;
-               txn->mt_dbxs[txn->mt_numdbs].md_dirty = 0;
+               txn->mt_dbxs[txn->mt_numdbs].md_dirty = dirty;
                memcpy(&txn->mt_dbs[txn->mt_numdbs], data.mv_data, sizeof(MDB_db));
                *dbi = txn->mt_numdbs;
                txn->mt_numdbs++;