setupapi: Fix string conversion in SetupGetInfInformationA().

fe411406 · Pierre Schweitzer · Alexandre Julliard · 90aab41c · fe411406
Commit fe411406 authored Feb 08, 2015 by Pierre Schweitzer Committed by Alexandre Julliard Feb 09, 2015
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

query.c dlls/setupapi/query.c +6 -1

No files found.
--- a/dlls/setupapi/query.c
+++ b/dlls/setupapi/query.c
@@ -125,8 +125,13 @@ BOOL WINAPI SetupGetInfInformationA(LPCVOID InfSpec, DWORD SearchControl,
    if (InfSpec && SearchControl >= INFINFO_INF_NAME_IS_ABSOLUTE)
    {
-        len = lstrlenA(InfSpec) + 1;
+        len = MultiByteToWideChar(CP_ACP, 0, InfSpec, -1, NULL, 0);
        inf = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
+        if (!inf)
+        {
+            SetLastError(ERROR_NOT_ENOUGH_MEMORY);
+            return FALSE;
+        }
        MultiByteToWideChar(CP_ACP, 0, InfSpec, -1, inf, len);
    }