5 files changed, 104 insertions, 0 deletions
diff --git a/lib/esp-idf-lua/examples/alloc/CMakeLists.txt b/lib/esp-idf-lua/examples/alloc/CMakeLists.txt
new file mode 100644
index 00000000..771bb5bb
--- /dev/null
+++ b/lib/esp-idf-lua/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)
diff --git a/lib/esp-idf-lua/examples/alloc/Makefile b/lib/esp-idf-lua/examples/alloc/Makefile
new file mode 100644
index 00000000..2b16bdbf
--- /dev/null
+++ b/lib/esp-idf-lua/examples/alloc/Makefile
@@ -0,0 +1,5 @@
+PROJECT_NAME := example-lua-alloc
+
+EXTRA_COMPONENT_DIRS := $(CURDIR)/../..
+
+include $(IDF_PATH)/make/project.mk
diff --git a/lib/esp-idf-lua/examples/alloc/main/CMakeLists.txt b/lib/esp-idf-lua/examples/alloc/main/CMakeLists.txt
new file mode 100644
index 00000000..09a8616b
--- /dev/null
+++ b/lib/esp-idf-lua/examples/alloc/main/CMakeLists.txt
@@ -0,0 +1,2 @@
+idf_component_register(SRC_DIRS "."
+                    INCLUDE_DIRS ".")
diff --git a/lib/esp-idf-lua/examples/alloc/main/component.mk b/lib/esp-idf-lua/examples/alloc/main/component.mk
new file mode 100644
index 00000000..037751c1
--- /dev/null
+++ b/lib/esp-idf-lua/examples/alloc/main/component.mk
@@ -0,0 +1,2 @@
+COMPONENT_ADD_INCLUDEDIRS = .
+COMPONENT_SRCDIRS = .
diff --git a/lib/esp-idf-lua/examples/alloc/main/main.c b/lib/esp-idf-lua/examples/alloc/main/main.c
new file mode 100644
index 00000000..38b374ef
--- /dev/null
+++ b/lib/esp-idf-lua/examples/alloc/main/main.c
@@ -0,0 +1,89 @@
+#include <stdio.h>
+#include <inttypes.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 int lua_mem_size = 0;
+
+static void* l_alloc(void *ud, void *ptr, size_t osize, size_t nsize)
+{
+    int new_size = lua_mem_size - osize + nsize;
+    if (new_size > LUA_MAX_MEMSIZE)
+    {
+        printf("Error! Lua wants more memory than we can allocate: %u > %u\n", new_size, LUA_MAX_MEMSIZE);
+        return NULL;
+    }
+    //printf("Reallocating: old = %u, new = %u\n", lua_mem_size, new_size);
+    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: %" PRIu32 "\n", xPortGetFreeHeapSize());
+    while (1)
+    {
+        printf(".");
+        fflush(stdout);
+        vTaskDelay(pdMS_TO_TICKS(100));
+    }
+}
+
+void app_main()
+{
+    xTaskCreate(test, "test", 0x10000, NULL, 5, NULL);
+}