add product and file entries to database

diff --git a/src/libimcv/plugins/imv_attestation/attest.c b/src/libimcv/plugins/imv_attestation/attest.c
index 531cd532ed7638f158014be9ee5dc8e15e02f3d1..3c5f6e3533bcf77821c3b999643dacc714f98e04 100644 (file)
--- a/src/libimcv/plugins/imv_attestation/attest.c
+++ b/src/libimcv/plugins/imv_attestation/attest.c
@@ -62,6 +62,8 @@ static void do_args(int argc, char *argv[])
                struct option long_opts[] = {
                        { "help", no_argument, NULL, 'h' },
                        { "files", no_argument, NULL, 'f' },
+                       { "add", no_argument, NULL, 'a' },
+                       { "del", no_argument, NULL, 'd' },
                        { "products", no_argument, NULL, 'p' },
                        { "hashes", no_argument, NULL, 'H' },
                        { "add", no_argument, NULL, 'a' },
@@ -110,10 +112,16 @@ static void do_args(int argc, char *argv[])
                                }
                                continue;
                        case 'H':
-                               operation = OP_HASHES;
+                               op = OP_HASHES;
+                               continue;
+                       case 'a':
+                               op = OP_ADD;
+                               continue;
+                       case 'd':
+                               op = OP_DEL;
                                continue;
                        case 'D':
-                               if (!attest->set_directory(attest, optarg))
+                               if (!attest->set_directory(attest, optarg, op == OP_ADD))
                                {
                                        exit(EXIT_FAILURE);
                                }