Custom memory allocator example

examples/alloc/CMakeLists.txt

@@ -0,0 +1,6 @@
+cmake_minimum_required(VERSION 3.5)
+
+set(EXTRA_COMPONENT_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/../..)
+
+include($ENV{IDF_PATH}/tools/cmake/project.cmake)
+project(example-lua-alloc)

examples/alloc/Makefile

@@ -0,0 +1,5 @@
+PROJECT_NAME := example-lua-alloc
+
+EXTRA_COMPONENT_DIRS := $(CURDIR)/../..
+
+include $(IDF_PATH)/make/project.mk

examples/alloc/main/CMakeLists.txt

@@ -0,0 +1,2 @@
+idf_component_register(SRC_DIRS "."
+ INCLUDE_DIRS ".")

examples/alloc/main/component.mk

@@ -0,0 +1,2 @@
+COMPONENT_ADD_INCLUDEDIRS = .
+COMPONENT_SRCDIRS = .

examples/alloc/main/main.c

@@ -0,0 +1,87 @@
+#include <stdio.h>
+#include <freertos/FreeRTOS.h>
+#include <freertos/task.h>
+#include <string.h>
+#include <lua/lua.h>
+#include <lua/lauxlib.h>
+#include <lua/lualib.h>
+
+#define LUA_MAX_MEMSIZE 0x10000 // 64k
+
+static const char *prg =
+ "local t = {}\n"
+ "for i = 1, 10000 do\n"
+ " print(string.format(\"Iteration %d...\", i))\n"
+ " t[i] = string.format(\"Lorem ipsum asklfhskldjf hskljfh slkjfd sdklfjh slkfjh sdkljfjh sdklfjh sdklf %d...\", i)\n"
+ "end\n";
+
+static void report(lua_State *L, int status)
+{
+ if (status == LUA_OK)
+ return;
+
+ const char *msg = lua_tostring(L, -1);
+ printf("%s\n", msg);
+ lua_pop(L, 1);
+}
+
+static size_t lua_mem_size = 0;
+
+static void* l_alloc(void *ud, void *ptr, size_t osize, size_t nsize)
+{
+ size_t new_size = lua_mem_size - osize + nsize;
+ if (new_size > LUA_MAX_MEMSIZE)
+ {
+ printf("Error! Lua wants more memory than we can allocate!\n");
+ return NULL;
+ }
+ lua_mem_size = new_size;
+
+ if (nsize == 0)
+ {
+ free(ptr);
+ return NULL;
+ }
+ else
+ return realloc(ptr, nsize);
+}
+
+void test(void *arg)
+{
+ printf("Program:\n%s\n", prg);
+
+ lua_State *L = lua_newstate(l_alloc, NULL);
+ if (!L)
+ {
+ printf("Could not create state\n");
+ while (1) { vTaskDelay(1000); }
+ }
+
+ luaL_openlibs(L);
+
+ int r = luaL_loadstring(L, prg);
+ if (r)
+ {
+ report(L, r);
+ while (1) { vTaskDelay(1000); }
+ }
+
+ r = lua_pcall(L, 0, 0, 0);
+ if (r)
+ report(L, r);
+
+ lua_close(L);
+
+ printf("State closed, heap: %d\n", xPortGetFreeHeapSize());
+ while (1)
+ {
+ printf(".");
+ fflush(stdout);
+ vTaskDelay(pdMS_TO_TICKS(100));
+ }
+}
+
+void app_main()
+{
+ xTaskCreate(test, "test", 0x10000, NULL, 5, NULL);
+}