/src/testdir/build/lua-master/source/ltm.c

Source (jump to first uncovered line)
/*
** $Id: ltm.c $
** Tag methods
** See Copyright Notice in lua.h
*/

#define ltm_c
#define LUA_CORE

#include "lprefix.h"


#include <string.h>

#include "lua.h"

#include "ldebug.h"
#include "ldo.h"
#include "lgc.h"
#include "lobject.h"
#include "lstate.h"
#include "lstring.h"
#include "ltable.h"
#include "ltm.h"
#include "lvm.h"


static const char udatatypename[] = "userdata";

LUAI_DDEF const char *const luaT_typenames_[LUA_TOTALTYPES] = {
  "no value",
  "nil", "boolean", udatatypename, "number",
  "string", "table", "function", udatatypename, "thread",
  "upvalue", "proto" /* these last cases are used for tests only */
};


void luaT_init (lua_State *L) {
  static const char *const luaT_eventname[] = {  /* ORDER TM */
    "__index", "__newindex",
    "__gc", "__mode", "__len", "__eq",
    "__add", "__sub", "__mul", "__mod", "__pow",
    "__div", "__idiv",
    "__band", "__bor", "__bxor", "__shl", "__shr",
    "__unm", "__bnot", "__lt", "__le",
    "__concat", "__call", "__close"
  };
  int i;
  for (i=0; i<TM_N; i++) {
    G(L)->tmname[i] = luaS_new(L, luaT_eventname[i]);
    luaC_fix(L, obj2gco(G(L)->tmname[i]));  /* never collect these names */
  }
}


/*
** function to be used with macro "fasttm": optimized for absence of
** tag methods
*/
const TValue *luaT_gettm (Table *events, TMS event, TString *ename) {
  const TValue *tm = luaH_getshortstr(events, ename);
  lua_assert(event <= TM_EQ);
  if (notm(tm)) {  /* no tag method? */
    events->flags |= cast_byte(1u<<event);  /* cache this fact */
    return NULL;
  }
  else return tm;
}


const TValue *luaT_gettmbyobj (lua_State *L, const TValue *o, TMS event) {
  Table *mt;
  switch (ttype(o)) {
    case LUA_TTABLE:
      mt = hvalue(o)->metatable;
      break;
    case LUA_TUSERDATA:
      mt = uvalue(o)->metatable;
      break;
    default:
      mt = G(L)->mt[ttype(o)];
  }
  return (mt ? luaH_getshortstr(mt, G(L)->tmname[event]) : &G(L)->nilvalue);
}


/*
** Return the name of the type of an object. For tables and userdata
** with metatable, use their '__name' metafield, if present.
*/
const char *luaT_objtypename (lua_State *L, const TValue *o) {
  Table *mt;
  if ((ttistable(o) && (mt = hvalue(o)->metatable) != NULL) ||
      (ttisfulluserdata(o) && (mt = uvalue(o)->metatable) != NULL)) {
    const TValue *name = luaH_getshortstr(mt, luaS_new(L, "__name"));
    if (ttisstring(name))  /* is '__name' a string? */
      return getstr(tsvalue(name));  /* use it as type name */
  }
  return ttypename(ttype(o));  /* else use standard type name */
}


void luaT_callTM (lua_State *L, const TValue *f, const TValue *p1,
                  const TValue *p2, const TValue *p3) {
  StkId func = L->top.p;
  setobj2s(L, func, f);  /* push function (assume EXTRA_STACK) */
  setobj2s(L, func + 1, p1);  /* 1st argument */
  setobj2s(L, func + 2, p2);  /* 2nd argument */
  setobj2s(L, func + 3, p3);  /* 3rd argument */
  L->top.p = func + 4;
  /* metamethod may yield only when called from Lua code */
  if (isLuacode(L->ci))
    luaD_call(L, func, 0);
  else
    luaD_callnoyield(L, func, 0);
}


void luaT_callTMres (lua_State *L, const TValue *f, const TValue *p1,
                     const TValue *p2, StkId res) {
  ptrdiff_t result = savestack(L, res);
  StkId func = L->top.p;
  setobj2s(L, func, f);  /* push function (assume EXTRA_STACK) */
  setobj2s(L, func + 1, p1);  /* 1st argument */
  setobj2s(L, func + 2, p2);  /* 2nd argument */
  L->top.p += 3;
  /* metamethod may yield only when called from Lua code */
  if (isLuacode(L->ci))
    luaD_call(L, func, 1);
  else
    luaD_callnoyield(L, func, 1);
  res = restorestack(L, result);
  setobjs2s(L, res, --L->top.p);  /* move result to its place */
}


