Coverage Report

Created: 2025-12-14 06:44

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/src/testdir/build/lua-master/source/lparser.c
Line
Count
Source
1
/*
2
** $Id: lparser.c $
3
** Lua Parser
4
** See Copyright Notice in lua.h
5
*/
6
7
#define lparser_c
8
#define LUA_CORE
9
10
#include "lprefix.h"
11
12
13
#include <limits.h>
14
#include <string.h>
15
16
#include "lua.h"
17
18
#include "lcode.h"
19
#include "ldebug.h"
20
#include "ldo.h"
21
#include "lfunc.h"
22
#include "llex.h"
23
#include "lmem.h"
24
#include "lobject.h"
25
#include "lopcodes.h"
26
#include "lparser.h"
27
#include "lstate.h"
28
#include "lstring.h"
29
#include "ltable.h"
30
31
32
33
/* maximum number of variable declarations per function (must be
34
   smaller than 250, due to the bytecode format) */
35
3.86M
#define MAXVARS   200
36
37
38
7.93M
#define hasmultret(k)   ((k) == VCALL || (k) == VVARARG)
39
40
41
/* because all strings are unified by the scanner, the parser
42
   can use pointer equality for string equality */
43
1.98G
#define eqstr(a,b)  ((a) == (b))
44
45
46
/*
47
** nodes for block list (list of active blocks)
48
*/
49
typedef struct BlockCnt {
50
  struct BlockCnt *previous;  /* chain */
51
  int firstlabel;  /* index of first label in this block */
52
  int firstgoto;  /* index of first pending goto in this block */
53
  short nactvar;  /* number of active declarations at block entry */
54
  lu_byte upval;  /* true if some variable in the block is an upvalue */
55
  lu_byte isloop;  /* 1 if 'block' is a loop; 2 if it has pending breaks */
56
  lu_byte insidetbc;  /* true if inside the scope of a to-be-closed var. */
57
} BlockCnt;
58
59
60
61
/*
62
** prototypes for recursive non-terminal functions
63
*/
64
static void statement (LexState *ls);
65
static void expr (LexState *ls, expdesc *v);
66
67
68
748k
static l_noret error_expected (LexState *ls, int token) {
69
748k
  luaX_syntaxerror(ls,
70
748k
      luaO_pushfstring(ls->L, "%s expected", luaX_token2str(ls, token)));
71
748k
}
72
73
74
1.70k
static l_noret errorlimit (FuncState *fs, int limit, const char *what) {
75
1.70k
  lua_State *L = fs->ls->L;
76
1.70k
  const char *msg;
77
1.70k
  int line = fs->f->linedefined;
78
1.70k
  const char *where = (line == 0)
79
1.70k
                      ? "main function"
80
1.70k
                      : luaO_pushfstring(L, "function at line %d", line);
81
1.70k
  msg = luaO_pushfstring(L, "too many %s (limit is %d) in %s",
82
1.70k
                             what, limit, where);
83
1.70k
  luaX_syntaxerror(fs->ls, msg);
84
1.70k
}
85
86
87
36.5M
void luaY_checklimit (FuncState *fs, int v, int l, const char *what) {
88
36.5M
  if (l_unlikely(v > l)) errorlimit(fs, l, what);
89
36.5M
}
90
91
92
/*
93
** Test whether next token is 'c'; if so, skip it.
94
*/
95
42.6M
static int testnext (LexState *ls, int c) {
96
42.6M
  if (ls->t.token == c) {
97
25.1M
    luaX_next(ls);
98
25.1M
    return 1;
99
25.1M
  }
100
17.4M
  else return 0;
101
42.6M
}
102
103
104
/*
105
** Check that next token is 'c'.
106
*/
107
211M
static void check (LexState *ls, int c) {
108
211M
  if (ls->t.token != c)
109
455k
    error_expected(ls, c);
110
211M
}
111
112
113
/*
114
** Check that next token is 'c' and skip it.
115
*/
116
12.2M
static void checknext (LexState *ls, int c) {
117
12.2M
  check(ls, c);
118
12.2M
  luaX_next(ls);
119
12.2M
}
120
121
122
12.4M
#define check_condition(ls,c,msg) { if (!(c)) luaX_syntaxerror(ls, msg); }
123
124
125
/*
126
** Check that next token is 'what' and skip it. In case of error,
127
** raise an error that the expected 'what' should match a 'who'
128
** in line 'where' (if that is not the current line).
129
*/
130
5.37M
static void check_match (LexState *ls, int what, int who, int where) {
131
5.37M
  if (l_unlikely(!testnext(ls, what))) {
132
306k
    if (where == ls->linenumber)  /* all in the same line? */
133
293k
      error_expected(ls, what);  /* do not need a complex message */
134
12.5k
    else {
135
12.5k
      luaX_syntaxerror(ls, luaO_pushfstring(ls->L,
136
12.5k
             "%s expected (to close %s at line %d)",
137
12.5k
              luaX_token2str(ls, what), luaX_token2str(ls, who), where));
138
12.5k
    }
139
306k
  }
140
5.37M
}
141
142
143
196M
static TString *str_checkname (LexState *ls) {
144
196M
  TString *ts;
145
196M
  check(ls, TK_NAME);
146
196M
  ts = ls->t.seminfo.ts;
147
196M
  luaX_next(ls);
148
196M
  return ts;
149
196M
}
150
151
152
527M
static void init_exp (expdesc *e, expkind k, int i) {
153
527M
  e->f = e->t = NO_JUMP;
154
527M
  e->k = k;
155
527M
  e->u.info = i;
156
527M
}
157
158
159
160M
static void codestring (expdesc *e, TString *s) {
160
160M
  e->f = e->t = NO_JUMP;
161
160M
  e->k = VKSTR;
162
160M
  e->u.strval = s;
163
160M
}
164
165
166
10.8M
static void codename (LexState *ls, expdesc *e) {
167
10.8M
  codestring(e, str_checkname(ls));
168
10.8M
}
169
170
171
/*
172
** Register a new local variable in the active 'Proto' (for debug
173
** information).
174
*/
175
static short registerlocalvar (LexState *ls, FuncState *fs,
176
3.86M
                               TString *varname) {
177
3.86M
  Proto *f = fs->f;
178
3.86M
  int oldsize = f->sizelocvars;
179
3.86M
  luaM_growvector(ls->L, f->locvars, fs->ndebugvars, f->sizelocvars,
180
3.86M
                  LocVar, SHRT_MAX, "local variables");
181
11.4M
  while (oldsize < f->sizelocvars)
182
7.54M
    f->locvars[oldsize++].varname = NULL;
183
3.86M
  f->locvars[fs->ndebugvars].varname = varname;
184
3.86M
  f->locvars[fs->ndebugvars].startpc = fs->pc;
185
3.86M
  luaC_objbarrier(ls->L, f, varname);
186
3.86M
  return fs->ndebugvars++;
187
3.86M
}
188
189
190
/*
191
** Create a new variable with the given 'name' and given 'kind'.
192
** Return its index in the function.
193
*/
194
5.55M
static int new_varkind (LexState *ls, TString *name, lu_byte kind) {
195
5.55M
  lua_State *L = ls->L;
196
5.55M
  FuncState *fs = ls->fs;
197
5.55M
  Dyndata *dyd = ls->dyd;
198
5.55M
  Vardesc *var;
199
5.55M
  luaM_growvector(L, dyd->actvar.arr, dyd->actvar.n + 1,
200
5.55M
             dyd->actvar.size, Vardesc, SHRT_MAX, "variable declarations");
201
5.55M
  var = &dyd->actvar.arr[dyd->actvar.n++];
202
5.55M
  var->vd.kind = kind;  /* default */
203
5.55M
  var->vd.name = name;
204
5.55M
  return dyd->actvar.n - 1 - fs->firstlocal;
205
5.55M
}
206
207
208
/*
209
** Create a new local variable with the given 'name' and regular kind.
210
*/
211
2.23M
static int new_localvar (LexState *ls, TString *name) {
212
2.23M
  return new_varkind(ls, name, VDKREG);
213
2.23M
}
214
215
#define new_localvarliteral(ls,v) \
216
6.54k
    new_localvar(ls,  \
217
6.54k
      luaX_newstring(ls, "" v, (sizeof(v)/sizeof(char)) - 1));
