Coverage Report

Created: 2026-01-16 06:23

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
2.81M
#define MAXVARS   200
36
37
38
5.39M
#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.72G
#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
729k
static l_noret error_expected (LexState *ls, int token) {
69
729k
  luaX_syntaxerror(ls,
70
729k
      luaO_pushfstring(ls->L, "%s expected", luaX_token2str(ls, token)));
71
729k
}
72
73
74
1.36k
static l_noret errorlimit (FuncState *fs, int limit, const char *what) {
75
1.36k
  lua_State *L = fs->ls->L;
76
1.36k
  const char *msg;
77
1.36k
  int line = fs->f->linedefined;
78
1.36k
  const char *where = (line == 0)
79
1.36k
                      ? "main function"
80
1.36k
                      : luaO_pushfstring(L, "function at line %d", line);
81
1.36k
  msg = luaO_pushfstring(L, "too many %s (limit is %d) in %s",
82
1.36k
                             what, limit, where);
83
1.36k
  luaX_syntaxerror(fs->ls, msg);
84
1.36k
}
85
86
87
24.2M
void luaY_checklimit (FuncState *fs, int v, int l, const char *what) {
88
24.2M
  if (l_unlikely(v > l)) errorlimit(fs, l, what);
89
24.2M
}
90
91
92
/*
93
** Test whether next token is 'c'; if so, skip it.
94
*/
95
31.8M
static int testnext (LexState *ls, int c) {
96
31.8M
  if (ls->t.token == c) {
97
18.7M
    luaX_next(ls);
98
18.7M
    return 1;
99
18.7M
  }
100
13.1M
  else return 0;
101
31.8M
}
102
103
104
/*
105
** Check that next token is 'c'.
106
*/
107
197M
static void check (LexState *ls, int c) {
108
197M
  if (ls->t.token != c)
109
422k
    error_expected(ls, c);
110
197M
}
111
112
113
/*
114
** Check that next token is 'c' and skip it.
115
*/
116
8.47M
static void checknext (LexState *ls, int c) {
117
8.47M
  check(ls, c);
118
8.47M
  luaX_next(ls);
119
8.47M
}
120
121
122
9.85M
#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
3.58M
static void check_match (LexState *ls, int what, int who, int where) {
131
3.58M
  if (l_unlikely(!testnext(ls, what))) {
132
332k
    if (where == ls->linenumber)  /* all in the same line? */
133
306k
      error_expected(ls, what);  /* do not need a complex message */
134
25.6k
    else {
135
25.6k
      luaX_syntaxerror(ls, luaO_pushfstring(ls->L,
136
25.6k
             "%s expected (to close %s at line %d)",
137
25.6k
              luaX_token2str(ls, what), luaX_token2str(ls, who), where));
138
25.6k
    }
139
332k
  }
140
3.58M
}
141
142
143
187M
static TString *str_checkname (LexState *ls) {
144
187M
  TString *ts;
145
187M
  check(ls, TK_NAME);
146
187M
  ts = ls->t.seminfo.ts;
147
187M
  luaX_next(ls);
148
187M
  return ts;
149
187M
}
150
151
152
518M
static void init_exp (expdesc *e, expkind k, int i) {
153
518M
  e->f = e->t = NO_JUMP;
154
518M
  e->k = k;
155
518M
  e->u.info = i;
156
518M
}
157
158
159
157M
static void codestring (expdesc *e, TString *s) {
160
157M
  e->f = e->t = NO_JUMP;
161
157M
  e->k = VKSTR;
162
157M
  e->u.strval = s;
163
157M
}
164
165
166
6.85M
static void codename (LexState *ls, expdesc *e) {
167
6.85M
  codestring(e, str_checkname(ls));
168
6.85M
}
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
2.81M
                               TString *varname) {
177
2.81M
  Proto *f = fs->f;
178
2.81M
  int oldsize = f->sizelocvars;
179
2.81M
  luaM_growvector(ls->L, f->locvars, fs->ndebugvars, f->sizelocvars,
180
2.81M
                  LocVar, SHRT_MAX, "local variables");
181
8.44M
  while (oldsize < f->sizelocvars)
182
5.63M
    f->locvars[oldsize++].varname = NULL;
183
2.81M
  f->locvars[fs->ndebugvars].varname = varname;
184
2.81M
  f->locvars[fs->ndebugvars].startpc = fs->pc;
185
2.81M
  luaC_objbarrier(ls->L, f, varname);
186
2.81M
  return fs->ndebugvars++;
187
2.81M
}
188
189
190
/*
191
** Create a new variable with the given 'name' and given 'kind'.
192
** Return its index in the function.
193
*/
194
4.40M
static int new_varkind (LexState *ls, TString *name, lu_byte kind) {
195
4.40M
  lua_State *L = ls->L;
196
4.40M
  FuncState *fs = ls->fs;
197
4.40M
  Dyndata *dyd = ls->dyd;
198
4.40M
  Vardesc *var;
199
4.40M
  luaM_growvector(L, dyd->actvar.arr, dyd->actvar.n + 1,
200
4.40M
             dyd->actvar.size, Vardesc, SHRT_MAX, "variable declarations");
201
4.40M
  var = &dyd->actvar.arr[dyd->actvar.n++];
202
4.40M
  var->vd.kind = kind;  /* default */
203
4.40M
  var->vd.name = name;
204
4.40M
  return dyd->actvar.n - 1 - fs->firstlocal;
205
4.40M
}
206
207
208
/*
209
** Create a new local variable with the given 'name' and regular kind.
210
*/
211
1.37M
static int new_localvar (LexState *ls, TString *name) {
212
1.37M
  return new_varkind(ls, name, VDKREG);
213
1.37M
}
214
215
#define new_localvarliteral(ls,v) \
216
10.6k
    new_localvar(ls,  \
217
10.6k
      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
1.68G
static Vardesc *getlocalvardesc (FuncState *fs, int vidx) {
227
1.68G
  return &fs->ls->dyd->actvar.arr[fs->firstlocal + vidx];
228
1.68G
}
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
475M
static lu_byte reglevel (FuncState *fs, int nvar) {
237
724M
  while (nvar-- > 0) {
238
459M
    Vardesc *vd = getlocalvardesc(fs, nvar);  /* get previous variable */
239
459M
    if (varinreg(vd))  /* is in a register? */
240
210M
      return cast_byte(vd->vd.ridx + 1);
241
459M
  }
242
265M
  return 0;  /* no variables in registers */
243
475M
}
244
245
246
/*
247
** Return the number of variables in the register stack for the given
248
** function.
249
*/
250
469M
lu_byte luaY_nvarstack (FuncState *fs) {
251
469M
  return reglevel(fs, fs->nactvar);
252
469M
}
253
254
255
/*
256
** Get the debug-information entry for current variable 'vidx'.
257
*/
258
1.85M
static LocVar *localdebuginfo (FuncState *fs, int vidx) {
259
1.85M
  Vardesc *vd = getlocalvardesc(fs,  vidx);
260
1.85M
  if (!varinreg(vd))
261
132k
    return NULL;  /* no debug info. for constants */
262
1.72M
  else {
263
1.72M
    int idx = vd->vd.pidx;
264
1.72M
    lua_assert(idx < fs->ndebugvars);
265
1.72M
    return &fs->f->locvars[idx];
266
1.72M
  }
267
1.85M
}
268
269
270
/*
271
** Create an expression representing variable 'vidx'
272
*/
273
4.27M
static void init_var (FuncState *fs, expdesc *e, int vidx) {
274
4.27M
  e->f = e->t = NO_JUMP;
275
4.27M
  e->k = VLOCAL;
276
4.27M
  e->u.var.vidx = cast_short(vidx);
277
4.27M
  e->u.var.ridx = getlocalvardesc(fs, vidx)->vd.ridx;
278
4.27M
}
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
4.97M
static void check_readonly (LexState *ls, expdesc *e) {
287
4.97M
  FuncState *fs = ls->fs;
288
4.97M
  TString *varname = NULL;  /* to be set if variable is const */
289
4.97M
  switch (e->k) {
290
197
    case VCONST: {
291
197
      varname = ls->dyd->actvar.arr[e->u.info].vd.name;
292
197
      break;
293
0
    }
294
134k
    case VLOCAL: case VVARGVAR: {
295
134k
      Vardesc *vardesc = getlocalvardesc(fs, e->u.var.vidx);
296
134k
      if (vardesc->vd.kind != VDKREG)  /* not a regular variable? */
297
360
        varname = vardesc->vd.name;
298
134k
      break;
299
134k
    }
300
195k
    case VUPVAL: {
301
195k
      Upvaldesc *up = &fs->f->upvalues[e->u.info];
302
195k
      if (up->kind != VDKREG)
303
15
        varname = up->name;
304
195k
      break;
305
134k
    }
306
34.6k
    case VVARGIND: {
307
34.6k
      needvatab(fs->f);  /* function will need a vararg table */
308
34.6k
      e->k = VINDEXED;
309
34.6k
    }  /* FALLTHROUGH */
310
4.63M
    case VINDEXUP: case VINDEXSTR: case VINDEXED: {  /* global variable */
311
4.63M
      if (e->u.ind.ro)  /* read-only? */
312
1.16k
        varname = tsvalue(&fs->f->k[e->u.ind.keystr]);
313
4.63M
      break;
314
4.63M
    }
315
4.63M
    default:
316
11.2k
      lua_assert(e->k == VINDEXI);  /* this one doesn't need any check */
317
11.2k
      return;  /* integer index cannot be read-only */
318
4.97M
  }
319
4.96M
  if (varname)
320
1.74k
    luaK_semerror(ls, "attempt to assign to const variable '%s'",
321
1.74k
                      getstr(varname));
322
4.96M
}
323
324
325
/*
326
** Start the scope for the last 'nvars' created variables.
327
*/
328
2.11M
static void adjustlocalvars (LexState *ls, int nvars) {
329
2.11M
  FuncState *fs = ls->fs;
330
2.11M
  int reglevel = luaY_nvarstack(fs);
331
2.11M
  int i;
332
4.92M
  for (i = 0; i < nvars; i++) {
333
2.81M
    int vidx = fs->nactvar++;
334
2.81M
    Vardesc *var = getlocalvardesc(fs, vidx);
335
2.81M
    var->vd.ridx = cast_byte(reglevel++);
336
2.81M
    var->vd.pidx = registerlocalvar(ls, fs, var->vd.name);
337
2.81M
    luaY_checklimit(fs, reglevel, MAXVARS, "local variables");
338
2.81M
  }
339
2.11M
}
340
341
342
/*
343
** Close the scope for all variables up to level 'tolevel'.
344
** (debug info.)
345
*/
346
2.91M
static void removevars (FuncState *fs, int tolevel) {
347
2.91M
  fs->ls->dyd->actvar.n -= (fs->nactvar - tolevel);
348
4.72M
  while (fs->nactvar > tolevel) {
349
1.81M
    LocVar *var = localdebuginfo(fs, --fs->nactvar);
350
1.81M
    if (var)  /* does it have debug information? */
351
1.67M
      var->endpc = fs->pc;
352
1.81M
  }
353
2.91M
}
354
355
356
/*
357
** Search the upvalues of the function 'fs' for one
358
** with the given 'name'.
359
*/
360
430M
static int searchupvalue (FuncState *fs, TString *name) {
361
430M
  int i;
362
430M
  Upvaldesc *up = fs->f->upvalues;
363
820M
  for (i = 0; i < fs->nups; i++) {
364
507M
    if (eqstr(up[i].name, name)) return i;
365
507M
  }
366
313M
  return -1;  /* not found */
367
430M
}
368
369
370
5.69M
static Upvaldesc *allocupvalue (FuncState *fs) {
371
5.69M
  Proto *f = fs->f;
372
5.69M
  int oldsize = f->sizeupvalues;
373
5.69M
  luaY_checklimit(fs, fs->nups + 1, MAXUPVAL, "upvalues");
374
5.69M
  luaM_growvector(fs->ls->L, f->upvalues, fs->nups, f->sizeupvalues,
375
5.69M
                  Upvaldesc, MAXUPVAL, "upvalues");
376
27.4M
  while (oldsize < f->sizeupvalues)
377
21.7M
    f->upvalues[oldsize++].name = NULL;
378
5.69M
  return &f->upvalues[fs->nups++];
379
5.69M
}
380
381
382
945k
static int newupvalue (FuncState *fs, TString *name, expdesc *v) {
383
945k
  Upvaldesc *up = allocupvalue(fs);
384
945k
  FuncState *prev = fs->prev;
385
945k
  if (v->k == VLOCAL) {
386
124k
    up->instack = 1;
387
124k
    up->idx = v->u.var.ridx;
388
124k
    up->kind = getlocalvardesc(prev, v->u.var.vidx)->vd.kind;
389
124k
    lua_assert(eqstr(name, getlocalvardesc(prev, v->u.var.vidx)->vd.name));
390
124k
  }
391
821k
  else {
392
821k
    up->instack = 0;
393
821k
    up->idx = cast_byte(v->u.info);
394
821k
    up->kind = prev->f->upvalues[v->u.info].kind;
395
821k
    lua_assert(eqstr(name, prev->f->upvalues[v->u.info].name));
396
821k
  }
397
945k
  up->name = name;
398
945k
  luaC_objbarrier(fs->ls->L, fs->f, name);
399
945k
  return fs->nups - 1;
400
945k
}
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
489M
static int searchvar (FuncState *fs, TString *n, expdesc *var) {
415
489M
  int i;
416
1.64G
  for (i = cast_int(fs->nactvar) - 1; i >= 0; i--) {
417
1.21G
    Vardesc *vd = getlocalvardesc(fs, i);
418
1.21G
    if (varglobal(vd)) {  /* global declaration? */
419
110M
      if (vd->vd.name == NULL) {  /* collective declaration? */
420
3.10M
        if (var->u.info < 0)  /* no previous collective declaration? */
421
163k
          var->u.info = fs->firstlocal + i;  /* this is the first one */
422
3.10M
      }
423
107M
      else {  /* global name */
424
107M
        if (eqstr(n, vd->vd.name)) {  /* found? */
425
34.3k
          init_exp(var, VGLOBAL, fs->firstlocal + i);
426
34.3k
          return VGLOBAL;
427
34.3k
        }
428
107M
        else if (var->u.info == -1)  /* active preambular declaration? */
429
123k
          var->u.info = -2;  /* invalidate preambular declaration */
430
107M
      }
431
110M
    }
432
1.10G
    else if (eqstr(n, vd->vd.name)) {  /* found? */
433
59.5M
      if (vd->vd.kind == RDKCTC)  /* compile-time constant? */
434
55.2M
        init_exp(var, VCONST, fs->firstlocal + i);
435
4.27M
      else {  /* local variable */
436
4.27M
        init_var(fs, var, i);
437
4.27M
        if (vd->vd.kind == RDKVAVAR)  /* vararg parameter? */
438
181k
          var->k = VVARGVAR;
439
4.27M
      }
440
59.5M
      return cast_int(var->k);
441
59.5M
    }
442
1.21G
  }
443
430M
  return -1;  /* not found */
444
489M
}
445
446
447
/*
448
** Mark block where variable at given level was defined
449
** (to emit close instructions later).
450
*/
451
124k
static void markupval (FuncState *fs, int level) {
452
124k
  BlockCnt *bl = fs->bl;
453
146k
  while (bl->nactvar > level)
454
22.1k
    bl = bl->previous;
455
124k
  bl->upval = 1;
456
124k
  fs->needclose = 1;
457
124k
}
458
459
460
/*
461
** Mark that current block has a to-be-closed variable.
462
*/
463
48.1k
static void marktobeclosed (FuncState *fs) {
464
48.1k
  BlockCnt *bl = fs->bl;
465
48.1k
  bl->upval = 1;
466
48.1k
  bl->insidetbc = 1;
467
48.1k
  fs->needclose = 1;
468
48.1k
}
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
489M
static void singlevaraux (FuncState *fs, TString *n, expdesc *var, int base) {
477
489M
  int v = searchvar(fs, n, var);  /* look up variables at current level */
478
489M
  if (v >= 0) {  /* found? */
479
59.5M
    if (!base) {
480
4.16M
      if (var->k == VVARGVAR)  /* vararg parameter? */
481
821
        luaK_vapar2local(fs, var);  /* change it to a regular local */
482
4.16M
      if (var->k == VLOCAL)
483
124k
        markupval(fs, var->u.var.vidx);  /* will be used as an upvalue */
484
4.16M
    }
485
    /* else nothing else to be done */
486
59.5M
  }
487
430M
  else {  /* not found at current level; try upvalues */
488
430M
    int idx = searchupvalue(fs, n);  /* try existing upvalues */
489
430M
    if (idx < 0) {  /* not found? */
490
313M
      if (fs->prev != NULL)  /* more levels? */
491
165M
        singlevaraux(fs->prev, n, var, 0);  /* try upper levels */
492
313M
      if (var->k == VLOCAL || var->k == VUPVAL)  /* local or upvalue? */
493
945k
        idx  = newupvalue(fs, n, var);  /* will be a new upvalue */
494
312M
      else  /* it is a global or a constant */
495
312M
        return;  /* don't need to do anything at this level */
496
313M
    }
497
117M
    init_exp(var, VUPVAL, idx);  /* new or old upvalue */
498
117M
  }
499
489M
}
500
501
502
148M
static void buildglobal (LexState *ls, TString *varname, expdesc *var) {
503
148M
  FuncState *fs = ls->fs;
504
148M
  expdesc key;
505
148M
  init_exp(var, VGLOBAL, -1);  /* global by default */
506
148M
  singlevaraux(fs, ls->envn, var, 1);  /* get environment variable */
507
148M
  if (var->k == VGLOBAL)
508
355
    luaK_semerror(ls, "%s is global when accessing variable '%s'",
509
355
                      LUA_ENV, getstr(varname));
510
148M
  luaK_exp2anyregup(fs, var);  /* _ENV could be a constant */
511
148M
  codestring(&key, varname);  /* key is variable name */
512
148M
  luaK_indexed(fs, var, &key);  /* 'var' represents _ENV[varname] */
513
148M
}
514
515
516
/*
517
** Find a variable with the given name 'n', handling global variables
518
** too.
519
*/
520
175M
static void buildvar (LexState *ls, TString *varname, expdesc *var) {
521
175M
  FuncState *fs = ls->fs;
522
175M
  init_exp(var, VGLOBAL, -1);  /* global by default */
523
175M
  singlevaraux(fs, varname, var, 1);
524
175M
  if (var->k == VGLOBAL) {  /* global name? */
525
148M
    int info = var->u.info;
526
    /* global by default in the scope of a global declaration? */
527
148M
    if (info == -2)
528
3.67k
      luaK_semerror(ls, "variable '%s' not declared", getstr(varname));
529
148M
    buildglobal(ls, varname, var);
530
148M
    if (info != -1 && ls->dyd->actvar.arr[info].vd.kind == GDKCONST)
531
6.76k
      var->u.ind.ro = 1;  /* mark variable as read-only */
532
148M
    else  /* anyway must be a global */
533
148M
      lua_assert(info == -1 || ls->dyd->actvar.arr[info].vd.kind == GDKREG);
534
148M
  }
535
175M
}
536
537
538
176M
static void singlevar (LexState *ls, expdesc *var) {
539
176M
  buildvar(ls, str_checkname(ls), var);
540
176M
}
541
542
543
/*
544
** Adjust the number of results from an expression list 'e' with 'nexps'
545
** expressions to 'nvars' values.
546
*/
547
770k
static void adjust_assign (LexState *ls, int nvars, int nexps, expdesc *e) {
548
770k
  FuncState *fs = ls->fs;
549
770k
  int needed = nvars - nexps;  /* extra values needed */
550
770k
  luaK_checkstack(fs, needed);
551
770k
  if (hasmultret(e->k)) {  /* last expression has multiple returns? */
552
104k
    int extra = needed + 1;  /* discount last expression itself */
553
104k
    if (extra < 0)
554
6.65k
      extra = 0;
555
104k
    luaK_setreturns(fs, e, extra);  /* last exp. provides the difference */
556
104k
  }
557
666k
  else {
558
666k
    if (e->k != VVOID)  /* at least one expression? */
559
377k
      luaK_exp2nextreg(fs, e);  /* close last expression */
560
666k
    if (needed > 0)  /* missing values? */
561
419k
      luaK_nil(fs, fs->freereg, needed);  /* complete with nils */
562
666k
  }
563
770k
  if (needed > 0)
564
468k
    luaK_reserveregs(fs, needed);  /* registers for extra values */
565
302k
  else  /* adding 'needed' is actually a subtraction */
566
302k
    fs->freereg = cast_byte(fs->freereg + needed);  /* remove extra values */
567
770k
}
568
569
570
206M
#define enterlevel(ls)  luaE_incCstack(ls->L)
571
572
573
202M
#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
13
static l_noret jumpscopeerror (LexState *ls, Labeldesc *gt) {
581
13
  TString *tsname = getlocalvardesc(ls->fs, gt->nactvar)->vd.name;
582
13
  const char *varname = (tsname != NULL) ? getstr(tsname) : "*";
583
13
  luaK_semerror(ls,
584
13
     "<goto %s> at line %d jumps into the scope of '%s'",
585
13
      getstr(gt->name), gt->line, varname);  /* raise the error */
586
13
}
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
96.3k
static void closegoto (LexState *ls, int g, Labeldesc *label, int bup) {
598
96.3k
  int i;
599
96.3k
  FuncState *fs = ls->fs;
600
96.3k
  Labellist *gl = &ls->dyd->gt;  /* list of gotos */
601
96.3k
  Labeldesc *gt = &gl->arr[g];  /* goto to be resolved */
602
96.3k
  lua_assert(eqstr(gt->name, label->name));
603
96.3k
  if (l_unlikely(gt->nactvar < label->nactvar))  /* enter some scope? */
604
13
    jumpscopeerror(ls, gt);
605
96.2k
  if (gt->close ||
606
91.4k
      (label->nactvar < gt->nactvar && bup)) {  /* needs close? */
607
22.6k
    lu_byte stklevel = reglevel(fs, label->nactvar);
608
    /* move jump to CLOSE position */
609
22.6k
    fs->f->code[gt->pc + 1] = fs->f->code[gt->pc];
610
    /* put CLOSE instruction at original position */
611
22.6k
    fs->f->code[gt->pc] = CREATE_ABCk(OP_CLOSE, stklevel, 0, 0, 0);
612
22.6k
    gt->pc++;  /* must point to jump instruction */
613
22.6k
  }
614
96.2k
  luaK_patchlist(ls->fs, gt->pc, label->pc);  /* goto jumps to label */
615
874k
  for (i = g; i < gl->n - 1; i++)  /* remove goto from pending list */
616
778k
    gl->arr[i] = gl->arr[i + 1];
617
96.2k
  gl->n--;
618
96.2k
}
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
406k
static Labeldesc *findlabel (LexState *ls, TString *name, int ilb) {
627
406k
  Dyndata *dyd = ls->dyd;
628
443k
  for (; ilb < dyd->label.n; ilb++) {
629
143k
    Labeldesc *lb = &dyd->label.arr[ilb];
630
143k
    if (eqstr(lb->name, name))  /* correct label? */
631
106k
      return lb;
632
143k
  }
633
299k
  return NULL;  /* label not found */
634
406k
}
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
605k
                          int line, int pc) {
642
605k
  int n = l->n;
643
605k
  luaM_growvector(ls->L, l->arr, n, l->size,
644
605k
                  Labeldesc, SHRT_MAX, "labels/gotos");
645
605k
  l->arr[n].name = name;
646
605k
  l->arr[n].line = line;
647
605k
  l->arr[n].nactvar = ls->fs->nactvar;
648
605k
  l->arr[n].close = 0;
649
605k
  l->arr[n].pc = pc;
650
605k
  l->n = n + 1;
651
605k
  return n;
652
605k
}
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
480k
static int newgotoentry (LexState *ls, TString *name, int line) {
664
480k
  FuncState *fs = ls->fs;
665
480k
  int pc = luaK_jump(fs);  /* create jump */
666
480k
  luaK_codeABC(fs, OP_CLOSE, 0, 1, 0);  /* spaceholder, marked as dead */
667
480k
  return newlabelentry(ls, &ls->dyd->gt, name, line, pc);
668
480k
}
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
125k
static void createlabel (LexState *ls, TString *name, int line, int last) {
679
125k
  FuncState *fs = ls->fs;
680
125k
  Labellist *ll = &ls->dyd->label;
681
125k
  int l = newlabelentry(ls, ll, name, line, luaK_getlabel(fs));
682
125k
  if (last) {  /* label is last no-op statement in the block? */
683
    /* assume that locals are already out of scope */
684
24.0k
    ll->arr[l].nactvar = fs->bl->nactvar;
685
24.0k
  }
686
125k
}
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
2.91M
static void solvegotos (FuncState *fs, BlockCnt *bl) {
697
2.91M
  LexState *ls = fs->ls;
698
2.91M
  Labellist *gl = &ls->dyd->gt;
699
2.91M
  int outlevel = reglevel(fs, bl->nactvar);  /* level outside the block */
700
2.91M
  int igt = bl->firstgoto;  /* first goto in the finishing block */
701
3.26M
  while (igt < gl->n) {   /* for each pending goto */
702
353k
    Labeldesc *gt = &gl->arr[igt];
703
    /* search for a matching label in the current block */
704
353k
    Labeldesc *lb = findlabel(ls, gt->name, bl->firstlabel);
705
353k
    if (lb != NULL)  /* found a match? */
706
96.3k
      closegoto(ls, igt, lb, bl->upval);  /* close and remove goto */
707
257k
    else {  /* adjust 'goto' for outer block */
708
      /* block has variables to be closed and goto escapes the scope of
709
         some variable? */
710
257k
      if (bl->upval && reglevel(fs, gt->nactvar) > outlevel)
711
6.44k
        gt->close = 1;  /* jump may need a close */
712
257k
      gt->nactvar = bl->nactvar;  /* correct level for outer block */
713
257k
      igt++;  /* go to next goto */
714
257k
    }
715
353k
  }
716
2.91M
  ls->dyd->label.n = bl->firstlabel;  /* remove local labels */
717
2.91M
}
718
719
720
6.83M
static void enterblock (FuncState *fs, BlockCnt *bl, lu_byte isloop) {
721
6.83M
  bl->isloop = isloop;
722
6.83M
  bl->nactvar = fs->nactvar;
723
6.83M
  bl->firstlabel = fs->ls->dyd->label.n;
724
6.83M
  bl->firstgoto = fs->ls->dyd->gt.n;
725
6.83M
  bl->upval = 0;
726
  /* inherit 'insidetbc' from enclosing block */
727
6.83M
  bl->insidetbc = (fs->bl != NULL && fs->bl->insidetbc);
728
6.83M
  bl->previous = fs->bl;  /* link block in function's block list */
729
6.83M
  fs->bl = bl;
730
6.83M
  lua_assert(fs->freereg == luaY_nvarstack(fs));
731
6.83M
}
732
733
734
/*
735
** generates an error for an undefined 'goto'.
736
*/
737
1.47k
static l_noret undefgoto (LexState *ls, Labeldesc *gt) {
738
  /* breaks are checked when created, cannot be undefined */
739
1.47k
  lua_assert(!eqstr(gt->name, ls->brkn));
740
1.47k
  luaK_semerror(ls, "no visible label '%s' for <goto> at line %d",
741
1.47k
                    getstr(gt->name), gt->line);
742
1.47k
}
743
744
745
2.91M
static void leaveblock (FuncState *fs) {
746
2.91M
  BlockCnt *bl = fs->bl;
747
2.91M
  LexState *ls = fs->ls;
748
2.91M
  lu_byte stklevel = reglevel(fs, bl->nactvar);  /* level outside block */
749
2.91M
  if (bl->previous && bl->upval)  /* need a 'close'? */
750
37.3k
    luaK_codeABC(fs, OP_CLOSE, stklevel, 0, 0);
751
2.91M
  fs->freereg = stklevel;  /* free registers */
752
2.91M
  removevars(fs, bl->nactvar);  /* remove block locals */
753
2.91M
  lua_assert(bl->nactvar == fs->nactvar);  /* back to level on entry */
754
2.91M
  if (bl->isloop == 2)  /* has to fix pending breaks? */
755
83.3k
    createlabel(ls, ls->brkn, 0, 0);
756
2.91M
  solvegotos(fs, bl);
757
2.91M
  if (bl->previous == NULL) {  /* was it the last block? */
758
2.36M
    if (bl->firstgoto < ls->dyd->gt.n)  /* still pending gotos? */
759
1.47k
      undefgoto(ls, &ls->dyd->gt.arr[bl->firstgoto]);  /* error */
760
2.36M
  }
761
2.90M
  fs->bl = bl->previous;  /* current block now is previous one */
762
2.90M
}
763
764
765
/*
766
** adds a new prototype into list of prototypes
767
*/
768
1.39M
static Proto *addprototype (LexState *ls) {
769
1.39M
  Proto *clp;
770
1.39M
  lua_State *L = ls->L;
771
1.39M
  FuncState *fs = ls->fs;
772
1.39M
  Proto *f = fs->f;  /* prototype of current function */
773
1.39M
  if (fs->np >= f->sizep) {
774
533k
    int oldsize = f->sizep;
775
533k
    luaM_growvector(L, f->p, fs->np, f->sizep, Proto *, MAXARG_Bx, "functions");
776
3.87M
    while (oldsize < f->sizep)
777
3.34M
      f->p[oldsize++] = NULL;
778
533k
  }
779
1.39M
  f->p[fs->np++] = clp = luaF_newproto(L);
780
1.39M
  luaC_objbarrier(L, f, clp);
781
1.39M
  return clp;
782
1.39M
}
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
956k
static void codeclosure (LexState *ls, expdesc *v) {
793
956k
  FuncState *fs = ls->fs->prev;
794
956k
  init_exp(v, VRELOC, luaK_codeABx(fs, OP_CLOSURE, 0, fs->np - 1));
795
956k
  luaK_exp2nextreg(fs, v);  /* fix it at the last register */
796
956k
}
797
798
799
6.14M
static void open_func (LexState *ls, FuncState *fs, BlockCnt *bl) {
800
6.14M
  lua_State *L = ls->L;
801
6.14M
  Proto *f = fs->f;
802
6.14M
  fs->prev = ls->fs;  /* linked list of funcstates */
803
6.14M
  fs->ls = ls;
804
6.14M
  ls->fs = fs;
805
6.14M
  fs->pc = 0;
806
6.14M
  fs->previousline = f->linedefined;
807
6.14M
  fs->iwthabs = 0;
808
6.14M
  fs->lasttarget = 0;
809
6.14M
  fs->freereg = 0;
810
6.14M
  fs->nk = 0;
811
6.14M
  fs->nabslineinfo = 0;
812
6.14M
  fs->np = 0;
813
6.14M
  fs->nups = 0;
814
6.14M
  fs->ndebugvars = 0;
815
6.14M
  fs->nactvar = 0;
816
6.14M
  fs->needclose = 0;
817
6.14M
  fs->firstlocal = ls->dyd->actvar.n;
818
6.14M
  fs->firstlabel = ls->dyd->label.n;
819
6.14M
  fs->bl = NULL;
820
6.14M
  f->source = ls->source;
821
6.14M
  luaC_objbarrier(L, f, f->source);
822
6.14M
  f->maxstacksize = 2;  /* registers 0/1 are always valid */
823
6.14M
  fs->kcache = luaH_new(L);  /* create table for function */
824
6.14M
  sethvalue2s(L, L->top.p, fs->kcache);  /* anchor it */
825
6.14M
  luaD_inctop(L);
826
6.14M
  enterblock(fs, bl, 0);
827
6.14M
}
828
829
830
2.36M
static void close_func (LexState *ls) {
831
2.36M
  lua_State *L = ls->L;
832
2.36M
  FuncState *fs = ls->fs;
833
2.36M
  Proto *f = fs->f;
834
2.36M
  luaK_ret(fs, luaY_nvarstack(fs), 0);  /* final return */
835
2.36M
  leaveblock(fs);
836
2.36M
  lua_assert(fs->bl == NULL);
837
2.36M
  luaK_finish(fs);
838
2.36M
  luaM_shrinkvector(L, f->code, f->sizecode, fs->pc, Instruction);
839
2.36M
  luaM_shrinkvector(L, f->lineinfo, f->sizelineinfo, fs->pc, ls_byte);
840
2.36M
  luaM_shrinkvector(L, f->abslineinfo, f->sizeabslineinfo,
841
2.36M
                       fs->nabslineinfo, AbsLineInfo);
842
2.36M
  luaM_shrinkvector(L, f->k, f->sizek, fs->nk, TValue);
843
2.36M
  luaM_shrinkvector(L, f->p, f->sizep, fs->np, Proto *);
844
2.36M
  luaM_shrinkvector(L, f->locvars, f->sizelocvars, fs->ndebugvars, LocVar);
845
2.36M
  luaM_shrinkvector(L, f->upvalues, f->sizeupvalues, fs->nups, Upvaldesc);
846
2.36M
  ls->fs = fs->prev;
847
2.36M
  L->top.p--;  /* pop kcache table */
848
2.36M
  luaC_checkGC(L);
849
2.36M
}
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
12.3M
static int block_follow (LexState *ls, int withuntil) {
865
12.3M
  switch (ls->t.token) {
866
23.4k
    case TK_ELSE: case TK_ELSEIF:
867
2.52M
    case TK_END: case TK_EOS:
868
2.52M
      return 1;
869
57.3k
    case TK_UNTIL: return withuntil;
870
9.73M
    default: return 0;
871
12.3M
  }
872
12.3M
}
873
874
875
6.34M
static void statlist (LexState *ls) {
876
  /* statlist -> { stat [';'] } */
877
15.5M
  while (!block_follow(ls, 1)) {
878
9.50M
    if (ls->t.token == TK_RETURN) {
879
264k
      statement(ls);
880
264k
      return;  /* 'return' must be last statement */
881
264k
    }
882
9.24M
    statement(ls);
883
9.24M
  }
884
6.34M
}
885
886
887
5.44M
static void fieldsel (LexState *ls, expdesc *v) {
888
  /* fieldsel -> ['.' | ':'] NAME */
889
5.44M
  FuncState *fs = ls->fs;
890
5.44M
  expdesc key;
891
5.44M
  luaK_exp2anyregup(fs, v);
892
5.44M
  luaX_next(ls);  /* skip the dot or colon */
893
5.44M
  codename(ls, &key);
894
5.44M
  luaK_indexed(fs, v, &key);
895
5.44M
}
896
897
898
402k
static void yindex (LexState *ls, expdesc *v) {
899
  /* index -> '[' expr ']' */
900
402k
  luaX_next(ls);  /* skip the '[' */
901
402k
  expr(ls, v);
902
402k
  luaK_exp2val(ls->fs, v);
903
402k
  checknext(ls, ']');
904
402k
}
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
9.47M
#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
1.24M
static void recfield (LexState *ls, ConsControl *cc) {
937
  /* recfield -> (NAME | '['exp']') = exp */
938
1.24M
  FuncState *fs = ls->fs;
939
1.24M
  lu_byte reg = ls->fs->freereg;
940
1.24M
  expdesc tab, key, val;
941
1.24M
  if (ls->t.token == TK_NAME)
942
1.17M
    codename(ls, &key);
943
61.7k
  else  /* ls->t.token == '[' */
944
61.7k
    yindex(ls, &key);
945
1.24M
  cc->nh++;
946
1.24M
  checknext(ls, '=');
947
1.24M
  tab = *cc->t;
948
1.24M
  luaK_indexed(fs, &tab, &key);
949
1.24M
  expr(ls, &val);
950
1.24M
  luaK_storevar(fs, &tab, &val);
951
1.24M
  fs->freereg = reg;  /* free registers */
952
1.24M
}
953
954
955
6.73M
static void closelistfield (FuncState *fs, ConsControl *cc) {
956
6.73M
  lua_assert(cc->tostore > 0);
957
6.73M
  luaK_exp2nextreg(fs, &cc->v);
958
6.73M
  cc->v.k = VVOID;
959
6.73M
  if (cc->tostore >= cc->maxtostore) {
960
1.28M
    luaK_setlist(fs, cc->t->u.info, cc->na, cc->tostore);  /* flush */
961
1.28M
    cc->na += cc->tostore;
962
1.28M
    cc->tostore = 0;  /* no more items pending */
963
1.28M
  }
964
6.73M
}
965
966
967
216k
static void lastlistfield (FuncState *fs, ConsControl *cc) {
968
216k
  if (cc->tostore == 0) return;
969
70.0k
  if (hasmultret(cc->v.k)) {
970
15.4k
    luaK_setmultret(fs, &cc->v);
971
15.4k
    luaK_setlist(fs, cc->t->u.info, cc->na, LUA_MULTRET);
972
15.4k
    cc->na--;  /* do not count last expression (unknown number of elements) */
973
15.4k
  }
974
54.5k
  else {
975
54.5k
    if (cc->v.k != VVOID)
976
43.3k
      luaK_exp2nextreg(fs, &cc->v);
977
54.5k
    luaK_setlist(fs, cc->t->u.info, cc->na, cc->tostore);
978
54.5k
  }
979
70.0k
  cc->na += cc->tostore;
980
70.0k
}
981
982
983
8.23M
static void listfield (LexState *ls, ConsControl *cc) {
984
  /* listfield -> exp */
985
8.23M
  expr(ls, &cc->v);
986
8.23M
  cc->tostore++;
987
8.23M
}
988
989
990
9.47M
static void field (LexState *ls, ConsControl *cc) {
991
  /* field -> listfield | recfield */
992
9.47M
  switch(ls->t.token) {
993
8.53M
    case TK_NAME: {  /* may be 'listfield' or 'recfield' */
994
8.53M
      if (luaX_lookahead(ls) != '=')  /* expression? */
995
7.34M
        listfield(ls, cc);
996
1.18M
      else
997
1.18M
        recfield(ls, cc);
998
8.53M
      break;
999
0
    }
1000
61.7k
    case '[': {
1001
61.7k
      recfield(ls, cc);
1002
61.7k
      break;
1003
0
    }
1004
885k
    default: {
1005
885k
      listfield(ls, cc);
1006
885k
      break;
1007
0
    }
1008
9.47M
  }
1009
9.47M
}
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
1.68M
static int maxtostore (FuncState *fs) {
1018
1.68M
  int numfreeregs = MAX_FSTACK - fs->freereg;
1019
1.68M
  if (numfreeregs >= 160)  /* "lots" of registers? */
1020
1.38M
    return numfreeregs / 5;  /* use up to 1/5 of them */
1021
300k
  else if (numfreeregs >= 80)  /* still "enough" registers? */
1022
228k
    return 10;  /* one 'SETLIST' instruction for each 10 values */
1023
72.2k
  else  /* save registers for potential more nesting */
1024
72.2k
    return 1;
1025
1.68M
}
1026
1027
1028
1.68M
static void constructor (LexState *ls, expdesc *t) {
1029
  /* constructor -> '{' [ field { sep field } [sep] ] '}'
1030
     sep -> ',' | ';' */
1031
1.68M
  FuncState *fs = ls->fs;
1032
1.68M
  int line = ls->linenumber;
1033
1.68M
  int pc = luaK_codevABCk(fs, OP_NEWTABLE, 0, 0, 0, 0);
1034
1.68M
  ConsControl cc;
1035
1.68M
  luaK_code(fs, 0);  /* space for extra arg. */
1036
1.68M
  cc.na = cc.nh = cc.tostore = 0;
1037
1.68M
  cc.t = t;
1038
1.68M
  init_exp(t, VNONRELOC, fs->freereg);  /* table will be at stack top */
1039
1.68M
  luaK_reserveregs(fs, 1);
1040
1.68M
  init_exp(&cc.v, VVOID, 0);  /* no value (yet) */
1041
1.68M
  checknext(ls, '{' /*}*/);
1042
1.68M
  cc.maxtostore = maxtostore(fs);
1043
9.61M
  do {
1044
9.61M
    if (ls->t.token == /*{*/ '}') break;
1045
9.47M
    if (cc.v.k != VVOID)  /* is there a previous list item? */
1046
6.73M
      closelistfield(fs, &cc);  /* close it */
1047
9.47M
    field(ls, &cc);
1048
9.47M
    luaY_checklimit(fs, cc.tostore + cc.na + cc.nh, MAX_CNST,
1049
9.47M
                    "items in a constructor");
1050
9.47M
  } while (testnext(ls, ',') || testnext(ls, ';'));
1051
1.68M
  check_match(ls, /*{*/ '}', '{' /*}*/, line);
1052
1.68M
  lastlistfield(fs, &cc);
1053
1.68M
  luaK_settablesize(fs, pc, t->u.info, cc.na, cc.nh);
1054
1.68M
}
1055
1056
/* }====================================================================== */
1057
1058
1059
4.76M
static void setvararg (FuncState *fs) {
1060
4.76M
  fs->f->flag |= PF_VAHID;  /* by default, use hidden vararg arguments */
1061
4.76M
  luaK_codeABC(fs, OP_VARARGPREP, 0, 0, 0);
1062
4.76M
}
1063
1064
1065
1.38M
static void parlist (LexState *ls) {
1066
  /* parlist -> [ {NAME ','} (NAME | '...') ] */
1067
1.38M
  FuncState *fs = ls->fs;
1068
1.38M
  Proto *f = fs->f;
1069
1.38M
  int nparams = 0;
1070
1.38M
  int varargk = 0;
1071
1.38M
  if (ls->t.token != ')') {  /* is 'parlist' not empty? */
1072
1.01M
    do {
1073
1.01M
      switch (ls->t.token) {
1074
994k
        case TK_NAME: {
1075
994k
          new_localvar(ls, str_checkname(ls));
1076
994k
          nparams++;
1077
994k
          break;
1078
0
        }
1079
17.4k
        case TK_DOTS: {
1080
17.4k
          varargk = 1;
1081
17.4k
          luaX_next(ls);  /* skip '...' */
1082
17.4k
          if (ls->t.token == TK_NAME)
1083
6.85k
            new_varkind(ls, str_checkname(ls), RDKVAVAR);
1084
10.6k
          else
1085
10.6k
            new_localvarliteral(ls, "(vararg table)");
1086
17.4k
          break;
1087
0
        }
1088
4.82k
        default: luaX_syntaxerror(ls, "<name> or '...' expected");
1089
1.01M
      }
1090
1.01M
    } while (!varargk && testnext(ls, ','));
1091
493k
  }
1092
1.37M
  adjustlocalvars(ls, nparams);
1093
1.37M
  f->numparams = cast_byte(fs->nactvar);
1094
1.37M
  if (varargk) {
1095
17.4k
    setvararg(fs);  /* declared vararg */
1096
17.4k
    adjustlocalvars(ls, 1);  /* vararg parameter */
1097
17.4k
  }
1098
  /* reserve registers for parameters (plus vararg parameter, if present) */
1099
1.37M
  luaK_reserveregs(fs, fs->nactvar);
1100
1.37M
}
1101
1102
1103
1.39M
static void body (LexState *ls, expdesc *e, int ismethod, int line) {
1104
  /* body ->  '(' parlist ')' block END */
1105
1.39M
  FuncState new_fs;
1106
1.39M
  BlockCnt bl;
1107
1.39M
  new_fs.f = addprototype(ls);
1108
1.39M
  new_fs.f->linedefined = line;
1109
1.39M
  open_func(ls, &new_fs, &bl);
1110
1.39M
  checknext(ls, '(');
1111
1.39M
  if (ismethod) {
1112
12.7k
    new_localvarliteral(ls, "self");  /* create 'self' parameter */
1113
12.7k
    adjustlocalvars(ls, 1);
1114
12.7k
  }
1115
1.39M
  parlist(ls);
1116
1.39M
  checknext(ls, ')');
1117
1.39M
  statlist(ls);
1118
1.39M
  new_fs.f->lastlinedefined = ls->linenumber;
1119
1.39M
  check_match(ls, TK_END, TK_FUNCTION, line);
1120
1.39M
  codeclosure(ls, e);
1121
1.39M
  close_func(ls);
1122
1.39M
}
1123
1124
1125
3.95M
static int explist (LexState *ls, expdesc *v) {
1126
  /* explist -> expr { ',' expr } */
1127
3.95M
  int n = 1;  /* at least one expression */
1128
3.95M
  expr(ls, v);
1129
5.27M
  while (testnext(ls, ',')) {
1130
1.31M
    luaK_exp2nextreg(ls->fs, v);
1131
1.31M
    expr(ls, v);
1132
1.31M
    n++;
1133
1.31M
  }
1134
3.95M
  return n;
1135
3.95M
}
1136
1137
1138
3.94M
static void funcargs (LexState *ls, expdesc *f) {
1139
3.94M
  FuncState *fs = ls->fs;
1140
3.94M
  expdesc args;
1141
3.94M
  int base, nparams;
1142
3.94M
  int line = ls->linenumber;
1143
3.94M
  switch (ls->t.token) {
1144
1.74M
    case '(': {  /* funcargs -> '(' [ explist ] ')' */
1145
1.74M
      luaX_next(ls);
1146
1.74M
      if (ls->t.token == ')')  /* arg list is empty? */
1147
193k
        args.k = VVOID;
1148
1.55M
      else {
1149
1.55M
        explist(ls, &args);
1150
1.55M
        if (hasmultret(args.k))
1151
108k
          luaK_setmultret(fs, &args);
1152
1.55M
      }
1153
1.74M
      check_match(ls, ')', '(', line);
1154
1.74M
      break;
1155
0
    }
1156
702k
    case '{' /*}*/: {  /* funcargs -> constructor */
1157
702k
      constructor(ls, &args);
1158
702k
      break;
1159
0
    }
1160
1.48M
    case TK_STRING: {  /* funcargs -> STRING */
1161
1.48M
      codestring(&args, ls->t.seminfo.ts);
1162
1.48M
      luaX_next(ls);  /* must use 'seminfo' before 'next' */
1163
1.48M
      break;
1164
0
    }
1165
11.7k
    default: {
1166
11.7k
      luaX_syntaxerror(ls, "function arguments expected");
1167
0
    }
1168
3.94M
  }
1169
2.80M
  lua_assert(f->k == VNONRELOC);
1170
2.80M
  base = f->u.info;  /* base register for call */
1171
2.80M
  if (hasmultret(args.k))
1172
106k
    nparams = LUA_MULTRET;  /* open call */
1173
2.70M
  else {
1174
2.70M
    if (args.k != VVOID)
1175
2.50M
      luaK_exp2nextreg(fs, &args);  /* close last argument */
1176
2.70M
    nparams = fs->freereg - (base+1);
1177
2.70M
  }
1178
2.80M
  init_exp(f, VCALL, luaK_codeABC(fs, OP_CALL, base, nparams+1, 2));
1179
2.80M
  luaK_fixline(fs, line);
1180
  /* call removes function and arguments and leaves one result (unless
1181
     changed later) */
1182
2.80M
  fs->freereg = cast_byte(base + 1);
1183
2.80M
}
1184
1185
1186
1187
1188
/*
1189
** {======================================================================
1190
** Expression parsing
1191
** =======================================================================
1192
*/
1193
1194
1195
177M
static void primaryexp (LexState *ls, expdesc *v) {
1196
  /* primaryexp -> NAME | '(' expr ')' */
1197
177M
  switch (ls->t.token) {
1198
834k
    case '(': {
1199
834k
      int line = ls->linenumber;
1200
834k
      luaX_next(ls);
1201
834k
      expr(ls, v);
1202
834k
      check_match(ls, ')', '(', line);
1203
834k
      luaK_dischargevars(ls->fs, v);
1204
834k
      return;
1205
0
    }
1206
175M
    case TK_NAME: {
1207
175M
      singlevar(ls, v);
1208
175M
      return;
1209
0
    }
1210
1.04M
    default: {
1211
1.04M
      luaX_syntaxerror(ls, "unexpected symbol");
1212
0
    }
1213
177M
  }
1214
177M
}
1215
1216
1217
177M
static void suffixedexp (LexState *ls, expdesc *v) {
1218
  /* suffixedexp ->
1219
       primaryexp { '.' NAME | '[' exp ']' | ':' NAME funcargs | funcargs } */
1220
177M
  FuncState *fs = ls->fs;
1221
177M
  primaryexp(ls, v);
1222
184M
  for (;;) {
1223
184M
    switch (ls->t.token) {
1224
5.39M
      case '.': {  /* fieldsel */
1225
5.39M
        fieldsel(ls, v);
1226
5.39M
        break;
1227
0
      }
1228
340k
      case '[': {  /* '[' exp ']' */
1229
340k
        expdesc key;
1230
340k
        luaK_exp2anyregup(fs, v);
1231
340k
        yindex(ls, &key);
1232
340k
        luaK_indexed(fs, v, &key);
1233
340k
        break;
1234
0
      }
1235
256k
      case ':': {  /* ':' NAME funcargs */
1236
256k
        expdesc key;
1237
256k
        luaX_next(ls);
1238
256k
        codename(ls, &key);
1239
256k
        luaK_self(fs, v, &key);
1240
256k
        funcargs(ls, v);
1241
256k
        break;
1242
0
      }
1243
3.73M
      case '(': case TK_STRING: case '{' /*}*/: {  /* funcargs */
1244
3.73M
        luaK_exp2nextreg(fs, v);
1245
3.73M
        funcargs(ls, v);
1246
3.73M
        break;
1247
3.05M
      }
1248
174M
      default: return;
1249
184M
    }
1250
184M
  }
1251
177M
}
1252
1253
1254
183M
static void simpleexp (LexState *ls, expdesc *v) {
1255
  /* simpleexp -> FLT | INT | STRING | NIL | TRUE | FALSE | ... |
1256
                  constructor | FUNCTION body | suffixedexp */
1257
183M
  switch (ls->t.token) {
1258
1.49M
    case TK_FLT: {
1259
1.49M
      init_exp(v, VKFLT, 0);
1260
1.49M
      v->u.nval = ls->t.seminfo.r;
1261
1.49M
      break;
1262
0
    }
1263
9.96M
    case TK_INT: {
1264
9.96M
      init_exp(v, VKINT, 0);
1265
9.96M
      v->u.ival = ls->t.seminfo.i;
1266
9.96M
      break;
1267
0
    }
1268
556k
    case TK_STRING: {
1269
556k
      codestring(v, ls->t.seminfo.ts);
1270
556k
      break;
1271
0
    }
1272
176k
    case TK_NIL: {
1273
176k
      init_exp(v, VNIL, 0);
1274
176k
      break;
1275
0
    }
1276
63.8k
    case TK_TRUE: {
1277
63.8k
      init_exp(v, VTRUE, 0);
1278
63.8k
      break;
1279
0
    }
1280
15.0k
    case TK_FALSE: {
1281
15.0k
      init_exp(v, VFALSE, 0);
1282
15.0k
      break;
1283
0
    }
1284
102k
    case TK_DOTS: {  /* vararg */
1285
102k
      FuncState *fs = ls->fs;
1286
102k
      check_condition(ls, isvararg(fs->f),
1287
102k
                      "cannot use '...' outside a vararg function");
1288
98.1k
      init_exp(v, VVARARG, luaK_codeABC(fs, OP_VARARG, 0, fs->f->numparams, 1));
1289
98.1k
      break;
1290
102k
    }
1291
980k
    case '{' /*}*/: {  /* constructor */
1292
980k
      constructor(ls, v);
1293
980k
      return;
1294
102k
    }
1295
1.27M
    case TK_FUNCTION: {
1296
1.27M
      luaX_next(ls);
1297
1.27M
      body(ls, v, 0, ls->linenumber);
1298
1.27M
      return;
1299
102k
    }
1300
168M
    default: {
1301
168M
      suffixedexp(ls, v);
1302
168M
      return;
1303
102k
    }
1304
183M
  }
1305
12.3M
  luaX_next(ls);
1306
12.3M
}
1307
1308
1309
193M
static UnOpr getunopr (int op) {
1310
193M
  switch (op) {
1311
112k
    case TK_NOT: return OPR_NOT;
1312
1.59M
    case '-': return OPR_MINUS;
1313
8.27M
    case '~': return OPR_BNOT;
1314
217k
    case '#': return OPR_LEN;
1315
183M
    default: return OPR_NOUNOPR;
1316
193M
  }
1317
193M
}
1318
1319
1320
190M
static BinOpr getbinopr (int op) {
1321
190M
  switch (op) {
1322
360k
    case '+': return OPR_ADD;
1323
1.96M
    case '-': return OPR_SUB;
1324
4.76M
    case '*': return OPR_MUL;
1325
1.13M
    case '%': return OPR_MOD;
1326
3.04M
    case '^': return OPR_POW;
1327
6.96M
    case '/': return OPR_DIV;
1328
403k
    case TK_IDIV: return OPR_IDIV;
1329
124k
    case '&': return OPR_BAND;
1330
518k
    case '|': return OPR_BOR;
1331
4.73M
    case '~': return OPR_BXOR;
1332
645k
    case TK_SHL: return OPR_SHL;
1333
1.31M
    case TK_SHR: return OPR_SHR;
1334
2.69M
    case TK_CONCAT: return OPR_CONCAT;
1335
77.3k
    case TK_NE: return OPR_NE;
1336
192k
    case TK_EQ: return OPR_EQ;
1337
8.12M
    case '<': return OPR_LT;
1338
568k
    case TK_LE: return OPR_LE;
1339
137M
    case '>': return OPR_GT;
1340
31.3k
    case TK_GE: return OPR_GE;
1341
453k
    case TK_AND: return OPR_AND;
1342
646k
    case TK_OR: return OPR_OR;
1343
14.2M
    default: return OPR_NOBINOPR;
1344
190M
  }
1345
190M
}
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
10.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
193M
static BinOpr subexpr (LexState *ls, expdesc *v, int limit) {
1375
193M
  BinOpr op;
1376
193M
  UnOpr uop;
1377
193M
  enterlevel(ls);
1378
193M
  uop = getunopr(ls->t.token);
1379
193M
  if (uop != OPR_NOUNOPR) {  /* prefix (unary) operator? */
1380
10.1M
    int line = ls->linenumber;
1381
10.1M
    luaX_next(ls);  /* skip operator */
1382
10.1M
    subexpr(ls, v, UNARY_PRIORITY);
1383
10.1M
    luaK_prefix(ls->fs, uop, v, line);
1384
10.1M
  }
1385
183M
  else simpleexp(ls, v);
1386
  /* expand while operators have priorities higher than 'limit' */
1387
193M
  op = getbinopr(ls->t.token);
1388
360M
  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
193M
  leavelevel(ls);
1400
193M
  return op;  /* return first untreated operator */
1401
193M
}
1402
1403
1404
16.3M
static void expr (LexState *ls, expdesc *v) {
1405
16.3M
  subexpr(ls, v, 0);
1406
16.3M
}
1407
1408
/* }==================================================================== */
1409
1410
1411
1412
/*
1413
** {======================================================================
1414
** Rules for Statements
1415
** =======================================================================
1416
*/
1417
1418
1419
364k
static void block (LexState *ls) {
1420
  /* block -> statlist */
1421
364k
  FuncState *fs = ls->fs;
1422
364k
  BlockCnt bl;
1423
364k
  enterblock(fs, &bl, 0);
1424
364k
  statlist(ls);
1425
364k
  leaveblock(fs);
1426
364k
}
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
265k
static void check_conflict (LexState *ls, struct LHS_assign *lh, expdesc *v) {
1446
265k
  FuncState *fs = ls->fs;
1447
265k
  lu_byte extra = fs->freereg;  /* eventual position to save local variable */
1448
265k
  int conflict = 0;
1449
2.57M
  for (; lh; lh = lh->prev) {  /* check all previous assignments */
1450
2.30M
    if (vkisindexed(lh->v.k)) {  /* assignment to table field? */
1451
1.70M
      if (lh->v.k == VINDEXUP) {  /* is table an upvalue? */
1452
606k
        if (v->k == VUPVAL && lh->v.u.ind.t == v->u.info) {
1453
504k
          conflict = 1;  /* table is the upvalue being assigned now */
1454
504k
          lh->v.k = VINDEXSTR;
1455
504k
          lh->v.u.ind.t = extra;  /* assignment will use safe copy */
1456
504k
        }
1457
606k
      }
1458
1.09M
      else {  /* table is a register */
1459
1.09M
        if (v->k == VLOCAL && lh->v.u.ind.t == v->u.var.ridx) {
1460
3.49k
          conflict = 1;  /* table is the local being assigned now */
1461
3.49k
          lh->v.u.ind.t = extra;  /* assignment will use safe copy */
1462
3.49k
        }
1463
        /* is index the local being assigned? */
1464
1.09M
        if (lh->v.k == VINDEXED && v->k == VLOCAL &&
1465
11.2k
            lh->v.u.ind.idx == v->u.var.ridx) {
1466
584
          conflict = 1;
1467
584
          lh->v.u.ind.idx = extra;  /* previous assignment will use safe copy */
1468
584
        }
1469
1.09M
      }
1470
1.70M
    }
1471
2.30M
  }
1472
265k
  if (conflict) {
1473
    /* copy upvalue/local value to a temporary (in position 'extra') */
1474
132k
    if (v->k == VLOCAL)
1475
1.69k
      luaK_codeABC(fs, OP_MOVE, extra, v->u.var.ridx, 0);
1476
130k
    else
1477
130k
      luaK_codeABC(fs, OP_GETUPVAL, extra, v->u.info, 0);
1478
132k
    luaK_reserveregs(fs, 1);
1479
132k
  }
1480
265k
}
1481
1482
1483
/* Create code to store the "top" register in 'var' */
1484
1.79M
static void storevartop (FuncState *fs, expdesc *var) {
1485
1.79M
  expdesc e;
1486
1.79M
  init_exp(&e, VNONRELOC, fs->freereg - 1);
1487
1.79M
  luaK_storevar(fs, var, &e);  /* will also free the top register */
1488
1.79M
}
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
4.94M
static void restassign (LexState *ls, struct LHS_assign *lh, int nvars) {
1499
4.94M
  expdesc e;
1500
4.94M
  check_condition(ls, vkisvar(lh->v.k), "syntax error");
1501
4.90M
  check_readonly(ls, &lh->v);
1502
4.90M
  if (testnext(ls, ',')) {  /* restassign -> ',' suffixedexp restassign */
1503
2.85M
    struct LHS_assign nv;
1504
2.85M
    nv.prev = lh;
1505
2.85M
    suffixedexp(ls, &nv.v);
1506
2.85M
    if (!vkisindexed(nv.v.k))
1507
265k
      check_conflict(ls, lh, &nv.v);
1508
2.85M
    enterlevel(ls);  /* control recursion depth */
1509
2.85M
    restassign(ls, &nv, nvars+1);
1510
2.85M
    leavelevel(ls);
1511
2.85M
  }
1512
2.04M
  else {  /* restassign -> '=' explist */
1513
2.04M
    int nexps;
1514
2.04M
    checknext(ls, '=');
1515
2.04M
    nexps = explist(ls, &e);
1516
2.04M
    if (nexps != nvars)
1517
207k
      adjust_assign(ls, nvars, nexps, &e);
1518
1.83M
    else {
1519
1.83M
      luaK_setoneret(ls->fs, &e);  /* close last expression */
1520
1.83M
      luaK_storevar(ls->fs, &lh->v, &e);
1521
1.83M
      return;  /* avoid default */
1522
1.83M
    }
1523
2.04M
  }
1524
3.06M
  storevartop(ls->fs, &lh->v);  /* default assignment */
1525
3.06M
}
1526
1527
1528
291k
static int cond (LexState *ls) {
1529
  /* cond -> exp */
1530
291k
  expdesc v;
1531
291k
  expr(ls, &v);  /* read condition */
1532
291k
  if (v.k == VNIL) v.k = VFALSE;  /* 'falses' are all equal here */
1533
291k
  luaK_goiftrue(ls->fs, &v);
1534
291k
  return v.f;
1535
291k
}
1536
1537
1538
392k
static void gotostat (LexState *ls, int line) {
1539
392k
  TString *name = str_checkname(ls);  /* label's name */
1540
392k
  newgotoentry(ls, name, line);
1541
392k
}
1542
1543
1544
/*
1545
** Break statement. Semantically equivalent to "goto break".
1546
*/
1547
104k
static void breakstat (LexState *ls, int line) {
1548
104k
  BlockCnt *bl;  /* to look for an enclosing loop */
1549
339k
  for (bl = ls->fs->bl; bl != NULL; bl = bl->previous) {
1550
330k
    if (bl->isloop)  /* found one? */
1551
95.9k
      goto ok;
1552
330k
  }
1553
8.45k
  luaX_syntaxerror(ls, "break outside loop");
1554
95.9k
 ok:
1555
95.9k
  bl->isloop = 2;  /* signal that block has pending breaks */
1556
95.9k
  luaX_next(ls);  /* skip break */
1557
95.9k
  newgotoentry(ls, ls->brkn, line);
1558
95.9k
}
1559
1560
1561
/*
1562
** Check whether there is already a label with the given 'name' at
1563
** current function.
1564
*/
1565
52.1k
static void checkrepeated (LexState *ls, TString *name) {
1566
52.1k
  Labeldesc *lb = findlabel(ls, name, ls->fs->firstlabel);
1567
52.1k
  if (l_unlikely(lb != NULL))  /* already defined? */
1568
10.3k
    luaK_semerror(ls, "label '%s' already defined on line %d",
1569
10.3k
                      getstr(name), lb->line);  /* error */
1570
52.1k
}
1571
1572
1573
78.9k
static void labelstat (LexState *ls, TString *name, int line) {
1574
  /* label -> '::' NAME '::' */
1575
78.9k
  checknext(ls, TK_DBCOLON);  /* skip double colon */
1576
126k
  while (ls->t.token == ';' || ls->t.token == TK_DBCOLON)
1577
47.7k
    statement(ls);  /* skip other no-op statements */
1578
78.9k
  checkrepeated(ls, name);  /* check for repeated labels */
1579
78.9k
  createlabel(ls, name, line, block_follow(ls, 0));
1580
78.9k
}
1581
1582
1583
23.5k
static void whilestat (LexState *ls, int line) {
1584
  /* whilestat -> WHILE cond DO block END */
1585
23.5k
  FuncState *fs = ls->fs;
1586
23.5k
  int whileinit;
1587
23.5k
  int condexit;
1588
23.5k
  BlockCnt bl;
1589
23.5k
  luaX_next(ls);  /* skip WHILE */
1590
23.5k
  whileinit = luaK_getlabel(fs);
1591
23.5k
  condexit = cond(ls);
1592
23.5k
  enterblock(fs, &bl, 1);
1593
23.5k
  checknext(ls, TK_DO);
1594
23.5k
  block(ls);
1595
23.5k
  luaK_jumpto(fs, whileinit);
1596
23.5k
  check_match(ls, TK_END, TK_WHILE, line);
1597
23.5k
  leaveblock(fs);
1598
23.5k
  luaK_patchtohere(fs, condexit);  /* false conditions finish the loop */
1599
23.5k
}
1600
1601
1602
68.0k
static void repeatstat (LexState *ls, int line) {
1603
  /* repeatstat -> REPEAT block UNTIL cond */
1604
68.0k
  int condexit;
1605
68.0k
  FuncState *fs = ls->fs;
1606
68.0k
  int repeat_init = luaK_getlabel(fs);
1607
68.0k
  BlockCnt bl1, bl2;
1608
68.0k
  enterblock(fs, &bl1, 1);  /* loop block */
1609
68.0k
  enterblock(fs, &bl2, 0);  /* scope block */
1610
68.0k
  luaX_next(ls);  /* skip REPEAT */
1611
68.0k
  statlist(ls);
1612
68.0k
  check_match(ls, TK_UNTIL, TK_REPEAT, line);
1613
68.0k
  condexit = cond(ls);  /* read condition (inside scope block) */
1614
68.0k
  leaveblock(fs);  /* finish scope */
1615
68.0k
  if (bl2.upval) {  /* upvalues? */
1616
7.14k
    int exit = luaK_jump(fs);  /* normal exit must jump over fix */
1617
7.14k
    luaK_patchtohere(fs, condexit);  /* repetition must close upvalues */
1618
7.14k
    luaK_codeABC(fs, OP_CLOSE, reglevel(fs, bl2.nactvar), 0, 0);
1619
7.14k
    condexit = luaK_jump(fs);  /* repeat after closing upvalues */
1620
7.14k
    luaK_patchtohere(fs, exit);  /* normal exit comes to here */
1621
7.14k
  }
1622
68.0k
  luaK_patchlist(fs, condexit, repeat_init);  /* close the loop */
1623
68.0k
  leaveblock(fs);  /* finish loop */
1624
68.0k
}
1625
1626
1627
/*
1628
** Read an expression and generate code to put its results in next
1629
** stack slot.
1630
**
1631
*/
1632
76.0k
static void exp1 (LexState *ls) {
1633
76.0k
  expdesc e;
1634
76.0k
  expr(ls, &e);
1635
76.0k
  luaK_exp2nextreg(ls->fs, &e);
1636
76.0k
  lua_assert(e.k == VNONRELOC);
1637
76.0k
}
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
94.0k
static void fixforjump (FuncState *fs, int pc, int dest, int back) {
1646
94.0k
  Instruction *jmp = &fs->f->code[pc];
1647
94.0k
  int offset = dest - (pc + 1);
1648
94.0k
  if (back)
1649
47.0k
    offset = -offset;
1650
94.0k
  if (l_unlikely(offset > MAXARG_Bx))
1651
26
    luaX_syntaxerror(fs->ls, "control structure too long");
1652
94.0k
  SETARG_Bx(*jmp, offset);
1653
94.0k
}
1654
1655
1656
/*
1657
** Generate code for a 'for' loop.
1658
*/
1659
67.0k
static void forbody (LexState *ls, int base, int line, int nvars, int isgen) {
1660
  /* forbody -> DO block */
1661
67.0k
  static const OpCode forprep[2] = {OP_FORPREP, OP_TFORPREP};
1662
67.0k
  static const OpCode forloop[2] = {OP_FORLOOP, OP_TFORLOOP};
1663
67.0k
  BlockCnt bl;
1664
67.0k
  FuncState *fs = ls->fs;
1665
67.0k
  int prep, endfor;
1666
67.0k
  checknext(ls, TK_DO);
1667
67.0k
  prep = luaK_codeABx(fs, forprep[isgen], base, 0);
1668
67.0k
  fs->freereg--;  /* both 'forprep' remove one register from the stack */
1669
67.0k
  enterblock(fs, &bl, 0);  /* scope for declared variables */
1670
67.0k
  adjustlocalvars(ls, nvars);
1671
67.0k
  luaK_reserveregs(fs, nvars);
1672
67.0k
  block(ls);
1673
67.0k
  leaveblock(fs);  /* end of scope for declared variables */
1674
67.0k
  fixforjump(fs, prep, luaK_getlabel(fs), 0);
1675
67.0k
  if (isgen) {  /* generic for? */
1676
28.2k
    luaK_codeABC(fs, OP_TFORCALL, base, 0, nvars);
1677
28.2k
    luaK_fixline(fs, line);
1678
28.2k
  }
1679
67.0k
  endfor = luaK_codeABx(fs, forloop[isgen], base, 0);
1680
67.0k
  fixforjump(fs, endfor, prep + 1, 1);
1681
67.0k
  luaK_fixline(fs, line);
1682
67.0k
}
1683
1684
1685
36.5k
static void fornum (LexState *ls, TString *varname, int line) {
1686
  /* fornum -> NAME = exp,exp[,exp] forbody */
1687
36.5k
  FuncState *fs = ls->fs;
1688
36.5k
  int base = fs->freereg;
1689
36.5k
  new_localvarliteral(ls, "(for state)");
1690
36.5k
  new_localvarliteral(ls, "(for state)");
1691
36.5k
  new_varkind(ls, varname, RDKCONST);  /* control variable */
1692
36.5k
  checknext(ls, '=');
1693
36.5k
  exp1(ls);  /* initial value */
1694
36.5k
  checknext(ls, ',');
1695
36.5k
  exp1(ls);  /* limit */
1696
36.5k
  if (testnext(ls, ','))
1697
7.18k
    exp1(ls);  /* optional step */
1698
29.3k
  else {  /* default step = 1 */
1699
29.3k
    luaK_int(fs, fs->freereg, 1);
1700
29.3k
    luaK_reserveregs(fs, 1);
1701
29.3k
  }
1702
36.5k
  adjustlocalvars(ls, 2);  /* start scope for internal variables */
1703
36.5k
  forbody(ls, base, line, 1, 0);
1704
36.5k
}
1705
1706
1707
54.2k
static void forlist (LexState *ls, TString *indexname) {
1708
  /* forlist -> NAME {,NAME} IN explist forbody */
1709
54.2k
  FuncState *fs = ls->fs;
1710
54.2k
  expdesc e;
1711
54.2k
  int nvars = 4;  /* function, state, closing, control */
1712
54.2k
  int line;
1713
54.2k
  int base = fs->freereg;
1714
  /* create internal variables */
1715
54.2k
  new_localvarliteral(ls, "(for state)");  /* iterator function */
1716
54.2k
  new_localvarliteral(ls, "(for state)");  /* state */
1717
54.2k
  new_localvarliteral(ls, "(for state)");  /* closing var. (after swap) */
1718
54.2k
  new_varkind(ls, indexname, RDKCONST);  /* control variable */
1719
  /* other declared variables */
1720
129k
  while (testnext(ls, ',')) {
1721
75.6k
    new_localvar(ls, str_checkname(ls));
1722
75.6k
    nvars++;
1723
75.6k
  }
1724
54.2k
  checknext(ls, TK_IN);
1725
54.2k
  line = ls->linenumber;
1726
54.2k
  adjust_assign(ls, 4, explist(ls, &e), &e);
1727
54.2k
  adjustlocalvars(ls, 3);  /* start scope for internal variables */
1728
54.2k
  marktobeclosed(fs);  /* last internal var. must be closed */
1729
54.2k
  luaK_checkstack(fs, 2);  /* extra space to call iterator */
1730
54.2k
  forbody(ls, base, line, nvars - 3, 1);
1731
54.2k
}
1732
1733
1734
122k
static void forstat (LexState *ls, int line) {
1735
  /* forstat -> FOR (fornum | forlist) END */
1736
122k
  FuncState *fs = ls->fs;
1737
122k
  TString *varname;
1738
122k
  BlockCnt bl;
1739
122k
  enterblock(fs, &bl, 1);  /* scope for loop and control variables */
1740
122k
  luaX_next(ls);  /* skip 'for' */
1741
122k
  varname = str_checkname(ls);  /* first variable name */
1742
122k
  switch (ls->t.token) {
1743
36.5k
    case '=': fornum(ls, varname, line); break;
1744
54.2k
    case ',': case TK_IN: forlist(ls, varname); break;
1745
9.13k
    default: luaX_syntaxerror(ls, "'=' or 'in' expected");
1746
122k
  }
1747
47.0k
  check_match(ls, TK_END, TK_FOR, line);
1748
47.0k
  leaveblock(fs);  /* loop scope ('break' jumps to this point) */
1749
47.0k
}
1750
1751
1752
205k
static void test_then_block (LexState *ls, int *escapelist) {
1753
  /* test_then_block -> [IF | ELSEIF] cond THEN block */
1754
205k
  FuncState *fs = ls->fs;
1755
205k
  int condtrue;
1756
205k
  luaX_next(ls);  /* skip IF or ELSEIF */
1757
205k
  condtrue = cond(ls);  /* read condition */
1758
205k
  checknext(ls, TK_THEN);
1759
205k
  block(ls);  /* 'then' part */
1760
205k
  if (ls->t.token == TK_ELSE ||
1761
184k
      ls->t.token == TK_ELSEIF)  /* followed by 'else'/'elseif'? */
1762
33.0k
    luaK_concat(fs, escapelist, luaK_jump(fs));  /* must jump over it */
1763
205k
  luaK_patchtohere(fs, condtrue);
1764
205k
}
1765
1766
1767
187k
static void ifstat (LexState *ls, int line) {
1768
  /* ifstat -> IF cond THEN block {ELSEIF cond THEN block} [ELSE block] END */
1769
187k
  FuncState *fs = ls->fs;
1770
187k
  int escapelist = NO_JUMP;  /* exit list for finished parts */
1771
187k
  test_then_block(ls, &escapelist);  /* IF cond THEN block */
1772
205k
  while (ls->t.token == TK_ELSEIF)
1773
18.2k
    test_then_block(ls, &escapelist);  /* ELSEIF cond THEN block */
1774
187k
  if (testnext(ls, TK_ELSE))
1775
14.8k
    block(ls);  /* 'else' part */
1776
187k
  check_match(ls, TK_END, TK_IF, line);
1777
187k
  luaK_patchtohere(fs, escapelist);  /* patch escape list to 'if' end */
1778
187k
}
1779
1780
1781
53.7k
static void localfunc (LexState *ls) {
1782
53.7k
  expdesc b;
1783
53.7k
  FuncState *fs = ls->fs;
1784
53.7k
  int fvar = fs->nactvar;  /* function's variable index */
1785
53.7k
  new_localvar(ls, str_checkname(ls));  /* new local variable */
1786
53.7k
  adjustlocalvars(ls, 1);  /* enter its scope */
1787
53.7k
  body(ls, &b, 0, ls->linenumber);  /* function created in next register */
1788
  /* debug information will only see the variable after this point! */
1789
53.7k
  localdebuginfo(fs, fvar)->startpc = fs->pc;
1790
53.7k
}
1791
1792
1793
3.53M
static lu_byte getvarattribute (LexState *ls, lu_byte df) {
1794
  /* attrib -> ['<' NAME '>'] */
1795
3.53M
  if (testnext(ls, '<')) {
1796
49.9k
    TString *ts = str_checkname(ls);
1797
49.9k
    const char *attr = getstr(ts);
1798
49.9k
    checknext(ls, '>');
1799
49.9k
    if (strcmp(attr, "const") == 0)
1800
25.1k
      return RDKCONST;  /* read-only variable */
1801
24.8k
    else if (strcmp(attr, "close") == 0)
1802
14.6k
      return RDKTOCLOSE;  /* to-be-closed variable */
1803
10.2k
    else
1804
10.2k
      luaK_semerror(ls, "unknown attribute '%s'", attr);
1805
49.9k
  }
1806
3.48M
  return df;  /* return default value */
1807
3.53M
}
1808
1809
1810
532k
static void checktoclose (FuncState *fs, int level) {
1811
532k
  if (level != -1) {  /* is there a to-be-closed variable? */
1812
11.6k
    marktobeclosed(fs);
1813
11.6k
    luaK_codeABC(fs, OP_TBC, reglevel(fs, level), 0, 0);
1814
11.6k
  }
1815
532k
}
1816
1817
1818
572k
static void localstat (LexState *ls) {
1819
  /* stat -> LOCAL NAME attrib { ',' NAME attrib } ['=' explist] */
1820
572k
  FuncState *fs = ls->fs;
1821
572k
  int toclose = -1;  /* index of to-be-closed variable (if any) */
1822
572k
  Vardesc *var;  /* last variable */
1823
572k
  int vidx;  /* index of last variable */
1824
572k
  int nvars = 0;
1825
572k
  int nexps;
1826
572k
  expdesc e;
1827
  /* get prefixed attribute (if any); default is regular local variable */
1828
572k
  lu_byte defkind = getvarattribute(ls, VDKREG);
1829
2.00M
  do {  /* for each variable */
1830
2.00M
    TString *vname = str_checkname(ls);  /* get its name */
1831
2.00M
    lu_byte kind = getvarattribute(ls, defkind);  /* postfixed attribute */
1832
2.00M
    vidx = new_varkind(ls, vname, kind);  /* predeclare it */
1833
2.00M
    if (kind == RDKTOCLOSE) {  /* to-be-closed? */
1834
11.6k
      if (toclose != -1)  /* one already present? */
1835
12
        luaK_semerror(ls, "multiple to-be-closed variables in local list");
1836
11.6k
      toclose = fs->nactvar + nvars;
1837
11.6k
    }
1838
2.00M
    nvars++;
1839
2.00M
  } while (testnext(ls, ','));
1840
572k
  if (testnext(ls, '='))  /* initialization? */
1841
247k
    nexps = explist(ls, &e);
1842
324k
  else {
1843
324k
    e.k = VVOID;
1844
324k
    nexps = 0;
1845
324k
  }
1846
572k
  var = getlocalvardesc(fs, vidx);  /* retrieve last variable */
1847
572k
  if (nvars == nexps &&  /* no adjustments? */
1848
201k
      var->vd.kind == RDKCONST &&  /* last variable is const? */
1849
16.0k
      luaK_exp2const(fs, &e, &var->k)) {  /* compile-time constant? */
1850
12.2k
    var->vd.kind = RDKCTC;  /* variable is a compile-time constant */
1851
12.2k
    adjustlocalvars(ls, nvars - 1);  /* exclude last variable */
1852
12.2k
    fs->nactvar++;  /* but count it */
1853
12.2k
  }
1854
559k
  else {
1855
559k
    adjust_assign(ls, nvars, nexps, &e);
1856
559k
    adjustlocalvars(ls, nvars);
1857
559k
  }
1858
572k
  checktoclose(fs, toclose);
1859
572k
}
1860
1861
1862
982k
static lu_byte getglobalattribute (LexState *ls, lu_byte df) {
1863
982k
  lu_byte kind = getvarattribute(ls, df);
1864
982k
  switch (kind) {
1865
2.86k
    case RDKTOCLOSE:
1866
2.86k
      luaK_semerror(ls, "global variables cannot be to-be-closed");
1867
0
      return kind;  /* to avoid warnings */
1868
5.30k
    case RDKCONST:
1869
5.30k
      return GDKCONST;  /* adjust kind for global variable */
1870
972k
    default:
1871
972k
      return kind;
1872
982k
  }
1873
982k
}
1874
1875
1876
61.3k
static void checkglobal (LexState *ls, TString *varname, int line) {
1877
61.3k
  FuncState *fs = ls->fs;
1878
61.3k
  expdesc var;
1879
61.3k
  int k;
1880
61.3k
  buildglobal(ls, varname, &var);  /* create global variable in 'var' */
1881
61.3k
  k = var.u.ind.keystr;  /* index of global name in 'k' */
1882
61.3k
  luaK_codecheckglobal(fs, &var, k, line);
1883
61.3k
}
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
73.5k
                        int line) {
1896
73.5k
  if (n == nvars) {  /* traversed all variables? */
1897
6.71k
    expdesc e;
1898
6.71k
    int nexps = explist(ls, &e);  /* read list of expressions */
1899
6.71k
    adjust_assign(ls, nvars, nexps, &e);
1900
6.71k
  }
1901
66.8k
  else {  /* handle variable 'n' */
1902
66.8k
    FuncState *fs = ls->fs;
1903
66.8k
    expdesc var;
1904
66.8k
    TString *varname = getlocalvardesc(fs, firstidx + n)->vd.name;
1905
66.8k
    buildglobal(ls, varname, &var);  /* create global variable in 'var' */
1906
66.8k
    enterlevel(ls);  /* control recursion depth */
1907
66.8k
    initglobal(ls, nvars, firstidx, n + 1, line);
1908
66.8k
    leavelevel(ls);
1909
66.8k
    checkglobal(ls, varname, line);
1910
66.8k
    storevartop(fs, &var);
1911
66.8k
  }
1912
73.5k
}
1913
1914
1915
20.8k
static void globalnames (LexState *ls, lu_byte defkind) {
1916
20.8k
  FuncState *fs = ls->fs;
1917
20.8k
  int nvars = 0;
1918
20.8k
  int lastidx;  /* index of last registered variable */
1919
955k
  do {  /* for each name */
1920
955k
    TString *vname = str_checkname(ls);
1921
955k
    lu_byte kind = getglobalattribute(ls, defkind);
1922
955k
    lastidx = new_varkind(ls, vname, kind);
1923
955k
    nvars++;
1924
955k
  } while (testnext(ls, ','));
1925
20.8k
  if (testnext(ls, '='))  /* initialization? */
1926
6.72k
    initglobal(ls, nvars, lastidx - nvars + 1, 0, ls->linenumber);
1927
20.8k
  fs->nactvar = cast_short(fs->nactvar + nvars);  /* activate declaration */
1928
20.8k
}
1929
1930
1931
28.8k
static void globalstat (LexState *ls) {
1932
  /* globalstat -> (GLOBAL) attrib '*'
1933
     globalstat -> (GLOBAL) attrib NAME attrib {',' NAME attrib} */
1934
28.8k
  FuncState *fs = ls->fs;
1935
  /* get prefixed attribute (if any); default is regular global variable */
1936
28.8k
  lu_byte defkind = getglobalattribute(ls, GDKREG);
1937
28.8k
  if (!testnext(ls, '*'))
1938
20.8k
    globalnames(ls, defkind);
1939
7.95k
  else {
1940
    /* use NULL as name to represent '*' entries */
1941
7.95k
    new_varkind(ls, NULL, defkind);
1942
7.95k
    fs->nactvar++;  /* activate declaration */
1943
7.95k
  }
1944
28.8k
}
1945
1946
1947
6.17k
static void globalfunc (LexState *ls, int line) {
1948
  /* globalfunc -> (GLOBAL FUNCTION) NAME body */
1949
6.17k
  expdesc var, b;
1950
6.17k
  FuncState *fs = ls->fs;
1951
6.17k
  TString *fname = str_checkname(ls);
1952
6.17k
  new_varkind(ls, fname, GDKREG);  /* declare global variable */
1953
6.17k
  fs->nactvar++;  /* enter its scope */
1954
6.17k
  buildglobal(ls, fname, &var);
1955
6.17k
  body(ls, &b, 0, ls->linenumber);  /* compile and return closure in 'b' */
1956
6.17k
  checkglobal(ls, fname, line);
1957
6.17k
  luaK_storevar(fs, &var, &b);
1958
6.17k
  luaK_fixline(fs, line);  /* definition "happens" in the first line */
1959
6.17k
}
1960
1961
1962
34.9k
static void globalstatfunc (LexState *ls, int line) {
1963
  /* stat -> GLOBAL globalfunc | GLOBAL globalstat */
1964
34.9k
  luaX_next(ls);  /* skip 'global' */
1965
34.9k
  if (testnext(ls, TK_FUNCTION))
1966
6.17k
    globalfunc(ls, line);
1967
28.8k
  else
1968
28.8k
    globalstat(ls);
1969
34.9k
}
1970
1971
1972
99.3k
static int funcname (LexState *ls, expdesc *v) {
1973
  /* funcname -> NAME {fieldsel} [':' NAME] */
1974
99.3k
  int ismethod = 0;
1975
99.3k
  singlevar(ls, v);
1976
134k
  while (ls->t.token == '.')
1977
35.1k
    fieldsel(ls, v);
1978
99.3k
  if (ls->t.token == ':') {
1979
14.7k
    ismethod = 1;
1980
14.7k
    fieldsel(ls, v);
1981
14.7k
  }
1982
99.3k
  return ismethod;
1983
99.3k
}
1984
1985
1986
99.3k
static void funcstat (LexState *ls, int line) {
1987
  /* funcstat -> FUNCTION funcname body */
1988
99.3k
  int ismethod;
1989
99.3k
  expdesc v, b;
1990
99.3k
  luaX_next(ls);  /* skip FUNCTION */
1991
99.3k
  ismethod = funcname(ls, &v);
1992
99.3k
  check_readonly(ls, &v);
1993
99.3k
  body(ls, &b, ismethod, line);
1994
99.3k
  luaK_storevar(ls->fs, &v, &b);
1995
99.3k
  luaK_fixline(ls->fs, line);  /* definition "happens" in the first line */
1996
99.3k
}
1997
1998
1999
6.20M
static void exprstat (LexState *ls) {
2000
  /* stat -> func | assignment */
2001
6.20M
  FuncState *fs = ls->fs;
2002
6.20M
  struct LHS_assign v;
2003
6.20M
  suffixedexp(ls, &v.v);
2004
6.20M
  if (ls->t.token == '=' || ls->t.token == ',') { /* stat -> assignment ? */
2005
2.27M
    v.prev = NULL;
2006
2.27M
    restassign(ls, &v, 1);
2007
2.27M
  }
2008
3.93M
  else {  /* stat -> func */
2009
3.93M
    Instruction *inst;
2010
3.93M
    check_condition(ls, v.v.k == VCALL, "syntax error");
2011
3.18M
    inst = &getinstruction(fs, &v.v);
2012
3.18M
    SETARG_C(*inst, 1);  /* call statement uses no results */
2013
3.18M
  }
2014
6.20M
}
2015
2016
2017
264k
static void retstat (LexState *ls) {
2018
  /* stat -> RETURN [explist] [';'] */
2019
264k
  FuncState *fs = ls->fs;
2020
264k
  expdesc e;
2021
264k
  int nret;  /* number of values being returned */
2022
264k
  int first = luaY_nvarstack(fs);  /* first slot to be returned */
2023
264k
  if (block_follow(ls, 1) || ls->t.token == ';')
2024
65.9k
    nret = 0;  /* return no values */
2025
198k
  else {
2026
198k
    nret = explist(ls, &e);  /* optional return values */
2027
198k
    if (hasmultret(e.k)) {
2028
44.0k
      luaK_setmultret(fs, &e);
2029
44.0k
      if (e.k == VCALL && nret == 1 && !fs->bl->insidetbc) {  /* tail call? */
2030
28.5k
        SET_OPCODE(getinstruction(fs,&e), OP_TAILCALL);
2031
28.5k
        lua_assert(GETARG_A(getinstruction(fs,&e)) == luaY_nvarstack(fs));
2032
28.5k
      }
2033
44.0k
      nret = LUA_MULTRET;  /* return all values */
2034
44.0k
    }
2035
154k
    else {
2036
154k
      if (nret == 1)  /* only one single value? */
2037
136k
        first = luaK_exp2anyreg(fs, &e);  /* can use original slot */
2038
17.3k
      else {  /* values must go to the top of the stack */
2039
17.3k
        luaK_exp2nextreg(fs, &e);
2040
17.3k
        lua_assert(nret == fs->freereg - first);
2041
17.3k
      }
2042
154k
    }
2043
198k
  }
2044
264k
  luaK_ret(fs, first, nret);
2045
256k
  testnext(ls, ';');  /* skip optional semicolon */
2046
256k
}
2047
2048
2049
9.55M
static void statement (LexState *ls) {
2050
9.55M
  int line = ls->linenumber;  /* may be needed for error messages */
2051
9.55M
  enterlevel(ls);
2052
9.55M
  switch (ls->t.token) {
2053
1.24M
    case ';': {  /* stat -> ';' (empty statement) */
2054
1.24M
      luaX_next(ls);  /* skip ';' */
2055
1.24M
      break;
2056
0
    }
2057
187k
    case TK_IF: {  /* stat -> ifstat */
2058
187k
      ifstat(ls, line);
2059
187k
      break;
2060
0
    }
2061
23.5k
    case TK_WHILE: {  /* stat -> whilestat */
2062
23.5k
      whilestat(ls, line);
2063
23.5k
      break;
2064
0
    }
2065
73.0k
    case TK_DO: {  /* stat -> DO block END */
2066
73.0k
      luaX_next(ls);  /* skip DO */
2067
73.0k
      block(ls);
2068
73.0k
      check_match(ls, TK_END, TK_DO, line);
2069
73.0k
      break;
2070
0
    }
2071
122k
    case TK_FOR: {  /* stat -> forstat */
2072
122k
      forstat(ls, line);
2073
122k
      break;
2074
0
    }
2075
68.0k
    case TK_REPEAT: {  /* stat -> repeatstat */
2076
68.0k
      repeatstat(ls, line);
2077
68.0k
      break;
2078
0
    }
2079
99.3k
    case TK_FUNCTION: {  /* stat -> funcstat */
2080
99.3k
      funcstat(ls, line);
2081
99.3k
      break;
2082
0
    }
2083
625k
    case TK_LOCAL: {  /* stat -> localstat */
2084
625k
      luaX_next(ls);  /* skip LOCAL */
2085
625k
      if (testnext(ls, TK_FUNCTION))  /* local function? */
2086
53.7k
        localfunc(ls);
2087
572k
      else
2088
572k
        localstat(ls);
2089
625k
      break;
2090
0
    }
2091
0
    case TK_GLOBAL: {  /* stat -> globalstatfunc */
2092
0
      globalstatfunc(ls, line);
2093
0
      break;
2094
0
    }
2095
91.8k
    case TK_DBCOLON: {  /* stat -> label */
2096
91.8k
      luaX_next(ls);  /* skip double colon */
2097
91.8k
      labelstat(ls, str_checkname(ls), line);
2098
91.8k
      break;
2099
0
    }
2100
264k
    case TK_RETURN: {  /* stat -> retstat */
2101
264k
      luaX_next(ls);  /* skip RETURN */
2102
264k
      retstat(ls);
2103
264k
      break;
2104
0
    }
2105
104k
    case TK_BREAK: {  /* stat -> breakstat */
2106
104k
      breakstat(ls, line);
2107
104k
      break;
2108
0
    }
2109
393k
    case TK_GOTO: {  /* stat -> 'goto' NAME */
2110
393k
      luaX_next(ls);  /* skip 'goto' */
2111
393k
      gotostat(ls, line);
2112
393k
      break;
2113
0
    }
2114
0
#if defined(LUA_COMPAT_GLOBAL)
2115
5.35M
    case TK_NAME: {
2116
      /* compatibility code to parse global keyword when "global"
2117
         is not reserved */
2118
5.35M
      if (ls->t.seminfo.ts == ls->glbn) {  /* current = "global"? */
2119
41.4k
        int lk = luaX_lookahead(ls);
2120
41.4k
        if (lk == '<' || lk == TK_NAME || lk == '*' || lk == TK_FUNCTION) {
2121
          /* 'global <attrib>' or 'global name' or 'global *' or
2122
             'global function' */
2123
34.9k
          globalstatfunc(ls, line);
2124
34.9k
          break;
2125
34.9k
        }
2126
41.4k
      }  /* else... */
2127
5.35M
    }
2128
5.32M
#endif
2129
    /* FALLTHROUGH */
2130
6.20M
    default: {  /* stat -> func | assignment */
2131
6.20M
      exprstat(ls);
2132
6.20M
      break;
2133
5.35M
    }
2134
9.55M
  }
2135
5.95M
  lua_assert(ls->fs->f->maxstacksize >= ls->fs->freereg &&
2136
5.95M
             ls->fs->freereg >= luaY_nvarstack(ls->fs));
2137
5.95M
  ls->fs->freereg = luaY_nvarstack(ls->fs);  /* free registers */
2138
5.95M
  leavelevel(ls);
2139
5.95M
}
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
4.74M
static void mainfunc (LexState *ls, FuncState *fs) {
2151
4.74M
  BlockCnt bl;
2152
4.74M
  Upvaldesc *env;
2153
4.74M
  open_func(ls, fs, &bl);
2154
4.74M
  setvararg(fs);  /* main function is always vararg */
2155
4.74M
  env = allocupvalue(fs);  /* ...set environment upvalue */
2156
4.74M
  env->instack = 1;
2157
4.74M
  env->idx = 0;
2158
4.74M
  env->kind = VDKREG;
2159
4.74M
  env->name = ls->envn;
2160
4.74M
  luaC_objbarrier(ls->L, fs->f, env->name);
2161
4.74M
  luaX_next(ls);  /* read first token */
2162
4.74M
  statlist(ls);  /* parse main body */
2163
4.74M
  check(ls, TK_EOS);
2164
4.74M
  close_func(ls);
2165
4.74M
}
2166
2167
2168
LClosure *luaY_parser (lua_State *L, ZIO *z, Mbuffer *buff,
2169
4.74M
                       Dyndata *dyd, const char *name, int firstchar) {
2170
4.74M
  LexState lexstate;
2171
4.74M
  FuncState funcstate;
2172
4.74M
  LClosure *cl = luaF_newLclosure(L, 1);  /* create main closure */
2173
4.74M
  setclLvalue2s(L, L->top.p, cl);  /* anchor it (to avoid being collected) */
2174
4.74M
  luaD_inctop(L);
2175
4.74M
  lexstate.h = luaH_new(L);  /* create table for scanner */
2176
4.74M
  sethvalue2s(L, L->top.p, lexstate.h);  /* anchor it */
2177
4.74M
  luaD_inctop(L);
2178
4.74M
  funcstate.f = cl->p = luaF_newproto(L);
2179
4.74M
  luaC_objbarrier(L, cl, cl->p);
2180
4.74M
  funcstate.f->source = luaS_new(L, name);  /* create and anchor TString */
2181
4.74M
  luaC_objbarrier(L, funcstate.f, funcstate.f->source);
2182
4.74M
  lexstate.buff = buff;
2183
4.74M
  lexstate.dyd = dyd;
2184
4.74M
  dyd->actvar.n = dyd->gt.n = dyd->label.n = 0;
2185
4.74M
  luaX_setinput(L, &lexstate, z, funcstate.f->source, firstchar);
2186
4.74M
  mainfunc(&lexstate, &funcstate);
2187
4.74M
  lua_assert(!funcstate.prev && funcstate.nups == 1 && !lexstate.fs);
2188
  /* all scopes should be correctly finished */
2189
4.74M
  lua_assert(dyd->actvar.n == 0 && dyd->gt.n == 0 && dyd->label.n == 0);
2190
1.40M
  L->top.p--;  /* remove scanner's table */
2191
1.40M
  return cl;  /* closure is on the stack, too */
2192
1.40M
}
2193