static int callbinTM (lua_State *L, const TValue *p1, const TValue *p2,
                      StkId res, TMS event) {
  const TValue *tm = luaT_gettmbyobj(L, p1, event);  /* try first operand */
  if (notm(tm))
    tm = luaT_gettmbyobj(L, p2, event);  /* try second operand */
  if (notm(tm)) return 0;
  luaT_callTMres(L, tm, p1, p2, res);
  return 1;
}


void luaT_trybinTM (lua_State *L, const TValue *p1, const TValue *p2,
                    StkId res, TMS event) {
  if (l_unlikely(!callbinTM(L, p1, p2, res, event))) {
    switch (event) {
      case TM_BAND: case TM_BOR: case TM_BXOR:
      case TM_SHL: case TM_SHR: case TM_BNOT: {
        if (ttisnumber(p1) && ttisnumber(p2))
          luaG_tointerror(L, p1, p2);
        else
          luaG_opinterror(L, p1, p2, "perform bitwise operation on");
      }
      /* calls never return, but to avoid warnings: *//* FALLTHROUGH */
      default:
        luaG_opinterror(L, p1, p2, "perform arithmetic on");
    }
  }
}


void luaT_tryconcatTM (lua_State *L) {
  StkId top = L->top.p;
  if (l_unlikely(!callbinTM(L, s2v(top - 2), s2v(top - 1), top - 2,
                               TM_CONCAT)))
    luaG_concaterror(L, s2v(top - 2), s2v(top - 1));
}


void luaT_trybinassocTM (lua_State *L, const TValue *p1, const TValue *p2,
                                       int flip, StkId res, TMS event) {
  if (flip)
    luaT_trybinTM(L, p2, p1, res, event);
  else
    luaT_trybinTM(L, p1, p2, res, event);
}


void luaT_trybiniTM (lua_State *L, const TValue *p1, lua_Integer i2,
                                   int flip, StkId res, TMS event) {
  TValue aux;
  setivalue(&aux, i2);
  luaT_trybinassocTM(L, p1, &aux, flip, res, event);
}


/*
** Calls an order tag method.
** For lessequal, LUA_COMPAT_LT_LE keeps compatibility with old
** behavior: if there is no '__le', try '__lt', based on l <= r iff
** !(r < l) (assuming a total order). If the metamethod yields during
** this substitution, the continuation has to know about it (to negate
** the result of r<l); bit CIST_LEQ in the call status keeps that
** information.
*/
int luaT_callorderTM (lua_State *L, const TValue *p1, const TValue *p2,
                      TMS event) {
  if (callbinTM(L, p1, p2, L->top.p, event))  /* try original event */
    return !l_isfalse(s2v(L->top.p));
#if defined(LUA_COMPAT_LT_LE)
  else if (event == TM_LE) {
      /* try '!(p2 < p1)' for '(p1 <= p2)' */
      L->ci->callstatus |= CIST_LEQ;  /* mark it is doing 'lt' for 'le' */
      if (callbinTM(L, p2, p1, L->top.p, TM_LT)) {
        L->ci->callstatus ^= CIST_LEQ;  /* clear mark */
        return l_isfalse(s2v(L->top.p));
      }
      /* else error will remove this 'ci'; no need to clear mark */
  }
#endif
  luaG_ordererror(L, p1, p2);  /* no metamethod found */
  return 0;  /* to avoid warnings */
}


int luaT_callorderiTM (lua_State *L, const TValue *p1, int v2,
                       int flip, int isfloat, TMS event) {
  TValue aux; const TValue *p2;
  if (isfloat) {
    setfltvalue(&aux, cast_num(v2));
  }
  else
    setivalue(&aux, v2);
  if (flip) {  /* arguments were exchanged? */
    p2 = p1; p1 = &aux;  /* correct them */
  }
  else
    p2 = &aux;
  return luaT_callorderTM(L, p1, p2, event);
}


