Skip to content

Commit

Permalink
add part 10
Browse files Browse the repository at this point in the history
  • Loading branch information
@Manistein
Manistein committed Feb 23, 2021
1 parent 7f3b7df commit b365b1c
Show file tree
Hide file tree
Showing 70 changed files with 10,182 additions and 0 deletions.
277 changes: 277 additions & 0 deletions windows/project/part10/clib/luaaux.c
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,277 @@
/* Copyright (c) 2018 Manistein,https://manistein.github.io/blog/
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.*/

#include "luaaux.h"
#include "../vm/luado.h"
#include "../common/luastring.h"
#include "../common/luatable.h"
#include "../vm/luafunc.h"

static void* l_alloc(void* ud, void* ptr, size_t osize, size_t nsize) {
(void)ud;
(void)osize;

// printf("l_alloc nsize:%ld\n", nsize);
if (nsize == 0) {
free(ptr);
return NULL;
}

return realloc(ptr, nsize);
}

struct lua_State* luaL_newstate() {
struct lua_State* L = lua_newstate(&l_alloc, NULL);
return L;
}

void luaL_close(struct lua_State* L) {
lua_close(L);
}

// this function will load c modules into _LOADED table, if glb is true, then the module will also
// add to _G
int luaL_requiref(struct lua_State* L, const char* name, lua_CFunction func, int glb) {
luaL_getsubtable(L, LUA_REGISTRYINDEX, LUA_LOADED);

TValue* top = luaL_index2addr(L, lua_gettop(L));
if (novariant(top) == LUA_TNIL) {
luaL_pop(L);
luaL_createtable(L); // _LOADED = {}
lua_pushvalue(L, -1);
lua_setfield(L, LUA_REGISTRYINDEX, LUA_LOADED); // set _LOADED to registry
}

luaL_getsubtable(L, -1, name);
if (novariant(L->top - 1) == LUA_TNIL) {
luaL_pop(L);

luaL_pushcfunction(L, func);
luaL_pushstring(L, name);
luaL_pcall(L, 1, 1);

lua_pushvalue(L, -1);
lua_setfield(L, -3, name);
}

lua_remove(L, -2); // remove _LOADED

if (glb) {
lua_pushglobaltable(L); // push _G into stack top

TValue* o = index2addr(L, lua_gettop(L));
lua_insert(L, -2, o);
lua_pop(L);

lua_pushstring(L, name);
o = index2addr(L, lua_gettop(L));
lua_insert(L, -2, o);
lua_pop(L);

luaL_settable(L, -3);
}
lua_pop(L); // pop _G

return LUA_OK;
}

void luaL_pushinteger(struct lua_State* L, lua_Integer integer) {
lua_pushinteger(L, integer);
}

void luaL_pushnumber(struct lua_State* L, float number) {
lua_pushnumber(L, number);
}

void luaL_pushlightuserdata(struct lua_State* L, void* userdata) {
lua_pushlightuserdata(L, userdata);
}

void luaL_pushnil(struct lua_State* L) {
lua_pushnil(L);
}

void luaL_pushcfunction(struct lua_State* L, lua_CFunction f) {
lua_pushcfunction(L, f);
}

void luaL_pushboolean(struct lua_State* L, bool boolean) {
lua_pushboolean(L, boolean);
}

void luaL_pushstring(struct lua_State* L, const char* str) {
lua_pushstring(L, str);
}

// function call
typedef struct CallS {
StkId func;
int nresult;
} CallS;

static int f_call(lua_State* L, void* ud) {
CallS* c = cast(CallS*, ud);
luaD_call(L, c->func, c->nresult);
return LUA_OK;
}

int luaL_pcall(struct lua_State* L, int narg, int nresult) {
int status = LUA_OK;
CallS c;
c.func = L->top - (narg + 1);
c.nresult = nresult;

status = luaD_pcall(L, &f_call, &c, savestack(L, L->top), 0);
return status;
}

bool luaL_checkinteger(struct lua_State* L, int idx) {
int isnum = 0;
lua_tointegerx(L, idx, &isnum);
if (isnum) {
return true;
}
else {
return false;
}
}

lua_Integer luaL_tointeger(struct lua_State* L, int idx) {
int isnum = 0;
lua_Integer ret = lua_tointegerx(L, idx, &isnum);
return ret;
}

lua_Number luaL_tonumber(struct lua_State* L, int idx) {
int isnum = 0;
lua_Number ret = lua_tonumberx(L, idx, &isnum);
return ret;
}

void* luaL_touserdata(struct lua_State* L, int idx) {
// TODO
return NULL;
}

bool luaL_toboolean(struct lua_State* L, int idx) {
return lua_toboolean(L, idx);
}

int luaL_isnil(struct lua_State* L, int idx) {
return lua_isnil(L, idx);
}

char* luaL_tostring(struct lua_State* L, int idx) {
return lua_tostring(L, idx);
}

