setupapi: Fix buffer overflow in SetupGetFileCompressionInfoW.

b8c42020 · Thomas Faber · Alexandre Julliard · c8269c2c · b8c42020
Commit b8c42020 authored Mar 27, 2014 by Thomas Faber Committed by Alexandre Julliard Mar 27, 2014
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

misc.c dlls/setupapi/misc.c +2 -1

No files found.
--- a/dlls/setupapi/misc.c
+++ b/dlls/setupapi/misc.c
@@ -1412,7 +1412,8 @@ DWORD WINAPI SetupGetFileCompressionInfoW( PCWSTR source, PWSTR *name, PDWORD so
        return ERROR_INVALID_PARAMETER;
    ret = SetupGetFileCompressionInfoExW( source, NULL, 0, &required, NULL, NULL, NULL );
-    if (!(actual_name = MyMalloc( required ))) return ERROR_NOT_ENOUGH_MEMORY;
+    if (!(actual_name = MyMalloc( required * sizeof(WCHAR) )))
+        return ERROR_NOT_ENOUGH_MEMORY;
    ret = SetupGetFileCompressionInfoExW( source, actual_name, required, &required,
                                          source_size, target_size, type );