[PATCH v4 1/1] urlmon: FindMimeFromData return only the mime type

[Alistair Leslie-Hughes]

From: Alistair Leslie-Hughes

---
 dlls/urlmon/mimefilter.c | 10 ++++++----
 dlls/urlmon/tests/misc.c |  5 +++++
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/dlls/urlmon/mimefilter.c b/dlls/urlmon/mimefilter.c
index 793de18a853..9577d046275 100644
--- a/dlls/urlmon/mimefilter.c
+++ b/dlls/urlmon/mimefilter.c
@@ -544,12 +544,13 @@ static HRESULT find_mime_from_buffer(const BYTE *buf, DWORD size, const WCHAR *p
         if(!proposed_mime)
             return E_FAIL;

-        len = lstrlenW(proposed_mime)+1;
-        *ret_mime = CoTaskMemAlloc(len*sizeof(WCHAR));
+        len = wcscspn(proposed_mime, L";");
+        *ret_mime = CoTaskMemAlloc((len+1)*sizeof(WCHAR));
         if(!*ret_mime)
             return E_OUTOFMEMORY;

memcpy(*ret_mime, proposed_mime, len*sizeof(WCHAR));
+        (*ret_mime)[len] = 0;
         return S_OK;
     }

@@ -638,12 +639,13 @@ static HRESULT find_mime_from_buffer(const BYTE *buf, DWORD size, const WCHAR *p

TRACE("found %s for %s\n", debugstr_w(ret), debugstr_an((const char*)buf, min(32, size)));

-    len = lstrlenW(ret)+1;
-    *ret_mime = CoTaskMemAlloc(len*sizeof(WCHAR));
+    len = wcscspn(ret, L";");
+    *ret_mime = CoTaskMemAlloc((len+1)*sizeof(WCHAR));
     if(!*ret_mime)
         return E_OUTOFMEMORY;

memcpy(*ret_mime, ret, len*sizeof(WCHAR));
+    (*ret_mime)[len] = 0;
     return S_OK;
 }

diff --git a/dlls/urlmon/tests/misc.c b/dlls/urlmon/tests/misc.c
index 0926eb75224..2a61b892930 100644
--- a/dlls/urlmon/tests/misc.c
+++ b/dlls/urlmon/tests/misc.c
@@ -794,6 +794,11 @@ static void test_FindMimeFromData(void)
     ok(!lstrcmpW(mime, L"text/plain"), "wrong mime: %s\n", wine_dbgstr_w(mime));
     CoTaskMemFree(mime);

+    hres = pFindMimeFromData(NULL, L"winexml://content/brew=1", data1, 0, L"text/html; charset=unicode", 0, &mime, 0);
+    ok(hres == S_OK, "FindMimeFromData failed: %08lx\n", hres);
+    ok(!lstrcmpW(mime, L"text/html"), "wrong mime: %s\n", wine_dbgstr_w(mime));
+    CoTaskMemFree(mime);
+
     hres = pFindMimeFromData(NULL, NULL, data1, 0, L"text/plain", 0, NULL, 0);
     ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08lx, expected E_INVALIDARG\n", hres);
 }