TValue* luaL_index2addr(struct lua_State* L, int idx) {
return index2addr(L, idx);
}

int luaL_createtable(struct lua_State* L) {
return lua_createtable(L);
}

int luaL_settable(struct lua_State* L, int idx) {
return lua_settable(L, idx);
}

int luaL_gettable(struct lua_State* L, int idx) {
return lua_gettable(L, idx);
}

int luaL_getglobal(struct lua_State* L) {
return lua_pushglobaltable(L);
}

int luaL_getsubtable(struct lua_State* L, int idx, const char* name) {
TValue* o = luaL_index2addr(L, idx);
if (novariant(o) != LUA_TTABLE) {
return LUA_ERRERR;
}

struct Table* table = gco2tbl(gcvalue(o));
TValue* subtable = (TValue*)luaH_getstr(L, table, luaS_newliteral(L, name));
if (subtable == luaO_nilobject) {
lua_pushnil(L);
return LUA_OK;
}

setobj(L->top, subtable);
increase_top(L);

return LUA_OK;
}

void luaL_pop(struct lua_State* L) {
lua_pop(L);
}

int luaL_stacksize(struct lua_State* L) {
return lua_gettop(L);
}

static char* getF(struct lua_State* L, void* data, size_t* sz) {
LoadF* lf = (LoadF*)data;
if (lf->n > 0) {
*sz = lf->n;
lf->n = 0;
}
else {
*sz = fread(lf->buff, sizeof(char), BUFSIZE, lf->f);
lf->n = 0;
}

return lf->buff;
}

static void init_upval(struct lua_State* L) {
StkId top = L->top - 1;
LClosure* cl = gco2lclosure(gcvalue(top));
if (cl->nupvalues > 0) {
struct Table* t = gco2tbl(gcvalue(&G(L)->l_registry));
TValue* _G = &t->array[LUA_GLOBALTBLIDX];
setobj(cl->upvals[0]->v, _G);
}
}

int luaL_loadfile(struct lua_State* L, const char* filename) {
FILE* fptr = NULL;
l_fopen(&fptr, filename, "rb");
if (fptr == NULL)
{
return LUA_ERRERR;
}

// init LoadF
LoadF lf;
lf.f = fptr;
lf.n = 0;
memset(lf.buff, 0, BUFSIZE);

int ok = luaD_load(L, getF, &lf, filename);
if (ok == LUA_OK) {
init_upval(L);
}
fclose(fptr);

return ok;
}
67 changes: 67 additions & 0 deletions windows/project/part10/clib/luaaux.h
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,67 @@
/* Copyright (c) 2018 Manistein,https://manistein.github.io/blog/
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.*/

#ifndef luaaux_h
#define luaaux_h

#include "../common/luastate.h"

// a structure for us to load standard libraries
typedef struct lua_Reg {
const char* name;
lua_CFunction func;
}lua_Reg;

struct lua_State* luaL_newstate();
void luaL_close(struct lua_State* L);

void luaL_openlibs(struct lua_State* L); // open standard libraries
int luaL_requiref(struct lua_State* L, const char* name, lua_CFunction func, int glb); // try to load module into _LOADED table

void luaL_pushinteger(struct lua_State* L, lua_Integer integer);
void luaL_pushnumber(struct lua_State* L, float number);
void luaL_pushlightuserdata(struct lua_State* L, void* userdata);
void luaL_pushnil(struct lua_State* L);
void luaL_pushcfunction(struct lua_State* L, lua_CFunction f);
void luaL_pushboolean(struct lua_State* L, bool boolean);
void luaL_pushstring(struct lua_State* L, const char* str);
int luaL_pcall(struct lua_State* L, int narg, int nresult);

bool luaL_checkinteger(struct lua_State* L, int idx);
lua_Integer luaL_tointeger(struct lua_State* L, int idx);
lua_Number luaL_tonumber(struct lua_State* L, int idx);
void* luaL_touserdata(struct lua_State* L, int idx);
bool luaL_toboolean(struct lua_State* L, int idx);
char* luaL_tostring(struct lua_State* L, int index);
int luaL_isnil(struct lua_State* L, int idx);
TValue* luaL_index2addr(struct lua_State* L, int idx);

int luaL_createtable(struct lua_State* L); // create a new table, and push into the stack
int luaL_settable(struct lua_State* L, int idx);
int luaL_gettable(struct lua_State* L, int idx);
int luaL_getglobal(struct lua_State* L);
int luaL_getsubtable(struct lua_State* L, int idx, const char* name); // try to get a sub table by name and push into the stack

void luaL_pop(struct lua_State* L);
int luaL_stacksize(struct lua_State* L);

// load source and compile, if load success, then it returns 1, otherwise it returns 0
int luaL_loadfile(struct lua_State* L, const char* filename);
#endif
Loading

0 comments on commit b365b1c

Please sign in to comment.