From c3cbae6131079865719d889d8c95ec1b46db4811 Mon Sep 17 00:00:00 2001
From: "radha%netscape.com" <radha%netscape.com>
Date: Wed, 5 Mar 2003 22:23:47 +0000
Subject: [PATCH] Fix for bug #162128. Find a usable base uri for urls loaded
 through javascript. r=jst sr=darin.

---
 docshell/base/nsDocShell.cpp  |  3 +-
 dom/src/base/nsGlobalWindow.h |  1 +
 dom/src/base/nsLocation.cpp   | 70 +++++++++++++++++++++++++++++++++--
 3 files changed, 69 insertions(+), 5 deletions(-)

diff --git a/docshell/base/nsDocShell.cpp b/docshell/base/nsDocShell.cpp
index 65b8d014a5a1..a8e9dc0a2e5d 100644
--- a/docshell/base/nsDocShell.cpp
+++ b/docshell/base/nsDocShell.cpp
@@ -6119,7 +6119,8 @@ nsDocShell::AddToSessionHistory(nsIURI * aURI,
     }
     else {  
         // This is a subframe.
-        if (mLoadType != LOAD_NORMAL_REPLACE)
+        if ((mLoadType != LOAD_NORMAL_REPLACE) ||
+            (mLoadType == LOAD_NORMAL_REPLACE && !mOSHE))
             rv = AddChildSHEntry(nsnull, entry, mChildOffset);
     }
 
diff --git a/dom/src/base/nsGlobalWindow.h b/dom/src/base/nsGlobalWindow.h
index 33c24dbe6f28..f0370a5b0269 100644
--- a/dom/src/base/nsGlobalWindow.h
+++ b/dom/src/base/nsGlobalWindow.h
@@ -486,6 +486,7 @@ protected:
   nsresult GetSourceDocument(JSContext* cx, nsIDocument** aDocument);
 
   nsresult CheckURL(nsIURI *url, nsIDocShellLoadInfo** aLoadInfo);
+  nsresult FindUsableBaseURI(nsIURI * aBaseURI, nsIDocShell * aParent, nsIURI ** aUsableURI);
 
   nsIDocShell *mDocShell; // Weak Reference
 };
diff --git a/dom/src/base/nsLocation.cpp b/dom/src/base/nsLocation.cpp
index 47c87ac6d093..fc9d286be816 100644
--- a/dom/src/base/nsLocation.cpp
+++ b/dom/src/base/nsLocation.cpp
@@ -65,7 +65,7 @@
 #include "nsDOMError.h"
 #include "nsDOMClassInfo.h"
 #include "nsCRT.h"
-
+#include "nsIProtocolHandler.h"
 
 static nsresult GetDocumentCharacterSetForURI(const nsAString& aHref, nsACString& aCharset)
 {
@@ -200,6 +200,63 @@ LocationImpl::CheckURL(nsIURI* aURI, nsIDocShellLoadInfo** aLoadInfo)
   return NS_OK;
 }
 
+// Walk up the docshell hierarchy and find a usable base URI. Basically 
+// anything that would allow a relative uri.
+
+nsresult
+LocationImpl::FindUsableBaseURI(nsIURI * aBaseURI, nsIDocShell * aParent, nsIURI ** aUsableURI)
+{
+  if (!aBaseURI || !aParent)
+    return NS_ERROR_FAILURE;
+  NS_ENSURE_ARG_POINTER(aUsableURI);
+    
+  *aUsableURI = nsnull;
+  nsresult rv = NS_OK;    
+  nsCOMPtr<nsIDocShell> parentDS = aParent;
+  nsCOMPtr<nsIURI> baseURI = aBaseURI;
+  nsCOMPtr<nsIIOService> ioService(do_GetService(NS_IOSERVICE_CONTRACTID, &rv));
+
+  while(NS_SUCCEEDED(rv) && baseURI && ioService) {
+    // Check if the current base uri supports relative uris.
+    // We make this check by looking at the protocol flags of
+    // the protocol handler. If the protocol flags has URI_NORELATIVE,
+    // it means that the base uri does not support relative uris.
+    nsCAutoString scheme;
+    baseURI->GetScheme(scheme);
+    nsCOMPtr<nsIProtocolHandler> protocolHandler;
+    // Get the protocol handler for the base uri.
+    ioService->GetProtocolHandler(scheme.get(), getter_AddRefs(protocolHandler));
+    if (!protocolHandler)
+      return NS_ERROR_FAILURE;
+    PRUint32 pFlags; // Is there a default value for the protocol flags?
+    protocolHandler->GetProtocolFlags(&pFlags);
+    if (!(pFlags & nsIProtocolHandler::URI_NORELATIVE)) {
+      *aUsableURI = baseURI;
+      NS_ADDREF(*aUsableURI);
+      return NS_OK;
+    }
+
+    // Get the same type parent docshell
+    nsCOMPtr<nsIDocShellTreeItem> docShellAsTreeItem(do_QueryInterface(parentDS));
+    if (!docShellAsTreeItem)
+      return NS_ERROR_FAILURE;
+    nsCOMPtr<nsIDocShellTreeItem> parentDSTreeItem;
+    docShellAsTreeItem->GetSameTypeParent(getter_AddRefs(parentDSTreeItem));      
+    nsCOMPtr<nsIWebNavigation> webNav(do_QueryInterface(parentDSTreeItem));
+
+    // Get the parent docshell's uri
+    if (webNav) {
+      rv = webNav->GetCurrentURI(getter_AddRefs(baseURI));
+      parentDS = do_QueryInterface(parentDSTreeItem);
+    }
+    else
+      return NS_ERROR_FAILURE;
+  }  // while 
+
+    return rv;
+}
+
+
 nsresult
 LocationImpl::GetURI(nsIURI** aURI)
 {
@@ -475,13 +532,18 @@ LocationImpl::SetHrefWithBase(const nsAString& aHref,
                               nsIURI* aBase, PRBool aReplace)
 {
   nsresult result;
-  nsCOMPtr<nsIURI> newUri;
+  nsCOMPtr<nsIURI> newUri, baseURI;
+
+  // Make sure the base url is something that will be useful. 
+  result = FindUsableBaseURI(aBase,  mDocShell, getter_AddRefs(baseURI));
+  if (!baseURI)
+    return NS_ERROR_FAILURE;
 
   nsCAutoString docCharset;
   if (NS_SUCCEEDED(GetDocumentCharacterSetForURI(aHref, docCharset)))
-    result = NS_NewURI(getter_AddRefs(newUri), aHref, docCharset.get(), aBase);
+    result = NS_NewURI(getter_AddRefs(newUri), aHref, docCharset.get(), baseURI);
   else
-    result = NS_NewURI(getter_AddRefs(newUri), aHref, nsnull, aBase);
+    result = NS_NewURI(getter_AddRefs(newUri), aHref, nsnull, baseURI);
 
   if (newUri && mDocShell) {
     nsCOMPtr<nsIDocShellLoadInfo> loadInfo;