[3/3] mshtml: Implement IHTMLLocation::get_port

[aeikum at codeweavers.com]

---
 dlls/mshtml/htmllocation.c       |   23 +++++++++++++++++++++--
 dlls/mshtml/tests/htmllocation.c |   12 ++++++------
 2 files changed, 27 insertions(+), 8 deletions(-)

diff --git a/dlls/mshtml/htmllocation.c b/dlls/mshtml/htmllocation.c index
4769f03..e86cee6 100644
--- a/dlls/mshtml/htmllocation.c
+++ b/dlls/mshtml/htmllocation.c
@@ -245,12 +245,31 @@ static HRESULT WINAPI
HTMLLocation_put_port(IHTMLLocation *iface, BSTR v)
 static HRESULT WINAPI HTMLLocation_get_port(IHTMLLocation *iface, BSTR
*p)
 {
     HTMLLocation *This = HTMLLOCATION_THIS(iface);
-    FIXME("(%p)->(%p)\n", This, p);
+    URL_COMPONENTSW url = {sizeof(URL_COMPONENTSW)};
+    HRESULT hres;
+
+    TRACE("(%p)->(%p)\n", This, p);

     if(!p)
         return E_POINTER;

-    return E_NOTIMPL;
+    hres = get_url_components(This, &url);
+    if(FAILED(hres))
+        return hres;
+
+    if(url.nPort) {
+        const WCHAR format[] = {'%','d',0};
+        WCHAR buf[6];
+        snprintfW(buf, 6, format, url.nPort);
+        *p = SysAllocString(buf);
+    }else {
+        const WCHAR empty[] = {0};
+        *p = SysAllocString(empty);
+    }
+
+    if(!*p)
+        return E_OUTOFMEMORY;
+    return S_OK;
 }

 static HRESULT WINAPI HTMLLocation_put_pathname(IHTMLLocation *iface,
BSTR v)
diff --git a/dlls/mshtml/tests/htmllocation.c
b/dlls/mshtml/tests/htmllocation.c
index c3a5e82..778feb8 100644
--- a/dlls/mshtml/tests/htmllocation.c
+++ b/dlls/mshtml/tests/htmllocation.c
@@ -60,7 +60,7 @@ static const struct location_test http_test = {
             "http:", FALSE,
             "www.winehq.org:80", FALSE,
             "www.winehq.org", FALSE,
-            "80", FALSE,
+            "80", TRUE,
             "", FALSE,
             "?search", FALSE,
             "#hash", FALSE
@@ -74,7 +74,7 @@ static const struct location_test http_file_test = {
             "http:", FALSE,
             "www.winehq.org:80", FALSE,
             "www.winehq.org", FALSE,
-            "80", FALSE,
+            "80", TRUE,
             "file", FALSE,
             "?search", FALSE,
             "#hash", FALSE
@@ -88,7 +88,7 @@ static const struct location_test ftp_test = {
             "ftp:", FALSE,
             "ftp.winehq.org:21", FALSE,
             "ftp.winehq.org", FALSE,
-            "21", FALSE,
+            "21", TRUE,
             "", FALSE,
             NULL, FALSE,
             NULL, FALSE
@@ -102,7 +102,7 @@ static const struct location_test ftp_file_test = {
             "ftp:", FALSE,
             "ftp.winehq.org:21", FALSE,
             "ftp.winehq.org", FALSE,
-            "21", FALSE,
+            "21", TRUE,
             "file", FALSE,
             NULL, FALSE,
             NULL, FALSE
@@ -116,7 +116,7 @@ static const struct location_test file_test = {
             "file:", FALSE,
             NULL, FALSE,
             NULL, FALSE,
-            "", FALSE,
+            "", TRUE,
             "C:\\windows\\win.ini", TRUE,
             NULL, FALSE,
             NULL, FALSE
@@ -217,7 +217,7 @@ static void test_port(IHTMLLocation *loc, const struct
location_test *test)
     ok(hres == E_POINTER, "%s: get_port should have failed with E_POINTER
(0x%08x), was: 0x%08x\n", test->name, E_POINTER, hres);

     hres = IHTMLLocation_get_port(loc, &str);
-    todo_wine ok(hres == S_OK, "%s: get_port failed: 0x%08x\n",
test->name, hres);
+    ok(hres == S_OK, "%s: get_port failed: 0x%08x\n", test->name, hres);
     if(hres == S_OK){
         if(test->port_ok)
             ok(str_eq_wa(str, test->port), "%s: expected retrieved port
to be L\"%s\", was: %s\n", test->name, test->port,
wine_dbgstr_w(str));
-- 
1.6.4.4