Coverage Report

Created: 2025-07-18 06:59

/src/testdir/build/lua-master/source/lvm.c
Line
Count
Source (jump to first uncovered line)
1
/*
2
** $Id: lvm.c $
3
** Lua virtual machine
4
** See Copyright Notice in lua.h
5
*/
6
7
#define lvm_c
8
#define LUA_CORE
9
10
#include "lprefix.h"
11
12
#include <float.h>
13
#include <limits.h>
14
#include <math.h>
15
#include <stdio.h>
16
#include <stdlib.h>
17
#include <string.h>
18
19
#include "lua.h"
20
21
#include "lapi.h"
22
#include "ldebug.h"
23
#include "ldo.h"
24
#include "lfunc.h"
25
#include "lgc.h"
26
#include "lobject.h"
27
#include "lopcodes.h"
28
#include "lstate.h"
29
#include "lstring.h"
30
#include "ltable.h"
31
#include "ltm.h"
32
#include "lvm.h"
33
34
35
/*
36
** By default, use jump tables in the main interpreter loop on gcc
37
** and compatible compilers.
38
*/
39
#if !defined(LUA_USE_JUMPTABLE)
40
#if defined(__GNUC__)
41
#define LUA_USE_JUMPTABLE 1
42
#else
43
#define LUA_USE_JUMPTABLE 0
44
#endif
45
#endif
46
47
48
49
/* limit for table tag-method chains (to avoid infinite loops) */
50
445k
#define MAXTAGLOOP  2000
51
52
53
/*
54
** 'l_intfitsf' checks whether a given integer is in the range that
55
** can be converted to a float without rounding. Used in comparisons.
56
*/
57
58
/* number of bits in the mantissa of a float */
59
0
#define NBM   (l_floatatt(MANT_DIG))
60
61
/*
62
** Check whether some integers may not fit in a float, testing whether
63
** (maxinteger >> NBM) > 0. (That implies (1 << NBM) <= maxinteger.)
64
** (The shifts are done in parts, to avoid shifting by more than the size
65
** of an integer. In a worst case, NBM == 113 for long double and
66
** sizeof(long) == 32.)
67
*/
68
#if ((((LUA_MAXINTEGER >> (NBM / 4)) >> (NBM / 4)) >> (NBM / 4)) \
69
  >> (NBM - (3 * (NBM / 4))))  >  0
