src/frontend/store.c

   1 /*
   2  * SysDB - src/frontend/store.c
   3  * Copyright (C) 2015 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  */
  27
  28 #include "sysdb.h"
  29
  30 #include "core/store.h"
  31 #include "frontend/connection-private.h"
  32 #include "utils/error.h"
  33 #include "utils/proto.h"
  34 #include "utils/strbuf.h"
  35
  36 #include <ctype.h>
  37 #include <string.h>
  38
  39 /*
  40  * private helper functions
  41  */
  42
  43 static int
  44 store_reply(sdb_conn_t *conn, int type, const char *name, int status)
  45 {
  46         if (status < 0) {
  47                 sdb_strbuf_sprintf(conn->errbuf, "STORE: Failed to store %s object",
  48                                 SDB_STORE_TYPE_TO_NAME(type));
  49                 return -1;
  50         }
  51
  52         if (! status) {
  53                 char msg[strlen(name) + 64];
  54                 snprintf(msg, sizeof(msg), "Successfully stored %s %s",
  55                                 SDB_STORE_TYPE_TO_NAME(type), name);
  56                 msg[sizeof(msg) - 1] = '\0';
  57                 sdb_connection_send(conn, SDB_CONNECTION_OK,
  58                                 (uint32_t)strlen(msg), msg);
  59         }
  60         else {
  61                 char msg[strlen(name) + 64];
  62                 snprintf(msg, sizeof(msg), "%s %s already up to date",
  63                                 SDB_STORE_TYPE_TO_NAME(type), name);
  64                 msg[0] = (char)toupper((int)msg[0]);
  65                 msg[sizeof(msg) - 1] = '\0';
  66                 sdb_connection_send(conn, SDB_CONNECTION_OK,
  67                                 (uint32_t)strlen(msg), msg);
  68         }
  69         return 0;
  70 } /* store_reply */
  71
  72 /*
  73  * public API
  74  */
  75
  76 /*
  77  * TODO: move sdb_fe_store to frontend/query.c and let it build an AST
  78  */
  79
  80 int
  81 sdb_fe_store(sdb_conn_t *conn)
  82 {
  83         uint32_t type;
  84
  85         const char *buf = sdb_strbuf_string(conn->buf);
  86         size_t len = conn->cmd_len;
  87         ssize_t n;
  88
  89         if ((! conn) || (conn->cmd != SDB_CONNECTION_STORE))
  90                 return -1;
  91
  92         if ((n = sdb_proto_unmarshal_int32(buf, len, &type)) < 0) {
  93                 sdb_log(SDB_LOG_ERR, "frontend: Invalid command length %zu for "
  94                                 "STORE command", len);
  95                 sdb_strbuf_sprintf(conn->errbuf,
  96                                 "STORE: Invalid command length %zu", len);
  97                 return -1;
  98         }
  99
 100         switch (type) {
 101                 case SDB_HOST:
 102                 {
 103                         sdb_proto_host_t host;
 104                         if (sdb_proto_unmarshal_host(buf, len, &host) < 0) {
 105                                 sdb_strbuf_sprintf(conn->errbuf,
 106                                                 "STORE: Failed to unmarshal host object");
 107                                 return -1;
 108                         }
 109                         return sdb_fe_store_host(conn, &host);
 110                 }
 111                 case SDB_SERVICE:
 112                 {
 113                         sdb_proto_service_t svc;
 114                         if (sdb_proto_unmarshal_service(buf, len, &svc) < 0) {
 115                                 sdb_strbuf_sprintf(conn->errbuf,
 116                                                 "STORE: Failed to unmarshal service object");
 117                                 return -1;
 118                         }
 119                         return sdb_fe_store_service(conn, &svc);
 120                 }
 121                 case SDB_METRIC:
 122                 {
 123                         sdb_proto_metric_t metric;
 124                         if (sdb_proto_unmarshal_metric(buf, len, &metric) < 0) {
 125                                 sdb_strbuf_sprintf(conn->errbuf,
 126                                                 "STORE: Failed to unmarshal metric object");
 127                                 return -1;
 128                         }
 129                         return sdb_fe_store_metric(conn, &metric);
 130                 }
 131         }
 132         if (type & SDB_ATTRIBUTE) {
 133                 sdb_proto_attribute_t attr;
 134                 int status;
 135                 if (sdb_proto_unmarshal_attribute(buf, len, &attr) < 0) {
 136                         sdb_strbuf_sprintf(conn->errbuf,
 137                                         "STORE: Failed to unmarshal attribute object");
 138                         return -1;
 139                 }
 140                 status = sdb_fe_store_attribute(conn, &attr);
 141                 sdb_data_free_datum(&attr.value);
 142                 return status;
 143         }
 144
 145         sdb_log(SDB_LOG_ERR, "frontend: Invalid object type %d for "
 146                         "STORE COMMAND", type);
 147         sdb_strbuf_sprintf(conn->errbuf, "STORE: Invalid object type %d", type);
 148         return -1;
 149 } /* sdb_fe_store */
 150
 151 int
 152 sdb_fe_store_host(sdb_conn_t *conn, const sdb_proto_host_t *host)
 153 {
 154         if ((! conn) || (! host) || (! host->name))
 155                 return -1;
 156
 157         return store_reply(conn, SDB_HOST, host->name,
 158                         sdb_store_host(host->name, host->last_update));
 159 } /* sdb_fe_store_host */
 160
 161 int
 162 sdb_fe_store_service(sdb_conn_t *conn, const sdb_proto_service_t *svc)
 163 {
 164         char name[svc ? strlen(svc->hostname) + strlen(svc->name) + 2 : 2];
 165
 166         if ((! conn) || (! svc) || (! svc->hostname) || (! svc->name))
 167                 return -1;
 168
 169         snprintf(name, sizeof(name), "%s.%s", svc->hostname, svc->name);
 170         return store_reply(conn, SDB_SERVICE, name,
 171                         sdb_store_service(svc->hostname, svc->name, svc->last_update));
 172 } /* sdb_fe_store_service */
 173
 174 int
 175 sdb_fe_store_metric(sdb_conn_t *conn, const sdb_proto_metric_t *metric)
 176 {
 177         sdb_metric_store_t store;
 178         char name[metric ? strlen(metric->hostname) + strlen(metric->name) + 2 : 2];
 179
 180         if ((! conn) || (! metric) || (! metric->hostname) || (! metric->name))
 181                 return -1;
 182
 183         store.type = metric->store_type;
 184         store.id = metric->store_id;
 185         snprintf(name, sizeof(name), "%s.%s", metric->hostname, metric->name);
 186         return store_reply(conn, SDB_METRIC, name,
 187                         sdb_store_metric(metric->hostname, metric->name,
 188                                 &store, metric->last_update));
 189 } /* sdb_fe_store_metric */
 190
 191 int
 192 sdb_fe_store_attribute(sdb_conn_t *conn, const sdb_proto_attribute_t *attr)
 193 {
 194         char name[attr ? strlen(attr->parent) + strlen(attr->key) + 2 : 2];
 195         int status;
 196
 197         if ((! conn) || (! attr) || (! attr->parent) || (! attr->key))
 198                 return -1;
 199
 200         if (attr->parent_type == SDB_HOST)
 201                 status = sdb_store_attribute(attr->parent,
 202                                 attr->key, &attr->value, attr->last_update);
 203         else if (attr->parent_type == SDB_SERVICE)
 204                 status = sdb_store_service_attr(attr->hostname, attr->parent,
 205                                 attr->key, &attr->value, attr->last_update);
 206         else if (attr->parent_type == SDB_METRIC)
 207                 status = sdb_store_metric_attr(attr->hostname, attr->parent,
 208                                 attr->key, &attr->value, attr->last_update);
 209         else {
 210                 sdb_strbuf_sprintf(conn->errbuf,
 211                                 "STORE: Invalid parent object type %d",
 212                                 attr->parent_type);
 213                 return -1;
 214         }
 215
 216         snprintf(name, sizeof(name), "%s.%s", attr->parent, attr->key);
 217         return store_reply(conn, attr->parent_type | SDB_ATTRIBUTE, name, status);
 218 } /* sdb_fe_store_attribute */
 219
 220 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */
 221