| 1110 | |
| 1111 | |
| 1112 | static void forstat (LexState *ls, int line) { |
| 1113 | /* forstat -> FOR (fornum | forlist) END */ |
| 1114 | FuncState *fs = ls->fs; |
| 1115 | TString *varname; |
| 1116 | BlockCnt bl; |
| 1117 | enterblock(fs, &bl, 1); /* scope for loop and control variables */ |
| 1118 | luaX_next(ls); /* skip `for' */ |
| 1119 | varname = str_checkname(ls); /* first variable name */ |
| 1120 | switch (ls->t.token) { |
| 1121 | case '=': fornum(ls, varname, line); break; |
| 1122 | case ',': case TK_IN: forlist(ls, varname); break; |
| 1123 | default: luaX_syntaxerror(ls, LUA_QL("=") " or " LUA_QL("in") " expected"); |
| 1124 | } |
| 1125 | check_match(ls, TK_END, TK_FOR, line); |
| 1126 | leaveblock(fs); /* loop scope (`break' jumps to this point) */ |
| 1127 | } |
| 1128 | |
| 1129 | |
| 1130 | static int test_then_block (LexState *ls) { |
| 1131 | /* test_then_block -> [IF | ELSEIF] cond THEN block */ |
| 1132 | int condexit; |
| 1133 | luaX_next(ls); /* skip IF or ELSEIF */ |
| 1134 | condexit = cond(ls); |
| 1135 | checknext(ls, TK_THEN); |
| 1136 | block(ls); /* `then' part */ |
| 1137 | return condexit; |
| 1138 | } |
| 1139 | |
| 1140 | |
| 1141 | static void ifstat (LexState *ls, int line) { |
| 1142 | /* ifstat -> IF cond THEN block {ELSEIF cond THEN block} [ELSE block] END */ |
| 1143 | FuncState *fs = ls->fs; |
| 1144 | int flist; |
| 1145 | int escapelist = NO_JUMP; |
| 1146 | flist = test_then_block(ls); /* IF cond THEN block */ |
| 1147 | while (ls->t.token == TK_ELSEIF) { |
| 1148 | luaK_concat(fs, &escapelist, luaK_jump(fs)); |
| 1149 | luaK_patchtohere(fs, flist); |
| 1150 | flist = test_then_block(ls); /* ELSEIF cond THEN block */ |
| 1151 | } |
| 1152 | if (ls->t.token == TK_ELSE) { |
| 1153 | luaK_concat(fs, &escapelist, luaK_jump(fs)); |
| 1154 | luaK_patchtohere(fs, flist); |
| 1155 | luaX_next(ls); /* skip ELSE (after patch, for correct line info) */ |
| 1156 | block(ls); /* `else' part */ |
| 1157 | } |
| 1158 | else |
| 1159 | luaK_concat(fs, &escapelist, flist); |
| 1160 | luaK_patchtohere(fs, escapelist); |
| 1161 | check_match(ls, TK_END, TK_IF, line); |
| 1162 | } |
| 1163 | |
| 1164 | |
| 1165 | static void localfunc (LexState *ls) { |
| 1166 | expdesc v, b; |
| 1167 | FuncState *fs = ls->fs; |
| 1168 | new_localvar(ls, str_checkname(ls), 0); |
| 1169 | init_exp(&v, VLOCAL, fs->freereg); |
no test coverage detected