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#pragma once
27
28#include "JSCJSValue.h"
29#include "ProfilerBytecodes.h"
30#include "ProfilerCompilation.h"
31#include "ProfilerEvent.h"
32#include <wtf/FastMalloc.h>
33#include <wtf/HashMap.h>
34#include <wtf/Lock.h>
35#include <wtf/Noncopyable.h>
36#include <wtf/SegmentedVector.h>
37#include <wtf/text/WTFString.h>
38
39namespace JSC { namespace Profiler {
40
41class Database {
42 WTF_MAKE_FAST_ALLOCATED; WTF_MAKE_NONCOPYABLE(Database);
43public:
44 JS_EXPORT_PRIVATE Database(VM&);
45 JS_EXPORT_PRIVATE ~Database();
46
47 int databaseID() const { return m_databaseID; }
48
49 Bytecodes* ensureBytecodesFor(CodeBlock*);
50 void notifyDestruction(CodeBlock*);
51
52 void addCompilation(CodeBlock*, Ref<Compilation>&&);
53
54 // Converts the database to a JavaScript object that is suitable for JSON stringification.
55 // Note that it's probably a good idea to use an CallFrame* associated with a global
56 // object that is "clean" - i.e. array and object prototypes haven't had strange things
57 // done to them. And yes, it should be appropriate to just use a globalExec here.
58 JS_EXPORT_PRIVATE JSValue toJS(JSGlobalObject*) const;
59
60 // Converts the database to a JavaScript object using a private temporary global object,
61 // and then returns the JSON representation of that object.
62 JS_EXPORT_PRIVATE String toJSON() const;
63
64 // Saves the JSON representation (from toJSON()) to the given file. Returns false if the
65 // save failed.
66 JS_EXPORT_PRIVATE bool save(const char* filename) const;
67
68 void registerToSaveAtExit(const char* filename);
69
70 JS_EXPORT_PRIVATE void logEvent(CodeBlock* codeBlock, const char* summary, const CString& detail);
71
72private:
73 Bytecodes* ensureBytecodesFor(const AbstractLocker&, CodeBlock*);
74
75 void addDatabaseToAtExit();
76 void removeDatabaseFromAtExit();
77 void performAtExitSave() const;
78 static Database* removeFirstAtExitDatabase();
79 static void atExitCallback();
80
81 int m_databaseID;
82 VM& m_vm;
83 SegmentedVector<Bytecodes> m_bytecodes;
84 HashMap<CodeBlock*, Bytecodes*> m_bytecodesMap;
85 Vector<Ref<Compilation>> m_compilations;
86 HashMap<CodeBlock*, Ref<Compilation>> m_compilationMap;
87 Vector<Event> m_events;
88 bool m_shouldSaveAtExit;
89 CString m_atExitSaveFilename;
90 Database* m_nextRegisteredDatabase;
91 Lock m_lock;
92};
93
94} } // namespace JSC::Profiler
95