| /* |
| * Copyright (C) 2017 Apple Inc. All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * 1. Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * |
| * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' |
| * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, |
| * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
| * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS |
| * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
| * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
| * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
| * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
| * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
| * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF |
| * THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #include "config.h" |
| #include "PageNetworkAgent.h" |
| |
| #include "Document.h" |
| #include "DocumentLoader.h" |
| #include "Frame.h" |
| #include "Page.h" |
| #include "ScriptState.h" |
| #include "WebSocket.h" |
| #include "WebSocketChannel.h" |
| |
| namespace WebCore { |
| |
| using namespace Inspector; |
| |
| PageNetworkAgent::PageNetworkAgent(PageAgentContext& context, InspectorPageAgent* pageAgent) |
| : InspectorNetworkAgent(context) |
| , m_pageAgent(pageAgent) |
| { |
| ASSERT(m_pageAgent); |
| } |
| |
| String PageNetworkAgent::loaderIdentifier(DocumentLoader* loader) |
| { |
| return m_pageAgent->loaderId(loader); |
| } |
| |
| String PageNetworkAgent::frameIdentifier(DocumentLoader* loader) |
| { |
| if (!loader) |
| return { }; |
| return m_pageAgent->frameId(loader->frame()); |
| } |
| |
| Vector<WebSocket*> PageNetworkAgent::activeWebSockets(const LockHolder& lock) |
| { |
| Vector<WebSocket*> webSockets; |
| |
| for (WebSocket* webSocket : WebSocket::allActiveWebSockets(lock)) { |
| if (!is<WebSocketChannel>(webSocket->channel().get())) |
| continue; |
| |
| auto* channel = downcast<WebSocketChannel>(webSocket->channel().get()); |
| if (!channel) |
| continue; |
| |
| if (!is<Document>(webSocket->scriptExecutionContext())) |
| continue; |
| |
| // FIXME: <https://webkit.org/b/168475> Web Inspector: Correctly display iframe's WebSockets |
| auto* document = downcast<Document>(webSocket->scriptExecutionContext()); |
| if (document->page() != &m_pageAgent->page()) |
| continue; |
| |
| webSockets.append(webSocket); |
| } |
| |
| return webSockets; |
| } |
| |
| void PageNetworkAgent::setResourceCachingDisabled(bool disabled) |
| { |
| m_pageAgent->page().setResourceCachingDisabledOverride(disabled); |
| } |
| |
| ScriptExecutionContext* PageNetworkAgent::scriptExecutionContext(ErrorString& errorString, const String& frameId) |
| { |
| auto* frame = m_pageAgent->assertFrame(errorString, frameId); |
| if (!frame) |
| return nullptr; |
| |
| auto* document = frame->document(); |
| if (!document) { |
| errorString = ASCIILiteral("No Document instance for the specified frame"); |
| return nullptr; |
| } |
| |
| return document; |
| } |
| |
| } // namespace WebCore |