lmdb/0001-Add-extra-tools-and-CFEngine-modifications-for-LMDB.patch

355 lines
8.2 KiB
Diff

From 1701892d066c626bbd82c9340d6b442f733c33a1 Mon Sep 17 00:00:00 2001
From: Kristian Amlie <kristian.amlie@cfengine.com>
Date: Mon, 28 Jul 2014 11:28:38 +0200
Subject: [PATCH 1/4] Add extra tools and CFEngine modifications for LMDB.
---
libraries/liblmdb/lmdump.c | 128 +++++++++++++++++++++++++++
libraries/liblmdb/lmmgr.c | 201 +++++++++++++++++++++++++++++++++++++++++++
libraries/liblmdb/mdb_copy.c | 2 +-
3 files changed, 330 insertions(+), 1 deletion(-)
create mode 100644 libraries/liblmdb/lmdump.c
create mode 100644 libraries/liblmdb/lmmgr.c
diff --git a/libraries/liblmdb/lmdump.c b/libraries/liblmdb/lmdump.c
new file mode 100644
index 000000000000..0e3ee4bb61c2
--- /dev/null
+++ b/libraries/liblmdb/lmdump.c
@@ -0,0 +1,128 @@
+/* lmdump.c - Lmdb database dumper
+ Has three modes :
+ -a : print keys in ascii form
+ -A : print keys and values in ascii form
+ -x : print keys and values in hexadecimal form
+ -d : print only the size of keys and values
+ */
+
+#define _XOPEN_SOURCE 500 /* srandom(), random() */
+#include <stdio.h>
+#include <stdlib.h>
+#include <time.h>
+#include <string.h>
+#include "lmdb.h"
+
+static void print_hex(char *s, int len)
+{
+ int i = 0;
+ for (i=0; i<len; i++)
+ {
+ printf("%02x", s[i]);
+ }
+}
+
+static void print_usage(void)
+{
+ printf("Lmdb database dumper\n");
+ printf("Usage: dump -d|-x|-a|-A filename\n\n");
+ printf("Has three modes :\n");
+ printf(" -a : print keys in ascii form\n");
+ printf(" -A : print keys and values in ascii form\n");
+ printf(" -x : print keys and values in hexadecimal form\n");
+ printf(" -d : print only the size of keys and values\n");
+}
+
+static int report_error(int rc)
+{
+ printf("err(%d): %s\n", rc, mdb_strerror(rc));
+ return rc;
+}
+
+int main(int argc, char * argv[])
+{
+ int rc;
+ MDB_env *env;
+ MDB_dbi dbi;
+ MDB_val key, data;
+ MDB_txn *txn;
+ MDB_cursor *cursor;
+
+ if (argc<3)
+ {
+ print_usage();
+ return 1;
+ }
+ int mode = -1;
+ if (strcmp(argv[1],"-a") == 0)
+ {
+ mode = 'a';
+ }
+ else if (strcmp(argv[1],"-A") == 0)
+ {
+ mode = 'A';
+ }
+ else if(strcmp(argv[1],"-x") == 0)
+ {
+ mode = 'x';
+ }
+ else if(strcmp(argv[1],"-d") == 0)
+ {
+ mode = 'd';
+ }
+ else
+ {
+ print_usage();
+ return 1;
+ }
+ rc = mdb_env_create(&env);
+ if(rc) return report_error(rc);
+
+ rc = mdb_env_open(env, argv[2], MDB_NOSUBDIR | MDB_NOLOCK, 0644);
+ if(rc) return report_error(rc);
+
+ rc = mdb_txn_begin(env, NULL, MDB_RDONLY, &txn);
+ if(rc) return report_error(rc);
+
+ rc = mdb_open(txn, NULL, 0, &dbi);
+ if(rc) return report_error(rc);
+
+ rc = mdb_cursor_open(txn, dbi, &cursor);
+ if(rc) return report_error(rc);
+
+ while( (rc = mdb_cursor_get(cursor, &key, &data, MDB_NEXT)) ==0)
+ {
+ if (mode == 'A')
+ {
+ printf("key: %p[%d] %.*s\n",
+ key.mv_data, (int) key.mv_size, (int) key.mv_size, (char *) key.mv_data);
+ }
+ else if (mode == 'a')
+ {
+ printf("key: %p[%d] %.*s, data: %p[%d] %.*s\n",
+ key.mv_data, (int) key.mv_size, (int) key.mv_size, (char *) key.mv_data,
+ data.mv_data, (int) data.mv_size, (int) data.mv_size, (char *) data.mv_data);
+ }
+ else if (mode == 'd')
+ {
+ printf("key: %p[%d] ,data: %p[%d]\n",
+ key.mv_data, (int) key.mv_size,
+ data.mv_data, (int) data.mv_size);
+ }
+ else if (mode == 'x')
+ {
+ printf("key: %p[%d] ", key.mv_data, (int) key.mv_size);
+ print_hex(key.mv_data, (int) key.mv_size);
+ printf(" ,data: %p[%d] ", data.mv_data, (int) data.mv_size);
+ print_hex(data.mv_data, (int) data.mv_size);
+ printf("\n");
+ }
+ }
+ mdb_cursor_close(cursor);
+ mdb_close(env, dbi);
+
+ mdb_txn_abort(txn);
+ mdb_env_close(env);
+
+ return 0;
+}
diff --git a/libraries/liblmdb/lmmgr.c b/libraries/liblmdb/lmmgr.c
new file mode 100644
index 000000000000..1137d8dc388a
--- /dev/null
+++ b/libraries/liblmdb/lmmgr.c
@@ -0,0 +1,201 @@
+/*
+lmmgr.c : Add, remove or manage the maxreaders value of a given database file
+*/
+#include <stdio.h>
+#include <string.h>
+#include "lmdb.h"
+
+static int report_error(int rc)
+{
+ printf("err(%d): %s\n", rc, mdb_strerror(rc));
+ return rc;
+}
+
+int openconn(const char *dbfile, MDB_env **env, long int maxreaders)
+{
+/*init*/
+ int rc;
+ rc = mdb_env_create(env);
+ if(rc) return report_error(rc);
+
+ rc = mdb_env_open(*env, dbfile, MDB_NOSUBDIR, 0644);
+ if(rc) return report_error(rc);
+ return 0;
+}
+
+int gettxn(MDB_env *env, MDB_txn **txn, MDB_dbi *dbi)
+{
+ int rc;
+/*setup txn*/
+ rc = mdb_txn_begin(env, NULL, 0, txn);
+ if(rc) return report_error(rc);
+ rc = mdb_open(*txn, NULL, 0, dbi);
+ if(rc) return report_error(rc);
+ return 0;
+}
+
+int committxn(MDB_txn *txn)
+{
+/*commit*/
+ int rc;
+ rc = mdb_txn_commit(txn);
+ return 0;
+}
+
+int aborttxn(MDB_txn *txn)
+{
+/*abort*/
+ int rc;
+ mdb_txn_abort(txn);
+ return 0;
+}
+
+int closeall(MDB_env *env, MDB_txn *txn, MDB_dbi dbi)
+{
+/*end*/
+ mdb_close(env, dbi);
+ mdb_env_close(env);
+ return 0;
+}
+
+int do_put(char *dbfile, char *mykey, char *myval)
+{
+ MDB_env *env;
+ openconn(dbfile, &env, 0);
+
+ MDB_txn *txn;
+ MDB_dbi dbi;
+ gettxn(env, &txn, &dbi);
+
+ int rc;
+ MDB_val key, data;
+ key.mv_data = mykey;
+ data.mv_data = myval;
+ key.mv_size = strlen(mykey) + 1;
+ data.mv_size = strlen(myval) + 1;
+
+ rc = mdb_put(txn, dbi, &key, &data, MDB_NOOVERWRITE);
+ if (rc == MDB_SUCCESS)
+ {
+ rc = mdb_txn_commit(txn);
+ }
+ else
+ {
+ mdb_txn_abort(txn);
+ }
+
+ mdb_env_close(env);
+ return rc;
+}
+
+int do_del(char *dbfile, char *mykey)
+{
+ MDB_env *env;
+ openconn(dbfile, &env, 0);
+
+ MDB_txn *txn;
+ MDB_dbi dbi;
+ gettxn(env, &txn, &dbi);
+
+ int rc;
+ MDB_val key;
+ key.mv_data = mykey;
+ key.mv_size = strlen(mykey) + 1;
+
+ rc = mdb_del(txn, dbi, &key, NULL);
+ if (rc == MDB_SUCCESS)
+ {
+ rc = mdb_txn_commit(txn);
+ }
+ else
+ {
+ mdb_txn_abort(txn);
+ }
+
+ mdb_env_close(env);
+ return rc;
+}
+
+int do_init(char *dbfile, unsigned long maxr)
+{
+ MDB_env *env;
+ openconn(dbfile, &env, maxr);
+ mdb_env_close(env);
+ return 0;
+}
+
+int do_stat(char *dbfile)
+{
+ int rc;
+ MDB_env *env;
+ openconn(dbfile, &env, 0);
+
+ MDB_stat stat;
+ MDB_envinfo info;
+
+ rc = mdb_env_stat(env, &stat);
+ rc = mdb_env_info(env, &info);
+ printf("me_maxreaders=%ld\n", info.me_maxreaders);
+ mdb_env_close(env);
+ return 0;
+}
+
+int main(int argc, char *argv[])
+{
+ if (argc < 3)
+ {
+ goto fail;
+ }
+
+ if (!strcmp(argv[1], "put"))
+ {
+ if (argc == 5)
+ {
+ int rc = do_put(argv[2], argv[3], argv[4]);
+ return rc;
+ }
+ }
+ else if (!strcmp(argv[1], "del"))
+ {
+ if (argc == 4)
+ {
+ int rc = do_del(argv[2], argv[3]);
+ return rc;
+ }
+ }
+ else if (!strcmp(argv[1], "init"))
+ {
+ if (argc == 3)
+ {
+ int rc = do_init(argv[2], 0L);
+ return rc;
+ }
+ else if (argc == 5 && !strcmp(argv[3], "-m"))
+ {
+ int rc = do_init(argv[2], atol(argv[4]));
+ do_stat(argv[2]);
+ return rc;
+ }
+ }
+ else if (!strcmp(argv[1], "maxr"))
+ {
+ if (argc == 3)
+ {
+ int rc = do_stat(argv[2]);
+ return rc;
+ }
+ }
+
+fail:
+ printf("Usage :\n");
+ printf("Add a key and value to a DB file :\n");
+ printf("\tlmmgr put dbfile key value\n");
+ printf("Remove a key from a DB file :\n");
+ printf("\tlmmgr del dbfile key\n");
+ printf("Set a new maxreaders value of a DB file :\n");
+ printf("\tlmmgr init dbfile [-m maxreaders]\n");
+ printf("Give maxreaders value of a DB file :\n");
+ printf("\tlmmgr maxr dbfile\n");
+
+ return 1;
+}