.

lib/compiler.c

@@ -5,12 +5,14 @@
 
 #include "lua.h"
 #include "lauxlib.h"
+#include "lualib.h"
 
 #include "compiler/compiler.lua.h"
 
 TYC_RESULT code_assemble(const char* code, uint8_t** bytecode, size_t* bytecode_sz)
 {
     lua_State* L = luaL_newstate();
+    luaL_openlibs(L);
 
     int r = luaL_loadbufferx(L, (const char *) lib_compiler_compiler_out, lib_compiler_compiler_out_len, "compiler", "b");
     if (r == LUA_ERRSYNTAX)
@@ -18,6 +20,8 @@ TYC_RESULT code_assemble(const char* code, uint8_t** bytecode, size_t* bytecode_
     else if (r == LUA_ERRMEM)
         out_of_memory();
 
+    lua_call(L, 0, 1);
+
     lua_pushstring(L, code);
     r = lua_pcall(L, 1, 1, 0);
     if (r == LUA_ERRMEM) {
@@ -29,7 +33,15 @@ TYC_RESULT code_assemble(const char* code, uint8_t** bytecode, size_t* bytecode_
         return T_ERR_ASSEMBLER_SYNTAX_ERROR;
     }
 
-
+    if (!lua_istable(L, -1))
+        abort();
+    *bytecode_sz = luaL_len(L, -1);
+    *bytecode = malloc(*bytecode_sz);
+    for (size_t i = 0; i < *bytecode_sz; ++i) {
+        lua_geti(L, -1, i + 1);
+        (*bytecode)[i] = (uint8_t) lua_tointeger(L, -1);
+        lua_pop(L, 1);
+    }
 
     lua_close(L);
     return T_OK;