diff --git a/libraries/libmdb/mdb_stat.c b/libraries/libmdb/mdb_stat.c index 97bd8d4..73891cf 100644 --- a/libraries/libmdb/mdb_stat.c +++ b/libraries/libmdb/mdb_stat.c @@ -13,49 +13,123 @@ */ #include #include -#include +#include +#include #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 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 - * . - */ -#include -#include -#include -#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; -}