MeshCentral/MeshAgent
1
0
Fork 0
mirror of https://github.com/Ylianst/MeshAgent synced 2025-12-16 00:03:45 +00:00
Code Issues Releases Wiki Activity

1. Fixed compiler warning

Browse Source 
2. Updated so process.exit() throws exception when called in agent mode
This commit is contained in:
Bryan Roe
2019-05-14 14:52:39 -07:00
parent d0fb36d10b
commit 7cf000571d
6 changed files with 45 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
microscript/ILibDuktape_ScriptContainer.c
View File

@@ -491,22 +491,30 @@ duk_ret_t ILibDuktape_ScriptContainer_Process_Exit(duk_context *ctx)
void **tmp;
int nargs = duk_get_top(ctx);
duk_push_this(ctx); // [process]
if (nargs == 1)
if (duk_peval_string(ctx, "require('MeshAgent').agentMode") == 0 && duk_get_boolean(ctx, -1))
{
duk_push_int(ctx, duk_require_int(ctx, 0)); // [process][code]
duk_put_prop_string(ctx, -2, ILibDuktape_ScriptContainer_ExitCode); // [process]
// Running in Agent Mode, so exiting process is not allowed
return(ILibDuktape_Error(ctx, "Process.exit() not allowed when running in Agent Mode"));
}
if (duk_has_prop_string(ctx, -1, ILibDuktape_ScriptContainer_Exitting)) { return(ILibDuktape_Error(ctx, "Process.exit() forced script termination")); }
duk_push_int(ctx, 1);
duk_put_prop_string(ctx, -2, ILibDuktape_ScriptContainer_Exitting); // [process]
else
{
duk_push_this(ctx); // [process]
if (nargs == 1)
{
duk_push_int(ctx, duk_require_int(ctx, 0)); // [process][code]
duk_put_prop_string(ctx, -2, ILibDuktape_ScriptContainer_ExitCode); // [process]
}
if (duk_has_prop_string(ctx, -1, ILibDuktape_ScriptContainer_Exitting)) { return(ILibDuktape_Error(ctx, "Process.exit() forced script termination")); }
duk_push_int(ctx, 1);
duk_put_prop_string(ctx, -2, ILibDuktape_ScriptContainer_Exitting); // [process]
// Execute this later, so that this stack can unwind first, before we destroy the heap
tmp = (void**)Duktape_PushBuffer(ctx, sizeof(void*)); // [process][buffer]
duk_put_prop_string(ctx, -2, "\xFF_JUNK"); // [process]
tmp[0] = ctx;
ILibLifeTime_Add(ILibGetBaseTimer(Duktape_GetChain(ctx)), tmp, 0, ILibDuktape_ScriptContainer_Process_ExitCallback, NULL);
// Execute this later, so that this stack can unwind first, before we destroy the heap
tmp = (void**)Duktape_PushBuffer(ctx, sizeof(void*)); // [process][buffer]
duk_put_prop_string(ctx, -2, "\xFF_JUNK"); // [process]
tmp[0] = ctx;
ILibLifeTime_Add(ILibGetBaseTimer(Duktape_GetChain(ctx)), tmp, 0, ILibDuktape_ScriptContainer_Process_ExitCallback, NULL);
}
return(ILibDuktape_Error(ctx, "Process.exit() forced script termination"));
}