Web Inspector: Audit: provide a way to get the contents of resources
[WebKit-https.git] / Source / WebCore / inspector / agents / page / PageAuditAgent.cpp
1 /*
2  * Copyright (C) 2019 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 INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27 #include "PageAuditAgent.h"
28
29 #include "InspectorAuditAccessibilityObject.h"
30 #include "InspectorAuditDOMObject.h"
31 #include "InspectorAuditResourcesObject.h"
32 #include "JSInspectorAuditAccessibilityObject.h"
33 #include "JSInspectorAuditDOMObject.h"
34 #include "JSInspectorAuditResourcesObject.h"
35 #include "Page.h"
36 #include "PageConsoleClient.h"
37 #include <JavaScriptCore/CallFrame.h>
38 #include <JavaScriptCore/InjectedScript.h>
39 #include <JavaScriptCore/InjectedScriptManager.h>
40 #include <JavaScriptCore/JSLock.h>
41 #include <wtf/Ref.h>
42 #include <wtf/RefPtr.h>
43 #include <wtf/Vector.h>
44 #include <wtf/text/WTFString.h>
45
46 namespace WebCore {
47
48 using namespace Inspector;
49
50 PageAuditAgent::PageAuditAgent(PageAgentContext& context)
51     : InspectorAuditAgent(context)
52     , m_inspectedPage(context.inspectedPage)
53 {
54 }
55
56 InjectedScript PageAuditAgent::injectedScriptForEval(const int* executionContextId)
57 {
58     if (executionContextId)
59         return injectedScriptManager().injectedScriptForId(*executionContextId);
60
61     JSC::ExecState* scriptState = mainWorldExecState(&m_inspectedPage.mainFrame());
62     return injectedScriptManager().injectedScriptFor(scriptState);
63 }
64
65 InjectedScript PageAuditAgent::injectedScriptForEval(ErrorString& errorString, const int* executionContextId)
66 {
67     InjectedScript injectedScript = injectedScriptForEval(executionContextId);
68     if (injectedScript.hasNoValue()) {
69         if (executionContextId)
70             errorString = "Execution context with given id not found."_s;
71         else
72             errorString = "Internal error: main world execution context not found."_s;
73     }
74     return injectedScript;
75 }
76
77 void PageAuditAgent::populateAuditObject(JSC::ExecState* execState, JSC::Strong<JSC::JSObject>& auditObject)
78 {
79     InspectorAuditAgent::populateAuditObject(execState, auditObject);
80
81     ASSERT(execState);
82     if (!execState)
83         return;
84
85     if (auto* globalObject = JSC::jsCast<JSDOMGlobalObject*>(execState->lexicalGlobalObject())) {
86         JSC::JSLockHolder lock(execState);
87
88         if (JSC::JSValue jsInspectorAuditAccessibilityObject = toJSNewlyCreated(execState, globalObject, InspectorAuditAccessibilityObject::create(*this)))
89             auditObject->putDirect(execState->vm(), JSC::Identifier::fromString(execState, "Accessibility"), jsInspectorAuditAccessibilityObject);
90
91         if (JSC::JSValue jsInspectorAuditDOMObject = toJSNewlyCreated(execState, globalObject, InspectorAuditDOMObject::create(*this)))
92             auditObject->putDirect(execState->vm(), JSC::Identifier::fromString(execState, "DOM"), jsInspectorAuditDOMObject);
93
94         if (JSC::JSValue jsInspectorAuditResourcesObject = toJSNewlyCreated(execState, globalObject, InspectorAuditResourcesObject::create(*this)))
95             auditObject->putDirect(execState->vm(), JSC::Identifier::fromString(execState, "Resources"), jsInspectorAuditResourcesObject);
96     }
97 }
98
99 void PageAuditAgent::muteConsole()
100 {
101     InspectorAuditAgent::muteConsole();
102     PageConsoleClient::mute();
103 }
104
105 void PageAuditAgent::unmuteConsole()
106 {
107     PageConsoleClient::unmute();
108     InspectorAuditAgent::unmuteConsole();
109 }
110
111 } // namespace WebCore