mirror/The-Powder-Toy
1
0
Fork 0
mirror of https://github.com/The-Powder-Toy/The-Powder-Toy.git synced 2025-08-17 21:51:20 +02:00
Code Issues Projects Releases Wiki Activity

Protect new math.random against division by 0

Browse Source
This commit is contained in:
Tamás Bálint Misius
2023-04-15 23:20:25 +02:00
parent 34e4d90dac
commit 54d08f3496
1 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/lua/LuaScriptInterface.cpp
View File

@@ -175,12 +175,28 @@ static int mathRandom(lua_State *l)
{
// only thing that matters is that the rng not be luacon_sim->rng when !inSimEvent
auto &rng = inSimEvent ? luacon_sim->rng : interfaceRng;
int lower, upper;
switch (lua_gettop(l))
{
case 0: lua_pushnumber (l, rng.uniform01()); break;
case 1: lua_pushinteger(l, rng.between( 1, luaL_checkinteger(l, 1))); break;
default: lua_pushinteger(l, rng.between(luaL_checkinteger(l, 1), luaL_checkinteger(l, 2))); break;
case 0:
lua_pushnumber(l, rng.uniform01());
return 1;
case 1:
lower = 1;
upper = luaL_checkinteger(l, 1);
break;
default:
lower = luaL_checkinteger(l, 1);
upper = luaL_checkinteger(l, 2);
break;
}
if (upper < lower)
{
luaL_error(l, "interval is empty");
}
lua_pushinteger(l, rng.between(lower, upper));
return 1;
}