Coverage Report

Created: 2026-01-25 07:04

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