urlmon: Fixed CoInternetGetSecurityUrl implementation.

dlls/urlmon/sec_mgr.c

@@ -1240,19 +1240,32 @@ HRESULT WINAPI CoInternetCreateZoneManager(IServiceProvider* pSP, IInternetZoneM
  */
 HRESULT WINAPI CoInternetGetSecurityUrl(LPCWSTR pwzUrl, LPWSTR *ppwzSecUrl, PSUACTION psuAction, DWORD dwReserved)
 {
-    WCHAR url[INTERNET_MAX_URL_LENGTH], domain[INTERNET_MAX_URL_LENGTH];
+    WCHAR buf1[INTERNET_MAX_URL_LENGTH], buf2[INTERNET_MAX_URL_LENGTH];
+    LPWSTR url, domain;
     DWORD len;
     HRESULT hres;
 
     TRACE("(%p,%p,%u,%u)\n", pwzUrl, ppwzSecUrl, psuAction, dwReserved);
 
-    hres = CoInternetParseUrl(pwzUrl, PARSE_SECURITY_URL, 0, url, INTERNET_MAX_URL_LENGTH, &len, 0);
-    if(hres==S_OK) {
-        if(psuAction == PSU_DEFAULT)
-            hres = CoInternetParseUrl(url, PARSE_SECURITY_DOMAIN, 0, domain,
-                    INTERNET_MAX_URL_LENGTH, &len, 0);
+    url = buf1;
+    domain = buf2;
+    strcpyW(url, pwzUrl);
+
+    while(1) {
+        hres = CoInternetParseUrl(url, PARSE_SECURITY_URL, 0, domain, INTERNET_MAX_URL_LENGTH, &len, 0);
+        if(hres!=S_OK || !strcmpW(url, domain))
+            break;
+
+        if(url == buf1) {
+            url = buf2;
+            domain = buf1;
+        } else {
+            url = buf1;
+            domain = buf2;
+        }
+    }
 
-        if(psuAction==PSU_SECURITY_URL_ONLY || hres!=S_OK) {
+    if(psuAction==PSU_SECURITY_URL_ONLY) {
         len = lstrlenW(url)+1;
         *ppwzSecUrl = CoTaskMemAlloc(len*sizeof(WCHAR));
         if(!*ppwzSecUrl)
@@ -1262,6 +1275,9 @@ HRESULT WINAPI CoInternetGetSecurityUrl(LPCWSTR pwzUrl, LPWSTR *ppwzSecUrl, PSUA
         return S_OK;
     }
 
+    hres = CoInternetParseUrl(url, PARSE_SECURITY_DOMAIN, 0, domain,
+            INTERNET_MAX_URL_LENGTH, &len, 0);
+    if(SUCCEEDED(hres)) {
         len++;
         *ppwzSecUrl = CoTaskMemAlloc(len*sizeof(WCHAR));
         if(!*ppwzSecUrl)
@@ -1271,14 +1287,13 @@ HRESULT WINAPI CoInternetGetSecurityUrl(LPCWSTR pwzUrl, LPWSTR *ppwzSecUrl, PSUA
         return S_OK;
     }
 
-    if(psuAction == PSU_DEFAULT) {
-        hres = CoInternetParseUrl(pwzUrl, PARSE_ROOTDOCUMENT, 0, url, 0, &len, 0);
+    hres = CoInternetParseUrl(url, PARSE_ROOTDOCUMENT, 0, domain, 0, &len, 0);
     if(hres == S_FALSE) {
-            hres = CoInternetParseUrl(pwzUrl, PARSE_SCHEMA, 0, domain,
+        hres = CoInternetParseUrl(url, PARSE_SCHEMA, 0, domain,
                 INTERNET_MAX_URL_LENGTH, &len, 0);
         if(hres == S_OK) {
             domain[len] = ':';
-                hres = CoInternetParseUrl(pwzUrl, PARSE_DOMAIN, 0, domain+len+1,
+            hres = CoInternetParseUrl(url, PARSE_DOMAIN, 0, domain+len+1,
                     INTERNET_MAX_URL_LENGTH-len-1, &len, 0);
             if(hres == S_OK) {
                 len = lstrlenW(domain)+1;
@@ -1291,13 +1306,12 @@ HRESULT WINAPI CoInternetGetSecurityUrl(LPCWSTR pwzUrl, LPWSTR *ppwzSecUrl, PSUA
             }
         }
     }
-    }
 
-    len = lstrlenW(pwzUrl)+1;
+    len = lstrlenW(url)+1;
     *ppwzSecUrl = CoTaskMemAlloc(len*sizeof(WCHAR));
     if(!*ppwzSecUrl)
         return E_OUTOFMEMORY;
 
-    memcpy(*ppwzSecUrl, pwzUrl, len*sizeof(WCHAR));
+    memcpy(*ppwzSecUrl, url, len*sizeof(WCHAR));
     return S_OK;
 }