Coverage Report

Created: 2025-12-11 06:28

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