1 |
|
|
#include "diagnosticfilename-inl.h" |
2 |
|
|
#include "env-inl.h" |
3 |
|
|
#include "memory_tracker-inl.h" |
4 |
|
|
#include "stream_base-inl.h" |
5 |
|
|
#include "util-inl.h" |
6 |
|
|
|
7 |
|
|
using v8::Array; |
8 |
|
|
using v8::Boolean; |
9 |
|
|
using v8::Context; |
10 |
|
|
using v8::EmbedderGraph; |
11 |
|
|
using v8::EscapableHandleScope; |
12 |
|
|
using v8::FunctionCallbackInfo; |
13 |
|
|
using v8::FunctionTemplate; |
14 |
|
|
using v8::Global; |
15 |
|
|
using v8::HandleScope; |
16 |
|
|
using v8::HeapSnapshot; |
17 |
|
|
using v8::Isolate; |
18 |
|
|
using v8::Local; |
19 |
|
|
using v8::MaybeLocal; |
20 |
|
|
using v8::Number; |
21 |
|
|
using v8::Object; |
22 |
|
|
using v8::ObjectTemplate; |
23 |
|
|
using v8::String; |
24 |
|
|
using v8::Value; |
25 |
|
|
|
26 |
|
|
namespace node { |
27 |
|
|
namespace heap { |
28 |
|
|
|
29 |
|
|
class JSGraphJSNode : public EmbedderGraph::Node { |
30 |
|
|
public: |
31 |
|
|
const char* Name() override { return "<JS Node>"; } |
32 |
|
|
size_t SizeInBytes() override { return 0; } |
33 |
|
|
bool IsEmbedderNode() override { return false; } |
34 |
|
|
Local<Value> JSValue() { return PersistentToLocal::Strong(persistent_); } |
35 |
|
|
|
36 |
|
|
int IdentityHash() { |
37 |
|
|
Local<Value> v = JSValue(); |
38 |
|
|
if (v->IsObject()) return v.As<Object>()->GetIdentityHash(); |
39 |
|
|
if (v->IsName()) return v.As<v8::Name>()->GetIdentityHash(); |
40 |
|
|
if (v->IsInt32()) return v.As<v8::Int32>()->Value(); |
41 |
|
|
return 0; |
42 |
|
|
} |
43 |
|
|
|
44 |
|
|
JSGraphJSNode(Isolate* isolate, Local<Value> val) |
45 |
|
|
: persistent_(isolate, val) { |
46 |
|
|
CHECK(!val.IsEmpty()); |
47 |
|
|
} |
48 |
|
|
|
49 |
|
|
struct Hash { |
50 |
|
|
inline size_t operator()(JSGraphJSNode* n) const { |
51 |
|
|
return static_cast<size_t>(n->IdentityHash()); |
52 |
|
|
} |
53 |
|
|
}; |
54 |
|
|
|
55 |
|
|
struct Equal { |
56 |
|
|
inline bool operator()(JSGraphJSNode* a, JSGraphJSNode* b) const { |
57 |
|
|
return a->JSValue()->SameValue(b->JSValue()); |
58 |
|
|
} |
59 |
|
|
}; |
60 |
|
|
|
61 |
|
|
private: |
62 |
|
|
Global<Value> persistent_; |
63 |
|
|
}; |
64 |
|
|
|
65 |
|
|
class JSGraph : public EmbedderGraph { |
66 |
|
|
public: |
67 |
|
|
explicit JSGraph(Isolate* isolate) : isolate_(isolate) {} |
68 |
|
|
|
69 |
|
|
Node* V8Node(const Local<Value>& value) override { |
70 |
|
|
std::unique_ptr<JSGraphJSNode> n { new JSGraphJSNode(isolate_, value) }; |
71 |
|
|
auto it = engine_nodes_.find(n.get()); |
72 |
|
|
if (it != engine_nodes_.end()) |
73 |
|
|
return *it; |
74 |
|
|
engine_nodes_.insert(n.get()); |
75 |
|
|
return AddNode(std::unique_ptr<Node>(n.release())); |
76 |
|
|
} |
77 |
|
|
|
78 |
|
|
Node* AddNode(std::unique_ptr<Node> node) override { |
79 |
|
|
Node* n = node.get(); |
80 |
|
|
nodes_.emplace(std::move(node)); |
81 |
|
|
return n; |
82 |
|
|
} |
83 |
|
|
|
84 |
|
|
void AddEdge(Node* from, Node* to, const char* name = nullptr) override { |
85 |
|
|
edges_[from].insert(std::make_pair(name, to)); |
86 |
|
|
} |
87 |
|
|
|
88 |
|
|
MaybeLocal<Array> CreateObject() const { |
89 |
|
|
EscapableHandleScope handle_scope(isolate_); |
90 |
|
|
Local<Context> context = isolate_->GetCurrentContext(); |
91 |
|
|
Environment* env = Environment::GetCurrent(context); |
92 |
|
|
|
93 |
|
|
std::unordered_map<Node*, Local<Object>> info_objects; |
94 |
|
|
Local<Array> nodes = Array::New(isolate_, nodes_.size()); |
95 |
|
|
Local<String> edges_string = FIXED_ONE_BYTE_STRING(isolate_, "edges"); |
96 |
|
|
Local<String> is_root_string = FIXED_ONE_BYTE_STRING(isolate_, "isRoot"); |
97 |
|
|
Local<String> name_string = env->name_string(); |
98 |
|
|
Local<String> size_string = env->size_string(); |
99 |
|
|
Local<String> value_string = env->value_string(); |
100 |
|
|
Local<String> wraps_string = FIXED_ONE_BYTE_STRING(isolate_, "wraps"); |
101 |
|
|
Local<String> to_string = FIXED_ONE_BYTE_STRING(isolate_, "to"); |
102 |
|
|
|
103 |
|
|
for (const std::unique_ptr<Node>& n : nodes_) |
104 |
|
|
info_objects[n.get()] = Object::New(isolate_); |
105 |
|
|
|
106 |
|
|
{ |
107 |
|
|
HandleScope handle_scope(isolate_); |
108 |
|
|
size_t i = 0; |
109 |
|
|
for (const std::unique_ptr<Node>& n : nodes_) { |
110 |
|
|
Local<Object> obj = info_objects[n.get()]; |
111 |
|
|
Local<Value> value; |
112 |
|
|
std::string name_str; |
113 |
|
|
const char* prefix = n->NamePrefix(); |
114 |
|
|
if (prefix == nullptr) { |
115 |
|
|
name_str = n->Name(); |
116 |
|
|
} else { |
117 |
|
|
name_str = n->NamePrefix(); |
118 |
|
|
name_str += " "; |
119 |
|
|
name_str += n->Name(); |
120 |
|
|
} |
121 |
|
|
if (!String::NewFromUtf8(isolate_, name_str.c_str()) |
122 |
|
|
.ToLocal(&value) || |
123 |
|
|
obj->Set(context, name_string, value).IsNothing() || |
124 |
|
|
obj->Set(context, |
125 |
|
|
is_root_string, |
126 |
|
|
Boolean::New(isolate_, n->IsRootNode())) |
127 |
|
|
.IsNothing() || |
128 |
|
|
obj->Set(context, |
129 |
|
|
size_string, |
130 |
|
|
Number::New(isolate_, n->SizeInBytes())) |
131 |
|
|
.IsNothing() || |
132 |
|
|
obj->Set(context, edges_string, Array::New(isolate_)).IsNothing()) { |
133 |
|
|
return MaybeLocal<Array>(); |
134 |
|
|
} |
135 |
|
|
if (nodes->Set(context, i++, obj).IsNothing()) |
136 |
|
|
return MaybeLocal<Array>(); |
137 |
|
|
if (!n->IsEmbedderNode()) { |
138 |
|
|
value = static_cast<JSGraphJSNode*>(n.get())->JSValue(); |
139 |
|
|
if (obj->Set(context, value_string, value).IsNothing()) |
140 |
|
|
return MaybeLocal<Array>(); |
141 |
|
|
} |
142 |
|
|
} |
143 |
|
|
} |
144 |
|
|
|
145 |
|
|
for (const std::unique_ptr<Node>& n : nodes_) { |
146 |
|
|
Node* wraps = n->WrapperNode(); |
147 |
|
|
if (wraps == nullptr) continue; |
148 |
|
|
Local<Object> from = info_objects[n.get()]; |
149 |
|
|
Local<Object> to = info_objects[wraps]; |
150 |
|
|
if (from->Set(context, wraps_string, to).IsNothing()) |
151 |
|
|
return MaybeLocal<Array>(); |
152 |
|
|
} |
153 |
|
|
|
154 |
|
|
for (const auto& edge_info : edges_) { |
155 |
|
|
Node* source = edge_info.first; |
156 |
|
|
Local<Value> edges; |
157 |
|
|
if (!info_objects[source]->Get(context, edges_string).ToLocal(&edges) || |
158 |
|
|
!edges->IsArray()) { |
159 |
|
|
return MaybeLocal<Array>(); |
160 |
|
|
} |
161 |
|
|
|
162 |
|
|
size_t i = 0; |
163 |
|
|
size_t j = 0; |
164 |
|
|
for (const auto& edge : edge_info.second) { |
165 |
|
|
Local<Object> to_object = info_objects[edge.second]; |
166 |
|
|
Local<Object> edge_obj = Object::New(isolate_); |
167 |
|
|
Local<Value> edge_name_value; |
168 |
|
|
const char* edge_name = edge.first; |
169 |
|
|
if (edge_name != nullptr) { |
170 |
|
|
if (!String::NewFromUtf8(isolate_, edge_name) |
171 |
|
|
.ToLocal(&edge_name_value)) { |
172 |
|
|
return MaybeLocal<Array>(); |
173 |
|
|
} |
174 |
|
|
} else { |
175 |
|
|
edge_name_value = Number::New(isolate_, j++); |
176 |
|
|
} |
177 |
|
|
if (edge_obj->Set(context, name_string, edge_name_value).IsNothing() || |
178 |
|
|
edge_obj->Set(context, to_string, to_object).IsNothing() || |
179 |
|
|
edges.As<Array>()->Set(context, i++, edge_obj).IsNothing()) { |
180 |
|
|
return MaybeLocal<Array>(); |
181 |
|
|
} |
182 |
|
|
} |
183 |
|
|
} |
184 |
|
|
|
185 |
|
|
return handle_scope.Escape(nodes); |
186 |
|
|
} |
187 |
|
|
|
188 |
|
|
private: |
189 |
|
|
Isolate* isolate_; |
190 |
|
|
std::unordered_set<std::unique_ptr<Node>> nodes_; |
191 |
|
|
std::unordered_set<JSGraphJSNode*, JSGraphJSNode::Hash, JSGraphJSNode::Equal> |
192 |
|
|
engine_nodes_; |
193 |
|
|
std::unordered_map<Node*, std::set<std::pair<const char*, Node*>>> edges_; |
194 |
|
|
}; |
195 |
|
|
|
196 |
|
|
void BuildEmbedderGraph(const FunctionCallbackInfo<Value>& args) { |
197 |
|
|
Environment* env = Environment::GetCurrent(args); |
198 |
|
|
JSGraph graph(env->isolate()); |
199 |
|
|
Environment::BuildEmbedderGraph(env->isolate(), &graph, env); |
200 |
|
|
Local<Array> ret; |
201 |
|
|
if (graph.CreateObject().ToLocal(&ret)) |
202 |
|
|
args.GetReturnValue().Set(ret); |
203 |
|
|
} |
204 |
|
|
|
205 |
|
|
namespace { |
206 |
|
5 |
class FileOutputStream : public v8::OutputStream { |
207 |
|
|
public: |
208 |
|
5 |
explicit FileOutputStream(FILE* stream) : stream_(stream) {} |
209 |
|
|
|
210 |
|
10 |
int GetChunkSize() override { |
211 |
|
10 |
return 65536; // big chunks == faster |
212 |
|
|
} |
213 |
|
|
|
214 |
|
5 |
void EndOfStream() override {} |
215 |
|
|
|
216 |
|
289 |
WriteResult WriteAsciiChunk(char* data, int size) override { |
217 |
|
289 |
const size_t len = static_cast<size_t>(size); |
218 |
|
289 |
size_t off = 0; |
219 |
|
|
|
220 |
✓✓✓✗ ✓✗✓✓
|
867 |
while (off < len && !feof(stream_) && !ferror(stream_)) |
221 |
|
289 |
off += fwrite(data + off, 1, len - off, stream_); |
222 |
|
|
|
223 |
✓✗ |
289 |
return off == len ? kContinue : kAbort; |
224 |
|
|
} |
225 |
|
|
|
226 |
|
|
private: |
227 |
|
|
FILE* stream_; |
228 |
|
|
}; |
229 |
|
|
|
230 |
|
|
class HeapSnapshotStream : public AsyncWrap, |
231 |
|
|
public StreamBase, |
232 |
|
|
public v8::OutputStream { |
233 |
|
|
public: |
234 |
|
14 |
HeapSnapshotStream( |
235 |
|
|
Environment* env, |
236 |
|
|
HeapSnapshotPointer&& snapshot, |
237 |
|
14 |
Local<Object> obj) : |
238 |
|
|
AsyncWrap(env, obj, AsyncWrap::PROVIDER_HEAPSNAPSHOT), |
239 |
|
|
StreamBase(env), |
240 |
|
14 |
snapshot_(std::move(snapshot)) { |
241 |
|
14 |
MakeWeak(); |
242 |
|
14 |
StreamBase::AttachToObject(GetObject()); |
243 |
|
14 |
} |
244 |
|
|
|
245 |
|
28 |
~HeapSnapshotStream() override {} |
246 |
|
|
|
247 |
|
22 |
int GetChunkSize() override { |
248 |
|
22 |
return 65536; // big chunks == faster |
249 |
|
|
} |
250 |
|
|
|
251 |
|
11 |
void EndOfStream() override { |
252 |
|
11 |
EmitRead(UV_EOF); |
253 |
|
11 |
snapshot_.reset(); |
254 |
|
11 |
} |
255 |
|
|
|
256 |
|
601 |
WriteResult WriteAsciiChunk(char* data, int size) override { |
257 |
|
601 |
int len = size; |
258 |
✓✓ |
1803 |
while (len != 0) { |
259 |
|
601 |
uv_buf_t buf = EmitAlloc(size); |
260 |
|
601 |
ssize_t avail = len; |
261 |
✗✓ |
601 |
if (static_cast<ssize_t>(buf.len) < avail) |
262 |
|
|
avail = buf.len; |
263 |
|
601 |
memcpy(buf.base, data, avail); |
264 |
|
601 |
data += avail; |
265 |
|
601 |
len -= avail; |
266 |
|
601 |
EmitRead(size, buf); |
267 |
|
|
} |
268 |
|
601 |
return kContinue; |
269 |
|
|
} |
270 |
|
|
|
271 |
|
11 |
int ReadStart() override { |
272 |
✗✓ |
11 |
CHECK_NE(snapshot_, nullptr); |
273 |
|
11 |
snapshot_->Serialize(this, HeapSnapshot::kJSON); |
274 |
|
11 |
return 0; |
275 |
|
|
} |
276 |
|
|
|
277 |
|
612 |
int ReadStop() override { |
278 |
|
612 |
return 0; |
279 |
|
|
} |
280 |
|
|
|
281 |
|
|
int DoShutdown(ShutdownWrap* req_wrap) override { |
282 |
|
|
UNREACHABLE(); |
283 |
|
|
} |
284 |
|
|
|
285 |
|
|
int DoWrite(WriteWrap* w, |
286 |
|
|
uv_buf_t* bufs, |
287 |
|
|
size_t count, |
288 |
|
|
uv_stream_t* send_handle) override { |
289 |
|
|
UNREACHABLE(); |
290 |
|
|
} |
291 |
|
|
|
292 |
|
623 |
bool IsAlive() override { return snapshot_ != nullptr; } |
293 |
|
|
bool IsClosing() override { return snapshot_ == nullptr; } |
294 |
|
1249 |
AsyncWrap* GetAsyncWrap() override { return this; } |
295 |
|
|
|
296 |
|
7 |
void MemoryInfo(MemoryTracker* tracker) const override { |
297 |
✓✗ |
7 |
if (snapshot_ != nullptr) { |
298 |
|
|
tracker->TrackFieldWithSize( |
299 |
|
7 |
"snapshot", sizeof(*snapshot_), "HeapSnapshot"); |
300 |
|
|
} |
301 |
|
7 |
} |
302 |
|
|
|
303 |
|
7 |
SET_MEMORY_INFO_NAME(HeapSnapshotStream) |
304 |
|
7 |
SET_SELF_SIZE(HeapSnapshotStream) |
305 |
|
|
|
306 |
|
|
private: |
307 |
|
|
HeapSnapshotPointer snapshot_; |
308 |
|
|
}; |
309 |
|
|
|
310 |
|
5 |
inline void TakeSnapshot(Isolate* isolate, v8::OutputStream* out) { |
311 |
|
|
HeapSnapshotPointer snapshot { |
312 |
|
10 |
isolate->GetHeapProfiler()->TakeHeapSnapshot() }; |
313 |
|
5 |
snapshot->Serialize(out, HeapSnapshot::kJSON); |
314 |
|
5 |
} |
315 |
|
|
|
316 |
|
|
} // namespace |
317 |
|
|
|
318 |
|
5 |
bool WriteSnapshot(Isolate* isolate, const char* filename) { |
319 |
|
5 |
FILE* fp = fopen(filename, "w"); |
320 |
✗✓ |
5 |
if (fp == nullptr) |
321 |
|
|
return false; |
322 |
|
10 |
FileOutputStream stream(fp); |
323 |
|
5 |
TakeSnapshot(isolate, &stream); |
324 |
|
5 |
fclose(fp); |
325 |
|
5 |
return true; |
326 |
|
|
} |
327 |
|
|
|
328 |
|
19 |
void DeleteHeapSnapshot(const HeapSnapshot* snapshot) { |
329 |
|
19 |
const_cast<HeapSnapshot*>(snapshot)->Delete(); |
330 |
|
19 |
} |
331 |
|
|
|
332 |
|
14 |
BaseObjectPtr<AsyncWrap> CreateHeapSnapshotStream( |
333 |
|
|
Environment* env, HeapSnapshotPointer&& snapshot) { |
334 |
|
28 |
HandleScope scope(env->isolate()); |
335 |
|
|
|
336 |
✓✓ |
28 |
if (env->streambaseoutputstream_constructor_template().IsEmpty()) { |
337 |
|
|
// Create FunctionTemplate for HeapSnapshotStream |
338 |
|
5 |
Local<FunctionTemplate> os = FunctionTemplate::New(env->isolate()); |
339 |
|
10 |
os->Inherit(AsyncWrap::GetConstructorTemplate(env)); |
340 |
|
5 |
Local<ObjectTemplate> ost = os->InstanceTemplate(); |
341 |
|
5 |
ost->SetInternalFieldCount(StreamBase::kInternalFieldCount); |
342 |
|
10 |
os->SetClassName( |
343 |
|
5 |
FIXED_ONE_BYTE_STRING(env->isolate(), "HeapSnapshotStream")); |
344 |
|
5 |
StreamBase::AddMethods(env, os); |
345 |
|
5 |
env->set_streambaseoutputstream_constructor_template(ost); |
346 |
|
|
} |
347 |
|
|
|
348 |
|
|
Local<Object> obj; |
349 |
✗✓ |
42 |
if (!env->streambaseoutputstream_constructor_template() |
350 |
|
42 |
->NewInstance(env->context()) |
351 |
|
14 |
.ToLocal(&obj)) { |
352 |
|
|
return {}; |
353 |
|
|
} |
354 |
|
14 |
return MakeBaseObject<HeapSnapshotStream>(env, std::move(snapshot), obj); |
355 |
|
|
} |
356 |
|
|
|
357 |
|
14 |
void CreateHeapSnapshotStream(const FunctionCallbackInfo<Value>& args) { |
358 |
|
14 |
Environment* env = Environment::GetCurrent(args); |
359 |
|
|
HeapSnapshotPointer snapshot { |
360 |
|
28 |
env->isolate()->GetHeapProfiler()->TakeHeapSnapshot() }; |
361 |
✗✓ |
14 |
CHECK(snapshot); |
362 |
|
|
BaseObjectPtr<AsyncWrap> stream = |
363 |
|
28 |
CreateHeapSnapshotStream(env, std::move(snapshot)); |
364 |
✓✗ |
14 |
if (stream) |
365 |
|
42 |
args.GetReturnValue().Set(stream->object()); |
366 |
|
14 |
} |
367 |
|
|
|
368 |
|
4 |
void TriggerHeapSnapshot(const FunctionCallbackInfo<Value>& args) { |
369 |
|
4 |
Environment* env = Environment::GetCurrent(args); |
370 |
|
4 |
Isolate* isolate = args.GetIsolate(); |
371 |
|
|
|
372 |
|
4 |
Local<Value> filename_v = args[0]; |
373 |
|
|
|
374 |
✓✓ |
8 |
if (filename_v->IsUndefined()) { |
375 |
|
6 |
DiagnosticFilename name(env, "Heap", "heapsnapshot"); |
376 |
✗✓ |
3 |
if (!WriteSnapshot(isolate, *name)) |
377 |
|
|
return; |
378 |
✓✗ |
6 |
if (String::NewFromUtf8(isolate, *name).ToLocal(&filename_v)) { |
379 |
|
6 |
args.GetReturnValue().Set(filename_v); |
380 |
|
|
} |
381 |
|
3 |
return; |
382 |
|
|
} |
383 |
|
|
|
384 |
|
2 |
BufferValue path(isolate, filename_v); |
385 |
✗✓ |
1 |
CHECK_NOT_NULL(*path); |
386 |
✗✓ |
1 |
if (!WriteSnapshot(isolate, *path)) |
387 |
|
|
return; |
388 |
|
2 |
return args.GetReturnValue().Set(filename_v); |
389 |
|
|
} |
390 |
|
|
|
391 |
|
623 |
void Initialize(Local<Object> target, |
392 |
|
|
Local<Value> unused, |
393 |
|
|
Local<Context> context, |
394 |
|
|
void* priv) { |
395 |
|
623 |
Environment* env = Environment::GetCurrent(context); |
396 |
|
|
|
397 |
|
623 |
env->SetMethod(target, "buildEmbedderGraph", BuildEmbedderGraph); |
398 |
|
623 |
env->SetMethod(target, "triggerHeapSnapshot", TriggerHeapSnapshot); |
399 |
|
623 |
env->SetMethod(target, "createHeapSnapshotStream", CreateHeapSnapshotStream); |
400 |
|
623 |
} |
401 |
|
|
|
402 |
|
|
} // namespace heap |
403 |
|
|
} // namespace node |
404 |
|
|
|
405 |
✓✗✓✗
|
18704 |
NODE_MODULE_CONTEXT_AWARE_INTERNAL(heap_utils, node::heap::Initialize) |