From 53a4ee450b3699e5a0430aabb608dba90342d604 Mon Sep 17 00:00:00 2001
From: Alessandro Pilotti <apilotti@cloudbasesolutions.com>
Date: Sun, 5 Aug 2018 15:33:53 +0300
Subject: [PATCH] Fix access violation when calling get_volumes

Change-Id: I4b4ae831a096af1c94ee713549407de940232201
Closes-Bug: #1785473
---
 cloudbaseinit/osutils/windows.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/cloudbaseinit/osutils/windows.py b/cloudbaseinit/osutils/windows.py
index e031531c..649a2bbe 100644
--- a/cloudbaseinit/osutils/windows.py
+++ b/cloudbaseinit/osutils/windows.py
@@ -259,6 +259,17 @@ kernel32.GetVolumePathNamesForVolumeNameW.argtypes = [wintypes.LPCWSTR,
                                                           wintypes.DWORD)]
 kernel32.GetVolumePathNamesForVolumeNameW.restype = wintypes.BOOL
 
+kernel32.FindFirstVolumeW.argtypes = [wintypes.LPWSTR, wintypes.DWORD]
+kernel32.FindFirstVolumeW.restype = wintypes.HANDLE
+
+kernel32.FindNextVolumeW.argtypes = [wintypes.HANDLE,
+                                     wintypes.LPWSTR,
+                                     wintypes.DWORD]
+kernel32.FindNextVolumeW.restype = wintypes.BOOL
+
+kernel32.FindVolumeClose.argtypes = [wintypes.HANDLE]
+kernel32.FindVolumeClose.restype = wintypes.BOOL
+
 iphlpapi.GetIpForwardTable.argtypes = [
     ctypes.POINTER(Win32_MIB_IPFORWARDTABLE),
     ctypes.POINTER(wintypes.ULONG),
@@ -1365,7 +1376,6 @@ class WindowsUtils(base.BaseOSUtils):
         if handle_volumes == self.INVALID_HANDLE_VALUE:
             raise exception.WindowsCloudbaseInitException(
                 "FindFirstVolumeW failed: %r")
-
         try:
             while True:
                 volumes.append(volume.value)