70
71
/* limit for integers that fit in a float */
72
0
#define MAXINTFITSF ((lua_Unsigned)1 << NBM)
73
74
/* check whether 'i' is in the interval [-MAXINTFITSF, MAXINTFITSF] */
75
0
#define l_intfitsf(i) ((MAXINTFITSF + l_castS2U(i)) <= (2 * MAXINTFITSF))
76
77
#else  /* all integers fit in a float precisely */
78
79
#define l_intfitsf(i) 1
80
81
#endif
82
83
84
/*
85
** Try to convert a value from string to a number value.
86
** If the value is not a string or is a string not representing
87
** a valid numeral (or if coercions from strings to numbers
88
** are disabled via macro 'cvt2num'), do not modify 'result'
89
** and return 0.
90
*/
91
6
static int l_strton (const TValue *obj, TValue *result) {
92
6
  lua_assert(obj != result);
93
6
  if (!cvt2num(obj))  /* is object not a string? */
94
2
    return 0;
95
4
  else {
96
4
    TString *st = tsvalue(obj);
97
0
    size_t stlen;
98
4
    const char *s = getlstr(st, stlen);
99
4
    return (luaO_str2num(s, result) == stlen + 1);
100
4
  }
101
6
}
102
103
104
/*
105
** Try to convert a value to a float. The float case is already handled
106
** by the macro 'tonumber'.
107
*/
108
2
int luaV_tonumber_ (const TValue *obj, lua_Number *n) {
109
2
  TValue v;
110
2
  if (ttisinteger(obj)) {
111
0
    *n = cast_num(ivalue(obj));
112
0
    return 1;
113
0
  }
114
2
  else if (l_strton(obj, &v)) {  /* string coercible to number? */
115
0
    *n = nvalue(&v);  /* convert result of 'luaO_str2num' to a float */
116
0
    return 1;
117
0
  }
118
2
  else
119
2
    return 0;  /* conversion failed */
120
2
}
121
122
123
/*
124
** try to convert a float to an integer, rounding according to 'mode'.
125
*/
126
8.48M
int luaV_flttointeger (lua_Number n, lua_Integer *p, F2Imod mode) {
127
8.48M
  lua_Number f = l_floor(n);
128
8.48M
  if (n != f) {  /* not an integral value? */
129
37.4k
    if (mode == F2Ieq) return 0;  /* fails if mode demands integral value */
130
1
    else if (mode == F2Iceil)  /* needs ceiling? */
131
0
      f += 1;  /* convert floor to ceiling (remember: n != f) */
132
37.4k
  }
133
8.44M
  return lua_numbertointeger(f, p);
134
8.48M
}
135
136
137
/*
138
** try to convert a value to an integer, rounding according to 'mode',
139
** without string coercion.
140
** ("Fast track" handled by macro 'tointegerns'.)
141
*/
142
51.9k
int luaV_tointegerns (const TValue *obj, lua_Integer *p, F2Imod mode) {
143
51.9k
  if (ttisfloat(obj))
144
3.49k
    return luaV_flttointeger(fltvalue(obj), p, mode);
145
48.4k
  else if (ttisinteger(obj)) {
146
48.3k
    *p = ivalue(obj);
147
0
    return 1;
148
48.3k
  }
149
86
  else
150
86
    return 0;
151
51.9k
}
152
153
154
/*
155
** try to convert a value to an integer.
156
*/
157
4
int luaV_tointeger (const TValue *obj, lua_Integer *p, F2Imod mode) {
158
4
  TValue v;
159
4
  if (l_strton(obj, &v))  /* does 'obj' point to a numerical string? */
160
0
    obj = &v;  /* change it to point to its corresponding number */
161
4
  return luaV_tointegerns(obj, p, mode);
162
4
}
163
164
165
/*
166
** Try to convert a 'for' limit to an integer, preserving the semantics
167
** of the loop. Return true if the loop must not run; otherwise, '*p'
168
** gets the integer limit.
169
** (The following explanation assumes a positive step; it is valid for
170
** negative steps mutatis mutandis.)
171
** If the limit is an integer or can be converted to an integer,
172
** rounding down, that is the limit.
173
** Otherwise, check whether the limit can be converted to a float. If
174
** the float is too large, clip it to LUA_MAXINTEGER.  If the float
175
** is too negative, the loop should not run, because any initial
176
** integer value is greater than such limit; so, the function returns
177
** true to signal that. (For this latter case, no integer limit would be
178
** correct; even a limit of LUA_MININTEGER would run the loop once for
179
** an initial value equal to LUA_MININTEGER.)
180
*/
181
static int forlimit (lua_State *L, lua_Integer init, const TValue *lim,
182
4
                                   lua_Integer *p, lua_Integer step) {
183
4
  if (!luaV_tointeger(lim, p, (step < 0 ? F2Iceil : F2Ifloor))) {
184
    /* not coercible to in integer */
185
2
    lua_Number flim;  /* try to convert to float */
186
2
    if (!tonumber(lim, &flim)) /* cannot convert to float? */
187
2
      luaG_forerror(L, lim, "limit");
188
    /* else 'flim' is a float out of integer bounds */
189
0
    if (luai_numlt(0, flim)) {  /* if it is positive, it is too large */
190
0
      if (step < 0) return 1;  /* initial value must be less than it */
191
0
      *p = LUA_MAXINTEGER;  /* truncate */
192
0
    }
193
0
    else {  /* it is less than min integer */
194
0
      if (step > 0) return 1;  /* initial value must be greater than it */
195
0
      *p = LUA_MININTEGER;  /* truncate */
196
0
    }
197
0
  }
198
2
  return (step > 0 ? init > *p : init < *p);  /* not to run? */
199
4
}
200
201
202
/*
203
** Prepare a numerical for loop (opcode OP_FORPREP).
204
** Before execution, stack is as follows:
205
**   ra     : initial value
206
**   ra + 1 : limit
207
**   ra + 2 : step
208
** Return true to skip the loop. Otherwise,
209
** after preparation, stack will be as follows:
210
**   ra     : loop counter (integer loops) or limit (float loops)
211
**   ra + 1 : step
212
**   ra + 2 : control variable
213
*/
214
4
static int forprep (lua_State *L, StkId ra) {
215
4
  TValue *pinit = s2v(ra);
216
4
  TValue *plimit = s2v(ra + 1);
217
4
  TValue *pstep = s2v(ra + 2);
218
4
  if (ttisinteger(pinit) && ttisinteger(pstep)) { /* integer loop? */
219
4
    lua_Integer init = ivalue(pinit);
220
4
    lua_Integer step = ivalue(pstep);
221
0
    lua_Integer limit;
222
4
    if (step == 0)
223
0
      luaG_runerror(L, "'for' step is zero");
224
4
    if (forlimit(L, init, plimit, &limit, step))
225
0
      return 1;  /* skip the loop */
226
4
    else {  /* prepare loop counter */
227
4
      lua_Unsigned count;
228
4
      if (step > 0) {  /* ascending loop? */
229
2
        count = l_castS2U(limit) - l_castS2U(init);
230
2
        if (step != 1)  /* avoid division in the too common case */
231
0
          count /= l_castS2U(step);
232
2
      }
233
2
      else {  /* step < 0; descending loop */
234
2
        count = l_castS2U(init) - l_castS2U(limit);
235
        /* 'step+1' avoids negating 'mininteger' */
236
2
        count /= l_castS2U(-(step + 1)) + 1u;
237
2
      }
238
      /* use 'chgivalue' for places that for sure had integers */
239
4
      chgivalue(s2v(ra), l_castU2S(count));  /* change init to count */
240
2
      setivalue(s2v(ra + 1), step);  /* change limit to step */
241
2
      chgivalue(s2v(ra + 2), init);  /* change step to init */
242
2
    }
243
4
  }
244
0
  else {  /* try making all values floats */
245
0
    lua_Number init; lua_Number limit; lua_Number step;
246
0
    if (l_unlikely(!tonumber(plimit, &limit)))
247
0
      luaG_forerror(L, plimit, "limit");
248
0
    if (l_unlikely(!tonumber(pstep, &step)))
249
0
      luaG_forerror(L, pstep, "step");
250
0
    if (l_unlikely(!tonumber(pinit, &init)))
251
0
      luaG_forerror(L, pinit, "initial value");
252
0
    if (step == 0)
253
0
      luaG_runerror(L, "'for' step is zero");
254
0
    if (luai_numlt(0, step) ? luai_numlt(limit, init)
255
0
                            : luai_numlt(init, limit))
256
0
      return 1;  /* skip the loop */
257
0
    else {
258
      /* make sure all values are floats */
259
0
      setfltvalue(s2v(ra), limit);
260
0
      setfltvalue(s2v(ra + 1), step);
261
0
      setfltvalue(s2v(ra + 2), init);  /* control variable */
262
0
    }
263
0
  }
264
2
  return 0;
265
4
}
266
267
268
/*
269
** Execute a step of a float numerical for loop, returning
270
** true iff the loop must continue. (The integer case is
271
** written online with opcode OP_FORLOOP, for performance.)
272
*/
273
0
static int floatforloop (StkId ra) {
274
0
  lua_Number step = fltvalue(s2v(ra + 1));
275
0
  lua_Number limit = fltvalue(s2v(ra));
276
0
  lua_Number idx = fltvalue(s2v(ra + 2));  /* control variable */
277
0
  idx = luai_numadd(L, idx, step);  /* increment index */
278
0
  if (luai_numlt(0, step) ? luai_numle(idx, limit)
279
0
                          : luai_numle(limit, idx)) {
280
0
    chgfltvalue(s2v(ra + 2), idx);  /* update control variable */
281
0
    return 1;  /* jump back */
282
0
  }
283
0
  else
284
0
    return 0;  /* finish the loop */
285
0
}
286
287
288
/*
289
** Finish the table access 'val = t[key]' and return the tag of the result.
290
*/
291
lu_byte luaV_finishget (lua_State *L, const TValue *t, TValue *key,
292
445k
                                      StkId val, lu_byte tag) {
293
445k
  int loop;  /* counter to avoid infinite loops */
294
445k
  const TValue *tm;  /* metamethod */
295
445k
  for (loop = 0; loop < MAXTAGLOOP; loop++) {
296
445k
    if (tag == LUA_VNOTABLE) {  /* 't' is not a table? */
297
1
      lua_assert(!ttistable(t));
298
1
      tm = luaT_gettmbyobj(L, t, TM_INDEX);
299
1
      if (l_unlikely(notm(tm)))
300
1
        luaG_typeerror(L, t, "index");  /* no metamethod */
301
      /* else will try the metamethod */
302
1
    }
303
445k
    else {  /* 't' is a table */
304
445k
      tm = fasttm(L, hvalue(t)->metatable, TM_INDEX);  /* table's metamethod */
305
445k
      if (tm == NULL) {  /* no metamethod? */
306
445k
        setnilvalue(s2v(val));  /* result is nil */
307
445k
        return LUA_VNIL;
308
445k
      }
309
      /* else will try the metamethod */
310
445k
    }
311
0
    if (ttisfunction(tm)) {  /* is metamethod a function? */
312
0
      tag = luaT_callTMres(L, tm, t, key, val);  /* call it */
313
0
      return tag;  /* return tag of the result */
314
0
    }
315
0
    t = tm;  /* else try to access 'tm[key]' */
316
0
    luaV_fastget(t, key, s2v(val), luaH_get, tag);
317
0
    if (!tagisempty(tag))
318
0
      return tag;  /* done */
319
    /* else repeat (tail call 'luaV_finishget') */
320
0
  }
321
0
  luaG_runerror(L, "'__index' chain too long; possible loop");
322
0
  return 0;  /* to avoid warnings */
323
445k
}
324
325
326
/*
327
** Finish a table assignment 't[key] = val'.
328
** About anchoring the table before the call to 'luaH_finishset':
329
** This call may trigger an emergency collection. When loop>0,
330
** the table being accessed is a field in some metatable. If this
331
** metatable is weak and the table is not anchored, this collection
332
** could collect that table while it is being updated.
333
*/
334
void luaV_finishset (lua_State *L, const TValue *t, TValue *key,
335
179
                      TValue *val, int hres) {
336
179
  int loop;  /* counter to avoid infinite loops */
337
179
  for (loop = 0; loop < MAXTAGLOOP; loop++) {
338
179
    const TValue *tm;  /* '__newindex' metamethod */
339
179
    if (hres != HNOTATABLE) {  /* is 't' a table? */
340
358
      Table *h = hvalue(t);  /* save 't' table */
341
179
      tm = fasttm(L, h->metatable, TM_NEWINDEX);  /* get metamethod */
342
358
      if (tm == NULL) {  /* no metamethod? */
343
179
        sethvalue2s(L, L->top.p, h);  /* anchor 't' */
344
179
        L->top.p++;  /* assume EXTRA_STACK */
345
179
        luaH_finishset(L, h, key, val, hres);  /* set new value */
346
179
        L->top.p--;
347
179
        invalidateTMcache(h);
348
179
        luaC_barrierback(L, obj2gco(h), val);
349
179
        return;
350
179
      }
351
      /* else will try the metamethod */
352
358
    }
353
0
    else {  /* not a table; check metamethod */
354
0
      tm = luaT_gettmbyobj(L, t, TM_NEWINDEX);
355
0
      if (l_unlikely(notm(tm)))
356
0
        luaG_typeerror(L, t, "index");
357
0
    }
358
    /* try the metamethod */
359
0
    if (ttisfunction(tm)) {
360
0
      luaT_callTM(L, tm, t, key, val);
361
0
      return;
362
0
    }
363
0
    t = tm;  /* else repeat assignment over 'tm' */
364
0
    luaV_fastset(t, key, val, hres, luaH_pset);
365
0
    if (hres == HOK) {
366
0
      luaV_finishfastset(L, t, val);
367
0
      return;  /* done */
368
0
    }
369
    /* else 'return luaV_finishset(L, t, key, val, slot)' (loop) */
370
0
  }
371
0
  luaG_runerror(L, "'__newindex' chain too long; possible loop");
372
179
}
373
374
375
/*
376
** Compare two strings 'ts1' x 'ts2', returning an integer less-equal-
377
** -greater than zero if 'ts1' is less-equal-greater than 'ts2'.
378
** The code is a little tricky because it allows '\0' in the strings
379
** and it uses 'strcoll' (to respect locales) for each segment
380
** of the strings. Note that segments can compare equal but still
381
** have different lengths.
382
*/
383
43
static int l_strcmp (const TString *ts1, const TString *ts2) {
384
43
  size_t rl1;  /* real length */
385
43
  const char *s1 = getlstr(ts1, rl1);
386
43
  size_t rl2;
387
43
  const char *s2 = getlstr(ts2, rl2);
388
72
  for (;;) {  /* for each segment */
389
72
    int temp = strcoll(s1, s2);
390
72
    if (temp != 0)  /* not equal? */
391
43
      return temp;  /* done */
392
29
    else {  /* strings are equal up to a '\0' */
393
29
      size_t zl1 = strlen(s1);  /* index of first '\0' in 's1' */
394
29
      size_t zl2 = strlen(s2);  /* index of first '\0' in 's2' */
395
29
      if (zl2 == rl2)  /* 's2' is finished? */
396
0
        return (zl1 == rl1) ? 0 : 1;  /* check 's1' */
397
29
      else if (zl1 == rl1)  /* 's1' is finished? */
398
0
        return -1;  /* 's1' is less than 's2' ('s2' is not finished) */
399
      /* both strings longer than 'zl'; go on comparing after the '\0' */
400
29
      zl1++; zl2++;
401
29
      s1 += zl1; rl1 -= zl1; s2 += zl2; rl2 -= zl2;
402
29
    }
403
72
  }
404
43
}
405
406
407
/*
408
** Check whether integer 'i' is less than float 'f'. If 'i' has an
409
** exact representation as a float ('l_intfitsf'), compare numbers as
410
** floats. Otherwise, use the equivalence 'i < f <=> i < ceil(f)'.
411
** If 'ceil(f)' is out of integer range, either 'f' is greater than
412
** all integers or less than all integers.
413
** (The test with 'l_intfitsf' is only for performance; the else
414
** case is correct for all values, but it is slow due to the conversion
415
** from float to int.)
416
** When 'f' is NaN, comparisons must result in false.
417
*/
418
0
l_sinline int LTintfloat (lua_Integer i, lua_Number f) {
419
0
  if (l_intfitsf(i))
420
0
    return luai_numlt(cast_num(i), f);  /* compare them as floats */
421
0
  else {  /* i < f <=> i < ceil(f) */
422
0
    lua_Integer fi;
423
0
    if (luaV_flttointeger(f, &fi, F2Iceil))  /* fi = ceil(f) */
424
0
      return i < fi;   /* compare them as integers */
425
0
    else  /* 'f' is either greater or less than all integers */
426
0
      return f > 0;  /* greater? */
427
0
  }
428
0
}
429
430
431
/*
432
** Check whether integer 'i' is less than or equal to float 'f'.
433
** See comments on previous function.
434
*/
435
0
l_sinline int LEintfloat (lua_Integer i, lua_Number f) {
436
0
  if (l_intfitsf(i))
437
0
    return luai_numle(cast_num(i), f);  /* compare them as floats */
438
0
  else {  /* i <= f <=> i <= floor(f) */
439
0
    lua_Integer fi;
440
0
    if (luaV_flttointeger(f, &fi, F2Ifloor))  /* fi = floor(f) */
441
0
      return i <= fi;   /* compare them as integers */
442
0
    else  /* 'f' is either greater or less than all integers */
443
0
      return f > 0;  /* greater? */
444
0
  }
445
0
}
446
447
448
/*
449
** Check whether float 'f' is less than integer 'i'.
450
** See comments on previous function.
451
*/
452
0
l_sinline int LTfloatint (lua_Number f, lua_Integer i) {
453
0
  if (l_intfitsf(i))
454
0
    return luai_numlt(f, cast_num(i));  /* compare them as floats */
455
0
  else {  /* f < i <=> floor(f) < i */
456
0
    lua_Integer fi;
457
0
    if (luaV_flttointeger(f, &fi, F2Ifloor))  /* fi = floor(f) */
458
0
      return fi < i;   /* compare them as integers */
459
0
    else  /* 'f' is either greater or less than all integers */
460
0
      return f < 0;  /* less? */
461
0
  }
462
0
}
463
464
465
/*
466
** Check whether float 'f' is less than or equal to integer 'i'.
467
** See comments on previous function.
468
*/
469
0
l_sinline int LEfloatint (lua_Number f, lua_Integer i) {
470
0
  if (l_intfitsf(i))
471
0
    return luai_numle(f, cast_num(i));  /* compare them as floats */
472
0
  else {  /* f <= i <=> ceil(f) <= i */
473
0
    lua_Integer fi;
474
0
    if (luaV_flttointeger(f, &fi, F2Iceil))  /* fi = ceil(f) */
475
0
      return fi <= i;   /* compare them as integers */
476
0
    else  /* 'f' is either greater or less than all integers */
477
0
      return f < 0;  /* less? */
478
0
  }
479
0
}
480
481
482
/*
483
** Return 'l < r', for numbers.
484
*/
485
0
l_sinline int LTnum (const TValue *l, const TValue *r) {
486
0
  lua_assert(ttisnumber(l) && ttisnumber(r));
487
0
  if (ttisinteger(l)) {
488
0
    lua_Integer li = ivalue(l);
489
0
    if (ttisinteger(r))
490
0
      return li < ivalue(r);  /* both are integers */
491
0
    else  /* 'l' is int and 'r' is float */
492
0
      return LTintfloat(li, fltvalue(r));  /* l < r ? */
493
0
  }
494
0
  else {
495
0
    lua_Number lf = fltvalue(l);  /* 'l' must be float */
496
0
    if (ttisfloat(r))
497
0
      return luai_numlt(lf, fltvalue(r));  /* both are float */
498
0
    else  /* 'l' is float and 'r' is int */
499
0
      return LTfloatint(lf, ivalue(r));
500
0
  }
501
0
}
502
503
504
/*
505
** Return 'l <= r', for numbers.
506
*/
507
0
l_sinline int LEnum (const TValue *l, const TValue *r) {
508
0
  lua_assert(ttisnumber(l) && ttisnumber(r));
509
0
  if (ttisinteger(l)) {
510
0
    lua_Integer li = ivalue(l);
511
0
    if (ttisinteger(r))
512
0
      return li <= ivalue(r);  /* both are integers */
513
0
    else  /* 'l' is int and 'r' is float */
514
0
      return LEintfloat(li, fltvalue(r));  /* l <= r ? */
515
0
  }
516
0
  else {
517
0
    lua_Number lf = fltvalue(l);  /* 'l' must be float */
518
0
    if (ttisfloat(r))
519
0
      return luai_numle(lf, fltvalue(r));  /* both are float */
520
0
    else  /* 'l' is float and 'r' is int */
521
0
      return LEfloatint(lf, ivalue(r));
522
0
  }
523
0
}
524
525
526
/*
527
** return 'l < r' for non-numbers.
528
*/
529
27
static int lessthanothers (lua_State *L, const TValue *l, const TValue *r) {
530
27
  lua_assert(!ttisnumber(l) || !ttisnumber(r));
531
27
  if (ttisstring(l) && ttisstring(r))  /* both are strings? */
532
54
    return l_strcmp(tsvalue(l), tsvalue(r)) < 0;
533
0
  else
534
0
    return luaT_callorderTM(L, l, r, TM_LT);
535
27
}
536
537
538
/*
539
** Main operation less than; return 'l < r'.
540
*/
541
0
int luaV_lessthan (lua_State *L, const TValue *l, const TValue *r) {
542
0
  if (ttisnumber(l) && ttisnumber(r))  /* both operands are numbers? */
543
0
    return LTnum(l, r);
544
0
  else return lessthanothers(L, l, r);
545
0
}
546
547
548
/*
549
** return 'l <= r' for non-numbers.
550
*/
551
17
static int lessequalothers (lua_State *L, const TValue *l, const TValue *r) {
552
17
  lua_assert(!ttisnumber(l) || !ttisnumber(r));
553
17
  if (ttisstring(l) && ttisstring(r))  /* both are strings? */
554
32
    return l_strcmp(tsvalue(l), tsvalue(r)) <= 0;
555
1
  else
556
1
    return luaT_callorderTM(L, l, r, TM_LE);
557
17
}
558
559
560
/*
561
** Main operation less than or equal to; return 'l <= r'.
562
*/
563
0
int luaV_lessequal (lua_State *L, const TValue *l, const TValue *r) {
564
0
  if (ttisnumber(l) && ttisnumber(r))  /* both operands are numbers? */
565
0
    return LEnum(l, r);
566
0
  else return lessequalothers(L, l, r);
567
0
}
568
569
570
/*
571
** Main operation for equality of Lua values; return 't1 == t2'.
572
** L == NULL means raw equality (no metamethods)
573
*/
574
18.1M
int luaV_equalobj (lua_State *L, const TValue *t1, const TValue *t2) {
575
18.1M
  const TValue *tm;
576
18.1M
  if (ttypetag(t1) != ttypetag(t2)) {  /* not the same variant? */
577
0
    if (ttype(t1) != ttype(t2) || ttype(t1) != LUA_TNUMBER)
578
0
      return 0;  /* only numbers can be equal with different variants */
579
0
    else {  /* two numbers with different variants */
580
      /* One of them is an integer. If the other does not have an
581
         integer value, they cannot be equal; otherwise, compare their
582
         integer values. */
583
0
      lua_Integer i1, i2;
584
0
      return (luaV_tointegerns(t1, &i1, F2Ieq) &&
585
0
              luaV_tointegerns(t2, &i2, F2Ieq) &&
586
0
              i1 == i2);
587
0
    }
588
0
  }
589
  /* values have same type and same variant */
590
18.1M
  switch (ttypetag(t1)) {
591
88
    case LUA_VNIL: case LUA_VFALSE: case LUA_VTRUE: return 1;
592
99.6k
    case LUA_VNUMINT: return (ivalue(t1) == ivalue(t2));
593
23.5k
    case LUA_VNUMFLT: return luai_numeq(fltvalue(t1), fltvalue(t2));
594
0
    case LUA_VLIGHTUSERDATA: return pvalue(t1) == pvalue(t2);
595
0
    case LUA_VLCF: return fvalue(t1) == fvalue(t2);
596
72.1M
    case LUA_VSHRSTR: return eqshrstr(tsvalue(t1), tsvalue(t2));
597
7.38k
    case LUA_VLNGSTR: return luaS_eqlngstr(tsvalue(t1), tsvalue(t2));
598
0
    case LUA_VUSERDATA: {
599
0
      if (uvalue(t1) == uvalue(t2)) return 1;
600
0
      else if (L == NULL) return 0;
601
0
      tm = fasttm(L, uvalue(t1)->metatable, TM_EQ);
602
0
      if (tm == NULL)
603
0
        tm = fasttm(L, uvalue(t2)->metatable, TM_EQ);
604
0
      break;  /* will try TM */
605
0
    }
606
0
    case LUA_VTABLE: {
607
0
      if (hvalue(t1) == hvalue(t2)) return 1;
608
0
      else if (L == NULL) return 0;
609
0
      tm = fasttm(L, hvalue(t1)->metatable, TM_EQ);
610
0
      if (tm == NULL)
611
0
        tm = fasttm(L, hvalue(t2)->metatable, TM_EQ);
612
0
      break;  /* will try TM */
613
0
    }
614
0
    default:
615
18.1M
      return gcvalue(t1) == gcvalue(t2);
616
18.1M
  }
617
0
  if (tm == NULL)  /* no TM? */
618
0
    return 0;  /* objects are different */
619
0
  else {
620
0
    int tag = luaT_callTMres(L, tm, t1, t2, L->top.p);  /* call TM */
621
0
    return !tagisfalse(tag);
622
0
  }
623
0
}
624
625
626
/* macro used by 'luaV_concat' to ensure that element at 'o' is a string */
627
#define tostring(L,o)  \
628
140
  (ttisstring(o) || (cvt2str(o) && (luaO_tostring(L, o), 1)))
