Add support for 2 digit version number input

exploitdb/ctfsetup/qemu_setup.py

@@ -95,13 +95,22 @@ def get_kernel_release_number(kernel_version):
     """
     Return a version number string that is a valid kernel release (hopefully)
 
-    Replace all "x" in the number with "1".
-    Replace ".0" ending with ".1"
+    Example:
+        4.5.x   -> 4.5.1
+        3.6     -> 3.6.1
+        5.0     -> 5.0.1
+        5.10.0  -> 5.10.1
 
     This could fail in the rare case where there is a new kernel, without version 1
     FIXME: Use qemu.utils.get_kernel_version_table() to get a 100% valid version number
     """
-    result = re.sub('x', '1', kernel_version)
-    if result[-2:] == ".0":
-        result = result[:-2] + ".1"
+    result = re.sub('[xyz]', '1', kernel_version)
+    result_splitted = result.split('.')
+
+    # Make sure the length is 3 or 4
+    while len(result_splitted) < 3:
+        result_splitted.append('1')
+    if result_splitted[-1] == '0':
+        result_splitted[-1] = '1'
+    result = '.'.join(result_splitted)
     return result