fix(kernel_crawler): enforce kernelversion to be a string.

Signed-off-by: Federico Di Pierro <nierro92@gmail.com>
falcosecurity · Aug 3, 2023 · bb69a6f · bb69a6f
1 parent b056143
commit bb69a6f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/kernel_crawler/flatcar.py b/kernel_crawler/flatcar.py
@@ -55,4 +55,4 @@ def list_repos(self):
         return repos
 
     def to_driverkit_config(self, release, deps):
-        return repo.DriverKitConfig(release, "flatcar", None, 1, list(deps)[0])
+        return repo.DriverKitConfig(release, "flatcar", None, "1", list(deps)[0])
diff --git a/kernel_crawler/oracle.py b/kernel_crawler/oracle.py
@@ -59,6 +59,6 @@ def to_driverkit_config(self, release, deps):
                     # example:
                     #  # uname -a
                     #  Linux vm-ol8 5.15.0-100.96.32.el8uek.x86_64 #2 SMP Tue ...
-                    return repo.DriverKitConfig(release, "ol", dep, kernelversion=2)
+                    return repo.DriverKitConfig(release, "ol", dep, kernelversion="2")
                 else:  # else return default with kernelversion=1
                     return repo.DriverKitConfig(release, "ol", dep)
diff --git a/kernel_crawler/repo.py b/kernel_crawler/repo.py
@@ -13,6 +13,8 @@ def __str__(self):
 
 class DriverKitConfig(object):
     def __init__(self, kernelrelease, target, headers=None, kernelversion="1", kernelconfigdata=None):
+        if not isinstance(kernelversion, str):
+            raise TypeError('kernelversion should be a string')
         self.kernelversion = kernelversion
         self.kernelrelease = kernelrelease
         self.target = target