1 /* 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. 26 typedef struct { 27 int tcd_frame_idx; // ec_frame_idx at ISN_TRY 28 int tcd_stack_len; // size of ectx.ec_stack at ISN_TRY 29 int tcd_in_catch; // in catch or finally block 30 int tcd_did_throw; // set did_throw in :endtry 31 int tcd_catch_idx; // instruction of the first :catch or :finally 32 int tcd_finally_idx; // instruction of the :finally block or zero 33 int tcd_endtry_idx; // instruction of the :endtry 34 int tcd_caught; // catch block entered 35 int tcd_cont; // :continue encountered, jump here (minus one) 36 int tcd_return; // when TRUE return from end of :finally 37 } trycmd_T; 38 39 // Data local to a function. 40 // On a function call, if not empty, is saved on the stack and restored when 41 // returning. 42 typedef struct { 43 int floc_restore_cmdmod; 44 cmdmod_T floc_save_cmdmod; 45 int floc_restore_cmdmod_stacklen; 46 } funclocal_T; 47 48 // Structure to hold a reference to an outer_T, with information of whether it 49 // was allocated. 50 typedef struct { 51 outer_T *or_outer; 52 partial_T *or_partial; // decrement "or_partial->pt_refcount" later 53 int or_outer_allocated; // free "or_outer" later 54 } outer_ref_T; 55 56 // A stack is used to store: 57 // - arguments passed to a :def function 58 // - info about the calling function, to use when returning 59 // - local variables 60 // - temporary values 61 // 62 // In detail (FP == Frame Pointer): 63 // arg1 first argument from caller (if present) 64 // arg2 second argument from caller (if present) 65 // extra_arg1 any missing optional argument default value 66 // FP -> cur_func calling function 67 // current previous instruction pointer 68 // frame_ptr previous Frame Pointer 69 // var1 space for local variable 70 // var2 space for local variable 71 // .... fixed space for max. number of local variables 72 // temp temporary values 73 // .... flexible space for temporary values (can grow big) 74 75 /* 76 * Execution context. 77 */ 78 struct ectx_S { 79 garray_T ec_stack; // stack of typval_T values 80 int ec_frame_idx; // index in ec_stack: context of ec_dfunc_idx 81 int ec_initial_frame_idx; // frame index when called 82 83 outer_ref_T *ec_outer_ref; // outer scope used for closures, allocated 84 funclocal_T ec_funclocal; 85 86 garray_T ec_trystack; // stack of trycmd_T values 87 88 isn_T *ec_instr; // array with instructions 89 int ec_dfunc_idx; // current function index 90 int ec_iidx; // index in ec_instr: instruction to execute 91 92 garray_T ec_funcrefs; // partials that might be a closure 93 94 int ec_did_emsg_before; 95 int ec_trylevel_at_start; 96 where_T ec_where; 97 }; 98 99 #ifdef FEAT_PROFILE 100 // stack of profinfo_T used when profiling. 101 static garray_T profile_info_ga = {0, 0, sizeof(profinfo_T), 20, NULL}; 102 #endif 103 104 // Get pointer to item relative to the bottom of the stack, -1 is the last one. 105 #define STACK_TV_BOT(idx) (((typval_T *)ectx->ec_stack.ga_data) + ectx->ec_stack.ga_len + (idx)) 106 107 void 108 to_string_error(vartype_T vartype) 109 { 110 semsg(_(e_cannot_convert_str_to_string), vartype_name(vartype)); 111 } 112 113 /* 114 * Return the number of arguments, including optional arguments and any vararg. 115 */ 116 static int 117 ufunc_argcount(ufunc_T *ufunc) 118 { 119 return ufunc->uf_args.ga_len + (ufunc->uf_va_name != NULL ? 1 : 0); 120 } 121 122 /* 123 * Create a new list from "count" items at the bottom of the stack. 124 * When "count" is zero an empty list is added to the stack. 125 */ 126 static int 127 exe_newlist(int count, ectx_T *ectx) 128 { 129 list_T *list = list_alloc_with_items(count); 130 int idx; 131 typval_T *tv; 132 133 if (list == NULL) 134 return FAIL; 135 for (idx = 0; idx < count; ++idx) 136 list_set_item(list, idx, STACK_TV_BOT(idx - count)); 137 138 if (count > 0) 139 ectx->ec_stack.ga_len -= count - 1; 140 else if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 141 return FAIL; 142 else 143 ++ectx->ec_stack.ga_len; 144 tv = STACK_TV_BOT(-1); 145 tv->v_type = VAR_LIST; 146 tv->vval.v_list = list; 147 ++list->lv_refcount; 148 return OK; 149 } 150 151 /* 152 * If debug_tick changed check if "ufunc" has a breakpoint and update 153 * "uf_has_breakpoint". 154 */ 155 static void 156 update_has_breakpoint(ufunc_T *ufunc) 157 { 158 if (ufunc->uf_debug_tick != debug_tick) 159 { 160 linenr_T breakpoint; 161 162 ufunc->uf_debug_tick = debug_tick; 163 breakpoint = dbg_find_breakpoint(FALSE, ufunc->uf_name, 0); 164 ufunc->uf_has_breakpoint = breakpoint > 0; 165 } 166 } 167 168 /* 169 * Call compiled function "cdf_idx" from compiled code. 170 * This adds a stack frame and sets the instruction pointer to the start of the 171 * called function. 172 * If "pt" is not null use "pt->pt_outer" for ec_outer_ref->or_outer. 173 * 174 * Stack has: 175 * - current arguments (already there) 176 * - omitted optional argument (default values) added here 177 * - stack frame: 178 * - pointer to calling function 179 * - Index of next instruction in calling function 180 * - previous frame pointer 181 * - reserved space for local variables 182 */ 183 static int 184 call_dfunc( 185 int cdf_idx, 186 partial_T *pt, 187 int argcount_arg, 188 ectx_T *ectx) 189 { 190 int argcount = argcount_arg; 191 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data) + cdf_idx; 192 ufunc_T *ufunc = dfunc->df_ufunc; 193 int did_emsg_before = did_emsg_cumul + did_emsg; 194 int arg_to_add; 195 int vararg_count = 0; 196 int varcount; 197 int idx; 198 estack_T *entry; 199 funclocal_T *floc = NULL; 200 int res = OK; 201 202 if (dfunc->df_deleted) 203 { 204 // don't use ufunc->uf_name, it may have been freed 205 emsg_funcname(e_func_deleted, 206 dfunc->df_name == NULL ? (char_u *)"unknown" : dfunc->df_name); 207 return FAIL; 208 } 209 210 #ifdef FEAT_PROFILE 211 if (do_profiling == PROF_YES) 212 { 213 if (GA_GROW_OK(&profile_info_ga, 1)) 214 { 215 profinfo_T *info = ((profinfo_T *)profile_info_ga.ga_data) 216 + profile_info_ga.ga_len; 217 ++profile_info_ga.ga_len; 218 CLEAR_POINTER(info); 219 profile_may_start_func(info, ufunc, 220 (((dfunc_T *)def_functions.ga_data) 221 + ectx->ec_dfunc_idx)->df_ufunc); 222 } 223 } 224 #endif 225 226 // Update uf_has_breakpoint if needed. 227 update_has_breakpoint(ufunc); 228 229 // When debugging and using "cont" switches to the not-debugged 230 // instructions, may need to still compile them. 231 if (func_needs_compiling(ufunc, COMPILE_TYPE(ufunc))) 232 { 233 res = compile_def_function(ufunc, FALSE, COMPILE_TYPE(ufunc), NULL); 234 235 // compile_def_function() may cause def_functions.ga_data to change 236 dfunc = ((dfunc_T *)def_functions.ga_data) + cdf_idx; 237 } 238 if (res == FAIL || INSTRUCTIONS(dfunc) == NULL) 239 { 240 if (did_emsg_cumul + did_emsg == did_emsg_before) 241 semsg(_(e_function_is_not_compiled_str), 242 printable_func_name(ufunc)); 243 return FAIL; 244 } 245 246 if (ufunc->uf_va_name != NULL) 247 { 248 // Need to make a list out of the vararg arguments. 249 // Stack at time of call with 2 varargs: 250 // normal_arg 251 // optional_arg 252 // vararg_1 253 // vararg_2 254 // After creating the list: 255 // normal_arg 256 // optional_arg 257 // vararg-list 258 // With missing optional arguments we get: 259 // normal_arg 260 // After creating the list 261 // normal_arg 262 // (space for optional_arg) 263 // vararg-list 264 vararg_count = argcount - ufunc->uf_args.ga_len; 265 if (vararg_count < 0) 266 vararg_count = 0; 267 else 268 argcount -= vararg_count; 269 if (exe_newlist(vararg_count, ectx) == FAIL) 270 return FAIL; 271 272 vararg_count = 1; 273 } 274 275 arg_to_add = ufunc->uf_args.ga_len - argcount; 276 if (arg_to_add < 0) 277 { 278 if (arg_to_add == -1) 279 emsg(_(e_one_argument_too_many)); 280 else 281 semsg(_(e_nr_arguments_too_many), -arg_to_add); 282 return FAIL; 283 } 284 285 // Reserve space for: 286 // - missing arguments 287 // - stack frame 288 // - local variables 289 // - if needed: a counter for number of closures created in 290 // ectx->ec_funcrefs. 291 varcount = dfunc->df_varcount + dfunc->df_has_closure; 292 if (GA_GROW_FAILS(&ectx->ec_stack, arg_to_add + STACK_FRAME_SIZE + varcount)) 293 return FAIL; 294 295 // If depth of calling is getting too high, don't execute the function. 296 if (funcdepth_increment() == FAIL) 297 return FAIL; 298 ++ex_nesting_level; 299 300 // Only make a copy of funclocal if it contains something to restore. 301 if (ectx->ec_funclocal.floc_restore_cmdmod) 302 { 303 floc = ALLOC_ONE(funclocal_T); 304 if (floc == NULL) 305 return FAIL; 306 *floc = ectx->ec_funclocal; 307 ectx->ec_funclocal.floc_restore_cmdmod = FALSE; 308 } 309 310 // Move the vararg-list to below the missing optional arguments. 311 if (vararg_count > 0 && arg_to_add > 0) 312 *STACK_TV_BOT(arg_to_add - 1) = *STACK_TV_BOT(-1); 313 314 // Reserve space for omitted optional arguments, filled in soon. 315 for (idx = 0; idx < arg_to_add; ++idx) 316 STACK_TV_BOT(idx - vararg_count)->v_type = VAR_UNKNOWN; 317 ectx->ec_stack.ga_len += arg_to_add; 318 319 // Store current execution state in stack frame for ISN_RETURN. 320 STACK_TV_BOT(STACK_FRAME_FUNC_OFF)->vval.v_number = ectx->ec_dfunc_idx; 321 STACK_TV_BOT(STACK_FRAME_IIDX_OFF)->vval.v_number = ectx->ec_iidx; 322 STACK_TV_BOT(STACK_FRAME_INSTR_OFF)->vval.v_string = (void *)ectx->ec_instr; 323 STACK_TV_BOT(STACK_FRAME_OUTER_OFF)->vval.v_string = 324 (void *)ectx->ec_outer_ref; 325 STACK_TV_BOT(STACK_FRAME_FUNCLOCAL_OFF)->vval.v_string = (void *)floc; 326 STACK_TV_BOT(STACK_FRAME_IDX_OFF)->vval.v_number = ectx->ec_frame_idx; 327 ectx->ec_frame_idx = ectx->ec_stack.ga_len; 328 329 // Initialize local variables 330 for (idx = 0; idx < dfunc->df_varcount; ++idx) 331 STACK_TV_BOT(STACK_FRAME_SIZE + idx)->v_type = VAR_UNKNOWN; 332 if (dfunc->df_has_closure) 333 { 334 typval_T *tv = STACK_TV_BOT(STACK_FRAME_SIZE + dfunc->df_varcount); 335 336 tv->v_type = VAR_NUMBER; 337 tv->vval.v_number = 0; 338 } 339 ectx->ec_stack.ga_len += STACK_FRAME_SIZE + varcount; 340 341 if (pt != NULL || ufunc->uf_partial != NULL 342 || (ufunc->uf_flags & FC_CLOSURE)) 343 { 344 outer_ref_T *ref = ALLOC_CLEAR_ONE(outer_ref_T); 345 346 if (ref == NULL) 347 return FAIL; 348 if (pt != NULL) 349 { 350 ref->or_outer = &pt->pt_outer; 351 ++pt->pt_refcount; 352 ref->or_partial = pt; 353 } 354 else if (ufunc->uf_partial != NULL) 355 { 356 ref->or_outer = &ufunc->uf_partial->pt_outer; 357 ++ufunc->uf_partial->pt_refcount; 358 ref->or_partial = ufunc->uf_partial; 359 } 360 else 361 { 362 ref->or_outer = ALLOC_CLEAR_ONE(outer_T); 363 if (unlikely(ref->or_outer == NULL)) 364 { 365 vim_free(ref); 366 return FAIL; 367 } 368 ref->or_outer_allocated = TRUE; 369 ref->or_outer->out_stack = &ectx->ec_stack; 370 ref->or_outer->out_frame_idx = ectx->ec_frame_idx; 371 if (ectx->ec_outer_ref != NULL) 372 ref->or_outer->out_up = ectx->ec_outer_ref->or_outer; 373 } 374 ectx->ec_outer_ref = ref; 375 } 376 else 377 ectx->ec_outer_ref = NULL; 378 379 ++ufunc->uf_calls; 380 381 // Set execution state to the start of the called function. 382 ectx->ec_dfunc_idx = cdf_idx; 383 ectx->ec_instr = INSTRUCTIONS(dfunc); 384 entry = estack_push_ufunc(ufunc, 1); 385 if (entry != NULL) 386 { 387 // Set the script context to the script where the function was defined. 388 // Save the current context so it can be restored on return. 389 entry->es_save_sctx = current_sctx; 390 current_sctx = ufunc->uf_script_ctx; 391 } 392 393 // Start execution at the first instruction. 394 ectx->ec_iidx = 0; 395 396 return OK; 397 } 398 399 // Get pointer to item in the stack. 400 #define STACK_TV(idx) (((typval_T *)ectx->ec_stack.ga_data) + idx) 401 402 /* 403 * Used when returning from a function: Check if any closure is still 404 * referenced. If so then move the arguments and variables to a separate piece 405 * of stack to be used when the closure is called. 406 * When "free_arguments" is TRUE the arguments are to be freed. 407 * Returns FAIL when out of memory. 408 */ 409 static int 410 handle_closure_in_use(ectx_T *ectx, int free_arguments) 411 { 412 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data) 413 + ectx->ec_dfunc_idx; 414 int argcount; 415 int top; 416 int idx; 417 typval_T *tv; 418 int closure_in_use = FALSE; 419 garray_T *gap = &ectx->ec_funcrefs; 420 varnumber_T closure_count; 421 422 if (dfunc->df_ufunc == NULL) 423 return OK; // function was freed 424 if (dfunc->df_has_closure == 0) 425 return OK; // no closures 426 tv = STACK_TV(ectx->ec_frame_idx + STACK_FRAME_SIZE + dfunc->df_varcount); 427 closure_count = tv->vval.v_number; 428 if (closure_count == 0) 429 return OK; // no funcrefs created 430 431 argcount = ufunc_argcount(dfunc->df_ufunc); 432 top = ectx->ec_frame_idx - argcount; 433 434 // Check if any created closure is still in use. 435 for (idx = 0; idx < closure_count; ++idx) 436 { 437 partial_T *pt; 438 int off = gap->ga_len - closure_count + idx; 439 440 if (off < 0) 441 continue; // count is off or already done 442 pt = ((partial_T **)gap->ga_data)[off]; 443 if (pt->pt_refcount > 1) 444 { 445 int refcount = pt->pt_refcount; 446 int i; 447 448 // A Reference in a local variables doesn't count, it gets 449 // unreferenced on return. 450 for (i = 0; i < dfunc->df_varcount; ++i) 451 { 452 typval_T *stv = STACK_TV(ectx->ec_frame_idx 453 + STACK_FRAME_SIZE + i); 454 if (stv->v_type == VAR_PARTIAL && pt == stv->vval.v_partial) 455 --refcount; 456 } 457 if (refcount > 1) 458 { 459 closure_in_use = TRUE; 460 break; 461 } 462 } 463 } 464 465 if (closure_in_use) 466 { 467 funcstack_T *funcstack = ALLOC_CLEAR_ONE(funcstack_T); 468 typval_T *stack; 469 470 // A closure is using the arguments and/or local variables. 471 // Move them to the called function. 472 if (funcstack == NULL) 473 return FAIL; 474 funcstack->fs_var_offset = argcount + STACK_FRAME_SIZE; 475 funcstack->fs_ga.ga_len = funcstack->fs_var_offset + dfunc->df_varcount; 476 stack = ALLOC_CLEAR_MULT(typval_T, funcstack->fs_ga.ga_len); 477 funcstack->fs_ga.ga_data = stack; 478 if (stack == NULL) 479 { 480 vim_free(funcstack); 481 return FAIL; 482 } 483 484 // Move or copy the arguments. 485 for (idx = 0; idx < argcount; ++idx) 486 { 487 tv = STACK_TV(top + idx); 488 if (free_arguments) 489 { 490 *(stack + idx) = *tv; 491 tv->v_type = VAR_UNKNOWN; 492 } 493 else 494 copy_tv(tv, stack + idx); 495 } 496 // Move the local variables. 497 for (idx = 0; idx < dfunc->df_varcount; ++idx) 498 { 499 tv = STACK_TV(ectx->ec_frame_idx + STACK_FRAME_SIZE + idx); 500 501 // A partial created for a local function, that is also used as a 502 // local variable, has a reference count for the variable, thus 503 // will never go down to zero. When all these refcounts are one 504 // then the funcstack is unused. We need to count how many we have 505 // so we need when to check. 506 if (tv->v_type == VAR_PARTIAL && tv->vval.v_partial != NULL) 507 { 508 int i; 509 510 for (i = 0; i < closure_count; ++i) 511 if (tv->vval.v_partial == ((partial_T **)gap->ga_data)[ 512 gap->ga_len - closure_count + i]) 513 ++funcstack->fs_min_refcount; 514 } 515 516 *(stack + funcstack->fs_var_offset + idx) = *tv; 517 tv->v_type = VAR_UNKNOWN; 518 } 519 520 for (idx = 0; idx < closure_count; ++idx) 521 { 522 partial_T *pt = ((partial_T **)gap->ga_data)[gap->ga_len 523 - closure_count + idx]; 524 if (pt->pt_refcount > 1) 525 { 526 ++funcstack->fs_refcount; 527 pt->pt_funcstack = funcstack; 528 pt->pt_outer.out_stack = &funcstack->fs_ga; 529 pt->pt_outer.out_frame_idx = ectx->ec_frame_idx - top; 530 } 531 } 532 } 533 534 for (idx = 0; idx < closure_count; ++idx) 535 partial_unref(((partial_T **)gap->ga_data)[gap->ga_len 536 - closure_count + idx]); 537 gap->ga_len -= closure_count; 538 if (gap->ga_len == 0) 539 ga_clear(gap); 540 541 return OK; 542 } 543 544 /* 545 * Called when a partial is freed or its reference count goes down to one. The 546 * funcstack may be the only reference to the partials in the local variables. 547 * Go over all of them, the funcref and can be freed if all partials 548 * referencing the funcstack have a reference count of one. 549 */ 550 void 551 funcstack_check_refcount(funcstack_T *funcstack) 552 { 553 int i; 554 garray_T *gap = &funcstack->fs_ga; 555 int done = 0; 556 557 if (funcstack->fs_refcount > funcstack->fs_min_refcount) 558 return; 559 for (i = funcstack->fs_var_offset; i < gap->ga_len; ++i) 560 { 561 typval_T *tv = ((typval_T *)gap->ga_data) + i; 562 563 if (tv->v_type == VAR_PARTIAL && tv->vval.v_partial != NULL 564 && tv->vval.v_partial->pt_funcstack == funcstack 565 && tv->vval.v_partial->pt_refcount == 1) 566 ++done; 567 } 568 if (done == funcstack->fs_min_refcount) 569 { 570 typval_T *stack = gap->ga_data; 571 572 // All partials referencing the funcstack have a reference count of 573 // one, thus the funcstack is no longer of use. 574 for (i = 0; i < gap->ga_len; ++i) 575 clear_tv(stack + i); 576 vim_free(stack); 577 vim_free(funcstack); 578 } 579 } 580 581 /* 582 * Return from the current function. 583 */ 584 static int 585 func_return(ectx_T *ectx) 586 { 587 int idx; 588 int ret_idx; 589 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data) 590 + ectx->ec_dfunc_idx; 591 int argcount = ufunc_argcount(dfunc->df_ufunc); 592 int top = ectx->ec_frame_idx - argcount; 593 estack_T *entry; 594 int prev_dfunc_idx = STACK_TV(ectx->ec_frame_idx 595 + STACK_FRAME_FUNC_OFF)->vval.v_number; 596 funclocal_T *floc; 597 #ifdef FEAT_PROFILE 598 dfunc_T *prev_dfunc = ((dfunc_T *)def_functions.ga_data) 599 + prev_dfunc_idx; 600 601 if (do_profiling == PROF_YES) 602 { 603 ufunc_T *caller = prev_dfunc->df_ufunc; 604 605 if (dfunc->df_ufunc->uf_profiling 606 || (caller != NULL && caller->uf_profiling)) 607 { 608 profile_may_end_func(((profinfo_T *)profile_info_ga.ga_data) 609 + profile_info_ga.ga_len - 1, dfunc->df_ufunc, caller); 610 --profile_info_ga.ga_len; 611 } 612 } 613 #endif 614 // TODO: when is it safe to delete the function when it is no longer used? 615 --dfunc->df_ufunc->uf_calls; 616 617 // execution context goes one level up 618 entry = estack_pop(); 619 if (entry != NULL) 620 current_sctx = entry->es_save_sctx; 621 622 if (handle_closure_in_use(ectx, TRUE) == FAIL) 623 return FAIL; 624 625 // Clear the arguments. 626 for (idx = top; idx < ectx->ec_frame_idx; ++idx) 627 clear_tv(STACK_TV(idx)); 628 629 // Clear local variables and temp values, but not the return value. 630 for (idx = ectx->ec_frame_idx + STACK_FRAME_SIZE; 631 idx < ectx->ec_stack.ga_len - 1; ++idx) 632 clear_tv(STACK_TV(idx)); 633 634 // The return value should be on top of the stack. However, when aborting 635 // it may not be there and ec_frame_idx is the top of the stack. 636 ret_idx = ectx->ec_stack.ga_len - 1; 637 if (ret_idx == ectx->ec_frame_idx + STACK_FRAME_IDX_OFF) 638 ret_idx = 0; 639 640 if (ectx->ec_outer_ref != NULL) 641 { 642 if (ectx->ec_outer_ref->or_outer_allocated) 643 vim_free(ectx->ec_outer_ref->or_outer); 644 partial_unref(ectx->ec_outer_ref->or_partial); 645 vim_free(ectx->ec_outer_ref); 646 } 647 648 // Restore the previous frame. 649 ectx->ec_dfunc_idx = prev_dfunc_idx; 650 ectx->ec_iidx = STACK_TV(ectx->ec_frame_idx 651 + STACK_FRAME_IIDX_OFF)->vval.v_number; 652 ectx->ec_instr = (void *)STACK_TV(ectx->ec_frame_idx 653 + STACK_FRAME_INSTR_OFF)->vval.v_string; 654 ectx->ec_outer_ref = (void *)STACK_TV(ectx->ec_frame_idx 655 + STACK_FRAME_OUTER_OFF)->vval.v_string; 656 floc = (void *)STACK_TV(ectx->ec_frame_idx 657 + STACK_FRAME_FUNCLOCAL_OFF)->vval.v_string; 658 // restoring ec_frame_idx must be last 659 ectx->ec_frame_idx = STACK_TV(ectx->ec_frame_idx 660 + STACK_FRAME_IDX_OFF)->vval.v_number; 661 662 if (floc == NULL) 663 ectx->ec_funclocal.floc_restore_cmdmod = FALSE; 664 else 665 { 666 ectx->ec_funclocal = *floc; 667 vim_free(floc); 668 } 669 670 if (ret_idx > 0) 671 { 672 // Reset the stack to the position before the call, with a spot for the 673 // return value, moved there from above the frame. 674 ectx->ec_stack.ga_len = top + 1; 675 *STACK_TV_BOT(-1) = *STACK_TV(ret_idx); 676 } 677 else 678 // Reset the stack to the position before the call. 679 ectx->ec_stack.ga_len = top; 680 681 funcdepth_decrement(); 682 --ex_nesting_level; 683 return OK; 684 } 685 686 #undef STACK_TV 687 688 /* 689 * Prepare arguments and rettv for calling a builtin or user function. 690 */ 691 static int 692 call_prepare(int argcount, typval_T *argvars, ectx_T *ectx) 693 { 694 int idx; 695 typval_T *tv; 696 697 // Move arguments from bottom of the stack to argvars[] and add terminator. 698 for (idx = 0; idx < argcount; ++idx) 699 argvars[idx] = *STACK_TV_BOT(idx - argcount); 700 argvars[argcount].v_type = VAR_UNKNOWN; 701 702 // Result replaces the arguments on the stack. 703 if (argcount > 0) 704 ectx->ec_stack.ga_len -= argcount - 1; 705 else if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 706 return FAIL; 707 else 708 ++ectx->ec_stack.ga_len; 709 710 // Default return value is zero. 711 tv = STACK_TV_BOT(-1); 712 tv->v_type = VAR_NUMBER; 713 tv->vval.v_number = 0; 714 715 return OK; 716 } 717 718 // Ugly global to avoid passing the execution context around through many 719 // layers. 720 static ectx_T *current_ectx = NULL; 721 722 /* 723 * Call a builtin function by index. 724 */ 725 static int 726 call_bfunc(int func_idx, int argcount, ectx_T *ectx) 727 { 728 typval_T argvars[MAX_FUNC_ARGS]; 729 int idx; 730 int did_emsg_before = did_emsg; 731 ectx_T *prev_ectx = current_ectx; 732 char *save_func_name = ectx->ec_where.wt_func_name; 733 734 if (call_prepare(argcount, argvars, ectx) == FAIL) 735 return FAIL; 736 ectx->ec_where.wt_func_name = internal_func_name(func_idx); 737 738 // Call the builtin function. Set "current_ectx" so that when it 739 // recursively invokes call_def_function() a closure context can be set. 740 current_ectx = ectx; 741 call_internal_func_by_idx(func_idx, argvars, STACK_TV_BOT(-1)); 742 current_ectx = prev_ectx; 743 ectx->ec_where.wt_func_name = save_func_name; 744 745 // Clear the arguments. 746 for (idx = 0; idx < argcount; ++idx) 747 clear_tv(&argvars[idx]); 748 749 if (did_emsg > did_emsg_before) 750 return FAIL; 751 return OK; 752 } 753 754 /* 755 * Execute a user defined function. 756 * If the function is compiled this will add a stack frame and set the 757 * instruction pointer at the start of the function. 758 * Otherwise the function is called here. 759 * If "pt" is not null use "pt->pt_outer" for ec_outer_ref->or_outer. 760 * "iptr" can be used to replace the instruction with a more efficient one. 761 */ 762 static int 763 call_ufunc( 764 ufunc_T *ufunc, 765 partial_T *pt, 766 int argcount, 767 ectx_T *ectx, 768 isn_T *iptr) 769 { 770 typval_T argvars[MAX_FUNC_ARGS]; 771 funcexe_T funcexe; 772 int error; 773 int idx; 774 int did_emsg_before = did_emsg; 775 compiletype_T compile_type = COMPILE_TYPE(ufunc); 776 777 if (func_needs_compiling(ufunc, compile_type) 778 && compile_def_function(ufunc, FALSE, compile_type, NULL) 779 == FAIL) 780 return FAIL; 781 if (ufunc->uf_def_status == UF_COMPILED) 782 { 783 error = check_user_func_argcount(ufunc, argcount); 784 if (error != FCERR_UNKNOWN) 785 { 786 if (error == FCERR_TOOMANY) 787 semsg(_(e_toomanyarg), ufunc->uf_name); 788 else 789 semsg(_(e_toofewarg), ufunc->uf_name); 790 return FAIL; 791 } 792 793 // The function has been compiled, can call it quickly. For a function 794 // that was defined later: we can call it directly next time. 795 // TODO: what if the function was deleted and then defined again? 796 if (iptr != NULL) 797 { 798 delete_instr(iptr); 799 iptr->isn_type = ISN_DCALL; 800 iptr->isn_arg.dfunc.cdf_idx = ufunc->uf_dfunc_idx; 801 iptr->isn_arg.dfunc.cdf_argcount = argcount; 802 } 803 return call_dfunc(ufunc->uf_dfunc_idx, pt, argcount, ectx); 804 } 805 806 if (call_prepare(argcount, argvars, ectx) == FAIL) 807 return FAIL; 808 CLEAR_FIELD(funcexe); 809 funcexe.evaluate = TRUE; 810 811 // Call the user function. Result goes in last position on the stack. 812 // TODO: add selfdict if there is one 813 error = call_user_func_check(ufunc, argcount, argvars, 814 STACK_TV_BOT(-1), &funcexe, NULL); 815 816 // Clear the arguments. 817 for (idx = 0; idx < argcount; ++idx) 818 clear_tv(&argvars[idx]); 819 820 if (error != FCERR_NONE) 821 { 822 user_func_error(error, ufunc->uf_name); 823 return FAIL; 824 } 825 if (did_emsg > did_emsg_before) 826 // Error other than from calling the function itself. 827 return FAIL; 828 return OK; 829 } 830 831 /* 832 * If command modifiers were applied restore them. 833 */ 834 static void 835 may_restore_cmdmod(funclocal_T *funclocal) 836 { 837 if (funclocal->floc_restore_cmdmod) 838 { 839 cmdmod.cmod_filter_regmatch.regprog = NULL; 840 undo_cmdmod(&cmdmod); 841 cmdmod = funclocal->floc_save_cmdmod; 842 funclocal->floc_restore_cmdmod = FALSE; 843 } 844 } 845 846 /* 847 * Return TRUE if an error was given or CTRL-C was pressed. 848 */ 849 static int 850 vim9_aborting(int prev_called_emsg) 851 { 852 return called_emsg > prev_called_emsg || got_int || did_throw; 853 } 854 855 /* 856 * Execute a function by "name". 857 * This can be a builtin function or a user function. 858 * "iptr" can be used to replace the instruction with a more efficient one. 859 * Returns FAIL if not found without an error message. 860 */ 861 static int 862 call_by_name( 863 char_u *name, 864 int argcount, 865 ectx_T *ectx, 866 isn_T *iptr) 867 { 868 ufunc_T *ufunc; 869 870 if (builtin_function(name, -1)) 871 { 872 int func_idx = find_internal_func(name); 873 874 if (func_idx < 0) 875 return FAIL; 876 if (check_internal_func(func_idx, argcount) < 0) 877 return FAIL; 878 return call_bfunc(func_idx, argcount, ectx); 879 } 880 881 ufunc = find_func(name, FALSE, NULL); 882 883 if (ufunc == NULL) 884 { 885 int called_emsg_before = called_emsg; 886 887 if (script_autoload(name, TRUE)) 888 // loaded a package, search for the function again 889 ufunc = find_func(name, FALSE, NULL); 890 if (vim9_aborting(called_emsg_before)) 891 return FAIL; // bail out if loading the script caused an error 892 } 893 894 if (ufunc != NULL) 895 { 896 if (ufunc->uf_arg_types != NULL || ufunc->uf_va_type != NULL) 897 { 898 int i; 899 typval_T *argv = STACK_TV_BOT(0) - argcount; 900 901 // The function can change at runtime, check that the argument 902 // types are correct. 903 for (i = 0; i < argcount; ++i) 904 { 905 type_T *type = NULL; 906 907 if (i < ufunc->uf_args.ga_len && ufunc->uf_arg_types != NULL) 908 type = ufunc->uf_arg_types[i]; 909 else if (ufunc->uf_va_type != NULL) 910 type = ufunc->uf_va_type->tt_member; 911 if (type != NULL && check_typval_arg_type(type, 912 &argv[i], NULL, i + 1) == FAIL) 913 return FAIL; 914 } 915 } 916 917 return call_ufunc(ufunc, NULL, argcount, ectx, iptr); 918 } 919 920 return FAIL; 921 } 922 923 static int 924 call_partial( 925 typval_T *tv, 926 int argcount_arg, 927 ectx_T *ectx) 928 { 929 int argcount = argcount_arg; 930 char_u *name = NULL; 931 int called_emsg_before = called_emsg; 932 int res = FAIL; 933 934 if (tv->v_type == VAR_PARTIAL) 935 { 936 partial_T *pt = tv->vval.v_partial; 937 int i; 938 939 if (pt->pt_argc > 0) 940 { 941 // Make space for arguments from the partial, shift the "argcount" 942 // arguments up. 943 if (GA_GROW_FAILS(&ectx->ec_stack, pt->pt_argc)) 944 return FAIL; 945 for (i = 1; i <= argcount; ++i) 946 *STACK_TV_BOT(-i + pt->pt_argc) = *STACK_TV_BOT(-i); 947 ectx->ec_stack.ga_len += pt->pt_argc; 948 argcount += pt->pt_argc; 949 950 // copy the arguments from the partial onto the stack 951 for (i = 0; i < pt->pt_argc; ++i) 952 copy_tv(&pt->pt_argv[i], STACK_TV_BOT(-argcount + i)); 953 } 954 955 if (pt->pt_func != NULL) 956 return call_ufunc(pt->pt_func, pt, argcount, ectx, NULL); 957 958 name = pt->pt_name; 959 } 960 else if (tv->v_type == VAR_FUNC) 961 name = tv->vval.v_string; 962 if (name != NULL) 963 { 964 char_u fname_buf[FLEN_FIXED + 1]; 965 char_u *tofree = NULL; 966 int error = FCERR_NONE; 967 char_u *fname; 968 969 // May need to translate <SNR>123_ to K_SNR. 970 fname = fname_trans_sid(name, fname_buf, &tofree, &error); 971 if (error != FCERR_NONE) 972 res = FAIL; 973 else 974 res = call_by_name(fname, argcount, ectx, NULL); 975 vim_free(tofree); 976 } 977 978 if (res == FAIL) 979 { 980 if (called_emsg == called_emsg_before) 981 semsg(_(e_unknownfunc), 982 name == NULL ? (char_u *)"[unknown]" : name); 983 return FAIL; 984 } 985 return OK; 986 } 987 988 /* 989 * Check if "lock" is VAR_LOCKED or VAR_FIXED. If so give an error and return 990 * TRUE. 991 */ 992 static int 993 error_if_locked(int lock, char *error) 994 { 995 if (lock & (VAR_LOCKED | VAR_FIXED)) 996 { 997 emsg(_(error)); 998 return TRUE; 999 } 1000 return FALSE; 1001 } 1002 1003 /* 1004 * Give an error if "tv" is not a number and return FAIL. 1005 */ 1006 static int 1007 check_for_number(typval_T *tv) 1008 { 1009 if (tv->v_type != VAR_NUMBER) 1010 { 1011 semsg(_(e_expected_str_but_got_str), 1012 vartype_name(VAR_NUMBER), vartype_name(tv->v_type)); 1013 return FAIL; 1014 } 1015 return OK; 1016 } 1017 1018 /* 1019 * Store "tv" in variable "name". 1020 * This is for s: and g: variables. 1021 */ 1022 static void 1023 store_var(char_u *name, typval_T *tv) 1024 { 1025 funccal_entry_T entry; 1026 int flags = ASSIGN_DECL; 1027 1028 if (tv->v_lock) 1029 flags |= ASSIGN_CONST; 1030 save_funccal(&entry); 1031 set_var_const(name, NULL, tv, FALSE, flags, 0); 1032 restore_funccal(); 1033 } 1034 1035 /* 1036 * Convert "tv" to a string. 1037 * Return FAIL if not allowed. 1038 */ 1039 static int 1040 do_2string(typval_T *tv, int is_2string_any, int tolerant) 1041 { 1042 if (tv->v_type != VAR_STRING) 1043 { 1044 char_u *str; 1045 1046 if (is_2string_any) 1047 { 1048 switch (tv->v_type) 1049 { 1050 case VAR_SPECIAL: 1051 case VAR_BOOL: 1052 case VAR_NUMBER: 1053 case VAR_FLOAT: 1054 case VAR_BLOB: break; 1055 1056 case VAR_LIST: 1057 if (tolerant) 1058 { 1059 char_u *s, *e, *p; 1060 garray_T ga; 1061 1062 ga_init2(&ga, sizeof(char_u *), 1); 1063 1064 // Convert to NL separated items, then 1065 // escape the items and replace the NL with 1066 // a space. 1067 str = typval2string(tv, TRUE); 1068 if (str == NULL) 1069 return FAIL; 1070 s = str; 1071 while ((e = vim_strchr(s, '\n')) != NULL) 1072 { 1073 *e = NUL; 1074 p = vim_strsave_fnameescape(s, FALSE); 1075 if (p != NULL) 1076 { 1077 ga_concat(&ga, p); 1078 ga_concat(&ga, (char_u *)" "); 1079 vim_free(p); 1080 } 1081 s = e + 1; 1082 } 1083 vim_free(str); 1084 clear_tv(tv); 1085 tv->v_type = VAR_STRING; 1086 tv->vval.v_string = ga.ga_data; 1087 return OK; 1088 } 1089 // FALLTHROUGH 1090 default: to_string_error(tv->v_type); 1091 return FAIL; 1092 } 1093 } 1094 str = typval_tostring(tv, TRUE); 1095 clear_tv(tv); 1096 tv->v_type = VAR_STRING; 1097 tv->vval.v_string = str; 1098 } 1099 return OK; 1100 } 1101 1102 /* 1103 * When the value of "sv" is a null list of dict, allocate it. 1104 */ 1105 static void 1106 allocate_if_null(typval_T *tv) 1107 { 1108 switch (tv->v_type) 1109 { 1110 case VAR_LIST: 1111 if (tv->vval.v_list == NULL) 1112 (void)rettv_list_alloc(tv); 1113 break; 1114 case VAR_DICT: 1115 if (tv->vval.v_dict == NULL) 1116 (void)rettv_dict_alloc(tv); 1117 break; 1118 case VAR_BLOB: 1119 if (tv->vval.v_blob == NULL) 1120 (void)rettv_blob_alloc(tv); 1121 break; 1122 default: 1123 break; 1124 } 1125 } 1126 1127 /* 1128 * Return the character "str[index]" where "index" is the character index, 1129 * including composing characters. 1130 * If "index" is out of range NULL is returned. 1131 */ 1132 char_u * 1133 char_from_string(char_u *str, varnumber_T index) 1134 { 1135 size_t nbyte = 0; 1136 varnumber_T nchar = index; 1137 size_t slen; 1138 1139 if (str == NULL) 1140 return NULL; 1141 slen = STRLEN(str); 1142 1143 // Do the same as for a list: a negative index counts from the end. 1144 // Optimization to check the first byte to be below 0x80 (and no composing 1145 // character follows) makes this a lot faster. 1146 if (index < 0) 1147 { 1148 int clen = 0; 1149 1150 for (nbyte = 0; nbyte < slen; ++clen) 1151 { 1152 if (str[nbyte] < 0x80 && str[nbyte + 1] < 0x80) 1153 ++nbyte; 1154 else if (enc_utf8) 1155 nbyte += utfc_ptr2len(str + nbyte); 1156 else 1157 nbyte += mb_ptr2len(str + nbyte); 1158 } 1159 nchar = clen + index; 1160 if (nchar < 0) 1161 // unlike list: index out of range results in empty string 1162 return NULL; 1163 } 1164 1165 for (nbyte = 0; nchar > 0 && nbyte < slen; --nchar) 1166 { 1167 if (str[nbyte] < 0x80 && str[nbyte + 1] < 0x80) 1168 ++nbyte; 1169 else if (enc_utf8) 1170 nbyte += utfc_ptr2len(str + nbyte); 1171 else 1172 nbyte += mb_ptr2len(str + nbyte); 1173 } 1174 if (nbyte >= slen) 1175 return NULL; 1176 return vim_strnsave(str + nbyte, mb_ptr2len(str + nbyte)); 1177 } 1178 1179 /* 1180 * Get the byte index for character index "idx" in string "str" with length 1181 * "str_len". Composing characters are included. 1182 * If going over the end return "str_len". 1183 * If "idx" is negative count from the end, -1 is the last character. 1184 * When going over the start return -1. 1185 */ 1186 static long 1187 char_idx2byte(char_u *str, size_t str_len, varnumber_T idx) 1188 { 1189 varnumber_T nchar = idx; 1190 size_t nbyte = 0; 1191 1192 if (nchar >= 0) 1193 { 1194 while (nchar > 0 && nbyte < str_len) 1195 { 1196 nbyte += mb_ptr2len(str + nbyte); 1197 --nchar; 1198 } 1199 } 1200 else 1201 { 1202 nbyte = str_len; 1203 while (nchar < 0 && nbyte > 0) 1204 { 1205 --nbyte; 1206 nbyte -= mb_head_off(str, str + nbyte); 1207 ++nchar; 1208 } 1209 if (nchar < 0) 1210 return -1; 1211 } 1212 return (long)nbyte; 1213 } 1214 1215 /* 1216 * Return the slice "str[first : last]" using character indexes. Composing 1217 * characters are included. 1218 * "exclusive" is TRUE for slice(). 1219 * Return NULL when the result is empty. 1220 */ 1221 char_u * 1222 string_slice(char_u *str, varnumber_T first, varnumber_T last, int exclusive) 1223 { 1224 long start_byte, end_byte; 1225 size_t slen; 1226 1227 if (str == NULL) 1228 return NULL; 1229 slen = STRLEN(str); 1230 start_byte = char_idx2byte(str, slen, first); 1231 if (start_byte < 0) 1232 start_byte = 0; // first index very negative: use zero 1233 if ((last == -1 && !exclusive) || last == VARNUM_MAX) 1234 end_byte = (long)slen; 1235 else 1236 { 1237 end_byte = char_idx2byte(str, slen, last); 1238 if (!exclusive && end_byte >= 0 && end_byte < (long)slen) 1239 // end index is inclusive 1240 end_byte += mb_ptr2len(str + end_byte); 1241 } 1242 1243 if (start_byte >= (long)slen || end_byte <= start_byte) 1244 return NULL; 1245 return vim_strnsave(str + start_byte, end_byte - start_byte); 1246 } 1247 1248 /* 1249 * Get a script variable for ISN_STORESCRIPT and ISN_LOADSCRIPT. 1250 * When "dfunc_idx" is negative don't give an error. 1251 * Returns NULL for an error. 1252 */ 1253 static svar_T * 1254 get_script_svar(scriptref_T *sref, int dfunc_idx) 1255 { 1256 scriptitem_T *si = SCRIPT_ITEM(sref->sref_sid); 1257 dfunc_T *dfunc = dfunc_idx < 0 ? NULL 1258 : ((dfunc_T *)def_functions.ga_data) + dfunc_idx; 1259 svar_T *sv; 1260 1261 if (sref->sref_seq != si->sn_script_seq) 1262 { 1263 // The script was reloaded after the function was compiled, the 1264 // script_idx may not be valid. 1265 if (dfunc != NULL) 1266 semsg(_(e_script_variable_invalid_after_reload_in_function_str), 1267 printable_func_name(dfunc->df_ufunc)); 1268 return NULL; 1269 } 1270 sv = ((svar_T *)si->sn_var_vals.ga_data) + sref->sref_idx; 1271 if (!equal_type(sv->sv_type, sref->sref_type, 0)) 1272 { 1273 if (dfunc != NULL) 1274 emsg(_(e_script_variable_type_changed)); 1275 return NULL; 1276 } 1277 return sv; 1278 } 1279 1280 /* 1281 * Function passed to do_cmdline() for splitting a script joined by NL 1282 * characters. 1283 */ 1284 static char_u * 1285 get_split_sourceline( 1286 int c UNUSED, 1287 void *cookie, 1288 int indent UNUSED, 1289 getline_opt_T options UNUSED) 1290 { 1291 source_cookie_T *sp = (source_cookie_T *)cookie; 1292 char_u *p; 1293 char_u *line; 1294 1295 if (*sp->nextline == NUL) 1296 return NULL; 1297 p = vim_strchr(sp->nextline, '\n'); 1298 if (p == NULL) 1299 { 1300 line = vim_strsave(sp->nextline); 1301 sp->nextline += STRLEN(sp->nextline); 1302 } 1303 else 1304 { 1305 line = vim_strnsave(sp->nextline, p - sp->nextline); 1306 sp->nextline = p + 1; 1307 } 1308 return line; 1309 } 1310 1311 /* 1312 * Execute a function by "name". 1313 * This can be a builtin function, user function or a funcref. 1314 * "iptr" can be used to replace the instruction with a more efficient one. 1315 */ 1316 static int 1317 call_eval_func( 1318 char_u *name, 1319 int argcount, 1320 ectx_T *ectx, 1321 isn_T *iptr) 1322 { 1323 int called_emsg_before = called_emsg; 1324 int res; 1325 1326 res = call_by_name(name, argcount, ectx, iptr); 1327 if (res == FAIL && called_emsg == called_emsg_before) 1328 { 1329 dictitem_T *v; 1330 1331 v = find_var(name, NULL, FALSE); 1332 if (v == NULL) 1333 { 1334 semsg(_(e_unknownfunc), name); 1335 return FAIL; 1336 } 1337 if (v->di_tv.v_type != VAR_PARTIAL && v->di_tv.v_type != VAR_FUNC) 1338 { 1339 semsg(_(e_unknownfunc), name); 1340 return FAIL; 1341 } 1342 return call_partial(&v->di_tv, argcount, ectx); 1343 } 1344 return res; 1345 } 1346 1347 /* 1348 * When a function reference is used, fill a partial with the information 1349 * needed, especially when it is used as a closure. 1350 */ 1351 int 1352 fill_partial_and_closure(partial_T *pt, ufunc_T *ufunc, ectx_T *ectx) 1353 { 1354 pt->pt_func = ufunc; 1355 pt->pt_refcount = 1; 1356 1357 if (ufunc->uf_flags & FC_CLOSURE) 1358 { 1359 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data) 1360 + ectx->ec_dfunc_idx; 1361 1362 // The closure may need to find arguments and local variables in the 1363 // current stack. 1364 pt->pt_outer.out_stack = &ectx->ec_stack; 1365 pt->pt_outer.out_frame_idx = ectx->ec_frame_idx; 1366 if (ectx->ec_outer_ref != NULL) 1367 { 1368 // The current context already has a context, link to that one. 1369 pt->pt_outer.out_up = ectx->ec_outer_ref->or_outer; 1370 if (ectx->ec_outer_ref->or_partial != NULL) 1371 { 1372 pt->pt_outer.out_up_partial = ectx->ec_outer_ref->or_partial; 1373 ++pt->pt_outer.out_up_partial->pt_refcount; 1374 } 1375 } 1376 1377 // If this function returns and the closure is still being used, we 1378 // need to make a copy of the context (arguments and local variables). 1379 // Store a reference to the partial so we can handle that. 1380 if (GA_GROW_FAILS(&ectx->ec_funcrefs, 1)) 1381 { 1382 vim_free(pt); 1383 return FAIL; 1384 } 1385 // Extra variable keeps the count of closures created in the current 1386 // function call. 1387 ++(((typval_T *)ectx->ec_stack.ga_data) + ectx->ec_frame_idx 1388 + STACK_FRAME_SIZE + dfunc->df_varcount)->vval.v_number; 1389 1390 ((partial_T **)ectx->ec_funcrefs.ga_data) 1391 [ectx->ec_funcrefs.ga_len] = pt; 1392 ++pt->pt_refcount; 1393 ++ectx->ec_funcrefs.ga_len; 1394 } 1395 ++ufunc->uf_refcount; 1396 return OK; 1397 } 1398 1399 // used for v_instr of typval of VAR_INSTR 1400 struct instr_S { 1401 ectx_T *instr_ectx; 1402 isn_T *instr_instr; 1403 }; 1404 1405 // used for substitute_instr 1406 typedef struct subs_expr_S { 1407 ectx_T *subs_ectx; 1408 isn_T *subs_instr; 1409 int subs_status; 1410 } subs_expr_T; 1411 1412 // Get pointer to item in the stack. 1413 #define STACK_TV(idx) (((typval_T *)ectx->ec_stack.ga_data) + idx) 1414 1415 // Get pointer to item at the bottom of the stack, -1 is the bottom. 1416 #undef STACK_TV_BOT 1417 #define STACK_TV_BOT(idx) (((typval_T *)ectx->ec_stack.ga_data) + ectx->ec_stack.ga_len + idx) 1418 1419 // Get pointer to a local variable on the stack. Negative for arguments. 1420 #define STACK_TV_VAR(idx) (((typval_T *)ectx->ec_stack.ga_data) + ectx->ec_frame_idx + STACK_FRAME_SIZE + idx) 1421 1422 // Set when calling do_debug(). 1423 static ectx_T *debug_context = NULL; 1424 static int debug_var_count; 1425 1426 /* 1427 * When debugging lookup "name" and return the typeval. 1428 * When not found return NULL. 1429 */ 1430 typval_T * 1431 lookup_debug_var(char_u *name) 1432 { 1433 int idx; 1434 dfunc_T *dfunc; 1435 ufunc_T *ufunc; 1436 ectx_T *ectx = debug_context; 1437 int varargs_off; 1438 1439 if (ectx == NULL) 1440 return NULL; 1441 dfunc = ((dfunc_T *)def_functions.ga_data) + ectx->ec_dfunc_idx; 1442 1443 // Go through the local variable names, from last to first. 1444 for (idx = debug_var_count - 1; idx >= 0; --idx) 1445 { 1446 if (STRCMP(((char_u **)dfunc->df_var_names.ga_data)[idx], name) == 0) 1447 return STACK_TV_VAR(idx); 1448 } 1449 1450 // Go through argument names. 1451 ufunc = dfunc->df_ufunc; 1452 varargs_off = ufunc->uf_va_name == NULL ? 0 : 1; 1453 for (idx = 0; idx < ufunc->uf_args.ga_len; ++idx) 1454 if (STRCMP(((char_u **)(ufunc->uf_args.ga_data))[idx], name) == 0) 1455 return STACK_TV(ectx->ec_frame_idx - ufunc->uf_args.ga_len 1456 - varargs_off + idx); 1457 if (ufunc->uf_va_name != NULL && STRCMP(ufunc->uf_va_name, name) == 0) 1458 return STACK_TV(ectx->ec_frame_idx - 1); 1459 1460 return NULL; 1461 } 1462 1463 static void 1464 handle_debug(isn_T *iptr, ectx_T *ectx) 1465 { 1466 char_u *line; 1467 ufunc_T *ufunc = (((dfunc_T *)def_functions.ga_data) 1468 + ectx->ec_dfunc_idx)->df_ufunc; 1469 isn_T *ni; 1470 int end_lnum = iptr->isn_lnum; 1471 garray_T ga; 1472 int lnum; 1473 1474 if (ex_nesting_level > debug_break_level) 1475 { 1476 linenr_T breakpoint; 1477 1478 if (!ufunc->uf_has_breakpoint) 1479 return; 1480 1481 // check for the next breakpoint if needed 1482 breakpoint = dbg_find_breakpoint(FALSE, ufunc->uf_name, 1483 iptr->isn_arg.debug.dbg_break_lnum); 1484 if (breakpoint <= 0 || breakpoint > iptr->isn_lnum) 1485 return; 1486 } 1487 1488 SOURCING_LNUM = iptr->isn_lnum; 1489 debug_context = ectx; 1490 debug_var_count = iptr->isn_arg.debug.dbg_var_names_len; 1491 1492 for (ni = iptr + 1; ni->isn_type != ISN_FINISH; ++ni) 1493 if (ni->isn_type == ISN_DEBUG 1494 || ni->isn_type == ISN_RETURN 1495 || ni->isn_type == ISN_RETURN_VOID) 1496 { 1497 end_lnum = ni->isn_lnum; 1498 break; 1499 } 1500 1501 if (end_lnum > iptr->isn_lnum) 1502 { 1503 ga_init2(&ga, sizeof(char_u *), 10); 1504 for (lnum = iptr->isn_lnum; lnum < end_lnum; ++lnum) 1505 { 1506 char_u *p = ((char_u **)ufunc->uf_lines.ga_data)[lnum - 1]; 1507 1508 if (p == NULL) 1509 continue; // left over from continuation line 1510 p = skipwhite(p); 1511 if (*p == '#') 1512 break; 1513 if (GA_GROW_OK(&ga, 1)) 1514 ((char_u **)(ga.ga_data))[ga.ga_len++] = p; 1515 if (STRNCMP(p, "def ", 4) == 0) 1516 break; 1517 } 1518 line = ga_concat_strings(&ga, " "); 1519 vim_free(ga.ga_data); 1520 } 1521 else 1522 line = ((char_u **)ufunc->uf_lines.ga_data)[iptr->isn_lnum - 1]; 1523 1524 do_debug(line == NULL ? (char_u *)"[empty]" : line); 1525 debug_context = NULL; 1526 1527 if (end_lnum > iptr->isn_lnum) 1528 vim_free(line); 1529 } 1530 1531 /* 1532 * Execute instructions in execution context "ectx". 1533 * Return OK or FAIL; 1534 */ 1535 static int 1536 exec_instructions(ectx_T *ectx) 1537 { 1538 int ret = FAIL; 1539 int save_trylevel_at_start = ectx->ec_trylevel_at_start; 1540 1541 // Start execution at the first instruction. 1542 ectx->ec_iidx = 0; 1543 1544 // Only catch exceptions in this instruction list. 1545 ectx->ec_trylevel_at_start = trylevel; 1546 1547 for (;;) 1548 { 1549 static int breakcheck_count = 0; // using "static" makes it faster 1550 isn_T *iptr; 1551 typval_T *tv; 1552 1553 if (unlikely(++breakcheck_count >= 100)) 1554 { 1555 line_breakcheck(); 1556 breakcheck_count = 0; 1557 } 1558 if (unlikely(got_int)) 1559 { 1560 // Turn CTRL-C into an exception. 1561 got_int = FALSE; 1562 if (throw_exception("Vim:Interrupt", ET_INTERRUPT, NULL) == FAIL) 1563 goto theend; 1564 did_throw = TRUE; 1565 } 1566 1567 if (unlikely(did_emsg && msg_list != NULL && *msg_list != NULL)) 1568 { 1569 // Turn an error message into an exception. 1570 did_emsg = FALSE; 1571 if (throw_exception(*msg_list, ET_ERROR, NULL) == FAIL) 1572 goto theend; 1573 did_throw = TRUE; 1574 *msg_list = NULL; 1575 } 1576 1577 if (unlikely(did_throw)) 1578 { 1579 garray_T *trystack = &ectx->ec_trystack; 1580 trycmd_T *trycmd = NULL; 1581 int index = trystack->ga_len; 1582 1583 // An exception jumps to the first catch, finally, or returns from 1584 // the current function. 1585 while (index > 0) 1586 { 1587 trycmd = ((trycmd_T *)trystack->ga_data) + index - 1; 1588 if (!trycmd->tcd_in_catch || trycmd->tcd_finally_idx != 0) 1589 break; 1590 // In the catch and finally block of this try we have to go up 1591 // one level. 1592 --index; 1593 trycmd = NULL; 1594 } 1595 if (trycmd != NULL && trycmd->tcd_frame_idx == ectx->ec_frame_idx) 1596 { 1597 if (trycmd->tcd_in_catch) 1598 { 1599 // exception inside ":catch", jump to ":finally" once 1600 ectx->ec_iidx = trycmd->tcd_finally_idx; 1601 trycmd->tcd_finally_idx = 0; 1602 } 1603 else 1604 // jump to first ":catch" 1605 ectx->ec_iidx = trycmd->tcd_catch_idx; 1606 trycmd->tcd_in_catch = TRUE; 1607 did_throw = FALSE; // don't come back here until :endtry 1608 trycmd->tcd_did_throw = TRUE; 1609 } 1610 else 1611 { 1612 // Not inside try or need to return from current functions. 1613 // Push a dummy return value. 1614 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 1615 goto theend; 1616 tv = STACK_TV_BOT(0); 1617 tv->v_type = VAR_NUMBER; 1618 tv->vval.v_number = 0; 1619 ++ectx->ec_stack.ga_len; 1620 if (ectx->ec_frame_idx == ectx->ec_initial_frame_idx) 1621 { 1622 // At the toplevel we are done. 1623 need_rethrow = TRUE; 1624 if (handle_closure_in_use(ectx, FALSE) == FAIL) 1625 goto theend; 1626 goto done; 1627 } 1628 1629 if (func_return(ectx) == FAIL) 1630 goto theend; 1631 } 1632 continue; 1633 } 1634 1635 iptr = &ectx->ec_instr[ectx->ec_iidx++]; 1636 switch (iptr->isn_type) 1637 { 1638 // execute Ex command line 1639 case ISN_EXEC: 1640 { 1641 source_cookie_T cookie; 1642 1643 SOURCING_LNUM = iptr->isn_lnum; 1644 // Pass getsourceline to get an error for a missing ":end" 1645 // command. 1646 CLEAR_FIELD(cookie); 1647 cookie.sourcing_lnum = iptr->isn_lnum - 1; 1648 if (do_cmdline(iptr->isn_arg.string, 1649 getsourceline, &cookie, 1650 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_KEYTYPED) 1651 == FAIL 1652 || did_emsg) 1653 goto on_error; 1654 } 1655 break; 1656 1657 // execute Ex command line split at NL characters. 1658 case ISN_EXEC_SPLIT: 1659 { 1660 source_cookie_T cookie; 1661 char_u *line; 1662 1663 SOURCING_LNUM = iptr->isn_lnum; 1664 CLEAR_FIELD(cookie); 1665 cookie.sourcing_lnum = iptr->isn_lnum - 1; 1666 cookie.nextline = iptr->isn_arg.string; 1667 line = get_split_sourceline(0, &cookie, 0, 0); 1668 if (do_cmdline(line, 1669 get_split_sourceline, &cookie, 1670 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_KEYTYPED) 1671 == FAIL 1672 || did_emsg) 1673 { 1674 vim_free(line); 1675 goto on_error; 1676 } 1677 vim_free(line); 1678 } 1679 break; 1680 1681 // Evaluate an expression with legacy syntax, push it onto the 1682 // stack. 1683 case ISN_LEGACY_EVAL: 1684 { 1685 char_u *arg = iptr->isn_arg.string; 1686 int res; 1687 int save_flags = cmdmod.cmod_flags; 1688 1689 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 1690 goto theend; 1691 tv = STACK_TV_BOT(0); 1692 init_tv(tv); 1693 cmdmod.cmod_flags |= CMOD_LEGACY; 1694 res = eval0(arg, tv, NULL, &EVALARG_EVALUATE); 1695 cmdmod.cmod_flags = save_flags; 1696 if (res == FAIL) 1697 goto on_error; 1698 ++ectx->ec_stack.ga_len; 1699 } 1700 break; 1701 1702 // push typeval VAR_INSTR with instructions to be executed 1703 case ISN_INSTR: 1704 { 1705 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 1706 goto theend; 1707 tv = STACK_TV_BOT(0); 1708 tv->vval.v_instr = ALLOC_ONE(instr_T); 1709 if (tv->vval.v_instr == NULL) 1710 goto on_error; 1711 ++ectx->ec_stack.ga_len; 1712 1713 tv->v_type = VAR_INSTR; 1714 tv->vval.v_instr->instr_ectx = ectx; 1715 tv->vval.v_instr->instr_instr = iptr->isn_arg.instr; 1716 } 1717 break; 1718 1719 // execute :substitute with an expression 1720 case ISN_SUBSTITUTE: 1721 { 1722 subs_T *subs = &iptr->isn_arg.subs; 1723 source_cookie_T cookie; 1724 struct subs_expr_S *save_instr = substitute_instr; 1725 struct subs_expr_S subs_instr; 1726 int res; 1727 1728 subs_instr.subs_ectx = ectx; 1729 subs_instr.subs_instr = subs->subs_instr; 1730 subs_instr.subs_status = OK; 1731 substitute_instr = &subs_instr; 1732 1733 SOURCING_LNUM = iptr->isn_lnum; 1734 // This is very much like ISN_EXEC 1735 CLEAR_FIELD(cookie); 1736 cookie.sourcing_lnum = iptr->isn_lnum - 1; 1737 res = do_cmdline(subs->subs_cmd, 1738 getsourceline, &cookie, 1739 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_KEYTYPED); 1740 substitute_instr = save_instr; 1741 1742 if (res == FAIL || did_emsg 1743 || subs_instr.subs_status == FAIL) 1744 goto on_error; 1745 } 1746 break; 1747 1748 case ISN_FINISH: 1749 goto done; 1750 1751 case ISN_REDIRSTART: 1752 // create a dummy entry for var_redir_str() 1753 if (alloc_redir_lval() == FAIL) 1754 goto on_error; 1755 1756 // The output is stored in growarray "redir_ga" until 1757 // redirection ends. 1758 init_redir_ga(); 1759 redir_vname = 1; 1760 break; 1761 1762 case ISN_REDIREND: 1763 { 1764 char_u *res = get_clear_redir_ga(); 1765 1766 // End redirection, put redirected text on the stack. 1767 clear_redir_lval(); 1768 redir_vname = 0; 1769 1770 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 1771 { 1772 vim_free(res); 1773 goto theend; 1774 } 1775 tv = STACK_TV_BOT(0); 1776 tv->v_type = VAR_STRING; 1777 tv->vval.v_string = res; 1778 ++ectx->ec_stack.ga_len; 1779 } 1780 break; 1781 1782 case ISN_CEXPR_AUCMD: 1783 #ifdef FEAT_QUICKFIX 1784 if (trigger_cexpr_autocmd(iptr->isn_arg.number) == FAIL) 1785 goto on_error; 1786 #endif 1787 break; 1788 1789 case ISN_CEXPR_CORE: 1790 #ifdef FEAT_QUICKFIX 1791 { 1792 exarg_T ea; 1793 int res; 1794 1795 CLEAR_FIELD(ea); 1796 ea.cmdidx = iptr->isn_arg.cexpr.cexpr_ref->cer_cmdidx; 1797 ea.forceit = iptr->isn_arg.cexpr.cexpr_ref->cer_forceit; 1798 ea.cmdlinep = &iptr->isn_arg.cexpr.cexpr_ref->cer_cmdline; 1799 --ectx->ec_stack.ga_len; 1800 tv = STACK_TV_BOT(0); 1801 res = cexpr_core(&ea, tv); 1802 clear_tv(tv); 1803 if (res == FAIL) 1804 goto on_error; 1805 } 1806 #endif 1807 break; 1808 1809 // execute Ex command from pieces on the stack 1810 case ISN_EXECCONCAT: 1811 { 1812 int count = iptr->isn_arg.number; 1813 size_t len = 0; 1814 int pass; 1815 int i; 1816 char_u *cmd = NULL; 1817 char_u *str; 1818 1819 for (pass = 1; pass <= 2; ++pass) 1820 { 1821 for (i = 0; i < count; ++i) 1822 { 1823 tv = STACK_TV_BOT(i - count); 1824 str = tv->vval.v_string; 1825 if (str != NULL && *str != NUL) 1826 { 1827 if (pass == 2) 1828 STRCPY(cmd + len, str); 1829 len += STRLEN(str); 1830 } 1831 if (pass == 2) 1832 clear_tv(tv); 1833 } 1834 if (pass == 1) 1835 { 1836 cmd = alloc(len + 1); 1837 if (unlikely(cmd == NULL)) 1838 goto theend; 1839 len = 0; 1840 } 1841 } 1842 1843 SOURCING_LNUM = iptr->isn_lnum; 1844 do_cmdline_cmd(cmd); 1845 vim_free(cmd); 1846 } 1847 break; 1848 1849 // execute :echo {string} ... 1850 case ISN_ECHO: 1851 { 1852 int count = iptr->isn_arg.echo.echo_count; 1853 int atstart = TRUE; 1854 int needclr = TRUE; 1855 int idx; 1856 1857 for (idx = 0; idx < count; ++idx) 1858 { 1859 tv = STACK_TV_BOT(idx - count); 1860 echo_one(tv, iptr->isn_arg.echo.echo_with_white, 1861 &atstart, &needclr); 1862 clear_tv(tv); 1863 } 1864 if (needclr) 1865 msg_clr_eos(); 1866 ectx->ec_stack.ga_len -= count; 1867 } 1868 break; 1869 1870 // :execute {string} ... 1871 // :echomsg {string} ... 1872 // :echoconsole {string} ... 1873 // :echoerr {string} ... 1874 case ISN_EXECUTE: 1875 case ISN_ECHOMSG: 1876 case ISN_ECHOCONSOLE: 1877 case ISN_ECHOERR: 1878 { 1879 int count = iptr->isn_arg.number; 1880 garray_T ga; 1881 char_u buf[NUMBUFLEN]; 1882 char_u *p; 1883 int len; 1884 int failed = FALSE; 1885 int idx; 1886 1887 ga_init2(&ga, 1, 80); 1888 for (idx = 0; idx < count; ++idx) 1889 { 1890 tv = STACK_TV_BOT(idx - count); 1891 if (iptr->isn_type == ISN_EXECUTE) 1892 { 1893 if (tv->v_type == VAR_CHANNEL 1894 || tv->v_type == VAR_JOB) 1895 { 1896 SOURCING_LNUM = iptr->isn_lnum; 1897 semsg(_(e_using_invalid_value_as_string_str), 1898 vartype_name(tv->v_type)); 1899 break; 1900 } 1901 else 1902 p = tv_get_string_buf(tv, buf); 1903 } 1904 else 1905 p = tv_stringify(tv, buf); 1906 1907 len = (int)STRLEN(p); 1908 if (GA_GROW_FAILS(&ga, len + 2)) 1909 failed = TRUE; 1910 else 1911 { 1912 if (ga.ga_len > 0) 1913 ((char_u *)(ga.ga_data))[ga.ga_len++] = ' '; 1914 STRCPY((char_u *)(ga.ga_data) + ga.ga_len, p); 1915 ga.ga_len += len; 1916 } 1917 clear_tv(tv); 1918 } 1919 ectx->ec_stack.ga_len -= count; 1920 if (failed) 1921 { 1922 ga_clear(&ga); 1923 goto on_error; 1924 } 1925 1926 if (ga.ga_data != NULL) 1927 { 1928 if (iptr->isn_type == ISN_EXECUTE) 1929 { 1930 SOURCING_LNUM = iptr->isn_lnum; 1931 do_cmdline_cmd((char_u *)ga.ga_data); 1932 if (did_emsg) 1933 { 1934 ga_clear(&ga); 1935 goto on_error; 1936 } 1937 } 1938 else 1939 { 1940 msg_sb_eol(); 1941 if (iptr->isn_type == ISN_ECHOMSG) 1942 { 1943 msg_attr(ga.ga_data, echo_attr); 1944 out_flush(); 1945 } 1946 else if (iptr->isn_type == ISN_ECHOCONSOLE) 1947 { 1948 ui_write(ga.ga_data, (int)STRLEN(ga.ga_data), 1949 TRUE); 1950 ui_write((char_u *)"\r\n", 2, TRUE); 1951 } 1952 else 1953 { 1954 SOURCING_LNUM = iptr->isn_lnum; 1955 emsg(ga.ga_data); 1956 } 1957 } 1958 } 1959 ga_clear(&ga); 1960 } 1961 break; 1962 1963 // load local variable or argument 1964 case ISN_LOAD: 1965 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 1966 goto theend; 1967 copy_tv(STACK_TV_VAR(iptr->isn_arg.number), STACK_TV_BOT(0)); 1968 ++ectx->ec_stack.ga_len; 1969 break; 1970 1971 // load v: variable 1972 case ISN_LOADV: 1973 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 1974 goto theend; 1975 copy_tv(get_vim_var_tv(iptr->isn_arg.number), STACK_TV_BOT(0)); 1976 ++ectx->ec_stack.ga_len; 1977 break; 1978 1979 // load s: variable in Vim9 script 1980 case ISN_LOADSCRIPT: 1981 { 1982 scriptref_T *sref = iptr->isn_arg.script.scriptref; 1983 svar_T *sv; 1984 1985 sv = get_script_svar(sref, ectx->ec_dfunc_idx); 1986 if (sv == NULL) 1987 goto theend; 1988 allocate_if_null(sv->sv_tv); 1989 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 1990 goto theend; 1991 copy_tv(sv->sv_tv, STACK_TV_BOT(0)); 1992 ++ectx->ec_stack.ga_len; 1993 } 1994 break; 1995 1996 // load s: variable in old script 1997 case ISN_LOADS: 1998 { 1999 hashtab_T *ht = &SCRIPT_VARS( 2000 iptr->isn_arg.loadstore.ls_sid); 2001 char_u *name = iptr->isn_arg.loadstore.ls_name; 2002 dictitem_T *di = find_var_in_ht(ht, 0, name, TRUE); 2003 2004 if (di == NULL) 2005 { 2006 SOURCING_LNUM = iptr->isn_lnum; 2007 semsg(_(e_undefined_variable_str), name); 2008 goto on_error; 2009 } 2010 else 2011 { 2012 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 2013 goto theend; 2014 copy_tv(&di->di_tv, STACK_TV_BOT(0)); 2015 ++ectx->ec_stack.ga_len; 2016 } 2017 } 2018 break; 2019 2020 // load g:/b:/w:/t: variable 2021 case ISN_LOADG: 2022 case ISN_LOADB: 2023 case ISN_LOADW: 2024 case ISN_LOADT: 2025 { 2026 dictitem_T *di = NULL; 2027 hashtab_T *ht = NULL; 2028 char namespace; 2029 2030 switch (iptr->isn_type) 2031 { 2032 case ISN_LOADG: 2033 ht = get_globvar_ht(); 2034 namespace = 'g'; 2035 break; 2036 case ISN_LOADB: 2037 ht = &curbuf->b_vars->dv_hashtab; 2038 namespace = 'b'; 2039 break; 2040 case ISN_LOADW: 2041 ht = &curwin->w_vars->dv_hashtab; 2042 namespace = 'w'; 2043 break; 2044 case ISN_LOADT: 2045 ht = &curtab->tp_vars->dv_hashtab; 2046 namespace = 't'; 2047 break; 2048 default: // Cannot reach here 2049 goto theend; 2050 } 2051 di = find_var_in_ht(ht, 0, iptr->isn_arg.string, TRUE); 2052 2053 if (di == NULL) 2054 { 2055 SOURCING_LNUM = iptr->isn_lnum; 2056 semsg(_(e_undefined_variable_char_str), 2057 namespace, iptr->isn_arg.string); 2058 goto on_error; 2059 } 2060 else 2061 { 2062 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 2063 goto theend; 2064 copy_tv(&di->di_tv, STACK_TV_BOT(0)); 2065 ++ectx->ec_stack.ga_len; 2066 } 2067 } 2068 break; 2069 2070 // load autoload variable 2071 case ISN_LOADAUTO: 2072 { 2073 char_u *name = iptr->isn_arg.string; 2074 2075 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 2076 goto theend; 2077 SOURCING_LNUM = iptr->isn_lnum; 2078 if (eval_variable(name, (int)STRLEN(name), 2079 STACK_TV_BOT(0), NULL, EVAL_VAR_VERBOSE) == FAIL) 2080 goto on_error; 2081 ++ectx->ec_stack.ga_len; 2082 } 2083 break; 2084 2085 // load g:/b:/w:/t: namespace 2086 case ISN_LOADGDICT: 2087 case ISN_LOADBDICT: 2088 case ISN_LOADWDICT: 2089 case ISN_LOADTDICT: 2090 { 2091 dict_T *d = NULL; 2092 2093 switch (iptr->isn_type) 2094 { 2095 case ISN_LOADGDICT: d = get_globvar_dict(); break; 2096 case ISN_LOADBDICT: d = curbuf->b_vars; break; 2097 case ISN_LOADWDICT: d = curwin->w_vars; break; 2098 case ISN_LOADTDICT: d = curtab->tp_vars; break; 2099 default: // Cannot reach here 2100 goto theend; 2101 } 2102 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 2103 goto theend; 2104 tv = STACK_TV_BOT(0); 2105 tv->v_type = VAR_DICT; 2106 tv->v_lock = 0; 2107 tv->vval.v_dict = d; 2108 ++d->dv_refcount; 2109 ++ectx->ec_stack.ga_len; 2110 } 2111 break; 2112 2113 // load &option 2114 case ISN_LOADOPT: 2115 { 2116 typval_T optval; 2117 char_u *name = iptr->isn_arg.string; 2118 2119 // This is not expected to fail, name is checked during 2120 // compilation: don't set SOURCING_LNUM. 2121 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 2122 goto theend; 2123 if (eval_option(&name, &optval, TRUE) == FAIL) 2124 goto theend; 2125 *STACK_TV_BOT(0) = optval; 2126 ++ectx->ec_stack.ga_len; 2127 } 2128 break; 2129 2130 // load $ENV 2131 case ISN_LOADENV: 2132 { 2133 typval_T optval; 2134 char_u *name = iptr->isn_arg.string; 2135 2136 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 2137 goto theend; 2138 // name is always valid, checked when compiling 2139 (void)eval_env_var(&name, &optval, TRUE); 2140 *STACK_TV_BOT(0) = optval; 2141 ++ectx->ec_stack.ga_len; 2142 } 2143 break; 2144 2145 // load @register 2146 case ISN_LOADREG: 2147 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 2148 goto theend; 2149 tv = STACK_TV_BOT(0); 2150 tv->v_type = VAR_STRING; 2151 tv->v_lock = 0; 2152 // This may result in NULL, which should be equivalent to an 2153 // empty string. 2154 tv->vval.v_string = get_reg_contents( 2155 iptr->isn_arg.number, GREG_EXPR_SRC); 2156 ++ectx->ec_stack.ga_len; 2157 break; 2158 2159 // store local variable 2160 case ISN_STORE: 2161 --ectx->ec_stack.ga_len; 2162 tv = STACK_TV_VAR(iptr->isn_arg.number); 2163 clear_tv(tv); 2164 *tv = *STACK_TV_BOT(0); 2165 break; 2166 2167 // store s: variable in old script 2168 case ISN_STORES: 2169 { 2170 hashtab_T *ht = &SCRIPT_VARS( 2171 iptr->isn_arg.loadstore.ls_sid); 2172 char_u *name = iptr->isn_arg.loadstore.ls_name; 2173 dictitem_T *di = find_var_in_ht(ht, 0, name + 2, TRUE); 2174 2175 --ectx->ec_stack.ga_len; 2176 if (di == NULL) 2177 store_var(name, STACK_TV_BOT(0)); 2178 else 2179 { 2180 SOURCING_LNUM = iptr->isn_lnum; 2181 if (var_check_permission(di, name) == FAIL) 2182 { 2183 clear_tv(STACK_TV_BOT(0)); 2184 goto on_error; 2185 } 2186 clear_tv(&di->di_tv); 2187 di->di_tv = *STACK_TV_BOT(0); 2188 } 2189 } 2190 break; 2191 2192 // store script-local variable in Vim9 script 2193 case ISN_STORESCRIPT: 2194 { 2195 scriptref_T *sref = iptr->isn_arg.script.scriptref; 2196 svar_T *sv; 2197 2198 sv = get_script_svar(sref, ectx->ec_dfunc_idx); 2199 if (sv == NULL) 2200 goto theend; 2201 --ectx->ec_stack.ga_len; 2202 2203 // "const" and "final" are checked at compile time, locking 2204 // the value needs to be checked here. 2205 SOURCING_LNUM = iptr->isn_lnum; 2206 if (value_check_lock(sv->sv_tv->v_lock, sv->sv_name, FALSE)) 2207 { 2208 clear_tv(STACK_TV_BOT(0)); 2209 goto on_error; 2210 } 2211 2212 clear_tv(sv->sv_tv); 2213 *sv->sv_tv = *STACK_TV_BOT(0); 2214 } 2215 break; 2216 2217 // store option 2218 case ISN_STOREOPT: 2219 { 2220 long n = 0; 2221 char_u *s = NULL; 2222 char *msg; 2223 2224 --ectx->ec_stack.ga_len; 2225 tv = STACK_TV_BOT(0); 2226 if (tv->v_type == VAR_STRING) 2227 { 2228 s = tv->vval.v_string; 2229 if (s == NULL) 2230 s = (char_u *)""; 2231 } 2232 else 2233 // must be VAR_NUMBER, CHECKTYPE makes sure 2234 n = tv->vval.v_number; 2235 msg = set_option_value(iptr->isn_arg.storeopt.so_name, 2236 n, s, iptr->isn_arg.storeopt.so_flags); 2237 clear_tv(tv); 2238 if (msg != NULL) 2239 { 2240 SOURCING_LNUM = iptr->isn_lnum; 2241 emsg(_(msg)); 2242 goto on_error; 2243 } 2244 } 2245 break; 2246 2247 // store $ENV 2248 case ISN_STOREENV: 2249 --ectx->ec_stack.ga_len; 2250 tv = STACK_TV_BOT(0); 2251 vim_setenv_ext(iptr->isn_arg.string, tv_get_string(tv)); 2252 clear_tv(tv); 2253 break; 2254 2255 // store @r 2256 case ISN_STOREREG: 2257 { 2258 int reg = iptr->isn_arg.number; 2259 2260 --ectx->ec_stack.ga_len; 2261 tv = STACK_TV_BOT(0); 2262 write_reg_contents(reg, tv_get_string(tv), -1, FALSE); 2263 clear_tv(tv); 2264 } 2265 break; 2266 2267 // store v: variable 2268 case ISN_STOREV: 2269 --ectx->ec_stack.ga_len; 2270 if (set_vim_var_tv(iptr->isn_arg.number, STACK_TV_BOT(0)) 2271 == FAIL) 2272 // should not happen, type is checked when compiling 2273 goto on_error; 2274 break; 2275 2276 // store g:/b:/w:/t: variable 2277 case ISN_STOREG: 2278 case ISN_STOREB: 2279 case ISN_STOREW: 2280 case ISN_STORET: 2281 { 2282 dictitem_T *di; 2283 hashtab_T *ht; 2284 char_u *name = iptr->isn_arg.string + 2; 2285 2286 switch (iptr->isn_type) 2287 { 2288 case ISN_STOREG: 2289 ht = get_globvar_ht(); 2290 break; 2291 case ISN_STOREB: 2292 ht = &curbuf->b_vars->dv_hashtab; 2293 break; 2294 case ISN_STOREW: 2295 ht = &curwin->w_vars->dv_hashtab; 2296 break; 2297 case ISN_STORET: 2298 ht = &curtab->tp_vars->dv_hashtab; 2299 break; 2300 default: // Cannot reach here 2301 goto theend; 2302 } 2303 2304 --ectx->ec_stack.ga_len; 2305 di = find_var_in_ht(ht, 0, name, TRUE); 2306 if (di == NULL) 2307 store_var(iptr->isn_arg.string, STACK_TV_BOT(0)); 2308 else 2309 { 2310 SOURCING_LNUM = iptr->isn_lnum; 2311 if (var_check_permission(di, name) == FAIL) 2312 goto on_error; 2313 clear_tv(&di->di_tv); 2314 di->di_tv = *STACK_TV_BOT(0); 2315 } 2316 } 2317 break; 2318 2319 // store an autoload variable 2320 case ISN_STOREAUTO: 2321 SOURCING_LNUM = iptr->isn_lnum; 2322 set_var(iptr->isn_arg.string, STACK_TV_BOT(-1), TRUE); 2323 clear_tv(STACK_TV_BOT(-1)); 2324 --ectx->ec_stack.ga_len; 2325 break; 2326 2327 // store number in local variable 2328 case ISN_STORENR: 2329 tv = STACK_TV_VAR(iptr->isn_arg.storenr.stnr_idx); 2330 clear_tv(tv); 2331 tv->v_type = VAR_NUMBER; 2332 tv->vval.v_number = iptr->isn_arg.storenr.stnr_val; 2333 break; 2334 2335 // store value in list or dict variable 2336 case ISN_STOREINDEX: 2337 { 2338 vartype_T dest_type = iptr->isn_arg.vartype; 2339 typval_T *tv_idx = STACK_TV_BOT(-2); 2340 typval_T *tv_dest = STACK_TV_BOT(-1); 2341 int status = OK; 2342 2343 // Stack contains: 2344 // -3 value to be stored 2345 // -2 index 2346 // -1 dict or list 2347 tv = STACK_TV_BOT(-3); 2348 SOURCING_LNUM = iptr->isn_lnum; 2349 if (dest_type == VAR_ANY) 2350 { 2351 dest_type = tv_dest->v_type; 2352 if (dest_type == VAR_DICT) 2353 status = do_2string(tv_idx, TRUE, FALSE); 2354 else if (dest_type == VAR_LIST 2355 && tv_idx->v_type != VAR_NUMBER) 2356 { 2357 emsg(_(e_number_expected)); 2358 status = FAIL; 2359 } 2360 } 2361 else if (dest_type != tv_dest->v_type) 2362 { 2363 // just in case, should be OK 2364 semsg(_(e_expected_str_but_got_str), 2365 vartype_name(dest_type), 2366 vartype_name(tv_dest->v_type)); 2367 status = FAIL; 2368 } 2369 2370 if (status == OK && dest_type == VAR_LIST) 2371 { 2372 long lidx = (long)tv_idx->vval.v_number; 2373 list_T *list = tv_dest->vval.v_list; 2374 2375 if (list == NULL) 2376 { 2377 emsg(_(e_list_not_set)); 2378 goto on_error; 2379 } 2380 if (lidx < 0 && list->lv_len + lidx >= 0) 2381 // negative index is relative to the end 2382 lidx = list->lv_len + lidx; 2383 if (lidx < 0 || lidx > list->lv_len) 2384 { 2385 semsg(_(e_listidx), lidx); 2386 goto on_error; 2387 } 2388 if (lidx < list->lv_len) 2389 { 2390 listitem_T *li = list_find(list, lidx); 2391 2392 if (error_if_locked(li->li_tv.v_lock, 2393 e_cannot_change_list_item)) 2394 goto on_error; 2395 // overwrite existing list item 2396 clear_tv(&li->li_tv); 2397 li->li_tv = *tv; 2398 } 2399 else 2400 { 2401 if (error_if_locked(list->lv_lock, 2402 e_cannot_change_list)) 2403 goto on_error; 2404 // append to list, only fails when out of memory 2405 if (list_append_tv(list, tv) == FAIL) 2406 goto theend; 2407 clear_tv(tv); 2408 } 2409 } 2410 else if (status == OK && dest_type == VAR_DICT) 2411 { 2412 char_u *key = tv_idx->vval.v_string; 2413 dict_T *dict = tv_dest->vval.v_dict; 2414 dictitem_T *di; 2415 2416 SOURCING_LNUM = iptr->isn_lnum; 2417 if (dict == NULL) 2418 { 2419 emsg(_(e_dictionary_not_set)); 2420 goto on_error; 2421 } 2422 if (key == NULL) 2423 key = (char_u *)""; 2424 di = dict_find(dict, key, -1); 2425 if (di != NULL) 2426 { 2427 if (error_if_locked(di->di_tv.v_lock, 2428 e_cannot_change_dict_item)) 2429 goto on_error; 2430 // overwrite existing value 2431 clear_tv(&di->di_tv); 2432 di->di_tv = *tv; 2433 } 2434 else 2435 { 2436 if (error_if_locked(dict->dv_lock, 2437 e_cannot_change_dict)) 2438 goto on_error; 2439 // add to dict, only fails when out of memory 2440 if (dict_add_tv(dict, (char *)key, tv) == FAIL) 2441 goto theend; 2442 clear_tv(tv); 2443 } 2444 } 2445 else if (status == OK && dest_type == VAR_BLOB) 2446 { 2447 long lidx = (long)tv_idx->vval.v_number; 2448 blob_T *blob = tv_dest->vval.v_blob; 2449 varnumber_T nr; 2450 int error = FALSE; 2451 int len; 2452 2453 if (blob == NULL) 2454 { 2455 emsg(_(e_blob_not_set)); 2456 goto on_error; 2457 } 2458 len = blob_len(blob); 2459 if (lidx < 0 && len + lidx >= 0) 2460 // negative index is relative to the end 2461 lidx = len + lidx; 2462 2463 // Can add one byte at the end. 2464 if (lidx < 0 || lidx > len) 2465 { 2466 semsg(_(e_blobidx), lidx); 2467 goto on_error; 2468 } 2469 if (value_check_lock(blob->bv_lock, 2470 (char_u *)"blob", FALSE)) 2471 goto on_error; 2472 nr = tv_get_number_chk(tv, &error); 2473 if (error) 2474 goto on_error; 2475 blob_set_append(blob, lidx, nr); 2476 } 2477 else 2478 { 2479 status = FAIL; 2480 semsg(_(e_cannot_index_str), vartype_name(dest_type)); 2481 } 2482 2483 clear_tv(tv_idx); 2484 clear_tv(tv_dest); 2485 ectx->ec_stack.ga_len -= 3; 2486 if (status == FAIL) 2487 { 2488 clear_tv(tv); 2489 goto on_error; 2490 } 2491 } 2492 break; 2493 2494 // store value in blob range 2495 case ISN_STORERANGE: 2496 { 2497 typval_T *tv_idx1 = STACK_TV_BOT(-3); 2498 typval_T *tv_idx2 = STACK_TV_BOT(-2); 2499 typval_T *tv_dest = STACK_TV_BOT(-1); 2500 int status = OK; 2501 2502 // Stack contains: 2503 // -4 value to be stored 2504 // -3 first index or "none" 2505 // -2 second index or "none" 2506 // -1 destination blob 2507 tv = STACK_TV_BOT(-4); 2508 if (tv_dest->v_type != VAR_BLOB) 2509 { 2510 status = FAIL; 2511 emsg(_(e_blob_required)); 2512 } 2513 else 2514 { 2515 varnumber_T n1; 2516 varnumber_T n2; 2517 int error = FALSE; 2518 2519 n1 = tv_get_number_chk(tv_idx1, &error); 2520 if (error) 2521 status = FAIL; 2522 else 2523 { 2524 if (tv_idx2->v_type == VAR_SPECIAL 2525 && tv_idx2->vval.v_number == VVAL_NONE) 2526 n2 = blob_len(tv_dest->vval.v_blob) - 1; 2527 else 2528 n2 = tv_get_number_chk(tv_idx2, &error); 2529 if (error) 2530 status = FAIL; 2531 else 2532 { 2533 long bloblen = blob_len(tv_dest->vval.v_blob); 2534 2535 if (check_blob_index(bloblen, 2536 n1, FALSE) == FAIL 2537 || check_blob_range(bloblen, 2538 n1, n2, FALSE) == FAIL) 2539 status = FAIL; 2540 else 2541 status = blob_set_range( 2542 tv_dest->vval.v_blob, n1, n2, tv); 2543 } 2544 } 2545 } 2546 2547 clear_tv(tv_idx1); 2548 clear_tv(tv_idx2); 2549 clear_tv(tv_dest); 2550 ectx->ec_stack.ga_len -= 4; 2551 clear_tv(tv); 2552 2553 if (status == FAIL) 2554 goto on_error; 2555 } 2556 break; 2557 2558 // load or store variable or argument from outer scope 2559 case ISN_LOADOUTER: 2560 case ISN_STOREOUTER: 2561 { 2562 int depth = iptr->isn_arg.outer.outer_depth; 2563 outer_T *outer = ectx->ec_outer_ref == NULL ? NULL 2564 : ectx->ec_outer_ref->or_outer; 2565 2566 while (depth > 1 && outer != NULL) 2567 { 2568 outer = outer->out_up; 2569 --depth; 2570 } 2571 if (outer == NULL) 2572 { 2573 SOURCING_LNUM = iptr->isn_lnum; 2574 iemsg("LOADOUTER depth more than scope levels"); 2575 goto theend; 2576 } 2577 tv = ((typval_T *)outer->out_stack->ga_data) 2578 + outer->out_frame_idx + STACK_FRAME_SIZE 2579 + iptr->isn_arg.outer.outer_idx; 2580 if (iptr->isn_type == ISN_LOADOUTER) 2581 { 2582 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 2583 goto theend; 2584 copy_tv(tv, STACK_TV_BOT(0)); 2585 ++ectx->ec_stack.ga_len; 2586 } 2587 else 2588 { 2589 --ectx->ec_stack.ga_len; 2590 clear_tv(tv); 2591 *tv = *STACK_TV_BOT(0); 2592 } 2593 } 2594 break; 2595 2596 // unlet item in list or dict variable 2597 case ISN_UNLETINDEX: 2598 { 2599 typval_T *tv_idx = STACK_TV_BOT(-2); 2600 typval_T *tv_dest = STACK_TV_BOT(-1); 2601 int status = OK; 2602 2603 // Stack contains: 2604 // -2 index 2605 // -1 dict or list 2606 if (tv_dest->v_type == VAR_DICT) 2607 { 2608 // unlet a dict item, index must be a string 2609 if (tv_idx->v_type != VAR_STRING) 2610 { 2611 SOURCING_LNUM = iptr->isn_lnum; 2612 semsg(_(e_expected_str_but_got_str), 2613 vartype_name(VAR_STRING), 2614 vartype_name(tv_idx->v_type)); 2615 status = FAIL; 2616 } 2617 else 2618 { 2619 dict_T *d = tv_dest->vval.v_dict; 2620 char_u *key = tv_idx->vval.v_string; 2621 dictitem_T *di = NULL; 2622 2623 if (key == NULL) 2624 key = (char_u *)""; 2625 if (d != NULL) 2626 di = dict_find(d, key, (int)STRLEN(key)); 2627 if (di == NULL) 2628 { 2629 // NULL dict is equivalent to empty dict 2630 SOURCING_LNUM = iptr->isn_lnum; 2631 semsg(_(e_dictkey), key); 2632 status = FAIL; 2633 } 2634 else 2635 { 2636 // TODO: check for dict or item locked 2637 dictitem_remove(d, di); 2638 } 2639 } 2640 } 2641 else if (tv_dest->v_type == VAR_LIST) 2642 { 2643 // unlet a List item, index must be a number 2644 SOURCING_LNUM = iptr->isn_lnum; 2645 if (check_for_number(tv_idx) == FAIL) 2646 { 2647 status = FAIL; 2648 } 2649 else 2650 { 2651 list_T *l = tv_dest->vval.v_list; 2652 long n = (long)tv_idx->vval.v_number; 2653 listitem_T *li = NULL; 2654 2655 li = list_find(l, n); 2656 if (li == NULL) 2657 { 2658 SOURCING_LNUM = iptr->isn_lnum; 2659 semsg(_(e_listidx), n); 2660 status = FAIL; 2661 } 2662 else 2663 // TODO: check for list or item locked 2664 listitem_remove(l, li); 2665 } 2666 } 2667 else 2668 { 2669 status = FAIL; 2670 semsg(_(e_cannot_index_str), 2671 vartype_name(tv_dest->v_type)); 2672 } 2673 2674 clear_tv(tv_idx); 2675 clear_tv(tv_dest); 2676 ectx->ec_stack.ga_len -= 2; 2677 if (status == FAIL) 2678 goto on_error; 2679 } 2680 break; 2681 2682 // unlet range of items in list variable 2683 case ISN_UNLETRANGE: 2684 { 2685 // Stack contains: 2686 // -3 index1 2687 // -2 index2 2688 // -1 dict or list 2689 typval_T *tv_idx1 = STACK_TV_BOT(-3); 2690 typval_T *tv_idx2 = STACK_TV_BOT(-2); 2691 typval_T *tv_dest = STACK_TV_BOT(-1); 2692 int status = OK; 2693 2694 if (tv_dest->v_type == VAR_LIST) 2695 { 2696 // indexes must be a number 2697 SOURCING_LNUM = iptr->isn_lnum; 2698 if (check_for_number(tv_idx1) == FAIL 2699 || (tv_idx2->v_type != VAR_SPECIAL 2700 && check_for_number(tv_idx2) == FAIL)) 2701 { 2702 status = FAIL; 2703 } 2704 else 2705 { 2706 list_T *l = tv_dest->vval.v_list; 2707 long n1 = (long)tv_idx1->vval.v_number; 2708 long n2 = tv_idx2->v_type == VAR_SPECIAL 2709 ? 0 : (long)tv_idx2->vval.v_number; 2710 listitem_T *li; 2711 2712 li = list_find_index(l, &n1); 2713 if (li == NULL) 2714 status = FAIL; 2715 else 2716 { 2717 if (n1 < 0) 2718 n1 = list_idx_of_item(l, li); 2719 if (n2 < 0) 2720 { 2721 listitem_T *li2 = list_find(l, n2); 2722 2723 if (li2 == NULL) 2724 status = FAIL; 2725 else 2726 n2 = list_idx_of_item(l, li2); 2727 } 2728 if (status != FAIL 2729 && tv_idx2->v_type != VAR_SPECIAL 2730 && n2 < n1) 2731 { 2732 semsg(_(e_listidx), n2); 2733 status = FAIL; 2734 } 2735 if (status != FAIL 2736 && list_unlet_range(l, li, NULL, n1, 2737 tv_idx2->v_type != VAR_SPECIAL, n2) 2738 == FAIL) 2739 status = FAIL; 2740 } 2741 } 2742 } 2743 else 2744 { 2745 status = FAIL; 2746 SOURCING_LNUM = iptr->isn_lnum; 2747 semsg(_(e_cannot_index_str), 2748 vartype_name(tv_dest->v_type)); 2749 } 2750 2751 clear_tv(tv_idx1); 2752 clear_tv(tv_idx2); 2753 clear_tv(tv_dest); 2754 ectx->ec_stack.ga_len -= 3; 2755 if (status == FAIL) 2756 goto on_error; 2757 } 2758 break; 2759 2760 // push constant 2761 case ISN_PUSHNR: 2762 case ISN_PUSHBOOL: 2763 case ISN_PUSHSPEC: 2764 case ISN_PUSHF: 2765 case ISN_PUSHS: 2766 case ISN_PUSHBLOB: 2767 case ISN_PUSHFUNC: 2768 case ISN_PUSHCHANNEL: 2769 case ISN_PUSHJOB: 2770 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 2771 goto theend; 2772 tv = STACK_TV_BOT(0); 2773 tv->v_lock = 0; 2774 ++ectx->ec_stack.ga_len; 2775 switch (iptr->isn_type) 2776 { 2777 case ISN_PUSHNR: 2778 tv->v_type = VAR_NUMBER; 2779 tv->vval.v_number = iptr->isn_arg.number; 2780 break; 2781 case ISN_PUSHBOOL: 2782 tv->v_type = VAR_BOOL; 2783 tv->vval.v_number = iptr->isn_arg.number; 2784 break; 2785 case ISN_PUSHSPEC: 2786 tv->v_type = VAR_SPECIAL; 2787 tv->vval.v_number = iptr->isn_arg.number; 2788 break; 2789 #ifdef FEAT_FLOAT 2790 case ISN_PUSHF: 2791 tv->v_type = VAR_FLOAT; 2792 tv->vval.v_float = iptr->isn_arg.fnumber; 2793 break; 2794 #endif 2795 case ISN_PUSHBLOB: 2796 blob_copy(iptr->isn_arg.blob, tv); 2797 break; 2798 case ISN_PUSHFUNC: 2799 tv->v_type = VAR_FUNC; 2800 if (iptr->isn_arg.string == NULL) 2801 tv->vval.v_string = NULL; 2802 else 2803 tv->vval.v_string = 2804 vim_strsave(iptr->isn_arg.string); 2805 break; 2806 case ISN_PUSHCHANNEL: 2807 #ifdef FEAT_JOB_CHANNEL 2808 tv->v_type = VAR_CHANNEL; 2809 tv->vval.v_channel = iptr->isn_arg.channel; 2810 if (tv->vval.v_channel != NULL) 2811 ++tv->vval.v_channel->ch_refcount; 2812 #endif 2813 break; 2814 case ISN_PUSHJOB: 2815 #ifdef FEAT_JOB_CHANNEL 2816 tv->v_type = VAR_JOB; 2817 tv->vval.v_job = iptr->isn_arg.job; 2818 if (tv->vval.v_job != NULL) 2819 ++tv->vval.v_job->jv_refcount; 2820 #endif 2821 break; 2822 default: 2823 tv->v_type = VAR_STRING; 2824 tv->vval.v_string = vim_strsave( 2825 iptr->isn_arg.string == NULL 2826 ? (char_u *)"" : iptr->isn_arg.string); 2827 } 2828 break; 2829 2830 case ISN_UNLET: 2831 if (do_unlet(iptr->isn_arg.unlet.ul_name, 2832 iptr->isn_arg.unlet.ul_forceit) == FAIL) 2833 goto on_error; 2834 break; 2835 case ISN_UNLETENV: 2836 vim_unsetenv(iptr->isn_arg.unlet.ul_name); 2837 break; 2838 2839 case ISN_LOCKCONST: 2840 item_lock(STACK_TV_BOT(-1), 100, TRUE, TRUE); 2841 break; 2842 2843 // create a list from items on the stack; uses a single allocation 2844 // for the list header and the items 2845 case ISN_NEWLIST: 2846 if (exe_newlist(iptr->isn_arg.number, ectx) == FAIL) 2847 goto theend; 2848 break; 2849 2850 // create a dict from items on the stack 2851 case ISN_NEWDICT: 2852 { 2853 int count = iptr->isn_arg.number; 2854 dict_T *dict = dict_alloc(); 2855 dictitem_T *item; 2856 char_u *key; 2857 int idx; 2858 2859 if (unlikely(dict == NULL)) 2860 goto theend; 2861 for (idx = 0; idx < count; ++idx) 2862 { 2863 // have already checked key type is VAR_STRING 2864 tv = STACK_TV_BOT(2 * (idx - count)); 2865 // check key is unique 2866 key = tv->vval.v_string == NULL 2867 ? (char_u *)"" : tv->vval.v_string; 2868 item = dict_find(dict, key, -1); 2869 if (item != NULL) 2870 { 2871 SOURCING_LNUM = iptr->isn_lnum; 2872 semsg(_(e_duplicate_key), key); 2873 dict_unref(dict); 2874 goto on_error; 2875 } 2876 item = dictitem_alloc(key); 2877 clear_tv(tv); 2878 if (unlikely(item == NULL)) 2879 { 2880 dict_unref(dict); 2881 goto theend; 2882 } 2883 item->di_tv = *STACK_TV_BOT(2 * (idx - count) + 1); 2884 item->di_tv.v_lock = 0; 2885 if (dict_add(dict, item) == FAIL) 2886 { 2887 // can this ever happen? 2888 dict_unref(dict); 2889 goto theend; 2890 } 2891 } 2892 2893 if (count > 0) 2894 ectx->ec_stack.ga_len -= 2 * count - 1; 2895 else if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 2896 goto theend; 2897 else 2898 ++ectx->ec_stack.ga_len; 2899 tv = STACK_TV_BOT(-1); 2900 tv->v_type = VAR_DICT; 2901 tv->v_lock = 0; 2902 tv->vval.v_dict = dict; 2903 ++dict->dv_refcount; 2904 } 2905 break; 2906 2907 // call a :def function 2908 case ISN_DCALL: 2909 SOURCING_LNUM = iptr->isn_lnum; 2910 if (call_dfunc(iptr->isn_arg.dfunc.cdf_idx, 2911 NULL, 2912 iptr->isn_arg.dfunc.cdf_argcount, 2913 ectx) == FAIL) 2914 goto on_error; 2915 break; 2916 2917 // call a builtin function 2918 case ISN_BCALL: 2919 SOURCING_LNUM = iptr->isn_lnum; 2920 if (call_bfunc(iptr->isn_arg.bfunc.cbf_idx, 2921 iptr->isn_arg.bfunc.cbf_argcount, 2922 ectx) == FAIL) 2923 goto on_error; 2924 break; 2925 2926 // call a funcref or partial 2927 case ISN_PCALL: 2928 { 2929 cpfunc_T *pfunc = &iptr->isn_arg.pfunc; 2930 int r; 2931 typval_T partial_tv; 2932 2933 SOURCING_LNUM = iptr->isn_lnum; 2934 if (pfunc->cpf_top) 2935 { 2936 // funcref is above the arguments 2937 tv = STACK_TV_BOT(-pfunc->cpf_argcount - 1); 2938 } 2939 else 2940 { 2941 // Get the funcref from the stack. 2942 --ectx->ec_stack.ga_len; 2943 partial_tv = *STACK_TV_BOT(0); 2944 tv = &partial_tv; 2945 } 2946 r = call_partial(tv, pfunc->cpf_argcount, ectx); 2947 if (tv == &partial_tv) 2948 clear_tv(&partial_tv); 2949 if (r == FAIL) 2950 goto on_error; 2951 } 2952 break; 2953 2954 case ISN_PCALL_END: 2955 // PCALL finished, arguments have been consumed and replaced by 2956 // the return value. Now clear the funcref from the stack, 2957 // and move the return value in its place. 2958 --ectx->ec_stack.ga_len; 2959 clear_tv(STACK_TV_BOT(-1)); 2960 *STACK_TV_BOT(-1) = *STACK_TV_BOT(0); 2961 break; 2962 2963 // call a user defined function or funcref/partial 2964 case ISN_UCALL: 2965 { 2966 cufunc_T *cufunc = &iptr->isn_arg.ufunc; 2967 2968 SOURCING_LNUM = iptr->isn_lnum; 2969 if (call_eval_func(cufunc->cuf_name, cufunc->cuf_argcount, 2970 ectx, iptr) == FAIL) 2971 goto on_error; 2972 } 2973 break; 2974 2975 // return from a :def function call without a value 2976 case ISN_RETURN_VOID: 2977 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 2978 goto theend; 2979 tv = STACK_TV_BOT(0); 2980 ++ectx->ec_stack.ga_len; 2981 tv->v_type = VAR_VOID; 2982 tv->vval.v_number = 0; 2983 tv->v_lock = 0; 2984 // FALLTHROUGH 2985 2986 // return from a :def function call with what is on the stack 2987 case ISN_RETURN: 2988 { 2989 garray_T *trystack = &ectx->ec_trystack; 2990 trycmd_T *trycmd = NULL; 2991 2992 if (trystack->ga_len > 0) 2993 trycmd = ((trycmd_T *)trystack->ga_data) 2994 + trystack->ga_len - 1; 2995 if (trycmd != NULL 2996 && trycmd->tcd_frame_idx == ectx->ec_frame_idx) 2997 { 2998 // jump to ":finally" or ":endtry" 2999 if (trycmd->tcd_finally_idx != 0) 3000 ectx->ec_iidx = trycmd->tcd_finally_idx; 3001 else 3002 ectx->ec_iidx = trycmd->tcd_endtry_idx; 3003 trycmd->tcd_return = TRUE; 3004 } 3005 else 3006 goto func_return; 3007 } 3008 break; 3009 3010 // push a partial, a reference to a compiled function 3011 case ISN_FUNCREF: 3012 { 3013 partial_T *pt = ALLOC_CLEAR_ONE(partial_T); 3014 dfunc_T *pt_dfunc = ((dfunc_T *)def_functions.ga_data) 3015 + iptr->isn_arg.funcref.fr_func; 3016 3017 if (pt == NULL) 3018 goto theend; 3019 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 3020 { 3021 vim_free(pt); 3022 goto theend; 3023 } 3024 if (fill_partial_and_closure(pt, pt_dfunc->df_ufunc, 3025 ectx) == FAIL) 3026 goto theend; 3027 tv = STACK_TV_BOT(0); 3028 ++ectx->ec_stack.ga_len; 3029 tv->vval.v_partial = pt; 3030 tv->v_type = VAR_PARTIAL; 3031 tv->v_lock = 0; 3032 } 3033 break; 3034 3035 // Create a global function from a lambda. 3036 case ISN_NEWFUNC: 3037 { 3038 newfunc_T *newfunc = &iptr->isn_arg.newfunc; 3039 3040 if (copy_func(newfunc->nf_lambda, newfunc->nf_global, 3041 ectx) == FAIL) 3042 goto theend; 3043 } 3044 break; 3045 3046 // List functions 3047 case ISN_DEF: 3048 if (iptr->isn_arg.string == NULL) 3049 list_functions(NULL); 3050 else 3051 { 3052 exarg_T ea; 3053 3054 CLEAR_FIELD(ea); 3055 ea.cmd = ea.arg = iptr->isn_arg.string; 3056 define_function(&ea, NULL); 3057 } 3058 break; 3059 3060 // jump if a condition is met 3061 case ISN_JUMP: 3062 { 3063 jumpwhen_T when = iptr->isn_arg.jump.jump_when; 3064 int error = FALSE; 3065 int jump = TRUE; 3066 3067 if (when != JUMP_ALWAYS) 3068 { 3069 tv = STACK_TV_BOT(-1); 3070 if (when == JUMP_IF_COND_FALSE 3071 || when == JUMP_IF_FALSE 3072 || when == JUMP_IF_COND_TRUE) 3073 { 3074 SOURCING_LNUM = iptr->isn_lnum; 3075 jump = tv_get_bool_chk(tv, &error); 3076 if (error) 3077 goto on_error; 3078 } 3079 else 3080 jump = tv2bool(tv); 3081 if (when == JUMP_IF_FALSE 3082 || when == JUMP_AND_KEEP_IF_FALSE 3083 || when == JUMP_IF_COND_FALSE) 3084 jump = !jump; 3085 if (when == JUMP_IF_FALSE || !jump) 3086 { 3087 // drop the value from the stack 3088 clear_tv(tv); 3089 --ectx->ec_stack.ga_len; 3090 } 3091 } 3092 if (jump) 3093 ectx->ec_iidx = iptr->isn_arg.jump.jump_where; 3094 } 3095 break; 3096 3097 // Jump if an argument with a default value was already set and not 3098 // v:none. 3099 case ISN_JUMP_IF_ARG_SET: 3100 tv = STACK_TV_VAR(iptr->isn_arg.jumparg.jump_arg_off); 3101 if (tv->v_type != VAR_UNKNOWN 3102 && !(tv->v_type == VAR_SPECIAL 3103 && tv->vval.v_number == VVAL_NONE)) 3104 ectx->ec_iidx = iptr->isn_arg.jumparg.jump_where; 3105 break; 3106 3107 // top of a for loop 3108 case ISN_FOR: 3109 { 3110 typval_T *ltv = STACK_TV_BOT(-1); 3111 typval_T *idxtv = 3112 STACK_TV_VAR(iptr->isn_arg.forloop.for_idx); 3113 3114 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 3115 goto theend; 3116 if (ltv->v_type == VAR_LIST) 3117 { 3118 list_T *list = ltv->vval.v_list; 3119 3120 // push the next item from the list 3121 ++idxtv->vval.v_number; 3122 if (list == NULL 3123 || idxtv->vval.v_number >= list->lv_len) 3124 { 3125 // past the end of the list, jump to "endfor" 3126 ectx->ec_iidx = iptr->isn_arg.forloop.for_end; 3127 may_restore_cmdmod(&ectx->ec_funclocal); 3128 } 3129 else if (list->lv_first == &range_list_item) 3130 { 3131 // non-materialized range() list 3132 tv = STACK_TV_BOT(0); 3133 tv->v_type = VAR_NUMBER; 3134 tv->v_lock = 0; 3135 tv->vval.v_number = list_find_nr( 3136 list, idxtv->vval.v_number, NULL); 3137 ++ectx->ec_stack.ga_len; 3138 } 3139 else 3140 { 3141 listitem_T *li = list_find(list, 3142 idxtv->vval.v_number); 3143 3144 copy_tv(&li->li_tv, STACK_TV_BOT(0)); 3145 ++ectx->ec_stack.ga_len; 3146 } 3147 } 3148 else if (ltv->v_type == VAR_STRING) 3149 { 3150 char_u *str = ltv->vval.v_string; 3151 3152 // The index is for the last byte of the previous 3153 // character. 3154 ++idxtv->vval.v_number; 3155 if (str == NULL || str[idxtv->vval.v_number] == NUL) 3156 { 3157 // past the end of the string, jump to "endfor" 3158 ectx->ec_iidx = iptr->isn_arg.forloop.for_end; 3159 may_restore_cmdmod(&ectx->ec_funclocal); 3160 } 3161 else 3162 { 3163 int clen = mb_ptr2len(str + idxtv->vval.v_number); 3164 3165 // Push the next character from the string. 3166 tv = STACK_TV_BOT(0); 3167 tv->v_type = VAR_STRING; 3168 tv->vval.v_string = vim_strnsave( 3169 str + idxtv->vval.v_number, clen); 3170 ++ectx->ec_stack.ga_len; 3171 idxtv->vval.v_number += clen - 1; 3172 } 3173 } 3174 else if (ltv->v_type == VAR_BLOB) 3175 { 3176 blob_T *blob = ltv->vval.v_blob; 3177 3178 // When we get here the first time make a copy of the 3179 // blob, so that the iteration still works when it is 3180 // changed. 3181 if (idxtv->vval.v_number == -1 && blob != NULL) 3182 { 3183 blob_copy(blob, ltv); 3184 blob_unref(blob); 3185 blob = ltv->vval.v_blob; 3186 } 3187 3188 // The index is for the previous byte. 3189 ++idxtv->vval.v_number; 3190 if (blob == NULL 3191 || idxtv->vval.v_number >= blob_len(blob)) 3192 { 3193 // past the end of the blob, jump to "endfor" 3194 ectx->ec_iidx = iptr->isn_arg.forloop.for_end; 3195 may_restore_cmdmod(&ectx->ec_funclocal); 3196 } 3197 else 3198 { 3199 // Push the next byte from the blob. 3200 tv = STACK_TV_BOT(0); 3201 tv->v_type = VAR_NUMBER; 3202 tv->vval.v_number = blob_get(blob, 3203 idxtv->vval.v_number); 3204 ++ectx->ec_stack.ga_len; 3205 } 3206 } 3207 else 3208 { 3209 semsg(_(e_for_loop_on_str_not_supported), 3210 vartype_name(ltv->v_type)); 3211 goto theend; 3212 } 3213 } 3214 break; 3215 3216 // start of ":try" block 3217 case ISN_TRY: 3218 { 3219 trycmd_T *trycmd = NULL; 3220 3221 if (GA_GROW_FAILS(&ectx->ec_trystack, 1)) 3222 goto theend; 3223 trycmd = ((trycmd_T *)ectx->ec_trystack.ga_data) 3224 + ectx->ec_trystack.ga_len; 3225 ++ectx->ec_trystack.ga_len; 3226 ++trylevel; 3227 CLEAR_POINTER(trycmd); 3228 trycmd->tcd_frame_idx = ectx->ec_frame_idx; 3229 trycmd->tcd_stack_len = ectx->ec_stack.ga_len; 3230 trycmd->tcd_catch_idx = 3231 iptr->isn_arg.try.try_ref->try_catch; 3232 trycmd->tcd_finally_idx = 3233 iptr->isn_arg.try.try_ref->try_finally; 3234 trycmd->tcd_endtry_idx = 3235 iptr->isn_arg.try.try_ref->try_endtry; 3236 } 3237 break; 3238 3239 case ISN_PUSHEXC: 3240 if (current_exception == NULL) 3241 { 3242 SOURCING_LNUM = iptr->isn_lnum; 3243 iemsg("Evaluating catch while current_exception is NULL"); 3244 goto theend; 3245 } 3246 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 3247 goto theend; 3248 tv = STACK_TV_BOT(0); 3249 ++ectx->ec_stack.ga_len; 3250 tv->v_type = VAR_STRING; 3251 tv->v_lock = 0; 3252 tv->vval.v_string = vim_strsave( 3253 (char_u *)current_exception->value); 3254 break; 3255 3256 case ISN_CATCH: 3257 { 3258 garray_T *trystack = &ectx->ec_trystack; 3259 3260 may_restore_cmdmod(&ectx->ec_funclocal); 3261 if (trystack->ga_len > 0) 3262 { 3263 trycmd_T *trycmd = ((trycmd_T *)trystack->ga_data) 3264 + trystack->ga_len - 1; 3265 trycmd->tcd_caught = TRUE; 3266 trycmd->tcd_did_throw = FALSE; 3267 } 3268 did_emsg = got_int = did_throw = FALSE; 3269 force_abort = need_rethrow = FALSE; 3270 catch_exception(current_exception); 3271 } 3272 break; 3273 3274 case ISN_TRYCONT: 3275 { 3276 garray_T *trystack = &ectx->ec_trystack; 3277 trycont_T *trycont = &iptr->isn_arg.trycont; 3278 int i; 3279 trycmd_T *trycmd; 3280 int iidx = trycont->tct_where; 3281 3282 if (trystack->ga_len < trycont->tct_levels) 3283 { 3284 siemsg("TRYCONT: expected %d levels, found %d", 3285 trycont->tct_levels, trystack->ga_len); 3286 goto theend; 3287 } 3288 // Make :endtry jump to any outer try block and the last 3289 // :endtry inside the loop to the loop start. 3290 for (i = trycont->tct_levels; i > 0; --i) 3291 { 3292 trycmd = ((trycmd_T *)trystack->ga_data) 3293 + trystack->ga_len - i; 3294 // Add one to tcd_cont to be able to jump to 3295 // instruction with index zero. 3296 trycmd->tcd_cont = iidx + 1; 3297 iidx = trycmd->tcd_finally_idx == 0 3298 ? trycmd->tcd_endtry_idx : trycmd->tcd_finally_idx; 3299 } 3300 // jump to :finally or :endtry of current try statement 3301 ectx->ec_iidx = iidx; 3302 } 3303 break; 3304 3305 case ISN_FINALLY: 3306 { 3307 garray_T *trystack = &ectx->ec_trystack; 3308 trycmd_T *trycmd = ((trycmd_T *)trystack->ga_data) 3309 + trystack->ga_len - 1; 3310 3311 // Reset the index to avoid a return statement jumps here 3312 // again. 3313 trycmd->tcd_finally_idx = 0; 3314 break; 3315 } 3316 3317 // end of ":try" block 3318 case ISN_ENDTRY: 3319 { 3320 garray_T *trystack = &ectx->ec_trystack; 3321 3322 if (trystack->ga_len > 0) 3323 { 3324 trycmd_T *trycmd; 3325 3326 --trystack->ga_len; 3327 --trylevel; 3328 trycmd = ((trycmd_T *)trystack->ga_data) 3329 + trystack->ga_len; 3330 if (trycmd->tcd_did_throw) 3331 did_throw = TRUE; 3332 if (trycmd->tcd_caught && current_exception != NULL) 3333 { 3334 // discard the exception 3335 if (caught_stack == current_exception) 3336 caught_stack = caught_stack->caught; 3337 discard_current_exception(); 3338 } 3339 3340 if (trycmd->tcd_return) 3341 goto func_return; 3342 3343 while (ectx->ec_stack.ga_len > trycmd->tcd_stack_len) 3344 { 3345 --ectx->ec_stack.ga_len; 3346 clear_tv(STACK_TV_BOT(0)); 3347 } 3348 if (trycmd->tcd_cont != 0) 3349 // handling :continue: jump to outer try block or 3350 // start of the loop 3351 ectx->ec_iidx = trycmd->tcd_cont - 1; 3352 } 3353 } 3354 break; 3355 3356 case ISN_THROW: 3357 { 3358 garray_T *trystack = &ectx->ec_trystack; 3359 3360 if (trystack->ga_len == 0 && trylevel == 0 && emsg_silent) 3361 { 3362 // throwing an exception while using "silent!" causes 3363 // the function to abort but not display an error. 3364 tv = STACK_TV_BOT(-1); 3365 clear_tv(tv); 3366 tv->v_type = VAR_NUMBER; 3367 tv->vval.v_number = 0; 3368 goto done; 3369 } 3370 --ectx->ec_stack.ga_len; 3371 tv = STACK_TV_BOT(0); 3372 if (tv->vval.v_string == NULL 3373 || *skipwhite(tv->vval.v_string) == NUL) 3374 { 3375 vim_free(tv->vval.v_string); 3376 SOURCING_LNUM = iptr->isn_lnum; 3377 emsg(_(e_throw_with_empty_string)); 3378 goto theend; 3379 } 3380 3381 // Inside a "catch" we need to first discard the caught 3382 // exception. 3383 if (trystack->ga_len > 0) 3384 { 3385 trycmd_T *trycmd = ((trycmd_T *)trystack->ga_data) 3386 + trystack->ga_len - 1; 3387 if (trycmd->tcd_caught && current_exception != NULL) 3388 { 3389 // discard the exception 3390 if (caught_stack == current_exception) 3391 caught_stack = caught_stack->caught; 3392 discard_current_exception(); 3393 trycmd->tcd_caught = FALSE; 3394 } 3395 } 3396 3397 if (throw_exception(tv->vval.v_string, ET_USER, NULL) 3398 == FAIL) 3399 { 3400 vim_free(tv->vval.v_string); 3401 goto theend; 3402 } 3403 did_throw = TRUE; 3404 } 3405 break; 3406 3407 // compare with special values 3408 case ISN_COMPAREBOOL: 3409 case ISN_COMPARESPECIAL: 3410 { 3411 typval_T *tv1 = STACK_TV_BOT(-2); 3412 typval_T *tv2 = STACK_TV_BOT(-1); 3413 varnumber_T arg1 = tv1->vval.v_number; 3414 varnumber_T arg2 = tv2->vval.v_number; 3415 int res; 3416 3417 switch (iptr->isn_arg.op.op_type) 3418 { 3419 case EXPR_EQUAL: res = arg1 == arg2; break; 3420 case EXPR_NEQUAL: res = arg1 != arg2; break; 3421 default: res = 0; break; 3422 } 3423 3424 --ectx->ec_stack.ga_len; 3425 tv1->v_type = VAR_BOOL; 3426 tv1->vval.v_number = res ? VVAL_TRUE : VVAL_FALSE; 3427 } 3428 break; 3429 3430 // Operation with two number arguments 3431 case ISN_OPNR: 3432 case ISN_COMPARENR: 3433 { 3434 typval_T *tv1 = STACK_TV_BOT(-2); 3435 typval_T *tv2 = STACK_TV_BOT(-1); 3436 varnumber_T arg1 = tv1->vval.v_number; 3437 varnumber_T arg2 = tv2->vval.v_number; 3438 varnumber_T res = 0; 3439 int div_zero = FALSE; 3440 3441 switch (iptr->isn_arg.op.op_type) 3442 { 3443 case EXPR_MULT: res = arg1 * arg2; break; 3444 case EXPR_DIV: if (arg2 == 0) 3445 div_zero = TRUE; 3446 else 3447 res = arg1 / arg2; 3448 break; 3449 case EXPR_REM: if (arg2 == 0) 3450 div_zero = TRUE; 3451 else 3452 res = arg1 % arg2; 3453 break; 3454 case EXPR_SUB: res = arg1 - arg2; break; 3455 case EXPR_ADD: res = arg1 + arg2; break; 3456 3457 case EXPR_EQUAL: res = arg1 == arg2; break; 3458 case EXPR_NEQUAL: res = arg1 != arg2; break; 3459 case EXPR_GREATER: res = arg1 > arg2; break; 3460 case EXPR_GEQUAL: res = arg1 >= arg2; break; 3461 case EXPR_SMALLER: res = arg1 < arg2; break; 3462 case EXPR_SEQUAL: res = arg1 <= arg2; break; 3463 default: break; 3464 } 3465 3466 --ectx->ec_stack.ga_len; 3467 if (iptr->isn_type == ISN_COMPARENR) 3468 { 3469 tv1->v_type = VAR_BOOL; 3470 tv1->vval.v_number = res ? VVAL_TRUE : VVAL_FALSE; 3471 } 3472 else 3473 tv1->vval.v_number = res; 3474 if (div_zero) 3475 { 3476 SOURCING_LNUM = iptr->isn_lnum; 3477 emsg(_(e_divide_by_zero)); 3478 goto on_error; 3479 } 3480 } 3481 break; 3482 3483 // Computation with two float arguments 3484 case ISN_OPFLOAT: 3485 case ISN_COMPAREFLOAT: 3486 #ifdef FEAT_FLOAT 3487 { 3488 typval_T *tv1 = STACK_TV_BOT(-2); 3489 typval_T *tv2 = STACK_TV_BOT(-1); 3490 float_T arg1 = tv1->vval.v_float; 3491 float_T arg2 = tv2->vval.v_float; 3492 float_T res = 0; 3493 int cmp = FALSE; 3494 3495 switch (iptr->isn_arg.op.op_type) 3496 { 3497 case EXPR_MULT: res = arg1 * arg2; break; 3498 case EXPR_DIV: res = arg1 / arg2; break; 3499 case EXPR_SUB: res = arg1 - arg2; break; 3500 case EXPR_ADD: res = arg1 + arg2; break; 3501 3502 case EXPR_EQUAL: cmp = arg1 == arg2; break; 3503 case EXPR_NEQUAL: cmp = arg1 != arg2; break; 3504 case EXPR_GREATER: cmp = arg1 > arg2; break; 3505 case EXPR_GEQUAL: cmp = arg1 >= arg2; break; 3506 case EXPR_SMALLER: cmp = arg1 < arg2; break; 3507 case EXPR_SEQUAL: cmp = arg1 <= arg2; break; 3508 default: cmp = 0; break; 3509 } 3510 --ectx->ec_stack.ga_len; 3511 if (iptr->isn_type == ISN_COMPAREFLOAT) 3512 { 3513 tv1->v_type = VAR_BOOL; 3514 tv1->vval.v_number = cmp ? VVAL_TRUE : VVAL_FALSE; 3515 } 3516 else 3517 tv1->vval.v_float = res; 3518 } 3519 #endif 3520 break; 3521 3522 case ISN_COMPARELIST: 3523 { 3524 typval_T *tv1 = STACK_TV_BOT(-2); 3525 typval_T *tv2 = STACK_TV_BOT(-1); 3526 list_T *arg1 = tv1->vval.v_list; 3527 list_T *arg2 = tv2->vval.v_list; 3528 int cmp = FALSE; 3529 int ic = iptr->isn_arg.op.op_ic; 3530 3531 switch (iptr->isn_arg.op.op_type) 3532 { 3533 case EXPR_EQUAL: cmp = 3534 list_equal(arg1, arg2, ic, FALSE); break; 3535 case EXPR_NEQUAL: cmp = 3536 !list_equal(arg1, arg2, ic, FALSE); break; 3537 case EXPR_IS: cmp = arg1 == arg2; break; 3538 case EXPR_ISNOT: cmp = arg1 != arg2; break; 3539 default: cmp = 0; break; 3540 } 3541 --ectx->ec_stack.ga_len; 3542 clear_tv(tv1); 3543 clear_tv(tv2); 3544 tv1->v_type = VAR_BOOL; 3545 tv1->vval.v_number = cmp ? VVAL_TRUE : VVAL_FALSE; 3546 } 3547 break; 3548 3549 case ISN_COMPAREBLOB: 3550 { 3551 typval_T *tv1 = STACK_TV_BOT(-2); 3552 typval_T *tv2 = STACK_TV_BOT(-1); 3553 blob_T *arg1 = tv1->vval.v_blob; 3554 blob_T *arg2 = tv2->vval.v_blob; 3555 int cmp = FALSE; 3556 3557 switch (iptr->isn_arg.op.op_type) 3558 { 3559 case EXPR_EQUAL: cmp = blob_equal(arg1, arg2); break; 3560 case EXPR_NEQUAL: cmp = !blob_equal(arg1, arg2); break; 3561 case EXPR_IS: cmp = arg1 == arg2; break; 3562 case EXPR_ISNOT: cmp = arg1 != arg2; break; 3563 default: cmp = 0; break; 3564 } 3565 --ectx->ec_stack.ga_len; 3566 clear_tv(tv1); 3567 clear_tv(tv2); 3568 tv1->v_type = VAR_BOOL; 3569 tv1->vval.v_number = cmp ? VVAL_TRUE : VVAL_FALSE; 3570 } 3571 break; 3572 3573 // TODO: handle separately 3574 case ISN_COMPARESTRING: 3575 case ISN_COMPAREDICT: 3576 case ISN_COMPAREFUNC: 3577 case ISN_COMPAREANY: 3578 { 3579 typval_T *tv1 = STACK_TV_BOT(-2); 3580 typval_T *tv2 = STACK_TV_BOT(-1); 3581 exprtype_T exprtype = iptr->isn_arg.op.op_type; 3582 int ic = iptr->isn_arg.op.op_ic; 3583 3584 SOURCING_LNUM = iptr->isn_lnum; 3585 typval_compare(tv1, tv2, exprtype, ic); 3586 clear_tv(tv2); 3587 --ectx->ec_stack.ga_len; 3588 } 3589 break; 3590 3591 case ISN_ADDLIST: 3592 case ISN_ADDBLOB: 3593 { 3594 typval_T *tv1 = STACK_TV_BOT(-2); 3595 typval_T *tv2 = STACK_TV_BOT(-1); 3596 3597 // add two lists or blobs 3598 if (iptr->isn_type == ISN_ADDLIST) 3599 eval_addlist(tv1, tv2); 3600 else 3601 eval_addblob(tv1, tv2); 3602 clear_tv(tv2); 3603 --ectx->ec_stack.ga_len; 3604 } 3605 break; 3606 3607 case ISN_LISTAPPEND: 3608 { 3609 typval_T *tv1 = STACK_TV_BOT(-2); 3610 typval_T *tv2 = STACK_TV_BOT(-1); 3611 list_T *l = tv1->vval.v_list; 3612 3613 // add an item to a list 3614 if (l == NULL) 3615 { 3616 SOURCING_LNUM = iptr->isn_lnum; 3617 emsg(_(e_cannot_add_to_null_list)); 3618 goto on_error; 3619 } 3620 if (list_append_tv(l, tv2) == FAIL) 3621 goto theend; 3622 clear_tv(tv2); 3623 --ectx->ec_stack.ga_len; 3624 } 3625 break; 3626 3627 case ISN_BLOBAPPEND: 3628 { 3629 typval_T *tv1 = STACK_TV_BOT(-2); 3630 typval_T *tv2 = STACK_TV_BOT(-1); 3631 blob_T *b = tv1->vval.v_blob; 3632 int error = FALSE; 3633 varnumber_T n; 3634 3635 // add a number to a blob 3636 if (b == NULL) 3637 { 3638 SOURCING_LNUM = iptr->isn_lnum; 3639 emsg(_(e_cannot_add_to_null_blob)); 3640 goto on_error; 3641 } 3642 n = tv_get_number_chk(tv2, &error); 3643 if (error) 3644 goto on_error; 3645 ga_append(&b->bv_ga, (int)n); 3646 --ectx->ec_stack.ga_len; 3647 } 3648 break; 3649 3650 // Computation with two arguments of unknown type 3651 case ISN_OPANY: 3652 { 3653 typval_T *tv1 = STACK_TV_BOT(-2); 3654 typval_T *tv2 = STACK_TV_BOT(-1); 3655 varnumber_T n1, n2; 3656 #ifdef FEAT_FLOAT 3657 float_T f1 = 0, f2 = 0; 3658 #endif 3659 int error = FALSE; 3660 3661 if (iptr->isn_arg.op.op_type == EXPR_ADD) 3662 { 3663 if (tv1->v_type == VAR_LIST && tv2->v_type == VAR_LIST) 3664 { 3665 eval_addlist(tv1, tv2); 3666 clear_tv(tv2); 3667 --ectx->ec_stack.ga_len; 3668 break; 3669 } 3670 else if (tv1->v_type == VAR_BLOB 3671 && tv2->v_type == VAR_BLOB) 3672 { 3673 eval_addblob(tv1, tv2); 3674 clear_tv(tv2); 3675 --ectx->ec_stack.ga_len; 3676 break; 3677 } 3678 } 3679 #ifdef FEAT_FLOAT 3680 if (tv1->v_type == VAR_FLOAT) 3681 { 3682 f1 = tv1->vval.v_float; 3683 n1 = 0; 3684 } 3685 else 3686 #endif 3687 { 3688 SOURCING_LNUM = iptr->isn_lnum; 3689 n1 = tv_get_number_chk(tv1, &error); 3690 if (error) 3691 goto on_error; 3692 #ifdef FEAT_FLOAT 3693 if (tv2->v_type == VAR_FLOAT) 3694 f1 = n1; 3695 #endif 3696 } 3697 #ifdef FEAT_FLOAT 3698 if (tv2->v_type == VAR_FLOAT) 3699 { 3700 f2 = tv2->vval.v_float; 3701 n2 = 0; 3702 } 3703 else 3704 #endif 3705 { 3706 n2 = tv_get_number_chk(tv2, &error); 3707 if (error) 3708 goto on_error; 3709 #ifdef FEAT_FLOAT 3710 if (tv1->v_type == VAR_FLOAT) 3711 f2 = n2; 3712 #endif 3713 } 3714 #ifdef FEAT_FLOAT 3715 // if there is a float on either side the result is a float 3716 if (tv1->v_type == VAR_FLOAT || tv2->v_type == VAR_FLOAT) 3717 { 3718 switch (iptr->isn_arg.op.op_type) 3719 { 3720 case EXPR_MULT: f1 = f1 * f2; break; 3721 case EXPR_DIV: f1 = f1 / f2; break; 3722 case EXPR_SUB: f1 = f1 - f2; break; 3723 case EXPR_ADD: f1 = f1 + f2; break; 3724 default: SOURCING_LNUM = iptr->isn_lnum; 3725 emsg(_(e_modulus)); 3726 goto on_error; 3727 } 3728 clear_tv(tv1); 3729 clear_tv(tv2); 3730 tv1->v_type = VAR_FLOAT; 3731 tv1->vval.v_float = f1; 3732 --ectx->ec_stack.ga_len; 3733 } 3734 else 3735 #endif 3736 { 3737 int failed = FALSE; 3738 3739 switch (iptr->isn_arg.op.op_type) 3740 { 3741 case EXPR_MULT: n1 = n1 * n2; break; 3742 case EXPR_DIV: n1 = num_divide(n1, n2, &failed); 3743 if (failed) 3744 goto on_error; 3745 break; 3746 case EXPR_SUB: n1 = n1 - n2; break; 3747 case EXPR_ADD: n1 = n1 + n2; break; 3748 default: n1 = num_modulus(n1, n2, &failed); 3749 if (failed) 3750 goto on_error; 3751 break; 3752 } 3753 clear_tv(tv1); 3754 clear_tv(tv2); 3755 tv1->v_type = VAR_NUMBER; 3756 tv1->vval.v_number = n1; 3757 --ectx->ec_stack.ga_len; 3758 } 3759 } 3760 break; 3761 3762 case ISN_CONCAT: 3763 { 3764 char_u *str1 = STACK_TV_BOT(-2)->vval.v_string; 3765 char_u *str2 = STACK_TV_BOT(-1)->vval.v_string; 3766 char_u *res; 3767 3768 res = concat_str(str1, str2); 3769 clear_tv(STACK_TV_BOT(-2)); 3770 clear_tv(STACK_TV_BOT(-1)); 3771 --ectx->ec_stack.ga_len; 3772 STACK_TV_BOT(-1)->vval.v_string = res; 3773 } 3774 break; 3775 3776 case ISN_STRINDEX: 3777 case ISN_STRSLICE: 3778 { 3779 int is_slice = iptr->isn_type == ISN_STRSLICE; 3780 varnumber_T n1 = 0, n2; 3781 char_u *res; 3782 3783 // string index: string is at stack-2, index at stack-1 3784 // string slice: string is at stack-3, first index at 3785 // stack-2, second index at stack-1 3786 if (is_slice) 3787 { 3788 tv = STACK_TV_BOT(-2); 3789 n1 = tv->vval.v_number; 3790 } 3791 3792 tv = STACK_TV_BOT(-1); 3793 n2 = tv->vval.v_number; 3794 3795 ectx->ec_stack.ga_len -= is_slice ? 2 : 1; 3796 tv = STACK_TV_BOT(-1); 3797 if (is_slice) 3798 // Slice: Select the characters from the string 3799 res = string_slice(tv->vval.v_string, n1, n2, FALSE); 3800 else 3801 // Index: The resulting variable is a string of a 3802 // single character (including composing characters). 3803 // If the index is too big or negative the result is 3804 // empty. 3805 res = char_from_string(tv->vval.v_string, n2); 3806 vim_free(tv->vval.v_string); 3807 tv->vval.v_string = res; 3808 } 3809 break; 3810 3811 case ISN_LISTINDEX: 3812 case ISN_LISTSLICE: 3813 case ISN_BLOBINDEX: 3814 case ISN_BLOBSLICE: 3815 { 3816 int is_slice = iptr->isn_type == ISN_LISTSLICE 3817 || iptr->isn_type == ISN_BLOBSLICE; 3818 int is_blob = iptr->isn_type == ISN_BLOBINDEX 3819 || iptr->isn_type == ISN_BLOBSLICE; 3820 varnumber_T n1, n2; 3821 typval_T *val_tv; 3822 3823 // list index: list is at stack-2, index at stack-1 3824 // list slice: list is at stack-3, indexes at stack-2 and 3825 // stack-1 3826 // Same for blob. 3827 val_tv = is_slice ? STACK_TV_BOT(-3) : STACK_TV_BOT(-2); 3828 3829 tv = STACK_TV_BOT(-1); 3830 n1 = n2 = tv->vval.v_number; 3831 clear_tv(tv); 3832 3833 if (is_slice) 3834 { 3835 tv = STACK_TV_BOT(-2); 3836 n1 = tv->vval.v_number; 3837 clear_tv(tv); 3838 } 3839 3840 ectx->ec_stack.ga_len -= is_slice ? 2 : 1; 3841 tv = STACK_TV_BOT(-1); 3842 SOURCING_LNUM = iptr->isn_lnum; 3843 if (is_blob) 3844 { 3845 if (blob_slice_or_index(val_tv->vval.v_blob, is_slice, 3846 n1, n2, FALSE, tv) == FAIL) 3847 goto on_error; 3848 } 3849 else 3850 { 3851 if (list_slice_or_index(val_tv->vval.v_list, is_slice, 3852 n1, n2, FALSE, tv, TRUE) == FAIL) 3853 goto on_error; 3854 } 3855 } 3856 break; 3857 3858 case ISN_ANYINDEX: 3859 case ISN_ANYSLICE: 3860 { 3861 int is_slice = iptr->isn_type == ISN_ANYSLICE; 3862 typval_T *var1, *var2; 3863 int res; 3864 3865 // index: composite is at stack-2, index at stack-1 3866 // slice: composite is at stack-3, indexes at stack-2 and 3867 // stack-1 3868 tv = is_slice ? STACK_TV_BOT(-3) : STACK_TV_BOT(-2); 3869 SOURCING_LNUM = iptr->isn_lnum; 3870 if (check_can_index(tv, TRUE, TRUE) == FAIL) 3871 goto on_error; 3872 var1 = is_slice ? STACK_TV_BOT(-2) : STACK_TV_BOT(-1); 3873 var2 = is_slice ? STACK_TV_BOT(-1) : NULL; 3874 res = eval_index_inner(tv, is_slice, var1, var2, 3875 FALSE, NULL, -1, TRUE); 3876 clear_tv(var1); 3877 if (is_slice) 3878 clear_tv(var2); 3879 ectx->ec_stack.ga_len -= is_slice ? 2 : 1; 3880 if (res == FAIL) 3881 goto on_error; 3882 } 3883 break; 3884 3885 case ISN_SLICE: 3886 { 3887 list_T *list; 3888 int count = iptr->isn_arg.number; 3889 3890 // type will have been checked to be a list 3891 tv = STACK_TV_BOT(-1); 3892 list = tv->vval.v_list; 3893 3894 // no error for short list, expect it to be checked earlier 3895 if (list != NULL && list->lv_len >= count) 3896 { 3897 list_T *newlist = list_slice(list, 3898 count, list->lv_len - 1); 3899 3900 if (newlist != NULL) 3901 { 3902 list_unref(list); 3903 tv->vval.v_list = newlist; 3904 ++newlist->lv_refcount; 3905 } 3906 } 3907 } 3908 break; 3909 3910 case ISN_GETITEM: 3911 { 3912 listitem_T *li; 3913 getitem_T *gi = &iptr->isn_arg.getitem; 3914 3915 // Get list item: list is at stack-1, push item. 3916 // List type and length is checked for when compiling. 3917 tv = STACK_TV_BOT(-1 - gi->gi_with_op); 3918 li = list_find(tv->vval.v_list, gi->gi_index); 3919 3920 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 3921 goto theend; 3922 ++ectx->ec_stack.ga_len; 3923 copy_tv(&li->li_tv, STACK_TV_BOT(-1)); 3924 3925 // Useful when used in unpack assignment. Reset at 3926 // ISN_DROP. 3927 ectx->ec_where.wt_index = gi->gi_index + 1; 3928 ectx->ec_where.wt_variable = TRUE; 3929 } 3930 break; 3931 3932 case ISN_MEMBER: 3933 { 3934 dict_T *dict; 3935 char_u *key; 3936 dictitem_T *di; 3937 typval_T temp_tv; 3938 3939 // dict member: dict is at stack-2, key at stack-1 3940 tv = STACK_TV_BOT(-2); 3941 // no need to check for VAR_DICT, CHECKTYPE will check. 3942 dict = tv->vval.v_dict; 3943 3944 tv = STACK_TV_BOT(-1); 3945 // no need to check for VAR_STRING, 2STRING will check. 3946 key = tv->vval.v_string; 3947 if (key == NULL) 3948 key = (char_u *)""; 3949 3950 if ((di = dict_find(dict, key, -1)) == NULL) 3951 { 3952 SOURCING_LNUM = iptr->isn_lnum; 3953 semsg(_(e_dictkey), key); 3954 3955 // If :silent! is used we will continue, make sure the 3956 // stack contents makes sense. 3957 clear_tv(tv); 3958 --ectx->ec_stack.ga_len; 3959 tv = STACK_TV_BOT(-1); 3960 clear_tv(tv); 3961 tv->v_type = VAR_NUMBER; 3962 tv->vval.v_number = 0; 3963 goto on_fatal_error; 3964 } 3965 clear_tv(tv); 3966 --ectx->ec_stack.ga_len; 3967 // Clear the dict only after getting the item, to avoid 3968 // that it makes the item invalid. 3969 tv = STACK_TV_BOT(-1); 3970 temp_tv = *tv; 3971 copy_tv(&di->di_tv, tv); 3972 clear_tv(&temp_tv); 3973 } 3974 break; 3975 3976 // dict member with string key 3977 case ISN_STRINGMEMBER: 3978 { 3979 dict_T *dict; 3980 dictitem_T *di; 3981 typval_T temp_tv; 3982 3983 tv = STACK_TV_BOT(-1); 3984 if (tv->v_type != VAR_DICT || tv->vval.v_dict == NULL) 3985 { 3986 SOURCING_LNUM = iptr->isn_lnum; 3987 emsg(_(e_dictreq)); 3988 goto on_error; 3989 } 3990 dict = tv->vval.v_dict; 3991 3992 if ((di = dict_find(dict, iptr->isn_arg.string, -1)) 3993 == NULL) 3994 { 3995 SOURCING_LNUM = iptr->isn_lnum; 3996 semsg(_(e_dictkey), iptr->isn_arg.string); 3997 goto on_error; 3998 } 3999 // Clear the dict after getting the item, to avoid that it 4000 // make the item invalid. 4001 temp_tv = *tv; 4002 copy_tv(&di->di_tv, tv); 4003 clear_tv(&temp_tv); 4004 } 4005 break; 4006 4007 case ISN_NEGATENR: 4008 tv = STACK_TV_BOT(-1); 4009 if (tv->v_type != VAR_NUMBER 4010 #ifdef FEAT_FLOAT 4011 && tv->v_type != VAR_FLOAT 4012 #endif 4013 ) 4014 { 4015 SOURCING_LNUM = iptr->isn_lnum; 4016 emsg(_(e_number_expected)); 4017 goto on_error; 4018 } 4019 #ifdef FEAT_FLOAT 4020 if (tv->v_type == VAR_FLOAT) 4021 tv->vval.v_float = -tv->vval.v_float; 4022 else 4023 #endif 4024 tv->vval.v_number = -tv->vval.v_number; 4025 break; 4026 4027 case ISN_CHECKNR: 4028 { 4029 int error = FALSE; 4030 4031 tv = STACK_TV_BOT(-1); 4032 SOURCING_LNUM = iptr->isn_lnum; 4033 if (check_not_string(tv) == FAIL) 4034 goto on_error; 4035 (void)tv_get_number_chk(tv, &error); 4036 if (error) 4037 goto on_error; 4038 } 4039 break; 4040 4041 case ISN_CHECKTYPE: 4042 { 4043 checktype_T *ct = &iptr->isn_arg.type; 4044 4045 tv = STACK_TV_BOT((int)ct->ct_off); 4046 SOURCING_LNUM = iptr->isn_lnum; 4047 if (!ectx->ec_where.wt_variable) 4048 ectx->ec_where.wt_index = ct->ct_arg_idx; 4049 if (check_typval_type(ct->ct_type, tv, ectx->ec_where) 4050 == FAIL) 4051 goto on_error; 4052 if (!ectx->ec_where.wt_variable) 4053 ectx->ec_where.wt_index = 0; 4054 4055 // number 0 is FALSE, number 1 is TRUE 4056 if (tv->v_type == VAR_NUMBER 4057 && ct->ct_type->tt_type == VAR_BOOL 4058 && (tv->vval.v_number == 0 4059 || tv->vval.v_number == 1)) 4060 { 4061 tv->v_type = VAR_BOOL; 4062 tv->vval.v_number = tv->vval.v_number 4063 ? VVAL_TRUE : VVAL_FALSE; 4064 } 4065 } 4066 break; 4067 4068 case ISN_CHECKLEN: 4069 { 4070 int min_len = iptr->isn_arg.checklen.cl_min_len; 4071 list_T *list = NULL; 4072 4073 tv = STACK_TV_BOT(-1); 4074 if (tv->v_type == VAR_LIST) 4075 list = tv->vval.v_list; 4076 if (list == NULL || list->lv_len < min_len 4077 || (list->lv_len > min_len 4078 && !iptr->isn_arg.checklen.cl_more_OK)) 4079 { 4080 SOURCING_LNUM = iptr->isn_lnum; 4081 semsg(_(e_expected_nr_items_but_got_nr), 4082 min_len, list == NULL ? 0 : list->lv_len); 4083 goto on_error; 4084 } 4085 } 4086 break; 4087 4088 case ISN_SETTYPE: 4089 { 4090 checktype_T *ct = &iptr->isn_arg.type; 4091 4092 tv = STACK_TV_BOT(-1); 4093 if (tv->v_type == VAR_DICT && tv->vval.v_dict != NULL) 4094 { 4095 free_type(tv->vval.v_dict->dv_type); 4096 tv->vval.v_dict->dv_type = alloc_type(ct->ct_type); 4097 } 4098 else if (tv->v_type == VAR_LIST && tv->vval.v_list != NULL) 4099 { 4100 free_type(tv->vval.v_list->lv_type); 4101 tv->vval.v_list->lv_type = alloc_type(ct->ct_type); 4102 } 4103 } 4104 break; 4105 4106 case ISN_2BOOL: 4107 case ISN_COND2BOOL: 4108 { 4109 int n; 4110 int error = FALSE; 4111 4112 if (iptr->isn_type == ISN_2BOOL) 4113 { 4114 tv = STACK_TV_BOT(iptr->isn_arg.tobool.offset); 4115 n = tv2bool(tv); 4116 if (iptr->isn_arg.tobool.invert) 4117 n = !n; 4118 } 4119 else 4120 { 4121 tv = STACK_TV_BOT(-1); 4122 SOURCING_LNUM = iptr->isn_lnum; 4123 n = tv_get_bool_chk(tv, &error); 4124 if (error) 4125 goto on_error; 4126 } 4127 clear_tv(tv); 4128 tv->v_type = VAR_BOOL; 4129 tv->vval.v_number = n ? VVAL_TRUE : VVAL_FALSE; 4130 } 4131 break; 4132 4133 case ISN_2STRING: 4134 case ISN_2STRING_ANY: 4135 SOURCING_LNUM = iptr->isn_lnum; 4136 if (do_2string(STACK_TV_BOT(iptr->isn_arg.tostring.offset), 4137 iptr->isn_type == ISN_2STRING_ANY, 4138 iptr->isn_arg.tostring.tolerant) == FAIL) 4139 goto on_error; 4140 break; 4141 4142 case ISN_RANGE: 4143 { 4144 exarg_T ea; 4145 char *errormsg; 4146 4147 ea.line2 = 0; 4148 ea.addr_count = 0; 4149 ea.addr_type = ADDR_LINES; 4150 ea.cmd = iptr->isn_arg.string; 4151 ea.skip = FALSE; 4152 if (parse_cmd_address(&ea, &errormsg, FALSE) == FAIL) 4153 goto on_error; 4154 4155 if (GA_GROW_FAILS(&ectx->ec_stack, 1)) 4156 goto theend; 4157 ++ectx->ec_stack.ga_len; 4158 tv = STACK_TV_BOT(-1); 4159 tv->v_type = VAR_NUMBER; 4160 tv->v_lock = 0; 4161 if (ea.addr_count == 0) 4162 tv->vval.v_number = curwin->w_cursor.lnum; 4163 else 4164 tv->vval.v_number = ea.line2; 4165 } 4166 break; 4167 4168 case ISN_PUT: 4169 { 4170 int regname = iptr->isn_arg.put.put_regname; 4171 linenr_T lnum = iptr->isn_arg.put.put_lnum; 4172 char_u *expr = NULL; 4173 int dir = FORWARD; 4174 4175 if (lnum < -2) 4176 { 4177 // line number was put on the stack by ISN_RANGE 4178 tv = STACK_TV_BOT(-1); 4179 curwin->w_cursor.lnum = tv->vval.v_number; 4180 if (lnum == LNUM_VARIABLE_RANGE_ABOVE) 4181 dir = BACKWARD; 4182 --ectx->ec_stack.ga_len; 4183 } 4184 else if (lnum == -2) 4185 // :put! above cursor 4186 dir = BACKWARD; 4187 else if (lnum >= 0) 4188 curwin->w_cursor.lnum = iptr->isn_arg.put.put_lnum; 4189 4190 if (regname == '=') 4191 { 4192 tv = STACK_TV_BOT(-1); 4193 if (tv->v_type == VAR_STRING) 4194 expr = tv->vval.v_string; 4195 else 4196 { 4197 expr = typval2string(tv, TRUE); // allocates value 4198 clear_tv(tv); 4199 } 4200 --ectx->ec_stack.ga_len; 4201 } 4202 check_cursor(); 4203 do_put(regname, expr, dir, 1L, PUT_LINE|PUT_CURSLINE); 4204 vim_free(expr); 4205 } 4206 break; 4207 4208 case ISN_CMDMOD: 4209 ectx->ec_funclocal.floc_save_cmdmod = cmdmod; 4210 ectx->ec_funclocal.floc_restore_cmdmod = TRUE; 4211 ectx->ec_funclocal.floc_restore_cmdmod_stacklen = 4212 ectx->ec_stack.ga_len; 4213 cmdmod = *iptr->isn_arg.cmdmod.cf_cmdmod; 4214 apply_cmdmod(&cmdmod); 4215 break; 4216 4217 case ISN_CMDMOD_REV: 4218 // filter regprog is owned by the instruction, don't free it 4219 cmdmod.cmod_filter_regmatch.regprog = NULL; 4220 undo_cmdmod(&cmdmod); 4221 cmdmod = ectx->ec_funclocal.floc_save_cmdmod; 4222 ectx->ec_funclocal.floc_restore_cmdmod = FALSE; 4223 break; 4224 4225 case ISN_UNPACK: 4226 { 4227 int count = iptr->isn_arg.unpack.unp_count; 4228 int semicolon = iptr->isn_arg.unpack.unp_semicolon; 4229 list_T *l; 4230 listitem_T *li; 4231 int i; 4232 4233 // Check there is a valid list to unpack. 4234 tv = STACK_TV_BOT(-1); 4235 if (tv->v_type != VAR_LIST) 4236 { 4237 SOURCING_LNUM = iptr->isn_lnum; 4238 emsg(_(e_for_argument_must_be_sequence_of_lists)); 4239 goto on_error; 4240 } 4241 l = tv->vval.v_list; 4242 if (l == NULL 4243 || l->lv_len < (semicolon ? count - 1 : count)) 4244 { 4245 SOURCING_LNUM = iptr->isn_lnum; 4246 emsg(_(e_list_value_does_not_have_enough_items)); 4247 goto on_error; 4248 } 4249 else if (!semicolon && l->lv_len > count) 4250 { 4251 SOURCING_LNUM = iptr->isn_lnum; 4252 emsg(_(e_list_value_has_more_items_than_targets)); 4253 goto on_error; 4254 } 4255 4256 CHECK_LIST_MATERIALIZE(l); 4257 if (GA_GROW_FAILS(&ectx->ec_stack, count - 1)) 4258 goto theend; 4259 ectx->ec_stack.ga_len += count - 1; 4260 4261 // Variable after semicolon gets a list with the remaining 4262 // items. 4263 if (semicolon) 4264 { 4265 list_T *rem_list = 4266 list_alloc_with_items(l->lv_len - count + 1); 4267 4268 if (rem_list == NULL) 4269 goto theend; 4270 tv = STACK_TV_BOT(-count); 4271 tv->vval.v_list = rem_list; 4272 ++rem_list->lv_refcount; 4273 tv->v_lock = 0; 4274 li = l->lv_first; 4275 for (i = 0; i < count - 1; ++i) 4276 li = li->li_next; 4277 for (i = 0; li != NULL; ++i) 4278 { 4279 list_set_item(rem_list, i, &li->li_tv); 4280 li = li->li_next; 4281 } 4282 --count; 4283 } 4284 4285 // Produce the values in reverse order, first item last. 4286 li = l->lv_first; 4287 for (i = 0; i < count; ++i) 4288 { 4289 tv = STACK_TV_BOT(-i - 1); 4290 copy_tv(&li->li_tv, tv); 4291 li = li->li_next; 4292 } 4293 4294 list_unref(l); 4295 } 4296 break; 4297 4298 case ISN_PROF_START: 4299 case ISN_PROF_END: 4300 { 4301 #ifdef FEAT_PROFILE 4302 funccall_T cookie; 4303 ufunc_T *cur_ufunc = 4304 (((dfunc_T *)def_functions.ga_data) 4305 + ectx->ec_dfunc_idx)->df_ufunc; 4306 4307 cookie.func = cur_ufunc; 4308 if (iptr->isn_type == ISN_PROF_START) 4309 { 4310 func_line_start(&cookie, iptr->isn_lnum); 4311 // if we get here the instruction is executed 4312 func_line_exec(&cookie); 4313 } 4314 else 4315 func_line_end(&cookie); 4316 #endif 4317 } 4318 break; 4319 4320 case ISN_DEBUG: 4321 handle_debug(iptr, ectx); 4322 break; 4323 4324 case ISN_SHUFFLE: 4325 { 4326 typval_T tmp_tv; 4327 int item = iptr->isn_arg.shuffle.shfl_item; 4328 int up = iptr->isn_arg.shuffle.shfl_up; 4329 4330 tmp_tv = *STACK_TV_BOT(-item); 4331 for ( ; up > 0 && item > 1; --up) 4332 { 4333 *STACK_TV_BOT(-item) = *STACK_TV_BOT(-item + 1); 4334 --item; 4335 } 4336 *STACK_TV_BOT(-item) = tmp_tv; 4337 } 4338 break; 4339 4340 case ISN_DROP: 4341 --ectx->ec_stack.ga_len; 4342 clear_tv(STACK_TV_BOT(0)); 4343 ectx->ec_where.wt_index = 0; 4344 ectx->ec_where.wt_variable = FALSE; 4345 break; 4346 } 4347 continue; 4348 4349 func_return: 4350 // Restore previous function. If the frame pointer is where we started 4351 // then there is none and we are done. 4352 if (ectx->ec_frame_idx == ectx->ec_initial_frame_idx) 4353 goto done; 4354 4355 if (func_return(ectx) == FAIL) 4356 // only fails when out of memory 4357 goto theend; 4358 continue; 4359 4360 on_error: 4361 // Jump here for an error that does not require aborting execution. 4362 // If "emsg_silent" is set then ignore the error, unless it was set 4363 // when calling the function. 4364 if (did_emsg_cumul + did_emsg == ectx->ec_did_emsg_before 4365 && emsg_silent && did_emsg_def == 0) 4366 { 4367 // If a sequence of instructions causes an error while ":silent!" 4368 // was used, restore the stack length and jump ahead to restoring 4369 // the cmdmod. 4370 if (ectx->ec_funclocal.floc_restore_cmdmod) 4371 { 4372 while (ectx->ec_stack.ga_len 4373 > ectx->ec_funclocal.floc_restore_cmdmod_stacklen) 4374 { 4375 --ectx->ec_stack.ga_len; 4376 clear_tv(STACK_TV_BOT(0)); 4377 } 4378 while (ectx->ec_instr[ectx->ec_iidx].isn_type != ISN_CMDMOD_REV) 4379 ++ectx->ec_iidx; 4380 } 4381 continue; 4382 } 4383 on_fatal_error: 4384 // Jump here for an error that messes up the stack. 4385 // If we are not inside a try-catch started here, abort execution. 4386 if (trylevel <= ectx->ec_trylevel_at_start) 4387 goto theend; 4388 } 4389 4390 done: 4391 ret = OK; 4392 theend: 4393 ectx->ec_trylevel_at_start = save_trylevel_at_start; 4394 return ret; 4395 } 4396 4397 /* 4398 * Execute the instructions from a VAR_INSTR typeval and put the result in 4399 * "rettv". 4400 * Return OK or FAIL. 4401 */ 4402 int 4403 exe_typval_instr(typval_T *tv, typval_T *rettv) 4404 { 4405 ectx_T *ectx = tv->vval.v_instr->instr_ectx; 4406 isn_T *save_instr = ectx->ec_instr; 4407 int save_iidx = ectx->ec_iidx; 4408 int res; 4409 4410 ectx->ec_instr = tv->vval.v_instr->instr_instr; 4411 res = exec_instructions(ectx); 4412 if (res == OK) 4413 { 4414 *rettv = *STACK_TV_BOT(-1); 4415 --ectx->ec_stack.ga_len; 4416 } 4417 4418 ectx->ec_instr = save_instr; 4419 ectx->ec_iidx = save_iidx; 4420 4421 return res; 4422 } 4423 4424 /* 4425 * Execute the instructions from an ISN_SUBSTITUTE command, which are in 4426 * "substitute_instr". 4427 */ 4428 char_u * 4429 exe_substitute_instr(void) 4430 { 4431 ectx_T *ectx = substitute_instr->subs_ectx; 4432 isn_T *save_instr = ectx->ec_instr; 4433 int save_iidx = ectx->ec_iidx; 4434 char_u *res; 4435 4436 ectx->ec_instr = substitute_instr->subs_instr; 4437 if (exec_instructions(ectx) == OK) 4438 { 4439 typval_T *tv = STACK_TV_BOT(-1); 4440 4441 res = typval2string(tv, TRUE); 4442 --ectx->ec_stack.ga_len; 4443 clear_tv(tv); 4444 } 4445 else 4446 { 4447 substitute_instr->subs_status = FAIL; 4448 res = vim_strsave((char_u *)""); 4449 } 4450 4451 ectx->ec_instr = save_instr; 4452 ectx->ec_iidx = save_iidx; 4453 4454 return res; 4455 } 4456 4457 /* 4458 * Call a "def" function from old Vim script. 4459 * Return OK or FAIL. 4460 */ 4461 int 4462 call_def_function( 4463 ufunc_T *ufunc, 4464 int argc_arg, // nr of arguments 4465 typval_T *argv, // arguments 4466 partial_T *partial, // optional partial for context 4467 typval_T *rettv) // return value 4468 { 4469 ectx_T ectx; // execution context 4470 int argc = argc_arg; 4471 typval_T *tv; 4472 int idx; 4473 int ret = FAIL; 4474 int defcount = ufunc->uf_args.ga_len - argc; 4475 sctx_T save_current_sctx = current_sctx; 4476 int did_emsg_before = did_emsg_cumul + did_emsg; 4477 int save_suppress_errthrow = suppress_errthrow; 4478 msglist_T **saved_msg_list = NULL; 4479 msglist_T *private_msg_list = NULL; 4480 int save_emsg_silent_def = emsg_silent_def; 4481 int save_did_emsg_def = did_emsg_def; 4482 int orig_funcdepth; 4483 int orig_nesting_level = ex_nesting_level; 4484 4485 // Get pointer to item in the stack. 4486 #undef STACK_TV 4487 #define STACK_TV(idx) (((typval_T *)ectx.ec_stack.ga_data) + idx) 4488 4489 // Get pointer to item at the bottom of the stack, -1 is the bottom. 4490 #undef STACK_TV_BOT 4491 #define STACK_TV_BOT(idx) (((typval_T *)ectx.ec_stack.ga_data) + ectx.ec_stack.ga_len + idx) 4492 4493 // Get pointer to a local variable on the stack. Negative for arguments. 4494 #undef STACK_TV_VAR 4495 #define STACK_TV_VAR(idx) (((typval_T *)ectx.ec_stack.ga_data) + ectx.ec_frame_idx + STACK_FRAME_SIZE + idx) 4496 4497 // Update uf_has_breakpoint if needed. 4498 update_has_breakpoint(ufunc); 4499 4500 if (ufunc->uf_def_status == UF_NOT_COMPILED 4501 || ufunc->uf_def_status == UF_COMPILE_ERROR 4502 || (func_needs_compiling(ufunc, COMPILE_TYPE(ufunc)) 4503 && compile_def_function(ufunc, FALSE, COMPILE_TYPE(ufunc), NULL) 4504 == FAIL)) 4505 { 4506 if (did_emsg_cumul + did_emsg == did_emsg_before) 4507 semsg(_(e_function_is_not_compiled_str), 4508 printable_func_name(ufunc)); 4509 return FAIL; 4510 } 4511 4512 { 4513 // Check the function was really compiled. 4514 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data) 4515 + ufunc->uf_dfunc_idx; 4516 if (INSTRUCTIONS(dfunc) == NULL) 4517 { 4518 iemsg("using call_def_function() on not compiled function"); 4519 return FAIL; 4520 } 4521 } 4522 4523 // If depth of calling is getting too high, don't execute the function. 4524 orig_funcdepth = funcdepth_get(); 4525 if (funcdepth_increment() == FAIL) 4526 return FAIL; 4527 4528 CLEAR_FIELD(ectx); 4529 ectx.ec_dfunc_idx = ufunc->uf_dfunc_idx; 4530 ga_init2(&ectx.ec_stack, sizeof(typval_T), 500); 4531 if (GA_GROW_FAILS(&ectx.ec_stack, 20)) 4532 { 4533 funcdepth_decrement(); 4534 return FAIL; 4535 } 4536 ga_init2(&ectx.ec_trystack, sizeof(trycmd_T), 10); 4537 ga_init2(&ectx.ec_funcrefs, sizeof(partial_T *), 10); 4538 ectx.ec_did_emsg_before = did_emsg_before; 4539 ++ex_nesting_level; 4540 4541 idx = argc - ufunc->uf_args.ga_len; 4542 if (idx > 0 && ufunc->uf_va_name == NULL) 4543 { 4544 if (idx == 1) 4545 emsg(_(e_one_argument_too_many)); 4546 else 4547 semsg(_(e_nr_arguments_too_many), idx); 4548 goto failed_early; 4549 } 4550 idx = argc - ufunc->uf_args.ga_len + ufunc->uf_def_args.ga_len; 4551 if (idx < 0) 4552 { 4553 if (idx == -1) 4554 emsg(_(e_one_argument_too_few)); 4555 else 4556 semsg(_(e_nr_arguments_too_few), -idx); 4557 goto failed_early; 4558 } 4559 4560 // Put arguments on the stack, but no more than what the function expects. 4561 // A lambda can be called with more arguments than it uses. 4562 for (idx = 0; idx < argc 4563 && (ufunc->uf_va_name != NULL || idx < ufunc->uf_args.ga_len); 4564 ++idx) 4565 { 4566 if (idx >= ufunc->uf_args.ga_len - ufunc->uf_def_args.ga_len 4567 && argv[idx].v_type == VAR_SPECIAL 4568 && argv[idx].vval.v_number == VVAL_NONE) 4569 { 4570 // Use the default value. 4571 STACK_TV_BOT(0)->v_type = VAR_UNKNOWN; 4572 } 4573 else 4574 { 4575 if (ufunc->uf_arg_types != NULL && idx < ufunc->uf_args.ga_len 4576 && check_typval_arg_type( 4577 ufunc->uf_arg_types[idx], &argv[idx], 4578 NULL, idx + 1) == FAIL) 4579 goto failed_early; 4580 copy_tv(&argv[idx], STACK_TV_BOT(0)); 4581 } 4582 ++ectx.ec_stack.ga_len; 4583 } 4584 4585 // Turn varargs into a list. Empty list if no args. 4586 if (ufunc->uf_va_name != NULL) 4587 { 4588 int vararg_count = argc - ufunc->uf_args.ga_len; 4589 4590 if (vararg_count < 0) 4591 vararg_count = 0; 4592 else 4593 argc -= vararg_count; 4594 if (exe_newlist(vararg_count, &ectx) == FAIL) 4595 goto failed_early; 4596 4597 // Check the type of the list items. 4598 tv = STACK_TV_BOT(-1); 4599 if (ufunc->uf_va_type != NULL 4600 && ufunc->uf_va_type != &t_list_any 4601 && ufunc->uf_va_type->tt_member != &t_any 4602 && tv->vval.v_list != NULL) 4603 { 4604 type_T *expected = ufunc->uf_va_type->tt_member; 4605 listitem_T *li = tv->vval.v_list->lv_first; 4606 4607 for (idx = 0; idx < vararg_count; ++idx) 4608 { 4609 if (check_typval_arg_type(expected, &li->li_tv, 4610 NULL, argc + idx + 1) == FAIL) 4611 goto failed_early; 4612 li = li->li_next; 4613 } 4614 } 4615 4616 if (defcount > 0) 4617 // Move varargs list to below missing default arguments. 4618 *STACK_TV_BOT(defcount - 1) = *STACK_TV_BOT(-1); 4619 --ectx.ec_stack.ga_len; 4620 } 4621 4622 // Make space for omitted arguments, will store default value below. 4623 // Any varargs list goes after them. 4624 if (defcount > 0) 4625 for (idx = 0; idx < defcount; ++idx) 4626 { 4627 STACK_TV_BOT(0)->v_type = VAR_UNKNOWN; 4628 ++ectx.ec_stack.ga_len; 4629 } 4630 if (ufunc->uf_va_name != NULL) 4631 ++ectx.ec_stack.ga_len; 4632 4633 // Frame pointer points to just after arguments. 4634 ectx.ec_frame_idx = ectx.ec_stack.ga_len; 4635 ectx.ec_initial_frame_idx = ectx.ec_frame_idx; 4636 4637 { 4638 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data) 4639 + ufunc->uf_dfunc_idx; 4640 ufunc_T *base_ufunc = dfunc->df_ufunc; 4641 4642 // "uf_partial" is on the ufunc that "df_ufunc" points to, as is done 4643 // by copy_func(). 4644 if (partial != NULL || base_ufunc->uf_partial != NULL) 4645 { 4646 ectx.ec_outer_ref = ALLOC_CLEAR_ONE(outer_ref_T); 4647 if (ectx.ec_outer_ref == NULL) 4648 goto failed_early; 4649 if (partial != NULL) 4650 { 4651 if (partial->pt_outer.out_stack == NULL && current_ectx != NULL) 4652 { 4653 if (current_ectx->ec_outer_ref != NULL 4654 && current_ectx->ec_outer_ref->or_outer != NULL) 4655 ectx.ec_outer_ref->or_outer = 4656 current_ectx->ec_outer_ref->or_outer; 4657 } 4658 else 4659 { 4660 ectx.ec_outer_ref->or_outer = &partial->pt_outer; 4661 ++partial->pt_refcount; 4662 ectx.ec_outer_ref->or_partial = partial; 4663 } 4664 } 4665 else 4666 { 4667 ectx.ec_outer_ref->or_outer = &base_ufunc->uf_partial->pt_outer; 4668 ++base_ufunc->uf_partial->pt_refcount; 4669 ectx.ec_outer_ref->or_partial = base_ufunc->uf_partial; 4670 } 4671 } 4672 } 4673 4674 // dummy frame entries 4675 for (idx = 0; idx < STACK_FRAME_SIZE; ++idx) 4676 { 4677 STACK_TV(ectx.ec_stack.ga_len)->v_type = VAR_UNKNOWN; 4678 ++ectx.ec_stack.ga_len; 4679 } 4680 4681 { 4682 // Reserve space for local variables and any closure reference count. 4683 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data) 4684 + ufunc->uf_dfunc_idx; 4685 4686 for (idx = 0; idx < dfunc->df_varcount; ++idx) 4687 STACK_TV_VAR(idx)->v_type = VAR_UNKNOWN; 4688 ectx.ec_stack.ga_len += dfunc->df_varcount; 4689 if (dfunc->df_has_closure) 4690 { 4691 STACK_TV_VAR(idx)->v_type = VAR_NUMBER; 4692 STACK_TV_VAR(idx)->vval.v_number = 0; 4693 ++ectx.ec_stack.ga_len; 4694 } 4695 4696 ectx.ec_instr = INSTRUCTIONS(dfunc); 4697 } 4698 4699 // Following errors are in the function, not the caller. 4700 // Commands behave like vim9script. 4701 estack_push_ufunc(ufunc, 1); 4702 current_sctx = ufunc->uf_script_ctx; 4703 current_sctx.sc_version = SCRIPT_VERSION_VIM9; 4704 4705 // Use a specific location for storing error messages to be converted to an 4706 // exception. 4707 saved_msg_list = msg_list; 4708 msg_list = &private_msg_list; 4709 4710 // Do turn errors into exceptions. 4711 suppress_errthrow = FALSE; 4712 4713 // Do not delete the function while executing it. 4714 ++ufunc->uf_calls; 4715 4716 // When ":silent!" was used before calling then we still abort the 4717 // function. If ":silent!" is used in the function then we don't. 4718 emsg_silent_def = emsg_silent; 4719 did_emsg_def = 0; 4720 4721 ectx.ec_where.wt_index = 0; 4722 ectx.ec_where.wt_variable = FALSE; 4723 4724 // Execute the instructions until done. 4725 ret = exec_instructions(&ectx); 4726 if (ret == OK) 4727 { 4728 // function finished, get result from the stack. 4729 if (ufunc->uf_ret_type == &t_void) 4730 { 4731 rettv->v_type = VAR_VOID; 4732 } 4733 else 4734 { 4735 tv = STACK_TV_BOT(-1); 4736 *rettv = *tv; 4737 tv->v_type = VAR_UNKNOWN; 4738 } 4739 } 4740 4741 // When failed need to unwind the call stack. 4742 while (ectx.ec_frame_idx != ectx.ec_initial_frame_idx) 4743 func_return(&ectx); 4744 4745 // Deal with any remaining closures, they may be in use somewhere. 4746 if (ectx.ec_funcrefs.ga_len > 0) 4747 { 4748 handle_closure_in_use(&ectx, FALSE); 4749 ga_clear(&ectx.ec_funcrefs); // TODO: should not be needed? 4750 } 4751 4752 estack_pop(); 4753 current_sctx = save_current_sctx; 4754 4755 // TODO: when is it safe to delete the function if it is no longer used? 4756 --ufunc->uf_calls; 4757 4758 if (*msg_list != NULL && saved_msg_list != NULL) 4759 { 4760 msglist_T **plist = saved_msg_list; 4761 4762 // Append entries from the current msg_list (uncaught exceptions) to 4763 // the saved msg_list. 4764 while (*plist != NULL) 4765 plist = &(*plist)->next; 4766 4767 *plist = *msg_list; 4768 } 4769 msg_list = saved_msg_list; 4770 4771 if (ectx.ec_funclocal.floc_restore_cmdmod) 4772 { 4773 cmdmod.cmod_filter_regmatch.regprog = NULL; 4774 undo_cmdmod(&cmdmod); 4775 cmdmod = ectx.ec_funclocal.floc_save_cmdmod; 4776 } 4777 emsg_silent_def = save_emsg_silent_def; 4778 did_emsg_def += save_did_emsg_def; 4779 4780 failed_early: 4781 // Free all local variables, but not arguments. 4782 for (idx = 0; idx < ectx.ec_stack.ga_len; ++idx) 4783 clear_tv(STACK_TV(idx)); 4784 ex_nesting_level = orig_nesting_level; 4785 4786 vim_free(ectx.ec_stack.ga_data); 4787 vim_free(ectx.ec_trystack.ga_data); 4788 if (ectx.ec_outer_ref != NULL) 4789 { 4790 if (ectx.ec_outer_ref->or_outer_allocated) 4791 vim_free(ectx.ec_outer_ref->or_outer); 4792 partial_unref(ectx.ec_outer_ref->or_partial); 4793 vim_free(ectx.ec_outer_ref); 4794 } 4795 4796 // Not sure if this is necessary. 4797 suppress_errthrow = save_suppress_errthrow; 4798 4799 if (ret != OK && did_emsg_cumul + did_emsg == did_emsg_before 4800 && !need_rethrow) 4801 semsg(_(e_unknown_error_while_executing_str), 4802 printable_func_name(ufunc)); 4803 funcdepth_restore(orig_funcdepth); 4804 return ret; 4805 } 4806 4807 /* 4808 * List instructions "instr" up to "instr_count" or until ISN_FINISH. 4809 * "ufunc" has the source lines, NULL for the instructions of ISN_SUBSTITUTE. 4810 * "pfx" is prefixed to every line. 4811 */ 4812 static void 4813 list_instructions(char *pfx, isn_T *instr, int instr_count, ufunc_T *ufunc) 4814 { 4815 int line_idx = 0; 4816 int prev_current = 0; 4817 int current; 4818 int def_arg_idx = 0; 4819 4820 for (current = 0; current < instr_count; ++current) 4821 { 4822 isn_T *iptr = &instr[current]; 4823 char *line; 4824 4825 if (ufunc != NULL) 4826 { 4827 while (line_idx < iptr->isn_lnum 4828 && line_idx < ufunc->uf_lines.ga_len) 4829 { 4830 if (current > prev_current) 4831 { 4832 msg_puts("\n\n"); 4833 prev_current = current; 4834 } 4835 line = ((char **)ufunc->uf_lines.ga_data)[line_idx++]; 4836 if (line != NULL) 4837 msg(line); 4838 } 4839 if (iptr->isn_type == ISN_JUMP_IF_ARG_SET) 4840 { 4841 int first_def_arg = ufunc->uf_args.ga_len 4842 - ufunc->uf_def_args.ga_len; 4843 4844 if (def_arg_idx > 0) 4845 msg_puts("\n\n"); 4846 msg_start(); 4847 msg_puts(" "); 4848 msg_puts(((char **)(ufunc->uf_args.ga_data))[ 4849 first_def_arg + def_arg_idx]); 4850 msg_puts(" = "); 4851 msg_puts(((char **)(ufunc->uf_def_args.ga_data))[def_arg_idx++]); 4852 msg_clr_eos(); 4853 msg_end(); 4854 } 4855 } 4856 4857 switch (iptr->isn_type) 4858 { 4859 case ISN_EXEC: 4860 smsg("%s%4d EXEC %s", pfx, current, iptr->isn_arg.string); 4861 break; 4862 case ISN_EXEC_SPLIT: 4863 smsg("%s%4d EXEC_SPLIT %s", pfx, current, iptr->isn_arg.string); 4864 break; 4865 case ISN_LEGACY_EVAL: 4866 smsg("%s%4d EVAL legacy %s", pfx, current, 4867 iptr->isn_arg.string); 4868 break; 4869 case ISN_REDIRSTART: 4870 smsg("%s%4d REDIR", pfx, current); 4871 break; 4872 case ISN_REDIREND: 4873 smsg("%s%4d REDIR END%s", pfx, current, 4874 iptr->isn_arg.number ? " append" : ""); 4875 break; 4876 case ISN_CEXPR_AUCMD: 4877 #ifdef FEAT_QUICKFIX 4878 smsg("%s%4d CEXPR pre %s", pfx, current, 4879 cexpr_get_auname(iptr->isn_arg.number)); 4880 #endif 4881 break; 4882 case ISN_CEXPR_CORE: 4883 #ifdef FEAT_QUICKFIX 4884 { 4885 cexprref_T *cer = iptr->isn_arg.cexpr.cexpr_ref; 4886 4887 smsg("%s%4d CEXPR core %s%s \"%s\"", pfx, current, 4888 cexpr_get_auname(cer->cer_cmdidx), 4889 cer->cer_forceit ? "!" : "", 4890 cer->cer_cmdline); 4891 } 4892 #endif 4893 break; 4894 case ISN_INSTR: 4895 { 4896 smsg("%s%4d INSTR", pfx, current); 4897 list_instructions(" ", iptr->isn_arg.instr, 4898 INT_MAX, NULL); 4899 msg(" -------------"); 4900 } 4901 break; 4902 case ISN_SUBSTITUTE: 4903 { 4904 subs_T *subs = &iptr->isn_arg.subs; 4905 4906 smsg("%s%4d SUBSTITUTE %s", pfx, current, subs->subs_cmd); 4907 list_instructions(" ", subs->subs_instr, INT_MAX, NULL); 4908 msg(" -------------"); 4909 } 4910 break; 4911 case ISN_EXECCONCAT: 4912 smsg("%s%4d EXECCONCAT %lld", pfx, current, 4913 (varnumber_T)iptr->isn_arg.number); 4914 break; 4915 case ISN_ECHO: 4916 { 4917 echo_T *echo = &iptr->isn_arg.echo; 4918 4919 smsg("%s%4d %s %d", pfx, current, 4920 echo->echo_with_white ? "ECHO" : "ECHON", 4921 echo->echo_count); 4922 } 4923 break; 4924 case ISN_EXECUTE: 4925 smsg("%s%4d EXECUTE %lld", pfx, current, 4926 (varnumber_T)(iptr->isn_arg.number)); 4927 break; 4928 case ISN_ECHOMSG: 4929 smsg("%s%4d ECHOMSG %lld", pfx, current, 4930 (varnumber_T)(iptr->isn_arg.number)); 4931 break; 4932 case ISN_ECHOCONSOLE: 4933 smsg("%s%4d ECHOCONSOLE %lld", pfx, current, 4934 (varnumber_T)(iptr->isn_arg.number)); 4935 break; 4936 case ISN_ECHOERR: 4937 smsg("%s%4d ECHOERR %lld", pfx, current, 4938 (varnumber_T)(iptr->isn_arg.number)); 4939 break; 4940 case ISN_LOAD: 4941 { 4942 if (iptr->isn_arg.number < 0) 4943 smsg("%s%4d LOAD arg[%lld]", pfx, current, 4944 (varnumber_T)(iptr->isn_arg.number 4945 + STACK_FRAME_SIZE)); 4946 else 4947 smsg("%s%4d LOAD $%lld", pfx, current, 4948 (varnumber_T)(iptr->isn_arg.number)); 4949 } 4950 break; 4951 case ISN_LOADOUTER: 4952 { 4953 if (iptr->isn_arg.number < 0) 4954 smsg("%s%4d LOADOUTER level %d arg[%d]", pfx, current, 4955 iptr->isn_arg.outer.outer_depth, 4956 iptr->isn_arg.outer.outer_idx 4957 + STACK_FRAME_SIZE); 4958 else 4959 smsg("%s%4d LOADOUTER level %d $%d", pfx, current, 4960 iptr->isn_arg.outer.outer_depth, 4961 iptr->isn_arg.outer.outer_idx); 4962 } 4963 break; 4964 case ISN_LOADV: 4965 smsg("%s%4d LOADV v:%s", pfx, current, 4966 get_vim_var_name(iptr->isn_arg.number)); 4967 break; 4968 case ISN_LOADSCRIPT: 4969 { 4970 scriptref_T *sref = iptr->isn_arg.script.scriptref; 4971 scriptitem_T *si = SCRIPT_ITEM(sref->sref_sid); 4972 svar_T *sv; 4973 4974 sv = get_script_svar(sref, -1); 4975 if (sv == NULL) 4976 smsg("%s%4d LOADSCRIPT [deleted] from %s", 4977 pfx, current, si->sn_name); 4978 else 4979 smsg("%s%4d LOADSCRIPT %s-%d from %s", pfx, current, 4980 sv->sv_name, 4981 sref->sref_idx, 4982 si->sn_name); 4983 } 4984 break; 4985 case ISN_LOADS: 4986 { 4987 scriptitem_T *si = SCRIPT_ITEM( 4988 iptr->isn_arg.loadstore.ls_sid); 4989 4990 smsg("%s%4d LOADS s:%s from %s", pfx, current, 4991 iptr->isn_arg.loadstore.ls_name, si->sn_name); 4992 } 4993 break; 4994 case ISN_LOADAUTO: 4995 smsg("%s%4d LOADAUTO %s", pfx, current, iptr->isn_arg.string); 4996 break; 4997 case ISN_LOADG: 4998 smsg("%s%4d LOADG g:%s", pfx, current, iptr->isn_arg.string); 4999 break; 5000 case ISN_LOADB: 5001 smsg("%s%4d LOADB b:%s", pfx, current, iptr->isn_arg.string); 5002 break; 5003 case ISN_LOADW: 5004 smsg("%s%4d LOADW w:%s", pfx, current, iptr->isn_arg.string); 5005 break; 5006 case ISN_LOADT: 5007 smsg("%s%4d LOADT t:%s", pfx, current, iptr->isn_arg.string); 5008 break; 5009 case ISN_LOADGDICT: 5010 smsg("%s%4d LOAD g:", pfx, current); 5011 break; 5012 case ISN_LOADBDICT: 5013 smsg("%s%4d LOAD b:", pfx, current); 5014 break; 5015 case ISN_LOADWDICT: 5016 smsg("%s%4d LOAD w:", pfx, current); 5017 break; 5018 case ISN_LOADTDICT: 5019 smsg("%s%4d LOAD t:", pfx, current); 5020 break; 5021 case ISN_LOADOPT: 5022 smsg("%s%4d LOADOPT %s", pfx, current, iptr->isn_arg.string); 5023 break; 5024 case ISN_LOADENV: 5025 smsg("%s%4d LOADENV %s", pfx, current, iptr->isn_arg.string); 5026 break; 5027 case ISN_LOADREG: 5028 smsg("%s%4d LOADREG @%c", pfx, current, 5029 (int)(iptr->isn_arg.number)); 5030 break; 5031 5032 case ISN_STORE: 5033 if (iptr->isn_arg.number < 0) 5034 smsg("%s%4d STORE arg[%lld]", pfx, current, 5035 iptr->isn_arg.number + STACK_FRAME_SIZE); 5036 else 5037 smsg("%s%4d STORE $%lld", pfx, current, 5038 iptr->isn_arg.number); 5039 break; 5040 case ISN_STOREOUTER: 5041 { 5042 if (iptr->isn_arg.number < 0) 5043 smsg("%s%4d STOREOUTEr level %d arg[%d]", pfx, current, 5044 iptr->isn_arg.outer.outer_depth, 5045 iptr->isn_arg.outer.outer_idx + STACK_FRAME_SIZE); 5046 else 5047 smsg("%s%4d STOREOUTER level %d $%d", pfx, current, 5048 iptr->isn_arg.outer.outer_depth, 5049 iptr->isn_arg.outer.outer_idx); 5050 } 5051 break; 5052 case ISN_STOREV: 5053 smsg("%s%4d STOREV v:%s", pfx, current, 5054 get_vim_var_name(iptr->isn_arg.number)); 5055 break; 5056 case ISN_STOREAUTO: 5057 smsg("%s%4d STOREAUTO %s", pfx, current, iptr->isn_arg.string); 5058 break; 5059 case ISN_STOREG: 5060 smsg("%s%4d STOREG %s", pfx, current, iptr->isn_arg.string); 5061 break; 5062 case ISN_STOREB: 5063 smsg("%s%4d STOREB %s", pfx, current, iptr->isn_arg.string); 5064 break; 5065 case ISN_STOREW: 5066 smsg("%s%4d STOREW %s", pfx, current, iptr->isn_arg.string); 5067 break; 5068 case ISN_STORET: 5069 smsg("%s%4d STORET %s", pfx, current, iptr->isn_arg.string); 5070 break; 5071 case ISN_STORES: 5072 { 5073 scriptitem_T *si = SCRIPT_ITEM( 5074 iptr->isn_arg.loadstore.ls_sid); 5075 5076 smsg("%s%4d STORES %s in %s", pfx, current, 5077 iptr->isn_arg.loadstore.ls_name, si->sn_name); 5078 } 5079 break; 5080 case ISN_STORESCRIPT: 5081 { 5082 scriptref_T *sref = iptr->isn_arg.script.scriptref; 5083 scriptitem_T *si = SCRIPT_ITEM(sref->sref_sid); 5084 svar_T *sv; 5085 5086 sv = get_script_svar(sref, -1); 5087 if (sv == NULL) 5088 smsg("%s%4d STORESCRIPT [deleted] in %s", 5089 pfx, current, si->sn_name); 5090 else 5091 smsg("%s%4d STORESCRIPT %s-%d in %s", pfx, current, 5092 sv->sv_name, 5093 sref->sref_idx, 5094 si->sn_name); 5095 } 5096 break; 5097 case ISN_STOREOPT: 5098 smsg("%s%4d STOREOPT &%s", pfx, current, 5099 iptr->isn_arg.storeopt.so_name); 5100 break; 5101 case ISN_STOREENV: 5102 smsg("%s%4d STOREENV $%s", pfx, current, iptr->isn_arg.string); 5103 break; 5104 case ISN_STOREREG: 5105 smsg("%s%4d STOREREG @%c", pfx, current, 5106 (int)iptr->isn_arg.number); 5107 break; 5108 case ISN_STORENR: 5109 smsg("%s%4d STORE %lld in $%d", pfx, current, 5110 iptr->isn_arg.storenr.stnr_val, 5111 iptr->isn_arg.storenr.stnr_idx); 5112 break; 5113 5114 case ISN_STOREINDEX: 5115 smsg("%s%4d STOREINDEX %s", pfx, current, 5116 vartype_name(iptr->isn_arg.vartype)); 5117 break; 5118 5119 case ISN_STORERANGE: 5120 smsg("%s%4d STORERANGE", pfx, current); 5121 break; 5122 5123 // constants 5124 case ISN_PUSHNR: 5125 smsg("%s%4d PUSHNR %lld", pfx, current, 5126 (varnumber_T)(iptr->isn_arg.number)); 5127 break; 5128 case ISN_PUSHBOOL: 5129 case ISN_PUSHSPEC: 5130 smsg("%s%4d PUSH %s", pfx, current, 5131 get_var_special_name(iptr->isn_arg.number)); 5132 break; 5133 case ISN_PUSHF: 5134 #ifdef FEAT_FLOAT 5135 smsg("%s%4d PUSHF %g", pfx, current, iptr->isn_arg.fnumber); 5136 #endif 5137 break; 5138 case ISN_PUSHS: 5139 smsg("%s%4d PUSHS \"%s\"", pfx, current, iptr->isn_arg.string); 5140 break; 5141 case ISN_PUSHBLOB: 5142 { 5143 char_u *r; 5144 char_u numbuf[NUMBUFLEN]; 5145 char_u *tofree; 5146 5147 r = blob2string(iptr->isn_arg.blob, &tofree, numbuf); 5148 smsg("%s%4d PUSHBLOB %s", pfx, current, r); 5149 vim_free(tofree); 5150 } 5151 break; 5152 case ISN_PUSHFUNC: 5153 { 5154 char *name = (char *)iptr->isn_arg.string; 5155 5156 smsg("%s%4d PUSHFUNC \"%s\"", pfx, current, 5157 name == NULL ? "[none]" : name); 5158 } 5159 break; 5160 case ISN_PUSHCHANNEL: 5161 #ifdef FEAT_JOB_CHANNEL 5162 { 5163 channel_T *channel = iptr->isn_arg.channel; 5164 5165 smsg("%s%4d PUSHCHANNEL %d", pfx, current, 5166 channel == NULL ? 0 : channel->ch_id); 5167 } 5168 #endif 5169 break; 5170 case ISN_PUSHJOB: 5171 #ifdef FEAT_JOB_CHANNEL 5172 { 5173 typval_T tv; 5174 char_u *name; 5175 char_u buf[NUMBUFLEN]; 5176 5177 tv.v_type = VAR_JOB; 5178 tv.vval.v_job = iptr->isn_arg.job; 5179 name = job_to_string_buf(&tv, buf); 5180 smsg("%s%4d PUSHJOB \"%s\"", pfx, current, name); 5181 } 5182 #endif 5183 break; 5184 case ISN_PUSHEXC: 5185 smsg("%s%4d PUSH v:exception", pfx, current); 5186 break; 5187 case ISN_UNLET: 5188 smsg("%s%4d UNLET%s %s", pfx, current, 5189 iptr->isn_arg.unlet.ul_forceit ? "!" : "", 5190 iptr->isn_arg.unlet.ul_name); 5191 break; 5192 case ISN_UNLETENV: 5193 smsg("%s%4d UNLETENV%s $%s", pfx, current, 5194 iptr->isn_arg.unlet.ul_forceit ? "!" : "", 5195 iptr->isn_arg.unlet.ul_name); 5196 break; 5197 case ISN_UNLETINDEX: 5198 smsg("%s%4d UNLETINDEX", pfx, current); 5199 break; 5200 case ISN_UNLETRANGE: 5201 smsg("%s%4d UNLETRANGE", pfx, current); 5202 break; 5203 case ISN_LOCKCONST: 5204 smsg("%s%4d LOCKCONST", pfx, current); 5205 break; 5206 case ISN_NEWLIST: 5207 smsg("%s%4d NEWLIST size %lld", pfx, current, 5208 (varnumber_T)(iptr->isn_arg.number)); 5209 break; 5210 case ISN_NEWDICT: 5211 smsg("%s%4d NEWDICT size %lld", pfx, current, 5212 (varnumber_T)(iptr->isn_arg.number)); 5213 break; 5214 5215 // function call 5216 case ISN_BCALL: 5217 { 5218 cbfunc_T *cbfunc = &iptr->isn_arg.bfunc; 5219 5220 smsg("%s%4d BCALL %s(argc %d)", pfx, current, 5221 internal_func_name(cbfunc->cbf_idx), 5222 cbfunc->cbf_argcount); 5223 } 5224 break; 5225 case ISN_DCALL: 5226 { 5227 cdfunc_T *cdfunc = &iptr->isn_arg.dfunc; 5228 dfunc_T *df = ((dfunc_T *)def_functions.ga_data) 5229 + cdfunc->cdf_idx; 5230 5231 smsg("%s%4d DCALL %s(argc %d)", pfx, current, 5232 printable_func_name(df->df_ufunc), 5233 cdfunc->cdf_argcount); 5234 } 5235 break; 5236 case ISN_UCALL: 5237 { 5238 cufunc_T *cufunc = &iptr->isn_arg.ufunc; 5239 5240 smsg("%s%4d UCALL %s(argc %d)", pfx, current, 5241 cufunc->cuf_name, cufunc->cuf_argcount); 5242 } 5243 break; 5244 case ISN_PCALL: 5245 { 5246 cpfunc_T *cpfunc = &iptr->isn_arg.pfunc; 5247 5248 smsg("%s%4d PCALL%s (argc %d)", pfx, current, 5249 cpfunc->cpf_top ? " top" : "", cpfunc->cpf_argcount); 5250 } 5251 break; 5252 case ISN_PCALL_END: 5253 smsg("%s%4d PCALL end", pfx, current); 5254 break; 5255 case ISN_RETURN: 5256 smsg("%s%4d RETURN", pfx, current); 5257 break; 5258 case ISN_RETURN_VOID: 5259 smsg("%s%4d RETURN void", pfx, current); 5260 break; 5261 case ISN_FUNCREF: 5262 { 5263 funcref_T *funcref = &iptr->isn_arg.funcref; 5264 dfunc_T *df = ((dfunc_T *)def_functions.ga_data) 5265 + funcref->fr_func; 5266 5267 smsg("%s%4d FUNCREF %s", pfx, current, df->df_ufunc->uf_name); 5268 } 5269 break; 5270 5271 case ISN_NEWFUNC: 5272 { 5273 newfunc_T *newfunc = &iptr->isn_arg.newfunc; 5274 5275 smsg("%s%4d NEWFUNC %s %s", pfx, current, 5276 newfunc->nf_lambda, newfunc->nf_global); 5277 } 5278 break; 5279 5280 case ISN_DEF: 5281 { 5282 char_u *name = iptr->isn_arg.string; 5283 5284 smsg("%s%4d DEF %s", pfx, current, 5285 name == NULL ? (char_u *)"" : name); 5286 } 5287 break; 5288 5289 case ISN_JUMP: 5290 { 5291 char *when = "?"; 5292 5293 switch (iptr->isn_arg.jump.jump_when) 5294 { 5295 case JUMP_ALWAYS: 5296 when = "JUMP"; 5297 break; 5298 case JUMP_AND_KEEP_IF_TRUE: 5299 when = "JUMP_AND_KEEP_IF_TRUE"; 5300 break; 5301 case JUMP_IF_FALSE: 5302 when = "JUMP_IF_FALSE"; 5303 break; 5304 case JUMP_AND_KEEP_IF_FALSE: 5305 when = "JUMP_AND_KEEP_IF_FALSE"; 5306 break; 5307 case JUMP_IF_COND_FALSE: 5308 when = "JUMP_IF_COND_FALSE"; 5309 break; 5310 case JUMP_IF_COND_TRUE: 5311 when = "JUMP_IF_COND_TRUE"; 5312 break; 5313 } 5314 smsg("%s%4d %s -> %d", pfx, current, when, 5315 iptr->isn_arg.jump.jump_where); 5316 } 5317 break; 5318 5319 case ISN_JUMP_IF_ARG_SET: 5320 smsg("%s%4d JUMP_IF_ARG_SET arg[%d] -> %d", pfx, current, 5321 iptr->isn_arg.jumparg.jump_arg_off + STACK_FRAME_SIZE, 5322 iptr->isn_arg.jump.jump_where); 5323 break; 5324 5325 case ISN_FOR: 5326 { 5327 forloop_T *forloop = &iptr->isn_arg.forloop; 5328 5329 smsg("%s%4d FOR $%d -> %d", pfx, current, 5330 forloop->for_idx, forloop->for_end); 5331 } 5332 break; 5333 5334 case ISN_TRY: 5335 { 5336 try_T *try = &iptr->isn_arg.try; 5337 5338 if (try->try_ref->try_finally == 0) 5339 smsg("%s%4d TRY catch -> %d, endtry -> %d", 5340 pfx, current, 5341 try->try_ref->try_catch, 5342 try->try_ref->try_endtry); 5343 else 5344 smsg("%s%4d TRY catch -> %d, finally -> %d, endtry -> %d", 5345 pfx, current, 5346 try->try_ref->try_catch, 5347 try->try_ref->try_finally, 5348 try->try_ref->try_endtry); 5349 } 5350 break; 5351 case ISN_CATCH: 5352 // TODO 5353 smsg("%s%4d CATCH", pfx, current); 5354 break; 5355 case ISN_TRYCONT: 5356 { 5357 trycont_T *trycont = &iptr->isn_arg.trycont; 5358 5359 smsg("%s%4d TRY-CONTINUE %d level%s -> %d", pfx, current, 5360 trycont->tct_levels, 5361 trycont->tct_levels == 1 ? "" : "s", 5362 trycont->tct_where); 5363 } 5364 break; 5365 case ISN_FINALLY: 5366 smsg("%s%4d FINALLY", pfx, current); 5367 break; 5368 case ISN_ENDTRY: 5369 smsg("%s%4d ENDTRY", pfx, current); 5370 break; 5371 case ISN_THROW: 5372 smsg("%s%4d THROW", pfx, current); 5373 break; 5374 5375 // expression operations on number 5376 case ISN_OPNR: 5377 case ISN_OPFLOAT: 5378 case ISN_OPANY: 5379 { 5380 char *what; 5381 char *ins; 5382 5383 switch (iptr->isn_arg.op.op_type) 5384 { 5385 case EXPR_MULT: what = "*"; break; 5386 case EXPR_DIV: what = "/"; break; 5387 case EXPR_REM: what = "%"; break; 5388 case EXPR_SUB: what = "-"; break; 5389 case EXPR_ADD: what = "+"; break; 5390 default: what = "???"; break; 5391 } 5392 switch (iptr->isn_type) 5393 { 5394 case ISN_OPNR: ins = "OPNR"; break; 5395 case ISN_OPFLOAT: ins = "OPFLOAT"; break; 5396 case ISN_OPANY: ins = "OPANY"; break; 5397 default: ins = "???"; break; 5398 } 5399 smsg("%s%4d %s %s", pfx, current, ins, what); 5400 } 5401 break; 5402 5403 case ISN_COMPAREBOOL: 5404 case ISN_COMPARESPECIAL: 5405 case ISN_COMPARENR: 5406 case ISN_COMPAREFLOAT: 5407 case ISN_COMPARESTRING: 5408 case ISN_COMPAREBLOB: 5409 case ISN_COMPARELIST: 5410 case ISN_COMPAREDICT: 5411 case ISN_COMPAREFUNC: 5412 case ISN_COMPAREANY: 5413 { 5414 char *p; 5415 char buf[10]; 5416 char *type; 5417 5418 switch (iptr->isn_arg.op.op_type) 5419 { 5420 case EXPR_EQUAL: p = "=="; break; 5421 case EXPR_NEQUAL: p = "!="; break; 5422 case EXPR_GREATER: p = ">"; break; 5423 case EXPR_GEQUAL: p = ">="; break; 5424 case EXPR_SMALLER: p = "<"; break; 5425 case EXPR_SEQUAL: p = "<="; break; 5426 case EXPR_MATCH: p = "=~"; break; 5427 case EXPR_IS: p = "is"; break; 5428 case EXPR_ISNOT: p = "isnot"; break; 5429 case EXPR_NOMATCH: p = "!~"; break; 5430 default: p = "???"; break; 5431 } 5432 STRCPY(buf, p); 5433 if (iptr->isn_arg.op.op_ic == TRUE) 5434 strcat(buf, "?"); 5435 switch(iptr->isn_type) 5436 { 5437 case ISN_COMPAREBOOL: type = "COMPAREBOOL"; break; 5438 case ISN_COMPARESPECIAL: 5439 type = "COMPARESPECIAL"; break; 5440 case ISN_COMPARENR: type = "COMPARENR"; break; 5441 case ISN_COMPAREFLOAT: type = "COMPAREFLOAT"; break; 5442 case ISN_COMPARESTRING: 5443 type = "COMPARESTRING"; break; 5444 case ISN_COMPAREBLOB: type = "COMPAREBLOB"; break; 5445 case ISN_COMPARELIST: type = "COMPARELIST"; break; 5446 case ISN_COMPAREDICT: type = "COMPAREDICT"; break; 5447 case ISN_COMPAREFUNC: type = "COMPAREFUNC"; break; 5448 case ISN_COMPAREANY: type = "COMPAREANY"; break; 5449 default: type = "???"; break; 5450 } 5451 5452 smsg("%s%4d %s %s", pfx, current, type, buf); 5453 } 5454 break; 5455 5456 case ISN_ADDLIST: smsg("%s%4d ADDLIST", pfx, current); break; 5457 case ISN_ADDBLOB: smsg("%s%4d ADDBLOB", pfx, current); break; 5458 5459 // expression operations 5460 case ISN_CONCAT: smsg("%s%4d CONCAT", pfx, current); break; 5461 case ISN_STRINDEX: smsg("%s%4d STRINDEX", pfx, current); break; 5462 case ISN_STRSLICE: smsg("%s%4d STRSLICE", pfx, current); break; 5463 case ISN_BLOBINDEX: smsg("%s%4d BLOBINDEX", pfx, current); break; 5464 case ISN_BLOBSLICE: smsg("%s%4d BLOBSLICE", pfx, current); break; 5465 case ISN_LISTAPPEND: smsg("%s%4d LISTAPPEND", pfx, current); break; 5466 case ISN_BLOBAPPEND: smsg("%s%4d BLOBAPPEND", pfx, current); break; 5467 case ISN_LISTINDEX: smsg("%s%4d LISTINDEX", pfx, current); break; 5468 case ISN_LISTSLICE: smsg("%s%4d LISTSLICE", pfx, current); break; 5469 case ISN_ANYINDEX: smsg("%s%4d ANYINDEX", pfx, current); break; 5470 case ISN_ANYSLICE: smsg("%s%4d ANYSLICE", pfx, current); break; 5471 case ISN_SLICE: smsg("%s%4d SLICE %lld", 5472 pfx, current, iptr->isn_arg.number); break; 5473 case ISN_GETITEM: smsg("%s%4d ITEM %lld%s", pfx, current, 5474 iptr->isn_arg.getitem.gi_index, 5475 iptr->isn_arg.getitem.gi_with_op ? 5476 " with op" : ""); break; 5477 case ISN_MEMBER: smsg("%s%4d MEMBER", pfx, current); break; 5478 case ISN_STRINGMEMBER: smsg("%s%4d MEMBER %s", pfx, current, 5479 iptr->isn_arg.string); break; 5480 case ISN_NEGATENR: smsg("%s%4d NEGATENR", pfx, current); break; 5481 5482 case ISN_CHECKNR: smsg("%s%4d CHECKNR", pfx, current); break; 5483 case ISN_CHECKTYPE: 5484 { 5485 checktype_T *ct = &iptr->isn_arg.type; 5486 char *tofree; 5487 5488 if (ct->ct_arg_idx == 0) 5489 smsg("%s%4d CHECKTYPE %s stack[%d]", pfx, current, 5490 type_name(ct->ct_type, &tofree), 5491 (int)ct->ct_off); 5492 else 5493 smsg("%s%4d CHECKTYPE %s stack[%d] arg %d", pfx, current, 5494 type_name(ct->ct_type, &tofree), 5495 (int)ct->ct_off, 5496 (int)ct->ct_arg_idx); 5497 vim_free(tofree); 5498 break; 5499 } 5500 case ISN_CHECKLEN: smsg("%s%4d CHECKLEN %s%d", pfx, current, 5501 iptr->isn_arg.checklen.cl_more_OK ? ">= " : "", 5502 iptr->isn_arg.checklen.cl_min_len); 5503 break; 5504 case ISN_SETTYPE: 5505 { 5506 char *tofree; 5507 5508 smsg("%s%4d SETTYPE %s", pfx, current, 5509 type_name(iptr->isn_arg.type.ct_type, &tofree)); 5510 vim_free(tofree); 5511 break; 5512 } 5513 case ISN_COND2BOOL: smsg("%s%4d COND2BOOL", pfx, current); break; 5514 case ISN_2BOOL: if (iptr->isn_arg.tobool.invert) 5515 smsg("%s%4d INVERT %d (!val)", pfx, current, 5516 iptr->isn_arg.tobool.offset); 5517 else 5518 smsg("%s%4d 2BOOL %d (!!val)", pfx, current, 5519 iptr->isn_arg.tobool.offset); 5520 break; 5521 case ISN_2STRING: smsg("%s%4d 2STRING stack[%lld]", pfx, current, 5522 (varnumber_T)(iptr->isn_arg.tostring.offset)); 5523 break; 5524 case ISN_2STRING_ANY: smsg("%s%4d 2STRING_ANY stack[%lld]", 5525 pfx, current, 5526 (varnumber_T)(iptr->isn_arg.tostring.offset)); 5527 break; 5528 case ISN_RANGE: smsg("%s%4d RANGE %s", pfx, current, 5529 iptr->isn_arg.string); 5530 break; 5531 case ISN_PUT: 5532 if (iptr->isn_arg.put.put_lnum == LNUM_VARIABLE_RANGE_ABOVE) 5533 smsg("%s%4d PUT %c above range", 5534 pfx, current, iptr->isn_arg.put.put_regname); 5535 else if (iptr->isn_arg.put.put_lnum == LNUM_VARIABLE_RANGE) 5536 smsg("%s%4d PUT %c range", 5537 pfx, current, iptr->isn_arg.put.put_regname); 5538 else 5539 smsg("%s%4d PUT %c %ld", pfx, current, 5540 iptr->isn_arg.put.put_regname, 5541 (long)iptr->isn_arg.put.put_lnum); 5542 break; 5543 5544 // TODO: summarize modifiers 5545 case ISN_CMDMOD: 5546 { 5547 char_u *buf; 5548 size_t len = produce_cmdmods( 5549 NULL, iptr->isn_arg.cmdmod.cf_cmdmod, FALSE); 5550 5551 buf = alloc(len + 1); 5552 if (likely(buf != NULL)) 5553 { 5554 (void)produce_cmdmods( 5555 buf, iptr->isn_arg.cmdmod.cf_cmdmod, FALSE); 5556 smsg("%s%4d CMDMOD %s", pfx, current, buf); 5557 vim_free(buf); 5558 } 5559 break; 5560 } 5561 case ISN_CMDMOD_REV: smsg("%s%4d CMDMOD_REV", pfx, current); break; 5562 5563 case ISN_PROF_START: 5564 smsg("%s%4d PROFILE START line %d", pfx, current, 5565 iptr->isn_lnum); 5566 break; 5567 5568 case ISN_PROF_END: 5569 smsg("%s%4d PROFILE END", pfx, current); 5570 break; 5571 5572 case ISN_DEBUG: 5573 smsg("%s%4d DEBUG line %d-%d varcount %lld", pfx, current, 5574 iptr->isn_arg.debug.dbg_break_lnum + 1, 5575 iptr->isn_lnum, 5576 iptr->isn_arg.debug.dbg_var_names_len); 5577 break; 5578 5579 case ISN_UNPACK: smsg("%s%4d UNPACK %d%s", pfx, current, 5580 iptr->isn_arg.unpack.unp_count, 5581 iptr->isn_arg.unpack.unp_semicolon ? " semicolon" : ""); 5582 break; 5583 case ISN_SHUFFLE: smsg("%s%4d SHUFFLE %d up %d", pfx, current, 5584 iptr->isn_arg.shuffle.shfl_item, 5585 iptr->isn_arg.shuffle.shfl_up); 5586 break; 5587 case ISN_DROP: smsg("%s%4d DROP", pfx, current); break; 5588 5589 case ISN_FINISH: // End of list of instructions for ISN_SUBSTITUTE. 5590 return; 5591 } 5592 5593 out_flush(); // output one line at a time 5594 ui_breakcheck(); 5595 if (got_int) 5596 break; 5597 } 5598 } 5599 5600 /* 5601 * Handle command line completion for the :disassemble command. 5602 */ 5603 void 5604 set_context_in_disassemble_cmd(expand_T *xp, char_u *arg) 5605 { 5606 char_u *p; 5607 5608 // Default: expand user functions, "debug" and "profile" 5609 xp->xp_context = EXPAND_DISASSEMBLE; 5610 xp->xp_pattern = arg; 5611 5612 // first argument already typed: only user function names 5613 if (*arg != NUL && *(p = skiptowhite(arg)) != NUL) 5614 { 5615 xp->xp_context = EXPAND_USER_FUNC; 5616 xp->xp_pattern = skipwhite(p); 5617 } 5618 } 5619 5620 /* 5621 * Function given to ExpandGeneric() to obtain the list of :disassemble 5622 * arguments. 5623 */ 5624 char_u * 5625 get_disassemble_argument(expand_T *xp, int idx) 5626 { 5627 if (idx == 0) 5628 return (char_u *)"debug"; 5629 if (idx == 1) 5630 return (char_u *)"profile"; 5631 return get_user_func_name(xp, idx - 2); 5632 } 5633 5634 /* 5635 * ":disassemble". 5636 * We don't really need this at runtime, but we do have tests that require it, 5637 * so always include this. 5638 */ 5639 void 5640 ex_disassemble(exarg_T *eap) 5641 { 5642 char_u *arg = eap->arg; 5643 char_u *fname; 5644 ufunc_T *ufunc; 5645 dfunc_T *dfunc; 5646 isn_T *instr; 5647 int instr_count; 5648 int is_global = FALSE; 5649 compiletype_T compile_type = CT_NONE; 5650 5651 if (STRNCMP(arg, "profile", 7) == 0) 5652 { 5653 compile_type = CT_PROFILE; 5654 arg = skipwhite(arg + 7); 5655 } 5656 else if (STRNCMP(arg, "debug", 5) == 0) 5657 { 5658 compile_type = CT_DEBUG; 5659 arg = skipwhite(arg + 5); 5660 } 5661 5662 if (STRNCMP(arg, "<lambda>", 8) == 0) 5663 { 5664 arg += 8; 5665 (void)getdigits(&arg); 5666 fname = vim_strnsave(eap->arg, arg - eap->arg); 5667 } 5668 else 5669 fname = trans_function_name(&arg, &is_global, FALSE, 5670 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD, NULL, NULL, NULL); 5671 if (fname == NULL) 5672 { 5673 semsg(_(e_invarg2), eap->arg); 5674 return; 5675 } 5676 5677 ufunc = find_func(fname, is_global, NULL); 5678 if (ufunc == NULL) 5679 { 5680 char_u *p = untrans_function_name(fname); 5681 5682 if (p != NULL) 5683 // Try again without making it script-local. 5684 ufunc = find_func(p, FALSE, NULL); 5685 } 5686 vim_free(fname); 5687 if (ufunc == NULL) 5688 { 5689 semsg(_(e_cannot_find_function_str), eap->arg); 5690 return; 5691 } 5692 if (func_needs_compiling(ufunc, compile_type) 5693 && compile_def_function(ufunc, FALSE, compile_type, NULL) == FAIL) 5694 return; 5695 if (ufunc->uf_def_status != UF_COMPILED) 5696 { 5697 semsg(_(e_function_is_not_compiled_str), eap->arg); 5698 return; 5699 } 5700 msg((char *)printable_func_name(ufunc)); 5701 5702 dfunc = ((dfunc_T *)def_functions.ga_data) + ufunc->uf_dfunc_idx; 5703 switch (compile_type) 5704 { 5705 case CT_PROFILE: 5706 #ifdef FEAT_PROFILE 5707 instr = dfunc->df_instr_prof; 5708 instr_count = dfunc->df_instr_prof_count; 5709 break; 5710 #endif 5711 // FALLTHROUGH 5712 case CT_NONE: 5713 instr = dfunc->df_instr; 5714 instr_count = dfunc->df_instr_count; 5715 break; 5716 case CT_DEBUG: 5717 instr = dfunc->df_instr_debug; 5718 instr_count = dfunc->df_instr_debug_count; 5719 break; 5720 } 5721 5722 list_instructions("", instr, instr_count, ufunc); 5723 } 5724 5725 /* 5726 * Return TRUE when "tv" is not falsy: non-zero, non-empty string, non-empty 5727 * list, etc. Mostly like what JavaScript does, except that empty list and 5728 * empty dictionary are FALSE. 5729 */ 5730 int 5731 tv2bool(typval_T *tv) 5732 { 5733 switch (tv->v_type) 5734 { 5735 case VAR_NUMBER: 5736 return tv->vval.v_number != 0; 5737 case VAR_FLOAT: 5738 #ifdef FEAT_FLOAT 5739 return tv->vval.v_float != 0.0; 5740 #else 5741 break; 5742 #endif 5743 case VAR_PARTIAL: 5744 return tv->vval.v_partial != NULL; 5745 case VAR_FUNC: 5746 case VAR_STRING: 5747 return tv->vval.v_string != NULL && *tv->vval.v_string != NUL; 5748 case VAR_LIST: 5749 return tv->vval.v_list != NULL && tv->vval.v_list->lv_len > 0; 5750 case VAR_DICT: 5751 return tv->vval.v_dict != NULL 5752 && tv->vval.v_dict->dv_hashtab.ht_used > 0; 5753 case VAR_BOOL: 5754 case VAR_SPECIAL: 5755 return tv->vval.v_number == VVAL_TRUE ? TRUE : FALSE; 5756 case VAR_JOB: 5757 #ifdef FEAT_JOB_CHANNEL 5758 return tv->vval.v_job != NULL; 5759 #else 5760 break; 5761 #endif 5762 case VAR_CHANNEL: 5763 #ifdef FEAT_JOB_CHANNEL 5764 return tv->vval.v_channel != NULL; 5765 #else 5766 break; 5767 #endif 5768 case VAR_BLOB: 5769 return tv->vval.v_blob != NULL && tv->vval.v_blob->bv_ga.ga_len > 0; 5770 case VAR_UNKNOWN: 5771 case VAR_ANY: 5772 case VAR_VOID: 5773 case VAR_INSTR: 5774 break; 5775 } 5776 return FALSE; 5777 } 5778 5779 void 5780 emsg_using_string_as(typval_T *tv, int as_number) 5781 { 5782 semsg(_(as_number ? e_using_string_as_number_str 5783 : e_using_string_as_bool_str), 5784 tv->vval.v_string == NULL 5785 ? (char_u *)"" : tv->vval.v_string); 5786 } 5787 5788 /* 5789 * If "tv" is a string give an error and return FAIL. 5790 */ 5791 int 5792 check_not_string(typval_T *tv) 5793 { 5794 if (tv->v_type == VAR_STRING) 5795 { 5796 emsg_using_string_as(tv, TRUE); 5797 clear_tv(tv); 5798 return FAIL; 5799 } 5800 return OK; 5801 } 5802 5803 5804 #endif // FEAT_EVAL 5805