Code

data: Let sdb_data_expr_eval() check it's arguments.
[sysdb.git] / src / core / data.c
index b990f397013ece515a90cad2ee634039a9863d37..0f10538fe6313f798eef54dd9c1ce081974f4108 100644 (file)
  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
+#if HAVE_CONFIG_H
+#      include "config.h"
+#endif /* HAVE_CONFIG_H */
+
+#include "sysdb.h"
+
 #include "core/data.h"
+#include "utils/error.h"
+
+#include <errno.h>
 
 #include <inttypes.h>
 
 #include <stdlib.h>
 #include <string.h>
 
+#include <math.h>
+
+/*
+ * private helper functions
+ */
+
+static int
+data_concat(const sdb_data_t *d1, const sdb_data_t *d2, sdb_data_t *res)
+{
+       unsigned char *new;
+       unsigned char *s1, *s2;
+       size_t len1, len2;
+
+       if (d1->type == SDB_TYPE_STRING) {
+               s1 = (unsigned char *)d1->data.string;
+               s2 = (unsigned char *)d2->data.string;
+               len1 = s1 ? strlen((char *)s1) : 0;
+               len2 = s2 ? strlen((char *)s2) : 0;
+       }
+       else if (d1->type == SDB_TYPE_BINARY) {
+               s1 = d1->data.binary.datum;
+               s2 = d2->data.binary.datum;
+               len1 = d1->data.binary.length;
+               len2 = d2->data.binary.length;
+       }
+       else
+               return -1;
+
+       if (s1 || s2) {
+               new = malloc(len1 + len2 + 1);
+               if (! new)
+                       return -1;
+       }
+       else
+               new = NULL;
+
+       if (len1)
+               memcpy(new, s1, len1);
+       if (len2)
+               memcpy(new + len1, s2, len2);
+       if (new)
+               new[len1 + len2] = '\0';
+
+       res->type = d1->type;
+       if (res->type == SDB_TYPE_STRING) {
+               res->data.string = (char *)new;
+       }
+       else {
+               res->data.binary.datum = new;
+               res->data.binary.length = len1 + len2;
+       }
+       return 0;
+} /* data_concat */
+
 /*
  * public API
  */
@@ -61,6 +124,7 @@ sdb_data_copy(sdb_data_t *dst, const sdb_data_t *src)
                        break;
        }
 
+       sdb_data_free_datum(dst);
        *dst = tmp;
        return 0;
 } /* sdb_data_copy */
@@ -86,8 +150,238 @@ sdb_data_free_datum(sdb_data_t *datum)
        }
 } /* sdb_data_free_datum */
 
