JSC Heap snapshot and capture

Reviewed By: astreet

Differential Revision: D2699681

fb-gh-sync-id: 81617187abec6d9bc1ffe3da02513c6875d0b93b

ReactAndroid/src/main/jni/react/JSCExecutor.cpp

@@ -30,6 +30,7 @@ using fbsystrace::FbSystraceSection;
 
 // Add native performance markers support
 #include <react/JSCPerfLogging.h>
+#include <react/JSCMemory.h>
 
 #ifdef WITH_FB_JSC_TUNING
 #include <jsc_config_android.h>
@@ -132,6 +133,8 @@ JSCExecutor::JSCExecutor(FlushImmediateCallback cb) :
   addNativeProfilingHooks(m_context);
   addNativePerfLoggingHooks(m_context);
   #endif
+
+  addNativeMemoryHooks(m_context);
 }
 
 JSCExecutor::~JSCExecutor() {

ReactAndroid/src/main/jni/react/JSCMemory.cpp

@@ -0,0 +1,157 @@
+// Copyright 2004-present Facebook. All Rights Reserved.
+
+#include <stdio.h>
+#include <string.h>
+#include <JavaScriptCore/JavaScript.h>
+#include <JavaScriptCore/API/JSProfilerPrivate.h>
+#include "JSCHelpers.h"
+
+#include "Value.h"
+
+#if WITH_FB_MEMORY_PROFILING
+
+static JSValueRef nativeEnableAllocationTag(
+    JSContextRef ctx,
+    JSObjectRef function,
+    JSObjectRef thisObject,
+    size_t argumentCount,
+    const JSValueRef arguments[],
+    JSValueRef* exception) {
+
+  if (argumentCount < 1) {
+    if (exception) {
+      *exception = facebook::react::makeJSCException(
+        ctx,
+        "nativeEnableAllocationTag requires a single boolean argument");
+    }
+    return JSValueMakeUndefined(ctx);
+  }
+
+  JSEnableAllocationTag(ctx, JSValueToBoolean(ctx, arguments[0]));
+  return JSValueMakeUndefined(ctx);
+}
+
+static JSValueRef nativeAllocationPushTag(
+    JSContextRef ctx,
+    JSObjectRef function,
+    JSObjectRef thisObject,
+    size_t argumentCount,
+    const JSValueRef arguments[],
+    JSValueRef* exception) {
+  std::string marker;
+
+  if (argumentCount < 1) {
+    if (exception) {
+      *exception = facebook::react::makeJSCException(
+        ctx,
+        "nativeAllocationPushTag requires at least 1 argument");
+    }
+    return JSValueMakeUndefined(ctx);
+  }
+
+  JSStringRef tag = JSValueToStringCopy(ctx, arguments[0], exception);
+  JSPushAllocationTag(ctx, facebook::react::String::ref(tag).str().c_str());
+  JSStringRelease(tag);
+  return JSValueMakeUndefined(ctx);
+}
+
+static JSValueRef nativeAllocationPopTag(
+    JSContextRef ctx,
+    JSObjectRef function,
+    JSObjectRef thisObject,
+    size_t argumentCount,
+    const JSValueRef arguments[],
+    JSValueRef* exception) {
+  JSPopAllocationTag(ctx);
+  return JSValueMakeUndefined(ctx);
+}
+
+static JSValueRef nativeForceSyncGC(
+    JSContextRef ctx,
+    JSObjectRef function,
+    JSObjectRef thisObject,
+    size_t argumentCount,
+    const JSValueRef arguments[],
+    JSValueRef* exception) {
+  JSSynchronousGarbageCollectForDebugging(ctx);
+  return JSValueMakeUndefined(ctx);
+}
+
+static JSValueRef nativeCaptureStart(
+    JSContextRef ctx,
+    JSObjectRef function,
+    JSObjectRef thisObject,
+    size_t argumentCount,
+    const JSValueRef arguments[],
+    JSValueRef* exception) {
+  if (argumentCount < 1) {
+    if (exception) {
+      *exception = facebook::react::makeJSCException(
+        ctx,
+        "nativeCaptureStart requires at least 1 argument");
+    }
+    return JSValueMakeUndefined(ctx);
+  }
+
+  JSStringRef outputFilename = JSValueToStringCopy(ctx, arguments[0], exception);
+  std::string finalFilename =
+    std::string("/sdcard/") +
+    facebook::react::String::ref(outputFilename).str();
+  JSHeapCaptureStart(ctx, finalFilename.c_str());
+  JSStringRelease(outputFilename);
+  return JSValueMakeUndefined(ctx);
+}
+
+static JSValueRef nativeCaptureEnd(
+    JSContextRef ctx,
+    JSObjectRef function,
+    JSObjectRef thisObject,
+    size_t argumentCount,
+    const JSValueRef arguments[],
+    JSValueRef* exception) {
+  JSHeapCaptureEnd(ctx);
+  return JSValueMakeUndefined(ctx);
+}
+
+static JSValueRef nativeHeapDump(
+    JSContextRef ctx,
+    JSObjectRef function,
+    JSObjectRef thisObject,
+    size_t argumentCount,
+    const JSValueRef arguments[],
+    JSValueRef* exception) {
+  if (argumentCount < 1) {
+    if (exception) {
+      *exception = facebook::react::makeJSCException(
+        ctx,
+        "nativeHeapDump requires at least 1 argument");
+    }
+    return JSValueMakeUndefined(ctx);
+  }
+
+  JSStringRef outputFilename = JSValueToStringCopy(ctx, arguments[0], exception);
+  std::string finalFilename =
+    std::string("/sdcard/") +
+    facebook::react::String::ref(outputFilename).str();
+  JSHeapDump(ctx, finalFilename.c_str());
+  JSStringRelease(outputFilename);
+  return JSValueMakeUndefined(ctx);
+}
+#endif
+
+namespace facebook {
+namespace react {
+
+void addNativeMemoryHooks(JSGlobalContextRef ctx) {
+#if WITH_FB_MEMORY_PROFILING
+  installGlobalFunction(ctx, "nativeEnableAllocationTag", nativeEnableAllocationTag);
+  installGlobalFunction(ctx, "nativeAllocationPushTag", nativeAllocationPushTag);
+  installGlobalFunction(ctx, "nativeAllocationPopTag", nativeAllocationPopTag);
+  installGlobalFunction(ctx, "nativeForceSyncGC", nativeForceSyncGC);
+  installGlobalFunction(ctx, "nativeCaptureStart", nativeCaptureStart);
+  installGlobalFunction(ctx, "nativeCaptureEnd", nativeCaptureEnd);
+  installGlobalFunction(ctx, "nativeHeapDump", nativeHeapDump);
+#endif
+}
+
+} }

ReactAndroid/src/main/jni/react/JSCMemory.h

@@ -0,0 +1,11 @@
+// Copyright 2004-present Facebook. All Rights Reserved.
+
+#pragma once
+
+#include <JavaScriptCore/JSContextRef.h>
+namespace facebook {
+namespace react {
+
+void addNativeMemoryHooks(JSGlobalContextRef ctx);
+
+} }