commit a2bf4d009b9c9f0ec288b4ea9768b4be06ada347
author Alexandre Julliard <julliard@winehq.org> Wed Mar 31 20:09:54 2004 +0000
committer Alexandre Julliard <julliard@winehq.org> Wed Mar 31 20:09:54 2004 +0000
tree 65a38f4ccead0c6a8334e00ac7050f5a3f0fc11b
parent 840d7de5d475fb59bbb29eb26aee354608a3cd16

Fixed length handling in HttpQueryInfoW.

dlls/wininet/http.c

diff --git a/dlls/wininet/http.c b/dlls/wininet/http.c
index 9eb8ed9..f914a49 100644
--- a/dlls/wininet/http.c
+++ b/dlls/wininet/http.c
@@ -953,9 +953,9 @@
            }
            size += delim;
 
-           if (size + 1 > *lpdwBufferLength)
+           if (size + 1 > *lpdwBufferLength/sizeof(WCHAR))
            {
-              *lpdwBufferLength = size + 1;
+              *lpdwBufferLength = (size + 1) * sizeof(WCHAR);
               INTERNET_SetLastError(ERROR_INSUFFICIENT_BUFFER);
               goto lend;
            }
@@ -988,9 +988,9 @@
 
             strcpyW((WCHAR*)lpBuffer + cnt, index == HTTP_QUERY_RAW_HEADERS_CRLF ? szcrlf : sznul);
 
-           *lpdwBufferLength = cnt + delim;
-           bSuccess = TRUE;
-	        goto lend;
+            *lpdwBufferLength = (cnt + delim) * sizeof(WCHAR);
+            bSuccess = TRUE;
+            goto lend;
         }
 	else if (index >= 0 && index <= HTTP_QUERY_MAX && lpwhr->StdHeaders[index].lpszValue)
 	{
@@ -1055,18 +1055,17 @@
     }
     else
     {
-        INT len = strlenW(lphttpHdr->lpszValue);
+        INT len = (strlenW(lphttpHdr->lpszValue) + 1) * sizeof(WCHAR);
 
-        if (len + 1 > *lpdwBufferLength)
+        if (len > *lpdwBufferLength)
         {
-            *lpdwBufferLength = len + 1;
+            *lpdwBufferLength = len;
             INTERNET_SetLastError(ERROR_INSUFFICIENT_BUFFER);
             goto lend;
         }
 
-        strncpyW(lpBuffer, lphttpHdr->lpszValue, len);
-        ((WCHAR*)lpBuffer)[len]=0;
-        *lpdwBufferLength = len;
+        memcpy(lpBuffer, lphttpHdr->lpszValue, len);
+        *lpdwBufferLength = len - sizeof(WCHAR);
         bSuccess = TRUE;
 
 	TRACE(" returning string : '%s'\n", debugstr_w(lpBuffer));