From e4282a59dd8cdeba0004cc3adf81c68e2c82c01a Mon Sep 17 00:00:00 2001 From: Misha Koshelev Date: Mon, 9 Jul 2007 21:49:58 -0500 Subject: [PATCH] urlmon: Improve ObtainUserAgentString conformance to native. --- dlls/urlmon/urlmon_main.c | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/dlls/urlmon/urlmon_main.c b/dlls/urlmon/urlmon_main.c index 0f12011d777..bbd7cc59208 100644 --- a/dlls/urlmon/urlmon_main.c +++ b/dlls/urlmon/urlmon_main.c @@ -304,13 +304,18 @@ HRESULT WINAPI ObtainUserAgentString(DWORD dwOption, LPSTR pcszUAOut, DWORD *cbS { FIXME("(%d, %p, %p): stub\n", dwOption, pcszUAOut, cbSize); - if(dwOption) { - ERR("dwOption: %d, must be zero\n", dwOption); + if (pcszUAOut == NULL || cbSize == NULL) + return E_INVALIDARG; + + if (*cbSize < sizeof(Agent)) + { + *cbSize = sizeof(Agent); + return E_OUTOFMEMORY; } if (sizeof(Agent) < *cbSize) *cbSize = sizeof(Agent); - lstrcpynA(pcszUAOut, Agent, *cbSize); + lstrcpynA(pcszUAOut, Agent, *cbSize); return S_OK; }