1 /* 2 * Copyright (C) 2009 Apple Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 1. Redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer. 9 * 2. Redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution. 12 * 13 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY 14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR 17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY 21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26 #include "config.h" 27 #include "HistoryDelegate.h" 28 29 #include "DumpRenderTree.h" 30 #include "DumpRenderTreeWin.h" 31 #include "LayoutTestController.h" 32 #include <string> 33 #include <WebKit/WebKit.h> 34 35 using std::wstring; 36 37 static inline wstring wstringFromBSTR(BSTR str) 38 { 39 return wstring(str, ::SysStringLen(str)); 40 } 41 42 HistoryDelegate::HistoryDelegate() 43 : m_refCount(1) 44 { 45 } 46 47 HistoryDelegate::~HistoryDelegate() 48 { 49 } 50 51 // IUnknown 52 HRESULT HistoryDelegate::QueryInterface(REFIID riid, void** ppvObject) 53 { 54 *ppvObject = 0; 55 if (IsEqualGUID(riid, IID_IUnknown)) 56 *ppvObject = static_cast<IWebHistoryDelegate*>(this); 57 else if (IsEqualGUID(riid, IID_IWebHistoryDelegate)) 58 *ppvObject = static_cast<IWebHistoryDelegate*>(this); 59 else 60 return E_NOINTERFACE; 61 62 AddRef(); 63 return S_OK; 64 } 65 66 ULONG HistoryDelegate::AddRef(void) 67 { 68 return ++m_refCount; 69 } 70 71 ULONG HistoryDelegate::Release(void) 72 { 73 ULONG newRef = --m_refCount; 74 if (!newRef) 75 delete(this); 76 77 return newRef; 78 } 79 80 // IWebHistoryDelegate 81 HRESULT HistoryDelegate::didNavigateWithNavigationData(IWebView* webView, IWebNavigationData* navigationData, IWebFrame* webFrame) 82 { 83 if (!gLayoutTestController->dumpHistoryDelegateCallbacks()) 84 return S_OK; 85 86 BSTR urlBSTR; 87 if (FAILED(navigationData->url(&urlBSTR))) 88 return E_FAIL; 89 wstring url; 90 if (urlBSTR) 91 url = urlSuitableForTestResult(wstringFromBSTR(urlBSTR)); 92 SysFreeString(urlBSTR); 93 94 BSTR titleBSTR; 95 if (FAILED(navigationData->title(&titleBSTR))) 96 return E_FAIL; 97 wstring title; 98 if (titleBSTR) 99 title = wstringFromBSTR(titleBSTR); 100 SysFreeString(titleBSTR); 101 102 COMPtr<IWebURLRequest> request; 103 if (FAILED(navigationData->originalRequest(&request))) 104 return E_FAIL; 105 106 BSTR httpMethodBSTR; 107 if (FAILED(request->HTTPMethod(&httpMethodBSTR))) 108 return E_FAIL; 109 wstring httpMethod; 110 if (httpMethodBSTR) 111 httpMethod = wstringFromBSTR(httpMethodBSTR); 112 SysFreeString(httpMethodBSTR); 113 114 COMPtr<IWebURLResponse> response; 115 if (FAILED(navigationData->response(&response))) 116 return E_FAIL; 117 118 COMPtr<IWebHTTPURLResponse> httpResponse; 119 if (FAILED(response->QueryInterface(&httpResponse))) 120 return E_FAIL; 121 122 int statusCode = 0; 123 if (FAILED(httpResponse->statusCode(&statusCode))) 124 return E_FAIL; 125 126 BOOL hasSubstituteData; 127 if (FAILED(navigationData->hasSubstituteData(&hasSubstituteData))) 128 return E_FAIL; 129 130 BSTR clientRedirectSourceBSTR; 131 if (FAILED(navigationData->clientRedirectSource(&clientRedirectSourceBSTR))) 132 return E_FAIL; 133 bool hasClientRedirect = clientRedirectSourceBSTR && SysStringLen(clientRedirectSourceBSTR); 134 wstring redirectSource; 135 if (clientRedirectSourceBSTR) 136 redirectSource = urlSuitableForTestResult(wstringFromBSTR(clientRedirectSourceBSTR)); 137 SysFreeString(clientRedirectSourceBSTR); 138 139 bool wasFailure = hasSubstituteData || (httpResponse && statusCode >= 400); 140 141 printf("WebView navigated to url \"%S\" with title \"%S\" with HTTP equivalent method \"%S\". The navigation was %s and was %s%S.\n", 142 url.c_str(), 143 title.c_str(), 144 httpMethod.c_str(), 145 wasFailure ? "a failure" : "successful", 146 hasClientRedirect ? "a client redirect from " : "not a client redirect", 147 redirectSource.c_str()); 148 149 return S_OK; 150 } 151 152 HRESULT HistoryDelegate::didPerformClientRedirectFromURL(IWebView*, BSTR sourceURL, BSTR destinationURL, IWebFrame*) 153 { 154 if (!gLayoutTestController->dumpHistoryDelegateCallbacks()) 155 return S_OK; 156 157 wstring source; 158 if (sourceURL) 159 source = urlSuitableForTestResult(wstringFromBSTR(sourceURL)); 160 161 wstring destination; 162 if (destinationURL) 163 destination = urlSuitableForTestResult(wstringFromBSTR(destinationURL)); 164 165 printf("WebView performed a client redirect from \"%S\" to \"%S\".\n", source.c_str(), destination.c_str()); 166 return S_OK; 167 } 168 169 HRESULT HistoryDelegate::didPerformServerRedirectFromURL(IWebView* webView, BSTR sourceURL, BSTR destinationURL, IWebFrame* webFrame) 170 { 171 if (!gLayoutTestController->dumpHistoryDelegateCallbacks()) 172 return S_OK; 173 174 wstring source; 175 if (sourceURL) 176 source = urlSuitableForTestResult(wstringFromBSTR(sourceURL)); 177 178 wstring destination; 179 if (destinationURL) 180 destination = urlSuitableForTestResult(wstringFromBSTR(destinationURL)); 181 182 printf("WebView performed a server redirect from \"%S\" to \"%S\".\n", source.c_str(), destination.c_str()); 183 return S_OK; 184 } 185 186 HRESULT HistoryDelegate::updateHistoryTitle(IWebView* webView, BSTR titleBSTR, BSTR urlBSTR) 187 { 188 if (!gLayoutTestController->dumpHistoryDelegateCallbacks()) 189 return S_OK; 190 191 wstring url; 192 if (urlBSTR) 193 url = urlSuitableForTestResult(wstringFromBSTR(urlBSTR)); 194 195 wstring title; 196 if (titleBSTR) 197 title = wstringFromBSTR(titleBSTR); 198 199 printf("WebView updated the title for history URL \"%S\" to \"%S\".\n", url.c_str(), title.c_str()); 200 return S_OK; 201 } 202 203 HRESULT HistoryDelegate::populateVisitedLinksForWebView(IWebView* webView) 204 { 205 if (!gLayoutTestController->dumpHistoryDelegateCallbacks()) 206 return S_OK; 207 208 BSTR urlBSTR; 209 if (FAILED(webView->mainFrameURL(&urlBSTR))) 210 return E_FAIL; 211 212 wstring url; 213 if (urlBSTR) 214 url = urlSuitableForTestResult(wstringFromBSTR(urlBSTR)); 215 SysFreeString(urlBSTR); 216 217 if (gLayoutTestController->dumpVisitedLinksCallback()) 218 printf("Asked to populate visited links for WebView \"%S\"\n", url.c_str()); 219 220 return S_OK; 221 } 222
