3 ** Copyright (C) 2005-2017 Mike Pall. See Copyright Notice in luajit.h
23 #include "lj_bcdump.h"
26 /* -- Load Lua source code and bytecode ----------------------------------- */
28 static TValue
*cpparser(lua_State
*L
, lua_CFunction dummy
, void *ud
)
30 LexState
*ls
= (LexState
*)ud
;
35 cframe_errfunc(L
->cframe
) = -1; /* Inherit error function. */
36 bc
= lj_lex_setup(L
, ls
);
37 if (ls
->mode
&& !strchr(ls
->mode
, bc
? 'b' : 't')) {
38 setstrV(L
, L
->top
++, lj_err_str(L
, LJ_ERR_XMODE
));
39 lj_err_throw(L
, LUA_ERRSYNTAX
);
41 pt
= bc
? lj_bcread(ls
) : lj_parse(ls
);
42 fn
= lj_func_newL_empty(L
, pt
, tabref(L
->env
));
43 /* Don't combine above/below into one statement. */
44 setfuncV(L
, L
->top
++, fn
);
48 LUA_API
int lua_loadx(lua_State
*L
, lua_Reader reader
, void *data
,
49 const char *chunkname
, const char *mode
)
55 ls
.chunkarg
= chunkname
? chunkname
: "?";
57 lj_str_initbuf(&ls
.sb
);
58 status
= lj_vm_cpcall(L
, NULL
, &ls
, cpparser
);
59 lj_lex_cleanup(L
, &ls
);
64 LUA_API
int lua_load(lua_State
*L
, lua_Reader reader
, void *data
,
65 const char *chunkname
)
67 return lua_loadx(L
, reader
, data
, chunkname
, NULL
);
70 typedef struct FileReaderCtx
{
72 char buf
[LUAL_BUFFERSIZE
];
75 static const char *reader_file(lua_State
*L
, void *ud
, size_t *size
)
77 FileReaderCtx
*ctx
= (FileReaderCtx
*)ud
;
79 if (feof(ctx
->fp
)) return NULL
;
80 *size
= fread(ctx
->buf
, 1, sizeof(ctx
->buf
), ctx
->fp
);
81 return *size
> 0 ? ctx
->buf
: NULL
;
84 LUALIB_API
int luaL_loadfilex(lua_State
*L
, const char *filename
,
89 const char *chunkname
;
91 ctx
.fp
= fopen(filename
, "rb");
93 lua_pushfstring(L
, "cannot open %s: %s", filename
, strerror(errno
));
96 chunkname
= lua_pushfstring(L
, "@%s", filename
);
101 status
= lua_loadx(L
, reader_file
, &ctx
, chunkname
, mode
);
102 if (ferror(ctx
.fp
)) {
103 L
->top
-= filename
? 2 : 1;
104 lua_pushfstring(L
, "cannot read %s: %s", chunkname
+1, strerror(errno
));
111 copyTV(L
, L
->top
-1, L
->top
);
117 LUALIB_API
int luaL_loadfile(lua_State
*L
, const char *filename
)
119 return luaL_loadfilex(L
, filename
, NULL
);
122 typedef struct StringReaderCtx
{
127 static const char *reader_string(lua_State
*L
, void *ud
, size_t *size
)
129 StringReaderCtx
*ctx
= (StringReaderCtx
*)ud
;
131 if (ctx
->size
== 0) return NULL
;
137 LUALIB_API
int luaL_loadbufferx(lua_State
*L
, const char *buf
, size_t size
,
138 const char *name
, const char *mode
)
143 return lua_loadx(L
, reader_string
, &ctx
, name
, mode
);
146 LUALIB_API
int luaL_loadbuffer(lua_State
*L
, const char *buf
, size_t size
,
149 return luaL_loadbufferx(L
, buf
, size
, name
, NULL
);
152 LUALIB_API
int luaL_loadstring(lua_State
*L
, const char *s
)
154 return luaL_loadbuffer(L
, s
, strlen(s
), s
);
157 /* -- Dump bytecode ------------------------------------------------------- */
159 LUA_API
int lua_dump(lua_State
*L
, lua_Writer writer
, void *data
)
161 cTValue
*o
= L
->top
-1;
162 api_check(L
, L
->top
> L
->base
);
163 if (tvisfunc(o
) && isluafunc(funcV(o
)))
164 return lj_bcwrite(L
, funcproto(funcV(o
)), writer
, data
, 0);