1 #include "node_sea.h"
2
3 #include "env-inl.h"
4 #include "node_external_reference.h"
5 #include "node_internals.h"
6 #include "node_union_bytes.h"
7 #include "simdutf.h"
8 #include "v8.h"
9
10 // The POSTJECT_SENTINEL_FUSE macro is a string of random characters selected by
11 // the Node.js project that is present only once in the entire binary. It is
12 // used by the postject_has_resource() function to efficiently detect if a
13 // resource has been injected. See
14 // https://github.com/nodejs/postject/blob/35343439cac8c488f2596d7c4c1dddfec1fddcae/postject-api.h#L42-L45.
15 #define POSTJECT_SENTINEL_FUSE "NODE_JS_FUSE_fce680ab2cc467b6e072b8b5df1996b2"
16 #include "postject-api.h"
17 #undef POSTJECT_SENTINEL_FUSE
18
19 #include <memory>
20 #include <string_view>
21 #include <tuple>
22 #include <vector>
23
24 #if !defined(DISABLE_SINGLE_EXECUTABLE_APPLICATION)
25
26 using v8::Context;
27 using v8::FunctionCallbackInfo;
28 using v8::Local;
29 using v8::Object;
30 using v8::Value;
31
32 namespace {
33
FindSingleExecutableCode()34 const std::string_view FindSingleExecutableCode() {
35 static const std::string_view sea_code = []() -> std::string_view {
36 size_t size;
37 #ifdef __APPLE__
38 postject_options options;
39 postject_options_init(&options);
40 options.macho_segment_name = "NODE_JS";
41 const char* code = static_cast<const char*>(
42 postject_find_resource("NODE_JS_CODE", &size, &options));
43 #else
44 const char* code = static_cast<const char*>(
45 postject_find_resource("NODE_JS_CODE", &size, nullptr));
46 #endif
47 return {code, size};
48 }();
49 return sea_code;
50 }
51
GetSingleExecutableCode(const FunctionCallbackInfo<Value> & args)52 void GetSingleExecutableCode(const FunctionCallbackInfo<Value>& args) {
53 node::Environment* env = node::Environment::GetCurrent(args);
54
55 static const std::string_view sea_code = FindSingleExecutableCode();
56
57 if (sea_code.empty()) {
58 return;
59 }
60
61 // TODO(joyeecheung): Use one-byte strings for ASCII-only source to save
62 // memory/binary size - using UTF16 by default results in twice of the size
63 // than necessary.
64 static const node::UnionBytes sea_code_union_bytes =
65 []() -> node::UnionBytes {
66 size_t expected_u16_length =
67 simdutf::utf16_length_from_utf8(sea_code.data(), sea_code.size());
68 auto out = std::make_shared<std::vector<uint16_t>>(expected_u16_length);
69 size_t u16_length = simdutf::convert_utf8_to_utf16(
70 sea_code.data(),
71 sea_code.size(),
72 reinterpret_cast<char16_t*>(out->data()));
73 out->resize(u16_length);
74 return node::UnionBytes{out};
75 }();
76
77 args.GetReturnValue().Set(
78 sea_code_union_bytes.ToStringChecked(env->isolate()));
79 }
80
81 } // namespace
82
83 namespace node {
84 namespace sea {
85
IsSingleExecutable()86 bool IsSingleExecutable() {
87 return postject_has_resource();
88 }
89
FixupArgsForSEA(int argc,char ** argv)90 std::tuple<int, char**> FixupArgsForSEA(int argc, char** argv) {
91 // Repeats argv[0] at position 1 on argv as a replacement for the missing
92 // entry point file path.
93 if (IsSingleExecutable()) {
94 static std::vector<char*> new_argv;
95 new_argv.reserve(argc + 2);
96 new_argv.emplace_back(argv[0]);
97 new_argv.insert(new_argv.end(), argv, argv + argc);
98 new_argv.emplace_back(nullptr);
99 argc = new_argv.size() - 1;
100 argv = new_argv.data();
101 }
102
103 return {argc, argv};
104 }
105
Initialize(Local<Object> target,Local<Value> unused,Local<Context> context,void * priv)106 void Initialize(Local<Object> target,
107 Local<Value> unused,
108 Local<Context> context,
109 void* priv) {
110 SetMethod(
111 context, target, "getSingleExecutableCode", GetSingleExecutableCode);
112 }
113
RegisterExternalReferences(ExternalReferenceRegistry * registry)114 void RegisterExternalReferences(ExternalReferenceRegistry* registry) {
115 registry->Register(GetSingleExecutableCode);
116 }
117
118 } // namespace sea
119 } // namespace node
120
121 NODE_BINDING_CONTEXT_AWARE_INTERNAL(sea, node::sea::Initialize)
122 NODE_BINDING_EXTERNAL_REFERENCE(sea, node::sea::RegisterExternalReferences)
123
124 #endif // !defined(DISABLE_SINGLE_EXECUTABLE_APPLICATION)
125