218
219
220
221
/*
222
** Return the "variable description" (Vardesc) of a given variable.
223
** (Unless noted otherwise, all variables are referred to by their
224
** compiler indices.)
225
*/
226
2.00G
static Vardesc *getlocalvardesc (FuncState *fs, int vidx) {
227
2.00G
  return &fs->ls->dyd->actvar.arr[fs->firstlocal + vidx];
228
2.00G
}
229
230
231
/*
232
** Convert 'nvar', a compiler index level, to its corresponding
233
** register. For that, search for the highest variable below that level
234
** that is in a register and uses its register index ('ridx') plus one.
235
*/
236
482M
static lu_byte reglevel (FuncState *fs, int nvar) {
237
771M
  while (nvar-- > 0) {
238
492M
    Vardesc *vd = getlocalvardesc(fs, nvar);  /* get previous variable */
239
492M
    if (varinreg(vd))  /* is in a register? */
240
203M
      return cast_byte(vd->vd.ridx + 1);
241
492M
  }
242
278M
  return 0;  /* no variables in registers */
243
482M
}
244
245
246
/*
247
** Return the number of variables in the register stack for the given
248
** function.
249
*/
250
472M
lu_byte luaY_nvarstack (FuncState *fs) {
251
472M
  return reglevel(fs, fs->nactvar);
252
472M
}
253
254
255
/*
256
** Get the debug-information entry for current variable 'vidx'.
257
*/
258
2.54M
static LocVar *localdebuginfo (FuncState *fs, int vidx) {
259
2.54M
  Vardesc *vd = getlocalvardesc(fs,  vidx);
260
2.54M
  if (!varinreg(vd))
261
71.9k
    return NULL;  /* no debug info. for constants */
262
2.47M
  else {
263
2.47M
    int idx = vd->vd.pidx;
264
2.47M
    lua_assert(idx < fs->ndebugvars);
265
2.47M
    return &fs->f->locvars[idx];
266
2.47M
  }
267
2.54M
}
268
269
270
/*
271
** Create an expression representing variable 'vidx'
272
*/
273
6.52M
static void init_var (FuncState *fs, expdesc *e, int vidx) {
274
6.52M
  e->f = e->t = NO_JUMP;
275
6.52M
  e->k = VLOCAL;
276
6.52M
  e->u.var.vidx = cast_short(vidx);
277
6.52M
  e->u.var.ridx = getlocalvardesc(fs, vidx)->vd.ridx;
278
6.52M
}
279
280
281
/*
282
** Raises an error if variable described by 'e' is read only; moreover,
283
** if 'e' is t[exp] where t is the vararg parameter, change it to index
284
** a real table. (Virtual vararg tables cannot be changed.)
285
*/
286
6.16M
static void check_readonly (LexState *ls, expdesc *e) {
287
6.16M
  FuncState *fs = ls->fs;
288
6.16M
  TString *varname = NULL;  /* to be set if variable is const */
289
6.16M
  switch (e->k) {
290
2
    case VCONST: {
291
2
      varname = ls->dyd->actvar.arr[e->u.info].vd.name;
292
2
      break;
293
0
    }
294
152k
    case VLOCAL: case VVARGVAR: {
295
152k
      Vardesc *vardesc = getlocalvardesc(fs, e->u.var.vidx);
296
152k
      if (vardesc->vd.kind != VDKREG)  /* not a regular variable? */
297
388
        varname = vardesc->vd.name;
298
152k
      break;
299
152k
    }
300
246k
    case VUPVAL: {
301
246k
      Upvaldesc *up = &fs->f->upvalues[e->u.info];
302
246k
      if (up->kind != VDKREG)
303
5
        varname = up->name;
304
246k
      break;
305
152k
    }
306
56.9k
    case VVARGIND: {
307
56.9k
      needvatab(fs->f);  /* function will need a vararg table */
308
56.9k
      e->k = VINDEXED;
309
56.9k
    }  /* FALLTHROUGH */
310
5.75M
    case VINDEXUP: case VINDEXSTR: case VINDEXED: {  /* global variable */
311
5.75M
      if (e->u.ind.ro)  /* read-only? */
312
787
        varname = tsvalue(&fs->f->k[e->u.ind.keystr]);
313
5.75M
      break;
314
5.75M
    }
315
5.75M
    default:
316
7.96k
      lua_assert(e->k == VINDEXI);  /* this one doesn't need any check */
317
7.96k
      return;  /* integer index cannot be read-only */
318
6.16M
  }
319
6.15M
  if (varname)
320
1.18k
    luaK_semerror(ls, "attempt to assign to const variable '%s'",
321
1.18k
                      getstr(varname));
322
6.15M
}
323
324
325
/*
326
** Start the scope for the last 'nvars' created variables.
327
*/
328
2.67M
static void adjustlocalvars (LexState *ls, int nvars) {
329
2.67M
  FuncState *fs = ls->fs;
330
2.67M
  int reglevel = luaY_nvarstack(fs);
331
2.67M
  int i;
332
6.54M
  for (i = 0; i < nvars; i++) {
333
3.86M
    int vidx = fs->nactvar++;
334
3.86M
    Vardesc *var = getlocalvardesc(fs, vidx);
335
3.86M
    var->vd.ridx = cast_byte(reglevel++);
336
3.86M
    var->vd.pidx = registerlocalvar(ls, fs, var->vd.name);
337
3.86M
    luaY_checklimit(fs, reglevel, MAXVARS, "local variables");
338
3.86M
  }
339
2.67M
}
340
341
342
/*
343
** Close the scope for all variables up to level 'tolevel'.
344
** (debug info.)
345
*/
346
4.85M
static void removevars (FuncState *fs, int tolevel) {
347
4.85M
  fs->ls->dyd->actvar.n -= (fs->nactvar - tolevel);
348
7.36M
  while (fs->nactvar > tolevel) {
349
2.50M
    LocVar *var = localdebuginfo(fs, --fs->nactvar);
350
2.50M
    if (var)  /* does it have debug information? */
351
2.43M
      var->endpc = fs->pc;
352
2.50M
  }
353
4.85M
}
354
355
356
/*
357
** Search the upvalues of the function 'fs' for one
358
** with the given 'name'.
359
*/
360
417M
static int searchupvalue (FuncState *fs, TString *name) {
361
417M
  int i;
362
417M
  Upvaldesc *up = fs->f->upvalues;
363
793M
  for (i = 0; i < fs->nups; i++) {
364
493M
    if (eqstr(up[i].name, name)) return i;
365
493M
  }
366
299M
  return -1;  /* not found */
367
417M
}
368
369
370
9.58M
static Upvaldesc *allocupvalue (FuncState *fs) {
371
9.58M
  Proto *f = fs->f;
372
9.58M
  int oldsize = f->sizeupvalues;
373
9.58M
  luaY_checklimit(fs, fs->nups + 1, MAXUPVAL, "upvalues");
374
9.58M
  luaM_growvector(fs->ls->L, f->upvalues, fs->nups, f->sizeupvalues,
375
9.58M
                  Upvaldesc, MAXUPVAL, "upvalues");
376
46.4M
  while (oldsize < f->sizeupvalues)
377
36.8M
    f->upvalues[oldsize++].name = NULL;
378
9.58M
  return &f->upvalues[fs->nups++];
379
9.58M
}
380
381
382
1.42M
static int newupvalue (FuncState *fs, TString *name, expdesc *v) {
383
1.42M
  Upvaldesc *up = allocupvalue(fs);
384
1.42M
  FuncState *prev = fs->prev;
385
1.42M
  if (v->k == VLOCAL) {
386
310k
    up->instack = 1;
387
310k
    up->idx = v->u.var.ridx;
388
310k
    up->kind = getlocalvardesc(prev, v->u.var.vidx)->vd.kind;
389
310k
    lua_assert(eqstr(name, getlocalvardesc(prev, v->u.var.vidx)->vd.name));
390
310k
  }
391
1.10M
  else {
392
1.10M
    up->instack = 0;
393
1.10M
    up->idx = cast_byte(v->u.info);
394
1.10M
    up->kind = prev->f->upvalues[v->u.info].kind;
395
1.10M
    lua_assert(eqstr(name, prev->f->upvalues[v->u.info].name));
396
1.10M
  }
397
1.42M
  up->name = name;
398
1.42M
  luaC_objbarrier(fs->ls->L, fs->f, name);
399
1.42M
  return fs->nups - 1;
400
1.42M
}
401
402
403
/*
404
** Look for an active variable with the name 'n' in the
405
** function 'fs'. If found, initialize 'var' with it and return
406
** its expression kind; otherwise return -1. While searching,
407
** var->u.info==-1 means that the preambular global declaration is
408
** active (the default while there is no other global declaration);
409
** var->u.info==-2 means there is no active collective declaration
410
** (some previous global declaration but no collective declaration);
411
** and var->u.info>=0 points to the inner-most (the first one found)
412
** collective declaration, if there is one.
413
*/
414
480M
static int searchvar (FuncState *fs, TString *n, expdesc *var) {
415
480M
  int i;
416
1.91G
  for (i = cast_int(fs->nactvar) - 1; i >= 0; i--) {
417
1.50G
    Vardesc *vd = getlocalvardesc(fs, i);
418
1.50G
    if (varglobal(vd)) {  /* global declaration? */
419
212M
      if (vd->vd.name == NULL) {  /* collective declaration? */
420
6.67M
        if (var->u.info < 0)  /* no previous collective declaration? */
421
265k
          var->u.info = fs->firstlocal + i;  /* this is the first one */
422
6.67M
      }
423
206M
      else {  /* global name */
424
206M
        if (eqstr(n, vd->vd.name)) {  /* found? */
425
58.5k
          init_exp(var, VGLOBAL, fs->firstlocal + i);
426
58.5k
          return VGLOBAL;
427
58.5k
        }
428
206M
        else if (var->u.info == -1)  /* active preambular declaration? */
429
196k
          var->u.info = -2;  /* invalidate preambular declaration */
430
206M
      }
431
212M
    }
432
1.28G
    else if (eqstr(n, vd->vd.name)) {  /* found? */
433
62.2M
      if (vd->vd.kind == RDKCTC)  /* compile-time constant? */
434
55.7M
        init_exp(var, VCONST, fs->firstlocal + i);
435
6.52M
      else {  /* local variable */
436
6.52M
        init_var(fs, var, i);
437
6.52M
        if (vd->vd.kind == RDKVAVAR)  /* vararg parameter? */
438
202k
          var->k = VVARGVAR;
439
6.52M
      }
440
62.2M
      return cast_int(var->k);
441
62.2M
    }
442
1.50G
  }
443
417M
  return -1;  /* not found */
444
480M
}
445
446
447
/*
448
** Mark block where variable at given level was defined
449
** (to emit close instructions later).
450
*/
451
310k
static void markupval (FuncState *fs, int level) {
452
310k
  BlockCnt *bl = fs->bl;
453
321k
  while (bl->nactvar > level)
454
11.7k
    bl = bl->previous;
455
310k
  bl->upval = 1;
456
310k
  fs->needclose = 1;
457
310k
}
458
459
460
/*
461
** Mark that current block has a to-be-closed variable.
462
*/
463
43.5k
static void marktobeclosed (FuncState *fs) {
464
43.5k
  BlockCnt *bl = fs->bl;
465
43.5k
  bl->upval = 1;
466
43.5k
  bl->insidetbc = 1;
467
43.5k
  fs->needclose = 1;
468
43.5k
}
469
470
471
/*
472
** Find a variable with the given name 'n'. If it is an upvalue, add
473
** this upvalue into all intermediate functions. If it is a global, set
474
** 'var' as 'void' as a flag.
475
*/
476
480M
static void singlevaraux (FuncState *fs, TString *n, expdesc *var, int base) {
477
480M
  int v = searchvar(fs, n, var);  /* look up variables at current level */
478
480M
  if (v >= 0) {  /* found? */
479
62.3M
    if (!base) {
480
3.97M
      if (var->k == VVARGVAR)  /* vararg parameter? */
481
342
        luaK_vapar2local(fs, var);  /* change it to a regular local */
482
3.97M
      if (var->k == VLOCAL)
483
310k
        markupval(fs, var->u.var.vidx);  /* will be used as an upvalue */
484
3.97M
    }
485
    /* else nothing else to be done */
486
62.3M
  }
487
417M
  else {  /* not found at current level; try upvalues */
488
417M
    int idx = searchupvalue(fs, n);  /* try existing upvalues */
489
417M
    if (idx < 0) {  /* not found? */
490
299M
      if (fs->prev != NULL)  /* more levels? */
491
152M
        singlevaraux(fs->prev, n, var, 0);  /* try upper levels */
492
299M
      if (var->k == VLOCAL || var->k == VUPVAL)  /* local or upvalue? */
493
1.42M
        idx  = newupvalue(fs, n, var);  /* will be a new upvalue */
494
298M
      else  /* it is a global or a constant */
495
298M
        return;  /* don't need to do anything at this level */
496
299M
    }
497
119M
    init_exp(var, VUPVAL, idx);  /* new or old upvalue */
498
119M
  }
499
480M
}
500
501
502
147M
static void buildglobal (LexState *ls, TString *varname, expdesc *var) {
503
147M
  FuncState *fs = ls->fs;
504
147M
  expdesc key;
505
147M
  init_exp(var, VGLOBAL, -1);  /* global by default */
506
147M
  singlevaraux(fs, ls->envn, var, 1);  /* get environment variable */
507
147M
  if (var->k == VGLOBAL)
508
285
    luaK_semerror(ls, "%s is global when accessing variable '%s'",
509
285
                      LUA_ENV, getstr(varname));
510
147M
  luaK_exp2anyregup(fs, var);  /* _ENV could be a constant */
511
147M
  codestring(&key, varname);  /* key is variable name */
512
147M
  luaK_indexed(fs, var, &key);  /* 'var' represents _ENV[varname] */
513
147M
}
514
515
516
/*
517
** Find a variable with the given name 'n', handling global variables
518
** too.
519
*/
520
179M
static void buildvar (LexState *ls, TString *varname, expdesc *var) {
521
179M
  FuncState *fs = ls->fs;
522
179M
  init_exp(var, VGLOBAL, -1);  /* global by default */
523
179M
  singlevaraux(fs, varname, var, 1);
524
179M
  if (var->k == VGLOBAL) {  /* global name? */
525
147M
    int info = var->u.info;
526
    /* global by default in the scope of a global declaration? */
527
147M
    if (info == -2)
528
7.83k
      luaK_semerror(ls, "variable '%s' not declared", getstr(varname));
529
147M
    buildglobal(ls, varname, var);
530
147M
    if (info != -1 && ls->dyd->actvar.arr[info].vd.kind == GDKCONST)
531
5.21k
      var->u.ind.ro = 1;  /* mark variable as read-only */
532
147M
    else  /* anyway must be a global */
533
147M
      lua_assert(info == -1 || ls->dyd->actvar.arr[info].vd.kind == GDKREG);
534
147M
  }
535
179M
}
536
537
538
180M
static void singlevar (LexState *ls, expdesc *var) {
539
180M
  buildvar(ls, str_checkname(ls), var);
540
180M
}
541
542
543
/*
544
** Adjust the number of results from an expression list 'e' with 'nexps'
545
** expressions to 'nvars' values.
546
*/
547
1.01M
static void adjust_assign (LexState *ls, int nvars, int nexps, expdesc *e) {
548
1.01M
  FuncState *fs = ls->fs;
549
1.01M
  int needed = nvars - nexps;  /* extra values needed */
550
1.01M
  luaK_checkstack(fs, needed);
551
1.01M
  if (hasmultret(e->k)) {  /* last expression has multiple returns? */
552
301k
    int extra = needed + 1;  /* discount last expression itself */
553
301k
    if (extra < 0)
554
5.43k
      extra = 0;
555
301k
    luaK_setreturns(fs, e, extra);  /* last exp. provides the difference */
556
301k
  }
557
711k
  else {
558
711k
    if (e->k != VVOID)  /* at least one expression? */
559
636k
      luaK_exp2nextreg(fs, e);  /* close last expression */
560
711k
    if (needed > 0)  /* missing values? */
561
198k
      luaK_nil(fs, fs->freereg, needed);  /* complete with nils */
562
711k
  }
563
1.01M
  if (needed > 0)
564
329k
    luaK_reserveregs(fs, needed);  /* registers for extra values */
565
682k
  else  /* adding 'needed' is actually a subtraction */
566
682k
    fs->freereg = cast_byte(fs->freereg + needed);  /* remove extra values */
567
1.01M
}
568
569
570
216M
#define enterlevel(ls)  luaE_incCstack(ls->L)
571
572
573
210M
#define leavelevel(ls) ((ls)->L->nCcalls--)
574
575
576
/*
577
** Generates an error that a goto jumps into the scope of some
578
** variable declaration.
579
*/
580
15
static l_noret jumpscopeerror (LexState *ls, Labeldesc *gt) {
581
15
  TString *tsname = getlocalvardesc(ls->fs, gt->nactvar)->vd.name;
582
15
  const char *varname = (tsname != NULL) ? getstr(tsname) : "*";
583
15
  luaK_semerror(ls,
584
15
     "<goto %s> at line %d jumps into the scope of '%s'",
585
15
      getstr(gt->name), gt->line, varname);  /* raise the error */
586
15
}
587
588
589
/*
590
** Closes the goto at index 'g' to given 'label' and removes it
591
** from the list of pending gotos.
592
** If it jumps into the scope of some variable, raises an error.
593
** The goto needs a CLOSE if it jumps out of a block with upvalues,
594
** or out of the scope of some variable and the block has upvalues
595
** (signaled by parameter 'bup').
596
*/
597
113k
static void closegoto (LexState *ls, int g, Labeldesc *label, int bup) {
598
113k
  int i;
599
113k
  FuncState *fs = ls->fs;
600
113k
  Labellist *gl = &ls->dyd->gt;  /* list of gotos */
601
113k
  Labeldesc *gt = &gl->arr[g];  /* goto to be resolved */
602
113k
  lua_assert(eqstr(gt->name, label->name));
603
113k
  if (l_unlikely(gt->nactvar < label->nactvar))  /* enter some scope? */
604
15
    jumpscopeerror(ls, gt);
605
113k
  if (gt->close ||
606
110k
      (label->nactvar < gt->nactvar && bup)) {  /* needs close? */
607
24.0k
    lu_byte stklevel = reglevel(fs, label->nactvar);
608
    /* move jump to CLOSE position */
609
24.0k
    fs->f->code[gt->pc + 1] = fs->f->code[gt->pc];
610
    /* put CLOSE instruction at original position */
611
24.0k
    fs->f->code[gt->pc] = CREATE_ABCk(OP_CLOSE, stklevel, 0, 0, 0);
612
24.0k
    gt->pc++;  /* must point to jump instruction */
613
24.0k
  }
614
113k
  luaK_patchlist(ls->fs, gt->pc, label->pc);  /* goto jumps to label */
615
4.41M
  for (i = g; i < gl->n - 1; i++)  /* remove goto from pending list */
616
4.30M
    gl->arr[i] = gl->arr[i + 1];
617
113k
  gl->n--;
618
113k
}
619
620
621
/*
622
** Search for an active label with the given name, starting at
623
** index 'ilb' (so that it can search for all labels in current block
624
** or all labels in current function).
625
*/
626
461k
static Labeldesc *findlabel (LexState *ls, TString *name, int ilb) {
627
461k
  Dyndata *dyd = ls->dyd;
628
539k
  for (; ilb < dyd->label.n; ilb++) {
629
197k
    Labeldesc *lb = &dyd->label.arr[ilb];
630
197k
    if (eqstr(lb->name, name))  /* correct label? */
631
119k
      return lb;
632
197k
  }
633
342k
  return NULL;  /* label not found */
634
461k
}
635
636
637
/*
638
** Adds a new label/goto in the corresponding list.
639
*/
640
static int newlabelentry (LexState *ls, Labellist *l, TString *name,
641
534k
                          int line, int pc) {
642
534k
  int n = l->n;
643
534k
  luaM_growvector(ls->L, l->arr, n, l->size,
644
534k
                  Labeldesc, SHRT_MAX, "labels/gotos");
645
534k
  l->arr[n].name = name;
646
534k
  l->arr[n].line = line;
647
534k
  l->arr[n].nactvar = ls->fs->nactvar;
648
534k
  l->arr[n].close = 0;
649
534k
  l->arr[n].pc = pc;
650
534k
  l->n = n + 1;
651
534k
  return n;
652
534k
}
653
654
655
/*
656
** Create an entry for the goto and the code for it. As it is not known
657
** at this point whether the goto may need a CLOSE, the code has a jump
658
** followed by an CLOSE. (As the CLOSE comes after the jump, it is a
659
** dead instruction; it works as a placeholder.) When the goto is closed
660
** against a label, if it needs a CLOSE, the two instructions swap
661
** positions, so that the CLOSE comes before the jump.
662
*/
663
367k
static int newgotoentry (LexState *ls, TString *name, int line) {
664
367k
  FuncState *fs = ls->fs;
665
367k
  int pc = luaK_jump(fs);  /* create jump */
666
367k
  luaK_codeABC(fs, OP_CLOSE, 0, 1, 0);  /* spaceholder, marked as dead */
667
367k
  return newlabelentry(ls, &ls->dyd->gt, name, line, pc);
668
367k
}
669
670
671
/*
672
** Create a new label with the given 'name' at the given 'line'.
673
** 'last' tells whether label is the last non-op statement in its
674
** block. Solves all pending gotos to this new label and adds
675
** a close instruction if necessary.
676
** Returns true iff it added a close instruction.
677
*/
678
166k
static void createlabel (LexState *ls, TString *name, int line, int last) {
679
166k
  FuncState *fs = ls->fs;
680
166k
  Labellist *ll = &ls->dyd->label;
681
166k
  int l = newlabelentry(ls, ll, name, line, luaK_getlabel(fs));
682
166k
  if (last) {  /* label is last no-op statement in the block? */
683
    /* assume that locals are already out of scope */
684
14.1k
    ll->arr[l].nactvar = fs->bl->nactvar;
685
14.1k
  }
686
166k
}
687
688
689
/*
690
** Traverse the pending gotos of the finishing block checking whether
691
** each match some label of that block. Those that do not match are
692
** "exported" to the outer block, to be solved there. In particular,
693
** its 'nactvar' is updated with the level of the inner block,
694
** as the variables of the inner block are now out of scope.
695
*/
696
4.85M
static void solvegotos (FuncState *fs, BlockCnt *bl) {
697
4.85M
  LexState *ls = fs->ls;
698
4.85M
  Labellist *gl = &ls->dyd->gt;
699
4.85M
  int outlevel = reglevel(fs, bl->nactvar);  /* level outside the block */
700
4.85M
  int igt = bl->firstgoto;  /* first goto in the finishing block */
701
5.24M
  while (igt < gl->n) {   /* for each pending goto */
702
386k
    Labeldesc *gt = &gl->arr[igt];
703
    /* search for a matching label in the current block */
704
386k
    Labeldesc *lb = findlabel(ls, gt->name, bl->firstlabel);
705
386k
    if (lb != NULL)  /* found a match? */
706
113k
      closegoto(ls, igt, lb, bl->upval);  /* close and remove goto */
707
272k
    else {  /* adjust 'goto' for outer block */
708
      /* block has variables to be closed and goto escapes the scope of
709
         some variable? */
710
272k
      if (bl->upval && reglevel(fs, gt->nactvar) > outlevel)
711
5.08k
        gt->close = 1;  /* jump may need a close */
712
272k
      gt->nactvar = bl->nactvar;  /* correct level for outer block */
713
272k
      igt++;  /* go to next goto */
714
272k
    }
715
386k
  }
716
4.85M
  ls->dyd->label.n = bl->firstlabel;  /* remove local labels */
717
4.85M
}
718
719
720
10.7M
static void enterblock (FuncState *fs, BlockCnt *bl, lu_byte isloop) {
721
10.7M
  bl->isloop = isloop;
722
10.7M
  bl->nactvar = fs->nactvar;
723
10.7M
  bl->firstlabel = fs->ls->dyd->label.n;
724
10.7M
  bl->firstgoto = fs->ls->dyd->gt.n;
725
10.7M
  bl->upval = 0;
726
  /* inherit 'insidetbc' from enclosing block */
727
10.7M
  bl->insidetbc = (fs->bl != NULL && fs->bl->insidetbc);
728
10.7M
  bl->previous = fs->bl;  /* link block in function's block list */
729
10.7M
  fs->bl = bl;
730
10.7M
  lua_assert(fs->freereg == luaY_nvarstack(fs));
731
10.7M
}
732
733
734
/*
735
** generates an error for an undefined 'goto'.
736
*/
737
1.46k
static l_noret undefgoto (LexState *ls, Labeldesc *gt) {
738
  /* breaks are checked when created, cannot be undefined */
739
1.46k
  lua_assert(!eqstr(gt->name, ls->brkn));
740
1.46k
  luaK_semerror(ls, "no visible label '%s' for <goto> at line %d",
741
1.46k
                    getstr(gt->name), gt->line);
742
1.46k
}
743
744
745
4.85M
static void leaveblock (FuncState *fs) {
746
4.85M
  BlockCnt *bl = fs->bl;
747
4.85M
  LexState *ls = fs->ls;
748
4.85M
  lu_byte stklevel = reglevel(fs, bl->nactvar);  /* level outside block */
749
4.85M
  if (bl->previous && bl->upval)  /* need a 'close'? */
750
31.7k
    luaK_codeABC(fs, OP_CLOSE, stklevel, 0, 0);
751
4.85M
  fs->freereg = stklevel;  /* free registers */
752
4.85M
  removevars(fs, bl->nactvar);  /* remove block locals */
753
4.85M
  lua_assert(bl->nactvar == fs->nactvar);  /* back to level on entry */
754
4.85M
  if (bl->isloop == 2)  /* has to fix pending breaks? */
755
96.4k
    createlabel(ls, ls->brkn, 0, 0);
756
4.85M
  solvegotos(fs, bl);
757
4.85M
  if (bl->previous == NULL) {  /* was it the last block? */
758
4.14M
    if (bl->firstgoto < ls->dyd->gt.n)  /* still pending gotos? */
759
1.46k
      undefgoto(ls, &ls->dyd->gt.arr[bl->firstgoto]);  /* error */
760
4.14M
  }
761
4.85M
  fs->bl = bl->previous;  /* current block now is previous one */
762
4.85M
}
763
764
765
/*
766
** adds a new prototype into list of prototypes
767
*/
768
1.74M
static Proto *addprototype (LexState *ls) {
769
1.74M
  Proto *clp;
770
1.74M
  lua_State *L = ls->L;
771
1.74M
  FuncState *fs = ls->fs;
772
1.74M
  Proto *f = fs->f;  /* prototype of current function */
773
1.74M
  if (fs->np >= f->sizep) {
774
733k
    int oldsize = f->sizep;
775
733k
    luaM_growvector(L, f->p, fs->np, f->sizep, Proto *, MAXARG_Bx, "functions");
776
4.93M
    while (oldsize < f->sizep)
777
4.19M
      f->p[oldsize++] = NULL;
778
733k
  }
779
1.74M
  f->p[fs->np++] = clp = luaF_newproto(L);
780
1.74M
  luaC_objbarrier(L, f, clp);
781
1.74M
  return clp;
782
1.74M
}
783
784
785
/*
786
** codes instruction to create new closure in parent function.
787
** The OP_CLOSURE instruction uses the last available register,
788
** so that, if it invokes the GC, the GC knows which registers
789
** are in use at that time.
790
791
*/
792
1.16M
static void codeclosure (LexState *ls, expdesc *v) {
793
1.16M
  FuncState *fs = ls->fs->prev;
794
1.16M
  init_exp(v, VRELOC, luaK_codeABx(fs, OP_CLOSURE, 0, fs->np - 1));
795
1.16M
  luaK_exp2nextreg(fs, v);  /* fix it at the last register */
796
1.16M
}
797
798
799
9.90M
static void open_func (LexState *ls, FuncState *fs, BlockCnt *bl) {
800
9.90M
  lua_State *L = ls->L;
801
9.90M
  Proto *f = fs->f;
802
9.90M
  fs->prev = ls->fs;  /* linked list of funcstates */
803
9.90M
  fs->ls = ls;
804
9.90M
  ls->fs = fs;
805
9.90M
  fs->pc = 0;
806
9.90M
  fs->previousline = f->linedefined;
807
9.90M
  fs->iwthabs = 0;
808
9.90M
  fs->lasttarget = 0;
809
9.90M
  fs->freereg = 0;
810
9.90M
  fs->nk = 0;
811
9.90M
  fs->nabslineinfo = 0;
812
9.90M
  fs->np = 0;
813
9.90M
  fs->nups = 0;
814
9.90M
  fs->ndebugvars = 0;
815
9.90M
  fs->nactvar = 0;
816
9.90M
  fs->needclose = 0;
817
9.90M
  fs->firstlocal = ls->dyd->actvar.n;
818
9.90M
  fs->firstlabel = ls->dyd->label.n;
819
9.90M
  fs->bl = NULL;
820
9.90M
  f->source = ls->source;
821
9.90M
  luaC_objbarrier(L, f, f->source);
822
9.90M
  f->maxstacksize = 2;  /* registers 0/1 are always valid */
823
9.90M
  fs->kcache = luaH_new(L);  /* create table for function */
824
9.90M
  sethvalue2s(L, L->top.p, fs->kcache);  /* anchor it */
825
9.90M
  luaD_inctop(L);
826
9.90M
  enterblock(fs, bl, 0);
827
9.90M
}
828
829
830
4.14M
static void close_func (LexState *ls) {
831
4.14M
  lua_State *L = ls->L;
832
4.14M
  FuncState *fs = ls->fs;
833
4.14M
  Proto *f = fs->f;
834
4.14M
  luaK_ret(fs, luaY_nvarstack(fs), 0);  /* final return */
835
4.14M
  leaveblock(fs);
836
4.14M
  lua_assert(fs->bl == NULL);
837
4.14M
  luaK_finish(fs);
838
4.14M
  luaM_shrinkvector(L, f->code, f->sizecode, fs->pc, Instruction);
839
4.14M
  luaM_shrinkvector(L, f->lineinfo, f->sizelineinfo, fs->pc, ls_byte);
840
4.14M
  luaM_shrinkvector(L, f->abslineinfo, f->sizeabslineinfo,
841
4.14M
                       fs->nabslineinfo, AbsLineInfo);
842
4.14M
  luaM_shrinkvector(L, f->k, f->sizek, fs->nk, TValue);
843
4.14M
  luaM_shrinkvector(L, f->p, f->sizep, fs->np, Proto *);
844
4.14M
  luaM_shrinkvector(L, f->locvars, f->sizelocvars, fs->ndebugvars, LocVar);
845
4.14M
  luaM_shrinkvector(L, f->upvalues, f->sizeupvalues, fs->nups, Upvaldesc);
846
4.14M
  ls->fs = fs->prev;
847
4.14M
  L->top.p--;  /* pop kcache table */
848
4.14M
  luaC_checkGC(L);
849
4.14M
}
850
851
852
/*
853
** {======================================================================
854
** GRAMMAR RULES
855
** =======================================================================
856
*/
857
858
859
/*
860
** check whether current token is in the follow set of a block.
861
** 'until' closes syntactical blocks, but do not close scope,
862
** so it is handled in separate.
863
*/
864
16.8M
static int block_follow (LexState *ls, int withuntil) {
865
16.8M
  switch (ls->t.token) {
866
19.6k
    case TK_ELSE: case TK_ELSEIF:
867
4.00M
    case TK_END: case TK_EOS:
868
4.00M
      return 1;
869
58.7k
    case TK_UNTIL: return withuntil;
870
12.7M
    default: return 0;
871
16.8M
  }
872
16.8M
}
873
874
875
10.3M
static void statlist (LexState *ls) {
876
  /* statlist -> { stat [';'] } */
877
21.7M
  while (!block_follow(ls, 1)) {
878
12.0M
    if (ls->t.token == TK_RETURN) {
879
721k
      statement(ls);
880
721k
      return;  /* 'return' must be last statement */
881
721k
    }
882
11.3M
    statement(ls);
883
11.3M
  }
884
10.3M
}
885
886
887
7.77M
static void fieldsel (LexState *ls, expdesc *v) {
888
  /* fieldsel -> ['.' | ':'] NAME */
889
7.77M
  FuncState *fs = ls->fs;
890
7.77M
  expdesc key;
891
7.77M
  luaK_exp2anyregup(fs, v);
892
7.77M
  luaX_next(ls);  /* skip the dot or colon */
893
7.77M
  codename(ls, &key);
894
7.77M
  luaK_indexed(fs, v, &key);
895
7.77M
}
896
897
898
527k
static void yindex (LexState *ls, expdesc *v) {
899
  /* index -> '[' expr ']' */
900
527k
  luaX_next(ls);  /* skip the '[' */
901
527k
  expr(ls, v);
902
527k
  luaK_exp2val(ls->fs, v);
903
527k
  checknext(ls, ']');
904
527k
}
905
906
907
/*
908
** {======================================================================
909
** Rules for Constructors
910
** =======================================================================
911
*/
912
913
typedef struct ConsControl {
914
  expdesc v;  /* last list item read */
915
  expdesc *t;  /* table descriptor */
916
  int nh;  /* total number of 'record' elements */
917
  int na;  /* number of array elements already stored */
918
  int tostore;  /* number of array elements pending to be stored */
919
  int maxtostore;  /* maximum number of pending elements */
920
} ConsControl;
921
922
923
/*
924
** Maximum number of elements in a constructor, to control the following:
925
** * counter overflows;
926
** * overflows in 'extra' for OP_NEWTABLE and OP_SETLIST;
927
** * overflows when adding multiple returns in OP_SETLIST.
928
*/
929
12.3M
#define MAX_CNST  (INT_MAX/2)
930
#if MAX_CNST/(MAXARG_vC + 1) > MAXARG_Ax
931
#undef MAX_CNST
932
#define MAX_CNST  (MAXARG_Ax * (MAXARG_vC + 1))
933
#endif
934
935
936
2.95M
static void recfield (LexState *ls, ConsControl *cc) {
937
  /* recfield -> (NAME | '['exp']') = exp */
938
2.95M
  FuncState *fs = ls->fs;
939
2.95M
  lu_byte reg = ls->fs->freereg;
940
2.95M
  expdesc tab, key, val;
941
2.95M
  if (ls->t.token == TK_NAME)
942
2.82M
    codename(ls, &key);
943
128k
  else  /* ls->t.token == '[' */
944
128k
    yindex(ls, &key);
945
2.95M
  cc->nh++;
946
2.95M
  checknext(ls, '=');
947
2.95M
  tab = *cc->t;
948
2.95M
  luaK_indexed(fs, &tab, &key);
949
2.95M
  expr(ls, &val);
950
2.95M
  luaK_storevar(fs, &tab, &val);
951
2.95M
  fs->freereg = reg;  /* free registers */
952
2.95M
}
953
954
955
7.57M
static void closelistfield (FuncState *fs, ConsControl *cc) {
956
7.57M
  lua_assert(cc->tostore > 0);
957
7.57M
  luaK_exp2nextreg(fs, &cc->v);
958
7.57M
  cc->v.k = VVOID;
959
7.57M
  if (cc->tostore >= cc->maxtostore) {
960
883k
    luaK_setlist(fs, cc->t->u.info, cc->na, cc->tostore);  /* flush */
961
883k
    cc->na += cc->tostore;
962
883k
    cc->tostore = 0;  /* no more items pending */
963
883k
  }
964
7.57M
}
965
966
967
392k
static void lastlistfield (FuncState *fs, ConsControl *cc) {
968
392k
  if (cc->tostore == 0) return;
969
78.7k
  if (hasmultret(cc->v.k)) {
970
13.8k
    luaK_setmultret(fs, &cc->v);
971
13.8k
    luaK_setlist(fs, cc->t->u.info, cc->na, LUA_MULTRET);
972
13.8k
    cc->na--;  /* do not count last expression (unknown number of elements) */
973
13.8k
  }
974
64.9k
  else {
975
64.9k
    if (cc->v.k != VVOID)
976
53.6k
      luaK_exp2nextreg(fs, &cc->v);
977
64.9k
    luaK_setlist(fs, cc->t->u.info, cc->na, cc->tostore);
978
64.9k
  }
979
78.7k
  cc->na += cc->tostore;
980
78.7k
}
981
982
983
9.41M
static void listfield (LexState *ls, ConsControl *cc) {
984
  /* listfield -> exp */
985
9.41M
  expr(ls, &cc->v);
986
9.41M
  cc->tostore++;
987
9.41M
}
988
989
990
12.3M
static void field (LexState *ls, ConsControl *cc) {
991
  /* field -> listfield | recfield */
992
12.3M
  switch(ls->t.token) {
993
11.1M
    case TK_NAME: {  /* may be 'listfield' or 'recfield' */
994
11.1M
      if (luaX_lookahead(ls) != '=')  /* expression? */
995
8.31M
        listfield(ls, cc);
996
2.83M
      else
997
2.83M
        recfield(ls, cc);
998
11.1M
      break;
999
0
    }
1000
128k
    case '[': {
1001
128k
      recfield(ls, cc);
1002
128k
      break;
1003
0
    }
1004
1.10M
    default: {
1005
1.10M
      listfield(ls, cc);
1006
1.10M
      break;
1007
0
    }
1008
12.3M
  }
1009
12.3M
}
1010
1011
1012
/*
1013
** Compute a limit for how many registers a constructor can use before
1014
** emitting a 'SETLIST' instruction, based on how many registers are
1015
** available.
1016
*/
1017
2.21M
static int maxtostore (FuncState *fs) {
1018
2.21M
  int numfreeregs = MAX_FSTACK - fs->freereg;
1019
2.21M
  if (numfreeregs >= 160)  /* "lots" of registers? */
1020
1.88M
    return numfreeregs / 5;  /* use up to 1/5 of them */
1021
334k
  else if (numfreeregs >= 80)  /* still "enough" registers? */
1022
255k
    return 10;  /* one 'SETLIST' instruction for each 10 values */
1023
79.1k
  else  /* save registers for potential more nesting */
1024
79.1k
    return 1;
1025
2.21M
}
1026
1027
1028
2.21M
static void constructor (LexState *ls, expdesc *t) {
1029
  /* constructor -> '{' [ field { sep field } [sep] ] '}'
1030
     sep -> ',' | ';' */
1031
2.21M
  FuncState *fs = ls->fs;
1032
2.21M
  int line = ls->linenumber;
1033
2.21M
  int pc = luaK_codevABCk(fs, OP_NEWTABLE, 0, 0, 0, 0);
1034
2.21M
  ConsControl cc;
1035
2.21M
  luaK_code(fs, 0);  /* space for extra arg. */
1036
2.21M
  cc.na = cc.nh = cc.tostore = 0;
1037
2.21M
  cc.t = t;
1038
2.21M
  init_exp(t, VNONRELOC, fs->freereg);  /* table will be at stack top */
1039
2.21M
  luaK_reserveregs(fs, 1);
1040
2.21M
  init_exp(&cc.v, VVOID, 0);  /* no value (yet) */
1041
2.21M
  checknext(ls, '{' /*}*/);
1042
2.21M
  cc.maxtostore = maxtostore(fs);
1043
12.6M
  do {
1044
12.6M
    if (ls->t.token == /*{*/ '}') break;
1045
12.3M
    if (cc.v.k != VVOID)  /* is there a previous list item? */
1046
7.57M
      closelistfield(fs, &cc);  /* close it */
1047
12.3M
    field(ls, &cc);
1048
12.3M
    luaY_checklimit(fs, cc.tostore + cc.na + cc.nh, MAX_CNST,
1049
12.3M
                    "items in a constructor");
1050
12.3M
  } while (testnext(ls, ',') || testnext(ls, ';'));
1051
2.21M
  check_match(ls, /*{*/ '}', '{' /*}*/, line);
1052
2.21M
  lastlistfield(fs, &cc);
1053
2.21M
  luaK_settablesize(fs, pc, t->u.info, cc.na, cc.nh);
1054
2.21M
}
1055
1056
/* }====================================================================== */
1057
1058
1059
8.17M
static void setvararg (FuncState *fs) {
1060
8.17M
  fs->f->flag |= PF_VAHID;  /* by default, use hidden vararg arguments */
1061
8.17M
  luaK_codeABC(fs, OP_VARARGPREP, 0, 0, 0);
1062
8.17M
}
1063
1064
1065
1.72M
static void parlist (LexState *ls) {
1066
  /* parlist -> [ {NAME ','} (NAME | '...') ] */
1067
1.72M
  FuncState *fs = ls->fs;
1068
1.72M
  Proto *f = fs->f;
1069
1.72M
  int nparams = 0;
1070
1.72M
  int varargk = 0;
1071
1.72M
  if (ls->t.token != ')') {  /* is 'parlist' not empty? */
1072
1.89M
    do {
1073
1.89M
      switch (ls->t.token) {
1074
1.87M
        case TK_NAME: {
1075
1.87M
          new_localvar(ls, str_checkname(ls));
1076
1.87M
          nparams++;
1077
1.87M
          break;
1078
0
        }
1079
8.79k
        case TK_DOTS: {
1080
8.79k
          varargk = 1;
1081
8.79k
          luaX_next(ls);  /* skip '...' */
1082
8.79k
          if (ls->t.token == TK_NAME)
1083
2.25k
            new_varkind(ls, str_checkname(ls), RDKVAVAR);
1084
6.54k
          else
1085
6.54k
            new_localvarliteral(ls, "(vararg table)");
1086
8.79k
          break;
1087
0
        }
1088
7.21k
        default: luaX_syntaxerror(ls, "<name> or '...' expected");
1089
1.89M
      }
1090
1.89M
    } while (!varargk && testnext(ls, ','));
1091
925k
  }
1092
1.72M
  adjustlocalvars(ls, nparams);
1093
1.72M
  f->numparams = cast_byte(fs->nactvar);
1094
1.72M
  if (varargk) {
1095
8.79k
    setvararg(fs);  /* declared vararg */
1096
8.79k
    adjustlocalvars(ls, 1);  /* vararg parameter */
1097
8.79k
  }
1098
  /* reserve registers for parameters (plus vararg parameter, if present) */
1099
1.72M
  luaK_reserveregs(fs, fs->nactvar);
1100
1.72M
}
1101
1102
1103
1.74M
static void body (LexState *ls, expdesc *e, int ismethod, int line) {
1104
  /* body ->  '(' parlist ')' block END */
1105
1.74M
  FuncState new_fs;
1106
1.74M
  BlockCnt bl;
1107
1.74M
  new_fs.f = addprototype(ls);
1108
1.74M
  new_fs.f->linedefined = line;
1109
1.74M
  open_func(ls, &new_fs, &bl);
1110
1.74M
  checknext(ls, '(');
1111
1.74M
  if (ismethod) {
1112
7.61k
    new_localvarliteral(ls, "self");  /* create 'self' parameter */
1113
7.61k
    adjustlocalvars(ls, 1);
1114
7.61k
  }
1115
1.74M
  parlist(ls);
1116
1.74M
  checknext(ls, ')');
1117
1.74M
  statlist(ls);
1118
1.74M
  new_fs.f->lastlinedefined = ls->linenumber;
1119
1.74M
  check_match(ls, TK_END, TK_FUNCTION, line);
1120
1.74M
  codeclosure(ls, e);
1121
1.74M
  close_func(ls);
1122
1.74M
}
1123
1124
1125
6.60M
static int explist (LexState *ls, expdesc *v) {
1126
  /* explist -> expr { ',' expr } */
1127
6.60M
  int n = 1;  /* at least one expression */
1128
6.60M
  expr(ls, v);
1129
8.46M
  while (testnext(ls, ',')) {
1130
1.85M
    luaK_exp2nextreg(ls->fs, v);
1131
1.85M
    expr(ls, v);
1132
1.85M
    n++;
1133
1.85M
  }
1134
6.60M
  return n;
1135
6.60M
}
1136
1137
1138
4.78M
static void funcargs (LexState *ls, expdesc *f) {
1139
4.78M
  FuncState *fs = ls->fs;
1140
4.78M
  expdesc args;
1141
4.78M
  int base, nparams;
1142
4.78M
  int line = ls->linenumber;
1143
4.78M
  switch (ls->t.token) {
1144
2.94M
    case '(': {  /* funcargs -> '(' [ explist ] ')' */
1145
2.94M
      luaX_next(ls);
1146
2.94M
      if (ls->t.token == ')')  /* arg list is empty? */
1147
260k
        args.k = VVOID;
1148
2.68M
      else {
1149
2.68M
        explist(ls, &args);
1150
2.68M
        if (hasmultret(args.k))
1151
175k
          luaK_setmultret(fs, &args);
1152
2.68M
      }
1153
2.94M
      check_match(ls, ')', '(', line);
1154
2.94M
      break;
1155
0
    }
1156
846k
    case '{' /*}*/: {  /* funcargs -> constructor */
1157
846k
      constructor(ls, &args);
1158
846k
      break;
1159
0
    }
1160
982k
    case TK_STRING: {  /* funcargs -> STRING */
1161
982k
      codestring(&args, ls->t.seminfo.ts);
1162
982k
      luaX_next(ls);  /* must use 'seminfo' before 'next' */
1163
982k
      break;
1164
0
    }
1165
10.5k
    default: {
1166
10.5k
      luaX_syntaxerror(ls, "function arguments expected");
1167
0
    }
1168
4.78M
  }
1169
3.50M
  lua_assert(f->k == VNONRELOC);
1170
3.50M
  base = f->u.info;  /* base register for call */
1171
3.50M
  if (hasmultret(args.k))
1172
171k
    nparams = LUA_MULTRET;  /* open call */
1173
3.33M
  else {
1174
3.33M
    if (args.k != VVOID)
1175
3.07M
      luaK_exp2nextreg(fs, &args);  /* close last argument */
1176
3.33M
    nparams = fs->freereg - (base+1);
1177
3.33M
  }
1178
3.50M
  init_exp(f, VCALL, luaK_codeABC(fs, OP_CALL, base, nparams+1, 2));
1179
3.50M
  luaK_fixline(fs, line);
1180
  /* call removes function and arguments and leaves one result (unless
1181
     changed later) */
1182
3.50M
  fs->freereg = cast_byte(base + 1);
1183
3.50M
}
1184
1185
1186
1187
1188
/*
1189
** {======================================================================
1190
** Expression parsing
1191
** =======================================================================
1192
*/
1193
1194
1195
183M
static void primaryexp (LexState *ls, expdesc *v) {
1196
  /* primaryexp -> NAME | '(' expr ')' */
1197
183M
  switch (ls->t.token) {
1198
974k
    case '(': {
1199
974k
      int line = ls->linenumber;
1200
974k
      luaX_next(ls);
1201
974k
      expr(ls, v);
1202
974k
      check_match(ls, ')', '(', line);
1203
974k
      luaK_dischargevars(ls->fs, v);
1204
974k
      return;
1205
0
    }
1206
180M
    case TK_NAME: {
1207
180M
      singlevar(ls, v);
1208
180M
      return;
1209
0
    }
1210
2.75M
    default: {
1211
2.75M
      luaX_syntaxerror(ls, "unexpected symbol");
1212
0
    }
1213
183M
  }
1214
183M
}
1215
1216
1217
183M
static void suffixedexp (LexState *ls, expdesc *v) {
1218
  /* suffixedexp ->
1219
       primaryexp { '.' NAME | '[' exp ']' | ':' NAME funcargs | funcargs } */
1220
183M
  FuncState *fs = ls->fs;
1221
183M
  primaryexp(ls, v);
1222
191M
  for (;;) {
1223
191M
    switch (ls->t.token) {
1224
7.74M
      case '.': {  /* fieldsel */
1225
7.74M
        fieldsel(ls, v);
1226
7.74M
        break;
1227
0
      }
1228
398k
      case '[': {  /* '[' exp ']' */
1229
398k
        expdesc key;
1230
398k
        luaK_exp2anyregup(fs, v);
1231
398k
        yindex(ls, &key);
1232
398k
        luaK_indexed(fs, v, &key);
1233
398k
        break;
1234
0
      }
1235
281k
      case ':': {  /* ':' NAME funcargs */
1236
281k
        expdesc key;
1237
281k
        luaX_next(ls);
1238
281k
        codename(ls, &key);
1239
281k
        luaK_self(fs, v, &key);
1240
281k
        funcargs(ls, v);
1241
281k
        break;
1242
0
      }
1243
4.59M
      case '(': case TK_STRING: case '{' /*}*/: {  /* funcargs */
1244
4.59M
        luaK_exp2nextreg(fs, v);
1245
4.59M
        funcargs(ls, v);
1246
4.59M
        break;
1247
3.76M
      }
1248
178M
      default: return;
1249
191M
    }
1250
191M
  }
1251
183M
}
1252
1253
1254
189M
static void simpleexp (LexState *ls, expdesc *v) {
1255
  /* simpleexp -> FLT | INT | STRING | NIL | TRUE | FALSE | ... |
1256
                  constructor | FUNCTION body | suffixedexp */
1257
189M
  switch (ls->t.token) {
1258
1.95M
    case TK_FLT: {
1259
1.95M
      init_exp(v, VKFLT, 0);
1260
1.95M
      v->u.nval = ls->t.seminfo.r;
1261
1.95M
      break;
1262
0
    }
1263
11.0M
    case TK_INT: {
1264
11.0M
      init_exp(v, VKINT, 0);
1265
11.0M
      v->u.ival = ls->t.seminfo.i;
1266
11.0M
      break;
1267
0
    }
1268
937k
    case TK_STRING: {
1269
937k
      codestring(v, ls->t.seminfo.ts);
1270
937k
      break;
1271
0
    }
1272
282k
    case TK_NIL: {
1273
282k
      init_exp(v, VNIL, 0);
1274
282k
      break;
1275
0
    }
1276
217k
    case TK_TRUE: {
1277
217k
      init_exp(v, VTRUE, 0);
1278
217k
      break;
1279
0
    }
1280
54.1k
    case TK_FALSE: {
1281
54.1k
      init_exp(v, VFALSE, 0);
1282
54.1k
      break;
1283
0
    }
1284
222k
    case TK_DOTS: {  /* vararg */
1285
222k
      FuncState *fs = ls->fs;
1286
222k
      check_condition(ls, isvararg(fs->f),
1287
222k
                      "cannot use '...' outside a vararg function");
1288
218k
      init_exp(v, VVARARG, luaK_codeABC(fs, OP_VARARG, 0, fs->f->numparams, 1));
1289
218k
      break;
1290
222k
    }
1291
1.37M
    case '{' /*}*/: {  /* constructor */
1292
1.37M
      constructor(ls, v);
1293
1.37M
      return;
1294
222k
    }
1295
1.63M
    case TK_FUNCTION: {
1296
1.63M
      luaX_next(ls);
1297
1.63M
      body(ls, v, 0, ls->linenumber);
1298
1.63M
      return;
1299
222k
    }
1300
172M
    default: {
1301
172M
      suffixedexp(ls, v);
1302
172M
      return;
1303
222k
    }
1304
189M
  }
1305
14.7M
  luaX_next(ls);
1306
14.7M
}
1307
1308
1309
200M
static UnOpr getunopr (int op) {
1310
200M
  switch (op) {
1311
127k
    case TK_NOT: return OPR_NOT;
1312
1.95M
    case '-': return OPR_MINUS;
1313
8.83M
    case '~': return OPR_BNOT;
1314
214k
    case '#': return OPR_LEN;
1315
189M
    default: return OPR_NOUNOPR;
1316
200M
  }
1317
200M
}
1318
1319
1320
197M
static BinOpr getbinopr (int op) {
1321
197M
  switch (op) {
1322
410k
    case '+': return OPR_ADD;
1323
2.02M
    case '-': return OPR_SUB;
1324
5.61M
    case '*': return OPR_MUL;
1325
1.01M
    case '%': return OPR_MOD;
1326
3.61M
    case '^': return OPR_POW;
1327
8.00M
    case '/': return OPR_DIV;
1328
493k
    case TK_IDIV: return OPR_IDIV;
1329
141k
    case '&': return OPR_BAND;
1330
933k
    case '|': return OPR_BOR;
1331
5.13M
    case '~': return OPR_BXOR;
1332
699k
    case TK_SHL: return OPR_SHL;
1333
1.40M
    case TK_SHR: return OPR_SHR;
1334
2.28M
    case TK_CONCAT: return OPR_CONCAT;
1335
111k
    case TK_NE: return OPR_NE;
1336
434k
    case TK_EQ: return OPR_EQ;
1337
8.69M
    case '<': return OPR_LT;
1338
801k
    case TK_LE: return OPR_LE;
1339
134M
    case '>': return OPR_GT;
1340
44.9k
    case TK_GE: return OPR_GE;
1341
804k
    case TK_AND: return OPR_AND;
1342
710k
    case TK_OR: return OPR_OR;
1343
20.1M
    default: return OPR_NOBINOPR;
1344
197M
  }
1345
197M
}
1346
1347
1348
/*
1349
** Priority table for binary operators.
1350
*/
1351
static const struct {
1352
  lu_byte left;  /* left priority for each binary operator */
1353
  lu_byte right; /* right priority */
1354
} priority[] = {  /* ORDER OPR */
1355
   {10, 10}, {10, 10},           /* '+' '-' */
1356
   {11, 11}, {11, 11},           /* '*' '%' */
1357
   {14, 13},                  /* '^' (right associative) */
1358
   {11, 11}, {11, 11},           /* '/' '//' */
1359
   {6, 6}, {4, 4}, {5, 5},   /* '&' '|' '~' */
1360
   {7, 7}, {7, 7},           /* '<<' '>>' */
1361
   {9, 8},                   /* '..' (right associative) */
1362
   {3, 3}, {3, 3}, {3, 3},   /* ==, <, <= */
1363
   {3, 3}, {3, 3}, {3, 3},   /* ~=, >, >= */
1364
   {2, 2}, {1, 1}            /* and, or */
1365
};
1366
1367
11.1M
#define UNARY_PRIORITY  12  /* priority for unary operators */
1368
1369
1370
/*
1371
** subexpr -> (simpleexp | unop subexpr) { binop subexpr }
1372
** where 'binop' is any binary operator with a priority higher than 'limit'
1373
*/
1374
200M
static BinOpr subexpr (LexState *ls, expdesc *v, int limit) {
1375
200M
  BinOpr op;
1376
200M
  UnOpr uop;
1377
200M
  enterlevel(ls);
1378
200M
  uop = getunopr(ls->t.token);
1379
200M
  if (uop != OPR_NOUNOPR) {  /* prefix (unary) operator? */
1380
11.1M
    int line = ls->linenumber;
1381
11.1M
    luaX_next(ls);  /* skip operator */
1382
11.1M
    subexpr(ls, v, UNARY_PRIORITY);
1383
11.1M
    luaK_prefix(ls->fs, uop, v, line);
1384
11.1M
  }
1385
189M
  else simpleexp(ls, v);
1386
  /* expand while operators have priorities higher than 'limit' */
1387
200M
  op = getbinopr(ls->t.token);
1388
368M
  while (op != OPR_NOBINOPR && priority[op].left > limit) {
1389
167M
    expdesc v2;
1390
167M
    BinOpr nextop;
1391
167M
    int line = ls->linenumber;
1392
167M
    luaX_next(ls);  /* skip operator */
1393
167M
    luaK_infix(ls->fs, op, v);
1394
    /* read sub-expression with higher priority */
1395
167M
    nextop = subexpr(ls, &v2, priority[op].right);
1396
167M
    luaK_posfix(ls->fs, op, v, &v2, line);
1397
167M
    op = nextop;
1398
167M
  }
1399
200M
  leavelevel(ls);
1400
200M
  return op;  /* return first untreated operator */
1401
200M
}
1402
1403
1404
22.7M
static void expr (LexState *ls, expdesc *v) {
1405
22.7M
  subexpr(ls, v, 0);
1406
22.7M
}
1407
1408
/* }==================================================================== */
1409
1410
1411
1412
/*
1413
** {======================================================================
1414
** Rules for Statements
1415
** =======================================================================
1416
*/
1417
1418
1419
507k
static void block (LexState *ls) {
1420
  /* block -> statlist */
1421
507k
  FuncState *fs = ls->fs;
1422
507k
  BlockCnt bl;
1423
507k
  enterblock(fs, &bl, 0);
1424
507k
  statlist(ls);
1425
507k
  leaveblock(fs);
1426
507k
}
1427
1428
1429
/*
1430
** structure to chain all variables in the left-hand side of an
1431
** assignment
1432
*/
1433
struct LHS_assign {
1434
  struct LHS_assign *prev;
1435
  expdesc v;  /* variable (global, local, upvalue, or indexed) */
1436
};
1437
1438
1439
/*
1440
** check whether, in an assignment to an upvalue/local variable, the
1441
** upvalue/local variable is begin used in a previous assignment to a
1442
** table. If so, save original upvalue/local value in a safe place and
1443
** use this safe copy in the previous assignment.
1444
*/
1445
307k
static void check_conflict (LexState *ls, struct LHS_assign *lh, expdesc *v) {
1446
307k
  FuncState *fs = ls->fs;
1447
307k
  lu_byte extra = fs->freereg;  /* eventual position to save local variable */
1448
307k
  int conflict = 0;
1449
4.61M
  for (; lh; lh = lh->prev) {  /* check all previous assignments */
1450
4.30M
    if (vkisindexed(lh->v.k)) {  /* assignment to table field? */
1451
3.25M
      if (lh->v.k == VINDEXUP) {  /* is table an upvalue? */
1452
877k
        if (v->k == VUPVAL && lh->v.u.ind.t == v->u.info) {
1453
776k
          conflict = 1;  /* table is the upvalue being assigned now */
1454
776k
          lh->v.k = VINDEXSTR;
1455
776k
          lh->v.u.ind.t = extra;  /* assignment will use safe copy */
1456
776k
        }
1457
877k
      }
1458
2.37M
      else {  /* table is a register */
1459
2.37M
        if (v->k == VLOCAL && lh->v.u.ind.t == v->u.var.ridx) {
1460
3.82k
          conflict = 1;  /* table is the local being assigned now */
1461
3.82k
          lh->v.u.ind.t = extra;  /* assignment will use safe copy */
1462
3.82k
        }
1463
        /* is index the local being assigned? */
1464
2.37M
        if (lh->v.k == VINDEXED && v->k == VLOCAL &&
1465
10.5k
            lh->v.u.ind.idx == v->u.var.ridx) {
1466
636
          conflict = 1;
1467
636
          lh->v.u.ind.idx = extra;  /* previous assignment will use safe copy */
1468
636
        }
1469
2.37M
      }
1470
3.25M
    }
1471
4.30M
  }
1472
307k
  if (conflict) {
1473
    /* copy upvalue/local value to a temporary (in position 'extra') */
1474
181k
    if (v->k == VLOCAL)
1475
1.77k
      luaK_codeABC(fs, OP_MOVE, extra, v->u.var.ridx, 0);
1476
179k
    else
1477
179k
      luaK_codeABC(fs, OP_GETUPVAL, extra, v->u.info, 0);
1478
181k
    luaK_reserveregs(fs, 1);
1479
181k
  }
1480
307k
}
1481
1482
1483
/* Create code to store the "top" register in 'var' */
1484
2.20M
static void storevartop (FuncState *fs, expdesc *var) {
1485
2.20M
  expdesc e;
1486
2.20M
  init_exp(&e, VNONRELOC, fs->freereg - 1);
1487
2.20M
  luaK_storevar(fs, var, &e);  /* will also free the top register */
1488
2.20M
}
1489
1490
1491
/*
1492
** Parse and compile a multiple assignment. The first "variable"
1493
** (a 'suffixedexp') was already read by the caller.
1494
**
1495
** assignment -> suffixedexp restassign
1496
** restassign -> ',' suffixedexp restassign | '=' explist
1497
*/
1498
6.26M
static void restassign (LexState *ls, struct LHS_assign *lh, int nvars) {
1499
6.26M
  expdesc e;
1500
6.26M
  check_condition(ls, vkisvar(lh->v.k), "syntax error");
1501
6.09M
  check_readonly(ls, &lh->v);
1502
6.09M
  if (testnext(ls, ',')) {  /* restassign -> ',' suffixedexp restassign */
1503
3.46M
    struct LHS_assign nv;
1504
3.46M
    nv.prev = lh;
1505
3.46M
    suffixedexp(ls, &nv.v);
1506
3.46M
    if (!vkisindexed(nv.v.k))
1507
307k
      check_conflict(ls, lh, &nv.v);
1508
3.46M
    enterlevel(ls);  /* control recursion depth */
1509
3.46M
    restassign(ls, &nv, nvars+1);
1510
3.46M
    leavelevel(ls);
1511
3.46M
  }
1512
2.63M
  else {  /* restassign -> '=' explist */
1513
2.63M
    int nexps;
1514
2.63M
    checknext(ls, '=');
1515
2.63M
    nexps = explist(ls, &e);
1516
2.63M
    if (nexps != nvars)
1517
202k
      adjust_assign(ls, nvars, nexps, &e);
1518
2.43M
    else {
1519
2.43M
      luaK_setoneret(ls->fs, &e);  /* close last expression */
1520
2.43M
      luaK_storevar(ls->fs, &lh->v, &e);
1521
2.43M
      return;  /* avoid default */
1522
2.43M
    }
1523
2.63M
  }
1524
3.66M
  storevartop(ls->fs, &lh->v);  /* default assignment */
1525
3.66M
}
1526
1527
1528
416k
static int cond (LexState *ls) {
1529
  /* cond -> exp */
1530
416k
  expdesc v;
1531
416k
  expr(ls, &v);  /* read condition */
1532
416k
  if (v.k == VNIL) v.k = VFALSE;  /* 'falses' are all equal here */
1533
416k
  luaK_goiftrue(ls->fs, &v);
1534
416k
  return v.f;
1535
416k
}
1536
1537
1538
261k
static void gotostat (LexState *ls, int line) {
1539
261k
  TString *name = str_checkname(ls);  /* label's name */
1540
261k
  newgotoentry(ls, name, line);
1541
261k
}
1542
1543
1544
/*
1545
** Break statement. Semantically equivalent to "goto break".
1546
*/
1547
113k
static void breakstat (LexState *ls, int line) {
1548
113k
  BlockCnt *bl;  /* to look for an enclosing loop */
1549
386k
  for (bl = ls->fs->bl; bl != NULL; bl = bl->previous) {
1550
384k
    if (bl->isloop)  /* found one? */
1551
110k
      goto ok;
1552
384k
  }
1553
2.15k
  luaX_syntaxerror(ls, "break outside loop");
1554
110k
 ok:
1555
110k
  bl->isloop = 2;  /* signal that block has pending breaks */
1556
110k
  luaX_next(ls);  /* skip break */
1557
110k
  newgotoentry(ls, ls->brkn, line);
1558
110k
}
1559
1560
1561
/*
1562
** Check whether there is already a label with the given 'name' at
1563
** current function.
1564
*/
1565
75.3k
static void checkrepeated (LexState *ls, TString *name) {
1566
75.3k
  Labeldesc *lb = findlabel(ls, name, ls->fs->firstlabel);
1567
75.3k
  if (l_unlikely(lb != NULL))  /* already defined? */
1568
5.28k
    luaK_semerror(ls, "label '%s' already defined on line %d",
1569
5.28k
                      getstr(name), lb->line);  /* error */
1570
75.3k
}
1571
1572
1573
92.3k
static void labelstat (LexState *ls, TString *name, int line) {
1574
  /* label -> '::' NAME '::' */
1575
92.3k
  checknext(ls, TK_DBCOLON);  /* skip double colon */
1576
152k
  while (ls->t.token == ';' || ls->t.token == TK_DBCOLON)
1577
60.3k
    statement(ls);  /* skip other no-op statements */
1578
92.3k
  checkrepeated(ls, name);  /* check for repeated labels */
1579
92.3k
  createlabel(ls, name, line, block_follow(ls, 0));
1580
92.3k
}
1581
1582
1583
25.0k
static void whilestat (LexState *ls, int line) {
1584
  /* whilestat -> WHILE cond DO block END */
1585
25.0k
  FuncState *fs = ls->fs;
1586
25.0k
  int whileinit;
1587
25.0k
  int condexit;
1588
25.0k
  BlockCnt bl;
1589
25.0k
  luaX_next(ls);  /* skip WHILE */
1590
25.0k
  whileinit = luaK_getlabel(fs);
1591
25.0k
  condexit = cond(ls);
1592
25.0k
  enterblock(fs, &bl, 1);
1593
25.0k
  checknext(ls, TK_DO);
1594
25.0k
  block(ls);
1595
25.0k
  luaK_jumpto(fs, whileinit);
1596
25.0k
  check_match(ls, TK_END, TK_WHILE, line);
1597
25.0k
  leaveblock(fs);
1598
25.0k
  luaK_patchtohere(fs, condexit);  /* false conditions finish the loop */
1599
25.0k
}
1600
1601
1602
62.0k
static void repeatstat (LexState *ls, int line) {
1603
  /* repeatstat -> REPEAT block UNTIL cond */
1604
62.0k
  int condexit;
1605
62.0k
  FuncState *fs = ls->fs;
1606
62.0k
  int repeat_init = luaK_getlabel(fs);
1607
62.0k
  BlockCnt bl1, bl2;
1608
62.0k
  enterblock(fs, &bl1, 1);  /* loop block */
1609
62.0k
  enterblock(fs, &bl2, 0);  /* scope block */
1610
62.0k
  luaX_next(ls);  /* skip REPEAT */
1611
62.0k
  statlist(ls);
1612
62.0k
  check_match(ls, TK_UNTIL, TK_REPEAT, line);
1613
62.0k
  condexit = cond(ls);  /* read condition (inside scope block) */
1614
62.0k
  leaveblock(fs);  /* finish scope */
1615
62.0k
  if (bl2.upval) {  /* upvalues? */
1616
2.72k
    int exit = luaK_jump(fs);  /* normal exit must jump over fix */
1617
2.72k
    luaK_patchtohere(fs, condexit);  /* repetition must close upvalues */
1618
2.72k
    luaK_codeABC(fs, OP_CLOSE, reglevel(fs, bl2.nactvar), 0, 0);
1619
2.72k
    condexit = luaK_jump(fs);  /* repeat after closing upvalues */
1620
2.72k
    luaK_patchtohere(fs, exit);  /* normal exit comes to here */
1621
2.72k
  }
1622
62.0k
  luaK_patchlist(fs, condexit, repeat_init);  /* close the loop */
1623
62.0k
  leaveblock(fs);  /* finish loop */
1624
62.0k
}
1625
1626
1627
/*
1628
** Read an expression and generate code to put its results in next
1629
** stack slot.
1630
**
1631
*/
1632
72.8k
static void exp1 (LexState *ls) {
1633
72.8k
  expdesc e;
1634
72.8k
  expr(ls, &e);
1635
72.8k
  luaK_exp2nextreg(ls->fs, &e);
1636
72.8k
  lua_assert(e.k == VNONRELOC);
1637
72.8k
}
1638
1639
1640
/*
1641
** Fix for instruction at position 'pc' to jump to 'dest'.
1642
** (Jump addresses are relative in Lua). 'back' true means
1643
** a back jump.
1644
*/
1645
95.6k
static void fixforjump (FuncState *fs, int pc, int dest, int back) {
1646
95.6k
  Instruction *jmp = &fs->f->code[pc];
1647
95.6k
  int offset = dest - (pc + 1);
1648
95.6k
  if (back)
1649
47.7k
    offset = -offset;
1650
95.6k
  if (l_unlikely(offset > MAXARG_Bx))
1651
28
    luaX_syntaxerror(fs->ls, "control structure too long");
1652
95.5k
  SETARG_Bx(*jmp, offset);
1653
95.5k
}
1654
1655
1656
/*
1657
** Generate code for a 'for' loop.
1658
*/
1659
69.7k
static void forbody (LexState *ls, int base, int line, int nvars, int isgen) {
1660
  /* forbody -> DO block */
1661
69.7k
  static const OpCode forprep[2] = {OP_FORPREP, OP_TFORPREP};
1662
69.7k
  static const OpCode forloop[2] = {OP_FORLOOP, OP_TFORLOOP};
1663
69.7k
  BlockCnt bl;
1664
69.7k
  FuncState *fs = ls->fs;
1665
69.7k
  int prep, endfor;
1666
69.7k
  checknext(ls, TK_DO);
1667
69.7k
  prep = luaK_codeABx(fs, forprep[isgen], base, 0);
1668
69.7k
  fs->freereg--;  /* both 'forprep' remove one register from the stack */
1669
69.7k
  enterblock(fs, &bl, 0);  /* scope for declared variables */
1670
69.7k
  adjustlocalvars(ls, nvars);
1671
69.7k
  luaK_reserveregs(fs, nvars);
1672
69.7k
  block(ls);
1673
69.7k
  leaveblock(fs);  /* end of scope for declared variables */
1674
69.7k
  fixforjump(fs, prep, luaK_getlabel(fs), 0);
1675
69.7k
  if (isgen) {  /* generic for? */
1676
27.0k
    luaK_codeABC(fs, OP_TFORCALL, base, 0, nvars);
1677
27.0k
    luaK_fixline(fs, line);
1678
27.0k
  }
1679
69.7k
  endfor = luaK_codeABx(fs, forloop[isgen], base, 0);
1680
69.7k
  fixforjump(fs, endfor, prep + 1, 1);
1681
69.7k
  luaK_fixline(fs, line);
1682
69.7k
}
1683
1684
1685
33.4k
static void fornum (LexState *ls, TString *varname, int line) {
1686
  /* fornum -> NAME = exp,exp[,exp] forbody */
1687
33.4k
  FuncState *fs = ls->fs;
1688
33.4k
  int base = fs->freereg;
1689
33.4k
  new_localvarliteral(ls, "(for state)");
1690
33.4k
  new_localvarliteral(ls, "(for state)");
1691
33.4k
  new_varkind(ls, varname, RDKCONST);  /* control variable */
1692
33.4k
  checknext(ls, '=');
1693
33.4k
  exp1(ls);  /* initial value */
1694
33.4k
  checknext(ls, ',');
1695
33.4k
  exp1(ls);  /* limit */
1696
33.4k
  if (testnext(ls, ','))
1697
8.07k
    exp1(ls);  /* optional step */
1698
25.3k
  else {  /* default step = 1 */
1699
25.3k
    luaK_int(fs, fs->freereg, 1);
1700
25.3k
    luaK_reserveregs(fs, 1);
1701
25.3k
  }
1702
33.4k
  adjustlocalvars(ls, 2);  /* start scope for internal variables */
1703
33.4k
  forbody(ls, base, line, 1, 0);
1704
33.4k
}
1705
1706
1707
62.8k
static void forlist (LexState *ls, TString *indexname) {
1708
  /* forlist -> NAME {,NAME} IN explist forbody */
1709
62.8k
  FuncState *fs = ls->fs;
1710
62.8k
  expdesc e;
1711
62.8k
  int nvars = 4;  /* function, state, closing, control */
1712
62.8k
  int line;
1713
62.8k
  int base = fs->freereg;
1714
  /* create internal variables */
1715
62.8k
  new_localvarliteral(ls, "(for state)");  /* iterator function */
1716
62.8k
  new_localvarliteral(ls, "(for state)");  /* state */
1717
62.8k
  new_localvarliteral(ls, "(for state)");  /* closing var. (after swap) */
1718
62.8k
  new_varkind(ls, indexname, RDKCONST);  /* control variable */
1719
  /* other declared variables */
1720
118k
  while (testnext(ls, ',')) {
1721
55.7k
    new_localvar(ls, str_checkname(ls));
1722
55.7k
    nvars++;
1723
55.7k
  }
1724
62.8k
  checknext(ls, TK_IN);
1725
62.8k
  line = ls->linenumber;
1726
62.8k
  adjust_assign(ls, 4, explist(ls, &e), &e);
1727
62.8k
  adjustlocalvars(ls, 3);  /* start scope for internal variables */
1728
62.8k
  marktobeclosed(fs);  /* last internal var. must be closed */
1729
62.8k
  luaK_checkstack(fs, 2);  /* extra space to call iterator */
1730
62.8k
  forbody(ls, base, line, nvars - 3, 1);
1731
62.8k
}
1732
1733
1734
135k
static void forstat (LexState *ls, int line) {
1735
  /* forstat -> FOR (fornum | forlist) END */
1736
135k
  FuncState *fs = ls->fs;
1737
135k
  TString *varname;
1738
135k
  BlockCnt bl;
1739
135k
  enterblock(fs, &bl, 1);  /* scope for loop and control variables */
1740
135k
  luaX_next(ls);  /* skip 'for' */
1741
135k
  varname = str_checkname(ls);  /* first variable name */
1742
135k
  switch (ls->t.token) {
1743
33.4k
    case '=': fornum(ls, varname, line); break;
1744
62.8k
    case ',': case TK_IN: forlist(ls, varname); break;
1745
8.49k
    default: luaX_syntaxerror(ls, "'=' or 'in' expected");
1746
135k
  }
1747
47.7k
  check_match(ls, TK_END, TK_FOR, line);
1748
47.7k
  leaveblock(fs);  /* loop scope ('break' jumps to this point) */
1749
47.7k
}
1750
1751
1752
331k
static void test_then_block (LexState *ls, int *escapelist) {
1753
  /* test_then_block -> [IF | ELSEIF] cond THEN block */
1754
331k
  FuncState *fs = ls->fs;
1755
331k
  int condtrue;
1756
331k
  luaX_next(ls);  /* skip IF or ELSEIF */
1757
331k
  condtrue = cond(ls);  /* read condition */
1758
331k
  checknext(ls, TK_THEN);
1759
331k
  block(ls);  /* 'then' part */
1760
331k
  if (ls->t.token == TK_ELSE ||
1761
281k
      ls->t.token == TK_ELSEIF)  /* followed by 'else'/'elseif'? */
1762
86.9k
    luaK_concat(fs, escapelist, luaK_jump(fs));  /* must jump over it */
1763
331k
  luaK_patchtohere(fs, condtrue);
1764
331k
}
1765
1766
1767
286k
static void ifstat (LexState *ls, int line) {
1768
  /* ifstat -> IF cond THEN block {ELSEIF cond THEN block} [ELSE block] END */
1769
286k
  FuncState *fs = ls->fs;
1770
286k
  int escapelist = NO_JUMP;  /* exit list for finished parts */
1771
286k
  test_then_block(ls, &escapelist);  /* IF cond THEN block */
1772
331k
  while (ls->t.token == TK_ELSEIF)
1773
45.1k
    test_then_block(ls, &escapelist);  /* ELSEIF cond THEN block */
1774
286k
  if (testnext(ls, TK_ELSE))
1775
41.7k
    block(ls);  /* 'else' part */
1776
286k
  check_match(ls, TK_END, TK_IF, line);
1777
286k
  luaK_patchtohere(fs, escapelist);  /* patch escape list to 'if' end */
1778
286k
}
1779
1780
1781
43.6k
static void localfunc (LexState *ls) {
1782
43.6k
  expdesc b;
1783
43.6k
  FuncState *fs = ls->fs;
1784
43.6k
  int fvar = fs->nactvar;  /* function's variable index */
1785
43.6k
  new_localvar(ls, str_checkname(ls));  /* new local variable */
1786
43.6k
  adjustlocalvars(ls, 1);  /* enter its scope */
1787
43.6k
  body(ls, &b, 0, ls->linenumber);  /* function created in next register */
1788
  /* debug information will only see the variable after this point! */
1789
43.6k
  localdebuginfo(fs, fvar)->startpc = fs->pc;
1790
43.6k
}
1791
1792
1793
4.04M
static lu_byte getvarattribute (LexState *ls, lu_byte df) {
1794
  /* attrib -> ['<' NAME '>'] */
1795
4.04M
  if (testnext(ls, '<')) {
1796
37.1k
    TString *ts = str_checkname(ls);
1797
37.1k
    const char *attr = getstr(ts);
1798
37.1k
    checknext(ls, '>');
1799
37.1k
    if (strcmp(attr, "const") == 0)
1800
20.8k
      return RDKCONST;  /* read-only variable */
1801
16.3k
    else if (strcmp(attr, "close") == 0)
1802
5.01k
      return RDKTOCLOSE;  /* to-be-closed variable */
1803
11.3k
    else
1804
11.3k
      luaK_semerror(ls, "unknown attribute '%s'", attr);
1805
37.1k
  }
1806
4.00M
  return df;  /* return default value */
1807
4.04M
}
1808
1809
1810
778k
static void checktoclose (FuncState *fs, int level) {
1811
778k
  if (level != -1) {  /* is there a to-be-closed variable? */
1812
4.70k
    marktobeclosed(fs);
1813
4.70k
    luaK_codeABC(fs, OP_TBC, reglevel(fs, level), 0, 0);
1814
4.70k
  }
1815
778k
}
1816
1817
1818
798k
static void localstat (LexState *ls) {
1819
  /* stat -> LOCAL NAME attrib { ',' NAME attrib } ['=' explist] */
1820
798k
  FuncState *fs = ls->fs;
1821
798k
  int toclose = -1;  /* index of to-be-closed variable (if any) */
1822
798k
  Vardesc *var;  /* last variable */
1823
798k
  int vidx;  /* index of last variable */
1824
798k
  int nvars = 0;
1825
798k
  int nexps;
1826
798k
  expdesc e;
1827
  /* get prefixed attribute (if any); default is regular local variable */
1828
798k
  lu_byte defkind = getvarattribute(ls, VDKREG);
1829
2.14M
  do {  /* for each variable */
1830
2.14M
    TString *vname = str_checkname(ls);  /* get its name */
1831
2.14M
    lu_byte kind = getvarattribute(ls, defkind);  /* postfixed attribute */
1832
2.14M
    vidx = new_varkind(ls, vname, kind);  /* predeclare it */
1833
2.14M
    if (kind == RDKTOCLOSE) {  /* to-be-closed? */
1834
4.96k
      if (toclose != -1)  /* one already present? */
1835
132
        luaK_semerror(ls, "multiple to-be-closed variables in local list");
1836
4.83k
      toclose = fs->nactvar + nvars;
1837
4.83k
    }
1838
2.14M
    nvars++;
1839
2.14M
  } while (testnext(ls, ','));
1840
798k
  if (testnext(ls, '='))  /* initialization? */
1841
707k
    nexps = explist(ls, &e);
1842
91.2k
  else {
1843
91.2k
    e.k = VVOID;
1844
91.2k
    nexps = 0;
1845
91.2k
  }
1846
798k
  var = getlocalvardesc(fs, vidx);  /* retrieve last variable */
1847
798k
  if (nvars == nexps &&  /* no adjustments? */
1848
580k
      var->vd.kind == RDKCONST &&  /* last variable is const? */
1849
16.9k
      luaK_exp2const(fs, &e, &var->k)) {  /* compile-time constant? */
1850
14.5k
    var->vd.kind = RDKCTC;  /* variable is a compile-time constant */
1851
14.5k
    adjustlocalvars(ls, nvars - 1);  /* exclude last variable */
1852
14.5k
    fs->nactvar++;  /* but count it */
1853
14.5k
  }
1854
783k
  else {
1855
783k
    adjust_assign(ls, nvars, nexps, &e);
1856
783k
    adjustlocalvars(ls, nvars);
1857
783k
  }
1858
798k
  checktoclose(fs, toclose);
1859
798k
}
1860
1861
1862
1.11M
static lu_byte getglobalattribute (LexState *ls, lu_byte df) {
1863
1.11M
  lu_byte kind = getvarattribute(ls, df);
1864
1.11M
  switch (kind) {
1865
4
    case RDKTOCLOSE:
1866
4
      luaK_semerror(ls, "global variables cannot be to-be-closed");
1867
0
      return kind;  /* to avoid warnings */
1868
2.27k
    case RDKCONST:
1869
2.27k
      return GDKCONST;  /* adjust kind for global variable */
1870
1.10M
    default:
1871
1.10M
      return kind;
1872
1.11M
  }
1873
1.11M
}
1874
1875
1876
117k
static void checkglobal (LexState *ls, TString *varname, int line) {
1877
117k
  FuncState *fs = ls->fs;
1878
117k
  expdesc var;
1879
117k
  int k;
1880
117k
  buildglobal(ls, varname, &var);  /* create global variable in 'var' */
1881
117k
  k = var.u.ind.keystr;  /* index of global name in 'k' */
1882
117k
  luaK_codecheckglobal(fs, &var, k, line);
1883
117k
}
1884
1885
1886
/*
1887
** Recursively traverse list of globals to be initalized. When
1888
** going, generate table description for the global. In the end,
1889
** after all indices have been generated, read list of initializing
1890
** expressions. When returning, generate the assignment of the value on
1891
** the stack to the corresponding table description. 'n' is the variable
1892
** being handled, range [0, nvars - 1].
1893
*/
1894
static void initglobal (LexState *ls, int nvars, int firstidx, int n,
1895
130k
                        int line) {
1896
130k
  if (n == nvars) {  /* traversed all variables? */
1897
6.55k
    expdesc e;
1898
6.55k
    int nexps = explist(ls, &e);  /* read list of expressions */
1899
6.55k
    adjust_assign(ls, nvars, nexps, &e);
1900
6.55k
  }
1901
124k
  else {  /* handle variable 'n' */
1902
124k
    FuncState *fs = ls->fs;
1903
124k
    expdesc var;
1904
124k
    TString *varname = getlocalvardesc(fs, firstidx + n)->vd.name;
1905
124k
    buildglobal(ls, varname, &var);  /* create global variable in 'var' */
1906
124k
    enterlevel(ls);  /* control recursion depth */
1907
124k
    initglobal(ls, nvars, firstidx, n + 1, line);
1908
124k
    leavelevel(ls);
1909
124k
    checkglobal(ls, varname, line);
1910
124k
    storevartop(fs, &var);
1911
124k
  }
1912
130k
}
1913
1914
1915
23.7k
static void globalnames (LexState *ls, lu_byte defkind) {
1916
23.7k
  FuncState *fs = ls->fs;
1917
23.7k
  int nvars = 0;
1918
23.7k
  int lastidx;  /* index of last registered variable */
1919
1.08M
  do {  /* for each name */
1920
1.08M
    TString *vname = str_checkname(ls);
1921
1.08M
    lu_byte kind = getglobalattribute(ls, defkind);
1922
1.08M
    lastidx = new_varkind(ls, vname, kind);
1923
1.08M
    nvars++;
1924
1.08M
  } while (testnext(ls, ','));
1925
23.7k
  if (testnext(ls, '='))  /* initialization? */
1926
6.56k
    initglobal(ls, nvars, lastidx - nvars + 1, 0, ls->linenumber);
1927
23.7k
  fs->nactvar = cast_short(fs->nactvar + nvars);  /* activate declaration */
1928
23.7k
}
1929
1930
1931
30.5k
static void globalstat (LexState *ls) {
1932
  /* globalstat -> (GLOBAL) attrib '*'
1933
     globalstat -> (GLOBAL) attrib NAME attrib {',' NAME attrib} */
1934
30.5k
  FuncState *fs = ls->fs;
1935
  /* get prefixed attribute (if any); default is regular global variable */
1936
30.5k
  lu_byte defkind = getglobalattribute(ls, GDKREG);
1937
30.5k
  if (!testnext(ls, '*'))
1938
23.7k
    globalnames(ls, defkind);
1939
6.85k
  else {
1940
    /* use NULL as name to represent '*' entries */
1941
6.85k
    new_varkind(ls, NULL, defkind);
1942
6.85k
    fs->nactvar++;  /* activate declaration */
1943
6.85k
  }
1944
30.5k
}
1945
1946
1947
3.35k
static void globalfunc (LexState *ls, int line) {
1948
  /* globalfunc -> (GLOBAL FUNCTION) NAME body */
1949
3.35k
  expdesc var, b;
1950
3.35k
  FuncState *fs = ls->fs;
1951
3.35k
  TString *fname = str_checkname(ls);
1952
3.35k
  new_varkind(ls, fname, GDKREG);  /* declare global variable */
1953
3.35k
  fs->nactvar++;  /* enter its scope */
1954
3.35k
  buildglobal(ls, fname, &var);
1955
3.35k
  body(ls, &b, 0, ls->linenumber);  /* compile and return closure in 'b' */
1956
3.35k
  checkglobal(ls, fname, line);
1957
3.35k
  luaK_storevar(fs, &var, &b);
1958
3.35k
  luaK_fixline(fs, line);  /* definition "happens" in the first line */
1959
3.35k
}
1960
1961
1962
33.9k
static void globalstatfunc (LexState *ls, int line) {
1963
  /* stat -> GLOBAL globalfunc | GLOBAL globalstat */
1964
33.9k
  luaX_next(ls);  /* skip 'global' */
1965
33.9k
  if (testnext(ls, TK_FUNCTION))
1966
3.35k
    globalfunc(ls, line);
1967
30.5k
  else
1968
30.5k
    globalstat(ls);
1969
33.9k
}
1970
1971
1972
86.3k
static int funcname (LexState *ls, expdesc *v) {
1973
  /* funcname -> NAME {fieldsel} [':' NAME] */
1974
86.3k
  int ismethod = 0;
1975
86.3k
  singlevar(ls, v);
1976
114k
  while (ls->t.token == '.')
1977
27.7k
    fieldsel(ls, v);
1978
86.3k
  if (ls->t.token == ':') {
1979
9.08k
    ismethod = 1;
1980
9.08k
    fieldsel(ls, v);
1981
9.08k
  }
1982
86.3k
  return ismethod;
1983
86.3k
}
1984
1985
1986
86.3k
static void funcstat (LexState *ls, int line) {
1987
  /* funcstat -> FUNCTION funcname body */
1988
86.3k
  int ismethod;
1989
86.3k
  expdesc v, b;
1990
86.3k
  luaX_next(ls);  /* skip FUNCTION */
1991
86.3k
  ismethod = funcname(ls, &v);
1992
86.3k
  check_readonly(ls, &v);
1993
86.3k
  body(ls, &b, ismethod, line);
1994
86.3k
  luaK_storevar(ls->fs, &v, &b);
1995
86.3k
  luaK_fixline(ls->fs, line);  /* definition "happens" in the first line */
1996
86.3k
}
1997
1998
1999
8.29M
static void exprstat (LexState *ls) {
2000
  /* stat -> func | assignment */
2001
8.29M
  FuncState *fs = ls->fs;
2002
8.29M
  struct LHS_assign v;
2003
8.29M
  suffixedexp(ls, &v.v);
2004
8.29M
  if (ls->t.token == '=' || ls->t.token == ',') { /* stat -> assignment ? */
2005
2.99M
    v.prev = NULL;
2006
2.99M
    restassign(ls, &v, 1);
2007
2.99M
  }
2008
5.30M
  else {  /* stat -> func */
2009
5.30M
    Instruction *inst;
2010
5.30M
    check_condition(ls, v.v.k == VCALL, "syntax error");
2011
4.53M
    inst = &getinstruction(fs, &v.v);
2012
4.53M
    SETARG_C(*inst, 1);  /* call statement uses no results */
2013
4.53M
  }
2014
8.29M
}
2015
2016
2017
721k
static void retstat (LexState *ls) {
2018
  /* stat -> RETURN [explist] [';'] */
2019
721k
  FuncState *fs = ls->fs;
2020
721k
  expdesc e;
2021
721k
  int nret;  /* number of values being returned */
2022
721k
  int first = luaY_nvarstack(fs);  /* first slot to be returned */
2023
721k
  if (block_follow(ls, 1) || ls->t.token == ';')
2024
69.1k
    nret = 0;  /* return no values */
2025
651k
  else {
2026
651k
    nret = explist(ls, &e);  /* optional return values */
2027
651k
    if (hasmultret(e.k)) {
2028
88.9k
      luaK_setmultret(fs, &e);
2029
88.9k
      if (e.k == VCALL && nret == 1 && !fs->bl->insidetbc) {  /* tail call? */
2030
73.4k
        SET_OPCODE(getinstruction(fs,&e), OP_TAILCALL);
2031
73.4k
        lua_assert(GETARG_A(getinstruction(fs,&e)) == luaY_nvarstack(fs));
2032
73.4k
      }
2033
88.9k
      nret = LUA_MULTRET;  /* return all values */
2034
88.9k
    }
2035
562k
    else {
2036
562k
      if (nret == 1)  /* only one single value? */
2037
549k
        first = luaK_exp2anyreg(fs, &e);  /* can use original slot */
2038
13.0k
      else {  /* values must go to the top of the stack */
2039
13.0k
        luaK_exp2nextreg(fs, &e);
2040
13.0k
        lua_assert(nret == fs->freereg - first);
2041
13.0k
      }
2042
562k
    }
2043
651k
  }
2044
721k
  luaK_ret(fs, first, nret);
2045
714k
  testnext(ls, ';');  /* skip optional semicolon */
2046
714k
}
2047
2048
2049
12.1M
static void statement (LexState *ls) {
2050
12.1M
  int line = ls->linenumber;  /* may be needed for error messages */
2051
12.1M
  enterlevel(ls);
2052
12.1M
  switch (ls->t.token) {
2053
1.08M
    case ';': {  /* stat -> ';' (empty statement) */
2054
1.08M
      luaX_next(ls);  /* skip ';' */
2055
1.08M
      break;
2056
0
    }
2057
286k
    case TK_IF: {  /* stat -> ifstat */
2058
286k
      ifstat(ls, line);
2059
286k
      break;
2060
0
    }
2061
25.0k
    case TK_WHILE: {  /* stat -> whilestat */
2062
25.0k
      whilestat(ls, line);
2063
25.0k
      break;
2064
0
    }
2065
61.3k
    case TK_DO: {  /* stat -> DO block END */
2066
61.3k
      luaX_next(ls);  /* skip DO */
2067
61.3k
      block(ls);
2068
61.3k
      check_match(ls, TK_END, TK_DO, line);
2069
61.3k
      break;
2070
0
    }
2071
135k
    case TK_FOR: {  /* stat -> forstat */
2072
135k
      forstat(ls, line);
2073
135k
      break;
2074
0
    }
2075
62.0k
    case TK_REPEAT: {  /* stat -> repeatstat */
2076
62.0k
      repeatstat(ls, line);
2077
62.0k
      break;
2078
0
    }
2079
86.3k
    case TK_FUNCTION: {  /* stat -> funcstat */
2080
86.3k
      funcstat(ls, line);
2081
86.3k
      break;
2082
0
    }
2083
842k
    case TK_LOCAL: {  /* stat -> localstat */
2084
842k
      luaX_next(ls);  /* skip LOCAL */
2085
842k
      if (testnext(ls, TK_FUNCTION))  /* local function? */
2086
43.6k
        localfunc(ls);
2087
798k
      else
2088
798k
        localstat(ls);
2089
842k
      break;
2090
0
    }
2091
0
    case TK_GLOBAL: {  /* stat -> globalstatfunc */
2092
0
      globalstatfunc(ls, line);
2093
0
      break;
2094
0
    }
2095
100k
    case TK_DBCOLON: {  /* stat -> label */
2096
100k
      luaX_next(ls);  /* skip double colon */
2097
100k
      labelstat(ls, str_checkname(ls), line);
2098
100k
      break;
2099
0
    }
2100
721k
    case TK_RETURN: {  /* stat -> retstat */
2101
721k
      luaX_next(ls);  /* skip RETURN */
2102
721k
      retstat(ls);
2103
721k
      break;
2104
0
    }
2105
113k
    case TK_BREAK: {  /* stat -> breakstat */
2106
113k
      breakstat(ls, line);
2107
113k
      break;
2108
0
    }
2109
264k
    case TK_GOTO: {  /* stat -> 'goto' NAME */
2110
264k
      luaX_next(ls);  /* skip 'goto' */
2111
264k
      gotostat(ls, line);
2112
264k
      break;
2113
0
    }
2114
0
#if defined(LUA_COMPAT_GLOBAL)
2115
5.71M
    case TK_NAME: {
2116
      /* compatibility code to parse global keyword when "global"
2117
         is not reserved */
2118
5.71M
      if (ls->t.seminfo.ts == ls->glbn) {  /* current = "global"? */
2119
39.7k
        int lk = luaX_lookahead(ls);
2120
39.7k
        if (lk == '<' || lk == TK_NAME || lk == '*' || lk == TK_FUNCTION) {
2121
          /* 'global <attrib>' or 'global name' or 'global *' or
2122
             'global function' */
2123
33.9k
          globalstatfunc(ls, line);
2124
33.9k
          break;
2125
33.9k
        }
2126
39.7k
      }  /* else... */
2127
5.71M
    }
2128
5.68M
#endif
2129
    /* FALLTHROUGH */
2130
8.29M
    default: {  /* stat -> func | assignment */
2131
8.29M
      exprstat(ls);
2132
8.29M
      break;
2133
5.71M
    }
2134
12.1M
  }
2135
6.47M
  lua_assert(ls->fs->f->maxstacksize >= ls->fs->freereg &&
2136
6.47M
             ls->fs->freereg >= luaY_nvarstack(ls->fs));
2137
6.47M
  ls->fs->freereg = luaY_nvarstack(ls->fs);  /* free registers */
2138
6.47M
  leavelevel(ls);
2139
6.47M
}
2140
2141
/* }====================================================================== */
2142
2143
/* }====================================================================== */
2144
2145
2146
/*
2147
** compiles the main function, which is a regular vararg function with an
2148
** upvalue named LUA_ENV
2149
*/
2150
8.16M
static void mainfunc (LexState *ls, FuncState *fs) {
2151
8.16M
  BlockCnt bl;
2152
8.16M
  Upvaldesc *env;
2153
8.16M
  open_func(ls, fs, &bl);
2154
8.16M
  setvararg(fs);  /* main function is always vararg */
2155
8.16M
  env = allocupvalue(fs);  /* ...set environment upvalue */
2156
8.16M
  env->instack = 1;
2157
8.16M
  env->idx = 0;
2158
8.16M
  env->kind = VDKREG;
2159
8.16M
  env->name = ls->envn;
2160
8.16M
  luaC_objbarrier(ls->L, fs->f, env->name);
2161
8.16M
  luaX_next(ls);  /* read first token */
2162
8.16M
  statlist(ls);  /* parse main body */
2163
8.16M
  check(ls, TK_EOS);
2164
8.16M
  close_func(ls);
2165
8.16M
}
2166
2167
2168
LClosure *luaY_parser (lua_State *L, ZIO *z, Mbuffer *buff,
2169
8.16M
                       Dyndata *dyd, const char *name, int firstchar) {
2170
8.16M
  LexState lexstate;
2171
8.16M
  FuncState funcstate;
2172
8.16M
  LClosure *cl = luaF_newLclosure(L, 1);  /* create main closure */
2173
8.16M
  setclLvalue2s(L, L->top.p, cl);  /* anchor it (to avoid being collected) */
2174
8.16M
  luaD_inctop(L);
2175
8.16M
  lexstate.h = luaH_new(L);  /* create table for scanner */
2176
8.16M
  sethvalue2s(L, L->top.p, lexstate.h);  /* anchor it */
2177
8.16M
  luaD_inctop(L);
2178
8.16M
  funcstate.f = cl->p = luaF_newproto(L);
2179
8.16M
  luaC_objbarrier(L, cl, cl->p);
2180
8.16M
  funcstate.f->source = luaS_new(L, name);  /* create and anchor TString */
2181
8.16M
  luaC_objbarrier(L, funcstate.f, funcstate.f->source);
2182
8.16M
  lexstate.buff = buff;
2183
8.16M
  lexstate.dyd = dyd;
2184
8.16M
  dyd->actvar.n = dyd->gt.n = dyd->label.n = 0;
2185
8.16M
  luaX_setinput(L, &lexstate, z, funcstate.f->source, firstchar);
2186
8.16M
  mainfunc(&lexstate, &funcstate);
2187
8.16M
  lua_assert(!funcstate.prev && funcstate.nups == 1 && !lexstate.fs);
2188
  /* all scopes should be correctly finished */
2189
8.16M
  lua_assert(dyd->actvar.n == 0 && dyd->gt.n == 0 && dyd->label.n == 0);
2190
2.98M
  L->top.p--;  /* remove scanner's table */
2191
2.98M
  return cl;  /* closure is on the stack, too */
2192
2.98M
}
2193