void luaT_adjustvarargs (lua_State *L, int nfixparams, CallInfo *ci,
                         const Proto *p) {
  int i;
  int actual = cast_int(L->top.p - ci->func.p) - 1;  /* number of arguments */
  int nextra = actual - nfixparams;  /* number of extra arguments */
  ci->u.l.nextraargs = nextra;
  luaD_checkstack(L, p->maxstacksize + 1);
  /* copy function to the top of the stack */
  setobjs2s(L, L->top.p++, ci->func.p);
  /* move fixed parameters to the top of the stack */
  for (i = 1; i <= nfixparams; i++) {
    setobjs2s(L, L->top.p++, ci->func.p + i);
    setnilvalue(s2v(ci->func.p + i));  /* erase original parameter (for GC) */
  }
  ci->func.p += actual + 1;
  ci->top.p += actual + 1;
  lua_assert(L->top.p <= ci->top.p && ci->top.p <= L->stack_last.p);
}


void luaT_getvarargs (lua_State *L, CallInfo *ci, StkId where, int wanted) {
  int i;
  int nextra = ci->u.l.nextraargs;
  if (wanted < 0) {
    wanted = nextra;  /* get all extra arguments available */
    checkstackGCp(L, nextra, where);  /* ensure stack space */
    L->top.p = where + nextra;  /* next instruction will need top */
  }
  for (i = 0; i < wanted && i < nextra; i++)
    setobjs2s(L, where + i, ci->func.p - nextra + i);
  for (; i < wanted; i++)   /* complete required results with nil */
    setnilvalue(s2v(where + i));
}


Coverage Report

Created: 2023-09-15 06:20