629
630
132
#define isemptystr(o) (ttisshrstring(o) && tsvalue(o)->shrlen == 0)
631
632
/* copy strings in stack from top - n up to top - 1 to buffer */
633
63
static void copy2buff (StkId top, int n, char *buff) {
634
63
  size_t tl = 0;  /* size already copied */
635
140
  do {
636
280
    TString *st = tsvalue(s2v(top - n));
637
0
    size_t l;  /* length of string being copied */
638
280
    const char *s = getlstr(st, l);
639
280
    memcpy(buff + tl, s, l * sizeof(char));
640
280
    tl += l;
641
280
  } while (--n > 0);
642
63
}
643
644
645
/*
646
** Main operation for concatenation: concat 'total' values in the stack,
647
** from 'L->top.p - total' up to 'L->top.p - 1'.
648
*/
649
63
void luaV_concat (lua_State *L, int total) {
650
63
  if (total == 1)
651
0
    return;  /* "all" values already concatenated */
652
63
  do {
653
63
    StkId top = L->top.p;
654
63
    int n = 2;  /* number of elements handled in this pass (at least 2) */
655
63
    if (!(ttisstring(s2v(top - 2)) || cvt2str(s2v(top - 2))) ||
656
63
        !tostring(L, s2v(top - 1)))
657
0
      luaT_tryconcatTM(L);  /* may invalidate 'top' */
658
63
    else if (isemptystr(s2v(top - 1)))  /* second operand is empty? */
659
63
      cast_void(tostring(L, s2v(top - 2)));  /* result is first operand */
660
63
    else if (isemptystr(s2v(top - 2))) {  /* first operand is empty string? */
661
0
      setobjs2s(L, top - 2, top - 1);  /* result is second op. */
662
0
    }
663
63
    else {
664
      /* at least two non-empty string values; get as many as possible */
665
63
      size_t tl = tsslen(tsvalue(s2v(top - 1)));
666
63
      TString *ts;
667
      /* collect total length and number of strings */
668
140
      for (n = 1; n < total && tostring(L, s2v(top - n - 1)); n++) {
669
77
        size_t l = tsslen(tsvalue(s2v(top - n - 1)));
670
77
        if (l_unlikely(l >= MAX_SIZE - sizeof(TString) - tl)) {
671
0
          L->top.p = top - total;  /* pop strings to avoid wasting stack */
672
0
          luaG_runerror(L, "string length overflow");
673
0
        }
674
77
        tl += l;
675
77
      }
676
63
      if (tl <= LUAI_MAXSHORTLEN) {  /* is result a short string? */
677
0
        char buff[LUAI_MAXSHORTLEN];
678
0
        copy2buff(top, n, buff);  /* copy strings to buffer */
679
0
        ts = luaS_newlstr(L, buff, tl);
680
0
      }
681
63
      else {  /* long string; copy strings directly to final result */
682
63
        ts = luaS_createlngstrobj(L, tl);
683
63
        copy2buff(top, n, getlngstr(ts));
684
63
      }
685
126
      setsvalue2s(L, top - n, ts);  /* create result */
686
63
    }
687
63
    total -= n - 1;  /* got 'n' strings to create one new */
688
63
    L->top.p -= n - 1;  /* popped 'n' strings and pushed one */
689
63
  } while (total > 1);  /* repeat until only 1 result left */
690
63
}
691
692
693
/*
694
** Main operation 'ra = #rb'.
695
*/
696
39
void luaV_objlen (lua_State *L, StkId ra, const TValue *rb) {
697
39
  const TValue *tm;
698
39
  switch (ttypetag(rb)) {
699
0
    case LUA_VTABLE: {
700
0
      Table *h = hvalue(rb);
701
0
      tm = fasttm(L, h->metatable, TM_LEN);
702
0
      if (tm) break;  /* metamethod? break switch to call it */
703
0
      setivalue(s2v(ra), l_castU2S(luaH_getn(h)));  /* else primitive len */
704
0
      return;
705
0
    }
706
0
    case LUA_VSHRSTR: {
707
0
      setivalue(s2v(ra), tsvalue(rb)->shrlen);
708
0
      return;
709
0
    }
710
39
    case LUA_VLNGSTR: {
711
39
      setivalue(s2v(ra), cast_st2S(tsvalue(rb)->u.lnglen));
712
39
      return;
713
39
    }
714
0
    default: {  /* try metamethod */
715
0
      tm = luaT_gettmbyobj(L, rb, TM_LEN);
716
0
      if (l_unlikely(notm(tm)))  /* no metamethod? */
717
0
        luaG_typeerror(L, rb, "get length of");
718
0
      break;
719
0
    }
720
39
  }
721
0
  luaT_callTMres(L, tm, rb, rb, ra);
722
0
}
723
724
725
/*
726
** Integer division; return 'm // n', that is, floor(m/n).
727
** C division truncates its result (rounds towards zero).
728
** 'floor(q) == trunc(q)' when 'q >= 0' or when 'q' is integer,
729
** otherwise 'floor(q) == trunc(q) - 1'.
730
*/
731
2
lua_Integer luaV_idiv (lua_State *L, lua_Integer m, lua_Integer n) {
732
2
  if (l_unlikely(l_castS2U(n) + 1u <= 1u)) {  /* special cases: -1 or 0 */
733
0
    if (n == 0)
734
0
      luaG_runerror(L, "attempt to divide by zero");
735
0
    return intop(-, 0, m);   /* n==-1; avoid overflow with 0x80000...//-1 */
736
0
  }
737
2
  else {
738
2
    lua_Integer q = m / n;  /* perform C division */
739
2
    if ((m ^ n) < 0 && m % n != 0)  /* 'm/n' would be negative non-integer? */
740
1
      q -= 1;  /* correct result for different rounding */
741
2
    return q;
742
2
  }
743
2
}
744
745
746
/*
747
** Integer modulus; return 'm % n'. (Assume that C '%' with
748
** negative operands follows C99 behavior. See previous comment
749
** about luaV_idiv.)
750
*/
751
6.31k
lua_Integer luaV_mod (lua_State *L, lua_Integer m, lua_Integer n) {
752
6.31k
  if (l_unlikely(l_castS2U(n) + 1u <= 1u)) {  /* special cases: -1 or 0 */
753
360
    if (n == 0)
754
0
      luaG_runerror(L, "attempt to perform 'n%%0'");
755
360
    return 0;   /* m % -1 == 0; avoid overflow with 0x80000...%-1 */
756
360
  }
757
5.95k
  else {
758
5.95k
    lua_Integer r = m % n;
759
5.95k
    if (r != 0 && (r ^ n) < 0)  /* 'm/n' would be non-integer negative? */
760
20
      r += n;  /* correct result for different rounding */
761
5.95k
    return r;
762
5.95k
  }
763
6.31k
}
764
765
766
/*
767
** Float modulus
768
*/
769
6.16k
lua_Number luaV_modf (lua_State *L, lua_Number m, lua_Number n) {
770
6.16k
  lua_Number r;
771
6.16k
  luai_nummod(L, m, n, r);
772
6.16k
  return r;
773
6.16k
}
774
775
776
/* number of bits in an integer */
777
917
#define NBITS l_numbits(lua_Integer)
778
779
780
/*
781
** Shift left operation. (Shift right just negates 'y'.)
782
*/
783
917
lua_Integer luaV_shiftl (lua_Integer x, lua_Integer y) {
784
917
  if (y < 0) {  /* shift right? */
785
83
    if (y <= -NBITS) return 0;
786
0
    else return intop(>>, x, -y);
787
83
  }
788
834
  else {  /* shift left */
789
834
    if (y >= NBITS) return 0;
790
793
    else return intop(<<, x, y);
791
834
  }
792
917
}
793
794
795
/*
796
** create a new Lua closure, push it in the stack, and initialize
797
** its upvalues.
798
*/
799
static void pushclosure (lua_State *L, Proto *p, UpVal **encup, StkId base,
800
0
                         StkId ra) {
801
0
  int nup = p->sizeupvalues;
802
0
  Upvaldesc *uv = p->upvalues;
803
0
  int i;
804
0
  LClosure *ncl = luaF_newLclosure(L, nup);
805
0
  ncl->p = p;
806
0
  setclLvalue2s(L, ra, ncl);  /* anchor new closure in stack */
807
0
  for (i = 0; i < nup; i++) {  /* fill in its upvalues */
808
0
    if (uv[i].instack)  /* upvalue refers to local variable? */
809
0
      ncl->upvals[i] = luaF_findupval(L, base + uv[i].idx);
810
0
    else  /* get upvalue from enclosing function */
811
0
      ncl->upvals[i] = encup[uv[i].idx];
812
0
    luaC_objbarrier(L, ncl, ncl->upvals[i]);
813
0
  }
814
0
}
815
816
817
/*
818
** finish execution of an opcode interrupted by a yield
819
*/
820
0
void luaV_finishOp (lua_State *L) {
821
0
  CallInfo *ci = L->ci;
822
0
  StkId base = ci->func.p + 1;
823
0
  Instruction inst = *(ci->u.l.savedpc - 1);  /* interrupted instruction */
824
0
  OpCode op = GET_OPCODE(inst);
825
0
  switch (op) {  /* finish its execution */
826
0
    case OP_MMBIN: case OP_MMBINI: case OP_MMBINK: {
827
0
      setobjs2s(L, base + GETARG_A(*(ci->u.l.savedpc - 2)), --L->top.p);
828
0
      break;
829
0
    }
830
0
    case OP_UNM: case OP_BNOT: case OP_LEN:
831
0
    case OP_GETTABUP: case OP_GETTABLE: case OP_GETI:
832
0
    case OP_GETFIELD: case OP_SELF: {
833
0
      setobjs2s(L, base + GETARG_A(inst), --L->top.p);
834
0
      break;
835
0
    }
836
0
    case OP_LT: case OP_LE:
837
0
    case OP_LTI: case OP_LEI:
838
0
    case OP_GTI: case OP_GEI:
839
0
    case OP_EQ: {  /* note that 'OP_EQI'/'OP_EQK' cannot yield */
840
0
      int res = !l_isfalse(s2v(L->top.p - 1));
841
0
      L->top.p--;
842
#if defined(LUA_COMPAT_LT_LE)
843
      if (ci->callstatus & CIST_LEQ) {  /* "<=" using "<" instead? */
844
        ci->callstatus ^= CIST_LEQ;  /* clear mark */
845
        res = !res;  /* negate result */
846
      }
847
#endif
848
0
      lua_assert(GET_OPCODE(*ci->u.l.savedpc) == OP_JMP);
849
0
      if (res != GETARG_k(inst))  /* condition failed? */
850
0
        ci->u.l.savedpc++;  /* skip jump instruction */
851
0
      break;
852
0
    }
853
0
    case OP_CONCAT: {
854
0
      StkId top = L->top.p - 1;  /* top when 'luaT_tryconcatTM' was called */
855
0
      int a = GETARG_A(inst);      /* first element to concatenate */
856
0
      int total = cast_int(top - 1 - (base + a));  /* yet to concatenate */
857
0
      setobjs2s(L, top - 2, top);  /* put TM result in proper position */
858
0
      L->top.p = top - 1;  /* top is one after last element (at top-2) */
859
0
      luaV_concat(L, total);  /* concat them (may yield again) */
860
0
      break;
861
0
    }
862
0
    case OP_CLOSE: {  /* yielded closing variables */
863
0
      ci->u.l.savedpc--;  /* repeat instruction to close other vars. */
864
0
      break;
865
0
    }
866
0
    case OP_RETURN: {  /* yielded closing variables */
867
0
      StkId ra = base + GETARG_A(inst);
868
      /* adjust top to signal correct number of returns, in case the
869
         return is "up to top" ('isIT') */
870
0
      L->top.p = ra + ci->u2.nres;
871
      /* repeat instruction to close other vars. and complete the return */
872
0
      ci->u.l.savedpc--;
873
0
      break;
874
0
    }
875
0
    default: {
876
      /* only these other opcodes can yield */
877
0
      lua_assert(op == OP_TFORCALL || op == OP_CALL ||
878
0
           op == OP_TAILCALL || op == OP_SETTABUP || op == OP_SETTABLE ||
879
0
           op == OP_SETI || op == OP_SETFIELD);
880
0
      break;
881
0
    }
882
0
  }
883
0
}
884
885
886
887
888
/*
889
** {==================================================================
890
** Macros for arithmetic/bitwise/comparison opcodes in 'luaV_execute'
891
** ===================================================================
892
*/
893
894
#define l_addi(L,a,b) intop(+, a, b)
895
#define l_subi(L,a,b) intop(-, a, b)
896
#define l_muli(L,a,b) intop(*, a, b)
897
#define l_band(a,b) intop(&, a, b)
898
#define l_bor(a,b)  intop(|, a, b)
899
#define l_bxor(a,b) intop(^, a, b)
900
901
4
#define l_lti(a,b)  (a < b)
902
0
#define l_lei(a,b)  (a <= b)
903
0
#define l_gti(a,b)  (a > b)
904
0
#define l_gei(a,b)  (a >= b)
905
906
907
/*
908
** Arithmetic operations with immediate operands. 'iop' is the integer
909
** operation, 'fop' is the float operation.
910
*/
911
0
#define op_arithI(L,iop,fop) {  \
912
0
  StkId ra = RA(i); \
