summaryrefslogtreecommitdiffstats
path: root/test/drive_sql.cc
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--test/drive_sql.cc78
1 files changed, 78 insertions, 0 deletions
diff --git a/test/drive_sql.cc b/test/drive_sql.cc
new file mode 100644
index 0000000..243e78a
--- /dev/null
+++ b/test/drive_sql.cc
@@ -0,0 +1,78 @@
+#include <iostream>
+
+#include <assert.h>
+#include <sqlite3.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "base/auto_mem.hh"
+#include "base/injector.hh"
+#include "regexp_vtab.hh"
+#include "sqlite-extension-func.hh"
+#include "xpath_vtab.hh"
+
+struct callback_state {
+ int cs_row;
+};
+
+static int
+sql_callback(void* ptr, int ncols, char** colvalues, char** colnames)
+{
+ struct callback_state* cs = (struct callback_state*) ptr;
+
+ printf("Row %d:\n", cs->cs_row);
+ for (int lpc = 0; lpc < ncols; lpc++) {
+ printf(" Column %10s: %s\n", colnames[lpc], colvalues[lpc]);
+ }
+
+ cs->cs_row += 1;
+
+ return 0;
+}
+
+int
+main(int argc, char* argv[])
+{
+ int retval = EXIT_SUCCESS;
+ auto_mem<sqlite3> db(sqlite3_close);
+ std::string stmt;
+
+ log_argv(argc, argv);
+
+ if (argc == 2) {
+ stmt = argv[1];
+ } else {
+ std::getline(std::cin, stmt, '\0');
+ }
+
+ if (sqlite3_open(":memory:", db.out()) != SQLITE_OK) {
+ fprintf(stderr, "error: unable to make sqlite memory database\n");
+ retval = EXIT_FAILURE;
+ } else {
+ auto_mem<char> errmsg(sqlite3_free);
+ struct callback_state state;
+
+ memset(&state, 0, sizeof(state));
+
+ {
+ int register_collation_functions(sqlite3 * db);
+
+ register_sqlite_funcs(db.in(), sqlite_registration_funcs);
+ register_collation_functions(db.in());
+ }
+
+ register_regexp_vtab(db.in());
+ register_xpath_vtab(db.in());
+
+ if (sqlite3_exec(
+ db.in(), stmt.c_str(), sql_callback, &state, errmsg.out())
+ != SQLITE_OK)
+ {
+ fprintf(stderr, "error: sqlite3_exec failed -- %s\n", errmsg.in());
+ retval = EXIT_FAILURE;
+ }
+ }
+
+ return retval;
+}