blob: 0d17b9ed79d28c697625758427496b60d052c8cf [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * evalfunc.c: Builtin functions
12 */
13#define USING_FLOAT_STUFF
14
15#include "vim.h"
16
17#if defined(FEAT_EVAL) || defined(PROTO)
18
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020019#ifdef VMS
20# include <float.h>
21#endif
22
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020023#ifdef FEAT_FLOAT
24static void f_abs(typval_T *argvars, typval_T *rettv);
25static void f_acos(typval_T *argvars, typval_T *rettv);
26#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020027static void f_and(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020028#ifdef FEAT_FLOAT
29static void f_asin(typval_T *argvars, typval_T *rettv);
30static void f_atan(typval_T *argvars, typval_T *rettv);
31static void f_atan2(typval_T *argvars, typval_T *rettv);
32#endif
Bram Moolenaar59716a22017-03-01 20:32:44 +010033#ifdef FEAT_BEVAL
Bram Moolenaarbe0a2592019-05-09 13:50:16 +020034static void f_balloon_gettext(typval_T *argvars, typval_T *rettv);
Bram Moolenaar59716a22017-03-01 20:32:44 +010035static void f_balloon_show(typval_T *argvars, typval_T *rettv);
Bram Moolenaar669a8282017-11-19 20:13:05 +010036# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +010037static void f_balloon_split(typval_T *argvars, typval_T *rettv);
Bram Moolenaar669a8282017-11-19 20:13:05 +010038# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +010039#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020040static void f_byte2line(typval_T *argvars, typval_T *rettv);
41static void byteidx(typval_T *argvars, typval_T *rettv, int comp);
42static void f_byteidx(typval_T *argvars, typval_T *rettv);
43static void f_byteidxcomp(typval_T *argvars, typval_T *rettv);
44static void f_call(typval_T *argvars, typval_T *rettv);
45#ifdef FEAT_FLOAT
46static void f_ceil(typval_T *argvars, typval_T *rettv);
47#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020048static void f_changenr(typval_T *argvars, typval_T *rettv);
49static void f_char2nr(typval_T *argvars, typval_T *rettv);
Bram Moolenaar6f02b002021-01-10 20:22:54 +010050static void f_charcol(typval_T *argvars, typval_T *rettv);
Bram Moolenaar17793ef2020-12-28 12:56:58 +010051static void f_charidx(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020052static void f_col(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020053static void f_confirm(typval_T *argvars, typval_T *rettv);
54static void f_copy(typval_T *argvars, typval_T *rettv);
55#ifdef FEAT_FLOAT
56static void f_cos(typval_T *argvars, typval_T *rettv);
57static void f_cosh(typval_T *argvars, typval_T *rettv);
58#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020059static void f_cursor(typval_T *argsvars, typval_T *rettv);
Bram Moolenaar4f974752019-02-17 17:44:42 +010060#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +020061static void f_debugbreak(typval_T *argvars, typval_T *rettv);
62#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020063static void f_deepcopy(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020064static void f_did_filetype(typval_T *argvars, typval_T *rettv);
Bram Moolenaar4132eb52020-02-14 16:53:00 +010065static void f_echoraw(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020066static void f_empty(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +020067static void f_environ(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020068static void f_escape(typval_T *argvars, typval_T *rettv);
69static void f_eval(typval_T *argvars, typval_T *rettv);
70static void f_eventhandler(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020071static void f_execute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020072static void f_exists(typval_T *argvars, typval_T *rettv);
73#ifdef FEAT_FLOAT
74static void f_exp(typval_T *argvars, typval_T *rettv);
75#endif
76static void f_expand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar80dad482019-06-09 17:22:31 +020077static void f_expandcmd(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020078static void f_feedkeys(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020079#ifdef FEAT_FLOAT
80static void f_float2nr(typval_T *argvars, typval_T *rettv);
81static void f_floor(typval_T *argvars, typval_T *rettv);
82static void f_fmod(typval_T *argvars, typval_T *rettv);
83#endif
84static void f_fnameescape(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020085static void f_foreground(typval_T *argvars, typval_T *rettv);
Bram Moolenaar437bafe2016-08-01 15:40:54 +020086static void f_funcref(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020087static void f_function(typval_T *argvars, typval_T *rettv);
88static void f_garbagecollect(typval_T *argvars, typval_T *rettv);
89static void f_get(typval_T *argvars, typval_T *rettv);
Bram Moolenaar07ad8162018-02-13 13:59:59 +010090static void f_getchangelist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar6f02b002021-01-10 20:22:54 +010091static void f_getcharpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020092static void f_getcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +020093static void f_getenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020094static void f_getfontname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar4f505882018-02-10 21:06:32 +010095static void f_getjumplist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020096static void f_getpid(typval_T *argvars, typval_T *rettv);
97static void f_getcurpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar6f02b002021-01-10 20:22:54 +010098static void f_getcursorcharpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020099static void f_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200100static void f_getreg(typval_T *argvars, typval_T *rettv);
Bram Moolenaarbb861e22020-06-07 18:16:36 +0200101static void f_getreginfo(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200102static void f_getregtype(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100103static void f_gettagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0b39c3f2020-08-30 15:52:10 +0200104static void f_gettext(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200105static void f_haslocaldir(typval_T *argvars, typval_T *rettv);
106static void f_hasmapto(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200107static void f_hlID(typval_T *argvars, typval_T *rettv);
108static void f_hlexists(typval_T *argvars, typval_T *rettv);
109static void f_hostname(typval_T *argvars, typval_T *rettv);
110static void f_iconv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200111static void f_index(typval_T *argvars, typval_T *rettv);
112static void f_input(typval_T *argvars, typval_T *rettv);
113static void f_inputdialog(typval_T *argvars, typval_T *rettv);
114static void f_inputlist(typval_T *argvars, typval_T *rettv);
115static void f_inputrestore(typval_T *argvars, typval_T *rettv);
116static void f_inputsave(typval_T *argvars, typval_T *rettv);
117static void f_inputsecret(typval_T *argvars, typval_T *rettv);
Bram Moolenaar67a2deb2019-11-25 00:05:32 +0100118static void f_interrupt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200119static void f_invert(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200120static void f_islocked(typval_T *argvars, typval_T *rettv);
121#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
Bram Moolenaarfda1bff2019-04-04 13:44:37 +0200122static void f_isinf(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200123static void f_isnan(typval_T *argvars, typval_T *rettv);
124#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200125static void f_last_buffer_nr(typval_T *argvars, typval_T *rettv);
126static void f_len(typval_T *argvars, typval_T *rettv);
127static void f_libcall(typval_T *argvars, typval_T *rettv);
128static void f_libcallnr(typval_T *argvars, typval_T *rettv);
129static void f_line(typval_T *argvars, typval_T *rettv);
130static void f_line2byte(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200131#ifdef FEAT_FLOAT
132static void f_log(typval_T *argvars, typval_T *rettv);
133static void f_log10(typval_T *argvars, typval_T *rettv);
134#endif
135#ifdef FEAT_LUA
136static void f_luaeval(typval_T *argvars, typval_T *rettv);
137#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200138static void f_maparg(typval_T *argvars, typval_T *rettv);
139static void f_mapcheck(typval_T *argvars, typval_T *rettv);
140static void f_match(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200141static void f_matchend(typval_T *argvars, typval_T *rettv);
142static void f_matchlist(typval_T *argvars, typval_T *rettv);
143static void f_matchstr(typval_T *argvars, typval_T *rettv);
144static void f_matchstrpos(typval_T *argvars, typval_T *rettv);
145static void f_max(typval_T *argvars, typval_T *rettv);
146static void f_min(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200147#ifdef FEAT_MZSCHEME
148static void f_mzeval(typval_T *argvars, typval_T *rettv);
149#endif
150static void f_nextnonblank(typval_T *argvars, typval_T *rettv);
151static void f_nr2char(typval_T *argvars, typval_T *rettv);
152static void f_or(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200153#ifdef FEAT_PERL
154static void f_perleval(typval_T *argvars, typval_T *rettv);
155#endif
156#ifdef FEAT_FLOAT
157static void f_pow(typval_T *argvars, typval_T *rettv);
158#endif
159static void f_prevnonblank(typval_T *argvars, typval_T *rettv);
160static void f_printf(typval_T *argvars, typval_T *rettv);
Bram Moolenaare9bd5722019-08-17 19:36:06 +0200161static void f_pum_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200162static void f_pumvisible(typval_T *argvars, typval_T *rettv);
163#ifdef FEAT_PYTHON3
164static void f_py3eval(typval_T *argvars, typval_T *rettv);
165#endif
166#ifdef FEAT_PYTHON
167static void f_pyeval(typval_T *argvars, typval_T *rettv);
168#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100169#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
170static void f_pyxeval(typval_T *argvars, typval_T *rettv);
171#endif
Bram Moolenaar4f645c52020-02-08 16:40:39 +0100172static void f_test_srand_seed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100173static void f_rand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200174static void f_range(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0b6d9112018-05-22 20:35:17 +0200175static void f_reg_executing(typval_T *argvars, typval_T *rettv);
176static void f_reg_recording(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200177static void f_rename(typval_T *argvars, typval_T *rettv);
178static void f_repeat(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200179#ifdef FEAT_FLOAT
180static void f_round(typval_T *argvars, typval_T *rettv);
181#endif
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100182#ifdef FEAT_RUBY
183static void f_rubyeval(typval_T *argvars, typval_T *rettv);
184#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200185static void f_screenattr(typval_T *argvars, typval_T *rettv);
186static void f_screenchar(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100187static void f_screenchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200188static void f_screencol(typval_T *argvars, typval_T *rettv);
189static void f_screenrow(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100190static void f_screenstring(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200191static void f_search(typval_T *argvars, typval_T *rettv);
192static void f_searchdecl(typval_T *argvars, typval_T *rettv);
193static void f_searchpair(typval_T *argvars, typval_T *rettv);
194static void f_searchpairpos(typval_T *argvars, typval_T *rettv);
195static void f_searchpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100196static void f_setcharpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200197static void f_setcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100198static void f_setcursorcharpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +0200199static void f_setenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200200static void f_setfperm(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200201static void f_setpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200202static void f_setreg(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100203static void f_settagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200204#ifdef FEAT_CRYPT
205static void f_sha256(typval_T *argvars, typval_T *rettv);
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200206#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200207static void f_shellescape(typval_T *argvars, typval_T *rettv);
208static void f_shiftwidth(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200209#ifdef FEAT_FLOAT
210static void f_sin(typval_T *argvars, typval_T *rettv);
211static void f_sinh(typval_T *argvars, typval_T *rettv);
212#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200213static void f_soundfold(typval_T *argvars, typval_T *rettv);
214static void f_spellbadword(typval_T *argvars, typval_T *rettv);
215static void f_spellsuggest(typval_T *argvars, typval_T *rettv);
216static void f_split(typval_T *argvars, typval_T *rettv);
217#ifdef FEAT_FLOAT
218static void f_sqrt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100219#endif
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100220static void f_srand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100221#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200222static void f_str2float(typval_T *argvars, typval_T *rettv);
223#endif
Bram Moolenaar9d401282019-04-06 13:18:12 +0200224static void f_str2list(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200225static void f_str2nr(typval_T *argvars, typval_T *rettv);
226static void f_strchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200227static void f_strgetchar(typval_T *argvars, typval_T *rettv);
228static void f_stridx(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200229static void f_strlen(typval_T *argvars, typval_T *rettv);
230static void f_strcharpart(typval_T *argvars, typval_T *rettv);
231static void f_strpart(typval_T *argvars, typval_T *rettv);
232static void f_strridx(typval_T *argvars, typval_T *rettv);
233static void f_strtrans(typval_T *argvars, typval_T *rettv);
234static void f_strdisplaywidth(typval_T *argvars, typval_T *rettv);
235static void f_strwidth(typval_T *argvars, typval_T *rettv);
236static void f_submatch(typval_T *argvars, typval_T *rettv);
237static void f_substitute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar00f123a2018-08-21 20:28:54 +0200238static void f_swapinfo(typval_T *argvars, typval_T *rettv);
Bram Moolenaar110bd602018-09-16 18:46:59 +0200239static void f_swapname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200240static void f_synID(typval_T *argvars, typval_T *rettv);
241static void f_synIDattr(typval_T *argvars, typval_T *rettv);
242static void f_synIDtrans(typval_T *argvars, typval_T *rettv);
243static void f_synstack(typval_T *argvars, typval_T *rettv);
244static void f_synconcealed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200245static void f_tabpagebuflist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200246static void f_taglist(typval_T *argvars, typval_T *rettv);
247static void f_tagfiles(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200248#ifdef FEAT_FLOAT
249static void f_tan(typval_T *argvars, typval_T *rettv);
250static void f_tanh(typval_T *argvars, typval_T *rettv);
251#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200252static void f_tolower(typval_T *argvars, typval_T *rettv);
253static void f_toupper(typval_T *argvars, typval_T *rettv);
254static void f_tr(typval_T *argvars, typval_T *rettv);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +0100255static void f_trim(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200256#ifdef FEAT_FLOAT
257static void f_trunc(typval_T *argvars, typval_T *rettv);
258#endif
259static void f_type(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200260static void f_virtcol(typval_T *argvars, typval_T *rettv);
261static void f_visualmode(typval_T *argvars, typval_T *rettv);
262static void f_wildmenumode(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0c1e3742019-12-27 13:49:24 +0100263static void f_windowsversion(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200264static void f_wordcount(typval_T *argvars, typval_T *rettv);
265static void f_xor(typval_T *argvars, typval_T *rettv);
266
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100267
Bram Moolenaar94738d82020-10-21 14:25:07 +0200268/*
269 * Functions that check the argument type of a builtin function.
270 * Each function returns FAIL and gives an error message if the type is wrong.
271 */
272
273// Context passed to an arg_ function.
274typedef struct {
Bram Moolenaarca174532020-10-21 16:42:22 +0200275 int arg_count; // actual argument count
276 type_T **arg_types; // list of argument types
277 int arg_idx; // current argument index (first arg is zero)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200278} argcontext_T;
279
280// A function to check one argument type. The first argument is the type to
281// check. If needed, other argument types can be obtained with the context.
282// E.g. if "arg_idx" is 1, then (type - 1) is the first argument type.
283typedef int (*argcheck_T)(type_T *, argcontext_T *);
284
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100285/*
286 * Check "type" is a float or a number.
287 */
Bram Moolenaar94738d82020-10-21 14:25:07 +0200288 static int
289arg_float_or_nr(type_T *type, argcontext_T *context)
290{
Bram Moolenaarca174532020-10-21 16:42:22 +0200291 if (type->tt_type == VAR_ANY
292 || type->tt_type == VAR_FLOAT || type->tt_type == VAR_NUMBER)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200293 return OK;
294 arg_type_mismatch(&t_number, type, context->arg_idx + 1);
295 return FAIL;
296}
297
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100298/*
299 * Check "type" is a number.
300 */
Bram Moolenaarca174532020-10-21 16:42:22 +0200301 static int
302arg_number(type_T *type, argcontext_T *context)
303{
Bram Moolenaar193f6202020-11-16 20:08:35 +0100304 return check_arg_type(&t_number, type, context->arg_idx + 1);
Bram Moolenaarca174532020-10-21 16:42:22 +0200305}
306
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100307/*
308 * Check "type" is a string.
309 */
310 static int
311arg_string(type_T *type, argcontext_T *context)
312{
Bram Moolenaar193f6202020-11-16 20:08:35 +0100313 return check_arg_type(&t_string, type, context->arg_idx + 1);
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100314}
315
316/*
317 * Check "type" is a list or a blob.
318 */
Bram Moolenaarca174532020-10-21 16:42:22 +0200319 static int
320arg_list_or_blob(type_T *type, argcontext_T *context)
321{
322 if (type->tt_type == VAR_ANY
323 || type->tt_type == VAR_LIST || type->tt_type == VAR_BLOB)
324 return OK;
325 arg_type_mismatch(&t_list_any, type, context->arg_idx + 1);
326 return FAIL;
327}
328
329/*
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100330 * Check "type" is a list or a dict.
331 */
332 static int
333arg_list_or_dict(type_T *type, argcontext_T *context)
334{
335 if (type->tt_type == VAR_ANY
336 || type->tt_type == VAR_LIST || type->tt_type == VAR_DICT)
337 return OK;
338 arg_type_mismatch(&t_list_any, type, context->arg_idx + 1);
339 return FAIL;
340}
341
342/*
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100343 * Check "type" is the same type as the previous argument.
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100344 * Must not be used for the first argcheck_T entry.
345 */
346 static int
347arg_same_as_prev(type_T *type, argcontext_T *context)
348{
349 type_T *prev_type = context->arg_types[context->arg_idx - 1];
350
Bram Moolenaar193f6202020-11-16 20:08:35 +0100351 return check_arg_type(prev_type, type, context->arg_idx + 1);
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100352}
353
354/*
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100355 * Check "type" is the same basic type as the previous argument, checks list or
356 * dict vs other type, but not member type.
357 * Must not be used for the first argcheck_T entry.
358 */
359 static int
360arg_same_struct_as_prev(type_T *type, argcontext_T *context)
361{
362 type_T *prev_type = context->arg_types[context->arg_idx - 1];
363
364 if (prev_type->tt_type != context->arg_types[context->arg_idx]->tt_type)
365 return check_arg_type(prev_type, type, context->arg_idx + 1);
366 return OK;
367}
368
369/*
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100370 * Check "type" is an item of the list or blob of the previous arg.
Bram Moolenaarca174532020-10-21 16:42:22 +0200371 * Must not be used for the first argcheck_T entry.
372 */
373 static int
374arg_item_of_prev(type_T *type, argcontext_T *context)
375{
376 type_T *prev_type = context->arg_types[context->arg_idx - 1];
377 type_T *expected;
378
379 if (prev_type->tt_type == VAR_LIST)
380 expected = prev_type->tt_member;
381 else if (prev_type->tt_type == VAR_BLOB)
382 expected = &t_number;
383 else
384 // probably VAR_ANY, can't check
385 return OK;
386
Bram Moolenaar193f6202020-11-16 20:08:35 +0100387 return check_arg_type(expected, type, context->arg_idx + 1);
Bram Moolenaarca174532020-10-21 16:42:22 +0200388}
389
Bram Moolenaar94738d82020-10-21 14:25:07 +0200390/*
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100391 * Check "type" which is the third argument of extend().
392 */
393 static int
394arg_extend3(type_T *type, argcontext_T *context)
395{
396 type_T *first_type = context->arg_types[context->arg_idx - 2];
397
398 if (first_type->tt_type == VAR_LIST)
399 return arg_number(type, context);
400 if (first_type->tt_type == VAR_DICT)
401 return arg_string(type, context);
402 return OK;
403}
404
405
406/*
Bram Moolenaar94738d82020-10-21 14:25:07 +0200407 * Lists of functions that check the argument types of a builtin function.
408 */
409argcheck_T arg1_float_or_nr[] = {arg_float_or_nr};
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100410argcheck_T arg2_listblob_item[] = {arg_list_or_blob, arg_item_of_prev};
411argcheck_T arg23_extend[] = {arg_list_or_dict, arg_same_as_prev, arg_extend3};
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100412argcheck_T arg23_extendnew[] = {arg_list_or_dict, arg_same_struct_as_prev, arg_extend3};
Bram Moolenaarca174532020-10-21 16:42:22 +0200413argcheck_T arg3_insert[] = {arg_list_or_blob, arg_item_of_prev, arg_number};
Bram Moolenaar94738d82020-10-21 14:25:07 +0200414
415/*
416 * Functions that return the return type of a builtin function.
Bram Moolenaara1224cb2020-10-22 12:31:49 +0200417 * Note that "argtypes" is NULL if "argcount" is zero.
Bram Moolenaar94738d82020-10-21 14:25:07 +0200418 */
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100419 static type_T *
420ret_void(int argcount UNUSED, type_T **argtypes UNUSED)
421{
422 return &t_void;
423}
424 static type_T *
425ret_any(int argcount UNUSED, type_T **argtypes UNUSED)
426{
427 return &t_any;
428}
429 static type_T *
Bram Moolenaar403dc312020-10-17 19:29:51 +0200430ret_bool(int argcount UNUSED, type_T **argtypes UNUSED)
431{
432 return &t_bool;
433}
434 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100435ret_number(int argcount UNUSED, type_T **argtypes UNUSED)
436{
437 return &t_number;
438}
439 static type_T *
440ret_float(int argcount UNUSED, type_T **argtypes UNUSED)
441{
442 return &t_float;
443}
444 static type_T *
445ret_string(int argcount UNUSED, type_T **argtypes UNUSED)
446{
447 return &t_string;
448}
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200449 static type_T *
450ret_list_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100451{
452 return &t_list_any;
453}
454 static type_T *
455ret_list_number(int argcount UNUSED, type_T **argtypes UNUSED)
456{
457 return &t_list_number;
458}
459 static type_T *
460ret_list_string(int argcount UNUSED, type_T **argtypes UNUSED)
461{
462 return &t_list_string;
463}
464 static type_T *
465ret_list_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
466{
467 return &t_list_dict_any;
468}
469 static type_T *
470ret_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
471{
472 return &t_dict_any;
473}
474 static type_T *
475ret_dict_number(int argcount UNUSED, type_T **argtypes UNUSED)
476{
477 return &t_dict_number;
478}
479 static type_T *
480ret_dict_string(int argcount UNUSED, type_T **argtypes UNUSED)
481{
482 return &t_dict_string;
483}
484 static type_T *
485ret_blob(int argcount UNUSED, type_T **argtypes UNUSED)
486{
487 return &t_blob;
488}
489 static type_T *
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200490ret_func_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100491{
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200492 return &t_func_any;
493}
494 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100495ret_channel(int argcount UNUSED, type_T **argtypes UNUSED)
496{
497 return &t_channel;
498}
499 static type_T *
500ret_job(int argcount UNUSED, type_T **argtypes UNUSED)
501{
502 return &t_job;
503}
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200504 static type_T *
505ret_first_arg(int argcount, type_T **argtypes)
506{
507 if (argcount > 0)
508 return argtypes[0];
509 return &t_void;
510}
Bram Moolenaarea696852020-11-09 18:31:39 +0100511// for map(): returns first argument but item type may differ
512 static type_T *
513ret_first_cont(int argcount UNUSED, type_T **argtypes)
514{
515 if (argtypes[0]->tt_type == VAR_LIST)
516 return &t_list_any;
517 if (argtypes[0]->tt_type == VAR_DICT)
518 return &t_dict_any;
519 if (argtypes[0]->tt_type == VAR_BLOB)
520 return argtypes[0];
521 return &t_any;
522}
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200523
Bram Moolenaarf151ad12020-06-30 13:38:01 +0200524/*
525 * Used for getqflist(): returns list if there is no argument, dict if there is
526 * one.
527 */
528 static type_T *
529ret_list_or_dict_0(int argcount, type_T **argtypes UNUSED)
530{
531 if (argcount > 0)
532 return &t_dict_any;
533 return &t_list_dict_any;
534}
535
536/*
537 * Used for getloclist(): returns list if there is one argument, dict if there
538 * are two.
539 */
540 static type_T *
541ret_list_or_dict_1(int argcount, type_T **argtypes UNUSED)
542{
543 if (argcount > 1)
544 return &t_dict_any;
545 return &t_list_dict_any;
546}
547
Bram Moolenaar846178a2020-07-05 17:04:13 +0200548 static type_T *
549ret_argv(int argcount, type_T **argtypes UNUSED)
550{
551 // argv() returns list of strings
552 if (argcount == 0)
553 return &t_list_string;
554
555 // argv(0) returns a string, but argv(-1] returns a list
556 return &t_any;
557}
558
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200559 static type_T *
560ret_remove(int argcount UNUSED, type_T **argtypes)
561{
Bram Moolenaar5e654232020-09-16 15:22:00 +0200562 if (argtypes != NULL)
563 {
564 if (argtypes[0]->tt_type == VAR_LIST
565 || argtypes[0]->tt_type == VAR_DICT)
566 return argtypes[0]->tt_member;
567 if (argtypes[0]->tt_type == VAR_BLOB)
568 return &t_number;
569 }
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200570 return &t_any;
571}
572
Bram Moolenaar3d945cc2020-08-06 21:26:59 +0200573 static type_T *
574ret_getreg(int argcount, type_T **argtypes UNUSED)
575{
576 // Assume that if the third argument is passed it's non-zero
577 if (argcount == 3)
578 return &t_list_string;
579 return &t_string;
580}
581
Bram Moolenaar4a6d1b62020-08-08 17:55:49 +0200582 static type_T *
583ret_maparg(int argcount, type_T **argtypes UNUSED)
584{
585 // Assume that if the fourth argument is passed it's non-zero
586 if (argcount == 4)
587 return &t_dict_any;
588 return &t_string;
589}
590
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100591static type_T *ret_f_function(int argcount, type_T **argtypes);
592
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200593/*
594 * Array with names and number of arguments of all internal functions
595 * MUST BE KEPT SORTED IN strcmp() ORDER FOR BINARY SEARCH!
596 */
Bram Moolenaarac92e252019-08-03 21:58:38 +0200597typedef struct
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200598{
Bram Moolenaar25e42232019-08-04 15:04:10 +0200599 char *f_name; // function name
600 char f_min_argc; // minimal number of arguments
601 char f_max_argc; // maximal number of arguments
602 char f_argtype; // for method: FEARG_ values
Bram Moolenaar94738d82020-10-21 14:25:07 +0200603 argcheck_T *f_argcheck; // list of functions to check argument types
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100604 type_T *(*f_retfunc)(int argcount, type_T **argtypes);
605 // return type function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200606 void (*f_func)(typval_T *args, typval_T *rvar);
Bram Moolenaar25e42232019-08-04 15:04:10 +0200607 // implementation of function
Bram Moolenaarac92e252019-08-03 21:58:38 +0200608} funcentry_T;
609
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200610// values for f_argtype; zero means it cannot be used as a method
611#define FEARG_1 1 // base is the first argument
612#define FEARG_2 2 // base is the second argument
Bram Moolenaar24278d22019-08-16 21:49:22 +0200613#define FEARG_3 3 // base is the third argument
Bram Moolenaaraad222c2019-09-06 22:46:09 +0200614#define FEARG_4 4 // base is the fourth argument
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200615#define FEARG_LAST 9 // base is the last argument
616
Bram Moolenaar15c47602020-03-26 22:16:48 +0100617#ifdef FEAT_FLOAT
618# define FLOAT_FUNC(name) name
619#else
620# define FLOAT_FUNC(name) NULL
621#endif
622#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
623# define MATH_FUNC(name) name
624#else
625# define MATH_FUNC(name) NULL
626#endif
627#ifdef FEAT_TIMERS
628# define TIMER_FUNC(name) name
629#else
630# define TIMER_FUNC(name) NULL
631#endif
632#ifdef FEAT_JOB_CHANNEL
633# define JOB_FUNC(name) name
634#else
635# define JOB_FUNC(name) NULL
636#endif
637#ifdef FEAT_PROP_POPUP
638# define PROP_FUNC(name) name
639#else
640# define PROP_FUNC(name) NULL
641#endif
642#ifdef FEAT_SIGNS
643# define SIGN_FUNC(name) name
644#else
645# define SIGN_FUNC(name) NULL
646#endif
647#ifdef FEAT_SOUND
648# define SOUND_FUNC(name) name
649#else
650# define SOUND_FUNC(name) NULL
651#endif
652#ifdef FEAT_TERMINAL
653# define TERM_FUNC(name) name
654#else
655# define TERM_FUNC(name) NULL
656#endif
657
Bram Moolenaarac92e252019-08-03 21:58:38 +0200658static funcentry_T global_functions[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200659{
Bram Moolenaar94738d82020-10-21 14:25:07 +0200660 {"abs", 1, 1, FEARG_1, arg1_float_or_nr,
661 ret_any, FLOAT_FUNC(f_abs)},
662 {"acos", 1, 1, FEARG_1, NULL,
663 ret_float, FLOAT_FUNC(f_acos)},
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100664 {"add", 2, 2, FEARG_1, NULL /* arg2_listblob_item */,
Bram Moolenaar94738d82020-10-21 14:25:07 +0200665 ret_first_arg, f_add},
666 {"and", 2, 2, FEARG_1, NULL,
667 ret_number, f_and},
668 {"append", 2, 2, FEARG_2, NULL,
669 ret_number, f_append},
670 {"appendbufline", 3, 3, FEARG_3, NULL,
671 ret_number, f_appendbufline},
672 {"argc", 0, 1, 0, NULL,
673 ret_number, f_argc},
674 {"argidx", 0, 0, 0, NULL,
675 ret_number, f_argidx},
676 {"arglistid", 0, 2, 0, NULL,
677 ret_number, f_arglistid},
678 {"argv", 0, 2, 0, NULL,
679 ret_argv, f_argv},
680 {"asin", 1, 1, FEARG_1, NULL,
681 ret_float, FLOAT_FUNC(f_asin)},
682 {"assert_beeps", 1, 2, FEARG_1, NULL,
683 ret_number, f_assert_beeps},
684 {"assert_equal", 2, 3, FEARG_2, NULL,
685 ret_number, f_assert_equal},
686 {"assert_equalfile", 2, 3, FEARG_1, NULL,
687 ret_number, f_assert_equalfile},
688 {"assert_exception", 1, 2, 0, NULL,
689 ret_number, f_assert_exception},
690 {"assert_fails", 1, 5, FEARG_1, NULL,
691 ret_number, f_assert_fails},
692 {"assert_false", 1, 2, FEARG_1, NULL,
693 ret_number, f_assert_false},
694 {"assert_inrange", 3, 4, FEARG_3, NULL,
695 ret_number, f_assert_inrange},
696 {"assert_match", 2, 3, FEARG_2, NULL,
697 ret_number, f_assert_match},
698 {"assert_notequal", 2, 3, FEARG_2, NULL,
699 ret_number, f_assert_notequal},
700 {"assert_notmatch", 2, 3, FEARG_2, NULL,
701 ret_number, f_assert_notmatch},
702 {"assert_report", 1, 1, FEARG_1, NULL,
703 ret_number, f_assert_report},
704 {"assert_true", 1, 2, FEARG_1, NULL,
705 ret_number, f_assert_true},
706 {"atan", 1, 1, FEARG_1, NULL,
707 ret_float, FLOAT_FUNC(f_atan)},
708 {"atan2", 2, 2, FEARG_1, NULL,
709 ret_float, FLOAT_FUNC(f_atan2)},
710 {"balloon_gettext", 0, 0, 0, NULL,
711 ret_string,
Bram Moolenaar59716a22017-03-01 20:32:44 +0100712#ifdef FEAT_BEVAL
Bram Moolenaar15c47602020-03-26 22:16:48 +0100713 f_balloon_gettext
714#else
715 NULL
Bram Moolenaar59716a22017-03-01 20:32:44 +0100716#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100717 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200718 {"balloon_show", 1, 1, FEARG_1, NULL,
719 ret_void,
Bram Moolenaar15c47602020-03-26 22:16:48 +0100720#ifdef FEAT_BEVAL
721 f_balloon_show
722#else
723 NULL
724#endif
725 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200726 {"balloon_split", 1, 1, FEARG_1, NULL,
727 ret_list_string,
Bram Moolenaar15c47602020-03-26 22:16:48 +0100728#if defined(FEAT_BEVAL_TERM)
729 f_balloon_split
730#else
731 NULL
732#endif
733 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200734 {"browse", 4, 4, 0, NULL,
735 ret_string, f_browse},
736 {"browsedir", 2, 2, 0, NULL,
737 ret_string, f_browsedir},
738 {"bufadd", 1, 1, FEARG_1, NULL,
739 ret_number, f_bufadd},
740 {"bufexists", 1, 1, FEARG_1, NULL,
741 ret_number, f_bufexists},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +0200742 {"buffer_exists", 1, 1, FEARG_1, NULL, // obsolete
743 ret_number, f_bufexists},
744 {"buffer_name", 0, 1, FEARG_1, NULL, // obsolete
745 ret_string, f_bufname},
746 {"buffer_number", 0, 1, FEARG_1, NULL, // obsolete
747 ret_number, f_bufnr},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200748 {"buflisted", 1, 1, FEARG_1, NULL,
749 ret_number, f_buflisted},
750 {"bufload", 1, 1, FEARG_1, NULL,
751 ret_void, f_bufload},
752 {"bufloaded", 1, 1, FEARG_1, NULL,
753 ret_number, f_bufloaded},
754 {"bufname", 0, 1, FEARG_1, NULL,
755 ret_string, f_bufname},
756 {"bufnr", 0, 2, FEARG_1, NULL,
757 ret_number, f_bufnr},
758 {"bufwinid", 1, 1, FEARG_1, NULL,
759 ret_number, f_bufwinid},
760 {"bufwinnr", 1, 1, FEARG_1, NULL,
761 ret_number, f_bufwinnr},
762 {"byte2line", 1, 1, FEARG_1, NULL,
763 ret_number, f_byte2line},
764 {"byteidx", 2, 2, FEARG_1, NULL,
765 ret_number, f_byteidx},
766 {"byteidxcomp", 2, 2, FEARG_1, NULL,
767 ret_number, f_byteidxcomp},
768 {"call", 2, 3, FEARG_1, NULL,
769 ret_any, f_call},
770 {"ceil", 1, 1, FEARG_1, NULL,
771 ret_float, FLOAT_FUNC(f_ceil)},
772 {"ch_canread", 1, 1, FEARG_1, NULL,
773 ret_number, JOB_FUNC(f_ch_canread)},
774 {"ch_close", 1, 1, FEARG_1, NULL,
775 ret_void, JOB_FUNC(f_ch_close)},
776 {"ch_close_in", 1, 1, FEARG_1, NULL,
777 ret_void, JOB_FUNC(f_ch_close_in)},
778 {"ch_evalexpr", 2, 3, FEARG_1, NULL,
779 ret_any, JOB_FUNC(f_ch_evalexpr)},
780 {"ch_evalraw", 2, 3, FEARG_1, NULL,
781 ret_any, JOB_FUNC(f_ch_evalraw)},
782 {"ch_getbufnr", 2, 2, FEARG_1, NULL,
783 ret_number, JOB_FUNC(f_ch_getbufnr)},
784 {"ch_getjob", 1, 1, FEARG_1, NULL,
785 ret_job, JOB_FUNC(f_ch_getjob)},
786 {"ch_info", 1, 1, FEARG_1, NULL,
787 ret_dict_any, JOB_FUNC(f_ch_info)},
788 {"ch_log", 1, 2, FEARG_1, NULL,
789 ret_void, JOB_FUNC(f_ch_log)},
790 {"ch_logfile", 1, 2, FEARG_1, NULL,
791 ret_void, JOB_FUNC(f_ch_logfile)},
792 {"ch_open", 1, 2, FEARG_1, NULL,
793 ret_channel, JOB_FUNC(f_ch_open)},
794 {"ch_read", 1, 2, FEARG_1, NULL,
795 ret_string, JOB_FUNC(f_ch_read)},
796 {"ch_readblob", 1, 2, FEARG_1, NULL,
797 ret_blob, JOB_FUNC(f_ch_readblob)},
798 {"ch_readraw", 1, 2, FEARG_1, NULL,
799 ret_string, JOB_FUNC(f_ch_readraw)},
800 {"ch_sendexpr", 2, 3, FEARG_1, NULL,
801 ret_void, JOB_FUNC(f_ch_sendexpr)},
802 {"ch_sendraw", 2, 3, FEARG_1, NULL,
803 ret_void, JOB_FUNC(f_ch_sendraw)},
804 {"ch_setoptions", 2, 2, FEARG_1, NULL,
805 ret_void, JOB_FUNC(f_ch_setoptions)},
806 {"ch_status", 1, 2, FEARG_1, NULL,
807 ret_string, JOB_FUNC(f_ch_status)},
808 {"changenr", 0, 0, 0, NULL,
809 ret_number, f_changenr},
810 {"char2nr", 1, 2, FEARG_1, NULL,
811 ret_number, f_char2nr},
812 {"charclass", 1, 1, FEARG_1, NULL,
813 ret_number, f_charclass},
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100814 {"charcol", 1, 1, FEARG_1, NULL,
815 ret_number, f_charcol},
Bram Moolenaar17793ef2020-12-28 12:56:58 +0100816 {"charidx", 2, 3, FEARG_1, NULL,
817 ret_number, f_charidx},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200818 {"chdir", 1, 1, FEARG_1, NULL,
819 ret_string, f_chdir},
820 {"cindent", 1, 1, FEARG_1, NULL,
821 ret_number, f_cindent},
822 {"clearmatches", 0, 1, FEARG_1, NULL,
823 ret_void, f_clearmatches},
824 {"col", 1, 1, FEARG_1, NULL,
825 ret_number, f_col},
826 {"complete", 2, 2, FEARG_2, NULL,
827 ret_void, f_complete},
828 {"complete_add", 1, 1, FEARG_1, NULL,
829 ret_number, f_complete_add},
830 {"complete_check", 0, 0, 0, NULL,
831 ret_number, f_complete_check},
832 {"complete_info", 0, 1, FEARG_1, NULL,
833 ret_dict_any, f_complete_info},
834 {"confirm", 1, 4, FEARG_1, NULL,
835 ret_number, f_confirm},
836 {"copy", 1, 1, FEARG_1, NULL,
837 ret_first_arg, f_copy},
838 {"cos", 1, 1, FEARG_1, NULL,
839 ret_float, FLOAT_FUNC(f_cos)},
840 {"cosh", 1, 1, FEARG_1, NULL,
841 ret_float, FLOAT_FUNC(f_cosh)},
842 {"count", 2, 4, FEARG_1, NULL,
843 ret_number, f_count},
844 {"cscope_connection",0,3, 0, NULL,
845 ret_number, f_cscope_connection},
846 {"cursor", 1, 3, FEARG_1, NULL,
847 ret_number, f_cursor},
848 {"debugbreak", 1, 1, FEARG_1, NULL,
849 ret_number,
Bram Moolenaar4f974752019-02-17 17:44:42 +0100850#ifdef MSWIN
Bram Moolenaar15c47602020-03-26 22:16:48 +0100851 f_debugbreak
852#else
853 NULL
Bram Moolenaar4551c0a2018-06-20 22:38:21 +0200854#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100855 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200856 {"deepcopy", 1, 2, FEARG_1, NULL,
857 ret_first_arg, f_deepcopy},
858 {"delete", 1, 2, FEARG_1, NULL,
859 ret_number, f_delete},
860 {"deletebufline", 2, 3, FEARG_1, NULL,
861 ret_number, f_deletebufline},
862 {"did_filetype", 0, 0, 0, NULL,
863 ret_number, f_did_filetype},
864 {"diff_filler", 1, 1, FEARG_1, NULL,
865 ret_number, f_diff_filler},
866 {"diff_hlID", 2, 2, FEARG_1, NULL,
867 ret_number, f_diff_hlID},
868 {"echoraw", 1, 1, FEARG_1, NULL,
869 ret_number, f_echoraw},
870 {"empty", 1, 1, FEARG_1, NULL,
871 ret_number, f_empty},
872 {"environ", 0, 0, 0, NULL,
873 ret_dict_string, f_environ},
874 {"escape", 2, 2, FEARG_1, NULL,
875 ret_string, f_escape},
876 {"eval", 1, 1, FEARG_1, NULL,
877 ret_any, f_eval},
878 {"eventhandler", 0, 0, 0, NULL,
879 ret_number, f_eventhandler},
880 {"executable", 1, 1, FEARG_1, NULL,
881 ret_number, f_executable},
882 {"execute", 1, 2, FEARG_1, NULL,
883 ret_string, f_execute},
884 {"exepath", 1, 1, FEARG_1, NULL,
885 ret_string, f_exepath},
886 {"exists", 1, 1, FEARG_1, NULL,
887 ret_number, f_exists},
888 {"exp", 1, 1, FEARG_1, NULL,
889 ret_float, FLOAT_FUNC(f_exp)},
890 {"expand", 1, 3, FEARG_1, NULL,
891 ret_any, f_expand},
892 {"expandcmd", 1, 1, FEARG_1, NULL,
893 ret_string, f_expandcmd},
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100894 {"extend", 2, 3, FEARG_1, arg23_extend,
Bram Moolenaar94738d82020-10-21 14:25:07 +0200895 ret_first_arg, f_extend},
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100896 {"extendnew", 2, 3, FEARG_1, arg23_extendnew,
897 ret_first_cont, f_extendnew},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200898 {"feedkeys", 1, 2, FEARG_1, NULL,
899 ret_void, f_feedkeys},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +0200900 {"file_readable", 1, 1, FEARG_1, NULL, // obsolete
901 ret_number, f_filereadable},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200902 {"filereadable", 1, 1, FEARG_1, NULL,
903 ret_number, f_filereadable},
904 {"filewritable", 1, 1, FEARG_1, NULL,
905 ret_number, f_filewritable},
906 {"filter", 2, 2, FEARG_1, NULL,
907 ret_first_arg, f_filter},
908 {"finddir", 1, 3, FEARG_1, NULL,
909 ret_string, f_finddir},
910 {"findfile", 1, 3, FEARG_1, NULL,
911 ret_string, f_findfile},
912 {"flatten", 1, 2, FEARG_1, NULL,
913 ret_list_any, f_flatten},
914 {"float2nr", 1, 1, FEARG_1, NULL,
915 ret_number, FLOAT_FUNC(f_float2nr)},
916 {"floor", 1, 1, FEARG_1, NULL,
917 ret_float, FLOAT_FUNC(f_floor)},
918 {"fmod", 2, 2, FEARG_1, NULL,
919 ret_float, FLOAT_FUNC(f_fmod)},
920 {"fnameescape", 1, 1, FEARG_1, NULL,
921 ret_string, f_fnameescape},
922 {"fnamemodify", 2, 2, FEARG_1, NULL,
923 ret_string, f_fnamemodify},
924 {"foldclosed", 1, 1, FEARG_1, NULL,
925 ret_number, f_foldclosed},
926 {"foldclosedend", 1, 1, FEARG_1, NULL,
927 ret_number, f_foldclosedend},
928 {"foldlevel", 1, 1, FEARG_1, NULL,
929 ret_number, f_foldlevel},
930 {"foldtext", 0, 0, 0, NULL,
931 ret_string, f_foldtext},
932 {"foldtextresult", 1, 1, FEARG_1, NULL,
933 ret_string, f_foldtextresult},
934 {"foreground", 0, 0, 0, NULL,
935 ret_void, f_foreground},
936 {"funcref", 1, 3, FEARG_1, NULL,
937 ret_func_any, f_funcref},
938 {"function", 1, 3, FEARG_1, NULL,
939 ret_f_function, f_function},
940 {"garbagecollect", 0, 1, 0, NULL,
941 ret_void, f_garbagecollect},
942 {"get", 2, 3, FEARG_1, NULL,
943 ret_any, f_get},
944 {"getbufinfo", 0, 1, FEARG_1, NULL,
945 ret_list_dict_any, f_getbufinfo},
946 {"getbufline", 2, 3, FEARG_1, NULL,
947 ret_list_string, f_getbufline},
948 {"getbufvar", 2, 3, FEARG_1, NULL,
949 ret_any, f_getbufvar},
950 {"getchangelist", 0, 1, FEARG_1, NULL,
951 ret_list_any, f_getchangelist},
952 {"getchar", 0, 1, 0, NULL,
953 ret_number, f_getchar},
954 {"getcharmod", 0, 0, 0, NULL,
955 ret_number, f_getcharmod},
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100956 {"getcharpos", 1, 1, FEARG_1, NULL,
957 ret_list_number, f_getcharpos},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200958 {"getcharsearch", 0, 0, 0, NULL,
959 ret_dict_any, f_getcharsearch},
960 {"getcmdline", 0, 0, 0, NULL,
961 ret_string, f_getcmdline},
962 {"getcmdpos", 0, 0, 0, NULL,
963 ret_number, f_getcmdpos},
964 {"getcmdtype", 0, 0, 0, NULL,
965 ret_string, f_getcmdtype},
966 {"getcmdwintype", 0, 0, 0, NULL,
967 ret_string, f_getcmdwintype},
968 {"getcompletion", 2, 3, FEARG_1, NULL,
969 ret_list_string, f_getcompletion},
970 {"getcurpos", 0, 1, FEARG_1, NULL,
971 ret_list_number, f_getcurpos},
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100972 {"getcursorcharpos", 0, 1, FEARG_1, NULL,
973 ret_list_number, f_getcursorcharpos},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200974 {"getcwd", 0, 2, FEARG_1, NULL,
975 ret_string, f_getcwd},
976 {"getenv", 1, 1, FEARG_1, NULL,
977 ret_string, f_getenv},
978 {"getfontname", 0, 1, 0, NULL,
979 ret_string, f_getfontname},
980 {"getfperm", 1, 1, FEARG_1, NULL,
981 ret_string, f_getfperm},
982 {"getfsize", 1, 1, FEARG_1, NULL,
983 ret_number, f_getfsize},
984 {"getftime", 1, 1, FEARG_1, NULL,
985 ret_number, f_getftime},
986 {"getftype", 1, 1, FEARG_1, NULL,
987 ret_string, f_getftype},
988 {"getimstatus", 0, 0, 0, NULL,
989 ret_number, f_getimstatus},
990 {"getjumplist", 0, 2, FEARG_1, NULL,
991 ret_list_any, f_getjumplist},
992 {"getline", 1, 2, FEARG_1, NULL,
993 ret_f_getline, f_getline},
994 {"getloclist", 1, 2, 0, NULL,
995 ret_list_or_dict_1, f_getloclist},
996 {"getmarklist", 0, 1, FEARG_1, NULL,
997 ret_list_dict_any, f_getmarklist},
998 {"getmatches", 0, 1, 0, NULL,
999 ret_list_dict_any, f_getmatches},
1000 {"getmousepos", 0, 0, 0, NULL,
1001 ret_dict_number, f_getmousepos},
1002 {"getpid", 0, 0, 0, NULL,
1003 ret_number, f_getpid},
1004 {"getpos", 1, 1, FEARG_1, NULL,
1005 ret_list_number, f_getpos},
1006 {"getqflist", 0, 1, 0, NULL,
1007 ret_list_or_dict_0, f_getqflist},
1008 {"getreg", 0, 3, FEARG_1, NULL,
1009 ret_getreg, f_getreg},
1010 {"getreginfo", 0, 1, FEARG_1, NULL,
1011 ret_dict_any, f_getreginfo},
1012 {"getregtype", 0, 1, FEARG_1, NULL,
1013 ret_string, f_getregtype},
1014 {"gettabinfo", 0, 1, FEARG_1, NULL,
1015 ret_list_dict_any, f_gettabinfo},
1016 {"gettabvar", 2, 3, FEARG_1, NULL,
1017 ret_any, f_gettabvar},
1018 {"gettabwinvar", 3, 4, FEARG_1, NULL,
1019 ret_any, f_gettabwinvar},
1020 {"gettagstack", 0, 1, FEARG_1, NULL,
1021 ret_dict_any, f_gettagstack},
1022 {"gettext", 1, 1, FEARG_1, NULL,
1023 ret_string, f_gettext},
1024 {"getwininfo", 0, 1, FEARG_1, NULL,
1025 ret_list_dict_any, f_getwininfo},
1026 {"getwinpos", 0, 1, FEARG_1, NULL,
1027 ret_list_number, f_getwinpos},
1028 {"getwinposx", 0, 0, 0, NULL,
1029 ret_number, f_getwinposx},
1030 {"getwinposy", 0, 0, 0, NULL,
1031 ret_number, f_getwinposy},
1032 {"getwinvar", 2, 3, FEARG_1, NULL,
1033 ret_any, f_getwinvar},
1034 {"glob", 1, 4, FEARG_1, NULL,
1035 ret_any, f_glob},
1036 {"glob2regpat", 1, 1, FEARG_1, NULL,
1037 ret_string, f_glob2regpat},
1038 {"globpath", 2, 5, FEARG_2, NULL,
1039 ret_any, f_globpath},
1040 {"has", 1, 2, 0, NULL,
1041 ret_number, f_has},
1042 {"has_key", 2, 2, FEARG_1, NULL,
1043 ret_number, f_has_key},
1044 {"haslocaldir", 0, 2, FEARG_1, NULL,
1045 ret_number, f_haslocaldir},
1046 {"hasmapto", 1, 3, FEARG_1, NULL,
1047 ret_number, f_hasmapto},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +02001048 {"highlightID", 1, 1, FEARG_1, NULL, // obsolete
1049 ret_number, f_hlID},
1050 {"highlight_exists",1, 1, FEARG_1, NULL, // obsolete
1051 ret_number, f_hlexists},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001052 {"histadd", 2, 2, FEARG_2, NULL,
1053 ret_number, f_histadd},
1054 {"histdel", 1, 2, FEARG_1, NULL,
1055 ret_number, f_histdel},
1056 {"histget", 1, 2, FEARG_1, NULL,
1057 ret_string, f_histget},
1058 {"histnr", 1, 1, FEARG_1, NULL,
1059 ret_number, f_histnr},
1060 {"hlID", 1, 1, FEARG_1, NULL,
1061 ret_number, f_hlID},
1062 {"hlexists", 1, 1, FEARG_1, NULL,
1063 ret_number, f_hlexists},
1064 {"hostname", 0, 0, 0, NULL,
1065 ret_string, f_hostname},
1066 {"iconv", 3, 3, FEARG_1, NULL,
1067 ret_string, f_iconv},
1068 {"indent", 1, 1, FEARG_1, NULL,
1069 ret_number, f_indent},
1070 {"index", 2, 4, FEARG_1, NULL,
1071 ret_number, f_index},
1072 {"input", 1, 3, FEARG_1, NULL,
1073 ret_string, f_input},
1074 {"inputdialog", 1, 3, FEARG_1, NULL,
1075 ret_string, f_inputdialog},
1076 {"inputlist", 1, 1, FEARG_1, NULL,
1077 ret_number, f_inputlist},
1078 {"inputrestore", 0, 0, 0, NULL,
1079 ret_number, f_inputrestore},
1080 {"inputsave", 0, 0, 0, NULL,
1081 ret_number, f_inputsave},
1082 {"inputsecret", 1, 2, FEARG_1, NULL,
1083 ret_string, f_inputsecret},
Bram Moolenaarca174532020-10-21 16:42:22 +02001084 {"insert", 2, 3, FEARG_1, arg3_insert,
Bram Moolenaar94738d82020-10-21 14:25:07 +02001085 ret_first_arg, f_insert},
1086 {"interrupt", 0, 0, 0, NULL,
1087 ret_void, f_interrupt},
1088 {"invert", 1, 1, FEARG_1, NULL,
1089 ret_number, f_invert},
1090 {"isdirectory", 1, 1, FEARG_1, NULL,
1091 ret_number, f_isdirectory},
1092 {"isinf", 1, 1, FEARG_1, NULL,
1093 ret_number, MATH_FUNC(f_isinf)},
1094 {"islocked", 1, 1, FEARG_1, NULL,
1095 ret_number, f_islocked},
1096 {"isnan", 1, 1, FEARG_1, NULL,
1097 ret_number, MATH_FUNC(f_isnan)},
1098 {"items", 1, 1, FEARG_1, NULL,
1099 ret_list_any, f_items},
1100 {"job_getchannel", 1, 1, FEARG_1, NULL,
1101 ret_channel, JOB_FUNC(f_job_getchannel)},
1102 {"job_info", 0, 1, FEARG_1, NULL,
1103 ret_dict_any, JOB_FUNC(f_job_info)},
1104 {"job_setoptions", 2, 2, FEARG_1, NULL,
1105 ret_void, JOB_FUNC(f_job_setoptions)},
1106 {"job_start", 1, 2, FEARG_1, NULL,
1107 ret_job, JOB_FUNC(f_job_start)},
1108 {"job_status", 1, 1, FEARG_1, NULL,
1109 ret_string, JOB_FUNC(f_job_status)},
1110 {"job_stop", 1, 2, FEARG_1, NULL,
1111 ret_number, JOB_FUNC(f_job_stop)},
1112 {"join", 1, 2, FEARG_1, NULL,
1113 ret_string, f_join},
1114 {"js_decode", 1, 1, FEARG_1, NULL,
1115 ret_any, f_js_decode},
1116 {"js_encode", 1, 1, FEARG_1, NULL,
1117 ret_string, f_js_encode},
1118 {"json_decode", 1, 1, FEARG_1, NULL,
1119 ret_any, f_json_decode},
1120 {"json_encode", 1, 1, FEARG_1, NULL,
1121 ret_string, f_json_encode},
1122 {"keys", 1, 1, FEARG_1, NULL,
1123 ret_list_string, f_keys},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +02001124 {"last_buffer_nr", 0, 0, 0, NULL, // obsolete
1125 ret_number, f_last_buffer_nr},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001126 {"len", 1, 1, FEARG_1, NULL,
1127 ret_number, f_len},
1128 {"libcall", 3, 3, FEARG_3, NULL,
1129 ret_string, f_libcall},
1130 {"libcallnr", 3, 3, FEARG_3, NULL,
1131 ret_number, f_libcallnr},
1132 {"line", 1, 2, FEARG_1, NULL,
1133 ret_number, f_line},
1134 {"line2byte", 1, 1, FEARG_1, NULL,
1135 ret_number, f_line2byte},
1136 {"lispindent", 1, 1, FEARG_1, NULL,
1137 ret_number, f_lispindent},
1138 {"list2str", 1, 2, FEARG_1, NULL,
1139 ret_string, f_list2str},
1140 {"listener_add", 1, 2, FEARG_2, NULL,
1141 ret_number, f_listener_add},
1142 {"listener_flush", 0, 1, FEARG_1, NULL,
1143 ret_void, f_listener_flush},
1144 {"listener_remove", 1, 1, FEARG_1, NULL,
1145 ret_number, f_listener_remove},
1146 {"localtime", 0, 0, 0, NULL,
1147 ret_number, f_localtime},
1148 {"log", 1, 1, FEARG_1, NULL,
1149 ret_float, FLOAT_FUNC(f_log)},
1150 {"log10", 1, 1, FEARG_1, NULL,
1151 ret_float, FLOAT_FUNC(f_log10)},
1152 {"luaeval", 1, 2, FEARG_1, NULL,
1153 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001154#ifdef FEAT_LUA
Bram Moolenaar15c47602020-03-26 22:16:48 +01001155 f_luaeval
1156#else
1157 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001158#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001159 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001160 {"map", 2, 2, FEARG_1, NULL,
Bram Moolenaarea696852020-11-09 18:31:39 +01001161 ret_first_cont, f_map},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001162 {"maparg", 1, 4, FEARG_1, NULL,
1163 ret_maparg, f_maparg},
1164 {"mapcheck", 1, 3, FEARG_1, NULL,
1165 ret_string, f_mapcheck},
Bram Moolenaarea696852020-11-09 18:31:39 +01001166 {"mapnew", 2, 2, FEARG_1, NULL,
1167 ret_first_cont, f_mapnew},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001168 {"mapset", 3, 3, FEARG_1, NULL,
1169 ret_void, f_mapset},
1170 {"match", 2, 4, FEARG_1, NULL,
1171 ret_any, f_match},
1172 {"matchadd", 2, 5, FEARG_1, NULL,
1173 ret_number, f_matchadd},
1174 {"matchaddpos", 2, 5, FEARG_1, NULL,
1175 ret_number, f_matchaddpos},
1176 {"matcharg", 1, 1, FEARG_1, NULL,
1177 ret_list_string, f_matcharg},
1178 {"matchdelete", 1, 2, FEARG_1, NULL,
1179 ret_number, f_matchdelete},
1180 {"matchend", 2, 4, FEARG_1, NULL,
1181 ret_number, f_matchend},
1182 {"matchfuzzy", 2, 3, FEARG_1, NULL,
1183 ret_list_string, f_matchfuzzy},
1184 {"matchfuzzypos", 2, 3, FEARG_1, NULL,
1185 ret_list_any, f_matchfuzzypos},
1186 {"matchlist", 2, 4, FEARG_1, NULL,
1187 ret_list_string, f_matchlist},
1188 {"matchstr", 2, 4, FEARG_1, NULL,
1189 ret_string, f_matchstr},
1190 {"matchstrpos", 2, 4, FEARG_1, NULL,
1191 ret_list_any, f_matchstrpos},
1192 {"max", 1, 1, FEARG_1, NULL,
1193 ret_any, f_max},
1194 {"menu_info", 1, 2, FEARG_1, NULL,
1195 ret_dict_any,
Bram Moolenaara2cbdea2020-03-16 21:08:31 +01001196#ifdef FEAT_MENU
Bram Moolenaar15c47602020-03-26 22:16:48 +01001197 f_menu_info
1198#else
1199 NULL
Bram Moolenaara2cbdea2020-03-16 21:08:31 +01001200#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001201 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001202 {"min", 1, 1, FEARG_1, NULL,
1203 ret_any, f_min},
1204 {"mkdir", 1, 3, FEARG_1, NULL,
1205 ret_number, f_mkdir},
1206 {"mode", 0, 1, FEARG_1, NULL,
1207 ret_string, f_mode},
1208 {"mzeval", 1, 1, FEARG_1, NULL,
1209 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001210#ifdef FEAT_MZSCHEME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001211 f_mzeval
1212#else
1213 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001214#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001215 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001216 {"nextnonblank", 1, 1, FEARG_1, NULL,
1217 ret_number, f_nextnonblank},
1218 {"nr2char", 1, 2, FEARG_1, NULL,
1219 ret_string, f_nr2char},
1220 {"or", 2, 2, FEARG_1, NULL,
1221 ret_number, f_or},
1222 {"pathshorten", 1, 2, FEARG_1, NULL,
1223 ret_string, f_pathshorten},
1224 {"perleval", 1, 1, FEARG_1, NULL,
1225 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001226#ifdef FEAT_PERL
Bram Moolenaar15c47602020-03-26 22:16:48 +01001227 f_perleval
1228#else
1229 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001230#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001231 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001232 {"popup_atcursor", 2, 2, FEARG_1, NULL,
1233 ret_number, PROP_FUNC(f_popup_atcursor)},
1234 {"popup_beval", 2, 2, FEARG_1, NULL,
1235 ret_number, PROP_FUNC(f_popup_beval)},
1236 {"popup_clear", 0, 1, 0, NULL,
1237 ret_void, PROP_FUNC(f_popup_clear)},
1238 {"popup_close", 1, 2, FEARG_1, NULL,
1239 ret_void, PROP_FUNC(f_popup_close)},
1240 {"popup_create", 2, 2, FEARG_1, NULL,
1241 ret_number, PROP_FUNC(f_popup_create)},
1242 {"popup_dialog", 2, 2, FEARG_1, NULL,
1243 ret_number, PROP_FUNC(f_popup_dialog)},
1244 {"popup_filter_menu", 2, 2, 0, NULL,
1245 ret_bool, PROP_FUNC(f_popup_filter_menu)},
1246 {"popup_filter_yesno", 2, 2, 0, NULL,
1247 ret_bool, PROP_FUNC(f_popup_filter_yesno)},
1248 {"popup_findinfo", 0, 0, 0, NULL,
1249 ret_number, PROP_FUNC(f_popup_findinfo)},
1250 {"popup_findpreview", 0, 0, 0, NULL,
1251 ret_number, PROP_FUNC(f_popup_findpreview)},
1252 {"popup_getoptions", 1, 1, FEARG_1, NULL,
1253 ret_dict_any, PROP_FUNC(f_popup_getoptions)},
1254 {"popup_getpos", 1, 1, FEARG_1, NULL,
1255 ret_dict_any, PROP_FUNC(f_popup_getpos)},
1256 {"popup_hide", 1, 1, FEARG_1, NULL,
1257 ret_void, PROP_FUNC(f_popup_hide)},
1258 {"popup_list", 0, 0, 0, NULL,
1259 ret_list_number, PROP_FUNC(f_popup_list)},
1260 {"popup_locate", 2, 2, 0, NULL,
1261 ret_number, PROP_FUNC(f_popup_locate)},
1262 {"popup_menu", 2, 2, FEARG_1, NULL,
1263 ret_number, PROP_FUNC(f_popup_menu)},
1264 {"popup_move", 2, 2, FEARG_1, NULL,
1265 ret_void, PROP_FUNC(f_popup_move)},
1266 {"popup_notification", 2, 2, FEARG_1, NULL,
1267 ret_number, PROP_FUNC(f_popup_notification)},
1268 {"popup_setoptions", 2, 2, FEARG_1, NULL,
1269 ret_void, PROP_FUNC(f_popup_setoptions)},
1270 {"popup_settext", 2, 2, FEARG_1, NULL,
1271 ret_void, PROP_FUNC(f_popup_settext)},
1272 {"popup_show", 1, 1, FEARG_1, NULL,
1273 ret_void, PROP_FUNC(f_popup_show)},
1274 {"pow", 2, 2, FEARG_1, NULL,
1275 ret_float, FLOAT_FUNC(f_pow)},
1276 {"prevnonblank", 1, 1, FEARG_1, NULL,
1277 ret_number, f_prevnonblank},
1278 {"printf", 1, 19, FEARG_2, NULL,
1279 ret_string, f_printf},
1280 {"prompt_getprompt", 1, 1, FEARG_1, NULL,
1281 ret_string, JOB_FUNC(f_prompt_getprompt)},
1282 {"prompt_setcallback", 2, 2, FEARG_1, NULL,
1283 ret_void, JOB_FUNC(f_prompt_setcallback)},
1284 {"prompt_setinterrupt", 2, 2, FEARG_1, NULL,
1285 ret_void, JOB_FUNC(f_prompt_setinterrupt)},
1286 {"prompt_setprompt", 2, 2, FEARG_1, NULL,
1287 ret_void, JOB_FUNC(f_prompt_setprompt)},
1288 {"prop_add", 3, 3, FEARG_1, NULL,
1289 ret_void, PROP_FUNC(f_prop_add)},
1290 {"prop_clear", 1, 3, FEARG_1, NULL,
1291 ret_void, PROP_FUNC(f_prop_clear)},
1292 {"prop_find", 1, 2, FEARG_1, NULL,
1293 ret_dict_any, PROP_FUNC(f_prop_find)},
1294 {"prop_list", 1, 2, FEARG_1, NULL,
1295 ret_list_dict_any, PROP_FUNC(f_prop_list)},
1296 {"prop_remove", 1, 3, FEARG_1, NULL,
1297 ret_number, PROP_FUNC(f_prop_remove)},
1298 {"prop_type_add", 2, 2, FEARG_1, NULL,
1299 ret_void, PROP_FUNC(f_prop_type_add)},
1300 {"prop_type_change", 2, 2, FEARG_1, NULL,
1301 ret_void, PROP_FUNC(f_prop_type_change)},
1302 {"prop_type_delete", 1, 2, FEARG_1, NULL,
1303 ret_void, PROP_FUNC(f_prop_type_delete)},
1304 {"prop_type_get", 1, 2, FEARG_1, NULL,
1305 ret_dict_any, PROP_FUNC(f_prop_type_get)},
1306 {"prop_type_list", 0, 1, FEARG_1, NULL,
1307 ret_list_string, PROP_FUNC(f_prop_type_list)},
1308 {"pum_getpos", 0, 0, 0, NULL,
1309 ret_dict_number, f_pum_getpos},
1310 {"pumvisible", 0, 0, 0, NULL,
1311 ret_number, f_pumvisible},
1312 {"py3eval", 1, 1, FEARG_1, NULL,
1313 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001314#ifdef FEAT_PYTHON3
Bram Moolenaar15c47602020-03-26 22:16:48 +01001315 f_py3eval
1316#else
1317 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001318#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001319 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001320 {"pyeval", 1, 1, FEARG_1, NULL,
1321 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001322#ifdef FEAT_PYTHON
Bram Moolenaar15c47602020-03-26 22:16:48 +01001323 f_pyeval
1324#else
1325 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001326#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001327 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001328 {"pyxeval", 1, 1, FEARG_1, NULL,
1329 ret_any,
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01001330#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
Bram Moolenaar15c47602020-03-26 22:16:48 +01001331 f_pyxeval
1332#else
1333 NULL
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01001334#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001335 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001336 {"rand", 0, 1, FEARG_1, NULL,
1337 ret_number, f_rand},
1338 {"range", 1, 3, FEARG_1, NULL,
1339 ret_list_number, f_range},
1340 {"readdir", 1, 3, FEARG_1, NULL,
1341 ret_list_string, f_readdir},
1342 {"readdirex", 1, 3, FEARG_1, NULL,
1343 ret_list_dict_any, f_readdirex},
1344 {"readfile", 1, 3, FEARG_1, NULL,
1345 ret_any, f_readfile},
1346 {"reduce", 2, 3, FEARG_1, NULL,
1347 ret_any, f_reduce},
1348 {"reg_executing", 0, 0, 0, NULL,
1349 ret_string, f_reg_executing},
1350 {"reg_recording", 0, 0, 0, NULL,
1351 ret_string, f_reg_recording},
1352 {"reltime", 0, 2, FEARG_1, NULL,
1353 ret_list_any, f_reltime},
1354 {"reltimefloat", 1, 1, FEARG_1, NULL,
1355 ret_float, FLOAT_FUNC(f_reltimefloat)},
1356 {"reltimestr", 1, 1, FEARG_1, NULL,
1357 ret_string, f_reltimestr},
1358 {"remote_expr", 2, 4, FEARG_1, NULL,
1359 ret_string, f_remote_expr},
1360 {"remote_foreground", 1, 1, FEARG_1, NULL,
1361 ret_string, f_remote_foreground},
1362 {"remote_peek", 1, 2, FEARG_1, NULL,
1363 ret_number, f_remote_peek},
1364 {"remote_read", 1, 2, FEARG_1, NULL,
1365 ret_string, f_remote_read},
1366 {"remote_send", 2, 3, FEARG_1, NULL,
1367 ret_string, f_remote_send},
1368 {"remote_startserver", 1, 1, FEARG_1, NULL,
1369 ret_void, f_remote_startserver},
1370 {"remove", 2, 3, FEARG_1, NULL,
1371 ret_remove, f_remove},
1372 {"rename", 2, 2, FEARG_1, NULL,
1373 ret_number, f_rename},
1374 {"repeat", 2, 2, FEARG_1, NULL,
1375 ret_first_arg, f_repeat},
1376 {"resolve", 1, 1, FEARG_1, NULL,
1377 ret_string, f_resolve},
1378 {"reverse", 1, 1, FEARG_1, NULL,
1379 ret_first_arg, f_reverse},
1380 {"round", 1, 1, FEARG_1, NULL,
1381 ret_float, FLOAT_FUNC(f_round)},
1382 {"rubyeval", 1, 1, FEARG_1, NULL,
1383 ret_any,
Bram Moolenaare99be0e2019-03-26 22:51:09 +01001384#ifdef FEAT_RUBY
Bram Moolenaar15c47602020-03-26 22:16:48 +01001385 f_rubyeval
1386#else
1387 NULL
Bram Moolenaare99be0e2019-03-26 22:51:09 +01001388#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001389 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001390 {"screenattr", 2, 2, FEARG_1, NULL,
1391 ret_number, f_screenattr},
1392 {"screenchar", 2, 2, FEARG_1, NULL,
1393 ret_number, f_screenchar},
1394 {"screenchars", 2, 2, FEARG_1, NULL,
1395 ret_list_number, f_screenchars},
1396 {"screencol", 0, 0, 0, NULL,
1397 ret_number, f_screencol},
1398 {"screenpos", 3, 3, FEARG_1, NULL,
1399 ret_dict_number, f_screenpos},
1400 {"screenrow", 0, 0, 0, NULL,
1401 ret_number, f_screenrow},
1402 {"screenstring", 2, 2, FEARG_1, NULL,
1403 ret_string, f_screenstring},
1404 {"search", 1, 5, FEARG_1, NULL,
1405 ret_number, f_search},
1406 {"searchcount", 0, 1, FEARG_1, NULL,
1407 ret_dict_any, f_searchcount},
1408 {"searchdecl", 1, 3, FEARG_1, NULL,
1409 ret_number, f_searchdecl},
1410 {"searchpair", 3, 7, 0, NULL,
1411 ret_number, f_searchpair},
1412 {"searchpairpos", 3, 7, 0, NULL,
1413 ret_list_number, f_searchpairpos},
1414 {"searchpos", 1, 5, FEARG_1, NULL,
1415 ret_list_number, f_searchpos},
1416 {"server2client", 2, 2, FEARG_1, NULL,
1417 ret_number, f_server2client},
1418 {"serverlist", 0, 0, 0, NULL,
1419 ret_string, f_serverlist},
1420 {"setbufline", 3, 3, FEARG_3, NULL,
1421 ret_number, f_setbufline},
1422 {"setbufvar", 3, 3, FEARG_3, NULL,
1423 ret_void, f_setbufvar},
1424 {"setcellwidths", 1, 1, FEARG_1, NULL,
1425 ret_void, f_setcellwidths},
Bram Moolenaar6f02b002021-01-10 20:22:54 +01001426 {"setcharpos", 2, 2, FEARG_2, NULL,
1427 ret_number, f_setcharpos},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001428 {"setcharsearch", 1, 1, FEARG_1, NULL,
1429 ret_void, f_setcharsearch},
1430 {"setcmdpos", 1, 1, FEARG_1, NULL,
1431 ret_number, f_setcmdpos},
Bram Moolenaar6f02b002021-01-10 20:22:54 +01001432 {"setcursorcharpos", 1, 3, FEARG_1, NULL,
1433 ret_number, f_setcursorcharpos},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001434 {"setenv", 2, 2, FEARG_2, NULL,
1435 ret_void, f_setenv},
1436 {"setfperm", 2, 2, FEARG_1, NULL,
1437 ret_number, f_setfperm},
1438 {"setline", 2, 2, FEARG_2, NULL,
1439 ret_number, f_setline},
1440 {"setloclist", 2, 4, FEARG_2, NULL,
1441 ret_number, f_setloclist},
1442 {"setmatches", 1, 2, FEARG_1, NULL,
1443 ret_number, f_setmatches},
1444 {"setpos", 2, 2, FEARG_2, NULL,
1445 ret_number, f_setpos},
1446 {"setqflist", 1, 3, FEARG_1, NULL,
1447 ret_number, f_setqflist},
1448 {"setreg", 2, 3, FEARG_2, NULL,
1449 ret_number, f_setreg},
1450 {"settabvar", 3, 3, FEARG_3, NULL,
1451 ret_void, f_settabvar},
1452 {"settabwinvar", 4, 4, FEARG_4, NULL,
1453 ret_void, f_settabwinvar},
1454 {"settagstack", 2, 3, FEARG_2, NULL,
1455 ret_number, f_settagstack},
1456 {"setwinvar", 3, 3, FEARG_3, NULL,
1457 ret_void, f_setwinvar},
1458 {"sha256", 1, 1, FEARG_1, NULL,
1459 ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001460#ifdef FEAT_CRYPT
Bram Moolenaar15c47602020-03-26 22:16:48 +01001461 f_sha256
1462#else
1463 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001464#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001465 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001466 {"shellescape", 1, 2, FEARG_1, NULL,
1467 ret_string, f_shellescape},
1468 {"shiftwidth", 0, 1, FEARG_1, NULL,
1469 ret_number, f_shiftwidth},
1470 {"sign_define", 1, 2, FEARG_1, NULL,
1471 ret_any, SIGN_FUNC(f_sign_define)},
1472 {"sign_getdefined", 0, 1, FEARG_1, NULL,
1473 ret_list_dict_any, SIGN_FUNC(f_sign_getdefined)},
1474 {"sign_getplaced", 0, 2, FEARG_1, NULL,
1475 ret_list_dict_any, SIGN_FUNC(f_sign_getplaced)},
1476 {"sign_jump", 3, 3, FEARG_1, NULL,
1477 ret_number, SIGN_FUNC(f_sign_jump)},
1478 {"sign_place", 4, 5, FEARG_1, NULL,
1479 ret_number, SIGN_FUNC(f_sign_place)},
1480 {"sign_placelist", 1, 1, FEARG_1, NULL,
1481 ret_list_number, SIGN_FUNC(f_sign_placelist)},
1482 {"sign_undefine", 0, 1, FEARG_1, NULL,
1483 ret_number, SIGN_FUNC(f_sign_undefine)},
1484 {"sign_unplace", 1, 2, FEARG_1, NULL,
1485 ret_number, SIGN_FUNC(f_sign_unplace)},
1486 {"sign_unplacelist", 1, 2, FEARG_1, NULL,
1487 ret_list_number, SIGN_FUNC(f_sign_unplacelist)},
1488 {"simplify", 1, 1, FEARG_1, NULL,
1489 ret_string, f_simplify},
1490 {"sin", 1, 1, FEARG_1, NULL,
1491 ret_float, FLOAT_FUNC(f_sin)},
1492 {"sinh", 1, 1, FEARG_1, NULL,
1493 ret_float, FLOAT_FUNC(f_sinh)},
1494 {"sort", 1, 3, FEARG_1, NULL,
1495 ret_first_arg, f_sort},
1496 {"sound_clear", 0, 0, 0, NULL,
1497 ret_void, SOUND_FUNC(f_sound_clear)},
1498 {"sound_playevent", 1, 2, FEARG_1, NULL,
1499 ret_number, SOUND_FUNC(f_sound_playevent)},
1500 {"sound_playfile", 1, 2, FEARG_1, NULL,
1501 ret_number, SOUND_FUNC(f_sound_playfile)},
1502 {"sound_stop", 1, 1, FEARG_1, NULL,
1503 ret_void, SOUND_FUNC(f_sound_stop)},
1504 {"soundfold", 1, 1, FEARG_1, NULL,
1505 ret_string, f_soundfold},
1506 {"spellbadword", 0, 1, FEARG_1, NULL,
1507 ret_list_string, f_spellbadword},
1508 {"spellsuggest", 1, 3, FEARG_1, NULL,
1509 ret_list_string, f_spellsuggest},
1510 {"split", 1, 3, FEARG_1, NULL,
1511 ret_list_string, f_split},
1512 {"sqrt", 1, 1, FEARG_1, NULL,
1513 ret_float, FLOAT_FUNC(f_sqrt)},
1514 {"srand", 0, 1, FEARG_1, NULL,
1515 ret_list_number, f_srand},
1516 {"state", 0, 1, FEARG_1, NULL,
1517 ret_string, f_state},
1518 {"str2float", 1, 1, FEARG_1, NULL,
1519 ret_float, FLOAT_FUNC(f_str2float)},
1520 {"str2list", 1, 2, FEARG_1, NULL,
1521 ret_list_number, f_str2list},
1522 {"str2nr", 1, 3, FEARG_1, NULL,
1523 ret_number, f_str2nr},
1524 {"strcharpart", 2, 3, FEARG_1, NULL,
1525 ret_string, f_strcharpart},
1526 {"strchars", 1, 2, FEARG_1, NULL,
1527 ret_number, f_strchars},
1528 {"strdisplaywidth", 1, 2, FEARG_1, NULL,
1529 ret_number, f_strdisplaywidth},
1530 {"strftime", 1, 2, FEARG_1, NULL,
1531 ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001532#ifdef HAVE_STRFTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001533 f_strftime
1534#else
1535 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001536#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001537 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001538 {"strgetchar", 2, 2, FEARG_1, NULL,
1539 ret_number, f_strgetchar},
1540 {"stridx", 2, 3, FEARG_1, NULL,
1541 ret_number, f_stridx},
1542 {"string", 1, 1, FEARG_1, NULL,
1543 ret_string, f_string},
1544 {"strlen", 1, 1, FEARG_1, NULL,
1545 ret_number, f_strlen},
1546 {"strpart", 2, 4, FEARG_1, NULL,
1547 ret_string, f_strpart},
1548 {"strptime", 2, 2, FEARG_1, NULL,
1549 ret_number,
Bram Moolenaar10455d42019-11-21 15:36:18 +01001550#ifdef HAVE_STRPTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001551 f_strptime
1552#else
1553 NULL
Bram Moolenaar10455d42019-11-21 15:36:18 +01001554#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001555 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001556 {"strridx", 2, 3, FEARG_1, NULL,
1557 ret_number, f_strridx},
1558 {"strtrans", 1, 1, FEARG_1, NULL,
1559 ret_string, f_strtrans},
1560 {"strwidth", 1, 1, FEARG_1, NULL,
1561 ret_number, f_strwidth},
1562 {"submatch", 1, 2, FEARG_1, NULL,
1563 ret_string, f_submatch},
1564 {"substitute", 4, 4, FEARG_1, NULL,
1565 ret_string, f_substitute},
1566 {"swapinfo", 1, 1, FEARG_1, NULL,
1567 ret_dict_any, f_swapinfo},
1568 {"swapname", 1, 1, FEARG_1, NULL,
1569 ret_string, f_swapname},
1570 {"synID", 3, 3, 0, NULL,
1571 ret_number, f_synID},
1572 {"synIDattr", 2, 3, FEARG_1, NULL,
1573 ret_string, f_synIDattr},
1574 {"synIDtrans", 1, 1, FEARG_1, NULL,
1575 ret_number, f_synIDtrans},
1576 {"synconcealed", 2, 2, 0, NULL,
1577 ret_list_any, f_synconcealed},
1578 {"synstack", 2, 2, 0, NULL,
1579 ret_list_number, f_synstack},
1580 {"system", 1, 2, FEARG_1, NULL,
1581 ret_string, f_system},
1582 {"systemlist", 1, 2, FEARG_1, NULL,
1583 ret_list_string, f_systemlist},
1584 {"tabpagebuflist", 0, 1, FEARG_1, NULL,
1585 ret_list_number, f_tabpagebuflist},
1586 {"tabpagenr", 0, 1, 0, NULL,
1587 ret_number, f_tabpagenr},
1588 {"tabpagewinnr", 1, 2, FEARG_1, NULL,
1589 ret_number, f_tabpagewinnr},
1590 {"tagfiles", 0, 0, 0, NULL,
1591 ret_list_string, f_tagfiles},
1592 {"taglist", 1, 2, FEARG_1, NULL,
1593 ret_list_dict_any, f_taglist},
1594 {"tan", 1, 1, FEARG_1, NULL,
1595 ret_float, FLOAT_FUNC(f_tan)},
1596 {"tanh", 1, 1, FEARG_1, NULL,
1597 ret_float, FLOAT_FUNC(f_tanh)},
1598 {"tempname", 0, 0, 0, NULL,
1599 ret_string, f_tempname},
1600 {"term_dumpdiff", 2, 3, FEARG_1, NULL,
1601 ret_number, TERM_FUNC(f_term_dumpdiff)},
1602 {"term_dumpload", 1, 2, FEARG_1, NULL,
1603 ret_number, TERM_FUNC(f_term_dumpload)},
1604 {"term_dumpwrite", 2, 3, FEARG_2, NULL,
1605 ret_void, TERM_FUNC(f_term_dumpwrite)},
1606 {"term_getaltscreen", 1, 1, FEARG_1, NULL,
1607 ret_number, TERM_FUNC(f_term_getaltscreen)},
1608 {"term_getansicolors", 1, 1, FEARG_1, NULL,
1609 ret_list_string,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +01001610#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +01001611 f_term_getansicolors
1612#else
1613 NULL
Bram Moolenaarc6df10e2017-07-29 20:15:08 +02001614#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001615 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001616 {"term_getattr", 2, 2, FEARG_1, NULL,
1617 ret_number, TERM_FUNC(f_term_getattr)},
1618 {"term_getcursor", 1, 1, FEARG_1, NULL,
1619 ret_list_any, TERM_FUNC(f_term_getcursor)},
1620 {"term_getjob", 1, 1, FEARG_1, NULL,
1621 ret_job, TERM_FUNC(f_term_getjob)},
1622 {"term_getline", 2, 2, FEARG_1, NULL,
1623 ret_string, TERM_FUNC(f_term_getline)},
1624 {"term_getscrolled", 1, 1, FEARG_1, NULL,
1625 ret_number, TERM_FUNC(f_term_getscrolled)},
1626 {"term_getsize", 1, 1, FEARG_1, NULL,
1627 ret_list_number, TERM_FUNC(f_term_getsize)},
1628 {"term_getstatus", 1, 1, FEARG_1, NULL,
1629 ret_string, TERM_FUNC(f_term_getstatus)},
1630 {"term_gettitle", 1, 1, FEARG_1, NULL,
1631 ret_string, TERM_FUNC(f_term_gettitle)},
1632 {"term_gettty", 1, 2, FEARG_1, NULL,
1633 ret_string, TERM_FUNC(f_term_gettty)},
1634 {"term_list", 0, 0, 0, NULL,
1635 ret_list_number, TERM_FUNC(f_term_list)},
1636 {"term_scrape", 2, 2, FEARG_1, NULL,
1637 ret_list_dict_any, TERM_FUNC(f_term_scrape)},
1638 {"term_sendkeys", 2, 2, FEARG_1, NULL,
1639 ret_void, TERM_FUNC(f_term_sendkeys)},
1640 {"term_setansicolors", 2, 2, FEARG_1, NULL,
1641 ret_void,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +01001642#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +01001643 f_term_setansicolors
1644#else
1645 NULL
1646#endif
1647 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001648 {"term_setapi", 2, 2, FEARG_1, NULL,
1649 ret_void, TERM_FUNC(f_term_setapi)},
1650 {"term_setkill", 2, 2, FEARG_1, NULL,
1651 ret_void, TERM_FUNC(f_term_setkill)},
1652 {"term_setrestore", 2, 2, FEARG_1, NULL,
1653 ret_void, TERM_FUNC(f_term_setrestore)},
1654 {"term_setsize", 3, 3, FEARG_1, NULL,
1655 ret_void, TERM_FUNC(f_term_setsize)},
1656 {"term_start", 1, 2, FEARG_1, NULL,
1657 ret_number, TERM_FUNC(f_term_start)},
1658 {"term_wait", 1, 2, FEARG_1, NULL,
1659 ret_void, TERM_FUNC(f_term_wait)},
1660 {"terminalprops", 0, 0, 0, NULL,
1661 ret_dict_string, f_terminalprops},
1662 {"test_alloc_fail", 3, 3, FEARG_1, NULL,
1663 ret_void, f_test_alloc_fail},
1664 {"test_autochdir", 0, 0, 0, NULL,
1665 ret_void, f_test_autochdir},
1666 {"test_feedinput", 1, 1, FEARG_1, NULL,
1667 ret_void, f_test_feedinput},
1668 {"test_garbagecollect_now", 0, 0, 0, NULL,
1669 ret_void, f_test_garbagecollect_now},
1670 {"test_garbagecollect_soon", 0, 0, 0, NULL,
1671 ret_void, f_test_garbagecollect_soon},
1672 {"test_getvalue", 1, 1, FEARG_1, NULL,
1673 ret_number, f_test_getvalue},
1674 {"test_ignore_error", 1, 1, FEARG_1, NULL,
1675 ret_void, f_test_ignore_error},
1676 {"test_null_blob", 0, 0, 0, NULL,
1677 ret_blob, f_test_null_blob},
1678 {"test_null_channel", 0, 0, 0, NULL,
1679 ret_channel, JOB_FUNC(f_test_null_channel)},
1680 {"test_null_dict", 0, 0, 0, NULL,
1681 ret_dict_any, f_test_null_dict},
1682 {"test_null_function", 0, 0, 0, NULL,
1683 ret_func_any, f_test_null_function},
1684 {"test_null_job", 0, 0, 0, NULL,
1685 ret_job, JOB_FUNC(f_test_null_job)},
1686 {"test_null_list", 0, 0, 0, NULL,
1687 ret_list_any, f_test_null_list},
1688 {"test_null_partial", 0, 0, 0, NULL,
1689 ret_func_any, f_test_null_partial},
1690 {"test_null_string", 0, 0, 0, NULL,
1691 ret_string, f_test_null_string},
1692 {"test_option_not_set", 1, 1, FEARG_1, NULL,
1693 ret_void, f_test_option_not_set},
1694 {"test_override", 2, 2, FEARG_2, NULL,
1695 ret_void, f_test_override},
1696 {"test_refcount", 1, 1, FEARG_1, NULL,
1697 ret_number, f_test_refcount},
1698 {"test_scrollbar", 3, 3, FEARG_2, NULL,
1699 ret_void,
Bram Moolenaarab186732018-09-14 21:27:06 +02001700#ifdef FEAT_GUI
Bram Moolenaar15c47602020-03-26 22:16:48 +01001701 f_test_scrollbar
1702#else
1703 NULL
Bram Moolenaarab186732018-09-14 21:27:06 +02001704#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001705 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001706 {"test_setmouse", 2, 2, 0, NULL,
1707 ret_void, f_test_setmouse},
1708 {"test_settime", 1, 1, FEARG_1, NULL,
1709 ret_void, f_test_settime},
1710 {"test_srand_seed", 0, 1, FEARG_1, NULL,
1711 ret_void, f_test_srand_seed},
1712 {"test_unknown", 0, 0, 0, NULL,
1713 ret_any, f_test_unknown},
1714 {"test_void", 0, 0, 0, NULL,
1715 ret_void, f_test_void},
1716 {"timer_info", 0, 1, FEARG_1, NULL,
1717 ret_list_dict_any, TIMER_FUNC(f_timer_info)},
1718 {"timer_pause", 2, 2, FEARG_1, NULL,
1719 ret_void, TIMER_FUNC(f_timer_pause)},
1720 {"timer_start", 2, 3, FEARG_1, NULL,
1721 ret_number, TIMER_FUNC(f_timer_start)},
1722 {"timer_stop", 1, 1, FEARG_1, NULL,
1723 ret_void, TIMER_FUNC(f_timer_stop)},
1724 {"timer_stopall", 0, 0, 0, NULL,
1725 ret_void, TIMER_FUNC(f_timer_stopall)},
1726 {"tolower", 1, 1, FEARG_1, NULL,
1727 ret_string, f_tolower},
1728 {"toupper", 1, 1, FEARG_1, NULL,
1729 ret_string, f_toupper},
1730 {"tr", 3, 3, FEARG_1, NULL,
1731 ret_string, f_tr},
1732 {"trim", 1, 3, FEARG_1, NULL,
1733 ret_string, f_trim},
1734 {"trunc", 1, 1, FEARG_1, NULL,
1735 ret_float, FLOAT_FUNC(f_trunc)},
1736 {"type", 1, 1, FEARG_1, NULL,
1737 ret_number, f_type},
1738 {"undofile", 1, 1, FEARG_1, NULL,
1739 ret_string, f_undofile},
1740 {"undotree", 0, 0, 0, NULL,
1741 ret_dict_any, f_undotree},
1742 {"uniq", 1, 3, FEARG_1, NULL,
1743 ret_list_any, f_uniq},
1744 {"values", 1, 1, FEARG_1, NULL,
1745 ret_list_any, f_values},
1746 {"virtcol", 1, 1, FEARG_1, NULL,
1747 ret_number, f_virtcol},
1748 {"visualmode", 0, 1, 0, NULL,
1749 ret_string, f_visualmode},
1750 {"wildmenumode", 0, 0, 0, NULL,
1751 ret_number, f_wildmenumode},
1752 {"win_execute", 2, 3, FEARG_2, NULL,
1753 ret_string, f_win_execute},
1754 {"win_findbuf", 1, 1, FEARG_1, NULL,
1755 ret_list_number, f_win_findbuf},
1756 {"win_getid", 0, 2, FEARG_1, NULL,
1757 ret_number, f_win_getid},
1758 {"win_gettype", 0, 1, FEARG_1, NULL,
1759 ret_string, f_win_gettype},
1760 {"win_gotoid", 1, 1, FEARG_1, NULL,
1761 ret_number, f_win_gotoid},
1762 {"win_id2tabwin", 1, 1, FEARG_1, NULL,
1763 ret_list_number, f_win_id2tabwin},
1764 {"win_id2win", 1, 1, FEARG_1, NULL,
1765 ret_number, f_win_id2win},
1766 {"win_screenpos", 1, 1, FEARG_1, NULL,
1767 ret_list_number, f_win_screenpos},
1768 {"win_splitmove", 2, 3, FEARG_1, NULL,
1769 ret_number, f_win_splitmove},
1770 {"winbufnr", 1, 1, FEARG_1, NULL,
1771 ret_number, f_winbufnr},
1772 {"wincol", 0, 0, 0, NULL,
1773 ret_number, f_wincol},
1774 {"windowsversion", 0, 0, 0, NULL,
1775 ret_string, f_windowsversion},
1776 {"winheight", 1, 1, FEARG_1, NULL,
1777 ret_number, f_winheight},
1778 {"winlayout", 0, 1, FEARG_1, NULL,
1779 ret_list_any, f_winlayout},
1780 {"winline", 0, 0, 0, NULL,
1781 ret_number, f_winline},
1782 {"winnr", 0, 1, FEARG_1, NULL,
1783 ret_number, f_winnr},
1784 {"winrestcmd", 0, 0, 0, NULL,
1785 ret_string, f_winrestcmd},
1786 {"winrestview", 1, 1, FEARG_1, NULL,
1787 ret_void, f_winrestview},
1788 {"winsaveview", 0, 0, 0, NULL,
Bram Moolenaar43b69b32021-01-07 20:23:33 +01001789 ret_dict_number, f_winsaveview},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001790 {"winwidth", 1, 1, FEARG_1, NULL,
1791 ret_number, f_winwidth},
1792 {"wordcount", 0, 0, 0, NULL,
1793 ret_dict_number, f_wordcount},
1794 {"writefile", 2, 3, FEARG_1, NULL,
1795 ret_number, f_writefile},
1796 {"xor", 2, 2, FEARG_1, NULL,
1797 ret_number, f_xor},
Bram Moolenaarac92e252019-08-03 21:58:38 +02001798};
1799
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001800/*
1801 * Function given to ExpandGeneric() to obtain the list of internal
1802 * or user defined function names.
1803 */
1804 char_u *
1805get_function_name(expand_T *xp, int idx)
1806{
1807 static int intidx = -1;
1808 char_u *name;
1809
1810 if (idx == 0)
1811 intidx = -1;
1812 if (intidx < 0)
1813 {
1814 name = get_user_func_name(xp, idx);
1815 if (name != NULL)
1816 return name;
1817 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001818 if (++intidx < (int)(sizeof(global_functions) / sizeof(funcentry_T)))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001819 {
Bram Moolenaarac92e252019-08-03 21:58:38 +02001820 STRCPY(IObuff, global_functions[intidx].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001821 STRCAT(IObuff, "(");
Bram Moolenaarac92e252019-08-03 21:58:38 +02001822 if (global_functions[intidx].f_max_argc == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001823 STRCAT(IObuff, ")");
1824 return IObuff;
1825 }
1826
1827 return NULL;
1828}
1829
1830/*
1831 * Function given to ExpandGeneric() to obtain the list of internal or
1832 * user defined variable or function names.
1833 */
1834 char_u *
1835get_expr_name(expand_T *xp, int idx)
1836{
1837 static int intidx = -1;
1838 char_u *name;
1839
1840 if (idx == 0)
1841 intidx = -1;
1842 if (intidx < 0)
1843 {
1844 name = get_function_name(xp, idx);
1845 if (name != NULL)
1846 return name;
1847 }
1848 return get_user_var_name(xp, ++intidx);
1849}
1850
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001851/*
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001852 * Find internal function "name" in table "global_functions".
Bram Moolenaar15c47602020-03-26 22:16:48 +01001853 * Return index, or -1 if not found or "implemented" is TRUE and the function
1854 * is not implemented.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001855 */
Bram Moolenaar15c47602020-03-26 22:16:48 +01001856 static int
1857find_internal_func_opt(char_u *name, int implemented)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001858{
1859 int first = 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001860 int last;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001861 int cmp;
1862 int x;
1863
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001864 last = (int)(sizeof(global_functions) / sizeof(funcentry_T)) - 1;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001865
1866 // Find the function name in the table. Binary search.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001867 while (first <= last)
1868 {
1869 x = first + ((unsigned)(last - first) >> 1);
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001870 cmp = STRCMP(name, global_functions[x].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001871 if (cmp < 0)
1872 last = x - 1;
1873 else if (cmp > 0)
1874 first = x + 1;
Bram Moolenaar15c47602020-03-26 22:16:48 +01001875 else if (implemented && global_functions[x].f_func == NULL)
1876 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001877 else
1878 return x;
1879 }
1880 return -1;
1881}
1882
Bram Moolenaar15c47602020-03-26 22:16:48 +01001883/*
1884 * Find internal function "name" in table "global_functions".
1885 * Return index, or -1 if not found or the function is not implemented.
1886 */
1887 int
1888find_internal_func(char_u *name)
1889{
1890 return find_internal_func_opt(name, TRUE);
1891}
1892
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001893 int
Bram Moolenaarac92e252019-08-03 21:58:38 +02001894has_internal_func(char_u *name)
1895{
Bram Moolenaar15c47602020-03-26 22:16:48 +01001896 return find_internal_func_opt(name, TRUE) >= 0;
1897}
1898
1899 static int
1900has_internal_func_name(char_u *name)
1901{
1902 return find_internal_func_opt(name, FALSE) >= 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001903}
1904
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001905 char *
1906internal_func_name(int idx)
1907{
1908 return global_functions[idx].f_name;
1909}
1910
Bram Moolenaar94738d82020-10-21 14:25:07 +02001911/*
1912 * Check the argument types for builting function "idx".
1913 * Uses the list of types on the type stack: "types".
1914 * Return FAIL and gives an error message when a type is wrong.
1915 */
1916 int
Bram Moolenaarca174532020-10-21 16:42:22 +02001917internal_func_check_arg_types(type_T **types, int idx, int argcount)
Bram Moolenaar94738d82020-10-21 14:25:07 +02001918{
1919 argcheck_T *argchecks = global_functions[idx].f_argcheck;
1920 int i;
1921
1922 if (argchecks != NULL)
1923 {
1924 argcontext_T context;
1925
1926 context.arg_count = argcount;
Bram Moolenaarca174532020-10-21 16:42:22 +02001927 context.arg_types = types;
Bram Moolenaar94738d82020-10-21 14:25:07 +02001928 for (i = 0; i < argcount; ++i)
1929 if (argchecks[i] != NULL)
1930 {
1931 context.arg_idx = i;
Bram Moolenaarca174532020-10-21 16:42:22 +02001932 if (argchecks[i](types[i], &context) == FAIL)
Bram Moolenaar94738d82020-10-21 14:25:07 +02001933 return FAIL;
1934 }
1935 }
1936 return OK;
1937}
1938
Bram Moolenaara1224cb2020-10-22 12:31:49 +02001939/*
1940 * Call the "f_retfunc" function to obtain the return type of function "idx".
1941 * "argtypes" is the list of argument types or NULL when there are no
1942 * arguments.
1943 */
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001944 type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001945internal_func_ret_type(int idx, int argcount, type_T **argtypes)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001946{
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001947 return global_functions[idx].f_retfunc(argcount, argtypes);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001948}
1949
1950/*
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01001951 * Return TRUE if "idx" is for the map() function.
1952 */
1953 int
1954internal_func_is_map(int idx)
1955{
1956 return global_functions[idx].f_func == f_map;
1957}
1958
1959/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001960 * Check the argument count to use for internal function "idx".
Bram Moolenaar389df252020-07-09 21:20:47 +02001961 * Returns -1 for failure, 0 if no method base accepted, 1 if method base is
1962 * first argument, 2 if method base is second argument, etc. 9 if method base
1963 * is last argument.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001964 */
1965 int
1966check_internal_func(int idx, int argcount)
1967{
1968 int res;
1969 char *name;
1970
1971 if (argcount < global_functions[idx].f_min_argc)
1972 res = FCERR_TOOFEW;
1973 else if (argcount > global_functions[idx].f_max_argc)
1974 res = FCERR_TOOMANY;
1975 else
Bram Moolenaar389df252020-07-09 21:20:47 +02001976 return global_functions[idx].f_argtype;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001977
1978 name = internal_func_name(idx);
1979 if (res == FCERR_TOOMANY)
1980 semsg(_(e_toomanyarg), name);
1981 else
1982 semsg(_(e_toofewarg), name);
Bram Moolenaar389df252020-07-09 21:20:47 +02001983 return -1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001984}
1985
Bram Moolenaarac92e252019-08-03 21:58:38 +02001986 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001987call_internal_func(
1988 char_u *name,
1989 int argcount,
1990 typval_T *argvars,
1991 typval_T *rettv)
1992{
1993 int i;
1994
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001995 i = find_internal_func(name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001996 if (i < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001997 return FCERR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001998 if (argcount < global_functions[i].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001999 return FCERR_TOOFEW;
Bram Moolenaarac92e252019-08-03 21:58:38 +02002000 if (argcount > global_functions[i].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01002001 return FCERR_TOOMANY;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002002 argvars[argcount].v_type = VAR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02002003 global_functions[i].f_func(argvars, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01002004 return FCERR_NONE;
Bram Moolenaarac92e252019-08-03 21:58:38 +02002005}
2006
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002007 void
2008call_internal_func_by_idx(
2009 int idx,
2010 typval_T *argvars,
2011 typval_T *rettv)
2012{
2013 global_functions[idx].f_func(argvars, rettv);
2014}
2015
Bram Moolenaarac92e252019-08-03 21:58:38 +02002016/*
2017 * Invoke a method for base->method().
2018 */
2019 int
2020call_internal_method(
2021 char_u *name,
2022 int argcount,
2023 typval_T *argvars,
2024 typval_T *rettv,
2025 typval_T *basetv)
2026{
2027 int i;
2028 int fi;
2029 typval_T argv[MAX_FUNC_ARGS + 1];
2030
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002031 fi = find_internal_func(name);
Bram Moolenaar91746392019-08-16 22:22:31 +02002032 if (fi < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01002033 return FCERR_UNKNOWN;
Bram Moolenaar91746392019-08-16 22:22:31 +02002034 if (global_functions[fi].f_argtype == 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01002035 return FCERR_NOTMETHOD;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002036 if (argcount + 1 < global_functions[fi].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01002037 return FCERR_TOOFEW;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002038 if (argcount + 1 > global_functions[fi].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01002039 return FCERR_TOOMANY;
Bram Moolenaarac92e252019-08-03 21:58:38 +02002040
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002041 if (global_functions[fi].f_argtype == FEARG_LAST)
Bram Moolenaar25e42232019-08-04 15:04:10 +02002042 {
2043 // base value goes last
2044 for (i = 0; i < argcount; ++i)
2045 argv[i] = argvars[i];
2046 argv[argcount] = *basetv;
2047 }
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002048 else if (global_functions[fi].f_argtype == FEARG_2)
Bram Moolenaar25e42232019-08-04 15:04:10 +02002049 {
2050 // base value goes second
2051 argv[0] = argvars[0];
2052 argv[1] = *basetv;
2053 for (i = 1; i < argcount; ++i)
2054 argv[i + 1] = argvars[i];
2055 }
Bram Moolenaar24278d22019-08-16 21:49:22 +02002056 else if (global_functions[fi].f_argtype == FEARG_3)
2057 {
2058 // base value goes third
2059 argv[0] = argvars[0];
2060 argv[1] = argvars[1];
2061 argv[2] = *basetv;
2062 for (i = 2; i < argcount; ++i)
2063 argv[i + 1] = argvars[i];
2064 }
Bram Moolenaaraad222c2019-09-06 22:46:09 +02002065 else if (global_functions[fi].f_argtype == FEARG_4)
2066 {
2067 // base value goes fourth
2068 argv[0] = argvars[0];
2069 argv[1] = argvars[1];
2070 argv[2] = argvars[2];
2071 argv[3] = *basetv;
2072 for (i = 3; i < argcount; ++i)
2073 argv[i + 1] = argvars[i];
2074 }
Bram Moolenaar25e42232019-08-04 15:04:10 +02002075 else
2076 {
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002077 // FEARG_1: base value goes first
Bram Moolenaar25e42232019-08-04 15:04:10 +02002078 argv[0] = *basetv;
2079 for (i = 0; i < argcount; ++i)
2080 argv[i + 1] = argvars[i];
2081 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02002082 argv[argcount + 1].v_type = VAR_UNKNOWN;
2083
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002084 global_functions[fi].f_func(argv, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01002085 return FCERR_NONE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002086}
2087
2088/*
2089 * Return TRUE for a non-zero Number and a non-empty String.
2090 */
Bram Moolenaar0e57dd82019-09-16 22:56:03 +02002091 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002092non_zero_arg(typval_T *argvars)
2093{
2094 return ((argvars[0].v_type == VAR_NUMBER
2095 && argvars[0].vval.v_number != 0)
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002096 || (argvars[0].v_type == VAR_BOOL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002097 && argvars[0].vval.v_number == VVAL_TRUE)
2098 || (argvars[0].v_type == VAR_STRING
2099 && argvars[0].vval.v_string != NULL
2100 && *argvars[0].vval.v_string != NUL));
2101}
2102
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002103#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002104/*
2105 * Get the float value of "argvars[0]" into "f".
2106 * Returns FAIL when the argument is not a Number or Float.
2107 */
2108 static int
2109get_float_arg(typval_T *argvars, float_T *f)
2110{
2111 if (argvars[0].v_type == VAR_FLOAT)
2112 {
2113 *f = argvars[0].vval.v_float;
2114 return OK;
2115 }
2116 if (argvars[0].v_type == VAR_NUMBER)
2117 {
2118 *f = (float_T)argvars[0].vval.v_number;
2119 return OK;
2120 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002121 emsg(_("E808: Number or Float required"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002122 return FAIL;
2123}
2124
2125/*
2126 * "abs(expr)" function
2127 */
2128 static void
2129f_abs(typval_T *argvars, typval_T *rettv)
2130{
2131 if (argvars[0].v_type == VAR_FLOAT)
2132 {
2133 rettv->v_type = VAR_FLOAT;
2134 rettv->vval.v_float = fabs(argvars[0].vval.v_float);
2135 }
2136 else
2137 {
2138 varnumber_T n;
2139 int error = FALSE;
2140
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002141 n = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002142 if (error)
2143 rettv->vval.v_number = -1;
2144 else if (n > 0)
2145 rettv->vval.v_number = n;
2146 else
2147 rettv->vval.v_number = -n;
2148 }
2149}
2150
2151/*
2152 * "acos()" function
2153 */
2154 static void
2155f_acos(typval_T *argvars, typval_T *rettv)
2156{
2157 float_T f = 0.0;
2158
2159 rettv->v_type = VAR_FLOAT;
2160 if (get_float_arg(argvars, &f) == OK)
2161 rettv->vval.v_float = acos(f);
2162 else
2163 rettv->vval.v_float = 0.0;
2164}
2165#endif
2166
2167/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002168 * "and(expr, expr)" function
2169 */
2170 static void
2171f_and(typval_T *argvars, typval_T *rettv)
2172{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002173 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
2174 & tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaarca851592018-06-06 21:04:07 +02002175}
2176
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002177#ifdef FEAT_FLOAT
2178/*
2179 * "asin()" function
2180 */
2181 static void
2182f_asin(typval_T *argvars, typval_T *rettv)
2183{
2184 float_T f = 0.0;
2185
2186 rettv->v_type = VAR_FLOAT;
2187 if (get_float_arg(argvars, &f) == OK)
2188 rettv->vval.v_float = asin(f);
2189 else
2190 rettv->vval.v_float = 0.0;
2191}
2192
2193/*
2194 * "atan()" function
2195 */
2196 static void
2197f_atan(typval_T *argvars, typval_T *rettv)
2198{
2199 float_T f = 0.0;
2200
2201 rettv->v_type = VAR_FLOAT;
2202 if (get_float_arg(argvars, &f) == OK)
2203 rettv->vval.v_float = atan(f);
2204 else
2205 rettv->vval.v_float = 0.0;
2206}
2207
2208/*
2209 * "atan2()" function
2210 */
2211 static void
2212f_atan2(typval_T *argvars, typval_T *rettv)
2213{
2214 float_T fx = 0.0, fy = 0.0;
2215
2216 rettv->v_type = VAR_FLOAT;
2217 if (get_float_arg(argvars, &fx) == OK
2218 && get_float_arg(&argvars[1], &fy) == OK)
2219 rettv->vval.v_float = atan2(fx, fy);
2220 else
2221 rettv->vval.v_float = 0.0;
2222}
2223#endif
2224
2225/*
Bram Moolenaar59716a22017-03-01 20:32:44 +01002226 * "balloon_show()" function
2227 */
2228#ifdef FEAT_BEVAL
2229 static void
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002230f_balloon_gettext(typval_T *argvars UNUSED, typval_T *rettv)
2231{
2232 rettv->v_type = VAR_STRING;
2233 if (balloonEval != NULL)
2234 {
2235 if (balloonEval->msg == NULL)
2236 rettv->vval.v_string = NULL;
2237 else
2238 rettv->vval.v_string = vim_strsave(balloonEval->msg);
2239 }
2240}
2241
2242 static void
Bram Moolenaar59716a22017-03-01 20:32:44 +01002243f_balloon_show(typval_T *argvars, typval_T *rettv UNUSED)
2244{
Bram Moolenaarcaf64342017-03-02 22:11:33 +01002245 if (balloonEval != NULL)
Bram Moolenaar246fe032017-11-19 19:56:27 +01002246 {
2247 if (argvars[0].v_type == VAR_LIST
2248# ifdef FEAT_GUI
2249 && !gui.in_use
2250# endif
2251 )
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002252 {
2253 list_T *l = argvars[0].vval.v_list;
2254
2255 // empty list removes the balloon
2256 post_balloon(balloonEval, NULL,
2257 l == NULL || l->lv_len == 0 ? NULL : l);
2258 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01002259 else
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002260 {
2261 char_u *mesg = tv_get_string_chk(&argvars[0]);
2262
2263 if (mesg != NULL)
2264 // empty string removes the balloon
2265 post_balloon(balloonEval, *mesg == NUL ? NULL : mesg, NULL);
2266 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01002267 }
2268}
2269
Bram Moolenaar669a8282017-11-19 20:13:05 +01002270# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +01002271 static void
2272f_balloon_split(typval_T *argvars, typval_T *rettv UNUSED)
2273{
2274 if (rettv_list_alloc(rettv) == OK)
2275 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002276 char_u *msg = tv_get_string_chk(&argvars[0]);
Bram Moolenaar246fe032017-11-19 19:56:27 +01002277
2278 if (msg != NULL)
2279 {
2280 pumitem_T *array;
2281 int size = split_message(msg, &array);
2282 int i;
2283
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002284 // Skip the first and last item, they are always empty.
Bram Moolenaar246fe032017-11-19 19:56:27 +01002285 for (i = 1; i < size - 1; ++i)
2286 list_append_string(rettv->vval.v_list, array[i].pum_text, -1);
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01002287 while (size > 0)
2288 vim_free(array[--size].pum_text);
Bram Moolenaar246fe032017-11-19 19:56:27 +01002289 vim_free(array);
2290 }
2291 }
Bram Moolenaar59716a22017-03-01 20:32:44 +01002292}
Bram Moolenaar669a8282017-11-19 20:13:05 +01002293# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +01002294#endif
2295
2296/*
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002297 * Get the buffer from "arg" and give an error and return NULL if it is not
2298 * valid.
2299 */
Bram Moolenaara3347722019-05-11 21:14:24 +02002300 buf_T *
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002301get_buf_arg(typval_T *arg)
2302{
2303 buf_T *buf;
2304
2305 ++emsg_off;
2306 buf = tv_get_buf(arg, FALSE);
2307 --emsg_off;
2308 if (buf == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002309 semsg(_("E158: Invalid buffer name: %s"), tv_get_string(arg));
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002310 return buf;
2311}
2312
2313/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002314 * "byte2line(byte)" function
2315 */
2316 static void
2317f_byte2line(typval_T *argvars UNUSED, typval_T *rettv)
2318{
2319#ifndef FEAT_BYTEOFF
2320 rettv->vval.v_number = -1;
2321#else
2322 long boff = 0;
2323
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002324 boff = tv_get_number(&argvars[0]) - 1; // boff gets -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002325 if (boff < 0)
2326 rettv->vval.v_number = -1;
2327 else
2328 rettv->vval.v_number = ml_find_line_or_offset(curbuf,
2329 (linenr_T)0, &boff);
2330#endif
2331}
2332
2333 static void
2334byteidx(typval_T *argvars, typval_T *rettv, int comp UNUSED)
2335{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002336 char_u *t;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002337 char_u *str;
2338 varnumber_T idx;
2339
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002340 str = tv_get_string_chk(&argvars[0]);
2341 idx = tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002342 rettv->vval.v_number = -1;
2343 if (str == NULL || idx < 0)
2344 return;
2345
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002346 t = str;
2347 for ( ; idx > 0; idx--)
2348 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002349 if (*t == NUL) // EOL reached
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002350 return;
2351 if (enc_utf8 && comp)
2352 t += utf_ptr2len(t);
2353 else
2354 t += (*mb_ptr2len)(t);
2355 }
2356 rettv->vval.v_number = (varnumber_T)(t - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002357}
2358
2359/*
2360 * "byteidx()" function
2361 */
2362 static void
2363f_byteidx(typval_T *argvars, typval_T *rettv)
2364{
2365 byteidx(argvars, rettv, FALSE);
2366}
2367
2368/*
2369 * "byteidxcomp()" function
2370 */
2371 static void
2372f_byteidxcomp(typval_T *argvars, typval_T *rettv)
2373{
2374 byteidx(argvars, rettv, TRUE);
2375}
2376
2377/*
2378 * "call(func, arglist [, dict])" function
2379 */
2380 static void
2381f_call(typval_T *argvars, typval_T *rettv)
2382{
2383 char_u *func;
2384 partial_T *partial = NULL;
2385 dict_T *selfdict = NULL;
2386
2387 if (argvars[1].v_type != VAR_LIST)
2388 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002389 emsg(_(e_listreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002390 return;
2391 }
2392 if (argvars[1].vval.v_list == NULL)
2393 return;
2394
2395 if (argvars[0].v_type == VAR_FUNC)
2396 func = argvars[0].vval.v_string;
2397 else if (argvars[0].v_type == VAR_PARTIAL)
2398 {
2399 partial = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002400 func = partial_name(partial);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002401 }
2402 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002403 func = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002404 if (*func == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002405 return; // type error or empty name
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002406
2407 if (argvars[2].v_type != VAR_UNKNOWN)
2408 {
2409 if (argvars[2].v_type != VAR_DICT)
2410 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002411 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002412 return;
2413 }
2414 selfdict = argvars[2].vval.v_dict;
2415 }
2416
2417 (void)func_call(func, &argvars[1], partial, selfdict, rettv);
2418}
2419
2420#ifdef FEAT_FLOAT
2421/*
2422 * "ceil({float})" function
2423 */
2424 static void
2425f_ceil(typval_T *argvars, typval_T *rettv)
2426{
2427 float_T f = 0.0;
2428
2429 rettv->v_type = VAR_FLOAT;
2430 if (get_float_arg(argvars, &f) == OK)
2431 rettv->vval.v_float = ceil(f);
2432 else
2433 rettv->vval.v_float = 0.0;
2434}
2435#endif
2436
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002437/*
2438 * "changenr()" function
2439 */
2440 static void
2441f_changenr(typval_T *argvars UNUSED, typval_T *rettv)
2442{
2443 rettv->vval.v_number = curbuf->b_u_seq_cur;
2444}
2445
2446/*
2447 * "char2nr(string)" function
2448 */
2449 static void
2450f_char2nr(typval_T *argvars, typval_T *rettv)
2451{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002452 if (has_mbyte)
2453 {
2454 int utf8 = 0;
2455
2456 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar24f77502020-09-04 19:50:57 +02002457 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002458
2459 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01002460 rettv->vval.v_number = utf_ptr2char(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002461 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002462 rettv->vval.v_number = (*mb_ptr2char)(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002463 }
2464 else
Bram Moolenaar13505972019-01-24 15:04:48 +01002465 rettv->vval.v_number = tv_get_string(&argvars[0])[0];
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002466}
2467
Bram Moolenaar17793ef2020-12-28 12:56:58 +01002468/*
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002469 * Get the current cursor column and store it in 'rettv'. If 'charcol' is TRUE,
2470 * returns the character index of the column. Otherwise, returns the byte index
2471 * of the column.
2472 */
2473 static void
2474get_col(typval_T *argvars, typval_T *rettv, int charcol)
2475{
2476 colnr_T col = 0;
2477 pos_T *fp;
2478 int fnum = curbuf->b_fnum;
2479
2480 fp = var2fpos(&argvars[0], FALSE, &fnum, charcol);
2481 if (fp != NULL && fnum == curbuf->b_fnum)
2482 {
2483 if (fp->col == MAXCOL)
2484 {
2485 // '> can be MAXCOL, get the length of the line then
2486 if (fp->lnum <= curbuf->b_ml.ml_line_count)
2487 col = (colnr_T)STRLEN(ml_get(fp->lnum)) + 1;
2488 else
2489 col = MAXCOL;
2490 }
2491 else
2492 {
2493 col = fp->col + 1;
2494 // col(".") when the cursor is on the NUL at the end of the line
2495 // because of "coladd" can be seen as an extra column.
2496 if (virtual_active() && fp == &curwin->w_cursor)
2497 {
2498 char_u *p = ml_get_cursor();
2499
2500 if (curwin->w_cursor.coladd >= (colnr_T)chartabsize(p,
2501 curwin->w_virtcol - curwin->w_cursor.coladd))
2502 {
2503 int l;
2504
2505 if (*p != NUL && p[(l = (*mb_ptr2len)(p))] == NUL)
2506 col += l;
2507 }
2508 }
2509 }
2510 }
2511 rettv->vval.v_number = col;
2512}
2513
2514/*
2515 * "charcol()" function
2516 */
2517 static void
2518f_charcol(typval_T *argvars, typval_T *rettv)
2519{
2520 get_col(argvars, rettv, TRUE);
2521}
2522
2523/*
Bram Moolenaar17793ef2020-12-28 12:56:58 +01002524 * "charidx()" function
2525 */
2526 static void
2527f_charidx(typval_T *argvars, typval_T *rettv)
2528{
2529 char_u *str;
2530 varnumber_T idx;
2531 int countcc = FALSE;
2532 char_u *p;
2533 int len;
2534 int (*ptr2len)(char_u *);
2535
2536 rettv->vval.v_number = -1;
2537
2538 if (argvars[0].v_type != VAR_STRING || argvars[1].v_type != VAR_NUMBER
2539 || (argvars[2].v_type != VAR_UNKNOWN
2540 && argvars[2].v_type != VAR_NUMBER))
2541 {
2542 emsg(_(e_invarg));
2543 return;
2544 }
2545
2546 str = tv_get_string_chk(&argvars[0]);
2547 idx = tv_get_number_chk(&argvars[1], NULL);
2548 if (str == NULL || idx < 0)
2549 return;
2550
2551 if (argvars[2].v_type != VAR_UNKNOWN)
2552 countcc = (int)tv_get_bool(&argvars[2]);
2553 if (countcc < 0 || countcc > 1)
2554 {
2555 semsg(_(e_using_number_as_bool_nr), countcc);
2556 return;
2557 }
2558
2559 if (enc_utf8 && countcc)
2560 ptr2len = utf_ptr2len;
2561 else
2562 ptr2len = mb_ptr2len;
2563
2564 for (p = str, len = 0; p <= str + idx; len++)
2565 {
2566 if (*p == NUL)
2567 return;
2568 p += ptr2len(p);
2569 }
2570
2571 rettv->vval.v_number = len > 0 ? len - 1 : 0;
2572}
2573
Bram Moolenaar29b7d7a2019-07-22 23:03:57 +02002574 win_T *
Bram Moolenaaraff74912019-03-30 18:11:49 +01002575get_optional_window(typval_T *argvars, int idx)
2576{
2577 win_T *win = curwin;
2578
2579 if (argvars[idx].v_type != VAR_UNKNOWN)
2580 {
2581 win = find_win_by_nr_or_id(&argvars[idx]);
2582 if (win == NULL)
2583 {
2584 emsg(_(e_invalwindow));
2585 return NULL;
2586 }
2587 }
2588 return win;
2589}
2590
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002591/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002592 * "col(string)" function
2593 */
2594 static void
2595f_col(typval_T *argvars, typval_T *rettv)
2596{
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002597 get_col(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002598}
2599
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002600/*
2601 * "confirm(message, buttons[, default [, type]])" function
2602 */
2603 static void
2604f_confirm(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2605{
2606#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2607 char_u *message;
2608 char_u *buttons = NULL;
2609 char_u buf[NUMBUFLEN];
2610 char_u buf2[NUMBUFLEN];
2611 int def = 1;
2612 int type = VIM_GENERIC;
2613 char_u *typestr;
2614 int error = FALSE;
2615
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002616 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002617 if (message == NULL)
2618 error = TRUE;
2619 if (argvars[1].v_type != VAR_UNKNOWN)
2620 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002621 buttons = tv_get_string_buf_chk(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002622 if (buttons == NULL)
2623 error = TRUE;
2624 if (argvars[2].v_type != VAR_UNKNOWN)
2625 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002626 def = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002627 if (argvars[3].v_type != VAR_UNKNOWN)
2628 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002629 typestr = tv_get_string_buf_chk(&argvars[3], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002630 if (typestr == NULL)
2631 error = TRUE;
2632 else
2633 {
2634 switch (TOUPPER_ASC(*typestr))
2635 {
2636 case 'E': type = VIM_ERROR; break;
2637 case 'Q': type = VIM_QUESTION; break;
2638 case 'I': type = VIM_INFO; break;
2639 case 'W': type = VIM_WARNING; break;
2640 case 'G': type = VIM_GENERIC; break;
2641 }
2642 }
2643 }
2644 }
2645 }
2646
2647 if (buttons == NULL || *buttons == NUL)
2648 buttons = (char_u *)_("&Ok");
2649
2650 if (!error)
2651 rettv->vval.v_number = do_dialog(type, NULL, message, buttons,
2652 def, NULL, FALSE);
2653#endif
2654}
2655
2656/*
2657 * "copy()" function
2658 */
2659 static void
2660f_copy(typval_T *argvars, typval_T *rettv)
2661{
2662 item_copy(&argvars[0], rettv, FALSE, 0);
2663}
2664
2665#ifdef FEAT_FLOAT
2666/*
2667 * "cos()" function
2668 */
2669 static void
2670f_cos(typval_T *argvars, typval_T *rettv)
2671{
2672 float_T f = 0.0;
2673
2674 rettv->v_type = VAR_FLOAT;
2675 if (get_float_arg(argvars, &f) == OK)
2676 rettv->vval.v_float = cos(f);
2677 else
2678 rettv->vval.v_float = 0.0;
2679}
2680
2681/*
2682 * "cosh()" function
2683 */
2684 static void
2685f_cosh(typval_T *argvars, typval_T *rettv)
2686{
2687 float_T f = 0.0;
2688
2689 rettv->v_type = VAR_FLOAT;
2690 if (get_float_arg(argvars, &f) == OK)
2691 rettv->vval.v_float = cosh(f);
2692 else
2693 rettv->vval.v_float = 0.0;
2694}
2695#endif
2696
2697/*
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002698 * Set the cursor position.
2699 * If 'charcol' is TRUE, then use the column number as a character offet.
2700 * Otherwise use the column number as a byte offset.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002701 */
2702 static void
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002703set_cursorpos(typval_T *argvars, typval_T *rettv, int charcol)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002704{
2705 long line, col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002706 long coladd = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002707 int set_curswant = TRUE;
2708
2709 rettv->vval.v_number = -1;
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002710 if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002711 {
2712 pos_T pos;
2713 colnr_T curswant = -1;
2714
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002715 if (list2fpos(argvars, &pos, NULL, &curswant, charcol) == FAIL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002716 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002717 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002718 return;
2719 }
2720 line = pos.lnum;
2721 col = pos.col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002722 coladd = pos.coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002723 if (curswant >= 0)
2724 {
2725 curwin->w_curswant = curswant - 1;
2726 set_curswant = FALSE;
2727 }
2728 }
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002729 else if ((argvars[0].v_type == VAR_NUMBER ||
2730 argvars[0].v_type == VAR_STRING)
2731 && argvars[1].v_type == VAR_NUMBER)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002732 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002733 line = tv_get_lnum(argvars);
Bram Moolenaar9a963372020-12-21 21:58:46 +01002734 if (line < 0)
2735 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002736 col = (long)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002737 if (charcol)
2738 col = buf_charidx_to_byteidx(curbuf, line, col);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002739 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002740 coladd = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002741 }
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002742 else
2743 {
2744 emsg(_(e_invarg));
2745 return;
2746 }
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01002747 if (line < 0 || col < 0 || coladd < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002748 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002749 if (line > 0)
2750 curwin->w_cursor.lnum = line;
2751 if (col > 0)
2752 curwin->w_cursor.col = col - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002753 curwin->w_cursor.coladd = coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002754
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002755 // Make sure the cursor is in a valid position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002756 check_cursor();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002757 // Correct cursor for multi-byte character.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002758 if (has_mbyte)
2759 mb_adjust_cursor();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002760
2761 curwin->w_set_curswant = set_curswant;
2762 rettv->vval.v_number = 0;
2763}
2764
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002765/*
2766 * "cursor(lnum, col)" function, or
2767 * "cursor(list)"
2768 *
2769 * Moves the cursor to the specified line and column.
2770 * Returns 0 when the position could be set, -1 otherwise.
2771 */
2772 static void
2773f_cursor(typval_T *argvars, typval_T *rettv)
2774{
2775 set_cursorpos(argvars, rettv, FALSE);
2776}
2777
Bram Moolenaar4f974752019-02-17 17:44:42 +01002778#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002779/*
2780 * "debugbreak()" function
2781 */
2782 static void
2783f_debugbreak(typval_T *argvars, typval_T *rettv)
2784{
2785 int pid;
2786
2787 rettv->vval.v_number = FAIL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002788 pid = (int)tv_get_number(&argvars[0]);
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002789 if (pid == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002790 emsg(_(e_invarg));
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002791 else
2792 {
2793 HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, 0, pid);
2794
2795 if (hProcess != NULL)
2796 {
2797 DebugBreakProcess(hProcess);
2798 CloseHandle(hProcess);
2799 rettv->vval.v_number = OK;
2800 }
2801 }
2802}
2803#endif
2804
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002805/*
2806 * "deepcopy()" function
2807 */
2808 static void
2809f_deepcopy(typval_T *argvars, typval_T *rettv)
2810{
2811 int noref = 0;
2812 int copyID;
2813
2814 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar44b4a242020-09-05 17:18:28 +02002815 noref = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002816 if (noref < 0 || noref > 1)
Bram Moolenaarbade44e2020-09-26 22:39:24 +02002817 semsg(_(e_using_number_as_bool_nr), noref);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002818 else
2819 {
2820 copyID = get_copyID();
2821 item_copy(&argvars[0], rettv, TRUE, noref == 0 ? copyID : 0);
2822 }
2823}
2824
2825/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002826 * "did_filetype()" function
2827 */
2828 static void
2829f_did_filetype(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2830{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002831 rettv->vval.v_number = did_filetype;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002832}
2833
2834/*
Bram Moolenaar4132eb52020-02-14 16:53:00 +01002835 * "echoraw({expr})" function
2836 */
2837 static void
2838f_echoraw(typval_T *argvars, typval_T *rettv UNUSED)
2839{
2840 char_u *str = tv_get_string_chk(&argvars[0]);
2841
2842 if (str != NULL && *str != NUL)
2843 {
2844 out_str(str);
2845 out_flush();
2846 }
2847}
2848
2849/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002850 * "empty({expr})" function
2851 */
2852 static void
2853f_empty(typval_T *argvars, typval_T *rettv)
2854{
2855 int n = FALSE;
2856
2857 switch (argvars[0].v_type)
2858 {
2859 case VAR_STRING:
2860 case VAR_FUNC:
2861 n = argvars[0].vval.v_string == NULL
2862 || *argvars[0].vval.v_string == NUL;
2863 break;
2864 case VAR_PARTIAL:
2865 n = FALSE;
2866 break;
2867 case VAR_NUMBER:
2868 n = argvars[0].vval.v_number == 0;
2869 break;
2870 case VAR_FLOAT:
2871#ifdef FEAT_FLOAT
2872 n = argvars[0].vval.v_float == 0.0;
2873 break;
2874#endif
2875 case VAR_LIST:
2876 n = argvars[0].vval.v_list == NULL
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002877 || argvars[0].vval.v_list->lv_len == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002878 break;
2879 case VAR_DICT:
2880 n = argvars[0].vval.v_dict == NULL
2881 || argvars[0].vval.v_dict->dv_hashtab.ht_used == 0;
2882 break;
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002883 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002884 case VAR_SPECIAL:
2885 n = argvars[0].vval.v_number != VVAL_TRUE;
2886 break;
2887
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002888 case VAR_BLOB:
2889 n = argvars[0].vval.v_blob == NULL
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002890 || argvars[0].vval.v_blob->bv_ga.ga_len == 0;
2891 break;
2892
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002893 case VAR_JOB:
2894#ifdef FEAT_JOB_CHANNEL
2895 n = argvars[0].vval.v_job == NULL
2896 || argvars[0].vval.v_job->jv_status != JOB_STARTED;
2897 break;
2898#endif
2899 case VAR_CHANNEL:
2900#ifdef FEAT_JOB_CHANNEL
2901 n = argvars[0].vval.v_channel == NULL
2902 || !channel_is_open(argvars[0].vval.v_channel);
2903 break;
2904#endif
2905 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02002906 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002907 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01002908 internal_error_no_abort("f_empty(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002909 n = TRUE;
2910 break;
2911 }
2912
2913 rettv->vval.v_number = n;
2914}
2915
2916/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02002917 * "environ()" function
2918 */
2919 static void
2920f_environ(typval_T *argvars UNUSED, typval_T *rettv)
2921{
2922#if !defined(AMIGA)
2923 int i = 0;
2924 char_u *entry, *value;
2925# ifdef MSWIN
2926 extern wchar_t **_wenviron;
2927# else
2928 extern char **environ;
2929# endif
2930
2931 if (rettv_dict_alloc(rettv) != OK)
2932 return;
2933
2934# ifdef MSWIN
2935 if (*_wenviron == NULL)
2936 return;
2937# else
2938 if (*environ == NULL)
2939 return;
2940# endif
2941
2942 for (i = 0; ; ++i)
2943 {
2944# ifdef MSWIN
2945 short_u *p;
2946
2947 if ((p = (short_u *)_wenviron[i]) == NULL)
2948 return;
2949 entry = utf16_to_enc(p, NULL);
2950# else
2951 if ((entry = (char_u *)environ[i]) == NULL)
2952 return;
2953 entry = vim_strsave(entry);
2954# endif
2955 if (entry == NULL) // out of memory
2956 return;
2957 if ((value = vim_strchr(entry, '=')) == NULL)
2958 {
2959 vim_free(entry);
2960 continue;
2961 }
2962 *value++ = NUL;
2963 dict_add_string(rettv->vval.v_dict, (char *)entry, value);
2964 vim_free(entry);
2965 }
2966#endif
2967}
2968
2969/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002970 * "escape({string}, {chars})" function
2971 */
2972 static void
2973f_escape(typval_T *argvars, typval_T *rettv)
2974{
2975 char_u buf[NUMBUFLEN];
2976
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002977 rettv->vval.v_string = vim_strsave_escaped(tv_get_string(&argvars[0]),
2978 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002979 rettv->v_type = VAR_STRING;
2980}
2981
2982/*
2983 * "eval()" function
2984 */
2985 static void
2986f_eval(typval_T *argvars, typval_T *rettv)
2987{
2988 char_u *s, *p;
2989
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002990 s = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002991 if (s != NULL)
2992 s = skipwhite(s);
2993
2994 p = s;
Bram Moolenaar5409f5d2020-06-24 18:37:35 +02002995 if (s == NULL || eval1(&s, rettv, &EVALARG_EVALUATE) == FAIL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002996 {
2997 if (p != NULL && !aborting())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002998 semsg(_(e_invexpr2), p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002999 need_clr_eos = FALSE;
3000 rettv->v_type = VAR_NUMBER;
3001 rettv->vval.v_number = 0;
3002 }
3003 else if (*s != NUL)
Bram Moolenaar2d06bfd2020-07-23 17:16:18 +02003004 semsg(_(e_trailing_arg), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003005}
3006
3007/*
3008 * "eventhandler()" function
3009 */
3010 static void
3011f_eventhandler(typval_T *argvars UNUSED, typval_T *rettv)
3012{
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003013 rettv->vval.v_number = vgetc_busy || input_busy;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003014}
3015
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003016static garray_T redir_execute_ga;
3017
3018/*
3019 * Append "value[value_len]" to the execute() output.
3020 */
3021 void
3022execute_redir_str(char_u *value, int value_len)
3023{
3024 int len;
3025
3026 if (value_len == -1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003027 len = (int)STRLEN(value); // Append the entire string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003028 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003029 len = value_len; // Append only "value_len" characters
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003030 if (ga_grow(&redir_execute_ga, len) == OK)
3031 {
3032 mch_memmove((char *)redir_execute_ga.ga_data
3033 + redir_execute_ga.ga_len, value, len);
3034 redir_execute_ga.ga_len += len;
3035 }
3036}
3037
3038/*
3039 * Get next line from a list.
3040 * Called by do_cmdline() to get the next line.
3041 * Returns allocated string, or NULL for end of function.
3042 */
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003043 static char_u *
3044get_list_line(
3045 int c UNUSED,
3046 void *cookie,
Bram Moolenaare96a2492019-06-25 04:12:16 +02003047 int indent UNUSED,
Bram Moolenaar66250c92020-08-20 15:02:42 +02003048 getline_opt_T options UNUSED)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003049{
3050 listitem_T **p = (listitem_T **)cookie;
3051 listitem_T *item = *p;
3052 char_u buf[NUMBUFLEN];
3053 char_u *s;
3054
3055 if (item == NULL)
3056 return NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003057 s = tv_get_string_buf_chk(&item->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003058 *p = item->li_next;
3059 return s == NULL ? NULL : vim_strsave(s);
3060}
3061
3062/*
3063 * "execute()" function
3064 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02003065 void
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003066execute_common(typval_T *argvars, typval_T *rettv, int arg_off)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003067{
3068 char_u *cmd = NULL;
3069 list_T *list = NULL;
3070 int save_msg_silent = msg_silent;
3071 int save_emsg_silent = emsg_silent;
3072 int save_emsg_noredir = emsg_noredir;
3073 int save_redir_execute = redir_execute;
Bram Moolenaar20951482017-12-25 13:44:43 +01003074 int save_redir_off = redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003075 garray_T save_ga;
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01003076 int save_msg_col = msg_col;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003077 int echo_output = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003078
3079 rettv->vval.v_string = NULL;
3080 rettv->v_type = VAR_STRING;
3081
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003082 if (argvars[arg_off].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003083 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003084 list = argvars[arg_off].vval.v_list;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003085 if (list == NULL || list->lv_len == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003086 // empty list, no commands, empty output
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003087 return;
3088 ++list->lv_refcount;
3089 }
Bram Moolenaare2a8f072020-01-08 19:32:18 +01003090 else if (argvars[arg_off].v_type == VAR_JOB
3091 || argvars[arg_off].v_type == VAR_CHANNEL)
3092 {
3093 emsg(_(e_inval_string));
3094 return;
3095 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003096 else
3097 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003098 cmd = tv_get_string_chk(&argvars[arg_off]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003099 if (cmd == NULL)
3100 return;
3101 }
3102
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003103 if (argvars[arg_off + 1].v_type != VAR_UNKNOWN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003104 {
3105 char_u buf[NUMBUFLEN];
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003106 char_u *s = tv_get_string_buf_chk(&argvars[arg_off + 1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003107
3108 if (s == NULL)
3109 return;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003110 if (*s == NUL)
3111 echo_output = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003112 if (STRNCMP(s, "silent", 6) == 0)
3113 ++msg_silent;
3114 if (STRCMP(s, "silent!") == 0)
3115 {
3116 emsg_silent = TRUE;
3117 emsg_noredir = TRUE;
3118 }
3119 }
3120 else
3121 ++msg_silent;
3122
3123 if (redir_execute)
3124 save_ga = redir_execute_ga;
3125 ga_init2(&redir_execute_ga, (int)sizeof(char), 500);
3126 redir_execute = TRUE;
Bram Moolenaar20951482017-12-25 13:44:43 +01003127 redir_off = FALSE;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003128 if (!echo_output)
3129 msg_col = 0; // prevent leading spaces
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003130
3131 if (cmd != NULL)
3132 do_cmdline_cmd(cmd);
3133 else
3134 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003135 listitem_T *item;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003136
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003137 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003138 item = list->lv_first;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003139 do_cmdline(NULL, get_list_line, (void *)&item,
3140 DOCMD_NOWAIT|DOCMD_VERBOSE|DOCMD_REPEAT|DOCMD_KEYTYPED);
3141 --list->lv_refcount;
3142 }
3143
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003144 // Need to append a NUL to the result.
Bram Moolenaard297f352017-01-29 20:31:21 +01003145 if (ga_grow(&redir_execute_ga, 1) == OK)
3146 {
3147 ((char *)redir_execute_ga.ga_data)[redir_execute_ga.ga_len] = NUL;
3148 rettv->vval.v_string = redir_execute_ga.ga_data;
3149 }
3150 else
3151 {
3152 ga_clear(&redir_execute_ga);
3153 rettv->vval.v_string = NULL;
3154 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003155 msg_silent = save_msg_silent;
3156 emsg_silent = save_emsg_silent;
3157 emsg_noredir = save_emsg_noredir;
3158
3159 redir_execute = save_redir_execute;
3160 if (redir_execute)
3161 redir_execute_ga = save_ga;
Bram Moolenaar20951482017-12-25 13:44:43 +01003162 redir_off = save_redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003163
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01003164 // "silent reg" or "silent echo x" leaves msg_col somewhere in the line.
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003165 if (echo_output)
3166 // When not working silently: put it in column zero. A following
3167 // "echon" will overwrite the message, unavoidably.
3168 msg_col = 0;
3169 else
3170 // When working silently: Put it back where it was, since nothing
3171 // should have been written.
3172 msg_col = save_msg_col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003173}
3174
3175/*
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003176 * "execute()" function
3177 */
3178 static void
3179f_execute(typval_T *argvars, typval_T *rettv)
3180{
3181 execute_common(argvars, rettv, 0);
3182}
3183
3184/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003185 * "exists()" function
3186 */
3187 static void
3188f_exists(typval_T *argvars, typval_T *rettv)
3189{
3190 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003191 int n = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003192
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003193 p = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003194 if (*p == '$') // environment variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003195 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003196 // first try "normal" environment variables (fast)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003197 if (mch_getenv(p + 1) != NULL)
3198 n = TRUE;
3199 else
3200 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003201 // try expanding things like $VIM and ${HOME}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003202 p = expand_env_save(p);
3203 if (p != NULL && *p != '$')
3204 n = TRUE;
3205 vim_free(p);
3206 }
3207 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003208 else if (*p == '&' || *p == '+') // option
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003209 {
Bram Moolenaar9a78e6d2020-07-01 18:29:55 +02003210 n = (eval_option(&p, NULL, TRUE) == OK);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003211 if (*skipwhite(p) != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003212 n = FALSE; // trailing garbage
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003213 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003214 else if (*p == '*') // internal or user defined function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003215 {
Bram Moolenaarb54c3ff2016-07-31 14:11:58 +02003216 n = function_exists(p + 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003217 }
Bram Moolenaar15c47602020-03-26 22:16:48 +01003218 else if (*p == '?') // internal function only
3219 {
3220 n = has_internal_func_name(p + 1);
3221 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003222 else if (*p == ':')
3223 {
3224 n = cmd_exists(p + 1);
3225 }
3226 else if (*p == '#')
3227 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003228 if (p[1] == '#')
3229 n = autocmd_supported(p + 2);
3230 else
3231 n = au_exists(p + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003232 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003233 else // internal variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003234 {
Bram Moolenaarc6f9f732018-02-11 19:06:26 +01003235 n = var_exists(p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003236 }
3237
3238 rettv->vval.v_number = n;
3239}
3240
3241#ifdef FEAT_FLOAT
3242/*
3243 * "exp()" function
3244 */
3245 static void
3246f_exp(typval_T *argvars, typval_T *rettv)
3247{
3248 float_T f = 0.0;
3249
3250 rettv->v_type = VAR_FLOAT;
3251 if (get_float_arg(argvars, &f) == OK)
3252 rettv->vval.v_float = exp(f);
3253 else
3254 rettv->vval.v_float = 0.0;
3255}
3256#endif
3257
3258/*
3259 * "expand()" function
3260 */
3261 static void
3262f_expand(typval_T *argvars, typval_T *rettv)
3263{
3264 char_u *s;
3265 int len;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003266 char *errormsg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003267 int options = WILD_SILENT|WILD_USE_NL|WILD_LIST_NOTFOUND;
3268 expand_T xpc;
3269 int error = FALSE;
3270 char_u *result;
Bram Moolenaar8f187fc2020-09-26 18:47:11 +02003271#ifdef BACKSLASH_IN_FILENAME
3272 char_u *p_csl_save = p_csl;
3273
3274 // avoid using 'completeslash' here
3275 p_csl = empty_option;
3276#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003277
3278 rettv->v_type = VAR_STRING;
3279 if (argvars[1].v_type != VAR_UNKNOWN
3280 && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaar551d25e2020-09-02 21:37:56 +02003281 && tv_get_bool_chk(&argvars[2], &error)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003282 && !error)
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02003283 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003284
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003285 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003286 if (*s == '%' || *s == '#' || *s == '<')
3287 {
3288 ++emsg_off;
3289 result = eval_vars(s, s, &len, NULL, &errormsg, NULL);
3290 --emsg_off;
3291 if (rettv->v_type == VAR_LIST)
3292 {
3293 if (rettv_list_alloc(rettv) != FAIL && result != NULL)
3294 list_append_string(rettv->vval.v_list, result, -1);
Bram Moolenaar86173482019-10-01 17:02:16 +02003295 vim_free(result);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003296 }
3297 else
3298 rettv->vval.v_string = result;
3299 }
3300 else
3301 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003302 // When the optional second argument is non-zero, don't remove matches
3303 // for 'wildignore' and don't put matches for 'suffixes' at the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003304 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaar551d25e2020-09-02 21:37:56 +02003305 && tv_get_bool_chk(&argvars[1], &error))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003306 options |= WILD_KEEP_ALL;
3307 if (!error)
3308 {
3309 ExpandInit(&xpc);
3310 xpc.xp_context = EXPAND_FILES;
3311 if (p_wic)
3312 options += WILD_ICASE;
3313 if (rettv->v_type == VAR_STRING)
3314 rettv->vval.v_string = ExpandOne(&xpc, s, NULL,
3315 options, WILD_ALL);
3316 else if (rettv_list_alloc(rettv) != FAIL)
3317 {
3318 int i;
3319
3320 ExpandOne(&xpc, s, NULL, options, WILD_ALL_KEEP);
3321 for (i = 0; i < xpc.xp_numfiles; i++)
3322 list_append_string(rettv->vval.v_list, xpc.xp_files[i], -1);
3323 ExpandCleanup(&xpc);
3324 }
3325 }
3326 else
3327 rettv->vval.v_string = NULL;
3328 }
Bram Moolenaar8f187fc2020-09-26 18:47:11 +02003329#ifdef BACKSLASH_IN_FILENAME
3330 p_csl = p_csl_save;
3331#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003332}
3333
3334/*
Bram Moolenaar80dad482019-06-09 17:22:31 +02003335 * "expandcmd()" function
3336 * Expand all the special characters in a command string.
3337 */
3338 static void
3339f_expandcmd(typval_T *argvars, typval_T *rettv)
3340{
3341 exarg_T eap;
3342 char_u *cmdstr;
3343 char *errormsg = NULL;
3344
3345 rettv->v_type = VAR_STRING;
3346 cmdstr = vim_strsave(tv_get_string(&argvars[0]));
3347
3348 memset(&eap, 0, sizeof(eap));
3349 eap.cmd = cmdstr;
3350 eap.arg = cmdstr;
Bram Moolenaar8071cb22019-07-12 17:58:01 +02003351 eap.argt |= EX_NOSPC;
Bram Moolenaar80dad482019-06-09 17:22:31 +02003352 eap.usefilter = FALSE;
3353 eap.nextcmd = NULL;
3354 eap.cmdidx = CMD_USER;
3355
3356 expand_filename(&eap, &cmdstr, &errormsg);
3357 if (errormsg != NULL && *errormsg != NUL)
3358 emsg(errormsg);
3359
3360 rettv->vval.v_string = cmdstr;
3361}
3362
3363/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003364 * "feedkeys()" function
3365 */
3366 static void
3367f_feedkeys(typval_T *argvars, typval_T *rettv UNUSED)
3368{
3369 int remap = TRUE;
3370 int insert = FALSE;
3371 char_u *keys, *flags;
3372 char_u nbuf[NUMBUFLEN];
3373 int typed = FALSE;
3374 int execute = FALSE;
3375 int dangerous = FALSE;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003376 int lowlevel = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003377 char_u *keys_esc;
3378
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003379 // This is not allowed in the sandbox. If the commands would still be
3380 // executed in the sandbox it would be OK, but it probably happens later,
3381 // when "sandbox" is no longer set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003382 if (check_secure())
3383 return;
3384
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003385 keys = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003386
3387 if (argvars[1].v_type != VAR_UNKNOWN)
3388 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003389 flags = tv_get_string_buf(&argvars[1], nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003390 for ( ; *flags != NUL; ++flags)
3391 {
3392 switch (*flags)
3393 {
3394 case 'n': remap = FALSE; break;
3395 case 'm': remap = TRUE; break;
3396 case 't': typed = TRUE; break;
3397 case 'i': insert = TRUE; break;
3398 case 'x': execute = TRUE; break;
3399 case '!': dangerous = TRUE; break;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003400 case 'L': lowlevel = TRUE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003401 }
3402 }
3403 }
3404
3405 if (*keys != NUL || execute)
3406 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003407 // Need to escape K_SPECIAL and CSI before putting the string in the
3408 // typeahead buffer.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003409 keys_esc = vim_strsave_escape_csi(keys);
3410 if (keys_esc != NULL)
3411 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003412 if (lowlevel)
3413 {
3414#ifdef USE_INPUT_BUF
Bram Moolenaar9645e2d2020-03-20 20:48:49 +01003415 int idx;
3416 int len = (int)STRLEN(keys);
3417
3418 for (idx = 0; idx < len; ++idx)
3419 {
3420 // if a CTRL-C was typed, set got_int, similar to what
3421 // happens in fill_input_buf()
3422 if (keys[idx] == 3 && ctrl_c_interrupts && typed)
3423 got_int = TRUE;
3424 add_to_input_buf(keys + idx, 1);
3425 }
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003426#else
3427 emsg(_("E980: lowlevel input not supported"));
3428#endif
3429 }
3430 else
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003431 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003432 ins_typebuf(keys_esc, (remap ? REMAP_YES : REMAP_NONE),
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003433 insert ? 0 : typebuf.tb_len, !typed, FALSE);
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003434 if (vgetc_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02003435#ifdef FEAT_TIMERS
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003436 || timer_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02003437#endif
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003438 || input_busy)
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003439 typebuf_was_filled = TRUE;
3440 }
3441 vim_free(keys_esc);
3442
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003443 if (execute)
3444 {
3445 int save_msg_scroll = msg_scroll;
3446
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003447 // Avoid a 1 second delay when the keys start Insert mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003448 msg_scroll = FALSE;
3449
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003450 if (!dangerous)
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003451 {
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003452 ++ex_normal_busy;
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003453 ++in_feedkeys;
3454 }
Bram Moolenaar905dd902019-04-07 14:21:47 +02003455 exec_normal(TRUE, lowlevel, TRUE);
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003456 if (!dangerous)
Bram Moolenaar189832b2020-09-23 12:29:11 +02003457 {
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003458 --ex_normal_busy;
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003459 --in_feedkeys;
Bram Moolenaar189832b2020-09-23 12:29:11 +02003460 }
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003461
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003462 msg_scroll |= save_msg_scroll;
3463 }
3464 }
3465 }
3466}
3467
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003468#ifdef FEAT_FLOAT
3469/*
3470 * "float2nr({float})" function
3471 */
3472 static void
3473f_float2nr(typval_T *argvars, typval_T *rettv)
3474{
3475 float_T f = 0.0;
3476
3477 if (get_float_arg(argvars, &f) == OK)
3478 {
Bram Moolenaar37184272020-05-23 19:30:05 +02003479 if (f <= (float_T)-VARNUM_MAX + DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01003480 rettv->vval.v_number = -VARNUM_MAX;
Bram Moolenaar37184272020-05-23 19:30:05 +02003481 else if (f >= (float_T)VARNUM_MAX - DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01003482 rettv->vval.v_number = VARNUM_MAX;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003483 else
3484 rettv->vval.v_number = (varnumber_T)f;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003485 }
3486}
3487
3488/*
3489 * "floor({float})" function
3490 */
3491 static void
3492f_floor(typval_T *argvars, typval_T *rettv)
3493{
3494 float_T f = 0.0;
3495
3496 rettv->v_type = VAR_FLOAT;
3497 if (get_float_arg(argvars, &f) == OK)
3498 rettv->vval.v_float = floor(f);
3499 else
3500 rettv->vval.v_float = 0.0;
3501}
3502
3503/*
3504 * "fmod()" function
3505 */
3506 static void
3507f_fmod(typval_T *argvars, typval_T *rettv)
3508{
3509 float_T fx = 0.0, fy = 0.0;
3510
3511 rettv->v_type = VAR_FLOAT;
3512 if (get_float_arg(argvars, &fx) == OK
3513 && get_float_arg(&argvars[1], &fy) == OK)
3514 rettv->vval.v_float = fmod(fx, fy);
3515 else
3516 rettv->vval.v_float = 0.0;
3517}
3518#endif
3519
3520/*
3521 * "fnameescape({string})" function
3522 */
3523 static void
3524f_fnameescape(typval_T *argvars, typval_T *rettv)
3525{
3526 rettv->vval.v_string = vim_strsave_fnameescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003527 tv_get_string(&argvars[0]), FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003528 rettv->v_type = VAR_STRING;
3529}
3530
3531/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003532 * "foreground()" function
3533 */
3534 static void
3535f_foreground(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3536{
3537#ifdef FEAT_GUI
3538 if (gui.in_use)
Bram Moolenaarafde13b2019-04-28 19:46:49 +02003539 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003540 gui_mch_set_foreground();
Bram Moolenaarafde13b2019-04-28 19:46:49 +02003541 return;
3542 }
3543#endif
3544#if defined(MSWIN) && (!defined(FEAT_GUI) || defined(VIMDLL))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003545 win32_set_foreground();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003546#endif
3547}
3548
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003549 static void
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003550common_function(typval_T *argvars, typval_T *rettv, int is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003551{
3552 char_u *s;
3553 char_u *name;
3554 int use_string = FALSE;
3555 partial_T *arg_pt = NULL;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003556 char_u *trans_name = NULL;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003557 int is_global = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003558
3559 if (argvars[0].v_type == VAR_FUNC)
3560 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003561 // function(MyFunc, [arg], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003562 s = argvars[0].vval.v_string;
3563 }
3564 else if (argvars[0].v_type == VAR_PARTIAL
3565 && argvars[0].vval.v_partial != NULL)
3566 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003567 // function(dict.MyFunc, [arg])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003568 arg_pt = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003569 s = partial_name(arg_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003570 }
3571 else
3572 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003573 // function('MyFunc', [arg], dict)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003574 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003575 use_string = TRUE;
3576 }
3577
Bram Moolenaar843b8842016-08-21 14:36:15 +02003578 if ((use_string && vim_strchr(s, AUTOLOAD_CHAR) == NULL) || is_funcref)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003579 {
3580 name = s;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003581 trans_name = trans_function_name(&name, &is_global, FALSE,
Bram Moolenaar32b3f822021-01-06 21:59:39 +01003582 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF,
3583 NULL, NULL, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003584 if (*name != NUL)
3585 s = NULL;
3586 }
3587
Bram Moolenaar843b8842016-08-21 14:36:15 +02003588 if (s == NULL || *s == NUL || (use_string && VIM_ISDIGIT(*s))
3589 || (is_funcref && trans_name == NULL))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003590 semsg(_(e_invarg2), use_string ? tv_get_string(&argvars[0]) : s);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003591 // Don't check an autoload name for existence here.
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003592 else if (trans_name != NULL && (is_funcref
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003593 ? find_func(trans_name, is_global, NULL) == NULL
3594 : !translated_function_exists(trans_name, is_global)))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003595 semsg(_("E700: Unknown function: %s"), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003596 else
3597 {
3598 int dict_idx = 0;
3599 int arg_idx = 0;
3600 list_T *list = NULL;
3601
3602 if (STRNCMP(s, "s:", 2) == 0 || STRNCMP(s, "<SID>", 5) == 0)
3603 {
3604 char sid_buf[25];
3605 int off = *s == 's' ? 2 : 5;
3606
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003607 // Expand s: and <SID> into <SNR>nr_, so that the function can
3608 // also be called from another script. Using trans_function_name()
3609 // would also work, but some plugins depend on the name being
3610 // printable text.
Bram Moolenaarf29c1c62018-09-10 21:05:02 +02003611 sprintf(sid_buf, "<SNR>%ld_", (long)current_sctx.sc_sid);
Bram Moolenaar51e14382019-05-25 20:21:28 +02003612 name = alloc(STRLEN(sid_buf) + STRLEN(s + off) + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003613 if (name != NULL)
3614 {
3615 STRCPY(name, sid_buf);
3616 STRCAT(name, s + off);
3617 }
3618 }
3619 else
3620 name = vim_strsave(s);
3621
3622 if (argvars[1].v_type != VAR_UNKNOWN)
3623 {
3624 if (argvars[2].v_type != VAR_UNKNOWN)
3625 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003626 // function(name, [args], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003627 arg_idx = 1;
3628 dict_idx = 2;
3629 }
3630 else if (argvars[1].v_type == VAR_DICT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003631 // function(name, dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003632 dict_idx = 1;
3633 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003634 // function(name, [args])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003635 arg_idx = 1;
3636 if (dict_idx > 0)
3637 {
3638 if (argvars[dict_idx].v_type != VAR_DICT)
3639 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003640 emsg(_("E922: expected a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003641 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003642 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003643 }
3644 if (argvars[dict_idx].vval.v_dict == NULL)
3645 dict_idx = 0;
3646 }
3647 if (arg_idx > 0)
3648 {
3649 if (argvars[arg_idx].v_type != VAR_LIST)
3650 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003651 emsg(_("E923: Second argument of function() must be a list or a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003652 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003653 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003654 }
3655 list = argvars[arg_idx].vval.v_list;
3656 if (list == NULL || list->lv_len == 0)
3657 arg_idx = 0;
Bram Moolenaar4c054e92019-11-10 00:13:50 +01003658 else if (list->lv_len > MAX_FUNC_ARGS)
3659 {
Bram Moolenaar2118a302019-11-22 19:29:45 +01003660 emsg_funcname((char *)e_toomanyarg, s);
Bram Moolenaar4c054e92019-11-10 00:13:50 +01003661 vim_free(name);
3662 goto theend;
3663 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003664 }
3665 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003666 if (dict_idx > 0 || arg_idx > 0 || arg_pt != NULL || is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003667 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003668 partial_T *pt = ALLOC_CLEAR_ONE(partial_T);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003669
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003670 // result is a VAR_PARTIAL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003671 if (pt == NULL)
3672 vim_free(name);
3673 else
3674 {
3675 if (arg_idx > 0 || (arg_pt != NULL && arg_pt->pt_argc > 0))
3676 {
3677 listitem_T *li;
3678 int i = 0;
3679 int arg_len = 0;
3680 int lv_len = 0;
3681
3682 if (arg_pt != NULL)
3683 arg_len = arg_pt->pt_argc;
3684 if (list != NULL)
3685 lv_len = list->lv_len;
3686 pt->pt_argc = arg_len + lv_len;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003687 pt->pt_argv = ALLOC_MULT(typval_T, pt->pt_argc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003688 if (pt->pt_argv == NULL)
3689 {
3690 vim_free(pt);
3691 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003692 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003693 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003694 for (i = 0; i < arg_len; i++)
3695 copy_tv(&arg_pt->pt_argv[i], &pt->pt_argv[i]);
3696 if (lv_len > 0)
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003697 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003698 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003699 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003700 copy_tv(&li->li_tv, &pt->pt_argv[i++]);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003701 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003702 }
3703
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003704 // For "function(dict.func, [], dict)" and "func" is a partial
3705 // use "dict". That is backwards compatible.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003706 if (dict_idx > 0)
3707 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003708 // The dict is bound explicitly, pt_auto is FALSE.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003709 pt->pt_dict = argvars[dict_idx].vval.v_dict;
3710 ++pt->pt_dict->dv_refcount;
3711 }
3712 else if (arg_pt != NULL)
3713 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003714 // If the dict was bound automatically the result is also
3715 // bound automatically.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003716 pt->pt_dict = arg_pt->pt_dict;
3717 pt->pt_auto = arg_pt->pt_auto;
3718 if (pt->pt_dict != NULL)
3719 ++pt->pt_dict->dv_refcount;
3720 }
3721
3722 pt->pt_refcount = 1;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003723 if (arg_pt != NULL && arg_pt->pt_func != NULL)
3724 {
3725 pt->pt_func = arg_pt->pt_func;
3726 func_ptr_ref(pt->pt_func);
3727 vim_free(name);
3728 }
3729 else if (is_funcref)
3730 {
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003731 pt->pt_func = find_func(trans_name, is_global, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003732 func_ptr_ref(pt->pt_func);
3733 vim_free(name);
3734 }
3735 else
3736 {
3737 pt->pt_name = name;
3738 func_ref(name);
3739 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003740 }
3741 rettv->v_type = VAR_PARTIAL;
3742 rettv->vval.v_partial = pt;
3743 }
3744 else
3745 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003746 // result is a VAR_FUNC
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003747 rettv->v_type = VAR_FUNC;
3748 rettv->vval.v_string = name;
3749 func_ref(name);
3750 }
3751 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003752theend:
3753 vim_free(trans_name);
3754}
3755
3756/*
3757 * "funcref()" function
3758 */
3759 static void
3760f_funcref(typval_T *argvars, typval_T *rettv)
3761{
3762 common_function(argvars, rettv, TRUE);
3763}
3764
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003765 static type_T *
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003766ret_f_function(int argcount, type_T **argtypes)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003767{
3768 if (argcount == 1 && argtypes[0]->tt_type == VAR_STRING)
3769 return &t_func_any;
Bram Moolenaar5e654232020-09-16 15:22:00 +02003770 return &t_func_unknown;
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003771}
3772
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003773/*
3774 * "function()" function
3775 */
3776 static void
3777f_function(typval_T *argvars, typval_T *rettv)
3778{
3779 common_function(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003780}
3781
3782/*
3783 * "garbagecollect()" function
3784 */
3785 static void
3786f_garbagecollect(typval_T *argvars, typval_T *rettv UNUSED)
3787{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003788 // This is postponed until we are back at the toplevel, because we may be
3789 // using Lists and Dicts internally. E.g.: ":echo [garbagecollect()]".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003790 want_garbage_collect = TRUE;
3791
Bram Moolenaar2df47312020-09-05 17:30:44 +02003792 if (argvars[0].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[0]) == 1)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003793 garbage_collect_at_exit = TRUE;
3794}
3795
3796/*
3797 * "get()" function
3798 */
3799 static void
3800f_get(typval_T *argvars, typval_T *rettv)
3801{
3802 listitem_T *li;
3803 list_T *l;
3804 dictitem_T *di;
3805 dict_T *d;
3806 typval_T *tv = NULL;
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003807 int what_is_dict = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003808
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003809 if (argvars[0].v_type == VAR_BLOB)
3810 {
3811 int error = FALSE;
3812 int idx = tv_get_number_chk(&argvars[1], &error);
3813
3814 if (!error)
3815 {
3816 rettv->v_type = VAR_NUMBER;
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003817 if (idx < 0)
3818 idx = blob_len(argvars[0].vval.v_blob) + idx;
3819 if (idx < 0 || idx >= blob_len(argvars[0].vval.v_blob))
3820 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003821 else
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003822 {
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003823 rettv->vval.v_number = blob_get(argvars[0].vval.v_blob, idx);
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003824 tv = rettv;
3825 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003826 }
3827 }
3828 else if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003829 {
3830 if ((l = argvars[0].vval.v_list) != NULL)
3831 {
3832 int error = FALSE;
3833
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003834 li = list_find(l, (long)tv_get_number_chk(&argvars[1], &error));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003835 if (!error && li != NULL)
3836 tv = &li->li_tv;
3837 }
3838 }
3839 else if (argvars[0].v_type == VAR_DICT)
3840 {
3841 if ((d = argvars[0].vval.v_dict) != NULL)
3842 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003843 di = dict_find(d, tv_get_string(&argvars[1]), -1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003844 if (di != NULL)
3845 tv = &di->di_tv;
3846 }
3847 }
3848 else if (argvars[0].v_type == VAR_PARTIAL || argvars[0].v_type == VAR_FUNC)
3849 {
3850 partial_T *pt;
3851 partial_T fref_pt;
3852
3853 if (argvars[0].v_type == VAR_PARTIAL)
3854 pt = argvars[0].vval.v_partial;
3855 else
3856 {
Bram Moolenaara80faa82020-04-12 19:37:17 +02003857 CLEAR_FIELD(fref_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003858 fref_pt.pt_name = argvars[0].vval.v_string;
3859 pt = &fref_pt;
3860 }
3861
3862 if (pt != NULL)
3863 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003864 char_u *what = tv_get_string(&argvars[1]);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003865 char_u *n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003866
3867 if (STRCMP(what, "func") == 0 || STRCMP(what, "name") == 0)
3868 {
3869 rettv->v_type = (*what == 'f' ? VAR_FUNC : VAR_STRING);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003870 n = partial_name(pt);
3871 if (n == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003872 rettv->vval.v_string = NULL;
3873 else
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003874 {
3875 rettv->vval.v_string = vim_strsave(n);
3876 if (rettv->v_type == VAR_FUNC)
3877 func_ref(rettv->vval.v_string);
3878 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003879 }
3880 else if (STRCMP(what, "dict") == 0)
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003881 {
3882 what_is_dict = TRUE;
3883 if (pt->pt_dict != NULL)
3884 rettv_dict_set(rettv, pt->pt_dict);
3885 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003886 else if (STRCMP(what, "args") == 0)
3887 {
3888 rettv->v_type = VAR_LIST;
3889 if (rettv_list_alloc(rettv) == OK)
3890 {
3891 int i;
3892
3893 for (i = 0; i < pt->pt_argc; ++i)
3894 list_append_tv(rettv->vval.v_list, &pt->pt_argv[i]);
3895 }
3896 }
3897 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003898 semsg(_(e_invarg2), what);
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003899
3900 // When {what} == "dict" and pt->pt_dict == NULL, evaluate the
3901 // third argument
3902 if (!what_is_dict)
3903 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003904 }
3905 }
3906 else
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01003907 semsg(_(e_listdictblobarg), "get()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003908
3909 if (tv == NULL)
3910 {
3911 if (argvars[2].v_type != VAR_UNKNOWN)
3912 copy_tv(&argvars[2], rettv);
3913 }
3914 else
3915 copy_tv(tv, rettv);
3916}
3917
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003918/*
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003919 * "getchangelist()" function
3920 */
3921 static void
3922f_getchangelist(typval_T *argvars, typval_T *rettv)
3923{
3924#ifdef FEAT_JUMPLIST
3925 buf_T *buf;
3926 int i;
3927 list_T *l;
3928 dict_T *d;
3929#endif
3930
3931 if (rettv_list_alloc(rettv) != OK)
3932 return;
3933
3934#ifdef FEAT_JUMPLIST
Bram Moolenaar4c313b12019-08-24 22:58:31 +02003935 if (argvars[0].v_type == VAR_UNKNOWN)
3936 buf = curbuf;
3937 else
Bram Moolenaara5d38412020-09-02 21:02:35 +02003938 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003939 if (buf == NULL)
3940 return;
3941
3942 l = list_alloc();
3943 if (l == NULL)
3944 return;
3945
3946 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3947 return;
3948 /*
3949 * The current window change list index tracks only the position in the
3950 * current buffer change list. For other buffers, use the change list
3951 * length as the current index.
3952 */
3953 list_append_number(rettv->vval.v_list,
3954 (varnumber_T)((buf == curwin->w_buffer)
3955 ? curwin->w_changelistidx : buf->b_changelistlen));
3956
3957 for (i = 0; i < buf->b_changelistlen; ++i)
3958 {
3959 if (buf->b_changelist[i].lnum == 0)
3960 continue;
3961 if ((d = dict_alloc()) == NULL)
3962 return;
3963 if (list_append_dict(l, d) == FAIL)
3964 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003965 dict_add_number(d, "lnum", (long)buf->b_changelist[i].lnum);
3966 dict_add_number(d, "col", (long)buf->b_changelist[i].col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003967 dict_add_number(d, "coladd", (long)buf->b_changelist[i].coladd);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003968 }
3969#endif
3970}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003971
Bram Moolenaar6f02b002021-01-10 20:22:54 +01003972 static void
3973getpos_both(
3974 typval_T *argvars,
3975 typval_T *rettv,
3976 int getcurpos,
3977 int charcol)
3978{
3979 pos_T *fp = NULL;
3980 pos_T pos;
3981 win_T *wp = curwin;
3982 list_T *l;
3983 int fnum = -1;
3984
3985 if (rettv_list_alloc(rettv) == OK)
3986 {
3987 l = rettv->vval.v_list;
3988 if (getcurpos)
3989 {
3990 if (argvars[0].v_type != VAR_UNKNOWN)
3991 {
3992 wp = find_win_by_nr_or_id(&argvars[0]);
3993 if (wp != NULL)
3994 fp = &wp->w_cursor;
3995 }
3996 else
3997 fp = &curwin->w_cursor;
3998 if (fp != NULL && charcol)
3999 {
4000 pos = *fp;
4001 pos.col = buf_byteidx_to_charidx(wp->w_buffer, pos.lnum,
4002 pos.col) - 1;
4003 fp = &pos;
4004 }
4005 }
4006 else
4007 fp = var2fpos(&argvars[0], TRUE, &fnum, charcol);
4008 if (fnum != -1)
4009 list_append_number(l, (varnumber_T)fnum);
4010 else
4011 list_append_number(l, (varnumber_T)0);
4012 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->lnum
4013 : (varnumber_T)0);
4014 list_append_number(l, (fp != NULL)
4015 ? (varnumber_T)(fp->col == MAXCOL ? MAXCOL : fp->col + 1)
4016 : (varnumber_T)0);
4017 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->coladd :
4018 (varnumber_T)0);
4019 if (getcurpos)
4020 {
4021 int save_set_curswant = curwin->w_set_curswant;
4022 colnr_T save_curswant = curwin->w_curswant;
4023 colnr_T save_virtcol = curwin->w_virtcol;
4024
4025 if (wp == curwin)
4026 update_curswant();
4027 list_append_number(l, wp == NULL ? 0 : wp->w_curswant == MAXCOL
4028 ? (varnumber_T)MAXCOL : (varnumber_T)wp->w_curswant + 1);
4029
4030 // Do not change "curswant", as it is unexpected that a get
4031 // function has a side effect.
4032 if (wp == curwin && save_set_curswant)
4033 {
4034 curwin->w_set_curswant = save_set_curswant;
4035 curwin->w_curswant = save_curswant;
4036 curwin->w_virtcol = save_virtcol;
4037 curwin->w_valid &= ~VALID_VIRTCOL;
4038 }
4039 }
4040 }
4041 else
4042 rettv->vval.v_number = FALSE;
4043}
4044
4045/*
4046 * "getcharpos()" function
4047 */
4048 static void
4049f_getcharpos(typval_T *argvars UNUSED, typval_T *rettv)
4050{
4051 getpos_both(argvars, rettv, FALSE, TRUE);
4052}
4053
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004054/*
4055 * "getcharsearch()" function
4056 */
4057 static void
4058f_getcharsearch(typval_T *argvars UNUSED, typval_T *rettv)
4059{
4060 if (rettv_dict_alloc(rettv) != FAIL)
4061 {
4062 dict_T *dict = rettv->vval.v_dict;
4063
Bram Moolenaare0be1672018-07-08 16:50:37 +02004064 dict_add_string(dict, "char", last_csearch());
4065 dict_add_number(dict, "forward", last_csearch_forward());
4066 dict_add_number(dict, "until", last_csearch_until());
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004067 }
4068}
4069
4070/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02004071 * "getenv()" function
4072 */
4073 static void
4074f_getenv(typval_T *argvars, typval_T *rettv)
4075{
4076 int mustfree = FALSE;
4077 char_u *p = vim_getenv(tv_get_string(&argvars[0]), &mustfree);
4078
4079 if (p == NULL)
4080 {
4081 rettv->v_type = VAR_SPECIAL;
4082 rettv->vval.v_number = VVAL_NULL;
4083 return;
4084 }
4085 if (!mustfree)
4086 p = vim_strsave(p);
4087 rettv->vval.v_string = p;
4088 rettv->v_type = VAR_STRING;
4089}
4090
4091/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004092 * "getfontname()" function
4093 */
4094 static void
4095f_getfontname(typval_T *argvars UNUSED, typval_T *rettv)
4096{
4097 rettv->v_type = VAR_STRING;
4098 rettv->vval.v_string = NULL;
4099#ifdef FEAT_GUI
4100 if (gui.in_use)
4101 {
4102 GuiFont font;
4103 char_u *name = NULL;
4104
4105 if (argvars[0].v_type == VAR_UNKNOWN)
4106 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004107 // Get the "Normal" font. Either the name saved by
4108 // hl_set_font_name() or from the font ID.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004109 font = gui.norm_font;
4110 name = hl_get_font_name();
4111 }
4112 else
4113 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004114 name = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004115 if (STRCMP(name, "*") == 0) // don't use font dialog
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004116 return;
4117 font = gui_mch_get_font(name, FALSE);
4118 if (font == NOFONT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004119 return; // Invalid font name, return empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004120 }
4121 rettv->vval.v_string = gui_mch_get_fontname(font, name);
4122 if (argvars[0].v_type != VAR_UNKNOWN)
4123 gui_mch_free_font(font);
4124 }
4125#endif
4126}
4127
4128/*
Bram Moolenaar4f505882018-02-10 21:06:32 +01004129 * "getjumplist()" function
4130 */
4131 static void
4132f_getjumplist(typval_T *argvars, typval_T *rettv)
4133{
4134#ifdef FEAT_JUMPLIST
4135 win_T *wp;
4136 int i;
4137 list_T *l;
4138 dict_T *d;
4139#endif
4140
4141 if (rettv_list_alloc(rettv) != OK)
4142 return;
4143
4144#ifdef FEAT_JUMPLIST
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004145 wp = find_tabwin(&argvars[0], &argvars[1], NULL);
Bram Moolenaar4f505882018-02-10 21:06:32 +01004146 if (wp == NULL)
4147 return;
4148
Bram Moolenaar57ee2b62019-02-12 22:15:06 +01004149 cleanup_jumplist(wp, TRUE);
4150
Bram Moolenaar4f505882018-02-10 21:06:32 +01004151 l = list_alloc();
4152 if (l == NULL)
4153 return;
4154
4155 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4156 return;
4157 list_append_number(rettv->vval.v_list, (varnumber_T)wp->w_jumplistidx);
4158
4159 for (i = 0; i < wp->w_jumplistlen; ++i)
4160 {
Bram Moolenaara7e18d22018-02-11 14:29:49 +01004161 if (wp->w_jumplist[i].fmark.mark.lnum == 0)
4162 continue;
Bram Moolenaar4f505882018-02-10 21:06:32 +01004163 if ((d = dict_alloc()) == NULL)
4164 return;
4165 if (list_append_dict(l, d) == FAIL)
4166 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02004167 dict_add_number(d, "lnum", (long)wp->w_jumplist[i].fmark.mark.lnum);
4168 dict_add_number(d, "col", (long)wp->w_jumplist[i].fmark.mark.col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02004169 dict_add_number(d, "coladd", (long)wp->w_jumplist[i].fmark.mark.coladd);
Bram Moolenaare0be1672018-07-08 16:50:37 +02004170 dict_add_number(d, "bufnr", (long)wp->w_jumplist[i].fmark.fnum);
Bram Moolenaara7e18d22018-02-11 14:29:49 +01004171 if (wp->w_jumplist[i].fname != NULL)
Bram Moolenaare0be1672018-07-08 16:50:37 +02004172 dict_add_string(d, "filename", wp->w_jumplist[i].fname);
Bram Moolenaar4f505882018-02-10 21:06:32 +01004173 }
4174#endif
4175}
4176
4177/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004178 * "getpid()" function
4179 */
4180 static void
4181f_getpid(typval_T *argvars UNUSED, typval_T *rettv)
4182{
4183 rettv->vval.v_number = mch_get_pid();
4184}
4185
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004186/*
4187 * "getcurpos()" function
4188 */
4189 static void
4190f_getcurpos(typval_T *argvars, typval_T *rettv)
4191{
Bram Moolenaar6f02b002021-01-10 20:22:54 +01004192 getpos_both(argvars, rettv, TRUE, FALSE);
4193}
4194
4195 static void
4196f_getcursorcharpos(typval_T *argvars, typval_T *rettv)
4197{
4198 getpos_both(argvars, rettv, TRUE, TRUE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004199}
4200
4201/*
4202 * "getpos(string)" function
4203 */
4204 static void
4205f_getpos(typval_T *argvars, typval_T *rettv)
4206{
Bram Moolenaar6f02b002021-01-10 20:22:54 +01004207 getpos_both(argvars, rettv, FALSE, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004208}
4209
4210/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004211 * "getreg()" function
4212 */
4213 static void
4214f_getreg(typval_T *argvars, typval_T *rettv)
4215{
4216 char_u *strregname;
4217 int regname;
4218 int arg2 = FALSE;
4219 int return_list = FALSE;
4220 int error = FALSE;
4221
4222 if (argvars[0].v_type != VAR_UNKNOWN)
4223 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004224 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004225 error = strregname == NULL;
4226 if (argvars[1].v_type != VAR_UNKNOWN)
4227 {
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02004228 arg2 = (int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004229 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02004230 return_list = (int)tv_get_bool_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004231 }
4232 }
4233 else
4234 strregname = get_vim_var_str(VV_REG);
4235
4236 if (error)
4237 return;
4238
4239 regname = (strregname == NULL ? '"' : *strregname);
4240 if (regname == 0)
4241 regname = '"';
4242
4243 if (return_list)
4244 {
4245 rettv->v_type = VAR_LIST;
4246 rettv->vval.v_list = (list_T *)get_reg_contents(regname,
4247 (arg2 ? GREG_EXPR_SRC : 0) | GREG_LIST);
4248 if (rettv->vval.v_list == NULL)
4249 (void)rettv_list_alloc(rettv);
4250 else
4251 ++rettv->vval.v_list->lv_refcount;
4252 }
4253 else
4254 {
4255 rettv->v_type = VAR_STRING;
4256 rettv->vval.v_string = get_reg_contents(regname,
4257 arg2 ? GREG_EXPR_SRC : 0);
4258 }
4259}
4260
4261/*
4262 * "getregtype()" function
4263 */
4264 static void
4265f_getregtype(typval_T *argvars, typval_T *rettv)
4266{
4267 char_u *strregname;
4268 int regname;
4269 char_u buf[NUMBUFLEN + 2];
4270 long reglen = 0;
4271
4272 if (argvars[0].v_type != VAR_UNKNOWN)
4273 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004274 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004275 if (strregname == NULL) // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004276 {
4277 rettv->v_type = VAR_STRING;
4278 rettv->vval.v_string = NULL;
4279 return;
4280 }
4281 }
4282 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004283 // Default to v:register
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004284 strregname = get_vim_var_str(VV_REG);
4285
4286 regname = (strregname == NULL ? '"' : *strregname);
4287 if (regname == 0)
4288 regname = '"';
4289
4290 buf[0] = NUL;
4291 buf[1] = NUL;
4292 switch (get_reg_type(regname, &reglen))
4293 {
4294 case MLINE: buf[0] = 'V'; break;
4295 case MCHAR: buf[0] = 'v'; break;
4296 case MBLOCK:
4297 buf[0] = Ctrl_V;
4298 sprintf((char *)buf + 1, "%ld", reglen + 1);
4299 break;
4300 }
4301 rettv->v_type = VAR_STRING;
4302 rettv->vval.v_string = vim_strsave(buf);
4303}
4304
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02004305/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01004306 * "gettagstack()" function
4307 */
4308 static void
4309f_gettagstack(typval_T *argvars, typval_T *rettv)
4310{
4311 win_T *wp = curwin; // default is current window
4312
4313 if (rettv_dict_alloc(rettv) != OK)
4314 return;
4315
4316 if (argvars[0].v_type != VAR_UNKNOWN)
4317 {
4318 wp = find_win_by_nr_or_id(&argvars[0]);
4319 if (wp == NULL)
4320 return;
4321 }
4322
4323 get_tagstack(wp, rettv->vval.v_dict);
4324}
4325
Bram Moolenaar0b39c3f2020-08-30 15:52:10 +02004326/*
4327 * "gettext()" function
4328 */
4329 static void
4330f_gettext(typval_T *argvars, typval_T *rettv)
4331{
4332 if (argvars[0].v_type != VAR_STRING
4333 || argvars[0].vval.v_string == NULL
4334 || *argvars[0].vval.v_string == NUL)
4335 {
4336 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
4337 }
4338 else
4339 {
4340 rettv->v_type = VAR_STRING;
4341 rettv->vval.v_string = vim_strsave(
4342 (char_u *)_(argvars[0].vval.v_string));
4343 }
4344}
4345
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004346// for VIM_VERSION_ defines
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004347#include "version.h"
4348
4349/*
4350 * "has()" function
4351 */
Bram Moolenaara259d8d2020-01-31 20:10:50 +01004352 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004353f_has(typval_T *argvars, typval_T *rettv)
4354{
4355 int i;
4356 char_u *name;
Bram Moolenaar79296512020-03-22 16:17:14 +01004357 int x = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004358 int n = FALSE;
Bram Moolenaar79296512020-03-22 16:17:14 +01004359 typedef struct {
4360 char *name;
4361 short present;
4362 } has_item_T;
4363 static has_item_T has_list[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004364 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004365 {"amiga",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004366#ifdef AMIGA
Bram Moolenaar79296512020-03-22 16:17:14 +01004367 1
Bram Moolenaar39536dd2019-01-29 22:58:21 +01004368#else
Bram Moolenaar79296512020-03-22 16:17:14 +01004369 0
Bram Moolenaar39536dd2019-01-29 22:58:21 +01004370#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004371 },
4372 {"arp",
4373#if defined(AMIGA) && defined(FEAT_ARP)
4374 1
4375#else
4376 0
4377#endif
4378 },
Bram Moolenaar79296512020-03-22 16:17:14 +01004379 {"haiku",
4380#ifdef __HAIKU__
4381 1
4382#else
4383 0
4384#endif
4385 },
4386 {"bsd",
4387#if defined(BSD) && !defined(MACOS_X)
4388 1
4389#else
4390 0
4391#endif
4392 },
4393 {"hpux",
4394#ifdef hpux
4395 1
4396#else
4397 0
4398#endif
4399 },
4400 {"linux",
4401#ifdef __linux__
4402 1
4403#else
4404 0
4405#endif
4406 },
4407 {"mac", // Mac OS X (and, once, Mac OS Classic)
4408#ifdef MACOS_X
4409 1
4410#else
4411 0
4412#endif
4413 },
4414 {"osx", // Mac OS X
4415#ifdef MACOS_X
4416 1
4417#else
4418 0
4419#endif
4420 },
4421 {"macunix", // Mac OS X, with the darwin feature
4422#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
4423 1
4424#else
4425 0
4426#endif
4427 },
4428 {"osxdarwin", // synonym for macunix
4429#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
4430 1
4431#else
4432 0
4433#endif
4434 },
4435 {"qnx",
4436#ifdef __QNX__
4437 1
4438#else
4439 0
4440#endif
4441 },
4442 {"sun",
4443#ifdef SUN_SYSTEM
4444 1
4445#else
4446 0
4447#endif
4448 },
4449 {"unix",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004450#ifdef UNIX
Bram Moolenaar79296512020-03-22 16:17:14 +01004451 1
4452#else
4453 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004454#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004455 },
4456 {"vms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004457#ifdef VMS
Bram Moolenaar79296512020-03-22 16:17:14 +01004458 1
4459#else
4460 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004461#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004462 },
4463 {"win32",
Bram Moolenaar4f974752019-02-17 17:44:42 +01004464#ifdef MSWIN
Bram Moolenaar79296512020-03-22 16:17:14 +01004465 1
4466#else
4467 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004468#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004469 },
4470 {"win32unix",
Bram Moolenaar1eed5322019-02-26 17:03:54 +01004471#if defined(UNIX) && defined(__CYGWIN__)
Bram Moolenaar79296512020-03-22 16:17:14 +01004472 1
4473#else
4474 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004475#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004476 },
4477 {"win64",
Bram Moolenaar44b443c2019-02-18 22:14:18 +01004478#ifdef _WIN64
Bram Moolenaar79296512020-03-22 16:17:14 +01004479 1
4480#else
4481 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004482#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004483 },
4484 {"ebcdic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004485#ifdef EBCDIC
Bram Moolenaar79296512020-03-22 16:17:14 +01004486 1
4487#else
4488 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004489#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004490 },
4491 {"fname_case",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004492#ifndef CASE_INSENSITIVE_FILENAME
Bram Moolenaar79296512020-03-22 16:17:14 +01004493 1
4494#else
4495 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004496#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004497 },
4498 {"acl",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004499#ifdef HAVE_ACL
Bram Moolenaar79296512020-03-22 16:17:14 +01004500 1
4501#else
4502 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004503#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004504 },
4505 {"arabic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004506#ifdef FEAT_ARABIC
Bram Moolenaar79296512020-03-22 16:17:14 +01004507 1
4508#else
4509 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004510#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004511 },
4512 {"autocmd", 1},
4513 {"autochdir",
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02004514#ifdef FEAT_AUTOCHDIR
Bram Moolenaar79296512020-03-22 16:17:14 +01004515 1
4516#else
4517 0
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02004518#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004519 },
4520 {"autoservername",
Bram Moolenaare42a6d22017-11-12 19:21:51 +01004521#ifdef FEAT_AUTOSERVERNAME
Bram Moolenaar79296512020-03-22 16:17:14 +01004522 1
4523#else
4524 0
Bram Moolenaare42a6d22017-11-12 19:21:51 +01004525#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004526 },
4527 {"balloon_eval",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01004528#ifdef FEAT_BEVAL_GUI
Bram Moolenaar79296512020-03-22 16:17:14 +01004529 1
4530#else
4531 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004532#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004533 },
4534 {"balloon_multiline",
4535#if defined(FEAT_BEVAL_GUI) && !defined(FEAT_GUI_MSWIN)
4536 // MS-Windows requires runtime check, see below
4537 1
4538#else
4539 0
4540#endif
4541 },
4542 {"balloon_eval_term",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01004543#ifdef FEAT_BEVAL_TERM
Bram Moolenaar79296512020-03-22 16:17:14 +01004544 1
4545#else
4546 0
Bram Moolenaar51b0f372017-11-18 18:52:04 +01004547#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004548 },
4549 {"builtin_terms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004550#if defined(SOME_BUILTIN_TCAPS) || defined(ALL_BUILTIN_TCAPS)
Bram Moolenaar79296512020-03-22 16:17:14 +01004551 1
4552#else
4553 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004554#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004555 },
4556 {"all_builtin_terms",
4557#if defined(ALL_BUILTIN_TCAPS)
4558 1
4559#else
4560 0
4561#endif
4562 },
4563 {"browsefilter",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004564#if defined(FEAT_BROWSE) && (defined(USE_FILE_CHOOSER) \
Bram Moolenaar4f974752019-02-17 17:44:42 +01004565 || defined(FEAT_GUI_MSWIN) \
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004566 || defined(FEAT_GUI_MOTIF))
Bram Moolenaar79296512020-03-22 16:17:14 +01004567 1
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004568#else
Bram Moolenaar79296512020-03-22 16:17:14 +01004569 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004570#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004571 },
4572 {"byte_offset",
4573#ifdef FEAT_BYTEOFF
4574 1
4575#else
4576 0
4577#endif
4578 },
4579 {"channel",
4580#ifdef FEAT_JOB_CHANNEL
4581 1
4582#else
4583 0
4584#endif
4585 },
4586 {"cindent",
4587#ifdef FEAT_CINDENT
4588 1
4589#else
4590 0
4591#endif
4592 },
4593 {"clientserver",
4594#ifdef FEAT_CLIENTSERVER
4595 1
4596#else
4597 0
4598#endif
4599 },
4600 {"clipboard",
4601#ifdef FEAT_CLIPBOARD
4602 1
4603#else
4604 0
4605#endif
4606 },
4607 {"cmdline_compl", 1},
4608 {"cmdline_hist", 1},
4609 {"comments", 1},
4610 {"conceal",
4611#ifdef FEAT_CONCEAL
4612 1
4613#else
4614 0
4615#endif
4616 },
4617 {"cryptv",
4618#ifdef FEAT_CRYPT
4619 1
4620#else
4621 0
4622#endif
4623 },
4624 {"crypt-blowfish",
4625#ifdef FEAT_CRYPT
4626 1
4627#else
4628 0
4629#endif
4630 },
4631 {"crypt-blowfish2",
4632#ifdef FEAT_CRYPT
4633 1
4634#else
4635 0
4636#endif
4637 },
4638 {"cscope",
4639#ifdef FEAT_CSCOPE
4640 1
4641#else
4642 0
4643#endif
4644 },
4645 {"cursorbind", 1},
4646 {"cursorshape",
4647#ifdef CURSOR_SHAPE
4648 1
4649#else
4650 0
4651#endif
4652 },
4653 {"debug",
4654#ifdef DEBUG
4655 1
4656#else
4657 0
4658#endif
4659 },
4660 {"dialog_con",
4661#ifdef FEAT_CON_DIALOG
4662 1
4663#else
4664 0
4665#endif
4666 },
4667 {"dialog_gui",
4668#ifdef FEAT_GUI_DIALOG
4669 1
4670#else
4671 0
4672#endif
4673 },
4674 {"diff",
4675#ifdef FEAT_DIFF
4676 1
4677#else
4678 0
4679#endif
4680 },
4681 {"digraphs",
4682#ifdef FEAT_DIGRAPHS
4683 1
4684#else
4685 0
4686#endif
4687 },
4688 {"directx",
4689#ifdef FEAT_DIRECTX
4690 1
4691#else
4692 0
4693#endif
4694 },
4695 {"dnd",
4696#ifdef FEAT_DND
4697 1
4698#else
4699 0
4700#endif
4701 },
4702 {"emacs_tags",
4703#ifdef FEAT_EMACS_TAGS
4704 1
4705#else
4706 0
4707#endif
4708 },
4709 {"eval", 1}, // always present, of course!
4710 {"ex_extra", 1}, // graduated feature
4711 {"extra_search",
4712#ifdef FEAT_SEARCH_EXTRA
4713 1
4714#else
4715 0
4716#endif
4717 },
4718 {"file_in_path",
4719#ifdef FEAT_SEARCHPATH
4720 1
4721#else
4722 0
4723#endif
4724 },
4725 {"filterpipe",
4726#if defined(FEAT_FILTERPIPE) && !defined(VIMDLL)
4727 1
4728#else
4729 0
4730#endif
4731 },
4732 {"find_in_path",
4733#ifdef FEAT_FIND_ID
4734 1
4735#else
4736 0
4737#endif
4738 },
4739 {"float",
4740#ifdef FEAT_FLOAT
4741 1
4742#else
4743 0
4744#endif
4745 },
4746 {"folding",
4747#ifdef FEAT_FOLDING
4748 1
4749#else
4750 0
4751#endif
4752 },
4753 {"footer",
4754#ifdef FEAT_FOOTER
4755 1
4756#else
4757 0
4758#endif
4759 },
4760 {"fork",
4761#if !defined(USE_SYSTEM) && defined(UNIX)
4762 1
4763#else
4764 0
4765#endif
4766 },
4767 {"gettext",
4768#ifdef FEAT_GETTEXT
4769 1
4770#else
4771 0
4772#endif
4773 },
4774 {"gui",
4775#ifdef FEAT_GUI
4776 1
4777#else
4778 0
4779#endif
4780 },
4781 {"gui_neXtaw",
4782#if defined(FEAT_GUI_ATHENA) && defined(FEAT_GUI_NEXTAW)
4783 1
4784#else
4785 0
4786#endif
4787 },
4788 {"gui_athena",
4789#if defined(FEAT_GUI_ATHENA) && !defined(FEAT_GUI_NEXTAW)
4790 1
4791#else
4792 0
4793#endif
4794 },
4795 {"gui_gtk",
4796#ifdef FEAT_GUI_GTK
4797 1
4798#else
4799 0
4800#endif
4801 },
4802 {"gui_gtk2",
4803#if defined(FEAT_GUI_GTK) && !defined(USE_GTK3)
4804 1
4805#else
4806 0
4807#endif
4808 },
4809 {"gui_gtk3",
4810#if defined(FEAT_GUI_GTK) && defined(USE_GTK3)
4811 1
4812#else
4813 0
4814#endif
4815 },
4816 {"gui_gnome",
4817#ifdef FEAT_GUI_GNOME
4818 1
4819#else
4820 0
4821#endif
4822 },
4823 {"gui_haiku",
4824#ifdef FEAT_GUI_HAIKU
4825 1
4826#else
4827 0
4828#endif
4829 },
Bram Moolenaar097148e2020-08-11 21:58:20 +02004830 {"gui_mac", 0},
Bram Moolenaar79296512020-03-22 16:17:14 +01004831 {"gui_motif",
4832#ifdef FEAT_GUI_MOTIF
4833 1
4834#else
4835 0
4836#endif
4837 },
4838 {"gui_photon",
4839#ifdef FEAT_GUI_PHOTON
4840 1
4841#else
4842 0
4843#endif
4844 },
4845 {"gui_win32",
4846#ifdef FEAT_GUI_MSWIN
4847 1
4848#else
4849 0
4850#endif
4851 },
4852 {"iconv",
4853#if defined(HAVE_ICONV_H) && defined(USE_ICONV)
4854 1
4855#else
4856 0
4857#endif
4858 },
4859 {"insert_expand", 1},
Bram Moolenaarbfe13cc2020-04-12 17:53:12 +02004860 {"ipv6",
4861#ifdef FEAT_IPV6
4862 1
4863#else
4864 0
4865#endif
4866 },
Bram Moolenaar79296512020-03-22 16:17:14 +01004867 {"job",
4868#ifdef FEAT_JOB_CHANNEL
4869 1
4870#else
4871 0
4872#endif
4873 },
4874 {"jumplist",
4875#ifdef FEAT_JUMPLIST
4876 1
4877#else
4878 0
4879#endif
4880 },
4881 {"keymap",
4882#ifdef FEAT_KEYMAP
4883 1
4884#else
4885 0
4886#endif
4887 },
4888 {"lambda", 1}, // always with FEAT_EVAL, since 7.4.2120 with closure
4889 {"langmap",
4890#ifdef FEAT_LANGMAP
4891 1
4892#else
4893 0
4894#endif
4895 },
4896 {"libcall",
4897#ifdef FEAT_LIBCALL
4898 1
4899#else
4900 0
4901#endif
4902 },
4903 {"linebreak",
4904#ifdef FEAT_LINEBREAK
4905 1
4906#else
4907 0
4908#endif
4909 },
4910 {"lispindent",
4911#ifdef FEAT_LISP
4912 1
4913#else
4914 0
4915#endif
4916 },
4917 {"listcmds", 1},
4918 {"localmap", 1},
4919 {"lua",
4920#if defined(FEAT_LUA) && !defined(DYNAMIC_LUA)
4921 1
4922#else
4923 0
4924#endif
4925 },
4926 {"menu",
4927#ifdef FEAT_MENU
4928 1
4929#else
4930 0
4931#endif
4932 },
4933 {"mksession",
4934#ifdef FEAT_SESSION
4935 1
4936#else
4937 0
4938#endif
4939 },
4940 {"modify_fname", 1},
4941 {"mouse", 1},
4942 {"mouseshape",
4943#ifdef FEAT_MOUSESHAPE
4944 1
4945#else
4946 0
4947#endif
4948 },
4949 {"mouse_dec",
4950#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_DEC)
4951 1
4952#else
4953 0
4954#endif
4955 },
4956 {"mouse_gpm",
4957#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_GPM)
4958 1
4959#else
4960 0
4961#endif
4962 },
4963 {"mouse_jsbterm",
4964#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_JSB)
4965 1
4966#else
4967 0
4968#endif
4969 },
4970 {"mouse_netterm",
4971#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_NET)
4972 1
4973#else
4974 0
4975#endif
4976 },
4977 {"mouse_pterm",
4978#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_PTERM)
4979 1
4980#else
4981 0
4982#endif
4983 },
4984 {"mouse_sgr",
4985#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4986 1
4987#else
4988 0
4989#endif
4990 },
4991 {"mouse_sysmouse",
4992#if (defined(UNIX) || defined(VMS)) && defined(FEAT_SYSMOUSE)
4993 1
4994#else
4995 0
4996#endif
4997 },
4998 {"mouse_urxvt",
4999#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_URXVT)
5000 1
5001#else
5002 0
5003#endif
5004 },
5005 {"mouse_xterm",
5006#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
5007 1
5008#else
5009 0
5010#endif
5011 },
5012 {"multi_byte", 1},
5013 {"multi_byte_ime",
5014#ifdef FEAT_MBYTE_IME
5015 1
5016#else
5017 0
5018#endif
5019 },
5020 {"multi_lang",
5021#ifdef FEAT_MULTI_LANG
5022 1
5023#else
5024 0
5025#endif
5026 },
5027 {"mzscheme",
5028#if defined(FEAT_MZSCHEME) && !defined(DYNAMIC_MZSCHEME)
5029 1
5030#else
5031 0
5032#endif
5033 },
5034 {"num64", 1},
5035 {"ole",
5036#ifdef FEAT_OLE
5037 1
5038#else
5039 0
5040#endif
5041 },
5042 {"packages",
5043#ifdef FEAT_EVAL
5044 1
5045#else
5046 0
5047#endif
5048 },
5049 {"path_extra",
5050#ifdef FEAT_PATH_EXTRA
5051 1
5052#else
5053 0
5054#endif
5055 },
5056 {"perl",
5057#if defined(FEAT_PERL) && !defined(DYNAMIC_PERL)
5058 1
5059#else
5060 0
5061#endif
5062 },
5063 {"persistent_undo",
5064#ifdef FEAT_PERSISTENT_UNDO
5065 1
5066#else
5067 0
5068#endif
5069 },
5070 {"python_compiled",
5071#if defined(FEAT_PYTHON)
5072 1
5073#else
5074 0
5075#endif
5076 },
5077 {"python_dynamic",
5078#if defined(FEAT_PYTHON) && defined(DYNAMIC_PYTHON)
5079 1
5080#else
5081 0
5082#endif
5083 },
5084 {"python",
5085#if defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)
5086 1
5087#else
5088 0
5089#endif
5090 },
5091 {"pythonx",
5092#if (defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)) \
5093 || (defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3))
5094 1
5095#else
5096 0
5097#endif
5098 },
5099 {"python3_compiled",
5100#if defined(FEAT_PYTHON3)
5101 1
5102#else
5103 0
5104#endif
5105 },
5106 {"python3_dynamic",
5107#if defined(FEAT_PYTHON3) && defined(DYNAMIC_PYTHON3)
5108 1
5109#else
5110 0
5111#endif
5112 },
5113 {"python3",
5114#if defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3)
5115 1
5116#else
5117 0
5118#endif
5119 },
5120 {"popupwin",
5121#ifdef FEAT_PROP_POPUP
5122 1
5123#else
5124 0
5125#endif
5126 },
5127 {"postscript",
5128#ifdef FEAT_POSTSCRIPT
5129 1
5130#else
5131 0
5132#endif
5133 },
5134 {"printer",
5135#ifdef FEAT_PRINTER
5136 1
5137#else
5138 0
5139#endif
5140 },
5141 {"profile",
5142#ifdef FEAT_PROFILE
5143 1
5144#else
5145 0
5146#endif
5147 },
5148 {"reltime",
5149#ifdef FEAT_RELTIME
5150 1
5151#else
5152 0
5153#endif
5154 },
5155 {"quickfix",
5156#ifdef FEAT_QUICKFIX
5157 1
5158#else
5159 0
5160#endif
5161 },
5162 {"rightleft",
5163#ifdef FEAT_RIGHTLEFT
5164 1
5165#else
5166 0
5167#endif
5168 },
5169 {"ruby",
5170#if defined(FEAT_RUBY) && !defined(DYNAMIC_RUBY)
5171 1
5172#else
5173 0
5174#endif
5175 },
5176 {"scrollbind", 1},
5177 {"showcmd",
5178#ifdef FEAT_CMDL_INFO
5179 1
5180#else
5181 0
5182#endif
5183 },
5184 {"cmdline_info",
5185#ifdef FEAT_CMDL_INFO
5186 1
5187#else
5188 0
5189#endif
5190 },
5191 {"signs",
5192#ifdef FEAT_SIGNS
5193 1
5194#else
5195 0
5196#endif
5197 },
5198 {"smartindent",
5199#ifdef FEAT_SMARTINDENT
5200 1
5201#else
5202 0
5203#endif
5204 },
5205 {"startuptime",
5206#ifdef STARTUPTIME
5207 1
5208#else
5209 0
5210#endif
5211 },
5212 {"statusline",
5213#ifdef FEAT_STL_OPT
5214 1
5215#else
5216 0
5217#endif
5218 },
5219 {"netbeans_intg",
5220#ifdef FEAT_NETBEANS_INTG
5221 1
5222#else
5223 0
5224#endif
5225 },
5226 {"sound",
5227#ifdef FEAT_SOUND
5228 1
5229#else
5230 0
5231#endif
5232 },
5233 {"spell",
5234#ifdef FEAT_SPELL
5235 1
5236#else
5237 0
5238#endif
5239 },
5240 {"syntax",
5241#ifdef FEAT_SYN_HL
5242 1
5243#else
5244 0
5245#endif
5246 },
5247 {"system",
5248#if defined(USE_SYSTEM) || !defined(UNIX)
5249 1
5250#else
5251 0
5252#endif
5253 },
5254 {"tag_binary",
5255#ifdef FEAT_TAG_BINS
5256 1
5257#else
5258 0
5259#endif
5260 },
5261 {"tcl",
5262#if defined(FEAT_TCL) && !defined(DYNAMIC_TCL)
5263 1
5264#else
5265 0
5266#endif
5267 },
5268 {"termguicolors",
5269#ifdef FEAT_TERMGUICOLORS
5270 1
5271#else
5272 0
5273#endif
5274 },
5275 {"terminal",
5276#if defined(FEAT_TERMINAL) && !defined(MSWIN)
5277 1
5278#else
5279 0
5280#endif
5281 },
5282 {"terminfo",
5283#ifdef TERMINFO
5284 1
5285#else
5286 0
5287#endif
5288 },
5289 {"termresponse",
5290#ifdef FEAT_TERMRESPONSE
5291 1
5292#else
5293 0
5294#endif
5295 },
5296 {"textobjects",
5297#ifdef FEAT_TEXTOBJ
5298 1
5299#else
5300 0
5301#endif
5302 },
5303 {"textprop",
5304#ifdef FEAT_PROP_POPUP
5305 1
5306#else
5307 0
5308#endif
5309 },
5310 {"tgetent",
5311#ifdef HAVE_TGETENT
5312 1
5313#else
5314 0
5315#endif
5316 },
5317 {"timers",
5318#ifdef FEAT_TIMERS
5319 1
5320#else
5321 0
5322#endif
5323 },
5324 {"title",
5325#ifdef FEAT_TITLE
5326 1
5327#else
5328 0
5329#endif
5330 },
5331 {"toolbar",
5332#ifdef FEAT_TOOLBAR
5333 1
5334#else
5335 0
5336#endif
5337 },
5338 {"unnamedplus",
5339#if defined(FEAT_CLIPBOARD) && defined(FEAT_X11)
5340 1
5341#else
5342 0
5343#endif
5344 },
5345 {"user-commands", 1}, // was accidentally included in 5.4
5346 {"user_commands", 1},
5347 {"vartabs",
5348#ifdef FEAT_VARTABS
5349 1
5350#else
5351 0
5352#endif
5353 },
5354 {"vertsplit", 1},
5355 {"viminfo",
5356#ifdef FEAT_VIMINFO
5357 1
5358#else
5359 0
5360#endif
5361 },
5362 {"vimscript-1", 1},
5363 {"vimscript-2", 1},
5364 {"vimscript-3", 1},
5365 {"vimscript-4", 1},
5366 {"virtualedit", 1},
5367 {"visual", 1},
5368 {"visualextra", 1},
5369 {"vreplace", 1},
5370 {"vtp",
5371#ifdef FEAT_VTP
5372 1
5373#else
5374 0
5375#endif
5376 },
5377 {"wildignore",
5378#ifdef FEAT_WILDIGN
5379 1
5380#else
5381 0
5382#endif
5383 },
5384 {"wildmenu",
5385#ifdef FEAT_WILDMENU
5386 1
5387#else
5388 0
5389#endif
5390 },
5391 {"windows", 1},
5392 {"winaltkeys",
5393#ifdef FEAT_WAK
5394 1
5395#else
5396 0
5397#endif
5398 },
5399 {"writebackup",
5400#ifdef FEAT_WRITEBACKUP
5401 1
5402#else
5403 0
5404#endif
5405 },
5406 {"xim",
5407#ifdef FEAT_XIM
5408 1
5409#else
5410 0
5411#endif
5412 },
5413 {"xfontset",
5414#ifdef FEAT_XFONTSET
5415 1
5416#else
5417 0
5418#endif
5419 },
5420 {"xpm",
5421#if defined(FEAT_XPM_W32) || defined(HAVE_XPM)
5422 1
5423#else
5424 0
5425#endif
5426 },
5427 {"xpm_w32", // for backward compatibility
5428#ifdef FEAT_XPM_W32
5429 1
5430#else
5431 0
5432#endif
5433 },
5434 {"xsmp",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005435#ifdef USE_XSMP
Bram Moolenaar79296512020-03-22 16:17:14 +01005436 1
5437#else
5438 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005439#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005440 },
5441 {"xsmp_interact",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005442#ifdef USE_XSMP_INTERACT
Bram Moolenaar79296512020-03-22 16:17:14 +01005443 1
5444#else
5445 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005446#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005447 },
5448 {"xterm_clipboard",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005449#ifdef FEAT_XCLIPBOARD
Bram Moolenaar79296512020-03-22 16:17:14 +01005450 1
5451#else
5452 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005453#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005454 },
5455 {"xterm_save",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005456#ifdef FEAT_XTERM_SAVE
Bram Moolenaar79296512020-03-22 16:17:14 +01005457 1
5458#else
5459 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005460#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005461 },
5462 {"X11",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005463#if defined(UNIX) && defined(FEAT_X11)
Bram Moolenaar79296512020-03-22 16:17:14 +01005464 1
5465#else
5466 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005467#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005468 },
5469 {NULL, 0}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005470 };
5471
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005472 name = tv_get_string(&argvars[0]);
Bram Moolenaar79296512020-03-22 16:17:14 +01005473 for (i = 0; has_list[i].name != NULL; ++i)
5474 if (STRICMP(name, has_list[i].name) == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005475 {
Bram Moolenaar79296512020-03-22 16:17:14 +01005476 x = TRUE;
5477 n = has_list[i].present;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005478 break;
5479 }
5480
Bram Moolenaar79296512020-03-22 16:17:14 +01005481 // features also in has_list[] but sometimes enabled at runtime
5482 if (x == TRUE && n == FALSE)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005483 {
Bram Moolenaar79296512020-03-22 16:17:14 +01005484 if (0)
Bram Moolenaar86b9a3e2020-04-07 19:57:29 +02005485 {
5486 // intentionally empty
5487 }
Bram Moolenaar4f974752019-02-17 17:44:42 +01005488#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005489 else if (STRICMP(name, "balloon_multiline") == 0)
5490 n = multiline_balloon_available();
5491#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005492#ifdef VIMDLL
5493 else if (STRICMP(name, "filterpipe") == 0)
5494 n = gui.in_use || gui.starting;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005495#endif
5496#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
5497 else if (STRICMP(name, "iconv") == 0)
5498 n = iconv_enabled(FALSE);
5499#endif
5500#ifdef DYNAMIC_LUA
5501 else if (STRICMP(name, "lua") == 0)
5502 n = lua_enabled(FALSE);
5503#endif
5504#ifdef DYNAMIC_MZSCHEME
5505 else if (STRICMP(name, "mzscheme") == 0)
5506 n = mzscheme_enabled(FALSE);
5507#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005508#ifdef DYNAMIC_PERL
5509 else if (STRICMP(name, "perl") == 0)
5510 n = perl_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005511#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005512#ifdef DYNAMIC_PYTHON
5513 else if (STRICMP(name, "python") == 0)
5514 n = python_enabled(FALSE);
5515#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005516#ifdef DYNAMIC_PYTHON3
5517 else if (STRICMP(name, "python3") == 0)
5518 n = python3_enabled(FALSE);
5519#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005520#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
5521 else if (STRICMP(name, "pythonx") == 0)
5522 {
5523# if defined(DYNAMIC_PYTHON) && defined(DYNAMIC_PYTHON3)
5524 if (p_pyx == 0)
5525 n = python3_enabled(FALSE) || python_enabled(FALSE);
5526 else if (p_pyx == 3)
5527 n = python3_enabled(FALSE);
5528 else if (p_pyx == 2)
5529 n = python_enabled(FALSE);
5530# elif defined(DYNAMIC_PYTHON)
5531 n = python_enabled(FALSE);
5532# elif defined(DYNAMIC_PYTHON3)
5533 n = python3_enabled(FALSE);
5534# endif
5535 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005536#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005537#ifdef DYNAMIC_RUBY
5538 else if (STRICMP(name, "ruby") == 0)
5539 n = ruby_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005540#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005541#ifdef DYNAMIC_TCL
5542 else if (STRICMP(name, "tcl") == 0)
5543 n = tcl_enabled(FALSE);
Bram Moolenaar4b8366b2019-05-04 17:34:34 +02005544#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +01005545#if defined(FEAT_TERMINAL) && defined(MSWIN)
Bram Moolenaara83e3962017-08-17 14:39:07 +02005546 else if (STRICMP(name, "terminal") == 0)
5547 n = terminal_enabled();
5548#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005549 }
5550
Bram Moolenaar79296512020-03-22 16:17:14 +01005551 // features not in has_list[]
5552 if (x == FALSE)
5553 {
5554 if (STRNICMP(name, "patch", 5) == 0)
5555 {
5556 x = TRUE;
5557 if (name[5] == '-'
5558 && STRLEN(name) >= 11
5559 && vim_isdigit(name[6])
5560 && vim_isdigit(name[8])
5561 && vim_isdigit(name[10]))
5562 {
5563 int major = atoi((char *)name + 6);
5564 int minor = atoi((char *)name + 8);
5565
5566 // Expect "patch-9.9.01234".
5567 n = (major < VIM_VERSION_MAJOR
5568 || (major == VIM_VERSION_MAJOR
5569 && (minor < VIM_VERSION_MINOR
5570 || (minor == VIM_VERSION_MINOR
5571 && has_patch(atoi((char *)name + 10))))));
5572 }
5573 else
5574 n = has_patch(atoi((char *)name + 5));
5575 }
5576 else if (STRICMP(name, "vim_starting") == 0)
5577 {
5578 x = TRUE;
5579 n = (starting != 0);
5580 }
5581 else if (STRICMP(name, "ttyin") == 0)
5582 {
5583 x = TRUE;
5584 n = mch_input_isatty();
5585 }
5586 else if (STRICMP(name, "ttyout") == 0)
5587 {
5588 x = TRUE;
5589 n = stdout_isatty;
5590 }
5591 else if (STRICMP(name, "multi_byte_encoding") == 0)
5592 {
5593 x = TRUE;
5594 n = has_mbyte;
5595 }
5596 else if (STRICMP(name, "gui_running") == 0)
5597 {
5598 x = TRUE;
5599#ifdef FEAT_GUI
5600 n = (gui.in_use || gui.starting);
5601#endif
5602 }
5603 else if (STRICMP(name, "browse") == 0)
5604 {
5605 x = TRUE;
5606#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
5607 n = gui.in_use; // gui_mch_browse() works when GUI is running
5608#endif
5609 }
5610 else if (STRICMP(name, "syntax_items") == 0)
5611 {
5612 x = TRUE;
5613#ifdef FEAT_SYN_HL
5614 n = syntax_present(curwin);
5615#endif
5616 }
5617 else if (STRICMP(name, "vcon") == 0)
5618 {
5619 x = TRUE;
5620#ifdef FEAT_VTP
5621 n = is_term_win32() && has_vtp_working();
5622#endif
5623 }
5624 else if (STRICMP(name, "netbeans_enabled") == 0)
5625 {
5626 x = TRUE;
5627#ifdef FEAT_NETBEANS_INTG
5628 n = netbeans_active();
5629#endif
5630 }
5631 else if (STRICMP(name, "mouse_gpm_enabled") == 0)
5632 {
5633 x = TRUE;
5634#ifdef FEAT_MOUSE_GPM
5635 n = gpm_enabled();
5636#endif
5637 }
5638 else if (STRICMP(name, "conpty") == 0)
5639 {
5640 x = TRUE;
5641#if defined(FEAT_TERMINAL) && defined(MSWIN)
5642 n = use_conpty();
5643#endif
5644 }
5645 else if (STRICMP(name, "clipboard_working") == 0)
5646 {
5647 x = TRUE;
5648#ifdef FEAT_CLIPBOARD
5649 n = clip_star.available;
5650#endif
5651 }
5652 }
5653
Bram Moolenaar04637e22020-09-05 18:45:29 +02005654 if (argvars[1].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[1]))
Bram Moolenaar79296512020-03-22 16:17:14 +01005655 // return whether feature could ever be enabled
5656 rettv->vval.v_number = x;
5657 else
5658 // return whether feature is enabled
5659 rettv->vval.v_number = n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005660}
5661
5662/*
Bram Moolenaar8cebd432020-11-08 12:49:47 +01005663 * Return TRUE if "feature" can change later.
5664 * Also when checking for the feature has side effects, such as loading a DLL.
5665 */
5666 int
5667dynamic_feature(char_u *feature)
5668{
5669 return (feature == NULL
5670#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
5671 || STRICMP(feature, "balloon_multiline") == 0
5672#endif
5673#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
5674 || (STRICMP(feature, "browse") == 0 && !gui.in_use)
5675#endif
5676#ifdef VIMDLL
5677 || STRICMP(feature, "filterpipe") == 0
5678#endif
Bram Moolenaar29b281b2020-11-10 20:58:00 +01005679#if defined(FEAT_GUI) && !defined(ALWAYS_USE_GUI) && !defined(VIMDLL)
Bram Moolenaar8cebd432020-11-08 12:49:47 +01005680 // this can only change on Unix where the ":gui" command could be
5681 // used.
5682 || (STRICMP(feature, "gui_running") == 0 && !gui.in_use)
5683#endif
5684#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
5685 || STRICMP(feature, "iconv") == 0
5686#endif
5687#ifdef DYNAMIC_LUA
5688 || STRICMP(feature, "lua") == 0
5689#endif
5690#ifdef FEAT_MOUSE_GPM
5691 || (STRICMP(feature, "mouse_gpm_enabled") == 0 && !gpm_enabled())
5692#endif
5693#ifdef DYNAMIC_MZSCHEME
5694 || STRICMP(feature, "mzscheme") == 0
5695#endif
5696#ifdef FEAT_NETBEANS_INTG
5697 || STRICMP(feature, "netbeans_enabled") == 0
5698#endif
5699#ifdef DYNAMIC_PERL
5700 || STRICMP(feature, "perl") == 0
5701#endif
5702#ifdef DYNAMIC_PYTHON
5703 || STRICMP(feature, "python") == 0
5704#endif
5705#ifdef DYNAMIC_PYTHON3
5706 || STRICMP(feature, "python3") == 0
5707#endif
5708#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
5709 || STRICMP(feature, "pythonx") == 0
5710#endif
5711#ifdef DYNAMIC_RUBY
5712 || STRICMP(feature, "ruby") == 0
5713#endif
5714#ifdef FEAT_SYN_HL
5715 || STRICMP(feature, "syntax_items") == 0
5716#endif
5717#ifdef DYNAMIC_TCL
5718 || STRICMP(feature, "tcl") == 0
5719#endif
5720 // once "starting" is zero it will stay that way
5721 || (STRICMP(feature, "vim_starting") == 0 && starting != 0)
5722 || STRICMP(feature, "multi_byte_encoding") == 0
5723#if defined(FEAT_TERMINAL) && defined(MSWIN)
5724 || STRICMP(feature, "conpty") == 0
5725#endif
5726 );
5727}
5728
5729/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005730 * "haslocaldir()" function
5731 */
5732 static void
5733f_haslocaldir(typval_T *argvars, typval_T *rettv)
5734{
Bram Moolenaar00aa0692019-04-27 20:37:57 +02005735 tabpage_T *tp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005736 win_T *wp = NULL;
5737
Bram Moolenaar00aa0692019-04-27 20:37:57 +02005738 wp = find_tabwin(&argvars[0], &argvars[1], &tp);
5739
5740 // Check for window-local and tab-local directories
5741 if (wp != NULL && wp->w_localdir != NULL)
5742 rettv->vval.v_number = 1;
5743 else if (tp != NULL && tp->tp_localdir != NULL)
5744 rettv->vval.v_number = 2;
5745 else
5746 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005747}
5748
5749/*
5750 * "hasmapto()" function
5751 */
5752 static void
5753f_hasmapto(typval_T *argvars, typval_T *rettv)
5754{
5755 char_u *name;
5756 char_u *mode;
5757 char_u buf[NUMBUFLEN];
5758 int abbr = FALSE;
5759
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005760 name = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005761 if (argvars[1].v_type == VAR_UNKNOWN)
5762 mode = (char_u *)"nvo";
5763 else
5764 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005765 mode = tv_get_string_buf(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005766 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar04d594b2020-09-02 22:25:35 +02005767 abbr = (int)tv_get_bool(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005768 }
5769
5770 if (map_to_exists(name, mode, abbr))
5771 rettv->vval.v_number = TRUE;
5772 else
5773 rettv->vval.v_number = FALSE;
5774}
5775
5776/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005777 * "highlightID(name)" function
5778 */
5779 static void
5780f_hlID(typval_T *argvars, typval_T *rettv)
5781{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005782 rettv->vval.v_number = syn_name2id(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005783}
5784
5785/*
5786 * "highlight_exists()" function
5787 */
5788 static void
5789f_hlexists(typval_T *argvars, typval_T *rettv)
5790{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005791 rettv->vval.v_number = highlight_exists(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005792}
5793
5794/*
5795 * "hostname()" function
5796 */
5797 static void
5798f_hostname(typval_T *argvars UNUSED, typval_T *rettv)
5799{
5800 char_u hostname[256];
5801
5802 mch_get_host_name(hostname, 256);
5803 rettv->v_type = VAR_STRING;
5804 rettv->vval.v_string = vim_strsave(hostname);
5805}
5806
5807/*
5808 * iconv() function
5809 */
5810 static void
5811f_iconv(typval_T *argvars UNUSED, typval_T *rettv)
5812{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005813 char_u buf1[NUMBUFLEN];
5814 char_u buf2[NUMBUFLEN];
5815 char_u *from, *to, *str;
5816 vimconv_T vimconv;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005817
5818 rettv->v_type = VAR_STRING;
5819 rettv->vval.v_string = NULL;
5820
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005821 str = tv_get_string(&argvars[0]);
5822 from = enc_canonize(enc_skip(tv_get_string_buf(&argvars[1], buf1)));
5823 to = enc_canonize(enc_skip(tv_get_string_buf(&argvars[2], buf2)));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005824 vimconv.vc_type = CONV_NONE;
5825 convert_setup(&vimconv, from, to);
5826
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005827 // If the encodings are equal, no conversion needed.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005828 if (vimconv.vc_type == CONV_NONE)
5829 rettv->vval.v_string = vim_strsave(str);
5830 else
5831 rettv->vval.v_string = string_convert(&vimconv, str, NULL);
5832
5833 convert_setup(&vimconv, NULL, NULL);
5834 vim_free(from);
5835 vim_free(to);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005836}
5837
5838/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005839 * "index()" function
5840 */
5841 static void
5842f_index(typval_T *argvars, typval_T *rettv)
5843{
5844 list_T *l;
5845 listitem_T *item;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005846 blob_T *b;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005847 long idx = 0;
5848 int ic = FALSE;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005849 int error = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005850
5851 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005852 if (argvars[0].v_type == VAR_BLOB)
5853 {
5854 typval_T tv;
5855 int start = 0;
5856
5857 if (argvars[2].v_type != VAR_UNKNOWN)
5858 {
5859 start = tv_get_number_chk(&argvars[2], &error);
5860 if (error)
5861 return;
5862 }
5863 b = argvars[0].vval.v_blob;
5864 if (b == NULL)
5865 return;
Bram Moolenaar05500ec2019-01-13 19:10:33 +01005866 if (start < 0)
5867 {
5868 start = blob_len(b) + start;
5869 if (start < 0)
5870 start = 0;
5871 }
5872
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005873 for (idx = start; idx < blob_len(b); ++idx)
5874 {
5875 tv.v_type = VAR_NUMBER;
5876 tv.vval.v_number = blob_get(b, idx);
5877 if (tv_equal(&tv, &argvars[1], ic, FALSE))
5878 {
5879 rettv->vval.v_number = idx;
5880 return;
5881 }
5882 }
5883 return;
5884 }
5885 else if (argvars[0].v_type != VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005886 {
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01005887 emsg(_(e_listblobreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005888 return;
5889 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005890
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005891 l = argvars[0].vval.v_list;
5892 if (l != NULL)
5893 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02005894 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005895 item = l->lv_first;
5896 if (argvars[2].v_type != VAR_UNKNOWN)
5897 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005898 // Start at specified item. Use the cached index that list_find()
5899 // sets, so that a negative number also works.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005900 item = list_find(l, (long)tv_get_number_chk(&argvars[2], &error));
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01005901 idx = l->lv_u.mat.lv_idx;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005902 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaar6c553f92020-09-02 22:10:34 +02005903 ic = (int)tv_get_bool_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005904 if (error)
5905 item = NULL;
5906 }
5907
5908 for ( ; item != NULL; item = item->li_next, ++idx)
5909 if (tv_equal(&item->li_tv, &argvars[1], ic, FALSE))
5910 {
5911 rettv->vval.v_number = idx;
5912 break;
5913 }
5914 }
5915}
5916
5917static int inputsecret_flag = 0;
5918
5919/*
5920 * "input()" function
5921 * Also handles inputsecret() when inputsecret is set.
5922 */
5923 static void
5924f_input(typval_T *argvars, typval_T *rettv)
5925{
5926 get_user_input(argvars, rettv, FALSE, inputsecret_flag);
5927}
5928
5929/*
5930 * "inputdialog()" function
5931 */
5932 static void
5933f_inputdialog(typval_T *argvars, typval_T *rettv)
5934{
5935#if defined(FEAT_GUI_TEXTDIALOG)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005936 // Use a GUI dialog if the GUI is running and 'c' is not in 'guioptions'
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005937 if (gui.in_use && vim_strchr(p_go, GO_CONDIALOG) == NULL)
5938 {
5939 char_u *message;
5940 char_u buf[NUMBUFLEN];
5941 char_u *defstr = (char_u *)"";
5942
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005943 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005944 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005945 && (defstr = tv_get_string_buf_chk(&argvars[1], buf)) != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005946 vim_strncpy(IObuff, defstr, IOSIZE - 1);
5947 else
5948 IObuff[0] = NUL;
5949 if (message != NULL && defstr != NULL
5950 && do_dialog(VIM_QUESTION, NULL, message,
5951 (char_u *)_("&OK\n&Cancel"), 1, IObuff, FALSE) == 1)
5952 rettv->vval.v_string = vim_strsave(IObuff);
5953 else
5954 {
5955 if (message != NULL && defstr != NULL
5956 && argvars[1].v_type != VAR_UNKNOWN
5957 && argvars[2].v_type != VAR_UNKNOWN)
5958 rettv->vval.v_string = vim_strsave(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005959 tv_get_string_buf(&argvars[2], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005960 else
5961 rettv->vval.v_string = NULL;
5962 }
5963 rettv->v_type = VAR_STRING;
5964 }
5965 else
5966#endif
5967 get_user_input(argvars, rettv, TRUE, inputsecret_flag);
5968}
5969
5970/*
5971 * "inputlist()" function
5972 */
5973 static void
5974f_inputlist(typval_T *argvars, typval_T *rettv)
5975{
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005976 list_T *l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005977 listitem_T *li;
5978 int selected;
5979 int mouse_used;
5980
5981#ifdef NO_CONSOLE_INPUT
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005982 // While starting up, there is no place to enter text. When running tests
5983 // with --not-a-term we assume feedkeys() will be used.
Bram Moolenaar91d348a2017-07-29 20:16:03 +02005984 if (no_console_input() && !is_not_a_term())
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005985 return;
5986#endif
5987 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
5988 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005989 semsg(_(e_listarg), "inputlist()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005990 return;
5991 }
5992
5993 msg_start();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005994 msg_row = Rows - 1; // for when 'cmdheight' > 1
5995 lines_left = Rows; // avoid more prompt
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005996 msg_scroll = TRUE;
5997 msg_clr_eos();
5998
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005999 l = argvars[0].vval.v_list;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006000 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02006001 FOR_ALL_LIST_ITEMS(l, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006002 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01006003 msg_puts((char *)tv_get_string(&li->li_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006004 msg_putchar('\n');
6005 }
6006
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006007 // Ask for choice.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006008 selected = prompt_for_number(&mouse_used);
6009 if (mouse_used)
6010 selected -= lines_left;
6011
6012 rettv->vval.v_number = selected;
6013}
6014
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006015static garray_T ga_userinput = {0, 0, sizeof(tasave_T), 4, NULL};
6016
6017/*
6018 * "inputrestore()" function
6019 */
6020 static void
6021f_inputrestore(typval_T *argvars UNUSED, typval_T *rettv)
6022{
6023 if (ga_userinput.ga_len > 0)
6024 {
6025 --ga_userinput.ga_len;
6026 restore_typeahead((tasave_T *)(ga_userinput.ga_data)
6027 + ga_userinput.ga_len);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006028 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006029 }
6030 else if (p_verbose > 1)
6031 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01006032 verb_msg(_("called inputrestore() more often than inputsave()"));
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006033 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006034 }
6035}
6036
6037/*
6038 * "inputsave()" function
6039 */
6040 static void
6041f_inputsave(typval_T *argvars UNUSED, typval_T *rettv)
6042{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006043 // Add an entry to the stack of typeahead storage.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006044 if (ga_grow(&ga_userinput, 1) == OK)
6045 {
6046 save_typeahead((tasave_T *)(ga_userinput.ga_data)
6047 + ga_userinput.ga_len);
6048 ++ga_userinput.ga_len;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006049 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006050 }
6051 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006052 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006053}
6054
6055/*
6056 * "inputsecret()" function
6057 */
6058 static void
6059f_inputsecret(typval_T *argvars, typval_T *rettv)
6060{
6061 ++cmdline_star;
6062 ++inputsecret_flag;
6063 f_input(argvars, rettv);
6064 --cmdline_star;
6065 --inputsecret_flag;
6066}
6067
6068/*
Bram Moolenaar67a2deb2019-11-25 00:05:32 +01006069 * "interrupt()" function
6070 */
6071 static void
6072f_interrupt(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6073{
6074 got_int = TRUE;
6075}
6076
6077/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006078 * "invert(expr)" function
6079 */
6080 static void
6081f_invert(typval_T *argvars, typval_T *rettv)
6082{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006083 rettv->vval.v_number = ~tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006084}
6085
6086/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006087 * "islocked()" function
6088 */
6089 static void
6090f_islocked(typval_T *argvars, typval_T *rettv)
6091{
6092 lval_T lv;
6093 char_u *end;
6094 dictitem_T *di;
6095
6096 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006097 end = get_lval(tv_get_string(&argvars[0]), NULL, &lv, FALSE, FALSE,
Bram Moolenaar3a257732017-02-21 20:47:13 +01006098 GLV_NO_AUTOLOAD | GLV_READ_ONLY, FNE_CHECK_START);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006099 if (end != NULL && lv.ll_name != NULL)
6100 {
6101 if (*end != NUL)
Bram Moolenaar2d06bfd2020-07-23 17:16:18 +02006102 semsg(_(e_trailing_arg), end);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006103 else
6104 {
6105 if (lv.ll_tv == NULL)
6106 {
Bram Moolenaar79518e22017-02-17 16:31:35 +01006107 di = find_var(lv.ll_name, NULL, TRUE);
6108 if (di != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006109 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006110 // Consider a variable locked when:
6111 // 1. the variable itself is locked
6112 // 2. the value of the variable is locked.
6113 // 3. the List or Dict value is locked.
Bram Moolenaar79518e22017-02-17 16:31:35 +01006114 rettv->vval.v_number = ((di->di_flags & DI_FLAGS_LOCK)
6115 || tv_islocked(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006116 }
6117 }
6118 else if (lv.ll_range)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006119 emsg(_("E786: Range not allowed"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006120 else if (lv.ll_newkey != NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006121 semsg(_(e_dictkey), lv.ll_newkey);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006122 else if (lv.ll_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006123 // List item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006124 rettv->vval.v_number = tv_islocked(&lv.ll_li->li_tv);
6125 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006126 // Dictionary item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006127 rettv->vval.v_number = tv_islocked(&lv.ll_di->di_tv);
6128 }
6129 }
6130
6131 clear_lval(&lv);
6132}
6133
6134#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
6135/*
Bram Moolenaarfda1bff2019-04-04 13:44:37 +02006136 * "isinf()" function
6137 */
6138 static void
6139f_isinf(typval_T *argvars, typval_T *rettv)
6140{
6141 if (argvars[0].v_type == VAR_FLOAT && isinf(argvars[0].vval.v_float))
6142 rettv->vval.v_number = argvars[0].vval.v_float > 0.0 ? 1 : -1;
6143}
6144
6145/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006146 * "isnan()" function
6147 */
6148 static void
6149f_isnan(typval_T *argvars, typval_T *rettv)
6150{
6151 rettv->vval.v_number = argvars[0].v_type == VAR_FLOAT
6152 && isnan(argvars[0].vval.v_float);
6153}
6154#endif
6155
6156/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006157 * "last_buffer_nr()" function.
6158 */
6159 static void
6160f_last_buffer_nr(typval_T *argvars UNUSED, typval_T *rettv)
6161{
6162 int n = 0;
6163 buf_T *buf;
6164
Bram Moolenaar29323592016-07-24 22:04:11 +02006165 FOR_ALL_BUFFERS(buf)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006166 if (n < buf->b_fnum)
6167 n = buf->b_fnum;
6168
6169 rettv->vval.v_number = n;
6170}
6171
6172/*
6173 * "len()" function
6174 */
6175 static void
6176f_len(typval_T *argvars, typval_T *rettv)
6177{
6178 switch (argvars[0].v_type)
6179 {
6180 case VAR_STRING:
6181 case VAR_NUMBER:
6182 rettv->vval.v_number = (varnumber_T)STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006183 tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006184 break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01006185 case VAR_BLOB:
6186 rettv->vval.v_number = blob_len(argvars[0].vval.v_blob);
6187 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006188 case VAR_LIST:
6189 rettv->vval.v_number = list_len(argvars[0].vval.v_list);
6190 break;
6191 case VAR_DICT:
6192 rettv->vval.v_number = dict_len(argvars[0].vval.v_dict);
6193 break;
6194 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02006195 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006196 case VAR_VOID:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01006197 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006198 case VAR_SPECIAL:
6199 case VAR_FLOAT:
6200 case VAR_FUNC:
6201 case VAR_PARTIAL:
6202 case VAR_JOB:
6203 case VAR_CHANNEL:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006204 emsg(_("E701: Invalid type for len()"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006205 break;
6206 }
6207}
6208
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006209 static void
Bram Moolenaar6d721c72017-01-17 16:56:28 +01006210libcall_common(typval_T *argvars UNUSED, typval_T *rettv, int type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006211{
6212#ifdef FEAT_LIBCALL
6213 char_u *string_in;
6214 char_u **string_result;
6215 int nr_result;
6216#endif
6217
6218 rettv->v_type = type;
6219 if (type != VAR_NUMBER)
6220 rettv->vval.v_string = NULL;
6221
6222 if (check_restricted() || check_secure())
6223 return;
6224
6225#ifdef FEAT_LIBCALL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006226 // The first two args must be strings, otherwise it's meaningless
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006227 if (argvars[0].v_type == VAR_STRING && argvars[1].v_type == VAR_STRING)
6228 {
6229 string_in = NULL;
6230 if (argvars[2].v_type == VAR_STRING)
6231 string_in = argvars[2].vval.v_string;
6232 if (type == VAR_NUMBER)
6233 string_result = NULL;
6234 else
6235 string_result = &rettv->vval.v_string;
6236 if (mch_libcall(argvars[0].vval.v_string,
6237 argvars[1].vval.v_string,
6238 string_in,
6239 argvars[2].vval.v_number,
6240 string_result,
6241 &nr_result) == OK
6242 && type == VAR_NUMBER)
6243 rettv->vval.v_number = nr_result;
6244 }
6245#endif
6246}
6247
6248/*
6249 * "libcall()" function
6250 */
6251 static void
6252f_libcall(typval_T *argvars, typval_T *rettv)
6253{
6254 libcall_common(argvars, rettv, VAR_STRING);
6255}
6256
6257/*
6258 * "libcallnr()" function
6259 */
6260 static void
6261f_libcallnr(typval_T *argvars, typval_T *rettv)
6262{
6263 libcall_common(argvars, rettv, VAR_NUMBER);
6264}
6265
6266/*
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006267 * "line(string, [winid])" function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006268 */
6269 static void
6270f_line(typval_T *argvars, typval_T *rettv)
6271{
6272 linenr_T lnum = 0;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006273 pos_T *fp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006274 int fnum;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006275 int id;
6276 tabpage_T *tp;
6277 win_T *wp;
6278 win_T *save_curwin;
6279 tabpage_T *save_curtab;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006280
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006281 if (argvars[1].v_type != VAR_UNKNOWN)
6282 {
6283 // use window specified in the second argument
6284 id = (int)tv_get_number(&argvars[1]);
6285 wp = win_id2wp_tp(id, &tp);
6286 if (wp != NULL && tp != NULL)
6287 {
6288 if (switch_win_noblock(&save_curwin, &save_curtab, wp, tp, TRUE)
6289 == OK)
6290 {
6291 check_cursor();
Bram Moolenaar6f02b002021-01-10 20:22:54 +01006292 fp = var2fpos(&argvars[0], TRUE, &fnum, FALSE);
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006293 }
6294 restore_win_noblock(save_curwin, save_curtab, TRUE);
6295 }
6296 }
6297 else
6298 // use current window
Bram Moolenaar6f02b002021-01-10 20:22:54 +01006299 fp = var2fpos(&argvars[0], TRUE, &fnum, FALSE);
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006300
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006301 if (fp != NULL)
6302 lnum = fp->lnum;
6303 rettv->vval.v_number = lnum;
6304}
6305
6306/*
6307 * "line2byte(lnum)" function
6308 */
6309 static void
6310f_line2byte(typval_T *argvars UNUSED, typval_T *rettv)
6311{
6312#ifndef FEAT_BYTEOFF
6313 rettv->vval.v_number = -1;
6314#else
6315 linenr_T lnum;
6316
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006317 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006318 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count + 1)
6319 rettv->vval.v_number = -1;
6320 else
6321 rettv->vval.v_number = ml_find_line_or_offset(curbuf, lnum, NULL);
6322 if (rettv->vval.v_number >= 0)
6323 ++rettv->vval.v_number;
6324#endif
6325}
6326
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006327#ifdef FEAT_FLOAT
6328/*
6329 * "log()" function
6330 */
6331 static void
6332f_log(typval_T *argvars, typval_T *rettv)
6333{
6334 float_T f = 0.0;
6335
6336 rettv->v_type = VAR_FLOAT;
6337 if (get_float_arg(argvars, &f) == OK)
6338 rettv->vval.v_float = log(f);
6339 else
6340 rettv->vval.v_float = 0.0;
6341}
6342
6343/*
6344 * "log10()" function
6345 */
6346 static void
6347f_log10(typval_T *argvars, typval_T *rettv)
6348{
6349 float_T f = 0.0;
6350
6351 rettv->v_type = VAR_FLOAT;
6352 if (get_float_arg(argvars, &f) == OK)
6353 rettv->vval.v_float = log10(f);
6354 else
6355 rettv->vval.v_float = 0.0;
6356}
6357#endif
6358
6359#ifdef FEAT_LUA
6360/*
6361 * "luaeval()" function
6362 */
6363 static void
6364f_luaeval(typval_T *argvars, typval_T *rettv)
6365{
6366 char_u *str;
6367 char_u buf[NUMBUFLEN];
6368
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006369 if (check_restricted() || check_secure())
6370 return;
6371
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006372 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006373 do_luaeval(str, argvars + 1, rettv);
6374}
6375#endif
6376
6377/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006378 * "maparg()" function
6379 */
6380 static void
6381f_maparg(typval_T *argvars, typval_T *rettv)
6382{
6383 get_maparg(argvars, rettv, TRUE);
6384}
6385
6386/*
6387 * "mapcheck()" function
6388 */
6389 static void
6390f_mapcheck(typval_T *argvars, typval_T *rettv)
6391{
6392 get_maparg(argvars, rettv, FALSE);
6393}
6394
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006395typedef enum
6396{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006397 MATCH_END, // matchend()
6398 MATCH_MATCH, // match()
6399 MATCH_STR, // matchstr()
6400 MATCH_LIST, // matchlist()
6401 MATCH_POS // matchstrpos()
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006402} matchtype_T;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006403
6404 static void
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006405find_some_match(typval_T *argvars, typval_T *rettv, matchtype_T type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006406{
6407 char_u *str = NULL;
6408 long len = 0;
6409 char_u *expr = NULL;
6410 char_u *pat;
6411 regmatch_T regmatch;
6412 char_u patbuf[NUMBUFLEN];
6413 char_u strbuf[NUMBUFLEN];
6414 char_u *save_cpo;
6415 long start = 0;
6416 long nth = 1;
6417 colnr_T startcol = 0;
6418 int match = 0;
6419 list_T *l = NULL;
6420 listitem_T *li = NULL;
6421 long idx = 0;
6422 char_u *tofree = NULL;
6423
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006424 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006425 save_cpo = p_cpo;
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01006426 p_cpo = empty_option;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006427
6428 rettv->vval.v_number = -1;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006429 if (type == MATCH_LIST || type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006430 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006431 // type MATCH_LIST: return empty list when there are no matches.
6432 // type MATCH_POS: return ["", -1, -1, -1]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006433 if (rettv_list_alloc(rettv) == FAIL)
6434 goto theend;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006435 if (type == MATCH_POS
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006436 && (list_append_string(rettv->vval.v_list,
6437 (char_u *)"", 0) == FAIL
6438 || list_append_number(rettv->vval.v_list,
6439 (varnumber_T)-1) == FAIL
6440 || list_append_number(rettv->vval.v_list,
6441 (varnumber_T)-1) == FAIL
6442 || list_append_number(rettv->vval.v_list,
6443 (varnumber_T)-1) == FAIL))
6444 {
6445 list_free(rettv->vval.v_list);
6446 rettv->vval.v_list = NULL;
6447 goto theend;
6448 }
6449 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006450 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006451 {
6452 rettv->v_type = VAR_STRING;
6453 rettv->vval.v_string = NULL;
6454 }
6455
6456 if (argvars[0].v_type == VAR_LIST)
6457 {
6458 if ((l = argvars[0].vval.v_list) == NULL)
6459 goto theend;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006460 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006461 li = l->lv_first;
6462 }
6463 else
6464 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006465 expr = str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006466 len = (long)STRLEN(str);
6467 }
6468
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006469 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006470 if (pat == NULL)
6471 goto theend;
6472
6473 if (argvars[2].v_type != VAR_UNKNOWN)
6474 {
6475 int error = FALSE;
6476
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006477 start = (long)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006478 if (error)
6479 goto theend;
6480 if (l != NULL)
6481 {
6482 li = list_find(l, start);
6483 if (li == NULL)
6484 goto theend;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006485 idx = l->lv_u.mat.lv_idx; // use the cached index
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006486 }
6487 else
6488 {
6489 if (start < 0)
6490 start = 0;
6491 if (start > len)
6492 goto theend;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006493 // When "count" argument is there ignore matches before "start",
6494 // otherwise skip part of the string. Differs when pattern is "^"
6495 // or "\<".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006496 if (argvars[3].v_type != VAR_UNKNOWN)
6497 startcol = start;
6498 else
6499 {
6500 str += start;
6501 len -= start;
6502 }
6503 }
6504
6505 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006506 nth = (long)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006507 if (error)
6508 goto theend;
6509 }
6510
6511 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
6512 if (regmatch.regprog != NULL)
6513 {
6514 regmatch.rm_ic = p_ic;
6515
6516 for (;;)
6517 {
6518 if (l != NULL)
6519 {
6520 if (li == NULL)
6521 {
6522 match = FALSE;
6523 break;
6524 }
6525 vim_free(tofree);
6526 expr = str = echo_string(&li->li_tv, &tofree, strbuf, 0);
6527 if (str == NULL)
6528 break;
6529 }
6530
6531 match = vim_regexec_nl(&regmatch, str, (colnr_T)startcol);
6532
6533 if (match && --nth <= 0)
6534 break;
6535 if (l == NULL && !match)
6536 break;
6537
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006538 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006539 if (l != NULL)
6540 {
6541 li = li->li_next;
6542 ++idx;
6543 }
6544 else
6545 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006546 startcol = (colnr_T)(regmatch.startp[0]
6547 + (*mb_ptr2len)(regmatch.startp[0]) - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006548 if (startcol > (colnr_T)len
6549 || str + startcol <= regmatch.startp[0])
6550 {
6551 match = FALSE;
6552 break;
6553 }
6554 }
6555 }
6556
6557 if (match)
6558 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006559 if (type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006560 {
6561 listitem_T *li1 = rettv->vval.v_list->lv_first;
6562 listitem_T *li2 = li1->li_next;
6563 listitem_T *li3 = li2->li_next;
6564 listitem_T *li4 = li3->li_next;
6565
6566 vim_free(li1->li_tv.vval.v_string);
6567 li1->li_tv.vval.v_string = vim_strnsave(regmatch.startp[0],
Bram Moolenaardf44a272020-06-07 20:49:05 +02006568 regmatch.endp[0] - regmatch.startp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006569 li3->li_tv.vval.v_number =
6570 (varnumber_T)(regmatch.startp[0] - expr);
6571 li4->li_tv.vval.v_number =
6572 (varnumber_T)(regmatch.endp[0] - expr);
6573 if (l != NULL)
6574 li2->li_tv.vval.v_number = (varnumber_T)idx;
6575 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006576 else if (type == MATCH_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006577 {
6578 int i;
6579
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006580 // return list with matched string and submatches
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006581 for (i = 0; i < NSUBEXP; ++i)
6582 {
6583 if (regmatch.endp[i] == NULL)
6584 {
6585 if (list_append_string(rettv->vval.v_list,
6586 (char_u *)"", 0) == FAIL)
6587 break;
6588 }
6589 else if (list_append_string(rettv->vval.v_list,
6590 regmatch.startp[i],
6591 (int)(regmatch.endp[i] - regmatch.startp[i]))
6592 == FAIL)
6593 break;
6594 }
6595 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006596 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006597 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006598 // return matched string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006599 if (l != NULL)
6600 copy_tv(&li->li_tv, rettv);
6601 else
6602 rettv->vval.v_string = vim_strnsave(regmatch.startp[0],
Bram Moolenaardf44a272020-06-07 20:49:05 +02006603 regmatch.endp[0] - regmatch.startp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006604 }
6605 else if (l != NULL)
6606 rettv->vval.v_number = idx;
6607 else
6608 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006609 if (type != MATCH_END)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006610 rettv->vval.v_number =
6611 (varnumber_T)(regmatch.startp[0] - str);
6612 else
6613 rettv->vval.v_number =
6614 (varnumber_T)(regmatch.endp[0] - str);
6615 rettv->vval.v_number += (varnumber_T)(str - expr);
6616 }
6617 }
6618 vim_regfree(regmatch.regprog);
6619 }
6620
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006621theend:
6622 if (type == MATCH_POS && l == NULL && rettv->vval.v_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006623 // matchstrpos() without a list: drop the second item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006624 listitem_remove(rettv->vval.v_list,
6625 rettv->vval.v_list->lv_first->li_next);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006626 vim_free(tofree);
6627 p_cpo = save_cpo;
6628}
6629
6630/*
6631 * "match()" function
6632 */
6633 static void
6634f_match(typval_T *argvars, typval_T *rettv)
6635{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006636 find_some_match(argvars, rettv, MATCH_MATCH);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006637}
6638
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006639/*
6640 * "matchend()" function
6641 */
6642 static void
6643f_matchend(typval_T *argvars, typval_T *rettv)
6644{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006645 find_some_match(argvars, rettv, MATCH_END);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006646}
6647
6648/*
6649 * "matchlist()" function
6650 */
6651 static void
6652f_matchlist(typval_T *argvars, typval_T *rettv)
6653{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006654 find_some_match(argvars, rettv, MATCH_LIST);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006655}
6656
6657/*
6658 * "matchstr()" function
6659 */
6660 static void
6661f_matchstr(typval_T *argvars, typval_T *rettv)
6662{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006663 find_some_match(argvars, rettv, MATCH_STR);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006664}
6665
6666/*
6667 * "matchstrpos()" function
6668 */
6669 static void
6670f_matchstrpos(typval_T *argvars, typval_T *rettv)
6671{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006672 find_some_match(argvars, rettv, MATCH_POS);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006673}
6674
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006675 static void
6676max_min(typval_T *argvars, typval_T *rettv, int domax)
6677{
6678 varnumber_T n = 0;
6679 varnumber_T i;
6680 int error = FALSE;
6681
6682 if (argvars[0].v_type == VAR_LIST)
6683 {
6684 list_T *l;
6685 listitem_T *li;
6686
6687 l = argvars[0].vval.v_list;
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006688 if (l != NULL && l->lv_len > 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006689 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006690 if (l->lv_first == &range_list_item)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006691 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006692 if ((l->lv_u.nonmat.lv_stride > 0) ^ domax)
6693 n = l->lv_u.nonmat.lv_start;
6694 else
6695 n = l->lv_u.nonmat.lv_start + (l->lv_len - 1)
6696 * l->lv_u.nonmat.lv_stride;
6697 }
6698 else
6699 {
6700 li = l->lv_first;
6701 if (li != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006702 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006703 n = tv_get_number_chk(&li->li_tv, &error);
6704 for (;;)
6705 {
6706 li = li->li_next;
6707 if (li == NULL)
6708 break;
6709 i = tv_get_number_chk(&li->li_tv, &error);
6710 if (domax ? i > n : i < n)
6711 n = i;
6712 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006713 }
6714 }
6715 }
6716 }
6717 else if (argvars[0].v_type == VAR_DICT)
6718 {
6719 dict_T *d;
6720 int first = TRUE;
6721 hashitem_T *hi;
6722 int todo;
6723
6724 d = argvars[0].vval.v_dict;
6725 if (d != NULL)
6726 {
6727 todo = (int)d->dv_hashtab.ht_used;
6728 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
6729 {
6730 if (!HASHITEM_EMPTY(hi))
6731 {
6732 --todo;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006733 i = tv_get_number_chk(&HI2DI(hi)->di_tv, &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006734 if (first)
6735 {
6736 n = i;
6737 first = FALSE;
6738 }
6739 else if (domax ? i > n : i < n)
6740 n = i;
6741 }
6742 }
6743 }
6744 }
6745 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006746 semsg(_(e_listdictarg), domax ? "max()" : "min()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006747 rettv->vval.v_number = error ? 0 : n;
6748}
6749
6750/*
6751 * "max()" function
6752 */
6753 static void
6754f_max(typval_T *argvars, typval_T *rettv)
6755{
6756 max_min(argvars, rettv, TRUE);
6757}
6758
6759/*
6760 * "min()" function
6761 */
6762 static void
6763f_min(typval_T *argvars, typval_T *rettv)
6764{
6765 max_min(argvars, rettv, FALSE);
6766}
6767
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006768#if defined(FEAT_MZSCHEME) || defined(PROTO)
6769/*
6770 * "mzeval()" function
6771 */
6772 static void
6773f_mzeval(typval_T *argvars, typval_T *rettv)
6774{
6775 char_u *str;
6776 char_u buf[NUMBUFLEN];
6777
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006778 if (check_restricted() || check_secure())
6779 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006780 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006781 do_mzeval(str, rettv);
6782}
6783
6784 void
6785mzscheme_call_vim(char_u *name, typval_T *args, typval_T *rettv)
6786{
6787 typval_T argvars[3];
6788
6789 argvars[0].v_type = VAR_STRING;
6790 argvars[0].vval.v_string = name;
6791 copy_tv(args, &argvars[1]);
6792 argvars[2].v_type = VAR_UNKNOWN;
6793 f_call(argvars, rettv);
6794 clear_tv(&argvars[1]);
6795}
6796#endif
6797
6798/*
6799 * "nextnonblank()" function
6800 */
6801 static void
6802f_nextnonblank(typval_T *argvars, typval_T *rettv)
6803{
6804 linenr_T lnum;
6805
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006806 for (lnum = tv_get_lnum(argvars); ; ++lnum)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006807 {
6808 if (lnum < 0 || lnum > curbuf->b_ml.ml_line_count)
6809 {
6810 lnum = 0;
6811 break;
6812 }
6813 if (*skipwhite(ml_get(lnum)) != NUL)
6814 break;
6815 }
6816 rettv->vval.v_number = lnum;
6817}
6818
6819/*
6820 * "nr2char()" function
6821 */
6822 static void
6823f_nr2char(typval_T *argvars, typval_T *rettv)
6824{
6825 char_u buf[NUMBUFLEN];
6826
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006827 if (has_mbyte)
6828 {
6829 int utf8 = 0;
6830
6831 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaared6a4302020-09-05 20:29:41 +02006832 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006833 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01006834 buf[utf_char2bytes((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006835 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006836 buf[(*mb_char2bytes)((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006837 }
6838 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006839 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006840 buf[0] = (char_u)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006841 buf[1] = NUL;
6842 }
6843 rettv->v_type = VAR_STRING;
6844 rettv->vval.v_string = vim_strsave(buf);
6845}
6846
6847/*
6848 * "or(expr, expr)" function
6849 */
6850 static void
6851f_or(typval_T *argvars, typval_T *rettv)
6852{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006853 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
6854 | tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006855}
6856
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006857#ifdef FEAT_PERL
6858/*
6859 * "perleval()" function
6860 */
6861 static void
6862f_perleval(typval_T *argvars, typval_T *rettv)
6863{
6864 char_u *str;
6865 char_u buf[NUMBUFLEN];
6866
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006867 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006868 do_perleval(str, rettv);
6869}
6870#endif
6871
6872#ifdef FEAT_FLOAT
6873/*
6874 * "pow()" function
6875 */
6876 static void
6877f_pow(typval_T *argvars, typval_T *rettv)
6878{
6879 float_T fx = 0.0, fy = 0.0;
6880
6881 rettv->v_type = VAR_FLOAT;
6882 if (get_float_arg(argvars, &fx) == OK
6883 && get_float_arg(&argvars[1], &fy) == OK)
6884 rettv->vval.v_float = pow(fx, fy);
6885 else
6886 rettv->vval.v_float = 0.0;
6887}
6888#endif
6889
6890/*
6891 * "prevnonblank()" function
6892 */
6893 static void
6894f_prevnonblank(typval_T *argvars, typval_T *rettv)
6895{
6896 linenr_T lnum;
6897
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006898 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006899 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count)
6900 lnum = 0;
6901 else
6902 while (lnum >= 1 && *skipwhite(ml_get(lnum)) == NUL)
6903 --lnum;
6904 rettv->vval.v_number = lnum;
6905}
6906
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006907// This dummy va_list is here because:
6908// - passing a NULL pointer doesn't work when va_list isn't a pointer
6909// - locally in the function results in a "used before set" warning
6910// - using va_start() to initialize it gives "function with fixed args" error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006911static va_list ap;
6912
6913/*
6914 * "printf()" function
6915 */
6916 static void
6917f_printf(typval_T *argvars, typval_T *rettv)
6918{
6919 char_u buf[NUMBUFLEN];
6920 int len;
6921 char_u *s;
6922 int saved_did_emsg = did_emsg;
6923 char *fmt;
6924
6925 rettv->v_type = VAR_STRING;
6926 rettv->vval.v_string = NULL;
6927
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006928 // Get the required length, allocate the buffer and do it for real.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006929 did_emsg = FALSE;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006930 fmt = (char *)tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02006931 len = vim_vsnprintf_typval(NULL, 0, fmt, ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006932 if (!did_emsg)
6933 {
6934 s = alloc(len + 1);
6935 if (s != NULL)
6936 {
6937 rettv->vval.v_string = s;
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02006938 (void)vim_vsnprintf_typval((char *)s, len + 1, fmt,
6939 ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006940 }
6941 }
6942 did_emsg |= saved_did_emsg;
6943}
6944
6945/*
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006946 * "pum_getpos()" function
6947 */
6948 static void
6949f_pum_getpos(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6950{
6951 if (rettv_dict_alloc(rettv) != OK)
6952 return;
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006953 pum_set_event_info(rettv->vval.v_dict);
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006954}
6955
6956/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006957 * "pumvisible()" function
6958 */
6959 static void
6960f_pumvisible(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6961{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006962 if (pum_visible())
6963 rettv->vval.v_number = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006964}
6965
6966#ifdef FEAT_PYTHON3
6967/*
6968 * "py3eval()" function
6969 */
6970 static void
6971f_py3eval(typval_T *argvars, typval_T *rettv)
6972{
6973 char_u *str;
6974 char_u buf[NUMBUFLEN];
6975
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006976 if (check_restricted() || check_secure())
6977 return;
6978
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006979 if (p_pyx == 0)
6980 p_pyx = 3;
6981
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006982 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006983 do_py3eval(str, rettv);
6984}
6985#endif
6986
6987#ifdef FEAT_PYTHON
6988/*
6989 * "pyeval()" function
6990 */
6991 static void
6992f_pyeval(typval_T *argvars, typval_T *rettv)
6993{
6994 char_u *str;
6995 char_u buf[NUMBUFLEN];
6996
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006997 if (check_restricted() || check_secure())
6998 return;
6999
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01007000 if (p_pyx == 0)
7001 p_pyx = 2;
7002
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007003 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007004 do_pyeval(str, rettv);
7005}
7006#endif
7007
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01007008#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
7009/*
7010 * "pyxeval()" function
7011 */
7012 static void
7013f_pyxeval(typval_T *argvars, typval_T *rettv)
7014{
Bram Moolenaar8c62a082019-02-08 14:34:10 +01007015 if (check_restricted() || check_secure())
7016 return;
7017
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01007018# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
7019 init_pyxversion();
7020 if (p_pyx == 2)
7021 f_pyeval(argvars, rettv);
7022 else
7023 f_py3eval(argvars, rettv);
7024# elif defined(FEAT_PYTHON)
7025 f_pyeval(argvars, rettv);
7026# elif defined(FEAT_PYTHON3)
7027 f_py3eval(argvars, rettv);
7028# endif
7029}
7030#endif
7031
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007032static UINT32_T srand_seed_for_testing = 0;
7033static int srand_seed_for_testing_is_used = FALSE;
7034
7035 static void
7036f_test_srand_seed(typval_T *argvars, typval_T *rettv UNUSED)
7037{
7038 if (argvars[0].v_type == VAR_UNKNOWN)
Bram Moolenaar7633fe52020-06-22 19:10:56 +02007039 srand_seed_for_testing_is_used = FALSE;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007040 else
7041 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02007042 srand_seed_for_testing = (UINT32_T)tv_get_number(&argvars[0]);
7043 srand_seed_for_testing_is_used = TRUE;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007044 }
7045}
7046
7047 static void
7048init_srand(UINT32_T *x)
7049{
7050#ifndef MSWIN
7051 static int dev_urandom_state = NOTDONE; // FAIL or OK once tried
7052#endif
7053
7054 if (srand_seed_for_testing_is_used)
7055 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02007056 *x = srand_seed_for_testing;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007057 return;
7058 }
7059#ifndef MSWIN
7060 if (dev_urandom_state != FAIL)
7061 {
7062 int fd = open("/dev/urandom", O_RDONLY);
7063 struct {
7064 union {
7065 UINT32_T number;
7066 char bytes[sizeof(UINT32_T)];
7067 } contents;
7068 } buf;
7069
7070 // Attempt reading /dev/urandom.
7071 if (fd == -1)
7072 dev_urandom_state = FAIL;
7073 else
7074 {
7075 buf.contents.number = 0;
7076 if (read(fd, buf.contents.bytes, sizeof(UINT32_T))
7077 != sizeof(UINT32_T))
7078 dev_urandom_state = FAIL;
7079 else
7080 {
7081 dev_urandom_state = OK;
7082 *x = buf.contents.number;
7083 }
7084 close(fd);
7085 }
7086 }
7087 if (dev_urandom_state != OK)
7088 // Reading /dev/urandom doesn't work, fall back to time().
7089#endif
7090 *x = vim_time();
7091}
7092
7093#define ROTL(x, k) ((x << k) | (x >> (32 - k)))
7094#define SPLITMIX32(x, z) ( \
7095 z = (x += 0x9e3779b9), \
7096 z = (z ^ (z >> 16)) * 0x85ebca6b, \
7097 z = (z ^ (z >> 13)) * 0xc2b2ae35, \
7098 z ^ (z >> 16) \
7099 )
7100#define SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w) \
7101 result = ROTL(y * 5, 7) * 9; \
7102 t = y << 9; \
7103 z ^= x; \
7104 w ^= y; \
7105 y ^= z, x ^= w; \
7106 z ^= t; \
7107 w = ROTL(w, 11);
7108
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007109/*
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007110 * "rand()" function
7111 */
7112 static void
7113f_rand(typval_T *argvars, typval_T *rettv)
7114{
7115 list_T *l = NULL;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007116 static UINT32_T gx, gy, gz, gw;
7117 static int initialized = FALSE;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007118 listitem_T *lx, *ly, *lz, *lw;
Bram Moolenaar0fd797e2020-11-05 20:46:32 +01007119 UINT32_T x = 0, y, z, w, t, result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007120
7121 if (argvars[0].v_type == VAR_UNKNOWN)
7122 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007123 // When no argument is given use the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007124 if (initialized == FALSE)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007125 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007126 // Initialize the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007127 init_srand(&x);
7128
7129 gx = SPLITMIX32(x, z);
7130 gy = SPLITMIX32(x, z);
7131 gz = SPLITMIX32(x, z);
7132 gw = SPLITMIX32(x, z);
7133 initialized = TRUE;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007134 }
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007135
7136 SHUFFLE_XOSHIRO128STARSTAR(gx, gy, gz, gw);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007137 }
7138 else if (argvars[0].v_type == VAR_LIST)
7139 {
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007140 l = argvars[0].vval.v_list;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007141 if (l == NULL || list_len(l) != 4)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007142 goto theend;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007143
7144 lx = list_find(l, 0L);
7145 ly = list_find(l, 1L);
7146 lz = list_find(l, 2L);
7147 lw = list_find(l, 3L);
7148 if (lx->li_tv.v_type != VAR_NUMBER) goto theend;
7149 if (ly->li_tv.v_type != VAR_NUMBER) goto theend;
7150 if (lz->li_tv.v_type != VAR_NUMBER) goto theend;
7151 if (lw->li_tv.v_type != VAR_NUMBER) goto theend;
7152 x = (UINT32_T)lx->li_tv.vval.v_number;
7153 y = (UINT32_T)ly->li_tv.vval.v_number;
7154 z = (UINT32_T)lz->li_tv.vval.v_number;
7155 w = (UINT32_T)lw->li_tv.vval.v_number;
7156
7157 SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w);
7158
7159 lx->li_tv.vval.v_number = (varnumber_T)x;
7160 ly->li_tv.vval.v_number = (varnumber_T)y;
7161 lz->li_tv.vval.v_number = (varnumber_T)z;
7162 lw->li_tv.vval.v_number = (varnumber_T)w;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007163 }
7164 else
7165 goto theend;
7166
7167 rettv->v_type = VAR_NUMBER;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007168 rettv->vval.v_number = (varnumber_T)result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007169 return;
7170
7171theend:
7172 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007173 rettv->v_type = VAR_NUMBER;
7174 rettv->vval.v_number = -1;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007175}
7176
7177/*
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007178 * "srand()" function
7179 */
7180 static void
7181f_srand(typval_T *argvars, typval_T *rettv)
7182{
7183 UINT32_T x = 0, z;
7184
7185 if (rettv_list_alloc(rettv) == FAIL)
7186 return;
7187 if (argvars[0].v_type == VAR_UNKNOWN)
7188 {
7189 init_srand(&x);
7190 }
7191 else
7192 {
7193 int error = FALSE;
7194
7195 x = (UINT32_T)tv_get_number_chk(&argvars[0], &error);
7196 if (error)
7197 return;
7198 }
7199
7200 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7201 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7202 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7203 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7204}
7205
7206#undef ROTL
7207#undef SPLITMIX32
7208#undef SHUFFLE_XOSHIRO128STARSTAR
7209
7210/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007211 * "range()" function
7212 */
7213 static void
7214f_range(typval_T *argvars, typval_T *rettv)
7215{
7216 varnumber_T start;
7217 varnumber_T end;
7218 varnumber_T stride = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007219 int error = FALSE;
7220
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007221 start = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007222 if (argvars[1].v_type == VAR_UNKNOWN)
7223 {
7224 end = start - 1;
7225 start = 0;
7226 }
7227 else
7228 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007229 end = tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007230 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007231 stride = tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007232 }
7233
7234 if (error)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007235 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007236 if (stride == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007237 emsg(_("E726: Stride is zero"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007238 else if (stride > 0 ? end + 1 < start : end - 1 > start)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007239 emsg(_("E727: Start past end"));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007240 else if (rettv_list_alloc(rettv) == OK)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007241 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007242 list_T *list = rettv->vval.v_list;
7243
7244 // Create a non-materialized list. This is much more efficient and
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007245 // works with ":for". If used otherwise CHECK_LIST_MATERIALIZE() must
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007246 // be called.
7247 list->lv_first = &range_list_item;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01007248 list->lv_u.nonmat.lv_start = start;
7249 list->lv_u.nonmat.lv_end = end;
7250 list->lv_u.nonmat.lv_stride = stride;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007251 list->lv_len = (end - start) / stride + 1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007252 }
7253}
7254
7255/*
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007256 * Materialize "list".
7257 * Do not call directly, use CHECK_LIST_MATERIALIZE()
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007258 */
7259 void
7260range_list_materialize(list_T *list)
7261{
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007262 varnumber_T start = list->lv_u.nonmat.lv_start;
7263 varnumber_T end = list->lv_u.nonmat.lv_end;
7264 int stride = list->lv_u.nonmat.lv_stride;
7265 varnumber_T i;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007266
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007267 list->lv_first = NULL;
7268 list->lv_u.mat.lv_last = NULL;
7269 list->lv_len = 0;
7270 list->lv_u.mat.lv_idx_item = NULL;
7271 for (i = start; stride > 0 ? i <= end : i >= end; i += stride)
7272 if (list_append_number(list, (varnumber_T)i) == FAIL)
7273 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007274}
7275
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007276/*
7277 * "getreginfo()" function
7278 */
7279 static void
7280f_getreginfo(typval_T *argvars, typval_T *rettv)
7281{
7282 char_u *strregname;
7283 int regname;
7284 char_u buf[NUMBUFLEN + 2];
7285 long reglen = 0;
7286 dict_T *dict;
7287 list_T *list;
7288
7289 if (argvars[0].v_type != VAR_UNKNOWN)
7290 {
7291 strregname = tv_get_string_chk(&argvars[0]);
7292 if (strregname == NULL)
7293 return;
7294 }
7295 else
7296 strregname = get_vim_var_str(VV_REG);
7297
7298 regname = (strregname == NULL ? '"' : *strregname);
7299 if (regname == 0 || regname == '@')
7300 regname = '"';
7301
7302 if (rettv_dict_alloc(rettv) == FAIL)
7303 return;
7304 dict = rettv->vval.v_dict;
7305
7306 list = (list_T *)get_reg_contents(regname, GREG_EXPR_SRC | GREG_LIST);
7307 if (list == NULL)
7308 return;
Bram Moolenaar91639192020-06-29 19:55:58 +02007309 (void)dict_add_list(dict, "regcontents", list);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007310
7311 buf[0] = NUL;
7312 buf[1] = NUL;
7313 switch (get_reg_type(regname, &reglen))
7314 {
7315 case MLINE: buf[0] = 'V'; break;
7316 case MCHAR: buf[0] = 'v'; break;
7317 case MBLOCK:
7318 vim_snprintf((char *)buf, sizeof(buf), "%c%ld", Ctrl_V,
7319 reglen + 1);
7320 break;
7321 }
Bram Moolenaar91639192020-06-29 19:55:58 +02007322 (void)dict_add_string(dict, (char *)"regtype", buf);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007323
7324 buf[0] = get_register_name(get_unname_register());
7325 buf[1] = NUL;
7326 if (regname == '"')
Bram Moolenaar91639192020-06-29 19:55:58 +02007327 (void)dict_add_string(dict, (char *)"points_to", buf);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007328 else
7329 {
7330 dictitem_T *item = dictitem_alloc((char_u *)"isunnamed");
7331
7332 if (item != NULL)
7333 {
7334 item->di_tv.v_type = VAR_SPECIAL;
7335 item->di_tv.vval.v_number = regname == buf[0]
7336 ? VVAL_TRUE : VVAL_FALSE;
Bram Moolenaar91639192020-06-29 19:55:58 +02007337 (void)dict_add(dict, item);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007338 }
7339 }
7340}
7341
Bram Moolenaar0b6d9112018-05-22 20:35:17 +02007342 static void
7343return_register(int regname, typval_T *rettv)
7344{
7345 char_u buf[2] = {0, 0};
7346
7347 buf[0] = (char_u)regname;
7348 rettv->v_type = VAR_STRING;
7349 rettv->vval.v_string = vim_strsave(buf);
7350}
7351
7352/*
7353 * "reg_executing()" function
7354 */
7355 static void
7356f_reg_executing(typval_T *argvars UNUSED, typval_T *rettv)
7357{
7358 return_register(reg_executing, rettv);
7359}
7360
7361/*
7362 * "reg_recording()" function
7363 */
7364 static void
7365f_reg_recording(typval_T *argvars UNUSED, typval_T *rettv)
7366{
7367 return_register(reg_recording, rettv);
7368}
7369
Bram Moolenaar7416f3e2017-03-18 18:10:13 +01007370/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007371 * "rename({from}, {to})" function
7372 */
7373 static void
7374f_rename(typval_T *argvars, typval_T *rettv)
7375{
7376 char_u buf[NUMBUFLEN];
7377
7378 if (check_restricted() || check_secure())
7379 rettv->vval.v_number = -1;
7380 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007381 rettv->vval.v_number = vim_rename(tv_get_string(&argvars[0]),
7382 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007383}
7384
7385/*
7386 * "repeat()" function
7387 */
7388 static void
7389f_repeat(typval_T *argvars, typval_T *rettv)
7390{
7391 char_u *p;
7392 int n;
7393 int slen;
7394 int len;
7395 char_u *r;
7396 int i;
7397
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007398 n = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007399 if (argvars[0].v_type == VAR_LIST)
7400 {
7401 if (rettv_list_alloc(rettv) == OK && argvars[0].vval.v_list != NULL)
7402 while (n-- > 0)
7403 if (list_extend(rettv->vval.v_list,
7404 argvars[0].vval.v_list, NULL) == FAIL)
7405 break;
7406 }
7407 else
7408 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007409 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007410 rettv->v_type = VAR_STRING;
7411 rettv->vval.v_string = NULL;
7412
7413 slen = (int)STRLEN(p);
7414 len = slen * n;
7415 if (len <= 0)
7416 return;
7417
7418 r = alloc(len + 1);
7419 if (r != NULL)
7420 {
7421 for (i = 0; i < n; i++)
7422 mch_memmove(r + i * slen, p, (size_t)slen);
7423 r[len] = NUL;
7424 }
7425
7426 rettv->vval.v_string = r;
7427 }
7428}
7429
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007430#define SP_NOMOVE 0x01 // don't move cursor
7431#define SP_REPEAT 0x02 // repeat to find outer pair
7432#define SP_RETCOUNT 0x04 // return matchcount
7433#define SP_SETPCMARK 0x08 // set previous context mark
7434#define SP_START 0x10 // accept match at start position
7435#define SP_SUBPAT 0x20 // return nr of matching sub-pattern
7436#define SP_END 0x40 // leave cursor at end of match
7437#define SP_COLUMN 0x80 // start at cursor column
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007438
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007439/*
7440 * Get flags for a search function.
7441 * Possibly sets "p_ws".
7442 * Returns BACKWARD, FORWARD or zero (for an error).
7443 */
7444 static int
7445get_search_arg(typval_T *varp, int *flagsp)
7446{
7447 int dir = FORWARD;
7448 char_u *flags;
7449 char_u nbuf[NUMBUFLEN];
7450 int mask;
7451
7452 if (varp->v_type != VAR_UNKNOWN)
7453 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007454 flags = tv_get_string_buf_chk(varp, nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007455 if (flags == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007456 return 0; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007457 while (*flags != NUL)
7458 {
7459 switch (*flags)
7460 {
7461 case 'b': dir = BACKWARD; break;
7462 case 'w': p_ws = TRUE; break;
7463 case 'W': p_ws = FALSE; break;
7464 default: mask = 0;
7465 if (flagsp != NULL)
7466 switch (*flags)
7467 {
7468 case 'c': mask = SP_START; break;
7469 case 'e': mask = SP_END; break;
7470 case 'm': mask = SP_RETCOUNT; break;
7471 case 'n': mask = SP_NOMOVE; break;
7472 case 'p': mask = SP_SUBPAT; break;
7473 case 'r': mask = SP_REPEAT; break;
7474 case 's': mask = SP_SETPCMARK; break;
7475 case 'z': mask = SP_COLUMN; break;
7476 }
7477 if (mask == 0)
7478 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007479 semsg(_(e_invarg2), flags);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007480 dir = 0;
7481 }
7482 else
7483 *flagsp |= mask;
7484 }
7485 if (dir == 0)
7486 break;
7487 ++flags;
7488 }
7489 }
7490 return dir;
7491}
7492
7493/*
7494 * Shared by search() and searchpos() functions.
7495 */
7496 static int
7497search_cmn(typval_T *argvars, pos_T *match_pos, int *flagsp)
7498{
7499 int flags;
7500 char_u *pat;
7501 pos_T pos;
7502 pos_T save_cursor;
7503 int save_p_ws = p_ws;
7504 int dir;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007505 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007506 long lnum_stop = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007507#ifdef FEAT_RELTIME
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007508 proftime_T tm;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007509 long time_limit = 0;
7510#endif
7511 int options = SEARCH_KEEP;
7512 int subpatnum;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007513 searchit_arg_T sia;
Bram Moolenaara9c01042020-06-07 14:50:50 +02007514 int use_skip = FALSE;
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007515 pos_T firstpos;
7516
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007517 pat = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007518 dir = get_search_arg(&argvars[1], flagsp); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007519 if (dir == 0)
7520 goto theend;
7521 flags = *flagsp;
7522 if (flags & SP_START)
7523 options |= SEARCH_START;
7524 if (flags & SP_END)
7525 options |= SEARCH_END;
7526 if (flags & SP_COLUMN)
7527 options |= SEARCH_COL;
7528
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007529 // Optional arguments: line number to stop searching, timeout and skip.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007530 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN)
7531 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007532 lnum_stop = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007533 if (lnum_stop < 0)
7534 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007535 if (argvars[3].v_type != VAR_UNKNOWN)
7536 {
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007537#ifdef FEAT_RELTIME
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007538 time_limit = (long)tv_get_number_chk(&argvars[3], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007539 if (time_limit < 0)
7540 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007541#endif
Bram Moolenaara9c01042020-06-07 14:50:50 +02007542 use_skip = eval_expr_valid_arg(&argvars[4]);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007543 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007544 }
7545
7546#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007547 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007548 profile_setlimit(time_limit, &tm);
7549#endif
7550
7551 /*
7552 * This function does not accept SP_REPEAT and SP_RETCOUNT flags.
7553 * Check to make sure only those flags are set.
7554 * Also, Only the SP_NOMOVE or the SP_SETPCMARK flag can be set. Both
7555 * flags cannot be set. Check for that condition also.
7556 */
7557 if (((flags & (SP_REPEAT | SP_RETCOUNT)) != 0)
7558 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
7559 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007560 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007561 goto theend;
7562 }
7563
7564 pos = save_cursor = curwin->w_cursor;
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007565 CLEAR_FIELD(firstpos);
Bram Moolenaara80faa82020-04-12 19:37:17 +02007566 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007567 sia.sa_stop_lnum = (linenr_T)lnum_stop;
7568#ifdef FEAT_RELTIME
7569 sia.sa_tm = &tm;
7570#endif
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007571
7572 // Repeat until {skip} returns FALSE.
7573 for (;;)
7574 {
7575 subpatnum = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007576 options, RE_SEARCH, &sia);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007577 // finding the first match again means there is no match where {skip}
7578 // evaluates to zero.
7579 if (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos))
7580 subpatnum = FAIL;
7581
Bram Moolenaara9c01042020-06-07 14:50:50 +02007582 if (subpatnum == FAIL || !use_skip)
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007583 // didn't find it or no skip argument
7584 break;
7585 firstpos = pos;
7586
Bram Moolenaara9c01042020-06-07 14:50:50 +02007587 // If the skip expression matches, ignore this match.
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007588 {
7589 int do_skip;
7590 int err;
7591 pos_T save_pos = curwin->w_cursor;
7592
7593 curwin->w_cursor = pos;
Bram Moolenaara9c01042020-06-07 14:50:50 +02007594 err = FALSE;
7595 do_skip = eval_expr_to_bool(&argvars[4], &err);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007596 curwin->w_cursor = save_pos;
7597 if (err)
7598 {
7599 // Evaluating {skip} caused an error, break here.
7600 subpatnum = FAIL;
7601 break;
7602 }
7603 if (!do_skip)
7604 break;
7605 }
7606 }
7607
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007608 if (subpatnum != FAIL)
7609 {
7610 if (flags & SP_SUBPAT)
7611 retval = subpatnum;
7612 else
7613 retval = pos.lnum;
7614 if (flags & SP_SETPCMARK)
7615 setpcmark();
7616 curwin->w_cursor = pos;
7617 if (match_pos != NULL)
7618 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007619 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007620 match_pos->lnum = pos.lnum;
7621 match_pos->col = pos.col + 1;
7622 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007623 // "/$" will put the cursor after the end of the line, may need to
7624 // correct that here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007625 check_cursor();
7626 }
7627
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007628 // If 'n' flag is used: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007629 if (flags & SP_NOMOVE)
7630 curwin->w_cursor = save_cursor;
7631 else
7632 curwin->w_set_curswant = TRUE;
7633theend:
7634 p_ws = save_p_ws;
7635
7636 return retval;
7637}
7638
7639#ifdef FEAT_FLOAT
7640
7641/*
7642 * round() is not in C90, use ceil() or floor() instead.
7643 */
7644 float_T
7645vim_round(float_T f)
7646{
7647 return f > 0 ? floor(f + 0.5) : ceil(f - 0.5);
7648}
7649
7650/*
7651 * "round({float})" function
7652 */
7653 static void
7654f_round(typval_T *argvars, typval_T *rettv)
7655{
7656 float_T f = 0.0;
7657
7658 rettv->v_type = VAR_FLOAT;
7659 if (get_float_arg(argvars, &f) == OK)
7660 rettv->vval.v_float = vim_round(f);
7661 else
7662 rettv->vval.v_float = 0.0;
7663}
7664#endif
7665
Bram Moolenaare99be0e2019-03-26 22:51:09 +01007666#ifdef FEAT_RUBY
7667/*
7668 * "rubyeval()" function
7669 */
7670 static void
7671f_rubyeval(typval_T *argvars, typval_T *rettv)
7672{
7673 char_u *str;
7674 char_u buf[NUMBUFLEN];
7675
7676 str = tv_get_string_buf(&argvars[0], buf);
7677 do_rubyeval(str, rettv);
7678}
7679#endif
7680
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007681/*
7682 * "screenattr()" function
7683 */
7684 static void
7685f_screenattr(typval_T *argvars, typval_T *rettv)
7686{
7687 int row;
7688 int col;
7689 int c;
7690
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007691 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7692 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007693 if (row < 0 || row >= screen_Rows
7694 || col < 0 || col >= screen_Columns)
7695 c = -1;
7696 else
7697 c = ScreenAttrs[LineOffset[row] + col];
7698 rettv->vval.v_number = c;
7699}
7700
7701/*
7702 * "screenchar()" function
7703 */
7704 static void
7705f_screenchar(typval_T *argvars, typval_T *rettv)
7706{
7707 int row;
7708 int col;
7709 int off;
7710 int c;
7711
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007712 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7713 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007714 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007715 c = -1;
7716 else
7717 {
7718 off = LineOffset[row] + col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007719 if (enc_utf8 && ScreenLinesUC[off] != 0)
7720 c = ScreenLinesUC[off];
7721 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007722 c = ScreenLines[off];
7723 }
7724 rettv->vval.v_number = c;
7725}
7726
7727/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007728 * "screenchars()" function
7729 */
7730 static void
7731f_screenchars(typval_T *argvars, typval_T *rettv)
7732{
7733 int row;
7734 int col;
7735 int off;
7736 int c;
7737 int i;
7738
7739 if (rettv_list_alloc(rettv) == FAIL)
7740 return;
7741 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7742 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
7743 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
7744 return;
7745
7746 off = LineOffset[row] + col;
7747 if (enc_utf8 && ScreenLinesUC[off] != 0)
7748 c = ScreenLinesUC[off];
7749 else
7750 c = ScreenLines[off];
7751 list_append_number(rettv->vval.v_list, (varnumber_T)c);
7752
7753 if (enc_utf8)
7754
7755 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
7756 list_append_number(rettv->vval.v_list,
7757 (varnumber_T)ScreenLinesC[i][off]);
7758}
7759
7760/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007761 * "screencol()" function
7762 *
7763 * First column is 1 to be consistent with virtcol().
7764 */
7765 static void
7766f_screencol(typval_T *argvars UNUSED, typval_T *rettv)
7767{
7768 rettv->vval.v_number = screen_screencol() + 1;
7769}
7770
7771/*
7772 * "screenrow()" function
7773 */
7774 static void
7775f_screenrow(typval_T *argvars UNUSED, typval_T *rettv)
7776{
7777 rettv->vval.v_number = screen_screenrow() + 1;
7778}
7779
7780/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007781 * "screenstring()" function
7782 */
7783 static void
7784f_screenstring(typval_T *argvars, typval_T *rettv)
7785{
7786 int row;
7787 int col;
7788 int off;
7789 int c;
7790 int i;
7791 char_u buf[MB_MAXBYTES + 1];
7792 int buflen = 0;
7793
7794 rettv->vval.v_string = NULL;
7795 rettv->v_type = VAR_STRING;
7796
7797 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7798 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
7799 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
7800 return;
7801
7802 off = LineOffset[row] + col;
7803 if (enc_utf8 && ScreenLinesUC[off] != 0)
7804 c = ScreenLinesUC[off];
7805 else
7806 c = ScreenLines[off];
7807 buflen += mb_char2bytes(c, buf);
7808
7809 if (enc_utf8)
7810 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
7811 buflen += mb_char2bytes(ScreenLinesC[i][off], buf + buflen);
7812
7813 buf[buflen] = NUL;
7814 rettv->vval.v_string = vim_strsave(buf);
7815}
7816
7817/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007818 * "search()" function
7819 */
7820 static void
7821f_search(typval_T *argvars, typval_T *rettv)
7822{
7823 int flags = 0;
7824
7825 rettv->vval.v_number = search_cmn(argvars, NULL, &flags);
7826}
7827
7828/*
7829 * "searchdecl()" function
7830 */
7831 static void
7832f_searchdecl(typval_T *argvars, typval_T *rettv)
7833{
Bram Moolenaar30788d32020-09-05 21:35:16 +02007834 int locally = TRUE;
7835 int thisblock = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007836 int error = FALSE;
7837 char_u *name;
7838
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007839 rettv->vval.v_number = 1; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007840
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007841 name = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007842 if (argvars[1].v_type != VAR_UNKNOWN)
7843 {
Bram Moolenaar30788d32020-09-05 21:35:16 +02007844 locally = !(int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007845 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar30788d32020-09-05 21:35:16 +02007846 thisblock = (int)tv_get_bool_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007847 }
7848 if (!error && name != NULL)
7849 rettv->vval.v_number = find_decl(name, (int)STRLEN(name),
7850 locally, thisblock, SEARCH_KEEP) == FAIL;
7851}
7852
7853/*
7854 * Used by searchpair() and searchpairpos()
7855 */
7856 static int
7857searchpair_cmn(typval_T *argvars, pos_T *match_pos)
7858{
7859 char_u *spat, *mpat, *epat;
Bram Moolenaar48570482017-10-30 21:48:41 +01007860 typval_T *skip;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007861 int save_p_ws = p_ws;
7862 int dir;
7863 int flags = 0;
7864 char_u nbuf1[NUMBUFLEN];
7865 char_u nbuf2[NUMBUFLEN];
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007866 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007867 long lnum_stop = 0;
7868 long time_limit = 0;
7869
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007870 // Get the three pattern arguments: start, middle, end. Will result in an
7871 // error if not a valid argument.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007872 spat = tv_get_string_chk(&argvars[0]);
7873 mpat = tv_get_string_buf_chk(&argvars[1], nbuf1);
7874 epat = tv_get_string_buf_chk(&argvars[2], nbuf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007875 if (spat == NULL || mpat == NULL || epat == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007876 goto theend; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007877
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007878 // Handle the optional fourth argument: flags
7879 dir = get_search_arg(&argvars[3], &flags); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007880 if (dir == 0)
7881 goto theend;
7882
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007883 // Don't accept SP_END or SP_SUBPAT.
7884 // Only one of the SP_NOMOVE or SP_SETPCMARK flags can be set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007885 if ((flags & (SP_END | SP_SUBPAT)) != 0
7886 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
7887 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007888 semsg(_(e_invarg2), tv_get_string(&argvars[3]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007889 goto theend;
7890 }
7891
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007892 // Using 'r' implies 'W', otherwise it doesn't work.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007893 if (flags & SP_REPEAT)
7894 p_ws = FALSE;
7895
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007896 // Optional fifth argument: skip expression
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007897 if (argvars[3].v_type == VAR_UNKNOWN
7898 || argvars[4].v_type == VAR_UNKNOWN)
Bram Moolenaar48570482017-10-30 21:48:41 +01007899 skip = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007900 else
7901 {
Bram Moolenaara9c01042020-06-07 14:50:50 +02007902 // Type is checked later.
Bram Moolenaar48570482017-10-30 21:48:41 +01007903 skip = &argvars[4];
Bram Moolenaara9c01042020-06-07 14:50:50 +02007904
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007905 if (argvars[5].v_type != VAR_UNKNOWN)
7906 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007907 lnum_stop = (long)tv_get_number_chk(&argvars[5], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007908 if (lnum_stop < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007909 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007910 semsg(_(e_invarg2), tv_get_string(&argvars[5]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007911 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007912 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007913#ifdef FEAT_RELTIME
7914 if (argvars[6].v_type != VAR_UNKNOWN)
7915 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007916 time_limit = (long)tv_get_number_chk(&argvars[6], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007917 if (time_limit < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007918 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007919 semsg(_(e_invarg2), tv_get_string(&argvars[6]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007920 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007921 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007922 }
7923#endif
7924 }
7925 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007926
7927 retval = do_searchpair(spat, mpat, epat, dir, skip, flags,
7928 match_pos, lnum_stop, time_limit);
7929
7930theend:
7931 p_ws = save_p_ws;
7932
7933 return retval;
7934}
7935
7936/*
7937 * "searchpair()" function
7938 */
7939 static void
7940f_searchpair(typval_T *argvars, typval_T *rettv)
7941{
7942 rettv->vval.v_number = searchpair_cmn(argvars, NULL);
7943}
7944
7945/*
7946 * "searchpairpos()" function
7947 */
7948 static void
7949f_searchpairpos(typval_T *argvars, typval_T *rettv)
7950{
7951 pos_T match_pos;
7952 int lnum = 0;
7953 int col = 0;
7954
7955 if (rettv_list_alloc(rettv) == FAIL)
7956 return;
7957
7958 if (searchpair_cmn(argvars, &match_pos) > 0)
7959 {
7960 lnum = match_pos.lnum;
7961 col = match_pos.col;
7962 }
7963
7964 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
7965 list_append_number(rettv->vval.v_list, (varnumber_T)col);
7966}
7967
7968/*
7969 * Search for a start/middle/end thing.
7970 * Used by searchpair(), see its documentation for the details.
7971 * Returns 0 or -1 for no match,
7972 */
7973 long
7974do_searchpair(
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007975 char_u *spat, // start pattern
7976 char_u *mpat, // middle pattern
7977 char_u *epat, // end pattern
7978 int dir, // BACKWARD or FORWARD
7979 typval_T *skip, // skip expression
7980 int flags, // SP_SETPCMARK and other SP_ values
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007981 pos_T *match_pos,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007982 linenr_T lnum_stop, // stop at this line if not zero
7983 long time_limit UNUSED) // stop after this many msec
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007984{
7985 char_u *save_cpo;
7986 char_u *pat, *pat2 = NULL, *pat3 = NULL;
7987 long retval = 0;
7988 pos_T pos;
7989 pos_T firstpos;
7990 pos_T foundpos;
7991 pos_T save_cursor;
7992 pos_T save_pos;
7993 int n;
7994 int r;
7995 int nest = 1;
Bram Moolenaar48570482017-10-30 21:48:41 +01007996 int use_skip = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007997 int err;
7998 int options = SEARCH_KEEP;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007999#ifdef FEAT_RELTIME
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008000 proftime_T tm;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02008001#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008002
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008003 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008004 save_cpo = p_cpo;
8005 p_cpo = empty_option;
8006
8007#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008008 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008009 profile_setlimit(time_limit, &tm);
8010#endif
8011
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008012 // Make two search patterns: start/end (pat2, for in nested pairs) and
8013 // start/middle/end (pat3, for the top pair).
Bram Moolenaar964b3742019-05-24 18:54:09 +02008014 pat2 = alloc(STRLEN(spat) + STRLEN(epat) + 17);
8015 pat3 = alloc(STRLEN(spat) + STRLEN(mpat) + STRLEN(epat) + 25);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008016 if (pat2 == NULL || pat3 == NULL)
8017 goto theend;
Bram Moolenaar6e450a52017-01-06 20:03:58 +01008018 sprintf((char *)pat2, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)", spat, epat);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008019 if (*mpat == NUL)
8020 STRCPY(pat3, pat2);
8021 else
Bram Moolenaar6e450a52017-01-06 20:03:58 +01008022 sprintf((char *)pat3, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)\\|\\(%s\\m\\)",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008023 spat, epat, mpat);
8024 if (flags & SP_START)
8025 options |= SEARCH_START;
8026
Bram Moolenaar48570482017-10-30 21:48:41 +01008027 if (skip != NULL)
Bram Moolenaara9c01042020-06-07 14:50:50 +02008028 use_skip = eval_expr_valid_arg(skip);
Bram Moolenaar48570482017-10-30 21:48:41 +01008029
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008030 save_cursor = curwin->w_cursor;
8031 pos = curwin->w_cursor;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01008032 CLEAR_POS(&firstpos);
8033 CLEAR_POS(&foundpos);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008034 pat = pat3;
8035 for (;;)
8036 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02008037 searchit_arg_T sia;
8038
Bram Moolenaara80faa82020-04-12 19:37:17 +02008039 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02008040 sia.sa_stop_lnum = lnum_stop;
8041#ifdef FEAT_RELTIME
8042 sia.sa_tm = &tm;
8043#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01008044 n = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02008045 options, RE_SEARCH, &sia);
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01008046 if (n == FAIL || (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos)))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008047 // didn't find it or found the first match again: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008048 break;
8049
8050 if (firstpos.lnum == 0)
8051 firstpos = pos;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01008052 if (EQUAL_POS(pos, foundpos))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008053 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008054 // Found the same position again. Can happen with a pattern that
8055 // has "\zs" at the end and searching backwards. Advance one
8056 // character and try again.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008057 if (dir == BACKWARD)
8058 decl(&pos);
8059 else
8060 incl(&pos);
8061 }
8062 foundpos = pos;
8063
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008064 // clear the start flag to avoid getting stuck here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008065 options &= ~SEARCH_START;
8066
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008067 // If the skip pattern matches, ignore this match.
Bram Moolenaar48570482017-10-30 21:48:41 +01008068 if (use_skip)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008069 {
8070 save_pos = curwin->w_cursor;
8071 curwin->w_cursor = pos;
Bram Moolenaar48570482017-10-30 21:48:41 +01008072 err = FALSE;
8073 r = eval_expr_to_bool(skip, &err);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008074 curwin->w_cursor = save_pos;
8075 if (err)
8076 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008077 // Evaluating {skip} caused an error, break here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008078 curwin->w_cursor = save_cursor;
8079 retval = -1;
8080 break;
8081 }
8082 if (r)
8083 continue;
8084 }
8085
8086 if ((dir == BACKWARD && n == 3) || (dir == FORWARD && n == 2))
8087 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008088 // Found end when searching backwards or start when searching
8089 // forward: nested pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008090 ++nest;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008091 pat = pat2; // nested, don't search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008092 }
8093 else
8094 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008095 // Found end when searching forward or start when searching
8096 // backward: end of (nested) pair; or found middle in outer pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008097 if (--nest == 1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008098 pat = pat3; // outer level, search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008099 }
8100
8101 if (nest == 0)
8102 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008103 // Found the match: return matchcount or line number.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008104 if (flags & SP_RETCOUNT)
8105 ++retval;
8106 else
8107 retval = pos.lnum;
8108 if (flags & SP_SETPCMARK)
8109 setpcmark();
8110 curwin->w_cursor = pos;
8111 if (!(flags & SP_REPEAT))
8112 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008113 nest = 1; // search for next unmatched
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008114 }
8115 }
8116
8117 if (match_pos != NULL)
8118 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008119 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008120 match_pos->lnum = curwin->w_cursor.lnum;
8121 match_pos->col = curwin->w_cursor.col + 1;
8122 }
8123
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008124 // If 'n' flag is used or search failed: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008125 if ((flags & SP_NOMOVE) || retval == 0)
8126 curwin->w_cursor = save_cursor;
8127
8128theend:
8129 vim_free(pat2);
8130 vim_free(pat3);
8131 if (p_cpo == empty_option)
8132 p_cpo = save_cpo;
8133 else
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008134 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008135 // Darn, evaluating the {skip} expression changed the value.
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008136 // If it's still empty it was changed and restored, need to restore in
8137 // the complicated way.
8138 if (*p_cpo == NUL)
8139 set_option_value((char_u *)"cpo", 0L, save_cpo, 0);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008140 free_string_option(save_cpo);
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008141 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008142
8143 return retval;
8144}
8145
8146/*
8147 * "searchpos()" function
8148 */
8149 static void
8150f_searchpos(typval_T *argvars, typval_T *rettv)
8151{
8152 pos_T match_pos;
8153 int lnum = 0;
8154 int col = 0;
8155 int n;
8156 int flags = 0;
8157
8158 if (rettv_list_alloc(rettv) == FAIL)
8159 return;
8160
8161 n = search_cmn(argvars, &match_pos, &flags);
8162 if (n > 0)
8163 {
8164 lnum = match_pos.lnum;
8165 col = match_pos.col;
8166 }
8167
8168 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
8169 list_append_number(rettv->vval.v_list, (varnumber_T)col);
8170 if (flags & SP_SUBPAT)
8171 list_append_number(rettv->vval.v_list, (varnumber_T)n);
8172}
8173
Bram Moolenaar6f02b002021-01-10 20:22:54 +01008174/*
8175 * Set the cursor or mark position.
8176 * If 'charpos' is TRUE, then use the column number as a character offet.
8177 * Otherwise use the column number as a byte offset.
8178 */
8179 static void
8180set_position(typval_T *argvars, typval_T *rettv, int charpos)
8181{
8182 pos_T pos;
8183 int fnum;
8184 char_u *name;
8185 colnr_T curswant = -1;
8186
8187 rettv->vval.v_number = -1;
8188
8189 name = tv_get_string_chk(argvars);
8190 if (name != NULL)
8191 {
8192 if (list2fpos(&argvars[1], &pos, &fnum, &curswant, charpos) == OK)
8193 {
8194 if (pos.col != MAXCOL && --pos.col < 0)
8195 pos.col = 0;
8196 if ((name[0] == '.' && name[1] == NUL))
8197 {
8198 // set cursor; "fnum" is ignored
8199 curwin->w_cursor = pos;
8200 if (curswant >= 0)
8201 {
8202 curwin->w_curswant = curswant - 1;
8203 curwin->w_set_curswant = FALSE;
8204 }
8205 check_cursor();
8206 rettv->vval.v_number = 0;
8207 }
8208 else if (name[0] == '\'' && name[1] != NUL && name[2] == NUL)
8209 {
8210 // set mark
8211 if (setmark_pos(name[1], &pos, fnum) == OK)
8212 rettv->vval.v_number = 0;
8213 }
8214 else
8215 emsg(_(e_invarg));
8216 }
8217 }
8218}
8219/*
8220 * "setcharpos()" function
8221 */
8222 static void
8223f_setcharpos(typval_T *argvars, typval_T *rettv)
8224{
8225 set_position(argvars, rettv, TRUE);
8226}
8227
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008228 static void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008229f_setcharsearch(typval_T *argvars, typval_T *rettv UNUSED)
8230{
8231 dict_T *d;
8232 dictitem_T *di;
8233 char_u *csearch;
8234
8235 if (argvars[0].v_type != VAR_DICT)
8236 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008237 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008238 return;
8239 }
8240
8241 if ((d = argvars[0].vval.v_dict) != NULL)
8242 {
Bram Moolenaar8f667172018-12-14 15:38:31 +01008243 csearch = dict_get_string(d, (char_u *)"char", FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008244 if (csearch != NULL)
8245 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008246 if (enc_utf8)
8247 {
8248 int pcc[MAX_MCO];
8249 int c = utfc_ptr2char(csearch, pcc);
8250
8251 set_last_csearch(c, csearch, utfc_ptr2len(csearch));
8252 }
8253 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008254 set_last_csearch(PTR2CHAR(csearch),
Bram Moolenaar1614a142019-10-06 22:00:13 +02008255 csearch, mb_ptr2len(csearch));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008256 }
8257
8258 di = dict_find(d, (char_u *)"forward", -1);
8259 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008260 set_csearch_direction((int)tv_get_number(&di->di_tv)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008261 ? FORWARD : BACKWARD);
8262
8263 di = dict_find(d, (char_u *)"until", -1);
8264 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008265 set_csearch_until(!!tv_get_number(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008266 }
8267}
8268
8269/*
Bram Moolenaar6f02b002021-01-10 20:22:54 +01008270 * "setcursorcharpos" function
8271 */
8272 static void
8273f_setcursorcharpos(typval_T *argvars, typval_T *rettv UNUSED)
8274{
8275 set_cursorpos(argvars, rettv, TRUE);
8276}
8277
8278/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02008279 * "setenv()" function
8280 */
8281 static void
8282f_setenv(typval_T *argvars, typval_T *rettv UNUSED)
8283{
8284 char_u namebuf[NUMBUFLEN];
8285 char_u valbuf[NUMBUFLEN];
8286 char_u *name = tv_get_string_buf(&argvars[0], namebuf);
8287
8288 if (argvars[1].v_type == VAR_SPECIAL
8289 && argvars[1].vval.v_number == VVAL_NULL)
8290 vim_unsetenv(name);
8291 else
8292 vim_setenv(name, tv_get_string_buf(&argvars[1], valbuf));
8293}
8294
8295/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008296 * "setfperm({fname}, {mode})" function
8297 */
8298 static void
8299f_setfperm(typval_T *argvars, typval_T *rettv)
8300{
8301 char_u *fname;
8302 char_u modebuf[NUMBUFLEN];
8303 char_u *mode_str;
8304 int i;
8305 int mask;
8306 int mode = 0;
8307
8308 rettv->vval.v_number = 0;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008309 fname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008310 if (fname == NULL)
8311 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008312 mode_str = tv_get_string_buf_chk(&argvars[1], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008313 if (mode_str == NULL)
8314 return;
8315 if (STRLEN(mode_str) != 9)
8316 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008317 semsg(_(e_invarg2), mode_str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008318 return;
8319 }
8320
8321 mask = 1;
8322 for (i = 8; i >= 0; --i)
8323 {
8324 if (mode_str[i] != '-')
8325 mode |= mask;
8326 mask = mask << 1;
8327 }
8328 rettv->vval.v_number = mch_setperm(fname, mode) == OK;
8329}
8330
8331/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008332 * "setpos()" function
8333 */
8334 static void
8335f_setpos(typval_T *argvars, typval_T *rettv)
8336{
Bram Moolenaar6f02b002021-01-10 20:22:54 +01008337 set_position(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008338}
8339
8340/*
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008341 * Translate a register type string to the yank type and block length
8342 */
8343 static int
8344get_yank_type(char_u **pp, char_u *yank_type, long *block_len)
8345{
8346 char_u *stropt = *pp;
8347 switch (*stropt)
8348 {
8349 case 'v': case 'c': // character-wise selection
8350 *yank_type = MCHAR;
8351 break;
8352 case 'V': case 'l': // line-wise selection
8353 *yank_type = MLINE;
8354 break;
8355 case 'b': case Ctrl_V: // block-wise selection
8356 *yank_type = MBLOCK;
8357 if (VIM_ISDIGIT(stropt[1]))
8358 {
8359 ++stropt;
8360 *block_len = getdigits(&stropt) - 1;
8361 --stropt;
8362 }
8363 break;
8364 default:
8365 return FAIL;
8366 }
8367 *pp = stropt;
8368 return OK;
8369}
8370
8371/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008372 * "setreg()" function
8373 */
8374 static void
8375f_setreg(typval_T *argvars, typval_T *rettv)
8376{
8377 int regname;
8378 char_u *strregname;
8379 char_u *stropt;
8380 char_u *strval;
8381 int append;
8382 char_u yank_type;
8383 long block_len;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008384 typval_T *regcontents;
8385 int pointreg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008386
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008387 pointreg = 0;
8388 regcontents = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008389 block_len = -1;
8390 yank_type = MAUTO;
8391 append = FALSE;
8392
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008393 strregname = tv_get_string_chk(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008394 rettv->vval.v_number = 1; // FAIL is default
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008395
8396 if (strregname == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008397 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008398 regname = *strregname;
8399 if (regname == 0 || regname == '@')
8400 regname = '"';
8401
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008402 if (argvars[1].v_type == VAR_DICT)
8403 {
8404 dict_T *d = argvars[1].vval.v_dict;
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008405 dictitem_T *di;
8406
8407 if (d == NULL || d->dv_hashtab.ht_used == 0)
8408 {
8409 // Empty dict, clear the register (like setreg(0, []))
8410 char_u *lstval[2] = {NULL, NULL};
8411 write_reg_contents_lst(regname, lstval, 0, FALSE, MAUTO, -1);
8412 return;
8413 }
8414
8415 di = dict_find(d, (char_u *)"regcontents", -1);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008416 if (di != NULL)
8417 regcontents = &di->di_tv;
8418
8419 stropt = dict_get_string(d, (char_u *)"regtype", FALSE);
8420 if (stropt != NULL)
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008421 {
8422 int ret = get_yank_type(&stropt, &yank_type, &block_len);
8423
8424 if (ret == FAIL || *++stropt != NUL)
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008425 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008426 semsg(_(e_invargval), "value");
8427 return;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008428 }
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008429 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008430
8431 if (regname == '"')
8432 {
8433 stropt = dict_get_string(d, (char_u *)"points_to", FALSE);
8434 if (stropt != NULL)
8435 {
8436 pointreg = *stropt;
8437 regname = pointreg;
8438 }
8439 }
Bram Moolenaar6a950582020-08-28 16:39:33 +02008440 else if (dict_get_bool(d, (char_u *)"isunnamed", -1) > 0)
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008441 pointreg = regname;
8442 }
8443 else
8444 regcontents = &argvars[1];
8445
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008446 if (argvars[2].v_type != VAR_UNKNOWN)
8447 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008448 if (yank_type != MAUTO)
8449 {
8450 semsg(_(e_toomanyarg), "setreg");
8451 return;
8452 }
8453
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008454 stropt = tv_get_string_chk(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008455 if (stropt == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008456 return; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008457 for (; *stropt != NUL; ++stropt)
8458 switch (*stropt)
8459 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008460 case 'a': case 'A': // append
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008461 append = TRUE;
8462 break;
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008463 default:
8464 get_yank_type(&stropt, &yank_type, &block_len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008465 }
8466 }
8467
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008468 if (regcontents && regcontents->v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008469 {
8470 char_u **lstval;
8471 char_u **allocval;
8472 char_u buf[NUMBUFLEN];
8473 char_u **curval;
8474 char_u **curallocval;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008475 list_T *ll = regcontents->vval.v_list;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008476 listitem_T *li;
8477 int len;
8478
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008479 // If the list is NULL handle like an empty list.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008480 len = ll == NULL ? 0 : ll->lv_len;
8481
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008482 // First half: use for pointers to result lines; second half: use for
8483 // pointers to allocated copies.
Bram Moolenaarc799fe22019-05-28 23:08:19 +02008484 lstval = ALLOC_MULT(char_u *, (len + 1) * 2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008485 if (lstval == NULL)
8486 return;
8487 curval = lstval;
8488 allocval = lstval + len + 2;
8489 curallocval = allocval;
8490
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008491 if (ll != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008492 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02008493 CHECK_LIST_MATERIALIZE(ll);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02008494 FOR_ALL_LIST_ITEMS(ll, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008495 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008496 strval = tv_get_string_buf_chk(&li->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008497 if (strval == NULL)
8498 goto free_lstval;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008499 if (strval == buf)
8500 {
8501 // Need to make a copy, next tv_get_string_buf_chk() will
8502 // overwrite the string.
8503 strval = vim_strsave(buf);
8504 if (strval == NULL)
8505 goto free_lstval;
8506 *curallocval++ = strval;
8507 }
8508 *curval++ = strval;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008509 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008510 }
8511 *curval++ = NULL;
8512
8513 write_reg_contents_lst(regname, lstval, -1,
8514 append, yank_type, block_len);
8515free_lstval:
8516 while (curallocval > allocval)
8517 vim_free(*--curallocval);
8518 vim_free(lstval);
8519 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008520 else if (regcontents)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008521 {
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008522 strval = tv_get_string_chk(regcontents);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008523 if (strval == NULL)
8524 return;
8525 write_reg_contents_ex(regname, strval, -1,
8526 append, yank_type, block_len);
8527 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008528 if (pointreg != 0)
8529 get_yank_register(pointreg, TRUE);
8530
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008531 rettv->vval.v_number = 0;
8532}
8533
8534/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008535 * "settagstack()" function
8536 */
8537 static void
8538f_settagstack(typval_T *argvars, typval_T *rettv)
8539{
8540 static char *e_invact2 = N_("E962: Invalid action: '%s'");
8541 win_T *wp;
8542 dict_T *d;
8543 int action = 'r';
8544
8545 rettv->vval.v_number = -1;
8546
8547 // first argument: window number or id
8548 wp = find_win_by_nr_or_id(&argvars[0]);
8549 if (wp == NULL)
8550 return;
8551
8552 // second argument: dict with items to set in the tag stack
8553 if (argvars[1].v_type != VAR_DICT)
8554 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008555 emsg(_(e_dictreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008556 return;
8557 }
8558 d = argvars[1].vval.v_dict;
8559 if (d == NULL)
8560 return;
8561
8562 // third argument: action - 'a' for append and 'r' for replace.
8563 // default is to replace the stack.
8564 if (argvars[2].v_type == VAR_UNKNOWN)
8565 action = 'r';
8566 else if (argvars[2].v_type == VAR_STRING)
8567 {
8568 char_u *actstr;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008569 actstr = tv_get_string_chk(&argvars[2]);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008570 if (actstr == NULL)
8571 return;
Bram Moolenaar271fa082020-01-02 14:02:16 +01008572 if ((*actstr == 'r' || *actstr == 'a' || *actstr == 't')
8573 && actstr[1] == NUL)
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008574 action = *actstr;
8575 else
8576 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008577 semsg(_(e_invact2), actstr);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008578 return;
8579 }
8580 }
8581 else
8582 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008583 emsg(_(e_stringreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008584 return;
8585 }
8586
8587 if (set_tagstack(wp, d, action) == OK)
8588 rettv->vval.v_number = 0;
8589}
8590
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008591#ifdef FEAT_CRYPT
8592/*
8593 * "sha256({string})" function
8594 */
8595 static void
8596f_sha256(typval_T *argvars, typval_T *rettv)
8597{
8598 char_u *p;
8599
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008600 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008601 rettv->vval.v_string = vim_strsave(
8602 sha256_bytes(p, (int)STRLEN(p), NULL, 0));
8603 rettv->v_type = VAR_STRING;
8604}
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008605#endif // FEAT_CRYPT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008606
8607/*
8608 * "shellescape({string})" function
8609 */
8610 static void
8611f_shellescape(typval_T *argvars, typval_T *rettv)
8612{
Bram Moolenaar20615522017-06-05 18:46:26 +02008613 int do_special = non_zero_arg(&argvars[1]);
8614
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008615 rettv->vval.v_string = vim_strsave_shellescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008616 tv_get_string(&argvars[0]), do_special, do_special);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008617 rettv->v_type = VAR_STRING;
8618}
8619
8620/*
8621 * shiftwidth() function
8622 */
8623 static void
8624f_shiftwidth(typval_T *argvars UNUSED, typval_T *rettv)
8625{
Bram Moolenaarf9514162018-11-22 03:08:29 +01008626 rettv->vval.v_number = 0;
8627
8628 if (argvars[0].v_type != VAR_UNKNOWN)
8629 {
8630 long col;
8631
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008632 col = (long)tv_get_number_chk(argvars, NULL);
Bram Moolenaarf9514162018-11-22 03:08:29 +01008633 if (col < 0)
8634 return; // type error; errmsg already given
8635#ifdef FEAT_VARTABS
8636 rettv->vval.v_number = get_sw_value_col(curbuf, col);
8637 return;
8638#endif
8639 }
8640
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008641 rettv->vval.v_number = get_sw_value(curbuf);
8642}
8643
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008644#ifdef FEAT_FLOAT
8645/*
8646 * "sin()" function
8647 */
8648 static void
8649f_sin(typval_T *argvars, typval_T *rettv)
8650{
8651 float_T f = 0.0;
8652
8653 rettv->v_type = VAR_FLOAT;
8654 if (get_float_arg(argvars, &f) == OK)
8655 rettv->vval.v_float = sin(f);
8656 else
8657 rettv->vval.v_float = 0.0;
8658}
8659
8660/*
8661 * "sinh()" function
8662 */
8663 static void
8664f_sinh(typval_T *argvars, typval_T *rettv)
8665{
8666 float_T f = 0.0;
8667
8668 rettv->v_type = VAR_FLOAT;
8669 if (get_float_arg(argvars, &f) == OK)
8670 rettv->vval.v_float = sinh(f);
8671 else
8672 rettv->vval.v_float = 0.0;
8673}
8674#endif
8675
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008676/*
8677 * "soundfold({word})" function
8678 */
8679 static void
8680f_soundfold(typval_T *argvars, typval_T *rettv)
8681{
8682 char_u *s;
8683
8684 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008685 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008686#ifdef FEAT_SPELL
8687 rettv->vval.v_string = eval_soundfold(s);
8688#else
8689 rettv->vval.v_string = vim_strsave(s);
8690#endif
8691}
8692
8693/*
8694 * "spellbadword()" function
8695 */
8696 static void
8697f_spellbadword(typval_T *argvars UNUSED, typval_T *rettv)
8698{
8699 char_u *word = (char_u *)"";
8700 hlf_T attr = HLF_COUNT;
8701 int len = 0;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008702#ifdef FEAT_SPELL
8703 int wo_spell_save = curwin->w_p_spell;
8704
8705 if (!curwin->w_p_spell)
8706 {
8707 did_set_spelllang(curwin);
8708 curwin->w_p_spell = TRUE;
8709 }
8710
8711 if (*curwin->w_s->b_p_spl == NUL)
8712 {
8713 emsg(_(e_no_spell));
8714 curwin->w_p_spell = wo_spell_save;
8715 return;
8716 }
8717#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008718
8719 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008720 {
8721#ifdef FEAT_SPELL
8722 curwin->w_p_spell = wo_spell_save;
8723#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008724 return;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008725 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008726
8727#ifdef FEAT_SPELL
8728 if (argvars[0].v_type == VAR_UNKNOWN)
8729 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008730 // Find the start and length of the badly spelled word.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008731 len = spell_move_to(curwin, FORWARD, TRUE, TRUE, &attr);
8732 if (len != 0)
Bram Moolenaarb73fa622017-12-21 20:27:47 +01008733 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008734 word = ml_get_cursor();
Bram Moolenaarb73fa622017-12-21 20:27:47 +01008735 curwin->w_set_curswant = TRUE;
8736 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008737 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008738 else if (*curbuf->b_s.b_p_spl != NUL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008739 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008740 char_u *str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008741 int capcol = -1;
8742
8743 if (str != NULL)
8744 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008745 // Check the argument for spelling.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008746 while (*str != NUL)
8747 {
8748 len = spell_check(curwin, str, &attr, &capcol, FALSE);
8749 if (attr != HLF_COUNT)
8750 {
8751 word = str;
8752 break;
8753 }
8754 str += len;
Bram Moolenaar66ab9162018-07-20 20:28:48 +02008755 capcol -= len;
Bram Moolenaar0c779e82019-08-09 17:01:02 +02008756 len = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008757 }
8758 }
8759 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008760 curwin->w_p_spell = wo_spell_save;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008761#endif
8762
8763 list_append_string(rettv->vval.v_list, word, len);
8764 list_append_string(rettv->vval.v_list, (char_u *)(
8765 attr == HLF_SPB ? "bad" :
8766 attr == HLF_SPR ? "rare" :
8767 attr == HLF_SPL ? "local" :
8768 attr == HLF_SPC ? "caps" :
8769 ""), -1);
8770}
8771
8772/*
8773 * "spellsuggest()" function
8774 */
8775 static void
8776f_spellsuggest(typval_T *argvars UNUSED, typval_T *rettv)
8777{
8778#ifdef FEAT_SPELL
8779 char_u *str;
8780 int typeerr = FALSE;
8781 int maxcount;
8782 garray_T ga;
8783 int i;
8784 listitem_T *li;
8785 int need_capital = FALSE;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008786 int wo_spell_save = curwin->w_p_spell;
8787
8788 if (!curwin->w_p_spell)
8789 {
8790 did_set_spelllang(curwin);
8791 curwin->w_p_spell = TRUE;
8792 }
8793
8794 if (*curwin->w_s->b_p_spl == NUL)
8795 {
8796 emsg(_(e_no_spell));
8797 curwin->w_p_spell = wo_spell_save;
8798 return;
8799 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008800#endif
8801
8802 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008803 {
8804#ifdef FEAT_SPELL
8805 curwin->w_p_spell = wo_spell_save;
8806#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008807 return;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008808 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008809
8810#ifdef FEAT_SPELL
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008811 if (*curwin->w_s->b_p_spl != NUL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008812 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008813 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008814 if (argvars[1].v_type != VAR_UNKNOWN)
8815 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008816 maxcount = (int)tv_get_number_chk(&argvars[1], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008817 if (maxcount <= 0)
8818 return;
8819 if (argvars[2].v_type != VAR_UNKNOWN)
8820 {
Bram Moolenaar7c27f332020-09-05 22:45:55 +02008821 need_capital = (int)tv_get_bool_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008822 if (typeerr)
8823 return;
8824 }
8825 }
8826 else
8827 maxcount = 25;
8828
8829 spell_suggest_list(&ga, str, maxcount, need_capital, FALSE);
8830
8831 for (i = 0; i < ga.ga_len; ++i)
8832 {
8833 str = ((char_u **)ga.ga_data)[i];
8834
8835 li = listitem_alloc();
8836 if (li == NULL)
8837 vim_free(str);
8838 else
8839 {
8840 li->li_tv.v_type = VAR_STRING;
8841 li->li_tv.v_lock = 0;
8842 li->li_tv.vval.v_string = str;
8843 list_append(rettv->vval.v_list, li);
8844 }
8845 }
8846 ga_clear(&ga);
8847 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008848 curwin->w_p_spell = wo_spell_save;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008849#endif
8850}
8851
8852 static void
8853f_split(typval_T *argvars, typval_T *rettv)
8854{
8855 char_u *str;
8856 char_u *end;
8857 char_u *pat = NULL;
8858 regmatch_T regmatch;
8859 char_u patbuf[NUMBUFLEN];
8860 char_u *save_cpo;
8861 int match;
8862 colnr_T col = 0;
8863 int keepempty = FALSE;
8864 int typeerr = FALSE;
8865
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008866 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008867 save_cpo = p_cpo;
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008868 p_cpo = empty_option;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008869
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008870 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008871 if (argvars[1].v_type != VAR_UNKNOWN)
8872 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008873 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008874 if (pat == NULL)
8875 typeerr = TRUE;
8876 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar3986b942020-09-06 16:09:04 +02008877 keepempty = (int)tv_get_bool_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008878 }
8879 if (pat == NULL || *pat == NUL)
8880 pat = (char_u *)"[\\x01- ]\\+";
8881
8882 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008883 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008884 if (typeerr)
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008885 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008886
8887 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
8888 if (regmatch.regprog != NULL)
8889 {
8890 regmatch.rm_ic = FALSE;
8891 while (*str != NUL || keepempty)
8892 {
8893 if (*str == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008894 match = FALSE; // empty item at the end
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008895 else
8896 match = vim_regexec_nl(&regmatch, str, col);
8897 if (match)
8898 end = regmatch.startp[0];
8899 else
8900 end = str + STRLEN(str);
8901 if (keepempty || end > str || (rettv->vval.v_list->lv_len > 0
8902 && *str != NUL && match && end < regmatch.endp[0]))
8903 {
8904 if (list_append_string(rettv->vval.v_list, str,
8905 (int)(end - str)) == FAIL)
8906 break;
8907 }
8908 if (!match)
8909 break;
Bram Moolenaar13505972019-01-24 15:04:48 +01008910 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008911 if (regmatch.endp[0] > str)
8912 col = 0;
8913 else
Bram Moolenaar13505972019-01-24 15:04:48 +01008914 // Don't get stuck at the same match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008915 col = (*mb_ptr2len)(regmatch.endp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008916 str = regmatch.endp[0];
8917 }
8918
8919 vim_regfree(regmatch.regprog);
8920 }
8921
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008922theend:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008923 p_cpo = save_cpo;
8924}
8925
8926#ifdef FEAT_FLOAT
8927/*
8928 * "sqrt()" function
8929 */
8930 static void
8931f_sqrt(typval_T *argvars, typval_T *rettv)
8932{
8933 float_T f = 0.0;
8934
8935 rettv->v_type = VAR_FLOAT;
8936 if (get_float_arg(argvars, &f) == OK)
8937 rettv->vval.v_float = sqrt(f);
8938 else
8939 rettv->vval.v_float = 0.0;
8940}
Bram Moolenaar0387cae2019-11-29 21:07:58 +01008941#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008942
Bram Moolenaar0387cae2019-11-29 21:07:58 +01008943#ifdef FEAT_FLOAT
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01008944/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008945 * "str2float()" function
8946 */
8947 static void
8948f_str2float(typval_T *argvars, typval_T *rettv)
8949{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008950 char_u *p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01008951 int isneg = (*p == '-');
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008952
Bram Moolenaar08243d22017-01-10 16:12:29 +01008953 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008954 p = skipwhite(p + 1);
8955 (void)string2float(p, &rettv->vval.v_float);
Bram Moolenaar08243d22017-01-10 16:12:29 +01008956 if (isneg)
8957 rettv->vval.v_float *= -1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008958 rettv->v_type = VAR_FLOAT;
8959}
8960#endif
8961
8962/*
Bram Moolenaar9d401282019-04-06 13:18:12 +02008963 * "str2list()" function
8964 */
8965 static void
8966f_str2list(typval_T *argvars, typval_T *rettv)
8967{
8968 char_u *p;
8969 int utf8 = FALSE;
8970
8971 if (rettv_list_alloc(rettv) == FAIL)
8972 return;
8973
8974 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaara48f7862020-09-05 20:16:57 +02008975 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar9d401282019-04-06 13:18:12 +02008976
8977 p = tv_get_string(&argvars[0]);
8978
8979 if (has_mbyte || utf8)
8980 {
8981 int (*ptr2len)(char_u *);
8982 int (*ptr2char)(char_u *);
8983
8984 if (utf8 || enc_utf8)
8985 {
8986 ptr2len = utf_ptr2len;
8987 ptr2char = utf_ptr2char;
8988 }
8989 else
8990 {
8991 ptr2len = mb_ptr2len;
8992 ptr2char = mb_ptr2char;
8993 }
8994
8995 for ( ; *p != NUL; p += (*ptr2len)(p))
8996 list_append_number(rettv->vval.v_list, (*ptr2char)(p));
8997 }
8998 else
8999 for ( ; *p != NUL; ++p)
9000 list_append_number(rettv->vval.v_list, *p);
9001}
9002
9003/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009004 * "str2nr()" function
9005 */
9006 static void
9007f_str2nr(typval_T *argvars, typval_T *rettv)
9008{
9009 int base = 10;
9010 char_u *p;
9011 varnumber_T n;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02009012 int what = 0;
Bram Moolenaar08243d22017-01-10 16:12:29 +01009013 int isneg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009014
9015 if (argvars[1].v_type != VAR_UNKNOWN)
9016 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009017 base = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009018 if (base != 2 && base != 8 && base != 10 && base != 16)
9019 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009020 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009021 return;
9022 }
Bram Moolenaar3986b942020-09-06 16:09:04 +02009023 if (argvars[2].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[2]))
Bram Moolenaar60a8de22019-09-15 14:33:22 +02009024 what |= STR2NR_QUOTE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009025 }
9026
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009027 p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01009028 isneg = (*p == '-');
9029 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009030 p = skipwhite(p + 1);
9031 switch (base)
9032 {
Bram Moolenaar60a8de22019-09-15 14:33:22 +02009033 case 2: what |= STR2NR_BIN + STR2NR_FORCE; break;
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02009034 case 8: what |= STR2NR_OCT + STR2NR_OOCT + STR2NR_FORCE; break;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02009035 case 16: what |= STR2NR_HEX + STR2NR_FORCE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009036 }
Bram Moolenaar16e9b852019-05-19 19:59:35 +02009037 vim_str2nr(p, NULL, NULL, what, &n, NULL, 0, FALSE);
9038 // Text after the number is silently ignored.
Bram Moolenaar08243d22017-01-10 16:12:29 +01009039 if (isneg)
9040 rettv->vval.v_number = -n;
9041 else
9042 rettv->vval.v_number = n;
9043
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009044}
9045
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009046/*
9047 * "strgetchar()" function
9048 */
9049 static void
9050f_strgetchar(typval_T *argvars, typval_T *rettv)
9051{
9052 char_u *str;
9053 int len;
9054 int error = FALSE;
9055 int charidx;
Bram Moolenaar13505972019-01-24 15:04:48 +01009056 int byteidx = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009057
9058 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009059 str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009060 if (str == NULL)
9061 return;
9062 len = (int)STRLEN(str);
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009063 charidx = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009064 if (error)
9065 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009066
Bram Moolenaar13505972019-01-24 15:04:48 +01009067 while (charidx >= 0 && byteidx < len)
9068 {
9069 if (charidx == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009070 {
Bram Moolenaar13505972019-01-24 15:04:48 +01009071 rettv->vval.v_number = mb_ptr2char(str + byteidx);
9072 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009073 }
Bram Moolenaar13505972019-01-24 15:04:48 +01009074 --charidx;
9075 byteidx += MB_CPTR2LEN(str + byteidx);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009076 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009077}
9078
9079/*
9080 * "stridx()" function
9081 */
9082 static void
9083f_stridx(typval_T *argvars, typval_T *rettv)
9084{
9085 char_u buf[NUMBUFLEN];
9086 char_u *needle;
9087 char_u *haystack;
9088 char_u *save_haystack;
9089 char_u *pos;
9090 int start_idx;
9091
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009092 needle = tv_get_string_chk(&argvars[1]);
9093 save_haystack = haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009094 rettv->vval.v_number = -1;
9095 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009096 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009097
9098 if (argvars[2].v_type != VAR_UNKNOWN)
9099 {
9100 int error = FALSE;
9101
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009102 start_idx = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009103 if (error || start_idx >= (int)STRLEN(haystack))
9104 return;
9105 if (start_idx >= 0)
9106 haystack += start_idx;
9107 }
9108
9109 pos = (char_u *)strstr((char *)haystack, (char *)needle);
9110 if (pos != NULL)
9111 rettv->vval.v_number = (varnumber_T)(pos - save_haystack);
9112}
9113
9114/*
9115 * "string()" function
9116 */
Bram Moolenaar461a7fc2018-12-22 13:28:07 +01009117 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009118f_string(typval_T *argvars, typval_T *rettv)
9119{
9120 char_u *tofree;
9121 char_u numbuf[NUMBUFLEN];
9122
9123 rettv->v_type = VAR_STRING;
9124 rettv->vval.v_string = tv2string(&argvars[0], &tofree, numbuf,
9125 get_copyID());
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009126 // Make a copy if we have a value but it's not in allocated memory.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009127 if (rettv->vval.v_string != NULL && tofree == NULL)
9128 rettv->vval.v_string = vim_strsave(rettv->vval.v_string);
9129}
9130
9131/*
9132 * "strlen()" function
9133 */
9134 static void
9135f_strlen(typval_T *argvars, typval_T *rettv)
9136{
9137 rettv->vval.v_number = (varnumber_T)(STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009138 tv_get_string(&argvars[0])));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009139}
9140
9141/*
9142 * "strchars()" function
9143 */
9144 static void
9145f_strchars(typval_T *argvars, typval_T *rettv)
9146{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009147 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar3986b942020-09-06 16:09:04 +02009148 int skipcc = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009149 varnumber_T len = 0;
9150 int (*func_mb_ptr2char_adv)(char_u **pp);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009151
9152 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar3986b942020-09-06 16:09:04 +02009153 skipcc = (int)tv_get_bool(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009154 if (skipcc < 0 || skipcc > 1)
Bram Moolenaarbade44e2020-09-26 22:39:24 +02009155 semsg(_(e_using_number_as_bool_nr), skipcc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009156 else
9157 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009158 func_mb_ptr2char_adv = skipcc ? mb_ptr2char_adv : mb_cptr2char_adv;
9159 while (*s != NUL)
9160 {
9161 func_mb_ptr2char_adv(&s);
9162 ++len;
9163 }
9164 rettv->vval.v_number = len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009165 }
9166}
9167
9168/*
9169 * "strdisplaywidth()" function
9170 */
9171 static void
9172f_strdisplaywidth(typval_T *argvars, typval_T *rettv)
9173{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009174 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009175 int col = 0;
9176
9177 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009178 col = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009179
9180 rettv->vval.v_number = (varnumber_T)(linetabsize_col(col, s) - col);
9181}
9182
9183/*
9184 * "strwidth()" function
9185 */
9186 static void
9187f_strwidth(typval_T *argvars, typval_T *rettv)
9188{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009189 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009190
Bram Moolenaar13505972019-01-24 15:04:48 +01009191 rettv->vval.v_number = (varnumber_T)(mb_string2cells(s, -1));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009192}
9193
9194/*
9195 * "strcharpart()" function
9196 */
9197 static void
9198f_strcharpart(typval_T *argvars, typval_T *rettv)
9199{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009200 char_u *p;
9201 int nchar;
9202 int nbyte = 0;
9203 int charlen;
9204 int len = 0;
9205 int slen;
9206 int error = FALSE;
9207
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009208 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009209 slen = (int)STRLEN(p);
9210
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009211 nchar = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009212 if (!error)
9213 {
9214 if (nchar > 0)
9215 while (nchar > 0 && nbyte < slen)
9216 {
Bram Moolenaard3c907b2016-08-17 21:32:09 +02009217 nbyte += MB_CPTR2LEN(p + nbyte);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009218 --nchar;
9219 }
9220 else
9221 nbyte = nchar;
9222 if (argvars[2].v_type != VAR_UNKNOWN)
9223 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009224 charlen = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009225 while (charlen > 0 && nbyte + len < slen)
9226 {
9227 int off = nbyte + len;
9228
9229 if (off < 0)
9230 len += 1;
9231 else
Bram Moolenaard3c907b2016-08-17 21:32:09 +02009232 len += MB_CPTR2LEN(p + off);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009233 --charlen;
9234 }
9235 }
9236 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009237 len = slen - nbyte; // default: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009238 }
9239
9240 /*
9241 * Only return the overlap between the specified part and the actual
9242 * string.
9243 */
9244 if (nbyte < 0)
9245 {
9246 len += nbyte;
9247 nbyte = 0;
9248 }
9249 else if (nbyte > slen)
9250 nbyte = slen;
9251 if (len < 0)
9252 len = 0;
9253 else if (nbyte + len > slen)
9254 len = slen - nbyte;
9255
9256 rettv->v_type = VAR_STRING;
9257 rettv->vval.v_string = vim_strnsave(p + nbyte, len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009258}
9259
9260/*
9261 * "strpart()" function
9262 */
9263 static void
9264f_strpart(typval_T *argvars, typval_T *rettv)
9265{
9266 char_u *p;
9267 int n;
9268 int len;
9269 int slen;
9270 int error = FALSE;
9271
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009272 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009273 slen = (int)STRLEN(p);
9274
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009275 n = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009276 if (error)
9277 len = 0;
9278 else if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009279 len = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009280 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009281 len = slen - n; // default len: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009282
Bram Moolenaar6c53fca2020-08-23 17:34:46 +02009283 // Only return the overlap between the specified part and the actual
9284 // string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009285 if (n < 0)
9286 {
9287 len += n;
9288 n = 0;
9289 }
9290 else if (n > slen)
9291 n = slen;
9292 if (len < 0)
9293 len = 0;
9294 else if (n + len > slen)
9295 len = slen - n;
9296
Bram Moolenaar6c53fca2020-08-23 17:34:46 +02009297 if (argvars[2].v_type != VAR_UNKNOWN && argvars[3].v_type != VAR_UNKNOWN)
9298 {
9299 int off;
9300
9301 // length in characters
9302 for (off = n; off < slen && len > 0; --len)
9303 off += mb_ptr2len(p + off);
9304 len = off - n;
9305 }
9306
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009307 rettv->v_type = VAR_STRING;
9308 rettv->vval.v_string = vim_strnsave(p + n, len);
9309}
9310
9311/*
9312 * "strridx()" function
9313 */
9314 static void
9315f_strridx(typval_T *argvars, typval_T *rettv)
9316{
9317 char_u buf[NUMBUFLEN];
9318 char_u *needle;
9319 char_u *haystack;
9320 char_u *rest;
9321 char_u *lastmatch = NULL;
9322 int haystack_len, end_idx;
9323
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009324 needle = tv_get_string_chk(&argvars[1]);
9325 haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009326
9327 rettv->vval.v_number = -1;
9328 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009329 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009330
9331 haystack_len = (int)STRLEN(haystack);
9332 if (argvars[2].v_type != VAR_UNKNOWN)
9333 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009334 // Third argument: upper limit for index
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009335 end_idx = (int)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009336 if (end_idx < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009337 return; // can never find a match
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009338 }
9339 else
9340 end_idx = haystack_len;
9341
9342 if (*needle == NUL)
9343 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009344 // Empty string matches past the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009345 lastmatch = haystack + end_idx;
9346 }
9347 else
9348 {
9349 for (rest = haystack; *rest != '\0'; ++rest)
9350 {
9351 rest = (char_u *)strstr((char *)rest, (char *)needle);
9352 if (rest == NULL || rest > haystack + end_idx)
9353 break;
9354 lastmatch = rest;
9355 }
9356 }
9357
9358 if (lastmatch == NULL)
9359 rettv->vval.v_number = -1;
9360 else
9361 rettv->vval.v_number = (varnumber_T)(lastmatch - haystack);
9362}
9363
9364/*
9365 * "strtrans()" function
9366 */
9367 static void
9368f_strtrans(typval_T *argvars, typval_T *rettv)
9369{
9370 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009371 rettv->vval.v_string = transstr(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009372}
9373
9374/*
9375 * "submatch()" function
9376 */
9377 static void
9378f_submatch(typval_T *argvars, typval_T *rettv)
9379{
9380 int error = FALSE;
9381 int no;
9382 int retList = 0;
9383
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009384 no = (int)tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009385 if (error)
9386 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02009387 if (no < 0 || no >= NSUBEXP)
9388 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009389 semsg(_("E935: invalid submatch number: %d"), no);
Bram Moolenaar79518e22017-02-17 16:31:35 +01009390 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02009391 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009392 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaarad304702020-09-06 18:22:53 +02009393 retList = (int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009394 if (error)
9395 return;
9396
9397 if (retList == 0)
9398 {
9399 rettv->v_type = VAR_STRING;
9400 rettv->vval.v_string = reg_submatch(no);
9401 }
9402 else
9403 {
9404 rettv->v_type = VAR_LIST;
9405 rettv->vval.v_list = reg_submatch_list(no);
9406 }
9407}
9408
9409/*
9410 * "substitute()" function
9411 */
9412 static void
9413f_substitute(typval_T *argvars, typval_T *rettv)
9414{
9415 char_u patbuf[NUMBUFLEN];
9416 char_u subbuf[NUMBUFLEN];
9417 char_u flagsbuf[NUMBUFLEN];
9418
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009419 char_u *str = tv_get_string_chk(&argvars[0]);
9420 char_u *pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009421 char_u *sub = NULL;
9422 typval_T *expr = NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009423 char_u *flg = tv_get_string_buf_chk(&argvars[3], flagsbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009424
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009425 if (argvars[2].v_type == VAR_FUNC || argvars[2].v_type == VAR_PARTIAL)
9426 expr = &argvars[2];
9427 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009428 sub = tv_get_string_buf_chk(&argvars[2], subbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009429
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009430 rettv->v_type = VAR_STRING;
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009431 if (str == NULL || pat == NULL || (sub == NULL && expr == NULL)
9432 || flg == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009433 rettv->vval.v_string = NULL;
9434 else
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009435 rettv->vval.v_string = do_string_sub(str, pat, sub, expr, flg);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009436}
9437
9438/*
Bram Moolenaar00f123a2018-08-21 20:28:54 +02009439 * "swapinfo(swap_filename)" function
9440 */
9441 static void
9442f_swapinfo(typval_T *argvars, typval_T *rettv)
9443{
9444 if (rettv_dict_alloc(rettv) == OK)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009445 get_b0_dict(tv_get_string(argvars), rettv->vval.v_dict);
Bram Moolenaar00f123a2018-08-21 20:28:54 +02009446}
9447
9448/*
Bram Moolenaar110bd602018-09-16 18:46:59 +02009449 * "swapname(expr)" function
9450 */
9451 static void
9452f_swapname(typval_T *argvars, typval_T *rettv)
9453{
9454 buf_T *buf;
9455
9456 rettv->v_type = VAR_STRING;
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01009457 buf = tv_get_buf(&argvars[0], FALSE);
Bram Moolenaar110bd602018-09-16 18:46:59 +02009458 if (buf == NULL || buf->b_ml.ml_mfp == NULL
9459 || buf->b_ml.ml_mfp->mf_fname == NULL)
9460 rettv->vval.v_string = NULL;
9461 else
9462 rettv->vval.v_string = vim_strsave(buf->b_ml.ml_mfp->mf_fname);
9463}
9464
9465/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009466 * "synID(lnum, col, trans)" function
9467 */
9468 static void
9469f_synID(typval_T *argvars UNUSED, typval_T *rettv)
9470{
9471 int id = 0;
9472#ifdef FEAT_SYN_HL
9473 linenr_T lnum;
9474 colnr_T col;
9475 int trans;
9476 int transerr = FALSE;
9477
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009478 lnum = tv_get_lnum(argvars); // -1 on type error
9479 col = (linenr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaarfcb6d702020-09-05 21:41:56 +02009480 trans = (int)tv_get_bool_chk(&argvars[2], &transerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009481
9482 if (!transerr && lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9483 && col >= 0 && col < (long)STRLEN(ml_get(lnum)))
9484 id = syn_get_id(curwin, lnum, (colnr_T)col, trans, NULL, FALSE);
9485#endif
9486
9487 rettv->vval.v_number = id;
9488}
9489
9490/*
9491 * "synIDattr(id, what [, mode])" function
9492 */
9493 static void
9494f_synIDattr(typval_T *argvars UNUSED, typval_T *rettv)
9495{
9496 char_u *p = NULL;
9497#ifdef FEAT_SYN_HL
9498 int id;
9499 char_u *what;
9500 char_u *mode;
9501 char_u modebuf[NUMBUFLEN];
9502 int modec;
9503
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009504 id = (int)tv_get_number(&argvars[0]);
9505 what = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009506 if (argvars[2].v_type != VAR_UNKNOWN)
9507 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009508 mode = tv_get_string_buf(&argvars[2], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009509 modec = TOLOWER_ASC(mode[0]);
9510 if (modec != 't' && modec != 'c' && modec != 'g')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009511 modec = 0; // replace invalid with current
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009512 }
9513 else
9514 {
9515#if defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS)
9516 if (USE_24BIT)
9517 modec = 'g';
9518 else
9519#endif
9520 if (t_colors > 1)
9521 modec = 'c';
9522 else
9523 modec = 't';
9524 }
9525
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009526 switch (TOLOWER_ASC(what[0]))
9527 {
9528 case 'b':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009529 if (TOLOWER_ASC(what[1]) == 'g') // bg[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009530 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009531 else // bold
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009532 p = highlight_has_attr(id, HL_BOLD, modec);
9533 break;
9534
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009535 case 'f': // fg[#] or font
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009536 p = highlight_color(id, what, modec);
9537 break;
9538
9539 case 'i':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009540 if (TOLOWER_ASC(what[1]) == 'n') // inverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009541 p = highlight_has_attr(id, HL_INVERSE, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009542 else // italic
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009543 p = highlight_has_attr(id, HL_ITALIC, modec);
9544 break;
9545
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009546 case 'n': // name
Bram Moolenaarc96272e2017-03-26 13:50:09 +02009547 p = get_highlight_name_ext(NULL, id - 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009548 break;
9549
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009550 case 'r': // reverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009551 p = highlight_has_attr(id, HL_INVERSE, modec);
9552 break;
9553
9554 case 's':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009555 if (TOLOWER_ASC(what[1]) == 'p') // sp[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009556 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009557 // strikeout
Bram Moolenaarcf4b00c2017-09-02 18:33:56 +02009558 else if (TOLOWER_ASC(what[1]) == 't' &&
9559 TOLOWER_ASC(what[2]) == 'r')
9560 p = highlight_has_attr(id, HL_STRIKETHROUGH, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009561 else // standout
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009562 p = highlight_has_attr(id, HL_STANDOUT, modec);
9563 break;
9564
9565 case 'u':
Bram Moolenaar391c3622020-09-29 20:59:17 +02009566 if (TOLOWER_ASC(what[1]) == 'l') // ul
9567 p = highlight_color(id, what, modec);
9568 else if (STRLEN(what) <= 5 || TOLOWER_ASC(what[5]) != 'c')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009569 // underline
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009570 p = highlight_has_attr(id, HL_UNDERLINE, modec);
9571 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009572 // undercurl
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009573 p = highlight_has_attr(id, HL_UNDERCURL, modec);
9574 break;
9575 }
9576
9577 if (p != NULL)
9578 p = vim_strsave(p);
9579#endif
9580 rettv->v_type = VAR_STRING;
9581 rettv->vval.v_string = p;
9582}
9583
9584/*
9585 * "synIDtrans(id)" function
9586 */
9587 static void
9588f_synIDtrans(typval_T *argvars UNUSED, typval_T *rettv)
9589{
9590 int id;
9591
9592#ifdef FEAT_SYN_HL
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009593 id = (int)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009594
9595 if (id > 0)
9596 id = syn_get_final_id(id);
9597 else
9598#endif
9599 id = 0;
9600
9601 rettv->vval.v_number = id;
9602}
9603
9604/*
9605 * "synconcealed(lnum, col)" function
9606 */
9607 static void
9608f_synconcealed(typval_T *argvars UNUSED, typval_T *rettv)
9609{
9610#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
9611 linenr_T lnum;
9612 colnr_T col;
9613 int syntax_flags = 0;
9614 int cchar;
9615 int matchid = 0;
9616 char_u str[NUMBUFLEN];
9617#endif
9618
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02009619 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009620
9621#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009622 lnum = tv_get_lnum(argvars); // -1 on type error
9623 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009624
Bram Moolenaara80faa82020-04-12 19:37:17 +02009625 CLEAR_FIELD(str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009626
9627 if (rettv_list_alloc(rettv) != FAIL)
9628 {
9629 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9630 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
9631 && curwin->w_p_cole > 0)
9632 {
9633 (void)syn_get_id(curwin, lnum, col, FALSE, NULL, FALSE);
9634 syntax_flags = get_syntax_info(&matchid);
9635
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009636 // get the conceal character
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009637 if ((syntax_flags & HL_CONCEAL) && curwin->w_p_cole < 3)
9638 {
9639 cchar = syn_get_sub_char();
Bram Moolenaar4d785892017-06-22 22:00:50 +02009640 if (cchar == NUL && curwin->w_p_cole == 1)
9641 cchar = (lcs_conceal == NUL) ? ' ' : lcs_conceal;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009642 if (cchar != NUL)
9643 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009644 if (has_mbyte)
9645 (*mb_char2bytes)(cchar, str);
9646 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009647 str[0] = cchar;
9648 }
9649 }
9650 }
9651
9652 list_append_number(rettv->vval.v_list,
9653 (syntax_flags & HL_CONCEAL) != 0);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009654 // -1 to auto-determine strlen
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009655 list_append_string(rettv->vval.v_list, str, -1);
9656 list_append_number(rettv->vval.v_list, matchid);
9657 }
9658#endif
9659}
9660
9661/*
9662 * "synstack(lnum, col)" function
9663 */
9664 static void
9665f_synstack(typval_T *argvars UNUSED, typval_T *rettv)
9666{
9667#ifdef FEAT_SYN_HL
9668 linenr_T lnum;
9669 colnr_T col;
9670 int i;
9671 int id;
9672#endif
9673
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02009674 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009675
9676#ifdef FEAT_SYN_HL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009677 lnum = tv_get_lnum(argvars); // -1 on type error
9678 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009679
9680 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9681 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
9682 && rettv_list_alloc(rettv) != FAIL)
9683 {
9684 (void)syn_get_id(curwin, lnum, (colnr_T)col, FALSE, NULL, TRUE);
9685 for (i = 0; ; ++i)
9686 {
9687 id = syn_get_stack_item(i);
9688 if (id < 0)
9689 break;
9690 if (list_append_number(rettv->vval.v_list, id) == FAIL)
9691 break;
9692 }
9693 }
9694#endif
9695}
9696
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009697/*
9698 * "tabpagebuflist()" function
9699 */
9700 static void
9701f_tabpagebuflist(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
9702{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009703 tabpage_T *tp;
9704 win_T *wp = NULL;
9705
9706 if (argvars[0].v_type == VAR_UNKNOWN)
9707 wp = firstwin;
9708 else
9709 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009710 tp = find_tabpage((int)tv_get_number(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009711 if (tp != NULL)
9712 wp = (tp == curtab) ? firstwin : tp->tp_firstwin;
9713 }
9714 if (wp != NULL && rettv_list_alloc(rettv) != FAIL)
9715 {
9716 for (; wp != NULL; wp = wp->w_next)
9717 if (list_append_number(rettv->vval.v_list,
9718 wp->w_buffer->b_fnum) == FAIL)
9719 break;
9720 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009721}
9722
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009723/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009724 * "tagfiles()" function
9725 */
9726 static void
9727f_tagfiles(typval_T *argvars UNUSED, typval_T *rettv)
9728{
9729 char_u *fname;
9730 tagname_T tn;
9731 int first;
9732
9733 if (rettv_list_alloc(rettv) == FAIL)
9734 return;
9735 fname = alloc(MAXPATHL);
9736 if (fname == NULL)
9737 return;
9738
9739 for (first = TRUE; ; first = FALSE)
9740 if (get_tagfname(&tn, first, fname) == FAIL
9741 || list_append_string(rettv->vval.v_list, fname, -1) == FAIL)
9742 break;
9743 tagname_free(&tn);
9744 vim_free(fname);
9745}
9746
9747/*
9748 * "taglist()" function
9749 */
9750 static void
9751f_taglist(typval_T *argvars, typval_T *rettv)
9752{
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009753 char_u *fname = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009754 char_u *tag_pattern;
9755
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009756 tag_pattern = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009757
9758 rettv->vval.v_number = FALSE;
9759 if (*tag_pattern == NUL)
9760 return;
9761
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009762 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009763 fname = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009764 if (rettv_list_alloc(rettv) == OK)
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009765 (void)get_tags(rettv->vval.v_list, tag_pattern, fname);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009766}
9767
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009768#ifdef FEAT_FLOAT
9769/*
9770 * "tan()" function
9771 */
9772 static void
9773f_tan(typval_T *argvars, typval_T *rettv)
9774{
9775 float_T f = 0.0;
9776
9777 rettv->v_type = VAR_FLOAT;
9778 if (get_float_arg(argvars, &f) == OK)
9779 rettv->vval.v_float = tan(f);
9780 else
9781 rettv->vval.v_float = 0.0;
9782}
9783
9784/*
9785 * "tanh()" function
9786 */
9787 static void
9788f_tanh(typval_T *argvars, typval_T *rettv)
9789{
9790 float_T f = 0.0;
9791
9792 rettv->v_type = VAR_FLOAT;
9793 if (get_float_arg(argvars, &f) == OK)
9794 rettv->vval.v_float = tanh(f);
9795 else
9796 rettv->vval.v_float = 0.0;
9797}
9798#endif
9799
9800/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009801 * "tolower(string)" function
9802 */
9803 static void
9804f_tolower(typval_T *argvars, typval_T *rettv)
9805{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009806 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009807 rettv->vval.v_string = strlow_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009808}
9809
9810/*
9811 * "toupper(string)" function
9812 */
9813 static void
9814f_toupper(typval_T *argvars, typval_T *rettv)
9815{
9816 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009817 rettv->vval.v_string = strup_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009818}
9819
9820/*
9821 * "tr(string, fromstr, tostr)" function
9822 */
9823 static void
9824f_tr(typval_T *argvars, typval_T *rettv)
9825{
9826 char_u *in_str;
9827 char_u *fromstr;
9828 char_u *tostr;
9829 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009830 int inlen;
9831 int fromlen;
9832 int tolen;
9833 int idx;
9834 char_u *cpstr;
9835 int cplen;
9836 int first = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009837 char_u buf[NUMBUFLEN];
9838 char_u buf2[NUMBUFLEN];
9839 garray_T ga;
9840
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009841 in_str = tv_get_string(&argvars[0]);
9842 fromstr = tv_get_string_buf_chk(&argvars[1], buf);
9843 tostr = tv_get_string_buf_chk(&argvars[2], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009844
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009845 // Default return value: empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009846 rettv->v_type = VAR_STRING;
9847 rettv->vval.v_string = NULL;
9848 if (fromstr == NULL || tostr == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009849 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009850 ga_init2(&ga, (int)sizeof(char), 80);
9851
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009852 if (!has_mbyte)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009853 // not multi-byte: fromstr and tostr must be the same length
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009854 if (STRLEN(fromstr) != STRLEN(tostr))
9855 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009856error:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009857 semsg(_(e_invarg2), fromstr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009858 ga_clear(&ga);
9859 return;
9860 }
9861
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009862 // fromstr and tostr have to contain the same number of chars
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009863 while (*in_str != NUL)
9864 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009865 if (has_mbyte)
9866 {
9867 inlen = (*mb_ptr2len)(in_str);
9868 cpstr = in_str;
9869 cplen = inlen;
9870 idx = 0;
9871 for (p = fromstr; *p != NUL; p += fromlen)
9872 {
9873 fromlen = (*mb_ptr2len)(p);
9874 if (fromlen == inlen && STRNCMP(in_str, p, inlen) == 0)
9875 {
9876 for (p = tostr; *p != NUL; p += tolen)
9877 {
9878 tolen = (*mb_ptr2len)(p);
9879 if (idx-- == 0)
9880 {
9881 cplen = tolen;
9882 cpstr = p;
9883 break;
9884 }
9885 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009886 if (*p == NUL) // tostr is shorter than fromstr
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009887 goto error;
9888 break;
9889 }
9890 ++idx;
9891 }
9892
9893 if (first && cpstr == in_str)
9894 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009895 // Check that fromstr and tostr have the same number of
9896 // (multi-byte) characters. Done only once when a character
9897 // of in_str doesn't appear in fromstr.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009898 first = FALSE;
9899 for (p = tostr; *p != NUL; p += tolen)
9900 {
9901 tolen = (*mb_ptr2len)(p);
9902 --idx;
9903 }
9904 if (idx != 0)
9905 goto error;
9906 }
9907
9908 (void)ga_grow(&ga, cplen);
9909 mch_memmove((char *)ga.ga_data + ga.ga_len, cpstr, (size_t)cplen);
9910 ga.ga_len += cplen;
9911
9912 in_str += inlen;
9913 }
9914 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009915 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009916 // When not using multi-byte chars we can do it faster.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009917 p = vim_strchr(fromstr, *in_str);
9918 if (p != NULL)
9919 ga_append(&ga, tostr[p - fromstr]);
9920 else
9921 ga_append(&ga, *in_str);
9922 ++in_str;
9923 }
9924 }
9925
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009926 // add a terminating NUL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009927 (void)ga_grow(&ga, 1);
9928 ga_append(&ga, NUL);
9929
9930 rettv->vval.v_string = ga.ga_data;
9931}
9932
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009933/*
9934 * "trim({expr})" function
9935 */
9936 static void
9937f_trim(typval_T *argvars, typval_T *rettv)
9938{
9939 char_u buf1[NUMBUFLEN];
9940 char_u buf2[NUMBUFLEN];
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009941 char_u *head = tv_get_string_buf_chk(&argvars[0], buf1);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009942 char_u *mask = NULL;
9943 char_u *tail;
9944 char_u *prev;
9945 char_u *p;
9946 int c1;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009947 int dir = 0;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009948
9949 rettv->v_type = VAR_STRING;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009950 rettv->vval.v_string = NULL;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009951 if (head == NULL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009952 return;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009953
9954 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009955 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009956 mask = tv_get_string_buf_chk(&argvars[1], buf2);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009957
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009958 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009959 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009960 int error = 0;
9961
9962 // leading or trailing characters to trim
9963 dir = (int)tv_get_number_chk(&argvars[2], &error);
9964 if (error)
9965 return;
9966 if (dir < 0 || dir > 2)
9967 {
9968 semsg(_(e_invarg2), tv_get_string(&argvars[2]));
9969 return;
9970 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009971 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009972 }
9973
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009974 if (dir == 0 || dir == 1)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009975 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009976 // Trim leading characters
9977 while (*head != NUL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009978 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009979 c1 = PTR2CHAR(head);
9980 if (mask == NULL)
9981 {
9982 if (c1 > ' ' && c1 != 0xa0)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009983 break;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009984 }
9985 else
9986 {
9987 for (p = mask; *p != NUL; MB_PTR_ADV(p))
9988 if (c1 == PTR2CHAR(p))
9989 break;
9990 if (*p == NUL)
9991 break;
9992 }
9993 MB_PTR_ADV(head);
9994 }
9995 }
9996
9997 tail = head + STRLEN(head);
9998 if (dir == 0 || dir == 2)
9999 {
10000 // Trim trailing characters
10001 for (; tail > head; tail = prev)
10002 {
10003 prev = tail;
10004 MB_PTR_BACK(head, prev);
10005 c1 = PTR2CHAR(prev);
10006 if (mask == NULL)
10007 {
10008 if (c1 > ' ' && c1 != 0xa0)
10009 break;
10010 }
10011 else
10012 {
10013 for (p = mask; *p != NUL; MB_PTR_ADV(p))
10014 if (c1 == PTR2CHAR(p))
10015 break;
10016 if (*p == NUL)
10017 break;
10018 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +010010019 }
10020 }
Bram Moolenaardf44a272020-06-07 20:49:05 +020010021 rettv->vval.v_string = vim_strnsave(head, tail - head);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +010010022}
10023
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010024#ifdef FEAT_FLOAT
10025/*
10026 * "trunc({float})" function
10027 */
10028 static void
10029f_trunc(typval_T *argvars, typval_T *rettv)
10030{
10031 float_T f = 0.0;
10032
10033 rettv->v_type = VAR_FLOAT;
10034 if (get_float_arg(argvars, &f) == OK)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010010035 // trunc() is not in C90, use floor() or ceil() instead.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010036 rettv->vval.v_float = f > 0 ? floor(f) : ceil(f);
10037 else
10038 rettv->vval.v_float = 0.0;
10039}
10040#endif
10041
10042/*
10043 * "type(expr)" function
10044 */
10045 static void
10046f_type(typval_T *argvars, typval_T *rettv)
10047{
10048 int n = -1;
10049
10050 switch (argvars[0].v_type)
10051 {
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +010010052 case VAR_NUMBER: n = VAR_TYPE_NUMBER; break;
10053 case VAR_STRING: n = VAR_TYPE_STRING; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010054 case VAR_PARTIAL:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +010010055 case VAR_FUNC: n = VAR_TYPE_FUNC; break;
10056 case VAR_LIST: n = VAR_TYPE_LIST; break;
10057 case VAR_DICT: n = VAR_TYPE_DICT; break;
10058 case VAR_FLOAT: n = VAR_TYPE_FLOAT; break;
10059 case VAR_BOOL: n = VAR_TYPE_BOOL; break;
10060 case VAR_SPECIAL: n = VAR_TYPE_NONE; break;
Bram Moolenaarf562e722016-07-19 17:25:25 +020010061 case VAR_JOB: n = VAR_TYPE_JOB; break;
10062 case VAR_CHANNEL: n = VAR_TYPE_CHANNEL; break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +010010063 case VAR_BLOB: n = VAR_TYPE_BLOB; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010064 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +020010065 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010010066 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +010010067 internal_error_no_abort("f_type(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010068 n = -1;
10069 break;
10070 }
10071 rettv->vval.v_number = n;
10072}
10073
10074/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010075 * "virtcol(string)" function
10076 */
10077 static void
10078f_virtcol(typval_T *argvars, typval_T *rettv)
10079{
10080 colnr_T vcol = 0;
10081 pos_T *fp;
10082 int fnum = curbuf->b_fnum;
Bram Moolenaarb3d33d82020-01-15 20:36:55 +010010083 int len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010084
Bram Moolenaar6f02b002021-01-10 20:22:54 +010010085 fp = var2fpos(&argvars[0], FALSE, &fnum, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010086 if (fp != NULL && fp->lnum <= curbuf->b_ml.ml_line_count
10087 && fnum == curbuf->b_fnum)
10088 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +010010089 // Limit the column to a valid value, getvvcol() doesn't check.
10090 if (fp->col < 0)
10091 fp->col = 0;
10092 else
10093 {
10094 len = (int)STRLEN(ml_get(fp->lnum));
10095 if (fp->col > len)
10096 fp->col = len;
10097 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010098 getvvcol(curwin, fp, NULL, NULL, &vcol);
10099 ++vcol;
10100 }
10101
10102 rettv->vval.v_number = vcol;
10103}
10104
10105/*
10106 * "visualmode()" function
10107 */
10108 static void
10109f_visualmode(typval_T *argvars, typval_T *rettv)
10110{
10111 char_u str[2];
10112
10113 rettv->v_type = VAR_STRING;
10114 str[0] = curbuf->b_visual_mode_eval;
10115 str[1] = NUL;
10116 rettv->vval.v_string = vim_strsave(str);
10117
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010010118 // A non-zero number or non-empty string argument: reset mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010119 if (non_zero_arg(&argvars[0]))
10120 curbuf->b_visual_mode_eval = NUL;
10121}
10122
10123/*
10124 * "wildmenumode()" function
10125 */
10126 static void
10127f_wildmenumode(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
10128{
10129#ifdef FEAT_WILDMENU
10130 if (wild_menu_showing)
10131 rettv->vval.v_number = 1;
10132#endif
10133}
10134
10135/*
Bram Moolenaar0c1e3742019-12-27 13:49:24 +010010136 * "windowsversion()" function
10137 */
10138 static void
10139f_windowsversion(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
10140{
10141 rettv->v_type = VAR_STRING;
10142 rettv->vval.v_string = vim_strsave((char_u *)windowsVersion);
10143}
10144
10145/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010146 * "wordcount()" function
10147 */
10148 static void
10149f_wordcount(typval_T *argvars UNUSED, typval_T *rettv)
10150{
10151 if (rettv_dict_alloc(rettv) == FAIL)
10152 return;
10153 cursor_pos_info(rettv->vval.v_dict);
10154}
10155
10156/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010157 * "xor(expr, expr)" function
10158 */
10159 static void
10160f_xor(typval_T *argvars, typval_T *rettv)
10161{
Bram Moolenaard155d7a2018-12-21 16:04:21 +010010162 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
10163 ^ tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010164}
10165
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010010166#endif // FEAT_EVAL