913
0
  TValue *v1 = vRB(i);  \
914
0
  int imm = GETARG_sC(i);  \
915
0
  if (ttisinteger(v1)) {  \
916
0
    lua_Integer iv1 = ivalue(v1);  \
917
0
    pc++; setivalue(s2v(ra), iop(L, iv1, imm));  \
918
0
  }  \
919
0
  else if (ttisfloat(v1)) {  \
920
0
    lua_Number nb = fltvalue(v1);  \
921
0
    lua_Number fimm = cast_num(imm);  \
922
0
    pc++; setfltvalue(s2v(ra), fop(L, nb, fimm)); \
923
0
  }}
924
925
926
/*
927
** Auxiliary function for arithmetic operations over floats and others
928
** with two operands.
929
*/
930
6
#define op_arithf_aux(L,v1,v2,fop) {  \
931
6
  lua_Number n1; lua_Number n2;  \
932
6
  if (tonumberns(v1, n1) && tonumberns(v2, n2)) {  \
933
2
    pc++; setfltvalue(s2v(ra), fop(L, n1, n2));  \
934
2
  }}
935
936
937
/*
938
** Arithmetic operations over floats and others with register operands.
939
*/
940
1
#define op_arithf(L,fop) {  \
941
1
  StkId ra = RA(i); \
942
1
  TValue *v1 = vRB(i);  \
943
1
  TValue *v2 = vRC(i);  \
944
1
  op_arithf_aux(L, v1, v2, fop); }
945
946
947
/*
948
** Arithmetic operations with K operands for floats.
949
*/
950
1
#define op_arithfK(L,fop) {  \
951
1
  StkId ra = RA(i); \
952
1
  TValue *v1 = vRB(i);  \
953
1
  TValue *v2 = KC(i); lua_assert(ttisnumber(v2));  \
954
1
  op_arithf_aux(L, v1, v2, fop); }
955
956
957
/*
958
** Arithmetic operations over integers and floats.
959
*/
960
4
#define op_arith_aux(L,v1,v2,iop,fop) {  \
961
4
  StkId ra = RA(i); \
962
4
  if (ttisinteger(v1) && ttisinteger(v2)) {  \
963
0
    lua_Integer i1 = ivalue(v1); lua_Integer i2 = ivalue(v2);  \
964
0
    pc++; setivalue(s2v(ra), iop(L, i1, i2));  \
965
0
  }  \
966
4
  else op_arithf_aux(L, v1, v2, fop); }
967
968
969
/*
970
** Arithmetic operations with register operands.
971
*/
972
3
#define op_arith(L,iop,fop) {  \
973
3
  TValue *v1 = vRB(i);  \
974
3
  TValue *v2 = vRC(i);  \
975
3
  op_arith_aux(L, v1, v2, iop, fop); }
976
977
978
/*
979
** Arithmetic operations with K operands.
980
*/
981
1
#define op_arithK(L,iop,fop) {  \
982
1
  TValue *v1 = vRB(i);  \
983
1
  TValue *v2 = KC(i); lua_assert(ttisnumber(v2));  \
984
1
  op_arith_aux(L, v1, v2, iop, fop); }