Line	Count	Source (jump to first uncovered line)
1		/*
2		** $Id: ltm.c $
3		** Tag methods
4		** See Copyright Notice in lua.h
5		*/
6
7		#define ltm_c
8		#define LUA_CORE
9
10		#include "lprefix.h"
11
12
13		#include <string.h>
14
15		#include "lua.h"
16
17		#include "ldebug.h"
18		#include "ldo.h"
19		#include "lgc.h"
20		#include "lobject.h"
21		#include "lstate.h"
22		#include "lstring.h"
23		#include "ltable.h"
24		#include "ltm.h"
25		#include "lvm.h"
26
27
28		static const char udatatypename[] = "userdata";
29
30		LUAI_DDEF const char *const luaT_typenames_[LUA_TOTALTYPES] = {
31		"no value",
32		"nil", "boolean", udatatypename, "number",
33		"string", "table", "function", udatatypename, "thread",
34		"upvalue", "proto" /* these last cases are used for tests only */
35		};
36
37
38	390	void luaT_init (lua_State *L) {
39	390	static const char const luaT_eventname[] = { / ORDER TM */
40	390	"__index", "__newindex",
41	390	"__gc", "__mode", "__len", "__eq",
42	390	"__add", "__sub", "__mul", "__mod", "__pow",
43	390	"__div", "__idiv",
44	390	"__band", "__bor", "__bxor", "__shl", "__shr",
45	390	"__unm", "__bnot", "__lt", "__le",
46	390	"__concat", "__call", "__close"
47	390	};
48	390	int i;
49	10.1k	for (i=0; i<TM_N; i++) {
50	9.75k	G(L)->tmname[i] = luaS_new(L, luaT_eventname[i]);
51	9.75k	luaC_fix(L, obj2gco(G(L)->tmname[i])); /* never collect these names */
52	9.75k	}
53	390	}
54
55
56		/*
57		** function to be used with macro "fasttm": optimized for absence of
58		** tag methods
59		*/
60	16	const TValue luaT_gettm (Table events, TMS event, TString *ename) {
61	16	const TValue *tm = luaH_getshortstr(events, ename);
62	16	lua_assert(event <= TM_EQ);
63	16	if (notm(tm)) { /* no tag method? */
64	0	events->flags \|= cast_byte(1u<<event); /* cache this fact */
65	0	return NULL;
66	0	}
67	16	else return tm;
68	16	}
69
70
71	151	const TValue luaT_gettmbyobj (lua_State L, const TValue *o, TMS event) {
72	151	Table *mt;
73	151	switch (ttype(o)) {
74	0	case LUA_TTABLE:
75	0	mt = hvalue(o)->metatable;
76	0	break;
77	48	case LUA_TUSERDATA:
78	48	mt = uvalue(o)->metatable;
79	48	break;
80	103	default:
81	103	mt = G(L)->mt[ttype(o)];
82	151	}
83	151	return (mt ? luaH_getshortstr(mt, G(L)->tmname[event]) : &G(L)->nilvalue);
84	151	}
85
86
87		/*
88		** Return the name of the type of an object. For tables and userdata
89		** with metatable, use their '__name' metafield, if present.
90		*/
91	55	const char luaT_objtypename (lua_State L, const TValue *o) {
92	55	Table *mt;
93	55	if ((ttistable(o) && (mt = hvalue(o)->metatable) != NULL) \|\|
94	55	(ttisfulluserdata(o) && (mt = uvalue(o)->metatable) != NULL)) {
95	0	const TValue *name = luaH_getshortstr(mt, luaS_new(L, "__name"));
96	0	if (ttisstring(name)) /* is '__name' a string? */
97	0	return getstr(tsvalue(name)); /* use it as type name */
98	0	}
99	55	return ttypename(ttype(o)); /* else use standard type name */
100	55	}
101
102
103		void luaT_callTM (lua_State L, const TValue f, const TValue *p1,
104	0	const TValue p2, const TValue p3) {
105	0	StkId func = L->top.p;
106	0	setobj2s(L, func, f); /* push function (assume EXTRA_STACK) */
107	0	setobj2s(L, func + 1, p1); /* 1st argument */
108	0	setobj2s(L, func + 2, p2); /* 2nd argument */
109	0	setobj2s(L, func + 3, p3); /* 3rd argument */
110	0	L->top.p = func + 4;
111		/* metamethod may yield only when called from Lua code */
112	0	if (isLuacode(L->ci))
113	0	luaD_call(L, func, 0);
114	0	else
115	0	luaD_callnoyield(L, func, 0);
116	0	}
117
118
119		void luaT_callTMres (lua_State L, const TValue f, const TValue *p1,
120	0	const TValue *p2, StkId res) {
121	0	ptrdiff_t result = savestack(L, res);
122	0	StkId func = L->top.p;
123	0	setobj2s(L, func, f); /* push function (assume EXTRA_STACK) */
124	0	setobj2s(L, func + 1, p1); /* 1st argument */
125	0	setobj2s(L, func + 2, p2); /* 2nd argument */
126	0	L->top.p += 3;
127		/* metamethod may yield only when called from Lua code */
128	0	if (isLuacode(L->ci))
129	0	luaD_call(L, func, 1);
130	0	else
131	0	luaD_callnoyield(L, func, 1);
132	0	res = restorestack(L, result);
133	0	setobjs2s(L, res, --L->top.p); /* move result to its place */
134	0	}
135
136
137		static int callbinTM (lua_State L, const TValue p1, const TValue *p2,
138	50	StkId res, TMS event) {
139	50	const TValue tm = luaT_gettmbyobj(L, p1, event); / try first operand */
140	50	if (notm(tm))
141	50	tm = luaT_gettmbyobj(L, p2, event); /* try second operand */
142	50	if (notm(tm)) return 0;
143	0	luaT_callTMres(L, tm, p1, p2, res);
144	0	return 1;
145	50	}
146
147
148		void luaT_trybinTM (lua_State L, const TValue p1, const TValue *p2,
149	47	StkId res, TMS event) {
150	47	if (l_unlikely(!callbinTM(L, p1, p2, res, event))) {
151	47	switch (event) {
152	16	case TM_BAND: case TM_BOR: case TM_BXOR:
153	38	case TM_SHL: case TM_SHR: case TM_BNOT: {
154	38	if (ttisnumber(p1) && ttisnumber(p2))
155	3	luaG_tointerror(L, p1, p2);
156	35	else
157	35	luaG_opinterror(L, p1, p2, "perform bitwise operation on");
158	38	}
159		/* calls never return, but to avoid warnings: // FALLTHROUGH */
160	9	default:
161	9	luaG_opinterror(L, p1, p2, "perform arithmetic on");
162	47	}
163	47	}
164	47	}
165
166
167	1	void luaT_tryconcatTM (lua_State *L) {
168	1	StkId top = L->top.p;
169	1	if (l_unlikely(!callbinTM(L, s2v(top - 2), s2v(top - 1), top - 2,
170	1	TM_CONCAT)))
171	1	luaG_concaterror(L, s2v(top - 2), s2v(top - 1));
172	1	}
173
174
175		void luaT_trybinassocTM (lua_State L, const TValue p1, const TValue *p2,
176	5	int flip, StkId res, TMS event) {
177	5	if (flip)
178	1	luaT_trybinTM(L, p2, p1, res, event);
179	4	else
180	4	luaT_trybinTM(L, p1, p2, res, event);
181	5	}
182
183
184		void luaT_trybiniTM (lua_State L, const TValue p1, lua_Integer i2,
185	1	int flip, StkId res, TMS event) {
186	1	TValue aux;
187	1	setivalue(&aux, i2);
188	1	luaT_trybinassocTM(L, p1, &aux, flip, res, event);
189	1	}
190
191
192		/*
193		** Calls an order tag method.
194		** For lessequal, LUA_COMPAT_LT_LE keeps compatibility with old
195		** behavior: if there is no '__le', try '__lt', based on l <= r iff
196		** !(r < l) (assuming a total order). If the metamethod yields during
197		** this substitution, the continuation has to know about it (to negate
198		** the result of r<l); bit CIST_LEQ in the call status keeps that
199		** information.
200		*/
201		int luaT_callorderTM (lua_State L, const TValue p1, const TValue *p2,
202	2	TMS event) {
203	2	if (callbinTM(L, p1, p2, L->top.p, event)) /* try original event */
204	0	return !l_isfalse(s2v(L->top.p));
205		#if defined(LUA_COMPAT_LT_LE)
206		else if (event == TM_LE) {
207		/* try '!(p2 < p1)' for '(p1 <= p2)' */
208		L->ci->callstatus \|= CIST_LEQ; /* mark it is doing 'lt' for 'le' */
209		if (callbinTM(L, p2, p1, L->top.p, TM_LT)) {
210		L->ci->callstatus ^= CIST_LEQ; /* clear mark */
211		return l_isfalse(s2v(L->top.p));
212		}
213		/* else error will remove this 'ci'; no need to clear mark */
214		}
215		#endif
216	2	luaG_ordererror(L, p1, p2); /* no metamethod found */
217	0	return 0; /* to avoid warnings */
218	2	}
219
220
221		int luaT_callorderiTM (lua_State L, const TValue p1, int v2,
222	0	int flip, int isfloat, TMS event) {
223	0	TValue aux; const TValue *p2;
224	0	if (isfloat) {
225	0	setfltvalue(&aux, cast_num(v2));
226	0	}
227	0	else
228	0	setivalue(&aux, v2);
229	0	if (flip) { /* arguments were exchanged? */
230	0	p2 = p1; p1 = &aux; /* correct them */
231	0	}
232	0	else
233	0	p2 = &aux;
234	0	return luaT_callorderTM(L, p1, p2, event);
235	0	}
236
237
238		void luaT_adjustvarargs (lua_State L, int nfixparams, CallInfo ci,
239	8.07k	const Proto *p) {
240	8.07k	int i;
241	8.07k	int actual = cast_int(L->top.p - ci->func.p) - 1; /* number of arguments */
242	8.07k	int nextra = actual - nfixparams; /* number of extra arguments */
243	8.07k	ci->u.l.nextraargs = nextra;
244	8.07k	luaD_checkstack(L, p->maxstacksize + 1);
245		/* copy function to the top of the stack */
246	8.07k	setobjs2s(L, L->top.p++, ci->func.p);
247		/* move fixed parameters to the top of the stack */
248	8.07k	for (i = 1; i <= nfixparams; i++) {
249	0	setobjs2s(L, L->top.p++, ci->func.p + i);
250	0	setnilvalue(s2v(ci->func.p + i)); /* erase original parameter (for GC) */
251	0	}
252	8.07k	ci->func.p += actual + 1;
253	8.07k	ci->top.p += actual + 1;
254	8.07k	lua_assert(L->top.p <= ci->top.p && ci->top.p <= L->stack_last.p);
255	8.07k	}
256
257
258	8.00k	void luaT_getvarargs (lua_State L, CallInfo ci, StkId where, int wanted) {
259	8.00k	int i;
260	8.00k	int nextra = ci->u.l.nextraargs;
261	8.00k	if (wanted < 0) {
262	8.00k	wanted = nextra; /* get all extra arguments available */
263	8.00k	checkstackGCp(L, nextra, where); /* ensure stack space */
264	8.00k	L->top.p = where + nextra; /* next instruction will need top */
265	8.00k	}
266	16.0k	for (i = 0; i < wanted && i < nextra; i++)
267	8.00k	setobjs2s(L, where + i, ci->func.p - nextra + i);
268	8.00k	for (; i < wanted; i++) /* complete required results with nil */
269	8.00k	setnilvalue(s2v(where + i));
270	8.00k	}
271