blob: f20d993728e9ea0c3f0cd9700647f24ddbd830c2 [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 == vartype2)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000359 {
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000360 switch (vartype1)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000361 {
362 case VAR_BOOL: isntype = ISN_COMPAREBOOL; break;
363 case VAR_SPECIAL: isntype = ISN_COMPARESPECIAL; break;
364 case VAR_NUMBER: isntype = ISN_COMPARENR; break;
365 case VAR_FLOAT: isntype = ISN_COMPAREFLOAT; break;
366 case VAR_STRING: isntype = ISN_COMPARESTRING; break;
367 case VAR_BLOB: isntype = ISN_COMPAREBLOB; break;
368 case VAR_LIST: isntype = ISN_COMPARELIST; break;
369 case VAR_DICT: isntype = ISN_COMPAREDICT; break;
370 case VAR_FUNC: isntype = ISN_COMPAREFUNC; break;
371 default: isntype = ISN_COMPAREANY; break;
372 }
373 }
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000374 else if (vartype1 == VAR_ANY || vartype2 == VAR_ANY
375 || ((vartype1 == VAR_NUMBER || vartype1 == VAR_FLOAT)
376 && (vartype2 == VAR_NUMBER || vartype2 == VAR_FLOAT))
377 || (vartype1 == VAR_FUNC && vartype2 == VAR_PARTIAL)
378 || (vartype1 == VAR_PARTIAL && vartype2 == VAR_FUNC))
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000379 isntype = ISN_COMPAREANY;
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000380 else if (vartype1 == VAR_SPECIAL || vartype2 == VAR_SPECIAL)
Bram Moolenaar7a222242022-03-01 19:23:24 +0000381 {
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000382 if ((vartype1 == VAR_SPECIAL
383 && (tv1 != NULL ? tv1->vval.v_number == VVAL_NONE
384 : type1 == &t_none)
385 && vartype2 != VAR_STRING)
386 || (vartype2 == VAR_SPECIAL
387 && (tv2 != NULL ? tv2->vval.v_number == VVAL_NONE
388 : type2 == &t_none)
389 && vartype1 != VAR_STRING))
390 {
391 semsg(_(e_cannot_compare_str_with_str),
392 vartype_name(vartype1), vartype_name(vartype2));
393 return ISN_DROP;
394 }
Bram Moolenaar05667812022-03-15 20:21:33 +0000395 // although comparing null with number, float or bool is not useful, we
396 // allow it
Bram Moolenaar7a222242022-03-01 19:23:24 +0000397 isntype = ISN_COMPARENULL;
398 }
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000399
400 if ((exprtype == EXPR_IS || exprtype == EXPR_ISNOT)
401 && (isntype == ISN_COMPAREBOOL
402 || isntype == ISN_COMPARESPECIAL
403 || isntype == ISN_COMPARENR
404 || isntype == ISN_COMPAREFLOAT))
405 {
406 semsg(_(e_cannot_use_str_with_str),
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000407 exprtype == EXPR_IS ? "is" : "isnot" , vartype_name(vartype1));
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000408 return ISN_DROP;
409 }
410 if (isntype == ISN_DROP
411 || ((exprtype != EXPR_EQUAL && exprtype != EXPR_NEQUAL
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000412 && (vartype1 == VAR_BOOL || vartype1 == VAR_SPECIAL
413 || vartype2 == VAR_BOOL || vartype2 == VAR_SPECIAL)))
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000414 || ((exprtype != EXPR_EQUAL && exprtype != EXPR_NEQUAL
Bram Moolenaar7a222242022-03-01 19:23:24 +0000415 && exprtype != EXPR_IS && exprtype != EXPR_ISNOT
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000416 && (vartype1 == VAR_BLOB || vartype2 == VAR_BLOB
417 || vartype1 == VAR_LIST || vartype2 == VAR_LIST))))
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000418 {
419 semsg(_(e_cannot_compare_str_with_str),
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000420 vartype_name(vartype1), vartype_name(vartype2));
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000421 return ISN_DROP;
422 }
423 return isntype;
424}
425
426 int
427check_compare_types(exprtype_T type, typval_T *tv1, typval_T *tv2)
428{
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000429 if (get_compare_isn(type, tv1, tv2, NULL, NULL) == ISN_DROP)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000430 return FAIL;
431 return OK;
432}
433
434/*
435 * Generate an ISN_COMPARE* instruction with a boolean result.
436 */
437 int
438generate_COMPARE(cctx_T *cctx, exprtype_T exprtype, int ic)
439{
440 isntype_T isntype;
441 isn_T *isn;
442 garray_T *stack = &cctx->ctx_type_stack;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000443
444 RETURN_OK_IF_SKIP(cctx);
445
446 // Get the known type of the two items on the stack. If they are matching
447 // use a type-specific instruction. Otherwise fall back to runtime type
448 // checking.
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000449 isntype = get_compare_isn(exprtype, NULL, NULL,
450 get_type_on_stack(cctx, 1), get_type_on_stack(cctx, 0));
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000451 if (isntype == ISN_DROP)
452 return FAIL;
453
454 if ((isn = generate_instr(cctx, isntype)) == NULL)
455 return FAIL;
456 isn->isn_arg.op.op_type = exprtype;
457 isn->isn_arg.op.op_ic = ic;
458
459 // takes two arguments, puts one bool back
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100460 --stack->ga_len;
461 set_type_on_stack(cctx, &t_bool, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000462
463 return OK;
464}
465
466/*
LemonBoy372bcce2022-04-25 12:43:20 +0100467 * Generate an ISN_CONCAT instruction.
468 * "count" is the number of stack elements to join together and it must be
469 * greater or equal to one.
470 * The caller ensures all the "count" elements on the stack have the right type.
471 */
472 int
473generate_CONCAT(cctx_T *cctx, int count)
474{
475 isn_T *isn;
476 garray_T *stack = &cctx->ctx_type_stack;
477
478 RETURN_OK_IF_SKIP(cctx);
479
LemonBoy372bcce2022-04-25 12:43:20 +0100480 if ((isn = generate_instr(cctx, ISN_CONCAT)) == NULL)
481 return FAIL;
482 isn->isn_arg.number = count;
483
484 // drop the argument types
485 stack->ga_len -= count - 1;
486
487 return OK;
488}
489
490/*
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000491 * Generate an ISN_2BOOL instruction.
492 * "offset" is the offset in the type stack.
493 */
494 int
495generate_2BOOL(cctx_T *cctx, int invert, int offset)
496{
497 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000498
499 RETURN_OK_IF_SKIP(cctx);
500 if ((isn = generate_instr(cctx, ISN_2BOOL)) == NULL)
501 return FAIL;
502 isn->isn_arg.tobool.invert = invert;
503 isn->isn_arg.tobool.offset = offset;
504
505 // type becomes bool
Bram Moolenaar078a4612022-01-04 15:17:03 +0000506 set_type_on_stack(cctx, &t_bool, -1 - offset);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000507
508 return OK;
509}
510
511/*
512 * Generate an ISN_COND2BOOL instruction.
513 */
514 int
515generate_COND2BOOL(cctx_T *cctx)
516{
517 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000518
519 RETURN_OK_IF_SKIP(cctx);
520 if ((isn = generate_instr(cctx, ISN_COND2BOOL)) == NULL)
521 return FAIL;
522
523 // type becomes bool
Bram Moolenaar078a4612022-01-04 15:17:03 +0000524 set_type_on_stack(cctx, &t_bool, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000525
526 return OK;
527}
528
529 int
530generate_TYPECHECK(
531 cctx_T *cctx,
532 type_T *expected,
533 int offset,
Bram Moolenaarbd3a9d22022-05-17 16:12:39 +0100534 int is_var,
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000535 int argidx)
536{
537 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000538
539 RETURN_OK_IF_SKIP(cctx);
540 if ((isn = generate_instr(cctx, ISN_CHECKTYPE)) == NULL)
541 return FAIL;
542 isn->isn_arg.type.ct_type = alloc_type(expected);
543 isn->isn_arg.type.ct_off = (int8_T)offset;
Bram Moolenaarbd3a9d22022-05-17 16:12:39 +0100544 isn->isn_arg.type.ct_is_var = is_var;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000545 isn->isn_arg.type.ct_arg_idx = (int8_T)argidx;
546
547 // type becomes expected
Bram Moolenaar078a4612022-01-04 15:17:03 +0000548 set_type_on_stack(cctx, expected, -1 - offset);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000549
550 return OK;
551}
552
553 int
554generate_SETTYPE(
555 cctx_T *cctx,
556 type_T *expected)
557{
558 isn_T *isn;
559
560 RETURN_OK_IF_SKIP(cctx);
561 if ((isn = generate_instr(cctx, ISN_SETTYPE)) == NULL)
562 return FAIL;
563 isn->isn_arg.type.ct_type = alloc_type(expected);
564 return OK;
565}
566
567/*
568 * Generate a PUSH instruction for "tv".
569 * "tv" will be consumed or cleared.
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000570 */
571 int
572generate_tv_PUSH(cctx_T *cctx, typval_T *tv)
573{
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100574 switch (tv->v_type)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000575 {
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100576 case VAR_BOOL:
577 generate_PUSHBOOL(cctx, tv->vval.v_number);
578 break;
579 case VAR_SPECIAL:
580 generate_PUSHSPEC(cctx, tv->vval.v_number);
581 break;
582 case VAR_NUMBER:
583 generate_PUSHNR(cctx, tv->vval.v_number);
584 break;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000585#ifdef FEAT_FLOAT
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100586 case VAR_FLOAT:
587 generate_PUSHF(cctx, tv->vval.v_float);
588 break;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000589#endif
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100590 case VAR_BLOB:
591 generate_PUSHBLOB(cctx, tv->vval.v_blob);
592 tv->vval.v_blob = NULL;
593 break;
594 case VAR_LIST:
595 if (tv->vval.v_list != NULL)
596 iemsg("non-empty list constant not supported");
597 generate_NEWLIST(cctx, 0, TRUE);
598 break;
599 case VAR_DICT:
600 if (tv->vval.v_dict != NULL)
601 iemsg("non-empty dict constant not supported");
602 generate_NEWDICT(cctx, 0, TRUE);
603 break;
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000604#ifdef FEAT_JOB_CHANNEL
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100605 case VAR_JOB:
606 if (tv->vval.v_job != NULL)
607 iemsg("non-null job constant not supported");
608 generate_PUSHJOB(cctx);
609 break;
610 case VAR_CHANNEL:
611 if (tv->vval.v_channel != NULL)
612 iemsg("non-null channel constant not supported");
613 generate_PUSHCHANNEL(cctx);
614 break;
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000615#endif
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100616 case VAR_FUNC:
617 if (tv->vval.v_string != NULL)
618 iemsg("non-null function constant not supported");
619 generate_PUSHFUNC(cctx, NULL, &t_func_unknown);
620 break;
621 case VAR_PARTIAL:
622 if (tv->vval.v_partial != NULL)
623 iemsg("non-null partial constant not supported");
624 if (generate_instr_type(cctx, ISN_NEWPARTIAL, &t_func_unknown)
625 == NULL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000626 return FAIL;
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100627 break;
628 case VAR_STRING:
629 generate_PUSHS(cctx, &tv->vval.v_string);
630 tv->vval.v_string = NULL;
631 break;
632 default:
633 siemsg("constant type %d not supported", tv->v_type);
634 clear_tv(tv);
635 return FAIL;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000636 }
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100637 tv->v_type = VAR_UNKNOWN;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000638 return OK;
639}
640
641/*
642 * Generate an ISN_PUSHNR instruction.
643 */
644 int
645generate_PUSHNR(cctx_T *cctx, varnumber_T number)
646{
647 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000648
649 RETURN_OK_IF_SKIP(cctx);
650 if ((isn = generate_instr_type(cctx, ISN_PUSHNR, &t_number)) == NULL)
651 return FAIL;
652 isn->isn_arg.number = number;
653
654 if (number == 0 || number == 1)
655 // A 0 or 1 number can also be used as a bool.
Bram Moolenaar078a4612022-01-04 15:17:03 +0000656 set_type_on_stack(cctx, &t_number_bool, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000657 return OK;
658}
659
660/*
661 * Generate an ISN_PUSHBOOL instruction.
662 */
663 int
664generate_PUSHBOOL(cctx_T *cctx, varnumber_T number)
665{
666 isn_T *isn;
667
668 RETURN_OK_IF_SKIP(cctx);
669 if ((isn = generate_instr_type(cctx, ISN_PUSHBOOL, &t_bool)) == NULL)
670 return FAIL;
671 isn->isn_arg.number = number;
672
673 return OK;
674}
675
676/*
677 * Generate an ISN_PUSHSPEC instruction.
678 */
679 int
680generate_PUSHSPEC(cctx_T *cctx, varnumber_T number)
681{
682 isn_T *isn;
683
684 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +0000685 if ((isn = generate_instr_type(cctx, ISN_PUSHSPEC,
686 number == VVAL_NULL ? &t_null : &t_none)) == NULL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000687 return FAIL;
688 isn->isn_arg.number = number;
689
690 return OK;
691}
692
693#if defined(FEAT_FLOAT) || defined(PROTO)
694/*
695 * Generate an ISN_PUSHF instruction.
696 */
697 int
698generate_PUSHF(cctx_T *cctx, float_T fnumber)
699{
700 isn_T *isn;
701
702 RETURN_OK_IF_SKIP(cctx);
703 if ((isn = generate_instr_type(cctx, ISN_PUSHF, &t_float)) == NULL)
704 return FAIL;
705 isn->isn_arg.fnumber = fnumber;
706
707 return OK;
708}
709#endif
710
711/*
712 * Generate an ISN_PUSHS instruction.
713 * Consumes "*str". When freed *str is set to NULL, unless "str" is NULL.
Bram Moolenaar0abc2872022-05-10 13:24:30 +0100714 * Note that if "str" is used in the instruction OK is returned and "*str" is
715 * not set to NULL.
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000716 */
717 int
718generate_PUSHS(cctx_T *cctx, char_u **str)
719{
720 isn_T *isn;
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100721 int ret = OK;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000722
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100723 if (cctx->ctx_skip != SKIP_YES)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000724 {
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100725 if ((isn = generate_instr_type(cctx, ISN_PUSHS, &t_string)) == NULL)
726 ret = FAIL;
727 else
728 {
729 isn->isn_arg.string = str == NULL ? NULL : *str;
730 return OK;
731 }
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000732 }
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100733 if (str != NULL)
734 VIM_CLEAR(*str);
735 return ret;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000736}
737
738/*
Bram Moolenaar397a87a2022-03-20 21:14:15 +0000739 * Generate an ISN_PUSHCHANNEL instruction. Channel is always NULL.
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000740 */
741 int
Bram Moolenaar397a87a2022-03-20 21:14:15 +0000742generate_PUSHCHANNEL(cctx_T *cctx)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000743{
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100744#ifdef FEAT_JOB_CHANNEL
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000745 isn_T *isn;
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100746#endif
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000747
748 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100749#ifdef FEAT_JOB_CHANNEL
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000750 if ((isn = generate_instr_type(cctx, ISN_PUSHCHANNEL, &t_channel)) == NULL)
751 return FAIL;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000752 return OK;
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100753#else
754 emsg(_(e_channel_job_feature_not_available));
755 return FAIL;
756#endif
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000757}
758
759/*
Bram Moolenaar397a87a2022-03-20 21:14:15 +0000760 * Generate an ISN_PUSHJOB instruction. Job is always NULL.
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000761 */
762 int
Bram Moolenaar397a87a2022-03-20 21:14:15 +0000763generate_PUSHJOB(cctx_T *cctx)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000764{
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100765#ifdef FEAT_JOB_CHANNEL
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000766 isn_T *isn;
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100767#endif
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000768
769 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100770#ifdef FEAT_JOB_CHANNEL
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000771 if ((isn = generate_instr_type(cctx, ISN_PUSHJOB, &t_job)) == NULL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000772 return FAIL;
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000773 return OK;
Bram Moolenaarc9af6172022-05-04 16:46:54 +0100774#else
775 emsg(_(e_channel_job_feature_not_available));
776 return FAIL;
777#endif
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000778}
779
780/*
781 * Generate an ISN_PUSHBLOB instruction.
782 * Consumes "blob".
783 */
784 int
785generate_PUSHBLOB(cctx_T *cctx, blob_T *blob)
786{
787 isn_T *isn;
788
789 RETURN_OK_IF_SKIP(cctx);
790 if ((isn = generate_instr_type(cctx, ISN_PUSHBLOB, &t_blob)) == NULL)
791 return FAIL;
792 isn->isn_arg.blob = blob;
793
794 return OK;
795}
796
797/*
798 * Generate an ISN_PUSHFUNC instruction with name "name".
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000799 */
800 int
801generate_PUSHFUNC(cctx_T *cctx, char_u *name, type_T *type)
802{
803 isn_T *isn;
804 char_u *funcname;
805
806 RETURN_OK_IF_SKIP(cctx);
807 if ((isn = generate_instr_type(cctx, ISN_PUSHFUNC, type)) == NULL)
808 return FAIL;
809 if (name == NULL)
810 funcname = NULL;
Bram Moolenaard041f422022-01-12 19:54:00 +0000811 else if (*name == K_SPECIAL // script-local
812 || vim_strchr(name, AUTOLOAD_CHAR) != NULL) // autoload
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000813 funcname = vim_strsave(name);
814 else
815 {
816 funcname = alloc(STRLEN(name) + 3);
817 if (funcname != NULL)
818 {
819 STRCPY(funcname, "g:");
820 STRCPY(funcname + 2, name);
821 }
822 }
823
824 isn->isn_arg.string = funcname;
825 return OK;
826}
827
828/*
Bram Moolenaar06b77222022-01-25 15:51:56 +0000829 * Generate an ISN_AUTOLOAD instruction.
830 */
831 int
832generate_AUTOLOAD(cctx_T *cctx, char_u *name, type_T *type)
833{
834 isn_T *isn;
835
836 RETURN_OK_IF_SKIP(cctx);
837 if ((isn = generate_instr_type(cctx, ISN_AUTOLOAD, type)) == NULL)
838 return FAIL;
839 isn->isn_arg.string = vim_strsave(name);
840 if (isn->isn_arg.string == NULL)
841 return FAIL;
842 return OK;
843}
844
845/*
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000846 * Generate an ISN_GETITEM instruction with "index".
847 * "with_op" is TRUE for "+=" and other operators, the stack has the current
848 * value below the list with values.
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +0100849 * Caller must check the type is a list.
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000850 */
851 int
852generate_GETITEM(cctx_T *cctx, int index, int with_op)
853{
854 isn_T *isn;
Bram Moolenaar078a4612022-01-04 15:17:03 +0000855 type_T *type = get_type_on_stack(cctx, with_op ? 1 : 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000856 type_T *item_type = &t_any;
857
858 RETURN_OK_IF_SKIP(cctx);
859
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000860 item_type = type->tt_member;
861 if ((isn = generate_instr(cctx, ISN_GETITEM)) == NULL)
862 return FAIL;
863 isn->isn_arg.getitem.gi_index = index;
864 isn->isn_arg.getitem.gi_with_op = with_op;
865
866 // add the item type to the type stack
Bram Moolenaar078a4612022-01-04 15:17:03 +0000867 return push_type_stack(cctx, item_type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000868}
869
870/*
871 * Generate an ISN_SLICE instruction with "count".
872 */
873 int
874generate_SLICE(cctx_T *cctx, int count)
875{
876 isn_T *isn;
877
878 RETURN_OK_IF_SKIP(cctx);
879 if ((isn = generate_instr(cctx, ISN_SLICE)) == NULL)
880 return FAIL;
881 isn->isn_arg.number = count;
882 return OK;
883}
884
885/*
886 * Generate an ISN_CHECKLEN instruction with "min_len".
887 */
888 int
889generate_CHECKLEN(cctx_T *cctx, int min_len, int more_OK)
890{
891 isn_T *isn;
892
893 RETURN_OK_IF_SKIP(cctx);
894
895 if ((isn = generate_instr(cctx, ISN_CHECKLEN)) == NULL)
896 return FAIL;
897 isn->isn_arg.checklen.cl_min_len = min_len;
898 isn->isn_arg.checklen.cl_more_OK = more_OK;
899
900 return OK;
901}
902
903/*
904 * Generate an ISN_STORE instruction.
905 */
906 int
907generate_STORE(cctx_T *cctx, isntype_T isn_type, int idx, char_u *name)
908{
909 isn_T *isn;
910
911 RETURN_OK_IF_SKIP(cctx);
912 if ((isn = generate_instr_drop(cctx, isn_type, 1)) == NULL)
913 return FAIL;
914 if (name != NULL)
915 isn->isn_arg.string = vim_strsave(name);
916 else
917 isn->isn_arg.number = idx;
918
919 return OK;
920}
921
922/*
923 * Generate an ISN_STOREOUTER instruction.
924 */
Yegappan Lakshmanan782b43d2022-01-08 18:43:40 +0000925 static int
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000926generate_STOREOUTER(cctx_T *cctx, int idx, int level)
927{
928 isn_T *isn;
929
930 RETURN_OK_IF_SKIP(cctx);
931 if ((isn = generate_instr_drop(cctx, ISN_STOREOUTER, 1)) == NULL)
932 return FAIL;
933 isn->isn_arg.outer.outer_idx = idx;
934 isn->isn_arg.outer.outer_depth = level;
935
936 return OK;
937}
938
939/*
940 * Generate an ISN_STORENR instruction (short for ISN_PUSHNR + ISN_STORE)
941 */
942 int
943generate_STORENR(cctx_T *cctx, int idx, varnumber_T value)
944{
945 isn_T *isn;
946
947 RETURN_OK_IF_SKIP(cctx);
948 if ((isn = generate_instr(cctx, ISN_STORENR)) == NULL)
949 return FAIL;
950 isn->isn_arg.storenr.stnr_idx = idx;
951 isn->isn_arg.storenr.stnr_val = value;
952
953 return OK;
954}
955
956/*
957 * Generate an ISN_STOREOPT or ISN_STOREFUNCOPT instruction
958 */
Yegappan Lakshmanan782b43d2022-01-08 18:43:40 +0000959 static int
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000960generate_STOREOPT(
961 cctx_T *cctx,
962 isntype_T isn_type,
963 char_u *name,
964 int opt_flags)
965{
966 isn_T *isn;
967
968 RETURN_OK_IF_SKIP(cctx);
969 if ((isn = generate_instr_drop(cctx, isn_type, 1)) == NULL)
970 return FAIL;
971 isn->isn_arg.storeopt.so_name = vim_strsave(name);
972 isn->isn_arg.storeopt.so_flags = opt_flags;
973
974 return OK;
975}
976
977/*
978 * Generate an ISN_LOAD or similar instruction.
979 */
980 int
981generate_LOAD(
982 cctx_T *cctx,
983 isntype_T isn_type,
984 int idx,
985 char_u *name,
986 type_T *type)
987{
988 isn_T *isn;
989
990 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaar078a4612022-01-04 15:17:03 +0000991 if ((isn = generate_instr_type2(cctx, isn_type, type, type)) == NULL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000992 return FAIL;
993 if (name != NULL)
994 isn->isn_arg.string = vim_strsave(name);
995 else
996 isn->isn_arg.number = idx;
997
998 return OK;
999}
1000
1001/*
1002 * Generate an ISN_LOADOUTER instruction
1003 */
1004 int
1005generate_LOADOUTER(
1006 cctx_T *cctx,
1007 int idx,
1008 int nesting,
1009 type_T *type)
1010{
1011 isn_T *isn;
1012
1013 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaar078a4612022-01-04 15:17:03 +00001014 if ((isn = generate_instr_type2(cctx, ISN_LOADOUTER, type, type)) == NULL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001015 return FAIL;
1016 isn->isn_arg.outer.outer_idx = idx;
1017 isn->isn_arg.outer.outer_depth = nesting;
1018
1019 return OK;
1020}
1021
1022/*
1023 * Generate an ISN_LOADV instruction for v:var.
1024 */
1025 int
1026generate_LOADV(
1027 cctx_T *cctx,
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +01001028 char_u *name)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001029{
1030 int di_flags;
1031 int vidx = find_vim_var(name, &di_flags);
1032 type_T *type;
1033
1034 RETURN_OK_IF_SKIP(cctx);
1035 if (vidx < 0)
1036 {
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +01001037 semsg(_(e_variable_not_found_str), name);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001038 return FAIL;
1039 }
Bram Moolenaard787e402021-12-24 21:36:12 +00001040 type = get_vim_var_type(vidx, cctx->ctx_type_list);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001041 return generate_LOAD(cctx, ISN_LOADV, vidx, NULL, type);
1042}
1043
1044/*
1045 * Generate an ISN_UNLET instruction.
1046 */
1047 int
1048generate_UNLET(cctx_T *cctx, isntype_T isn_type, char_u *name, int forceit)
1049{
1050 isn_T *isn;
1051
1052 RETURN_OK_IF_SKIP(cctx);
1053 if ((isn = generate_instr(cctx, isn_type)) == NULL)
1054 return FAIL;
1055 isn->isn_arg.unlet.ul_name = vim_strsave(name);
1056 isn->isn_arg.unlet.ul_forceit = forceit;
1057
1058 return OK;
1059}
1060
1061/*
1062 * Generate an ISN_LOCKCONST instruction.
1063 */
1064 int
1065generate_LOCKCONST(cctx_T *cctx)
1066{
1067 isn_T *isn;
1068
1069 RETURN_OK_IF_SKIP(cctx);
1070 if ((isn = generate_instr(cctx, ISN_LOCKCONST)) == NULL)
1071 return FAIL;
1072 return OK;
1073}
1074
1075/*
1076 * Generate an ISN_LOADS instruction.
1077 */
1078 int
1079generate_OLDSCRIPT(
1080 cctx_T *cctx,
1081 isntype_T isn_type,
1082 char_u *name,
1083 int sid,
1084 type_T *type)
1085{
1086 isn_T *isn;
1087
1088 RETURN_OK_IF_SKIP(cctx);
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01001089 if (isn_type == ISN_LOADS || isn_type == ISN_LOADEXPORT)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001090 isn = generate_instr_type(cctx, isn_type, type);
1091 else
1092 isn = generate_instr_drop(cctx, isn_type, 1);
1093 if (isn == NULL)
1094 return FAIL;
1095 isn->isn_arg.loadstore.ls_name = vim_strsave(name);
1096 isn->isn_arg.loadstore.ls_sid = sid;
1097
1098 return OK;
1099}
1100
1101/*
1102 * Generate an ISN_LOADSCRIPT or ISN_STORESCRIPT instruction.
1103 */
1104 int
1105generate_VIM9SCRIPT(
1106 cctx_T *cctx,
1107 isntype_T isn_type,
1108 int sid,
1109 int idx,
1110 type_T *type)
1111{
1112 isn_T *isn;
1113 scriptref_T *sref;
1114 scriptitem_T *si = SCRIPT_ITEM(sid);
1115
1116 RETURN_OK_IF_SKIP(cctx);
1117 if (isn_type == ISN_LOADSCRIPT)
Bram Moolenaar160afdb2022-02-06 17:17:02 +00001118 isn = generate_instr_type2(cctx, isn_type, type, type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001119 else
1120 isn = generate_instr_drop(cctx, isn_type, 1);
1121 if (isn == NULL)
1122 return FAIL;
1123
1124 // This requires three arguments, which doesn't fit in an instruction, thus
1125 // we need to allocate a struct for this.
1126 sref = ALLOC_ONE(scriptref_T);
1127 if (sref == NULL)
1128 return FAIL;
1129 isn->isn_arg.script.scriptref = sref;
1130 sref->sref_sid = sid;
1131 sref->sref_idx = idx;
1132 sref->sref_seq = si->sn_script_seq;
1133 sref->sref_type = type;
1134 return OK;
1135}
1136
1137/*
Bram Moolenaarec15b1c2022-03-27 16:29:53 +01001138 * Generate an ISN_NEWLIST instruction for "count" items.
1139 * "use_null" is TRUE for null_list.
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001140 */
1141 int
Bram Moolenaarec15b1c2022-03-27 16:29:53 +01001142generate_NEWLIST(cctx_T *cctx, int count, int use_null)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001143{
1144 isn_T *isn;
Bram Moolenaar078a4612022-01-04 15:17:03 +00001145 type_T *member_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001146 type_T *type;
Bram Moolenaar078a4612022-01-04 15:17:03 +00001147 type_T *decl_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001148
1149 RETURN_OK_IF_SKIP(cctx);
1150 if ((isn = generate_instr(cctx, ISN_NEWLIST)) == NULL)
1151 return FAIL;
Bram Moolenaarec15b1c2022-03-27 16:29:53 +01001152 isn->isn_arg.number = use_null ? -1 : count;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001153
Bram Moolenaar078a4612022-01-04 15:17:03 +00001154 // Get the member type and the declared member type from all the items on
1155 // the stack.
Bram Moolenaar2626d6a2022-02-06 15:49:35 +00001156 member_type = get_member_type_from_stack(count, 1, cctx);
Bram Moolenaar078a4612022-01-04 15:17:03 +00001157 type = get_list_type(member_type, cctx->ctx_type_list);
Bram Moolenaar2626d6a2022-02-06 15:49:35 +00001158 decl_type = get_list_type(&t_any, cctx->ctx_type_list);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001159
1160 // drop the value types
Bram Moolenaar078a4612022-01-04 15:17:03 +00001161 cctx->ctx_type_stack.ga_len -= count;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001162
1163 // add the list type to the type stack
Bram Moolenaar078a4612022-01-04 15:17:03 +00001164 return push_type_stack2(cctx, type, decl_type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001165}
1166
1167/*
1168 * Generate an ISN_NEWDICT instruction.
Bram Moolenaarec15b1c2022-03-27 16:29:53 +01001169 * "use_null" is TRUE for null_dict.
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001170 */
1171 int
Bram Moolenaarec15b1c2022-03-27 16:29:53 +01001172generate_NEWDICT(cctx_T *cctx, int count, int use_null)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001173{
1174 isn_T *isn;
Bram Moolenaar078a4612022-01-04 15:17:03 +00001175 type_T *member_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001176 type_T *type;
Bram Moolenaar078a4612022-01-04 15:17:03 +00001177 type_T *decl_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001178
1179 RETURN_OK_IF_SKIP(cctx);
1180 if ((isn = generate_instr(cctx, ISN_NEWDICT)) == NULL)
1181 return FAIL;
Bram Moolenaarec15b1c2022-03-27 16:29:53 +01001182 isn->isn_arg.number = use_null ? -1 : count;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001183
Bram Moolenaar2626d6a2022-02-06 15:49:35 +00001184 member_type = get_member_type_from_stack(count, 2, cctx);
Bram Moolenaar078a4612022-01-04 15:17:03 +00001185 type = get_dict_type(member_type, cctx->ctx_type_list);
Bram Moolenaar2626d6a2022-02-06 15:49:35 +00001186 decl_type = get_dict_type(&t_any, cctx->ctx_type_list);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001187
1188 // drop the key and value types
Bram Moolenaar078a4612022-01-04 15:17:03 +00001189 cctx->ctx_type_stack.ga_len -= 2 * count;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001190
1191 // add the dict type to the type stack
Bram Moolenaar078a4612022-01-04 15:17:03 +00001192 return push_type_stack2(cctx, type, decl_type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001193}
1194
1195/*
1196 * Generate an ISN_FUNCREF instruction.
Bram Moolenaara915fa02022-03-23 11:29:15 +00001197 * "isnp" is set to the instruction, so that fr_dfunc_idx can be set later.
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001198 */
1199 int
Bram Moolenaara915fa02022-03-23 11:29:15 +00001200generate_FUNCREF(cctx_T *cctx, ufunc_T *ufunc, isn_T **isnp)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001201{
1202 isn_T *isn;
Bram Moolenaar078a4612022-01-04 15:17:03 +00001203 type_T *type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001204
1205 RETURN_OK_IF_SKIP(cctx);
1206 if ((isn = generate_instr(cctx, ISN_FUNCREF)) == NULL)
1207 return FAIL;
Bram Moolenaara915fa02022-03-23 11:29:15 +00001208 if (isnp != NULL)
1209 *isnp = isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001210 if (ufunc->uf_def_status == UF_NOT_COMPILED)
1211 isn->isn_arg.funcref.fr_func_name = vim_strsave(ufunc->uf_name);
1212 else
1213 isn->isn_arg.funcref.fr_dfunc_idx = ufunc->uf_dfunc_idx;
1214 cctx->ctx_has_closure = 1;
1215
1216 // If the referenced function is a closure, it may use items further up in
Bram Moolenaar139575d2022-03-15 19:29:30 +00001217 // the nested context, including this one. But not a function defined at
1218 // the script level.
1219 if ((ufunc->uf_flags & FC_CLOSURE)
1220 && func_name_refcount(cctx->ctx_ufunc->uf_name))
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001221 cctx->ctx_ufunc->uf_flags |= FC_CLOSURE;
1222
Bram Moolenaar078a4612022-01-04 15:17:03 +00001223 type = ufunc->uf_func_type == NULL ? &t_func_any : ufunc->uf_func_type;
1224 return push_type_stack(cctx, type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001225}
1226
1227/*
1228 * Generate an ISN_NEWFUNC instruction.
1229 * "lambda_name" and "func_name" must be in allocated memory and will be
1230 * consumed.
1231 */
1232 int
1233generate_NEWFUNC(cctx_T *cctx, char_u *lambda_name, char_u *func_name)
1234{
1235 isn_T *isn;
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +01001236 int ret = OK;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001237
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +01001238 if (cctx->ctx_skip != SKIP_YES)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001239 {
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +01001240 if ((isn = generate_instr(cctx, ISN_NEWFUNC)) == NULL)
1241 ret = FAIL;
1242 else
1243 {
1244 isn->isn_arg.newfunc.nf_lambda = lambda_name;
1245 isn->isn_arg.newfunc.nf_global = func_name;
1246 return OK;
1247 }
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001248 }
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +01001249 vim_free(lambda_name);
1250 vim_free(func_name);
1251 return ret;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001252}
1253
1254/*
1255 * Generate an ISN_DEF instruction: list functions
1256 */
1257 int
1258generate_DEF(cctx_T *cctx, char_u *name, size_t len)
1259{
1260 isn_T *isn;
1261
1262 RETURN_OK_IF_SKIP(cctx);
1263 if ((isn = generate_instr(cctx, ISN_DEF)) == NULL)
1264 return FAIL;
1265 if (len > 0)
1266 {
1267 isn->isn_arg.string = vim_strnsave(name, len);
1268 if (isn->isn_arg.string == NULL)
1269 return FAIL;
1270 }
1271 return OK;
1272}
1273
1274/*
1275 * Generate an ISN_JUMP instruction.
1276 */
1277 int
1278generate_JUMP(cctx_T *cctx, jumpwhen_T when, int where)
1279{
1280 isn_T *isn;
1281 garray_T *stack = &cctx->ctx_type_stack;
1282
1283 RETURN_OK_IF_SKIP(cctx);
1284 if ((isn = generate_instr(cctx, ISN_JUMP)) == NULL)
1285 return FAIL;
1286 isn->isn_arg.jump.jump_when = when;
1287 isn->isn_arg.jump.jump_where = where;
1288
1289 if (when != JUMP_ALWAYS && stack->ga_len > 0)
1290 --stack->ga_len;
1291
1292 return OK;
1293}
1294
1295/*
1296 * Generate an ISN_JUMP_IF_ARG_SET instruction.
1297 */
1298 int
1299generate_JUMP_IF_ARG_SET(cctx_T *cctx, int arg_off)
1300{
1301 isn_T *isn;
1302
1303 RETURN_OK_IF_SKIP(cctx);
1304 if ((isn = generate_instr(cctx, ISN_JUMP_IF_ARG_SET)) == NULL)
1305 return FAIL;
1306 isn->isn_arg.jumparg.jump_arg_off = arg_off;
1307 // jump_where is set later
1308 return OK;
1309}
1310
1311 int
1312generate_FOR(cctx_T *cctx, int loop_idx)
1313{
1314 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001315
1316 RETURN_OK_IF_SKIP(cctx);
1317 if ((isn = generate_instr(cctx, ISN_FOR)) == NULL)
1318 return FAIL;
1319 isn->isn_arg.forloop.for_idx = loop_idx;
1320
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001321 // type doesn't matter, will be stored next
Bram Moolenaar078a4612022-01-04 15:17:03 +00001322 return push_type_stack(cctx, &t_any);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001323}
1324/*
1325 * Generate an ISN_TRYCONT instruction.
1326 */
1327 int
1328generate_TRYCONT(cctx_T *cctx, int levels, int where)
1329{
1330 isn_T *isn;
1331
1332 RETURN_OK_IF_SKIP(cctx);
1333 if ((isn = generate_instr(cctx, ISN_TRYCONT)) == NULL)
1334 return FAIL;
1335 isn->isn_arg.trycont.tct_levels = levels;
1336 isn->isn_arg.trycont.tct_where = where;
1337
1338 return OK;
1339}
1340
1341
1342/*
1343 * Generate an ISN_BCALL instruction.
1344 * "method_call" is TRUE for "value->method()"
1345 * Return FAIL if the number of arguments is wrong.
1346 */
1347 int
1348generate_BCALL(cctx_T *cctx, int func_idx, int argcount, int method_call)
1349{
1350 isn_T *isn;
1351 garray_T *stack = &cctx->ctx_type_stack;
1352 int argoff;
Bram Moolenaar078a4612022-01-04 15:17:03 +00001353 type2_T *typep;
1354 type2_T *argtypes = NULL;
1355 type2_T shuffled_argtypes[MAX_FUNC_ARGS];
1356 type2_T *maptype = NULL;
1357 type_T *type;
Bram Moolenaar81330182022-02-01 12:11:58 +00001358 type_T *decl_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001359
1360 RETURN_OK_IF_SKIP(cctx);
1361 argoff = check_internal_func(func_idx, argcount);
1362 if (argoff < 0)
1363 return FAIL;
1364
1365 if (method_call && argoff > 1)
1366 {
1367 if ((isn = generate_instr(cctx, ISN_SHUFFLE)) == NULL)
1368 return FAIL;
1369 isn->isn_arg.shuffle.shfl_item = argcount;
1370 isn->isn_arg.shuffle.shfl_up = argoff - 1;
1371 }
1372
1373 if (argcount > 0)
1374 {
1375 // Check the types of the arguments.
Bram Moolenaar078a4612022-01-04 15:17:03 +00001376 typep = ((type2_T *)stack->ga_data) + stack->ga_len - argcount;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001377 if (method_call && argoff > 1)
1378 {
1379 int i;
1380
1381 for (i = 0; i < argcount; ++i)
1382 shuffled_argtypes[i] = (i < argoff - 1)
Bram Moolenaar078a4612022-01-04 15:17:03 +00001383 ? typep[i + 1]
1384 : (i == argoff - 1) ? typep[0] : typep[i];
1385 argtypes = shuffled_argtypes;
1386 }
1387 else
1388 {
1389 int i;
1390
1391 for (i = 0; i < argcount; ++i)
1392 shuffled_argtypes[i] = typep[i];
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001393 argtypes = shuffled_argtypes;
1394 }
1395 if (internal_func_check_arg_types(argtypes, func_idx, argcount,
1396 cctx) == FAIL)
1397 return FAIL;
1398 if (internal_func_is_map(func_idx))
Bram Moolenaar078a4612022-01-04 15:17:03 +00001399 maptype = argtypes;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001400 }
1401
1402 if ((isn = generate_instr(cctx, ISN_BCALL)) == NULL)
1403 return FAIL;
1404 isn->isn_arg.bfunc.cbf_idx = func_idx;
1405 isn->isn_arg.bfunc.cbf_argcount = argcount;
1406
1407 // Drop the argument types and push the return type.
1408 stack->ga_len -= argcount;
Bram Moolenaar81330182022-02-01 12:11:58 +00001409 type = internal_func_ret_type(func_idx, argcount, argtypes, &decl_type);
1410 if (push_type_stack2(cctx, type, decl_type) == FAIL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001411 return FAIL;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001412
Bram Moolenaar35c807d2022-01-27 16:36:29 +00001413 if (maptype != NULL && maptype[0].type_decl->tt_member != NULL
1414 && maptype[0].type_decl->tt_member != &t_any)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001415 // Check that map() didn't change the item types.
Bram Moolenaarbd3a9d22022-05-17 16:12:39 +01001416 generate_TYPECHECK(cctx, maptype[0].type_decl, -1, FALSE, 1);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001417
1418 return OK;
1419}
1420
1421/*
1422 * Generate an ISN_LISTAPPEND instruction. Works like add().
1423 * Argument count is already checked.
1424 */
1425 int
1426generate_LISTAPPEND(cctx_T *cctx)
1427{
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001428 type_T *list_type;
1429 type_T *item_type;
1430 type_T *expected;
1431
1432 // Caller already checked that list_type is a list.
Bram Moolenaar81330182022-02-01 12:11:58 +00001433 // For checking the item type we use the declared type of the list and the
1434 // current type of the added item, adding a string to [1, 2] is OK.
1435 list_type = get_decl_type_on_stack(cctx, 1);
Bram Moolenaar078a4612022-01-04 15:17:03 +00001436 item_type = get_type_on_stack(cctx, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001437 expected = list_type->tt_member;
1438 if (need_type(item_type, expected, -1, 0, cctx, FALSE, FALSE) == FAIL)
1439 return FAIL;
1440
1441 if (generate_instr(cctx, ISN_LISTAPPEND) == NULL)
1442 return FAIL;
1443
Bram Moolenaar078a4612022-01-04 15:17:03 +00001444 --cctx->ctx_type_stack.ga_len; // drop the argument
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001445 return OK;
1446}
1447
1448/*
1449 * Generate an ISN_BLOBAPPEND instruction. Works like add().
1450 * Argument count is already checked.
1451 */
1452 int
1453generate_BLOBAPPEND(cctx_T *cctx)
1454{
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001455 type_T *item_type;
1456
1457 // Caller already checked that blob_type is a blob.
Bram Moolenaar078a4612022-01-04 15:17:03 +00001458 item_type = get_type_on_stack(cctx, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001459 if (need_type(item_type, &t_number, -1, 0, cctx, FALSE, FALSE) == FAIL)
1460 return FAIL;
1461
1462 if (generate_instr(cctx, ISN_BLOBAPPEND) == NULL)
1463 return FAIL;
1464
Bram Moolenaar078a4612022-01-04 15:17:03 +00001465 --cctx->ctx_type_stack.ga_len; // drop the argument
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001466 return OK;
1467}
1468
1469/*
1470 * Generate an ISN_DCALL or ISN_UCALL instruction.
1471 * Return FAIL if the number of arguments is wrong.
1472 */
1473 int
1474generate_CALL(cctx_T *cctx, ufunc_T *ufunc, int pushed_argcount)
1475{
1476 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001477 int regular_args = ufunc->uf_args.ga_len;
1478 int argcount = pushed_argcount;
1479
1480 RETURN_OK_IF_SKIP(cctx);
1481 if (argcount > regular_args && !has_varargs(ufunc))
1482 {
1483 semsg(_(e_too_many_arguments_for_function_str),
1484 printable_func_name(ufunc));
1485 return FAIL;
1486 }
1487 if (argcount < regular_args - ufunc->uf_def_args.ga_len)
1488 {
1489 semsg(_(e_not_enough_arguments_for_function_str),
1490 printable_func_name(ufunc));
1491 return FAIL;
1492 }
1493
1494 if (ufunc->uf_def_status != UF_NOT_COMPILED
1495 && ufunc->uf_def_status != UF_COMPILE_ERROR)
1496 {
1497 int i;
Bram Moolenaar139575d2022-03-15 19:29:30 +00001498 compiletype_T compile_type;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001499
1500 for (i = 0; i < argcount; ++i)
1501 {
1502 type_T *expected;
1503 type_T *actual;
1504
Bram Moolenaar078a4612022-01-04 15:17:03 +00001505 actual = get_type_on_stack(cctx, argcount - i - 1);
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +00001506 if (actual->tt_type == VAR_SPECIAL
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001507 && i >= regular_args - ufunc->uf_def_args.ga_len)
1508 {
1509 // assume v:none used for default argument value
1510 continue;
1511 }
1512 if (i < regular_args)
1513 {
1514 if (ufunc->uf_arg_types == NULL)
1515 continue;
1516 expected = ufunc->uf_arg_types[i];
1517 }
1518 else if (ufunc->uf_va_type == NULL
1519 || ufunc->uf_va_type == &t_list_any)
1520 // possibly a lambda or "...: any"
1521 expected = &t_any;
1522 else
1523 expected = ufunc->uf_va_type->tt_member;
1524 if (need_type(actual, expected, -argcount + i, i + 1, cctx,
1525 TRUE, FALSE) == FAIL)
1526 {
1527 arg_type_mismatch(expected, actual, i + 1);
1528 return FAIL;
1529 }
1530 }
Bram Moolenaar139575d2022-03-15 19:29:30 +00001531 compile_type = get_compile_type(ufunc);
1532 if (func_needs_compiling(ufunc, compile_type)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001533 && compile_def_function(ufunc, ufunc->uf_ret_type == NULL,
Bram Moolenaar139575d2022-03-15 19:29:30 +00001534 compile_type, NULL) == FAIL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001535 return FAIL;
1536 }
1537 if (ufunc->uf_def_status == UF_COMPILE_ERROR)
1538 {
Bram Moolenaara6c18d32022-03-31 20:02:56 +01001539 emsg_funcname(e_call_to_function_that_failed_to_compile_str,
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001540 ufunc->uf_name);
1541 return FAIL;
1542 }
1543
1544 if ((isn = generate_instr(cctx,
1545 ufunc->uf_def_status != UF_NOT_COMPILED ? ISN_DCALL
1546 : ISN_UCALL)) == NULL)
1547 return FAIL;
1548 if (isn->isn_type == ISN_DCALL)
1549 {
1550 isn->isn_arg.dfunc.cdf_idx = ufunc->uf_dfunc_idx;
1551 isn->isn_arg.dfunc.cdf_argcount = argcount;
1552 }
1553 else
1554 {
1555 // A user function may be deleted and redefined later, can't use the
1556 // ufunc pointer, need to look it up again at runtime.
1557 isn->isn_arg.ufunc.cuf_name = vim_strsave(ufunc->uf_name);
1558 isn->isn_arg.ufunc.cuf_argcount = argcount;
1559 }
1560
Bram Moolenaar078a4612022-01-04 15:17:03 +00001561 // drop the argument types
1562 cctx->ctx_type_stack.ga_len -= argcount;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001563
Bram Moolenaar078a4612022-01-04 15:17:03 +00001564 // add return type
1565 return push_type_stack(cctx, ufunc->uf_ret_type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001566}
1567
1568/*
1569 * Generate an ISN_UCALL instruction when the function isn't defined yet.
1570 */
1571 int
1572generate_UCALL(cctx_T *cctx, char_u *name, int argcount)
1573{
1574 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001575
1576 RETURN_OK_IF_SKIP(cctx);
1577 if ((isn = generate_instr(cctx, ISN_UCALL)) == NULL)
1578 return FAIL;
1579 isn->isn_arg.ufunc.cuf_name = vim_strsave(name);
1580 isn->isn_arg.ufunc.cuf_argcount = argcount;
1581
Bram Moolenaar078a4612022-01-04 15:17:03 +00001582 // drop the argument types
1583 cctx->ctx_type_stack.ga_len -= argcount;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001584
Bram Moolenaar078a4612022-01-04 15:17:03 +00001585 // add return value
1586 return push_type_stack(cctx, &t_any);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001587}
1588
1589/*
1590 * Generate an ISN_PCALL instruction.
1591 * "type" is the type of the FuncRef.
1592 */
1593 int
1594generate_PCALL(
1595 cctx_T *cctx,
1596 int argcount,
1597 char_u *name,
1598 type_T *type,
1599 int at_top)
1600{
1601 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001602 type_T *ret_type;
1603
1604 RETURN_OK_IF_SKIP(cctx);
1605
Bram Moolenaar59618fe2021-12-21 12:32:17 +00001606 if (type->tt_type == VAR_ANY || type->tt_type == VAR_UNKNOWN)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001607 ret_type = &t_any;
1608 else if (type->tt_type == VAR_FUNC || type->tt_type == VAR_PARTIAL)
1609 {
1610 if (type->tt_argcount != -1)
1611 {
1612 int varargs = (type->tt_flags & TTFLAG_VARARGS) ? 1 : 0;
1613
1614 if (argcount < type->tt_min_argcount - varargs)
1615 {
Bram Moolenaara6c18d32022-03-31 20:02:56 +01001616 emsg_funcname(e_not_enough_arguments_for_function_str, name);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001617 return FAIL;
1618 }
1619 if (!varargs && argcount > type->tt_argcount)
1620 {
Bram Moolenaara6c18d32022-03-31 20:02:56 +01001621 emsg_funcname(e_too_many_arguments_for_function_str, name);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001622 return FAIL;
1623 }
1624 if (type->tt_args != NULL)
1625 {
1626 int i;
1627
1628 for (i = 0; i < argcount; ++i)
1629 {
1630 int offset = -argcount + i - (at_top ? 0 : 1);
Bram Moolenaar078a4612022-01-04 15:17:03 +00001631 type_T *actual = get_type_on_stack(cctx, -1 - offset);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001632 type_T *expected;
1633
1634 if (varargs && i >= type->tt_argcount - 1)
1635 expected = type->tt_args[
1636 type->tt_argcount - 1]->tt_member;
1637 else if (i >= type->tt_min_argcount
Bram Moolenaar53ba6ca2022-03-10 19:23:28 +00001638 && actual->tt_type == VAR_SPECIAL)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001639 expected = &t_any;
1640 else
1641 expected = type->tt_args[i];
1642 if (need_type(actual, expected, offset, i + 1,
1643 cctx, TRUE, FALSE) == FAIL)
1644 {
1645 arg_type_mismatch(expected, actual, i + 1);
1646 return FAIL;
1647 }
1648 }
1649 }
1650 }
1651 ret_type = type->tt_member;
1652 if (ret_type == &t_unknown)
1653 // return type not known yet, use a runtime check
1654 ret_type = &t_any;
1655 }
1656 else
1657 {
1658 semsg(_(e_not_callable_type_str), name);
1659 return FAIL;
1660 }
1661
1662 if ((isn = generate_instr(cctx, ISN_PCALL)) == NULL)
1663 return FAIL;
1664 isn->isn_arg.pfunc.cpf_top = at_top;
1665 isn->isn_arg.pfunc.cpf_argcount = argcount;
1666
Bram Moolenaar078a4612022-01-04 15:17:03 +00001667 // drop the arguments and the funcref/partial
1668 cctx->ctx_type_stack.ga_len -= argcount + 1;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001669
Bram Moolenaar078a4612022-01-04 15:17:03 +00001670 // push the return value
1671 push_type_stack(cctx, ret_type);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001672
1673 // If partial is above the arguments it must be cleared and replaced with
1674 // the return value.
1675 if (at_top && generate_instr(cctx, ISN_PCALL_END) == NULL)
1676 return FAIL;
1677
1678 return OK;
1679}
1680
1681/*
1682 * Generate an ISN_STRINGMEMBER instruction.
1683 */
1684 int
1685generate_STRINGMEMBER(cctx_T *cctx, char_u *name, size_t len)
1686{
1687 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001688 type_T *type;
1689
1690 RETURN_OK_IF_SKIP(cctx);
1691 if ((isn = generate_instr(cctx, ISN_STRINGMEMBER)) == NULL)
1692 return FAIL;
1693 isn->isn_arg.string = vim_strnsave(name, len);
1694
1695 // check for dict type
Bram Moolenaar078a4612022-01-04 15:17:03 +00001696 type = get_type_on_stack(cctx, 0);
Bram Moolenaar59618fe2021-12-21 12:32:17 +00001697 if (type->tt_type != VAR_DICT && type != &t_any && type != &t_unknown)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001698 {
1699 char *tofree;
1700
1701 semsg(_(e_expected_dictionary_for_using_key_str_but_got_str),
1702 name, type_name(type, &tofree));
1703 vim_free(tofree);
1704 return FAIL;
1705 }
1706 // change dict type to dict member type
1707 if (type->tt_type == VAR_DICT)
1708 {
Bram Moolenaar078a4612022-01-04 15:17:03 +00001709 type_T *ntype = type->tt_member == &t_unknown
1710 ? &t_any : type->tt_member;
1711 set_type_on_stack(cctx, ntype, 0);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001712 }
1713
1714 return OK;
1715}
1716
1717/*
1718 * Generate an ISN_ECHO instruction.
1719 */
1720 int
1721generate_ECHO(cctx_T *cctx, int with_white, int count)
1722{
1723 isn_T *isn;
1724
1725 RETURN_OK_IF_SKIP(cctx);
1726 if ((isn = generate_instr_drop(cctx, ISN_ECHO, count)) == NULL)
1727 return FAIL;
1728 isn->isn_arg.echo.echo_with_white = with_white;
1729 isn->isn_arg.echo.echo_count = count;
1730
1731 return OK;
1732}
1733
1734/*
1735 * Generate an ISN_EXECUTE/ISN_ECHOMSG/ISN_ECHOERR instruction.
1736 */
1737 int
1738generate_MULT_EXPR(cctx_T *cctx, isntype_T isn_type, int count)
1739{
1740 isn_T *isn;
1741
1742 if ((isn = generate_instr_drop(cctx, isn_type, count)) == NULL)
1743 return FAIL;
1744 isn->isn_arg.number = count;
1745
1746 return OK;
1747}
1748
1749/*
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01001750 * Generate an ISN_SOURCE instruction.
1751 */
1752 int
1753generate_SOURCE(cctx_T *cctx, int sid)
1754{
1755 isn_T *isn;
1756
1757 if ((isn = generate_instr(cctx, ISN_SOURCE)) == NULL)
1758 return FAIL;
1759 isn->isn_arg.number = sid;
1760
1761 return OK;
1762}
1763
1764/*
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001765 * Generate an ISN_PUT instruction.
1766 */
1767 int
1768generate_PUT(cctx_T *cctx, int regname, linenr_T lnum)
1769{
1770 isn_T *isn;
1771
1772 RETURN_OK_IF_SKIP(cctx);
1773 if ((isn = generate_instr(cctx, ISN_PUT)) == NULL)
1774 return FAIL;
1775 isn->isn_arg.put.put_regname = regname;
1776 isn->isn_arg.put.put_lnum = lnum;
1777 return OK;
1778}
1779
1780/*
1781 * Generate an EXEC instruction that takes a string argument.
1782 * A copy is made of "line".
1783 */
1784 int
1785generate_EXEC_copy(cctx_T *cctx, isntype_T isntype, char_u *line)
1786{
1787 isn_T *isn;
1788
1789 RETURN_OK_IF_SKIP(cctx);
1790 if ((isn = generate_instr(cctx, isntype)) == NULL)
1791 return FAIL;
1792 isn->isn_arg.string = vim_strsave(line);
1793 return OK;
1794}
1795
1796/*
1797 * Generate an EXEC instruction that takes a string argument.
1798 * "str" must be allocated, it is consumed.
1799 */
1800 int
1801generate_EXEC(cctx_T *cctx, isntype_T isntype, char_u *str)
1802{
1803 isn_T *isn;
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +01001804 int ret = OK;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001805
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +01001806 if (cctx->ctx_skip != SKIP_YES)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001807 {
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +01001808 if ((isn = generate_instr(cctx, isntype)) == NULL)
1809 ret = FAIL;
1810 else
1811 {
1812 isn->isn_arg.string = str;
1813 return OK;
1814 }
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001815 }
Bram Moolenaarc3caa7f2022-05-25 19:15:10 +01001816 vim_free(str);
1817 return ret;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001818}
1819
1820 int
1821generate_LEGACY_EVAL(cctx_T *cctx, char_u *line)
1822{
1823 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001824
1825 RETURN_OK_IF_SKIP(cctx);
1826 if ((isn = generate_instr(cctx, ISN_LEGACY_EVAL)) == NULL)
1827 return FAIL;
1828 isn->isn_arg.string = vim_strsave(line);
1829
Bram Moolenaar078a4612022-01-04 15:17:03 +00001830 return push_type_stack(cctx, &t_any);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001831}
1832
1833 int
1834generate_EXECCONCAT(cctx_T *cctx, int count)
1835{
1836 isn_T *isn;
1837
1838 if ((isn = generate_instr_drop(cctx, ISN_EXECCONCAT, count)) == NULL)
1839 return FAIL;
1840 isn->isn_arg.number = count;
1841 return OK;
1842}
1843
1844/*
1845 * Generate ISN_RANGE. Consumes "range". Return OK/FAIL.
1846 */
1847 int
1848generate_RANGE(cctx_T *cctx, char_u *range)
1849{
1850 isn_T *isn;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001851
1852 if ((isn = generate_instr(cctx, ISN_RANGE)) == NULL)
1853 return FAIL;
1854 isn->isn_arg.string = range;
1855
Bram Moolenaar078a4612022-01-04 15:17:03 +00001856 return push_type_stack(cctx, &t_number);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001857}
1858
1859 int
1860generate_UNPACK(cctx_T *cctx, int var_count, int semicolon)
1861{
1862 isn_T *isn;
1863
1864 RETURN_OK_IF_SKIP(cctx);
1865 if ((isn = generate_instr(cctx, ISN_UNPACK)) == NULL)
1866 return FAIL;
1867 isn->isn_arg.unpack.unp_count = var_count;
1868 isn->isn_arg.unpack.unp_semicolon = semicolon;
1869 return OK;
1870}
1871
1872/*
1873 * Generate an instruction for any command modifiers.
1874 */
1875 int
1876generate_cmdmods(cctx_T *cctx, cmdmod_T *cmod)
1877{
1878 isn_T *isn;
1879
1880 if (has_cmdmod(cmod, FALSE))
1881 {
1882 cctx->ctx_has_cmdmod = TRUE;
1883
1884 if ((isn = generate_instr(cctx, ISN_CMDMOD)) == NULL)
1885 return FAIL;
1886 isn->isn_arg.cmdmod.cf_cmdmod = ALLOC_ONE(cmdmod_T);
1887 if (isn->isn_arg.cmdmod.cf_cmdmod == NULL)
1888 return FAIL;
1889 mch_memmove(isn->isn_arg.cmdmod.cf_cmdmod, cmod, sizeof(cmdmod_T));
1890 // filter program now belongs to the instruction
1891 cmod->cmod_filter_regmatch.regprog = NULL;
1892 }
1893
1894 return OK;
1895}
1896
1897 int
1898generate_undo_cmdmods(cctx_T *cctx)
1899{
1900 if (cctx->ctx_has_cmdmod && generate_instr(cctx, ISN_CMDMOD_REV) == NULL)
1901 return FAIL;
1902 cctx->ctx_has_cmdmod = FALSE;
1903 return OK;
1904}
1905
1906/*
1907 * Generate a STORE instruction for "dest", not being "dest_local".
1908 * Return FAIL when out of memory.
1909 */
1910 int
1911generate_store_var(
1912 cctx_T *cctx,
1913 assign_dest_T dest,
1914 int opt_flags,
1915 int vimvaridx,
1916 int scriptvar_idx,
1917 int scriptvar_sid,
1918 type_T *type,
1919 char_u *name)
1920{
1921 switch (dest)
1922 {
1923 case dest_option:
1924 return generate_STOREOPT(cctx, ISN_STOREOPT,
1925 skip_option_env_lead(name), opt_flags);
1926 case dest_func_option:
1927 return generate_STOREOPT(cctx, ISN_STOREFUNCOPT,
1928 skip_option_env_lead(name), opt_flags);
1929 case dest_global:
1930 // include g: with the name, easier to execute that way
1931 return generate_STORE(cctx, vim_strchr(name, AUTOLOAD_CHAR) == NULL
1932 ? ISN_STOREG : ISN_STOREAUTO, 0, name);
1933 case dest_buffer:
1934 // include b: with the name, easier to execute that way
1935 return generate_STORE(cctx, ISN_STOREB, 0, name);
1936 case dest_window:
1937 // include w: with the name, easier to execute that way
1938 return generate_STORE(cctx, ISN_STOREW, 0, name);
1939 case dest_tab:
1940 // include t: with the name, easier to execute that way
1941 return generate_STORE(cctx, ISN_STORET, 0, name);
1942 case dest_env:
1943 return generate_STORE(cctx, ISN_STOREENV, 0, name + 1);
1944 case dest_reg:
1945 return generate_STORE(cctx, ISN_STOREREG,
1946 name[1] == '@' ? '"' : name[1], NULL);
1947 case dest_vimvar:
1948 return generate_STORE(cctx, ISN_STOREV, vimvaridx, NULL);
1949 case dest_script:
1950 if (scriptvar_idx < 0)
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01001951 {
1952 isntype_T isn_type = ISN_STORES;
1953
1954 if (SCRIPT_ID_VALID(scriptvar_sid)
Bram Moolenaarccbfd482022-03-31 16:18:23 +01001955 && SCRIPT_ITEM(scriptvar_sid)->sn_import_autoload
1956 && SCRIPT_ITEM(scriptvar_sid)->sn_autoload_prefix
1957 == NULL)
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01001958 {
1959 // "import autoload './dir/script.vim'" - load script first
1960 if (generate_SOURCE(cctx, scriptvar_sid) == FAIL)
1961 return FAIL;
1962 isn_type = ISN_STOREEXPORT;
1963 }
1964
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001965 // "s:" may be included in the name.
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01001966 return generate_OLDSCRIPT(cctx, isn_type, name,
1967 scriptvar_sid, type);
1968 }
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001969 return generate_VIM9SCRIPT(cctx, ISN_STORESCRIPT,
1970 scriptvar_sid, scriptvar_idx, type);
1971 case dest_local:
1972 case dest_expr:
1973 // cannot happen
1974 break;
1975 }
1976 return FAIL;
1977}
1978
Bram Moolenaar38ecd972022-01-15 21:44:44 +00001979/*
1980 * Return TRUE when inside a "for" or "while" loop.
1981 */
1982 int
1983inside_loop_scope(cctx_T *cctx)
1984{
1985 scope_T *scope = cctx->ctx_scope;
1986
1987 for (;;)
1988 {
1989 if (scope == NULL)
1990 break;
1991 if (scope->se_type == FOR_SCOPE || scope->se_type == WHILE_SCOPE)
1992 return TRUE;
1993 scope = scope->se_outer;
1994 }
1995 return FALSE;
1996}
1997
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001998 int
Bram Moolenaar5cd64792021-12-25 18:23:24 +00001999generate_store_lhs(cctx_T *cctx, lhs_T *lhs, int instr_count, int is_decl)
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002000{
2001 if (lhs->lhs_dest != dest_local)
2002 return generate_store_var(cctx, lhs->lhs_dest,
2003 lhs->lhs_opt_flags, lhs->lhs_vimvaridx,
2004 lhs->lhs_scriptvar_idx, lhs->lhs_scriptvar_sid,
2005 lhs->lhs_type, lhs->lhs_name);
2006
2007 if (lhs->lhs_lvar != NULL)
2008 {
2009 garray_T *instr = &cctx->ctx_instr;
2010 isn_T *isn = ((isn_T *)instr->ga_data) + instr->ga_len - 1;
2011
Bram Moolenaar5cd64792021-12-25 18:23:24 +00002012 // Optimization: turn "var = 123" from ISN_PUSHNR + ISN_STORE into
2013 // ISN_STORENR.
2014 // And "var = 0" does not need any instruction.
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002015 if (lhs->lhs_lvar->lv_from_outer == 0
2016 && instr->ga_len == instr_count + 1
2017 && isn->isn_type == ISN_PUSHNR)
2018 {
2019 varnumber_T val = isn->isn_arg.number;
2020 garray_T *stack = &cctx->ctx_type_stack;
2021
Bram Moolenaar38ecd972022-01-15 21:44:44 +00002022 if (val == 0 && is_decl && !inside_loop_scope(cctx))
Bram Moolenaar5cd64792021-12-25 18:23:24 +00002023 {
Bram Moolenaar38ecd972022-01-15 21:44:44 +00002024 // zero is the default value, no need to do anything
Bram Moolenaar5cd64792021-12-25 18:23:24 +00002025 --instr->ga_len;
2026 }
2027 else
2028 {
2029 isn->isn_type = ISN_STORENR;
2030 isn->isn_arg.storenr.stnr_idx = lhs->lhs_lvar->lv_idx;
2031 isn->isn_arg.storenr.stnr_val = val;
2032 }
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002033 if (stack->ga_len > 0)
2034 --stack->ga_len;
2035 }
2036 else if (lhs->lhs_lvar->lv_from_outer > 0)
2037 generate_STOREOUTER(cctx, lhs->lhs_lvar->lv_idx,
2038 lhs->lhs_lvar->lv_from_outer);
2039 else
2040 generate_STORE(cctx, ISN_STORE, lhs->lhs_lvar->lv_idx, NULL);
2041 }
2042 return OK;
2043}
2044
2045#if defined(FEAT_PROFILE) || defined(PROTO)
2046 void
2047may_generate_prof_end(cctx_T *cctx, int prof_lnum)
2048{
2049 if (cctx->ctx_compile_type == CT_PROFILE && prof_lnum >= 0)
2050 generate_instr(cctx, ISN_PROF_END);
2051}
2052#endif
2053
2054
2055/*
2056 * Delete an instruction, free what it contains.
2057 */
2058 void
2059delete_instr(isn_T *isn)
2060{
2061 switch (isn->isn_type)
2062 {
Bram Moolenaar06b77222022-01-25 15:51:56 +00002063 case ISN_AUTOLOAD:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002064 case ISN_DEF:
2065 case ISN_EXEC:
2066 case ISN_EXECRANGE:
2067 case ISN_EXEC_SPLIT:
2068 case ISN_LEGACY_EVAL:
2069 case ISN_LOADAUTO:
2070 case ISN_LOADB:
2071 case ISN_LOADENV:
2072 case ISN_LOADG:
2073 case ISN_LOADOPT:
2074 case ISN_LOADT:
2075 case ISN_LOADW:
2076 case ISN_LOCKUNLOCK:
2077 case ISN_PUSHEXC:
2078 case ISN_PUSHFUNC:
2079 case ISN_PUSHS:
2080 case ISN_RANGE:
2081 case ISN_STOREAUTO:
2082 case ISN_STOREB:
2083 case ISN_STOREENV:
2084 case ISN_STOREG:
2085 case ISN_STORET:
2086 case ISN_STOREW:
2087 case ISN_STRINGMEMBER:
2088 vim_free(isn->isn_arg.string);
2089 break;
2090
2091 case ISN_SUBSTITUTE:
2092 {
2093 int idx;
2094 isn_T *list = isn->isn_arg.subs.subs_instr;
2095
2096 vim_free(isn->isn_arg.subs.subs_cmd);
2097 for (idx = 0; list[idx].isn_type != ISN_FINISH; ++idx)
2098 delete_instr(list + idx);
2099 vim_free(list);
2100 }
2101 break;
2102
2103 case ISN_INSTR:
2104 {
2105 int idx;
2106 isn_T *list = isn->isn_arg.instr;
2107
2108 for (idx = 0; list[idx].isn_type != ISN_FINISH; ++idx)
2109 delete_instr(list + idx);
2110 vim_free(list);
2111 }
2112 break;
2113
2114 case ISN_LOADS:
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002115 case ISN_LOADEXPORT:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002116 case ISN_STORES:
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002117 case ISN_STOREEXPORT:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002118 vim_free(isn->isn_arg.loadstore.ls_name);
2119 break;
2120
2121 case ISN_UNLET:
2122 case ISN_UNLETENV:
2123 vim_free(isn->isn_arg.unlet.ul_name);
2124 break;
2125
2126 case ISN_STOREOPT:
2127 case ISN_STOREFUNCOPT:
2128 vim_free(isn->isn_arg.storeopt.so_name);
2129 break;
2130
2131 case ISN_PUSHBLOB: // push blob isn_arg.blob
2132 blob_unref(isn->isn_arg.blob);
2133 break;
2134
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002135 case ISN_UCALL:
2136 vim_free(isn->isn_arg.ufunc.cuf_name);
2137 break;
2138
2139 case ISN_FUNCREF:
2140 {
2141 if (isn->isn_arg.funcref.fr_func_name == NULL)
2142 {
2143 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002144 + isn->isn_arg.funcref.fr_dfunc_idx;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002145 ufunc_T *ufunc = dfunc->df_ufunc;
2146
2147 if (ufunc != NULL && func_name_refcount(ufunc->uf_name))
2148 func_ptr_unref(ufunc);
2149 }
2150 else
2151 {
2152 char_u *name = isn->isn_arg.funcref.fr_func_name;
2153
2154 if (name != NULL)
2155 func_unref(name);
2156 vim_free(isn->isn_arg.funcref.fr_func_name);
2157 }
2158 }
2159 break;
2160
2161 case ISN_DCALL:
2162 {
2163 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002164 + isn->isn_arg.dfunc.cdf_idx;
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002165
2166 if (dfunc->df_ufunc != NULL
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002167 && func_name_refcount(dfunc->df_ufunc->uf_name))
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002168 func_ptr_unref(dfunc->df_ufunc);
2169 }
2170 break;
2171
2172 case ISN_NEWFUNC:
2173 {
2174 char_u *lambda = isn->isn_arg.newfunc.nf_lambda;
Bram Moolenaaracc4b562022-01-24 13:54:45 +00002175 ufunc_T *ufunc = find_func_even_dead(lambda, FFED_IS_GLOBAL);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002176
2177 if (ufunc != NULL)
2178 {
2179 unlink_def_function(ufunc);
2180 func_ptr_unref(ufunc);
2181 }
2182
2183 vim_free(lambda);
2184 vim_free(isn->isn_arg.newfunc.nf_global);
2185 }
2186 break;
2187
2188 case ISN_CHECKTYPE:
2189 case ISN_SETTYPE:
2190 free_type(isn->isn_arg.type.ct_type);
2191 break;
2192
2193 case ISN_CMDMOD:
2194 vim_regfree(isn->isn_arg.cmdmod.cf_cmdmod
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002195 ->cmod_filter_regmatch.regprog);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002196 vim_free(isn->isn_arg.cmdmod.cf_cmdmod);
2197 break;
2198
2199 case ISN_LOADSCRIPT:
2200 case ISN_STORESCRIPT:
2201 vim_free(isn->isn_arg.script.scriptref);
2202 break;
2203
2204 case ISN_TRY:
Bram Moolenaar0d807102021-12-21 09:42:09 +00002205 vim_free(isn->isn_arg.tryref.try_ref);
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002206 break;
2207
2208 case ISN_CEXPR_CORE:
2209 vim_free(isn->isn_arg.cexpr.cexpr_ref->cer_cmdline);
2210 vim_free(isn->isn_arg.cexpr.cexpr_ref);
2211 break;
2212
2213 case ISN_2BOOL:
2214 case ISN_2STRING:
2215 case ISN_2STRING_ANY:
2216 case ISN_ADDBLOB:
2217 case ISN_ADDLIST:
2218 case ISN_ANYINDEX:
2219 case ISN_ANYSLICE:
2220 case ISN_BCALL:
2221 case ISN_BLOBAPPEND:
2222 case ISN_BLOBINDEX:
2223 case ISN_BLOBSLICE:
2224 case ISN_CATCH:
2225 case ISN_CEXPR_AUCMD:
2226 case ISN_CHECKLEN:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002227 case ISN_CLEARDICT:
2228 case ISN_CMDMOD_REV:
2229 case ISN_COMPAREANY:
2230 case ISN_COMPAREBLOB:
2231 case ISN_COMPAREBOOL:
2232 case ISN_COMPAREDICT:
2233 case ISN_COMPAREFLOAT:
2234 case ISN_COMPAREFUNC:
2235 case ISN_COMPARELIST:
2236 case ISN_COMPARENR:
Bram Moolenaar7a222242022-03-01 19:23:24 +00002237 case ISN_COMPARENULL:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002238 case ISN_COMPARESPECIAL:
2239 case ISN_COMPARESTRING:
2240 case ISN_CONCAT:
2241 case ISN_COND2BOOL:
2242 case ISN_DEBUG:
2243 case ISN_DROP:
2244 case ISN_ECHO:
2245 case ISN_ECHOCONSOLE:
2246 case ISN_ECHOERR:
2247 case ISN_ECHOMSG:
2248 case ISN_ENDTRY:
2249 case ISN_EXECCONCAT:
2250 case ISN_EXECUTE:
2251 case ISN_FINALLY:
2252 case ISN_FINISH:
2253 case ISN_FOR:
2254 case ISN_GETITEM:
2255 case ISN_JUMP:
2256 case ISN_JUMP_IF_ARG_SET:
2257 case ISN_LISTAPPEND:
2258 case ISN_LISTINDEX:
2259 case ISN_LISTSLICE:
2260 case ISN_LOAD:
2261 case ISN_LOADBDICT:
2262 case ISN_LOADGDICT:
2263 case ISN_LOADOUTER:
2264 case ISN_LOADREG:
2265 case ISN_LOADTDICT:
2266 case ISN_LOADV:
2267 case ISN_LOADWDICT:
2268 case ISN_LOCKCONST:
2269 case ISN_MEMBER:
2270 case ISN_NEGATENR:
2271 case ISN_NEWDICT:
2272 case ISN_NEWLIST:
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +00002273 case ISN_NEWPARTIAL:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002274 case ISN_OPANY:
2275 case ISN_OPFLOAT:
2276 case ISN_OPNR:
2277 case ISN_PCALL:
2278 case ISN_PCALL_END:
2279 case ISN_PROF_END:
2280 case ISN_PROF_START:
2281 case ISN_PUSHBOOL:
Bram Moolenaar397a87a2022-03-20 21:14:15 +00002282 case ISN_PUSHCHANNEL:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002283 case ISN_PUSHF:
Bram Moolenaar397a87a2022-03-20 21:14:15 +00002284 case ISN_PUSHJOB:
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002285 case ISN_PUSHNR:
2286 case ISN_PUSHSPEC:
2287 case ISN_PUT:
2288 case ISN_REDIREND:
2289 case ISN_REDIRSTART:
2290 case ISN_RETURN:
2291 case ISN_RETURN_VOID:
2292 case ISN_SHUFFLE:
2293 case ISN_SLICE:
2294 case ISN_STORE:
2295 case ISN_STOREINDEX:
2296 case ISN_STORENR:
Bram Moolenaarc0ceeeb2022-03-30 21:12:27 +01002297 case ISN_SOURCE:
2298 case ISN_STOREOUTER:
2299 case ISN_STORERANGE:
2300 case ISN_STOREREG:
2301 case ISN_STOREV:
2302 case ISN_STRINDEX:
2303 case ISN_STRSLICE:
2304 case ISN_THROW:
2305 case ISN_TRYCONT:
2306 case ISN_UNLETINDEX:
2307 case ISN_UNLETRANGE:
2308 case ISN_UNPACK:
2309 case ISN_USEDICT:
2310 // nothing allocated
2311 break;
2312}
Bram Moolenaardc7c3662021-12-20 15:04:29 +00002313}
2314
2315 void
2316clear_instr_ga(garray_T *gap)
2317{
2318 int idx;
2319
2320 for (idx = 0; idx < gap->ga_len; ++idx)
2321 delete_instr(((isn_T *)gap->ga_data) + idx);
2322 ga_clear(gap);
2323}
2324
2325
2326#endif // defined(FEAT_EVAL)