setupapi: Allocate the proper number of bytes for the drive spec.

Spotted with Valgrind.

setupapi: Allocate the proper number of bytes for the drive spec.
a91fd8fb · Andrew Nguyen · Alexandre Julliard · dd1b9372 · a91fd8fb
Commit a91fd8fb authored Jan 15, 2011 by Andrew Nguyen Committed by Alexandre Julliard Jan 17, 2011
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

diskspace.c dlls/setupapi/diskspace.c +1 -1

No files found.
--- a/dlls/setupapi/diskspace.c
+++ b/dlls/setupapi/diskspace.c
@@ -177,7 +177,7 @@ BOOL WINAPI SetupQuerySpaceRequiredOnDriveW(HDSKSPC DiskSpace,
        return FALSE;
    }

-    driveW = HeapAlloc(GetProcessHeap(), 0, lstrlenW(DriveSpec) + 2);
+    driveW = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(DriveSpec) + 2) * sizeof(WCHAR));
    if (!driveW)
    {
        SetLastError(ERROR_NOT_ENOUGH_MEMORY);