18a7d6542SBram Moolenaar /* vi:set ts=8 sts=4 sw=4 noet: 28a7d6542SBram Moolenaar * 38a7d6542SBram Moolenaar * VIM - Vi IMproved by Bram Moolenaar 48a7d6542SBram Moolenaar * 58a7d6542SBram Moolenaar * Do ":help uganda" in Vim to read copying and usage conditions. 68a7d6542SBram Moolenaar * Do ":help credits" in Vim to see a list of people who contributed. 78a7d6542SBram Moolenaar * See README.txt for an overview of the Vim source code. 88a7d6542SBram Moolenaar */ 98a7d6542SBram Moolenaar 108a7d6542SBram Moolenaar /* 118a7d6542SBram Moolenaar * vim9execute.c: execute Vim9 script instructions 128a7d6542SBram Moolenaar */ 138a7d6542SBram Moolenaar 148a7d6542SBram Moolenaar #define USING_FLOAT_STUFF 158a7d6542SBram Moolenaar #include "vim.h" 168a7d6542SBram Moolenaar 178a7d6542SBram Moolenaar #if defined(FEAT_EVAL) || defined(PROTO) 188a7d6542SBram Moolenaar 198a7d6542SBram Moolenaar #ifdef VMS 208a7d6542SBram Moolenaar # include <float.h> 218a7d6542SBram Moolenaar #endif 228a7d6542SBram Moolenaar 238a7d6542SBram Moolenaar #include "vim9.h" 248a7d6542SBram Moolenaar 258a7d6542SBram Moolenaar // Structure put on ec_trystack when ISN_TRY is encountered. 268a7d6542SBram Moolenaar typedef struct { 278a7d6542SBram Moolenaar int tcd_frame; // ec_frame when ISN_TRY was encountered 288a7d6542SBram Moolenaar int tcd_catch_idx; // instruction of the first catch 298a7d6542SBram Moolenaar int tcd_finally_idx; // instruction of the finally block 308a7d6542SBram Moolenaar int tcd_caught; // catch block entered 318a7d6542SBram Moolenaar int tcd_return; // when TRUE return from end of :finally 328a7d6542SBram Moolenaar } trycmd_T; 338a7d6542SBram Moolenaar 348a7d6542SBram Moolenaar 358a7d6542SBram Moolenaar // A stack is used to store: 368a7d6542SBram Moolenaar // - arguments passed to a :def function 378a7d6542SBram Moolenaar // - info about the calling function, to use when returning 388a7d6542SBram Moolenaar // - local variables 398a7d6542SBram Moolenaar // - temporary values 408a7d6542SBram Moolenaar // 418a7d6542SBram Moolenaar // In detail (FP == Frame Pointer): 428a7d6542SBram Moolenaar // arg1 first argument from caller (if present) 438a7d6542SBram Moolenaar // arg2 second argument from caller (if present) 448a7d6542SBram Moolenaar // extra_arg1 any missing optional argument default value 458a7d6542SBram Moolenaar // FP -> cur_func calling function 468a7d6542SBram Moolenaar // current previous instruction pointer 478a7d6542SBram Moolenaar // frame_ptr previous Frame Pointer 488a7d6542SBram Moolenaar // var1 space for local variable 498a7d6542SBram Moolenaar // var2 space for local variable 508a7d6542SBram Moolenaar // .... fixed space for max. number of local variables 518a7d6542SBram Moolenaar // temp temporary values 528a7d6542SBram Moolenaar // .... flexible space for temporary values (can grow big) 538a7d6542SBram Moolenaar 548a7d6542SBram Moolenaar /* 558a7d6542SBram Moolenaar * Execution context. 568a7d6542SBram Moolenaar */ 578a7d6542SBram Moolenaar typedef struct { 588a7d6542SBram Moolenaar garray_T ec_stack; // stack of typval_T values 598a7d6542SBram Moolenaar int ec_frame; // index in ec_stack: context of ec_dfunc_idx 608a7d6542SBram Moolenaar 618a7d6542SBram Moolenaar garray_T ec_trystack; // stack of trycmd_T values 628a7d6542SBram Moolenaar int ec_in_catch; // when TRUE in catch or finally block 638a7d6542SBram Moolenaar 648a7d6542SBram Moolenaar int ec_dfunc_idx; // current function index 658a7d6542SBram Moolenaar isn_T *ec_instr; // array with instructions 668a7d6542SBram Moolenaar int ec_iidx; // index in ec_instr: instruction to execute 678a7d6542SBram Moolenaar } ectx_T; 688a7d6542SBram Moolenaar 698a7d6542SBram Moolenaar // Get pointer to item relative to the bottom of the stack, -1 is the last one. 708a7d6542SBram Moolenaar #define STACK_TV_BOT(idx) (((typval_T *)ectx->ec_stack.ga_data) + ectx->ec_stack.ga_len + idx) 718a7d6542SBram Moolenaar 728a7d6542SBram Moolenaar /* 738a7d6542SBram Moolenaar * Return the number of arguments, including any vararg. 748a7d6542SBram Moolenaar */ 758a7d6542SBram Moolenaar static int 768a7d6542SBram Moolenaar ufunc_argcount(ufunc_T *ufunc) 778a7d6542SBram Moolenaar { 788a7d6542SBram Moolenaar return ufunc->uf_args.ga_len + (ufunc->uf_va_name != NULL ? 1 : 0); 798a7d6542SBram Moolenaar } 808a7d6542SBram Moolenaar 818a7d6542SBram Moolenaar /* 828a7d6542SBram Moolenaar * Call compiled function "cdf_idx" from compiled code. 838a7d6542SBram Moolenaar * 848a7d6542SBram Moolenaar * Stack has: 858a7d6542SBram Moolenaar * - current arguments (already there) 868a7d6542SBram Moolenaar * - omitted optional argument (default values) added here 878a7d6542SBram Moolenaar * - stack frame: 888a7d6542SBram Moolenaar * - pointer to calling function 898a7d6542SBram Moolenaar * - Index of next instruction in calling function 908a7d6542SBram Moolenaar * - previous frame pointer 918a7d6542SBram Moolenaar * - reserved space for local variables 928a7d6542SBram Moolenaar */ 938a7d6542SBram Moolenaar static int 948a7d6542SBram Moolenaar call_dfunc(int cdf_idx, int argcount, ectx_T *ectx) 958a7d6542SBram Moolenaar { 968a7d6542SBram Moolenaar dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data) + cdf_idx; 978a7d6542SBram Moolenaar ufunc_T *ufunc = dfunc->df_ufunc; 988a7d6542SBram Moolenaar int optcount = ufunc_argcount(ufunc) - argcount; 998a7d6542SBram Moolenaar int idx; 1008a7d6542SBram Moolenaar 1018a7d6542SBram Moolenaar if (dfunc->df_deleted) 1028a7d6542SBram Moolenaar { 1038a7d6542SBram Moolenaar emsg_funcname(e_func_deleted, ufunc->uf_name); 1048a7d6542SBram Moolenaar return FAIL; 1058a7d6542SBram Moolenaar } 1068a7d6542SBram Moolenaar 1078a7d6542SBram Moolenaar if (ga_grow(&ectx->ec_stack, optcount + 3 + dfunc->df_varcount) == FAIL) 1088a7d6542SBram Moolenaar return FAIL; 1098a7d6542SBram Moolenaar 1108a7d6542SBram Moolenaar // TODO: Put omitted argument default values on the stack. 1118a7d6542SBram Moolenaar if (optcount > 0) 1128a7d6542SBram Moolenaar { 1138a7d6542SBram Moolenaar emsg("optional arguments not implemented yet"); 1148a7d6542SBram Moolenaar return FAIL; 1158a7d6542SBram Moolenaar } 1168a7d6542SBram Moolenaar if (optcount < 0) 1178a7d6542SBram Moolenaar { 1188a7d6542SBram Moolenaar emsg("argument count wrong?"); 1198a7d6542SBram Moolenaar return FAIL; 1208a7d6542SBram Moolenaar } 1218a7d6542SBram Moolenaar // for (idx = argcount - dfunc->df_minarg; 1228a7d6542SBram Moolenaar // idx < dfunc->df_maxarg; ++idx) 1238a7d6542SBram Moolenaar // { 1248a7d6542SBram Moolenaar // copy_tv(&dfunc->df_defarg[idx], STACK_TV_BOT(0)); 1258a7d6542SBram Moolenaar // ++ectx->ec_stack.ga_len; 1268a7d6542SBram Moolenaar // } 1278a7d6542SBram Moolenaar 1288a7d6542SBram Moolenaar // Store current execution state in stack frame for ISN_RETURN. 1298a7d6542SBram Moolenaar // TODO: If the actual number of arguments doesn't match what the called 1308a7d6542SBram Moolenaar // function expects things go bad. 1318a7d6542SBram Moolenaar STACK_TV_BOT(0)->vval.v_number = ectx->ec_dfunc_idx; 1328a7d6542SBram Moolenaar STACK_TV_BOT(1)->vval.v_number = ectx->ec_iidx; 1338a7d6542SBram Moolenaar STACK_TV_BOT(2)->vval.v_number = ectx->ec_frame; 1348a7d6542SBram Moolenaar ectx->ec_frame = ectx->ec_stack.ga_len; 1358a7d6542SBram Moolenaar 1368a7d6542SBram Moolenaar // Initialize local variables 1378a7d6542SBram Moolenaar for (idx = 0; idx < dfunc->df_varcount; ++idx) 1388a7d6542SBram Moolenaar STACK_TV_BOT(STACK_FRAME_SIZE + idx)->v_type = VAR_UNKNOWN; 1398a7d6542SBram Moolenaar ectx->ec_stack.ga_len += STACK_FRAME_SIZE + dfunc->df_varcount; 1408a7d6542SBram Moolenaar 1418a7d6542SBram Moolenaar // Set execution state to the start of the called function. 1428a7d6542SBram Moolenaar ectx->ec_dfunc_idx = cdf_idx; 1438a7d6542SBram Moolenaar ectx->ec_instr = dfunc->df_instr; 1448a7d6542SBram Moolenaar estack_push_ufunc(ETYPE_UFUNC, dfunc->df_ufunc, 1); 1458a7d6542SBram Moolenaar ectx->ec_iidx = 0; 1468a7d6542SBram Moolenaar 1478a7d6542SBram Moolenaar return OK; 1488a7d6542SBram Moolenaar } 1498a7d6542SBram Moolenaar 1508a7d6542SBram Moolenaar // Get pointer to item in the stack. 1518a7d6542SBram Moolenaar #define STACK_TV(idx) (((typval_T *)ectx->ec_stack.ga_data) + idx) 1528a7d6542SBram Moolenaar 1538a7d6542SBram Moolenaar /* 1548a7d6542SBram Moolenaar * Return from the current function. 1558a7d6542SBram Moolenaar */ 1568a7d6542SBram Moolenaar static void 1578a7d6542SBram Moolenaar func_return(ectx_T *ectx) 1588a7d6542SBram Moolenaar { 1598a7d6542SBram Moolenaar int ret_idx = ectx->ec_stack.ga_len - 1; 1608a7d6542SBram Moolenaar int idx; 1618a7d6542SBram Moolenaar dfunc_T *dfunc; 1628a7d6542SBram Moolenaar 1638a7d6542SBram Moolenaar // execution context goes one level up 1648a7d6542SBram Moolenaar estack_pop(); 1658a7d6542SBram Moolenaar 1668a7d6542SBram Moolenaar // Clear the local variables and temporary values, but not 1678a7d6542SBram Moolenaar // the return value. 1688a7d6542SBram Moolenaar for (idx = ectx->ec_frame + STACK_FRAME_SIZE; 1698a7d6542SBram Moolenaar idx < ectx->ec_stack.ga_len - 1; ++idx) 1708a7d6542SBram Moolenaar clear_tv(STACK_TV(idx)); 1718a7d6542SBram Moolenaar dfunc = ((dfunc_T *)def_functions.ga_data) + ectx->ec_dfunc_idx; 1728a7d6542SBram Moolenaar ectx->ec_stack.ga_len = ectx->ec_frame 1738a7d6542SBram Moolenaar - ufunc_argcount(dfunc->df_ufunc) + 1; 1748a7d6542SBram Moolenaar ectx->ec_dfunc_idx = STACK_TV(ectx->ec_frame)->vval.v_number; 1758a7d6542SBram Moolenaar ectx->ec_iidx = STACK_TV(ectx->ec_frame + 1)->vval.v_number; 1768a7d6542SBram Moolenaar ectx->ec_frame = STACK_TV(ectx->ec_frame + 2)->vval.v_number; 1778a7d6542SBram Moolenaar *STACK_TV_BOT(-1) = *STACK_TV(ret_idx); 1788a7d6542SBram Moolenaar dfunc = ((dfunc_T *)def_functions.ga_data) + ectx->ec_dfunc_idx; 1798a7d6542SBram Moolenaar ectx->ec_instr = dfunc->df_instr; 1808a7d6542SBram Moolenaar } 1818a7d6542SBram Moolenaar 1828a7d6542SBram Moolenaar #undef STACK_TV 1838a7d6542SBram Moolenaar 1848a7d6542SBram Moolenaar /* 1858a7d6542SBram Moolenaar * Prepare arguments and rettv for calling a builtin or user function. 1868a7d6542SBram Moolenaar */ 1878a7d6542SBram Moolenaar static int 1888a7d6542SBram Moolenaar call_prepare(int argcount, typval_T *argvars, ectx_T *ectx) 1898a7d6542SBram Moolenaar { 1908a7d6542SBram Moolenaar int idx; 1918a7d6542SBram Moolenaar typval_T *tv; 1928a7d6542SBram Moolenaar 1938a7d6542SBram Moolenaar // Move arguments from bottom of the stack to argvars[] and add terminator. 1948a7d6542SBram Moolenaar for (idx = 0; idx < argcount; ++idx) 1958a7d6542SBram Moolenaar argvars[idx] = *STACK_TV_BOT(idx - argcount); 1968a7d6542SBram Moolenaar argvars[argcount].v_type = VAR_UNKNOWN; 1978a7d6542SBram Moolenaar 1988a7d6542SBram Moolenaar // Result replaces the arguments on the stack. 1998a7d6542SBram Moolenaar if (argcount > 0) 2008a7d6542SBram Moolenaar ectx->ec_stack.ga_len -= argcount - 1; 2018a7d6542SBram Moolenaar else if (ga_grow(&ectx->ec_stack, 1) == FAIL) 2028a7d6542SBram Moolenaar return FAIL; 2038a7d6542SBram Moolenaar else 2048a7d6542SBram Moolenaar ++ectx->ec_stack.ga_len; 2058a7d6542SBram Moolenaar 2068a7d6542SBram Moolenaar // Default return value is zero. 2078a7d6542SBram Moolenaar tv = STACK_TV_BOT(-1); 2088a7d6542SBram Moolenaar tv->v_type = VAR_NUMBER; 2098a7d6542SBram Moolenaar tv->vval.v_number = 0; 2108a7d6542SBram Moolenaar 2118a7d6542SBram Moolenaar return OK; 2128a7d6542SBram Moolenaar } 2138a7d6542SBram Moolenaar 2148a7d6542SBram Moolenaar /* 2158a7d6542SBram Moolenaar * Call a builtin function by index. 2168a7d6542SBram Moolenaar */ 2178a7d6542SBram Moolenaar static int 2188a7d6542SBram Moolenaar call_bfunc(int func_idx, int argcount, ectx_T *ectx) 2198a7d6542SBram Moolenaar { 2208a7d6542SBram Moolenaar typval_T argvars[MAX_FUNC_ARGS]; 2218a7d6542SBram Moolenaar int idx; 2228a7d6542SBram Moolenaar 2238a7d6542SBram Moolenaar if (call_prepare(argcount, argvars, ectx) == FAIL) 2248a7d6542SBram Moolenaar return FAIL; 2258a7d6542SBram Moolenaar 2268a7d6542SBram Moolenaar // Call the builtin function. 2278a7d6542SBram Moolenaar call_internal_func_by_idx(func_idx, argvars, STACK_TV_BOT(-1)); 2288a7d6542SBram Moolenaar 2298a7d6542SBram Moolenaar // Clear the arguments. 2308a7d6542SBram Moolenaar for (idx = 0; idx < argcount; ++idx) 2318a7d6542SBram Moolenaar clear_tv(&argvars[idx]); 2328a7d6542SBram Moolenaar return OK; 2338a7d6542SBram Moolenaar } 2348a7d6542SBram Moolenaar 2358a7d6542SBram Moolenaar /* 2368a7d6542SBram Moolenaar * Execute a user defined function. 2378a7d6542SBram Moolenaar */ 2388a7d6542SBram Moolenaar static int 2398a7d6542SBram Moolenaar call_ufunc(ufunc_T *ufunc, int argcount, ectx_T *ectx) 2408a7d6542SBram Moolenaar { 2418a7d6542SBram Moolenaar typval_T argvars[MAX_FUNC_ARGS]; 2428a7d6542SBram Moolenaar funcexe_T funcexe; 2438a7d6542SBram Moolenaar int error; 2448a7d6542SBram Moolenaar int idx; 2458a7d6542SBram Moolenaar 2468a7d6542SBram Moolenaar if (ufunc->uf_dfunc_idx >= 0) 2478a7d6542SBram Moolenaar // The function has been compiled, can call it quickly. 2488a7d6542SBram Moolenaar return call_dfunc(ufunc->uf_dfunc_idx, argcount, ectx); 2498a7d6542SBram Moolenaar 2508a7d6542SBram Moolenaar if (call_prepare(argcount, argvars, ectx) == FAIL) 2518a7d6542SBram Moolenaar return FAIL; 2528a7d6542SBram Moolenaar vim_memset(&funcexe, 0, sizeof(funcexe)); 2538a7d6542SBram Moolenaar funcexe.evaluate = TRUE; 2548a7d6542SBram Moolenaar 2558a7d6542SBram Moolenaar // Call the user function. Result goes in last position on the stack. 2568a7d6542SBram Moolenaar // TODO: add selfdict if there is one 2578a7d6542SBram Moolenaar error = call_user_func_check(ufunc, argcount, argvars, 2588a7d6542SBram Moolenaar STACK_TV_BOT(-1), &funcexe, NULL); 2598a7d6542SBram Moolenaar 2608a7d6542SBram Moolenaar // Clear the arguments. 2618a7d6542SBram Moolenaar for (idx = 0; idx < argcount; ++idx) 2628a7d6542SBram Moolenaar clear_tv(&argvars[idx]); 2638a7d6542SBram Moolenaar 2648a7d6542SBram Moolenaar if (error != FCERR_NONE) 2658a7d6542SBram Moolenaar { 2668a7d6542SBram Moolenaar user_func_error(error, ufunc->uf_name); 2678a7d6542SBram Moolenaar return FAIL; 2688a7d6542SBram Moolenaar } 2698a7d6542SBram Moolenaar return OK; 2708a7d6542SBram Moolenaar } 2718a7d6542SBram Moolenaar 2728a7d6542SBram Moolenaar /* 2738a7d6542SBram Moolenaar * Execute a function by "name". 2748a7d6542SBram Moolenaar * This can be a builtin function or a user function. 2758a7d6542SBram Moolenaar * Returns FAIL if not found without an error message. 2768a7d6542SBram Moolenaar */ 2778a7d6542SBram Moolenaar static int 2788a7d6542SBram Moolenaar call_by_name(char_u *name, int argcount, ectx_T *ectx) 2798a7d6542SBram Moolenaar { 2808a7d6542SBram Moolenaar ufunc_T *ufunc; 2818a7d6542SBram Moolenaar 2828a7d6542SBram Moolenaar if (builtin_function(name, -1)) 2838a7d6542SBram Moolenaar { 2848a7d6542SBram Moolenaar int func_idx = find_internal_func(name); 2858a7d6542SBram Moolenaar 2868a7d6542SBram Moolenaar if (func_idx < 0) 2878a7d6542SBram Moolenaar return FAIL; 2888a7d6542SBram Moolenaar if (check_internal_func(func_idx, argcount) == FAIL) 2898a7d6542SBram Moolenaar return FAIL; 2908a7d6542SBram Moolenaar return call_bfunc(func_idx, argcount, ectx); 2918a7d6542SBram Moolenaar } 2928a7d6542SBram Moolenaar 2938a7d6542SBram Moolenaar ufunc = find_func(name, NULL); 2948a7d6542SBram Moolenaar if (ufunc != NULL) 2958a7d6542SBram Moolenaar return call_ufunc(ufunc, argcount, ectx); 2968a7d6542SBram Moolenaar 2978a7d6542SBram Moolenaar return FAIL; 2988a7d6542SBram Moolenaar } 2998a7d6542SBram Moolenaar 3008a7d6542SBram Moolenaar static int 3018a7d6542SBram Moolenaar call_partial(typval_T *tv, int argcount, ectx_T *ectx) 3028a7d6542SBram Moolenaar { 3038a7d6542SBram Moolenaar char_u *name; 3048a7d6542SBram Moolenaar int called_emsg_before = called_emsg; 3058a7d6542SBram Moolenaar 3068a7d6542SBram Moolenaar if (tv->v_type == VAR_PARTIAL) 3078a7d6542SBram Moolenaar { 3088a7d6542SBram Moolenaar partial_T *pt = tv->vval.v_partial; 3098a7d6542SBram Moolenaar 3108a7d6542SBram Moolenaar if (pt->pt_func != NULL) 3118a7d6542SBram Moolenaar return call_ufunc(pt->pt_func, argcount, ectx); 3128a7d6542SBram Moolenaar name = pt->pt_name; 3138a7d6542SBram Moolenaar } 3148a7d6542SBram Moolenaar else 3158a7d6542SBram Moolenaar name = tv->vval.v_string; 3168a7d6542SBram Moolenaar if (call_by_name(name, argcount, ectx) == FAIL) 3178a7d6542SBram Moolenaar { 3188a7d6542SBram Moolenaar if (called_emsg == called_emsg_before) 3198a7d6542SBram Moolenaar semsg(_(e_unknownfunc), name); 3208a7d6542SBram Moolenaar return FAIL; 3218a7d6542SBram Moolenaar } 3228a7d6542SBram Moolenaar return OK; 3238a7d6542SBram Moolenaar } 3248a7d6542SBram Moolenaar 3258a7d6542SBram Moolenaar /* 3268a7d6542SBram Moolenaar * Execute a function by "name". 3278a7d6542SBram Moolenaar * This can be a builtin function, user function or a funcref. 3288a7d6542SBram Moolenaar */ 3298a7d6542SBram Moolenaar static int 3308a7d6542SBram Moolenaar call_eval_func(char_u *name, int argcount, ectx_T *ectx) 3318a7d6542SBram Moolenaar { 3328a7d6542SBram Moolenaar int called_emsg_before = called_emsg; 3338a7d6542SBram Moolenaar 3348a7d6542SBram Moolenaar if (call_by_name(name, argcount, ectx) == FAIL 3358a7d6542SBram Moolenaar && called_emsg == called_emsg_before) 3368a7d6542SBram Moolenaar { 3378a7d6542SBram Moolenaar // "name" may be a variable that is a funcref or partial 3388a7d6542SBram Moolenaar // if find variable 3398a7d6542SBram Moolenaar // call_partial() 3408a7d6542SBram Moolenaar // else 3418a7d6542SBram Moolenaar // semsg(_(e_unknownfunc), name); 3428a7d6542SBram Moolenaar emsg("call_eval_func(partial) not implemented yet"); 3438a7d6542SBram Moolenaar return FAIL; 3448a7d6542SBram Moolenaar } 3458a7d6542SBram Moolenaar return OK; 3468a7d6542SBram Moolenaar } 3478a7d6542SBram Moolenaar 3488a7d6542SBram Moolenaar /* 3498a7d6542SBram Moolenaar * Call a "def" function from old Vim script. 3508a7d6542SBram Moolenaar * Return OK or FAIL. 3518a7d6542SBram Moolenaar */ 3528a7d6542SBram Moolenaar int 3538a7d6542SBram Moolenaar call_def_function( 3548a7d6542SBram Moolenaar ufunc_T *ufunc, 3558a7d6542SBram Moolenaar int argc, // nr of arguments 3568a7d6542SBram Moolenaar typval_T *argv, // arguments 3578a7d6542SBram Moolenaar typval_T *rettv) // return value 3588a7d6542SBram Moolenaar { 3598a7d6542SBram Moolenaar ectx_T ectx; // execution context 3608a7d6542SBram Moolenaar int initial_frame_ptr; 3618a7d6542SBram Moolenaar typval_T *tv; 3628a7d6542SBram Moolenaar int idx; 3638a7d6542SBram Moolenaar int ret = FAIL; 3648a7d6542SBram Moolenaar dfunc_T *dfunc; 36526e117e9SBram Moolenaar int optcount = ufunc_argcount(ufunc) - argc; 3668a7d6542SBram Moolenaar 3678a7d6542SBram Moolenaar // Get pointer to item in the stack. 3688a7d6542SBram Moolenaar #define STACK_TV(idx) (((typval_T *)ectx.ec_stack.ga_data) + idx) 3698a7d6542SBram Moolenaar 3708a7d6542SBram Moolenaar // Get pointer to item at the bottom of the stack, -1 is the bottom. 3718a7d6542SBram Moolenaar #undef STACK_TV_BOT 3728a7d6542SBram Moolenaar #define STACK_TV_BOT(idx) (((typval_T *)ectx.ec_stack.ga_data) + ectx.ec_stack.ga_len + idx) 3738a7d6542SBram Moolenaar 3748a7d6542SBram Moolenaar // Get pointer to local variable on the stack. 3758a7d6542SBram Moolenaar #define STACK_TV_VAR(idx) (((typval_T *)ectx.ec_stack.ga_data) + ectx.ec_frame + STACK_FRAME_SIZE + idx) 3768a7d6542SBram Moolenaar 3778a7d6542SBram Moolenaar vim_memset(&ectx, 0, sizeof(ectx)); 3788a7d6542SBram Moolenaar ga_init2(&ectx.ec_stack, sizeof(typval_T), 500); 3798a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 20) == FAIL) 3808a7d6542SBram Moolenaar goto failed; 3818a7d6542SBram Moolenaar ectx.ec_dfunc_idx = ufunc->uf_dfunc_idx; 3828a7d6542SBram Moolenaar 3838a7d6542SBram Moolenaar ga_init2(&ectx.ec_trystack, sizeof(trycmd_T), 10); 3848a7d6542SBram Moolenaar 3858a7d6542SBram Moolenaar // Put arguments on the stack. 3868a7d6542SBram Moolenaar for (idx = 0; idx < argc; ++idx) 3878a7d6542SBram Moolenaar { 3888a7d6542SBram Moolenaar copy_tv(&argv[idx], STACK_TV_BOT(0)); 3898a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 3908a7d6542SBram Moolenaar } 3918a7d6542SBram Moolenaar 3928a7d6542SBram Moolenaar // Frame pointer points to just after arguments. 3938a7d6542SBram Moolenaar ectx.ec_frame = ectx.ec_stack.ga_len; 3948a7d6542SBram Moolenaar initial_frame_ptr = ectx.ec_frame; 3958a7d6542SBram Moolenaar 39626e117e9SBram Moolenaar // TODO: Put omitted argument default values on the stack. 39726e117e9SBram Moolenaar if (optcount > 0) 39826e117e9SBram Moolenaar { 39926e117e9SBram Moolenaar emsg("optional arguments not implemented yet"); 40026e117e9SBram Moolenaar return FAIL; 40126e117e9SBram Moolenaar } 4028a7d6542SBram Moolenaar // dummy frame entries 4038a7d6542SBram Moolenaar for (idx = 0; idx < STACK_FRAME_SIZE; ++idx) 4048a7d6542SBram Moolenaar { 4058a7d6542SBram Moolenaar STACK_TV(ectx.ec_stack.ga_len)->v_type = VAR_UNKNOWN; 4068a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 4078a7d6542SBram Moolenaar } 4088a7d6542SBram Moolenaar 4098a7d6542SBram Moolenaar // Reserve space for local variables. 4108a7d6542SBram Moolenaar dfunc = ((dfunc_T *)def_functions.ga_data) + ufunc->uf_dfunc_idx; 4118a7d6542SBram Moolenaar for (idx = 0; idx < dfunc->df_varcount; ++idx) 4128a7d6542SBram Moolenaar STACK_TV_VAR(idx)->v_type = VAR_UNKNOWN; 4138a7d6542SBram Moolenaar ectx.ec_stack.ga_len += dfunc->df_varcount; 4148a7d6542SBram Moolenaar 4158a7d6542SBram Moolenaar ectx.ec_instr = dfunc->df_instr; 4168a7d6542SBram Moolenaar ectx.ec_iidx = 0; 4178a7d6542SBram Moolenaar for (;;) 4188a7d6542SBram Moolenaar { 4198a7d6542SBram Moolenaar isn_T *iptr; 4208a7d6542SBram Moolenaar trycmd_T *trycmd = NULL; 4218a7d6542SBram Moolenaar 4228a7d6542SBram Moolenaar if (did_throw && !ectx.ec_in_catch) 4238a7d6542SBram Moolenaar { 4248a7d6542SBram Moolenaar garray_T *trystack = &ectx.ec_trystack; 4258a7d6542SBram Moolenaar 4268a7d6542SBram Moolenaar // An exception jumps to the first catch, finally, or returns from 4278a7d6542SBram Moolenaar // the current function. 4288a7d6542SBram Moolenaar if (trystack->ga_len > 0) 4298a7d6542SBram Moolenaar trycmd = ((trycmd_T *)trystack->ga_data) + trystack->ga_len - 1; 4308a7d6542SBram Moolenaar if (trycmd != NULL && trycmd->tcd_frame == ectx.ec_frame) 4318a7d6542SBram Moolenaar { 4328a7d6542SBram Moolenaar // jump to ":catch" or ":finally" 4338a7d6542SBram Moolenaar ectx.ec_in_catch = TRUE; 4348a7d6542SBram Moolenaar ectx.ec_iidx = trycmd->tcd_catch_idx; 4358a7d6542SBram Moolenaar } 4368a7d6542SBram Moolenaar else 4378a7d6542SBram Moolenaar { 4388a7d6542SBram Moolenaar // not inside try or need to return from current functions. 4398a7d6542SBram Moolenaar if (ectx.ec_frame == initial_frame_ptr) 4408a7d6542SBram Moolenaar { 4418a7d6542SBram Moolenaar // At the toplevel we are done. Push a dummy return value. 4428a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 4438a7d6542SBram Moolenaar goto failed; 4448a7d6542SBram Moolenaar tv = STACK_TV_BOT(0); 4458a7d6542SBram Moolenaar tv->v_type = VAR_NUMBER; 4468a7d6542SBram Moolenaar tv->vval.v_number = 0; 4478a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 4488a7d6542SBram Moolenaar goto done; 4498a7d6542SBram Moolenaar } 4508a7d6542SBram Moolenaar 4518a7d6542SBram Moolenaar func_return(&ectx); 4528a7d6542SBram Moolenaar } 4538a7d6542SBram Moolenaar continue; 4548a7d6542SBram Moolenaar } 4558a7d6542SBram Moolenaar 4568a7d6542SBram Moolenaar iptr = &ectx.ec_instr[ectx.ec_iidx++]; 4578a7d6542SBram Moolenaar switch (iptr->isn_type) 4588a7d6542SBram Moolenaar { 4598a7d6542SBram Moolenaar // execute Ex command line 4608a7d6542SBram Moolenaar case ISN_EXEC: 4618a7d6542SBram Moolenaar do_cmdline_cmd(iptr->isn_arg.string); 4628a7d6542SBram Moolenaar break; 4638a7d6542SBram Moolenaar 4648a7d6542SBram Moolenaar // execute :echo {string} ... 4658a7d6542SBram Moolenaar case ISN_ECHO: 4668a7d6542SBram Moolenaar { 4678a7d6542SBram Moolenaar int count = iptr->isn_arg.echo.echo_count; 4688a7d6542SBram Moolenaar int atstart = TRUE; 4698a7d6542SBram Moolenaar int needclr = TRUE; 4708a7d6542SBram Moolenaar 4718a7d6542SBram Moolenaar for (idx = 0; idx < count; ++idx) 4728a7d6542SBram Moolenaar { 4738a7d6542SBram Moolenaar tv = STACK_TV_BOT(idx - count); 4748a7d6542SBram Moolenaar echo_one(tv, iptr->isn_arg.echo.echo_with_white, 4758a7d6542SBram Moolenaar &atstart, &needclr); 4768a7d6542SBram Moolenaar clear_tv(tv); 4778a7d6542SBram Moolenaar } 4788a7d6542SBram Moolenaar ectx.ec_stack.ga_len -= count; 4798a7d6542SBram Moolenaar } 4808a7d6542SBram Moolenaar break; 4818a7d6542SBram Moolenaar 4828a7d6542SBram Moolenaar // load local variable or argument 4838a7d6542SBram Moolenaar case ISN_LOAD: 4848a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 4858a7d6542SBram Moolenaar goto failed; 4868a7d6542SBram Moolenaar copy_tv(STACK_TV_VAR(iptr->isn_arg.number), STACK_TV_BOT(0)); 4878a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 4888a7d6542SBram Moolenaar break; 4898a7d6542SBram Moolenaar 4908a7d6542SBram Moolenaar // load v: variable 4918a7d6542SBram Moolenaar case ISN_LOADV: 4928a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 4938a7d6542SBram Moolenaar goto failed; 4948a7d6542SBram Moolenaar copy_tv(get_vim_var_tv(iptr->isn_arg.number), STACK_TV_BOT(0)); 4958a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 4968a7d6542SBram Moolenaar break; 4978a7d6542SBram Moolenaar 498b283a8a6SBram Moolenaar // load s: variable in Vim9 script 4998a7d6542SBram Moolenaar case ISN_LOADSCRIPT: 5008a7d6542SBram Moolenaar { 5018a7d6542SBram Moolenaar scriptitem_T *si = 50221b9e977SBram Moolenaar SCRIPT_ITEM(iptr->isn_arg.script.script_sid); 5038a7d6542SBram Moolenaar svar_T *sv; 5048a7d6542SBram Moolenaar 5058a7d6542SBram Moolenaar sv = ((svar_T *)si->sn_var_vals.ga_data) 5068a7d6542SBram Moolenaar + iptr->isn_arg.script.script_idx; 5078a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 5088a7d6542SBram Moolenaar goto failed; 5098a7d6542SBram Moolenaar copy_tv(sv->sv_tv, STACK_TV_BOT(0)); 5108a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 5118a7d6542SBram Moolenaar } 5128a7d6542SBram Moolenaar break; 5138a7d6542SBram Moolenaar 5148a7d6542SBram Moolenaar // load s: variable in old script 5158a7d6542SBram Moolenaar case ISN_LOADS: 5168a7d6542SBram Moolenaar { 517b283a8a6SBram Moolenaar hashtab_T *ht = &SCRIPT_VARS( 518b283a8a6SBram Moolenaar iptr->isn_arg.loadstore.ls_sid); 519b283a8a6SBram Moolenaar char_u *name = iptr->isn_arg.loadstore.ls_name; 5208a7d6542SBram Moolenaar dictitem_T *di = find_var_in_ht(ht, 0, name, TRUE); 5218a7d6542SBram Moolenaar if (di == NULL) 5228a7d6542SBram Moolenaar { 523b283a8a6SBram Moolenaar semsg(_(e_undefvar), name); 5248a7d6542SBram Moolenaar goto failed; 5258a7d6542SBram Moolenaar } 5268a7d6542SBram Moolenaar else 5278a7d6542SBram Moolenaar { 5288a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 5298a7d6542SBram Moolenaar goto failed; 5308a7d6542SBram Moolenaar copy_tv(&di->di_tv, STACK_TV_BOT(0)); 5318a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 5328a7d6542SBram Moolenaar } 5338a7d6542SBram Moolenaar } 5348a7d6542SBram Moolenaar break; 5358a7d6542SBram Moolenaar 5368a7d6542SBram Moolenaar // load g: variable 5378a7d6542SBram Moolenaar case ISN_LOADG: 5388a7d6542SBram Moolenaar { 5398a7d6542SBram Moolenaar dictitem_T *di; 5408a7d6542SBram Moolenaar 5418a7d6542SBram Moolenaar di = find_var_in_ht(get_globvar_ht(), 0, 5428a7d6542SBram Moolenaar iptr->isn_arg.string, TRUE); 5438a7d6542SBram Moolenaar if (di == NULL) 5448a7d6542SBram Moolenaar { 5458a7d6542SBram Moolenaar semsg(_("E121: Undefined variable: g:%s"), 5468a7d6542SBram Moolenaar iptr->isn_arg.string); 5478a7d6542SBram Moolenaar goto failed; 5488a7d6542SBram Moolenaar } 5498a7d6542SBram Moolenaar else 5508a7d6542SBram Moolenaar { 5518a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 5528a7d6542SBram Moolenaar goto failed; 5538a7d6542SBram Moolenaar copy_tv(&di->di_tv, STACK_TV_BOT(0)); 5548a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 5558a7d6542SBram Moolenaar } 5568a7d6542SBram Moolenaar } 5578a7d6542SBram Moolenaar break; 5588a7d6542SBram Moolenaar 5598a7d6542SBram Moolenaar // load &option 5608a7d6542SBram Moolenaar case ISN_LOADOPT: 5618a7d6542SBram Moolenaar { 5628a7d6542SBram Moolenaar typval_T optval; 5638a7d6542SBram Moolenaar char_u *name = iptr->isn_arg.string; 5648a7d6542SBram Moolenaar 5658a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 5668a7d6542SBram Moolenaar goto failed; 56758ceca5cSBram Moolenaar if (get_option_tv(&name, &optval, TRUE) == FAIL) 56858ceca5cSBram Moolenaar goto failed; 5698a7d6542SBram Moolenaar *STACK_TV_BOT(0) = optval; 5708a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 5718a7d6542SBram Moolenaar } 5728a7d6542SBram Moolenaar break; 5738a7d6542SBram Moolenaar 5748a7d6542SBram Moolenaar // load $ENV 5758a7d6542SBram Moolenaar case ISN_LOADENV: 5768a7d6542SBram Moolenaar { 5778a7d6542SBram Moolenaar typval_T optval; 5788a7d6542SBram Moolenaar char_u *name = iptr->isn_arg.string; 5798a7d6542SBram Moolenaar 5808a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 5818a7d6542SBram Moolenaar goto failed; 58207da94b0SBram Moolenaar if (get_env_tv(&name, &optval, TRUE) == FAIL) 58307da94b0SBram Moolenaar { 58407da94b0SBram Moolenaar semsg(_("E1060: Invalid environment variable name: %s"), 58507da94b0SBram Moolenaar iptr->isn_arg.string); 58607da94b0SBram Moolenaar goto failed; 58707da94b0SBram Moolenaar } 5888a7d6542SBram Moolenaar *STACK_TV_BOT(0) = optval; 5898a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 5908a7d6542SBram Moolenaar } 5918a7d6542SBram Moolenaar break; 5928a7d6542SBram Moolenaar 5938a7d6542SBram Moolenaar // load @register 5948a7d6542SBram Moolenaar case ISN_LOADREG: 5958a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 5968a7d6542SBram Moolenaar goto failed; 5978a7d6542SBram Moolenaar tv = STACK_TV_BOT(0); 5988a7d6542SBram Moolenaar tv->v_type = VAR_STRING; 5998a7d6542SBram Moolenaar tv->vval.v_string = get_reg_contents( 6008a7d6542SBram Moolenaar iptr->isn_arg.number, GREG_EXPR_SRC); 6018a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 6028a7d6542SBram Moolenaar break; 6038a7d6542SBram Moolenaar 6048a7d6542SBram Moolenaar // store local variable 6058a7d6542SBram Moolenaar case ISN_STORE: 6068a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 6078a7d6542SBram Moolenaar tv = STACK_TV_VAR(iptr->isn_arg.number); 6088a7d6542SBram Moolenaar clear_tv(tv); 6098a7d6542SBram Moolenaar *tv = *STACK_TV_BOT(0); 6108a7d6542SBram Moolenaar break; 6118a7d6542SBram Moolenaar 612b283a8a6SBram Moolenaar // store s: variable in old script 613b283a8a6SBram Moolenaar case ISN_STORES: 614b283a8a6SBram Moolenaar { 615b283a8a6SBram Moolenaar hashtab_T *ht = &SCRIPT_VARS( 616b283a8a6SBram Moolenaar iptr->isn_arg.loadstore.ls_sid); 617b283a8a6SBram Moolenaar char_u *name = iptr->isn_arg.loadstore.ls_name; 618b283a8a6SBram Moolenaar dictitem_T *di = find_var_in_ht(ht, 0, name, TRUE); 619b283a8a6SBram Moolenaar 620b283a8a6SBram Moolenaar if (di == NULL) 621b283a8a6SBram Moolenaar { 622b283a8a6SBram Moolenaar semsg(_(e_undefvar), name); 623b283a8a6SBram Moolenaar goto failed; 624b283a8a6SBram Moolenaar } 625b283a8a6SBram Moolenaar --ectx.ec_stack.ga_len; 626b283a8a6SBram Moolenaar clear_tv(&di->di_tv); 627b283a8a6SBram Moolenaar di->di_tv = *STACK_TV_BOT(0); 628b283a8a6SBram Moolenaar } 629b283a8a6SBram Moolenaar break; 630b283a8a6SBram Moolenaar 631b283a8a6SBram Moolenaar // store script-local variable in Vim9 script 6328a7d6542SBram Moolenaar case ISN_STORESCRIPT: 6338a7d6542SBram Moolenaar { 63421b9e977SBram Moolenaar scriptitem_T *si = SCRIPT_ITEM( 6358a7d6542SBram Moolenaar iptr->isn_arg.script.script_sid); 6368a7d6542SBram Moolenaar svar_T *sv = ((svar_T *)si->sn_var_vals.ga_data) 6378a7d6542SBram Moolenaar + iptr->isn_arg.script.script_idx; 6388a7d6542SBram Moolenaar 6398a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 6408a7d6542SBram Moolenaar clear_tv(sv->sv_tv); 6418a7d6542SBram Moolenaar *sv->sv_tv = *STACK_TV_BOT(0); 6428a7d6542SBram Moolenaar } 6438a7d6542SBram Moolenaar break; 6448a7d6542SBram Moolenaar 6458a7d6542SBram Moolenaar // store option 6468a7d6542SBram Moolenaar case ISN_STOREOPT: 6478a7d6542SBram Moolenaar { 6488a7d6542SBram Moolenaar long n = 0; 6498a7d6542SBram Moolenaar char_u *s = NULL; 6508a7d6542SBram Moolenaar char *msg; 6518a7d6542SBram Moolenaar 6528a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 6538a7d6542SBram Moolenaar tv = STACK_TV_BOT(0); 6548a7d6542SBram Moolenaar if (tv->v_type == VAR_STRING) 65597a2af39SBram Moolenaar { 6568a7d6542SBram Moolenaar s = tv->vval.v_string; 65797a2af39SBram Moolenaar if (s == NULL) 65897a2af39SBram Moolenaar s = (char_u *)""; 65997a2af39SBram Moolenaar } 6608a7d6542SBram Moolenaar else if (tv->v_type == VAR_NUMBER) 6618a7d6542SBram Moolenaar n = tv->vval.v_number; 6628a7d6542SBram Moolenaar else 6638a7d6542SBram Moolenaar { 6648a7d6542SBram Moolenaar emsg(_("E1051: Expected string or number")); 6658a7d6542SBram Moolenaar goto failed; 6668a7d6542SBram Moolenaar } 6678a7d6542SBram Moolenaar msg = set_option_value(iptr->isn_arg.storeopt.so_name, 6688a7d6542SBram Moolenaar n, s, iptr->isn_arg.storeopt.so_flags); 6698a7d6542SBram Moolenaar if (msg != NULL) 6708a7d6542SBram Moolenaar { 6718a7d6542SBram Moolenaar emsg(_(msg)); 6728a7d6542SBram Moolenaar goto failed; 6738a7d6542SBram Moolenaar } 6748a7d6542SBram Moolenaar clear_tv(tv); 6758a7d6542SBram Moolenaar } 6768a7d6542SBram Moolenaar break; 6778a7d6542SBram Moolenaar 678b283a8a6SBram Moolenaar // store $ENV 679b283a8a6SBram Moolenaar case ISN_STOREENV: 680b283a8a6SBram Moolenaar --ectx.ec_stack.ga_len; 681b283a8a6SBram Moolenaar vim_setenv_ext(iptr->isn_arg.string, 682b283a8a6SBram Moolenaar tv_get_string(STACK_TV_BOT(0))); 683b283a8a6SBram Moolenaar break; 684b283a8a6SBram Moolenaar 685b283a8a6SBram Moolenaar // store @r 686b283a8a6SBram Moolenaar case ISN_STOREREG: 687b283a8a6SBram Moolenaar { 688b283a8a6SBram Moolenaar int reg = iptr->isn_arg.number; 689b283a8a6SBram Moolenaar 690b283a8a6SBram Moolenaar --ectx.ec_stack.ga_len; 691b283a8a6SBram Moolenaar write_reg_contents(reg == '@' ? '"' : reg, 692b283a8a6SBram Moolenaar tv_get_string(STACK_TV_BOT(0)), -1, FALSE); 693b283a8a6SBram Moolenaar } 694b283a8a6SBram Moolenaar break; 695b283a8a6SBram Moolenaar 696b283a8a6SBram Moolenaar // store v: variable 697b283a8a6SBram Moolenaar case ISN_STOREV: 698b283a8a6SBram Moolenaar --ectx.ec_stack.ga_len; 699b283a8a6SBram Moolenaar if (set_vim_var_tv(iptr->isn_arg.number, STACK_TV_BOT(0)) 700b283a8a6SBram Moolenaar == FAIL) 701b283a8a6SBram Moolenaar goto failed; 702b283a8a6SBram Moolenaar break; 703b283a8a6SBram Moolenaar 7048a7d6542SBram Moolenaar // store g: variable 7058a7d6542SBram Moolenaar case ISN_STOREG: 7068a7d6542SBram Moolenaar { 7078a7d6542SBram Moolenaar dictitem_T *di; 7088a7d6542SBram Moolenaar 7098a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 7108a7d6542SBram Moolenaar di = find_var_in_ht(get_globvar_ht(), 0, 7118a7d6542SBram Moolenaar iptr->isn_arg.string, TRUE); 7128a7d6542SBram Moolenaar if (di == NULL) 7138a7d6542SBram Moolenaar { 7148a7d6542SBram Moolenaar funccal_entry_T entry; 7158a7d6542SBram Moolenaar 7168a7d6542SBram Moolenaar save_funccal(&entry); 7178a7d6542SBram Moolenaar set_var_const(iptr->isn_arg.string, NULL, 7188a7d6542SBram Moolenaar STACK_TV_BOT(0), FALSE, 0); 7198a7d6542SBram Moolenaar restore_funccal(); 7208a7d6542SBram Moolenaar } 7218a7d6542SBram Moolenaar else 7228a7d6542SBram Moolenaar { 7238a7d6542SBram Moolenaar clear_tv(&di->di_tv); 7248a7d6542SBram Moolenaar di->di_tv = *STACK_TV_BOT(0); 7258a7d6542SBram Moolenaar } 7268a7d6542SBram Moolenaar } 7278a7d6542SBram Moolenaar break; 7288a7d6542SBram Moolenaar 7298a7d6542SBram Moolenaar // store number in local variable 7308a7d6542SBram Moolenaar case ISN_STORENR: 7318a7d6542SBram Moolenaar tv = STACK_TV_VAR(iptr->isn_arg.storenr.str_idx); 7328a7d6542SBram Moolenaar clear_tv(tv); 7338a7d6542SBram Moolenaar tv->v_type = VAR_NUMBER; 7348a7d6542SBram Moolenaar tv->vval.v_number = iptr->isn_arg.storenr.str_val; 7358a7d6542SBram Moolenaar break; 7368a7d6542SBram Moolenaar 7378a7d6542SBram Moolenaar // push constant 7388a7d6542SBram Moolenaar case ISN_PUSHNR: 7398a7d6542SBram Moolenaar case ISN_PUSHBOOL: 7408a7d6542SBram Moolenaar case ISN_PUSHSPEC: 7418a7d6542SBram Moolenaar case ISN_PUSHF: 7428a7d6542SBram Moolenaar case ISN_PUSHS: 7438a7d6542SBram Moolenaar case ISN_PUSHBLOB: 7448a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 7458a7d6542SBram Moolenaar goto failed; 7468a7d6542SBram Moolenaar tv = STACK_TV_BOT(0); 7478a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 7488a7d6542SBram Moolenaar switch (iptr->isn_type) 7498a7d6542SBram Moolenaar { 7508a7d6542SBram Moolenaar case ISN_PUSHNR: 7518a7d6542SBram Moolenaar tv->v_type = VAR_NUMBER; 7528a7d6542SBram Moolenaar tv->vval.v_number = iptr->isn_arg.number; 7538a7d6542SBram Moolenaar break; 7548a7d6542SBram Moolenaar case ISN_PUSHBOOL: 7558a7d6542SBram Moolenaar tv->v_type = VAR_BOOL; 7568a7d6542SBram Moolenaar tv->vval.v_number = iptr->isn_arg.number; 7578a7d6542SBram Moolenaar break; 7588a7d6542SBram Moolenaar case ISN_PUSHSPEC: 7598a7d6542SBram Moolenaar tv->v_type = VAR_SPECIAL; 7608a7d6542SBram Moolenaar tv->vval.v_number = iptr->isn_arg.number; 7618a7d6542SBram Moolenaar break; 7628a7d6542SBram Moolenaar #ifdef FEAT_FLOAT 7638a7d6542SBram Moolenaar case ISN_PUSHF: 7648a7d6542SBram Moolenaar tv->v_type = VAR_FLOAT; 7658a7d6542SBram Moolenaar tv->vval.v_float = iptr->isn_arg.fnumber; 7668a7d6542SBram Moolenaar break; 7678a7d6542SBram Moolenaar #endif 7688a7d6542SBram Moolenaar case ISN_PUSHBLOB: 7698a7d6542SBram Moolenaar blob_copy(iptr->isn_arg.blob, tv); 7708a7d6542SBram Moolenaar break; 7718a7d6542SBram Moolenaar default: 7728a7d6542SBram Moolenaar tv->v_type = VAR_STRING; 7738a7d6542SBram Moolenaar tv->vval.v_string = vim_strsave(iptr->isn_arg.string); 7748a7d6542SBram Moolenaar } 7758a7d6542SBram Moolenaar break; 7768a7d6542SBram Moolenaar 7778a7d6542SBram Moolenaar // create a list from items on the stack; uses a single allocation 7788a7d6542SBram Moolenaar // for the list header and the items 7798a7d6542SBram Moolenaar case ISN_NEWLIST: 7808a7d6542SBram Moolenaar { 7818a7d6542SBram Moolenaar int count = iptr->isn_arg.number; 7828a7d6542SBram Moolenaar list_T *list = list_alloc_with_items(count); 7838a7d6542SBram Moolenaar 7848a7d6542SBram Moolenaar if (list == NULL) 7858a7d6542SBram Moolenaar goto failed; 7868a7d6542SBram Moolenaar for (idx = 0; idx < count; ++idx) 7878a7d6542SBram Moolenaar list_set_item(list, idx, STACK_TV_BOT(idx - count)); 7888a7d6542SBram Moolenaar 7898a7d6542SBram Moolenaar if (count > 0) 7908a7d6542SBram Moolenaar ectx.ec_stack.ga_len -= count - 1; 7918a7d6542SBram Moolenaar else if (ga_grow(&ectx.ec_stack, 1) == FAIL) 7928a7d6542SBram Moolenaar goto failed; 7938a7d6542SBram Moolenaar else 7948a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 7958a7d6542SBram Moolenaar tv = STACK_TV_BOT(-1); 7968a7d6542SBram Moolenaar tv->v_type = VAR_LIST; 7978a7d6542SBram Moolenaar tv->vval.v_list = list; 7988a7d6542SBram Moolenaar ++list->lv_refcount; 7998a7d6542SBram Moolenaar } 8008a7d6542SBram Moolenaar break; 8018a7d6542SBram Moolenaar 8028a7d6542SBram Moolenaar // create a dict from items on the stack 8038a7d6542SBram Moolenaar case ISN_NEWDICT: 8048a7d6542SBram Moolenaar { 8058a7d6542SBram Moolenaar int count = iptr->isn_arg.number; 8068a7d6542SBram Moolenaar dict_T *dict = dict_alloc(); 8078a7d6542SBram Moolenaar dictitem_T *item; 8088a7d6542SBram Moolenaar 8098a7d6542SBram Moolenaar if (dict == NULL) 8108a7d6542SBram Moolenaar goto failed; 8118a7d6542SBram Moolenaar for (idx = 0; idx < count; ++idx) 8128a7d6542SBram Moolenaar { 8138a7d6542SBram Moolenaar // check key type is VAR_STRING 8148a7d6542SBram Moolenaar tv = STACK_TV_BOT(2 * (idx - count)); 8158a7d6542SBram Moolenaar item = dictitem_alloc(tv->vval.v_string); 8168a7d6542SBram Moolenaar clear_tv(tv); 8178a7d6542SBram Moolenaar if (item == NULL) 8188a7d6542SBram Moolenaar goto failed; 8198a7d6542SBram Moolenaar item->di_tv = *STACK_TV_BOT(2 * (idx - count) + 1); 8208a7d6542SBram Moolenaar item->di_tv.v_lock = 0; 8218a7d6542SBram Moolenaar if (dict_add(dict, item) == FAIL) 8228a7d6542SBram Moolenaar goto failed; 8238a7d6542SBram Moolenaar } 8248a7d6542SBram Moolenaar 8258a7d6542SBram Moolenaar if (count > 0) 8268a7d6542SBram Moolenaar ectx.ec_stack.ga_len -= 2 * count - 1; 8278a7d6542SBram Moolenaar else if (ga_grow(&ectx.ec_stack, 1) == FAIL) 8288a7d6542SBram Moolenaar goto failed; 8298a7d6542SBram Moolenaar else 8308a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 8318a7d6542SBram Moolenaar tv = STACK_TV_BOT(-1); 8328a7d6542SBram Moolenaar tv->v_type = VAR_DICT; 8338a7d6542SBram Moolenaar tv->vval.v_dict = dict; 8348a7d6542SBram Moolenaar ++dict->dv_refcount; 8358a7d6542SBram Moolenaar } 8368a7d6542SBram Moolenaar break; 8378a7d6542SBram Moolenaar 8388a7d6542SBram Moolenaar // call a :def function 8398a7d6542SBram Moolenaar case ISN_DCALL: 8408a7d6542SBram Moolenaar if (call_dfunc(iptr->isn_arg.dfunc.cdf_idx, 8418a7d6542SBram Moolenaar iptr->isn_arg.dfunc.cdf_argcount, 8428a7d6542SBram Moolenaar &ectx) == FAIL) 8438a7d6542SBram Moolenaar goto failed; 8448a7d6542SBram Moolenaar break; 8458a7d6542SBram Moolenaar 8468a7d6542SBram Moolenaar // call a builtin function 8478a7d6542SBram Moolenaar case ISN_BCALL: 8488a7d6542SBram Moolenaar SOURCING_LNUM = iptr->isn_lnum; 8498a7d6542SBram Moolenaar if (call_bfunc(iptr->isn_arg.bfunc.cbf_idx, 8508a7d6542SBram Moolenaar iptr->isn_arg.bfunc.cbf_argcount, 8518a7d6542SBram Moolenaar &ectx) == FAIL) 8528a7d6542SBram Moolenaar goto failed; 8538a7d6542SBram Moolenaar break; 8548a7d6542SBram Moolenaar 8558a7d6542SBram Moolenaar // call a funcref or partial 8568a7d6542SBram Moolenaar case ISN_PCALL: 8578a7d6542SBram Moolenaar { 8588a7d6542SBram Moolenaar cpfunc_T *pfunc = &iptr->isn_arg.pfunc; 8598a7d6542SBram Moolenaar int r; 8608a7d6542SBram Moolenaar typval_T partial; 8618a7d6542SBram Moolenaar 8628a7d6542SBram Moolenaar SOURCING_LNUM = iptr->isn_lnum; 8638a7d6542SBram Moolenaar if (pfunc->cpf_top) 8648a7d6542SBram Moolenaar { 8658a7d6542SBram Moolenaar // funcref is above the arguments 8668a7d6542SBram Moolenaar tv = STACK_TV_BOT(-pfunc->cpf_argcount - 1); 8678a7d6542SBram Moolenaar } 8688a7d6542SBram Moolenaar else 8698a7d6542SBram Moolenaar { 8708a7d6542SBram Moolenaar // Get the funcref from the stack. 8718a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 8728a7d6542SBram Moolenaar partial = *STACK_TV_BOT(0); 8738a7d6542SBram Moolenaar tv = &partial; 8748a7d6542SBram Moolenaar } 8758a7d6542SBram Moolenaar r = call_partial(tv, pfunc->cpf_argcount, &ectx); 8768a7d6542SBram Moolenaar if (tv == &partial) 8778a7d6542SBram Moolenaar clear_tv(&partial); 8788a7d6542SBram Moolenaar if (r == FAIL) 8798a7d6542SBram Moolenaar goto failed; 8808a7d6542SBram Moolenaar 8818a7d6542SBram Moolenaar if (pfunc->cpf_top) 8828a7d6542SBram Moolenaar { 8838a7d6542SBram Moolenaar // Get the funcref from the stack, overwrite with the 8848a7d6542SBram Moolenaar // return value. 8858a7d6542SBram Moolenaar clear_tv(tv); 8868a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 8878a7d6542SBram Moolenaar *STACK_TV_BOT(-1) = *STACK_TV_BOT(0); 8888a7d6542SBram Moolenaar } 8898a7d6542SBram Moolenaar } 8908a7d6542SBram Moolenaar break; 8918a7d6542SBram Moolenaar 8928a7d6542SBram Moolenaar // call a user defined function or funcref/partial 8938a7d6542SBram Moolenaar case ISN_UCALL: 8948a7d6542SBram Moolenaar { 8958a7d6542SBram Moolenaar cufunc_T *cufunc = &iptr->isn_arg.ufunc; 8968a7d6542SBram Moolenaar 8978a7d6542SBram Moolenaar SOURCING_LNUM = iptr->isn_lnum; 8988a7d6542SBram Moolenaar if (call_eval_func(cufunc->cuf_name, 8998a7d6542SBram Moolenaar cufunc->cuf_argcount, &ectx) == FAIL) 9008a7d6542SBram Moolenaar goto failed; 9018a7d6542SBram Moolenaar } 9028a7d6542SBram Moolenaar break; 9038a7d6542SBram Moolenaar 9048a7d6542SBram Moolenaar // return from a :def function call 9058a7d6542SBram Moolenaar case ISN_RETURN: 9068a7d6542SBram Moolenaar { 9078cbd6dfcSBram Moolenaar garray_T *trystack = &ectx.ec_trystack; 9088cbd6dfcSBram Moolenaar 9098cbd6dfcSBram Moolenaar if (trystack->ga_len > 0) 9108cbd6dfcSBram Moolenaar trycmd = ((trycmd_T *)trystack->ga_data) 9118cbd6dfcSBram Moolenaar + trystack->ga_len - 1; 9128a7d6542SBram Moolenaar if (trycmd != NULL && trycmd->tcd_frame == ectx.ec_frame 9138a7d6542SBram Moolenaar && trycmd->tcd_finally_idx != 0) 9148a7d6542SBram Moolenaar { 9158a7d6542SBram Moolenaar // jump to ":finally" 9168a7d6542SBram Moolenaar ectx.ec_iidx = trycmd->tcd_finally_idx; 9178a7d6542SBram Moolenaar trycmd->tcd_return = TRUE; 9188a7d6542SBram Moolenaar } 9198a7d6542SBram Moolenaar else 9208a7d6542SBram Moolenaar { 9218a7d6542SBram Moolenaar // Restore previous function. If the frame pointer 9228a7d6542SBram Moolenaar // is zero then there is none and we are done. 9238a7d6542SBram Moolenaar if (ectx.ec_frame == initial_frame_ptr) 9248a7d6542SBram Moolenaar goto done; 9258a7d6542SBram Moolenaar 9268a7d6542SBram Moolenaar func_return(&ectx); 9278a7d6542SBram Moolenaar } 9288a7d6542SBram Moolenaar } 9298a7d6542SBram Moolenaar break; 9308a7d6542SBram Moolenaar 9318a7d6542SBram Moolenaar // push a function reference to a compiled function 9328a7d6542SBram Moolenaar case ISN_FUNCREF: 9338a7d6542SBram Moolenaar { 9348a7d6542SBram Moolenaar partial_T *pt = NULL; 9358a7d6542SBram Moolenaar 9368a7d6542SBram Moolenaar pt = ALLOC_CLEAR_ONE(partial_T); 9378a7d6542SBram Moolenaar if (pt == NULL) 9388a7d6542SBram Moolenaar goto failed; 9398a7d6542SBram Moolenaar dfunc = ((dfunc_T *)def_functions.ga_data) 9408a7d6542SBram Moolenaar + iptr->isn_arg.number; 9418a7d6542SBram Moolenaar pt->pt_func = dfunc->df_ufunc; 9428a7d6542SBram Moolenaar pt->pt_refcount = 1; 9438a7d6542SBram Moolenaar ++dfunc->df_ufunc->uf_refcount; 9448a7d6542SBram Moolenaar 9458a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 9468a7d6542SBram Moolenaar goto failed; 9478a7d6542SBram Moolenaar tv = STACK_TV_BOT(0); 9488a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 9498a7d6542SBram Moolenaar tv->vval.v_partial = pt; 9508a7d6542SBram Moolenaar tv->v_type = VAR_PARTIAL; 9518a7d6542SBram Moolenaar } 9528a7d6542SBram Moolenaar break; 9538a7d6542SBram Moolenaar 9548a7d6542SBram Moolenaar // jump if a condition is met 9558a7d6542SBram Moolenaar case ISN_JUMP: 9568a7d6542SBram Moolenaar { 9578a7d6542SBram Moolenaar jumpwhen_T when = iptr->isn_arg.jump.jump_when; 9588a7d6542SBram Moolenaar int jump = TRUE; 9598a7d6542SBram Moolenaar 9608a7d6542SBram Moolenaar if (when != JUMP_ALWAYS) 9618a7d6542SBram Moolenaar { 9628a7d6542SBram Moolenaar tv = STACK_TV_BOT(-1); 9638a7d6542SBram Moolenaar jump = tv2bool(tv); 9648a7d6542SBram Moolenaar if (when == JUMP_IF_FALSE 9658a7d6542SBram Moolenaar || when == JUMP_AND_KEEP_IF_FALSE) 9668a7d6542SBram Moolenaar jump = !jump; 9678a7d6542SBram Moolenaar if (when == JUMP_IF_FALSE || when == JUMP_IF_TRUE 9688a7d6542SBram Moolenaar || !jump) 9698a7d6542SBram Moolenaar { 9708a7d6542SBram Moolenaar // drop the value from the stack 9718a7d6542SBram Moolenaar clear_tv(tv); 9728a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 9738a7d6542SBram Moolenaar } 9748a7d6542SBram Moolenaar } 9758a7d6542SBram Moolenaar if (jump) 9768a7d6542SBram Moolenaar ectx.ec_iidx = iptr->isn_arg.jump.jump_where; 9778a7d6542SBram Moolenaar } 9788a7d6542SBram Moolenaar break; 9798a7d6542SBram Moolenaar 9808a7d6542SBram Moolenaar // top of a for loop 9818a7d6542SBram Moolenaar case ISN_FOR: 9828a7d6542SBram Moolenaar { 9838a7d6542SBram Moolenaar list_T *list = STACK_TV_BOT(-1)->vval.v_list; 9848a7d6542SBram Moolenaar typval_T *idxtv = 9858a7d6542SBram Moolenaar STACK_TV_VAR(iptr->isn_arg.forloop.for_idx); 9868a7d6542SBram Moolenaar 9878a7d6542SBram Moolenaar // push the next item from the list 9888a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 9898a7d6542SBram Moolenaar goto failed; 9908a7d6542SBram Moolenaar if (++idxtv->vval.v_number >= list->lv_len) 9918a7d6542SBram Moolenaar // past the end of the list, jump to "endfor" 9928a7d6542SBram Moolenaar ectx.ec_iidx = iptr->isn_arg.forloop.for_end; 9938a7d6542SBram Moolenaar else if (list->lv_first == &range_list_item) 9948a7d6542SBram Moolenaar { 9958a7d6542SBram Moolenaar // non-materialized range() list 9968a7d6542SBram Moolenaar tv = STACK_TV_BOT(0); 9978a7d6542SBram Moolenaar tv->v_type = VAR_NUMBER; 9988a7d6542SBram Moolenaar tv->vval.v_number = list_find_nr( 9998a7d6542SBram Moolenaar list, idxtv->vval.v_number, NULL); 10008a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 10018a7d6542SBram Moolenaar } 10028a7d6542SBram Moolenaar else 10038a7d6542SBram Moolenaar { 10048a7d6542SBram Moolenaar listitem_T *li = list_find(list, idxtv->vval.v_number); 10058a7d6542SBram Moolenaar 10068a7d6542SBram Moolenaar if (li == NULL) 10078a7d6542SBram Moolenaar goto failed; 10088a7d6542SBram Moolenaar copy_tv(&li->li_tv, STACK_TV_BOT(0)); 10098a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 10108a7d6542SBram Moolenaar } 10118a7d6542SBram Moolenaar } 10128a7d6542SBram Moolenaar break; 10138a7d6542SBram Moolenaar 10148a7d6542SBram Moolenaar // start of ":try" block 10158a7d6542SBram Moolenaar case ISN_TRY: 10168a7d6542SBram Moolenaar { 10178a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_trystack, 1) == FAIL) 10188a7d6542SBram Moolenaar goto failed; 10198a7d6542SBram Moolenaar trycmd = ((trycmd_T *)ectx.ec_trystack.ga_data) 10208a7d6542SBram Moolenaar + ectx.ec_trystack.ga_len; 10218a7d6542SBram Moolenaar ++ectx.ec_trystack.ga_len; 10228a7d6542SBram Moolenaar ++trylevel; 10238a7d6542SBram Moolenaar trycmd->tcd_frame = ectx.ec_frame; 10248a7d6542SBram Moolenaar trycmd->tcd_catch_idx = iptr->isn_arg.try.try_catch; 10258a7d6542SBram Moolenaar trycmd->tcd_finally_idx = iptr->isn_arg.try.try_finally; 10268a7d6542SBram Moolenaar } 10278a7d6542SBram Moolenaar break; 10288a7d6542SBram Moolenaar 10298a7d6542SBram Moolenaar case ISN_PUSHEXC: 10308a7d6542SBram Moolenaar if (current_exception == NULL) 10318a7d6542SBram Moolenaar { 10328a7d6542SBram Moolenaar iemsg("Evaluating catch while current_exception is NULL"); 10338a7d6542SBram Moolenaar goto failed; 10348a7d6542SBram Moolenaar } 10358a7d6542SBram Moolenaar if (ga_grow(&ectx.ec_stack, 1) == FAIL) 10368a7d6542SBram Moolenaar goto failed; 10378a7d6542SBram Moolenaar tv = STACK_TV_BOT(0); 10388a7d6542SBram Moolenaar ++ectx.ec_stack.ga_len; 10398a7d6542SBram Moolenaar tv->v_type = VAR_STRING; 10408a7d6542SBram Moolenaar tv->vval.v_string = vim_strsave( 10418a7d6542SBram Moolenaar (char_u *)current_exception->value); 10428a7d6542SBram Moolenaar break; 10438a7d6542SBram Moolenaar 10448a7d6542SBram Moolenaar case ISN_CATCH: 10458a7d6542SBram Moolenaar { 10468a7d6542SBram Moolenaar garray_T *trystack = &ectx.ec_trystack; 10478a7d6542SBram Moolenaar 10488a7d6542SBram Moolenaar if (trystack->ga_len > 0) 10498a7d6542SBram Moolenaar { 10508a7d6542SBram Moolenaar trycmd = ((trycmd_T *)trystack->ga_data) 10518a7d6542SBram Moolenaar + trystack->ga_len - 1; 10528a7d6542SBram Moolenaar trycmd->tcd_caught = TRUE; 10538a7d6542SBram Moolenaar } 10548a7d6542SBram Moolenaar did_emsg = got_int = did_throw = FALSE; 10558a7d6542SBram Moolenaar catch_exception(current_exception); 10568a7d6542SBram Moolenaar } 10578a7d6542SBram Moolenaar break; 10588a7d6542SBram Moolenaar 10598a7d6542SBram Moolenaar // end of ":try" block 10608a7d6542SBram Moolenaar case ISN_ENDTRY: 10618a7d6542SBram Moolenaar { 10628a7d6542SBram Moolenaar garray_T *trystack = &ectx.ec_trystack; 10638a7d6542SBram Moolenaar 10648a7d6542SBram Moolenaar if (trystack->ga_len > 0) 10658a7d6542SBram Moolenaar { 10668a7d6542SBram Moolenaar --trystack->ga_len; 10678a7d6542SBram Moolenaar --trylevel; 10688a7d6542SBram Moolenaar trycmd = ((trycmd_T *)trystack->ga_data) 10698a7d6542SBram Moolenaar + trystack->ga_len; 10708a7d6542SBram Moolenaar if (trycmd->tcd_caught) 10718a7d6542SBram Moolenaar { 10728a7d6542SBram Moolenaar // discard the exception 10738a7d6542SBram Moolenaar if (caught_stack == current_exception) 10748a7d6542SBram Moolenaar caught_stack = caught_stack->caught; 10758a7d6542SBram Moolenaar discard_current_exception(); 10768a7d6542SBram Moolenaar } 10778a7d6542SBram Moolenaar 10788a7d6542SBram Moolenaar if (trycmd->tcd_return) 10798a7d6542SBram Moolenaar { 10808a7d6542SBram Moolenaar // Restore previous function. If the frame pointer 10818a7d6542SBram Moolenaar // is zero then there is none and we are done. 10828a7d6542SBram Moolenaar if (ectx.ec_frame == initial_frame_ptr) 10838a7d6542SBram Moolenaar goto done; 10848a7d6542SBram Moolenaar 10858a7d6542SBram Moolenaar func_return(&ectx); 10868a7d6542SBram Moolenaar } 10878a7d6542SBram Moolenaar } 10888a7d6542SBram Moolenaar } 10898a7d6542SBram Moolenaar break; 10908a7d6542SBram Moolenaar 10918a7d6542SBram Moolenaar case ISN_THROW: 10928a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 10938a7d6542SBram Moolenaar tv = STACK_TV_BOT(0); 10948a7d6542SBram Moolenaar if (throw_exception(tv->vval.v_string, ET_USER, NULL) == FAIL) 10958a7d6542SBram Moolenaar { 10968a7d6542SBram Moolenaar vim_free(tv->vval.v_string); 10978a7d6542SBram Moolenaar goto failed; 10988a7d6542SBram Moolenaar } 10998a7d6542SBram Moolenaar did_throw = TRUE; 11008a7d6542SBram Moolenaar break; 11018a7d6542SBram Moolenaar 11028a7d6542SBram Moolenaar // compare with special values 11038a7d6542SBram Moolenaar case ISN_COMPAREBOOL: 11048a7d6542SBram Moolenaar case ISN_COMPARESPECIAL: 11058a7d6542SBram Moolenaar { 11068a7d6542SBram Moolenaar typval_T *tv1 = STACK_TV_BOT(-2); 11078a7d6542SBram Moolenaar typval_T *tv2 = STACK_TV_BOT(-1); 11088a7d6542SBram Moolenaar varnumber_T arg1 = tv1->vval.v_number; 11098a7d6542SBram Moolenaar varnumber_T arg2 = tv2->vval.v_number; 11108a7d6542SBram Moolenaar int res; 11118a7d6542SBram Moolenaar 11128a7d6542SBram Moolenaar switch (iptr->isn_arg.op.op_type) 11138a7d6542SBram Moolenaar { 11148a7d6542SBram Moolenaar case EXPR_EQUAL: res = arg1 == arg2; break; 11158a7d6542SBram Moolenaar case EXPR_NEQUAL: res = arg1 != arg2; break; 11168a7d6542SBram Moolenaar default: res = 0; break; 11178a7d6542SBram Moolenaar } 11188a7d6542SBram Moolenaar 11198a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 11208a7d6542SBram Moolenaar tv1->v_type = VAR_BOOL; 11218a7d6542SBram Moolenaar tv1->vval.v_number = res ? VVAL_TRUE : VVAL_FALSE; 11228a7d6542SBram Moolenaar } 11238a7d6542SBram Moolenaar break; 11248a7d6542SBram Moolenaar 11258a7d6542SBram Moolenaar // Operation with two number arguments 11268a7d6542SBram Moolenaar case ISN_OPNR: 11278a7d6542SBram Moolenaar case ISN_COMPARENR: 11288a7d6542SBram Moolenaar { 11298a7d6542SBram Moolenaar typval_T *tv1 = STACK_TV_BOT(-2); 11308a7d6542SBram Moolenaar typval_T *tv2 = STACK_TV_BOT(-1); 11318a7d6542SBram Moolenaar varnumber_T arg1 = tv1->vval.v_number; 11328a7d6542SBram Moolenaar varnumber_T arg2 = tv2->vval.v_number; 11338a7d6542SBram Moolenaar varnumber_T res; 11348a7d6542SBram Moolenaar 11358a7d6542SBram Moolenaar switch (iptr->isn_arg.op.op_type) 11368a7d6542SBram Moolenaar { 11378a7d6542SBram Moolenaar case EXPR_MULT: res = arg1 * arg2; break; 11388a7d6542SBram Moolenaar case EXPR_DIV: res = arg1 / arg2; break; 11398a7d6542SBram Moolenaar case EXPR_REM: res = arg1 % arg2; break; 11408a7d6542SBram Moolenaar case EXPR_SUB: res = arg1 - arg2; break; 11418a7d6542SBram Moolenaar case EXPR_ADD: res = arg1 + arg2; break; 11428a7d6542SBram Moolenaar 11438a7d6542SBram Moolenaar case EXPR_EQUAL: res = arg1 == arg2; break; 11448a7d6542SBram Moolenaar case EXPR_NEQUAL: res = arg1 != arg2; break; 11458a7d6542SBram Moolenaar case EXPR_GREATER: res = arg1 > arg2; break; 11468a7d6542SBram Moolenaar case EXPR_GEQUAL: res = arg1 >= arg2; break; 11478a7d6542SBram Moolenaar case EXPR_SMALLER: res = arg1 < arg2; break; 11488a7d6542SBram Moolenaar case EXPR_SEQUAL: res = arg1 <= arg2; break; 11498a7d6542SBram Moolenaar default: res = 0; break; 11508a7d6542SBram Moolenaar } 11518a7d6542SBram Moolenaar 11528a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 11538a7d6542SBram Moolenaar if (iptr->isn_type == ISN_COMPARENR) 11548a7d6542SBram Moolenaar { 11558a7d6542SBram Moolenaar tv1->v_type = VAR_BOOL; 11568a7d6542SBram Moolenaar tv1->vval.v_number = res ? VVAL_TRUE : VVAL_FALSE; 11578a7d6542SBram Moolenaar } 11588a7d6542SBram Moolenaar else 11598a7d6542SBram Moolenaar tv1->vval.v_number = res; 11608a7d6542SBram Moolenaar } 11618a7d6542SBram Moolenaar break; 11628a7d6542SBram Moolenaar 11638a7d6542SBram Moolenaar // Computation with two float arguments 11648a7d6542SBram Moolenaar case ISN_OPFLOAT: 11658a7d6542SBram Moolenaar case ISN_COMPAREFLOAT: 1166a5d5953dSBram Moolenaar #ifdef FEAT_FLOAT 11678a7d6542SBram Moolenaar { 11688a7d6542SBram Moolenaar typval_T *tv1 = STACK_TV_BOT(-2); 11698a7d6542SBram Moolenaar typval_T *tv2 = STACK_TV_BOT(-1); 11708a7d6542SBram Moolenaar float_T arg1 = tv1->vval.v_float; 11718a7d6542SBram Moolenaar float_T arg2 = tv2->vval.v_float; 11728a7d6542SBram Moolenaar float_T res = 0; 11738a7d6542SBram Moolenaar int cmp = FALSE; 11748a7d6542SBram Moolenaar 11758a7d6542SBram Moolenaar switch (iptr->isn_arg.op.op_type) 11768a7d6542SBram Moolenaar { 11778a7d6542SBram Moolenaar case EXPR_MULT: res = arg1 * arg2; break; 11788a7d6542SBram Moolenaar case EXPR_DIV: res = arg1 / arg2; break; 11798a7d6542SBram Moolenaar case EXPR_SUB: res = arg1 - arg2; break; 11808a7d6542SBram Moolenaar case EXPR_ADD: res = arg1 + arg2; break; 11818a7d6542SBram Moolenaar 11828a7d6542SBram Moolenaar case EXPR_EQUAL: cmp = arg1 == arg2; break; 11838a7d6542SBram Moolenaar case EXPR_NEQUAL: cmp = arg1 != arg2; break; 11848a7d6542SBram Moolenaar case EXPR_GREATER: cmp = arg1 > arg2; break; 11858a7d6542SBram Moolenaar case EXPR_GEQUAL: cmp = arg1 >= arg2; break; 11868a7d6542SBram Moolenaar case EXPR_SMALLER: cmp = arg1 < arg2; break; 11878a7d6542SBram Moolenaar case EXPR_SEQUAL: cmp = arg1 <= arg2; break; 11888a7d6542SBram Moolenaar default: cmp = 0; break; 11898a7d6542SBram Moolenaar } 11908a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 11918a7d6542SBram Moolenaar if (iptr->isn_type == ISN_COMPAREFLOAT) 11928a7d6542SBram Moolenaar { 11938a7d6542SBram Moolenaar tv1->v_type = VAR_BOOL; 11948a7d6542SBram Moolenaar tv1->vval.v_number = cmp ? VVAL_TRUE : VVAL_FALSE; 11958a7d6542SBram Moolenaar } 11968a7d6542SBram Moolenaar else 11978a7d6542SBram Moolenaar tv1->vval.v_float = res; 11988a7d6542SBram Moolenaar } 1199a5d5953dSBram Moolenaar #endif 12008a7d6542SBram Moolenaar break; 12018a7d6542SBram Moolenaar 12028a7d6542SBram Moolenaar case ISN_COMPARELIST: 12038a7d6542SBram Moolenaar { 12048a7d6542SBram Moolenaar typval_T *tv1 = STACK_TV_BOT(-2); 12058a7d6542SBram Moolenaar typval_T *tv2 = STACK_TV_BOT(-1); 12068a7d6542SBram Moolenaar list_T *arg1 = tv1->vval.v_list; 12078a7d6542SBram Moolenaar list_T *arg2 = tv2->vval.v_list; 12088a7d6542SBram Moolenaar int cmp = FALSE; 12098a7d6542SBram Moolenaar int ic = iptr->isn_arg.op.op_ic; 12108a7d6542SBram Moolenaar 12118a7d6542SBram Moolenaar switch (iptr->isn_arg.op.op_type) 12128a7d6542SBram Moolenaar { 12138a7d6542SBram Moolenaar case EXPR_EQUAL: cmp = 12148a7d6542SBram Moolenaar list_equal(arg1, arg2, ic, FALSE); break; 12158a7d6542SBram Moolenaar case EXPR_NEQUAL: cmp = 12168a7d6542SBram Moolenaar !list_equal(arg1, arg2, ic, FALSE); break; 12178a7d6542SBram Moolenaar case EXPR_IS: cmp = arg1 == arg2; break; 12188a7d6542SBram Moolenaar case EXPR_ISNOT: cmp = arg1 != arg2; break; 12198a7d6542SBram Moolenaar default: cmp = 0; break; 12208a7d6542SBram Moolenaar } 12218a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 12228a7d6542SBram Moolenaar clear_tv(tv1); 12238a7d6542SBram Moolenaar clear_tv(tv2); 12248a7d6542SBram Moolenaar tv1->v_type = VAR_BOOL; 12258a7d6542SBram Moolenaar tv1->vval.v_number = cmp ? VVAL_TRUE : VVAL_FALSE; 12268a7d6542SBram Moolenaar } 12278a7d6542SBram Moolenaar break; 12288a7d6542SBram Moolenaar 12298a7d6542SBram Moolenaar case ISN_COMPAREBLOB: 12308a7d6542SBram Moolenaar { 12318a7d6542SBram Moolenaar typval_T *tv1 = STACK_TV_BOT(-2); 12328a7d6542SBram Moolenaar typval_T *tv2 = STACK_TV_BOT(-1); 12338a7d6542SBram Moolenaar blob_T *arg1 = tv1->vval.v_blob; 12348a7d6542SBram Moolenaar blob_T *arg2 = tv2->vval.v_blob; 12358a7d6542SBram Moolenaar int cmp = FALSE; 12368a7d6542SBram Moolenaar 12378a7d6542SBram Moolenaar switch (iptr->isn_arg.op.op_type) 12388a7d6542SBram Moolenaar { 12398a7d6542SBram Moolenaar case EXPR_EQUAL: cmp = blob_equal(arg1, arg2); break; 12408a7d6542SBram Moolenaar case EXPR_NEQUAL: cmp = !blob_equal(arg1, arg2); break; 12418a7d6542SBram Moolenaar case EXPR_IS: cmp = arg1 == arg2; break; 12428a7d6542SBram Moolenaar case EXPR_ISNOT: cmp = arg1 != arg2; break; 12438a7d6542SBram Moolenaar default: cmp = 0; break; 12448a7d6542SBram Moolenaar } 12458a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 12468a7d6542SBram Moolenaar clear_tv(tv1); 12478a7d6542SBram Moolenaar clear_tv(tv2); 12488a7d6542SBram Moolenaar tv1->v_type = VAR_BOOL; 12498a7d6542SBram Moolenaar tv1->vval.v_number = cmp ? VVAL_TRUE : VVAL_FALSE; 12508a7d6542SBram Moolenaar } 12518a7d6542SBram Moolenaar break; 12528a7d6542SBram Moolenaar 12538a7d6542SBram Moolenaar // TODO: handle separately 12548a7d6542SBram Moolenaar case ISN_COMPARESTRING: 12558a7d6542SBram Moolenaar case ISN_COMPAREDICT: 12568a7d6542SBram Moolenaar case ISN_COMPAREFUNC: 12578a7d6542SBram Moolenaar case ISN_COMPAREPARTIAL: 12588a7d6542SBram Moolenaar case ISN_COMPAREANY: 12598a7d6542SBram Moolenaar { 12608a7d6542SBram Moolenaar typval_T *tv1 = STACK_TV_BOT(-2); 12618a7d6542SBram Moolenaar typval_T *tv2 = STACK_TV_BOT(-1); 12628a7d6542SBram Moolenaar exptype_T exptype = iptr->isn_arg.op.op_type; 12638a7d6542SBram Moolenaar int ic = iptr->isn_arg.op.op_ic; 12648a7d6542SBram Moolenaar 12658a7d6542SBram Moolenaar typval_compare(tv1, tv2, exptype, ic); 12668a7d6542SBram Moolenaar clear_tv(tv2); 12678a7d6542SBram Moolenaar tv1->v_type = VAR_BOOL; 12688a7d6542SBram Moolenaar tv1->vval.v_number = tv1->vval.v_number 12698a7d6542SBram Moolenaar ? VVAL_TRUE : VVAL_FALSE; 12708a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 12718a7d6542SBram Moolenaar } 12728a7d6542SBram Moolenaar break; 12738a7d6542SBram Moolenaar 12748a7d6542SBram Moolenaar case ISN_ADDLIST: 12758a7d6542SBram Moolenaar case ISN_ADDBLOB: 12768a7d6542SBram Moolenaar { 12778a7d6542SBram Moolenaar typval_T *tv1 = STACK_TV_BOT(-2); 12788a7d6542SBram Moolenaar typval_T *tv2 = STACK_TV_BOT(-1); 12798a7d6542SBram Moolenaar 12808a7d6542SBram Moolenaar if (iptr->isn_type == ISN_ADDLIST) 12818a7d6542SBram Moolenaar eval_addlist(tv1, tv2); 12828a7d6542SBram Moolenaar else 12838a7d6542SBram Moolenaar eval_addblob(tv1, tv2); 12848a7d6542SBram Moolenaar clear_tv(tv2); 12858a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 12868a7d6542SBram Moolenaar } 12878a7d6542SBram Moolenaar break; 12888a7d6542SBram Moolenaar 12898a7d6542SBram Moolenaar // Computation with two arguments of unknown type 12908a7d6542SBram Moolenaar case ISN_OPANY: 12918a7d6542SBram Moolenaar { 12928a7d6542SBram Moolenaar typval_T *tv1 = STACK_TV_BOT(-2); 12938a7d6542SBram Moolenaar typval_T *tv2 = STACK_TV_BOT(-1); 12948a7d6542SBram Moolenaar varnumber_T n1, n2; 12958a7d6542SBram Moolenaar #ifdef FEAT_FLOAT 12968a7d6542SBram Moolenaar float_T f1 = 0, f2 = 0; 12978a7d6542SBram Moolenaar #endif 12988a7d6542SBram Moolenaar int error = FALSE; 12998a7d6542SBram Moolenaar 13008a7d6542SBram Moolenaar if (iptr->isn_arg.op.op_type == EXPR_ADD) 13018a7d6542SBram Moolenaar { 13028a7d6542SBram Moolenaar if (tv1->v_type == VAR_LIST && tv2->v_type == VAR_LIST) 13038a7d6542SBram Moolenaar { 13048a7d6542SBram Moolenaar eval_addlist(tv1, tv2); 13058a7d6542SBram Moolenaar clear_tv(tv2); 13068a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 13078a7d6542SBram Moolenaar break; 13088a7d6542SBram Moolenaar } 13098a7d6542SBram Moolenaar else if (tv1->v_type == VAR_BLOB 13108a7d6542SBram Moolenaar && tv2->v_type == VAR_BLOB) 13118a7d6542SBram Moolenaar { 13128a7d6542SBram Moolenaar eval_addblob(tv1, tv2); 13138a7d6542SBram Moolenaar clear_tv(tv2); 13148a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 13158a7d6542SBram Moolenaar break; 13168a7d6542SBram Moolenaar } 13178a7d6542SBram Moolenaar } 13188a7d6542SBram Moolenaar #ifdef FEAT_FLOAT 13198a7d6542SBram Moolenaar if (tv1->v_type == VAR_FLOAT) 13208a7d6542SBram Moolenaar { 13218a7d6542SBram Moolenaar f1 = tv1->vval.v_float; 13228a7d6542SBram Moolenaar n1 = 0; 13238a7d6542SBram Moolenaar } 13248a7d6542SBram Moolenaar else 13258a7d6542SBram Moolenaar #endif 13268a7d6542SBram Moolenaar { 13278a7d6542SBram Moolenaar n1 = tv_get_number_chk(tv1, &error); 13288a7d6542SBram Moolenaar if (error) 13298a7d6542SBram Moolenaar goto failed; 13308a7d6542SBram Moolenaar #ifdef FEAT_FLOAT 13318a7d6542SBram Moolenaar if (tv2->v_type == VAR_FLOAT) 13328a7d6542SBram Moolenaar f1 = n1; 13338a7d6542SBram Moolenaar #endif 13348a7d6542SBram Moolenaar } 13358a7d6542SBram Moolenaar #ifdef FEAT_FLOAT 13368a7d6542SBram Moolenaar if (tv2->v_type == VAR_FLOAT) 13378a7d6542SBram Moolenaar { 13388a7d6542SBram Moolenaar f2 = tv2->vval.v_float; 13398a7d6542SBram Moolenaar n2 = 0; 13408a7d6542SBram Moolenaar } 13418a7d6542SBram Moolenaar else 13428a7d6542SBram Moolenaar #endif 13438a7d6542SBram Moolenaar { 13448a7d6542SBram Moolenaar n2 = tv_get_number_chk(tv2, &error); 13458a7d6542SBram Moolenaar if (error) 13468a7d6542SBram Moolenaar goto failed; 13478a7d6542SBram Moolenaar #ifdef FEAT_FLOAT 13488a7d6542SBram Moolenaar if (tv1->v_type == VAR_FLOAT) 13498a7d6542SBram Moolenaar f2 = n2; 13508a7d6542SBram Moolenaar #endif 13518a7d6542SBram Moolenaar } 13528a7d6542SBram Moolenaar #ifdef FEAT_FLOAT 13538a7d6542SBram Moolenaar // if there is a float on either side the result is a float 13548a7d6542SBram Moolenaar if (tv1->v_type == VAR_FLOAT || tv2->v_type == VAR_FLOAT) 13558a7d6542SBram Moolenaar { 13568a7d6542SBram Moolenaar switch (iptr->isn_arg.op.op_type) 13578a7d6542SBram Moolenaar { 13588a7d6542SBram Moolenaar case EXPR_MULT: f1 = f1 * f2; break; 13598a7d6542SBram Moolenaar case EXPR_DIV: f1 = f1 / f2; break; 13608a7d6542SBram Moolenaar case EXPR_SUB: f1 = f1 - f2; break; 13618a7d6542SBram Moolenaar case EXPR_ADD: f1 = f1 + f2; break; 13628a7d6542SBram Moolenaar default: emsg(_(e_modulus)); goto failed; 13638a7d6542SBram Moolenaar } 13648a7d6542SBram Moolenaar clear_tv(tv1); 13658a7d6542SBram Moolenaar clear_tv(tv2); 13668a7d6542SBram Moolenaar tv1->v_type = VAR_FLOAT; 13678a7d6542SBram Moolenaar tv1->vval.v_float = f1; 13688a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 13698a7d6542SBram Moolenaar } 13708a7d6542SBram Moolenaar else 13718a7d6542SBram Moolenaar #endif 13728a7d6542SBram Moolenaar { 13738a7d6542SBram Moolenaar switch (iptr->isn_arg.op.op_type) 13748a7d6542SBram Moolenaar { 13758a7d6542SBram Moolenaar case EXPR_MULT: n1 = n1 * n2; break; 13768a7d6542SBram Moolenaar case EXPR_DIV: n1 = num_divide(n1, n2); break; 13778a7d6542SBram Moolenaar case EXPR_SUB: n1 = n1 - n2; break; 13788a7d6542SBram Moolenaar case EXPR_ADD: n1 = n1 + n2; break; 13798a7d6542SBram Moolenaar default: n1 = num_modulus(n1, n2); break; 13808a7d6542SBram Moolenaar } 13818a7d6542SBram Moolenaar clear_tv(tv1); 13828a7d6542SBram Moolenaar clear_tv(tv2); 13838a7d6542SBram Moolenaar tv1->v_type = VAR_NUMBER; 13848a7d6542SBram Moolenaar tv1->vval.v_number = n1; 13858a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 13868a7d6542SBram Moolenaar } 13878a7d6542SBram Moolenaar } 13888a7d6542SBram Moolenaar break; 13898a7d6542SBram Moolenaar 13908a7d6542SBram Moolenaar case ISN_CONCAT: 13918a7d6542SBram Moolenaar { 13928a7d6542SBram Moolenaar char_u *str1 = STACK_TV_BOT(-2)->vval.v_string; 13938a7d6542SBram Moolenaar char_u *str2 = STACK_TV_BOT(-1)->vval.v_string; 13948a7d6542SBram Moolenaar char_u *res; 13958a7d6542SBram Moolenaar 13968a7d6542SBram Moolenaar res = concat_str(str1, str2); 13978a7d6542SBram Moolenaar clear_tv(STACK_TV_BOT(-2)); 13988a7d6542SBram Moolenaar clear_tv(STACK_TV_BOT(-1)); 13998a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 14008a7d6542SBram Moolenaar STACK_TV_BOT(-1)->vval.v_string = res; 14018a7d6542SBram Moolenaar } 14028a7d6542SBram Moolenaar break; 14038a7d6542SBram Moolenaar 14048a7d6542SBram Moolenaar case ISN_INDEX: 14058a7d6542SBram Moolenaar { 14068a7d6542SBram Moolenaar list_T *list; 14078a7d6542SBram Moolenaar varnumber_T n; 14088a7d6542SBram Moolenaar listitem_T *li; 14098a7d6542SBram Moolenaar 14108a7d6542SBram Moolenaar // list index: list is at stack-2, index at stack-1 14118a7d6542SBram Moolenaar tv = STACK_TV_BOT(-2); 14128a7d6542SBram Moolenaar if (tv->v_type != VAR_LIST) 14138a7d6542SBram Moolenaar { 14148a7d6542SBram Moolenaar emsg(_(e_listreq)); 14158a7d6542SBram Moolenaar goto failed; 14168a7d6542SBram Moolenaar } 14178a7d6542SBram Moolenaar list = tv->vval.v_list; 14188a7d6542SBram Moolenaar 14198a7d6542SBram Moolenaar tv = STACK_TV_BOT(-1); 14208a7d6542SBram Moolenaar if (tv->v_type != VAR_NUMBER) 14218a7d6542SBram Moolenaar { 14228a7d6542SBram Moolenaar emsg(_(e_number_exp)); 14238a7d6542SBram Moolenaar goto failed; 14248a7d6542SBram Moolenaar } 14258a7d6542SBram Moolenaar n = tv->vval.v_number; 14268a7d6542SBram Moolenaar clear_tv(tv); 14278a7d6542SBram Moolenaar if ((li = list_find(list, n)) == NULL) 14288a7d6542SBram Moolenaar { 14298a7d6542SBram Moolenaar semsg(_(e_listidx), n); 14308a7d6542SBram Moolenaar goto failed; 14318a7d6542SBram Moolenaar } 14328a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 14338a7d6542SBram Moolenaar clear_tv(STACK_TV_BOT(-1)); 14348a7d6542SBram Moolenaar copy_tv(&li->li_tv, STACK_TV_BOT(-1)); 14358a7d6542SBram Moolenaar } 14368a7d6542SBram Moolenaar break; 14378a7d6542SBram Moolenaar 14388a7d6542SBram Moolenaar // dict member with string key 14398a7d6542SBram Moolenaar case ISN_MEMBER: 14408a7d6542SBram Moolenaar { 14418a7d6542SBram Moolenaar dict_T *dict; 14428a7d6542SBram Moolenaar dictitem_T *di; 14438a7d6542SBram Moolenaar 14448a7d6542SBram Moolenaar tv = STACK_TV_BOT(-1); 14458a7d6542SBram Moolenaar if (tv->v_type != VAR_DICT || tv->vval.v_dict == NULL) 14468a7d6542SBram Moolenaar { 14478a7d6542SBram Moolenaar emsg(_(e_dictreq)); 14488a7d6542SBram Moolenaar goto failed; 14498a7d6542SBram Moolenaar } 14508a7d6542SBram Moolenaar dict = tv->vval.v_dict; 14518a7d6542SBram Moolenaar 14528a7d6542SBram Moolenaar if ((di = dict_find(dict, iptr->isn_arg.string, -1)) 14538a7d6542SBram Moolenaar == NULL) 14548a7d6542SBram Moolenaar { 14558a7d6542SBram Moolenaar semsg(_(e_dictkey), iptr->isn_arg.string); 14568a7d6542SBram Moolenaar goto failed; 14578a7d6542SBram Moolenaar } 14588a7d6542SBram Moolenaar clear_tv(tv); 14598a7d6542SBram Moolenaar copy_tv(&di->di_tv, tv); 14608a7d6542SBram Moolenaar } 14618a7d6542SBram Moolenaar break; 14628a7d6542SBram Moolenaar 14638a7d6542SBram Moolenaar case ISN_NEGATENR: 14648a7d6542SBram Moolenaar tv = STACK_TV_BOT(-1); 14658a7d6542SBram Moolenaar tv->vval.v_number = -tv->vval.v_number; 14668a7d6542SBram Moolenaar break; 14678a7d6542SBram Moolenaar 14688a7d6542SBram Moolenaar case ISN_CHECKNR: 14698a7d6542SBram Moolenaar { 14708a7d6542SBram Moolenaar int error = FALSE; 14718a7d6542SBram Moolenaar 14728a7d6542SBram Moolenaar tv = STACK_TV_BOT(-1); 14738a7d6542SBram Moolenaar if (check_not_string(tv) == FAIL) 14748a7d6542SBram Moolenaar { 14758a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 14768a7d6542SBram Moolenaar goto failed; 14778a7d6542SBram Moolenaar } 14788a7d6542SBram Moolenaar (void)tv_get_number_chk(tv, &error); 14798a7d6542SBram Moolenaar if (error) 14808a7d6542SBram Moolenaar goto failed; 14818a7d6542SBram Moolenaar } 14828a7d6542SBram Moolenaar break; 14838a7d6542SBram Moolenaar 14848a7d6542SBram Moolenaar case ISN_CHECKTYPE: 14858a7d6542SBram Moolenaar { 14868a7d6542SBram Moolenaar checktype_T *ct = &iptr->isn_arg.type; 14878a7d6542SBram Moolenaar 14888a7d6542SBram Moolenaar tv = STACK_TV_BOT(ct->ct_off); 14898a7d6542SBram Moolenaar if (tv->v_type != ct->ct_type) 14908a7d6542SBram Moolenaar { 14918a7d6542SBram Moolenaar semsg(_("E1029: Expected %s but got %s"), 14928a7d6542SBram Moolenaar vartype_name(ct->ct_type), 14938a7d6542SBram Moolenaar vartype_name(tv->v_type)); 14948a7d6542SBram Moolenaar goto failed; 14958a7d6542SBram Moolenaar } 14968a7d6542SBram Moolenaar } 14978a7d6542SBram Moolenaar break; 14988a7d6542SBram Moolenaar 14998a7d6542SBram Moolenaar case ISN_2BOOL: 15008a7d6542SBram Moolenaar { 15018a7d6542SBram Moolenaar int n; 15028a7d6542SBram Moolenaar 15038a7d6542SBram Moolenaar tv = STACK_TV_BOT(-1); 15048a7d6542SBram Moolenaar n = tv2bool(tv); 15058a7d6542SBram Moolenaar if (iptr->isn_arg.number) // invert 15068a7d6542SBram Moolenaar n = !n; 15078a7d6542SBram Moolenaar clear_tv(tv); 15088a7d6542SBram Moolenaar tv->v_type = VAR_BOOL; 15098a7d6542SBram Moolenaar tv->vval.v_number = n ? VVAL_TRUE : VVAL_FALSE; 15108a7d6542SBram Moolenaar } 15118a7d6542SBram Moolenaar break; 15128a7d6542SBram Moolenaar 15138a7d6542SBram Moolenaar case ISN_2STRING: 15148a7d6542SBram Moolenaar { 15158a7d6542SBram Moolenaar char_u *str; 15168a7d6542SBram Moolenaar 15178a7d6542SBram Moolenaar tv = STACK_TV_BOT(iptr->isn_arg.number); 15188a7d6542SBram Moolenaar if (tv->v_type != VAR_STRING) 15198a7d6542SBram Moolenaar { 15208a7d6542SBram Moolenaar str = typval_tostring(tv); 15218a7d6542SBram Moolenaar clear_tv(tv); 15228a7d6542SBram Moolenaar tv->v_type = VAR_STRING; 15238a7d6542SBram Moolenaar tv->vval.v_string = str; 15248a7d6542SBram Moolenaar } 15258a7d6542SBram Moolenaar } 15268a7d6542SBram Moolenaar break; 15278a7d6542SBram Moolenaar 15288a7d6542SBram Moolenaar case ISN_DROP: 15298a7d6542SBram Moolenaar --ectx.ec_stack.ga_len; 15308a7d6542SBram Moolenaar clear_tv(STACK_TV_BOT(0)); 15318a7d6542SBram Moolenaar break; 15328a7d6542SBram Moolenaar } 15338a7d6542SBram Moolenaar } 15348a7d6542SBram Moolenaar 15358a7d6542SBram Moolenaar done: 15368a7d6542SBram Moolenaar // function finished, get result from the stack. 15378a7d6542SBram Moolenaar tv = STACK_TV_BOT(-1); 15388a7d6542SBram Moolenaar *rettv = *tv; 15398a7d6542SBram Moolenaar tv->v_type = VAR_UNKNOWN; 15408a7d6542SBram Moolenaar ret = OK; 15418a7d6542SBram Moolenaar 15428a7d6542SBram Moolenaar failed: 15438a7d6542SBram Moolenaar for (idx = 0; idx < ectx.ec_stack.ga_len; ++idx) 15448a7d6542SBram Moolenaar clear_tv(STACK_TV(idx)); 15458a7d6542SBram Moolenaar vim_free(ectx.ec_stack.ga_data); 15468a7d6542SBram Moolenaar return ret; 15478a7d6542SBram Moolenaar } 15488a7d6542SBram Moolenaar 15498a7d6542SBram Moolenaar #define DISASSEMBLE 1 15508a7d6542SBram Moolenaar 15518a7d6542SBram Moolenaar /* 15528a7d6542SBram Moolenaar * ":dissassemble". 15538a7d6542SBram Moolenaar */ 15548a7d6542SBram Moolenaar void 15558a7d6542SBram Moolenaar ex_disassemble(exarg_T *eap) 15568a7d6542SBram Moolenaar { 15578a7d6542SBram Moolenaar #ifdef DISASSEMBLE 15580f18b6d1SBram Moolenaar char_u *fname; 15590f18b6d1SBram Moolenaar ufunc_T *ufunc; 15608a7d6542SBram Moolenaar dfunc_T *dfunc; 15618a7d6542SBram Moolenaar isn_T *instr; 15628a7d6542SBram Moolenaar int current; 15638a7d6542SBram Moolenaar int line_idx = 0; 15648a7d6542SBram Moolenaar int prev_current = 0; 15658a7d6542SBram Moolenaar 15660f18b6d1SBram Moolenaar fname = trans_function_name(&eap->arg, FALSE, 15670f18b6d1SBram Moolenaar TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF, NULL, NULL); 15680f18b6d1SBram Moolenaar ufunc = find_func(fname, NULL); 15690f18b6d1SBram Moolenaar vim_free(fname); 15708a7d6542SBram Moolenaar if (ufunc == NULL) 15718a7d6542SBram Moolenaar { 15720f18b6d1SBram Moolenaar semsg("E1061: Cannot find function %s", eap->arg); 15738a7d6542SBram Moolenaar return; 15748a7d6542SBram Moolenaar } 15758a7d6542SBram Moolenaar if (ufunc->uf_dfunc_idx < 0) 15768a7d6542SBram Moolenaar { 15770f18b6d1SBram Moolenaar semsg("E1062: Function %s is not compiled", eap->arg); 15788a7d6542SBram Moolenaar return; 15798a7d6542SBram Moolenaar } 15808a7d6542SBram Moolenaar if (ufunc->uf_name_exp != NULL) 15818a7d6542SBram Moolenaar msg((char *)ufunc->uf_name_exp); 15828a7d6542SBram Moolenaar else 15838a7d6542SBram Moolenaar msg((char *)ufunc->uf_name); 15848a7d6542SBram Moolenaar 15858a7d6542SBram Moolenaar dfunc = ((dfunc_T *)def_functions.ga_data) + ufunc->uf_dfunc_idx; 15868a7d6542SBram Moolenaar instr = dfunc->df_instr; 15878a7d6542SBram Moolenaar for (current = 0; current < dfunc->df_instr_count; ++current) 15888a7d6542SBram Moolenaar { 15898a7d6542SBram Moolenaar isn_T *iptr = &instr[current]; 15908a7d6542SBram Moolenaar 15918a7d6542SBram Moolenaar while (line_idx < iptr->isn_lnum && line_idx < ufunc->uf_lines.ga_len) 15928a7d6542SBram Moolenaar { 15938a7d6542SBram Moolenaar if (current > prev_current) 15948a7d6542SBram Moolenaar { 15958a7d6542SBram Moolenaar msg_puts("\n\n"); 15968a7d6542SBram Moolenaar prev_current = current; 15978a7d6542SBram Moolenaar } 15988a7d6542SBram Moolenaar msg(((char **)ufunc->uf_lines.ga_data)[line_idx++]); 15998a7d6542SBram Moolenaar } 16008a7d6542SBram Moolenaar 16018a7d6542SBram Moolenaar switch (iptr->isn_type) 16028a7d6542SBram Moolenaar { 16038a7d6542SBram Moolenaar case ISN_EXEC: 16048a7d6542SBram Moolenaar smsg("%4d EXEC %s", current, iptr->isn_arg.string); 16058a7d6542SBram Moolenaar break; 16068a7d6542SBram Moolenaar case ISN_ECHO: 16078a7d6542SBram Moolenaar { 16088a7d6542SBram Moolenaar echo_T *echo = &iptr->isn_arg.echo; 16098a7d6542SBram Moolenaar 16108a7d6542SBram Moolenaar smsg("%4d %s %d", current, 16118a7d6542SBram Moolenaar echo->echo_with_white ? "ECHO" : "ECHON", 16128a7d6542SBram Moolenaar echo->echo_count); 16138a7d6542SBram Moolenaar } 16148a7d6542SBram Moolenaar break; 16158a7d6542SBram Moolenaar case ISN_LOAD: 16168a7d6542SBram Moolenaar if (iptr->isn_arg.number < 0) 16178a7d6542SBram Moolenaar smsg("%4d LOAD arg[%lld]", current, 16188a7d6542SBram Moolenaar iptr->isn_arg.number + STACK_FRAME_SIZE); 16198a7d6542SBram Moolenaar else 16208a7d6542SBram Moolenaar smsg("%4d LOAD $%lld", current, iptr->isn_arg.number); 16218a7d6542SBram Moolenaar break; 16228a7d6542SBram Moolenaar case ISN_LOADV: 16238a7d6542SBram Moolenaar smsg("%4d LOADV v:%s", current, 16248a7d6542SBram Moolenaar get_vim_var_name(iptr->isn_arg.number)); 16258a7d6542SBram Moolenaar break; 16268a7d6542SBram Moolenaar case ISN_LOADSCRIPT: 16278a7d6542SBram Moolenaar { 16288a7d6542SBram Moolenaar scriptitem_T *si = 162921b9e977SBram Moolenaar SCRIPT_ITEM(iptr->isn_arg.script.script_sid); 16308a7d6542SBram Moolenaar svar_T *sv = ((svar_T *)si->sn_var_vals.ga_data) 16318a7d6542SBram Moolenaar + iptr->isn_arg.script.script_idx; 16328a7d6542SBram Moolenaar 16338a7d6542SBram Moolenaar smsg("%4d LOADSCRIPT %s from %s", current, 16348a7d6542SBram Moolenaar sv->sv_name, si->sn_name); 16358a7d6542SBram Moolenaar } 16368a7d6542SBram Moolenaar break; 16378a7d6542SBram Moolenaar case ISN_LOADS: 16388a7d6542SBram Moolenaar { 1639b283a8a6SBram Moolenaar scriptitem_T *si = SCRIPT_ITEM( 1640b283a8a6SBram Moolenaar iptr->isn_arg.loadstore.ls_sid); 16418a7d6542SBram Moolenaar 16428a7d6542SBram Moolenaar smsg("%4d LOADS s:%s from %s", current, 16438a7d6542SBram Moolenaar iptr->isn_arg.string, si->sn_name); 16448a7d6542SBram Moolenaar } 16458a7d6542SBram Moolenaar break; 16468a7d6542SBram Moolenaar case ISN_LOADG: 16478a7d6542SBram Moolenaar smsg("%4d LOADG g:%s", current, iptr->isn_arg.string); 16488a7d6542SBram Moolenaar break; 16498a7d6542SBram Moolenaar case ISN_LOADOPT: 16508a7d6542SBram Moolenaar smsg("%4d LOADOPT %s", current, iptr->isn_arg.string); 16518a7d6542SBram Moolenaar break; 16528a7d6542SBram Moolenaar case ISN_LOADENV: 16538a7d6542SBram Moolenaar smsg("%4d LOADENV %s", current, iptr->isn_arg.string); 16548a7d6542SBram Moolenaar break; 16558a7d6542SBram Moolenaar case ISN_LOADREG: 16568a7d6542SBram Moolenaar smsg("%4d LOADREG @%c", current, iptr->isn_arg.number); 16578a7d6542SBram Moolenaar break; 16588a7d6542SBram Moolenaar 16598a7d6542SBram Moolenaar case ISN_STORE: 16608a7d6542SBram Moolenaar smsg("%4d STORE $%lld", current, iptr->isn_arg.number); 16618a7d6542SBram Moolenaar break; 1662b283a8a6SBram Moolenaar case ISN_STOREV: 1663b283a8a6SBram Moolenaar smsg("%4d STOREV v:%s", current, 1664b283a8a6SBram Moolenaar get_vim_var_name(iptr->isn_arg.number)); 1665b283a8a6SBram Moolenaar break; 16668a7d6542SBram Moolenaar case ISN_STOREG: 1667b283a8a6SBram Moolenaar smsg("%4d STOREG %s", current, iptr->isn_arg.string); 1668b283a8a6SBram Moolenaar break; 1669b283a8a6SBram Moolenaar case ISN_STORES: 1670b283a8a6SBram Moolenaar { 1671b283a8a6SBram Moolenaar scriptitem_T *si = SCRIPT_ITEM( 1672b283a8a6SBram Moolenaar iptr->isn_arg.loadstore.ls_sid); 1673b283a8a6SBram Moolenaar 1674b283a8a6SBram Moolenaar smsg("%4d STORES s:%s in %s", current, 1675b283a8a6SBram Moolenaar iptr->isn_arg.string, si->sn_name); 1676b283a8a6SBram Moolenaar } 16778a7d6542SBram Moolenaar break; 16788a7d6542SBram Moolenaar case ISN_STORESCRIPT: 16798a7d6542SBram Moolenaar { 16808a7d6542SBram Moolenaar scriptitem_T *si = 168121b9e977SBram Moolenaar SCRIPT_ITEM(iptr->isn_arg.script.script_sid); 16828a7d6542SBram Moolenaar svar_T *sv = ((svar_T *)si->sn_var_vals.ga_data) 16838a7d6542SBram Moolenaar + iptr->isn_arg.script.script_idx; 16848a7d6542SBram Moolenaar 16858a7d6542SBram Moolenaar smsg("%4d STORESCRIPT %s in %s", current, 16868a7d6542SBram Moolenaar sv->sv_name, si->sn_name); 16878a7d6542SBram Moolenaar } 16888a7d6542SBram Moolenaar break; 16898a7d6542SBram Moolenaar case ISN_STOREOPT: 16908a7d6542SBram Moolenaar smsg("%4d STOREOPT &%s", current, 16918a7d6542SBram Moolenaar iptr->isn_arg.storeopt.so_name); 16928a7d6542SBram Moolenaar break; 1693b283a8a6SBram Moolenaar case ISN_STOREENV: 1694b283a8a6SBram Moolenaar smsg("%4d STOREENV $%s", current, iptr->isn_arg.string); 1695b283a8a6SBram Moolenaar break; 1696b283a8a6SBram Moolenaar case ISN_STOREREG: 1697b283a8a6SBram Moolenaar smsg("%4d STOREREG @%c", current, iptr->isn_arg.number); 1698b283a8a6SBram Moolenaar break; 16998a7d6542SBram Moolenaar case ISN_STORENR: 17008a7d6542SBram Moolenaar smsg("%4d STORE %lld in $%d", current, 17018a7d6542SBram Moolenaar iptr->isn_arg.storenr.str_val, 17028a7d6542SBram Moolenaar iptr->isn_arg.storenr.str_idx); 17038a7d6542SBram Moolenaar break; 17048a7d6542SBram Moolenaar 17058a7d6542SBram Moolenaar // constants 17068a7d6542SBram Moolenaar case ISN_PUSHNR: 17078a7d6542SBram Moolenaar smsg("%4d PUSHNR %lld", current, iptr->isn_arg.number); 17088a7d6542SBram Moolenaar break; 17098a7d6542SBram Moolenaar case ISN_PUSHBOOL: 17108a7d6542SBram Moolenaar case ISN_PUSHSPEC: 17118a7d6542SBram Moolenaar smsg("%4d PUSH %s", current, 17128a7d6542SBram Moolenaar get_var_special_name(iptr->isn_arg.number)); 17138a7d6542SBram Moolenaar break; 17148a7d6542SBram Moolenaar case ISN_PUSHF: 1715a5d5953dSBram Moolenaar #ifdef FEAT_FLOAT 17168a7d6542SBram Moolenaar smsg("%4d PUSHF %g", current, iptr->isn_arg.fnumber); 1717a5d5953dSBram Moolenaar #endif 17188a7d6542SBram Moolenaar break; 17198a7d6542SBram Moolenaar case ISN_PUSHS: 17208a7d6542SBram Moolenaar smsg("%4d PUSHS \"%s\"", current, iptr->isn_arg.string); 17218a7d6542SBram Moolenaar break; 17228a7d6542SBram Moolenaar case ISN_PUSHBLOB: 17238a7d6542SBram Moolenaar { 17248a7d6542SBram Moolenaar char_u *r; 17258a7d6542SBram Moolenaar char_u numbuf[NUMBUFLEN]; 17268a7d6542SBram Moolenaar char_u *tofree; 17278a7d6542SBram Moolenaar 17288a7d6542SBram Moolenaar r = blob2string(iptr->isn_arg.blob, &tofree, numbuf); 1729*ff80cb68SBram Moolenaar smsg("%4d PUSHBLOB %s", current, r); 17308a7d6542SBram Moolenaar vim_free(tofree); 17318a7d6542SBram Moolenaar } 17328a7d6542SBram Moolenaar break; 17338a7d6542SBram Moolenaar case ISN_PUSHEXC: 17348a7d6542SBram Moolenaar smsg("%4d PUSH v:exception", current); 17358a7d6542SBram Moolenaar break; 17368a7d6542SBram Moolenaar case ISN_NEWLIST: 17378a7d6542SBram Moolenaar smsg("%4d NEWLIST size %lld", current, iptr->isn_arg.number); 17388a7d6542SBram Moolenaar break; 17398a7d6542SBram Moolenaar case ISN_NEWDICT: 17408a7d6542SBram Moolenaar smsg("%4d NEWDICT size %lld", current, iptr->isn_arg.number); 17418a7d6542SBram Moolenaar break; 17428a7d6542SBram Moolenaar 17438a7d6542SBram Moolenaar // function call 17448a7d6542SBram Moolenaar case ISN_BCALL: 17458a7d6542SBram Moolenaar { 17468a7d6542SBram Moolenaar cbfunc_T *cbfunc = &iptr->isn_arg.bfunc; 17478a7d6542SBram Moolenaar 17488a7d6542SBram Moolenaar smsg("%4d BCALL %s(argc %d)", current, 17498a7d6542SBram Moolenaar internal_func_name(cbfunc->cbf_idx), 17508a7d6542SBram Moolenaar cbfunc->cbf_argcount); 17518a7d6542SBram Moolenaar } 17528a7d6542SBram Moolenaar break; 17538a7d6542SBram Moolenaar case ISN_DCALL: 17548a7d6542SBram Moolenaar { 17558a7d6542SBram Moolenaar cdfunc_T *cdfunc = &iptr->isn_arg.dfunc; 17568a7d6542SBram Moolenaar dfunc_T *df = ((dfunc_T *)def_functions.ga_data) 17578a7d6542SBram Moolenaar + cdfunc->cdf_idx; 17588a7d6542SBram Moolenaar 17598a7d6542SBram Moolenaar smsg("%4d DCALL %s(argc %d)", current, 17608a7d6542SBram Moolenaar df->df_ufunc->uf_name_exp != NULL 17618a7d6542SBram Moolenaar ? df->df_ufunc->uf_name_exp 17628a7d6542SBram Moolenaar : df->df_ufunc->uf_name, cdfunc->cdf_argcount); 17638a7d6542SBram Moolenaar } 17648a7d6542SBram Moolenaar break; 17658a7d6542SBram Moolenaar case ISN_UCALL: 17668a7d6542SBram Moolenaar { 17678a7d6542SBram Moolenaar cufunc_T *cufunc = &iptr->isn_arg.ufunc; 17688a7d6542SBram Moolenaar 17698a7d6542SBram Moolenaar smsg("%4d UCALL %s(argc %d)", current, 17708a7d6542SBram Moolenaar cufunc->cuf_name, cufunc->cuf_argcount); 17718a7d6542SBram Moolenaar } 17728a7d6542SBram Moolenaar break; 17738a7d6542SBram Moolenaar case ISN_PCALL: 17748a7d6542SBram Moolenaar { 17758a7d6542SBram Moolenaar cpfunc_T *cpfunc = &iptr->isn_arg.pfunc; 17768a7d6542SBram Moolenaar 17778a7d6542SBram Moolenaar smsg("%4d PCALL%s (argc %d)", current, 17788a7d6542SBram Moolenaar cpfunc->cpf_top ? " top" : "", cpfunc->cpf_argcount); 17798a7d6542SBram Moolenaar } 17808a7d6542SBram Moolenaar break; 17818a7d6542SBram Moolenaar case ISN_RETURN: 17828a7d6542SBram Moolenaar smsg("%4d RETURN", current); 17838a7d6542SBram Moolenaar break; 17848a7d6542SBram Moolenaar case ISN_FUNCREF: 17858a7d6542SBram Moolenaar { 17868a7d6542SBram Moolenaar dfunc_T *df = ((dfunc_T *)def_functions.ga_data) 17878a7d6542SBram Moolenaar + iptr->isn_arg.number; 17888a7d6542SBram Moolenaar 17898a7d6542SBram Moolenaar smsg("%4d FUNCREF %s", current, df->df_ufunc->uf_name); 17908a7d6542SBram Moolenaar } 17918a7d6542SBram Moolenaar break; 17928a7d6542SBram Moolenaar 17938a7d6542SBram Moolenaar case ISN_JUMP: 17948a7d6542SBram Moolenaar { 17958a7d6542SBram Moolenaar char *when = "?"; 17968a7d6542SBram Moolenaar 17978a7d6542SBram Moolenaar switch (iptr->isn_arg.jump.jump_when) 17988a7d6542SBram Moolenaar { 17998a7d6542SBram Moolenaar case JUMP_ALWAYS: 18008a7d6542SBram Moolenaar when = "JUMP"; 18018a7d6542SBram Moolenaar break; 18028a7d6542SBram Moolenaar case JUMP_IF_TRUE: 18038a7d6542SBram Moolenaar when = "JUMP_IF_TRUE"; 18048a7d6542SBram Moolenaar break; 18058a7d6542SBram Moolenaar case JUMP_AND_KEEP_IF_TRUE: 18068a7d6542SBram Moolenaar when = "JUMP_AND_KEEP_IF_TRUE"; 18078a7d6542SBram Moolenaar break; 18088a7d6542SBram Moolenaar case JUMP_IF_FALSE: 18098a7d6542SBram Moolenaar when = "JUMP_IF_FALSE"; 18108a7d6542SBram Moolenaar break; 18118a7d6542SBram Moolenaar case JUMP_AND_KEEP_IF_FALSE: 18128a7d6542SBram Moolenaar when = "JUMP_AND_KEEP_IF_FALSE"; 18138a7d6542SBram Moolenaar break; 18148a7d6542SBram Moolenaar } 18158a7d6542SBram Moolenaar smsg("%4d %s -> %lld", current, when, 18168a7d6542SBram Moolenaar iptr->isn_arg.jump.jump_where); 18178a7d6542SBram Moolenaar } 18188a7d6542SBram Moolenaar break; 18198a7d6542SBram Moolenaar 18208a7d6542SBram Moolenaar case ISN_FOR: 18218a7d6542SBram Moolenaar { 18228a7d6542SBram Moolenaar forloop_T *forloop = &iptr->isn_arg.forloop; 18238a7d6542SBram Moolenaar 18248a7d6542SBram Moolenaar smsg("%4d FOR $%d -> %d", current, 18258a7d6542SBram Moolenaar forloop->for_idx, forloop->for_end); 18268a7d6542SBram Moolenaar } 18278a7d6542SBram Moolenaar break; 18288a7d6542SBram Moolenaar 18298a7d6542SBram Moolenaar case ISN_TRY: 18308a7d6542SBram Moolenaar { 18318a7d6542SBram Moolenaar try_T *try = &iptr->isn_arg.try; 18328a7d6542SBram Moolenaar 18338a7d6542SBram Moolenaar smsg("%4d TRY catch -> %d, finally -> %d", current, 18348a7d6542SBram Moolenaar try->try_catch, try->try_finally); 18358a7d6542SBram Moolenaar } 18368a7d6542SBram Moolenaar break; 18378a7d6542SBram Moolenaar case ISN_CATCH: 18388a7d6542SBram Moolenaar // TODO 18398a7d6542SBram Moolenaar smsg("%4d CATCH", current); 18408a7d6542SBram Moolenaar break; 18418a7d6542SBram Moolenaar case ISN_ENDTRY: 18428a7d6542SBram Moolenaar smsg("%4d ENDTRY", current); 18438a7d6542SBram Moolenaar break; 18448a7d6542SBram Moolenaar case ISN_THROW: 18458a7d6542SBram Moolenaar smsg("%4d THROW", current); 18468a7d6542SBram Moolenaar break; 18478a7d6542SBram Moolenaar 18488a7d6542SBram Moolenaar // expression operations on number 18498a7d6542SBram Moolenaar case ISN_OPNR: 18508a7d6542SBram Moolenaar case ISN_OPFLOAT: 18518a7d6542SBram Moolenaar case ISN_OPANY: 18528a7d6542SBram Moolenaar { 18538a7d6542SBram Moolenaar char *what; 18548a7d6542SBram Moolenaar char *ins; 18558a7d6542SBram Moolenaar 18568a7d6542SBram Moolenaar switch (iptr->isn_arg.op.op_type) 18578a7d6542SBram Moolenaar { 18588a7d6542SBram Moolenaar case EXPR_MULT: what = "*"; break; 18598a7d6542SBram Moolenaar case EXPR_DIV: what = "/"; break; 18608a7d6542SBram Moolenaar case EXPR_REM: what = "%"; break; 18618a7d6542SBram Moolenaar case EXPR_SUB: what = "-"; break; 18628a7d6542SBram Moolenaar case EXPR_ADD: what = "+"; break; 18638a7d6542SBram Moolenaar default: what = "???"; break; 18648a7d6542SBram Moolenaar } 18658a7d6542SBram Moolenaar switch (iptr->isn_type) 18668a7d6542SBram Moolenaar { 18678a7d6542SBram Moolenaar case ISN_OPNR: ins = "OPNR"; break; 18688a7d6542SBram Moolenaar case ISN_OPFLOAT: ins = "OPFLOAT"; break; 18698a7d6542SBram Moolenaar case ISN_OPANY: ins = "OPANY"; break; 18708a7d6542SBram Moolenaar default: ins = "???"; break; 18718a7d6542SBram Moolenaar } 18728a7d6542SBram Moolenaar smsg("%4d %s %s", current, ins, what); 18738a7d6542SBram Moolenaar } 18748a7d6542SBram Moolenaar break; 18758a7d6542SBram Moolenaar 18768a7d6542SBram Moolenaar case ISN_COMPAREBOOL: 18778a7d6542SBram Moolenaar case ISN_COMPARESPECIAL: 18788a7d6542SBram Moolenaar case ISN_COMPARENR: 18798a7d6542SBram Moolenaar case ISN_COMPAREFLOAT: 18808a7d6542SBram Moolenaar case ISN_COMPARESTRING: 18818a7d6542SBram Moolenaar case ISN_COMPAREBLOB: 18828a7d6542SBram Moolenaar case ISN_COMPARELIST: 18838a7d6542SBram Moolenaar case ISN_COMPAREDICT: 18848a7d6542SBram Moolenaar case ISN_COMPAREFUNC: 18858a7d6542SBram Moolenaar case ISN_COMPAREPARTIAL: 18868a7d6542SBram Moolenaar case ISN_COMPAREANY: 18878a7d6542SBram Moolenaar { 18888a7d6542SBram Moolenaar char *p; 18898a7d6542SBram Moolenaar char buf[10]; 18908a7d6542SBram Moolenaar char *type; 18918a7d6542SBram Moolenaar 18928a7d6542SBram Moolenaar switch (iptr->isn_arg.op.op_type) 18938a7d6542SBram Moolenaar { 18948a7d6542SBram Moolenaar case EXPR_EQUAL: p = "=="; break; 18958a7d6542SBram Moolenaar case EXPR_NEQUAL: p = "!="; break; 18968a7d6542SBram Moolenaar case EXPR_GREATER: p = ">"; break; 18978a7d6542SBram Moolenaar case EXPR_GEQUAL: p = ">="; break; 18988a7d6542SBram Moolenaar case EXPR_SMALLER: p = "<"; break; 18998a7d6542SBram Moolenaar case EXPR_SEQUAL: p = "<="; break; 19008a7d6542SBram Moolenaar case EXPR_MATCH: p = "=~"; break; 19018a7d6542SBram Moolenaar case EXPR_IS: p = "is"; break; 19028a7d6542SBram Moolenaar case EXPR_ISNOT: p = "isnot"; break; 19038a7d6542SBram Moolenaar case EXPR_NOMATCH: p = "!~"; break; 19048a7d6542SBram Moolenaar default: p = "???"; break; 19058a7d6542SBram Moolenaar } 19068a7d6542SBram Moolenaar STRCPY(buf, p); 19078a7d6542SBram Moolenaar if (iptr->isn_arg.op.op_ic == TRUE) 19088a7d6542SBram Moolenaar strcat(buf, "?"); 19098a7d6542SBram Moolenaar switch(iptr->isn_type) 19108a7d6542SBram Moolenaar { 19118a7d6542SBram Moolenaar case ISN_COMPAREBOOL: type = "COMPAREBOOL"; break; 19128a7d6542SBram Moolenaar case ISN_COMPARESPECIAL: 19138a7d6542SBram Moolenaar type = "COMPARESPECIAL"; break; 19148a7d6542SBram Moolenaar case ISN_COMPARENR: type = "COMPARENR"; break; 19158a7d6542SBram Moolenaar case ISN_COMPAREFLOAT: type = "COMPAREFLOAT"; break; 19168a7d6542SBram Moolenaar case ISN_COMPARESTRING: 19178a7d6542SBram Moolenaar type = "COMPARESTRING"; break; 19188a7d6542SBram Moolenaar case ISN_COMPAREBLOB: type = "COMPAREBLOB"; break; 19198a7d6542SBram Moolenaar case ISN_COMPARELIST: type = "COMPARELIST"; break; 19208a7d6542SBram Moolenaar case ISN_COMPAREDICT: type = "COMPAREDICT"; break; 19218a7d6542SBram Moolenaar case ISN_COMPAREFUNC: type = "COMPAREFUNC"; break; 19228a7d6542SBram Moolenaar case ISN_COMPAREPARTIAL: 19238a7d6542SBram Moolenaar type = "COMPAREPARTIAL"; break; 19248a7d6542SBram Moolenaar case ISN_COMPAREANY: type = "COMPAREANY"; break; 19258a7d6542SBram Moolenaar default: type = "???"; break; 19268a7d6542SBram Moolenaar } 19278a7d6542SBram Moolenaar 19288a7d6542SBram Moolenaar smsg("%4d %s %s", current, type, buf); 19298a7d6542SBram Moolenaar } 19308a7d6542SBram Moolenaar break; 19318a7d6542SBram Moolenaar 19328a7d6542SBram Moolenaar case ISN_ADDLIST: smsg("%4d ADDLIST", current); break; 19338a7d6542SBram Moolenaar case ISN_ADDBLOB: smsg("%4d ADDBLOB", current); break; 19348a7d6542SBram Moolenaar 19358a7d6542SBram Moolenaar // expression operations 19368a7d6542SBram Moolenaar case ISN_CONCAT: smsg("%4d CONCAT", current); break; 19378a7d6542SBram Moolenaar case ISN_INDEX: smsg("%4d INDEX", current); break; 19388a7d6542SBram Moolenaar case ISN_MEMBER: smsg("%4d MEMBER %s", current, 19398a7d6542SBram Moolenaar iptr->isn_arg.string); break; 19408a7d6542SBram Moolenaar case ISN_NEGATENR: smsg("%4d NEGATENR", current); break; 19418a7d6542SBram Moolenaar 19428a7d6542SBram Moolenaar case ISN_CHECKNR: smsg("%4d CHECKNR", current); break; 19438a7d6542SBram Moolenaar case ISN_CHECKTYPE: smsg("%4d CHECKTYPE %s stack[%d]", current, 19448a7d6542SBram Moolenaar vartype_name(iptr->isn_arg.type.ct_type), 19458a7d6542SBram Moolenaar iptr->isn_arg.type.ct_off); 19468a7d6542SBram Moolenaar break; 19478a7d6542SBram Moolenaar case ISN_2BOOL: if (iptr->isn_arg.number) 19488a7d6542SBram Moolenaar smsg("%4d INVERT (!val)", current); 19498a7d6542SBram Moolenaar else 19508a7d6542SBram Moolenaar smsg("%4d 2BOOL (!!val)", current); 19518a7d6542SBram Moolenaar break; 19528a7d6542SBram Moolenaar case ISN_2STRING: smsg("%4d 2STRING stack[%d]", current, 19538a7d6542SBram Moolenaar iptr->isn_arg.number); 19548a7d6542SBram Moolenaar break; 19558a7d6542SBram Moolenaar 19568a7d6542SBram Moolenaar case ISN_DROP: smsg("%4d DROP", current); break; 19578a7d6542SBram Moolenaar } 19588a7d6542SBram Moolenaar } 19598a7d6542SBram Moolenaar #endif 19608a7d6542SBram Moolenaar } 19618a7d6542SBram Moolenaar 19628a7d6542SBram Moolenaar /* 19638a7d6542SBram Moolenaar * Return TRUE when "tv" is not falsey: non-zero, non-empty string, non-empty 19648a7d6542SBram Moolenaar * list, etc. Mostly like what JavaScript does, except that empty list and 19658a7d6542SBram Moolenaar * empty dictionary are FALSE. 19668a7d6542SBram Moolenaar */ 19678a7d6542SBram Moolenaar int 19688a7d6542SBram Moolenaar tv2bool(typval_T *tv) 19698a7d6542SBram Moolenaar { 19708a7d6542SBram Moolenaar switch (tv->v_type) 19718a7d6542SBram Moolenaar { 19728a7d6542SBram Moolenaar case VAR_NUMBER: 19738a7d6542SBram Moolenaar return tv->vval.v_number != 0; 19748a7d6542SBram Moolenaar case VAR_FLOAT: 19758a7d6542SBram Moolenaar #ifdef FEAT_FLOAT 19768a7d6542SBram Moolenaar return tv->vval.v_float != 0.0; 19778a7d6542SBram Moolenaar #else 19788a7d6542SBram Moolenaar break; 19798a7d6542SBram Moolenaar #endif 19808a7d6542SBram Moolenaar case VAR_PARTIAL: 19818a7d6542SBram Moolenaar return tv->vval.v_partial != NULL; 19828a7d6542SBram Moolenaar case VAR_FUNC: 19838a7d6542SBram Moolenaar case VAR_STRING: 19848a7d6542SBram Moolenaar return tv->vval.v_string != NULL && *tv->vval.v_string != NUL; 19858a7d6542SBram Moolenaar case VAR_LIST: 19868a7d6542SBram Moolenaar return tv->vval.v_list != NULL && tv->vval.v_list->lv_len > 0; 19878a7d6542SBram Moolenaar case VAR_DICT: 19888a7d6542SBram Moolenaar return tv->vval.v_dict != NULL 19898a7d6542SBram Moolenaar && tv->vval.v_dict->dv_hashtab.ht_used > 0; 19908a7d6542SBram Moolenaar case VAR_BOOL: 19918a7d6542SBram Moolenaar case VAR_SPECIAL: 19928a7d6542SBram Moolenaar return tv->vval.v_number == VVAL_TRUE ? TRUE : FALSE; 19938a7d6542SBram Moolenaar case VAR_JOB: 19948a7d6542SBram Moolenaar #ifdef FEAT_JOB_CHANNEL 19958a7d6542SBram Moolenaar return tv->vval.v_job != NULL; 19968a7d6542SBram Moolenaar #else 19978a7d6542SBram Moolenaar break; 19988a7d6542SBram Moolenaar #endif 19998a7d6542SBram Moolenaar case VAR_CHANNEL: 20008a7d6542SBram Moolenaar #ifdef FEAT_JOB_CHANNEL 20018a7d6542SBram Moolenaar return tv->vval.v_channel != NULL; 20028a7d6542SBram Moolenaar #else 20038a7d6542SBram Moolenaar break; 20048a7d6542SBram Moolenaar #endif 20058a7d6542SBram Moolenaar case VAR_BLOB: 20068a7d6542SBram Moolenaar return tv->vval.v_blob != NULL && tv->vval.v_blob->bv_ga.ga_len > 0; 20078a7d6542SBram Moolenaar case VAR_UNKNOWN: 20088a7d6542SBram Moolenaar case VAR_VOID: 20098a7d6542SBram Moolenaar break; 20108a7d6542SBram Moolenaar } 20118a7d6542SBram Moolenaar return FALSE; 20128a7d6542SBram Moolenaar } 20138a7d6542SBram Moolenaar 20148a7d6542SBram Moolenaar /* 20158a7d6542SBram Moolenaar * If "tv" is a string give an error and return FAIL. 20168a7d6542SBram Moolenaar */ 20178a7d6542SBram Moolenaar int 20188a7d6542SBram Moolenaar check_not_string(typval_T *tv) 20198a7d6542SBram Moolenaar { 20208a7d6542SBram Moolenaar if (tv->v_type == VAR_STRING) 20218a7d6542SBram Moolenaar { 20228a7d6542SBram Moolenaar emsg(_("E1030: Using a String as a Number")); 20238a7d6542SBram Moolenaar clear_tv(tv); 20248a7d6542SBram Moolenaar return FAIL; 20258a7d6542SBram Moolenaar } 20268a7d6542SBram Moolenaar return OK; 20278a7d6542SBram Moolenaar } 20288a7d6542SBram Moolenaar 20298a7d6542SBram Moolenaar 20308a7d6542SBram Moolenaar #endif // FEAT_EVAL 2031