Belated add the actual sqlit3 implementation

Signed-off-by: Shawn Webb <shawn.webb@hardenedbsd.org>

src/sqlite3.c

+/*-
+ * Copyright (c) 2021 Shawn Webb <shawn.webb@hardenedbsd.org>
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <unistd.h>
+
+#include <limits.h>
+
+#include "liblattutil.h"
+
+#define QUERY_CANLOG(q) ((q)->lsq_sql_ctx->lsq_logger != NULL)
+#define QUERY_GETLOGGER(q) ((q)->lsq_sql_ctx->lsq_logger)
+
+static bool _lattutil_sqlite_add_row(lattutil_sqlite_query_t *);
+static bool _lattutil_sqlite_add_column_names(lattutil_sqlite_query_t *, size_t);
+
+EXPORTED_SYM
+lattutil_sqlite_ctx_t *
+lattutil_sqlite_ctx_new(const char *path, lattutil_log_t *logger,
+    uint64_t flags)
+{
+	lattutil_sqlite_ctx_t *ctx;
+
+	if (path == NULL) {
+		return (NULL);
+	}
+
+	ctx = calloc(1, sizeof(*ctx));
+	if (ctx == NULL) {
+		return (NULL);
+	}
+
+	ctx->lsq_path = strdup(path);
+	if (ctx->lsq_path == NULL) {
+		free(ctx);
+		return (NULL);
+	}
+
+	ctx->lsq_logger = logger;
+
+	if (sqlite3_open(path, &(ctx->lsq_sqlctx)) != SQLITE_OK) {
+		free(ctx->lsq_path);
+		free(ctx);
+		return (NULL);
+	}
+
+	return (ctx);
+}
+
+EXPORTED_SYM
+void
+lattutil_sqlite_ctx_free(lattutil_sqlite_ctx_t **ctx)
+{
+	lattutil_sqlite_ctx_t *ctxp;
+
+	if (ctx == NULL || *ctx == NULL) {
+		return;
+	}
+
+	ctxp = *ctx;
+
+	if (ctxp->lsq_sqlctx != NULL) {
+		sqlite3_close(ctxp->lsq_sqlctx);
+	}
+
+	free(ctxp->lsq_path);
+	memset(ctxp, 0, sizeof(*ctxp));
+	free(ctxp);
+	*ctx = NULL;
+}
+
+EXPORTED_SYM
+lattutil_sqlite_query_t *
+lattutil_sqlite_prepare(lattutil_sqlite_ctx_t *ctx, const char *query_string)
+{
+	lattutil_sqlite_query_t *query;
+	bool bool_arg, sqlquery;
+	char *str_arg;
+	int int_arg;
+	va_list args;
+	size_t i;
+	int res;
+
+	if (ctx == NULL || query_string == NULL) {
+		printf("Nope\n");
+		return (NULL);
+	}
+
+	query = calloc(1, sizeof(*query));
+	if (query == NULL) {
+		perror("calloc");
+		return (NULL);
+	}
+
+	query->lsq_querystr = strdup(query_string);
+	if (query->lsq_querystr == NULL) {
+		perror("stdrdup");
+		free(query);
+		return (NULL);
+	}
+
+	query->lsq_result.lsr_rows = ucl_object_typed_new(UCL_ARRAY);
+	if (query->lsq_result.lsr_rows == NULL) {
+		printf("creating rows object failed\n");
+		free(query->lsq_querystr);
+		free(query);
+		return (NULL);
+	}
+
+	res = sqlite3_prepare(ctx->lsq_sqlctx, query_string, -1,
+	    &(query->lsq_stmt), NULL);
+	if (res != SQLITE_OK || query->lsq_stmt == NULL) {
+		printf("sqlite3 returned with: %d\n", res);
+		free(query);
+		return (NULL);
+	}
+
+	query->lsq_sql_ctx = ctx;
+
+	return (query);
+}
+
+EXPORTED_SYM
+void
+lattutil_sqlite_query_free(lattutil_sqlite_query_t **query)
+{
+	lattutil_sqlite_query_t *queryp;
+	size_t i;
+
+	if (query == NULL || *query == NULL) {
+		return;
+	}
+
+	queryp = *query;
+	free(queryp->lsq_querystr);
+
+	if (queryp->lsq_result.lsr_column_names != NULL) {
+		for (i = 0; i < queryp->lsq_result.lsr_ncolumns; i++) {
+			free(queryp->lsq_result.lsr_column_names[i]);
+		}
+
+		free(queryp->lsq_result.lsr_column_names);
+	}
+
+	memset(queryp, 0, sizeof(*queryp));
+	free(queryp);
+
+	*query = NULL;
+}
+
+EXPORTED_SYM
+lattutil_sql_res_t *
+lattutil_sqlite_get_result(lattutil_sqlite_query_t *query)
+{
+
+	if (query == NULL) {
+		return (NULL);
+	}
+
+	return (&(query->lsq_result));
+}
+
+EXPORTED_SYM
+bool
+lattutil_sqlite_bind_int(lattutil_sqlite_query_t *query, int paramno,
+    int val)
+{
+
+	if (query == NULL) {
+		return (false);
+	}
+
+	return (sqlite3_bind_int(query->lsq_stmt, paramno, val) ==
+	    SQLITE_OK);
+}
+
+EXPORTED_SYM
+bool
+lattutil_sqlite_bind_string(lattutil_sqlite_query_t *query, int paramno,
+    const char *val)
+{
+
+	if (query == NULL || val == NULL) {
+		return (false);
+	}
+
+	return (sqlite3_bind_text(query->lsq_stmt, paramno, val, -1,
+	    SQLITE_TRANSIENT) == SQLITE_OK);
+}
+
+EXPORTED_SYM
+bool
+lattutil_sqlite_bind_blob(lattutil_sqlite_query_t *query, int paramno,
+   void *val, size_t sz)
+{
+
+	if (query == NULL || val == NULL || sz > INT_MAX) {
+		return (false);
+	}
+
+	return (sqlite3_bind_blob(query->lsq_stmt, paramno, val,
+	    (int)sz, NULL) == SQLITE_OK);
+}
+
+EXPORTED_SYM
+bool
+lattutil_sqlite_bind_time(lattutil_sqlite_query_t *query, int paramno,
+    time_t val)
+{
+
+	if (query == NULL) {
+		return (NULL);
+	}
+
+	return (sqlite3_bind_int(query->lsq_stmt, paramno, val) ==
+	    SQLITE_OK);
+}
+
+EXPORTED_SYM
+bool
+lattutil_sqlite_exec(lattutil_sqlite_query_t *query)
+{
+	lattutil_log_t *logger;
+	bool ret;
+	int res;
+
+	if (query == NULL || query->lsq_stmt == NULL) {
+		return (false);
+	}
+
+	logger = QUERY_GETLOGGER(query);
+
+	ret = true;
+
+	while (true) {
+		res = sqlite3_step(query->lsq_stmt);
+		switch (res) {
+		case SQLITE_DONE:
+			goto end;
+		case SQLITE_ROW:
+			if (!_lattutil_sqlite_add_row(query)) {
+				if (QUERY_CANLOG(query)) {
+					logger->ll_log_err( logger, -1,
+					    "Unable to add row to sqlite object");
+				}
+				ret = false;
+				goto end;
+			}
+			break;
+		default:
+			if (QUERY_CANLOG(query)) {
+				logger->ll_log_err(logger, -1,
+				    "Unhandled sqlite3_step result: %d", res);
+			}
+			ret = false;
+			goto end;
+		}
+	}
+
+end:
+	sqlite3_finalize(query->lsq_stmt);
+	query->lsq_stmt = NULL;
+
+	return (ret);
+}
+
+static bool
+_lattutil_sqlite_add_row(lattutil_sqlite_query_t *query)
+{
+	ucl_object_t *colobj, *rowobj;
+	lattutil_log_t *logger;
+	const char *sval;
+	int *ival, res;
+	size_t i, ncols;
+	bool ret;
+
+	if (query == NULL) {
+		return (false);
+	}
+
+	ret = true;
+	colobj = NULL;
+	rowobj = NULL;
+
+	logger = QUERY_GETLOGGER(query);
+
+	ncols = sqlite3_column_count(query->lsq_stmt);
+	if (ncols <= 0) {
+		/* This should not happen */
+		ret = false;
+		goto end;
+	}
+
+	if (query->lsq_result.lsr_column_names == NULL) {
+		if (!_lattutil_sqlite_add_column_names(query, ncols)) {
+			ret = false;
+			goto end;
+		}
+	}
+
+	rowobj = ucl_object_typed_new(UCL_ARRAY);
+	if (rowobj == NULL) {
+		ret = false;
+		goto end;
+	}
+
+
+	for (i = 0; i < ncols; i++) {
+		switch (sqlite3_column_type(query->lsq_stmt, i)) {
+		case SQLITE_INTEGER:
+			colobj = ucl_object_fromint((int64_t)
+			    sqlite3_column_int(query->lsq_stmt, i));
+			break;
+		case SQLITE_TEXT:
+			sval = (const char *)sqlite3_column_text(query->lsq_stmt, i);
+			printf("Adding %s\n", sval);
+			colobj = ucl_object_fromstring(sval);
+			break;
+		case SQLITE_NULL:
+			colobj = ucl_object_typed_new(UCL_NULL);
+			break;
+		case SQLITE_FLOAT:
+			/* TODO: Support float */
+			break;
+		default:
+			colobj = NULL;
+			break;
+		}
+
+		if (colobj == NULL) {
+			ret = false;
+			if (QUERY_CANLOG(query)) {
+				logger->ll_log_err(logger, -1,
+				    "Unable to determine columnal data format");
+			}
+			goto end;
+		}
+
+		if (!ucl_array_append(rowobj, colobj)) {
+			ret = false;
+			if (QUERY_CANLOG(query)) {
+				logger->ll_log_err(logger, -1,
+				    "Unable to append column to row");
+				goto end;
+			}
+		}
+	}
+
+	if (!ucl_array_append(query->lsq_result.lsr_rows, rowobj)) {
+		if (QUERY_CANLOG(query)) {
+			logger->ll_log_err(logger, -1,
+			    "Unable to append row to rows object");
+		}
+	}
+
+end:
+	if (ret == false) {
+		if (rowobj != NULL) {
+			ucl_object_unref(rowobj);
+			rowobj = NULL;
+		}
+	}
+	return (ret);
+}
+
+static bool
+_lattutil_sqlite_add_column_names(lattutil_sqlite_query_t *query, size_t ncols)
+{
+	size_t i;
+
+	if (query == NULL || ncols == 0) {
+		return (false);
+	}
+
+	query->lsq_result.lsr_ncolumns = ncols;
+	query->lsq_result.lsr_column_names = calloc(ncols, sizeof(char **));
+	if (query->lsq_result.lsr_column_names == NULL) {
+		return (false);
+	}
+
+	for (i = 0; i < ncols; i++) {
+		query->lsq_result.lsr_column_names[i] = strdup(
+		    sqlite3_column_name(query->lsq_stmt, i));
+		if (query->lsq_result.lsr_column_names[i] == NULL) {
+			return (false);
+		}
+	}
+
+	return (true);
+}