sw-collector: Added --check option

diff --git a/src/sw-collector/sw-collector.8.in b/src/sw-collector/sw-collector.8.in
index b9041c77be77dec3be93369681f9100ac04618ae..5c8d2565664250c9863ff8616574fe6bc41160d3 100644 (file)
--- a/src/sw-collector/sw-collector.8.in
+++ b/src/sw-collector/sw-collector.8.in
@@ -34,6 +34,12 @@ sw-collector \- Extracts software installation events from dpkg history log
 .YS
 .
 .SY "sw-collector"
+.OP \-\-debug level
+.OP \-\-quiet
+.BR \-\-check
+.YS
+.
+.SY "sw-collector"
 .B \-h
 |
 .B \-\-help
@@ -76,7 +82,7 @@ installation status.
 .B "\-u, \-\-unregistered"
 Lists all software packages residing in the local collector database but for
 which no SWID tags exist yet in a central collector database reachable via a
-REST interface. 
+REST interface.
 .TP
 .B "\-g, \-\-generate"
 Generates ISO 19770-2:2015 SWID tags for all software packages residing in the
@@ -86,6 +92,10 @@ database reachable via a REST interface.
 .B "\-m, \-\-migrate"
 Can be used to migrate collector database versions. Currently all architecture
 suffixes are removed from dpkg package names.
+.TP
+.B "\-C, \-\-check"
+Checks the integrity of the collector database against the actual list of
+installed packages obtained with dpkg-query.
 .
 .SH "CONFIGURATION"
 .

diff --git a/src/sw-collector/sw-collector.c b/src/sw-collector/sw-collector.c
index a42f1068aed12b0677d69ffd545d6c71aceae99d..2b1783eff693c037cf931ae51f0e56ba648bb3f7 100644 (file)
--- a/src/sw-collector/sw-collector.c
+++ b/src/sw-collector/sw-collector.c
@@ -31,9 +31,10 @@
 #include <library.h>
 #include <utils/debug.h>
 #include <utils/lexparser.h>
+#include <collections/hashtable.h>
 
 #include <swid_gen/swid_gen.h>
-
+#include <swid_gen/swid_gen_info.h>
 /**
  * global debug output variables
  */
@@ -48,7 +49,8 @@ enum collector_op_t {
        COLLECTOR_OP_LIST,
        COLLECTOR_OP_UNREGISTERED,
        COLLECTOR_OP_GENERATE,
-       COLLECTOR_OP_MIGRATE
+       COLLECTOR_OP_MIGRATE,
+       COLLECTOR_OP_CHECK
 };
 
 /**
@@ -119,7 +121,8 @@ Usage:\n\
 --list|-unregistered\n\
   sw-collector [--debug <level>] [--quiet] [--installed|--removed] \
 [--full] --generate\n\
-  sw-collector [--debug <level>] [--quiet] --migrate\n");
+  sw-collector [--debug <level>] [--quiet] --migrate\n\
+  sw-collector [--debug <level>] [--quiet] --check\n");
 }
 
 /**
@@ -140,6 +143,7 @@ static collector_op_t do_args(int argc, char *argv[], bool *full_tags,
 
                struct option long_opts[] = {
                        { "help", no_argument, NULL, 'h' },
+                       { "check", no_argument, NULL, 'C' },
                        { "count", required_argument, NULL, 'c' },
                        { "debug", required_argument, NULL, 'd' },
                        { "full", no_argument, NULL, 'f' },
@@ -153,7 +157,7 @@ static collector_op_t do_args(int argc, char *argv[], bool *full_tags,
                        { 0,0,0,0 }
                };
 
-               c = getopt_long(argc, argv, "hc:d:fgilmqru", long_opts, NULL);
+               c = getopt_long(argc, argv, "hCc:d:fgilmqru", long_opts, NULL);
                switch (c)
                {
                        case EOF:
@@ -162,6 +166,9 @@ static collector_op_t do_args(int argc, char *argv[], bool *full_tags,
                                usage();
                                exit(SUCCESS);
                                break;
+                       case 'C':
+                               op = COLLECTOR_OP_CHECK;
+                               continue;
                        case 'c':
                                count = atoi(optarg);
                                continue;
@@ -537,7 +544,7 @@ end:
 }
 
 /**
- * Append missing architecture suffix to package entries in the database
+ * Remove architecture suffix from package entries in the database
  */
 static int migrate(sw_collector_db_t *db)
 {
@@ -582,6 +589,83 @@ static int migrate(sw_collector_db_t *db)
        return status;
 }
 
+/**
+ * Free hashtable entry
+ */
+static void free_entry(void *value, void *key)
+{
+       free(value);
+       free(key);
+}
+
+/**
+ * Check consistency of installed software identifiers in collector database
+ */
+static int check(sw_collector_db_t *db)
+{
+       sw_collector_dpkg_t *dpkg;
+       swid_gen_info_t *info;
+       hashtable_t *table;
+       enumerator_t *e;
+       char *dpkg_name, *name, *package, *arch, *version;
+       uint32_t sw_id, count = 0, installed;
+
+       dpkg = sw_collector_dpkg_create();
+       if (!dpkg)
+       {
+               return EXIT_FAILURE;
+       }
+       info = swid_gen_info_create();
+       table = hashtable_create(hashtable_hash_str, hashtable_equals_str, 4096);
+
+       /* Store all installed sw identifiers (according to dpkg) in hashtable */
+       e = dpkg->create_sw_enumerator(dpkg);
+       while (e->enumerate(e, &package, &arch, &version))
+       {
+               dpkg_name = info->create_sw_id(info, package, version);
+               table->put(table, strdup(package), dpkg_name);
+       }
+       e->destroy(e);
+
+       info->destroy(info);
+       dpkg->destroy(dpkg);
+
+       e = db->create_sw_enumerator(db, SW_QUERY_ALL, NULL);
+       if (!e)
+       {
+               return EXIT_FAILURE;
+       }
+       while (e->enumerate(e, &sw_id, &name, &package, &version, &installed))
+       {
+               dpkg_name = table->get(table, package);
+               if (installed)
+               {
+                       if (!dpkg_name)
+                       {
+                               printf("%4d %s erroneously noted as installed\n", sw_id, name);
+                       }
+                       else if (!streq(name, dpkg_name))
+                       {
+                               printf("%4d %s erroneously noted as installed instead of\n "
+                                          "    %s\n", sw_id, name, dpkg_name);
+                       }
+               }
+               else
+               {
+                       if (dpkg_name && streq(name, dpkg_name))
+                       {
+                               printf("%4d %s erroneously noted as removed\n", sw_id, name);
+                       }
+               }
+               count++;
+       }
+       e->destroy(e);
+
+       table->destroy_function(table, (void*)free_entry);
+       printf("checked %d software identifiers\n", count);
+
+       return EXIT_SUCCESS;
+}
 
 int main(int argc, char *argv[])
 {
@@ -646,6 +730,9 @@ int main(int argc, char *argv[])
                case COLLECTOR_OP_MIGRATE:
                        status = migrate(db);
                        break;
+               case COLLECTOR_OP_CHECK:
+                       status = check(db);
+                       break;
        }
        db->destroy(db);