Coverage Report

Created: 2023-08-27 06:20

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