985
986
987
/*
988
** Bitwise operations with constant operand.
989
*/
990
1
#define op_bitwiseK(L,op) {  \
991
1
  StkId ra = RA(i); \
992
1
  TValue *v1 = vRB(i);  \
993
1
  TValue *v2 = KC(i);  \
994
1
  lua_Integer i1;  \
995
1
  lua_Integer i2 = ivalue(v2);  \
996
1
  if (tointegerns(v1, &i1)) {  \
997
0
    pc++; setivalue(s2v(ra), op(i1, i2));  \
998
0
  }}
999
1000
1001
/*
1002
** Bitwise operations with register operands.
1003
*/
1004
2
#define op_bitwise(L,op) {  \
1005
2
  StkId ra = RA(i); \
1006
2
  TValue *v1 = vRB(i);  \
1007
2
  TValue *v2 = vRC(i);  \
1008
2
  lua_Integer i1; lua_Integer i2;  \
1009
2
  if (tointegerns(v1, &i1) && tointegerns(v2, &i2)) {  \
1010
0
    pc++; setivalue(s2v(ra), op(i1, i2));  \
1011
0
  }}
1012
1013
1014
/*
1015
** Order operations with register operands. 'opn' actually works
1016
** for all numbers, but the fast track improves performance for
1017
** integers.
1018
*/
1019
44
#define op_order(L,opi,opn,other) {  \
1020
44
  StkId ra = RA(i); \
1021
44
  int cond;  \
1022
44
  TValue *rb = vRB(i);  \
1023
44
  if (ttisinteger(s2v(ra)) && ttisinteger(rb)) {  \
1024
0
    lua_Integer ia = ivalue(s2v(ra));  \
1025
0
    lua_Integer ib = ivalue(rb);  \
1026
0
    cond = opi(ia, ib);  \
1027
0
  }  \
1028
44
  else if (ttisnumber(s2v(ra)) && ttisnumber(rb))  \
1029
44
    cond = opn(s2v(ra), rb);  \
1030
44
  else  \
1031
44
    Protect(cond = other(L, s2v(ra), rb));  \
1032
44
  docondjump(); }
1033
1034
1035
/*
1036
** Order operations with immediate operand. (Immediate operand is
1037
** always small enough to have an exact representation as a float.)
1038
*/
1039
1
#define op_orderI(L,opi,opf,inv,tm) {  \
1040
1
  StkId ra = RA(i); \
1041
1
  int cond;  \
1042
1
  int im = GETARG_sB(i);  \
1043
1
  if (ttisinteger(s2v(ra)))  \
1044
1
    cond = opi(ivalue(s2v(ra)), im);  \
1045
1
  else if (ttisfloat(s2v(ra))) {  \
1046
0
    lua_Number fa = fltvalue(s2v(ra));  \
1047
0
    lua_Number fim = cast_num(im);  \
1048
0
    cond = opf(fa, fim);  \
1049
0
  }  \
1050
0
  else {  \
1051
0
    int isf = GETARG_C(i);  \
1052
0
    Protect(cond = luaT_callorderiTM(L, s2v(ra), im, inv, isf, tm));  \
1053
0
  }  \
1054
1
  docondjump(); }
1055
1056
/* }================================================================== */
1057
1058
1059
/*
1060
** {==================================================================
1061
** Function 'luaV_execute': main interpreter loop
1062
** ===================================================================
1063
*/
1064
1065
/*
1066
** some macros for common tasks in 'luaV_execute'
1067
*/
1068
1069
1070
458k
#define RA(i) (base+GETARG_A(i))
1071
#define RB(i) (base+GETARG_B(i))
1072
284
#define vRB(i)  s2v(RB(i))
1073
129
#define KB(i) (k+GETARG_B(i))
1074
#define RC(i) (base+GETARG_C(i))
1075
6
#define vRC(i)  s2v(RC(i))
1076
445k
#define KC(i) (k+GETARG_C(i))
1077
194
#define RKC(i)  ((TESTARG_k(i)) ? k + GETARG_C(i) : s2v(base + GETARG_C(i)))
1078
1079
1080
1081
446k
#define updatetrap(ci)  (trap = ci->u.l.trap)
1082
1083
108
#define updatebase(ci)  (base = ci->func.p + 1)
1084
1085
1086
#define updatestack(ci)  \
1087
0
  { if (l_unlikely(trap)) { updatebase(ci); ra = RA(i); } }
1088
1089
1090
/*
1091
** Execute a jump instruction. The 'updatetrap' allows signals to stop
1092
** tight loops. (Without it, the local copy of 'trap' could never change.)
1093
*/
1094
80
#define dojump(ci,i,e)  { pc += GETARG_sJ(i) + e; updatetrap(ci); }
1095
1096
1097
/* for test instructions, execute the jump instruction that follows it */
1098
80
#define donextjump(ci)  { Instruction ni = *pc; dojump(ci, ni, 1); }
1099
1100
/*
1101
** do a conditional jump: skip next instruction if 'cond' is not what
1102
** was expected (parameter 'k'), else do next instruction, which must
1103
** be a jump.
1104
*/
1105
92
#define docondjump()  if (cond != GETARG_k(i)) pc++; else donextjump(ci);
1106
1107
1108
/*
1109
** Correct global 'pc'.
1110
*/
1111
446k
#define savepc(L) (ci->u.l.savedpc = pc)
1112
1113
1114
/*
1115
** Whenever code can raise errors, the global 'pc' and the global
1116
** 'top' must be correct to report occasional errors.
1117
*/
1118
445k
#define savestate(L,ci)   (savepc(L), L->top.p = ci->top.p)
1119
1120
1121
/*
1122
** Protect code that, in general, can raise errors, reallocate the
1123
** stack, and change the hooks.
1124
*/
1125
445k
#define Protect(exp)  (savestate(L,ci), (exp), updatetrap(ci))
1126
1127
/* special version that does not change the top */
1128
171
#define ProtectNT(exp)  (savepc(L), (exp), updatetrap(ci))
1129
1130
/*
1131
** Protect code that can only raise errors. (That is, it cannot change
1132
** the stack or hooks.)
1133
*/
1134
0
#define halfProtect(exp)  (savestate(L,ci), (exp))
1135
1136
/*
1137
** macro executed during Lua functions at points where the
1138
** function can yield.
1139
*/
1140
#if !defined(luai_threadyield)
1141
138
#define luai_threadyield(L) {lua_unlock(L); lua_lock(L);}
1142
#endif
1143
1144
/* 'c' is the limit of live values in the stack */
1145
#define checkGC(L,c)  \
1146
138
  { luaC_condGC(L, (savepc(L), L->top.p = (c)), \
1147
138
                         updatetrap(ci)); \
1148
138
           luai_threadyield(L); }
1149
1150
1151
/* fetch an instruction and prepare its execution */
1152
458k
#define vmfetch() { \
1153
458k
  if (l_unlikely(trap)) {  /* stack reallocation or hooks? */ \
1154
0
    trap = luaG_traceexec(L, pc);  /* handle hooks */ \
1155
0
    updatebase(ci);  /* correct stack */ \
1156
0
  } \
1157
458k
  i = *(pc++); \
