Misha Koshelev : urlmon: Improve ObtainUserAgentString conformance to native.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: e4282a59dd8cdeba0004cc3adf81c68e2c82c01a
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=e4282a59dd8cdeba0004cc3adf81c68e2c82c01a

Author: Misha Koshelev <mk144210 at bcm.edu>
Date:   Mon Jul  9 21:49:58 2007 -0500

urlmon: Improve ObtainUserAgentString conformance to native.

---

 dlls/urlmon/urlmon_main.c |   11 ++++++++---
 1 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/dlls/urlmon/urlmon_main.c b/dlls/urlmon/urlmon_main.c
index 0f12011..bbd7cc5 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;
 }