Merge mdb_stata into mdb_stat

diff --git a/libraries/libmdb/mdb_stat.c b/libraries/libmdb/mdb_stat.c
index 97bd8d4..73891cf 100644 (file)
--- a/libraries/libmdb/mdb_stat.c
+++ b/libraries/libmdb/mdb_stat.c
@@ -13,49 +13,123 @@
  */
 #include <stdio.h>
 #include <stdlib.h>
-#include <time.h>
+#include <string.h>
+#include <unistd.h>
 #include "mdb.h"
 
-int main(int argc,char * argv[])
+static void prstat(MDB_stat *ms)
 {
-       int rc;
+       printf("Page size: %u\n", ms->ms_psize);
+       printf("Tree depth: %u\n", ms->ms_depth);
+       printf("Branch pages: %zu\n", ms->ms_branch_pages);
+       printf("Leaf pages: %zu\n", ms->ms_leaf_pages);
+       printf("Overflow pages: %zu\n", ms->ms_overflow_pages);
+       printf("Entries: %zu\n", ms->ms_entries);
+}
+
+static void usage(char *prog)
+{
+       fprintf(stderr, "usage: %s dbpath [-a|-s subdb]\n", prog);
+       exit(EXIT_FAILURE);
+}
+
+int main(int argc, char *argv[])
+{
+       int i, rc;
        MDB_env *env;
        MDB_txn *txn;
        MDB_dbi dbi;
        MDB_stat mst;
-       char *envname = argv[1];
+       char *prog = argv[0];
+       char *envname;
        char *subname = NULL;
+       int alldbs = 0;
+
+       if (argc < 2) {
+               usage(prog);
+       }
+
+       /* -a: print stat of main DB and all subDBs
+        * -s: print stat of only the named subDB
+        * (default) print stat of only the main DB
+        */
+       while ((i = getopt(argc, argv, "as:")) != EOF) {
+               switch(i) {
+               case 'a':
+                       alldbs++;
+                       break;
+               case 's':
+                       subname = optarg;
+                       break;
+               default:
+                       fprintf(stderr, "%s: unrecognized option -%c\n", prog, optopt);
+                       usage(prog);
+               }
+       }
+
+       if (optind != argc - 1)
+               usage(prog);
 
+       envname = argv[optind];
        rc = mdb_env_create(&env);
 
-       if (argc > 2) {
+       if (alldbs || subname) {
                mdb_env_set_maxdbs(env, 4);
-               subname = argv[2];
        }
 
-       rc = mdb_env_open(env, envname, MDB_RDONLY, 0);
+       rc = mdb_env_open(env, envname, MDB_RDONLY, 0664);
        if (rc) {
-               printf("mdb_env_open failed, error %d\n", rc);
+               printf("mdb_env_open failed, error %d %s\n", rc, mdb_strerror(rc));
                goto env_close;
        }
        rc = mdb_txn_begin(env, NULL, MDB_RDONLY, &txn);
        if (rc) {
-               printf("mdb_txn_begin failed, error %d\n", rc);
+               printf("mdb_txn_begin failed, error %d %s\n", rc, mdb_strerror(rc));
                goto env_close;
        }
        rc = mdb_open(txn, subname, 0, &dbi);
        if (rc) {
-               printf("mdb_open failed, error %d\n", rc);
+               printf("mdb_open failed, error %d %s\n", rc, mdb_strerror(rc));
                goto txn_abort;
        }
    
        rc = mdb_stat(txn, dbi, &mst);
-       printf("Page size: %u\n", mst.ms_psize);
-       printf("Tree depth: %u\n", mst.ms_depth);
-       printf("Branch pages: %zu\n", mst.ms_branch_pages);
-       printf("Leaf pages: %zu\n", mst.ms_leaf_pages);
-       printf("Overflow pages: %zu\n", mst.ms_overflow_pages);
-       printf("Entries: %zu\n", mst.ms_entries);
+       if (rc) {
+               printf("mdb_stat failed, error %d %s\n", rc, mdb_strerror(rc));
+               goto txn_abort;
+       }
+       prstat(&mst);
+
+       if (alldbs) {
+               MDB_cursor *cursor;
+               MDB_val key;
+
+               rc = mdb_cursor_open(txn, dbi, &cursor);
+               if (rc) {
+                       printf("mdb_cursor_open failed, error %d %s\n", rc, mdb_strerror(rc));
+                       goto txn_abort;
+               }
+               while ((rc = mdb_cursor_get(cursor, &key, NULL, MDB_NEXT)) == 0) {
+                       char *str = malloc(key.mv_size+1);
+                       MDB_dbi db2;
+                       memcpy(str, key.mv_data, key.mv_size);
+                       str[key.mv_size] = '\0';
+                       rc = mdb_open(txn, str, 0, &db2);
+                       if (rc == MDB_SUCCESS)
+                               printf("\n%s\n", str);
+                       free(str);
+                       if (rc) continue;
+                       rc = mdb_stat(txn, db2, &mst);
+                       if (rc) {
+                               printf("mdb_stat failed, error %d %s\n", rc, mdb_strerror(rc));
+                               goto txn_abort;
+                       }
+                       prstat(&mst);
+                       mdb_close(env, db2);
+               }
+               mdb_cursor_close(cursor);
+       }
+
        mdb_close(env, dbi);
 txn_abort:
        mdb_txn_abort(txn);

diff --git a/libraries/libmdb/mdb_stata.c b/libraries/libmdb/mdb_stata.c
deleted file mode 100644 (file)
index 7cfebb4..0000000
--- a/libraries/libmdb/mdb_stata.c
+++ /dev/null
@@ -1,84 +0,0 @@
-/* mdb_stat.c - memory-mapped database status tool */
-/*
- * Copyright 2011 Howard Chu, Symas Corp.
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted only as authorized by the OpenLDAP
- * Public License.
- *
- * A copy of this license is available in the file LICENSE in the
- * top-level directory of the distribution or, alternatively, at
- * <http://www.OpenLDAP.org/license.html>.
- */
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include "mdb.h"
-
-int main(int argc,char * argv[])
-{
-       int rc;
-       MDB_env *env;
-       MDB_txn *txn;
-       MDB_dbi dbi;
-       MDB_stat mst;
-       MDB_cursor *cursor;
-       MDB_val key;
-       char *envname = argv[1];
-
-       rc = mdb_env_create(&env);
-
-       mdb_env_set_maxdbs(env, 4);
-
-       rc = mdb_env_open(env, envname, MDB_RDONLY, 0);
-       if (rc) {
-               printf("mdb_env_open failed, error %d\n", rc);
-               goto env_close;
-       }
-       rc = mdb_txn_begin(env, NULL, MDB_RDONLY, &txn);
-       if (rc) {
-               printf("mdb_txn_begin failed, error %d\n", rc);
-               goto env_close;
-       }
-       rc = mdb_open(txn, NULL, 0, &dbi);
-       if (rc) {
-               printf("mdb_open failed, error %d\n", rc);
-               goto txn_abort;
-       }
-   
-       rc = mdb_stat(txn, dbi, &mst);
-       printf("Page size: %u\n", mst.ms_psize);
-       printf("Tree depth: %u\n", mst.ms_depth);
-       printf("Branch pages: %zu\n", mst.ms_branch_pages);
-       printf("Leaf pages: %zu\n", mst.ms_leaf_pages);
-       printf("Overflow pages: %zu\n", mst.ms_overflow_pages);
-       printf("Entries: %zu\n", mst.ms_entries);
-
-       rc = mdb_cursor_open(txn, dbi, &cursor);
-       while ((rc = mdb_cursor_get(cursor, &key, NULL, MDB_NEXT)) == 0) {
-               char *str = malloc(key.mv_size+1);
-               MDB_dbi db2;
-               memcpy(str, key.mv_data, key.mv_size);
-               str[key.mv_size] = '\0';
-               printf("\n%s\n", str);
-               rc = mdb_open(txn, str, 0, &db2);
-               if (rc) break;
-               free(str);
-               rc = mdb_stat(txn, db2, &mst);
-               printf("Tree depth: %u\n", mst.ms_depth);
-               printf("Branch pages: %zu\n", mst.ms_branch_pages);
-               printf("Leaf pages: %zu\n", mst.ms_leaf_pages);
-               printf("Overflow pages: %zu\n", mst.ms_overflow_pages);
-               printf("Entries: %zu\n", mst.ms_entries);
-               mdb_close(env, db2);
-       }
-       mdb_cursor_close(cursor);
-       mdb_close(env, dbi);
-txn_abort:
-       mdb_txn_abort(txn);
-env_close:
-       mdb_env_close(env);
-
-       return rc ? EXIT_FAILURE : EXIT_SUCCESS;
-}