734 lines
22 KiB
Diff
734 lines
22 KiB
Diff
--- node/deps/v8/include/v8.h
|
|
+++ node/deps/v8/include/v8.h
|
|
@@ -6065,10 +6065,14 @@
|
|
*/
|
|
static void SetFlagsFromCommandLine(int* argc,
|
|
char** argv,
|
|
bool remove_flags);
|
|
|
|
+ static void EnableCompilationForSourcelessUse();
|
|
+ static void DisableCompilationForSourcelessUse();
|
|
+ static void FixSourcelessScript(Isolate* v8_isolate, Local<UnboundScript> script);
|
|
+
|
|
/** Get the version string. */
|
|
static const char* GetVersion();
|
|
|
|
/** Callback function for reporting failed access checks.*/
|
|
V8_INLINE static V8_DEPRECATE_SOON(
|
|
--- node/deps/v8/src/api.cc
|
|
+++ node/deps/v8/src/api.cc
|
|
@@ -428,10 +428,46 @@
|
|
void V8::SetFlagsFromCommandLine(int* argc, char** argv, bool remove_flags) {
|
|
i::FlagList::SetFlagsFromCommandLine(argc, argv, remove_flags);
|
|
}
|
|
|
|
|
|
+bool save_lazy;
|
|
+bool save_predictable;
|
|
+bool save_serialize_toplevel;
|
|
+
|
|
+
|
|
+void V8::EnableCompilationForSourcelessUse() {
|
|
+ save_lazy = i::FLAG_lazy;
|
|
+ i::FLAG_lazy = false;
|
|
+ save_predictable = i::FLAG_predictable;
|
|
+ i::FLAG_predictable = true;
|
|
+ save_serialize_toplevel = i::FLAG_serialize_toplevel;
|
|
+ i::FLAG_serialize_toplevel = true;
|
|
+ i::CpuFeatures::Reinitialize();
|
|
+ i::CpuFeatures::Probe(true);
|
|
+}
|
|
+
|
|
+
|
|
+void V8::DisableCompilationForSourcelessUse() {
|
|
+ i::FLAG_lazy = save_lazy;
|
|
+ i::FLAG_predictable = save_predictable;
|
|
+ i::FLAG_serialize_toplevel = save_serialize_toplevel;
|
|
+ i::CpuFeatures::Reinitialize();
|
|
+ i::CpuFeatures::Probe(false);
|
|
+}
|
|
+
|
|
+
|
|
+void V8::FixSourcelessScript(Isolate* v8_isolate, Local<UnboundScript> script) {
|
|
+ auto isolate = reinterpret_cast<i::Isolate*>(v8_isolate);
|
|
+ auto object = i::Handle<i::HeapObject>::cast(Utils::OpenHandle(*script));
|
|
+ i::Handle<i::SharedFunctionInfo> function_info(
|
|
+ i::SharedFunctionInfo::cast(*object), object->GetIsolate());
|
|
+ auto s = reinterpret_cast<i::Script*>(function_info->script());
|
|
+ s->set_source(isolate->heap()->undefined_value());
|
|
+}
|
|
+
|
|
+
|
|
RegisteredExtension* RegisteredExtension::first_extension_ = NULL;
|
|
|
|
|
|
RegisteredExtension::RegisteredExtension(Extension* extension)
|
|
: extension_(extension) { }
|
|
--- node/deps/v8/src/assembler.h
|
|
+++ node/deps/v8/src/assembler.h
|
|
@@ -223,10 +223,15 @@
|
|
static void PrintFeatures();
|
|
|
|
// Flush instruction cache.
|
|
static void FlushICache(void* start, size_t size);
|
|
|
|
+ static void Reinitialize() {
|
|
+ supported_ = 0;
|
|
+ initialized_ = false;
|
|
+ }
|
|
+
|
|
private:
|
|
// Platform-dependent implementation.
|
|
static void ProbeImpl(bool cross_compile);
|
|
|
|
static unsigned supported_;
|
|
--- node/deps/v8/src/parser.cc
|
|
+++ node/deps/v8/src/parser.cc
|
|
@@ -5636,10 +5636,11 @@
|
|
return false;
|
|
}
|
|
|
|
|
|
bool Parser::Parse(ParseInfo* info) {
|
|
+ if (info->script()->source()->IsUndefined()) return false;
|
|
DCHECK(info->function() == NULL);
|
|
FunctionLiteral* result = NULL;
|
|
// Ok to use Isolate here; this function is only called in the main thread.
|
|
DCHECK(parsing_on_main_thread_);
|
|
Isolate* isolate = info->isolate();
|
|
--- node/deps/v8/src/snapshot/serialize.cc
|
|
+++ node/deps/v8/src/snapshot/serialize.cc
|
|
@@ -2695,24 +2695,36 @@
|
|
|
|
|
|
SerializedCodeData::SanityCheckResult SerializedCodeData::SanityCheck(
|
|
Isolate* isolate, String* source) const {
|
|
uint32_t magic_number = GetMagicNumber();
|
|
- if (magic_number != ComputeMagicNumber(isolate)) return MAGIC_NUMBER_MISMATCH;
|
|
+ if (magic_number != ComputeMagicNumber(isolate)) {
|
|
+ base::OS::PrintError("Pkg: MAGIC_NUMBER_MISMATCH\n");
|
|
+ return MAGIC_NUMBER_MISMATCH;
|
|
+ }
|
|
uint32_t version_hash = GetHeaderValue(kVersionHashOffset);
|
|
- uint32_t source_hash = GetHeaderValue(kSourceHashOffset);
|
|
uint32_t cpu_features = GetHeaderValue(kCpuFeaturesOffset);
|
|
uint32_t flags_hash = GetHeaderValue(kFlagHashOffset);
|
|
uint32_t c1 = GetHeaderValue(kChecksum1Offset);
|
|
uint32_t c2 = GetHeaderValue(kChecksum2Offset);
|
|
- if (version_hash != Version::Hash()) return VERSION_MISMATCH;
|
|
- if (source_hash != SourceHash(source)) return SOURCE_MISMATCH;
|
|
- if (cpu_features != static_cast<uint32_t>(CpuFeatures::SupportedFeatures())) {
|
|
+ if (version_hash != Version::Hash()) {
|
|
+ base::OS::PrintError("Pkg: VERSION_MISMATCH\n");
|
|
+ return VERSION_MISMATCH;
|
|
+ }
|
|
+ uint32_t host_features = static_cast<uint32_t>(CpuFeatures::SupportedFeatures());
|
|
+ if (cpu_features & (~host_features)) {
|
|
+ base::OS::PrintError("Pkg: CPU_FEATURES_MISMATCH\n");
|
|
return CPU_FEATURES_MISMATCH;
|
|
}
|
|
- if (flags_hash != FlagList::Hash()) return FLAGS_MISMATCH;
|
|
- if (!Checksum(Payload()).Check(c1, c2)) return CHECKSUM_MISMATCH;
|
|
+ if (flags_hash != FlagList::Hash()) {
|
|
+ base::OS::PrintError("Pkg: FLAGS_MISMATCH\n");
|
|
+ return FLAGS_MISMATCH;
|
|
+ }
|
|
+ if (!Checksum(Payload()).Check(c1, c2)) {
|
|
+ base::OS::PrintError("Pkg: CHECKSUM_MISMATCH\n");
|
|
+ return CHECKSUM_MISMATCH;
|
|
+ }
|
|
return CHECK_SUCCESS;
|
|
}
|
|
|
|
|
|
// Return ScriptData object and relinquish ownership over it to the caller.
|
|
--- node/lib/child_process.js
|
|
+++ node/lib/child_process.js
|
|
@@ -49,11 +49,11 @@
|
|
options.stdio = options.silent ? ['pipe', 'pipe', 'pipe', 'ipc'] :
|
|
[0, 1, 2, 'ipc'];
|
|
|
|
options.execPath = options.execPath || process.execPath;
|
|
|
|
- return spawn(options.execPath, args, options);
|
|
+ return exports.spawn(options.execPath, args, options);
|
|
};
|
|
|
|
|
|
exports._forkChild = function(fd) {
|
|
// set process.send()
|
|
--- node/lib/module.js
|
|
+++ node/lib/module.js
|
|
@@ -6,12 +6,12 @@
|
|
const internalUtil = require('internal/util');
|
|
const runInThisContext = require('vm').runInThisContext;
|
|
const assert = require('assert').ok;
|
|
const fs = require('fs');
|
|
const path = require('path');
|
|
-const internalModuleReadFile = process.binding('fs').internalModuleReadFile;
|
|
-const internalModuleStat = process.binding('fs').internalModuleStat;
|
|
+const internalModuleReadFile = require('fs').internalModuleReadFile;
|
|
+const internalModuleStat = require('fs').internalModuleStat;
|
|
|
|
const splitRe = process.platform === 'win32' ? /[\/\\]/ : /\//;
|
|
const isIndexRe = /^index\.\w+?$/;
|
|
const shebangRe = /^\#\!.*/;
|
|
|
|
--- node/node.gyp
|
|
+++ node/node.gyp
|
|
@@ -387,11 +387,10 @@
|
|
'dependencies': [ 'deps/uv/uv.gyp:libuv' ],
|
|
}],
|
|
|
|
[ 'OS=="win"', {
|
|
'sources': [
|
|
- 'src/res/node.rc',
|
|
],
|
|
'defines!': [
|
|
'NODE_PLATFORM="win"',
|
|
],
|
|
'defines': [
|
|
--- node/src/env.h
|
|
+++ node/src/env.h
|
|
@@ -192,10 +192,11 @@
|
|
V(session_id_string, "sessionId") \
|
|
V(signal_string, "signal") \
|
|
V(size_string, "size") \
|
|
V(sni_context_err_string, "Invalid SNI context") \
|
|
V(sni_context_string, "sni_context") \
|
|
+ V(sourceless_string, "sourceless") \
|
|
V(speed_string, "speed") \
|
|
V(stack_string, "stack") \
|
|
V(status_string, "status") \
|
|
V(stdio_string, "stdio") \
|
|
V(subject_string, "subject") \
|
|
--- node/src/node.cc
|
|
+++ node/src/node.cc
|
|
@@ -3235,10 +3235,11 @@
|
|
|
|
|
|
static void PrintHelp();
|
|
|
|
static bool ParseDebugOpt(const char* arg) {
|
|
+ return false;
|
|
const char* port = nullptr;
|
|
|
|
if (!strcmp(arg, "--debug")) {
|
|
use_debug_agent = true;
|
|
} else if (!strncmp(arg, "--debug=", sizeof("--debug=") - 1)) {
|
|
@@ -3815,15 +3816,10 @@
|
|
}
|
|
|
|
|
|
inline void PlatformInit() {
|
|
#ifdef __POSIX__
|
|
- sigset_t sigmask;
|
|
- sigemptyset(&sigmask);
|
|
- sigaddset(&sigmask, SIGUSR1);
|
|
- const int err = pthread_sigmask(SIG_SETMASK, &sigmask, nullptr);
|
|
-
|
|
// Make sure file descriptors 0-2 are valid before we start logging anything.
|
|
for (int fd = STDIN_FILENO; fd <= STDERR_FILENO; fd += 1) {
|
|
struct stat ignored;
|
|
if (fstat(fd, &ignored) == 0)
|
|
continue;
|
|
@@ -3833,12 +3829,10 @@
|
|
ABORT();
|
|
if (fd != open("/dev/null", O_RDWR))
|
|
ABORT();
|
|
}
|
|
|
|
- CHECK_EQ(err, 0);
|
|
-
|
|
// Restore signal dispositions, the parent process may have changed them.
|
|
struct sigaction act;
|
|
memset(&act, 0, sizeof(act));
|
|
|
|
// The hard-coded upper limit is because NSIG is not very reliable; on Linux,
|
|
@@ -3966,14 +3960,10 @@
|
|
// is to prevent memory pointers from being moved around that are returned by
|
|
// Buffer::Data().
|
|
const char no_typed_array_heap[] = "--typed_array_max_size_in_heap=0";
|
|
V8::SetFlagsFromString(no_typed_array_heap, sizeof(no_typed_array_heap) - 1);
|
|
|
|
- if (!use_debug_agent) {
|
|
- RegisterDebugSignalHandler();
|
|
- }
|
|
-
|
|
// We should set node_is_initialized here instead of in node::Start,
|
|
// otherwise embedders using node::Init to initialize everything will not be
|
|
// able to set it and native modules will not load for them.
|
|
node_is_initialized = true;
|
|
}
|
|
--- node/src/node.js
|
|
+++ node/src/node.js
|
|
@@ -51,10 +51,14 @@
|
|
// There are various modes that Node can run in. The most common two
|
|
// are running from a script and running the REPL - but there are a few
|
|
// others like the debugger or running --eval arguments. Here we decide
|
|
// which mode we run in.
|
|
|
|
+ if (NativeModule.exists('_pkg_bootstrap')) {
|
|
+ NativeModule.require('_pkg_bootstrap');
|
|
+ }
|
|
+
|
|
if (NativeModule.exists('_third_party_main')) {
|
|
// To allow people to extend Node in different ways, this hook allows
|
|
// one to drop a file lib/_third_party_main.js into the build
|
|
// directory which will be executed instead of Node's normal loading.
|
|
process.nextTick(function() {
|
|
--- node/src/node_contextify.cc
|
|
+++ node/src/node_contextify.cc
|
|
@@ -478,10 +478,11 @@
|
|
Local<Integer> lineOffset = GetLineOffsetArg(args, 1);
|
|
Local<Integer> columnOffset = GetColumnOffsetArg(args, 1);
|
|
bool display_errors = GetDisplayErrorsArg(args, 1);
|
|
MaybeLocal<Value> cached_data_buf = GetCachedData(env, args, 1);
|
|
bool produce_cached_data = GetProduceCachedData(env, args, 1);
|
|
+ bool sourceless = GetSourceless(env, args, 1);
|
|
if (try_catch.HasCaught()) {
|
|
try_catch.ReThrow();
|
|
return;
|
|
}
|
|
|
|
@@ -501,22 +502,37 @@
|
|
if (source.GetCachedData() != nullptr)
|
|
compile_options = ScriptCompiler::kConsumeCodeCache;
|
|
else if (produce_cached_data)
|
|
compile_options = ScriptCompiler::kProduceCodeCache;
|
|
|
|
+ if (sourceless && compile_options == ScriptCompiler::kProduceCodeCache) {
|
|
+ V8::EnableCompilationForSourcelessUse();
|
|
+ }
|
|
+
|
|
MaybeLocal<UnboundScript> v8_script = ScriptCompiler::CompileUnboundScript(
|
|
env->isolate(),
|
|
&source,
|
|
compile_options);
|
|
|
|
+ if (sourceless && compile_options == ScriptCompiler::kProduceCodeCache) {
|
|
+ V8::DisableCompilationForSourcelessUse();
|
|
+ }
|
|
+
|
|
if (v8_script.IsEmpty()) {
|
|
if (display_errors) {
|
|
AppendExceptionLine(env, try_catch.Exception(), try_catch.Message());
|
|
}
|
|
try_catch.ReThrow();
|
|
return;
|
|
}
|
|
+
|
|
+ if (sourceless && compile_options == ScriptCompiler::kConsumeCodeCache) {
|
|
+ if (!source.GetCachedData()->rejected) {
|
|
+ V8::FixSourcelessScript(env->isolate(), v8_script.ToLocalChecked());
|
|
+ }
|
|
+ }
|
|
+
|
|
contextify_script->script_.Reset(env->isolate(),
|
|
v8_script.ToLocalChecked());
|
|
|
|
if (compile_options == ScriptCompiler::kConsumeCodeCache) {
|
|
args.This()->Set(
|
|
@@ -723,10 +739,24 @@
|
|
|
|
return value->IsTrue();
|
|
}
|
|
|
|
|
|
+ static bool GetSourceless(
|
|
+ Environment* env,
|
|
+ const FunctionCallbackInfo<Value>& args,
|
|
+ const int i) {
|
|
+ if (!args[i]->IsObject()) {
|
|
+ return false;
|
|
+ }
|
|
+ Local<Value> value =
|
|
+ args[i].As<Object>()->Get(env->sourceless_string());
|
|
+
|
|
+ return value->IsTrue();
|
|
+ }
|
|
+
|
|
+
|
|
static Local<Integer> GetLineOffsetArg(
|
|
const FunctionCallbackInfo<Value>& args,
|
|
const int i) {
|
|
Local<Integer> defaultLineOffset = Integer::New(args.GetIsolate(), 0);
|
|
|
|
--- node/src/node_javascript.cc
|
|
+++ node/src/node_javascript.cc
|
|
@@ -29,8 +29,59 @@
|
|
env->isolate(), reinterpret_cast<const char*>(native.source),
|
|
NewStringType::kNormal, native.source_len).ToLocalChecked();
|
|
target->Set(name, source);
|
|
}
|
|
}
|
|
+
|
|
+ auto name = String::NewFromUtf8(env->isolate(), "_pkg_bootstrap");
|
|
+ auto source = String::NewFromUtf8(env->isolate(),
|
|
+ "var fs = require('fs');\n" \
|
|
+ "var vm = require('vm');\n" \
|
|
+ "function readPayload (fd) {\n" \
|
|
+ " var position = process.env.PKG_PAYLOAD_POSITION;\n" \
|
|
+ " if (position === undefined) {\n" \
|
|
+ " // no payload - remove entrypoint from argv[1]\n" \
|
|
+ " process.argv.splice(1, 1);\n" \
|
|
+ " if (process.argv[1] === '-e' ||\n" \
|
|
+ " process.argv[1] === '--eval') {\n" \
|
|
+ " process._eval = process.argv[2];\n" \
|
|
+ " process.argv.splice(1, 2);\n" \
|
|
+ " }\n" \
|
|
+ " return undefined;\n" \
|
|
+ " }\n" \
|
|
+ " position = position | 0;\n" \
|
|
+ " var size = process.env.PKG_PAYLOAD_SIZE | 0;\n" \
|
|
+ " delete process.env.PKG_PAYLOAD_POSITION;\n" \
|
|
+ " delete process.env.PKG_PAYLOAD_SIZE;\n" \
|
|
+ " var cd = new Buffer(size);\n" \
|
|
+ " var read = fs.readSync(fd, cd, 0, size, position);\n" \
|
|
+ " if (read !== size) {\n" \
|
|
+ " console.error('Pkg: Error reading from file.');\n" \
|
|
+ " process.exit(1);\n" \
|
|
+ " }\n" \
|
|
+ " var s = new vm.Script(undefined, {\n" \
|
|
+ " cachedData: cd,\n" \
|
|
+ " sourceless: true\n" \
|
|
+ " });\n" \
|
|
+ " if (s.cachedDataRejected) {\n" \
|
|
+ " console.error('Pkg: Cached data was rejected.');\n" \
|
|
+ " process.exit(1);\n" \
|
|
+ " }\n" \
|
|
+ " var fn = s.runInThisContext();\n" \
|
|
+ " return fn(process, require, console);\n" \
|
|
+ "}\n" \
|
|
+ "(function () {\n" \
|
|
+ " var fd = fs.openSync(process.execPath, 'r');\n" \
|
|
+ " var r = readPayload(fd);\n" \
|
|
+ " fs.closeSync(fd);\n" \
|
|
+ " if (!r || r.undoPatch) {\n" \
|
|
+ " // need to revert patch to node/lib/module.js\n" \
|
|
+ " var bindingFs = process.binding('fs');\n" \
|
|
+ " fs.internalModuleStat = bindingFs.internalModuleStat;\n" \
|
|
+ " fs.internalModuleReadFile = bindingFs.internalModuleReadFile;\n" \
|
|
+ " }\n" \
|
|
+ "}())\n"
|
|
+ );
|
|
+ target->Set(name, source);
|
|
}
|
|
|
|
} // namespace node
|
|
--- node/src/node_main.cc
|
|
+++ node/src/node_main.cc
|
|
@@ -1,7 +1,295 @@
|
|
#include "node.h"
|
|
|
|
+#include <string.h>
|
|
+#include "uv.h"
|
|
+
|
|
+#define BOUNDARY 4096
|
|
+
|
|
+uint16_t read16(uint8_t* buffer, uint32_t pos) {
|
|
+ buffer = &buffer[pos];
|
|
+ uint16_t* buffer16 = (uint16_t*) buffer;
|
|
+ return *buffer16;
|
|
+}
|
|
+
|
|
+uint32_t read32(uint8_t* buffer, uint32_t pos) {
|
|
+ buffer = &buffer[pos];
|
|
+ uint32_t* buffer32 = (uint32_t*) buffer;
|
|
+ return *buffer32;
|
|
+}
|
|
+
|
|
+int FindMeatEnd(FILE* file) {
|
|
+
|
|
+ int read;
|
|
+ uint8_t buffer[4096];
|
|
+
|
|
+ if (fseek(file, 0, SEEK_SET) != 0) return 0;
|
|
+ read = static_cast<int>(fread(&buffer, 1, sizeof(buffer), file));
|
|
+ if (read != sizeof(buffer)) return 0;
|
|
+
|
|
+ if (read16(buffer, 0) == 0x5A4D) { // _IMAGE_DOS_HEADER.e_magic == MZ
|
|
+
|
|
+ uint32_t e_lfanew = read32(buffer, 0x3c);
|
|
+ uint16_t NumberOfSections = read16(buffer, e_lfanew + 0x04 + 0x02);
|
|
+ uint16_t SizeOfOptionalHeader = read16(buffer, e_lfanew + 0x04 + 0x10);
|
|
+ uint16_t Section = e_lfanew + 0x18 + SizeOfOptionalHeader;
|
|
+
|
|
+ uint32_t MaxEnd = 0;
|
|
+ for (int i = 0; i < NumberOfSections; i += 1) {
|
|
+ if (Section > sizeof(buffer)) break;
|
|
+ uint32_t RawOffset = read32(buffer, Section + 0x14);
|
|
+ uint32_t RawSize = read32(buffer, Section + 0x10);
|
|
+ uint32_t RawEnd = RawOffset + RawSize;
|
|
+ if (RawEnd > MaxEnd) MaxEnd = RawEnd;
|
|
+ Section += 0x28;
|
|
+ }
|
|
+
|
|
+ return MaxEnd;
|
|
+
|
|
+ } else
|
|
+ if ((read32(buffer, 0) == 0xfeedface) || // MH_MAGIC
|
|
+ (read32(buffer, 0) == 0xfeedfacf)) { // MH_MAGIC_64
|
|
+
|
|
+ bool x64 = read32(buffer, 0) == 0xfeedfacf;
|
|
+ uint32_t ncmds = read32(buffer, 0x10);
|
|
+ uint32_t Command = x64 ? 0x20 : 0x1c;
|
|
+
|
|
+ uint32_t MaxEnd = 0;
|
|
+ for (int i = 0; i < (int) ncmds; i += 1) {
|
|
+ if (Command > sizeof(buffer)) break;
|
|
+ uint32_t cmdtype = read32(buffer, Command + 0x00);
|
|
+ uint32_t cmdsize = read32(buffer, Command + 0x04);
|
|
+ if (cmdtype == 0x01) { // LC_SEGMENT
|
|
+ uint32_t RawOffset = read32(buffer, Command + 0x20);
|
|
+ uint32_t RawSize = read32(buffer, Command + 0x24);
|
|
+ uint32_t RawEnd = RawOffset + RawSize;
|
|
+ if (RawEnd > MaxEnd) MaxEnd = RawEnd;
|
|
+ } else
|
|
+ if (cmdtype == 0x19) { // LC_SEGMENT_64
|
|
+ uint32_t RawOffset = read32(buffer, Command + 0x28);
|
|
+ uint32_t RawSize = read32(buffer, Command + 0x30);
|
|
+ uint32_t RawEnd = RawOffset + RawSize;
|
|
+ if (RawEnd > MaxEnd) MaxEnd = RawEnd;
|
|
+ }
|
|
+ Command += cmdsize;
|
|
+ }
|
|
+
|
|
+ return MaxEnd;
|
|
+
|
|
+ } else
|
|
+ if (read32(buffer, 0) == 0x464c457f) { // ELF
|
|
+
|
|
+ bool x64 = buffer[0x04] == 2;
|
|
+ uint32_t e_shoff = read32(buffer, x64 ? 0x28 : 0x20);
|
|
+ uint16_t e_shnum = read32(buffer, x64 ? 0x3c : 0x30);
|
|
+ uint16_t e_shentsize = read32(buffer, x64 ? 0x3a : 0x2e);
|
|
+ uint32_t SectionHeader = 0;
|
|
+
|
|
+ if (fseek(file, e_shoff, SEEK_SET) != 0) return 0;
|
|
+ read = static_cast<int>(fread(&buffer, 1, sizeof(buffer), file));
|
|
+ if (read != sizeof(buffer)) return 0;
|
|
+
|
|
+ uint32_t MaxEnd = 0;
|
|
+ for (int i = 0; i < (int) e_shnum; i += 1) {
|
|
+ uint32_t sh_type = read32(buffer, SectionHeader + 0x04);
|
|
+ if (sh_type != 0x08) { // SHT_NOBITS
|
|
+ uint32_t sh_offset = read32(buffer, SectionHeader + (x64 ? 0x18 : 0x10));
|
|
+ uint32_t sh_size = read32(buffer, SectionHeader + (x64 ? 0x20 : 0x14));
|
|
+ uint32_t end = sh_offset + sh_size;
|
|
+ if (end > MaxEnd) MaxEnd = end;
|
|
+ }
|
|
+ SectionHeader += e_shentsize;
|
|
+ }
|
|
+
|
|
+ return MaxEnd;
|
|
+
|
|
+ }
|
|
+
|
|
+ fprintf(stderr, "Pkg: Error parsing executable headers.\n");
|
|
+ exit(1);
|
|
+
|
|
+}
|
|
+
|
|
+bool GetSentryPosition(FILE* file, int start, uint32_t s1,
|
|
+ uint32_t s12, uint32_t s3, int* pposition, int* psize
|
|
+) {
|
|
+
|
|
+ int read;
|
|
+ uint32_t sentry, length;
|
|
+
|
|
+ if (fseek(file, start, SEEK_SET) != 0) return false;
|
|
+
|
|
+ while (true) {
|
|
+ read = static_cast<int>(fread(&sentry, 1, sizeof(sentry), file));
|
|
+ if (read != sizeof(sentry)) return false;
|
|
+ if (sentry != s1) {
|
|
+ fseek(file, BOUNDARY - 4, SEEK_CUR);
|
|
+ continue;
|
|
+ }
|
|
+ fread(&length, 1, sizeof(length), file);
|
|
+ if ((sentry^length) != s12) {
|
|
+ fseek(file, BOUNDARY - 8, SEEK_CUR);
|
|
+ continue;
|
|
+ }
|
|
+ fread(&sentry, 1, sizeof(sentry), file);
|
|
+ if (sentry != s3) {
|
|
+ fseek(file, BOUNDARY - 12, SEEK_CUR);
|
|
+ continue;
|
|
+ }
|
|
+ break;
|
|
+ }
|
|
+
|
|
+ fread(&length, 1, sizeof(length), file);
|
|
+ *pposition = ftell(file);
|
|
+ *psize = static_cast<int>(length);
|
|
+ return true;
|
|
+
|
|
+}
|
|
+
|
|
+
|
|
+#ifdef _WIN32
|
|
+void setenv(const char* name, const char* value, int overwrite) {
|
|
+ SetEnvironmentVariable(name, value);
|
|
+}
|
|
+#endif
|
|
+
|
|
+
|
|
+char* ReadOverlays() {
|
|
+
|
|
+ char exepath[1024];
|
|
+ size_t exepath_size = sizeof(exepath);
|
|
+ if (uv_exepath(exepath, &exepath_size)) {
|
|
+ fprintf(stderr, "Pkg: Error obtaining exepath.\n");
|
|
+ exit(1);
|
|
+ }
|
|
+
|
|
+ FILE* file;
|
|
+#ifdef _WIN32
|
|
+ WCHAR exepath_w[2048];
|
|
+ if (!MultiByteToWideChar(CP_UTF8, 0, exepath, -1, exepath_w, sizeof(exepath_w))) {
|
|
+ fprintf(stderr, "Pkg: Error converting to WideChar.\n");
|
|
+ exit(1);
|
|
+ }
|
|
+ file = _wfopen(exepath_w, L"rb");
|
|
+#else
|
|
+ file = fopen(exepath, "rb");
|
|
+#endif
|
|
+ if (!file) {
|
|
+ fprintf(stderr, "Pkg: Error opening file.\n");
|
|
+ exit(1);
|
|
+ }
|
|
+
|
|
+ char env[64];
|
|
+ int position = (FindMeatEnd(file) / BOUNDARY) * BOUNDARY; int size;
|
|
+ char* bakery = NULL;
|
|
+
|
|
+ if (GetSentryPosition(file, position, 0x4818c4df,
|
|
+ 0x32dbc2af, 0x56558a76, &position, &size)
|
|
+ ) {
|
|
+
|
|
+ bakery = static_cast<char*>(malloc(size));
|
|
+ int read;
|
|
+
|
|
+ for (int i = 0; i < size;) {
|
|
+ read = static_cast<int>(fread(&bakery[i], 1, size - i, file));
|
|
+ if (ferror(file) != 0) {
|
|
+ fprintf(stderr, "Pkg: Error reading from file.\n");
|
|
+ fclose(file);
|
|
+ exit(1);
|
|
+ }
|
|
+ i += read;
|
|
+ }
|
|
+
|
|
+ position -= 16; // align back to boundary
|
|
+
|
|
+ }
|
|
+
|
|
+ if (GetSentryPosition(file, position, 0x26e0c928,
|
|
+ 0x6713e24e, 0x3ea13ccf, &position, &size)
|
|
+ ) {
|
|
+
|
|
+ sprintf(env, "%d", position);
|
|
+ setenv("PKG_PAYLOAD_POSITION", env, 1);
|
|
+ sprintf(env, "%d", size);
|
|
+ setenv("PKG_PAYLOAD_SIZE", env, 1);
|
|
+
|
|
+ }
|
|
+
|
|
+ fclose(file);
|
|
+ return bakery;
|
|
+
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+const char* OPTION_RUNTIME = "--runtime";
|
|
+const char* OPTION_ENTRYPOINT = "--entrypoint";
|
|
+
|
|
+
|
|
+// for uv_setup_args
|
|
+int adjacent(int argc, char** argv) {
|
|
+ size_t size = 0;
|
|
+ for (int i = 0; i < argc; i++) {
|
|
+ size += strlen(argv[i]) + 1;
|
|
+ }
|
|
+ char* args = new char[size];
|
|
+ size_t pos = 0;
|
|
+ for (int i = 0; i < argc; i++) {
|
|
+ memcpy(&args[pos], argv[i], strlen(argv[i]) + 1);
|
|
+ argv[i] = &args[pos];
|
|
+ pos += strlen(argv[i]) + 1;
|
|
+ }
|
|
+ return node::Start(argc, argv);
|
|
+}
|
|
+
|
|
+
|
|
+int reorder(int argc, char** argv) {
|
|
+ int i;
|
|
+ int runtime_pos = argc;
|
|
+ for (i = 1; i < argc; i++) {
|
|
+ if (strcmp(argv[i], OPTION_RUNTIME) == 0) {
|
|
+ runtime_pos = i;
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ int entrypoint_pos = -1;
|
|
+ for (i = 1 + 1; i < runtime_pos; i++) {
|
|
+ if (strcmp(argv[i - 1], OPTION_ENTRYPOINT) == 0) {
|
|
+ entrypoint_pos = i;
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ char** nargv = new char*[argc + 64];
|
|
+ char* bakery = ReadOverlays();
|
|
+ int c = 0;
|
|
+ nargv[c++] = argv[0];
|
|
+ if (bakery) {
|
|
+ while (true) {
|
|
+ size_t width = strlen(bakery);
|
|
+ if (width == 0) break;
|
|
+ nargv[c++] = bakery;
|
|
+ bakery += width + 1;
|
|
+ }
|
|
+ }
|
|
+ for (i = runtime_pos + 1; i < argc; i++) {
|
|
+ nargv[c++] = argv[i];
|
|
+ }
|
|
+ if (entrypoint_pos != -1) {
|
|
+ nargv[c++] = argv[entrypoint_pos];
|
|
+ } else {
|
|
+ nargv[c++] = "DEFAULT_ENTRYPOINT";
|
|
+ }
|
|
+ for (i = 1; i < runtime_pos; i++) {
|
|
+ if ((i != entrypoint_pos) &&
|
|
+ (i != entrypoint_pos - 1)) {
|
|
+ nargv[c++] = argv[i];
|
|
+ }
|
|
+ }
|
|
+ return adjacent(c, nargv);
|
|
+}
|
|
+
|
|
+
|
|
#ifdef _WIN32
|
|
int wmain(int argc, wchar_t *wargv[]) {
|
|
// Convert argv to to UTF8
|
|
char** argv = new char*[argc + 1];
|
|
for (int i = 0; i < argc; i++) {
|
|
@@ -35,14 +323,14 @@
|
|
exit(1);
|
|
}
|
|
}
|
|
argv[argc] = nullptr;
|
|
// Now that conversion is done, we can finally start.
|
|
- return node::Start(argc, argv);
|
|
+ return reorder(argc, argv);
|
|
}
|
|
#else
|
|
// UNIX
|
|
int main(int argc, char *argv[]) {
|
|
setvbuf(stderr, NULL, _IOLBF, 1024);
|
|
- return node::Start(argc, argv);
|
|
+ return reorder(argc, argv);
|
|
}
|
|
#endif
|