Parse pci.ids

Co-authored-by: jackun <jack.un@gmail.com>

src/meson.build

@@ -34,6 +34,7 @@ vklayer_files = files(
   'notify.cpp',
   'elfhacks.cpp',
   'real_dlsym.cpp',
+  'pci_ids.cpp',
 )
 
 opengl_files = files(

src/overlay.cpp

@@ -59,6 +59,7 @@
 #include "notify.h"
 #include "blacklist.h"
 #include "version.h"
+#include "pci_ids.h"
 
 #ifdef HAVE_DBUS
 #include "dbus_info.h"
@@ -832,6 +833,7 @@ void init_gpu_stats(uint32_t& vendorID, overlay_params& params)
          params.enabled[OVERLAY_PARAM_ENABLED_gpu_stats] = false;
       }
    }
+   parse_pciids();
 }
 
 void init_system_info(){

src/pci_ids.cpp

@@ -0,0 +1,70 @@
+#include <string>
+#include <iostream>
+#include <fstream>
+#include <sstream>
+#include <iomanip>
+
+#include "pci_ids.h"
+
+std::map<uint32_t /*vendor id*/, std::pair<std::string /*vendor desc*/, std::map<uint32_t /*device id*/, device>>> pci_ids;
+
+std::istream& get_uncommented_line(std::istream& is, std::string &line)
+{
+    while (std::getline(is, line)) {
+        auto c = line.find("#");
+        if (c!=std::string::npos)
+            line.erase(c, std::string::npos);
+        if (line.size())
+            break;
+    }
+    return is;
+}
+
+void parse_pciids()
+{
+    std::ifstream file("/usr/share/hwdata/pci.ids");
+    std::string line;
+    size_t tabs = 0;
+
+    uint32_t ven_id = 0, dev_id = 0, subsys_ven_id = 0, subsys_dev_id = 0;
+    std::string desc;
+    std::stringstream ss;
+    while(get_uncommented_line(file, line))
+    {
+        tabs = line.find_first_not_of("\t");
+
+        ss.str(""); ss.clear();
+        ss << line;
+        
+        switch(tabs)
+        {
+            case 0:
+                ss >> std::hex >> ven_id;
+                if (ven_id == 0xFFFF)
+                    return;
+
+                std::getline(ss, desc);
+                pci_ids[ven_id].first = desc;
+            break;
+            case 1:
+            {
+                ss >> std::hex >> dev_id;
+                std::getline(ss, desc); //TODO trim whitespace
+                auto& dev = pci_ids[ven_id].second[dev_id];
+                dev.desc = desc;
+            }
+            break;
+            case 2:
+            {
+                ss >> std::hex >> subsys_ven_id;
+                ss >> subsys_dev_id;
+                std::getline(ss, desc);
+                auto& dev = pci_ids[ven_id].second[dev_id];
+                dev.subsys.push_back({subsys_ven_id, subsys_dev_id, desc});
+            }
+            break;
+            default:
+            break;
+        }
+    }
+}

src/pci_ids.h

@@ -0,0 +1,19 @@
+#include <map>
+#include <vector>
+
+struct subsys_device
+{
+    uint32_t vendor_id;
+    uint32_t device_id;
+    std::string desc;
+};
+
+struct device
+{
+    std::string desc;
+    std::vector<subsys_device> subsys;
+};
+
+extern std::map<uint32_t /*vendor id*/, std::pair<std::string /*vendor desc*/, std::map<uint32_t /*device id*/, device>>> pci_ids;
+
+void parse_pciids(void);