[virt-tools-list] [libosinfo 2/2] Add osinfo_db_guess_os_from_location()

[Zeeshan Ali (Khattak)]

From: "Zeeshan Ali (Khattak)" <zeeshanak at gnome.org>

Add API to guess OS given an installation media location.
---
 docs/reference/Libosinfo-sections.txt |    1 +
 osinfo/libosinfo.syms                 |    1 +
 osinfo/osinfo_db.c                    |   71 ++++++++++++++++++++++++++++++++-
 osinfo/osinfo_db.h                    |    2 +
 4 files changed, 74 insertions(+), 1 deletions(-)

diff --git a/docs/reference/Libosinfo-sections.txt b/docs/reference/Libosinfo-sections.txt
index 0aee98f..80816e6 100644
--- a/docs/reference/Libosinfo-sections.txt
+++ b/docs/reference/Libosinfo-sections.txt
@@ -18,6 +18,7 @@ osinfo_db_add_os
 osinfo_db_add_platform
 osinfo_db_add_device
 osinfo_db_add_deployment
+osinfo_db_guess_os_from_location
 osinfo_db_unique_values_for_property_in_os
 osinfo_db_unique_values_for_property_in_platform
 osinfo_db_unique_values_for_property_in_device
diff --git a/osinfo/libosinfo.syms b/osinfo/libosinfo.syms
index aadcbc3..b318c9a 100644
--- a/osinfo/libosinfo.syms
+++ b/osinfo/libosinfo.syms
@@ -15,6 +15,7 @@ LIBOSINFO_0.0.1 {
 	osinfo_db_add_platform;
 	osinfo_db_add_device;
 	osinfo_db_add_deployment;
+        osinfo_db_guess_os_from_location;
 	osinfo_db_unique_values_for_property_in_os;
 	osinfo_db_unique_values_for_property_in_platform;
 	osinfo_db_unique_values_for_property_in_device;
diff --git a/osinfo/osinfo_db.c b/osinfo/osinfo_db.c
index 4c37365..2eee239 100644
--- a/osinfo/osinfo_db.c
+++ b/osinfo/osinfo_db.c
@@ -23,11 +23,17 @@
  */
 
 #include <osinfo/osinfo.h>
+#include <cdio/iso9660.h>
+#include <string.h>
 
 G_DEFINE_TYPE (OsinfoDb, osinfo_db, G_TYPE_OBJECT);
 
 #define OSINFO_DB_GET_PRIVATE(obj) (G_TYPE_INSTANCE_GET_PRIVATE ((obj), OSINFO_TYPE_DB, OsinfoDbPrivate))
 
+#define str_contains(str, substr) ((str) && \
+                                   (substr) && \
+                                   strstr((str), (substr)) != NULL)
+
 /**
  * SECTION:osinfo_db
  * @short_description: Database of all entities
@@ -144,7 +150,6 @@ OsinfoOs *osinfo_db_get_os(OsinfoDb *db, const gchar *id)
     return OSINFO_OS(osinfo_list_find_by_id(OSINFO_LIST(db->priv->oses), id));
 }
 
-
 /**
  * osinfo_db_get_deployment:
  * @db: the database
@@ -316,6 +321,70 @@ void osinfo_db_add_deployment(OsinfoDb *db, OsinfoDeployment *deployment)
     osinfo_list_add(OSINFO_LIST(db->priv->deployments), OSINFO_ENTITY(deployment));
 }
 
+/**
+ * osinfo_db_guess_os_from_location:
+ * @db: the database
+ * @location: the location of an installation media
+ *
+ * Returns: (transfer none): the operating system, or NULL if guessing failed
+ */
+OsinfoOs *osinfo_db_guess_os_from_location(OsinfoDb *db,
+                                           const gchar *location)
+{
+    OsinfoOs *ret = NULL;
+    gchar *volume = NULL;
+    gchar *system = NULL;
+    gchar *publisher = NULL;
+    iso9660_t *iso;
+    GList *os_iter;
+
+    g_return_val_if_fail(OSINFO_IS_DB(db), NULL);
+    g_return_val_if_fail(location != NULL, NULL);
+
+    iso = iso9660_open(location);
+    if (!iso)
+        return NULL;
+
+    iso9660_ifs_get_volume_id(iso, &volume);
+    iso9660_ifs_get_system_id(iso, &system);
+    iso9660_ifs_get_publisher_id(iso, &publisher);
+
+    if (!volume && (!system && !publisher))
+        return NULL;
+
+    GList *oss = osinfo_list_get_elements(OSINFO_LIST(db->priv->oses));
+    for (os_iter = oss; os_iter; os_iter = os_iter->next) {
+        OsinfoOs *os = OSINFO_OS(os_iter->data);
+        OsinfoMediaList *media_list = osinfo_os_get_media_list(os);
+        GList *medias = osinfo_list_get_elements(OSINFO_LIST(media_list));
+        GList *media_iter;
+
+        for (media_iter = medias; media_iter; media_iter = media_iter->next) {
+            OsinfoMedia *media = OSINFO_MEDIA(media_iter->data);
+            const gchar *media_volume = osinfo_media_get_volume_id(media);
+            const gchar *media_system = osinfo_media_get_system_id(media);
+            const gchar *media_publisher = osinfo_media_get_publisher_id(media);
+
+            if (str_contains(volume, media_volume) &&
+                (str_contains(system, media_system) ||
+                 str_contains(publisher, media_publisher))) {
+                ret = os;
+                break;
+            }
+        }
+
+        g_list_free(medias);
+        g_object_unref(media_list);
+
+        if (ret)
+            break;
+    }
+
+    g_list_free(oss);
+    iso9660_close(iso);
+
+    return ret;
+}
 
 struct osinfo_db_populate_values_args {
     GHashTable *values;
diff --git a/osinfo/osinfo_db.h b/osinfo/osinfo_db.h
index c9d506e..ca2a83e 100644
--- a/osinfo/osinfo_db.h
+++ b/osinfo/osinfo_db.h
@@ -92,6 +92,8 @@ void osinfo_db_add_platform(OsinfoDb *db, OsinfoPlatform *platform);
 void osinfo_db_add_device(OsinfoDb *db, OsinfoDevice *device);
 void osinfo_db_add_deployment(OsinfoDb *db, OsinfoDeployment *deployment);
 
+OsinfoOs *osinfo_db_guess_os_from_location(OsinfoDb *db, const gchar *location);
+
 // Get me all unique values for property "vendor" among operating systems
 GList *osinfo_db_unique_values_for_property_in_os(OsinfoDb *db, const gchar *propName);
 
-- 
1.7.6