src/lj_lib.h

   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
  33 LJ_FUNC TValue *lj_lib_checkany(lua_State *L, int narg);
  34 LJ_FUNC GCstr *lj_lib_checkstr(lua_State *L, int narg);
  35 LJ_FUNC GCstr *lj_lib_optstr(lua_State *L, int narg);
  36 #if LJ_DUALNUM
  37 LJ_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
  41 LJ_FUNC lua_Number lj_lib_checknum(lua_State *L, int narg);
  42 LJ_FUNC int32_t lj_lib_checkint(lua_State *L, int narg);
  43 LJ_FUNC int32_t lj_lib_optint(lua_State *L, int narg, int32_t def);
  44 LJ_FUNC GCfunc *lj_lib_checkfunc(lua_State *L, int narg);
  45 LJ_FUNC GCtab *lj_lib_checktab(lua_State *L, int narg);
  46 LJ_FUNC GCtab *lj_lib_checktabornil(lua_State *L, int narg);
  47 LJ_FUNC int lj_lib_checkopt(lua_State *L, int narg, int def, const char *lst);
  48
  49 /* Avoid including lj_frame.h. */
  50 #define lj_lib_upvalue(L, n) \
  51   (&gcref((L->base-1)->fr.func)->fn.c.upvalue[(n)-1])
  52
  53 #if LJ_TARGET_WINDOWS
  54 #define lj_lib_checkfpu(L) \
  55   do { setnumV(L->top++, (lua_Number)1437217655); \
  56     if (lua_tointeger(L, -1) != 1437217655) lj_err_caller(L, LJ_ERR_BADFPU); \
  57     L->top--; } while (0)
  58 #else
  59 #define lj_lib_checkfpu(L)      UNUSED(L)
  60 #endif
  61
  62 LJ_FUNC GCfunc *lj_lib_pushcc(lua_State *L, lua_CFunction f, int id, int n);
  63 #define lj_lib_pushcf(L, fn, id)        (lj_lib_pushcc(L, (fn), (id), 0))
  64
  65 /* Library function declarations. Scanned by buildvm. */
  66 #define LJLIB_CF(name)          static int lj_cf_##name(lua_State *L)
  67 #define LJLIB_ASM(name)         static int lj_ffh_##name(lua_State *L)
  68 #define LJLIB_ASM_(name)
  69 #define LJLIB_LUA(name)
  70 #define LJLIB_SET(name)
  71 #define LJLIB_PUSH(arg)
  72 #define LJLIB_REC(handler)
  73 #define LJLIB_NOREGUV
  74 #define LJLIB_NOREG
  75
  76 #define LJ_LIB_REG(L, regname, name) \
  77   lj_lib_register(L, regname, lj_lib_init_##name, lj_lib_cf_##name)
  78
  79 LJ_FUNC void lj_lib_register(lua_State *L, const char *libname,
  80                              const uint8_t *init, const lua_CFunction *cf);
  81 LJ_FUNC void lj_lib_prereg(lua_State *L, const char *name, lua_CFunction f,
  82                            GCtab *env);
  83 LJ_FUNC int lj_lib_postreg(lua_State *L, lua_CFunction cf, int id,
  84                            const char *name);
  85
  86 /* Library init data tags. */
  87 #define LIBINIT_LENMASK 0x3f
  88 #define LIBINIT_TAGMASK 0xc0
  89 #define LIBINIT_CF      0x00
  90 #define LIBINIT_ASM     0x40
  91 #define LIBINIT_ASM_    0x80
  92 #define LIBINIT_STRING  0xc0
  93 #define LIBINIT_MAXSTR  0x38
  94 #define LIBINIT_LUA     0xf9
  95 #define LIBINIT_SET     0xfa
  96 #define LIBINIT_NUMBER  0xfb
  97 #define LIBINIT_COPY    0xfc
  98 #define LIBINIT_LASTCL  0xfd
  99 #define LIBINIT_FFID    0xfe
 100 #define LIBINIT_END     0xff
 101
 102 /* Exported library functions. */
 103
 104 typedef struct RandomState RandomState;
 105 LJ_FUNC uint64_t LJ_FASTCALL lj_math_random_step(RandomState *rs);
 106
 107 #endif