blob: edc5db045110b23299ce7e8b4fce90043f6a100e [file] [log] [blame]
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001/* vi:set ts=8 sts=4 sw=4 noet:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * vim9instr.c: Dealing with instructions of a compiled function
12 */
13
14#define USING_FLOAT_STUFF
15#include "vim.h"
16
17#if defined(FEAT_EVAL) || defined(PROTO)
18
19// When not generating protos this is included in proto.h
20#ifdef PROTO
21# include "vim9.h"
22#endif
23
24
25/////////////////////////////////////////////////////////////////////
26// Following generate_ functions expect the caller to call ga_grow().
27
28#define RETURN_NULL_IF_SKIP(cctx) if (cctx->ctx_skip == SKIP_YES) return NULL
29#define RETURN_OK_IF_SKIP(cctx) if (cctx->ctx_skip == SKIP_YES) return OK
30
31/*
32 * Generate an instruction without arguments.
33 * Returns a pointer to the new instruction, NULL if failed.
34 */
35 isn_T *
36generate_instr(cctx_T *cctx, isntype_T isn_type)
37{
38 garray_T *instr = &cctx->ctx_instr;
39 isn_T *isn;
40
41 RETURN_NULL_IF_SKIP(cctx);
42 if (GA_GROW_FAILS(instr, 1))
43 return NULL;
44 isn = ((isn_T *)instr->ga_data) + instr->ga_len;
45 isn->isn_type = isn_type;
46 isn->isn_lnum = cctx->ctx_lnum + 1;
47 ++instr->ga_len;
48
49 return isn;
50}
51
52/*
53 * Generate an instruction without arguments.
54 * "drop" will be removed from the stack.
55 * Returns a pointer to the new instruction, NULL if failed.
56 */
57 isn_T *
58generate_instr_drop(cctx_T *cctx, isntype_T isn_type, int drop)
59{
Bram Moolenaardc7c3662021-12-20 15:04:29 +000060 RETURN_NULL_IF_SKIP(cctx);
Bram Moolenaar078a4612022-01-04 15:17:03 +000061 cctx->ctx_type_stack.ga_len -= drop;
Bram Moolenaardc7c3662021-12-20 15:04:29 +000062 return generate_instr(cctx, isn_type);
63}
64
65/*
Bram Moolenaar078a4612022-01-04 15:17:03 +000066 * Generate instruction "isn_type" and put "type" on the type stack,
67 * use "decl_type" for the declared type.
Bram Moolenaardc7c3662021-12-20 15:04:29 +000068 */
Yegappan Lakshmanan782b43d2022-01-08 18:43:40 +000069 static isn_T *
Bram Moolenaar078a4612022-01-04 15:17:03 +000070generate_instr_type2(
71 cctx_T *cctx,
72 isntype_T isn_type,
73 type_T *type,
74 type_T *decl_type)
Bram Moolenaardc7c3662021-12-20 15:04:29 +000075{
76 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +000077
78 if ((isn = generate_instr(cctx, isn_type)) == NULL)
79 return NULL;
80
Bram Moolenaar078a4612022-01-04 15:17:03 +000081 if (push_type_stack2(cctx, type == NULL ? &t_any : type,
82 decl_type == NULL ? &t_any : decl_type) == FAIL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +000083 return NULL;
Bram Moolenaardc7c3662021-12-20 15:04:29 +000084
85 return isn;
86}
87
88/*
Bram Moolenaar078a4612022-01-04 15:17:03 +000089 * Generate instruction "isn_type" and put "type" on the type stack.
90 * Uses "any" for the declared type, which works for constants. For declared
91 * variables use generate_instr_type2().
92 */
93 isn_T *
94generate_instr_type(cctx_T *cctx, isntype_T isn_type, type_T *type)
95{
96 return generate_instr_type2(cctx, isn_type, type, &t_any);
97}
98
99/*
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000100 * Generate an ISN_DEBUG instruction.
101 */
102 isn_T *
103generate_instr_debug(cctx_T *cctx)
104{
105 isn_T *isn;
106 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
107 + cctx->ctx_ufunc->uf_dfunc_idx;
108
109 if ((isn = generate_instr(cctx, ISN_DEBUG)) == NULL)
110 return NULL;
111 isn->isn_arg.debug.dbg_var_names_len = dfunc->df_var_names.ga_len;
112 isn->isn_arg.debug.dbg_break_lnum = cctx->ctx_prev_lnum;
113 return isn;
114}
115
116/*
117 * If type at "offset" isn't already VAR_STRING then generate ISN_2STRING.
118 * But only for simple types.
119 * When "tolerant" is TRUE convert most types to string, e.g. a List.
120 */
121 int
122may_generate_2STRING(int offset, int tolerant, cctx_T *cctx)
123{
124 isn_T *isn;
125 isntype_T isntype = ISN_2STRING;
Bram Moolenaar078a4612022-01-04 15:17:03 +0000126 type_T *type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000127
128 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaar078a4612022-01-04 15:17:03 +0000129 type = get_type_on_stack(cctx, -1 - offset);
130 switch (type->tt_type)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000131 {
132 // nothing to be done
133 case VAR_STRING: return OK;
134
135 // conversion possible
136 case VAR_SPECIAL:
137 case VAR_BOOL:
138 case VAR_NUMBER:
139 case VAR_FLOAT:
140 break;
141
142 // conversion possible (with runtime check)
143 case VAR_ANY:
144 case VAR_UNKNOWN:
145 isntype = ISN_2STRING_ANY;
146 break;
147
148 // conversion possible when tolerant
149 case VAR_LIST:
150 if (tolerant)
151 {
152 isntype = ISN_2STRING_ANY;
153 break;
154 }
155 // FALLTHROUGH
156
157 // conversion not possible
158 case VAR_VOID:
159 case VAR_BLOB:
160 case VAR_FUNC:
161 case VAR_PARTIAL:
162 case VAR_DICT:
163 case VAR_JOB:
164 case VAR_CHANNEL:
165 case VAR_INSTR:
Bram Moolenaar078a4612022-01-04 15:17:03 +0000166 to_string_error(type->tt_type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000167 return FAIL;
168 }
169
Bram Moolenaar078a4612022-01-04 15:17:03 +0000170 set_type_on_stack(cctx, &t_string, -1 - offset);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000171 if ((isn = generate_instr(cctx, isntype)) == NULL)
172 return FAIL;
173 isn->isn_arg.tostring.offset = offset;
174 isn->isn_arg.tostring.tolerant = tolerant;
175
176 return OK;
177}
178
179 static int
180check_number_or_float(vartype_T type1, vartype_T type2, char_u *op)
181{
Bram Moolenaar59618fe2021-12-21 12:32:17 +0000182 if (!((type1 == VAR_NUMBER || type1 == VAR_FLOAT
183 || type1 == VAR_ANY || type1 == VAR_UNKNOWN)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000184 && (type2 == VAR_NUMBER || type2 == VAR_FLOAT
Bram Moolenaar59618fe2021-12-21 12:32:17 +0000185 || type2 == VAR_ANY || type2 == VAR_UNKNOWN)))
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000186 {
187 if (*op == '+')
188 emsg(_(e_wrong_argument_type_for_plus));
189 else
190 semsg(_(e_char_requires_number_or_float_arguments), *op);
191 return FAIL;
192 }
193 return OK;
194}
195
196/*
197 * Generate instruction for "+". For a list this creates a new list.
198 */
199 int
200generate_add_instr(
201 cctx_T *cctx,
202 vartype_T vartype,
203 type_T *type1,
204 type_T *type2,
205 exprtype_T expr_type)
206{
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000207 isn_T *isn = generate_instr_drop(cctx,
208 vartype == VAR_NUMBER ? ISN_OPNR
209 : vartype == VAR_LIST ? ISN_ADDLIST
210 : vartype == VAR_BLOB ? ISN_ADDBLOB
211#ifdef FEAT_FLOAT
212 : vartype == VAR_FLOAT ? ISN_OPFLOAT
213#endif
214 : ISN_OPANY, 1);
215
216 if (vartype != VAR_LIST && vartype != VAR_BLOB
217 && type1->tt_type != VAR_ANY
Bram Moolenaar59618fe2021-12-21 12:32:17 +0000218 && type1->tt_type != VAR_UNKNOWN
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000219 && type2->tt_type != VAR_ANY
Bram Moolenaar59618fe2021-12-21 12:32:17 +0000220 && type2->tt_type != VAR_UNKNOWN
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000221 && check_number_or_float(
222 type1->tt_type, type2->tt_type, (char_u *)"+") == FAIL)
223 return FAIL;
224
225 if (isn != NULL)
226 {
227 if (isn->isn_type == ISN_ADDLIST)
228 isn->isn_arg.op.op_type = expr_type;
229 else
230 isn->isn_arg.op.op_type = EXPR_ADD;
231 }
232
233 // When concatenating two lists with different member types the member type
234 // becomes "any".
235 if (vartype == VAR_LIST
236 && type1->tt_type == VAR_LIST && type2->tt_type == VAR_LIST
237 && type1->tt_member != type2->tt_member)
Bram Moolenaar078a4612022-01-04 15:17:03 +0000238 set_type_on_stack(cctx, &t_list_any, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000239
240 return isn == NULL ? FAIL : OK;
241}
242
243/*
244 * Get the type to use for an instruction for an operation on "type1" and
245 * "type2". If they are matching use a type-specific instruction. Otherwise
246 * fall back to runtime type checking.
247 */
248 vartype_T
249operator_type(type_T *type1, type_T *type2)
250{
251 if (type1->tt_type == type2->tt_type
252 && (type1->tt_type == VAR_NUMBER
253 || type1->tt_type == VAR_LIST
254#ifdef FEAT_FLOAT
255 || type1->tt_type == VAR_FLOAT
256#endif
257 || type1->tt_type == VAR_BLOB))
258 return type1->tt_type;
259 return VAR_ANY;
260}
261
262/*
263 * Generate an instruction with two arguments. The instruction depends on the
264 * type of the arguments.
265 */
266 int
267generate_two_op(cctx_T *cctx, char_u *op)
268{
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000269 type_T *type1;
270 type_T *type2;
271 vartype_T vartype;
272 isn_T *isn;
273
274 RETURN_OK_IF_SKIP(cctx);
275
276 // Get the known type of the two items on the stack.
Bram Moolenaar078a4612022-01-04 15:17:03 +0000277 type1 = get_type_on_stack(cctx, 1);
278 type2 = get_type_on_stack(cctx, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000279 vartype = operator_type(type1, type2);
280
281 switch (*op)
282 {
283 case '+':
284 if (generate_add_instr(cctx, vartype, type1, type2,
285 EXPR_COPY) == FAIL)
286 return FAIL;
287 break;
288
289 case '-':
290 case '*':
291 case '/': if (check_number_or_float(type1->tt_type, type2->tt_type,
292 op) == FAIL)
293 return FAIL;
294 if (vartype == VAR_NUMBER)
295 isn = generate_instr_drop(cctx, ISN_OPNR, 1);
296#ifdef FEAT_FLOAT
297 else if (vartype == VAR_FLOAT)
298 isn = generate_instr_drop(cctx, ISN_OPFLOAT, 1);
299#endif
300 else
301 isn = generate_instr_drop(cctx, ISN_OPANY, 1);
302 if (isn != NULL)
303 isn->isn_arg.op.op_type = *op == '*'
304 ? EXPR_MULT : *op == '/'? EXPR_DIV : EXPR_SUB;
305 break;
306
307 case '%': if ((type1->tt_type != VAR_ANY
Bram Moolenaar59618fe2021-12-21 12:32:17 +0000308 && type1->tt_type != VAR_UNKNOWN
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000309 && type1->tt_type != VAR_NUMBER)
310 || (type2->tt_type != VAR_ANY
Bram Moolenaar59618fe2021-12-21 12:32:17 +0000311 && type2->tt_type != VAR_UNKNOWN
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000312 && type2->tt_type != VAR_NUMBER))
313 {
314 emsg(_(e_percent_requires_number_arguments));
315 return FAIL;
316 }
317 isn = generate_instr_drop(cctx,
318 vartype == VAR_NUMBER ? ISN_OPNR : ISN_OPANY, 1);
319 if (isn != NULL)
320 isn->isn_arg.op.op_type = EXPR_REM;
321 break;
322 }
323
324 // correct type of result
325 if (vartype == VAR_ANY)
326 {
327 type_T *type = &t_any;
328
329#ifdef FEAT_FLOAT
330 // float+number and number+float results in float
331 if ((type1->tt_type == VAR_NUMBER || type1->tt_type == VAR_FLOAT)
332 && (type2->tt_type == VAR_NUMBER || type2->tt_type == VAR_FLOAT))
333 type = &t_float;
334#endif
Bram Moolenaar078a4612022-01-04 15:17:03 +0000335 set_type_on_stack(cctx, type, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000336 }
337
338 return OK;
339}
340
341/*
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000342 * Get the instruction to use for comparing two values with specified types.
343 * Either "tv1" and "tv2" are passed or "type1" and "type2".
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000344 * Return ISN_DROP when failed.
345 */
346 static isntype_T
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000347get_compare_isn(
348 exprtype_T exprtype,
349 typval_T *tv1,
350 typval_T *tv2,
351 type_T *type1,
352 type_T *type2)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000353{
354 isntype_T isntype = ISN_DROP;
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000355 vartype_T vartype1 = tv1 != NULL ? tv1->v_type : type1->tt_type;
356 vartype_T vartype2 = tv2 != NULL ? tv2->v_type : type2->tt_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000357
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000358 if (vartype1 == VAR_UNKNOWN)
359 vartype1 = VAR_ANY;
360 if (vartype2 == VAR_UNKNOWN)
361 vartype2 = VAR_ANY;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000362
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000363 if (vartype1 == vartype2)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000364 {
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000365 switch (vartype1)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000366 {
367 case VAR_BOOL: isntype = ISN_COMPAREBOOL; break;
368 case VAR_SPECIAL: isntype = ISN_COMPARESPECIAL; break;
369 case VAR_NUMBER: isntype = ISN_COMPARENR; break;
370 case VAR_FLOAT: isntype = ISN_COMPAREFLOAT; break;
371 case VAR_STRING: isntype = ISN_COMPARESTRING; break;
372 case VAR_BLOB: isntype = ISN_COMPAREBLOB; break;
373 case VAR_LIST: isntype = ISN_COMPARELIST; break;
374 case VAR_DICT: isntype = ISN_COMPAREDICT; break;
375 case VAR_FUNC: isntype = ISN_COMPAREFUNC; break;
376 default: isntype = ISN_COMPAREANY; break;
377 }
378 }
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000379 else if (vartype1 == VAR_ANY || vartype2 == VAR_ANY
380 || ((vartype1 == VAR_NUMBER || vartype1 == VAR_FLOAT)
381 && (vartype2 == VAR_NUMBER || vartype2 == VAR_FLOAT))
382 || (vartype1 == VAR_FUNC && vartype2 == VAR_PARTIAL)
383 || (vartype1 == VAR_PARTIAL && vartype2 == VAR_FUNC))
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000384 isntype = ISN_COMPAREANY;
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000385 else if (vartype1 == VAR_SPECIAL || vartype2 == VAR_SPECIAL)
Bram Moolenaar7a222242022-03-01 19:23:24 +0000386 {
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000387 if ((vartype1 == VAR_SPECIAL
388 && (tv1 != NULL ? tv1->vval.v_number == VVAL_NONE
389 : type1 == &t_none)
390 && vartype2 != VAR_STRING)
391 || (vartype2 == VAR_SPECIAL
392 && (tv2 != NULL ? tv2->vval.v_number == VVAL_NONE
393 : type2 == &t_none)
394 && vartype1 != VAR_STRING))
395 {
396 semsg(_(e_cannot_compare_str_with_str),
397 vartype_name(vartype1), vartype_name(vartype2));
398 return ISN_DROP;
399 }
Bram Moolenaar05667812022-03-15 20:21:33 +0000400 // although comparing null with number, float or bool is not useful, we
401 // allow it
Bram Moolenaar7a222242022-03-01 19:23:24 +0000402 isntype = ISN_COMPARENULL;
403 }
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000404
405 if ((exprtype == EXPR_IS || exprtype == EXPR_ISNOT)
406 && (isntype == ISN_COMPAREBOOL
407 || isntype == ISN_COMPARESPECIAL
408 || isntype == ISN_COMPARENR
409 || isntype == ISN_COMPAREFLOAT))
410 {
411 semsg(_(e_cannot_use_str_with_str),
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000412 exprtype == EXPR_IS ? "is" : "isnot" , vartype_name(vartype1));
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000413 return ISN_DROP;
414 }
415 if (isntype == ISN_DROP
416 || ((exprtype != EXPR_EQUAL && exprtype != EXPR_NEQUAL
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000417 && (vartype1 == VAR_BOOL || vartype1 == VAR_SPECIAL
418 || vartype2 == VAR_BOOL || vartype2 == VAR_SPECIAL)))
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000419 || ((exprtype != EXPR_EQUAL && exprtype != EXPR_NEQUAL
Bram Moolenaar7a222242022-03-01 19:23:24 +0000420 && exprtype != EXPR_IS && exprtype != EXPR_ISNOT
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000421 && (vartype1 == VAR_BLOB || vartype2 == VAR_BLOB
422 || vartype1 == VAR_LIST || vartype2 == VAR_LIST))))
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000423 {
424 semsg(_(e_cannot_compare_str_with_str),
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000425 vartype_name(vartype1), vartype_name(vartype2));
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000426 return ISN_DROP;
427 }
428 return isntype;
429}
430
431 int
432check_compare_types(exprtype_T type, typval_T *tv1, typval_T *tv2)
433{
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000434 if (get_compare_isn(type, tv1, tv2, NULL, NULL) == ISN_DROP)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000435 return FAIL;
436 return OK;
437}
438
439/*
440 * Generate an ISN_COMPARE* instruction with a boolean result.
441 */
442 int
443generate_COMPARE(cctx_T *cctx, exprtype_T exprtype, int ic)
444{
445 isntype_T isntype;
446 isn_T *isn;
447 garray_T *stack = &cctx->ctx_type_stack;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000448
449 RETURN_OK_IF_SKIP(cctx);
450
451 // Get the known type of the two items on the stack. If they are matching
452 // use a type-specific instruction. Otherwise fall back to runtime type
453 // checking.
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000454 isntype = get_compare_isn(exprtype, NULL, NULL,
455 get_type_on_stack(cctx, 1), get_type_on_stack(cctx, 0));
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000456 if (isntype == ISN_DROP)
457 return FAIL;
458
459 if ((isn = generate_instr(cctx, isntype)) == NULL)
460 return FAIL;
461 isn->isn_arg.op.op_type = exprtype;
462 isn->isn_arg.op.op_ic = ic;
463
464 // takes two arguments, puts one bool back
465 if (stack->ga_len >= 2)
466 {
467 --stack->ga_len;
Bram Moolenaar078a4612022-01-04 15:17:03 +0000468 set_type_on_stack(cctx, &t_bool, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000469 }
470
471 return OK;
472}
473
474/*
LemonBoy372bcce2022-04-25 12:43:20 +0100475 * Generate an ISN_CONCAT instruction.
476 * "count" is the number of stack elements to join together and it must be
477 * greater or equal to one.
478 * The caller ensures all the "count" elements on the stack have the right type.
479 */
480 int
481generate_CONCAT(cctx_T *cctx, int count)
482{
483 isn_T *isn;
484 garray_T *stack = &cctx->ctx_type_stack;
485
486 RETURN_OK_IF_SKIP(cctx);
487
488 if (count < 1)
489 return FAIL;
490
491 if ((isn = generate_instr(cctx, ISN_CONCAT)) == NULL)
492 return FAIL;
493 isn->isn_arg.number = count;
494
495 // drop the argument types
496 stack->ga_len -= count - 1;
497
498 return OK;
499}
500
501/*
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000502 * Generate an ISN_2BOOL instruction.
503 * "offset" is the offset in the type stack.
504 */
505 int
506generate_2BOOL(cctx_T *cctx, int invert, int offset)
507{
508 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000509
510 RETURN_OK_IF_SKIP(cctx);
511 if ((isn = generate_instr(cctx, ISN_2BOOL)) == NULL)
512 return FAIL;
513 isn->isn_arg.tobool.invert = invert;
514 isn->isn_arg.tobool.offset = offset;
515
516 // type becomes bool
Bram Moolenaar078a4612022-01-04 15:17:03 +0000517 set_type_on_stack(cctx, &t_bool, -1 - offset);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000518
519 return OK;
520}
521
522/*
523 * Generate an ISN_COND2BOOL instruction.
524 */
525 int
526generate_COND2BOOL(cctx_T *cctx)
527{
528 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000529
530 RETURN_OK_IF_SKIP(cctx);
531 if ((isn = generate_instr(cctx, ISN_COND2BOOL)) == NULL)
532 return FAIL;
533
534 // type becomes bool
Bram Moolenaar078a4612022-01-04 15:17:03 +0000535 set_type_on_stack(cctx, &t_bool, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000536
537 return OK;
538}
539
540 int
541generate_TYPECHECK(
542 cctx_T *cctx,
543 type_T *expected,
544 int offset,
545 int argidx)
546{
547 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000548
549 RETURN_OK_IF_SKIP(cctx);
550 if ((isn = generate_instr(cctx, ISN_CHECKTYPE)) == NULL)
551 return FAIL;
552 isn->isn_arg.type.ct_type = alloc_type(expected);
553 isn->isn_arg.type.ct_off = (int8_T)offset;
554 isn->isn_arg.type.ct_arg_idx = (int8_T)argidx;
555
556 // type becomes expected
Bram Moolenaar078a4612022-01-04 15:17:03 +0000557 set_type_on_stack(cctx, expected, -1 - offset);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000558
559 return OK;
560}
561
562 int
563generate_SETTYPE(
564 cctx_T *cctx,
565 type_T *expected)
566{
567 isn_T *isn;
568
569 RETURN_OK_IF_SKIP(cctx);
570 if ((isn = generate_instr(cctx, ISN_SETTYPE)) == NULL)
571 return FAIL;
572 isn->isn_arg.type.ct_type = alloc_type(expected);
573 return OK;
574}
575
576/*
577 * Generate a PUSH instruction for "tv".
578 * "tv" will be consumed or cleared.
579 * Nothing happens if "tv" is NULL or of type VAR_UNKNOWN;
580 */
581 int
582generate_tv_PUSH(cctx_T *cctx, typval_T *tv)
583{
584 if (tv != NULL)
585 {
586 switch (tv->v_type)
587 {
588 case VAR_UNKNOWN:
589 break;
590 case VAR_BOOL:
591 generate_PUSHBOOL(cctx, tv->vval.v_number);
592 break;
593 case VAR_SPECIAL:
594 generate_PUSHSPEC(cctx, tv->vval.v_number);
595 break;
596 case VAR_NUMBER:
597 generate_PUSHNR(cctx, tv->vval.v_number);
598 break;
599#ifdef FEAT_FLOAT
600 case VAR_FLOAT:
601 generate_PUSHF(cctx, tv->vval.v_float);
602 break;
603#endif
604 case VAR_BLOB:
605 generate_PUSHBLOB(cctx, tv->vval.v_blob);
606 tv->vval.v_blob = NULL;
607 break;
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000608 case VAR_LIST:
609 if (tv->vval.v_list != NULL)
610 iemsg("non-empty list constant not supported");
Bram Moolenaarec15b1c2022-03-27 16:29:53 +0100611 generate_NEWLIST(cctx, 0, TRUE);
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000612 break;
613 case VAR_DICT:
614 if (tv->vval.v_dict != NULL)
615 iemsg("non-empty dict constant not supported");
Bram Moolenaarec15b1c2022-03-27 16:29:53 +0100616 generate_NEWDICT(cctx, 0, TRUE);
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000617 break;
618#ifdef FEAT_JOB_CHANNEL
619 case VAR_JOB:
620 if (tv->vval.v_job != NULL)
621 iemsg("non-null job constant not supported");
Bram Moolenaar397a87a2022-03-20 21:14:15 +0000622 generate_PUSHJOB(cctx);
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000623 break;
624 case VAR_CHANNEL:
625 if (tv->vval.v_channel != NULL)
626 iemsg("non-null channel constant not supported");
Bram Moolenaar397a87a2022-03-20 21:14:15 +0000627 generate_PUSHCHANNEL(cctx);
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000628 break;
629#endif
630 case VAR_FUNC:
631 if (tv->vval.v_string != NULL)
632 iemsg("non-null function constant not supported");
633 generate_PUSHFUNC(cctx, NULL, &t_func_unknown);
634 break;
635 case VAR_PARTIAL:
636 if (tv->vval.v_partial != NULL)
637 iemsg("non-null partial constant not supported");
638 if (generate_instr_type(cctx, ISN_NEWPARTIAL, &t_func_unknown)
639 == NULL)
640 return FAIL;
641 break;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000642 case VAR_STRING:
643 generate_PUSHS(cctx, &tv->vval.v_string);
644 tv->vval.v_string = NULL;
645 break;
646 default:
647 iemsg("constant type not supported");
648 clear_tv(tv);
649 return FAIL;
650 }
651 tv->v_type = VAR_UNKNOWN;
652 }
653 return OK;
654}
655
656/*
657 * Generate an ISN_PUSHNR instruction.
658 */
659 int
660generate_PUSHNR(cctx_T *cctx, varnumber_T number)
661{
662 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000663
664 RETURN_OK_IF_SKIP(cctx);
665 if ((isn = generate_instr_type(cctx, ISN_PUSHNR, &t_number)) == NULL)
666 return FAIL;
667 isn->isn_arg.number = number;
668
669 if (number == 0 || number == 1)
670 // A 0 or 1 number can also be used as a bool.
Bram Moolenaar078a4612022-01-04 15:17:03 +0000671 set_type_on_stack(cctx, &t_number_bool, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000672 return OK;
673}
674
675/*
676 * Generate an ISN_PUSHBOOL instruction.
677 */
678 int
679generate_PUSHBOOL(cctx_T *cctx, varnumber_T number)
680{
681 isn_T *isn;
682
683 RETURN_OK_IF_SKIP(cctx);
684 if ((isn = generate_instr_type(cctx, ISN_PUSHBOOL, &t_bool)) == NULL)
685 return FAIL;
686 isn->isn_arg.number = number;
687
688 return OK;
689}
690
691/*
692 * Generate an ISN_PUSHSPEC instruction.
693 */
694 int
695generate_PUSHSPEC(cctx_T *cctx, varnumber_T number)
696{
697 isn_T *isn;
698
699 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000700 if ((isn = generate_instr_type(cctx, ISN_PUSHSPEC,
701 number == VVAL_NULL ? &t_null : &t_none)) == NULL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000702 return FAIL;
703 isn->isn_arg.number = number;
704
705 return OK;
706}
707
708#if defined(FEAT_FLOAT) || defined(PROTO)
709/*
710 * Generate an ISN_PUSHF instruction.
711 */
712 int
713generate_PUSHF(cctx_T *cctx, float_T fnumber)
714{
715 isn_T *isn;
716
717 RETURN_OK_IF_SKIP(cctx);
718 if ((isn = generate_instr_type(cctx, ISN_PUSHF, &t_float)) == NULL)
719 return FAIL;
720 isn->isn_arg.fnumber = fnumber;
721
722 return OK;
723}
724#endif
725
726/*
727 * Generate an ISN_PUSHS instruction.
728 * Consumes "*str". When freed *str is set to NULL, unless "str" is NULL.
Bram Moolenaar0abc2872022-05-10 13:24:30 +0100729 * Note that if "str" is used in the instruction OK is returned and "*str" is
730 * not set to NULL.
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000731 */
732 int
733generate_PUSHS(cctx_T *cctx, char_u **str)
734{
735 isn_T *isn;
736
737 if (cctx->ctx_skip == SKIP_YES)
738 {
739 if (str != NULL)
740 VIM_CLEAR(*str);
741 return OK;
742 }
743 if ((isn = generate_instr_type(cctx, ISN_PUSHS, &t_string)) == NULL)
744 {
745 if (str != NULL)
746 VIM_CLEAR(*str);
747 return FAIL;
748 }
749 isn->isn_arg.string = str == NULL ? NULL : *str;
750
751 return OK;
752}
753
754/*
Bram Moolenaar397a87a2022-03-20 21:14:15 +0000755 * Generate an ISN_PUSHCHANNEL instruction. Channel is always NULL.
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000756 */
757 int
Bram Moolenaar397a87a2022-03-20 21:14:15 +0000758generate_PUSHCHANNEL(cctx_T *cctx)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000759{
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100760#ifdef FEAT_JOB_CHANNEL
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000761 isn_T *isn;
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100762#endif
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000763
764 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100765#ifdef FEAT_JOB_CHANNEL
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000766 if ((isn = generate_instr_type(cctx, ISN_PUSHCHANNEL, &t_channel)) == NULL)
767 return FAIL;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000768 return OK;
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100769#else
770 emsg(_(e_channel_job_feature_not_available));
771 return FAIL;
772#endif
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000773}
774
775/*
Bram Moolenaar397a87a2022-03-20 21:14:15 +0000776 * Generate an ISN_PUSHJOB instruction. Job is always NULL.
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000777 */
778 int
Bram Moolenaar397a87a2022-03-20 21:14:15 +0000779generate_PUSHJOB(cctx_T *cctx)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000780{
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100781#ifdef FEAT_JOB_CHANNEL
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000782 isn_T *isn;
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100783#endif
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000784
785 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100786#ifdef FEAT_JOB_CHANNEL
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000787 if ((isn = generate_instr_type(cctx, ISN_PUSHJOB, &t_job)) == NULL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000788 return FAIL;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000789 return OK;
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100790#else
791 emsg(_(e_channel_job_feature_not_available));
792 return FAIL;
793#endif
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000794}
795
796/*
797 * Generate an ISN_PUSHBLOB instruction.
798 * Consumes "blob".
799 */
800 int
801generate_PUSHBLOB(cctx_T *cctx, blob_T *blob)
802{
803 isn_T *isn;
804
805 RETURN_OK_IF_SKIP(cctx);
806 if ((isn = generate_instr_type(cctx, ISN_PUSHBLOB, &t_blob)) == NULL)
807 return FAIL;
808 isn->isn_arg.blob = blob;
809
810 return OK;
811}
812
813/*
814 * Generate an ISN_PUSHFUNC instruction with name "name".
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000815 */
816 int
817generate_PUSHFUNC(cctx_T *cctx, char_u *name, type_T *type)
818{
819 isn_T *isn;
820 char_u *funcname;
821
822 RETURN_OK_IF_SKIP(cctx);
823 if ((isn = generate_instr_type(cctx, ISN_PUSHFUNC, type)) == NULL)
824 return FAIL;
825 if (name == NULL)
826 funcname = NULL;
Bram Moolenaard041f422022-01-12 19:54:00 +0000827 else if (*name == K_SPECIAL // script-local
828 || vim_strchr(name, AUTOLOAD_CHAR) != NULL) // autoload
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000829 funcname = vim_strsave(name);
830 else
831 {
832 funcname = alloc(STRLEN(name) + 3);
833 if (funcname != NULL)
834 {
835 STRCPY(funcname, "g:");
836 STRCPY(funcname + 2, name);
837 }
838 }
839
840 isn->isn_arg.string = funcname;
841 return OK;
842}
843
844/*
Bram Moolenaar06b77222022-01-25 15:51:56 +0000845 * Generate an ISN_AUTOLOAD instruction.
846 */
847 int
848generate_AUTOLOAD(cctx_T *cctx, char_u *name, type_T *type)
849{
850 isn_T *isn;
851
852 RETURN_OK_IF_SKIP(cctx);
853 if ((isn = generate_instr_type(cctx, ISN_AUTOLOAD, type)) == NULL)
854 return FAIL;
855 isn->isn_arg.string = vim_strsave(name);
856 if (isn->isn_arg.string == NULL)
857 return FAIL;
858 return OK;
859}
860
861/*
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000862 * Generate an ISN_GETITEM instruction with "index".
863 * "with_op" is TRUE for "+=" and other operators, the stack has the current
864 * value below the list with values.
865 */
866 int
867generate_GETITEM(cctx_T *cctx, int index, int with_op)
868{
869 isn_T *isn;
Bram Moolenaar078a4612022-01-04 15:17:03 +0000870 type_T *type = get_type_on_stack(cctx, with_op ? 1 : 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000871 type_T *item_type = &t_any;
872
873 RETURN_OK_IF_SKIP(cctx);
874
875 if (type->tt_type != VAR_LIST)
876 {
877 // cannot happen, caller has checked the type
Bram Moolenaar460ae5d2022-01-01 14:19:49 +0000878 emsg(_(e_list_required));
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000879 return FAIL;
880 }
881 item_type = type->tt_member;
882 if ((isn = generate_instr(cctx, ISN_GETITEM)) == NULL)
883 return FAIL;
884 isn->isn_arg.getitem.gi_index = index;
885 isn->isn_arg.getitem.gi_with_op = with_op;
886
887 // add the item type to the type stack
Bram Moolenaar078a4612022-01-04 15:17:03 +0000888 return push_type_stack(cctx, item_type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000889}
890
891/*
892 * Generate an ISN_SLICE instruction with "count".
893 */
894 int
895generate_SLICE(cctx_T *cctx, int count)
896{
897 isn_T *isn;
898
899 RETURN_OK_IF_SKIP(cctx);
900 if ((isn = generate_instr(cctx, ISN_SLICE)) == NULL)
901 return FAIL;
902 isn->isn_arg.number = count;
903 return OK;
904}
905
906/*
907 * Generate an ISN_CHECKLEN instruction with "min_len".
908 */
909 int
910generate_CHECKLEN(cctx_T *cctx, int min_len, int more_OK)
911{
912 isn_T *isn;
913
914 RETURN_OK_IF_SKIP(cctx);
915
916 if ((isn = generate_instr(cctx, ISN_CHECKLEN)) == NULL)
917 return FAIL;
918 isn->isn_arg.checklen.cl_min_len = min_len;
919 isn->isn_arg.checklen.cl_more_OK = more_OK;
920
921 return OK;
922}
923
924/*
925 * Generate an ISN_STORE instruction.
926 */
927 int
928generate_STORE(cctx_T *cctx, isntype_T isn_type, int idx, char_u *name)
929{
930 isn_T *isn;
931
932 RETURN_OK_IF_SKIP(cctx);
933 if ((isn = generate_instr_drop(cctx, isn_type, 1)) == NULL)
934 return FAIL;
935 if (name != NULL)
936 isn->isn_arg.string = vim_strsave(name);
937 else
938 isn->isn_arg.number = idx;
939
940 return OK;
941}
942
943/*
944 * Generate an ISN_STOREOUTER instruction.
945 */
Yegappan Lakshmanan782b43d2022-01-08 18:43:40 +0000946 static int
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000947generate_STOREOUTER(cctx_T *cctx, int idx, int level)
948{
949 isn_T *isn;
950
951 RETURN_OK_IF_SKIP(cctx);
952 if ((isn = generate_instr_drop(cctx, ISN_STOREOUTER, 1)) == NULL)
953 return FAIL;
954 isn->isn_arg.outer.outer_idx = idx;
955 isn->isn_arg.outer.outer_depth = level;
956
957 return OK;
958}
959
960/*
961 * Generate an ISN_STORENR instruction (short for ISN_PUSHNR + ISN_STORE)
962 */
963 int
964generate_STORENR(cctx_T *cctx, int idx, varnumber_T value)
965{
966 isn_T *isn;
967
968 RETURN_OK_IF_SKIP(cctx);
969 if ((isn = generate_instr(cctx, ISN_STORENR)) == NULL)
970 return FAIL;
971 isn->isn_arg.storenr.stnr_idx = idx;
972 isn->isn_arg.storenr.stnr_val = value;
973
974 return OK;
975}
976
977/*
978 * Generate an ISN_STOREOPT or ISN_STOREFUNCOPT instruction
979 */
Yegappan Lakshmanan782b43d2022-01-08 18:43:40 +0000980 static int
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000981generate_STOREOPT(
982 cctx_T *cctx,
983 isntype_T isn_type,
984 char_u *name,
985 int opt_flags)
986{
987 isn_T *isn;
988
989 RETURN_OK_IF_SKIP(cctx);
990 if ((isn = generate_instr_drop(cctx, isn_type, 1)) == NULL)
991 return FAIL;
992 isn->isn_arg.storeopt.so_name = vim_strsave(name);
993 isn->isn_arg.storeopt.so_flags = opt_flags;
994
995 return OK;
996}
997
998/*
999 * Generate an ISN_LOAD or similar instruction.
1000 */
1001 int
1002generate_LOAD(
1003 cctx_T *cctx,
1004 isntype_T isn_type,
1005 int idx,
1006 char_u *name,
1007 type_T *type)
1008{
1009 isn_T *isn;
1010
1011 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaar078a4612022-01-04 15:17:03 +00001012 if ((isn = generate_instr_type2(cctx, isn_type, type, type)) == NULL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001013 return FAIL;
1014 if (name != NULL)
1015 isn->isn_arg.string = vim_strsave(name);
1016 else
1017 isn->isn_arg.number = idx;
1018
1019 return OK;
1020}
1021
1022/*
1023 * Generate an ISN_LOADOUTER instruction
1024 */
1025 int
1026generate_LOADOUTER(
1027 cctx_T *cctx,
1028 int idx,
1029 int nesting,
1030 type_T *type)
1031{
1032 isn_T *isn;
1033
1034 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaar078a4612022-01-04 15:17:03 +00001035 if ((isn = generate_instr_type2(cctx, ISN_LOADOUTER, type, type)) == NULL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001036 return FAIL;
1037 isn->isn_arg.outer.outer_idx = idx;
1038 isn->isn_arg.outer.outer_depth = nesting;
1039
1040 return OK;
1041}
1042
1043/*
1044 * Generate an ISN_LOADV instruction for v:var.
1045 */
1046 int
1047generate_LOADV(
1048 cctx_T *cctx,
1049 char_u *name,
1050 int error)
1051{
1052 int di_flags;
1053 int vidx = find_vim_var(name, &di_flags);
1054 type_T *type;
1055
1056 RETURN_OK_IF_SKIP(cctx);
1057 if (vidx < 0)
1058 {
1059 if (error)
1060 semsg(_(e_variable_not_found_str), name);
1061 return FAIL;
1062 }
Bram Moolenaard787e402021-12-24 21:36:12 +00001063 type = get_vim_var_type(vidx, cctx->ctx_type_list);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001064 return generate_LOAD(cctx, ISN_LOADV, vidx, NULL, type);
1065}
1066
1067/*
1068 * Generate an ISN_UNLET instruction.
1069 */
1070 int
1071generate_UNLET(cctx_T *cctx, isntype_T isn_type, char_u *name, int forceit)
1072{
1073 isn_T *isn;
1074
1075 RETURN_OK_IF_SKIP(cctx);
1076 if ((isn = generate_instr(cctx, isn_type)) == NULL)
1077 return FAIL;
1078 isn->isn_arg.unlet.ul_name = vim_strsave(name);
1079 isn->isn_arg.unlet.ul_forceit = forceit;
1080
1081 return OK;
1082}
1083
1084/*
1085 * Generate an ISN_LOCKCONST instruction.
1086 */
1087 int
1088generate_LOCKCONST(cctx_T *cctx)
1089{
1090 isn_T *isn;
1091
1092 RETURN_OK_IF_SKIP(cctx);
1093 if ((isn = generate_instr(cctx, ISN_LOCKCONST)) == NULL)
1094 return FAIL;
1095 return OK;
1096}
1097
1098/*
1099 * Generate an ISN_LOADS instruction.
1100 */
1101 int
1102generate_OLDSCRIPT(
1103 cctx_T *cctx,
1104 isntype_T isn_type,
1105 char_u *name,
1106 int sid,
1107 type_T *type)
1108{
1109 isn_T *isn;
1110
1111 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01001112 if (isn_type == ISN_LOADS || isn_type == ISN_LOADEXPORT)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001113 isn = generate_instr_type(cctx, isn_type, type);
1114 else
1115 isn = generate_instr_drop(cctx, isn_type, 1);
1116 if (isn == NULL)
1117 return FAIL;
1118 isn->isn_arg.loadstore.ls_name = vim_strsave(name);
1119 isn->isn_arg.loadstore.ls_sid = sid;
1120
1121 return OK;
1122}
1123
1124/*
1125 * Generate an ISN_LOADSCRIPT or ISN_STORESCRIPT instruction.
1126 */
1127 int
1128generate_VIM9SCRIPT(
1129 cctx_T *cctx,
1130 isntype_T isn_type,
1131 int sid,
1132 int idx,
1133 type_T *type)
1134{
1135 isn_T *isn;
1136 scriptref_T *sref;
1137 scriptitem_T *si = SCRIPT_ITEM(sid);
1138
1139 RETURN_OK_IF_SKIP(cctx);
1140 if (isn_type == ISN_LOADSCRIPT)
Bram Moolenaar160afdb2022-02-06 17:17:02 +00001141 isn = generate_instr_type2(cctx, isn_type, type, type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001142 else
1143 isn = generate_instr_drop(cctx, isn_type, 1);
1144 if (isn == NULL)
1145 return FAIL;
1146
1147 // This requires three arguments, which doesn't fit in an instruction, thus
1148 // we need to allocate a struct for this.
1149 sref = ALLOC_ONE(scriptref_T);
1150 if (sref == NULL)
1151 return FAIL;
1152 isn->isn_arg.script.scriptref = sref;
1153 sref->sref_sid = sid;
1154 sref->sref_idx = idx;
1155 sref->sref_seq = si->sn_script_seq;
1156 sref->sref_type = type;
1157 return OK;
1158}
1159
1160/*
Bram Moolenaarec15b1c2022-03-27 16:29:53 +01001161 * Generate an ISN_NEWLIST instruction for "count" items.
1162 * "use_null" is TRUE for null_list.
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001163 */
1164 int
Bram Moolenaarec15b1c2022-03-27 16:29:53 +01001165generate_NEWLIST(cctx_T *cctx, int count, int use_null)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001166{
1167 isn_T *isn;
Bram Moolenaar078a4612022-01-04 15:17:03 +00001168 type_T *member_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001169 type_T *type;
Bram Moolenaar078a4612022-01-04 15:17:03 +00001170 type_T *decl_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001171
1172 RETURN_OK_IF_SKIP(cctx);
1173 if ((isn = generate_instr(cctx, ISN_NEWLIST)) == NULL)
1174 return FAIL;
Bram Moolenaarec15b1c2022-03-27 16:29:53 +01001175 isn->isn_arg.number = use_null ? -1 : count;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001176
Bram Moolenaar078a4612022-01-04 15:17:03 +00001177 // Get the member type and the declared member type from all the items on
1178 // the stack.
Bram Moolenaar2626d6a2022-02-06 15:49:35 +00001179 member_type = get_member_type_from_stack(count, 1, cctx);
Bram Moolenaar078a4612022-01-04 15:17:03 +00001180 type = get_list_type(member_type, cctx->ctx_type_list);
Bram Moolenaar2626d6a2022-02-06 15:49:35 +00001181 decl_type = get_list_type(&t_any, cctx->ctx_type_list);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001182
1183 // drop the value types
Bram Moolenaar078a4612022-01-04 15:17:03 +00001184 cctx->ctx_type_stack.ga_len -= count;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001185
1186 // add the list type to the type stack
Bram Moolenaar078a4612022-01-04 15:17:03 +00001187 return push_type_stack2(cctx, type, decl_type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001188}
1189
1190/*
1191 * Generate an ISN_NEWDICT instruction.
Bram Moolenaarec15b1c2022-03-27 16:29:53 +01001192 * "use_null" is TRUE for null_dict.
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001193 */
1194 int
Bram Moolenaarec15b1c2022-03-27 16:29:53 +01001195generate_NEWDICT(cctx_T *cctx, int count, int use_null)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001196{
1197 isn_T *isn;
Bram Moolenaar078a4612022-01-04 15:17:03 +00001198 type_T *member_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001199 type_T *type;
Bram Moolenaar078a4612022-01-04 15:17:03 +00001200 type_T *decl_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001201
1202 RETURN_OK_IF_SKIP(cctx);
1203 if ((isn = generate_instr(cctx, ISN_NEWDICT)) == NULL)
1204 return FAIL;
Bram Moolenaarec15b1c2022-03-27 16:29:53 +01001205 isn->isn_arg.number = use_null ? -1 : count;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001206
Bram Moolenaar2626d6a2022-02-06 15:49:35 +00001207 member_type = get_member_type_from_stack(count, 2, cctx);
Bram Moolenaar078a4612022-01-04 15:17:03 +00001208 type = get_dict_type(member_type, cctx->ctx_type_list);
Bram Moolenaar2626d6a2022-02-06 15:49:35 +00001209 decl_type = get_dict_type(&t_any, cctx->ctx_type_list);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001210
1211 // drop the key and value types
Bram Moolenaar078a4612022-01-04 15:17:03 +00001212 cctx->ctx_type_stack.ga_len -= 2 * count;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001213
1214 // add the dict type to the type stack
Bram Moolenaar078a4612022-01-04 15:17:03 +00001215 return push_type_stack2(cctx, type, decl_type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001216}
1217
1218/*
1219 * Generate an ISN_FUNCREF instruction.
Bram Moolenaara915fa02022-03-23 11:29:15 +00001220 * "isnp" is set to the instruction, so that fr_dfunc_idx can be set later.
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001221 */
1222 int
Bram Moolenaara915fa02022-03-23 11:29:15 +00001223generate_FUNCREF(cctx_T *cctx, ufunc_T *ufunc, isn_T **isnp)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001224{
1225 isn_T *isn;
Bram Moolenaar078a4612022-01-04 15:17:03 +00001226 type_T *type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001227
1228 RETURN_OK_IF_SKIP(cctx);
1229 if ((isn = generate_instr(cctx, ISN_FUNCREF)) == NULL)
1230 return FAIL;
Bram Moolenaara915fa02022-03-23 11:29:15 +00001231 if (isnp != NULL)
1232 *isnp = isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001233 if (ufunc->uf_def_status == UF_NOT_COMPILED)
1234 isn->isn_arg.funcref.fr_func_name = vim_strsave(ufunc->uf_name);
1235 else
1236 isn->isn_arg.funcref.fr_dfunc_idx = ufunc->uf_dfunc_idx;
1237 cctx->ctx_has_closure = 1;
1238
1239 // If the referenced function is a closure, it may use items further up in
Bram Moolenaar139575d2022-03-15 19:29:30 +00001240 // the nested context, including this one. But not a function defined at
1241 // the script level.
1242 if ((ufunc->uf_flags & FC_CLOSURE)
1243 && func_name_refcount(cctx->ctx_ufunc->uf_name))
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001244 cctx->ctx_ufunc->uf_flags |= FC_CLOSURE;
1245
Bram Moolenaar078a4612022-01-04 15:17:03 +00001246 type = ufunc->uf_func_type == NULL ? &t_func_any : ufunc->uf_func_type;
1247 return push_type_stack(cctx, type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001248}
1249
1250/*
1251 * Generate an ISN_NEWFUNC instruction.
1252 * "lambda_name" and "func_name" must be in allocated memory and will be
1253 * consumed.
1254 */
1255 int
1256generate_NEWFUNC(cctx_T *cctx, char_u *lambda_name, char_u *func_name)
1257{
1258 isn_T *isn;
1259
1260 if (cctx->ctx_skip == SKIP_YES)
1261 {
1262 vim_free(lambda_name);
1263 vim_free(func_name);
1264 return OK;
1265 }
1266 if ((isn = generate_instr(cctx, ISN_NEWFUNC)) == NULL)
1267 {
1268 vim_free(lambda_name);
1269 vim_free(func_name);
1270 return FAIL;
1271 }
1272 isn->isn_arg.newfunc.nf_lambda = lambda_name;
1273 isn->isn_arg.newfunc.nf_global = func_name;
1274
1275 return OK;
1276}
1277
1278/*
1279 * Generate an ISN_DEF instruction: list functions
1280 */
1281 int
1282generate_DEF(cctx_T *cctx, char_u *name, size_t len)
1283{
1284 isn_T *isn;
1285
1286 RETURN_OK_IF_SKIP(cctx);
1287 if ((isn = generate_instr(cctx, ISN_DEF)) == NULL)
1288 return FAIL;
1289 if (len > 0)
1290 {
1291 isn->isn_arg.string = vim_strnsave(name, len);
1292 if (isn->isn_arg.string == NULL)
1293 return FAIL;
1294 }
1295 return OK;
1296}
1297
1298/*
1299 * Generate an ISN_JUMP instruction.
1300 */
1301 int
1302generate_JUMP(cctx_T *cctx, jumpwhen_T when, int where)
1303{
1304 isn_T *isn;
1305 garray_T *stack = &cctx->ctx_type_stack;
1306
1307 RETURN_OK_IF_SKIP(cctx);
1308 if ((isn = generate_instr(cctx, ISN_JUMP)) == NULL)
1309 return FAIL;
1310 isn->isn_arg.jump.jump_when = when;
1311 isn->isn_arg.jump.jump_where = where;
1312
1313 if (when != JUMP_ALWAYS && stack->ga_len > 0)
1314 --stack->ga_len;
1315
1316 return OK;
1317}
1318
1319/*
1320 * Generate an ISN_JUMP_IF_ARG_SET instruction.
1321 */
1322 int
1323generate_JUMP_IF_ARG_SET(cctx_T *cctx, int arg_off)
1324{
1325 isn_T *isn;
1326
1327 RETURN_OK_IF_SKIP(cctx);
1328 if ((isn = generate_instr(cctx, ISN_JUMP_IF_ARG_SET)) == NULL)
1329 return FAIL;
1330 isn->isn_arg.jumparg.jump_arg_off = arg_off;
1331 // jump_where is set later
1332 return OK;
1333}
1334
1335 int
1336generate_FOR(cctx_T *cctx, int loop_idx)
1337{
1338 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001339
1340 RETURN_OK_IF_SKIP(cctx);
1341 if ((isn = generate_instr(cctx, ISN_FOR)) == NULL)
1342 return FAIL;
1343 isn->isn_arg.forloop.for_idx = loop_idx;
1344
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001345 // type doesn't matter, will be stored next
Bram Moolenaar078a4612022-01-04 15:17:03 +00001346 return push_type_stack(cctx, &t_any);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001347}
1348/*
1349 * Generate an ISN_TRYCONT instruction.
1350 */
1351 int
1352generate_TRYCONT(cctx_T *cctx, int levels, int where)
1353{
1354 isn_T *isn;
1355
1356 RETURN_OK_IF_SKIP(cctx);
1357 if ((isn = generate_instr(cctx, ISN_TRYCONT)) == NULL)
1358 return FAIL;
1359 isn->isn_arg.trycont.tct_levels = levels;
1360 isn->isn_arg.trycont.tct_where = where;
1361
1362 return OK;
1363}
1364
1365
1366/*
1367 * Generate an ISN_BCALL instruction.
1368 * "method_call" is TRUE for "value->method()"
1369 * Return FAIL if the number of arguments is wrong.
1370 */
1371 int
1372generate_BCALL(cctx_T *cctx, int func_idx, int argcount, int method_call)
1373{
1374 isn_T *isn;
1375 garray_T *stack = &cctx->ctx_type_stack;
1376 int argoff;
Bram Moolenaar078a4612022-01-04 15:17:03 +00001377 type2_T *typep;
1378 type2_T *argtypes = NULL;
1379 type2_T shuffled_argtypes[MAX_FUNC_ARGS];
1380 type2_T *maptype = NULL;
1381 type_T *type;
Bram Moolenaar81330182022-02-01 12:11:58 +00001382 type_T *decl_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001383
1384 RETURN_OK_IF_SKIP(cctx);
1385 argoff = check_internal_func(func_idx, argcount);
1386 if (argoff < 0)
1387 return FAIL;
1388
1389 if (method_call && argoff > 1)
1390 {
1391 if ((isn = generate_instr(cctx, ISN_SHUFFLE)) == NULL)
1392 return FAIL;
1393 isn->isn_arg.shuffle.shfl_item = argcount;
1394 isn->isn_arg.shuffle.shfl_up = argoff - 1;
1395 }
1396
1397 if (argcount > 0)
1398 {
1399 // Check the types of the arguments.
Bram Moolenaar078a4612022-01-04 15:17:03 +00001400 typep = ((type2_T *)stack->ga_data) + stack->ga_len - argcount;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001401 if (method_call && argoff > 1)
1402 {
1403 int i;
1404
1405 for (i = 0; i < argcount; ++i)
1406 shuffled_argtypes[i] = (i < argoff - 1)
Bram Moolenaar078a4612022-01-04 15:17:03 +00001407 ? typep[i + 1]
1408 : (i == argoff - 1) ? typep[0] : typep[i];
1409 argtypes = shuffled_argtypes;
1410 }
1411 else
1412 {
1413 int i;
1414
1415 for (i = 0; i < argcount; ++i)
1416 shuffled_argtypes[i] = typep[i];
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001417 argtypes = shuffled_argtypes;
1418 }
1419 if (internal_func_check_arg_types(argtypes, func_idx, argcount,
1420 cctx) == FAIL)
1421 return FAIL;
1422 if (internal_func_is_map(func_idx))
Bram Moolenaar078a4612022-01-04 15:17:03 +00001423 maptype = argtypes;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001424 }
1425
1426 if ((isn = generate_instr(cctx, ISN_BCALL)) == NULL)
1427 return FAIL;
1428 isn->isn_arg.bfunc.cbf_idx = func_idx;
1429 isn->isn_arg.bfunc.cbf_argcount = argcount;
1430
1431 // Drop the argument types and push the return type.
1432 stack->ga_len -= argcount;
Bram Moolenaar81330182022-02-01 12:11:58 +00001433 type = internal_func_ret_type(func_idx, argcount, argtypes, &decl_type);
1434 if (push_type_stack2(cctx, type, decl_type) == FAIL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001435 return FAIL;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001436
Bram Moolenaar35c807d2022-01-27 16:36:29 +00001437 if (maptype != NULL && maptype[0].type_decl->tt_member != NULL
1438 && maptype[0].type_decl->tt_member != &t_any)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001439 // Check that map() didn't change the item types.
Bram Moolenaar35c807d2022-01-27 16:36:29 +00001440 generate_TYPECHECK(cctx, maptype[0].type_decl, -1, 1);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001441
1442 return OK;
1443}
1444
1445/*
1446 * Generate an ISN_LISTAPPEND instruction. Works like add().
1447 * Argument count is already checked.
1448 */
1449 int
1450generate_LISTAPPEND(cctx_T *cctx)
1451{
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001452 type_T *list_type;
1453 type_T *item_type;
1454 type_T *expected;
1455
1456 // Caller already checked that list_type is a list.
Bram Moolenaar81330182022-02-01 12:11:58 +00001457 // For checking the item type we use the declared type of the list and the
1458 // current type of the added item, adding a string to [1, 2] is OK.
1459 list_type = get_decl_type_on_stack(cctx, 1);
Bram Moolenaar078a4612022-01-04 15:17:03 +00001460 item_type = get_type_on_stack(cctx, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001461 expected = list_type->tt_member;
1462 if (need_type(item_type, expected, -1, 0, cctx, FALSE, FALSE) == FAIL)
1463 return FAIL;
1464
1465 if (generate_instr(cctx, ISN_LISTAPPEND) == NULL)
1466 return FAIL;
1467
Bram Moolenaar078a4612022-01-04 15:17:03 +00001468 --cctx->ctx_type_stack.ga_len; // drop the argument
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001469 return OK;
1470}
1471
1472/*
1473 * Generate an ISN_BLOBAPPEND instruction. Works like add().
1474 * Argument count is already checked.
1475 */
1476 int
1477generate_BLOBAPPEND(cctx_T *cctx)
1478{
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001479 type_T *item_type;
1480
1481 // Caller already checked that blob_type is a blob.
Bram Moolenaar078a4612022-01-04 15:17:03 +00001482 item_type = get_type_on_stack(cctx, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001483 if (need_type(item_type, &t_number, -1, 0, cctx, FALSE, FALSE) == FAIL)
1484 return FAIL;
1485
1486 if (generate_instr(cctx, ISN_BLOBAPPEND) == NULL)
1487 return FAIL;
1488
Bram Moolenaar078a4612022-01-04 15:17:03 +00001489 --cctx->ctx_type_stack.ga_len; // drop the argument
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001490 return OK;
1491}
1492
1493/*
1494 * Generate an ISN_DCALL or ISN_UCALL instruction.
1495 * Return FAIL if the number of arguments is wrong.
1496 */
1497 int
1498generate_CALL(cctx_T *cctx, ufunc_T *ufunc, int pushed_argcount)
1499{
1500 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001501 int regular_args = ufunc->uf_args.ga_len;
1502 int argcount = pushed_argcount;
1503
1504 RETURN_OK_IF_SKIP(cctx);
1505 if (argcount > regular_args && !has_varargs(ufunc))
1506 {
1507 semsg(_(e_too_many_arguments_for_function_str),
1508 printable_func_name(ufunc));
1509 return FAIL;
1510 }
1511 if (argcount < regular_args - ufunc->uf_def_args.ga_len)
1512 {
1513 semsg(_(e_not_enough_arguments_for_function_str),
1514 printable_func_name(ufunc));
1515 return FAIL;
1516 }
1517
1518 if (ufunc->uf_def_status != UF_NOT_COMPILED
1519 && ufunc->uf_def_status != UF_COMPILE_ERROR)
1520 {
1521 int i;
Bram Moolenaar139575d2022-03-15 19:29:30 +00001522 compiletype_T compile_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001523
1524 for (i = 0; i < argcount; ++i)
1525 {
1526 type_T *expected;
1527 type_T *actual;
1528
Bram Moolenaar078a4612022-01-04 15:17:03 +00001529 actual = get_type_on_stack(cctx, argcount - i - 1);
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +00001530 if (actual->tt_type == VAR_SPECIAL
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001531 && i >= regular_args - ufunc->uf_def_args.ga_len)
1532 {
1533 // assume v:none used for default argument value
1534 continue;
1535 }
1536 if (i < regular_args)
1537 {
1538 if (ufunc->uf_arg_types == NULL)
1539 continue;
1540 expected = ufunc->uf_arg_types[i];
1541 }
1542 else if (ufunc->uf_va_type == NULL
1543 || ufunc->uf_va_type == &t_list_any)
1544 // possibly a lambda or "...: any"
1545 expected = &t_any;
1546 else
1547 expected = ufunc->uf_va_type->tt_member;
1548 if (need_type(actual, expected, -argcount + i, i + 1, cctx,
1549 TRUE, FALSE) == FAIL)
1550 {
1551 arg_type_mismatch(expected, actual, i + 1);
1552 return FAIL;
1553 }
1554 }
Bram Moolenaar139575d2022-03-15 19:29:30 +00001555 compile_type = get_compile_type(ufunc);
1556 if (func_needs_compiling(ufunc, compile_type)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001557 && compile_def_function(ufunc, ufunc->uf_ret_type == NULL,
Bram Moolenaar139575d2022-03-15 19:29:30 +00001558 compile_type, NULL) == FAIL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001559 return FAIL;
1560 }
1561 if (ufunc->uf_def_status == UF_COMPILE_ERROR)
1562 {
Bram Moolenaara6c18d32022-03-31 20:02:56 +01001563 emsg_funcname(e_call_to_function_that_failed_to_compile_str,
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001564 ufunc->uf_name);
1565 return FAIL;
1566 }
1567
1568 if ((isn = generate_instr(cctx,
1569 ufunc->uf_def_status != UF_NOT_COMPILED ? ISN_DCALL
1570 : ISN_UCALL)) == NULL)
1571 return FAIL;
1572 if (isn->isn_type == ISN_DCALL)
1573 {
1574 isn->isn_arg.dfunc.cdf_idx = ufunc->uf_dfunc_idx;
1575 isn->isn_arg.dfunc.cdf_argcount = argcount;
1576 }
1577 else
1578 {
1579 // A user function may be deleted and redefined later, can't use the
1580 // ufunc pointer, need to look it up again at runtime.
1581 isn->isn_arg.ufunc.cuf_name = vim_strsave(ufunc->uf_name);
1582 isn->isn_arg.ufunc.cuf_argcount = argcount;
1583 }
1584
Bram Moolenaar078a4612022-01-04 15:17:03 +00001585 // drop the argument types
1586 cctx->ctx_type_stack.ga_len -= argcount;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001587
Bram Moolenaar078a4612022-01-04 15:17:03 +00001588 // add return type
1589 return push_type_stack(cctx, ufunc->uf_ret_type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001590}
1591
1592/*
1593 * Generate an ISN_UCALL instruction when the function isn't defined yet.
1594 */
1595 int
1596generate_UCALL(cctx_T *cctx, char_u *name, int argcount)
1597{
1598 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001599
1600 RETURN_OK_IF_SKIP(cctx);
1601 if ((isn = generate_instr(cctx, ISN_UCALL)) == NULL)
1602 return FAIL;
1603 isn->isn_arg.ufunc.cuf_name = vim_strsave(name);
1604 isn->isn_arg.ufunc.cuf_argcount = argcount;
1605
Bram Moolenaar078a4612022-01-04 15:17:03 +00001606 // drop the argument types
1607 cctx->ctx_type_stack.ga_len -= argcount;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001608
Bram Moolenaar078a4612022-01-04 15:17:03 +00001609 // add return value
1610 return push_type_stack(cctx, &t_any);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001611}
1612
1613/*
1614 * Generate an ISN_PCALL instruction.
1615 * "type" is the type of the FuncRef.
1616 */
1617 int
1618generate_PCALL(
1619 cctx_T *cctx,
1620 int argcount,
1621 char_u *name,
1622 type_T *type,
1623 int at_top)
1624{
1625 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001626 type_T *ret_type;
1627
1628 RETURN_OK_IF_SKIP(cctx);
1629
Bram Moolenaar59618fe2021-12-21 12:32:17 +00001630 if (type->tt_type == VAR_ANY || type->tt_type == VAR_UNKNOWN)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001631 ret_type = &t_any;
1632 else if (type->tt_type == VAR_FUNC || type->tt_type == VAR_PARTIAL)
1633 {
1634 if (type->tt_argcount != -1)
1635 {
1636 int varargs = (type->tt_flags & TTFLAG_VARARGS) ? 1 : 0;
1637
1638 if (argcount < type->tt_min_argcount - varargs)
1639 {
Bram Moolenaara6c18d32022-03-31 20:02:56 +01001640 emsg_funcname(e_not_enough_arguments_for_function_str, name);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001641 return FAIL;
1642 }
1643 if (!varargs && argcount > type->tt_argcount)
1644 {
Bram Moolenaara6c18d32022-03-31 20:02:56 +01001645 emsg_funcname(e_too_many_arguments_for_function_str, name);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001646 return FAIL;
1647 }
1648 if (type->tt_args != NULL)
1649 {
1650 int i;
1651
1652 for (i = 0; i < argcount; ++i)
1653 {
1654 int offset = -argcount + i - (at_top ? 0 : 1);
Bram Moolenaar078a4612022-01-04 15:17:03 +00001655 type_T *actual = get_type_on_stack(cctx, -1 - offset);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001656 type_T *expected;
1657
1658 if (varargs && i >= type->tt_argcount - 1)
1659 expected = type->tt_args[
1660 type->tt_argcount - 1]->tt_member;
1661 else if (i >= type->tt_min_argcount
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +00001662 && actual->tt_type == VAR_SPECIAL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001663 expected = &t_any;
1664 else
1665 expected = type->tt_args[i];
1666 if (need_type(actual, expected, offset, i + 1,
1667 cctx, TRUE, FALSE) == FAIL)
1668 {
1669 arg_type_mismatch(expected, actual, i + 1);
1670 return FAIL;
1671 }
1672 }
1673 }
1674 }
1675 ret_type = type->tt_member;
1676 if (ret_type == &t_unknown)
1677 // return type not known yet, use a runtime check
1678 ret_type = &t_any;
1679 }
1680 else
1681 {
1682 semsg(_(e_not_callable_type_str), name);
1683 return FAIL;
1684 }
1685
1686 if ((isn = generate_instr(cctx, ISN_PCALL)) == NULL)
1687 return FAIL;
1688 isn->isn_arg.pfunc.cpf_top = at_top;
1689 isn->isn_arg.pfunc.cpf_argcount = argcount;
1690
Bram Moolenaar078a4612022-01-04 15:17:03 +00001691 // drop the arguments and the funcref/partial
1692 cctx->ctx_type_stack.ga_len -= argcount + 1;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001693
Bram Moolenaar078a4612022-01-04 15:17:03 +00001694 // push the return value
1695 push_type_stack(cctx, ret_type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001696
1697 // If partial is above the arguments it must be cleared and replaced with
1698 // the return value.
1699 if (at_top && generate_instr(cctx, ISN_PCALL_END) == NULL)
1700 return FAIL;
1701
1702 return OK;
1703}
1704
1705/*
1706 * Generate an ISN_STRINGMEMBER instruction.
1707 */
1708 int
1709generate_STRINGMEMBER(cctx_T *cctx, char_u *name, size_t len)
1710{
1711 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001712 type_T *type;
1713
1714 RETURN_OK_IF_SKIP(cctx);
1715 if ((isn = generate_instr(cctx, ISN_STRINGMEMBER)) == NULL)
1716 return FAIL;
1717 isn->isn_arg.string = vim_strnsave(name, len);
1718
1719 // check for dict type
Bram Moolenaar078a4612022-01-04 15:17:03 +00001720 type = get_type_on_stack(cctx, 0);
Bram Moolenaar59618fe2021-12-21 12:32:17 +00001721 if (type->tt_type != VAR_DICT && type != &t_any && type != &t_unknown)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001722 {
1723 char *tofree;
1724
1725 semsg(_(e_expected_dictionary_for_using_key_str_but_got_str),
1726 name, type_name(type, &tofree));
1727 vim_free(tofree);
1728 return FAIL;
1729 }
1730 // change dict type to dict member type
1731 if (type->tt_type == VAR_DICT)
1732 {
Bram Moolenaar078a4612022-01-04 15:17:03 +00001733 type_T *ntype = type->tt_member == &t_unknown
1734 ? &t_any : type->tt_member;
1735 set_type_on_stack(cctx, ntype, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001736 }
1737
1738 return OK;
1739}
1740
1741/*
1742 * Generate an ISN_ECHO instruction.
1743 */
1744 int
1745generate_ECHO(cctx_T *cctx, int with_white, int count)
1746{
1747 isn_T *isn;
1748
1749 RETURN_OK_IF_SKIP(cctx);
1750 if ((isn = generate_instr_drop(cctx, ISN_ECHO, count)) == NULL)
1751 return FAIL;
1752 isn->isn_arg.echo.echo_with_white = with_white;
1753 isn->isn_arg.echo.echo_count = count;
1754
1755 return OK;
1756}
1757
1758/*
1759 * Generate an ISN_EXECUTE/ISN_ECHOMSG/ISN_ECHOERR instruction.
1760 */
1761 int
1762generate_MULT_EXPR(cctx_T *cctx, isntype_T isn_type, int count)
1763{
1764 isn_T *isn;
1765
1766 if ((isn = generate_instr_drop(cctx, isn_type, count)) == NULL)
1767 return FAIL;
1768 isn->isn_arg.number = count;
1769
1770 return OK;
1771}
1772
1773/*
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01001774 * Generate an ISN_SOURCE instruction.
1775 */
1776 int
1777generate_SOURCE(cctx_T *cctx, int sid)
1778{
1779 isn_T *isn;
1780
1781 if ((isn = generate_instr(cctx, ISN_SOURCE)) == NULL)
1782 return FAIL;
1783 isn->isn_arg.number = sid;
1784
1785 return OK;
1786}
1787
1788/*
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001789 * Generate an ISN_PUT instruction.
1790 */
1791 int
1792generate_PUT(cctx_T *cctx, int regname, linenr_T lnum)
1793{
1794 isn_T *isn;
1795
1796 RETURN_OK_IF_SKIP(cctx);
1797 if ((isn = generate_instr(cctx, ISN_PUT)) == NULL)
1798 return FAIL;
1799 isn->isn_arg.put.put_regname = regname;
1800 isn->isn_arg.put.put_lnum = lnum;
1801 return OK;
1802}
1803
1804/*
1805 * Generate an EXEC instruction that takes a string argument.
1806 * A copy is made of "line".
1807 */
1808 int
1809generate_EXEC_copy(cctx_T *cctx, isntype_T isntype, char_u *line)
1810{
1811 isn_T *isn;
1812
1813 RETURN_OK_IF_SKIP(cctx);
1814 if ((isn = generate_instr(cctx, isntype)) == NULL)
1815 return FAIL;
1816 isn->isn_arg.string = vim_strsave(line);
1817 return OK;
1818}
1819
1820/*
1821 * Generate an EXEC instruction that takes a string argument.
1822 * "str" must be allocated, it is consumed.
1823 */
1824 int
1825generate_EXEC(cctx_T *cctx, isntype_T isntype, char_u *str)
1826{
1827 isn_T *isn;
1828
1829 if (cctx->ctx_skip == SKIP_YES)
1830 {
1831 vim_free(str);
1832 return OK;
1833 }
1834 if ((isn = generate_instr(cctx, isntype)) == NULL)
1835 {
1836 vim_free(str);
1837 return FAIL;
1838 }
1839 isn->isn_arg.string = str;
1840 return OK;
1841}
1842
1843 int
1844generate_LEGACY_EVAL(cctx_T *cctx, char_u *line)
1845{
1846 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001847
1848 RETURN_OK_IF_SKIP(cctx);
1849 if ((isn = generate_instr(cctx, ISN_LEGACY_EVAL)) == NULL)
1850 return FAIL;
1851 isn->isn_arg.string = vim_strsave(line);
1852
Bram Moolenaar078a4612022-01-04 15:17:03 +00001853 return push_type_stack(cctx, &t_any);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001854}
1855
1856 int
1857generate_EXECCONCAT(cctx_T *cctx, int count)
1858{
1859 isn_T *isn;
1860
1861 if ((isn = generate_instr_drop(cctx, ISN_EXECCONCAT, count)) == NULL)
1862 return FAIL;
1863 isn->isn_arg.number = count;
1864 return OK;
1865}
1866
1867/*
1868 * Generate ISN_RANGE. Consumes "range". Return OK/FAIL.
1869 */
1870 int
1871generate_RANGE(cctx_T *cctx, char_u *range)
1872{
1873 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001874
1875 if ((isn = generate_instr(cctx, ISN_RANGE)) == NULL)
1876 return FAIL;
1877 isn->isn_arg.string = range;
1878
Bram Moolenaar078a4612022-01-04 15:17:03 +00001879 return push_type_stack(cctx, &t_number);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001880}
1881
1882 int
1883generate_UNPACK(cctx_T *cctx, int var_count, int semicolon)
1884{
1885 isn_T *isn;
1886
1887 RETURN_OK_IF_SKIP(cctx);
1888 if ((isn = generate_instr(cctx, ISN_UNPACK)) == NULL)
1889 return FAIL;
1890 isn->isn_arg.unpack.unp_count = var_count;
1891 isn->isn_arg.unpack.unp_semicolon = semicolon;
1892 return OK;
1893}
1894
1895/*
1896 * Generate an instruction for any command modifiers.
1897 */
1898 int
1899generate_cmdmods(cctx_T *cctx, cmdmod_T *cmod)
1900{
1901 isn_T *isn;
1902
1903 if (has_cmdmod(cmod, FALSE))
1904 {
1905 cctx->ctx_has_cmdmod = TRUE;
1906
1907 if ((isn = generate_instr(cctx, ISN_CMDMOD)) == NULL)
1908 return FAIL;
1909 isn->isn_arg.cmdmod.cf_cmdmod = ALLOC_ONE(cmdmod_T);
1910 if (isn->isn_arg.cmdmod.cf_cmdmod == NULL)
1911 return FAIL;
1912 mch_memmove(isn->isn_arg.cmdmod.cf_cmdmod, cmod, sizeof(cmdmod_T));
1913 // filter program now belongs to the instruction
1914 cmod->cmod_filter_regmatch.regprog = NULL;
1915 }
1916
1917 return OK;
1918}
1919
1920 int
1921generate_undo_cmdmods(cctx_T *cctx)
1922{
1923 if (cctx->ctx_has_cmdmod && generate_instr(cctx, ISN_CMDMOD_REV) == NULL)
1924 return FAIL;
1925 cctx->ctx_has_cmdmod = FALSE;
1926 return OK;
1927}
1928
1929/*
1930 * Generate a STORE instruction for "dest", not being "dest_local".
1931 * Return FAIL when out of memory.
1932 */
1933 int
1934generate_store_var(
1935 cctx_T *cctx,
1936 assign_dest_T dest,
1937 int opt_flags,
1938 int vimvaridx,
1939 int scriptvar_idx,
1940 int scriptvar_sid,
1941 type_T *type,
1942 char_u *name)
1943{
1944 switch (dest)
1945 {
1946 case dest_option:
1947 return generate_STOREOPT(cctx, ISN_STOREOPT,
1948 skip_option_env_lead(name), opt_flags);
1949 case dest_func_option:
1950 return generate_STOREOPT(cctx, ISN_STOREFUNCOPT,
1951 skip_option_env_lead(name), opt_flags);
1952 case dest_global:
1953 // include g: with the name, easier to execute that way
1954 return generate_STORE(cctx, vim_strchr(name, AUTOLOAD_CHAR) == NULL
1955 ? ISN_STOREG : ISN_STOREAUTO, 0, name);
1956 case dest_buffer:
1957 // include b: with the name, easier to execute that way
1958 return generate_STORE(cctx, ISN_STOREB, 0, name);
1959 case dest_window:
1960 // include w: with the name, easier to execute that way
1961 return generate_STORE(cctx, ISN_STOREW, 0, name);
1962 case dest_tab:
1963 // include t: with the name, easier to execute that way
1964 return generate_STORE(cctx, ISN_STORET, 0, name);
1965 case dest_env:
1966 return generate_STORE(cctx, ISN_STOREENV, 0, name + 1);
1967 case dest_reg:
1968 return generate_STORE(cctx, ISN_STOREREG,
1969 name[1] == '@' ? '"' : name[1], NULL);
1970 case dest_vimvar:
1971 return generate_STORE(cctx, ISN_STOREV, vimvaridx, NULL);
1972 case dest_script:
1973 if (scriptvar_idx < 0)
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01001974 {
1975 isntype_T isn_type = ISN_STORES;
1976
1977 if (SCRIPT_ID_VALID(scriptvar_sid)
Bram Moolenaarccbfd482022-03-31 16:18:23 +01001978 && SCRIPT_ITEM(scriptvar_sid)->sn_import_autoload
1979 && SCRIPT_ITEM(scriptvar_sid)->sn_autoload_prefix
1980 == NULL)
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01001981 {
1982 // "import autoload './dir/script.vim'" - load script first
1983 if (generate_SOURCE(cctx, scriptvar_sid) == FAIL)
1984 return FAIL;
1985 isn_type = ISN_STOREEXPORT;
1986 }
1987
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001988 // "s:" may be included in the name.
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01001989 return generate_OLDSCRIPT(cctx, isn_type, name,
1990 scriptvar_sid, type);
1991 }
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001992 return generate_VIM9SCRIPT(cctx, ISN_STORESCRIPT,
1993 scriptvar_sid, scriptvar_idx, type);
1994 case dest_local:
1995 case dest_expr:
1996 // cannot happen
1997 break;
1998 }
1999 return FAIL;
2000}
2001
Bram Moolenaar38ecd972022-01-15 21:44:44 +00002002/*
2003 * Return TRUE when inside a "for" or "while" loop.
2004 */
2005 int
2006inside_loop_scope(cctx_T *cctx)
2007{
2008 scope_T *scope = cctx->ctx_scope;
2009
2010 for (;;)
2011 {
2012 if (scope == NULL)
2013 break;
2014 if (scope->se_type == FOR_SCOPE || scope->se_type == WHILE_SCOPE)
2015 return TRUE;
2016 scope = scope->se_outer;
2017 }
2018 return FALSE;
2019}
2020
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002021 int
Bram Moolenaar5cd64792021-12-25 18:23:24 +00002022generate_store_lhs(cctx_T *cctx, lhs_T *lhs, int instr_count, int is_decl)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002023{
2024 if (lhs->lhs_dest != dest_local)
2025 return generate_store_var(cctx, lhs->lhs_dest,
2026 lhs->lhs_opt_flags, lhs->lhs_vimvaridx,
2027 lhs->lhs_scriptvar_idx, lhs->lhs_scriptvar_sid,
2028 lhs->lhs_type, lhs->lhs_name);
2029
2030 if (lhs->lhs_lvar != NULL)
2031 {
2032 garray_T *instr = &cctx->ctx_instr;
2033 isn_T *isn = ((isn_T *)instr->ga_data) + instr->ga_len - 1;
2034
Bram Moolenaar5cd64792021-12-25 18:23:24 +00002035 // Optimization: turn "var = 123" from ISN_PUSHNR + ISN_STORE into
2036 // ISN_STORENR.
2037 // And "var = 0" does not need any instruction.
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002038 if (lhs->lhs_lvar->lv_from_outer == 0
2039 && instr->ga_len == instr_count + 1
2040 && isn->isn_type == ISN_PUSHNR)
2041 {
2042 varnumber_T val = isn->isn_arg.number;
2043 garray_T *stack = &cctx->ctx_type_stack;
2044
Bram Moolenaar38ecd972022-01-15 21:44:44 +00002045 if (val == 0 && is_decl && !inside_loop_scope(cctx))
Bram Moolenaar5cd64792021-12-25 18:23:24 +00002046 {
Bram Moolenaar38ecd972022-01-15 21:44:44 +00002047 // zero is the default value, no need to do anything
Bram Moolenaar5cd64792021-12-25 18:23:24 +00002048 --instr->ga_len;
2049 }
2050 else
2051 {
2052 isn->isn_type = ISN_STORENR;
2053 isn->isn_arg.storenr.stnr_idx = lhs->lhs_lvar->lv_idx;
2054 isn->isn_arg.storenr.stnr_val = val;
2055 }
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002056 if (stack->ga_len > 0)
2057 --stack->ga_len;
2058 }
2059 else if (lhs->lhs_lvar->lv_from_outer > 0)
2060 generate_STOREOUTER(cctx, lhs->lhs_lvar->lv_idx,
2061 lhs->lhs_lvar->lv_from_outer);
2062 else
2063 generate_STORE(cctx, ISN_STORE, lhs->lhs_lvar->lv_idx, NULL);
2064 }
2065 return OK;
2066}
2067
2068#if defined(FEAT_PROFILE) || defined(PROTO)
2069 void
2070may_generate_prof_end(cctx_T *cctx, int prof_lnum)
2071{
2072 if (cctx->ctx_compile_type == CT_PROFILE && prof_lnum >= 0)
2073 generate_instr(cctx, ISN_PROF_END);
2074}
2075#endif
2076
2077
2078/*
2079 * Delete an instruction, free what it contains.
2080 */
2081 void
2082delete_instr(isn_T *isn)
2083{
2084 switch (isn->isn_type)
2085 {
Bram Moolenaar06b77222022-01-25 15:51:56 +00002086 case ISN_AUTOLOAD:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002087 case ISN_DEF:
2088 case ISN_EXEC:
2089 case ISN_EXECRANGE:
2090 case ISN_EXEC_SPLIT:
2091 case ISN_LEGACY_EVAL:
2092 case ISN_LOADAUTO:
2093 case ISN_LOADB:
2094 case ISN_LOADENV:
2095 case ISN_LOADG:
2096 case ISN_LOADOPT:
2097 case ISN_LOADT:
2098 case ISN_LOADW:
2099 case ISN_LOCKUNLOCK:
2100 case ISN_PUSHEXC:
2101 case ISN_PUSHFUNC:
2102 case ISN_PUSHS:
2103 case ISN_RANGE:
2104 case ISN_STOREAUTO:
2105 case ISN_STOREB:
2106 case ISN_STOREENV:
2107 case ISN_STOREG:
2108 case ISN_STORET:
2109 case ISN_STOREW:
2110 case ISN_STRINGMEMBER:
2111 vim_free(isn->isn_arg.string);
2112 break;
2113
2114 case ISN_SUBSTITUTE:
2115 {
2116 int idx;
2117 isn_T *list = isn->isn_arg.subs.subs_instr;
2118
2119 vim_free(isn->isn_arg.subs.subs_cmd);
2120 for (idx = 0; list[idx].isn_type != ISN_FINISH; ++idx)
2121 delete_instr(list + idx);
2122 vim_free(list);
2123 }
2124 break;
2125
2126 case ISN_INSTR:
2127 {
2128 int idx;
2129 isn_T *list = isn->isn_arg.instr;
2130
2131 for (idx = 0; list[idx].isn_type != ISN_FINISH; ++idx)
2132 delete_instr(list + idx);
2133 vim_free(list);
2134 }
2135 break;
2136
2137 case ISN_LOADS:
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002138 case ISN_LOADEXPORT:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002139 case ISN_STORES:
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002140 case ISN_STOREEXPORT:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002141 vim_free(isn->isn_arg.loadstore.ls_name);
2142 break;
2143
2144 case ISN_UNLET:
2145 case ISN_UNLETENV:
2146 vim_free(isn->isn_arg.unlet.ul_name);
2147 break;
2148
2149 case ISN_STOREOPT:
2150 case ISN_STOREFUNCOPT:
2151 vim_free(isn->isn_arg.storeopt.so_name);
2152 break;
2153
2154 case ISN_PUSHBLOB: // push blob isn_arg.blob
2155 blob_unref(isn->isn_arg.blob);
2156 break;
2157
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002158 case ISN_UCALL:
2159 vim_free(isn->isn_arg.ufunc.cuf_name);
2160 break;
2161
2162 case ISN_FUNCREF:
2163 {
2164 if (isn->isn_arg.funcref.fr_func_name == NULL)
2165 {
2166 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002167 + isn->isn_arg.funcref.fr_dfunc_idx;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002168 ufunc_T *ufunc = dfunc->df_ufunc;
2169
2170 if (ufunc != NULL && func_name_refcount(ufunc->uf_name))
2171 func_ptr_unref(ufunc);
2172 }
2173 else
2174 {
2175 char_u *name = isn->isn_arg.funcref.fr_func_name;
2176
2177 if (name != NULL)
2178 func_unref(name);
2179 vim_free(isn->isn_arg.funcref.fr_func_name);
2180 }
2181 }
2182 break;
2183
2184 case ISN_DCALL:
2185 {
2186 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002187 + isn->isn_arg.dfunc.cdf_idx;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002188
2189 if (dfunc->df_ufunc != NULL
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002190 && func_name_refcount(dfunc->df_ufunc->uf_name))
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002191 func_ptr_unref(dfunc->df_ufunc);
2192 }
2193 break;
2194
2195 case ISN_NEWFUNC:
2196 {
2197 char_u *lambda = isn->isn_arg.newfunc.nf_lambda;
Bram Moolenaaracc4b562022-01-24 13:54:45 +00002198 ufunc_T *ufunc = find_func_even_dead(lambda, FFED_IS_GLOBAL);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002199
2200 if (ufunc != NULL)
2201 {
2202 unlink_def_function(ufunc);
2203 func_ptr_unref(ufunc);
2204 }
2205
2206 vim_free(lambda);
2207 vim_free(isn->isn_arg.newfunc.nf_global);
2208 }
2209 break;
2210
2211 case ISN_CHECKTYPE:
2212 case ISN_SETTYPE:
2213 free_type(isn->isn_arg.type.ct_type);
2214 break;
2215
2216 case ISN_CMDMOD:
2217 vim_regfree(isn->isn_arg.cmdmod.cf_cmdmod
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002218 ->cmod_filter_regmatch.regprog);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002219 vim_free(isn->isn_arg.cmdmod.cf_cmdmod);
2220 break;
2221
2222 case ISN_LOADSCRIPT:
2223 case ISN_STORESCRIPT:
2224 vim_free(isn->isn_arg.script.scriptref);
2225 break;
2226
2227 case ISN_TRY:
Bram Moolenaar0d807102021-12-21 09:42:09 +00002228 vim_free(isn->isn_arg.tryref.try_ref);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002229 break;
2230
2231 case ISN_CEXPR_CORE:
2232 vim_free(isn->isn_arg.cexpr.cexpr_ref->cer_cmdline);
2233 vim_free(isn->isn_arg.cexpr.cexpr_ref);
2234 break;
2235
2236 case ISN_2BOOL:
2237 case ISN_2STRING:
2238 case ISN_2STRING_ANY:
2239 case ISN_ADDBLOB:
2240 case ISN_ADDLIST:
2241 case ISN_ANYINDEX:
2242 case ISN_ANYSLICE:
2243 case ISN_BCALL:
2244 case ISN_BLOBAPPEND:
2245 case ISN_BLOBINDEX:
2246 case ISN_BLOBSLICE:
2247 case ISN_CATCH:
2248 case ISN_CEXPR_AUCMD:
2249 case ISN_CHECKLEN:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002250 case ISN_CLEARDICT:
2251 case ISN_CMDMOD_REV:
2252 case ISN_COMPAREANY:
2253 case ISN_COMPAREBLOB:
2254 case ISN_COMPAREBOOL:
2255 case ISN_COMPAREDICT:
2256 case ISN_COMPAREFLOAT:
2257 case ISN_COMPAREFUNC:
2258 case ISN_COMPARELIST:
2259 case ISN_COMPARENR:
Bram Moolenaar7a222242022-03-01 19:23:24 +00002260 case ISN_COMPARENULL:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002261 case ISN_COMPARESPECIAL:
2262 case ISN_COMPARESTRING:
2263 case ISN_CONCAT:
2264 case ISN_COND2BOOL:
2265 case ISN_DEBUG:
2266 case ISN_DROP:
2267 case ISN_ECHO:
2268 case ISN_ECHOCONSOLE:
2269 case ISN_ECHOERR:
2270 case ISN_ECHOMSG:
2271 case ISN_ENDTRY:
2272 case ISN_EXECCONCAT:
2273 case ISN_EXECUTE:
2274 case ISN_FINALLY:
2275 case ISN_FINISH:
2276 case ISN_FOR:
2277 case ISN_GETITEM:
2278 case ISN_JUMP:
2279 case ISN_JUMP_IF_ARG_SET:
2280 case ISN_LISTAPPEND:
2281 case ISN_LISTINDEX:
2282 case ISN_LISTSLICE:
2283 case ISN_LOAD:
2284 case ISN_LOADBDICT:
2285 case ISN_LOADGDICT:
2286 case ISN_LOADOUTER:
2287 case ISN_LOADREG:
2288 case ISN_LOADTDICT:
2289 case ISN_LOADV:
2290 case ISN_LOADWDICT:
2291 case ISN_LOCKCONST:
2292 case ISN_MEMBER:
2293 case ISN_NEGATENR:
2294 case ISN_NEWDICT:
2295 case ISN_NEWLIST:
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +00002296 case ISN_NEWPARTIAL:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002297 case ISN_OPANY:
2298 case ISN_OPFLOAT:
2299 case ISN_OPNR:
2300 case ISN_PCALL:
2301 case ISN_PCALL_END:
2302 case ISN_PROF_END:
2303 case ISN_PROF_START:
2304 case ISN_PUSHBOOL:
Bram Moolenaar397a87a2022-03-20 21:14:15 +00002305 case ISN_PUSHCHANNEL:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002306 case ISN_PUSHF:
Bram Moolenaar397a87a2022-03-20 21:14:15 +00002307 case ISN_PUSHJOB:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002308 case ISN_PUSHNR:
2309 case ISN_PUSHSPEC:
2310 case ISN_PUT:
2311 case ISN_REDIREND:
2312 case ISN_REDIRSTART:
2313 case ISN_RETURN:
2314 case ISN_RETURN_VOID:
2315 case ISN_SHUFFLE:
2316 case ISN_SLICE:
2317 case ISN_STORE:
2318 case ISN_STOREINDEX:
2319 case ISN_STORENR:
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002320 case ISN_SOURCE:
2321 case ISN_STOREOUTER:
2322 case ISN_STORERANGE:
2323 case ISN_STOREREG:
2324 case ISN_STOREV:
2325 case ISN_STRINDEX:
2326 case ISN_STRSLICE:
2327 case ISN_THROW:
2328 case ISN_TRYCONT:
2329 case ISN_UNLETINDEX:
2330 case ISN_UNLETRANGE:
2331 case ISN_UNPACK:
2332 case ISN_USEDICT:
2333 // nothing allocated
2334 break;
2335}
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002336}
2337
2338 void
2339clear_instr_ga(garray_T *gap)
2340{
2341 int idx;
2342
2343 for (idx = 0; idx < gap->ga_len; ++idx)
2344 delete_instr(((isn_T *)gap->ga_data) + idx);
2345 ga_clear(gap);
2346}
2347
2348
2349#endif // defined(FEAT_EVAL)