X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=src%2Ffrontend%2Fquery.c;h=96c46d28db6f2b3425d90cd1079a363882779983;hb=192c24079685a32bcfff24e332c460ced0e51565;hp=9e3b66a3a5c2b55579dccb5fdc4741631930e95d;hpb=c2b7616c8a80c5de6356b675d55a4ee36415fd7b;p=sysdb.git diff --git a/src/frontend/query.c b/src/frontend/query.c index 9e3b66a..96c46d2 100644 --- a/src/frontend/query.c +++ b/src/frontend/query.c @@ -27,9 +27,9 @@ #include "sysdb.h" -#include "core/error.h" #include "core/store.h" #include "frontend/connection-private.h" +#include "utils/error.h" #include "utils/strbuf.h" #include @@ -38,6 +38,69 @@ * public API */ +int +sdb_fe_exec(sdb_conn_t *conn, sdb_conn_node_t *node) +{ + if (! node) + return -1; + + switch (node->cmd) { + case CONNECTION_FETCH: + return sdb_fe_fetch(conn, CONN_FETCH(node)->name); + case CONNECTION_LIST: + return sdb_fe_list(conn); + + default: + sdb_log(SDB_LOG_ERR, "frontend: Unknown command %i", node->cmd); + return -1; + } + return -1; +} /* sdb_fe_exec */ + +int +sdb_fe_fetch(sdb_conn_t *conn, const char *name) +{ + sdb_strbuf_t *buf; + sdb_store_base_t *host; + + host = sdb_store_get_host(name); + if (! host) { + sdb_log(SDB_LOG_DEBUG, "frontend: Failed to fetch host '%s': " + "not found", name); + + sdb_strbuf_sprintf(conn->errbuf, "Host %s not found", name); + return -1; + } + + buf = sdb_strbuf_create(1024); + if (! buf) { + char errbuf[1024]; + sdb_log(SDB_LOG_ERR, "frontend: Failed to create " + "buffer to handle FETCH command: %s", + sdb_strerror(errno, errbuf, sizeof(errbuf))); + + sdb_strbuf_sprintf(conn->errbuf, "Out of memory"); + sdb_strbuf_destroy(buf); + sdb_object_deref(SDB_OBJ(host)); + return -1; + } + + if (sdb_store_host_tojson(host, buf, /* flags = */ 0)) { + sdb_log(SDB_LOG_ERR, "frontend: Failed to serialize " + "host '%s' to JSON", name); + sdb_strbuf_sprintf(conn->errbuf, "Out of memory"); + sdb_strbuf_destroy(buf); + sdb_object_deref(SDB_OBJ(host)); + return -1; + } + + sdb_connection_send(conn, CONNECTION_OK, + (uint32_t)sdb_strbuf_len(buf), sdb_strbuf_string(buf)); + sdb_strbuf_destroy(buf); + sdb_object_deref(SDB_OBJ(host)); + return 0; +} /* sdb_fe_fetch */ + int sdb_fe_list(sdb_conn_t *conn) { @@ -50,16 +113,15 @@ sdb_fe_list(sdb_conn_t *conn) "buffer to handle LIST command: %s", sdb_strerror(errno, errbuf, sizeof(errbuf))); - /* XXX: send error message */ - sdb_connection_send(conn, CONNECTION_ERROR, 0, NULL); + sdb_strbuf_sprintf(conn->errbuf, "Out of memory"); sdb_strbuf_destroy(buf); return -1; } - if (sdb_store_tojson(buf)) { + if (sdb_store_tojson(buf, /* flags = */ 0)) { sdb_log(SDB_LOG_ERR, "frontend: Failed to serialize " "store to JSON"); - sdb_connection_send(conn, CONNECTION_ERROR, 0, NULL); + sdb_strbuf_sprintf(conn->errbuf, "Out of memory"); sdb_strbuf_destroy(buf); return -1; } @@ -68,7 +130,7 @@ sdb_fe_list(sdb_conn_t *conn) (uint32_t)sdb_strbuf_len(buf), sdb_strbuf_string(buf)); sdb_strbuf_destroy(buf); return 0; -} /* session_start */ +} /* sdb_fe_list */ /* vim: set tw=78 sw=4 ts=4 noexpandtab : */