4144bdaeb3a5fcefe6903aa4ecdada353cb315db
[WebKit-https.git] / Source / JavaScriptCore / profiler / ProfilerDatabase.cpp
1 /*
2  * Copyright (C) 2012-2013, 2016 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. ``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 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 "ProfilerDatabase.h"
28
29 #include "CodeBlock.h"
30 #include "JSONObject.h"
31 #include "ObjectConstructor.h"
32 #include "JSCInlines.h"
33 #include <wtf/CurrentTime.h>
34 #include <wtf/FilePrintStream.h>
35
36 namespace JSC { namespace Profiler {
37
38 static std::atomic<int> databaseCounter;
39
40 static StaticLock registrationLock;
41 static std::atomic<int> didRegisterAtExit;
42 static Database* firstDatabase;
43
44 Database::Database(VM& vm)
45     : m_databaseID(++databaseCounter)
46     , m_vm(vm)
47     , m_shouldSaveAtExit(false)
48     , m_nextRegisteredDatabase(0)
49 {
50 }
51
52 Database::~Database()
53 {
54     if (m_shouldSaveAtExit) {
55         removeDatabaseFromAtExit();
56         performAtExitSave();
57     }
58 }
59
60 Bytecodes* Database::ensureBytecodesFor(CodeBlock* codeBlock)
61 {
62     LockHolder locker(m_lock);
63     return ensureBytecodesFor(locker, codeBlock);
64 }
65
66 Bytecodes* Database::ensureBytecodesFor(const AbstractLocker&, CodeBlock* codeBlock)
67 {
68     codeBlock = codeBlock->baselineAlternative();
69     
70     HashMap<CodeBlock*, Bytecodes*>::iterator iter = m_bytecodesMap.find(codeBlock);
71     if (iter != m_bytecodesMap.end())
72         return iter->value;
73     
74     m_bytecodes.append(Bytecodes(m_bytecodes.size(), codeBlock));
75     Bytecodes* result = &m_bytecodes.last();
76     
77     m_bytecodesMap.add(codeBlock, result);
78     
79     return result;
80 }
81
82 void Database::notifyDestruction(CodeBlock* codeBlock)
83 {
84     LockHolder locker(m_lock);
85     
86     m_bytecodesMap.remove(codeBlock);
87     m_compilationMap.remove(codeBlock);
88 }
89
90 void Database::addCompilation(CodeBlock* codeBlock, Ref<Compilation>&& compilation)
91 {
92     LockHolder locker(m_lock);
93     ASSERT(!isCompilationThread());
94
95     m_compilations.append(compilation.copyRef());
96     m_compilationMap.set(codeBlock, WTFMove(compilation));
97 }
98
99 JSValue Database::toJS(ExecState* exec) const
100 {
101     VM& vm = exec->vm();
102     auto scope = DECLARE_THROW_SCOPE(vm);
103     JSObject* result = constructEmptyObject(exec);
104     
105     JSArray* bytecodes = constructEmptyArray(exec, 0);
106     RETURN_IF_EXCEPTION(scope, { });
107     for (unsigned i = 0; i < m_bytecodes.size(); ++i) {
108         auto value = m_bytecodes[i].toJS(exec);
109         RETURN_IF_EXCEPTION(scope, { });
110         bytecodes->putDirectIndex(exec, i, value);
111         RETURN_IF_EXCEPTION(scope, { });
112     }
113     result->putDirect(vm, exec->propertyNames().bytecodes, bytecodes);
114     
115     JSArray* compilations = constructEmptyArray(exec, 0);
116     RETURN_IF_EXCEPTION(scope, { });
117     for (unsigned i = 0; i < m_compilations.size(); ++i) {
118         auto value = m_compilations[i]->toJS(exec);
119         RETURN_IF_EXCEPTION(scope, { });
120         compilations->putDirectIndex(exec, i, value);
121         RETURN_IF_EXCEPTION(scope, { });
122     }
123     result->putDirect(vm, exec->propertyNames().compilations, compilations);
124     
125     JSArray* events = constructEmptyArray(exec, 0);
126     RETURN_IF_EXCEPTION(scope, { });
127     for (unsigned i = 0; i < m_events.size(); ++i) {
128         auto value = m_events[i].toJS(exec);
129         RETURN_IF_EXCEPTION(scope, { });
130         events->putDirectIndex(exec, i, value);
131         RETURN_IF_EXCEPTION(scope, { });
132     }
133     result->putDirect(vm, exec->propertyNames().events, events);
134     
135     return result;
136 }
137
138 String Database::toJSON() const
139 {
140     auto scope = DECLARE_THROW_SCOPE(m_vm);
141     JSGlobalObject* globalObject = JSGlobalObject::create(
142         m_vm, JSGlobalObject::createStructure(m_vm, jsNull()));
143
144     auto value = toJS(globalObject->globalExec());
145     RETURN_IF_EXCEPTION(scope, String());
146     scope.release();
147     return JSONStringify(globalObject->globalExec(), value, 0);
148 }
149
150 bool Database::save(const char* filename) const
151 {
152     auto scope = DECLARE_CATCH_SCOPE(m_vm);
153     auto out = FilePrintStream::open(filename, "w");
154     if (!out)
155         return false;
156     
157     String data = toJSON();
158     if (UNLIKELY(scope.exception())) {
159         scope.clearException();
160         return false;
161     }
162     out->print(data);
163     return true;
164 }
165
166 void Database::registerToSaveAtExit(const char* filename)
167 {
168     m_atExitSaveFilename = filename;
169     
170     if (m_shouldSaveAtExit)
171         return;
172     
173     addDatabaseToAtExit();
174     m_shouldSaveAtExit = true;
175 }
176
177 void Database::logEvent(CodeBlock* codeBlock, const char* summary, const CString& detail)
178 {
179     LockHolder locker(m_lock);
180     
181     Bytecodes* bytecodes = ensureBytecodesFor(locker, codeBlock);
182     Compilation* compilation = m_compilationMap.get(codeBlock);
183     m_events.append(Event(currentTime(), bytecodes, compilation, summary, detail));
184 }
185
186 void Database::addDatabaseToAtExit()
187 {
188     if (++didRegisterAtExit == 1)
189         atexit(atExitCallback);
190     
191     LockHolder holder(registrationLock);
192     m_nextRegisteredDatabase = firstDatabase;
193     firstDatabase = this;
194 }
195
196 void Database::removeDatabaseFromAtExit()
197 {
198     LockHolder holder(registrationLock);
199     for (Database** current = &firstDatabase; *current; current = &(*current)->m_nextRegisteredDatabase) {
200         if (*current != this)
201             continue;
202         *current = m_nextRegisteredDatabase;
203         m_nextRegisteredDatabase = 0;
204         m_shouldSaveAtExit = false;
205         break;
206     }
207 }
208
209 void Database::performAtExitSave() const
210 {
211     JSLockHolder lock(m_vm);
212     save(m_atExitSaveFilename.data());
213 }
214
215 Database* Database::removeFirstAtExitDatabase()
216 {
217     LockHolder holder(registrationLock);
218     Database* result = firstDatabase;
219     if (result) {
220         firstDatabase = result->m_nextRegisteredDatabase;
221         result->m_nextRegisteredDatabase = 0;
222         result->m_shouldSaveAtExit = false;
223     }
224     return result;
225 }
226
227 void Database::atExitCallback()
228 {
229     while (Database* database = removeFirstAtExitDatabase())
230         database->performAtExitSave();
231 }
232
233 } } // namespace JSC::Profiler
234