write to stdout instead of db file

column values are prefixed with type and length when -u is passed in
    gufi_query and gufi_sqlite3
    length is 4 digit human readable to avoid confusion between 0xa and newline

always pulling path, epath, fpath, and rpath/path into virtual table

contrib/treediff.c

@@ -255,12 +255,14 @@ static int processdir(QPTPool_t *ctx, const size_t id, void *data, void *args) {
     /* ********************************************** */
     const size_t next_level = cp->level + 1;
 
-    struct PrintArgs print;
-    print.output_buffer = &pa->obufs.buffers[id];
-    print.delim = '/';
-    print.mutex = pa->obufs.mutex;
-    print.outfile = stdout;
-    print.rows = 0;
+    struct PrintArgs print = {
+        .output_buffer = &pa->obufs.buffers[id],
+        .delim = '/',
+        .mutex = pa->obufs.mutex,
+        .outfile = stdout,
+        .rows = 0,
+        .types = NULL,
+    };
 
     char *buf[] = {NULL, NULL};     /* passed to print_parallel */
 

include/bf.h

@@ -178,6 +178,24 @@ struct input {
        refstr_t fin;
    } sql;
 
+   /*
+    * if outputting to STDOUT or OUTFILE, get list of
+    * types of final output to prefix columns with
+    *
+    * set up by gufi_query but cleaned up by input_fini
+    */
+   struct {
+       int prefix;
+
+       /* set if not aggregating */
+       int *tsum;
+       int *sum;
+       int *ent;
+
+       /* set if aggregating */
+       int *agg;
+   } types;
+
    int  printdir;
    int  printing;
    int  printheader;

include/dbutils.h

@@ -102,14 +102,14 @@ extern const char READDIRPLUS_INSERT[];
 /* prefer pentries over entries */
 #define ENTRIES           "entries"
 #define ENTRIES_SCHEMA(name, extra_cols)            \
-    "CREATE TABLE " name "(name TEXT, type TEXT, inode TEXT, mode INT64, nlink INT64, uid INT64, gid INT64, size INT64, blksize INT64, blocks INT64, atime INT64, mtime INT64, ctime INT64, linkname TEXT, xattr_names BLOB, crtime INT64, ossint1 INT64, ossint2 INT64, ossint3 INT64, ossint4 INT64, osstext1 TEXT, osstext2 TEXT" extra_cols ");"
+    "CREATE TABLE " name "(" extra_cols "name TEXT, type TEXT, inode TEXT, mode INT64, nlink INT64, uid INT64, gid INT64, size INT64, blksize INT64, blocks INT64, atime INT64, mtime INT64, ctime INT64, linkname TEXT, xattr_names BLOB, crtime INT64, ossint1 INT64, ossint2 INT64, ossint3 INT64, ossint4 INT64, osstext1 TEXT, osstext2 TEXT);"
 extern const char ENTRIES_CREATE[];
 extern const char ENTRIES_INSERT[];
 
 /* directory metadata + aggregate data */
 #define SUMMARY           "summary"
 #define SUMMARY_SCHEMA(name, extra_cols)            \
-    "CREATE TABLE " name "(name TEXT, type TEXT, inode TEXT, mode INT64, nlink INT64, uid INT64, gid INT64, size INT64, blksize INT64, blocks INT64, atime INT64, mtime INT64, ctime INT64, linkname TEXT, xattr_names BLOB, totfiles INT64, totlinks INT64, minuid INT64, maxuid INT64, mingid INT64, maxgid INT64, minsize INT64, maxsize INT64, totzero INT64, totltk INT64, totmtk INT64, totltm INT64, totmtm INT64, totmtg INT64, totmtt INT64, totsize INT64, minctime INT64, maxctime INT64, minmtime INT64, maxmtime INT64, minatime INT64, maxatime INT64, minblocks INT64, maxblocks INT64, totxattr INT64, depth INT64, mincrtime INT64, maxcrtime INT64, minossint1 INT64, maxossint1 INT64, totossint1 INT64, minossint2 INT64, maxossint2 INT64, totossint2 INT64, minossint3 INT64, maxossint3 INT64, totossint3 INT64, minossint4 INT64, maxossint4 INT64, totossint4 INT64, rectype INT64, pinode TEXT, isroot INT64, rollupscore INT64" extra_cols ");"
+    "CREATE TABLE " name "(" extra_cols "name TEXT, type TEXT, inode TEXT, mode INT64, nlink INT64, uid INT64, gid INT64, size INT64, blksize INT64, blocks INT64, atime INT64, mtime INT64, ctime INT64, linkname TEXT, xattr_names BLOB, totfiles INT64, totlinks INT64, minuid INT64, maxuid INT64, mingid INT64, maxgid INT64, minsize INT64, maxsize INT64, totzero INT64, totltk INT64, totmtk INT64, totltm INT64, totmtm INT64, totmtg INT64, totmtt INT64, totsize INT64, minctime INT64, maxctime INT64, minmtime INT64, maxmtime INT64, minatime INT64, maxatime INT64, minblocks INT64, maxblocks INT64, totxattr INT64, depth INT64, mincrtime INT64, maxcrtime INT64, minossint1 INT64, maxossint1 INT64, totossint1 INT64, minossint2 INT64, maxossint2 INT64, totossint2 INT64, minossint3 INT64, maxossint3 INT64, totossint3 INT64, minossint4 INT64, maxossint4 INT64, totossint4 INT64, rectype INT64, pinode TEXT, isroot INT64, rollupscore INT64);"
 extern const char SUMMARY_CREATE[];
 
 /* view of summary table with rollups */
@@ -136,7 +136,7 @@ extern const char VRPENTRIES_CREATE[];
 /* aggregate data of tree starting at current directory */
 #define TREESUMMARY       "treesummary"
 #define TREESUMMARY_SCHEMA(name, extra_cols)        \
-    "CREATE TABLE " name "(inode TEXT, pinode TEXT, totsubdirs INT64, maxsubdirfiles INT64, maxsubdirlinks INT64, maxsubdirsize INT64, totfiles INT64, totlinks INT64, minuid INT64, maxuid INT64, mingid INT64, maxgid INT64, minsize INT64, maxsize INT64, totzero INT64, totltk INT64, totmtk INT64, totltm INT64, totmtm INT64, totmtg INT64, totmtt INT64, totsize INT64, minctime INT64, maxctime INT64, minmtime INT64, maxmtime INT64, minatime INT64, maxatime INT64, minblocks INT64, maxblocks INT64, totxattr INT64, depth INT64, mincrtime INT64, maxcrtime INT64, minossint1 INT64, maxossint1 INT64, totossint1 INT64, minossint2 INT64, maxossint2 INT64, totossint2 INT64, minossint3 INT64, maxossint3 INT64, totossint3 INT64, minossint4 INT64, maxossint4 INT64, totossint4 INT64, totextdbs INT64, rectype INT64, uid INT64, gid INT64" extra_cols ");"
+    "CREATE TABLE " name "(" extra_cols "inode TEXT, pinode TEXT, totsubdirs INT64, maxsubdirfiles INT64, maxsubdirlinks INT64, maxsubdirsize INT64, totfiles INT64, totlinks INT64, minuid INT64, maxuid INT64, mingid INT64, maxgid INT64, minsize INT64, maxsize INT64, totzero INT64, totltk INT64, totmtk INT64, totltm INT64, totmtm INT64, totmtg INT64, totmtt INT64, totsize INT64, minctime INT64, maxctime INT64, minmtime INT64, maxmtime INT64, minatime INT64, maxatime INT64, minblocks INT64, maxblocks INT64, totxattr INT64, depth INT64, mincrtime INT64, maxcrtime INT64, minossint1 INT64, maxossint1 INT64, totossint1 INT64, minossint2 INT64, maxossint2 INT64, totossint2 INT64, minossint3 INT64, maxossint3 INT64, totossint3 INT64, minossint4 INT64, maxossint4 INT64, totossint4 INT64, totextdbs INT64, rectype INT64, uid INT64, gid INT64);"
 #define TREESUMMARY_CREATE                          \
     DROP_TABLE(TREESUMMARY)                         \
     TREESUMMARY_SCHEMA(TREESUMMARY, "")
@@ -268,6 +268,8 @@ enum CheckRollupScore {
 int bottomup_collect_treesummary(sqlite3 *db, const char *dirname, sll_t *subdirs,
                                  const enum CheckRollupScore check_rollupscore);
 
+int *get_col_types(sqlite3 *db, const refstr_t *sql, int *cols);
+
 #ifdef __cplusplus
 }
 #endif

include/gufi_query/validate_inputs.h → include/gufi_query/handle_sql.h

@@ -62,11 +62,11 @@ OF SUCH DAMAGE.
 
 
 
-#ifndef GUFI_QUERY_VALIDATE_INPUTS_H
-#define GUFI_QUERY_VALIDATE_INPUTS_H
+#ifndef GUFI_QUERY_HANDLE_SQL_H
+#define GUFI_QUERY_HANDLE_SQL_H
 
 #include "bf.h"
 
-int validate_inputs(struct input *in);
+int handle_sql(struct input *in);
 
 #endif

include/gufi_query/query.h

@@ -71,7 +71,7 @@ OF SUCH DAMAGE.
 
 void querydb(struct work *work,
              const char *dbname, const size_t dbname_len,
-             sqlite3 *db, const char *query,
+             sqlite3 *db, const char *query, int *types,
              PoolArgs_t *pa, int id,
              int (*callback)(void *, int, char **, char**), int *rc);
 

include/print.h

@@ -82,6 +82,7 @@ typedef struct PrintArgs {
     pthread_mutex_t *mutex;               /* mutex for printing to stdout */
     FILE *outfile;
     size_t rows;                          /* number of rows returned by the query */
+    int *types;                           /* if set, prefix output with 1 octet type and 4 digit human readable length */
     /* size_t printed;                    /\* number of records printed by the callback *\/ */
 } PrintArgs_t;
 

src/CMakeLists.txt

@@ -200,10 +200,10 @@ add_library(gufi_query_lib OBJECT
   gufi_query/aggregate.c
   gufi_query/external.c
   gufi_query/gqw.c
+  gufi_query/handle_sql.c
   gufi_query/process_queries.c
   gufi_query/processdir.c
   gufi_query/query.c
-  gufi_query/validate_inputs.c
 )
 
 add_dependencies(gufi_query_lib GUFI)

src/bf.c

@@ -124,6 +124,10 @@ struct input *input_init(struct input *in) {
 
 void input_fini(struct input *in) {
     if (in) {
+        free(in->types.agg);
+        free(in->types.ent);
+        free(in->types.sum);
+        free(in->types.tsum);
         sll_destroy(&in->external_attach, free);
         trie_free(in->skip);
     }
@@ -156,6 +160,7 @@ void print_help(const char* prog_name,
       case 'd': printf("  -d <delim>             delimiter (one char)  [use 'x' for 0x%02X]", (uint8_t)fielddelim); break;
       case 'o': printf("  -o <out_fname>         output file (one-per-thread, with thread-id suffix)"); break;
       case 'O': printf("  -O <out_DB>            output DB"); break;
+      case 'u': printf("  -u                     prefix output with 1 octet type and 4 digit human readable length"); break; /* need to use text to avoid \x0a confusion */
       case 'I': printf("  -I <SQL_init>          SQL init"); break;
       case 'T': printf("  -T <SQL_tsum>          SQL for tree-summary table"); break;
       case 'S': printf("  -S <SQL_sum>           SQL for summary table"); break;
@@ -208,6 +213,7 @@ void show_input(struct input* in, int retval) {
    printf("in.maxthreads               = %zu\n",           in->maxthreads);
    printf("in.delim                    = '%c'\n",          in->delim);
    printf("in.andor                    = %d\n",            (int) in->andor);
+   printf("in.types.prefix             = %d\n",            in->types.prefix);
    printf("in.process_xattrs           = %d\n",            in->process_xattrs);
    printf("in.nobody.uid               = %" STAT_uid "\n", in->nobody.uid);
    printf("in.nobody.gid               = %" STAT_gid "\n", in->nobody.gid);
@@ -353,6 +359,10 @@ int parse_cmd_line(int         argc,
          INSTALL_STR(&in->outname, optarg);
          break;
 
+      case 'u':
+         in->types.prefix = 1;
+         break;
+
       case 'I':               // SQL initializations
          INSTALL_STR(&in->sql.init, optarg);
          break;

src/dbutils.c

@@ -76,6 +76,7 @@ OF SUCH DAMAGE.
 #include "dbutils.h"
 #include "external.h"
 #include "histogram.h"
+#include "trie.h"
 
 static const char SQLITE_MEMORY_ARRAY[] = ":memory:";
 const char *SQLITE_MEMORY = SQLITE_MEMORY_ARRAY;
@@ -1190,3 +1191,88 @@ int bottomup_collect_treesummary(sqlite3 *db, const char *dirname, sll_t *subdir
 
     return inserttreesumdb(dirname, db, &tsum, 0, 0, 0);
 }
+
+/*
+ * subset of known string to SQLite type conversions
+ *
+ * https://www.sqlite.org/datatype3.html
+ * https://www.sqlite.org/c3ref/c_blob.html
+ */
+static trie_t *sqlite3_types(void) {
+    trie_t *types = trie_alloc();
+
+    trie_insert(types, "INT",     3, (void *) (uintptr_t) SQLITE_INTEGER, NULL);
+    trie_insert(types, "INTEGER", 7, (void *) (uintptr_t) SQLITE_INTEGER, NULL);
+    trie_insert(types, "INT64",   5, (void *) (uintptr_t) SQLITE_INTEGER, NULL);
+
+    trie_insert(types, "FLOAT",   5, (void *) (uintptr_t) SQLITE_FLOAT,   NULL);
+    trie_insert(types, "DOUBLE",  6, (void *) (uintptr_t) SQLITE_FLOAT,   NULL);
+    trie_insert(types, "REAL",    4, (void *) (uintptr_t) SQLITE_FLOAT,   NULL);
+
+    trie_insert(types, "TEXT",    4, (void *) (uintptr_t) SQLITE_TEXT,    NULL);
+
+    trie_insert(types, "BLOB",    4, (void *) (uintptr_t) SQLITE_BLOB,    NULL);
+
+    trie_insert(types, "NULL",    4, (void *) (uintptr_t) SQLITE_NULL,    NULL);
+
+    return types;
+}
+
+int *get_col_types(sqlite3 *db, const refstr_t *sql, int *cols) {
+    int rc = SQLITE_OK;
+
+    /* parse sql */
+    sqlite3_stmt *stmt = NULL;
+    rc = sqlite3_prepare_v2(db, sql->data, sql->len, &stmt, NULL);
+    if (rc != SQLITE_OK) {
+        fprintf(stderr, "Error: Could not prepare '%s' for getting column types: %s (%d)\n",
+                sql->data, sqlite3_errstr(rc), rc);
+        return NULL;
+    }
+
+    /* /\* */
+    /*  * need to step if calling sqlite3_column_type, but requires */
+    /*  * that the table has at least 1 row of actual values */
+    /*  *\/ */
+    /* rc = sqlite3_step(stmt); */
+    /* if (rc != SQLITE_ROW) { */
+    /*     fprintf(stderr, "Error: Failed to evaluate SQL statement '%s': %s (%d)\n", */
+    /*             sql->data, sqlite3_errstr(rc), rc); */
+    /*     return NULL; */
+    /* } */
+
+    /* get column count */
+    *cols = sqlite3_column_count(stmt);
+    if (*cols == 0) {
+        fprintf(stderr, "Error: '%s' was detected to have 0 columns\n", sql->data);
+        sqlite3_finalize(stmt);
+        return NULL;
+    }
+
+    trie_t *str2type = sqlite3_types();
+
+    /* get each column's type */
+    int *types = malloc(*cols * sizeof(int));
+    for(int i = 0; i < *cols; i++) {
+        const char *type = sqlite3_column_decltype(stmt, i);
+        if (!type) {
+            types[i] = SQLITE_NULL;
+            continue;
+        }
+
+        const size_t type_len = strlen(type);
+
+        void *sql_type = NULL;
+        if (trie_search(str2type, type, type_len, &sql_type) == 1) {
+            types[i] = (uintptr_t) sql_type;
+        }
+        else {
+            types[i] = 0; /* unknown type */
+        }
+    }
+
+    trie_free(str2type);
+
+    sqlite3_finalize(stmt);
+    return types;
+}

src/gufi_query/aggregate.c

@@ -157,12 +157,14 @@ int aggregate_process(Aggregate_t *aggregate, struct input *in) {
 
     /* normally expect STDOUT/OUTFILE to have SQL to run, but OUTDB can have SQL to run as well */
     if ((in->output != OUTDB) || in->sql.agg.len) {
-        PrintArgs_t pa;
-        pa.output_buffer = &aggregate->ob;
-        pa.delim = in->delim;
-        pa.mutex = NULL;
-        pa.outfile = aggregate->outfile;
-        pa.rows = 0;
+        PrintArgs_t pa = {
+            .output_buffer = &aggregate->ob,
+            .delim = in->delim,
+            .mutex = NULL,
+            .outfile = aggregate->outfile,
+            .rows = 0,
+            .types = in->types.agg,
+        };
 
         char *err = NULL;
         if (sqlite3_exec(aggregate->db, in->sql.agg.data, print_parallel, &pa, &err) != SQLITE_OK) {

src/gufi_query/validate_inputs.c → src/gufi_query/handle_sql.c

@@ -63,10 +63,13 @@ OF SUCH DAMAGE.
 
 
 #include <stdio.h>
+#include <stdlib.h>
 
-#include "gufi_query/validate_inputs.h"
+#include "dbutils.h"
+#include "gufi_query/handle_sql.h"
+#include "template_db.h"
 
-int validate_inputs(struct input *in) {
+int handle_sql(struct input *in) {
     /*
      * - Leaves are final outputs
      * - OUTFILE/OUTDB + aggregation will create per thread and final aggregation files
@@ -140,5 +143,43 @@ int validate_inputs(struct input *in) {
         }
     }
 
+    /* now that the SQL has been validated, generate types if necessary */
+    if ((in->types.prefix == 1) && ((in->output == STDOUT) || (in->output == OUTFILE))) {
+        /* have to create temporary db since there is no guarantee of a db yet */
+        sqlite3 *db = opendb(SQLITE_MEMORY, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE,
+                             0, 0, create_dbdb_tables, NULL);
+        if (!db) {
+            return -1;
+        }
+
+        int cols = 0; /* discarded */
+
+        /* if not aggregating, get types for T, S, and E */
+        if (!in->sql.init_agg.len) {
+            if (in->sql.tsum.len) {
+                in->types.tsum = get_col_types(db, &in->sql.tsum, &cols);
+            }
+            if (in->sql.sum.len) {
+                in->types.sum  = get_col_types(db, &in->sql.sum,  &cols);
+            }
+            if (in->sql.ent.len) {
+                in->types.ent  = get_col_types(db, &in->sql.ent,  &cols);
+            }
+        }
+        /* types for G */
+        else {
+            char *err = NULL;
+            if (sqlite3_exec(db, in->sql.init_agg.data, NULL, NULL, &err) != SQLITE_OK) {
+                fprintf(stderr, "Error: Init failed while getting column types: %s\n", err);
+                sqlite3_free(err);
+                closedb(db);
+                return -1;
+            }
+            in->types.agg = get_col_types(db, &in->sql.agg, &cols);
+        }
+
+        closedb(db);
+    }
+
     return 0;
 }

src/gufi_query/main.c

@@ -78,8 +78,8 @@ OF SUCH DAMAGE.
 
 #include "gufi_query/aggregate.h"
 #include "gufi_query/gqw.h"
+#include "gufi_query/handle_sql.h"
 #include "gufi_query/processdir.h"
-#include "gufi_query/validate_inputs.h"
 
 static void sub_help(void) {
    printf("GUFI_index        find GUFI index here\n");
@@ -93,9 +93,9 @@ int main(int argc, char *argv[])
     /* Callers provide the options-string for get_opt(), which will */
     /* control which options are parsed for each program. */
     struct input in;
-    process_args_and_maybe_exit("hHvT:S:E:an:jo:d:O:I:F:y:z:J:K:G:mB:wxk:M:s:" COMPRESS_OPT "Q:", 1, "GUFI_index ...", &in);
+    process_args_and_maybe_exit("hHvT:S:E:an:jo:d:O:uI:F:y:z:J:K:G:mB:wxk:M:s:" COMPRESS_OPT "Q:", 1, "GUFI_index ...", &in);
 
-    if (validate_inputs(&in) != 0) {
+    if (handle_sql(&in) != 0) {
         input_fini(&in);
         return EXIT_FAILURE;
     }

src/gufi_query/process_queries.c

@@ -255,7 +255,7 @@ int process_queries(PoolArgs_t *pa,
         if (in->sql.sum.len) {
             recs=1; /* set this to one record - if the sql succeeds it will set to 0 or 1 */
             /* put in the path relative to the user's input */
-            querydb(&gqw->work, dbname, dbname_len, db, in->sql.sum.data, pa, id, print_parallel, &recs);
+            querydb(&gqw->work, dbname, dbname_len, db, in->sql.sum.data, in->types.sum, pa, id, print_parallel, &recs);
         } else {
             recs = 1;
         }
@@ -265,7 +265,7 @@ int process_queries(PoolArgs_t *pa,
         /* if we have recs (or are running an OR) query the entries table */
         if (recs > 0) {
             if (in->sql.ent.len) {
-                querydb(&gqw->work, dbname, dbname_len, db, in->sql.ent.data, pa, id, print_parallel, &recs); /* recs is not used */
+                querydb(&gqw->work, dbname, dbname_len, db, in->sql.ent.data, in->types.ent, pa, id, print_parallel, &recs); /* recs is not used */
             }
         }
     }

src/gufi_query/processdir.c

@@ -214,14 +214,14 @@ int processdir(QPTPool_t *ctx, const size_t id, void *data, void *args) {
             if (in->andor == AND) {
                 /* make sure the treesummary table exists */
                 querydb(&gqw->work, dbname, dbname_len, db, "SELECT name FROM " ATTACH_NAME ".sqlite_master "
-                        "WHERE (type == 'table') AND (name == '" TREESUMMARY "');",
+                        "WHERE (type == 'table') AND (name == '" TREESUMMARY "');", NULL,
                         pa, id, count_rows, &recs);
                 if (recs < 1) {
                     recs = -1;
                 }
                 else {
                     /* run in->sql.tsum */
-                    querydb(&gqw->work, dbname, dbname_len, db, in->sql.tsum.data, pa, id, print_parallel, &recs);
+                    querydb(&gqw->work, dbname, dbname_len, db, in->sql.tsum.data, in->types.tsum, pa, id, print_parallel, &recs);
                 }
             }
             /* this is an OR or we got a record back. go on to summary/entries */