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
This commit is contained in:
William Wilgus 2020-10-05 03:28:02 -04:00
parent 74258fca31
commit ef34126913
3 changed files with 100 additions and 3 deletions

View file

@ -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}
};