#include <stdio.h>
|
#include <assert.h>
|
#include <stdlib.h>
|
#include <string.h>
|
#include <errno.h>
|
#include <stdbool.h>
|
#include <sys/types.h>
|
#include <unistd.h>
|
#include <ctype.h>
|
#include <inttypes.h>
|
|
#include <glib.h>
|
|
#include <pthread.h>
|
#include <gdbus.h>
|
|
#include "json-c/json.h"
|
#include "rkdb.h"
|
#include "common.h"
|
#include "dbus_helpers.h"
|
|
static gboolean signal_datachanged(DBusConnection *conn, char *interface, char *json_str)
|
{
|
DBusMessage *signal;
|
DBusMessageIter iter;
|
|
if (interface == NULL)
|
return FALSE;
|
|
signal = dbus_message_new_signal(DB_PATH,
|
interface, "DataChanged");
|
if (!signal)
|
return FALSE;
|
|
dbus_message_iter_init_append(signal, &iter);
|
dbus_message_iter_append_basic(&iter, DBUS_TYPE_STRING, &json_str);
|
|
dbus_connection_send(conn, signal, NULL);
|
dbus_message_unref(signal);
|
|
return TRUE;
|
}
|
|
static char *cmd_delete(json_object *j_cfg)
|
{
|
char *where = 0;
|
char *table = 0;
|
json_object *j_key = 0;
|
|
table = (char *)json_object_get_string(json_object_object_get(j_cfg, "table"));
|
j_key = json_object_object_get(j_cfg, "key");
|
|
json_object_object_foreach(j_key, key, val) {
|
if (where) {
|
char *tmp = where;
|
if (json_object_get_type(val) == json_type_int)
|
where = g_strdup_printf("%s and %s=%d", tmp, key, (int)json_object_get_int(val));
|
else
|
where = g_strdup_printf("%s and %s='%s'", tmp, key, (char *)json_object_get_string(val));
|
g_free(tmp);
|
} else {
|
if (json_object_get_type(val) == json_type_int)
|
where = g_strdup_printf("%s=%d", key, (int)json_object_get_int(val));
|
else
|
where = g_strdup_printf("%s='%s'", key, (char *)json_object_get_string(val));
|
}
|
}
|
return rkdb_delete(table, where);
|
}
|
|
static char *cmd_drop(json_object *j_cfg)
|
{
|
char *table = (char *)json_object_get_string(json_object_object_get(j_cfg, "table"));
|
return rkdb_drop(table);
|
}
|
|
static char *cmd_create(json_object *j_cfg)
|
{
|
char *table = (char *)json_object_get_string(json_object_object_get(j_cfg, "table"));
|
char *col = (char *)json_object_get_string(json_object_object_get(j_cfg, "col"));
|
|
return rkdb_create(table, col);
|
}
|
|
static char *cmd_select(json_object *j_cfg)
|
{
|
char *col = 0;
|
char *where = 0;
|
char *table = 0;
|
|
json_object *j_key;
|
json_object *j_data;
|
|
table = (char *)json_object_get_string(json_object_object_get(j_cfg, "table"));
|
j_key = json_object_object_get(j_cfg, "key");
|
j_data = json_object_object_get(j_cfg, "data");
|
|
json_object_object_foreach(j_key, key, val) {
|
if (where) {
|
char *tmp = where;
|
if (json_object_get_type(val) == json_type_int)
|
where = g_strdup_printf("%s and %s=%d", tmp, key, (int)json_object_get_int(val));
|
else
|
where = g_strdup_printf("%s and %s='%s'", tmp, key, (char *)json_object_get_string(val));
|
g_free(tmp);
|
} else {
|
if (json_object_get_type(val) == json_type_int)
|
where = g_strdup_printf("%s=%d", key, (int)json_object_get_int(val));
|
else
|
where = g_strdup_printf("%s='%s'", key, (char *)json_object_get_string(val));
|
}
|
}
|
|
col = (char *)json_object_get_string(j_data);
|
char *ret = rkdb_select(table, col, where, NULL, NULL);
|
g_free(where);
|
|
return ret;
|
}
|
|
static char *cmd_update(json_object *j_cfg)
|
{
|
int ret;
|
char *table = 0;
|
json_object *j_key = 0;
|
json_object *j_data = 0;
|
char *j_str;
|
json_object *j_ret;
|
|
table = (char *)json_object_get_string(json_object_object_get(j_cfg, "table"));
|
j_key = json_object_object_get(j_cfg, "key");
|
j_data = json_object_object_get(j_cfg, "data");
|
|
char *where = 0;
|
char *set = 0;
|
|
json_object_object_foreach(j_key, key, val) {
|
if (where) {
|
char *tmp = where;
|
if (json_object_get_type(val) == json_type_int)
|
where = g_strdup_printf("%s and %s=%d", tmp, key, (int)json_object_get_int(val));
|
else
|
where = g_strdup_printf("%s and %s='%s'", tmp, key, (char *)json_object_get_string(val));
|
g_free(tmp);
|
} else {
|
if (json_object_get_type(val) == json_type_int)
|
where = g_strdup_printf("%s=%d", key, (int)json_object_get_int(val));
|
else
|
where = g_strdup_printf("%s='%s'", key, (char *)json_object_get_string(val));
|
}
|
}
|
|
json_object_object_foreach(j_data, key1, val1) {
|
if (set) {
|
char *tmp = set;
|
if (json_object_get_type(val1) == json_type_int)
|
set = g_strdup_printf("%s,%s=%d", tmp, key1, (int)json_object_get_int(val1));
|
else
|
set = g_strdup_printf("%s,%s='%s'", tmp, key1, (char *)json_object_get_string(val1));
|
g_free(tmp);
|
} else {
|
if (json_object_get_type(val1) == json_type_int)
|
set = g_strdup_printf("%s=%d", key1, (int)json_object_get_int(val1));
|
else
|
set = g_strdup_printf("%s='%s'", key1, (char *)json_object_get_string(val1));
|
}
|
}
|
j_str = rkdb_select(table, "*", where, NULL, "0,1");
|
j_ret = json_tokener_parse(j_str);
|
json_object *j_array = json_object_object_get(j_ret, "jData");
|
int num = json_object_array_length(j_array);
|
json_object_put(j_ret);
|
g_free(j_str);
|
|
if (num == 0) {
|
char *cols = 0;
|
char *vals = 0;
|
|
json_object_object_foreach(j_key, key, val) {
|
if (cols) {
|
char *tmp = cols;
|
cols = g_strdup_printf("%s,%s", tmp, key);
|
g_free(tmp);
|
} else {
|
cols = g_strdup_printf("%s", key);
|
}
|
|
if (vals) {
|
char *tmp = vals;
|
if (json_object_get_type(val) == json_type_int)
|
vals = g_strdup_printf("%s,%d", tmp, (int)json_object_get_int(val));
|
else
|
vals = g_strdup_printf("%s,'%s'", tmp, (char *)json_object_get_string(val));
|
g_free(tmp);
|
} else {
|
if (json_object_get_type(val) == json_type_int)
|
vals = g_strdup_printf("%d", (int)json_object_get_int(val));
|
else
|
vals = g_strdup_printf("'%s'", (char *)json_object_get_string(val));
|
}
|
}
|
|
json_object_object_foreach(j_data, key1, val1) {
|
if (cols) {
|
char *tmp = cols;
|
cols = g_strdup_printf("%s,%s", tmp, key1);
|
g_free(tmp);
|
} else {
|
cols = g_strdup_printf("%s", key1);
|
}
|
if (vals) {
|
char *tmp = vals;
|
if (json_object_get_type(val1) == json_type_int)
|
vals = g_strdup_printf("%s,%d", tmp, (int)json_object_get_int(val1));
|
else
|
vals = g_strdup_printf("%s,'%s'", tmp, (char *)json_object_get_string(val1));
|
g_free(tmp);
|
} else {
|
if (json_object_get_type(val1) == json_type_int)
|
vals = g_strdup_printf("%d", (int)json_object_get_int(val1));
|
else
|
vals = g_strdup_printf("'%s'", (char *)json_object_get_string(val1));
|
}
|
}
|
j_str = rkdb_insert(table, cols, vals);
|
if (cols)
|
g_free(cols);
|
if (vals)
|
g_free(vals);
|
} else {
|
j_str = rkdb_update(table, set, where);
|
}
|
|
if (where)
|
g_free(where);
|
if (set)
|
g_free(set);
|
|
return j_str;
|
}
|
|
static char *cmd_insert(json_object *j_cfg)
|
{
|
char *cols = 0;
|
char *vals = 0;
|
char *ret;
|
char *table = 0;
|
json_object *j_key = 0;
|
json_object *j_data = 0;
|
|
table = (char *)json_object_get_string(json_object_object_get(j_cfg, "table"));
|
j_key = json_object_object_get(j_cfg, "key");
|
j_data = json_object_object_get(j_cfg, "data");
|
|
json_object_object_foreach(j_key, key, val) {
|
if (cols) {
|
char *tmp = cols;
|
cols = g_strdup_printf("%s,%s", tmp, key);
|
g_free(tmp);
|
} else {
|
cols = g_strdup_printf("%s", key);
|
}
|
|
if (vals) {
|
char *tmp = vals;
|
if (json_object_get_type(val) == json_type_int)
|
vals = g_strdup_printf("%s,%d", tmp, (int)json_object_get_int(val));
|
else
|
vals = g_strdup_printf("%s,'%s'", tmp, (char *)json_object_get_string(val));
|
g_free(tmp);
|
} else {
|
if (json_object_get_type(val) == json_type_int)
|
vals = g_strdup_printf("%d", (int)json_object_get_int(val));
|
else
|
vals = g_strdup_printf("'%s'", (char *)json_object_get_string(val));
|
}
|
}
|
|
json_object_object_foreach(j_data, key1, val1) {
|
if (cols) {
|
char *tmp = cols;
|
cols = g_strdup_printf("%s,%s", tmp, key1);
|
g_free(tmp);
|
} else {
|
cols = g_strdup_printf("%s", key1);
|
}
|
if (vals) {
|
char *tmp = vals;
|
if (json_object_get_type(val1) == json_type_int)
|
vals = g_strdup_printf("%s,%d", tmp, (int)json_object_get_int(val1));
|
else
|
vals = g_strdup_printf("%s,'%s'", tmp, (char *)json_object_get_string(val1));
|
g_free(tmp);
|
} else {
|
if (json_object_get_type(val1) == json_type_int)
|
vals = g_strdup_printf("%d", (int)json_object_get_int(val1));
|
else
|
vals = g_strdup_printf("'%s'", (char *)json_object_get_string(val1));
|
}
|
}
|
|
ret = rkdb_insert(table, cols, vals);
|
|
if (cols)
|
g_free(cols);
|
if (vals)
|
g_free(vals);
|
|
return ret;
|
}
|
|
DBusMessage *method_cmd(DBusConnection *conn,
|
DBusMessage *msg, void *data)
|
{
|
int ret;
|
const char *sender;
|
DBusMessage *reply;
|
char *json_str;
|
json_object *j_cfg;
|
DBusMessageIter array;
|
char *str = NULL;
|
|
sender = dbus_message_get_sender(msg);
|
|
dbus_message_get_args(msg, NULL, DBUS_TYPE_STRING, &json_str,
|
DBUS_TYPE_INVALID);
|
|
j_cfg = json_tokener_parse(json_str);
|
char *cmd = (char *)json_object_get_string(json_object_object_get(j_cfg, "cmd"));
|
int need_signal = 0;
|
if (g_str_equal(cmd, "Drop")) {
|
str = cmd_drop(j_cfg);
|
need_signal = 1;
|
} else if (g_str_equal(cmd, "Create")) {
|
str = cmd_create(j_cfg);
|
} else if (g_str_equal(cmd, "Select")) {
|
str = cmd_select(j_cfg);
|
} else if (g_str_equal(cmd, "Delete")) {
|
str = cmd_delete(j_cfg);
|
need_signal = 1;
|
} else if (g_str_equal(cmd, "Insert")) {
|
str = cmd_insert(j_cfg);
|
need_signal = 1;
|
} else if (g_str_equal(cmd, "Update")) {
|
str = cmd_update(j_cfg);
|
need_signal = 1;
|
}
|
|
if (need_signal && str) {
|
json_object *j_ret = json_tokener_parse(str);
|
if ((int)json_object_get_int(json_object_object_get(j_ret, "iReturn")) == 0)
|
signal_datachanged(conn, data, json_str);
|
json_object_put(j_ret);
|
}
|
|
json_object_put(j_cfg);
|
|
reply = dbus_message_new_method_return(msg);
|
if (!reply)
|
return NULL;
|
|
dbus_message_iter_init_append(reply, &array);
|
dbus_message_iter_append_basic(&array, DBUS_TYPE_STRING, &str);
|
g_free(str);
|
|
return reply;
|
}
|
|
DBusMessage *method_sql(DBusConnection *conn,
|
DBusMessage *msg, void *data)
|
{
|
const char *sender;
|
DBusMessage *reply;
|
DBusMessageIter array, dict;
|
char *str;
|
char *sql;
|
|
sender = dbus_message_get_sender(msg);
|
|
dbus_message_get_args(msg, NULL, DBUS_TYPE_STRING, &sql,
|
DBUS_TYPE_INVALID);
|
|
str = rkdb_sql(sql);
|
|
reply = dbus_message_new_method_return(msg);
|
if (!reply)
|
return NULL;
|
|
dbus_message_iter_init_append(reply, &array);
|
dbus_message_iter_append_basic(&array, DBUS_TYPE_STRING, &str);
|
g_free(str);
|
|
return reply;
|
}
|
|
void creat_version_table(char *table, char *version)
|
{
|
char *col_para = "id INTEGER PRIMARY KEY," \
|
"sVersion TEXT NOT NULL";
|
g_free(rkdb_create(table, col_para));
|
char *vals = g_strdup_printf("0,'%s'", version);
|
g_free(rkdb_insert(table, "id,sVersion", vals));
|
g_free(vals);
|
}
|
|
int equal_version(char *table, char *version)
|
{
|
int ret = 0;
|
|
char *j_str = rkdb_select(table, "sVersion", "id=0", NULL, NULL);
|
json_object *j_ret = json_tokener_parse(j_str);
|
json_object *j_array = json_object_object_get(j_ret, "jData");
|
int num = json_object_array_length(j_array);
|
if (num > 0) {
|
json_object *j_obj = json_object_array_get_idx(j_array, 0);
|
char *ver = (char *)json_object_get_string(json_object_object_get(j_obj, "sVersion"));
|
if (g_str_equal(ver, version))
|
ret = 1;
|
}
|
json_object_put(j_ret);
|
g_free(j_str);
|
|
return ret;
|
}
|
|
const GDBusMethodTable methods[] = {
|
{
|
GDBUS_ASYNC_METHOD("Cmd",
|
GDBUS_ARGS({ "json", "s" }), GDBUS_ARGS({ "json", "s" }),
|
method_cmd)
|
},
|
{
|
GDBUS_ASYNC_METHOD("Sql",
|
GDBUS_ARGS({ "json", "s" }), GDBUS_ARGS({ "json", "s" }),
|
method_sql)
|
},
|
{ },
|
};
|
|
const GDBusSignalTable signals[] = {
|
{
|
GDBUS_SIGNAL("DataChanged",
|
GDBUS_ARGS({ "json", "s" }))
|
},
|
{ },
|
};
|
