Add more extractor implementation

1 files changed, 8 insertions, 3 deletions

diff --git a/nfcollect.c b/nfcollect.c
index 5d7aaf1..10ad512 100644
--- a/nfcollect.c
+++ b/nfcollect.c
@@ -58,19 +58,20 @@ int main(int argc, char *argv[]) {
     uint32_t trunk_cnt = 0, trunk_size = 0;
     uint32_t entries_max;
     nflog_global_t g;
-    int nfl_group_id;
-    char *storage_dir = NULL;
+    int nfl_group_id = -1;
+    char *compression_flag = NULL, *storage_dir = NULL;
 
     struct option longopts[] = {/* name, has_args, flag, val */
                                 {"nflog-group", required_argument, NULL, 'g'},
                                 {"storage_dir", required_argument, NULL, 'd'},
                                 {"storage_size", required_argument, NULL, 's'},
+                                {"compression", optional_argument, NULL, 'z'},
                                 {"help", no_argument, NULL, 'h'},
                                 {"version", no_argument, NULL, 'v'},
                                 {0, 0, 0, 0}};
 
     int opt;
-    while ((opt = getopt_long(argc, argv, "g:d:s:hv", longopts, NULL)) != -1) {
+    while ((opt = getopt_long(argc, argv, "c:g:d:s:hv", longopts, NULL)) != -1) {
         switch (opt) {
         case 'h':
             printf("%s", help_text);
@@ -80,6 +81,9 @@ int main(int argc, char *argv[]) {
             printf("%s %s", PACKAGE, VERSION);
             exit(0);
             break;
+        case 'c':
+            compression_flag = optarg;
+            break;
         case 'd':
             storage_dir = optarg;
             break;
@@ -119,6 +123,7 @@ int main(int argc, char *argv[]) {
 
     nfl_cal_trunk(storage_size, &trunk_cnt, &trunk_size);
     nfl_cal_entries(trunk_size, &entries_max);
+    nfl_setup_compression(compression_flag, &g.compression_opt);
 
     // Set up commit worker
     g.nfl_commit_queue = malloc(sizeof(sem_t));