allow gufi_dir2index to use an optional user-specified plugin

Author: bertschinger

src/gufi_dir2index.c

@@ -118,6 +118,12 @@ struct NonDirArgs {
 
     /* list of xattr dbs */
     sll_t xattr_db_list;
+
+    /*
+     * an optional opaque pointer to user data for a plugin. if the plugin's db_init() function
+     * creates and returns a pointer, then the later plugin functions can use it to store state.
+     */
+    void *plugin_user_data;
 };
 
 static int process_external(struct input *in, void *args,
@@ -156,19 +162,16 @@ static int process_nondir(struct work *entry, struct entry_data *ed, void *args)
         external_read_file(in, entry, process_external, nda->db);
     }
 
-    /* get entry relative path (use extra buffer to prevent memcpy overlap) */
-    char relpath[MAXPATH];
-    const size_t relpath_len = SNFORMAT_S(relpath, MAXPATH, 1,
-                                          entry->name + entry->root_parent.len, entry->name_len - entry->root_parent.len);
-    /* overwrite full path with relative path */
-    entry->name_len = SNFORMAT_S(entry->name, MAXPATH, 1, relpath, relpath_len);
-
     /* update summary table */
     sumit(&nda->summary, ed);
 
     /* add entry + xattr names into bulk insert */
     insertdbgo(entry, ed, nda->entries_res);
 
+    if (in->plugin_ops->process_file) {
+        in->plugin_ops->process_file(entry->name, nda->db, nda->plugin_user_data);
+    }
+
 out:
     return rc;
 }
@@ -188,6 +191,7 @@ static int processdir(QPTPool_t *ctx, const size_t id, void *data, void *args) {
     memset(&nda.ed, 0, sizeof(nda.ed));
     nda.topath     = NULL;
     nda.ed.type    = 'd';
+    nda.plugin_user_data = NULL;
 
     DIR *dir = NULL;
 
@@ -269,6 +273,14 @@ static int processdir(QPTPool_t *ctx, const size_t id, void *data, void *args) {
         }
 
         startdb(nda.db);
+
+        if (pa->in.plugin_ops->db_init) {
+            nda.plugin_user_data = pa->in.plugin_ops->db_init(nda.db);
+        }
+    }
+
+    if (pa->in.plugin_ops->process_dir) {
+        pa->in.plugin_ops->process_dir(nda.work->name, nda.db, nda.plugin_user_data);
     }
 
     struct descend_counters ctrs;
@@ -277,6 +289,10 @@ static int processdir(QPTPool_t *ctx, const size_t id, void *data, void *args) {
             &ctrs);
 
     if (process_dbdb) {
+        if (pa->in.plugin_ops->db_exit) {
+            pa->in.plugin_ops->db_exit(nda.db, nda.plugin_user_data);
+        }
+
         stopdb(nda.db);
 
         /* entries and xattrs have been inserted */
@@ -407,7 +423,7 @@ static void sub_help(void) {
 
 int main(int argc, char *argv[]) {
     struct PoolArgs pa;
-    process_args_and_maybe_exit("hHvn:xy:z:k:M:s:C:" COMPRESS_OPT "q", 2, "input_dir... output_dir", &pa.in);
+    process_args_and_maybe_exit("hHvn:xy:z:k:M:s:C:U:" COMPRESS_OPT "q", 2, "input_dir... output_dir", &pa.in);
 
     /* parse positional args, following the options */
     /* does not have to be canonicalized */