Foolproofed config.allocator when it is a function

1 files changed, 3 insertions, 0 deletions

diff --git a/src/state.cpp b/src/state.cpp
index 3f6b3d7..ee0b199 100644
--- a/src/state.cpp
+++ b/src/state.cpp
@@ -208,6 +208,9 @@ namespace state {
                             lua_pushcclosure(from, U->provideAllocator, 0);
                             lua_call(from, 0, 1);
                             AllocatorDefinition* const _def{ luaG_tofulluserdata<AllocatorDefinition>(from, -1) };
+                            if (!_def || _def->version != AllocatorDefinition::kAllocatorVersion) {
+                                raise_luaL_error(from, "Bad config.allocator function, must provide a valid AllocatorDefinition");
+                            }
                             lua_State* const _L{ lua_newstate(_def->allocF, _def->allocUD) };
                             lua_pop(from, 1);
                             return _L;