+int
+sdb_data_cmp(const sdb_data_t *d1, const sdb_data_t *d2)
+{
+#define CMP_NULL(a, b) \
+       do { \
+               if (!(a) && !(b)) return 0; \
+               if (!(a)) return -1; \
+               if (!(b)) return 1; \
+       } while (0)
+
+       CMP_NULL(d1, d2);
+
+       if (d1->type != d2->type)
+               return -1;
+
+       switch (d1->type) {
+               case SDB_TYPE_INTEGER:
+                       return SDB_CMP(d1->data.integer, d2->data.integer);
+               case SDB_TYPE_DECIMAL:
+                       return SDB_CMP(d1->data.decimal, d2->data.decimal);
+               case SDB_TYPE_STRING:
+                       CMP_NULL(d1->data.string, d2->data.string);
+                       return strcasecmp(d1->data.string, d2->data.string);
+               case SDB_TYPE_DATETIME:
+                       return SDB_CMP(d1->data.datetime, d2->data.datetime);
+               case SDB_TYPE_BINARY:
+               {
+                       int diff;
+
+                       CMP_NULL(d1->data.binary.datum, d2->data.binary.datum);
+
+                       /* on a common prefix, the shorter datum sorts less */
+                       if (d1->data.binary.length < d2->data.binary.length) {
+                               diff = memcmp(d1->data.binary.datum, d2->data.binary.datum,
+                                               d1->data.binary.length);
+                               diff = diff ? diff : -1;
+                       }
+                       else if (d1->data.binary.length > d2->data.binary.length) {
+                               diff = memcmp(d1->data.binary.datum, d2->data.binary.datum,
+                                               d2->data.binary.length);
+                               diff = diff ? diff : 1;
+                       }
+                       else
+                               diff = memcmp(d1->data.binary.datum, d2->data.binary.datum,
+                                               d1->data.binary.length);
+
+                       return diff;
+               }
+               default:
+                       return -1;
+       }
+#undef CMP_NULL
+} /* sdb_data_cmp */
+
+int
+sdb_data_expr_eval(int op, const sdb_data_t *d1, const sdb_data_t *d2,
+               sdb_data_t *res)
+{
+       if ((! d1) || (! d2) || (! res))
+               return -1;
+       switch (op) {
+               case SDB_DATA_CONCAT:
+                       return data_concat(d1, d2, res);
+               case SDB_DATA_ADD:
+                       if (d1->type != d2->type)
+                               return -1;
+                       switch (d1->type) {
+                               case SDB_TYPE_INTEGER:
+                                       res->data.integer = d1->data.integer + d2->data.integer;
+                                       break;
+                               case SDB_TYPE_DECIMAL:
+                                       res->data.decimal = d1->data.decimal + d2->data.decimal;
+                                       break;
+                               case SDB_TYPE_DATETIME:
+                                       res->data.datetime = d1->data.datetime + d2->data.datetime;
+                                       break;
+                               default:
+                                       return -1;
+                       }
+                       break;
+               case SDB_DATA_SUB:
+                       if (d1->type != d2->type)
+                               return -1;
+                       switch (d1->type) {
+                               case SDB_TYPE_INTEGER:
+                                       res->data.integer = d1->data.integer - d2->data.integer;
+                                       break;
+                               case SDB_TYPE_DECIMAL:
+                                       res->data.decimal = d1->data.decimal - d2->data.decimal;
+                                       break;
+                               case SDB_TYPE_DATETIME:
+                                       res->data.datetime = d1->data.datetime - d2->data.datetime;
+                                       break;
+                               default:
+                                       return -1;
+                       }
+                       break;
+               case SDB_DATA_MUL:
+                       switch (d1->type) {
+                               case SDB_TYPE_INTEGER:
+                                       if (d2->type == SDB_TYPE_INTEGER)
+                                               res->data.integer = d1->data.integer
+                                                       * d2->data.integer;
+                                       else if (d2->type == SDB_TYPE_DATETIME) {
+                                               res->data.datetime = (sdb_time_t)d1->data.integer
+                                                       * d2->data.datetime;
+                                               res->type = SDB_TYPE_DATETIME;
+                                               return 0;
+                                       }
+                                       else
+                                               return -1;
+                                       break;
+                               case SDB_TYPE_DECIMAL:
+                                       if (d2->type == SDB_TYPE_DECIMAL)
+                                               res->data.decimal = d1->data.decimal
+                                                       * d2->data.decimal;
+                                       else if (d2->type == SDB_TYPE_DATETIME) {
+                                               double tmp = d1->data.decimal
+                                                       * (double)d2->data.datetime;
+                                               res->data.datetime = (sdb_time_t)tmp;
+                                               res->type = SDB_TYPE_DATETIME;
+                                               return 0;
+                                       }
+                                       else
+                                               return -1;
+                                       break;
+                               case SDB_TYPE_DATETIME:
+                                       if (d2->type == SDB_TYPE_DATETIME)
+                                               res->data.datetime = d1->data.datetime
+                                                       * d2->data.datetime;
+                                       else if (d2->type == SDB_TYPE_INTEGER) {
+                                               res->data.datetime = d1->data.datetime
+                                                       * (sdb_time_t)d2->data.integer;
+                                               res->type = SDB_TYPE_DATETIME;
+                                               return 0;
+                                       }
+                                       else if (d2->type == SDB_TYPE_DECIMAL) {
+                                               double tmp = (double)d1->data.datetime
+                                                       * d2->data.decimal;
+                                               res->data.datetime = (sdb_time_t)tmp;
+                                               res->type = SDB_TYPE_DATETIME;
+                                               return 0;
+                                       }
+                                       else
+                                               return -1;
+                                       break;
+                               default:
+                                       return -1;
+                       }
+                       break;
+               case SDB_DATA_DIV:
+                       switch (d1->type) {
+                               case SDB_TYPE_INTEGER:
+                                       if (d2->type != SDB_TYPE_INTEGER)
+                                               return -1;
+                                       res->data.integer = d1->data.integer / d2->data.integer;
+                                       break;
+                               case SDB_TYPE_DECIMAL:
+                                       if (d2->type != SDB_TYPE_DECIMAL)
+                                               return -1;
+                                       res->data.decimal = d1->data.decimal / d2->data.decimal;
+                                       break;
+                               case SDB_TYPE_DATETIME:
+                                       if (d2->type == SDB_TYPE_DATETIME)
+                                               res->data.datetime = d1->data.datetime
+                                                       / d2->data.datetime;
+                                       else if (d2->type == SDB_TYPE_INTEGER) {
+                                               res->data.datetime = d1->data.datetime
+                                                       / (sdb_time_t)d2->data.integer;
+                                               res->type = SDB_TYPE_DATETIME;
+                                               return 0;
+                                       }
+                                       else if (d2->type == SDB_TYPE_DECIMAL) {
+                                               double tmp = (double)d1->data.datetime
+                                                       / d2->data.decimal;
+                                               res->data.datetime = (sdb_time_t)tmp;
+                                               res->type = SDB_TYPE_DATETIME;
+                                               return 0;
+                                       }
+                                       else
+                                               return -1;
+                                       break;
+                               default:
+                                       return -1;
+                       }
+                       break;
+               case SDB_DATA_MOD:
+                       switch (d1->type) {
+                               case SDB_TYPE_INTEGER:
+                                       if (d2->type != SDB_TYPE_INTEGER)
+                                               return -1;
+                                       res->data.integer = d1->data.integer % d2->data.integer;
+                                       break;
+                               case SDB_TYPE_DECIMAL:
+                                       if (d2->type != SDB_TYPE_DECIMAL)
+                                               return -1;
+                                       res->data.decimal = fmod(d1->data.decimal, d2->data.decimal);
+                                       break;
+                               case SDB_TYPE_DATETIME:
+                                       if (d2->type == SDB_TYPE_DATETIME)
+                                               res->data.datetime = d1->data.datetime
+                                                       % d2->data.datetime;
+                                       else if (d2->type == SDB_TYPE_INTEGER) {
+                                               res->data.datetime = d1->data.datetime
+                                                       % (sdb_time_t)d2->data.integer;
+                                               res->type = SDB_TYPE_DATETIME;
+                                               return 0;
+                                       }
+                                       else if (d2->type == SDB_TYPE_DECIMAL) {
+                                               double tmp = fmod((double)d1->data.datetime,
+                                                       d2->data.decimal);
+                                               res->data.datetime = (sdb_time_t)tmp;
+                                               res->type = SDB_TYPE_DATETIME;
+                                               return 0;
+                                       }
+                                       else
+                                               return -1;
+                                       break;
+                               default:
+                                       return -1;
+                       }
+                       break;
+               default:
+                       return -1;
+       }
+
+       res->type = d1->type;
+       return 0;
+} /* sdb_data_expr_eval */
+
 size_t
