blob: 399720aaf5e421e496aad15790eaa992c7524002 [file] [log] [blame]
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001/* 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 * vim9execute.c: execute Vim9 script instructions
12 */
13
14#define USING_FLOAT_STUFF
15#include "vim.h"
16
17#if defined(FEAT_EVAL) || defined(PROTO)
18
19#ifdef VMS
20# include <float.h>
21#endif
22
23#include "vim9.h"
24
25// Structure put on ec_trystack when ISN_TRY is encountered.
26typedef struct {
27 int tcd_frame; // ec_frame when ISN_TRY was encountered
28 int tcd_catch_idx; // instruction of the first catch
29 int tcd_finally_idx; // instruction of the finally block
30 int tcd_caught; // catch block entered
31 int tcd_return; // when TRUE return from end of :finally
32} trycmd_T;
33
34
35// A stack is used to store:
36// - arguments passed to a :def function
37// - info about the calling function, to use when returning
38// - local variables
39// - temporary values
40//
41// In detail (FP == Frame Pointer):
42// arg1 first argument from caller (if present)
43// arg2 second argument from caller (if present)
44// extra_arg1 any missing optional argument default value
45// FP -> cur_func calling function
46// current previous instruction pointer
47// frame_ptr previous Frame Pointer
48// var1 space for local variable
49// var2 space for local variable
50// .... fixed space for max. number of local variables
51// temp temporary values
52// .... flexible space for temporary values (can grow big)
53
54/*
55 * Execution context.
56 */
57typedef struct {
58 garray_T ec_stack; // stack of typval_T values
59 int ec_frame; // index in ec_stack: context of ec_dfunc_idx
60
61 garray_T ec_trystack; // stack of trycmd_T values
62 int ec_in_catch; // when TRUE in catch or finally block
63
64 int ec_dfunc_idx; // current function index
65 isn_T *ec_instr; // array with instructions
66 int ec_iidx; // index in ec_instr: instruction to execute
67} ectx_T;
68
69// Get pointer to item relative to the bottom of the stack, -1 is the last one.
70#define STACK_TV_BOT(idx) (((typval_T *)ectx->ec_stack.ga_data) + ectx->ec_stack.ga_len + idx)
71
72/*
Bram Moolenaar170fcfc2020-02-06 17:51:35 +010073 * Return the number of arguments, including optional arguments and any vararg.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010074 */
75 static int
76ufunc_argcount(ufunc_T *ufunc)
77{
78 return ufunc->uf_args.ga_len + (ufunc->uf_va_name != NULL ? 1 : 0);
79}
80
81/*
Bram Moolenaar170fcfc2020-02-06 17:51:35 +010082 * Set the instruction index, depending on omitted arguments, where the default
83 * values are to be computed. If all optional arguments are present, start
84 * with the function body.
85 * The expression evaluation is at the start of the instructions:
86 * 0 -> EVAL default1
87 * STORE arg[-2]
88 * 1 -> EVAL default2
89 * STORE arg[-1]
90 * 2 -> function body
91 */
92 static void
93init_instr_idx(ufunc_T *ufunc, int argcount, ectx_T *ectx)
94{
95 if (ufunc->uf_def_args.ga_len == 0)
96 ectx->ec_iidx = 0;
97 else
98 {
99 int defcount = ufunc->uf_args.ga_len - argcount;
100
101 // If there is a varargs argument defcount can be negative, no defaults
102 // to evaluate then.
103 if (defcount < 0)
104 defcount = 0;
105 ectx->ec_iidx = ufunc->uf_def_arg_idx[
106 ufunc->uf_def_args.ga_len - defcount];
107 }
108}
109
110/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100111 * Call compiled function "cdf_idx" from compiled code.
112 *
113 * Stack has:
114 * - current arguments (already there)
115 * - omitted optional argument (default values) added here
116 * - stack frame:
117 * - pointer to calling function
118 * - Index of next instruction in calling function
119 * - previous frame pointer
120 * - reserved space for local variables
121 */
122 static int
123call_dfunc(int cdf_idx, int argcount, ectx_T *ectx)
124{
125 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data) + cdf_idx;
126 ufunc_T *ufunc = dfunc->df_ufunc;
127 int optcount = ufunc_argcount(ufunc) - argcount;
128 int idx;
129
130 if (dfunc->df_deleted)
131 {
132 emsg_funcname(e_func_deleted, ufunc->uf_name);
133 return FAIL;
134 }
135
136 if (ga_grow(&ectx->ec_stack, optcount + 3 + dfunc->df_varcount) == FAIL)
137 return FAIL;
138
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100139 if (optcount < 0)
140 {
141 emsg("argument count wrong?");
142 return FAIL;
143 }
Bram Moolenaar170fcfc2020-02-06 17:51:35 +0100144
145 // Reserve space for omitted optional arguments, filled in soon.
146 // Also any empty varargs argument.
147 ectx->ec_stack.ga_len += optcount;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100148
149 // Store current execution state in stack frame for ISN_RETURN.
150 // TODO: If the actual number of arguments doesn't match what the called
151 // function expects things go bad.
152 STACK_TV_BOT(0)->vval.v_number = ectx->ec_dfunc_idx;
153 STACK_TV_BOT(1)->vval.v_number = ectx->ec_iidx;
154 STACK_TV_BOT(2)->vval.v_number = ectx->ec_frame;
155 ectx->ec_frame = ectx->ec_stack.ga_len;
156
157 // Initialize local variables
158 for (idx = 0; idx < dfunc->df_varcount; ++idx)
159 STACK_TV_BOT(STACK_FRAME_SIZE + idx)->v_type = VAR_UNKNOWN;
160 ectx->ec_stack.ga_len += STACK_FRAME_SIZE + dfunc->df_varcount;
161
162 // Set execution state to the start of the called function.
163 ectx->ec_dfunc_idx = cdf_idx;
164 ectx->ec_instr = dfunc->df_instr;
165 estack_push_ufunc(ETYPE_UFUNC, dfunc->df_ufunc, 1);
Bram Moolenaar170fcfc2020-02-06 17:51:35 +0100166
167 // Decide where to start execution, handles optional arguments.
168 init_instr_idx(ufunc, argcount, ectx);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100169
170 return OK;
171}
172
173// Get pointer to item in the stack.
174#define STACK_TV(idx) (((typval_T *)ectx->ec_stack.ga_data) + idx)
175
176/*
177 * Return from the current function.
178 */
179 static void
180func_return(ectx_T *ectx)
181{
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100182 int idx;
183 dfunc_T *dfunc;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +0100184 int top;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100185
186 // execution context goes one level up
187 estack_pop();
188
189 // Clear the local variables and temporary values, but not
190 // the return value.
191 for (idx = ectx->ec_frame + STACK_FRAME_SIZE;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +0100192 idx < ectx->ec_stack.ga_len - 1; ++idx)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100193 clear_tv(STACK_TV(idx));
Bram Moolenaar170fcfc2020-02-06 17:51:35 +0100194
195 // Clear the arguments.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100196 dfunc = ((dfunc_T *)def_functions.ga_data) + ectx->ec_dfunc_idx;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +0100197 top = ectx->ec_frame - ufunc_argcount(dfunc->df_ufunc);
198 for (idx = top; idx < ectx->ec_frame; ++idx)
199 clear_tv(STACK_TV(idx));
200
201 // Restore the previous frame.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100202 ectx->ec_dfunc_idx = STACK_TV(ectx->ec_frame)->vval.v_number;
203 ectx->ec_iidx = STACK_TV(ectx->ec_frame + 1)->vval.v_number;
204 ectx->ec_frame = STACK_TV(ectx->ec_frame + 2)->vval.v_number;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100205 dfunc = ((dfunc_T *)def_functions.ga_data) + ectx->ec_dfunc_idx;
206 ectx->ec_instr = dfunc->df_instr;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +0100207
208 // Reset the stack to the position before the call, move the return value
209 // to the top of the stack.
210 idx = ectx->ec_stack.ga_len - 1;
211 ectx->ec_stack.ga_len = top + 1;
212 *STACK_TV_BOT(-1) = *STACK_TV(idx);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100213}
214
215#undef STACK_TV
216
217/*
218 * Prepare arguments and rettv for calling a builtin or user function.
219 */
220 static int
221call_prepare(int argcount, typval_T *argvars, ectx_T *ectx)
222{
223 int idx;
224 typval_T *tv;
225
226 // Move arguments from bottom of the stack to argvars[] and add terminator.
227 for (idx = 0; idx < argcount; ++idx)
228 argvars[idx] = *STACK_TV_BOT(idx - argcount);
229 argvars[argcount].v_type = VAR_UNKNOWN;
230
231 // Result replaces the arguments on the stack.
232 if (argcount > 0)
233 ectx->ec_stack.ga_len -= argcount - 1;
234 else if (ga_grow(&ectx->ec_stack, 1) == FAIL)
235 return FAIL;
236 else
237 ++ectx->ec_stack.ga_len;
238
239 // Default return value is zero.
240 tv = STACK_TV_BOT(-1);
241 tv->v_type = VAR_NUMBER;
242 tv->vval.v_number = 0;
243
244 return OK;
245}
246
247/*
248 * Call a builtin function by index.
249 */
250 static int
251call_bfunc(int func_idx, int argcount, ectx_T *ectx)
252{
253 typval_T argvars[MAX_FUNC_ARGS];
254 int idx;
255
256 if (call_prepare(argcount, argvars, ectx) == FAIL)
257 return FAIL;
258
259 // Call the builtin function.
260 call_internal_func_by_idx(func_idx, argvars, STACK_TV_BOT(-1));
261
262 // Clear the arguments.
263 for (idx = 0; idx < argcount; ++idx)
264 clear_tv(&argvars[idx]);
265 return OK;
266}
267
268/*
269 * Execute a user defined function.
270 */
271 static int
272call_ufunc(ufunc_T *ufunc, int argcount, ectx_T *ectx)
273{
274 typval_T argvars[MAX_FUNC_ARGS];
275 funcexe_T funcexe;
276 int error;
277 int idx;
278
279 if (ufunc->uf_dfunc_idx >= 0)
280 // The function has been compiled, can call it quickly.
281 return call_dfunc(ufunc->uf_dfunc_idx, argcount, ectx);
282
283 if (call_prepare(argcount, argvars, ectx) == FAIL)
284 return FAIL;
285 vim_memset(&funcexe, 0, sizeof(funcexe));
286 funcexe.evaluate = TRUE;
287
288 // Call the user function. Result goes in last position on the stack.
289 // TODO: add selfdict if there is one
290 error = call_user_func_check(ufunc, argcount, argvars,
291 STACK_TV_BOT(-1), &funcexe, NULL);
292
293 // Clear the arguments.
294 for (idx = 0; idx < argcount; ++idx)
295 clear_tv(&argvars[idx]);
296
297 if (error != FCERR_NONE)
298 {
299 user_func_error(error, ufunc->uf_name);
300 return FAIL;
301 }
302 return OK;
303}
304
305/*
306 * Execute a function by "name".
307 * This can be a builtin function or a user function.
308 * Returns FAIL if not found without an error message.
309 */
310 static int
311call_by_name(char_u *name, int argcount, ectx_T *ectx)
312{
313 ufunc_T *ufunc;
314
315 if (builtin_function(name, -1))
316 {
317 int func_idx = find_internal_func(name);
318
319 if (func_idx < 0)
320 return FAIL;
321 if (check_internal_func(func_idx, argcount) == FAIL)
322 return FAIL;
323 return call_bfunc(func_idx, argcount, ectx);
324 }
325
326 ufunc = find_func(name, NULL);
327 if (ufunc != NULL)
328 return call_ufunc(ufunc, argcount, ectx);
329
330 return FAIL;
331}
332
333 static int
334call_partial(typval_T *tv, int argcount, ectx_T *ectx)
335{
336 char_u *name;
337 int called_emsg_before = called_emsg;
338
339 if (tv->v_type == VAR_PARTIAL)
340 {
341 partial_T *pt = tv->vval.v_partial;
342
343 if (pt->pt_func != NULL)
344 return call_ufunc(pt->pt_func, argcount, ectx);
345 name = pt->pt_name;
346 }
347 else
348 name = tv->vval.v_string;
349 if (call_by_name(name, argcount, ectx) == FAIL)
350 {
351 if (called_emsg == called_emsg_before)
352 semsg(_(e_unknownfunc), name);
353 return FAIL;
354 }
355 return OK;
356}
357
358/*
359 * Execute a function by "name".
360 * This can be a builtin function, user function or a funcref.
361 */
362 static int
363call_eval_func(char_u *name, int argcount, ectx_T *ectx)
364{
365 int called_emsg_before = called_emsg;
366
367 if (call_by_name(name, argcount, ectx) == FAIL
368 && called_emsg == called_emsg_before)
369 {
370 // "name" may be a variable that is a funcref or partial
371 // if find variable
372 // call_partial()
373 // else
374 // semsg(_(e_unknownfunc), name);
375 emsg("call_eval_func(partial) not implemented yet");
376 return FAIL;
377 }
378 return OK;
379}
380
381/*
382 * Call a "def" function from old Vim script.
383 * Return OK or FAIL.
384 */
385 int
386call_def_function(
387 ufunc_T *ufunc,
388 int argc, // nr of arguments
389 typval_T *argv, // arguments
390 typval_T *rettv) // return value
391{
392 ectx_T ectx; // execution context
393 int initial_frame_ptr;
394 typval_T *tv;
395 int idx;
396 int ret = FAIL;
397 dfunc_T *dfunc;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +0100398 int defcount = ufunc->uf_args.ga_len - argc;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100399
400// Get pointer to item in the stack.
401#define STACK_TV(idx) (((typval_T *)ectx.ec_stack.ga_data) + idx)
402
403// Get pointer to item at the bottom of the stack, -1 is the bottom.
404#undef STACK_TV_BOT
405#define STACK_TV_BOT(idx) (((typval_T *)ectx.ec_stack.ga_data) + ectx.ec_stack.ga_len + idx)
406
407// Get pointer to local variable on the stack.
408#define STACK_TV_VAR(idx) (((typval_T *)ectx.ec_stack.ga_data) + ectx.ec_frame + STACK_FRAME_SIZE + idx)
409
410 vim_memset(&ectx, 0, sizeof(ectx));
411 ga_init2(&ectx.ec_stack, sizeof(typval_T), 500);
412 if (ga_grow(&ectx.ec_stack, 20) == FAIL)
413 goto failed;
414 ectx.ec_dfunc_idx = ufunc->uf_dfunc_idx;
415
416 ga_init2(&ectx.ec_trystack, sizeof(trycmd_T), 10);
417
418 // Put arguments on the stack.
419 for (idx = 0; idx < argc; ++idx)
420 {
421 copy_tv(&argv[idx], STACK_TV_BOT(0));
422 ++ectx.ec_stack.ga_len;
423 }
Bram Moolenaar170fcfc2020-02-06 17:51:35 +0100424 // Make space for omitted arguments, will store default value below.
425 if (defcount > 0)
426 for (idx = 0; idx < defcount; ++idx)
427 {
428 STACK_TV_BOT(0)->v_type = VAR_UNKNOWN;
429 ++ectx.ec_stack.ga_len;
430 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100431
432 // Frame pointer points to just after arguments.
433 ectx.ec_frame = ectx.ec_stack.ga_len;
434 initial_frame_ptr = ectx.ec_frame;
435
436 // dummy frame entries
437 for (idx = 0; idx < STACK_FRAME_SIZE; ++idx)
438 {
439 STACK_TV(ectx.ec_stack.ga_len)->v_type = VAR_UNKNOWN;
440 ++ectx.ec_stack.ga_len;
441 }
442
443 // Reserve space for local variables.
444 dfunc = ((dfunc_T *)def_functions.ga_data) + ufunc->uf_dfunc_idx;
445 for (idx = 0; idx < dfunc->df_varcount; ++idx)
446 STACK_TV_VAR(idx)->v_type = VAR_UNKNOWN;
447 ectx.ec_stack.ga_len += dfunc->df_varcount;
448
449 ectx.ec_instr = dfunc->df_instr;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +0100450
451 // Decide where to start execution, handles optional arguments.
452 init_instr_idx(ufunc, argc, &ectx);
453
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100454 for (;;)
455 {
456 isn_T *iptr;
457 trycmd_T *trycmd = NULL;
458
459 if (did_throw && !ectx.ec_in_catch)
460 {
461 garray_T *trystack = &ectx.ec_trystack;
462
463 // An exception jumps to the first catch, finally, or returns from
464 // the current function.
465 if (trystack->ga_len > 0)
466 trycmd = ((trycmd_T *)trystack->ga_data) + trystack->ga_len - 1;
467 if (trycmd != NULL && trycmd->tcd_frame == ectx.ec_frame)
468 {
469 // jump to ":catch" or ":finally"
470 ectx.ec_in_catch = TRUE;
471 ectx.ec_iidx = trycmd->tcd_catch_idx;
472 }
473 else
474 {
475 // not inside try or need to return from current functions.
476 if (ectx.ec_frame == initial_frame_ptr)
477 {
478 // At the toplevel we are done. Push a dummy return value.
479 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
480 goto failed;
481 tv = STACK_TV_BOT(0);
482 tv->v_type = VAR_NUMBER;
483 tv->vval.v_number = 0;
484 ++ectx.ec_stack.ga_len;
Bram Moolenaar257cc5e2020-02-19 17:06:11 +0100485 need_rethrow = TRUE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100486 goto done;
487 }
488
489 func_return(&ectx);
490 }
491 continue;
492 }
493
494 iptr = &ectx.ec_instr[ectx.ec_iidx++];
495 switch (iptr->isn_type)
496 {
497 // execute Ex command line
498 case ISN_EXEC:
499 do_cmdline_cmd(iptr->isn_arg.string);
500 break;
501
502 // execute :echo {string} ...
503 case ISN_ECHO:
504 {
505 int count = iptr->isn_arg.echo.echo_count;
506 int atstart = TRUE;
507 int needclr = TRUE;
508
509 for (idx = 0; idx < count; ++idx)
510 {
511 tv = STACK_TV_BOT(idx - count);
512 echo_one(tv, iptr->isn_arg.echo.echo_with_white,
513 &atstart, &needclr);
514 clear_tv(tv);
515 }
516 ectx.ec_stack.ga_len -= count;
517 }
518 break;
519
520 // load local variable or argument
521 case ISN_LOAD:
522 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
523 goto failed;
524 copy_tv(STACK_TV_VAR(iptr->isn_arg.number), STACK_TV_BOT(0));
525 ++ectx.ec_stack.ga_len;
526 break;
527
528 // load v: variable
529 case ISN_LOADV:
530 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
531 goto failed;
532 copy_tv(get_vim_var_tv(iptr->isn_arg.number), STACK_TV_BOT(0));
533 ++ectx.ec_stack.ga_len;
534 break;
535
Bram Moolenaarb283a8a2020-02-02 22:24:04 +0100536 // load s: variable in Vim9 script
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100537 case ISN_LOADSCRIPT:
538 {
539 scriptitem_T *si =
Bram Moolenaar21b9e972020-01-26 19:26:46 +0100540 SCRIPT_ITEM(iptr->isn_arg.script.script_sid);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100541 svar_T *sv;
542
543 sv = ((svar_T *)si->sn_var_vals.ga_data)
544 + iptr->isn_arg.script.script_idx;
545 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
546 goto failed;
547 copy_tv(sv->sv_tv, STACK_TV_BOT(0));
548 ++ectx.ec_stack.ga_len;
549 }
550 break;
551
552 // load s: variable in old script
553 case ISN_LOADS:
554 {
Bram Moolenaarb283a8a2020-02-02 22:24:04 +0100555 hashtab_T *ht = &SCRIPT_VARS(
556 iptr->isn_arg.loadstore.ls_sid);
557 char_u *name = iptr->isn_arg.loadstore.ls_name;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100558 dictitem_T *di = find_var_in_ht(ht, 0, name, TRUE);
559 if (di == NULL)
560 {
Bram Moolenaarb283a8a2020-02-02 22:24:04 +0100561 semsg(_(e_undefvar), name);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100562 goto failed;
563 }
564 else
565 {
566 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
567 goto failed;
568 copy_tv(&di->di_tv, STACK_TV_BOT(0));
569 ++ectx.ec_stack.ga_len;
570 }
571 }
572 break;
573
574 // load g: variable
575 case ISN_LOADG:
576 {
577 dictitem_T *di;
578
579 di = find_var_in_ht(get_globvar_ht(), 0,
580 iptr->isn_arg.string, TRUE);
581 if (di == NULL)
582 {
583 semsg(_("E121: Undefined variable: g:%s"),
584 iptr->isn_arg.string);
585 goto failed;
586 }
587 else
588 {
589 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
590 goto failed;
591 copy_tv(&di->di_tv, STACK_TV_BOT(0));
592 ++ectx.ec_stack.ga_len;
593 }
594 }
595 break;
596
597 // load &option
598 case ISN_LOADOPT:
599 {
600 typval_T optval;
601 char_u *name = iptr->isn_arg.string;
602
603 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
604 goto failed;
Bram Moolenaar58ceca52020-01-28 22:46:22 +0100605 if (get_option_tv(&name, &optval, TRUE) == FAIL)
606 goto failed;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100607 *STACK_TV_BOT(0) = optval;
608 ++ectx.ec_stack.ga_len;
609 }
610 break;
611
612 // load $ENV
613 case ISN_LOADENV:
614 {
615 typval_T optval;
616 char_u *name = iptr->isn_arg.string;
617
618 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
619 goto failed;
Bram Moolenaar07da94b2020-01-28 22:39:19 +0100620 if (get_env_tv(&name, &optval, TRUE) == FAIL)
621 {
622 semsg(_("E1060: Invalid environment variable name: %s"),
623 iptr->isn_arg.string);
624 goto failed;
625 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100626 *STACK_TV_BOT(0) = optval;
627 ++ectx.ec_stack.ga_len;
628 }
629 break;
630
631 // load @register
632 case ISN_LOADREG:
633 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
634 goto failed;
635 tv = STACK_TV_BOT(0);
636 tv->v_type = VAR_STRING;
637 tv->vval.v_string = get_reg_contents(
638 iptr->isn_arg.number, GREG_EXPR_SRC);
639 ++ectx.ec_stack.ga_len;
640 break;
641
642 // store local variable
643 case ISN_STORE:
644 --ectx.ec_stack.ga_len;
645 tv = STACK_TV_VAR(iptr->isn_arg.number);
646 clear_tv(tv);
647 *tv = *STACK_TV_BOT(0);
648 break;
649
Bram Moolenaarb283a8a2020-02-02 22:24:04 +0100650 // store s: variable in old script
651 case ISN_STORES:
652 {
653 hashtab_T *ht = &SCRIPT_VARS(
654 iptr->isn_arg.loadstore.ls_sid);
655 char_u *name = iptr->isn_arg.loadstore.ls_name;
656 dictitem_T *di = find_var_in_ht(ht, 0, name, TRUE);
657
658 if (di == NULL)
659 {
660 semsg(_(e_undefvar), name);
661 goto failed;
662 }
663 --ectx.ec_stack.ga_len;
664 clear_tv(&di->di_tv);
665 di->di_tv = *STACK_TV_BOT(0);
666 }
667 break;
668
669 // store script-local variable in Vim9 script
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100670 case ISN_STORESCRIPT:
671 {
Bram Moolenaar21b9e972020-01-26 19:26:46 +0100672 scriptitem_T *si = SCRIPT_ITEM(
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100673 iptr->isn_arg.script.script_sid);
674 svar_T *sv = ((svar_T *)si->sn_var_vals.ga_data)
675 + iptr->isn_arg.script.script_idx;
676
677 --ectx.ec_stack.ga_len;
678 clear_tv(sv->sv_tv);
679 *sv->sv_tv = *STACK_TV_BOT(0);
680 }
681 break;
682
683 // store option
684 case ISN_STOREOPT:
685 {
686 long n = 0;
687 char_u *s = NULL;
688 char *msg;
689
690 --ectx.ec_stack.ga_len;
691 tv = STACK_TV_BOT(0);
692 if (tv->v_type == VAR_STRING)
Bram Moolenaar97a2af32020-01-28 22:52:48 +0100693 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100694 s = tv->vval.v_string;
Bram Moolenaar97a2af32020-01-28 22:52:48 +0100695 if (s == NULL)
696 s = (char_u *)"";
697 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100698 else if (tv->v_type == VAR_NUMBER)
699 n = tv->vval.v_number;
700 else
701 {
702 emsg(_("E1051: Expected string or number"));
703 goto failed;
704 }
705 msg = set_option_value(iptr->isn_arg.storeopt.so_name,
706 n, s, iptr->isn_arg.storeopt.so_flags);
707 if (msg != NULL)
708 {
709 emsg(_(msg));
710 goto failed;
711 }
712 clear_tv(tv);
713 }
714 break;
715
Bram Moolenaarb283a8a2020-02-02 22:24:04 +0100716 // store $ENV
717 case ISN_STOREENV:
718 --ectx.ec_stack.ga_len;
719 vim_setenv_ext(iptr->isn_arg.string,
720 tv_get_string(STACK_TV_BOT(0)));
721 break;
722
723 // store @r
724 case ISN_STOREREG:
725 {
726 int reg = iptr->isn_arg.number;
727
728 --ectx.ec_stack.ga_len;
Bram Moolenaar401d9ff2020-02-19 18:14:44 +0100729 tv = STACK_TV_BOT(0);
Bram Moolenaarb283a8a2020-02-02 22:24:04 +0100730 write_reg_contents(reg == '@' ? '"' : reg,
Bram Moolenaar401d9ff2020-02-19 18:14:44 +0100731 tv_get_string(tv), -1, FALSE);
732 clear_tv(tv);
Bram Moolenaarb283a8a2020-02-02 22:24:04 +0100733 }
734 break;
735
736 // store v: variable
737 case ISN_STOREV:
738 --ectx.ec_stack.ga_len;
739 if (set_vim_var_tv(iptr->isn_arg.number, STACK_TV_BOT(0))
740 == FAIL)
741 goto failed;
742 break;
743
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100744 // store g: variable
745 case ISN_STOREG:
746 {
747 dictitem_T *di;
748
749 --ectx.ec_stack.ga_len;
750 di = find_var_in_ht(get_globvar_ht(), 0,
Bram Moolenaar401d9ff2020-02-19 18:14:44 +0100751 iptr->isn_arg.string + 2, TRUE);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100752 if (di == NULL)
753 {
754 funccal_entry_T entry;
755
756 save_funccal(&entry);
757 set_var_const(iptr->isn_arg.string, NULL,
758 STACK_TV_BOT(0), FALSE, 0);
759 restore_funccal();
760 }
761 else
762 {
763 clear_tv(&di->di_tv);
764 di->di_tv = *STACK_TV_BOT(0);
765 }
766 }
767 break;
768
769 // store number in local variable
770 case ISN_STORENR:
771 tv = STACK_TV_VAR(iptr->isn_arg.storenr.str_idx);
772 clear_tv(tv);
773 tv->v_type = VAR_NUMBER;
774 tv->vval.v_number = iptr->isn_arg.storenr.str_val;
775 break;
776
777 // push constant
778 case ISN_PUSHNR:
779 case ISN_PUSHBOOL:
780 case ISN_PUSHSPEC:
781 case ISN_PUSHF:
782 case ISN_PUSHS:
783 case ISN_PUSHBLOB:
784 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
785 goto failed;
786 tv = STACK_TV_BOT(0);
787 ++ectx.ec_stack.ga_len;
788 switch (iptr->isn_type)
789 {
790 case ISN_PUSHNR:
791 tv->v_type = VAR_NUMBER;
792 tv->vval.v_number = iptr->isn_arg.number;
793 break;
794 case ISN_PUSHBOOL:
795 tv->v_type = VAR_BOOL;
796 tv->vval.v_number = iptr->isn_arg.number;
797 break;
798 case ISN_PUSHSPEC:
799 tv->v_type = VAR_SPECIAL;
800 tv->vval.v_number = iptr->isn_arg.number;
801 break;
802#ifdef FEAT_FLOAT
803 case ISN_PUSHF:
804 tv->v_type = VAR_FLOAT;
805 tv->vval.v_float = iptr->isn_arg.fnumber;
806 break;
807#endif
808 case ISN_PUSHBLOB:
809 blob_copy(iptr->isn_arg.blob, tv);
810 break;
811 default:
812 tv->v_type = VAR_STRING;
813 tv->vval.v_string = vim_strsave(iptr->isn_arg.string);
814 }
815 break;
816
817 // create a list from items on the stack; uses a single allocation
818 // for the list header and the items
819 case ISN_NEWLIST:
820 {
821 int count = iptr->isn_arg.number;
822 list_T *list = list_alloc_with_items(count);
823
824 if (list == NULL)
825 goto failed;
826 for (idx = 0; idx < count; ++idx)
827 list_set_item(list, idx, STACK_TV_BOT(idx - count));
828
829 if (count > 0)
830 ectx.ec_stack.ga_len -= count - 1;
831 else if (ga_grow(&ectx.ec_stack, 1) == FAIL)
832 goto failed;
833 else
834 ++ectx.ec_stack.ga_len;
835 tv = STACK_TV_BOT(-1);
836 tv->v_type = VAR_LIST;
837 tv->vval.v_list = list;
838 ++list->lv_refcount;
839 }
840 break;
841
842 // create a dict from items on the stack
843 case ISN_NEWDICT:
844 {
845 int count = iptr->isn_arg.number;
846 dict_T *dict = dict_alloc();
847 dictitem_T *item;
848
849 if (dict == NULL)
850 goto failed;
851 for (idx = 0; idx < count; ++idx)
852 {
853 // check key type is VAR_STRING
854 tv = STACK_TV_BOT(2 * (idx - count));
855 item = dictitem_alloc(tv->vval.v_string);
856 clear_tv(tv);
857 if (item == NULL)
858 goto failed;
859 item->di_tv = *STACK_TV_BOT(2 * (idx - count) + 1);
860 item->di_tv.v_lock = 0;
861 if (dict_add(dict, item) == FAIL)
862 goto failed;
863 }
864
865 if (count > 0)
866 ectx.ec_stack.ga_len -= 2 * count - 1;
867 else if (ga_grow(&ectx.ec_stack, 1) == FAIL)
868 goto failed;
869 else
870 ++ectx.ec_stack.ga_len;
871 tv = STACK_TV_BOT(-1);
872 tv->v_type = VAR_DICT;
873 tv->vval.v_dict = dict;
874 ++dict->dv_refcount;
875 }
876 break;
877
878 // call a :def function
879 case ISN_DCALL:
880 if (call_dfunc(iptr->isn_arg.dfunc.cdf_idx,
881 iptr->isn_arg.dfunc.cdf_argcount,
882 &ectx) == FAIL)
883 goto failed;
884 break;
885
886 // call a builtin function
887 case ISN_BCALL:
888 SOURCING_LNUM = iptr->isn_lnum;
889 if (call_bfunc(iptr->isn_arg.bfunc.cbf_idx,
890 iptr->isn_arg.bfunc.cbf_argcount,
891 &ectx) == FAIL)
892 goto failed;
893 break;
894
895 // call a funcref or partial
896 case ISN_PCALL:
897 {
898 cpfunc_T *pfunc = &iptr->isn_arg.pfunc;
899 int r;
900 typval_T partial;
901
902 SOURCING_LNUM = iptr->isn_lnum;
903 if (pfunc->cpf_top)
904 {
905 // funcref is above the arguments
906 tv = STACK_TV_BOT(-pfunc->cpf_argcount - 1);
907 }
908 else
909 {
910 // Get the funcref from the stack.
911 --ectx.ec_stack.ga_len;
912 partial = *STACK_TV_BOT(0);
913 tv = &partial;
914 }
915 r = call_partial(tv, pfunc->cpf_argcount, &ectx);
916 if (tv == &partial)
917 clear_tv(&partial);
918 if (r == FAIL)
919 goto failed;
920
921 if (pfunc->cpf_top)
922 {
923 // Get the funcref from the stack, overwrite with the
924 // return value.
925 clear_tv(tv);
926 --ectx.ec_stack.ga_len;
927 *STACK_TV_BOT(-1) = *STACK_TV_BOT(0);
928 }
929 }
930 break;
931
932 // call a user defined function or funcref/partial
933 case ISN_UCALL:
934 {
935 cufunc_T *cufunc = &iptr->isn_arg.ufunc;
936
937 SOURCING_LNUM = iptr->isn_lnum;
938 if (call_eval_func(cufunc->cuf_name,
939 cufunc->cuf_argcount, &ectx) == FAIL)
940 goto failed;
941 }
942 break;
943
944 // return from a :def function call
945 case ISN_RETURN:
946 {
Bram Moolenaar8cbd6df2020-01-28 22:59:45 +0100947 garray_T *trystack = &ectx.ec_trystack;
948
949 if (trystack->ga_len > 0)
950 trycmd = ((trycmd_T *)trystack->ga_data)
951 + trystack->ga_len - 1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100952 if (trycmd != NULL && trycmd->tcd_frame == ectx.ec_frame
953 && trycmd->tcd_finally_idx != 0)
954 {
955 // jump to ":finally"
956 ectx.ec_iidx = trycmd->tcd_finally_idx;
957 trycmd->tcd_return = TRUE;
958 }
959 else
960 {
961 // Restore previous function. If the frame pointer
962 // is zero then there is none and we are done.
963 if (ectx.ec_frame == initial_frame_ptr)
964 goto done;
965
966 func_return(&ectx);
967 }
968 }
969 break;
970
971 // push a function reference to a compiled function
972 case ISN_FUNCREF:
973 {
974 partial_T *pt = NULL;
975
976 pt = ALLOC_CLEAR_ONE(partial_T);
977 if (pt == NULL)
978 goto failed;
979 dfunc = ((dfunc_T *)def_functions.ga_data)
980 + iptr->isn_arg.number;
981 pt->pt_func = dfunc->df_ufunc;
982 pt->pt_refcount = 1;
983 ++dfunc->df_ufunc->uf_refcount;
984
985 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
986 goto failed;
987 tv = STACK_TV_BOT(0);
988 ++ectx.ec_stack.ga_len;
989 tv->vval.v_partial = pt;
990 tv->v_type = VAR_PARTIAL;
991 }
992 break;
993
994 // jump if a condition is met
995 case ISN_JUMP:
996 {
997 jumpwhen_T when = iptr->isn_arg.jump.jump_when;
998 int jump = TRUE;
999
1000 if (when != JUMP_ALWAYS)
1001 {
1002 tv = STACK_TV_BOT(-1);
1003 jump = tv2bool(tv);
1004 if (when == JUMP_IF_FALSE
1005 || when == JUMP_AND_KEEP_IF_FALSE)
1006 jump = !jump;
Bram Moolenaar777770f2020-02-06 21:27:08 +01001007 if (when == JUMP_IF_FALSE || !jump)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001008 {
1009 // drop the value from the stack
1010 clear_tv(tv);
1011 --ectx.ec_stack.ga_len;
1012 }
1013 }
1014 if (jump)
1015 ectx.ec_iidx = iptr->isn_arg.jump.jump_where;
1016 }
1017 break;
1018
1019 // top of a for loop
1020 case ISN_FOR:
1021 {
1022 list_T *list = STACK_TV_BOT(-1)->vval.v_list;
1023 typval_T *idxtv =
1024 STACK_TV_VAR(iptr->isn_arg.forloop.for_idx);
1025
1026 // push the next item from the list
1027 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
1028 goto failed;
1029 if (++idxtv->vval.v_number >= list->lv_len)
1030 // past the end of the list, jump to "endfor"
1031 ectx.ec_iidx = iptr->isn_arg.forloop.for_end;
1032 else if (list->lv_first == &range_list_item)
1033 {
1034 // non-materialized range() list
1035 tv = STACK_TV_BOT(0);
1036 tv->v_type = VAR_NUMBER;
1037 tv->vval.v_number = list_find_nr(
1038 list, idxtv->vval.v_number, NULL);
1039 ++ectx.ec_stack.ga_len;
1040 }
1041 else
1042 {
1043 listitem_T *li = list_find(list, idxtv->vval.v_number);
1044
1045 if (li == NULL)
1046 goto failed;
1047 copy_tv(&li->li_tv, STACK_TV_BOT(0));
1048 ++ectx.ec_stack.ga_len;
1049 }
1050 }
1051 break;
1052
1053 // start of ":try" block
1054 case ISN_TRY:
1055 {
1056 if (ga_grow(&ectx.ec_trystack, 1) == FAIL)
1057 goto failed;
1058 trycmd = ((trycmd_T *)ectx.ec_trystack.ga_data)
1059 + ectx.ec_trystack.ga_len;
1060 ++ectx.ec_trystack.ga_len;
1061 ++trylevel;
1062 trycmd->tcd_frame = ectx.ec_frame;
1063 trycmd->tcd_catch_idx = iptr->isn_arg.try.try_catch;
1064 trycmd->tcd_finally_idx = iptr->isn_arg.try.try_finally;
1065 }
1066 break;
1067
1068 case ISN_PUSHEXC:
1069 if (current_exception == NULL)
1070 {
1071 iemsg("Evaluating catch while current_exception is NULL");
1072 goto failed;
1073 }
1074 if (ga_grow(&ectx.ec_stack, 1) == FAIL)
1075 goto failed;
1076 tv = STACK_TV_BOT(0);
1077 ++ectx.ec_stack.ga_len;
1078 tv->v_type = VAR_STRING;
1079 tv->vval.v_string = vim_strsave(
1080 (char_u *)current_exception->value);
1081 break;
1082
1083 case ISN_CATCH:
1084 {
1085 garray_T *trystack = &ectx.ec_trystack;
1086
1087 if (trystack->ga_len > 0)
1088 {
1089 trycmd = ((trycmd_T *)trystack->ga_data)
1090 + trystack->ga_len - 1;
1091 trycmd->tcd_caught = TRUE;
1092 }
1093 did_emsg = got_int = did_throw = FALSE;
1094 catch_exception(current_exception);
1095 }
1096 break;
1097
1098 // end of ":try" block
1099 case ISN_ENDTRY:
1100 {
1101 garray_T *trystack = &ectx.ec_trystack;
1102
1103 if (trystack->ga_len > 0)
1104 {
1105 --trystack->ga_len;
1106 --trylevel;
1107 trycmd = ((trycmd_T *)trystack->ga_data)
1108 + trystack->ga_len;
1109 if (trycmd->tcd_caught)
1110 {
1111 // discard the exception
1112 if (caught_stack == current_exception)
1113 caught_stack = caught_stack->caught;
1114 discard_current_exception();
1115 }
1116
1117 if (trycmd->tcd_return)
1118 {
1119 // Restore previous function. If the frame pointer
1120 // is zero then there is none and we are done.
1121 if (ectx.ec_frame == initial_frame_ptr)
1122 goto done;
1123
1124 func_return(&ectx);
1125 }
1126 }
1127 }
1128 break;
1129
1130 case ISN_THROW:
1131 --ectx.ec_stack.ga_len;
1132 tv = STACK_TV_BOT(0);
1133 if (throw_exception(tv->vval.v_string, ET_USER, NULL) == FAIL)
1134 {
1135 vim_free(tv->vval.v_string);
1136 goto failed;
1137 }
1138 did_throw = TRUE;
1139 break;
1140
1141 // compare with special values
1142 case ISN_COMPAREBOOL:
1143 case ISN_COMPARESPECIAL:
1144 {
1145 typval_T *tv1 = STACK_TV_BOT(-2);
1146 typval_T *tv2 = STACK_TV_BOT(-1);
1147 varnumber_T arg1 = tv1->vval.v_number;
1148 varnumber_T arg2 = tv2->vval.v_number;
1149 int res;
1150
1151 switch (iptr->isn_arg.op.op_type)
1152 {
1153 case EXPR_EQUAL: res = arg1 == arg2; break;
1154 case EXPR_NEQUAL: res = arg1 != arg2; break;
1155 default: res = 0; break;
1156 }
1157
1158 --ectx.ec_stack.ga_len;
1159 tv1->v_type = VAR_BOOL;
1160 tv1->vval.v_number = res ? VVAL_TRUE : VVAL_FALSE;
1161 }
1162 break;
1163
1164 // Operation with two number arguments
1165 case ISN_OPNR:
1166 case ISN_COMPARENR:
1167 {
1168 typval_T *tv1 = STACK_TV_BOT(-2);
1169 typval_T *tv2 = STACK_TV_BOT(-1);
1170 varnumber_T arg1 = tv1->vval.v_number;
1171 varnumber_T arg2 = tv2->vval.v_number;
1172 varnumber_T res;
1173
1174 switch (iptr->isn_arg.op.op_type)
1175 {
1176 case EXPR_MULT: res = arg1 * arg2; break;
1177 case EXPR_DIV: res = arg1 / arg2; break;
1178 case EXPR_REM: res = arg1 % arg2; break;
1179 case EXPR_SUB: res = arg1 - arg2; break;
1180 case EXPR_ADD: res = arg1 + arg2; break;
1181
1182 case EXPR_EQUAL: res = arg1 == arg2; break;
1183 case EXPR_NEQUAL: res = arg1 != arg2; break;
1184 case EXPR_GREATER: res = arg1 > arg2; break;
1185 case EXPR_GEQUAL: res = arg1 >= arg2; break;
1186 case EXPR_SMALLER: res = arg1 < arg2; break;
1187 case EXPR_SEQUAL: res = arg1 <= arg2; break;
1188 default: res = 0; break;
1189 }
1190
1191 --ectx.ec_stack.ga_len;
1192 if (iptr->isn_type == ISN_COMPARENR)
1193 {
1194 tv1->v_type = VAR_BOOL;
1195 tv1->vval.v_number = res ? VVAL_TRUE : VVAL_FALSE;
1196 }
1197 else
1198 tv1->vval.v_number = res;
1199 }
1200 break;
1201
1202 // Computation with two float arguments
1203 case ISN_OPFLOAT:
1204 case ISN_COMPAREFLOAT:
Bram Moolenaara5d59532020-01-26 21:42:03 +01001205#ifdef FEAT_FLOAT
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001206 {
1207 typval_T *tv1 = STACK_TV_BOT(-2);
1208 typval_T *tv2 = STACK_TV_BOT(-1);
1209 float_T arg1 = tv1->vval.v_float;
1210 float_T arg2 = tv2->vval.v_float;
1211 float_T res = 0;
1212 int cmp = FALSE;
1213
1214 switch (iptr->isn_arg.op.op_type)
1215 {
1216 case EXPR_MULT: res = arg1 * arg2; break;
1217 case EXPR_DIV: res = arg1 / arg2; break;
1218 case EXPR_SUB: res = arg1 - arg2; break;
1219 case EXPR_ADD: res = arg1 + arg2; break;
1220
1221 case EXPR_EQUAL: cmp = arg1 == arg2; break;
1222 case EXPR_NEQUAL: cmp = arg1 != arg2; break;
1223 case EXPR_GREATER: cmp = arg1 > arg2; break;
1224 case EXPR_GEQUAL: cmp = arg1 >= arg2; break;
1225 case EXPR_SMALLER: cmp = arg1 < arg2; break;
1226 case EXPR_SEQUAL: cmp = arg1 <= arg2; break;
1227 default: cmp = 0; break;
1228 }
1229 --ectx.ec_stack.ga_len;
1230 if (iptr->isn_type == ISN_COMPAREFLOAT)
1231 {
1232 tv1->v_type = VAR_BOOL;
1233 tv1->vval.v_number = cmp ? VVAL_TRUE : VVAL_FALSE;
1234 }
1235 else
1236 tv1->vval.v_float = res;
1237 }
Bram Moolenaara5d59532020-01-26 21:42:03 +01001238#endif
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001239 break;
1240
1241 case ISN_COMPARELIST:
1242 {
1243 typval_T *tv1 = STACK_TV_BOT(-2);
1244 typval_T *tv2 = STACK_TV_BOT(-1);
1245 list_T *arg1 = tv1->vval.v_list;
1246 list_T *arg2 = tv2->vval.v_list;
1247 int cmp = FALSE;
1248 int ic = iptr->isn_arg.op.op_ic;
1249
1250 switch (iptr->isn_arg.op.op_type)
1251 {
1252 case EXPR_EQUAL: cmp =
1253 list_equal(arg1, arg2, ic, FALSE); break;
1254 case EXPR_NEQUAL: cmp =
1255 !list_equal(arg1, arg2, ic, FALSE); break;
1256 case EXPR_IS: cmp = arg1 == arg2; break;
1257 case EXPR_ISNOT: cmp = arg1 != arg2; break;
1258 default: cmp = 0; break;
1259 }
1260 --ectx.ec_stack.ga_len;
1261 clear_tv(tv1);
1262 clear_tv(tv2);
1263 tv1->v_type = VAR_BOOL;
1264 tv1->vval.v_number = cmp ? VVAL_TRUE : VVAL_FALSE;
1265 }
1266 break;
1267
1268 case ISN_COMPAREBLOB:
1269 {
1270 typval_T *tv1 = STACK_TV_BOT(-2);
1271 typval_T *tv2 = STACK_TV_BOT(-1);
1272 blob_T *arg1 = tv1->vval.v_blob;
1273 blob_T *arg2 = tv2->vval.v_blob;
1274 int cmp = FALSE;
1275
1276 switch (iptr->isn_arg.op.op_type)
1277 {
1278 case EXPR_EQUAL: cmp = blob_equal(arg1, arg2); break;
1279 case EXPR_NEQUAL: cmp = !blob_equal(arg1, arg2); break;
1280 case EXPR_IS: cmp = arg1 == arg2; break;
1281 case EXPR_ISNOT: cmp = arg1 != arg2; break;
1282 default: cmp = 0; break;
1283 }
1284 --ectx.ec_stack.ga_len;
1285 clear_tv(tv1);
1286 clear_tv(tv2);
1287 tv1->v_type = VAR_BOOL;
1288 tv1->vval.v_number = cmp ? VVAL_TRUE : VVAL_FALSE;
1289 }
1290 break;
1291
1292 // TODO: handle separately
1293 case ISN_COMPARESTRING:
1294 case ISN_COMPAREDICT:
1295 case ISN_COMPAREFUNC:
1296 case ISN_COMPAREPARTIAL:
1297 case ISN_COMPAREANY:
1298 {
1299 typval_T *tv1 = STACK_TV_BOT(-2);
1300 typval_T *tv2 = STACK_TV_BOT(-1);
1301 exptype_T exptype = iptr->isn_arg.op.op_type;
1302 int ic = iptr->isn_arg.op.op_ic;
1303
1304 typval_compare(tv1, tv2, exptype, ic);
1305 clear_tv(tv2);
1306 tv1->v_type = VAR_BOOL;
1307 tv1->vval.v_number = tv1->vval.v_number
1308 ? VVAL_TRUE : VVAL_FALSE;
1309 --ectx.ec_stack.ga_len;
1310 }
1311 break;
1312
1313 case ISN_ADDLIST:
1314 case ISN_ADDBLOB:
1315 {
1316 typval_T *tv1 = STACK_TV_BOT(-2);
1317 typval_T *tv2 = STACK_TV_BOT(-1);
1318
1319 if (iptr->isn_type == ISN_ADDLIST)
1320 eval_addlist(tv1, tv2);
1321 else
1322 eval_addblob(tv1, tv2);
1323 clear_tv(tv2);
1324 --ectx.ec_stack.ga_len;
1325 }
1326 break;
1327
1328 // Computation with two arguments of unknown type
1329 case ISN_OPANY:
1330 {
1331 typval_T *tv1 = STACK_TV_BOT(-2);
1332 typval_T *tv2 = STACK_TV_BOT(-1);
1333 varnumber_T n1, n2;
1334#ifdef FEAT_FLOAT
1335 float_T f1 = 0, f2 = 0;
1336#endif
1337 int error = FALSE;
1338
1339 if (iptr->isn_arg.op.op_type == EXPR_ADD)
1340 {
1341 if (tv1->v_type == VAR_LIST && tv2->v_type == VAR_LIST)
1342 {
1343 eval_addlist(tv1, tv2);
1344 clear_tv(tv2);
1345 --ectx.ec_stack.ga_len;
1346 break;
1347 }
1348 else if (tv1->v_type == VAR_BLOB
1349 && tv2->v_type == VAR_BLOB)
1350 {
1351 eval_addblob(tv1, tv2);
1352 clear_tv(tv2);
1353 --ectx.ec_stack.ga_len;
1354 break;
1355 }
1356 }
1357#ifdef FEAT_FLOAT
1358 if (tv1->v_type == VAR_FLOAT)
1359 {
1360 f1 = tv1->vval.v_float;
1361 n1 = 0;
1362 }
1363 else
1364#endif
1365 {
1366 n1 = tv_get_number_chk(tv1, &error);
1367 if (error)
1368 goto failed;
1369#ifdef FEAT_FLOAT
1370 if (tv2->v_type == VAR_FLOAT)
1371 f1 = n1;
1372#endif
1373 }
1374#ifdef FEAT_FLOAT
1375 if (tv2->v_type == VAR_FLOAT)
1376 {
1377 f2 = tv2->vval.v_float;
1378 n2 = 0;
1379 }
1380 else
1381#endif
1382 {
1383 n2 = tv_get_number_chk(tv2, &error);
1384 if (error)
1385 goto failed;
1386#ifdef FEAT_FLOAT
1387 if (tv1->v_type == VAR_FLOAT)
1388 f2 = n2;
1389#endif
1390 }
1391#ifdef FEAT_FLOAT
1392 // if there is a float on either side the result is a float
1393 if (tv1->v_type == VAR_FLOAT || tv2->v_type == VAR_FLOAT)
1394 {
1395 switch (iptr->isn_arg.op.op_type)
1396 {
1397 case EXPR_MULT: f1 = f1 * f2; break;
1398 case EXPR_DIV: f1 = f1 / f2; break;
1399 case EXPR_SUB: f1 = f1 - f2; break;
1400 case EXPR_ADD: f1 = f1 + f2; break;
1401 default: emsg(_(e_modulus)); goto failed;
1402 }
1403 clear_tv(tv1);
1404 clear_tv(tv2);
1405 tv1->v_type = VAR_FLOAT;
1406 tv1->vval.v_float = f1;
1407 --ectx.ec_stack.ga_len;
1408 }
1409 else
1410#endif
1411 {
1412 switch (iptr->isn_arg.op.op_type)
1413 {
1414 case EXPR_MULT: n1 = n1 * n2; break;
1415 case EXPR_DIV: n1 = num_divide(n1, n2); break;
1416 case EXPR_SUB: n1 = n1 - n2; break;
1417 case EXPR_ADD: n1 = n1 + n2; break;
1418 default: n1 = num_modulus(n1, n2); break;
1419 }
1420 clear_tv(tv1);
1421 clear_tv(tv2);
1422 tv1->v_type = VAR_NUMBER;
1423 tv1->vval.v_number = n1;
1424 --ectx.ec_stack.ga_len;
1425 }
1426 }
1427 break;
1428
1429 case ISN_CONCAT:
1430 {
1431 char_u *str1 = STACK_TV_BOT(-2)->vval.v_string;
1432 char_u *str2 = STACK_TV_BOT(-1)->vval.v_string;
1433 char_u *res;
1434
1435 res = concat_str(str1, str2);
1436 clear_tv(STACK_TV_BOT(-2));
1437 clear_tv(STACK_TV_BOT(-1));
1438 --ectx.ec_stack.ga_len;
1439 STACK_TV_BOT(-1)->vval.v_string = res;
1440 }
1441 break;
1442
1443 case ISN_INDEX:
1444 {
1445 list_T *list;
1446 varnumber_T n;
1447 listitem_T *li;
1448
1449 // list index: list is at stack-2, index at stack-1
1450 tv = STACK_TV_BOT(-2);
1451 if (tv->v_type != VAR_LIST)
1452 {
1453 emsg(_(e_listreq));
1454 goto failed;
1455 }
1456 list = tv->vval.v_list;
1457
1458 tv = STACK_TV_BOT(-1);
1459 if (tv->v_type != VAR_NUMBER)
1460 {
1461 emsg(_(e_number_exp));
1462 goto failed;
1463 }
1464 n = tv->vval.v_number;
1465 clear_tv(tv);
1466 if ((li = list_find(list, n)) == NULL)
1467 {
1468 semsg(_(e_listidx), n);
1469 goto failed;
1470 }
1471 --ectx.ec_stack.ga_len;
1472 clear_tv(STACK_TV_BOT(-1));
1473 copy_tv(&li->li_tv, STACK_TV_BOT(-1));
1474 }
1475 break;
1476
1477 // dict member with string key
1478 case ISN_MEMBER:
1479 {
1480 dict_T *dict;
1481 dictitem_T *di;
1482
1483 tv = STACK_TV_BOT(-1);
1484 if (tv->v_type != VAR_DICT || tv->vval.v_dict == NULL)
1485 {
1486 emsg(_(e_dictreq));
1487 goto failed;
1488 }
1489 dict = tv->vval.v_dict;
1490
1491 if ((di = dict_find(dict, iptr->isn_arg.string, -1))
1492 == NULL)
1493 {
1494 semsg(_(e_dictkey), iptr->isn_arg.string);
1495 goto failed;
1496 }
1497 clear_tv(tv);
1498 copy_tv(&di->di_tv, tv);
1499 }
1500 break;
1501
1502 case ISN_NEGATENR:
1503 tv = STACK_TV_BOT(-1);
1504 tv->vval.v_number = -tv->vval.v_number;
1505 break;
1506
1507 case ISN_CHECKNR:
1508 {
1509 int error = FALSE;
1510
1511 tv = STACK_TV_BOT(-1);
1512 if (check_not_string(tv) == FAIL)
1513 {
1514 --ectx.ec_stack.ga_len;
1515 goto failed;
1516 }
1517 (void)tv_get_number_chk(tv, &error);
1518 if (error)
1519 goto failed;
1520 }
1521 break;
1522
1523 case ISN_CHECKTYPE:
1524 {
1525 checktype_T *ct = &iptr->isn_arg.type;
1526
1527 tv = STACK_TV_BOT(ct->ct_off);
1528 if (tv->v_type != ct->ct_type)
1529 {
1530 semsg(_("E1029: Expected %s but got %s"),
1531 vartype_name(ct->ct_type),
1532 vartype_name(tv->v_type));
1533 goto failed;
1534 }
1535 }
1536 break;
1537
1538 case ISN_2BOOL:
1539 {
1540 int n;
1541
1542 tv = STACK_TV_BOT(-1);
1543 n = tv2bool(tv);
1544 if (iptr->isn_arg.number) // invert
1545 n = !n;
1546 clear_tv(tv);
1547 tv->v_type = VAR_BOOL;
1548 tv->vval.v_number = n ? VVAL_TRUE : VVAL_FALSE;
1549 }
1550 break;
1551
1552 case ISN_2STRING:
1553 {
1554 char_u *str;
1555
1556 tv = STACK_TV_BOT(iptr->isn_arg.number);
1557 if (tv->v_type != VAR_STRING)
1558 {
1559 str = typval_tostring(tv);
1560 clear_tv(tv);
1561 tv->v_type = VAR_STRING;
1562 tv->vval.v_string = str;
1563 }
1564 }
1565 break;
1566
1567 case ISN_DROP:
1568 --ectx.ec_stack.ga_len;
1569 clear_tv(STACK_TV_BOT(0));
1570 break;
1571 }
1572 }
1573
1574done:
1575 // function finished, get result from the stack.
1576 tv = STACK_TV_BOT(-1);
1577 *rettv = *tv;
1578 tv->v_type = VAR_UNKNOWN;
1579 ret = OK;
1580
1581failed:
1582 for (idx = 0; idx < ectx.ec_stack.ga_len; ++idx)
1583 clear_tv(STACK_TV(idx));
1584 vim_free(ectx.ec_stack.ga_data);
1585 return ret;
1586}
1587
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001588/*
1589 * ":dissassemble".
Bram Moolenaar777770f2020-02-06 21:27:08 +01001590 * We don't really need this at runtime, but we do have tests that require it,
1591 * so always include this.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001592 */
1593 void
1594ex_disassemble(exarg_T *eap)
1595{
Bram Moolenaar21456cd2020-02-13 21:29:32 +01001596 char_u *arg = eap->arg;
Bram Moolenaar0f18b6d2020-02-02 17:22:27 +01001597 char_u *fname;
1598 ufunc_T *ufunc;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001599 dfunc_T *dfunc;
1600 isn_T *instr;
1601 int current;
1602 int line_idx = 0;
1603 int prev_current = 0;
1604
Bram Moolenaar21456cd2020-02-13 21:29:32 +01001605 fname = trans_function_name(&arg, FALSE,
Bram Moolenaar0f18b6d2020-02-02 17:22:27 +01001606 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF, NULL, NULL);
Bram Moolenaar21456cd2020-02-13 21:29:32 +01001607 if (fname == NULL)
1608 {
1609 semsg(_(e_invarg2), eap->arg);
1610 return;
1611 }
1612
Bram Moolenaar0f18b6d2020-02-02 17:22:27 +01001613 ufunc = find_func(fname, NULL);
1614 vim_free(fname);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001615 if (ufunc == NULL)
1616 {
Bram Moolenaardf2ecdd2020-02-16 15:03:48 +01001617 semsg(_("E1061: Cannot find function %s"), eap->arg);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001618 return;
1619 }
1620 if (ufunc->uf_dfunc_idx < 0)
1621 {
Bram Moolenaardf2ecdd2020-02-16 15:03:48 +01001622 semsg(_("E1062: Function %s is not compiled"), eap->arg);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001623 return;
1624 }
1625 if (ufunc->uf_name_exp != NULL)
1626 msg((char *)ufunc->uf_name_exp);
1627 else
1628 msg((char *)ufunc->uf_name);
1629
1630 dfunc = ((dfunc_T *)def_functions.ga_data) + ufunc->uf_dfunc_idx;
1631 instr = dfunc->df_instr;
1632 for (current = 0; current < dfunc->df_instr_count; ++current)
1633 {
1634 isn_T *iptr = &instr[current];
Bram Moolenaarf2460a32020-02-07 22:09:54 +01001635 char *line;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001636
1637 while (line_idx < iptr->isn_lnum && line_idx < ufunc->uf_lines.ga_len)
1638 {
1639 if (current > prev_current)
1640 {
1641 msg_puts("\n\n");
1642 prev_current = current;
1643 }
Bram Moolenaarf2460a32020-02-07 22:09:54 +01001644 line = ((char **)ufunc->uf_lines.ga_data)[line_idx++];
1645 if (line != NULL)
1646 msg(line);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001647 }
1648
1649 switch (iptr->isn_type)
1650 {
1651 case ISN_EXEC:
1652 smsg("%4d EXEC %s", current, iptr->isn_arg.string);
1653 break;
1654 case ISN_ECHO:
1655 {
1656 echo_T *echo = &iptr->isn_arg.echo;
1657
1658 smsg("%4d %s %d", current,
1659 echo->echo_with_white ? "ECHO" : "ECHON",
1660 echo->echo_count);
1661 }
1662 break;
1663 case ISN_LOAD:
1664 if (iptr->isn_arg.number < 0)
1665 smsg("%4d LOAD arg[%lld]", current,
1666 iptr->isn_arg.number + STACK_FRAME_SIZE);
1667 else
1668 smsg("%4d LOAD $%lld", current, iptr->isn_arg.number);
1669 break;
1670 case ISN_LOADV:
1671 smsg("%4d LOADV v:%s", current,
1672 get_vim_var_name(iptr->isn_arg.number));
1673 break;
1674 case ISN_LOADSCRIPT:
1675 {
1676 scriptitem_T *si =
Bram Moolenaar21b9e972020-01-26 19:26:46 +01001677 SCRIPT_ITEM(iptr->isn_arg.script.script_sid);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001678 svar_T *sv = ((svar_T *)si->sn_var_vals.ga_data)
1679 + iptr->isn_arg.script.script_idx;
1680
1681 smsg("%4d LOADSCRIPT %s from %s", current,
1682 sv->sv_name, si->sn_name);
1683 }
1684 break;
1685 case ISN_LOADS:
1686 {
Bram Moolenaarb283a8a2020-02-02 22:24:04 +01001687 scriptitem_T *si = SCRIPT_ITEM(
1688 iptr->isn_arg.loadstore.ls_sid);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001689
1690 smsg("%4d LOADS s:%s from %s", current,
1691 iptr->isn_arg.string, si->sn_name);
1692 }
1693 break;
1694 case ISN_LOADG:
1695 smsg("%4d LOADG g:%s", current, iptr->isn_arg.string);
1696 break;
1697 case ISN_LOADOPT:
1698 smsg("%4d LOADOPT %s", current, iptr->isn_arg.string);
1699 break;
1700 case ISN_LOADENV:
1701 smsg("%4d LOADENV %s", current, iptr->isn_arg.string);
1702 break;
1703 case ISN_LOADREG:
1704 smsg("%4d LOADREG @%c", current, iptr->isn_arg.number);
1705 break;
1706
1707 case ISN_STORE:
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01001708 if (iptr->isn_arg.number < 0)
1709 smsg("%4d STORE arg[%lld]", current,
1710 iptr->isn_arg.number + STACK_FRAME_SIZE);
1711 else
1712 smsg("%4d STORE $%lld", current, iptr->isn_arg.number);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001713 break;
Bram Moolenaarb283a8a2020-02-02 22:24:04 +01001714 case ISN_STOREV:
1715 smsg("%4d STOREV v:%s", current,
1716 get_vim_var_name(iptr->isn_arg.number));
1717 break;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001718 case ISN_STOREG:
Bram Moolenaarb283a8a2020-02-02 22:24:04 +01001719 smsg("%4d STOREG %s", current, iptr->isn_arg.string);
1720 break;
1721 case ISN_STORES:
1722 {
1723 scriptitem_T *si = SCRIPT_ITEM(
1724 iptr->isn_arg.loadstore.ls_sid);
1725
1726 smsg("%4d STORES s:%s in %s", current,
1727 iptr->isn_arg.string, si->sn_name);
1728 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001729 break;
1730 case ISN_STORESCRIPT:
1731 {
1732 scriptitem_T *si =
Bram Moolenaar21b9e972020-01-26 19:26:46 +01001733 SCRIPT_ITEM(iptr->isn_arg.script.script_sid);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001734 svar_T *sv = ((svar_T *)si->sn_var_vals.ga_data)
1735 + iptr->isn_arg.script.script_idx;
1736
1737 smsg("%4d STORESCRIPT %s in %s", current,
1738 sv->sv_name, si->sn_name);
1739 }
1740 break;
1741 case ISN_STOREOPT:
1742 smsg("%4d STOREOPT &%s", current,
1743 iptr->isn_arg.storeopt.so_name);
1744 break;
Bram Moolenaarb283a8a2020-02-02 22:24:04 +01001745 case ISN_STOREENV:
1746 smsg("%4d STOREENV $%s", current, iptr->isn_arg.string);
1747 break;
1748 case ISN_STOREREG:
1749 smsg("%4d STOREREG @%c", current, iptr->isn_arg.number);
1750 break;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001751 case ISN_STORENR:
1752 smsg("%4d STORE %lld in $%d", current,
1753 iptr->isn_arg.storenr.str_val,
1754 iptr->isn_arg.storenr.str_idx);
1755 break;
1756
1757 // constants
1758 case ISN_PUSHNR:
1759 smsg("%4d PUSHNR %lld", current, iptr->isn_arg.number);
1760 break;
1761 case ISN_PUSHBOOL:
1762 case ISN_PUSHSPEC:
1763 smsg("%4d PUSH %s", current,
1764 get_var_special_name(iptr->isn_arg.number));
1765 break;
1766 case ISN_PUSHF:
Bram Moolenaara5d59532020-01-26 21:42:03 +01001767#ifdef FEAT_FLOAT
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001768 smsg("%4d PUSHF %g", current, iptr->isn_arg.fnumber);
Bram Moolenaara5d59532020-01-26 21:42:03 +01001769#endif
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001770 break;
1771 case ISN_PUSHS:
1772 smsg("%4d PUSHS \"%s\"", current, iptr->isn_arg.string);
1773 break;
1774 case ISN_PUSHBLOB:
1775 {
1776 char_u *r;
1777 char_u numbuf[NUMBUFLEN];
1778 char_u *tofree;
1779
1780 r = blob2string(iptr->isn_arg.blob, &tofree, numbuf);
Bram Moolenaarff80cb62020-02-05 22:10:05 +01001781 smsg("%4d PUSHBLOB %s", current, r);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001782 vim_free(tofree);
1783 }
1784 break;
1785 case ISN_PUSHEXC:
1786 smsg("%4d PUSH v:exception", current);
1787 break;
1788 case ISN_NEWLIST:
1789 smsg("%4d NEWLIST size %lld", current, iptr->isn_arg.number);
1790 break;
1791 case ISN_NEWDICT:
1792 smsg("%4d NEWDICT size %lld", current, iptr->isn_arg.number);
1793 break;
1794
1795 // function call
1796 case ISN_BCALL:
1797 {
1798 cbfunc_T *cbfunc = &iptr->isn_arg.bfunc;
1799
1800 smsg("%4d BCALL %s(argc %d)", current,
1801 internal_func_name(cbfunc->cbf_idx),
1802 cbfunc->cbf_argcount);
1803 }
1804 break;
1805 case ISN_DCALL:
1806 {
1807 cdfunc_T *cdfunc = &iptr->isn_arg.dfunc;
1808 dfunc_T *df = ((dfunc_T *)def_functions.ga_data)
1809 + cdfunc->cdf_idx;
1810
1811 smsg("%4d DCALL %s(argc %d)", current,
1812 df->df_ufunc->uf_name_exp != NULL
1813 ? df->df_ufunc->uf_name_exp
1814 : df->df_ufunc->uf_name, cdfunc->cdf_argcount);
1815 }
1816 break;
1817 case ISN_UCALL:
1818 {
1819 cufunc_T *cufunc = &iptr->isn_arg.ufunc;
1820
1821 smsg("%4d UCALL %s(argc %d)", current,
1822 cufunc->cuf_name, cufunc->cuf_argcount);
1823 }
1824 break;
1825 case ISN_PCALL:
1826 {
1827 cpfunc_T *cpfunc = &iptr->isn_arg.pfunc;
1828
1829 smsg("%4d PCALL%s (argc %d)", current,
1830 cpfunc->cpf_top ? " top" : "", cpfunc->cpf_argcount);
1831 }
1832 break;
1833 case ISN_RETURN:
1834 smsg("%4d RETURN", current);
1835 break;
1836 case ISN_FUNCREF:
1837 {
1838 dfunc_T *df = ((dfunc_T *)def_functions.ga_data)
1839 + iptr->isn_arg.number;
1840
1841 smsg("%4d FUNCREF %s", current, df->df_ufunc->uf_name);
1842 }
1843 break;
1844
1845 case ISN_JUMP:
1846 {
1847 char *when = "?";
1848
1849 switch (iptr->isn_arg.jump.jump_when)
1850 {
1851 case JUMP_ALWAYS:
1852 when = "JUMP";
1853 break;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001854 case JUMP_AND_KEEP_IF_TRUE:
1855 when = "JUMP_AND_KEEP_IF_TRUE";
1856 break;
1857 case JUMP_IF_FALSE:
1858 when = "JUMP_IF_FALSE";
1859 break;
1860 case JUMP_AND_KEEP_IF_FALSE:
1861 when = "JUMP_AND_KEEP_IF_FALSE";
1862 break;
1863 }
1864 smsg("%4d %s -> %lld", current, when,
1865 iptr->isn_arg.jump.jump_where);
1866 }
1867 break;
1868
1869 case ISN_FOR:
1870 {
1871 forloop_T *forloop = &iptr->isn_arg.forloop;
1872
1873 smsg("%4d FOR $%d -> %d", current,
1874 forloop->for_idx, forloop->for_end);
1875 }
1876 break;
1877
1878 case ISN_TRY:
1879 {
1880 try_T *try = &iptr->isn_arg.try;
1881
1882 smsg("%4d TRY catch -> %d, finally -> %d", current,
1883 try->try_catch, try->try_finally);
1884 }
1885 break;
1886 case ISN_CATCH:
1887 // TODO
1888 smsg("%4d CATCH", current);
1889 break;
1890 case ISN_ENDTRY:
1891 smsg("%4d ENDTRY", current);
1892 break;
1893 case ISN_THROW:
1894 smsg("%4d THROW", current);
1895 break;
1896
1897 // expression operations on number
1898 case ISN_OPNR:
1899 case ISN_OPFLOAT:
1900 case ISN_OPANY:
1901 {
1902 char *what;
1903 char *ins;
1904
1905 switch (iptr->isn_arg.op.op_type)
1906 {
1907 case EXPR_MULT: what = "*"; break;
1908 case EXPR_DIV: what = "/"; break;
1909 case EXPR_REM: what = "%"; break;
1910 case EXPR_SUB: what = "-"; break;
1911 case EXPR_ADD: what = "+"; break;
1912 default: what = "???"; break;
1913 }
1914 switch (iptr->isn_type)
1915 {
1916 case ISN_OPNR: ins = "OPNR"; break;
1917 case ISN_OPFLOAT: ins = "OPFLOAT"; break;
1918 case ISN_OPANY: ins = "OPANY"; break;
1919 default: ins = "???"; break;
1920 }
1921 smsg("%4d %s %s", current, ins, what);
1922 }
1923 break;
1924
1925 case ISN_COMPAREBOOL:
1926 case ISN_COMPARESPECIAL:
1927 case ISN_COMPARENR:
1928 case ISN_COMPAREFLOAT:
1929 case ISN_COMPARESTRING:
1930 case ISN_COMPAREBLOB:
1931 case ISN_COMPARELIST:
1932 case ISN_COMPAREDICT:
1933 case ISN_COMPAREFUNC:
1934 case ISN_COMPAREPARTIAL:
1935 case ISN_COMPAREANY:
1936 {
1937 char *p;
1938 char buf[10];
1939 char *type;
1940
1941 switch (iptr->isn_arg.op.op_type)
1942 {
1943 case EXPR_EQUAL: p = "=="; break;
1944 case EXPR_NEQUAL: p = "!="; break;
1945 case EXPR_GREATER: p = ">"; break;
1946 case EXPR_GEQUAL: p = ">="; break;
1947 case EXPR_SMALLER: p = "<"; break;
1948 case EXPR_SEQUAL: p = "<="; break;
1949 case EXPR_MATCH: p = "=~"; break;
1950 case EXPR_IS: p = "is"; break;
1951 case EXPR_ISNOT: p = "isnot"; break;
1952 case EXPR_NOMATCH: p = "!~"; break;
1953 default: p = "???"; break;
1954 }
1955 STRCPY(buf, p);
1956 if (iptr->isn_arg.op.op_ic == TRUE)
1957 strcat(buf, "?");
1958 switch(iptr->isn_type)
1959 {
1960 case ISN_COMPAREBOOL: type = "COMPAREBOOL"; break;
1961 case ISN_COMPARESPECIAL:
1962 type = "COMPARESPECIAL"; break;
1963 case ISN_COMPARENR: type = "COMPARENR"; break;
1964 case ISN_COMPAREFLOAT: type = "COMPAREFLOAT"; break;
1965 case ISN_COMPARESTRING:
1966 type = "COMPARESTRING"; break;
1967 case ISN_COMPAREBLOB: type = "COMPAREBLOB"; break;
1968 case ISN_COMPARELIST: type = "COMPARELIST"; break;
1969 case ISN_COMPAREDICT: type = "COMPAREDICT"; break;
1970 case ISN_COMPAREFUNC: type = "COMPAREFUNC"; break;
1971 case ISN_COMPAREPARTIAL:
1972 type = "COMPAREPARTIAL"; break;
1973 case ISN_COMPAREANY: type = "COMPAREANY"; break;
1974 default: type = "???"; break;
1975 }
1976
1977 smsg("%4d %s %s", current, type, buf);
1978 }
1979 break;
1980
1981 case ISN_ADDLIST: smsg("%4d ADDLIST", current); break;
1982 case ISN_ADDBLOB: smsg("%4d ADDBLOB", current); break;
1983
1984 // expression operations
1985 case ISN_CONCAT: smsg("%4d CONCAT", current); break;
1986 case ISN_INDEX: smsg("%4d INDEX", current); break;
1987 case ISN_MEMBER: smsg("%4d MEMBER %s", current,
1988 iptr->isn_arg.string); break;
1989 case ISN_NEGATENR: smsg("%4d NEGATENR", current); break;
1990
1991 case ISN_CHECKNR: smsg("%4d CHECKNR", current); break;
1992 case ISN_CHECKTYPE: smsg("%4d CHECKTYPE %s stack[%d]", current,
1993 vartype_name(iptr->isn_arg.type.ct_type),
1994 iptr->isn_arg.type.ct_off);
1995 break;
1996 case ISN_2BOOL: if (iptr->isn_arg.number)
1997 smsg("%4d INVERT (!val)", current);
1998 else
1999 smsg("%4d 2BOOL (!!val)", current);
2000 break;
2001 case ISN_2STRING: smsg("%4d 2STRING stack[%d]", current,
2002 iptr->isn_arg.number);
2003 break;
2004
2005 case ISN_DROP: smsg("%4d DROP", current); break;
2006 }
2007 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002008}
2009
2010/*
2011 * Return TRUE when "tv" is not falsey: non-zero, non-empty string, non-empty
2012 * list, etc. Mostly like what JavaScript does, except that empty list and
2013 * empty dictionary are FALSE.
2014 */
2015 int
2016tv2bool(typval_T *tv)
2017{
2018 switch (tv->v_type)
2019 {
2020 case VAR_NUMBER:
2021 return tv->vval.v_number != 0;
2022 case VAR_FLOAT:
2023#ifdef FEAT_FLOAT
2024 return tv->vval.v_float != 0.0;
2025#else
2026 break;
2027#endif
2028 case VAR_PARTIAL:
2029 return tv->vval.v_partial != NULL;
2030 case VAR_FUNC:
2031 case VAR_STRING:
2032 return tv->vval.v_string != NULL && *tv->vval.v_string != NUL;
2033 case VAR_LIST:
2034 return tv->vval.v_list != NULL && tv->vval.v_list->lv_len > 0;
2035 case VAR_DICT:
2036 return tv->vval.v_dict != NULL
2037 && tv->vval.v_dict->dv_hashtab.ht_used > 0;
2038 case VAR_BOOL:
2039 case VAR_SPECIAL:
2040 return tv->vval.v_number == VVAL_TRUE ? TRUE : FALSE;
2041 case VAR_JOB:
2042#ifdef FEAT_JOB_CHANNEL
2043 return tv->vval.v_job != NULL;
2044#else
2045 break;
2046#endif
2047 case VAR_CHANNEL:
2048#ifdef FEAT_JOB_CHANNEL
2049 return tv->vval.v_channel != NULL;
2050#else
2051 break;
2052#endif
2053 case VAR_BLOB:
2054 return tv->vval.v_blob != NULL && tv->vval.v_blob->bv_ga.ga_len > 0;
2055 case VAR_UNKNOWN:
2056 case VAR_VOID:
2057 break;
2058 }
2059 return FALSE;
2060}
2061
2062/*
2063 * If "tv" is a string give an error and return FAIL.
2064 */
2065 int
2066check_not_string(typval_T *tv)
2067{
2068 if (tv->v_type == VAR_STRING)
2069 {
2070 emsg(_("E1030: Using a String as a Number"));
2071 clear_tv(tv);
2072 return FAIL;
2073 }
2074 return OK;
2075}
2076
2077
2078#endif // FEAT_EVAL