1/*
2** Library function support.
3** Copyright (C) 2005-2014 Mike Pall. See Copyright Notice in luajit.h
4*/
5
6#ifndef _LJ_LIB_H
7#define _LJ_LIB_H
8
9#include "lj_obj.h"
10
11/*
12** A fallback handler is called by the assembler VM if the fast path fails:
13**
14** - too few arguments: unrecoverable.
15** - wrong argument type: recoverable, if coercion succeeds.
16** - bad argument value: unrecoverable.
17** - stack overflow: recoverable, if stack reallocation succeeds.
18** - extra handling: recoverable.
19**
20** The unrecoverable cases throw an error with lj_err_arg(), lj_err_argtype(),
21** lj_err_caller() or lj_err_callermsg().
22** The recoverable cases return 0 or the number of results + 1.
23** The assembler VM retries the fast path only if 0 is returned.
24** This time the fallback must not be called again or it gets stuck in a loop.
25*/
26
27/* Return values from fallback handler. */
28#define FFH_RETRY 0
29#define FFH_UNREACHABLE FFH_RETRY
30#define FFH_RES(n) ((n)+1)
31#define FFH_TAILCALL (-1)
32
33LJ_FUNC TValue *lj_lib_checkany(lua_State *L, int narg);
34LJ_FUNC GCstr *lj_lib_checkstr(lua_State *L, int narg);
35LJ_FUNC GCstr *lj_lib_optstr(lua_State *L, int narg);
36#if LJ_DUALNUM
37LJ_FUNC void lj_lib_checknumber(lua_State *L, int narg);
38#else
39#define lj_lib_checknumber(L, narg) lj_lib_checknum((L), (narg))
40#endif
41LJ_FUNC lua_Number lj_lib_checknum(lua_State *L, int narg);
42LJ_FUNC int32_t lj_lib_checkint(lua_State *L, int narg);
43LJ_FUNC int32_t lj_lib_optint(lua_State *L, int narg, int32_t def);
44LJ_FUNC int32_t lj_lib_checkbit(lua_State *L, int narg);
45LJ_FUNC GCfunc *lj_lib_checkfunc(lua_State *L, int narg);
46LJ_FUNC GCtab *lj_lib_checktab(lua_State *L, int narg);
47LJ_FUNC GCtab *lj_lib_checktabornil(lua_State *L, int narg);
48LJ_FUNC int lj_lib_checkopt(lua_State *L, int narg, int def, const char *lst);
49
50/* Avoid including lj_frame.h. */
51#define lj_lib_upvalue(L, n) \
52 (&gcref((L->base-1)->fr.func)->fn.c.upvalue[(n)-1])
53
54#if LJ_TARGET_WINDOWS
55#define lj_lib_checkfpu(L) \
56 do { setnumV(L->top++, (lua_Number)1437217655); \
57 if (lua_tointeger(L, -1) != 1437217655) lj_err_caller(L, LJ_ERR_BADFPU); \
58 L->top--; } while (0)
59#else
60#define lj_lib_checkfpu(L) UNUSED(L)
61#endif
62
63/* Push internal function on the stack. */
64static LJ_AINLINE void lj_lib_pushcc(lua_State *L, lua_CFunction f,
65 int id, int n)
66{
67 GCfunc *fn;
68 lua_pushcclosure(L, f, n);
69 fn = funcV(L->top-1);
70 fn->c.ffid = (uint8_t)id;
71 setmref(fn->c.pc, &G(L)->bc_cfunc_int);
72}
73
74#define lj_lib_pushcf(L, fn, id) (lj_lib_pushcc(L, (fn), (id), 0))
75
76/* Library function declarations. Scanned by buildvm. */
77#define LJLIB_CF(name) static int lj_cf_##name(lua_State *L)
78#define LJLIB_ASM(name) static int lj_ffh_##name(lua_State *L)
79#define LJLIB_ASM_(name)
80#define LJLIB_SET(name)
81#define LJLIB_PUSH(arg)
82#define LJLIB_REC(handler)
83#define LJLIB_NOREGUV
84#define LJLIB_NOREG
85
86#define LJ_LIB_REG(L, regname, name) \
87 lj_lib_register(L, regname, lj_lib_init_##name, lj_lib_cf_##name)
88
89LJ_FUNC void lj_lib_register(lua_State *L, const char *libname,
90 const uint8_t *init, const lua_CFunction *cf);
91
92/* Library init data tags. */
93#define LIBINIT_LENMASK 0x3f
94#define LIBINIT_TAGMASK 0xc0
95#define LIBINIT_CF 0x00
96#define LIBINIT_ASM 0x40
97#define LIBINIT_ASM_ 0x80
98#define LIBINIT_STRING 0xc0
99#define LIBINIT_MAXSTR 0x39
100#define LIBINIT_SET 0xfa
101#define LIBINIT_NUMBER 0xfb
102#define LIBINIT_COPY 0xfc
103#define LIBINIT_LASTCL 0xfd
104#define LIBINIT_FFID 0xfe
105#define LIBINIT_END 0xff
106
107/* Exported library functions. */
108
109typedef struct RandomState RandomState;
110LJ_FUNC uint64_t LJ_FASTCALL lj_math_random_step(RandomState *rs);
111
112#endif
113