From dc57f3f089680adc766debeb7c2b1f65b0380d9d Mon Sep 17 00:00:00 2001
From: Alessandro Pilotti <ap@pilotti.it>
Date: Sat, 16 Nov 2013 16:44:12 +0200
Subject: [PATCH] Implements Windows OS version check with Win32 API

The goal is to support execution during serialize
after Sysprep where WMI is not available
---
 cloudbaseinit/init.py                         |  4 +-
 .../metadata/services/httpservice.py          |  3 +-
 cloudbaseinit/osutils/base.py                 |  2 +-
 cloudbaseinit/osutils/windows.py              | 61 ++++++++++++++++++-
 4 files changed, 62 insertions(+), 8 deletions(-)

diff --git a/cloudbaseinit/init.py b/cloudbaseinit/init.py
index d6b527c9..43c7627e 100644
--- a/cloudbaseinit/init.py
+++ b/cloudbaseinit/init.py
@@ -79,8 +79,8 @@ class InitManager(object):
             if not min_os_version:
                 supported = True
             else:
-                os_version = map(int, osutils.get_os_version().split('.'))
-                if os_version >= list(min_os_version):
+                os_major, os_minor = min_os_version
+                if osutils.check_os_version(os_major, os_minor):
                     supported = True
                 else:
                     LOG.debug('Skipping plugin: \'%s\'. OS version not '
diff --git a/cloudbaseinit/metadata/services/httpservice.py b/cloudbaseinit/metadata/services/httpservice.py
index 3cef1786..de7781ca 100644
--- a/cloudbaseinit/metadata/services/httpservice.py
+++ b/cloudbaseinit/metadata/services/httpservice.py
@@ -45,8 +45,7 @@ class HttpService(base.BaseMetadataService):
         '''
         osutils = osutils_factory.OSUtilsFactory().get_os_utils()
 
-        os_major_version = int(osutils.get_os_version().split('.')[0])
-        if os_major_version >= 6:
+        if osutils.check_os_version(6, 0):
             # 169.254.x.x addresses are not getting routed starting from
             # Windows Vista / 2008
             metadata_netloc = urlparse.urlparse(CONF.metadata_base_url).netloc
diff --git a/cloudbaseinit/osutils/base.py b/cloudbaseinit/osutils/base.py
index 3a38bf3b..7c90e022 100644
--- a/cloudbaseinit/osutils/base.py
+++ b/cloudbaseinit/osutils/base.py
@@ -88,7 +88,7 @@ class BaseOSUtils(object):
                          metric):
         pass
 
-    def get_os_version(self):
+    def check_os_version(self, major, minor, build=0):
         pass
 
     def get_volume_label(self, drive):
diff --git a/cloudbaseinit/osutils/windows.py b/cloudbaseinit/osutils/windows.py
index 072da28d..e2d194e4 100644
--- a/cloudbaseinit/osutils/windows.py
+++ b/cloudbaseinit/osutils/windows.py
@@ -83,6 +83,32 @@ class Win32_MIB_IPFORWARDTABLE(ctypes.Structure):
     ]
 
 
+class Win32_OSVERSIONINFOEX_W(ctypes.Structure):
+    _fields_ = [
+        ('dwOSVersionInfoSize', wintypes.DWORD),
+        ('dwMajorVersion', wintypes.DWORD),
+        ('dwMinorVersion', wintypes.DWORD),
+        ('dwBuildNumber', wintypes.DWORD),
+        ('dwPlatformId', wintypes.DWORD),
+        ('szCSDVersion', wintypes.WCHAR * 128),
+        ('wServicePackMajor', wintypes.DWORD),
+        ('wServicePackMinor', wintypes.DWORD),
+        ('wSuiteMask', wintypes.DWORD),
+        ('wProductType', wintypes.BYTE),
+        ('wReserved', wintypes.BYTE)
+    ]
+
+
+kernel32.VerifyVersionInfoW.argtypes = [
+    ctypes.POINTER(Win32_OSVERSIONINFOEX_W),
+    wintypes.DWORD, wintypes.ULARGE_INTEGER]
+kernel32.VerifyVersionInfoW.restype = wintypes.BOOL
+
+kernel32.VerSetConditionMask.argtypes = [wintypes.ULARGE_INTEGER,
+                                         wintypes.DWORD,
+                                         wintypes.BYTE]
+kernel32.VerSetConditionMask.restype = wintypes.ULARGE_INTEGER
+
 kernel32.SetComputerNameExW.argtypes = [ctypes.c_int, wintypes.LPCWSTR]
 kernel32.SetComputerNameExW.restype = wintypes.BOOL
 
@@ -112,6 +138,12 @@ iphlpapi.GetIpForwardTable.restype = wintypes.DWORD
 
 Ws2_32.inet_ntoa.restype = ctypes.c_char_p
 
+VER_MAJORVERSION = 1
+VER_MINORVERSION = 2
+VER_BUILDNUMBER = 4
+
+VER_GREATER_EQUAL = 3
+
 
 class WindowsUtils(base.BaseOSUtils):
     NERR_GroupNotFound = 2220
@@ -121,6 +153,7 @@ class WindowsUtils(base.BaseOSUtils):
     ERROR_NO_SUCH_MEMBER = 1387
     ERROR_MEMBER_IN_ALIAS = 1378
     ERROR_INVALID_MEMBER = 1388
+    ERROR_OLD_WIN_VERSION = 1150
 
     DRIVE_CDROM = 5
 
@@ -467,9 +500,31 @@ class WindowsUtils(base.BaseOSUtils):
         if err:
             raise Exception('Unable to add route: %(err)s' % locals())
 
-    def get_os_version(self):
-        conn = wmi.WMI(moniker='//./root/cimv2')
-        return conn.Win32_OperatingSystem()[0].Version
+    def check_os_version(self, major, minor, build=0):
+        vi = Win32_OSVERSIONINFOEX_W()
+        vi.dwOSVersionInfoSize = ctypes.sizeof(Win32_OSVERSIONINFOEX_W)
+
+        vi.dwMajorVersion = major
+        vi.dwMinorVersion = minor
+        vi.dwBuildNumber = build
+
+        mask = 0
+        for type_mask in [VER_MAJORVERSION, VER_MINORVERSION, VER_BUILDNUMBER]:
+            mask = kernel32.VerSetConditionMask(mask, type_mask,
+                                                VER_GREATER_EQUAL)
+
+        type_mask = VER_MAJORVERSION | VER_MINORVERSION | VER_BUILDNUMBER
+        ret_val = kernel32.VerifyVersionInfoW(ctypes.byref(vi), type_mask,
+                                              mask)
+        if ret_val:
+            return True
+        else:
+            err = kernel32.GetLastError()
+            if err == self.ERROR_OLD_WIN_VERSION:
+                return False
+            else:
+                raise Exception("VerifyVersionInfo failed with error: %s" %
+                                err)
 
     def get_volume_label(self, drive):
         max_label_size = 261