Thomas Mullaly : urlmon: Added implementation for PARSE_ROOTDOCUMENT for CoInternetParseIUri.

[Alexandre Julliard]

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

Author: Thomas Mullaly 
Date:   Sun Nov 28 15:42:16 2010 -0500

urlmon: Added implementation for PARSE_ROOTDOCUMENT for CoInternetParseIUri.

---

dlls/urlmon/tests/uri.c |   10 +++++++-
 dlls/urlmon/uri.c       |   60 +++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 69 insertions(+), 1 deletions(-)

diff --git a/dlls/urlmon/tests/uri.c b/dlls/urlmon/tests/uri.c
index dfeed9b..c03e516 100644
--- a/dlls/urlmon/tests/uri.c
+++ b/dlls/urlmon/tests/uri.c
@@ -5925,7 +5925,15 @@ static const uri_parse_test uri_parse_tests[] = {

/* PARSE_FRIENDLY tests. */
     {"http://[e-mail]/test#test%22,0,PARSE_FRIENDLY,0,%22http://google.com/...,
-    {"zip://[e-mail]/test",0,PARSE_FRIENDLY,0,"zip://[e-mail]/test",S_OK,FALSE}
+    {"zip://[e-mail]/test",0,PARSE_FRIENDLY,0,"zip://[e-mail]/test",S_OK,FALSE},
+
+    /* PARSE_ROOTDOCUMENT tests. */
+    {"http://google.com:200/test/test%22,0,PARSE_ROOTDOCUMENT,0,%22http://google.c...,
+    {"http://google.com%22,Uri_CREATE_NO_CANONICALIZE,PARSE_ROOTDOCUMENT,0,%22http...,
+    {"zip://google.com/",0,PARSE_ROOTDOCUMENT,0,"",S_OK,FALSE},
+    {"file:///c:/testing/",0,PARSE_ROOTDOCUMENT,0,"",S_OK,FALSE},
+    {"file://server/test",0,PARSE_ROOTDOCUMENT,0,"",S_OK,FALSE},
+    {"zip:test/test",0,PARSE_ROOTDOCUMENT,0,"",S_OK,FALSE}
 };

static inline LPWSTR a2w(LPCSTR str) {
diff --git a/dlls/urlmon/uri.c b/dlls/urlmon/uri.c
index 64eaba2..475db7f 100644
--- a/dlls/urlmon/uri.c
+++ b/dlls/urlmon/uri.c
@@ -6214,6 +6214,57 @@ static HRESULT parse_friendly(IUri *uri, LPWSTR output, DWORD output_len,
     return S_OK;
 }

+static HRESULT parse_rootdocument(const Uri *uri, LPWSTR output, DWORD output_len,
+                                  DWORD *result_len)
+{
+    static const WCHAR colon_slashesW[] = {':','/','/'};
+
+    WCHAR *ptr;
+    DWORD len = 0;
+
+    /* Windows only returns the root document if the URI has an authority
+     * and it's not an unknown scheme type or a file scheme type.
+     */
+    if(uri->authority_start == -1 ||
+       uri->scheme_type == URL_SCHEME_UNKNOWN ||
+       uri->scheme_type == URL_SCHEME_FILE) {
+        *result_len = 0;
+        if(!output_len)
+            return STRSAFE_E_INSUFFICIENT_BUFFER;
+
+        output[0] = 0;
+        return S_OK;
+    }
+
+    len = uri->scheme_len+uri->authority_len;
+    /* For the "://" and '/' which will be added. */
+    len += 4;
+
+    if(len+1 > output_len) {
+        *result_len = len;
+        return STRSAFE_E_INSUFFICIENT_BUFFER;
+    }
+
+    ptr = output;
+    memcpy(ptr, uri->canon_uri+uri->scheme_start, uri->scheme_len*sizeof(WCHAR));
+
+    /* Add the "://". */
+    ptr += uri->scheme_len;
+    memcpy(ptr, colon_slashesW, sizeof(colon_slashesW));
+
+    /* Add the authority. */
+    ptr += sizeof(colon_slashesW)/sizeof(WCHAR);
+    memcpy(ptr, uri->canon_uri+uri->authority_start, uri->authority_len*sizeof(WCHAR));
+
+    /* Add the '/' after the authority. */
+    ptr += uri->authority_len;
+    *ptr = '/';
+    ptr[1] = 0;
+
+    *result_len = len;
+    return S_OK;
+}
+
 /***********************************************************************
  *           CoInternetParseIUri (urlmon.@)
  */
@@ -6248,6 +6299,15 @@ HRESULT WINAPI CoInternetParseIUri(IUri *pIUri, PARSEACTION ParseAction, DWORD d
     case PARSE_FRIENDLY:
         hr = parse_friendly(pIUri, pwzResult, cchResult, pcchResult);
         break;
+    case PARSE_ROOTDOCUMENT:
+        if(!(uri = get_uri_obj(pIUri))) {
+            *pcchResult = 0;
+            FIXME("(%p %d %x %p %d %p %x) Unknown IUri's not supported for this action.\n",
+                pIUri, ParseAction, dwFlags, pwzResult, cchResult, pcchResult, (DWORD)dwReserved);
+            return E_NOTIMPL;
+        }
+        hr = parse_rootdocument(uri, pwzResult, cchResult, pcchResult);
+        break;
     case PARSE_SECURITY_URL:
     case PARSE_MIME:
     case PARSE_SERVER: