Setting a global variable on lua scripts to check if is running safely or not

This commit is contained in:
2025-09-03 22:26:39 -03:00
parent b6a67b30c5
commit b676ea873e

View File

@@ -674,6 +674,7 @@ func Install(packagepath string, serial uint) error {
L.SetGlobal("package", lua.LNil) L.SetGlobal("package", lua.LNil)
L.SetGlobal("require", lua.LNil) L.SetGlobal("require", lua.LNil)
L.SetGlobal("runningUnsafe", lua.LFalse)
L.SetGlobal("packets_package_dir", lua.LString(cfg.Config.DataDir)) L.SetGlobal("packets_package_dir", lua.LString(cfg.Config.DataDir))
L.SetGlobal("packets_bin_dir", lua.LString(cfg.Config.BinDir)) L.SetGlobal("packets_bin_dir", lua.LString(cfg.Config.BinDir))
@@ -709,6 +710,8 @@ func Install(packagepath string, serial uint) error {
ioObject.RawSetString("stdin", lua.LNil) ioObject.RawSetString("stdin", lua.LNil)
ioObject.RawSetString("lines", lua.LNil) ioObject.RawSetString("lines", lua.LNil)
ioObject.RawSetString("open", L.NewFunction(internal.SafeOpen)) ioObject.RawSetString("open", L.NewFunction(internal.SafeOpen))
} else {
L.SetGlobal("runningUnsafe", lua.LTrue)
} }
if err := L.DoFile(filepath.Join(cfg.Config.DataDir, name, manifest.Hooks.Install)); err != nil { if err := L.DoFile(filepath.Join(cfg.Config.DataDir, name, manifest.Hooks.Install)); err != nil {
@@ -1362,6 +1365,7 @@ func Unninstall(realname string) error {
L.SetGlobal("package", lua.LNil) L.SetGlobal("package", lua.LNil)
L.SetGlobal("require", lua.LNil) L.SetGlobal("require", lua.LNil)
L.SetGlobal("runningUnsafe", lua.LFalse)
L.SetGlobal("packets_package_dir", lua.LString(cfg.Config.DataDir)) L.SetGlobal("packets_package_dir", lua.LString(cfg.Config.DataDir))
L.SetGlobal("packets_bin_dir", lua.LString(cfg.Config.BinDir)) L.SetGlobal("packets_bin_dir", lua.LString(cfg.Config.BinDir))
@@ -1398,6 +1402,8 @@ func Unninstall(realname string) error {
ioObject.RawSetString("lines", lua.LNil) ioObject.RawSetString("lines", lua.LNil)
ioObject.RawSetString("open", L.NewFunction(internal.SafeOpen)) ioObject.RawSetString("open", L.NewFunction(internal.SafeOpen))
} else {
L.SetGlobal("runningUnsafe", lua.LTrue)
} }
if err := L.DoFile(filepath.Join(cfg.Config.DataDir, realname, manifest.Hooks.Remove)); err != nil { if err := L.DoFile(filepath.Join(cfg.Config.DataDir, realname, manifest.Hooks.Remove)); err != nil {
@@ -1687,6 +1693,7 @@ func Upgrade(packagepath string, og_realname string, serial uint) error {
L.SetGlobal("package", lua.LNil) L.SetGlobal("package", lua.LNil)
L.SetGlobal("require", lua.LNil) L.SetGlobal("require", lua.LNil)
L.SetGlobal("runningUnsafe", lua.LFalse)
L.SetGlobal("packets_package_dir", lua.LString(cfg.Config.DataDir)) L.SetGlobal("packets_package_dir", lua.LString(cfg.Config.DataDir))
L.SetGlobal("packets_bin_dir", lua.LString(cfg.Config.BinDir)) L.SetGlobal("packets_bin_dir", lua.LString(cfg.Config.BinDir))
@@ -1703,6 +1710,8 @@ func Upgrade(packagepath string, og_realname string, serial uint) error {
L.SetGlobal("build", build) L.SetGlobal("build", build)
// end of build functions
osObject.RawSetString("execute", lua.LNil) osObject.RawSetString("execute", lua.LNil)
osObject.RawSetString("exit", lua.LNil) osObject.RawSetString("exit", lua.LNil)
osObject.RawSetString("getenv", lua.LNil) osObject.RawSetString("getenv", lua.LNil)
@@ -1722,6 +1731,8 @@ func Upgrade(packagepath string, og_realname string, serial uint) error {
ioObject.RawSetString("stdin", lua.LNil) ioObject.RawSetString("stdin", lua.LNil)
ioObject.RawSetString("lines", lua.LNil) ioObject.RawSetString("lines", lua.LNil)
ioObject.RawSetString("open", L.NewFunction(internal.SafeOpen)) ioObject.RawSetString("open", L.NewFunction(internal.SafeOpen))
} else {
L.SetGlobal("runningUnsafe", lua.LTrue)
} }
if err := L.DoFile(filepath.Join(cfg.Config.DataDir, name, manifest.Hooks.Install)); err != nil { if err := L.DoFile(filepath.Join(cfg.Config.DataDir, name, manifest.Hooks.Install)); err != nil {