first somewhat working version.

Change-Id: I3101dc6b8add87eccac3bbf177b1320137f72463
Signed-off-by: Loïc Collignon <loic.collignon@iot.bzh>

1 files changed, 269 insertions, 0 deletions

diff --git a/ll-store-binding/src/ll-store-binding.c b/ll-store-binding/src/ll-store-binding.c
new file mode 100644
index 0000000..fd65552
--- /dev/null
+++ b/ll-store-binding/src/ll-store-binding.c
@@ -0,0 +1,269 @@
+#define _GNU_SOURCE
+#include <stdio.h>
+#include <string.h>
+#include <json-c/json.h>
+#include <lmdb.h>
+
+#define AFB_BINDING_VERSION 2
+#include <afb/afb-binding.h>
+
+#define DB_FILE "/home/agl/projects/ll-store-binding/build/src/ll-store-binding.lmdb"
+
+MDB_env* dbenv;
+
+/// @brief Get a string from a json object.
+/// @param[in] obj Json object from wich the string is queried.
+/// @param[in] name Name of the string to get.
+/// @return 
+const char* get_json_string(struct json_object* obj, const char* name)
+{
+	if (!obj || !name || !strlen(name)) return NULL;
+	
+	struct json_object* item = NULL;
+	if (!json_object_object_get_ex(obj, name, &item) || !item) return NULL;
+
+	return json_object_get_string(item);
+}
+
+char* make_key(const char* username, const char* appname, const char* tagname)
+{
+	size_t sz_username = username ? strlen(username) : 0;
+	size_t sz_appname = appname ? strlen(appname) : 0;
+	size_t sz_tagname = tagname ? strlen(tagname) : 0;
+	size_t sz_total = sz_username + sz_appname + sz_tagname + 3;
+
+	char* result = (char*)malloc(sz_total);
+	memset(result, sz_total, 0);
+
+	strcpy(result, username);
+	result[sz_username] = '.';
+	
+	strcpy(result + sz_username + 1, appname);
+	result[sz_username + 1 + sz_appname] = '.';
+
+	strcpy(result + sz_username + 1 + sz_appname + 1, tagname);
+	
+	return result;
+}
+
+static void verb_get(struct afb_req req)
+{
+	int r;
+	struct json_object* args = afb_req_json(req);
+
+	AFB_INFO("args:\n%s\n", json_object_to_json_string_ext(args, JSON_C_TO_STRING_PRETTY));
+
+	const char* username = get_json_string(args, "username");
+	const char* appname = get_json_string(args, "appname");
+	const char* tagname = get_json_string(args, "tagname");
+
+	if (!username || !appname || !tagname)
+	{
+		AFB_ERROR("[store] username, appname and tagname must be provided!");
+		afb_req_fail(req, "username, appname and tagname must be provided!", NULL);
+		return;
+	}
+
+	char* keyname = make_key(username, appname, tagname);
+
+	MDB_txn* txn;
+	r = mdb_txn_begin(dbenv, NULL, 0, &txn);
+	if (r)
+	{
+		free(keyname);
+		AFB_ERROR("Failed to begin a transaction!");
+		afb_req_fail(req, "Failed to begin a transaction!", NULL);
+		return;
+	}
+
+	MDB_dbi dbi;
+	r = mdb_dbi_open(txn, NULL, 0, &dbi);
+	if (r)
+	{
+		free(keyname);
+		mdb_txn_abort(txn);
+		AFB_ERROR("Failed to open the database!");
+		afb_req_fail(req, "Failed to open the database!", NULL);
+		return;
+	}
+
+	MDB_val k;
+	MDB_val v;
+	k.mv_size = strlen(keyname) + 1;
+	k.mv_data = keyname;
+
+	if(mdb_get(txn, dbi, &k, &v))
+	{
+		free(keyname);
+		mdb_txn_abort(txn);
+		mdb_dbi_close(dbenv, dbi);
+		AFB_ERROR("Failed to get the data!");
+		afb_req_fail(req, "Failed to get the data!", NULL);
+		return;
+	}
+
+	char* value = strndup(v.mv_data, v.mv_size + 1);
+	if(mdb_txn_commit(txn))
+	{
+		free(keyname);
+		free(value);
+		mdb_dbi_close(dbenv, dbi);
+		AFB_ERROR("Failed to commit the transaction!");
+		return;
+	}
+		
+	json_object* result = json_object_new_object();
+	json_object_object_add(result, "key", json_object_new_string(keyname));
+	json_object_object_add(result, "value", json_object_new_string(value));
+	afb_req_success(req, result, NULL);
+
+	free(value);
+	free(keyname);
+	return;
+}
+
+static void verb_set(struct afb_req req)
+{
+	int r;
+	struct json_object* args = afb_req_json(req);
+
+	AFB_INFO("args:\n%s\n", json_object_to_json_string_ext(args, JSON_C_TO_STRING_PRETTY));
+
+	const char* username = get_json_string(args, "username");
+	const char* appname = get_json_string(args, "appname");
+	const char* tagname = get_json_string(args, "tagname");
+	const char* value = get_json_string(args, "value");
+
+	if (!username || !appname || !tagname)
+	{
+		AFB_ERROR("[store] username, appname and tagname must be provided!");
+		afb_req_fail(req, "username, appname and tagname must be provided!", NULL);
+		return;
+	}
+
+	char* keyname = make_key(username, appname, tagname);
+
+	MDB_txn* txn;
+	r = mdb_txn_begin(dbenv, NULL, 0, &txn);
+	if (r)
+	{
+		free(keyname);
+		AFB_ERROR("Failed to begin a transaction!");
+		afb_req_fail(req, "Failed to begin a transaction!", NULL);
+		return;
+	}
+
+	MDB_dbi dbi;
+	r = mdb_dbi_open(txn, NULL, 0, &dbi);
+	if (r)
+	{
+		free(keyname);
+		mdb_txn_abort(txn);
+		AFB_ERROR("Failed to open the database!");
+		afb_req_fail(req, "Failed to open the database!", NULL);
+		return;
+	}
+
+	MDB_val k;
+	MDB_val v;
+	k.mv_size = strlen(keyname) + 1;
+	k.mv_data = keyname;
+	v.mv_size = value ? strlen(value) + 1 : 0;
+	v.mv_data = value;
+
+	if(mdb_put(txn, dbi, &k, &v, 0))
+	{
+		free(keyname);
+		mdb_txn_abort(txn);
+		mdb_dbi_close(dbenv, dbi);
+		AFB_ERROR("Failed to get the data!");
+		afb_req_fail(req, "Failed to get the data!", NULL);
+		return;
+	}
+
+	if(mdb_txn_commit(txn))
+	{
+		free(keyname);
+		mdb_dbi_close(dbenv, dbi);
+		AFB_ERROR("Failed to commit the transaction!");
+		return;
+	}
+		
+	json_object* result = json_object_new_object();
+	json_object_object_add(result, "key", json_object_new_string(keyname));
+	json_object_object_add(result, "value", json_object_new_string(value));
+	afb_req_success(req, result, NULL);
+
+	free(keyname);
+	return;
+}
+
+int ll_store_preinit()
+{
+	int r = mdb_env_create(&dbenv);
+	if (r)
+	{
+		AFB_INFO("Failed to create MDB environment!");
+		dbenv = NULL;
+		return r;
+	}
+
+	r = mdb_env_open(dbenv, "/home/agl/ll-store-binding.lmdb", MDB_NOSUBDIR, 0644);
+	if (r)
+	{
+		mdb_env_close(dbenv);
+		dbenv = NULL;
+		AFB_INFO("Failed to open MDB environment!");
+		return r;
+	}
+
+/*
+	MDB_txn* txn;
+	r = mdb_txn_begin(dbenv, NULL, 0, &txn);
+	if (r)
+	{
+		mdb_env_close(dbenv);
+		dbenv = NULL;
+		AFB_INFO("Failed to begin a transaction!");
+		return r;
+	}
+
+	MDB_dbi dbi;
+	r = mdb_dbi_open(txn, NULL, 0, &dbi);
+	if (r)
+	{
+		mdb_env_close(dbenv);
+		AFB_INFO("Failed to open the database!");
+		dbenv = NULL;
+	}
+*/
+	return 0;
+}
+
+static const afb_verb_v2 _ll_store_binding_verbs[]= {
+	{
+		.verb = "get",
+		.callback = verb_get,
+		.auth = NULL,
+		.info = NULL,
+		.session = AFB_SESSION_NONE_V2
+	},
+	{
+		.verb = "set",
+		.callback = verb_set,
+		.auth = NULL,
+		.info = NULL,
+		.session = AFB_SESSION_NONE_V2
+	},
+	{ .verb=NULL}
+};
+
+const struct afb_binding_v2 afbBindingV2 = {
+		.api = "ll-store",
+		.specification = NULL,
+		.verbs = _ll_store_binding_verbs,
+		.preinit = ll_store_preinit,
+		.init = NULL,
+		.onevent = NULL,
+		.noconcurrency = 0
+};