1158
458k
}
1159
1160
#define vmdispatch(o) switch(o)
1161
#define vmcase(l) case l:
1162
#define vmbreak   break
1163
1164
1165
108
void luaV_execute (lua_State *L, CallInfo *ci) {
1166
108
  LClosure *cl;
1167
108
  TValue *k;
1168
108
  StkId base;
1169
108
  const Instruction *pc;
1170
108
  int trap;
1171
108
#if LUA_USE_JUMPTABLE
1172
108
#include "ljumptab.h"
1173
108
#endif
1174
108
 startfunc:
1175
108
  trap = L->hookmask;
1176
108
 returning:  /* trap already set */
1177
216
  cl = ci_func(ci);
1178
0
  k = cl->p->k;
1179
216
  pc = ci->u.l.savedpc;
1180
216
  if (l_unlikely(trap))
1181
0
    trap = luaG_tracecall(L);
1182
216
  base = ci->func.p + 1;
1183
  /* main loop of interpreter */
1184
216
  for (;;) {
1185
108
    Instruction i;  /* instruction being executed */
1186
108
    vmfetch();
1187
    #if 0
1188
    { /* low-level line tracing for debugging Lua */
1189
      #include "lopnames.h"
1190
      int pcrel = pcRel(pc, cl->p);
1191
      printf("line: %d; %s (%d)\n", luaG_getfuncline(cl->p, pcrel),
1192
             opnames[GET_OPCODE(i)], pcrel);
1193
    }
1194
    #endif
1195
108
    lua_assert(base == ci->func.p + 1);
1196
108
    lua_assert(base <= L->top.p && L->top.p <= L->stack_last.p);
1197
    /* for tests, invalidate top for instructions not expecting it */
1198
108
    lua_assert(luaP_isIT(i) || (cast_void(L->top.p = base), 1));
1199
108
    vmdispatch (GET_OPCODE(i)) {
1200
108
      vmcase(OP_MOVE) {
1201
0
        StkId ra = RA(i);
1202
0
        setobjs2s(L, ra, RB(i));
1203
0
        vmbreak;
1204
0
      }
1205
2.52k
      vmcase(OP_LOADI) {
1206
2.52k
        StkId ra = RA(i);
1207
2.52k
        lua_Integer b = GETARG_sBx(i);
1208
2.52k
        setivalue(s2v(ra), b);
1209
2.52k
        vmbreak;
1210
2.52k
      }
1211
0
      vmcase(OP_LOADF) {
1212
0
        StkId ra = RA(i);
1213
0
        int b = GETARG_sBx(i);
1214
0
        setfltvalue(s2v(ra), cast_num(b));
1215
0
        vmbreak;
1216
0
      }
1217
325
      vmcase(OP_LOADK) {
1218
325
        StkId ra = RA(i);
1219
325
        TValue *rb = k + GETARG_Bx(i);
1220
325
        setobj2s(L, ra, rb);
1221
325
        vmbreak;
1222
325
      }
1223
0
      vmcase(OP_LOADKX) {
1224
0
        StkId ra = RA(i);
1225
0
        TValue *rb;
1226
0
        rb = k + GETARG_Ax(*pc); pc++;
1227
0
        setobj2s(L, ra, rb);
1228
0
        vmbreak;
1229
0
      }
1230
0
      vmcase(OP_LOADFALSE) {
1231
0
        StkId ra = RA(i);
1232
0
        setbfvalue(s2v(ra));
1233
0
        vmbreak;
1234
0
      }
1235
12
      vmcase(OP_LFALSESKIP) {
1236
12
        StkId ra = RA(i);
1237
12
        setbfvalue(s2v(ra));
1238
12
        pc++;  /* skip next instruction */
1239
12
        vmbreak;
1240
12
      }
1241
37
      vmcase(OP_LOADTRUE) {
1242
37
        StkId ra = RA(i);
1243
37
        setbtvalue(s2v(ra));
1244
37
        vmbreak;
1245
37
      }
1246
37
      vmcase(OP_LOADNIL) {
1247
3
        StkId ra = RA(i);
1248
3
        int b = GETARG_B(i);
1249
3
        do {
1250
3
          setnilvalue(s2v(ra++));
1251
3
        } while (b--);
1252
3
        vmbreak;
1253
3
      }
1254
0
      vmcase(OP_GETUPVAL) {
1255
0
        StkId ra = RA(i);
1256
0
        int b = GETARG_B(i);
1257
0
        setobj2s(L, ra, cl->upvals[b]->v.p);
1258
0
        vmbreak;
1259
0
      }
1260
0
      vmcase(OP_SETUPVAL) {
1261
0
        StkId ra = RA(i);
1262
0
        UpVal *uv = cl->upvals[GETARG_B(i)];
1263
0
        setobj(L, uv->v.p, s2v(ra));
1264
0
        luaC_barrier(L, uv, s2v(ra));
1265
0
        vmbreak;
1266
0
      }
1267
445k
      vmcase(OP_GETTABUP) {
1268
445k
        StkId ra = RA(i);
1269
445k
        TValue *upval = cl->upvals[GETARG_B(i)]->v.p;
1270
445k
        TValue *rc = KC(i);
1271
891k
        TString *key = tsvalue(rc);  /* key must be a short string */
1272
0
        lu_byte tag;
1273
891k
        luaV_fastget(upval, key, s2v(ra), luaH_getshortstr, tag);
1274
445k
        if (tagisempty(tag))
1275
445k
          Protect(luaV_finishget(L, upval, rc, ra, tag));
1276
445k
        vmbreak;
1277
445k
      }
1278
0
      vmcase(OP_GETTABLE) {
1279
0
        StkId ra = RA(i);
1280
0
        TValue *rb = vRB(i);
1281
0
        TValue *rc = vRC(i);
1282
0
        lu_byte tag;
1283
0
        if (ttisinteger(rc)) {  /* fast track for integers? */
1284
0
          luaV_fastgeti(rb, ivalue(rc), s2v(ra), tag);
1285
0
        }
1286
0
        else
1287
0
          luaV_fastget(rb, rc, s2v(ra), luaH_get, tag);
1288
0
        if (tagisempty(tag))
1289
0
          Protect(luaV_finishget(L, rb, rc, ra, tag));
1290
0
        vmbreak;
1291
0
      }
1292
0
      vmcase(OP_GETI) {
1293
0
        StkId ra = RA(i);
1294
0
        TValue *rb = vRB(i);
1295
0
        int c = GETARG_C(i);
1296
0
        lu_byte tag;
1297
0
        luaV_fastgeti(rb, c, s2v(ra), tag);
1298
0
        if (tagisempty(tag)) {
1299
0
          TValue key;
1300
0
          setivalue(&key, c);
1301
0
          Protect(luaV_finishget(L, rb, &key, ra, tag));
1302
0
        }
1303
0
        vmbreak;
1304
0
      }
1305
1
      vmcase(OP_GETFIELD) {
1306
1
        StkId ra = RA(i);
1307
1
        TValue *rb = vRB(i);
1308
1
        TValue *rc = KC(i);
1309
2
        TString *key = tsvalue(rc);  /* key must be a short string */
1310
0
        lu_byte tag;
1311
2
        luaV_fastget(rb, key, s2v(ra), luaH_getshortstr, tag);
1312
1
        if (tagisempty(tag))
1313
1
          Protect(luaV_finishget(L, rb, rc, ra, tag));
1314
1
        vmbreak;
1315
1
      }
1316
120
      vmcase(OP_SETTABUP) {
1317
120
        int hres;
1318
120
        TValue *upval = cl->upvals[GETARG_A(i)]->v.p;
1319
120
        TValue *rb = KB(i);
1320
120
        TValue *rc = RKC(i);
1321
240
        TString *key = tsvalue(rb);  /* key must be a short string */
1322
120
        luaV_fastset(upval, key, rc, hres, luaH_psetshortstr);
1323
120
        if (hres == HOK)
1324
120
          luaV_finishfastset(L, upval, rc);
1325
5
        else
1326
5
          Protect(luaV_finishset(L, upval, rb, rc, hres));
1327
120
        vmbreak;
1328
120
      }
1329
66
      vmcase(OP_SETTABLE) {
1330
66
        StkId ra = RA(i);
1331
66
        int hres;
1332
66
        TValue *rb = vRB(i);  /* key (table is in 'ra') */
1333
66
        TValue *rc = RKC(i);  /* value */
1334
66
        if (ttisinteger(rb)) {  /* fast track for integers? */
1335
61
          luaV_fastseti(s2v(ra), ivalue(rb), rc, hres);
1336
61
        }
1337
5
        else {
1338
5
          luaV_fastset(s2v(ra), rb, rc, hres, luaH_pset);
1339
5
        }
1340
66
        if (hres == HOK)
1341
66
          luaV_finishfastset(L, s2v(ra), rc);
1342
66
        else
1343
66
          Protect(luaV_finishset(L, s2v(ra), rb, rc, hres));
1344
66
        vmbreak;
1345
66
      }
1346
0
      vmcase(OP_SETI) {
1347
0
        StkId ra = RA(i);
1348
0
        int hres;
1349
0
        int b = GETARG_B(i);
1350
0
        TValue *rc = RKC(i);
1351
0
        luaV_fastseti(s2v(ra), b, rc, hres);
1352
0
        if (hres == HOK)
1353
0
          luaV_finishfastset(L, s2v(ra), rc);
1354
0
        else {
1355
0
          TValue key;
1356
0
          setivalue(&key, b);
1357
0
          Protect(luaV_finishset(L, s2v(ra), &key, rc, hres));
1358
0
        }
1359
0
        vmbreak;
1360
0
      }
1361
8
      vmcase(OP_SETFIELD) {
1362
8
        StkId ra = RA(i);
1363
8
        int hres;
1364
8
        TValue *rb = KB(i);
1365
8
        TValue *rc = RKC(i);
1366
16
        TString *key = tsvalue(rb);  /* key must be a short string */
1367
8
        luaV_fastset(s2v(ra), key, rc, hres, luaH_psetshortstr);
1368
8
        if (hres == HOK)
1369
8
          luaV_finishfastset(L, s2v(ra), rc);
1370
2
        else
1371
2
          Protect(luaV_finishset(L, s2v(ra), rb, rc, hres));
1372
8
        vmbreak;
1373
8
      }
1374
75
      vmcase(OP_NEWTABLE) {
1375
75
        StkId ra = RA(i);
1376
75
        unsigned b = cast_uint(GETARG_vB(i));  /* log2(hash size) + 1 */
1377
75
        unsigned c = cast_uint(GETARG_vC(i));  /* array size */
1378
0
        Table *t;
1379
75
        if (b > 0)
1380
2
          b = 1u << (b - 1);  /* hash size is 2^(b - 1) */
1381
75
        if (TESTARG_k(i)) {  /* non-zero extra argument? */
1382
4
          lua_assert(GETARG_Ax(*pc) != 0);
1383
          /* add it to array size */
1384
4
          c += cast_uint(GETARG_Ax(*pc)) * (MAXARG_vC + 1);
1385
4
        }
1386
75
        pc++;  /* skip extra argument */
1387
75
        L->top.p = ra + 1;  /* correct top in case of emergency GC */
1388
75
        t = luaH_new(L);  /* memory allocation */
1389
75
        sethvalue2s(L, ra, t);
1390
75
        if (b != 0 || c != 0)
1391
70
          luaH_resize(L, t, c, b);  /* idem */
1392
75
        checkGC(L, ra + 1);
1393
75
        vmbreak;
1394
75
      }
1395
0
      vmcase(OP_SELF) {
1396
0
        StkId ra = RA(i);
1397
0
        lu_byte tag;
1398
0
        TValue *rb = vRB(i);
1399
0
        TValue *rc = KC(i);
1400
0
        TString *key = tsvalue(rc);  /* key must be a short string */
1401
0
        setobj2s(L, ra + 1, rb);
1402
0
        luaV_fastget(rb, key, s2v(ra), luaH_getshortstr, tag);
1403
0
        if (tagisempty(tag))
1404
0
          Protect(luaV_finishget(L, rb, rc, ra, tag));
1405
0
        vmbreak;
1406
0
      }
1407
0
      vmcase(OP_ADDI) {
1408
0
        op_arithI(L, l_addi, luai_numadd);
1409
0
        vmbreak;
1410
0
      }
1411
0
      vmcase(OP_ADDK) {
1412
0
        op_arithK(L, l_addi, luai_numadd);
1413
0
        vmbreak;
1414
0
      }
1415
0
      vmcase(OP_SUBK) {
1416
0
        op_arithK(L, l_subi, luai_numsub);
1417
0
        vmbreak;
1418
0
      }
1419
1
      vmcase(OP_MULK) {
1420
4
        op_arithK(L, l_muli, luai_nummul);
1421
4
        vmbreak;
1422
4
      }
1423
0
      vmcase(OP_MODK) {
1424
0
        savestate(L, ci);  /* in case of division by 0 */
1425
0
        op_arithK(L, luaV_mod, luaV_modf);
1426
0
        vmbreak;
1427
0
      }
1428
1
      vmcase(OP_POWK) {
1429
3
        op_arithfK(L, luai_numpow);
1430
3
        vmbreak;
1431
3
      }
1432
0
      vmcase(OP_DIVK) {
1433
0
        op_arithfK(L, luai_numdiv);
1434
0
        vmbreak;
1435
0
      }
1436
0
      vmcase(OP_IDIVK) {
1437
0
        savestate(L, ci);  /* in case of division by 0 */
1438
0
        op_arithK(L, luaV_idiv, luai_numidiv);
1439
0
        vmbreak;
1440
0
      }
1441
0
      vmcase(OP_BANDK) {
1442
0
        op_bitwiseK(L, l_band);
1443
0
        vmbreak;
1444
0
      }
1445
0
      vmcase(OP_BORK) {
1446
0
        op_bitwiseK(L, l_bor);
1447
0
        vmbreak;
1448
0
      }
1449
1
      vmcase(OP_BXORK) {
1450
3
        op_bitwiseK(L, l_bxor);
1451
3
        vmbreak;
1452
3
      }
1453
0
      vmcase(OP_SHRI) {
1454
0
        StkId ra = RA(i);
1455
0
        TValue *rb = vRB(i);
1456
0
        int ic = GETARG_sC(i);
1457
0
        lua_Integer ib;
1458
0
        if (tointegerns(rb, &ib)) {
1459
0
          pc++; setivalue(s2v(ra), luaV_shiftl(ib, -ic));
1460
0
        }
1461
0
        vmbreak;
1462
0
      }
1463
0
      vmcase(OP_SHLI) {
1464
0
        StkId ra = RA(i);
1465
0
        TValue *rb = vRB(i);
1466
0
        int ic = GETARG_sC(i);
1467
0
        lua_Integer ib;
1468
0
        if (tointegerns(rb, &ib)) {
1469
0
          pc++; setivalue(s2v(ra), luaV_shiftl(ic, ib));
1470
0
        }
1471
0
        vmbreak;
1472
0
      }
1473
2
      vmcase(OP_ADD) {
1474
6
        op_arith(L, l_addi, luai_numadd);
1475
6
        vmbreak;
1476
6
      }
1477
1
      vmcase(OP_SUB) {
1478
3
        op_arith(L, l_subi, luai_numsub);
1479
3
        vmbreak;
1480
3
      }
1481
0
      vmcase(OP_MUL) {
1482
0
        op_arith(L, l_muli, luai_nummul);
1483
0
        vmbreak;
1484
0
      }
1485
0
      vmcase(OP_MOD) {
1486
0
        savestate(L, ci);  /* in case of division by 0 */
1487
0
        op_arith(L, luaV_mod, luaV_modf);
1488
0
        vmbreak;
1489
0
      }
1490
0
      vmcase(OP_POW) {
1491
0
        op_arithf(L, luai_numpow);
1492
0
        vmbreak;
1493
0
      }
1494
1
      vmcase(OP_DIV) {  /* float division (always with floats) */
1495
2
        op_arithf(L, luai_numdiv);
1496
2
        vmbreak;
1497
2
      }
1498
0
      vmcase(OP_IDIV) {  /* floor division */
1499
0
        savestate(L, ci);  /* in case of division by 0 */
1500
0
        op_arith(L, luaV_idiv, luai_numidiv);
1501
0
        vmbreak;
1502
0
      }
1503
2
      vmcase(OP_BAND) {
1504
4
        op_bitwise(L, l_band);
1505
4
        vmbreak;
1506
4
      }
1507
0
      vmcase(OP_BOR) {
1508
0
        op_bitwise(L, l_bor);
1509
0
        vmbreak;
1510
0
      }
1511
0
      vmcase(OP_BXOR) {
1512
0
        op_bitwise(L, l_bxor);
1513
0
        vmbreak;
1514
0
      }
1515
0
      vmcase(OP_SHR) {
1516
0
        op_bitwise(L, luaV_shiftr);
1517
0
        vmbreak;
1518
0
      }
1519
0
      vmcase(OP_SHL) {
1520
0
        op_bitwise(L, luaV_shiftl);
1521
0
        vmbreak;
1522
0
      }
1523
6
      vmcase(OP_MMBIN) {
1524
6
        StkId ra = RA(i);
1525
6
        Instruction pi = *(pc - 2);  /* original arith. expression */
1526
6
        TValue *rb = vRB(i);
1527
6
        TMS tm = (TMS)GETARG_C(i);
1528
6
        StkId result = RA(pi);
1529
6
        lua_assert(OP_ADD <= GET_OPCODE(pi) && GET_OPCODE(pi) <= OP_SHR);
1530
6
        Protect(luaT_trybinTM(L, s2v(ra), rb, result, tm));
1531
6
        vmbreak;
1532
6
      }
1533
0
      vmcase(OP_MMBINI) {
1534
0
        StkId ra = RA(i);
1535
0
        Instruction pi = *(pc - 2);  /* original arith. expression */
1536
0
        int imm = GETARG_sB(i);
1537
0
        TMS tm = (TMS)GETARG_C(i);
1538
0
        int flip = GETARG_k(i);
1539
0
        StkId result = RA(pi);
1540
0
        Protect(luaT_trybiniTM(L, s2v(ra), imm, flip, result, tm));
1541
0
        vmbreak;
1542
0
      }
1543
1
      vmcase(OP_MMBINK) {
1544
1
        StkId ra = RA(i);
1545
1
        Instruction pi = *(pc - 2);  /* original arith. expression */
1546
1
        TValue *imm = KB(i);
1547
1
        TMS tm = (TMS)GETARG_C(i);
1548
1
        int flip = GETARG_k(i);
1549
1
        StkId result = RA(pi);
1550
1
        Protect(luaT_trybinassocTM(L, s2v(ra), imm, flip, result, tm));
1551
1
        vmbreak;
1552
1
      }
1553
7
      vmcase(OP_UNM) {
1554
7
        StkId ra = RA(i);
1555
7
        TValue *rb = vRB(i);
1556
0
        lua_Number nb;
1557
7
        if (ttisinteger(rb)) {
1558
0
          lua_Integer ib = ivalue(rb);
1559
0
          setivalue(s2v(ra), intop(-, 0, ib));
1560
0
        }
1561
7
        else if (tonumberns(rb, nb)) {
1562
0
          setfltvalue(s2v(ra), luai_numunm(L, nb));
1563
0
        }
1564
7
        else
1565
7
          Protect(luaT_trybinTM(L, rb, rb, ra, TM_UNM));
1566
7
        vmbreak;
1567
7
      }
1568
118
      vmcase(OP_BNOT) {
1569
118
        StkId ra = RA(i);
1570
118
        TValue *rb = vRB(i);
1571
0
        lua_Integer ib;
1572
118
        if (tointegerns(rb, &ib)) {
1573
34
          setivalue(s2v(ra), intop(^, ~l_castS2U(0), ib));
1574
34
        }
1575
84
        else
1576
84
          Protect(luaT_trybinTM(L, rb, rb, ra, TM_BNOT));
1577
118
        vmbreak;
1578
118
      }
1579
33
      vmcase(OP_NOT) {
1580
33
        StkId ra = RA(i);
1581
33
        TValue *rb = vRB(i);
1582
33
        if (l_isfalse(rb))
1583
33
          setbtvalue(s2v(ra));
1584
33
        else
1585
33
          setbfvalue(s2v(ra));
1586
33
        vmbreak;
1587
33
      }
1588
39
      vmcase(OP_LEN) {
1589
39
        StkId ra = RA(i);
1590
39
        Protect(luaV_objlen(L, ra, vRB(i)));
1591
39
        vmbreak;
1592
39
      }
1593
63
      vmcase(OP_CONCAT) {
1594
63
        StkId ra = RA(i);
1595
63
        int n = GETARG_B(i);  /* number of elements to concatenate */
1596
0
        L->top.p = ra + n;  /* mark the end of concat operands */
1597
63
        ProtectNT(luaV_concat(L, n));
1598
63
        checkGC(L, L->top.p); /* 'luaV_concat' ensures correct top */
1599
63
        vmbreak;
1600
63
      }
1601
0
      vmcase(OP_CLOSE) {
1602
0
        StkId ra = RA(i);
1603
0
        lua_assert(!GETARG_B(i));  /* 'close must be alive */
1604
0
        Protect(luaF_close(L, ra, LUA_OK, 1));
1605
0
        vmbreak;
1606
0
      }
1607
0
      vmcase(OP_TBC) {
1608
0
        StkId ra = RA(i);
1609
        /* create new to-be-closed upvalue */
1610
0
        halfProtect(luaF_newtbcupval(L, ra));
1611
0
        vmbreak;
1612
0
      }
1613
0
      vmcase(OP_JMP) {
1614
0
        dojump(ci, i, 0);
1615
0
        vmbreak;
1616
0
      }
1617
0
      vmcase(OP_EQ) {
1618
0
        StkId ra = RA(i);
1619
0
        int cond;
1620
0
        TValue *rb = vRB(i);
1621
0
        Protect(cond = luaV_equalobj(L, s2v(ra), rb));
1622
0
        docondjump();
1623
0
        vmbreak;
1624
0
      }
1625
27
      vmcase(OP_LT) {
1626
81
        op_order(L, l_lti, LTnum, lessthanothers);
1627
81
        vmbreak;
1628
81
      }
1629
17
      vmcase(OP_LE) {
1630
50
        op_order(L, l_lei, LEnum, lessequalothers);
1631
50
        vmbreak;
1632
50
      }
1633
0
      vmcase(OP_EQK) {
1634
0
        StkId ra = RA(i);
1635
0
        TValue *rb = KB(i);
1636
        /* basic types do not use '__eq'; we can use raw equality */
1637
0
        int cond = luaV_rawequalobj(s2v(ra), rb);
1638
0
        docondjump();
1639
0
        vmbreak;
1640
0
      }
1641
5
      vmcase(OP_EQI) {
1642
5
        StkId ra = RA(i);
1643
5
        int cond;
1644
5
        int im = GETARG_sB(i);
1645
5
        if (ttisinteger(s2v(ra)))
1646
5
          cond = (ivalue(s2v(ra)) == im);
1647
0
        else if (ttisfloat(s2v(ra)))
1648
0
          cond = luai_numeq(fltvalue(s2v(ra)), cast_num(im));
1649
0
        else
1650
0
          cond = 0;  /* other types cannot be equal to a number */
1651
10
        docondjump();
1652
10
        vmbreak;
1653
10
      }
1654
1
      vmcase(OP_LTI) {
1655
3
        op_orderI(L, l_lti, luai_numlt, 0, TM_LT);
1656
3
        vmbreak;
1657
3
      }
1658
0
      vmcase(OP_LEI) {
1659
0
        op_orderI(L, l_lei, luai_numle, 0, TM_LE);
1660
0
        vmbreak;
1661
0
      }
1662
0
      vmcase(OP_GTI) {
1663
0
        op_orderI(L, l_gti, luai_numgt, 1, TM_LT);
1664
0
        vmbreak;
1665
0
      }
1666
0
      vmcase(OP_GEI) {
1667
0
        op_orderI(L, l_gei, luai_numge, 1, TM_LE);
1668
0
        vmbreak;
1669
0
      }
1670
42
      vmcase(OP_TEST) {
1671
42
        StkId ra = RA(i);
1672
42
        int cond = !l_isfalse(s2v(ra));
1673
42
        docondjump();
1674
42
        vmbreak;
1675
42
      }
1676
0
      vmcase(OP_TESTSET) {
1677
0
        StkId ra = RA(i);
1678
0
        TValue *rb = vRB(i);
1679
0
        if (l_isfalse(rb) == GETARG_k(i))
1680
0
          pc++;
1681
0
        else {
1682
0
          setobj2s(L, ra, rb);
1683
0
          donextjump(ci);
1684
0
        }
1685
0
        vmbreak;
1686
0
      }
1687
5
      vmcase(OP_CALL) {
1688
5
        StkId ra = RA(i);
1689
5
        CallInfo *newci;
1690
5
        int b = GETARG_B(i);
1691
5
        int nresults = GETARG_C(i) - 1;
1692
5
        if (b != 0)  /* fixed number of arguments? */
1693
5
          L->top.p = ra + b;  /* top signals number of arguments */
1694
        /* else previous instruction set top */
1695
5
        savepc(L);  /* in case of errors */
1696
5
        if ((newci = luaD_precall(L, ra, nresults)) == NULL)
1697
0
          updatetrap(ci);  /* C call; nothing else to be done */
1698
5
        else {  /* Lua call: run function in this same C frame */
1699
5
          ci = newci;
1700
5
          goto startfunc;
1701
5
        }
1702
5
        vmbreak;
1703
0
      }
1704
0
      vmcase(OP_TAILCALL) {
1705
0
        StkId ra = RA(i);
1706
0
        int b = GETARG_B(i);  /* number of arguments + 1 (function) */
1707
0
        int n;  /* number of results when calling a C function */
1708
0
        int nparams1 = GETARG_C(i);
1709
        /* delta is virtual 'func' - real 'func' (vararg functions) */
1710
0
        int delta = (nparams1) ? ci->u.l.nextraargs + nparams1 : 0;
1711
0
        if (b != 0)
1712
0
          L->top.p = ra + b;
1713
0
        else  /* previous instruction set top */
1714
0
          b = cast_int(L->top.p - ra);
1715
0
        savepc(ci);  /* several calls here can raise errors */
1716
0
        if (TESTARG_k(i)) {
1717
0
          luaF_closeupval(L, base);  /* close upvalues from current call */
1718
0
          lua_assert(L->tbclist.p < base);  /* no pending tbc variables */
1719
0
          lua_assert(base == ci->func.p + 1);
1720
0
        }
1721
0
        if ((n = luaD_pretailcall(L, ci, ra, b, delta)) < 0)  /* Lua function? */
1722
0
          goto startfunc;  /* execute the callee */
1723
0
        else {  /* C function? */
1724
0
          ci->func.p -= delta;  /* restore 'func' (if vararg) */
1725
0
          luaD_poscall(L, ci, n);  /* finish caller */
1726
0
          updatetrap(ci);  /* 'luaD_poscall' can change hooks */
1727
0
          goto ret;  /* caller returns after the tail call */
1728
0
        }
1729
0
      }
1730
1
      vmcase(OP_RETURN) {
1731
1
        StkId ra = RA(i);
1732
1
        int n = GETARG_B(i) - 1;  /* number of results */
1733
1
        int nparams1 = GETARG_C(i);
1734
1
        if (n < 0)  /* not fixed? */
1735
0
          n = cast_int(L->top.p - ra);  /* get what is available */
1736
1
        savepc(ci);
1737
1
        if (TESTARG_k(i)) {  /* may there be open upvalues? */
1738
0
          ci->u2.nres = n;  /* save number of returns */
1739
0
          if (L->top.p < ci->top.p)
1740
0
            L->top.p = ci->top.p;
1741
0
          luaF_close(L, base, CLOSEKTOP, 1);
1742
0
          updatetrap(ci);
1743
0
          updatestack(ci);
1744
0
        }
1745
1
        if (nparams1)  /* vararg function? */
1746
1
          ci->func.p -= ci->u.l.nextraargs + nparams1;
1747
1
        L->top.p = ra + n;  /* set call for 'luaD_poscall' */
1748
1
        luaD_poscall(L, ci, n);
1749
1
        updatetrap(ci);  /* 'luaD_poscall' can change hooks */
1750
1
        goto ret;
1751
1
      }
1752
0
      vmcase(OP_RETURN0) {
1753
0
        if (l_unlikely(L->hookmask)) {
1754
0
          StkId ra = RA(i);
1755
0
          L->top.p = ra;
1756
0
          savepc(ci);
1757
0
          luaD_poscall(L, ci, 0);  /* no hurry... */
1758
0
          trap = 1;
1759
0
        }
1760
0
        else {  /* do the 'poscall' here */
1761
0
          int nres = get_nresults(ci->callstatus);
1762
0
          L->ci = ci->previous;  /* back to caller */
1763
0
          L->top.p = base - 1;
1764
0
          for (; l_unlikely(nres > 0); nres--)
1765
0
            setnilvalue(s2v(L->top.p++));  /* all results are nil */
1766
0
        }
1767
0
        goto ret;
1768
1
      }
1769
0
      vmcase(OP_RETURN1) {
1770
0
        if (l_unlikely(L->hookmask)) {
1771
0
          StkId ra = RA(i);
1772
0
          L->top.p = ra + 1;
1773
0
          savepc(ci);
1774
0
          luaD_poscall(L, ci, 1);  /* no hurry... */
1775
0
          trap = 1;
1776
0
        }
1777
0
        else {  /* do the 'poscall' here */
1778
0
          int nres = get_nresults(ci->callstatus);
1779
0
          L->ci = ci->previous;  /* back to caller */
1780
0
          if (nres == 0)
1781
0
            L->top.p = base - 1;  /* asked for no results */
1782
0
          else {
1783
0
            StkId ra = RA(i);
1784
0
            setobjs2s(L, base - 1, ra);  /* at least this result */
1785
0
            L->top.p = base;
1786
0
            for (; l_unlikely(nres > 1); nres--)
1787
0
              setnilvalue(s2v(L->top.p++));  /* complete missing results */
1788
0
          }
1789
0
        }
1790
1
       ret:  /* return from a Lua function */
1791
1
        if (ci->callstatus & CIST_FRESH)
1792
1
          return;  /* end this frame */
1793
0
        else {
1794
0
          ci = ci->previous;
1795
0
          goto returning;  /* continue running caller in this frame */
1796
0
        }
1797
1
      }
1798
0
      vmcase(OP_FORLOOP) {
1799
0
        StkId ra = RA(i);
1800
0
        if (ttisinteger(s2v(ra + 1))) {  /* integer loop? */
1801
0
          lua_Unsigned count = l_castS2U(ivalue(s2v(ra)));
1802
0
          if (count > 0) {  /* still more iterations? */
1803
0
            lua_Integer step = ivalue(s2v(ra + 1));
1804
0
            lua_Integer idx = ivalue(s2v(ra + 2));  /* control variable */
1805
0
            chgivalue(s2v(ra), l_castU2S(count - 1));  /* update counter */
1806
0
            idx = intop(+, idx, step);  /* add step to index */
1807
0
            chgivalue(s2v(ra + 2), idx);  /* update control variable */
1808
0
            pc -= GETARG_Bx(i);  /* jump back */
1809
0
          }
1810
0
        }
1811
0
        else if (floatforloop(ra))  /* float loop */
1812
0
          pc -= GETARG_Bx(i);  /* jump back */
1813
0
        updatetrap(ci);  /* allows a signal to break the loop */
1814
0
        vmbreak;
1815
0
      }
1816
4
      vmcase(OP_FORPREP) {
1817
4
        StkId ra = RA(i);
1818
4
        savestate(L, ci);  /* in case of errors */
1819
4
        if (forprep(L, ra))
1820
0
          pc += GETARG_Bx(i) + 1;  /* skip the loop */
1821
4
        vmbreak;
1822
4
      }
1823
0
      vmcase(OP_TFORPREP) {
1824
       /* before: 'ra' has the iterator function, 'ra + 1' has the state,
1825
          'ra + 2' has the initial value for the control variable, and
1826
          'ra + 3' has the closing variable. This opcode then swaps the
1827
          control and the closing variables and marks the closing variable
1828
          as to-be-closed.
1829
       */
1830
0
       StkId ra = RA(i);
1831
0
       TValue temp;  /* to swap control and closing variables */
1832
0
       setobj(L, &temp, s2v(ra + 3));
1833
0
       setobjs2s(L, ra + 3, ra + 2);
1834
0
       setobj2s(L, ra + 2, &temp);
1835
        /* create to-be-closed upvalue (if closing var. is not nil) */
1836
0
        halfProtect(luaF_newtbcupval(L, ra + 2));
1837
0
        pc += GETARG_Bx(i);  /* go to end of the loop */
1838
0
        i = *(pc++);  /* fetch next instruction */
1839
0
        lua_assert(GET_OPCODE(i) == OP_TFORCALL && ra == RA(i));
1840
0
        goto l_tforcall;
1841
0
      }
1842
0
      vmcase(OP_TFORCALL) {
1843
0
       l_tforcall: {
1844
        /* 'ra' has the iterator function, 'ra + 1' has the state,
1845
           'ra + 2' has the closing variable, and 'ra + 3' has the control
1846
           variable. The call will use the stack starting at 'ra + 3',
1847
           so that it preserves the first three values, and the first
1848
           return will be the new value for the control variable.
1849
        */
1850
0
        StkId ra = RA(i);
1851
0
        setobjs2s(L, ra + 5, ra + 3);  /* copy the control variable */
1852
0
        setobjs2s(L, ra + 4, ra + 1);  /* copy state */
1853
0
        setobjs2s(L, ra + 3, ra);  /* copy function */
1854
0
        L->top.p = ra + 3 + 3;
1855
0
        ProtectNT(luaD_call(L, ra + 3, GETARG_C(i)));  /* do the call */
1856
0
        updatestack(ci);  /* stack may have changed */
1857
0
        i = *(pc++);  /* go to next instruction */
1858
0
        lua_assert(GET_OPCODE(i) == OP_TFORLOOP && ra == RA(i));
1859
0
        goto l_tforloop;
1860
0
      }}
1861
0
      vmcase(OP_TFORLOOP) {
1862
0
       l_tforloop: {
1863
0
        StkId ra = RA(i);
1864
0
        if (!ttisnil(s2v(ra + 3)))  /* continue loop? */
1865
0
          pc -= GETARG_Bx(i);  /* jump back */
1866
0
        vmbreak;
1867
0
      }}
1868
9.02k
      vmcase(OP_SETLIST) {
1869
9.02k
        StkId ra = RA(i);
1870
9.02k
        unsigned n = cast_uint(GETARG_vB(i));
1871
9.02k
        unsigned int last = cast_uint(GETARG_vC(i));
1872
18.0k
        Table *h = hvalue(s2v(ra));
1873
9.02k
        if (n == 0)
1874
0
          n = cast_uint(L->top.p - ra) - 1;  /* get up to the top */
1875
9.02k
        else
1876
9.02k
          L->top.p = ci->top.p;  /* correct top in case of emergency GC */
1877
18.0k
        last += n;
1878
18.0k
        if (TESTARG_k(i)) {
1879
8.87k
          last += cast_uint(GETARG_Ax(*pc)) * (MAXARG_vC + 1);
1880
0
          pc++;
1881
8.87k
        }
1882
        /* when 'n' is known, table should have proper size */
1883
9.02k
        if (last > h->asize) {  /* needs more space? */
1884
          /* fixed-size sets should have space preallocated */
1885
0
          lua_assert(GETARG_vB(i) == 0);
1886
0
          luaH_resizearray(L, h, last);  /* preallocate it at once */
1887
0
        }
1888
457k
        for (; n > 0; n--) {
1889
448k
          TValue *val = s2v(ra + n);
1890
448k
          obj2arr(h, last - 1, val);
1891
448k
          last--;
1892
448k
          luaC_barrierback(L, obj2gco(h), val);
1893
448k
        }
1894
9.02k
        vmbreak;
1895
9.02k
      }
1896
0
      vmcase(OP_CLOSURE) {
1897
0
        StkId ra = RA(i);
1898
0
        Proto *p = cl->p->p[GETARG_Bx(i)];
1899
0
        halfProtect(pushclosure(L, p, cl->upvals, base, ra));
1900
0
        checkGC(L, ra + 1);
1901
0
        vmbreak;
1902
0
      }
1903
0
      vmcase(OP_VARARG) {
1904
0
        StkId ra = RA(i);
1905
0
        int n = GETARG_C(i) - 1;  /* required results */
1906
0
        Protect(luaT_getvarargs(L, ci, ra, n));
1907
0
        vmbreak;
1908
0
      }
1909
108
      vmcase(OP_VARARGPREP) {
1910
108
        ProtectNT(luaT_adjustvarargs(L, GETARG_A(i), ci, cl->p));
1911
108
        if (l_unlikely(trap)) {  /* previous "Protect" updated trap */
1912
0
          luaD_hookcall(L, ci);
1913
0
          L->oldpc = 1;  /* next opcode will be seen as a "new" line */
1914
0
        }
1915
108
        updatebase(ci);  /* function has new base after adjustment */
1916
108
        vmbreak;
1917
108
      }
1918
108
      vmcase(OP_EXTRAARG) {
1919
0
        lua_assert(0);
1920
0
        vmbreak;
1921
0
      }
1922
0
    }
1923
0
  }
1924
216
}
1925
1926
/* }================================================================== */