xref: /illumos-gate/usr/src/uts/common/fs/zfs/lua/ltm.c (revision dfc11533)
1*dfc11533SChris Williamson /*
2*dfc11533SChris Williamson ** $Id: ltm.c,v 2.14.1.1 2013/04/12 18:48:47 roberto Exp $
3*dfc11533SChris Williamson ** Tag methods
4*dfc11533SChris Williamson ** See Copyright Notice in lua.h
5*dfc11533SChris Williamson */
6*dfc11533SChris Williamson 
7*dfc11533SChris Williamson 
8*dfc11533SChris Williamson #include <sys/zfs_context.h>
9*dfc11533SChris Williamson 
10*dfc11533SChris Williamson #define ltm_c
11*dfc11533SChris Williamson #define LUA_CORE
12*dfc11533SChris Williamson 
13*dfc11533SChris Williamson #include "lua.h"
14*dfc11533SChris Williamson 
15*dfc11533SChris Williamson #include "lobject.h"
16*dfc11533SChris Williamson #include "lstate.h"
17*dfc11533SChris Williamson #include "lstring.h"
18*dfc11533SChris Williamson #include "ltable.h"
19*dfc11533SChris Williamson #include "ltm.h"
20*dfc11533SChris Williamson 
21*dfc11533SChris Williamson 
22*dfc11533SChris Williamson static const char udatatypename[] = "userdata";
23*dfc11533SChris Williamson 
24*dfc11533SChris Williamson LUAI_DDEF const char *const luaT_typenames_[LUA_TOTALTAGS] = {
25*dfc11533SChris Williamson   "no value",
26*dfc11533SChris Williamson   "nil", "boolean", udatatypename, "number",
27*dfc11533SChris Williamson   "string", "table", "function", udatatypename, "thread",
28*dfc11533SChris Williamson   "proto", "upval"  /* these last two cases are used for tests only */
29*dfc11533SChris Williamson };
30*dfc11533SChris Williamson 
31*dfc11533SChris Williamson 
luaT_init(lua_State * L)32*dfc11533SChris Williamson void luaT_init (lua_State *L) {
33*dfc11533SChris Williamson   static const char *const luaT_eventname[] = {  /* ORDER TM */
34*dfc11533SChris Williamson     "__index", "__newindex",
35*dfc11533SChris Williamson     "__gc", "__mode", "__len", "__eq",
36*dfc11533SChris Williamson     "__add", "__sub", "__mul", "__div", "__mod",
37*dfc11533SChris Williamson     "__pow", "__unm", "__lt", "__le",
38*dfc11533SChris Williamson     "__concat", "__call"
39*dfc11533SChris Williamson   };
40*dfc11533SChris Williamson   int i;
41*dfc11533SChris Williamson   for (i=0; i<TM_N; i++) {
42*dfc11533SChris Williamson     G(L)->tmname[i] = luaS_new(L, luaT_eventname[i]);
43*dfc11533SChris Williamson     luaS_fix(G(L)->tmname[i]);  /* never collect these names */
44*dfc11533SChris Williamson   }
45*dfc11533SChris Williamson }
46*dfc11533SChris Williamson 
47*dfc11533SChris Williamson 
48*dfc11533SChris Williamson /*
49*dfc11533SChris Williamson ** function to be used with macro "fasttm": optimized for absence of
50*dfc11533SChris Williamson ** tag methods
51*dfc11533SChris Williamson */
luaT_gettm(Table * events,TMS event,TString * ename)52*dfc11533SChris Williamson const TValue *luaT_gettm (Table *events, TMS event, TString *ename) {
53*dfc11533SChris Williamson   const TValue *tm = luaH_getstr(events, ename);
54*dfc11533SChris Williamson   lua_assert(event <= TM_EQ);
55*dfc11533SChris Williamson   if (ttisnil(tm)) {  /* no tag method? */
56*dfc11533SChris Williamson     events->flags |= cast_byte(1u<<event);  /* cache this fact */
57*dfc11533SChris Williamson     return NULL;
58*dfc11533SChris Williamson   }
59*dfc11533SChris Williamson   else return tm;
60*dfc11533SChris Williamson }
61*dfc11533SChris Williamson 
62*dfc11533SChris Williamson 
luaT_gettmbyobj(lua_State * L,const TValue * o,TMS event)63*dfc11533SChris Williamson const TValue *luaT_gettmbyobj (lua_State *L, const TValue *o, TMS event) {
64*dfc11533SChris Williamson   Table *mt;
65*dfc11533SChris Williamson   switch (ttypenv(o)) {
66*dfc11533SChris Williamson     case LUA_TTABLE:
67*dfc11533SChris Williamson       mt = hvalue(o)->metatable;
68*dfc11533SChris Williamson       break;
69*dfc11533SChris Williamson     case LUA_TUSERDATA:
70*dfc11533SChris Williamson       mt = uvalue(o)->metatable;
71*dfc11533SChris Williamson       break;
72*dfc11533SChris Williamson     default:
73*dfc11533SChris Williamson       mt = G(L)->mt[ttypenv(o)];
74*dfc11533SChris Williamson   }
75*dfc11533SChris Williamson   return (mt ? luaH_getstr(mt, G(L)->tmname[event]) : luaO_nilobject);
76*dfc11533SChris Williamson }
77*dfc11533SChris Williamson 
78