1 /*
2 * SysDB - src/frontend/query.c
3 * Copyright (C) 2013 Sebastian 'tokkee' Harl <sh@tokkee.org>
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
17 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
18 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR
19 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
20 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
28 #include "sysdb.h"
30 #include "core/store.h"
31 #include "frontend/connection-private.h"
32 #include "utils/error.h"
33 #include "utils/strbuf.h"
35 #include <errno.h>
37 /*
38 * public API
39 */
41 int
42 sdb_fe_exec(sdb_conn_t *conn, sdb_conn_node_t *node)
43 {
44 if (! node)
45 return -1;
47 switch (node->cmd) {
48 case CONNECTION_FETCH:
49 return sdb_fe_fetch(conn, CONN_FETCH(node)->name);
50 case CONNECTION_LIST:
51 return sdb_fe_list(conn);
53 default:
54 sdb_log(SDB_LOG_ERR, "frontend: Unknown command %i", node->cmd);
55 return -1;
56 }
57 return -1;
58 } /* sdb_fe_exec */
60 int
61 sdb_fe_fetch(sdb_conn_t *conn, const char *name)
62 {
63 sdb_strbuf_t *buf;
64 sdb_store_base_t *host;
66 host = sdb_store_get_host(name);
67 if (! host) {
68 sdb_log(SDB_LOG_DEBUG, "frontend: Failed to fetch host '%s': "
69 "not found", name);
71 sdb_strbuf_sprintf(conn->errbuf, "Host %s not found", name);
72 return -1;
73 }
75 buf = sdb_strbuf_create(1024);
76 if (! buf) {
77 char errbuf[1024];
78 sdb_log(SDB_LOG_ERR, "frontend: Failed to create "
79 "buffer to handle FETCH command: %s",
80 sdb_strerror(errno, errbuf, sizeof(errbuf)));
82 sdb_strbuf_sprintf(conn->errbuf, "Out of memory");
83 sdb_strbuf_destroy(buf);
84 sdb_object_deref(SDB_OBJ(host));
85 return -1;
86 }
88 if (sdb_store_host_tojson(host, buf, /* flags = */ 0)) {
89 sdb_log(SDB_LOG_ERR, "frontend: Failed to serialize "
90 "host '%s' to JSON", name);
91 sdb_strbuf_sprintf(conn->errbuf, "Out of memory");
92 sdb_strbuf_destroy(buf);
93 sdb_object_deref(SDB_OBJ(host));
94 return -1;
95 }
97 sdb_connection_send(conn, CONNECTION_OK,
98 (uint32_t)sdb_strbuf_len(buf), sdb_strbuf_string(buf));
99 sdb_strbuf_destroy(buf);
100 sdb_object_deref(SDB_OBJ(host));
101 return 0;
102 } /* sdb_fe_fetch */
104 int
105 sdb_fe_list(sdb_conn_t *conn)
106 {
107 sdb_strbuf_t *buf;
109 buf = sdb_strbuf_create(1024);
110 if (! buf) {
111 char errbuf[1024];
112 sdb_log(SDB_LOG_ERR, "frontend: Failed to create "
113 "buffer to handle LIST command: %s",
114 sdb_strerror(errno, errbuf, sizeof(errbuf)));
116 sdb_strbuf_sprintf(conn->errbuf, "Out of memory");
117 sdb_strbuf_destroy(buf);
118 return -1;
119 }
121 if (sdb_store_tojson(buf, /* flags = */ 0)) {
122 sdb_log(SDB_LOG_ERR, "frontend: Failed to serialize "
123 "store to JSON");
124 sdb_strbuf_sprintf(conn->errbuf, "Out of memory");
125 sdb_strbuf_destroy(buf);
126 return -1;
127 }
129 sdb_connection_send(conn, CONNECTION_OK,
130 (uint32_t)sdb_strbuf_len(buf), sdb_strbuf_string(buf));
131 sdb_strbuf_destroy(buf);
132 return 0;
133 } /* sdb_fe_list */
135 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */