From ef34126913978c7cd6e5b0831f78ac8355f053f0 Mon Sep 17 00:00:00 2001 From: William Wilgus Date: Mon, 5 Oct 2020 03:28:02 -0400 Subject: lua add better memory stats lua gives you a memory used number that only reflects the current allocations if fact it doesn't even give you a way to get the amount of ram free rb.mem_stats() seeks to fill this gap by marking the memory allocated for lua with a sentinel value which can later be checked to get a high water mark of the ram used by lua and a pretty good idea of how much ram is available Also includes an example script usage: used, allocd, free = rb.mem_stats() Change-Id: Ia282869f989848324d7d88c7df4827fdbce4fb4e --- apps/plugins/lua/rocklib.c | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'apps/plugins/lua/rocklib.c') diff --git a/apps/plugins/lua/rocklib.c b/apps/plugins/lua/rocklib.c index 5f0143efbb..8921c0a4c3 100644 --- a/apps/plugins/lua/rocklib.c +++ b/apps/plugins/lua/rocklib.c @@ -53,6 +53,8 @@ * * ----------------------------- */ +extern size_t rock_get_allocated_bytes(void); /* tlsf_helper.c */ +extern size_t rock_get_unused_bytes(void); #define RB_WRAP(func) static int rock_##func(lua_State UNUSED_ATTR *L) #define SIMPLE_VOID_WRAPPER(func) RB_WRAP(func) { (void)L; func(); return 0; } @@ -931,6 +933,19 @@ RB_WRAP(show_logo) return 0; } +RB_WRAP(mem_stats) +{ + /* used, allocd, free = rb.mem_stats() */ + /* note free is the high watermark */ + size_t allocd = rock_get_allocated_bytes(); + size_t free = rock_get_unused_bytes(); + + lua_pushinteger(L, allocd - free); + lua_pushinteger(L, allocd); + lua_pushinteger(L, free); + return 3; +} + #define RB_FUNC(func) {#func, rock_##func} #define RB_ALIAS(name, func) {name, rock_##func} static const luaL_Reg rocklib[] = @@ -1015,6 +1030,7 @@ static const luaL_Reg rocklib[] = /* MISC */ RB_FUNC(restart_lua), RB_FUNC(show_logo), + RB_FUNC(mem_stats), {NULL, NULL} }; -- cgit v1.2.3