-sdb_data_strlen(sdb_data_t *datum)
+sdb_data_strlen(const sdb_data_t *datum)
 {
        if (! datum)
                return 0;
@@ -115,7 +409,7 @@ sdb_data_strlen(sdb_data_t *datum)
 } /* sdb_data_strlen */
 
 int
-sdb_data_format(sdb_data_t *datum, char *buf, size_t buflen, int quoted)
+sdb_data_format(const sdb_data_t *datum, char *buf, size_t buflen, int quoted)
 {
        char tmp[sdb_data_strlen(datum) + 1];
        char *data = NULL;
@@ -194,5 +488,61 @@ sdb_data_format(sdb_data_t *datum, char *buf, size_t buflen, int quoted)
        return ret;
 } /* sdb_data_format */
 
+int
+sdb_data_parse(char *str, int type, sdb_data_t *data)
+{
+       sdb_data_t tmp;
+
+       char *endptr = NULL;
+
+       errno = 0;
+       switch (type) {
+               case SDB_TYPE_INTEGER:
+                       tmp.data.integer = strtoll(str, &endptr, 0);
+                       break;
+               case SDB_TYPE_DECIMAL:
+                       tmp.data.decimal = strtod(str, &endptr);
+                       break;
+               case SDB_TYPE_STRING:
+                       tmp.data.string = str;
+                       break;
+               case SDB_TYPE_DATETIME:
+                       {
+                               double datetime = strtod(str, &endptr);
+                               tmp.data.datetime = DOUBLE_TO_SDB_TIME(datetime);
+                       }
+                       break;
+               case SDB_TYPE_BINARY:
+                       /* we don't support any binary information containing 0-bytes */
+                       tmp.data.binary.length = strlen(str);
+                       tmp.data.binary.datum = (unsigned char *)str;
+                       break;
+               default:
+                       errno = EINVAL;
+                       return -1;
+       }
+
+       if ((type == SDB_TYPE_INTEGER) || (type == SDB_TYPE_DECIMAL)
+                       || (type == SDB_TYPE_DATETIME)) {
+               if (errno || (str == endptr)) {
+                       char errbuf[1024];
+                       sdb_log(SDB_LOG_ERR, "core: Failed to parse string "
+                                       "'%s' as numeric value (type %i): %s", str, type,
+                                       sdb_strerror(errno, errbuf, sizeof(errbuf)));
+                       return -1;
+               }
+               else if (endptr && (*endptr != '\0'))
+                       sdb_log(SDB_LOG_WARNING, "core: Ignoring garbage after "
+                                       "number while parsing numeric value (type %i): %s.",
+                                       type, endptr);
+       }
+
+       if (data) {
+               *data = tmp;
+               data->type = type;
+       }
+       return 0;
+} /* sdb_data_parse */
+
 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */