From 07b9ef7fb7a29676da4f260bcc66d65bd12de461 Mon Sep 17 00:00:00 2001
From: "Wisniewski, Krzysztof2" <Krzysztof2.Wisniewski@intel.com>
Date: Tue, 9 Aug 2022 15:28:51 +0200
Subject: [PATCH] Add script which injects DSO into NodeJS process

---
 gprofiler/profilers/node.py | 122 ++++++++++++++++++++++++++++++++++++
 requirements.txt            |   2 +
 2 files changed, 124 insertions(+)
 create mode 100644 gprofiler/profilers/node.py

diff --git a/gprofiler/profilers/node.py b/gprofiler/profilers/node.py
new file mode 100644
index 000000000..58094cb0b
--- /dev/null
+++ b/gprofiler/profilers/node.py
@@ -0,0 +1,122 @@
+#
+# Copyright (c) Granulate. All rights reserved.
+# Licensed under the AGPL3 License. See LICENSE.md in the project root for license information.
+#
+"""
+This script allows user to inject node-linux-perf module and start it to generate perf maps at runtime.
+To prepare module run npm install --prefix <location> linux-perf or clone it from https://github.com/mmarchini-oss/node-linux-perf and run npm install 
+This only works for applications, that have CommonJS as entry script.
+
+It can be also easly modified to utilize any other module by changing expression in CDP request.
+"""
+
+import asyncio
+import websockets
+import json
+import requests
+import os
+import signal
+import time
+import psutil
+import argparse
+
+EXPECTED_CDP_RESPONSE = {
+    "id": 1,
+    "result": {"result": {"type": "boolean", "value": True}},
+}
+
+
+class NodeDebuggerUrlNotFound(Exception):
+    pass
+
+
+class NodeDebuggerUnexpectedResponse(Exception):
+    pass
+
+
+def retry(exception, retries, waitTime):
+    def wrap(f):
+        def wrapped(*args):
+            for _ in range(0, retries):
+                try:
+                    return f(*args)
+                except exception:
+                    time.sleep(waitTime)
+
+        return wrapped
+
+    return wrap
+
+
+def __start_debugger(pid):
+    os.kill(pid, signal.SIGUSR1)
+
+
+@retry(NodeDebuggerUrlNotFound, 5, 1)
+def __get_debugger_url(pid):
+    process = psutil.Process(pid)
+    possible_ports = [connection.laddr.port for connection in process.connections()]
+    debugger_url = ""
+    for port in possible_ports:
+        possible_port_response = requests.get(f"http://127.0.0.1:{port}/json/list")
+        if (
+            possible_port_response.status_code != 200
+            or not "application/json"
+            in possible_port_response.headers.get("Content-Type")
+        ):
+            continue
+
+        response_json = possible_port_response.json()
+        if (
+            not isinstance(response_json, list)
+            or len(response_json) == 0
+            or not isinstance(response_json[0], dict)
+            or not "webSocketDebuggerUrl" in response_json[0].keys()
+        ):
+            continue
+
+        debugger_url = response_json[0]["webSocketDebuggerUrl"]
+    if not debugger_url:
+        raise NodeDebuggerUrlNotFound
+    return debugger_url
+
+
+@retry(NodeDebuggerUnexpectedResponse, 5, 1)
+async def __load_dso(sock, module_path):
+    cdp_request = {
+        "id": 1,
+        "method": "Runtime.evaluate",
+        "params": {
+            "expression": f'process.mainModule.require("{module_path}").start()',
+            "replMode": True,
+        },
+    }
+    await sock.send(json.dumps(cdp_request))
+    message = await sock.recv()
+    if json.loads(message) != EXPECTED_CDP_RESPONSE:
+        raise NodeDebuggerUnexpectedResponse(json.loads(message))
+
+
+async def generate_perf_map(pid, module_path):
+    __start_debugger(pid)
+    debugger_url = __get_debugger_url(pid)
+    async with websockets.connect(debugger_url) as sock:
+        await __load_dso(sock, module_path)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument(
+        "-p",
+        "--pid",
+        type=int,
+        help="PID of nodejs process to which DSO should be injected",
+    )
+    parser.add_argument(
+        "-m",
+        "--module_path",
+        type=str,
+        help="Path to compiled node-linux-perf module. Module must be compiled for specific nodejs version",
+    )
+    args = parser.parse_args()
+    asyncio.run(generate_perf_map(args.pid, args.module_path))
diff --git a/requirements.txt b/requirements.txt
index 32bbe0bb6..b554ac3ae 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,4 +9,6 @@ dataclasses==0.8; python_version < '3.7'
 packaging==21.2
 pyelftools==0.28
 curlify==2.2.1
+websockets==10.3
+asyncio==3.4.3
 ./granulate-utils/