1/* 2 * Copyright (C) 2011 Google 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 are 6 * met: 7 * 8 * * Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * * Redistributions in binary form must reproduce the above 11 * copyright notice, this list of conditions and the following disclaimer 12 * in the documentation and/or other materials provided with the 13 * distribution. 14 * * Neither the name of Google Inc. nor the names of its 15 * contributors may be used to endorse or promote products derived from 16 * this software without specific prior written permission. 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 21 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 22 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 23 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 29 */ 30 31#include "config.h" 32 33#include "InspectorPageAgent.h" 34 35#if ENABLE(INSPECTOR) 36 37#include "CachedResourceLoader.h" 38#include "Cookie.h" 39#include "CookieJar.h" 40#include "DocumentLoader.h" 41#include "Frame.h" 42#include "FrameLoadRequest.h" 43#include "InjectedScriptManager.h" 44#include "InspectorFrontend.h" 45#include "InspectorValues.h" 46#include "InstrumentingAgents.h" 47#include "Page.h" 48#include "ScriptObject.h" 49#include "UserGestureIndicator.h" 50#include "WindowFeatures.h" 51#include <wtf/CurrentTime.h> 52#include <wtf/ListHashSet.h> 53 54namespace WebCore { 55 56PassOwnPtr<InspectorPageAgent> InspectorPageAgent::create(InstrumentingAgents* instrumentingAgents, Page* inspectedPage, InjectedScriptManager* injectedScriptManager) 57{ 58 return adoptPtr(new InspectorPageAgent(instrumentingAgents, inspectedPage, injectedScriptManager)); 59} 60 61InspectorPageAgent::InspectorPageAgent(InstrumentingAgents* instrumentingAgents, Page* inspectedPage, InjectedScriptManager* injectedScriptManager) 62 : m_instrumentingAgents(instrumentingAgents) 63 , m_inspectedPage(inspectedPage) 64 , m_injectedScriptManager(injectedScriptManager) 65 , m_frontend(0) 66{ 67} 68 69void InspectorPageAgent::setFrontend(InspectorFrontend* frontend) 70{ 71 m_frontend = frontend; 72 m_instrumentingAgents->setInspectorPageAgent(this); 73 74 // Initialize Web Inspector title. 75 m_frontend->page()->inspectedURLChanged(m_inspectedPage->mainFrame()->document()->url().string()); 76 77} 78 79void InspectorPageAgent::clearFrontend() 80{ 81 m_instrumentingAgents->setInspectorPageAgent(0); 82 m_userAgentOverride = ""; 83 m_frontend = 0; 84} 85 86void InspectorPageAgent::addScriptToEvaluateOnLoad(ErrorString*, const String& source) 87{ 88 m_scriptsToEvaluateOnLoad.append(source); 89} 90 91void InspectorPageAgent::removeAllScriptsToEvaluateOnLoad(ErrorString*) 92{ 93 m_scriptsToEvaluateOnLoad.clear(); 94} 95 96void InspectorPageAgent::reloadPage(ErrorString*, const bool* const optionalIgnoreCache) 97{ 98 m_inspectedPage->mainFrame()->loader()->reload(optionalIgnoreCache ? *optionalIgnoreCache : false); 99} 100 101void InspectorPageAgent::openInInspectedWindow(ErrorString*, const String& url) 102{ 103 Frame* mainFrame = m_inspectedPage->mainFrame(); 104 105 FrameLoadRequest request(mainFrame->document()->securityOrigin(), ResourceRequest(), "_blank"); 106 107 bool created; 108 WindowFeatures windowFeatures; 109 Frame* newFrame = WebCore::createWindow(mainFrame, mainFrame, request, windowFeatures, created); 110 if (!newFrame) 111 return; 112 113 UserGestureIndicator indicator(DefinitelyProcessingUserGesture); 114 newFrame->loader()->setOpener(mainFrame); 115 newFrame->page()->setOpenedByDOM(); 116 newFrame->loader()->changeLocation(mainFrame->document()->securityOrigin(), newFrame->loader()->completeURL(url), "", false, false); 117} 118 119void InspectorPageAgent::setUserAgentOverride(ErrorString*, const String& userAgent) 120{ 121 m_userAgentOverride = userAgent; 122} 123 124void InspectorPageAgent::applyUserAgentOverride(String* userAgent) const 125{ 126 if (!m_userAgentOverride.isEmpty()) 127 *userAgent = m_userAgentOverride; 128} 129 130static PassRefPtr<InspectorObject> buildObjectForCookie(const Cookie& cookie) 131{ 132 RefPtr<InspectorObject> value = InspectorObject::create(); 133 value->setString("name", cookie.name); 134 value->setString("value", cookie.value); 135 value->setString("domain", cookie.domain); 136 value->setString("path", cookie.path); 137 value->setNumber("expires", cookie.expires); 138 value->setNumber("size", (cookie.name.length() + cookie.value.length())); 139 value->setBoolean("httpOnly", cookie.httpOnly); 140 value->setBoolean("secure", cookie.secure); 141 value->setBoolean("session", cookie.session); 142 return value; 143} 144 145static PassRefPtr<InspectorArray> buildArrayForCookies(ListHashSet<Cookie>& cookiesList) 146{ 147 RefPtr<InspectorArray> cookies = InspectorArray::create(); 148 149 ListHashSet<Cookie>::iterator end = cookiesList.end(); 150 ListHashSet<Cookie>::iterator it = cookiesList.begin(); 151 for (int i = 0; it != end; ++it, i++) 152 cookies->pushObject(buildObjectForCookie(*it)); 153 154 return cookies; 155} 156 157void InspectorPageAgent::getCookies(ErrorString*, RefPtr<InspectorArray>* cookies, WTF::String* cookiesString) 158{ 159 // If we can get raw cookies. 160 ListHashSet<Cookie> rawCookiesList; 161 162 // If we can't get raw cookies - fall back to String representation 163 String stringCookiesList; 164 165 // Return value to getRawCookies should be the same for every call because 166 // the return value is platform/network backend specific, and the call will 167 // always return the same true/false value. 168 bool rawCookiesImplemented = false; 169 170 for (Frame* frame = m_inspectedPage->mainFrame(); frame; frame = frame->tree()->traverseNext(m_inspectedPage->mainFrame())) { 171 Document* document = frame->document(); 172 const CachedResourceLoader::DocumentResourceMap& allResources = document->cachedResourceLoader()->allCachedResources(); 173 CachedResourceLoader::DocumentResourceMap::const_iterator end = allResources.end(); 174 for (CachedResourceLoader::DocumentResourceMap::const_iterator it = allResources.begin(); it != end; ++it) { 175 Vector<Cookie> docCookiesList; 176 rawCookiesImplemented = getRawCookies(document, KURL(ParsedURLString, it->second->url()), docCookiesList); 177 178 if (!rawCookiesImplemented) { 179 // FIXME: We need duplication checking for the String representation of cookies. 180 ExceptionCode ec = 0; 181 stringCookiesList += document->cookie(ec); 182 // Exceptions are thrown by cookie() in sandboxed frames. That won't happen here 183 // because "document" is the document of the main frame of the page. 184 ASSERT(!ec); 185 } else { 186 int cookiesSize = docCookiesList.size(); 187 for (int i = 0; i < cookiesSize; i++) { 188 if (!rawCookiesList.contains(docCookiesList[i])) 189 rawCookiesList.add(docCookiesList[i]); 190 } 191 } 192 } 193 } 194 195 if (rawCookiesImplemented) 196 *cookies = buildArrayForCookies(rawCookiesList); 197 else 198 *cookiesString = stringCookiesList; 199} 200 201void InspectorPageAgent::deleteCookie(ErrorString*, const String& cookieName, const String& domain) 202{ 203 for (Frame* frame = m_inspectedPage->mainFrame(); frame; frame = frame->tree()->traverseNext(m_inspectedPage->mainFrame())) { 204 Document* document = frame->document(); 205 if (document->url().host() != domain) 206 continue; 207 const CachedResourceLoader::DocumentResourceMap& allResources = document->cachedResourceLoader()->allCachedResources(); 208 CachedResourceLoader::DocumentResourceMap::const_iterator end = allResources.end(); 209 for (CachedResourceLoader::DocumentResourceMap::const_iterator it = allResources.begin(); it != end; ++it) 210 WebCore::deleteCookie(document, KURL(ParsedURLString, it->second->url()), cookieName); 211 } 212} 213 214void InspectorPageAgent::inspectedURLChanged(const String& url) 215{ 216 m_frontend->page()->inspectedURLChanged(url); 217} 218 219void InspectorPageAgent::restore() 220{ 221 inspectedURLChanged(m_inspectedPage->mainFrame()->document()->url().string()); 222} 223 224void InspectorPageAgent::didCommitLoad(const String& url) 225{ 226 inspectedURLChanged(url); 227} 228 229void InspectorPageAgent::domContentEventFired() 230{ 231 m_frontend->page()->domContentEventFired(currentTime()); 232} 233 234void InspectorPageAgent::loadEventFired() 235{ 236 m_frontend->page()->loadEventFired(currentTime()); 237} 238 239void InspectorPageAgent::didClearWindowObjectInWorld(Frame* frame, DOMWrapperWorld* world) 240{ 241 if (world != mainThreadNormalWorld()) 242 return; 243 244 if (frame == m_inspectedPage->mainFrame()) 245 m_injectedScriptManager->discardInjectedScripts(); 246 247 if (m_scriptsToEvaluateOnLoad.size()) { 248 ScriptState* scriptState = mainWorldScriptState(frame); 249 for (Vector<String>::iterator it = m_scriptsToEvaluateOnLoad.begin(); 250 it != m_scriptsToEvaluateOnLoad.end(); ++it) { 251 m_injectedScriptManager->injectScript(*it, scriptState); 252 } 253 } 254} 255 256} // namespace WebCore 257 258#endif // ENABLE(INSPECTOR) 259