implement support for LMDB in ext/dba

don't abort txn if cursor is active

fix typos

1 files changed, 353 insertions, 0 deletions

diff --git a/ext/dba/dba_lmdb.c b/ext/dba/dba_lmdb.c
new file mode 100644
index 0000000000..91f42e1a76
--- /dev/null
+++ b/ext/dba/dba_lmdb.c
@@ -0,0 +1,353 @@
+/*
+  +----------------------------------------------------------------------+
+  | PHP Version 7                                                        |
+  +----------------------------------------------------------------------+
+  | Copyright (c) 2017 The PHP Group                                     |
+  +----------------------------------------------------------------------+
+  | This source file is subject to version 3.01 of the PHP license,      |
+  | that is bundled with this package in the file LICENSE, and is        |
+  | available through the world-wide-web at the following url:           |
+  | http://www.php.net/license/3_01.txt                                  |
+  | If you did not receive a copy of the PHP license and are unable to   |
+  | obtain it through the world-wide-web, please send a note to          |
+  | license@php.net so we can mail you a copy immediately.               |
+  +----------------------------------------------------------------------+
+  | Author: Anatol Belski <ab@php.net>                                   |
+  +----------------------------------------------------------------------+
+*/
+
+/* $Id$ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include "php.h"
+
+#if DBA_LMDB
+#include "php_lmdb.h"
+
+#ifdef LMDB_INCLUDE_FILE
+#include LMDB_INCLUDE_FILE
+#endif
+
+struct php_lmdb_info {
+	MDB_env *env;
+	MDB_txn *txn;
+	MDB_dbi dbi;
+	MDB_cursor *cur;
+};
+
+#define LMDB_IT(it) (((struct php_lmdb_info *)info->dbf)->it)
+
+DBA_OPEN_FUNC(lmdb)
+{
+	MDB_env *env;
+	MDB_txn *txn;
+	int rc, mode = 0644, flags = MDB_NOSUBDIR;
+
+	if(info->argc > 0) {
+		convert_to_long_ex(&info->argv[0]);
+		mode = Z_LVAL(info->argv[0]);
+
+		/* TODO implement handling of the additional flags. */
+	}
+
+	rc = mdb_env_create(&env);
+	if (rc) {
+		*error = mdb_strerror(rc);
+		return FAILURE;
+	}
+
+	rc = mdb_env_open(env, info->path, flags, mode);
+	if (rc) {
+		*error = mdb_strerror(rc);
+		return FAILURE;
+	}
+
+	rc = mdb_txn_begin(env, NULL, 0, &txn);
+	if (rc) {
+		mdb_env_close(env);
+		*error = mdb_strerror(rc);
+		return FAILURE;
+	}
+
+	info->dbf = pemalloc(sizeof(struct php_lmdb_info), info->flags & DBA_PERSISTENT);
+	if (!info->dbf) {
+		*error = "Failed to allocate php_lmdb_info.";
+		return FAILURE;
+	}
+	memset(info->dbf, 0, sizeof(struct php_lmdb_info));
+
+	rc = mdb_dbi_open(txn, NULL, 0, &LMDB_IT(dbi));
+	if (rc) {
+		mdb_env_close(env);
+		pefree(info->dbf, info->flags & DBA_PERSISTENT);
+		*error = mdb_strerror(rc);
+		return FAILURE;
+	}
+
+	LMDB_IT(env) = env;
+	LMDB_IT(txn) = txn;
+
+	mdb_txn_abort(LMDB_IT(txn));
+
+	return SUCCESS;
+}
+
+DBA_CLOSE_FUNC(lmdb)
+{
+	mdb_dbi_close(LMDB_IT(env), LMDB_IT(dbi));
+	mdb_env_close(LMDB_IT(env));
+
+	pefree(info->dbf, info->flags & DBA_PERSISTENT);
+}
+
+DBA_FETCH_FUNC(lmdb)
+{
+	int rc;
+	MDB_val k, v;
+	char *ret = NULL;
+	
+	if (LMDB_IT(cur)) {
+		rc = mdb_txn_renew(LMDB_IT(txn));
+	} else {
+		rc = mdb_txn_begin(LMDB_IT(env), NULL, MDB_RDONLY, &LMDB_IT(txn));
+	}
+	if (rc) {
+		php_error_docref1(NULL, key, E_WARNING, "%s", mdb_strerror(rc));
+		return NULL;
+	}
+
+	k.mv_size = keylen;
+	k.mv_data = key;
+
+	rc = mdb_get(LMDB_IT(txn), LMDB_IT(dbi), &k, &v);
+	if (rc) {
+		if (MDB_NOTFOUND != rc) {
+			php_error_docref1(NULL, key, E_WARNING, "%s", mdb_strerror(rc)); 
+		}
+		mdb_txn_abort(LMDB_IT(txn));
+		return NULL;
+	}
+
+	if (v.mv_data) {
+		if(newlen) *newlen = v.mv_size;
+		ret = estrndup(v.mv_data, v.mv_size);
+	}
+
+	if (LMDB_IT(cur)) {
+		mdb_txn_reset(LMDB_IT(txn));
+	} else {
+		mdb_txn_abort(LMDB_IT(txn));
+	}
+
+	return ret;
+}
+
+DBA_UPDATE_FUNC(lmdb)
+{
+	int rc;
+	MDB_val k, v;
+
+	rc = mdb_txn_begin(LMDB_IT(env), NULL, 0, &LMDB_IT(txn));
+	if (rc) {
+		php_error_docref2(NULL, key, val, E_WARNING, "%s", mdb_strerror(rc));
+		return FAILURE;
+	}
+
+	k.mv_size = keylen;
+	k.mv_data = key;
+	v.mv_size = vallen;
+	v.mv_data = val;
+
+	rc = mdb_put(LMDB_IT(txn), LMDB_IT(dbi), &k, &v, mode == 1 ? MDB_NOOVERWRITE : 0);
+	if (rc) {
+		if (MDB_KEYEXIST != rc) {
+			php_error_docref2(NULL, key, val, E_WARNING, "%s", mdb_strerror(rc));
+		}
+		mdb_txn_abort(LMDB_IT(txn));
+		return FAILURE;
+	}
+
+	rc = mdb_txn_commit(LMDB_IT(txn));
+	if (rc) {
+		php_error_docref2(NULL, key, val, E_WARNING, "%s", mdb_strerror(rc));
+		mdb_txn_abort(LMDB_IT(txn));
+		return FAILURE;
+	}
+
+	return SUCCESS;
+}
+
+DBA_EXISTS_FUNC(lmdb)
+{
+	int rc;
+	MDB_val k, v;
+
+	if (LMDB_IT(cur)) {
+		rc = mdb_txn_renew(LMDB_IT(txn));
+	} else {
+		rc = mdb_txn_begin(LMDB_IT(env), NULL, MDB_RDONLY, &LMDB_IT(txn));
+	}
+	if (rc) {
+		php_error_docref1(NULL, key, E_WARNING, "%s", mdb_strerror(rc));
+		return FAILURE;
+	}
+
+	k.mv_size = keylen;
+	k.mv_data = key;
+
+	rc = mdb_get(LMDB_IT(txn), LMDB_IT(dbi), &k, &v);
+	if (rc) {
+		if (MDB_NOTFOUND != rc) {
+			php_error_docref1(NULL, key, E_WARNING, "%s", mdb_strerror(rc)); 
+		}
+		mdb_txn_abort(LMDB_IT(txn));
+		return FAILURE;
+	}
+
+	if (LMDB_IT(cur)) {
+		mdb_txn_reset(LMDB_IT(txn));
+	} else {
+		mdb_txn_abort(LMDB_IT(txn));
+	}
+
+	return SUCCESS;
+}
+
+DBA_DELETE_FUNC(lmdb)
+{
+	int rc;
+	MDB_val k;
+
+	rc = mdb_txn_begin(LMDB_IT(env), NULL, 0, &LMDB_IT(txn));
+	if (rc) {
+		php_error_docref1(NULL, key, E_WARNING, "%s", mdb_strerror(rc));
+		return FAILURE;
+	}
+
+	k.mv_size = keylen;
+	k.mv_data = key;
+
+	rc = mdb_del(LMDB_IT(txn), LMDB_IT(dbi), &k, NULL);
+	if (!rc) {
+		rc = mdb_txn_commit(LMDB_IT(txn));
+		if (rc) {
+			php_error_docref1(NULL, key, E_WARNING, "%s", mdb_strerror(rc));
+			mdb_txn_abort(LMDB_IT(txn));
+			return FAILURE;
+		}
+		return SUCCESS;
+	}
+
+	php_error_docref1(NULL, key, E_WARNING, "%s", mdb_strerror(rc));
+
+	return FAILURE;
+}
+
+DBA_FIRSTKEY_FUNC(lmdb)
+{
+	int rc;
+	MDB_val k, v;
+	char *ret = NULL;
+
+	rc = mdb_txn_begin(LMDB_IT(env), NULL, MDB_RDONLY, &LMDB_IT(txn));
+	if (rc) {
+		php_error_docref0(NULL, E_WARNING, "%s", mdb_strerror(rc));
+		return NULL;
+	}
+
+	rc = mdb_cursor_open(LMDB_IT(txn), LMDB_IT(dbi), &LMDB_IT(cur));
+	if (rc) {
+		mdb_txn_abort(LMDB_IT(txn));
+		php_error_docref0(NULL, E_WARNING, "%s", mdb_strerror(rc));
+		return NULL;
+	}
+
+	rc = mdb_cursor_get(LMDB_IT(cur), &k, &v, MDB_FIRST); 
+	if (rc) {
+		mdb_txn_abort(LMDB_IT(txn));
+		mdb_cursor_close(LMDB_IT(cur));
+		if (MDB_NOTFOUND != rc) {
+			php_error_docref0(NULL, E_WARNING, "%s", mdb_strerror(rc));
+		}
+		return NULL;
+	}
+
+	if(k.mv_data) {
+		if(newlen) *newlen = k.mv_size;
+		ret = estrndup(k.mv_data, k.mv_size);
+	}
+
+	mdb_txn_reset(LMDB_IT(txn));
+
+	return ret;
+}
+
+DBA_NEXTKEY_FUNC(lmdb)
+{
+	int rc;
+	MDB_val k, v;
+	char *ret = NULL;
+
+	rc = mdb_txn_renew(LMDB_IT(txn));
+	if (rc) {
+		php_error_docref0(NULL, E_WARNING, "%s", mdb_strerror(rc));
+		return NULL;
+	}
+
+	rc = mdb_cursor_get(LMDB_IT(cur), &k, &v, MDB_NEXT); 
+	if (rc) {
+		mdb_txn_abort(LMDB_IT(txn));
+		mdb_cursor_close(LMDB_IT(cur));
+		LMDB_IT(cur) = NULL;
+		if (MDB_NOTFOUND != rc) {
+			php_error_docref0(NULL, E_WARNING, "%s", mdb_strerror(rc));
+		}
+		return NULL;
+	}
+
+	if(k.mv_data) {
+		if(newlen) *newlen = k.mv_size;
+		ret = estrndup(k.mv_data, k.mv_size);
+	}
+
+	mdb_txn_reset(LMDB_IT(txn));
+
+	return ret;
+}
+
+DBA_OPTIMIZE_FUNC(lmdb)
+{
+	return SUCCESS;
+}
+
+DBA_SYNC_FUNC(lmdb)
+{
+	int rc;
+
+	rc = mdb_env_sync(LMDB_IT(env), 1);
+	if (rc) {
+			php_error_docref0(NULL, E_WARNING, "%s", mdb_strerror(rc));
+			return FAILURE;
+	}
+
+	return SUCCESS;
+}
+
+DBA_INFO_FUNC(lmdb)
+{
+	return estrdup(MDB_VERSION_STRING);
+}
+
+#endif
+
+/*
+ * Local variables:
+ * tab-width: 4
+ * c-basic-offset: 4
+ * End:
+ * vim600: sw=4 ts=4 fdm=marker
+ * vim<600: sw=4 ts=4
+ */