blob: 45937fabf5e2627ef1dc832c216a780ce887cf1a [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 Moolenaar73dad1e2016-07-17 22:13:49 +020050static void f_col(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020051static void f_confirm(typval_T *argvars, typval_T *rettv);
52static void f_copy(typval_T *argvars, typval_T *rettv);
53#ifdef FEAT_FLOAT
54static void f_cos(typval_T *argvars, typval_T *rettv);
55static void f_cosh(typval_T *argvars, typval_T *rettv);
56#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020057static void f_cursor(typval_T *argsvars, typval_T *rettv);
Bram Moolenaar4f974752019-02-17 17:44:42 +010058#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +020059static void f_debugbreak(typval_T *argvars, typval_T *rettv);
60#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020061static void f_deepcopy(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020062static void f_did_filetype(typval_T *argvars, typval_T *rettv);
Bram Moolenaar4132eb52020-02-14 16:53:00 +010063static void f_echoraw(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020064static void f_empty(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +020065static void f_environ(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020066static void f_escape(typval_T *argvars, typval_T *rettv);
67static void f_eval(typval_T *argvars, typval_T *rettv);
68static void f_eventhandler(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020069static void f_execute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020070static void f_exists(typval_T *argvars, typval_T *rettv);
71#ifdef FEAT_FLOAT
72static void f_exp(typval_T *argvars, typval_T *rettv);
73#endif
74static void f_expand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar80dad482019-06-09 17:22:31 +020075static void f_expandcmd(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020076static void f_feedkeys(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020077#ifdef FEAT_FLOAT
78static void f_float2nr(typval_T *argvars, typval_T *rettv);
79static void f_floor(typval_T *argvars, typval_T *rettv);
80static void f_fmod(typval_T *argvars, typval_T *rettv);
81#endif
82static void f_fnameescape(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020083static void f_foreground(typval_T *argvars, typval_T *rettv);
Bram Moolenaar437bafe2016-08-01 15:40:54 +020084static void f_funcref(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020085static void f_function(typval_T *argvars, typval_T *rettv);
86static void f_garbagecollect(typval_T *argvars, typval_T *rettv);
87static void f_get(typval_T *argvars, typval_T *rettv);
Bram Moolenaar07ad8162018-02-13 13:59:59 +010088static void f_getchangelist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020089static void f_getcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +020090static void f_getenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020091static void f_getfontname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar4f505882018-02-10 21:06:32 +010092static void f_getjumplist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020093static void f_getpid(typval_T *argvars, typval_T *rettv);
94static void f_getcurpos(typval_T *argvars, typval_T *rettv);
95static void f_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020096static void f_getreg(typval_T *argvars, typval_T *rettv);
97static void f_getregtype(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +010098static void f_gettagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020099static void f_haslocaldir(typval_T *argvars, typval_T *rettv);
100static void f_hasmapto(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200101static void f_hlID(typval_T *argvars, typval_T *rettv);
102static void f_hlexists(typval_T *argvars, typval_T *rettv);
103static void f_hostname(typval_T *argvars, typval_T *rettv);
104static void f_iconv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200105static void f_index(typval_T *argvars, typval_T *rettv);
106static void f_input(typval_T *argvars, typval_T *rettv);
107static void f_inputdialog(typval_T *argvars, typval_T *rettv);
108static void f_inputlist(typval_T *argvars, typval_T *rettv);
109static void f_inputrestore(typval_T *argvars, typval_T *rettv);
110static void f_inputsave(typval_T *argvars, typval_T *rettv);
111static void f_inputsecret(typval_T *argvars, typval_T *rettv);
Bram Moolenaar67a2deb2019-11-25 00:05:32 +0100112static void f_interrupt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200113static void f_invert(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200114static void f_islocked(typval_T *argvars, typval_T *rettv);
115#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
Bram Moolenaarfda1bff2019-04-04 13:44:37 +0200116static void f_isinf(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200117static void f_isnan(typval_T *argvars, typval_T *rettv);
118#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200119static void f_last_buffer_nr(typval_T *argvars, typval_T *rettv);
120static void f_len(typval_T *argvars, typval_T *rettv);
121static void f_libcall(typval_T *argvars, typval_T *rettv);
122static void f_libcallnr(typval_T *argvars, typval_T *rettv);
123static void f_line(typval_T *argvars, typval_T *rettv);
124static void f_line2byte(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200125#ifdef FEAT_FLOAT
126static void f_log(typval_T *argvars, typval_T *rettv);
127static void f_log10(typval_T *argvars, typval_T *rettv);
128#endif
129#ifdef FEAT_LUA
130static void f_luaeval(typval_T *argvars, typval_T *rettv);
131#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200132static void f_maparg(typval_T *argvars, typval_T *rettv);
133static void f_mapcheck(typval_T *argvars, typval_T *rettv);
134static void f_match(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200135static void f_matchend(typval_T *argvars, typval_T *rettv);
136static void f_matchlist(typval_T *argvars, typval_T *rettv);
137static void f_matchstr(typval_T *argvars, typval_T *rettv);
138static void f_matchstrpos(typval_T *argvars, typval_T *rettv);
139static void f_max(typval_T *argvars, typval_T *rettv);
140static void f_min(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200141#ifdef FEAT_MZSCHEME
142static void f_mzeval(typval_T *argvars, typval_T *rettv);
143#endif
144static void f_nextnonblank(typval_T *argvars, typval_T *rettv);
145static void f_nr2char(typval_T *argvars, typval_T *rettv);
146static void f_or(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200147#ifdef FEAT_PERL
148static void f_perleval(typval_T *argvars, typval_T *rettv);
149#endif
150#ifdef FEAT_FLOAT
151static void f_pow(typval_T *argvars, typval_T *rettv);
152#endif
153static void f_prevnonblank(typval_T *argvars, typval_T *rettv);
154static void f_printf(typval_T *argvars, typval_T *rettv);
Bram Moolenaare9bd5722019-08-17 19:36:06 +0200155static void f_pum_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200156static void f_pumvisible(typval_T *argvars, typval_T *rettv);
157#ifdef FEAT_PYTHON3
158static void f_py3eval(typval_T *argvars, typval_T *rettv);
159#endif
160#ifdef FEAT_PYTHON
161static void f_pyeval(typval_T *argvars, typval_T *rettv);
162#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100163#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
164static void f_pyxeval(typval_T *argvars, typval_T *rettv);
165#endif
Bram Moolenaar4f645c52020-02-08 16:40:39 +0100166static void f_test_srand_seed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100167static void f_rand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200168static void f_range(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0b6d9112018-05-22 20:35:17 +0200169static void f_reg_executing(typval_T *argvars, typval_T *rettv);
170static void f_reg_recording(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200171static void f_rename(typval_T *argvars, typval_T *rettv);
172static void f_repeat(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200173#ifdef FEAT_FLOAT
174static void f_round(typval_T *argvars, typval_T *rettv);
175#endif
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100176#ifdef FEAT_RUBY
177static void f_rubyeval(typval_T *argvars, typval_T *rettv);
178#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200179static void f_screenattr(typval_T *argvars, typval_T *rettv);
180static void f_screenchar(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100181static void f_screenchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200182static void f_screencol(typval_T *argvars, typval_T *rettv);
183static void f_screenrow(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100184static void f_screenstring(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200185static void f_search(typval_T *argvars, typval_T *rettv);
186static void f_searchdecl(typval_T *argvars, typval_T *rettv);
187static void f_searchpair(typval_T *argvars, typval_T *rettv);
188static void f_searchpairpos(typval_T *argvars, typval_T *rettv);
189static void f_searchpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200190static void f_setcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +0200191static void f_setenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200192static void f_setfperm(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200193static void f_setpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200194static void f_setreg(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100195static void f_settagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200196#ifdef FEAT_CRYPT
197static void f_sha256(typval_T *argvars, typval_T *rettv);
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200198#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200199static void f_shellescape(typval_T *argvars, typval_T *rettv);
200static void f_shiftwidth(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200201#ifdef FEAT_FLOAT
202static void f_sin(typval_T *argvars, typval_T *rettv);
203static void f_sinh(typval_T *argvars, typval_T *rettv);
204#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200205static void f_soundfold(typval_T *argvars, typval_T *rettv);
206static void f_spellbadword(typval_T *argvars, typval_T *rettv);
207static void f_spellsuggest(typval_T *argvars, typval_T *rettv);
208static void f_split(typval_T *argvars, typval_T *rettv);
209#ifdef FEAT_FLOAT
210static void f_sqrt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100211#endif
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100212static void f_srand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100213#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200214static void f_str2float(typval_T *argvars, typval_T *rettv);
215#endif
Bram Moolenaar9d401282019-04-06 13:18:12 +0200216static void f_str2list(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200217static void f_str2nr(typval_T *argvars, typval_T *rettv);
218static void f_strchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200219static void f_strgetchar(typval_T *argvars, typval_T *rettv);
220static void f_stridx(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200221static void f_strlen(typval_T *argvars, typval_T *rettv);
222static void f_strcharpart(typval_T *argvars, typval_T *rettv);
223static void f_strpart(typval_T *argvars, typval_T *rettv);
224static void f_strridx(typval_T *argvars, typval_T *rettv);
225static void f_strtrans(typval_T *argvars, typval_T *rettv);
226static void f_strdisplaywidth(typval_T *argvars, typval_T *rettv);
227static void f_strwidth(typval_T *argvars, typval_T *rettv);
228static void f_submatch(typval_T *argvars, typval_T *rettv);
229static void f_substitute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar00f123a2018-08-21 20:28:54 +0200230static void f_swapinfo(typval_T *argvars, typval_T *rettv);
Bram Moolenaar110bd602018-09-16 18:46:59 +0200231static void f_swapname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200232static void f_synID(typval_T *argvars, typval_T *rettv);
233static void f_synIDattr(typval_T *argvars, typval_T *rettv);
234static void f_synIDtrans(typval_T *argvars, typval_T *rettv);
235static void f_synstack(typval_T *argvars, typval_T *rettv);
236static void f_synconcealed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200237static void f_tabpagebuflist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200238static void f_taglist(typval_T *argvars, typval_T *rettv);
239static void f_tagfiles(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200240#ifdef FEAT_FLOAT
241static void f_tan(typval_T *argvars, typval_T *rettv);
242static void f_tanh(typval_T *argvars, typval_T *rettv);
243#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200244static void f_tolower(typval_T *argvars, typval_T *rettv);
245static void f_toupper(typval_T *argvars, typval_T *rettv);
246static void f_tr(typval_T *argvars, typval_T *rettv);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +0100247static void f_trim(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200248#ifdef FEAT_FLOAT
249static void f_trunc(typval_T *argvars, typval_T *rettv);
250#endif
251static void f_type(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200252static void f_virtcol(typval_T *argvars, typval_T *rettv);
253static void f_visualmode(typval_T *argvars, typval_T *rettv);
254static void f_wildmenumode(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0c1e3742019-12-27 13:49:24 +0100255static void f_windowsversion(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200256static void f_wordcount(typval_T *argvars, typval_T *rettv);
257static void f_xor(typval_T *argvars, typval_T *rettv);
258
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100259
260 static type_T *
261ret_void(int argcount UNUSED, type_T **argtypes UNUSED)
262{
263 return &t_void;
264}
265 static type_T *
266ret_any(int argcount UNUSED, type_T **argtypes UNUSED)
267{
268 return &t_any;
269}
270 static type_T *
271ret_number(int argcount UNUSED, type_T **argtypes UNUSED)
272{
273 return &t_number;
274}
275 static type_T *
276ret_float(int argcount UNUSED, type_T **argtypes UNUSED)
277{
278 return &t_float;
279}
280 static type_T *
281ret_string(int argcount UNUSED, type_T **argtypes UNUSED)
282{
283 return &t_string;
284}
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200285 static type_T *
286ret_list_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100287{
288 return &t_list_any;
289}
290 static type_T *
291ret_list_number(int argcount UNUSED, type_T **argtypes UNUSED)
292{
293 return &t_list_number;
294}
295 static type_T *
296ret_list_string(int argcount UNUSED, type_T **argtypes UNUSED)
297{
298 return &t_list_string;
299}
300 static type_T *
301ret_list_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
302{
303 return &t_list_dict_any;
304}
305 static type_T *
306ret_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
307{
308 return &t_dict_any;
309}
310 static type_T *
311ret_dict_number(int argcount UNUSED, type_T **argtypes UNUSED)
312{
313 return &t_dict_number;
314}
315 static type_T *
316ret_dict_string(int argcount UNUSED, type_T **argtypes UNUSED)
317{
318 return &t_dict_string;
319}
320 static type_T *
321ret_blob(int argcount UNUSED, type_T **argtypes UNUSED)
322{
323 return &t_blob;
324}
325 static type_T *
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200326ret_func_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100327{
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200328 return &t_func_any;
329}
330 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100331ret_channel(int argcount UNUSED, type_T **argtypes UNUSED)
332{
333 return &t_channel;
334}
335 static type_T *
336ret_job(int argcount UNUSED, type_T **argtypes UNUSED)
337{
338 return &t_job;
339}
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100340
341static type_T *ret_f_function(int argcount, type_T **argtypes);
342
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200343/*
344 * Array with names and number of arguments of all internal functions
345 * MUST BE KEPT SORTED IN strcmp() ORDER FOR BINARY SEARCH!
346 */
Bram Moolenaarac92e252019-08-03 21:58:38 +0200347typedef struct
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200348{
Bram Moolenaar25e42232019-08-04 15:04:10 +0200349 char *f_name; // function name
350 char f_min_argc; // minimal number of arguments
351 char f_max_argc; // maximal number of arguments
352 char f_argtype; // for method: FEARG_ values
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100353 type_T *(*f_retfunc)(int argcount, type_T **argtypes);
354 // return type function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200355 void (*f_func)(typval_T *args, typval_T *rvar);
Bram Moolenaar25e42232019-08-04 15:04:10 +0200356 // implementation of function
Bram Moolenaarac92e252019-08-03 21:58:38 +0200357} funcentry_T;
358
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200359// values for f_argtype; zero means it cannot be used as a method
360#define FEARG_1 1 // base is the first argument
361#define FEARG_2 2 // base is the second argument
Bram Moolenaar24278d22019-08-16 21:49:22 +0200362#define FEARG_3 3 // base is the third argument
Bram Moolenaaraad222c2019-09-06 22:46:09 +0200363#define FEARG_4 4 // base is the fourth argument
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200364#define FEARG_LAST 9 // base is the last argument
365
Bram Moolenaar15c47602020-03-26 22:16:48 +0100366#ifdef FEAT_FLOAT
367# define FLOAT_FUNC(name) name
368#else
369# define FLOAT_FUNC(name) NULL
370#endif
371#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
372# define MATH_FUNC(name) name
373#else
374# define MATH_FUNC(name) NULL
375#endif
376#ifdef FEAT_TIMERS
377# define TIMER_FUNC(name) name
378#else
379# define TIMER_FUNC(name) NULL
380#endif
381#ifdef FEAT_JOB_CHANNEL
382# define JOB_FUNC(name) name
383#else
384# define JOB_FUNC(name) NULL
385#endif
386#ifdef FEAT_PROP_POPUP
387# define PROP_FUNC(name) name
388#else
389# define PROP_FUNC(name) NULL
390#endif
391#ifdef FEAT_SIGNS
392# define SIGN_FUNC(name) name
393#else
394# define SIGN_FUNC(name) NULL
395#endif
396#ifdef FEAT_SOUND
397# define SOUND_FUNC(name) name
398#else
399# define SOUND_FUNC(name) NULL
400#endif
401#ifdef FEAT_TERMINAL
402# define TERM_FUNC(name) name
403#else
404# define TERM_FUNC(name) NULL
405#endif
406
Bram Moolenaarac92e252019-08-03 21:58:38 +0200407static funcentry_T global_functions[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200408{
Bram Moolenaar15c47602020-03-26 22:16:48 +0100409 {"abs", 1, 1, FEARG_1, ret_any, FLOAT_FUNC(f_abs)},
410 {"acos", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_acos)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100411 {"add", 2, 2, FEARG_1, ret_any, f_add},
412 {"and", 2, 2, FEARG_1, ret_number, f_and},
413 {"append", 2, 2, FEARG_LAST, ret_number, f_append},
414 {"appendbufline", 3, 3, FEARG_LAST, ret_number, f_appendbufline},
415 {"argc", 0, 1, 0, ret_number, f_argc},
416 {"argidx", 0, 0, 0, ret_number, f_argidx},
417 {"arglistid", 0, 2, 0, ret_number, f_arglistid},
418 {"argv", 0, 2, 0, ret_any, f_argv},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100419 {"asin", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_asin)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100420 {"assert_beeps", 1, 2, FEARG_1, ret_number, f_assert_beeps},
421 {"assert_equal", 2, 3, FEARG_2, ret_number, f_assert_equal},
422 {"assert_equalfile", 2, 2, FEARG_1, ret_number, f_assert_equalfile},
423 {"assert_exception", 1, 2, 0, ret_number, f_assert_exception},
424 {"assert_fails", 1, 3, FEARG_1, ret_number, f_assert_fails},
425 {"assert_false", 1, 2, FEARG_1, ret_number, f_assert_false},
426 {"assert_inrange", 3, 4, FEARG_3, ret_number, f_assert_inrange},
427 {"assert_match", 2, 3, FEARG_2, ret_number, f_assert_match},
428 {"assert_notequal", 2, 3, FEARG_2, ret_number, f_assert_notequal},
429 {"assert_notmatch", 2, 3, FEARG_2, ret_number, f_assert_notmatch},
430 {"assert_report", 1, 1, FEARG_1, ret_number, f_assert_report},
431 {"assert_true", 1, 2, FEARG_1, ret_number, f_assert_true},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100432 {"atan", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_atan)},
433 {"atan2", 2, 2, FEARG_1, ret_float, FLOAT_FUNC(f_atan2)},
434 {"balloon_gettext", 0, 0, 0, ret_string,
Bram Moolenaar59716a22017-03-01 20:32:44 +0100435#ifdef FEAT_BEVAL
Bram Moolenaar15c47602020-03-26 22:16:48 +0100436 f_balloon_gettext
437#else
438 NULL
Bram Moolenaar59716a22017-03-01 20:32:44 +0100439#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100440 },
441 {"balloon_show", 1, 1, FEARG_1, ret_void,
442#ifdef FEAT_BEVAL
443 f_balloon_show
444#else
445 NULL
446#endif
447 },
448 {"balloon_split", 1, 1, FEARG_1, ret_list_string,
449#if defined(FEAT_BEVAL_TERM)
450 f_balloon_split
451#else
452 NULL
453#endif
454 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100455 {"browse", 4, 4, 0, ret_string, f_browse},
456 {"browsedir", 2, 2, 0, ret_string, f_browsedir},
457 {"bufadd", 1, 1, FEARG_1, ret_number, f_bufadd},
458 {"bufexists", 1, 1, FEARG_1, ret_number, f_bufexists},
459 {"buffer_exists", 1, 1, FEARG_1, ret_number, f_bufexists}, // obsolete
460 {"buffer_name", 0, 1, FEARG_1, ret_string, f_bufname}, // obsolete
461 {"buffer_number", 0, 1, FEARG_1, ret_number, f_bufnr}, // obsolete
462 {"buflisted", 1, 1, FEARG_1, ret_number, f_buflisted},
463 {"bufload", 1, 1, FEARG_1, ret_void, f_bufload},
464 {"bufloaded", 1, 1, FEARG_1, ret_number, f_bufloaded},
465 {"bufname", 0, 1, FEARG_1, ret_string, f_bufname},
466 {"bufnr", 0, 2, FEARG_1, ret_number, f_bufnr},
467 {"bufwinid", 1, 1, FEARG_1, ret_number, f_bufwinid},
468 {"bufwinnr", 1, 1, FEARG_1, ret_number, f_bufwinnr},
469 {"byte2line", 1, 1, FEARG_1, ret_number, f_byte2line},
470 {"byteidx", 2, 2, FEARG_1, ret_number, f_byteidx},
471 {"byteidxcomp", 2, 2, FEARG_1, ret_number, f_byteidxcomp},
472 {"call", 2, 3, FEARG_1, ret_any, f_call},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100473 {"ceil", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_ceil)},
474 {"ch_canread", 1, 1, FEARG_1, ret_number, JOB_FUNC(f_ch_canread)},
475 {"ch_close", 1, 1, FEARG_1, ret_void, JOB_FUNC(f_ch_close)},
476 {"ch_close_in", 1, 1, FEARG_1, ret_void, JOB_FUNC(f_ch_close_in)},
477 {"ch_evalexpr", 2, 3, FEARG_1, ret_any, JOB_FUNC(f_ch_evalexpr)},
478 {"ch_evalraw", 2, 3, FEARG_1, ret_any, JOB_FUNC(f_ch_evalraw)},
479 {"ch_getbufnr", 2, 2, FEARG_1, ret_number, JOB_FUNC(f_ch_getbufnr)},
480 {"ch_getjob", 1, 1, FEARG_1, ret_job, JOB_FUNC(f_ch_getjob)},
481 {"ch_info", 1, 1, FEARG_1, ret_dict_any, JOB_FUNC(f_ch_info)},
482 {"ch_log", 1, 2, FEARG_1, ret_void, JOB_FUNC(f_ch_log)},
483 {"ch_logfile", 1, 2, FEARG_1, ret_void, JOB_FUNC(f_ch_logfile)},
484 {"ch_open", 1, 2, FEARG_1, ret_channel, JOB_FUNC(f_ch_open)},
485 {"ch_read", 1, 2, FEARG_1, ret_string, JOB_FUNC(f_ch_read)},
486 {"ch_readblob", 1, 2, FEARG_1, ret_blob, JOB_FUNC(f_ch_readblob)},
487 {"ch_readraw", 1, 2, FEARG_1, ret_string, JOB_FUNC(f_ch_readraw)},
488 {"ch_sendexpr", 2, 3, FEARG_1, ret_void, JOB_FUNC(f_ch_sendexpr)},
489 {"ch_sendraw", 2, 3, FEARG_1, ret_void, JOB_FUNC(f_ch_sendraw)},
490 {"ch_setoptions", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_ch_setoptions)},
491 {"ch_status", 1, 2, FEARG_1, ret_string, JOB_FUNC(f_ch_status)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100492 {"changenr", 0, 0, 0, ret_number, f_changenr},
493 {"char2nr", 1, 2, FEARG_1, ret_number, f_char2nr},
494 {"chdir", 1, 1, FEARG_1, ret_string, f_chdir},
495 {"cindent", 1, 1, FEARG_1, ret_number, f_cindent},
496 {"clearmatches", 0, 1, FEARG_1, ret_void, f_clearmatches},
497 {"col", 1, 1, FEARG_1, ret_number, f_col},
498 {"complete", 2, 2, FEARG_2, ret_void, f_complete},
499 {"complete_add", 1, 1, FEARG_1, ret_number, f_complete_add},
500 {"complete_check", 0, 0, 0, ret_number, f_complete_check},
501 {"complete_info", 0, 1, FEARG_1, ret_dict_any, f_complete_info},
502 {"confirm", 1, 4, FEARG_1, ret_number, f_confirm},
503 {"copy", 1, 1, FEARG_1, ret_any, f_copy},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100504 {"cos", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_cos)},
505 {"cosh", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_cosh)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100506 {"count", 2, 4, FEARG_1, ret_number, f_count},
507 {"cscope_connection",0,3, 0, ret_number, f_cscope_connection},
508 {"cursor", 1, 3, FEARG_1, ret_number, f_cursor},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100509 {"debugbreak", 1, 1, FEARG_1, ret_number,
Bram Moolenaar4f974752019-02-17 17:44:42 +0100510#ifdef MSWIN
Bram Moolenaar15c47602020-03-26 22:16:48 +0100511 f_debugbreak
512#else
513 NULL
Bram Moolenaar4551c0a2018-06-20 22:38:21 +0200514#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100515 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100516 {"deepcopy", 1, 2, FEARG_1, ret_any, f_deepcopy},
517 {"delete", 1, 2, FEARG_1, ret_number, f_delete},
518 {"deletebufline", 2, 3, FEARG_1, ret_number, f_deletebufline},
519 {"did_filetype", 0, 0, 0, ret_number, f_did_filetype},
520 {"diff_filler", 1, 1, FEARG_1, ret_number, f_diff_filler},
521 {"diff_hlID", 2, 2, FEARG_1, ret_number, f_diff_hlID},
522 {"echoraw", 1, 1, FEARG_1, ret_number, f_echoraw},
523 {"empty", 1, 1, FEARG_1, ret_number, f_empty},
524 {"environ", 0, 0, 0, ret_dict_string, f_environ},
525 {"escape", 2, 2, FEARG_1, ret_string, f_escape},
526 {"eval", 1, 1, FEARG_1, ret_any, f_eval},
527 {"eventhandler", 0, 0, 0, ret_number, f_eventhandler},
528 {"executable", 1, 1, FEARG_1, ret_number, f_executable},
529 {"execute", 1, 2, FEARG_1, ret_string, f_execute},
530 {"exepath", 1, 1, FEARG_1, ret_string, f_exepath},
531 {"exists", 1, 1, FEARG_1, ret_number, f_exists},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100532 {"exp", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_exp)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100533 {"expand", 1, 3, FEARG_1, ret_any, f_expand},
534 {"expandcmd", 1, 1, FEARG_1, ret_string, f_expandcmd},
535 {"extend", 2, 3, FEARG_1, ret_any, f_extend},
536 {"feedkeys", 1, 2, FEARG_1, ret_void, f_feedkeys},
537 {"file_readable", 1, 1, FEARG_1, ret_number, f_filereadable}, // obsolete
538 {"filereadable", 1, 1, FEARG_1, ret_number, f_filereadable},
539 {"filewritable", 1, 1, FEARG_1, ret_number, f_filewritable},
540 {"filter", 2, 2, FEARG_1, ret_any, f_filter},
541 {"finddir", 1, 3, FEARG_1, ret_string, f_finddir},
542 {"findfile", 1, 3, FEARG_1, ret_string, f_findfile},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100543 {"float2nr", 1, 1, FEARG_1, ret_number, FLOAT_FUNC(f_float2nr)},
544 {"floor", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_floor)},
545 {"fmod", 2, 2, FEARG_1, ret_float, FLOAT_FUNC(f_fmod)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100546 {"fnameescape", 1, 1, FEARG_1, ret_string, f_fnameescape},
547 {"fnamemodify", 2, 2, FEARG_1, ret_string, f_fnamemodify},
548 {"foldclosed", 1, 1, FEARG_1, ret_number, f_foldclosed},
549 {"foldclosedend", 1, 1, FEARG_1, ret_number, f_foldclosedend},
550 {"foldlevel", 1, 1, FEARG_1, ret_number, f_foldlevel},
551 {"foldtext", 0, 0, 0, ret_string, f_foldtext},
552 {"foldtextresult", 1, 1, FEARG_1, ret_string, f_foldtextresult},
553 {"foreground", 0, 0, 0, ret_void, f_foreground},
Bram Moolenaard77a8522020-04-03 21:59:57 +0200554 {"funcref", 1, 3, FEARG_1, ret_func_any, f_funcref},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100555 {"function", 1, 3, FEARG_1, ret_f_function, f_function},
556 {"garbagecollect", 0, 1, 0, ret_void, f_garbagecollect},
557 {"get", 2, 3, FEARG_1, ret_any, f_get},
558 {"getbufinfo", 0, 1, 0, ret_list_dict_any, f_getbufinfo},
559 {"getbufline", 2, 3, FEARG_1, ret_list_string, f_getbufline},
560 {"getbufvar", 2, 3, FEARG_1, ret_any, f_getbufvar},
561 {"getchangelist", 0, 1, FEARG_1, ret_list_any, f_getchangelist},
562 {"getchar", 0, 1, 0, ret_number, f_getchar},
563 {"getcharmod", 0, 0, 0, ret_number, f_getcharmod},
564 {"getcharsearch", 0, 0, 0, ret_dict_any, f_getcharsearch},
565 {"getcmdline", 0, 0, 0, ret_string, f_getcmdline},
566 {"getcmdpos", 0, 0, 0, ret_number, f_getcmdpos},
567 {"getcmdtype", 0, 0, 0, ret_string, f_getcmdtype},
568 {"getcmdwintype", 0, 0, 0, ret_string, f_getcmdwintype},
569 {"getcompletion", 2, 3, FEARG_1, ret_list_string, f_getcompletion},
570 {"getcurpos", 0, 0, 0, ret_list_number, f_getcurpos},
571 {"getcwd", 0, 2, FEARG_1, ret_string, f_getcwd},
572 {"getenv", 1, 1, FEARG_1, ret_string, f_getenv},
573 {"getfontname", 0, 1, 0, ret_string, f_getfontname},
574 {"getfperm", 1, 1, FEARG_1, ret_string, f_getfperm},
575 {"getfsize", 1, 1, FEARG_1, ret_number, f_getfsize},
576 {"getftime", 1, 1, FEARG_1, ret_number, f_getftime},
577 {"getftype", 1, 1, FEARG_1, ret_string, f_getftype},
578 {"getimstatus", 0, 0, 0, ret_number, f_getimstatus},
579 {"getjumplist", 0, 2, FEARG_1, ret_list_any, f_getjumplist},
580 {"getline", 1, 2, FEARG_1, ret_f_getline, f_getline},
581 {"getloclist", 1, 2, 0, ret_list_dict_any, f_getloclist},
Bram Moolenaarf17e7ea2020-06-01 14:14:44 +0200582 {"getmarklist", 0, 1, FEARG_1, ret_list_dict_any, f_getmarklist},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100583 {"getmatches", 0, 1, 0, ret_list_dict_any, f_getmatches},
584 {"getmousepos", 0, 0, 0, ret_dict_number, f_getmousepos},
585 {"getpid", 0, 0, 0, ret_number, f_getpid},
586 {"getpos", 1, 1, FEARG_1, ret_list_number, f_getpos},
587 {"getqflist", 0, 1, 0, ret_list_dict_any, f_getqflist},
588 {"getreg", 0, 3, FEARG_1, ret_string, f_getreg},
589 {"getregtype", 0, 1, FEARG_1, ret_string, f_getregtype},
590 {"gettabinfo", 0, 1, FEARG_1, ret_list_dict_any, f_gettabinfo},
591 {"gettabvar", 2, 3, FEARG_1, ret_any, f_gettabvar},
592 {"gettabwinvar", 3, 4, FEARG_1, ret_any, f_gettabwinvar},
593 {"gettagstack", 0, 1, FEARG_1, ret_dict_any, f_gettagstack},
594 {"getwininfo", 0, 1, FEARG_1, ret_list_dict_any, f_getwininfo},
595 {"getwinpos", 0, 1, FEARG_1, ret_list_number, f_getwinpos},
596 {"getwinposx", 0, 0, 0, ret_number, f_getwinposx},
597 {"getwinposy", 0, 0, 0, ret_number, f_getwinposy},
598 {"getwinvar", 2, 3, FEARG_1, ret_any, f_getwinvar},
599 {"glob", 1, 4, FEARG_1, ret_any, f_glob},
600 {"glob2regpat", 1, 1, FEARG_1, ret_string, f_glob2regpat},
601 {"globpath", 2, 5, FEARG_2, ret_any, f_globpath},
Bram Moolenaar79296512020-03-22 16:17:14 +0100602 {"has", 1, 2, 0, ret_number, f_has},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100603 {"has_key", 2, 2, FEARG_1, ret_number, f_has_key},
604 {"haslocaldir", 0, 2, FEARG_1, ret_number, f_haslocaldir},
605 {"hasmapto", 1, 3, FEARG_1, ret_number, f_hasmapto},
606 {"highlightID", 1, 1, FEARG_1, ret_number, f_hlID}, // obsolete
607 {"highlight_exists",1, 1, FEARG_1, ret_number, f_hlexists}, // obsolete
608 {"histadd", 2, 2, FEARG_2, ret_number, f_histadd},
609 {"histdel", 1, 2, FEARG_1, ret_number, f_histdel},
610 {"histget", 1, 2, FEARG_1, ret_string, f_histget},
611 {"histnr", 1, 1, FEARG_1, ret_number, f_histnr},
612 {"hlID", 1, 1, FEARG_1, ret_number, f_hlID},
613 {"hlexists", 1, 1, FEARG_1, ret_number, f_hlexists},
614 {"hostname", 0, 0, 0, ret_string, f_hostname},
615 {"iconv", 3, 3, FEARG_1, ret_string, f_iconv},
616 {"indent", 1, 1, FEARG_1, ret_number, f_indent},
617 {"index", 2, 4, FEARG_1, ret_number, f_index},
618 {"input", 1, 3, FEARG_1, ret_string, f_input},
619 {"inputdialog", 1, 3, FEARG_1, ret_string, f_inputdialog},
620 {"inputlist", 1, 1, FEARG_1, ret_number, f_inputlist},
621 {"inputrestore", 0, 0, 0, ret_number, f_inputrestore},
622 {"inputsave", 0, 0, 0, ret_number, f_inputsave},
623 {"inputsecret", 1, 2, FEARG_1, ret_string, f_inputsecret},
624 {"insert", 2, 3, FEARG_1, ret_any, f_insert},
625 {"interrupt", 0, 0, 0, ret_void, f_interrupt},
626 {"invert", 1, 1, FEARG_1, ret_number, f_invert},
627 {"isdirectory", 1, 1, FEARG_1, ret_number, f_isdirectory},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100628 {"isinf", 1, 1, FEARG_1, ret_number, MATH_FUNC(f_isinf)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100629 {"islocked", 1, 1, FEARG_1, ret_number, f_islocked},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100630 {"isnan", 1, 1, FEARG_1, ret_number, MATH_FUNC(f_isnan)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100631 {"items", 1, 1, FEARG_1, ret_list_any, f_items},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100632 {"job_getchannel", 1, 1, FEARG_1, ret_channel, JOB_FUNC(f_job_getchannel)},
633 {"job_info", 0, 1, FEARG_1, ret_dict_any, JOB_FUNC(f_job_info)},
634 {"job_setoptions", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_job_setoptions)},
635 {"job_start", 1, 2, FEARG_1, ret_job, JOB_FUNC(f_job_start)},
636 {"job_status", 1, 1, FEARG_1, ret_string, JOB_FUNC(f_job_status)},
637 {"job_stop", 1, 2, FEARG_1, ret_number, JOB_FUNC(f_job_stop)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100638 {"join", 1, 2, FEARG_1, ret_string, f_join},
639 {"js_decode", 1, 1, FEARG_1, ret_any, f_js_decode},
640 {"js_encode", 1, 1, FEARG_1, ret_string, f_js_encode},
641 {"json_decode", 1, 1, FEARG_1, ret_any, f_json_decode},
642 {"json_encode", 1, 1, FEARG_1, ret_string, f_json_encode},
643 {"keys", 1, 1, FEARG_1, ret_list_any, f_keys},
644 {"last_buffer_nr", 0, 0, 0, ret_number, f_last_buffer_nr}, // obsolete
645 {"len", 1, 1, FEARG_1, ret_number, f_len},
646 {"libcall", 3, 3, FEARG_3, ret_string, f_libcall},
647 {"libcallnr", 3, 3, FEARG_3, ret_number, f_libcallnr},
648 {"line", 1, 2, FEARG_1, ret_number, f_line},
649 {"line2byte", 1, 1, FEARG_1, ret_number, f_line2byte},
650 {"lispindent", 1, 1, FEARG_1, ret_number, f_lispindent},
651 {"list2str", 1, 2, FEARG_1, ret_string, f_list2str},
652 {"listener_add", 1, 2, FEARG_2, ret_number, f_listener_add},
653 {"listener_flush", 0, 1, FEARG_1, ret_void, f_listener_flush},
654 {"listener_remove", 1, 1, FEARG_1, ret_number, f_listener_remove},
655 {"localtime", 0, 0, 0, ret_number, f_localtime},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100656 {"log", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_log)},
657 {"log10", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_log10)},
658 {"luaeval", 1, 2, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200659#ifdef FEAT_LUA
Bram Moolenaar15c47602020-03-26 22:16:48 +0100660 f_luaeval
661#else
662 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200663#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100664 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100665 {"map", 2, 2, FEARG_1, ret_any, f_map},
666 {"maparg", 1, 4, FEARG_1, ret_string, f_maparg},
667 {"mapcheck", 1, 3, FEARG_1, ret_string, f_mapcheck},
Bram Moolenaar4c9243f2020-05-22 13:10:44 +0200668 {"mapset", 3, 3, FEARG_1, ret_void, f_mapset},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100669 {"match", 2, 4, FEARG_1, ret_any, f_match},
670 {"matchadd", 2, 5, FEARG_1, ret_number, f_matchadd},
671 {"matchaddpos", 2, 5, FEARG_1, ret_number, f_matchaddpos},
672 {"matcharg", 1, 1, FEARG_1, ret_list_string, f_matcharg},
673 {"matchdelete", 1, 2, FEARG_1, ret_number, f_matchdelete},
674 {"matchend", 2, 4, FEARG_1, ret_number, f_matchend},
675 {"matchlist", 2, 4, FEARG_1, ret_list_string, f_matchlist},
676 {"matchstr", 2, 4, FEARG_1, ret_string, f_matchstr},
677 {"matchstrpos", 2, 4, FEARG_1, ret_list_any, f_matchstrpos},
678 {"max", 1, 1, FEARG_1, ret_any, f_max},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100679 {"menu_info", 1, 2, FEARG_1, ret_dict_any,
Bram Moolenaara2cbdea2020-03-16 21:08:31 +0100680#ifdef FEAT_MENU
Bram Moolenaar15c47602020-03-26 22:16:48 +0100681 f_menu_info
682#else
683 NULL
Bram Moolenaara2cbdea2020-03-16 21:08:31 +0100684#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100685 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100686 {"min", 1, 1, FEARG_1, ret_any, f_min},
687 {"mkdir", 1, 3, FEARG_1, ret_number, f_mkdir},
688 {"mode", 0, 1, FEARG_1, ret_string, f_mode},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100689 {"mzeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200690#ifdef FEAT_MZSCHEME
Bram Moolenaar15c47602020-03-26 22:16:48 +0100691 f_mzeval
692#else
693 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200694#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100695 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100696 {"nextnonblank", 1, 1, FEARG_1, ret_number, f_nextnonblank},
697 {"nr2char", 1, 2, FEARG_1, ret_string, f_nr2char},
698 {"or", 2, 2, FEARG_1, ret_number, f_or},
699 {"pathshorten", 1, 1, FEARG_1, ret_string, f_pathshorten},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100700 {"perleval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200701#ifdef FEAT_PERL
Bram Moolenaar15c47602020-03-26 22:16:48 +0100702 f_perleval
703#else
704 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200705#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100706 },
707 {"popup_atcursor", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_atcursor)},
708 {"popup_beval", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_beval)},
Bram Moolenaar03a9f842020-05-13 13:40:16 +0200709 {"popup_clear", 0, 1, 0, ret_void, PROP_FUNC(f_popup_clear)},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100710 {"popup_close", 1, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_close)},
711 {"popup_create", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_create)},
712 {"popup_dialog", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_dialog)},
713 {"popup_filter_menu", 2, 2, 0, ret_number, PROP_FUNC(f_popup_filter_menu)},
714 {"popup_filter_yesno", 2, 2, 0, ret_number, PROP_FUNC(f_popup_filter_yesno)},
715 {"popup_findinfo", 0, 0, 0, ret_number, PROP_FUNC(f_popup_findinfo)},
716 {"popup_findpreview", 0, 0, 0, ret_number, PROP_FUNC(f_popup_findpreview)},
717 {"popup_getoptions", 1, 1, FEARG_1, ret_dict_any, PROP_FUNC(f_popup_getoptions)},
718 {"popup_getpos", 1, 1, FEARG_1, ret_dict_any, PROP_FUNC(f_popup_getpos)},
719 {"popup_hide", 1, 1, FEARG_1, ret_void, PROP_FUNC(f_popup_hide)},
Bram Moolenaaref6b9792020-05-13 16:34:15 +0200720 {"popup_list", 0, 0, 0, ret_list_number, PROP_FUNC(f_popup_list)},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100721 {"popup_locate", 2, 2, 0, ret_number, PROP_FUNC(f_popup_locate)},
722 {"popup_menu", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_menu)},
723 {"popup_move", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_move)},
724 {"popup_notification", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_notification)},
725 {"popup_setoptions", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_setoptions)},
726 {"popup_settext", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_settext)},
727 {"popup_show", 1, 1, FEARG_1, ret_void, PROP_FUNC(f_popup_show)},
728 {"pow", 2, 2, FEARG_1, ret_float, FLOAT_FUNC(f_pow)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100729 {"prevnonblank", 1, 1, FEARG_1, ret_number, f_prevnonblank},
730 {"printf", 1, 19, FEARG_2, ret_string, f_printf},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100731 {"prompt_setcallback", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_prompt_setcallback)},
732 {"prompt_setinterrupt", 2, 2, FEARG_1,ret_void, JOB_FUNC(f_prompt_setinterrupt)},
733 {"prompt_setprompt", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_prompt_setprompt)},
734 {"prop_add", 3, 3, FEARG_1, ret_void, PROP_FUNC(f_prop_add)},
735 {"prop_clear", 1, 3, FEARG_1, ret_void, PROP_FUNC(f_prop_clear)},
736 {"prop_find", 1, 2, FEARG_1, ret_dict_any, PROP_FUNC(f_prop_find)},
737 {"prop_list", 1, 2, FEARG_1, ret_list_dict_any, PROP_FUNC(f_prop_list)},
738 {"prop_remove", 1, 3, FEARG_1, ret_number, PROP_FUNC(f_prop_remove)},
739 {"prop_type_add", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_prop_type_add)},
740 {"prop_type_change", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_prop_type_change)},
741 {"prop_type_delete", 1, 2, FEARG_1, ret_void, PROP_FUNC(f_prop_type_delete)},
742 {"prop_type_get", 1, 2, FEARG_1, ret_dict_any, PROP_FUNC(f_prop_type_get)},
743 {"prop_type_list", 0, 1, FEARG_1, ret_list_string, PROP_FUNC(f_prop_type_list)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100744 {"pum_getpos", 0, 0, 0, ret_dict_number, f_pum_getpos},
745 {"pumvisible", 0, 0, 0, ret_number, f_pumvisible},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100746 {"py3eval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200747#ifdef FEAT_PYTHON3
Bram Moolenaar15c47602020-03-26 22:16:48 +0100748 f_py3eval
749#else
750 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200751#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100752 },
753 {"pyeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200754#ifdef FEAT_PYTHON
Bram Moolenaar15c47602020-03-26 22:16:48 +0100755 f_pyeval
756#else
757 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200758#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100759 },
760 {"pyxeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100761#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
Bram Moolenaar15c47602020-03-26 22:16:48 +0100762 f_pyxeval
763#else
764 NULL
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100765#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100766 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100767 {"rand", 0, 1, FEARG_1, ret_number, f_rand},
768 {"range", 1, 3, FEARG_1, ret_list_number, f_range},
769 {"readdir", 1, 2, FEARG_1, ret_list_string, f_readdir},
Bram Moolenaar6c9ba042020-06-01 16:09:41 +0200770 {"readdirex", 1, 2, FEARG_1, ret_list_dict_any, f_readdirex},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100771 {"readfile", 1, 3, FEARG_1, ret_any, f_readfile},
772 {"reg_executing", 0, 0, 0, ret_string, f_reg_executing},
773 {"reg_recording", 0, 0, 0, ret_string, f_reg_recording},
774 {"reltime", 0, 2, FEARG_1, ret_list_any, f_reltime},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100775 {"reltimefloat", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_reltimefloat)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100776 {"reltimestr", 1, 1, FEARG_1, ret_string, f_reltimestr},
777 {"remote_expr", 2, 4, FEARG_1, ret_string, f_remote_expr},
778 {"remote_foreground", 1, 1, FEARG_1, ret_string, f_remote_foreground},
779 {"remote_peek", 1, 2, FEARG_1, ret_number, f_remote_peek},
780 {"remote_read", 1, 2, FEARG_1, ret_string, f_remote_read},
781 {"remote_send", 2, 3, FEARG_1, ret_string, f_remote_send},
782 {"remote_startserver", 1, 1, FEARG_1, ret_void, f_remote_startserver},
783 {"remove", 2, 3, FEARG_1, ret_any, f_remove},
784 {"rename", 2, 2, FEARG_1, ret_number, f_rename},
785 {"repeat", 2, 2, FEARG_1, ret_any, f_repeat},
786 {"resolve", 1, 1, FEARG_1, ret_string, f_resolve},
787 {"reverse", 1, 1, FEARG_1, ret_any, f_reverse},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100788 {"round", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_round)},
789 {"rubyeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100790#ifdef FEAT_RUBY
Bram Moolenaar15c47602020-03-26 22:16:48 +0100791 f_rubyeval
792#else
793 NULL
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100794#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100795 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100796 {"screenattr", 2, 2, FEARG_1, ret_number, f_screenattr},
797 {"screenchar", 2, 2, FEARG_1, ret_number, f_screenchar},
798 {"screenchars", 2, 2, FEARG_1, ret_list_number, f_screenchars},
799 {"screencol", 0, 0, 0, ret_number, f_screencol},
800 {"screenpos", 3, 3, FEARG_1, ret_dict_number, f_screenpos},
801 {"screenrow", 0, 0, 0, ret_number, f_screenrow},
802 {"screenstring", 2, 2, FEARG_1, ret_string, f_screenstring},
803 {"search", 1, 4, FEARG_1, ret_number, f_search},
804 {"searchdecl", 1, 3, FEARG_1, ret_number, f_searchdecl},
805 {"searchpair", 3, 7, 0, ret_number, f_searchpair},
806 {"searchpairpos", 3, 7, 0, ret_list_number, f_searchpairpos},
807 {"searchpos", 1, 4, FEARG_1, ret_list_number, f_searchpos},
808 {"server2client", 2, 2, FEARG_1, ret_number, f_server2client},
809 {"serverlist", 0, 0, 0, ret_string, f_serverlist},
810 {"setbufline", 3, 3, FEARG_3, ret_number, f_setbufline},
811 {"setbufvar", 3, 3, FEARG_3, ret_void, f_setbufvar},
812 {"setcharsearch", 1, 1, FEARG_1, ret_void, f_setcharsearch},
813 {"setcmdpos", 1, 1, FEARG_1, ret_number, f_setcmdpos},
814 {"setenv", 2, 2, FEARG_2, ret_void, f_setenv},
815 {"setfperm", 2, 2, FEARG_1, ret_number, f_setfperm},
816 {"setline", 2, 2, FEARG_2, ret_number, f_setline},
817 {"setloclist", 2, 4, FEARG_2, ret_number, f_setloclist},
818 {"setmatches", 1, 2, FEARG_1, ret_number, f_setmatches},
819 {"setpos", 2, 2, FEARG_2, ret_number, f_setpos},
820 {"setqflist", 1, 3, FEARG_1, ret_number, f_setqflist},
821 {"setreg", 2, 3, FEARG_2, ret_number, f_setreg},
822 {"settabvar", 3, 3, FEARG_3, ret_void, f_settabvar},
823 {"settabwinvar", 4, 4, FEARG_4, ret_void, f_settabwinvar},
824 {"settagstack", 2, 3, FEARG_2, ret_number, f_settagstack},
825 {"setwinvar", 3, 3, FEARG_3, ret_void, f_setwinvar},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100826 {"sha256", 1, 1, FEARG_1, ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200827#ifdef FEAT_CRYPT
Bram Moolenaar15c47602020-03-26 22:16:48 +0100828 f_sha256
829#else
830 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200831#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100832 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100833 {"shellescape", 1, 2, FEARG_1, ret_string, f_shellescape},
834 {"shiftwidth", 0, 1, FEARG_1, ret_number, f_shiftwidth},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100835 {"sign_define", 1, 2, FEARG_1, ret_any, SIGN_FUNC(f_sign_define)},
836 {"sign_getdefined", 0, 1, FEARG_1, ret_list_dict_any, SIGN_FUNC(f_sign_getdefined)},
837 {"sign_getplaced", 0, 2, FEARG_1, ret_list_dict_any, SIGN_FUNC(f_sign_getplaced)},
838 {"sign_jump", 3, 3, FEARG_1, ret_number, SIGN_FUNC(f_sign_jump)},
839 {"sign_place", 4, 5, FEARG_1, ret_number, SIGN_FUNC(f_sign_place)},
840 {"sign_placelist", 1, 1, FEARG_1, ret_list_number, SIGN_FUNC(f_sign_placelist)},
841 {"sign_undefine", 0, 1, FEARG_1, ret_number, SIGN_FUNC(f_sign_undefine)},
842 {"sign_unplace", 1, 2, FEARG_1, ret_number, SIGN_FUNC(f_sign_unplace)},
843 {"sign_unplacelist", 1, 2, FEARG_1, ret_list_number, SIGN_FUNC(f_sign_unplacelist)},
Bram Moolenaar7035fd92020-04-08 20:03:52 +0200844 {"simplify", 1, 1, FEARG_1, ret_string, f_simplify},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100845 {"sin", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_sin)},
846 {"sinh", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_sinh)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100847 {"sort", 1, 3, FEARG_1, ret_list_any, f_sort},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100848 {"sound_clear", 0, 0, 0, ret_void, SOUND_FUNC(f_sound_clear)},
849 {"sound_playevent", 1, 2, FEARG_1, ret_number, SOUND_FUNC(f_sound_playevent)},
850 {"sound_playfile", 1, 2, FEARG_1, ret_number, SOUND_FUNC(f_sound_playfile)},
851 {"sound_stop", 1, 1, FEARG_1, ret_void, SOUND_FUNC(f_sound_stop)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100852 {"soundfold", 1, 1, FEARG_1, ret_string, f_soundfold},
853 {"spellbadword", 0, 1, FEARG_1, ret_list_string, f_spellbadword},
854 {"spellsuggest", 1, 3, FEARG_1, ret_list_string, f_spellsuggest},
855 {"split", 1, 3, FEARG_1, ret_list_string, f_split},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100856 {"sqrt", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_sqrt)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100857 {"srand", 0, 1, FEARG_1, ret_list_number, f_srand},
858 {"state", 0, 1, FEARG_1, ret_string, f_state},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100859 {"str2float", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_str2float)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100860 {"str2list", 1, 2, FEARG_1, ret_list_number, f_str2list},
861 {"str2nr", 1, 3, FEARG_1, ret_number, f_str2nr},
862 {"strcharpart", 2, 3, FEARG_1, ret_string, f_strcharpart},
863 {"strchars", 1, 2, FEARG_1, ret_number, f_strchars},
864 {"strdisplaywidth", 1, 2, FEARG_1, ret_number, f_strdisplaywidth},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100865 {"strftime", 1, 2, FEARG_1, ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200866#ifdef HAVE_STRFTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +0100867 f_strftime
868#else
869 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200870#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100871 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100872 {"strgetchar", 2, 2, FEARG_1, ret_number, f_strgetchar},
873 {"stridx", 2, 3, FEARG_1, ret_number, f_stridx},
874 {"string", 1, 1, FEARG_1, ret_string, f_string},
875 {"strlen", 1, 1, FEARG_1, ret_number, f_strlen},
876 {"strpart", 2, 3, FEARG_1, ret_string, f_strpart},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100877 {"strptime", 2, 2, FEARG_1, ret_number,
Bram Moolenaar10455d42019-11-21 15:36:18 +0100878#ifdef HAVE_STRPTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +0100879 f_strptime
880#else
881 NULL
Bram Moolenaar10455d42019-11-21 15:36:18 +0100882#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100883 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100884 {"strridx", 2, 3, FEARG_1, ret_number, f_strridx},
885 {"strtrans", 1, 1, FEARG_1, ret_string, f_strtrans},
886 {"strwidth", 1, 1, FEARG_1, ret_number, f_strwidth},
887 {"submatch", 1, 2, FEARG_1, ret_string, f_submatch},
888 {"substitute", 4, 4, FEARG_1, ret_string, f_substitute},
889 {"swapinfo", 1, 1, FEARG_1, ret_dict_any, f_swapinfo},
890 {"swapname", 1, 1, FEARG_1, ret_string, f_swapname},
891 {"synID", 3, 3, 0, ret_number, f_synID},
892 {"synIDattr", 2, 3, FEARG_1, ret_string, f_synIDattr},
893 {"synIDtrans", 1, 1, FEARG_1, ret_number, f_synIDtrans},
894 {"synconcealed", 2, 2, 0, ret_list_any, f_synconcealed},
895 {"synstack", 2, 2, 0, ret_list_number, f_synstack},
896 {"system", 1, 2, FEARG_1, ret_string, f_system},
897 {"systemlist", 1, 2, FEARG_1, ret_list_string, f_systemlist},
898 {"tabpagebuflist", 0, 1, FEARG_1, ret_list_number, f_tabpagebuflist},
899 {"tabpagenr", 0, 1, 0, ret_number, f_tabpagenr},
900 {"tabpagewinnr", 1, 2, FEARG_1, ret_number, f_tabpagewinnr},
901 {"tagfiles", 0, 0, 0, ret_list_string, f_tagfiles},
902 {"taglist", 1, 2, FEARG_1, ret_list_dict_any, f_taglist},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100903 {"tan", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_tan)},
904 {"tanh", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_tanh)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100905 {"tempname", 0, 0, 0, ret_string, f_tempname},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100906 {"term_dumpdiff", 2, 3, FEARG_1, ret_number, TERM_FUNC(f_term_dumpdiff)},
907 {"term_dumpload", 1, 2, FEARG_1, ret_number, TERM_FUNC(f_term_dumpload)},
908 {"term_dumpwrite", 2, 3, FEARG_2, ret_void, TERM_FUNC(f_term_dumpwrite)},
909 {"term_getaltscreen", 1, 1, FEARG_1, ret_number, TERM_FUNC(f_term_getaltscreen)},
910 {"term_getansicolors", 1, 1, FEARG_1, ret_list_string,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +0100911#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +0100912 f_term_getansicolors
913#else
914 NULL
Bram Moolenaarc6df10e2017-07-29 20:15:08 +0200915#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100916 },
917 {"term_getattr", 2, 2, FEARG_1, ret_number, TERM_FUNC(f_term_getattr)},
918 {"term_getcursor", 1, 1, FEARG_1, ret_list_any, TERM_FUNC(f_term_getcursor)},
919 {"term_getjob", 1, 1, FEARG_1, ret_job, TERM_FUNC(f_term_getjob)},
920 {"term_getline", 2, 2, FEARG_1, ret_string, TERM_FUNC(f_term_getline)},
921 {"term_getscrolled", 1, 1, FEARG_1, ret_number, TERM_FUNC(f_term_getscrolled)},
922 {"term_getsize", 1, 1, FEARG_1, ret_list_number, TERM_FUNC(f_term_getsize)},
923 {"term_getstatus", 1, 1, FEARG_1, ret_string, TERM_FUNC(f_term_getstatus)},
924 {"term_gettitle", 1, 1, FEARG_1, ret_string, TERM_FUNC(f_term_gettitle)},
925 {"term_gettty", 1, 2, FEARG_1, ret_string, TERM_FUNC(f_term_gettty)},
926 {"term_list", 0, 0, 0, ret_list_number, TERM_FUNC(f_term_list)},
927 {"term_scrape", 2, 2, FEARG_1, ret_list_dict_any, TERM_FUNC(f_term_scrape)},
928 {"term_sendkeys", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_sendkeys)},
929 {"term_setansicolors", 2, 2, FEARG_1, ret_void,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +0100930#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +0100931 f_term_setansicolors
932#else
933 NULL
934#endif
935 },
936 {"term_setapi", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_setapi)},
937 {"term_setkill", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_setkill)},
938 {"term_setrestore", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_setrestore)},
939 {"term_setsize", 3, 3, FEARG_1, ret_void, TERM_FUNC(f_term_setsize)},
940 {"term_start", 1, 2, FEARG_1, ret_number, TERM_FUNC(f_term_start)},
941 {"term_wait", 1, 2, FEARG_1, ret_void, TERM_FUNC(f_term_wait)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100942 {"test_alloc_fail", 3, 3, FEARG_1, ret_void, f_test_alloc_fail},
943 {"test_autochdir", 0, 0, 0, ret_void, f_test_autochdir},
944 {"test_feedinput", 1, 1, FEARG_1, ret_void, f_test_feedinput},
945 {"test_garbagecollect_now", 0, 0, 0, ret_void, f_test_garbagecollect_now},
946 {"test_garbagecollect_soon", 0, 0, 0, ret_void, f_test_garbagecollect_soon},
947 {"test_getvalue", 1, 1, FEARG_1, ret_number, f_test_getvalue},
948 {"test_ignore_error", 1, 1, FEARG_1, ret_void, f_test_ignore_error},
949 {"test_null_blob", 0, 0, 0, ret_blob, f_test_null_blob},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100950 {"test_null_channel", 0, 0, 0, ret_channel, JOB_FUNC(f_test_null_channel)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100951 {"test_null_dict", 0, 0, 0, ret_dict_any, f_test_null_dict},
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200952 {"test_null_function", 0, 0, 0, ret_func_any, f_test_null_function},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100953 {"test_null_job", 0, 0, 0, ret_job, JOB_FUNC(f_test_null_job)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100954 {"test_null_list", 0, 0, 0, ret_list_any, f_test_null_list},
Bram Moolenaard77a8522020-04-03 21:59:57 +0200955 {"test_null_partial", 0, 0, 0, ret_func_any, f_test_null_partial},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100956 {"test_null_string", 0, 0, 0, ret_string, f_test_null_string},
957 {"test_option_not_set", 1, 1, FEARG_1,ret_void, f_test_option_not_set},
958 {"test_override", 2, 2, FEARG_2, ret_void, f_test_override},
959 {"test_refcount", 1, 1, FEARG_1, ret_number, f_test_refcount},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100960 {"test_scrollbar", 3, 3, FEARG_2, ret_void,
Bram Moolenaarab186732018-09-14 21:27:06 +0200961#ifdef FEAT_GUI
Bram Moolenaar15c47602020-03-26 22:16:48 +0100962 f_test_scrollbar
963#else
964 NULL
Bram Moolenaarab186732018-09-14 21:27:06 +0200965#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100966 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100967 {"test_setmouse", 2, 2, 0, ret_void, f_test_setmouse},
968 {"test_settime", 1, 1, FEARG_1, ret_void, f_test_settime},
969 {"test_srand_seed", 0, 1, FEARG_1, ret_void, f_test_srand_seed},
970 {"test_unknown", 0, 0, 0, ret_any, f_test_unknown},
971 {"test_void", 0, 0, 0, ret_any, f_test_void},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100972 {"timer_info", 0, 1, FEARG_1, ret_list_dict_any, TIMER_FUNC(f_timer_info)},
973 {"timer_pause", 2, 2, FEARG_1, ret_void, TIMER_FUNC(f_timer_pause)},
974 {"timer_start", 2, 3, FEARG_1, ret_number, TIMER_FUNC(f_timer_start)},
975 {"timer_stop", 1, 1, FEARG_1, ret_void, TIMER_FUNC(f_timer_stop)},
976 {"timer_stopall", 0, 0, 0, ret_void, TIMER_FUNC(f_timer_stopall)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100977 {"tolower", 1, 1, FEARG_1, ret_string, f_tolower},
978 {"toupper", 1, 1, FEARG_1, ret_string, f_toupper},
979 {"tr", 3, 3, FEARG_1, ret_string, f_tr},
Bram Moolenaar2245ae12020-05-31 22:20:36 +0200980 {"trim", 1, 3, FEARG_1, ret_string, f_trim},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100981 {"trunc", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_trunc)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100982 {"type", 1, 1, FEARG_1, ret_number, f_type},
983 {"undofile", 1, 1, FEARG_1, ret_string, f_undofile},
984 {"undotree", 0, 0, 0, ret_dict_any, f_undotree},
985 {"uniq", 1, 3, FEARG_1, ret_list_any, f_uniq},
986 {"values", 1, 1, FEARG_1, ret_list_any, f_values},
987 {"virtcol", 1, 1, FEARG_1, ret_number, f_virtcol},
988 {"visualmode", 0, 1, 0, ret_string, f_visualmode},
989 {"wildmenumode", 0, 0, 0, ret_number, f_wildmenumode},
990 {"win_execute", 2, 3, FEARG_2, ret_string, f_win_execute},
991 {"win_findbuf", 1, 1, FEARG_1, ret_list_number, f_win_findbuf},
992 {"win_getid", 0, 2, FEARG_1, ret_number, f_win_getid},
993 {"win_gettype", 0, 1, FEARG_1, ret_string, f_win_gettype},
994 {"win_gotoid", 1, 1, FEARG_1, ret_number, f_win_gotoid},
995 {"win_id2tabwin", 1, 1, FEARG_1, ret_list_number, f_win_id2tabwin},
996 {"win_id2win", 1, 1, FEARG_1, ret_number, f_win_id2win},
997 {"win_screenpos", 1, 1, FEARG_1, ret_list_number, f_win_screenpos},
998 {"win_splitmove", 2, 3, FEARG_1, ret_number, f_win_splitmove},
999 {"winbufnr", 1, 1, FEARG_1, ret_number, f_winbufnr},
1000 {"wincol", 0, 0, 0, ret_number, f_wincol},
1001 {"windowsversion", 0, 0, 0, ret_string, f_windowsversion},
1002 {"winheight", 1, 1, FEARG_1, ret_number, f_winheight},
1003 {"winlayout", 0, 1, FEARG_1, ret_list_any, f_winlayout},
1004 {"winline", 0, 0, 0, ret_number, f_winline},
1005 {"winnr", 0, 1, FEARG_1, ret_number, f_winnr},
1006 {"winrestcmd", 0, 0, 0, ret_string, f_winrestcmd},
1007 {"winrestview", 1, 1, FEARG_1, ret_void, f_winrestview},
1008 {"winsaveview", 0, 0, 0, ret_dict_any, f_winsaveview},
1009 {"winwidth", 1, 1, FEARG_1, ret_number, f_winwidth},
1010 {"wordcount", 0, 0, 0, ret_dict_number, f_wordcount},
1011 {"writefile", 2, 3, FEARG_1, ret_number, f_writefile},
1012 {"xor", 2, 2, FEARG_1, ret_number, f_xor},
Bram Moolenaarac92e252019-08-03 21:58:38 +02001013};
1014
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001015/*
1016 * Function given to ExpandGeneric() to obtain the list of internal
1017 * or user defined function names.
1018 */
1019 char_u *
1020get_function_name(expand_T *xp, int idx)
1021{
1022 static int intidx = -1;
1023 char_u *name;
1024
1025 if (idx == 0)
1026 intidx = -1;
1027 if (intidx < 0)
1028 {
1029 name = get_user_func_name(xp, idx);
1030 if (name != NULL)
1031 return name;
1032 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001033 if (++intidx < (int)(sizeof(global_functions) / sizeof(funcentry_T)))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001034 {
Bram Moolenaarac92e252019-08-03 21:58:38 +02001035 STRCPY(IObuff, global_functions[intidx].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001036 STRCAT(IObuff, "(");
Bram Moolenaarac92e252019-08-03 21:58:38 +02001037 if (global_functions[intidx].f_max_argc == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001038 STRCAT(IObuff, ")");
1039 return IObuff;
1040 }
1041
1042 return NULL;
1043}
1044
1045/*
1046 * Function given to ExpandGeneric() to obtain the list of internal or
1047 * user defined variable or function names.
1048 */
1049 char_u *
1050get_expr_name(expand_T *xp, int idx)
1051{
1052 static int intidx = -1;
1053 char_u *name;
1054
1055 if (idx == 0)
1056 intidx = -1;
1057 if (intidx < 0)
1058 {
1059 name = get_function_name(xp, idx);
1060 if (name != NULL)
1061 return name;
1062 }
1063 return get_user_var_name(xp, ++intidx);
1064}
1065
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001066/*
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001067 * Find internal function "name" in table "global_functions".
Bram Moolenaar15c47602020-03-26 22:16:48 +01001068 * Return index, or -1 if not found or "implemented" is TRUE and the function
1069 * is not implemented.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001070 */
Bram Moolenaar15c47602020-03-26 22:16:48 +01001071 static int
1072find_internal_func_opt(char_u *name, int implemented)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001073{
1074 int first = 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001075 int last;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001076 int cmp;
1077 int x;
1078
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001079 last = (int)(sizeof(global_functions) / sizeof(funcentry_T)) - 1;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001080
1081 // Find the function name in the table. Binary search.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001082 while (first <= last)
1083 {
1084 x = first + ((unsigned)(last - first) >> 1);
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001085 cmp = STRCMP(name, global_functions[x].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001086 if (cmp < 0)
1087 last = x - 1;
1088 else if (cmp > 0)
1089 first = x + 1;
Bram Moolenaar15c47602020-03-26 22:16:48 +01001090 else if (implemented && global_functions[x].f_func == NULL)
1091 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001092 else
1093 return x;
1094 }
1095 return -1;
1096}
1097
Bram Moolenaar15c47602020-03-26 22:16:48 +01001098/*
1099 * Find internal function "name" in table "global_functions".
1100 * Return index, or -1 if not found or the function is not implemented.
1101 */
1102 int
1103find_internal_func(char_u *name)
1104{
1105 return find_internal_func_opt(name, TRUE);
1106}
1107
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001108 int
Bram Moolenaarac92e252019-08-03 21:58:38 +02001109has_internal_func(char_u *name)
1110{
Bram Moolenaar15c47602020-03-26 22:16:48 +01001111 return find_internal_func_opt(name, TRUE) >= 0;
1112}
1113
1114 static int
1115has_internal_func_name(char_u *name)
1116{
1117 return find_internal_func_opt(name, FALSE) >= 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001118}
1119
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001120 char *
1121internal_func_name(int idx)
1122{
1123 return global_functions[idx].f_name;
1124}
1125
1126 type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001127internal_func_ret_type(int idx, int argcount, type_T **argtypes)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001128{
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001129 return global_functions[idx].f_retfunc(argcount, argtypes);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001130}
1131
1132/*
1133 * Check the argument count to use for internal function "idx".
1134 * Returns OK or FAIL;
1135 */
1136 int
1137check_internal_func(int idx, int argcount)
1138{
1139 int res;
1140 char *name;
1141
1142 if (argcount < global_functions[idx].f_min_argc)
1143 res = FCERR_TOOFEW;
1144 else if (argcount > global_functions[idx].f_max_argc)
1145 res = FCERR_TOOMANY;
1146 else
1147 return OK;
1148
1149 name = internal_func_name(idx);
1150 if (res == FCERR_TOOMANY)
1151 semsg(_(e_toomanyarg), name);
1152 else
1153 semsg(_(e_toofewarg), name);
1154 return FAIL;
1155}
1156
Bram Moolenaarac92e252019-08-03 21:58:38 +02001157 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001158call_internal_func(
1159 char_u *name,
1160 int argcount,
1161 typval_T *argvars,
1162 typval_T *rettv)
1163{
1164 int i;
1165
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001166 i = find_internal_func(name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001167 if (i < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001168 return FCERR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001169 if (argcount < global_functions[i].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001170 return FCERR_TOOFEW;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001171 if (argcount > global_functions[i].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001172 return FCERR_TOOMANY;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001173 argvars[argcount].v_type = VAR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001174 global_functions[i].f_func(argvars, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001175 return FCERR_NONE;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001176}
1177
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001178 void
1179call_internal_func_by_idx(
1180 int idx,
1181 typval_T *argvars,
1182 typval_T *rettv)
1183{
1184 global_functions[idx].f_func(argvars, rettv);
1185}
1186
Bram Moolenaarac92e252019-08-03 21:58:38 +02001187/*
1188 * Invoke a method for base->method().
1189 */
1190 int
1191call_internal_method(
1192 char_u *name,
1193 int argcount,
1194 typval_T *argvars,
1195 typval_T *rettv,
1196 typval_T *basetv)
1197{
1198 int i;
1199 int fi;
1200 typval_T argv[MAX_FUNC_ARGS + 1];
1201
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001202 fi = find_internal_func(name);
Bram Moolenaar91746392019-08-16 22:22:31 +02001203 if (fi < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001204 return FCERR_UNKNOWN;
Bram Moolenaar91746392019-08-16 22:22:31 +02001205 if (global_functions[fi].f_argtype == 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001206 return FCERR_NOTMETHOD;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001207 if (argcount + 1 < global_functions[fi].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001208 return FCERR_TOOFEW;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001209 if (argcount + 1 > global_functions[fi].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001210 return FCERR_TOOMANY;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001211
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001212 if (global_functions[fi].f_argtype == FEARG_LAST)
Bram Moolenaar25e42232019-08-04 15:04:10 +02001213 {
1214 // base value goes last
1215 for (i = 0; i < argcount; ++i)
1216 argv[i] = argvars[i];
1217 argv[argcount] = *basetv;
1218 }
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001219 else if (global_functions[fi].f_argtype == FEARG_2)
Bram Moolenaar25e42232019-08-04 15:04:10 +02001220 {
1221 // base value goes second
1222 argv[0] = argvars[0];
1223 argv[1] = *basetv;
1224 for (i = 1; i < argcount; ++i)
1225 argv[i + 1] = argvars[i];
1226 }
Bram Moolenaar24278d22019-08-16 21:49:22 +02001227 else if (global_functions[fi].f_argtype == FEARG_3)
1228 {
1229 // base value goes third
1230 argv[0] = argvars[0];
1231 argv[1] = argvars[1];
1232 argv[2] = *basetv;
1233 for (i = 2; i < argcount; ++i)
1234 argv[i + 1] = argvars[i];
1235 }
Bram Moolenaaraad222c2019-09-06 22:46:09 +02001236 else if (global_functions[fi].f_argtype == FEARG_4)
1237 {
1238 // base value goes fourth
1239 argv[0] = argvars[0];
1240 argv[1] = argvars[1];
1241 argv[2] = argvars[2];
1242 argv[3] = *basetv;
1243 for (i = 3; i < argcount; ++i)
1244 argv[i + 1] = argvars[i];
1245 }
Bram Moolenaar25e42232019-08-04 15:04:10 +02001246 else
1247 {
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001248 // FEARG_1: base value goes first
Bram Moolenaar25e42232019-08-04 15:04:10 +02001249 argv[0] = *basetv;
1250 for (i = 0; i < argcount; ++i)
1251 argv[i + 1] = argvars[i];
1252 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001253 argv[argcount + 1].v_type = VAR_UNKNOWN;
1254
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001255 global_functions[fi].f_func(argv, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001256 return FCERR_NONE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001257}
1258
1259/*
1260 * Return TRUE for a non-zero Number and a non-empty String.
1261 */
Bram Moolenaar0e57dd82019-09-16 22:56:03 +02001262 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001263non_zero_arg(typval_T *argvars)
1264{
1265 return ((argvars[0].v_type == VAR_NUMBER
1266 && argvars[0].vval.v_number != 0)
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01001267 || (argvars[0].v_type == VAR_BOOL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001268 && argvars[0].vval.v_number == VVAL_TRUE)
1269 || (argvars[0].v_type == VAR_STRING
1270 && argvars[0].vval.v_string != NULL
1271 && *argvars[0].vval.v_string != NUL));
1272}
1273
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001274#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001275/*
1276 * Get the float value of "argvars[0]" into "f".
1277 * Returns FAIL when the argument is not a Number or Float.
1278 */
1279 static int
1280get_float_arg(typval_T *argvars, float_T *f)
1281{
1282 if (argvars[0].v_type == VAR_FLOAT)
1283 {
1284 *f = argvars[0].vval.v_float;
1285 return OK;
1286 }
1287 if (argvars[0].v_type == VAR_NUMBER)
1288 {
1289 *f = (float_T)argvars[0].vval.v_number;
1290 return OK;
1291 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001292 emsg(_("E808: Number or Float required"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001293 return FAIL;
1294}
1295
1296/*
1297 * "abs(expr)" function
1298 */
1299 static void
1300f_abs(typval_T *argvars, typval_T *rettv)
1301{
1302 if (argvars[0].v_type == VAR_FLOAT)
1303 {
1304 rettv->v_type = VAR_FLOAT;
1305 rettv->vval.v_float = fabs(argvars[0].vval.v_float);
1306 }
1307 else
1308 {
1309 varnumber_T n;
1310 int error = FALSE;
1311
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001312 n = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001313 if (error)
1314 rettv->vval.v_number = -1;
1315 else if (n > 0)
1316 rettv->vval.v_number = n;
1317 else
1318 rettv->vval.v_number = -n;
1319 }
1320}
1321
1322/*
1323 * "acos()" function
1324 */
1325 static void
1326f_acos(typval_T *argvars, typval_T *rettv)
1327{
1328 float_T f = 0.0;
1329
1330 rettv->v_type = VAR_FLOAT;
1331 if (get_float_arg(argvars, &f) == OK)
1332 rettv->vval.v_float = acos(f);
1333 else
1334 rettv->vval.v_float = 0.0;
1335}
1336#endif
1337
1338/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001339 * "and(expr, expr)" function
1340 */
1341 static void
1342f_and(typval_T *argvars, typval_T *rettv)
1343{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001344 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
1345 & tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaarca851592018-06-06 21:04:07 +02001346}
1347
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001348#ifdef FEAT_FLOAT
1349/*
1350 * "asin()" function
1351 */
1352 static void
1353f_asin(typval_T *argvars, typval_T *rettv)
1354{
1355 float_T f = 0.0;
1356
1357 rettv->v_type = VAR_FLOAT;
1358 if (get_float_arg(argvars, &f) == OK)
1359 rettv->vval.v_float = asin(f);
1360 else
1361 rettv->vval.v_float = 0.0;
1362}
1363
1364/*
1365 * "atan()" function
1366 */
1367 static void
1368f_atan(typval_T *argvars, typval_T *rettv)
1369{
1370 float_T f = 0.0;
1371
1372 rettv->v_type = VAR_FLOAT;
1373 if (get_float_arg(argvars, &f) == OK)
1374 rettv->vval.v_float = atan(f);
1375 else
1376 rettv->vval.v_float = 0.0;
1377}
1378
1379/*
1380 * "atan2()" function
1381 */
1382 static void
1383f_atan2(typval_T *argvars, typval_T *rettv)
1384{
1385 float_T fx = 0.0, fy = 0.0;
1386
1387 rettv->v_type = VAR_FLOAT;
1388 if (get_float_arg(argvars, &fx) == OK
1389 && get_float_arg(&argvars[1], &fy) == OK)
1390 rettv->vval.v_float = atan2(fx, fy);
1391 else
1392 rettv->vval.v_float = 0.0;
1393}
1394#endif
1395
1396/*
Bram Moolenaar59716a22017-03-01 20:32:44 +01001397 * "balloon_show()" function
1398 */
1399#ifdef FEAT_BEVAL
1400 static void
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02001401f_balloon_gettext(typval_T *argvars UNUSED, typval_T *rettv)
1402{
1403 rettv->v_type = VAR_STRING;
1404 if (balloonEval != NULL)
1405 {
1406 if (balloonEval->msg == NULL)
1407 rettv->vval.v_string = NULL;
1408 else
1409 rettv->vval.v_string = vim_strsave(balloonEval->msg);
1410 }
1411}
1412
1413 static void
Bram Moolenaar59716a22017-03-01 20:32:44 +01001414f_balloon_show(typval_T *argvars, typval_T *rettv UNUSED)
1415{
Bram Moolenaarcaf64342017-03-02 22:11:33 +01001416 if (balloonEval != NULL)
Bram Moolenaar246fe032017-11-19 19:56:27 +01001417 {
1418 if (argvars[0].v_type == VAR_LIST
1419# ifdef FEAT_GUI
1420 && !gui.in_use
1421# endif
1422 )
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02001423 {
1424 list_T *l = argvars[0].vval.v_list;
1425
1426 // empty list removes the balloon
1427 post_balloon(balloonEval, NULL,
1428 l == NULL || l->lv_len == 0 ? NULL : l);
1429 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01001430 else
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02001431 {
1432 char_u *mesg = tv_get_string_chk(&argvars[0]);
1433
1434 if (mesg != NULL)
1435 // empty string removes the balloon
1436 post_balloon(balloonEval, *mesg == NUL ? NULL : mesg, NULL);
1437 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01001438 }
1439}
1440
Bram Moolenaar669a8282017-11-19 20:13:05 +01001441# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +01001442 static void
1443f_balloon_split(typval_T *argvars, typval_T *rettv UNUSED)
1444{
1445 if (rettv_list_alloc(rettv) == OK)
1446 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001447 char_u *msg = tv_get_string_chk(&argvars[0]);
Bram Moolenaar246fe032017-11-19 19:56:27 +01001448
1449 if (msg != NULL)
1450 {
1451 pumitem_T *array;
1452 int size = split_message(msg, &array);
1453 int i;
1454
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001455 // Skip the first and last item, they are always empty.
Bram Moolenaar246fe032017-11-19 19:56:27 +01001456 for (i = 1; i < size - 1; ++i)
1457 list_append_string(rettv->vval.v_list, array[i].pum_text, -1);
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01001458 while (size > 0)
1459 vim_free(array[--size].pum_text);
Bram Moolenaar246fe032017-11-19 19:56:27 +01001460 vim_free(array);
1461 }
1462 }
Bram Moolenaar59716a22017-03-01 20:32:44 +01001463}
Bram Moolenaar669a8282017-11-19 20:13:05 +01001464# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +01001465#endif
1466
1467/*
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01001468 * Get the buffer from "arg" and give an error and return NULL if it is not
1469 * valid.
1470 */
Bram Moolenaara3347722019-05-11 21:14:24 +02001471 buf_T *
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01001472get_buf_arg(typval_T *arg)
1473{
1474 buf_T *buf;
1475
1476 ++emsg_off;
1477 buf = tv_get_buf(arg, FALSE);
1478 --emsg_off;
1479 if (buf == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001480 semsg(_("E158: Invalid buffer name: %s"), tv_get_string(arg));
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01001481 return buf;
1482}
1483
1484/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001485 * "byte2line(byte)" function
1486 */
1487 static void
1488f_byte2line(typval_T *argvars UNUSED, typval_T *rettv)
1489{
1490#ifndef FEAT_BYTEOFF
1491 rettv->vval.v_number = -1;
1492#else
1493 long boff = 0;
1494
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001495 boff = tv_get_number(&argvars[0]) - 1; // boff gets -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001496 if (boff < 0)
1497 rettv->vval.v_number = -1;
1498 else
1499 rettv->vval.v_number = ml_find_line_or_offset(curbuf,
1500 (linenr_T)0, &boff);
1501#endif
1502}
1503
1504 static void
1505byteidx(typval_T *argvars, typval_T *rettv, int comp UNUSED)
1506{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001507 char_u *t;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001508 char_u *str;
1509 varnumber_T idx;
1510
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001511 str = tv_get_string_chk(&argvars[0]);
1512 idx = tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001513 rettv->vval.v_number = -1;
1514 if (str == NULL || idx < 0)
1515 return;
1516
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001517 t = str;
1518 for ( ; idx > 0; idx--)
1519 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001520 if (*t == NUL) // EOL reached
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001521 return;
1522 if (enc_utf8 && comp)
1523 t += utf_ptr2len(t);
1524 else
1525 t += (*mb_ptr2len)(t);
1526 }
1527 rettv->vval.v_number = (varnumber_T)(t - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001528}
1529
1530/*
1531 * "byteidx()" function
1532 */
1533 static void
1534f_byteidx(typval_T *argvars, typval_T *rettv)
1535{
1536 byteidx(argvars, rettv, FALSE);
1537}
1538
1539/*
1540 * "byteidxcomp()" function
1541 */
1542 static void
1543f_byteidxcomp(typval_T *argvars, typval_T *rettv)
1544{
1545 byteidx(argvars, rettv, TRUE);
1546}
1547
1548/*
1549 * "call(func, arglist [, dict])" function
1550 */
1551 static void
1552f_call(typval_T *argvars, typval_T *rettv)
1553{
1554 char_u *func;
1555 partial_T *partial = NULL;
1556 dict_T *selfdict = NULL;
1557
1558 if (argvars[1].v_type != VAR_LIST)
1559 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001560 emsg(_(e_listreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001561 return;
1562 }
1563 if (argvars[1].vval.v_list == NULL)
1564 return;
1565
1566 if (argvars[0].v_type == VAR_FUNC)
1567 func = argvars[0].vval.v_string;
1568 else if (argvars[0].v_type == VAR_PARTIAL)
1569 {
1570 partial = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02001571 func = partial_name(partial);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001572 }
1573 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001574 func = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001575 if (*func == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001576 return; // type error or empty name
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001577
1578 if (argvars[2].v_type != VAR_UNKNOWN)
1579 {
1580 if (argvars[2].v_type != VAR_DICT)
1581 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001582 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001583 return;
1584 }
1585 selfdict = argvars[2].vval.v_dict;
1586 }
1587
1588 (void)func_call(func, &argvars[1], partial, selfdict, rettv);
1589}
1590
1591#ifdef FEAT_FLOAT
1592/*
1593 * "ceil({float})" function
1594 */
1595 static void
1596f_ceil(typval_T *argvars, typval_T *rettv)
1597{
1598 float_T f = 0.0;
1599
1600 rettv->v_type = VAR_FLOAT;
1601 if (get_float_arg(argvars, &f) == OK)
1602 rettv->vval.v_float = ceil(f);
1603 else
1604 rettv->vval.v_float = 0.0;
1605}
1606#endif
1607
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001608/*
1609 * "changenr()" function
1610 */
1611 static void
1612f_changenr(typval_T *argvars UNUSED, typval_T *rettv)
1613{
1614 rettv->vval.v_number = curbuf->b_u_seq_cur;
1615}
1616
1617/*
1618 * "char2nr(string)" function
1619 */
1620 static void
1621f_char2nr(typval_T *argvars, typval_T *rettv)
1622{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001623 if (has_mbyte)
1624 {
1625 int utf8 = 0;
1626
1627 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001628 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001629
1630 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01001631 rettv->vval.v_number = utf_ptr2char(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001632 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001633 rettv->vval.v_number = (*mb_ptr2char)(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001634 }
1635 else
Bram Moolenaar13505972019-01-24 15:04:48 +01001636 rettv->vval.v_number = tv_get_string(&argvars[0])[0];
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001637}
1638
Bram Moolenaar29b7d7a2019-07-22 23:03:57 +02001639 win_T *
Bram Moolenaaraff74912019-03-30 18:11:49 +01001640get_optional_window(typval_T *argvars, int idx)
1641{
1642 win_T *win = curwin;
1643
1644 if (argvars[idx].v_type != VAR_UNKNOWN)
1645 {
1646 win = find_win_by_nr_or_id(&argvars[idx]);
1647 if (win == NULL)
1648 {
1649 emsg(_(e_invalwindow));
1650 return NULL;
1651 }
1652 }
1653 return win;
1654}
1655
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001656/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001657 * "col(string)" function
1658 */
1659 static void
1660f_col(typval_T *argvars, typval_T *rettv)
1661{
1662 colnr_T col = 0;
1663 pos_T *fp;
1664 int fnum = curbuf->b_fnum;
1665
1666 fp = var2fpos(&argvars[0], FALSE, &fnum);
1667 if (fp != NULL && fnum == curbuf->b_fnum)
1668 {
1669 if (fp->col == MAXCOL)
1670 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001671 // '> can be MAXCOL, get the length of the line then
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001672 if (fp->lnum <= curbuf->b_ml.ml_line_count)
1673 col = (colnr_T)STRLEN(ml_get(fp->lnum)) + 1;
1674 else
1675 col = MAXCOL;
1676 }
1677 else
1678 {
1679 col = fp->col + 1;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001680 // col(".") when the cursor is on the NUL at the end of the line
1681 // because of "coladd" can be seen as an extra column.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001682 if (virtual_active() && fp == &curwin->w_cursor)
1683 {
1684 char_u *p = ml_get_cursor();
1685
1686 if (curwin->w_cursor.coladd >= (colnr_T)chartabsize(p,
1687 curwin->w_virtcol - curwin->w_cursor.coladd))
1688 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001689 int l;
1690
1691 if (*p != NUL && p[(l = (*mb_ptr2len)(p))] == NUL)
1692 col += l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001693 }
1694 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001695 }
1696 }
1697 rettv->vval.v_number = col;
1698}
1699
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001700/*
1701 * "confirm(message, buttons[, default [, type]])" function
1702 */
1703 static void
1704f_confirm(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1705{
1706#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1707 char_u *message;
1708 char_u *buttons = NULL;
1709 char_u buf[NUMBUFLEN];
1710 char_u buf2[NUMBUFLEN];
1711 int def = 1;
1712 int type = VIM_GENERIC;
1713 char_u *typestr;
1714 int error = FALSE;
1715
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001716 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001717 if (message == NULL)
1718 error = TRUE;
1719 if (argvars[1].v_type != VAR_UNKNOWN)
1720 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001721 buttons = tv_get_string_buf_chk(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001722 if (buttons == NULL)
1723 error = TRUE;
1724 if (argvars[2].v_type != VAR_UNKNOWN)
1725 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001726 def = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001727 if (argvars[3].v_type != VAR_UNKNOWN)
1728 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001729 typestr = tv_get_string_buf_chk(&argvars[3], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001730 if (typestr == NULL)
1731 error = TRUE;
1732 else
1733 {
1734 switch (TOUPPER_ASC(*typestr))
1735 {
1736 case 'E': type = VIM_ERROR; break;
1737 case 'Q': type = VIM_QUESTION; break;
1738 case 'I': type = VIM_INFO; break;
1739 case 'W': type = VIM_WARNING; break;
1740 case 'G': type = VIM_GENERIC; break;
1741 }
1742 }
1743 }
1744 }
1745 }
1746
1747 if (buttons == NULL || *buttons == NUL)
1748 buttons = (char_u *)_("&Ok");
1749
1750 if (!error)
1751 rettv->vval.v_number = do_dialog(type, NULL, message, buttons,
1752 def, NULL, FALSE);
1753#endif
1754}
1755
1756/*
1757 * "copy()" function
1758 */
1759 static void
1760f_copy(typval_T *argvars, typval_T *rettv)
1761{
1762 item_copy(&argvars[0], rettv, FALSE, 0);
1763}
1764
1765#ifdef FEAT_FLOAT
1766/*
1767 * "cos()" function
1768 */
1769 static void
1770f_cos(typval_T *argvars, typval_T *rettv)
1771{
1772 float_T f = 0.0;
1773
1774 rettv->v_type = VAR_FLOAT;
1775 if (get_float_arg(argvars, &f) == OK)
1776 rettv->vval.v_float = cos(f);
1777 else
1778 rettv->vval.v_float = 0.0;
1779}
1780
1781/*
1782 * "cosh()" function
1783 */
1784 static void
1785f_cosh(typval_T *argvars, typval_T *rettv)
1786{
1787 float_T f = 0.0;
1788
1789 rettv->v_type = VAR_FLOAT;
1790 if (get_float_arg(argvars, &f) == OK)
1791 rettv->vval.v_float = cosh(f);
1792 else
1793 rettv->vval.v_float = 0.0;
1794}
1795#endif
1796
1797/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001798 * "cursor(lnum, col)" function, or
1799 * "cursor(list)"
1800 *
1801 * Moves the cursor to the specified line and column.
1802 * Returns 0 when the position could be set, -1 otherwise.
1803 */
1804 static void
1805f_cursor(typval_T *argvars, typval_T *rettv)
1806{
1807 long line, col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001808 long coladd = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001809 int set_curswant = TRUE;
1810
1811 rettv->vval.v_number = -1;
1812 if (argvars[1].v_type == VAR_UNKNOWN)
1813 {
1814 pos_T pos;
1815 colnr_T curswant = -1;
1816
1817 if (list2fpos(argvars, &pos, NULL, &curswant) == FAIL)
1818 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001819 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001820 return;
1821 }
1822 line = pos.lnum;
1823 col = pos.col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001824 coladd = pos.coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001825 if (curswant >= 0)
1826 {
1827 curwin->w_curswant = curswant - 1;
1828 set_curswant = FALSE;
1829 }
1830 }
1831 else
1832 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001833 line = tv_get_lnum(argvars);
1834 col = (long)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001835 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001836 coladd = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001837 }
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01001838 if (line < 0 || col < 0 || coladd < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001839 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001840 if (line > 0)
1841 curwin->w_cursor.lnum = line;
1842 if (col > 0)
1843 curwin->w_cursor.col = col - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001844 curwin->w_cursor.coladd = coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001845
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001846 // Make sure the cursor is in a valid position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001847 check_cursor();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001848 // Correct cursor for multi-byte character.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001849 if (has_mbyte)
1850 mb_adjust_cursor();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001851
1852 curwin->w_set_curswant = set_curswant;
1853 rettv->vval.v_number = 0;
1854}
1855
Bram Moolenaar4f974752019-02-17 17:44:42 +01001856#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02001857/*
1858 * "debugbreak()" function
1859 */
1860 static void
1861f_debugbreak(typval_T *argvars, typval_T *rettv)
1862{
1863 int pid;
1864
1865 rettv->vval.v_number = FAIL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001866 pid = (int)tv_get_number(&argvars[0]);
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02001867 if (pid == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001868 emsg(_(e_invarg));
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02001869 else
1870 {
1871 HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, 0, pid);
1872
1873 if (hProcess != NULL)
1874 {
1875 DebugBreakProcess(hProcess);
1876 CloseHandle(hProcess);
1877 rettv->vval.v_number = OK;
1878 }
1879 }
1880}
1881#endif
1882
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001883/*
1884 * "deepcopy()" function
1885 */
1886 static void
1887f_deepcopy(typval_T *argvars, typval_T *rettv)
1888{
1889 int noref = 0;
1890 int copyID;
1891
1892 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001893 noref = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001894 if (noref < 0 || noref > 1)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001895 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001896 else
1897 {
1898 copyID = get_copyID();
1899 item_copy(&argvars[0], rettv, TRUE, noref == 0 ? copyID : 0);
1900 }
1901}
1902
1903/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001904 * "did_filetype()" function
1905 */
1906 static void
1907f_did_filetype(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1908{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001909 rettv->vval.v_number = did_filetype;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001910}
1911
1912/*
Bram Moolenaar4132eb52020-02-14 16:53:00 +01001913 * "echoraw({expr})" function
1914 */
1915 static void
1916f_echoraw(typval_T *argvars, typval_T *rettv UNUSED)
1917{
1918 char_u *str = tv_get_string_chk(&argvars[0]);
1919
1920 if (str != NULL && *str != NUL)
1921 {
1922 out_str(str);
1923 out_flush();
1924 }
1925}
1926
1927/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001928 * "empty({expr})" function
1929 */
1930 static void
1931f_empty(typval_T *argvars, typval_T *rettv)
1932{
1933 int n = FALSE;
1934
1935 switch (argvars[0].v_type)
1936 {
1937 case VAR_STRING:
1938 case VAR_FUNC:
1939 n = argvars[0].vval.v_string == NULL
1940 || *argvars[0].vval.v_string == NUL;
1941 break;
1942 case VAR_PARTIAL:
1943 n = FALSE;
1944 break;
1945 case VAR_NUMBER:
1946 n = argvars[0].vval.v_number == 0;
1947 break;
1948 case VAR_FLOAT:
1949#ifdef FEAT_FLOAT
1950 n = argvars[0].vval.v_float == 0.0;
1951 break;
1952#endif
1953 case VAR_LIST:
1954 n = argvars[0].vval.v_list == NULL
Bram Moolenaar50985eb2020-01-27 22:09:39 +01001955 || argvars[0].vval.v_list->lv_len == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001956 break;
1957 case VAR_DICT:
1958 n = argvars[0].vval.v_dict == NULL
1959 || argvars[0].vval.v_dict->dv_hashtab.ht_used == 0;
1960 break;
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01001961 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001962 case VAR_SPECIAL:
1963 n = argvars[0].vval.v_number != VVAL_TRUE;
1964 break;
1965
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01001966 case VAR_BLOB:
1967 n = argvars[0].vval.v_blob == NULL
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01001968 || argvars[0].vval.v_blob->bv_ga.ga_len == 0;
1969 break;
1970
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001971 case VAR_JOB:
1972#ifdef FEAT_JOB_CHANNEL
1973 n = argvars[0].vval.v_job == NULL
1974 || argvars[0].vval.v_job->jv_status != JOB_STARTED;
1975 break;
1976#endif
1977 case VAR_CHANNEL:
1978#ifdef FEAT_JOB_CHANNEL
1979 n = argvars[0].vval.v_channel == NULL
1980 || !channel_is_open(argvars[0].vval.v_channel);
1981 break;
1982#endif
1983 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02001984 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001985 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01001986 internal_error_no_abort("f_empty(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001987 n = TRUE;
1988 break;
1989 }
1990
1991 rettv->vval.v_number = n;
1992}
1993
1994/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02001995 * "environ()" function
1996 */
1997 static void
1998f_environ(typval_T *argvars UNUSED, typval_T *rettv)
1999{
2000#if !defined(AMIGA)
2001 int i = 0;
2002 char_u *entry, *value;
2003# ifdef MSWIN
2004 extern wchar_t **_wenviron;
2005# else
2006 extern char **environ;
2007# endif
2008
2009 if (rettv_dict_alloc(rettv) != OK)
2010 return;
2011
2012# ifdef MSWIN
2013 if (*_wenviron == NULL)
2014 return;
2015# else
2016 if (*environ == NULL)
2017 return;
2018# endif
2019
2020 for (i = 0; ; ++i)
2021 {
2022# ifdef MSWIN
2023 short_u *p;
2024
2025 if ((p = (short_u *)_wenviron[i]) == NULL)
2026 return;
2027 entry = utf16_to_enc(p, NULL);
2028# else
2029 if ((entry = (char_u *)environ[i]) == NULL)
2030 return;
2031 entry = vim_strsave(entry);
2032# endif
2033 if (entry == NULL) // out of memory
2034 return;
2035 if ((value = vim_strchr(entry, '=')) == NULL)
2036 {
2037 vim_free(entry);
2038 continue;
2039 }
2040 *value++ = NUL;
2041 dict_add_string(rettv->vval.v_dict, (char *)entry, value);
2042 vim_free(entry);
2043 }
2044#endif
2045}
2046
2047/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002048 * "escape({string}, {chars})" function
2049 */
2050 static void
2051f_escape(typval_T *argvars, typval_T *rettv)
2052{
2053 char_u buf[NUMBUFLEN];
2054
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002055 rettv->vval.v_string = vim_strsave_escaped(tv_get_string(&argvars[0]),
2056 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002057 rettv->v_type = VAR_STRING;
2058}
2059
2060/*
2061 * "eval()" function
2062 */
2063 static void
2064f_eval(typval_T *argvars, typval_T *rettv)
2065{
2066 char_u *s, *p;
2067
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002068 s = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002069 if (s != NULL)
2070 s = skipwhite(s);
2071
2072 p = s;
Bram Moolenaar32e35112020-05-14 22:41:15 +02002073 if (s == NULL || eval1(&s, rettv, EVAL_EVALUATE) == FAIL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002074 {
2075 if (p != NULL && !aborting())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002076 semsg(_(e_invexpr2), p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002077 need_clr_eos = FALSE;
2078 rettv->v_type = VAR_NUMBER;
2079 rettv->vval.v_number = 0;
2080 }
2081 else if (*s != NUL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002082 emsg(_(e_trailing));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002083}
2084
2085/*
2086 * "eventhandler()" function
2087 */
2088 static void
2089f_eventhandler(typval_T *argvars UNUSED, typval_T *rettv)
2090{
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002091 rettv->vval.v_number = vgetc_busy || input_busy;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002092}
2093
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002094static garray_T redir_execute_ga;
2095
2096/*
2097 * Append "value[value_len]" to the execute() output.
2098 */
2099 void
2100execute_redir_str(char_u *value, int value_len)
2101{
2102 int len;
2103
2104 if (value_len == -1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002105 len = (int)STRLEN(value); // Append the entire string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002106 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002107 len = value_len; // Append only "value_len" characters
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002108 if (ga_grow(&redir_execute_ga, len) == OK)
2109 {
2110 mch_memmove((char *)redir_execute_ga.ga_data
2111 + redir_execute_ga.ga_len, value, len);
2112 redir_execute_ga.ga_len += len;
2113 }
2114}
2115
2116/*
2117 * Get next line from a list.
2118 * Called by do_cmdline() to get the next line.
2119 * Returns allocated string, or NULL for end of function.
2120 */
2121
2122 static char_u *
2123get_list_line(
2124 int c UNUSED,
2125 void *cookie,
Bram Moolenaare96a2492019-06-25 04:12:16 +02002126 int indent UNUSED,
2127 int do_concat UNUSED)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002128{
2129 listitem_T **p = (listitem_T **)cookie;
2130 listitem_T *item = *p;
2131 char_u buf[NUMBUFLEN];
2132 char_u *s;
2133
2134 if (item == NULL)
2135 return NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002136 s = tv_get_string_buf_chk(&item->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002137 *p = item->li_next;
2138 return s == NULL ? NULL : vim_strsave(s);
2139}
2140
2141/*
2142 * "execute()" function
2143 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02002144 void
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002145execute_common(typval_T *argvars, typval_T *rettv, int arg_off)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002146{
2147 char_u *cmd = NULL;
2148 list_T *list = NULL;
2149 int save_msg_silent = msg_silent;
2150 int save_emsg_silent = emsg_silent;
2151 int save_emsg_noredir = emsg_noredir;
2152 int save_redir_execute = redir_execute;
Bram Moolenaar20951482017-12-25 13:44:43 +01002153 int save_redir_off = redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002154 garray_T save_ga;
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002155 int save_msg_col = msg_col;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002156 int echo_output = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002157
2158 rettv->vval.v_string = NULL;
2159 rettv->v_type = VAR_STRING;
2160
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002161 if (argvars[arg_off].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002162 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002163 list = argvars[arg_off].vval.v_list;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002164 if (list == NULL || list->lv_len == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002165 // empty list, no commands, empty output
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002166 return;
2167 ++list->lv_refcount;
2168 }
Bram Moolenaare2a8f072020-01-08 19:32:18 +01002169 else if (argvars[arg_off].v_type == VAR_JOB
2170 || argvars[arg_off].v_type == VAR_CHANNEL)
2171 {
2172 emsg(_(e_inval_string));
2173 return;
2174 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002175 else
2176 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002177 cmd = tv_get_string_chk(&argvars[arg_off]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002178 if (cmd == NULL)
2179 return;
2180 }
2181
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002182 if (argvars[arg_off + 1].v_type != VAR_UNKNOWN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002183 {
2184 char_u buf[NUMBUFLEN];
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002185 char_u *s = tv_get_string_buf_chk(&argvars[arg_off + 1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002186
2187 if (s == NULL)
2188 return;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002189 if (*s == NUL)
2190 echo_output = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002191 if (STRNCMP(s, "silent", 6) == 0)
2192 ++msg_silent;
2193 if (STRCMP(s, "silent!") == 0)
2194 {
2195 emsg_silent = TRUE;
2196 emsg_noredir = TRUE;
2197 }
2198 }
2199 else
2200 ++msg_silent;
2201
2202 if (redir_execute)
2203 save_ga = redir_execute_ga;
2204 ga_init2(&redir_execute_ga, (int)sizeof(char), 500);
2205 redir_execute = TRUE;
Bram Moolenaar20951482017-12-25 13:44:43 +01002206 redir_off = FALSE;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002207 if (!echo_output)
2208 msg_col = 0; // prevent leading spaces
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002209
2210 if (cmd != NULL)
2211 do_cmdline_cmd(cmd);
2212 else
2213 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002214 listitem_T *item;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002215
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02002216 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002217 item = list->lv_first;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002218 do_cmdline(NULL, get_list_line, (void *)&item,
2219 DOCMD_NOWAIT|DOCMD_VERBOSE|DOCMD_REPEAT|DOCMD_KEYTYPED);
2220 --list->lv_refcount;
2221 }
2222
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002223 // Need to append a NUL to the result.
Bram Moolenaard297f352017-01-29 20:31:21 +01002224 if (ga_grow(&redir_execute_ga, 1) == OK)
2225 {
2226 ((char *)redir_execute_ga.ga_data)[redir_execute_ga.ga_len] = NUL;
2227 rettv->vval.v_string = redir_execute_ga.ga_data;
2228 }
2229 else
2230 {
2231 ga_clear(&redir_execute_ga);
2232 rettv->vval.v_string = NULL;
2233 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002234 msg_silent = save_msg_silent;
2235 emsg_silent = save_emsg_silent;
2236 emsg_noredir = save_emsg_noredir;
2237
2238 redir_execute = save_redir_execute;
2239 if (redir_execute)
2240 redir_execute_ga = save_ga;
Bram Moolenaar20951482017-12-25 13:44:43 +01002241 redir_off = save_redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002242
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002243 // "silent reg" or "silent echo x" leaves msg_col somewhere in the line.
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002244 if (echo_output)
2245 // When not working silently: put it in column zero. A following
2246 // "echon" will overwrite the message, unavoidably.
2247 msg_col = 0;
2248 else
2249 // When working silently: Put it back where it was, since nothing
2250 // should have been written.
2251 msg_col = save_msg_col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002252}
2253
2254/*
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002255 * "execute()" function
2256 */
2257 static void
2258f_execute(typval_T *argvars, typval_T *rettv)
2259{
2260 execute_common(argvars, rettv, 0);
2261}
2262
2263/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002264 * "exists()" function
2265 */
2266 static void
2267f_exists(typval_T *argvars, typval_T *rettv)
2268{
2269 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002270 int n = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002271
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002272 p = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002273 if (*p == '$') // environment variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002274 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002275 // first try "normal" environment variables (fast)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002276 if (mch_getenv(p + 1) != NULL)
2277 n = TRUE;
2278 else
2279 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002280 // try expanding things like $VIM and ${HOME}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002281 p = expand_env_save(p);
2282 if (p != NULL && *p != '$')
2283 n = TRUE;
2284 vim_free(p);
2285 }
2286 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002287 else if (*p == '&' || *p == '+') // option
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002288 {
2289 n = (get_option_tv(&p, NULL, TRUE) == OK);
2290 if (*skipwhite(p) != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002291 n = FALSE; // trailing garbage
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002292 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002293 else if (*p == '*') // internal or user defined function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002294 {
Bram Moolenaarb54c3ff2016-07-31 14:11:58 +02002295 n = function_exists(p + 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002296 }
Bram Moolenaar15c47602020-03-26 22:16:48 +01002297 else if (*p == '?') // internal function only
2298 {
2299 n = has_internal_func_name(p + 1);
2300 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002301 else if (*p == ':')
2302 {
2303 n = cmd_exists(p + 1);
2304 }
2305 else if (*p == '#')
2306 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002307 if (p[1] == '#')
2308 n = autocmd_supported(p + 2);
2309 else
2310 n = au_exists(p + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002311 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002312 else // internal variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002313 {
Bram Moolenaarc6f9f732018-02-11 19:06:26 +01002314 n = var_exists(p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002315 }
2316
2317 rettv->vval.v_number = n;
2318}
2319
2320#ifdef FEAT_FLOAT
2321/*
2322 * "exp()" function
2323 */
2324 static void
2325f_exp(typval_T *argvars, typval_T *rettv)
2326{
2327 float_T f = 0.0;
2328
2329 rettv->v_type = VAR_FLOAT;
2330 if (get_float_arg(argvars, &f) == OK)
2331 rettv->vval.v_float = exp(f);
2332 else
2333 rettv->vval.v_float = 0.0;
2334}
2335#endif
2336
2337/*
2338 * "expand()" function
2339 */
2340 static void
2341f_expand(typval_T *argvars, typval_T *rettv)
2342{
2343 char_u *s;
2344 int len;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002345 char *errormsg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002346 int options = WILD_SILENT|WILD_USE_NL|WILD_LIST_NOTFOUND;
2347 expand_T xpc;
2348 int error = FALSE;
2349 char_u *result;
2350
2351 rettv->v_type = VAR_STRING;
2352 if (argvars[1].v_type != VAR_UNKNOWN
2353 && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002354 && tv_get_number_chk(&argvars[2], &error)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002355 && !error)
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02002356 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002357
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002358 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002359 if (*s == '%' || *s == '#' || *s == '<')
2360 {
2361 ++emsg_off;
2362 result = eval_vars(s, s, &len, NULL, &errormsg, NULL);
2363 --emsg_off;
2364 if (rettv->v_type == VAR_LIST)
2365 {
2366 if (rettv_list_alloc(rettv) != FAIL && result != NULL)
2367 list_append_string(rettv->vval.v_list, result, -1);
Bram Moolenaar86173482019-10-01 17:02:16 +02002368 vim_free(result);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002369 }
2370 else
2371 rettv->vval.v_string = result;
2372 }
2373 else
2374 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002375 // When the optional second argument is non-zero, don't remove matches
2376 // for 'wildignore' and don't put matches for 'suffixes' at the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002377 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002378 && tv_get_number_chk(&argvars[1], &error))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002379 options |= WILD_KEEP_ALL;
2380 if (!error)
2381 {
2382 ExpandInit(&xpc);
2383 xpc.xp_context = EXPAND_FILES;
2384 if (p_wic)
2385 options += WILD_ICASE;
2386 if (rettv->v_type == VAR_STRING)
2387 rettv->vval.v_string = ExpandOne(&xpc, s, NULL,
2388 options, WILD_ALL);
2389 else if (rettv_list_alloc(rettv) != FAIL)
2390 {
2391 int i;
2392
2393 ExpandOne(&xpc, s, NULL, options, WILD_ALL_KEEP);
2394 for (i = 0; i < xpc.xp_numfiles; i++)
2395 list_append_string(rettv->vval.v_list, xpc.xp_files[i], -1);
2396 ExpandCleanup(&xpc);
2397 }
2398 }
2399 else
2400 rettv->vval.v_string = NULL;
2401 }
2402}
2403
2404/*
Bram Moolenaar80dad482019-06-09 17:22:31 +02002405 * "expandcmd()" function
2406 * Expand all the special characters in a command string.
2407 */
2408 static void
2409f_expandcmd(typval_T *argvars, typval_T *rettv)
2410{
2411 exarg_T eap;
2412 char_u *cmdstr;
2413 char *errormsg = NULL;
2414
2415 rettv->v_type = VAR_STRING;
2416 cmdstr = vim_strsave(tv_get_string(&argvars[0]));
2417
2418 memset(&eap, 0, sizeof(eap));
2419 eap.cmd = cmdstr;
2420 eap.arg = cmdstr;
Bram Moolenaar8071cb22019-07-12 17:58:01 +02002421 eap.argt |= EX_NOSPC;
Bram Moolenaar80dad482019-06-09 17:22:31 +02002422 eap.usefilter = FALSE;
2423 eap.nextcmd = NULL;
2424 eap.cmdidx = CMD_USER;
2425
2426 expand_filename(&eap, &cmdstr, &errormsg);
2427 if (errormsg != NULL && *errormsg != NUL)
2428 emsg(errormsg);
2429
2430 rettv->vval.v_string = cmdstr;
2431}
2432
2433/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002434 * "feedkeys()" function
2435 */
2436 static void
2437f_feedkeys(typval_T *argvars, typval_T *rettv UNUSED)
2438{
2439 int remap = TRUE;
2440 int insert = FALSE;
2441 char_u *keys, *flags;
2442 char_u nbuf[NUMBUFLEN];
2443 int typed = FALSE;
2444 int execute = FALSE;
2445 int dangerous = FALSE;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002446 int lowlevel = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002447 char_u *keys_esc;
2448
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002449 // This is not allowed in the sandbox. If the commands would still be
2450 // executed in the sandbox it would be OK, but it probably happens later,
2451 // when "sandbox" is no longer set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002452 if (check_secure())
2453 return;
2454
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002455 keys = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002456
2457 if (argvars[1].v_type != VAR_UNKNOWN)
2458 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002459 flags = tv_get_string_buf(&argvars[1], nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002460 for ( ; *flags != NUL; ++flags)
2461 {
2462 switch (*flags)
2463 {
2464 case 'n': remap = FALSE; break;
2465 case 'm': remap = TRUE; break;
2466 case 't': typed = TRUE; break;
2467 case 'i': insert = TRUE; break;
2468 case 'x': execute = TRUE; break;
2469 case '!': dangerous = TRUE; break;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002470 case 'L': lowlevel = TRUE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002471 }
2472 }
2473 }
2474
2475 if (*keys != NUL || execute)
2476 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002477 // Need to escape K_SPECIAL and CSI before putting the string in the
2478 // typeahead buffer.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002479 keys_esc = vim_strsave_escape_csi(keys);
2480 if (keys_esc != NULL)
2481 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002482 if (lowlevel)
2483 {
2484#ifdef USE_INPUT_BUF
Bram Moolenaar9645e2d2020-03-20 20:48:49 +01002485 int idx;
2486 int len = (int)STRLEN(keys);
2487
2488 for (idx = 0; idx < len; ++idx)
2489 {
2490 // if a CTRL-C was typed, set got_int, similar to what
2491 // happens in fill_input_buf()
2492 if (keys[idx] == 3 && ctrl_c_interrupts && typed)
2493 got_int = TRUE;
2494 add_to_input_buf(keys + idx, 1);
2495 }
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002496#else
2497 emsg(_("E980: lowlevel input not supported"));
2498#endif
2499 }
2500 else
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002501 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002502 ins_typebuf(keys_esc, (remap ? REMAP_YES : REMAP_NONE),
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002503 insert ? 0 : typebuf.tb_len, !typed, FALSE);
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002504 if (vgetc_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02002505#ifdef FEAT_TIMERS
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002506 || timer_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02002507#endif
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002508 || input_busy)
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002509 typebuf_was_filled = TRUE;
2510 }
2511 vim_free(keys_esc);
2512
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002513 if (execute)
2514 {
2515 int save_msg_scroll = msg_scroll;
2516
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002517 // Avoid a 1 second delay when the keys start Insert mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002518 msg_scroll = FALSE;
2519
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02002520 if (!dangerous)
2521 ++ex_normal_busy;
Bram Moolenaar905dd902019-04-07 14:21:47 +02002522 exec_normal(TRUE, lowlevel, TRUE);
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02002523 if (!dangerous)
2524 --ex_normal_busy;
2525
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002526 msg_scroll |= save_msg_scroll;
2527 }
2528 }
2529 }
2530}
2531
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002532#ifdef FEAT_FLOAT
2533/*
2534 * "float2nr({float})" function
2535 */
2536 static void
2537f_float2nr(typval_T *argvars, typval_T *rettv)
2538{
2539 float_T f = 0.0;
2540
2541 if (get_float_arg(argvars, &f) == OK)
2542 {
Bram Moolenaar37184272020-05-23 19:30:05 +02002543 if (f <= (float_T)-VARNUM_MAX + DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002544 rettv->vval.v_number = -VARNUM_MAX;
Bram Moolenaar37184272020-05-23 19:30:05 +02002545 else if (f >= (float_T)VARNUM_MAX - DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002546 rettv->vval.v_number = VARNUM_MAX;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002547 else
2548 rettv->vval.v_number = (varnumber_T)f;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002549 }
2550}
2551
2552/*
2553 * "floor({float})" function
2554 */
2555 static void
2556f_floor(typval_T *argvars, typval_T *rettv)
2557{
2558 float_T f = 0.0;
2559
2560 rettv->v_type = VAR_FLOAT;
2561 if (get_float_arg(argvars, &f) == OK)
2562 rettv->vval.v_float = floor(f);
2563 else
2564 rettv->vval.v_float = 0.0;
2565}
2566
2567/*
2568 * "fmod()" function
2569 */
2570 static void
2571f_fmod(typval_T *argvars, typval_T *rettv)
2572{
2573 float_T fx = 0.0, fy = 0.0;
2574
2575 rettv->v_type = VAR_FLOAT;
2576 if (get_float_arg(argvars, &fx) == OK
2577 && get_float_arg(&argvars[1], &fy) == OK)
2578 rettv->vval.v_float = fmod(fx, fy);
2579 else
2580 rettv->vval.v_float = 0.0;
2581}
2582#endif
2583
2584/*
2585 * "fnameescape({string})" function
2586 */
2587 static void
2588f_fnameescape(typval_T *argvars, typval_T *rettv)
2589{
2590 rettv->vval.v_string = vim_strsave_fnameescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002591 tv_get_string(&argvars[0]), FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002592 rettv->v_type = VAR_STRING;
2593}
2594
2595/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002596 * "foreground()" function
2597 */
2598 static void
2599f_foreground(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2600{
2601#ifdef FEAT_GUI
2602 if (gui.in_use)
Bram Moolenaarafde13b2019-04-28 19:46:49 +02002603 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002604 gui_mch_set_foreground();
Bram Moolenaarafde13b2019-04-28 19:46:49 +02002605 return;
2606 }
2607#endif
2608#if defined(MSWIN) && (!defined(FEAT_GUI) || defined(VIMDLL))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002609 win32_set_foreground();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002610#endif
2611}
2612
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002613 static void
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002614common_function(typval_T *argvars, typval_T *rettv, int is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002615{
2616 char_u *s;
2617 char_u *name;
2618 int use_string = FALSE;
2619 partial_T *arg_pt = NULL;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002620 char_u *trans_name = NULL;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002621 int is_global = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002622
2623 if (argvars[0].v_type == VAR_FUNC)
2624 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002625 // function(MyFunc, [arg], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002626 s = argvars[0].vval.v_string;
2627 }
2628 else if (argvars[0].v_type == VAR_PARTIAL
2629 && argvars[0].vval.v_partial != NULL)
2630 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002631 // function(dict.MyFunc, [arg])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002632 arg_pt = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002633 s = partial_name(arg_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002634 }
2635 else
2636 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002637 // function('MyFunc', [arg], dict)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002638 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002639 use_string = TRUE;
2640 }
2641
Bram Moolenaar843b8842016-08-21 14:36:15 +02002642 if ((use_string && vim_strchr(s, AUTOLOAD_CHAR) == NULL) || is_funcref)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002643 {
2644 name = s;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002645 trans_name = trans_function_name(&name, &is_global, FALSE,
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002646 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF, NULL, NULL);
2647 if (*name != NUL)
2648 s = NULL;
2649 }
2650
Bram Moolenaar843b8842016-08-21 14:36:15 +02002651 if (s == NULL || *s == NUL || (use_string && VIM_ISDIGIT(*s))
2652 || (is_funcref && trans_name == NULL))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002653 semsg(_(e_invarg2), use_string ? tv_get_string(&argvars[0]) : s);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002654 // Don't check an autoload name for existence here.
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002655 else if (trans_name != NULL && (is_funcref
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002656 ? find_func(trans_name, is_global, NULL) == NULL
2657 : !translated_function_exists(trans_name, is_global)))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002658 semsg(_("E700: Unknown function: %s"), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002659 else
2660 {
2661 int dict_idx = 0;
2662 int arg_idx = 0;
2663 list_T *list = NULL;
2664
2665 if (STRNCMP(s, "s:", 2) == 0 || STRNCMP(s, "<SID>", 5) == 0)
2666 {
2667 char sid_buf[25];
2668 int off = *s == 's' ? 2 : 5;
2669
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002670 // Expand s: and <SID> into <SNR>nr_, so that the function can
2671 // also be called from another script. Using trans_function_name()
2672 // would also work, but some plugins depend on the name being
2673 // printable text.
Bram Moolenaarf29c1c62018-09-10 21:05:02 +02002674 sprintf(sid_buf, "<SNR>%ld_", (long)current_sctx.sc_sid);
Bram Moolenaar51e14382019-05-25 20:21:28 +02002675 name = alloc(STRLEN(sid_buf) + STRLEN(s + off) + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002676 if (name != NULL)
2677 {
2678 STRCPY(name, sid_buf);
2679 STRCAT(name, s + off);
2680 }
2681 }
2682 else
2683 name = vim_strsave(s);
2684
2685 if (argvars[1].v_type != VAR_UNKNOWN)
2686 {
2687 if (argvars[2].v_type != VAR_UNKNOWN)
2688 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002689 // function(name, [args], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002690 arg_idx = 1;
2691 dict_idx = 2;
2692 }
2693 else if (argvars[1].v_type == VAR_DICT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002694 // function(name, dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002695 dict_idx = 1;
2696 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002697 // function(name, [args])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002698 arg_idx = 1;
2699 if (dict_idx > 0)
2700 {
2701 if (argvars[dict_idx].v_type != VAR_DICT)
2702 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002703 emsg(_("E922: expected a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002704 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002705 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002706 }
2707 if (argvars[dict_idx].vval.v_dict == NULL)
2708 dict_idx = 0;
2709 }
2710 if (arg_idx > 0)
2711 {
2712 if (argvars[arg_idx].v_type != VAR_LIST)
2713 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002714 emsg(_("E923: Second argument of function() must be a list or a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002715 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002716 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002717 }
2718 list = argvars[arg_idx].vval.v_list;
2719 if (list == NULL || list->lv_len == 0)
2720 arg_idx = 0;
Bram Moolenaar4c054e92019-11-10 00:13:50 +01002721 else if (list->lv_len > MAX_FUNC_ARGS)
2722 {
Bram Moolenaar2118a302019-11-22 19:29:45 +01002723 emsg_funcname((char *)e_toomanyarg, s);
Bram Moolenaar4c054e92019-11-10 00:13:50 +01002724 vim_free(name);
2725 goto theend;
2726 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002727 }
2728 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002729 if (dict_idx > 0 || arg_idx > 0 || arg_pt != NULL || is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002730 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02002731 partial_T *pt = ALLOC_CLEAR_ONE(partial_T);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002732
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002733 // result is a VAR_PARTIAL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002734 if (pt == NULL)
2735 vim_free(name);
2736 else
2737 {
2738 if (arg_idx > 0 || (arg_pt != NULL && arg_pt->pt_argc > 0))
2739 {
2740 listitem_T *li;
2741 int i = 0;
2742 int arg_len = 0;
2743 int lv_len = 0;
2744
2745 if (arg_pt != NULL)
2746 arg_len = arg_pt->pt_argc;
2747 if (list != NULL)
2748 lv_len = list->lv_len;
2749 pt->pt_argc = arg_len + lv_len;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02002750 pt->pt_argv = ALLOC_MULT(typval_T, pt->pt_argc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002751 if (pt->pt_argv == NULL)
2752 {
2753 vim_free(pt);
2754 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002755 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002756 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002757 for (i = 0; i < arg_len; i++)
2758 copy_tv(&arg_pt->pt_argv[i], &pt->pt_argv[i]);
2759 if (lv_len > 0)
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002760 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02002761 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002762 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002763 copy_tv(&li->li_tv, &pt->pt_argv[i++]);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002764 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002765 }
2766
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002767 // For "function(dict.func, [], dict)" and "func" is a partial
2768 // use "dict". That is backwards compatible.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002769 if (dict_idx > 0)
2770 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002771 // The dict is bound explicitly, pt_auto is FALSE.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002772 pt->pt_dict = argvars[dict_idx].vval.v_dict;
2773 ++pt->pt_dict->dv_refcount;
2774 }
2775 else if (arg_pt != NULL)
2776 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002777 // If the dict was bound automatically the result is also
2778 // bound automatically.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002779 pt->pt_dict = arg_pt->pt_dict;
2780 pt->pt_auto = arg_pt->pt_auto;
2781 if (pt->pt_dict != NULL)
2782 ++pt->pt_dict->dv_refcount;
2783 }
2784
2785 pt->pt_refcount = 1;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002786 if (arg_pt != NULL && arg_pt->pt_func != NULL)
2787 {
2788 pt->pt_func = arg_pt->pt_func;
2789 func_ptr_ref(pt->pt_func);
2790 vim_free(name);
2791 }
2792 else if (is_funcref)
2793 {
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002794 pt->pt_func = find_func(trans_name, is_global, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002795 func_ptr_ref(pt->pt_func);
2796 vim_free(name);
2797 }
2798 else
2799 {
2800 pt->pt_name = name;
2801 func_ref(name);
2802 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002803 }
2804 rettv->v_type = VAR_PARTIAL;
2805 rettv->vval.v_partial = pt;
2806 }
2807 else
2808 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002809 // result is a VAR_FUNC
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002810 rettv->v_type = VAR_FUNC;
2811 rettv->vval.v_string = name;
2812 func_ref(name);
2813 }
2814 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002815theend:
2816 vim_free(trans_name);
2817}
2818
2819/*
2820 * "funcref()" function
2821 */
2822 static void
2823f_funcref(typval_T *argvars, typval_T *rettv)
2824{
2825 common_function(argvars, rettv, TRUE);
2826}
2827
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002828 static type_T *
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002829ret_f_function(int argcount, type_T **argtypes)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002830{
2831 if (argcount == 1 && argtypes[0]->tt_type == VAR_STRING)
2832 return &t_func_any;
Bram Moolenaard77a8522020-04-03 21:59:57 +02002833 return &t_func_void;
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002834}
2835
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002836/*
2837 * "function()" function
2838 */
2839 static void
2840f_function(typval_T *argvars, typval_T *rettv)
2841{
2842 common_function(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002843}
2844
2845/*
2846 * "garbagecollect()" function
2847 */
2848 static void
2849f_garbagecollect(typval_T *argvars, typval_T *rettv UNUSED)
2850{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002851 // This is postponed until we are back at the toplevel, because we may be
2852 // using Lists and Dicts internally. E.g.: ":echo [garbagecollect()]".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002853 want_garbage_collect = TRUE;
2854
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002855 if (argvars[0].v_type != VAR_UNKNOWN && tv_get_number(&argvars[0]) == 1)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002856 garbage_collect_at_exit = TRUE;
2857}
2858
2859/*
2860 * "get()" function
2861 */
2862 static void
2863f_get(typval_T *argvars, typval_T *rettv)
2864{
2865 listitem_T *li;
2866 list_T *l;
2867 dictitem_T *di;
2868 dict_T *d;
2869 typval_T *tv = NULL;
Bram Moolenaarf91aac52019-07-28 13:21:01 +02002870 int what_is_dict = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002871
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002872 if (argvars[0].v_type == VAR_BLOB)
2873 {
2874 int error = FALSE;
2875 int idx = tv_get_number_chk(&argvars[1], &error);
2876
2877 if (!error)
2878 {
2879 rettv->v_type = VAR_NUMBER;
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002880 if (idx < 0)
2881 idx = blob_len(argvars[0].vval.v_blob) + idx;
2882 if (idx < 0 || idx >= blob_len(argvars[0].vval.v_blob))
2883 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002884 else
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002885 {
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002886 rettv->vval.v_number = blob_get(argvars[0].vval.v_blob, idx);
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002887 tv = rettv;
2888 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002889 }
2890 }
2891 else if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002892 {
2893 if ((l = argvars[0].vval.v_list) != NULL)
2894 {
2895 int error = FALSE;
2896
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002897 li = list_find(l, (long)tv_get_number_chk(&argvars[1], &error));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002898 if (!error && li != NULL)
2899 tv = &li->li_tv;
2900 }
2901 }
2902 else if (argvars[0].v_type == VAR_DICT)
2903 {
2904 if ((d = argvars[0].vval.v_dict) != NULL)
2905 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002906 di = dict_find(d, tv_get_string(&argvars[1]), -1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002907 if (di != NULL)
2908 tv = &di->di_tv;
2909 }
2910 }
2911 else if (argvars[0].v_type == VAR_PARTIAL || argvars[0].v_type == VAR_FUNC)
2912 {
2913 partial_T *pt;
2914 partial_T fref_pt;
2915
2916 if (argvars[0].v_type == VAR_PARTIAL)
2917 pt = argvars[0].vval.v_partial;
2918 else
2919 {
Bram Moolenaara80faa82020-04-12 19:37:17 +02002920 CLEAR_FIELD(fref_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002921 fref_pt.pt_name = argvars[0].vval.v_string;
2922 pt = &fref_pt;
2923 }
2924
2925 if (pt != NULL)
2926 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002927 char_u *what = tv_get_string(&argvars[1]);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002928 char_u *n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002929
2930 if (STRCMP(what, "func") == 0 || STRCMP(what, "name") == 0)
2931 {
2932 rettv->v_type = (*what == 'f' ? VAR_FUNC : VAR_STRING);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002933 n = partial_name(pt);
2934 if (n == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002935 rettv->vval.v_string = NULL;
2936 else
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002937 {
2938 rettv->vval.v_string = vim_strsave(n);
2939 if (rettv->v_type == VAR_FUNC)
2940 func_ref(rettv->vval.v_string);
2941 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002942 }
2943 else if (STRCMP(what, "dict") == 0)
Bram Moolenaarf91aac52019-07-28 13:21:01 +02002944 {
2945 what_is_dict = TRUE;
2946 if (pt->pt_dict != NULL)
2947 rettv_dict_set(rettv, pt->pt_dict);
2948 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002949 else if (STRCMP(what, "args") == 0)
2950 {
2951 rettv->v_type = VAR_LIST;
2952 if (rettv_list_alloc(rettv) == OK)
2953 {
2954 int i;
2955
2956 for (i = 0; i < pt->pt_argc; ++i)
2957 list_append_tv(rettv->vval.v_list, &pt->pt_argv[i]);
2958 }
2959 }
2960 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002961 semsg(_(e_invarg2), what);
Bram Moolenaarf91aac52019-07-28 13:21:01 +02002962
2963 // When {what} == "dict" and pt->pt_dict == NULL, evaluate the
2964 // third argument
2965 if (!what_is_dict)
2966 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002967 }
2968 }
2969 else
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01002970 semsg(_(e_listdictblobarg), "get()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002971
2972 if (tv == NULL)
2973 {
2974 if (argvars[2].v_type != VAR_UNKNOWN)
2975 copy_tv(&argvars[2], rettv);
2976 }
2977 else
2978 copy_tv(tv, rettv);
2979}
2980
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02002981/*
Bram Moolenaar07ad8162018-02-13 13:59:59 +01002982 * "getchangelist()" function
2983 */
2984 static void
2985f_getchangelist(typval_T *argvars, typval_T *rettv)
2986{
2987#ifdef FEAT_JUMPLIST
2988 buf_T *buf;
2989 int i;
2990 list_T *l;
2991 dict_T *d;
2992#endif
2993
2994 if (rettv_list_alloc(rettv) != OK)
2995 return;
2996
2997#ifdef FEAT_JUMPLIST
Bram Moolenaar4c313b12019-08-24 22:58:31 +02002998 if (argvars[0].v_type == VAR_UNKNOWN)
2999 buf = curbuf;
3000 else
3001 {
3002 (void)tv_get_number(&argvars[0]); // issue errmsg if type error
3003 ++emsg_off;
3004 buf = tv_get_buf(&argvars[0], FALSE);
3005 --emsg_off;
3006 }
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003007 if (buf == NULL)
3008 return;
3009
3010 l = list_alloc();
3011 if (l == NULL)
3012 return;
3013
3014 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3015 return;
3016 /*
3017 * The current window change list index tracks only the position in the
3018 * current buffer change list. For other buffers, use the change list
3019 * length as the current index.
3020 */
3021 list_append_number(rettv->vval.v_list,
3022 (varnumber_T)((buf == curwin->w_buffer)
3023 ? curwin->w_changelistidx : buf->b_changelistlen));
3024
3025 for (i = 0; i < buf->b_changelistlen; ++i)
3026 {
3027 if (buf->b_changelist[i].lnum == 0)
3028 continue;
3029 if ((d = dict_alloc()) == NULL)
3030 return;
3031 if (list_append_dict(l, d) == FAIL)
3032 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003033 dict_add_number(d, "lnum", (long)buf->b_changelist[i].lnum);
3034 dict_add_number(d, "col", (long)buf->b_changelist[i].col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003035 dict_add_number(d, "coladd", (long)buf->b_changelist[i].coladd);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003036 }
3037#endif
3038}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003039
3040/*
3041 * "getcharsearch()" function
3042 */
3043 static void
3044f_getcharsearch(typval_T *argvars UNUSED, typval_T *rettv)
3045{
3046 if (rettv_dict_alloc(rettv) != FAIL)
3047 {
3048 dict_T *dict = rettv->vval.v_dict;
3049
Bram Moolenaare0be1672018-07-08 16:50:37 +02003050 dict_add_string(dict, "char", last_csearch());
3051 dict_add_number(dict, "forward", last_csearch_forward());
3052 dict_add_number(dict, "until", last_csearch_until());
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003053 }
3054}
3055
3056/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02003057 * "getenv()" function
3058 */
3059 static void
3060f_getenv(typval_T *argvars, typval_T *rettv)
3061{
3062 int mustfree = FALSE;
3063 char_u *p = vim_getenv(tv_get_string(&argvars[0]), &mustfree);
3064
3065 if (p == NULL)
3066 {
3067 rettv->v_type = VAR_SPECIAL;
3068 rettv->vval.v_number = VVAL_NULL;
3069 return;
3070 }
3071 if (!mustfree)
3072 p = vim_strsave(p);
3073 rettv->vval.v_string = p;
3074 rettv->v_type = VAR_STRING;
3075}
3076
3077/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003078 * "getfontname()" function
3079 */
3080 static void
3081f_getfontname(typval_T *argvars UNUSED, typval_T *rettv)
3082{
3083 rettv->v_type = VAR_STRING;
3084 rettv->vval.v_string = NULL;
3085#ifdef FEAT_GUI
3086 if (gui.in_use)
3087 {
3088 GuiFont font;
3089 char_u *name = NULL;
3090
3091 if (argvars[0].v_type == VAR_UNKNOWN)
3092 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003093 // Get the "Normal" font. Either the name saved by
3094 // hl_set_font_name() or from the font ID.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003095 font = gui.norm_font;
3096 name = hl_get_font_name();
3097 }
3098 else
3099 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003100 name = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003101 if (STRCMP(name, "*") == 0) // don't use font dialog
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003102 return;
3103 font = gui_mch_get_font(name, FALSE);
3104 if (font == NOFONT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003105 return; // Invalid font name, return empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003106 }
3107 rettv->vval.v_string = gui_mch_get_fontname(font, name);
3108 if (argvars[0].v_type != VAR_UNKNOWN)
3109 gui_mch_free_font(font);
3110 }
3111#endif
3112}
3113
3114/*
Bram Moolenaar4f505882018-02-10 21:06:32 +01003115 * "getjumplist()" function
3116 */
3117 static void
3118f_getjumplist(typval_T *argvars, typval_T *rettv)
3119{
3120#ifdef FEAT_JUMPLIST
3121 win_T *wp;
3122 int i;
3123 list_T *l;
3124 dict_T *d;
3125#endif
3126
3127 if (rettv_list_alloc(rettv) != OK)
3128 return;
3129
3130#ifdef FEAT_JUMPLIST
Bram Moolenaar00aa0692019-04-27 20:37:57 +02003131 wp = find_tabwin(&argvars[0], &argvars[1], NULL);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003132 if (wp == NULL)
3133 return;
3134
Bram Moolenaar57ee2b62019-02-12 22:15:06 +01003135 cleanup_jumplist(wp, TRUE);
3136
Bram Moolenaar4f505882018-02-10 21:06:32 +01003137 l = list_alloc();
3138 if (l == NULL)
3139 return;
3140
3141 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3142 return;
3143 list_append_number(rettv->vval.v_list, (varnumber_T)wp->w_jumplistidx);
3144
3145 for (i = 0; i < wp->w_jumplistlen; ++i)
3146 {
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003147 if (wp->w_jumplist[i].fmark.mark.lnum == 0)
3148 continue;
Bram Moolenaar4f505882018-02-10 21:06:32 +01003149 if ((d = dict_alloc()) == NULL)
3150 return;
3151 if (list_append_dict(l, d) == FAIL)
3152 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003153 dict_add_number(d, "lnum", (long)wp->w_jumplist[i].fmark.mark.lnum);
3154 dict_add_number(d, "col", (long)wp->w_jumplist[i].fmark.mark.col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003155 dict_add_number(d, "coladd", (long)wp->w_jumplist[i].fmark.mark.coladd);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003156 dict_add_number(d, "bufnr", (long)wp->w_jumplist[i].fmark.fnum);
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003157 if (wp->w_jumplist[i].fname != NULL)
Bram Moolenaare0be1672018-07-08 16:50:37 +02003158 dict_add_string(d, "filename", wp->w_jumplist[i].fname);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003159 }
3160#endif
3161}
3162
3163/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003164 * "getpid()" function
3165 */
3166 static void
3167f_getpid(typval_T *argvars UNUSED, typval_T *rettv)
3168{
3169 rettv->vval.v_number = mch_get_pid();
3170}
3171
3172 static void
3173getpos_both(
3174 typval_T *argvars,
3175 typval_T *rettv,
3176 int getcurpos)
3177{
3178 pos_T *fp;
3179 list_T *l;
3180 int fnum = -1;
3181
3182 if (rettv_list_alloc(rettv) == OK)
3183 {
3184 l = rettv->vval.v_list;
3185 if (getcurpos)
3186 fp = &curwin->w_cursor;
3187 else
3188 fp = var2fpos(&argvars[0], TRUE, &fnum);
3189 if (fnum != -1)
3190 list_append_number(l, (varnumber_T)fnum);
3191 else
3192 list_append_number(l, (varnumber_T)0);
3193 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->lnum
3194 : (varnumber_T)0);
3195 list_append_number(l, (fp != NULL)
3196 ? (varnumber_T)(fp->col == MAXCOL ? MAXCOL : fp->col + 1)
3197 : (varnumber_T)0);
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01003198 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->coladd :
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003199 (varnumber_T)0);
3200 if (getcurpos)
3201 {
Bram Moolenaar19a66852019-03-07 11:25:32 +01003202 int save_set_curswant = curwin->w_set_curswant;
3203 colnr_T save_curswant = curwin->w_curswant;
3204 colnr_T save_virtcol = curwin->w_virtcol;
3205
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003206 update_curswant();
3207 list_append_number(l, curwin->w_curswant == MAXCOL ?
3208 (varnumber_T)MAXCOL : (varnumber_T)curwin->w_curswant + 1);
Bram Moolenaar19a66852019-03-07 11:25:32 +01003209
3210 // Do not change "curswant", as it is unexpected that a get
3211 // function has a side effect.
3212 if (save_set_curswant)
3213 {
3214 curwin->w_set_curswant = save_set_curswant;
3215 curwin->w_curswant = save_curswant;
3216 curwin->w_virtcol = save_virtcol;
3217 curwin->w_valid &= ~VALID_VIRTCOL;
3218 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003219 }
3220 }
3221 else
3222 rettv->vval.v_number = FALSE;
3223}
3224
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003225/*
3226 * "getcurpos()" function
3227 */
3228 static void
3229f_getcurpos(typval_T *argvars, typval_T *rettv)
3230{
3231 getpos_both(argvars, rettv, TRUE);
3232}
3233
3234/*
3235 * "getpos(string)" function
3236 */
3237 static void
3238f_getpos(typval_T *argvars, typval_T *rettv)
3239{
3240 getpos_both(argvars, rettv, FALSE);
3241}
3242
3243/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003244 * "getreg()" function
3245 */
3246 static void
3247f_getreg(typval_T *argvars, typval_T *rettv)
3248{
3249 char_u *strregname;
3250 int regname;
3251 int arg2 = FALSE;
3252 int return_list = FALSE;
3253 int error = FALSE;
3254
3255 if (argvars[0].v_type != VAR_UNKNOWN)
3256 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003257 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003258 error = strregname == NULL;
3259 if (argvars[1].v_type != VAR_UNKNOWN)
3260 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003261 arg2 = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003262 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003263 return_list = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003264 }
3265 }
3266 else
3267 strregname = get_vim_var_str(VV_REG);
3268
3269 if (error)
3270 return;
3271
3272 regname = (strregname == NULL ? '"' : *strregname);
3273 if (regname == 0)
3274 regname = '"';
3275
3276 if (return_list)
3277 {
3278 rettv->v_type = VAR_LIST;
3279 rettv->vval.v_list = (list_T *)get_reg_contents(regname,
3280 (arg2 ? GREG_EXPR_SRC : 0) | GREG_LIST);
3281 if (rettv->vval.v_list == NULL)
3282 (void)rettv_list_alloc(rettv);
3283 else
3284 ++rettv->vval.v_list->lv_refcount;
3285 }
3286 else
3287 {
3288 rettv->v_type = VAR_STRING;
3289 rettv->vval.v_string = get_reg_contents(regname,
3290 arg2 ? GREG_EXPR_SRC : 0);
3291 }
3292}
3293
3294/*
3295 * "getregtype()" function
3296 */
3297 static void
3298f_getregtype(typval_T *argvars, typval_T *rettv)
3299{
3300 char_u *strregname;
3301 int regname;
3302 char_u buf[NUMBUFLEN + 2];
3303 long reglen = 0;
3304
3305 if (argvars[0].v_type != VAR_UNKNOWN)
3306 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003307 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003308 if (strregname == NULL) // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003309 {
3310 rettv->v_type = VAR_STRING;
3311 rettv->vval.v_string = NULL;
3312 return;
3313 }
3314 }
3315 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003316 // Default to v:register
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003317 strregname = get_vim_var_str(VV_REG);
3318
3319 regname = (strregname == NULL ? '"' : *strregname);
3320 if (regname == 0)
3321 regname = '"';
3322
3323 buf[0] = NUL;
3324 buf[1] = NUL;
3325 switch (get_reg_type(regname, &reglen))
3326 {
3327 case MLINE: buf[0] = 'V'; break;
3328 case MCHAR: buf[0] = 'v'; break;
3329 case MBLOCK:
3330 buf[0] = Ctrl_V;
3331 sprintf((char *)buf + 1, "%ld", reglen + 1);
3332 break;
3333 }
3334 rettv->v_type = VAR_STRING;
3335 rettv->vval.v_string = vim_strsave(buf);
3336}
3337
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003338/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01003339 * "gettagstack()" function
3340 */
3341 static void
3342f_gettagstack(typval_T *argvars, typval_T *rettv)
3343{
3344 win_T *wp = curwin; // default is current window
3345
3346 if (rettv_dict_alloc(rettv) != OK)
3347 return;
3348
3349 if (argvars[0].v_type != VAR_UNKNOWN)
3350 {
3351 wp = find_win_by_nr_or_id(&argvars[0]);
3352 if (wp == NULL)
3353 return;
3354 }
3355
3356 get_tagstack(wp, rettv->vval.v_dict);
3357}
3358
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003359// for VIM_VERSION_ defines
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003360#include "version.h"
3361
3362/*
3363 * "has()" function
3364 */
Bram Moolenaara259d8d2020-01-31 20:10:50 +01003365 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003366f_has(typval_T *argvars, typval_T *rettv)
3367{
3368 int i;
3369 char_u *name;
Bram Moolenaar79296512020-03-22 16:17:14 +01003370 int x = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003371 int n = FALSE;
Bram Moolenaar79296512020-03-22 16:17:14 +01003372 typedef struct {
3373 char *name;
3374 short present;
3375 } has_item_T;
3376 static has_item_T has_list[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003377 {
Bram Moolenaar79296512020-03-22 16:17:14 +01003378 {"amiga",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003379#ifdef AMIGA
Bram Moolenaar79296512020-03-22 16:17:14 +01003380 1
Bram Moolenaar39536dd2019-01-29 22:58:21 +01003381#else
Bram Moolenaar79296512020-03-22 16:17:14 +01003382 0
Bram Moolenaar39536dd2019-01-29 22:58:21 +01003383#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003384 },
3385 {"arp",
3386#if defined(AMIGA) && defined(FEAT_ARP)
3387 1
3388#else
3389 0
3390#endif
3391 },
Bram Moolenaar79296512020-03-22 16:17:14 +01003392 {"haiku",
3393#ifdef __HAIKU__
3394 1
3395#else
3396 0
3397#endif
3398 },
3399 {"bsd",
3400#if defined(BSD) && !defined(MACOS_X)
3401 1
3402#else
3403 0
3404#endif
3405 },
3406 {"hpux",
3407#ifdef hpux
3408 1
3409#else
3410 0
3411#endif
3412 },
3413 {"linux",
3414#ifdef __linux__
3415 1
3416#else
3417 0
3418#endif
3419 },
3420 {"mac", // Mac OS X (and, once, Mac OS Classic)
3421#ifdef MACOS_X
3422 1
3423#else
3424 0
3425#endif
3426 },
3427 {"osx", // Mac OS X
3428#ifdef MACOS_X
3429 1
3430#else
3431 0
3432#endif
3433 },
3434 {"macunix", // Mac OS X, with the darwin feature
3435#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
3436 1
3437#else
3438 0
3439#endif
3440 },
3441 {"osxdarwin", // synonym for macunix
3442#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
3443 1
3444#else
3445 0
3446#endif
3447 },
3448 {"qnx",
3449#ifdef __QNX__
3450 1
3451#else
3452 0
3453#endif
3454 },
3455 {"sun",
3456#ifdef SUN_SYSTEM
3457 1
3458#else
3459 0
3460#endif
3461 },
3462 {"unix",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003463#ifdef UNIX
Bram Moolenaar79296512020-03-22 16:17:14 +01003464 1
3465#else
3466 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003467#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003468 },
3469 {"vms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003470#ifdef VMS
Bram Moolenaar79296512020-03-22 16:17:14 +01003471 1
3472#else
3473 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003474#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003475 },
3476 {"win32",
Bram Moolenaar4f974752019-02-17 17:44:42 +01003477#ifdef MSWIN
Bram Moolenaar79296512020-03-22 16:17:14 +01003478 1
3479#else
3480 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003481#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003482 },
3483 {"win32unix",
Bram Moolenaar1eed5322019-02-26 17:03:54 +01003484#if defined(UNIX) && defined(__CYGWIN__)
Bram Moolenaar79296512020-03-22 16:17:14 +01003485 1
3486#else
3487 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003488#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003489 },
3490 {"win64",
Bram Moolenaar44b443c2019-02-18 22:14:18 +01003491#ifdef _WIN64
Bram Moolenaar79296512020-03-22 16:17:14 +01003492 1
3493#else
3494 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003495#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003496 },
3497 {"ebcdic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003498#ifdef EBCDIC
Bram Moolenaar79296512020-03-22 16:17:14 +01003499 1
3500#else
3501 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003502#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003503 },
3504 {"fname_case",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003505#ifndef CASE_INSENSITIVE_FILENAME
Bram Moolenaar79296512020-03-22 16:17:14 +01003506 1
3507#else
3508 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003509#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003510 },
3511 {"acl",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003512#ifdef HAVE_ACL
Bram Moolenaar79296512020-03-22 16:17:14 +01003513 1
3514#else
3515 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003516#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003517 },
3518 {"arabic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003519#ifdef FEAT_ARABIC
Bram Moolenaar79296512020-03-22 16:17:14 +01003520 1
3521#else
3522 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003523#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003524 },
3525 {"autocmd", 1},
3526 {"autochdir",
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02003527#ifdef FEAT_AUTOCHDIR
Bram Moolenaar79296512020-03-22 16:17:14 +01003528 1
3529#else
3530 0
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02003531#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003532 },
3533 {"autoservername",
Bram Moolenaare42a6d22017-11-12 19:21:51 +01003534#ifdef FEAT_AUTOSERVERNAME
Bram Moolenaar79296512020-03-22 16:17:14 +01003535 1
3536#else
3537 0
Bram Moolenaare42a6d22017-11-12 19:21:51 +01003538#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003539 },
3540 {"balloon_eval",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01003541#ifdef FEAT_BEVAL_GUI
Bram Moolenaar79296512020-03-22 16:17:14 +01003542 1
3543#else
3544 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003545#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003546 },
3547 {"balloon_multiline",
3548#if defined(FEAT_BEVAL_GUI) && !defined(FEAT_GUI_MSWIN)
3549 // MS-Windows requires runtime check, see below
3550 1
3551#else
3552 0
3553#endif
3554 },
3555 {"balloon_eval_term",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01003556#ifdef FEAT_BEVAL_TERM
Bram Moolenaar79296512020-03-22 16:17:14 +01003557 1
3558#else
3559 0
Bram Moolenaar51b0f372017-11-18 18:52:04 +01003560#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003561 },
3562 {"builtin_terms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003563#if defined(SOME_BUILTIN_TCAPS) || defined(ALL_BUILTIN_TCAPS)
Bram Moolenaar79296512020-03-22 16:17:14 +01003564 1
3565#else
3566 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003567#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003568 },
3569 {"all_builtin_terms",
3570#if defined(ALL_BUILTIN_TCAPS)
3571 1
3572#else
3573 0
3574#endif
3575 },
3576 {"browsefilter",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003577#if defined(FEAT_BROWSE) && (defined(USE_FILE_CHOOSER) \
Bram Moolenaar4f974752019-02-17 17:44:42 +01003578 || defined(FEAT_GUI_MSWIN) \
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003579 || defined(FEAT_GUI_MOTIF))
Bram Moolenaar79296512020-03-22 16:17:14 +01003580 1
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003581#else
Bram Moolenaar79296512020-03-22 16:17:14 +01003582 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003583#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003584 },
3585 {"byte_offset",
3586#ifdef FEAT_BYTEOFF
3587 1
3588#else
3589 0
3590#endif
3591 },
3592 {"channel",
3593#ifdef FEAT_JOB_CHANNEL
3594 1
3595#else
3596 0
3597#endif
3598 },
3599 {"cindent",
3600#ifdef FEAT_CINDENT
3601 1
3602#else
3603 0
3604#endif
3605 },
3606 {"clientserver",
3607#ifdef FEAT_CLIENTSERVER
3608 1
3609#else
3610 0
3611#endif
3612 },
3613 {"clipboard",
3614#ifdef FEAT_CLIPBOARD
3615 1
3616#else
3617 0
3618#endif
3619 },
3620 {"cmdline_compl", 1},
3621 {"cmdline_hist", 1},
3622 {"comments", 1},
3623 {"conceal",
3624#ifdef FEAT_CONCEAL
3625 1
3626#else
3627 0
3628#endif
3629 },
3630 {"cryptv",
3631#ifdef FEAT_CRYPT
3632 1
3633#else
3634 0
3635#endif
3636 },
3637 {"crypt-blowfish",
3638#ifdef FEAT_CRYPT
3639 1
3640#else
3641 0
3642#endif
3643 },
3644 {"crypt-blowfish2",
3645#ifdef FEAT_CRYPT
3646 1
3647#else
3648 0
3649#endif
3650 },
3651 {"cscope",
3652#ifdef FEAT_CSCOPE
3653 1
3654#else
3655 0
3656#endif
3657 },
3658 {"cursorbind", 1},
3659 {"cursorshape",
3660#ifdef CURSOR_SHAPE
3661 1
3662#else
3663 0
3664#endif
3665 },
3666 {"debug",
3667#ifdef DEBUG
3668 1
3669#else
3670 0
3671#endif
3672 },
3673 {"dialog_con",
3674#ifdef FEAT_CON_DIALOG
3675 1
3676#else
3677 0
3678#endif
3679 },
3680 {"dialog_gui",
3681#ifdef FEAT_GUI_DIALOG
3682 1
3683#else
3684 0
3685#endif
3686 },
3687 {"diff",
3688#ifdef FEAT_DIFF
3689 1
3690#else
3691 0
3692#endif
3693 },
3694 {"digraphs",
3695#ifdef FEAT_DIGRAPHS
3696 1
3697#else
3698 0
3699#endif
3700 },
3701 {"directx",
3702#ifdef FEAT_DIRECTX
3703 1
3704#else
3705 0
3706#endif
3707 },
3708 {"dnd",
3709#ifdef FEAT_DND
3710 1
3711#else
3712 0
3713#endif
3714 },
3715 {"emacs_tags",
3716#ifdef FEAT_EMACS_TAGS
3717 1
3718#else
3719 0
3720#endif
3721 },
3722 {"eval", 1}, // always present, of course!
3723 {"ex_extra", 1}, // graduated feature
3724 {"extra_search",
3725#ifdef FEAT_SEARCH_EXTRA
3726 1
3727#else
3728 0
3729#endif
3730 },
3731 {"file_in_path",
3732#ifdef FEAT_SEARCHPATH
3733 1
3734#else
3735 0
3736#endif
3737 },
3738 {"filterpipe",
3739#if defined(FEAT_FILTERPIPE) && !defined(VIMDLL)
3740 1
3741#else
3742 0
3743#endif
3744 },
3745 {"find_in_path",
3746#ifdef FEAT_FIND_ID
3747 1
3748#else
3749 0
3750#endif
3751 },
3752 {"float",
3753#ifdef FEAT_FLOAT
3754 1
3755#else
3756 0
3757#endif
3758 },
3759 {"folding",
3760#ifdef FEAT_FOLDING
3761 1
3762#else
3763 0
3764#endif
3765 },
3766 {"footer",
3767#ifdef FEAT_FOOTER
3768 1
3769#else
3770 0
3771#endif
3772 },
3773 {"fork",
3774#if !defined(USE_SYSTEM) && defined(UNIX)
3775 1
3776#else
3777 0
3778#endif
3779 },
3780 {"gettext",
3781#ifdef FEAT_GETTEXT
3782 1
3783#else
3784 0
3785#endif
3786 },
3787 {"gui",
3788#ifdef FEAT_GUI
3789 1
3790#else
3791 0
3792#endif
3793 },
3794 {"gui_neXtaw",
3795#if defined(FEAT_GUI_ATHENA) && defined(FEAT_GUI_NEXTAW)
3796 1
3797#else
3798 0
3799#endif
3800 },
3801 {"gui_athena",
3802#if defined(FEAT_GUI_ATHENA) && !defined(FEAT_GUI_NEXTAW)
3803 1
3804#else
3805 0
3806#endif
3807 },
3808 {"gui_gtk",
3809#ifdef FEAT_GUI_GTK
3810 1
3811#else
3812 0
3813#endif
3814 },
3815 {"gui_gtk2",
3816#if defined(FEAT_GUI_GTK) && !defined(USE_GTK3)
3817 1
3818#else
3819 0
3820#endif
3821 },
3822 {"gui_gtk3",
3823#if defined(FEAT_GUI_GTK) && defined(USE_GTK3)
3824 1
3825#else
3826 0
3827#endif
3828 },
3829 {"gui_gnome",
3830#ifdef FEAT_GUI_GNOME
3831 1
3832#else
3833 0
3834#endif
3835 },
3836 {"gui_haiku",
3837#ifdef FEAT_GUI_HAIKU
3838 1
3839#else
3840 0
3841#endif
3842 },
3843 {"gui_mac",
3844#ifdef FEAT_GUI_MAC
3845 1
3846#else
3847 0
3848#endif
3849 },
3850 {"gui_motif",
3851#ifdef FEAT_GUI_MOTIF
3852 1
3853#else
3854 0
3855#endif
3856 },
3857 {"gui_photon",
3858#ifdef FEAT_GUI_PHOTON
3859 1
3860#else
3861 0
3862#endif
3863 },
3864 {"gui_win32",
3865#ifdef FEAT_GUI_MSWIN
3866 1
3867#else
3868 0
3869#endif
3870 },
3871 {"iconv",
3872#if defined(HAVE_ICONV_H) && defined(USE_ICONV)
3873 1
3874#else
3875 0
3876#endif
3877 },
3878 {"insert_expand", 1},
Bram Moolenaarbfe13cc2020-04-12 17:53:12 +02003879 {"ipv6",
3880#ifdef FEAT_IPV6
3881 1
3882#else
3883 0
3884#endif
3885 },
Bram Moolenaar79296512020-03-22 16:17:14 +01003886 {"job",
3887#ifdef FEAT_JOB_CHANNEL
3888 1
3889#else
3890 0
3891#endif
3892 },
3893 {"jumplist",
3894#ifdef FEAT_JUMPLIST
3895 1
3896#else
3897 0
3898#endif
3899 },
3900 {"keymap",
3901#ifdef FEAT_KEYMAP
3902 1
3903#else
3904 0
3905#endif
3906 },
3907 {"lambda", 1}, // always with FEAT_EVAL, since 7.4.2120 with closure
3908 {"langmap",
3909#ifdef FEAT_LANGMAP
3910 1
3911#else
3912 0
3913#endif
3914 },
3915 {"libcall",
3916#ifdef FEAT_LIBCALL
3917 1
3918#else
3919 0
3920#endif
3921 },
3922 {"linebreak",
3923#ifdef FEAT_LINEBREAK
3924 1
3925#else
3926 0
3927#endif
3928 },
3929 {"lispindent",
3930#ifdef FEAT_LISP
3931 1
3932#else
3933 0
3934#endif
3935 },
3936 {"listcmds", 1},
3937 {"localmap", 1},
3938 {"lua",
3939#if defined(FEAT_LUA) && !defined(DYNAMIC_LUA)
3940 1
3941#else
3942 0
3943#endif
3944 },
3945 {"menu",
3946#ifdef FEAT_MENU
3947 1
3948#else
3949 0
3950#endif
3951 },
3952 {"mksession",
3953#ifdef FEAT_SESSION
3954 1
3955#else
3956 0
3957#endif
3958 },
3959 {"modify_fname", 1},
3960 {"mouse", 1},
3961 {"mouseshape",
3962#ifdef FEAT_MOUSESHAPE
3963 1
3964#else
3965 0
3966#endif
3967 },
3968 {"mouse_dec",
3969#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_DEC)
3970 1
3971#else
3972 0
3973#endif
3974 },
3975 {"mouse_gpm",
3976#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_GPM)
3977 1
3978#else
3979 0
3980#endif
3981 },
3982 {"mouse_jsbterm",
3983#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_JSB)
3984 1
3985#else
3986 0
3987#endif
3988 },
3989 {"mouse_netterm",
3990#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_NET)
3991 1
3992#else
3993 0
3994#endif
3995 },
3996 {"mouse_pterm",
3997#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_PTERM)
3998 1
3999#else
4000 0
4001#endif
4002 },
4003 {"mouse_sgr",
4004#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4005 1
4006#else
4007 0
4008#endif
4009 },
4010 {"mouse_sysmouse",
4011#if (defined(UNIX) || defined(VMS)) && defined(FEAT_SYSMOUSE)
4012 1
4013#else
4014 0
4015#endif
4016 },
4017 {"mouse_urxvt",
4018#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_URXVT)
4019 1
4020#else
4021 0
4022#endif
4023 },
4024 {"mouse_xterm",
4025#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4026 1
4027#else
4028 0
4029#endif
4030 },
4031 {"multi_byte", 1},
4032 {"multi_byte_ime",
4033#ifdef FEAT_MBYTE_IME
4034 1
4035#else
4036 0
4037#endif
4038 },
4039 {"multi_lang",
4040#ifdef FEAT_MULTI_LANG
4041 1
4042#else
4043 0
4044#endif
4045 },
4046 {"mzscheme",
4047#if defined(FEAT_MZSCHEME) && !defined(DYNAMIC_MZSCHEME)
4048 1
4049#else
4050 0
4051#endif
4052 },
4053 {"num64", 1},
4054 {"ole",
4055#ifdef FEAT_OLE
4056 1
4057#else
4058 0
4059#endif
4060 },
4061 {"packages",
4062#ifdef FEAT_EVAL
4063 1
4064#else
4065 0
4066#endif
4067 },
4068 {"path_extra",
4069#ifdef FEAT_PATH_EXTRA
4070 1
4071#else
4072 0
4073#endif
4074 },
4075 {"perl",
4076#if defined(FEAT_PERL) && !defined(DYNAMIC_PERL)
4077 1
4078#else
4079 0
4080#endif
4081 },
4082 {"persistent_undo",
4083#ifdef FEAT_PERSISTENT_UNDO
4084 1
4085#else
4086 0
4087#endif
4088 },
4089 {"python_compiled",
4090#if defined(FEAT_PYTHON)
4091 1
4092#else
4093 0
4094#endif
4095 },
4096 {"python_dynamic",
4097#if defined(FEAT_PYTHON) && defined(DYNAMIC_PYTHON)
4098 1
4099#else
4100 0
4101#endif
4102 },
4103 {"python",
4104#if defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)
4105 1
4106#else
4107 0
4108#endif
4109 },
4110 {"pythonx",
4111#if (defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)) \
4112 || (defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3))
4113 1
4114#else
4115 0
4116#endif
4117 },
4118 {"python3_compiled",
4119#if defined(FEAT_PYTHON3)
4120 1
4121#else
4122 0
4123#endif
4124 },
4125 {"python3_dynamic",
4126#if defined(FEAT_PYTHON3) && defined(DYNAMIC_PYTHON3)
4127 1
4128#else
4129 0
4130#endif
4131 },
4132 {"python3",
4133#if defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3)
4134 1
4135#else
4136 0
4137#endif
4138 },
4139 {"popupwin",
4140#ifdef FEAT_PROP_POPUP
4141 1
4142#else
4143 0
4144#endif
4145 },
4146 {"postscript",
4147#ifdef FEAT_POSTSCRIPT
4148 1
4149#else
4150 0
4151#endif
4152 },
4153 {"printer",
4154#ifdef FEAT_PRINTER
4155 1
4156#else
4157 0
4158#endif
4159 },
4160 {"profile",
4161#ifdef FEAT_PROFILE
4162 1
4163#else
4164 0
4165#endif
4166 },
4167 {"reltime",
4168#ifdef FEAT_RELTIME
4169 1
4170#else
4171 0
4172#endif
4173 },
4174 {"quickfix",
4175#ifdef FEAT_QUICKFIX
4176 1
4177#else
4178 0
4179#endif
4180 },
4181 {"rightleft",
4182#ifdef FEAT_RIGHTLEFT
4183 1
4184#else
4185 0
4186#endif
4187 },
4188 {"ruby",
4189#if defined(FEAT_RUBY) && !defined(DYNAMIC_RUBY)
4190 1
4191#else
4192 0
4193#endif
4194 },
4195 {"scrollbind", 1},
4196 {"showcmd",
4197#ifdef FEAT_CMDL_INFO
4198 1
4199#else
4200 0
4201#endif
4202 },
4203 {"cmdline_info",
4204#ifdef FEAT_CMDL_INFO
4205 1
4206#else
4207 0
4208#endif
4209 },
4210 {"signs",
4211#ifdef FEAT_SIGNS
4212 1
4213#else
4214 0
4215#endif
4216 },
4217 {"smartindent",
4218#ifdef FEAT_SMARTINDENT
4219 1
4220#else
4221 0
4222#endif
4223 },
4224 {"startuptime",
4225#ifdef STARTUPTIME
4226 1
4227#else
4228 0
4229#endif
4230 },
4231 {"statusline",
4232#ifdef FEAT_STL_OPT
4233 1
4234#else
4235 0
4236#endif
4237 },
4238 {"netbeans_intg",
4239#ifdef FEAT_NETBEANS_INTG
4240 1
4241#else
4242 0
4243#endif
4244 },
4245 {"sound",
4246#ifdef FEAT_SOUND
4247 1
4248#else
4249 0
4250#endif
4251 },
4252 {"spell",
4253#ifdef FEAT_SPELL
4254 1
4255#else
4256 0
4257#endif
4258 },
4259 {"syntax",
4260#ifdef FEAT_SYN_HL
4261 1
4262#else
4263 0
4264#endif
4265 },
4266 {"system",
4267#if defined(USE_SYSTEM) || !defined(UNIX)
4268 1
4269#else
4270 0
4271#endif
4272 },
4273 {"tag_binary",
4274#ifdef FEAT_TAG_BINS
4275 1
4276#else
4277 0
4278#endif
4279 },
4280 {"tcl",
4281#if defined(FEAT_TCL) && !defined(DYNAMIC_TCL)
4282 1
4283#else
4284 0
4285#endif
4286 },
4287 {"termguicolors",
4288#ifdef FEAT_TERMGUICOLORS
4289 1
4290#else
4291 0
4292#endif
4293 },
4294 {"terminal",
4295#if defined(FEAT_TERMINAL) && !defined(MSWIN)
4296 1
4297#else
4298 0
4299#endif
4300 },
4301 {"terminfo",
4302#ifdef TERMINFO
4303 1
4304#else
4305 0
4306#endif
4307 },
4308 {"termresponse",
4309#ifdef FEAT_TERMRESPONSE
4310 1
4311#else
4312 0
4313#endif
4314 },
4315 {"textobjects",
4316#ifdef FEAT_TEXTOBJ
4317 1
4318#else
4319 0
4320#endif
4321 },
4322 {"textprop",
4323#ifdef FEAT_PROP_POPUP
4324 1
4325#else
4326 0
4327#endif
4328 },
4329 {"tgetent",
4330#ifdef HAVE_TGETENT
4331 1
4332#else
4333 0
4334#endif
4335 },
4336 {"timers",
4337#ifdef FEAT_TIMERS
4338 1
4339#else
4340 0
4341#endif
4342 },
4343 {"title",
4344#ifdef FEAT_TITLE
4345 1
4346#else
4347 0
4348#endif
4349 },
4350 {"toolbar",
4351#ifdef FEAT_TOOLBAR
4352 1
4353#else
4354 0
4355#endif
4356 },
4357 {"unnamedplus",
4358#if defined(FEAT_CLIPBOARD) && defined(FEAT_X11)
4359 1
4360#else
4361 0
4362#endif
4363 },
4364 {"user-commands", 1}, // was accidentally included in 5.4
4365 {"user_commands", 1},
4366 {"vartabs",
4367#ifdef FEAT_VARTABS
4368 1
4369#else
4370 0
4371#endif
4372 },
4373 {"vertsplit", 1},
4374 {"viminfo",
4375#ifdef FEAT_VIMINFO
4376 1
4377#else
4378 0
4379#endif
4380 },
4381 {"vimscript-1", 1},
4382 {"vimscript-2", 1},
4383 {"vimscript-3", 1},
4384 {"vimscript-4", 1},
4385 {"virtualedit", 1},
4386 {"visual", 1},
4387 {"visualextra", 1},
4388 {"vreplace", 1},
4389 {"vtp",
4390#ifdef FEAT_VTP
4391 1
4392#else
4393 0
4394#endif
4395 },
4396 {"wildignore",
4397#ifdef FEAT_WILDIGN
4398 1
4399#else
4400 0
4401#endif
4402 },
4403 {"wildmenu",
4404#ifdef FEAT_WILDMENU
4405 1
4406#else
4407 0
4408#endif
4409 },
4410 {"windows", 1},
4411 {"winaltkeys",
4412#ifdef FEAT_WAK
4413 1
4414#else
4415 0
4416#endif
4417 },
4418 {"writebackup",
4419#ifdef FEAT_WRITEBACKUP
4420 1
4421#else
4422 0
4423#endif
4424 },
4425 {"xim",
4426#ifdef FEAT_XIM
4427 1
4428#else
4429 0
4430#endif
4431 },
4432 {"xfontset",
4433#ifdef FEAT_XFONTSET
4434 1
4435#else
4436 0
4437#endif
4438 },
4439 {"xpm",
4440#if defined(FEAT_XPM_W32) || defined(HAVE_XPM)
4441 1
4442#else
4443 0
4444#endif
4445 },
4446 {"xpm_w32", // for backward compatibility
4447#ifdef FEAT_XPM_W32
4448 1
4449#else
4450 0
4451#endif
4452 },
4453 {"xsmp",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004454#ifdef USE_XSMP
Bram Moolenaar79296512020-03-22 16:17:14 +01004455 1
4456#else
4457 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004458#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004459 },
4460 {"xsmp_interact",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004461#ifdef USE_XSMP_INTERACT
Bram Moolenaar79296512020-03-22 16:17:14 +01004462 1
4463#else
4464 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004465#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004466 },
4467 {"xterm_clipboard",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004468#ifdef FEAT_XCLIPBOARD
Bram Moolenaar79296512020-03-22 16:17:14 +01004469 1
4470#else
4471 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004472#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004473 },
4474 {"xterm_save",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004475#ifdef FEAT_XTERM_SAVE
Bram Moolenaar79296512020-03-22 16:17:14 +01004476 1
4477#else
4478 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004479#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004480 },
4481 {"X11",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004482#if defined(UNIX) && defined(FEAT_X11)
Bram Moolenaar79296512020-03-22 16:17:14 +01004483 1
4484#else
4485 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004486#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004487 },
4488 {NULL, 0}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004489 };
4490
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004491 name = tv_get_string(&argvars[0]);
Bram Moolenaar79296512020-03-22 16:17:14 +01004492 for (i = 0; has_list[i].name != NULL; ++i)
4493 if (STRICMP(name, has_list[i].name) == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004494 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004495 x = TRUE;
4496 n = has_list[i].present;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004497 break;
4498 }
4499
Bram Moolenaar79296512020-03-22 16:17:14 +01004500 // features also in has_list[] but sometimes enabled at runtime
4501 if (x == TRUE && n == FALSE)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004502 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004503 if (0)
Bram Moolenaar86b9a3e2020-04-07 19:57:29 +02004504 {
4505 // intentionally empty
4506 }
Bram Moolenaar4f974752019-02-17 17:44:42 +01004507#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004508 else if (STRICMP(name, "balloon_multiline") == 0)
4509 n = multiline_balloon_available();
4510#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004511#ifdef VIMDLL
4512 else if (STRICMP(name, "filterpipe") == 0)
4513 n = gui.in_use || gui.starting;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004514#endif
4515#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
4516 else if (STRICMP(name, "iconv") == 0)
4517 n = iconv_enabled(FALSE);
4518#endif
4519#ifdef DYNAMIC_LUA
4520 else if (STRICMP(name, "lua") == 0)
4521 n = lua_enabled(FALSE);
4522#endif
4523#ifdef DYNAMIC_MZSCHEME
4524 else if (STRICMP(name, "mzscheme") == 0)
4525 n = mzscheme_enabled(FALSE);
4526#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004527#ifdef DYNAMIC_PERL
4528 else if (STRICMP(name, "perl") == 0)
4529 n = perl_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004530#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004531#ifdef DYNAMIC_PYTHON
4532 else if (STRICMP(name, "python") == 0)
4533 n = python_enabled(FALSE);
4534#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004535#ifdef DYNAMIC_PYTHON3
4536 else if (STRICMP(name, "python3") == 0)
4537 n = python3_enabled(FALSE);
4538#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01004539#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
4540 else if (STRICMP(name, "pythonx") == 0)
4541 {
4542# if defined(DYNAMIC_PYTHON) && defined(DYNAMIC_PYTHON3)
4543 if (p_pyx == 0)
4544 n = python3_enabled(FALSE) || python_enabled(FALSE);
4545 else if (p_pyx == 3)
4546 n = python3_enabled(FALSE);
4547 else if (p_pyx == 2)
4548 n = python_enabled(FALSE);
4549# elif defined(DYNAMIC_PYTHON)
4550 n = python_enabled(FALSE);
4551# elif defined(DYNAMIC_PYTHON3)
4552 n = python3_enabled(FALSE);
4553# endif
4554 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004555#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004556#ifdef DYNAMIC_RUBY
4557 else if (STRICMP(name, "ruby") == 0)
4558 n = ruby_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004559#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004560#ifdef DYNAMIC_TCL
4561 else if (STRICMP(name, "tcl") == 0)
4562 n = tcl_enabled(FALSE);
Bram Moolenaar4b8366b2019-05-04 17:34:34 +02004563#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +01004564#if defined(FEAT_TERMINAL) && defined(MSWIN)
Bram Moolenaara83e3962017-08-17 14:39:07 +02004565 else if (STRICMP(name, "terminal") == 0)
4566 n = terminal_enabled();
4567#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004568 }
4569
Bram Moolenaar79296512020-03-22 16:17:14 +01004570 // features not in has_list[]
4571 if (x == FALSE)
4572 {
4573 if (STRNICMP(name, "patch", 5) == 0)
4574 {
4575 x = TRUE;
4576 if (name[5] == '-'
4577 && STRLEN(name) >= 11
4578 && vim_isdigit(name[6])
4579 && vim_isdigit(name[8])
4580 && vim_isdigit(name[10]))
4581 {
4582 int major = atoi((char *)name + 6);
4583 int minor = atoi((char *)name + 8);
4584
4585 // Expect "patch-9.9.01234".
4586 n = (major < VIM_VERSION_MAJOR
4587 || (major == VIM_VERSION_MAJOR
4588 && (minor < VIM_VERSION_MINOR
4589 || (minor == VIM_VERSION_MINOR
4590 && has_patch(atoi((char *)name + 10))))));
4591 }
4592 else
4593 n = has_patch(atoi((char *)name + 5));
4594 }
4595 else if (STRICMP(name, "vim_starting") == 0)
4596 {
4597 x = TRUE;
4598 n = (starting != 0);
4599 }
4600 else if (STRICMP(name, "ttyin") == 0)
4601 {
4602 x = TRUE;
4603 n = mch_input_isatty();
4604 }
4605 else if (STRICMP(name, "ttyout") == 0)
4606 {
4607 x = TRUE;
4608 n = stdout_isatty;
4609 }
4610 else if (STRICMP(name, "multi_byte_encoding") == 0)
4611 {
4612 x = TRUE;
4613 n = has_mbyte;
4614 }
4615 else if (STRICMP(name, "gui_running") == 0)
4616 {
4617 x = TRUE;
4618#ifdef FEAT_GUI
4619 n = (gui.in_use || gui.starting);
4620#endif
4621 }
4622 else if (STRICMP(name, "browse") == 0)
4623 {
4624 x = TRUE;
4625#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
4626 n = gui.in_use; // gui_mch_browse() works when GUI is running
4627#endif
4628 }
4629 else if (STRICMP(name, "syntax_items") == 0)
4630 {
4631 x = TRUE;
4632#ifdef FEAT_SYN_HL
4633 n = syntax_present(curwin);
4634#endif
4635 }
4636 else if (STRICMP(name, "vcon") == 0)
4637 {
4638 x = TRUE;
4639#ifdef FEAT_VTP
4640 n = is_term_win32() && has_vtp_working();
4641#endif
4642 }
4643 else if (STRICMP(name, "netbeans_enabled") == 0)
4644 {
4645 x = TRUE;
4646#ifdef FEAT_NETBEANS_INTG
4647 n = netbeans_active();
4648#endif
4649 }
4650 else if (STRICMP(name, "mouse_gpm_enabled") == 0)
4651 {
4652 x = TRUE;
4653#ifdef FEAT_MOUSE_GPM
4654 n = gpm_enabled();
4655#endif
4656 }
4657 else if (STRICMP(name, "conpty") == 0)
4658 {
4659 x = TRUE;
4660#if defined(FEAT_TERMINAL) && defined(MSWIN)
4661 n = use_conpty();
4662#endif
4663 }
4664 else if (STRICMP(name, "clipboard_working") == 0)
4665 {
4666 x = TRUE;
4667#ifdef FEAT_CLIPBOARD
4668 n = clip_star.available;
4669#endif
4670 }
4671 }
4672
4673 if (argvars[1].v_type != VAR_UNKNOWN && tv_get_number(&argvars[1]) != 0)
4674 // return whether feature could ever be enabled
4675 rettv->vval.v_number = x;
4676 else
4677 // return whether feature is enabled
4678 rettv->vval.v_number = n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004679}
4680
4681/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004682 * "haslocaldir()" function
4683 */
4684 static void
4685f_haslocaldir(typval_T *argvars, typval_T *rettv)
4686{
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004687 tabpage_T *tp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004688 win_T *wp = NULL;
4689
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004690 wp = find_tabwin(&argvars[0], &argvars[1], &tp);
4691
4692 // Check for window-local and tab-local directories
4693 if (wp != NULL && wp->w_localdir != NULL)
4694 rettv->vval.v_number = 1;
4695 else if (tp != NULL && tp->tp_localdir != NULL)
4696 rettv->vval.v_number = 2;
4697 else
4698 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004699}
4700
4701/*
4702 * "hasmapto()" function
4703 */
4704 static void
4705f_hasmapto(typval_T *argvars, typval_T *rettv)
4706{
4707 char_u *name;
4708 char_u *mode;
4709 char_u buf[NUMBUFLEN];
4710 int abbr = FALSE;
4711
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004712 name = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004713 if (argvars[1].v_type == VAR_UNKNOWN)
4714 mode = (char_u *)"nvo";
4715 else
4716 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004717 mode = tv_get_string_buf(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004718 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004719 abbr = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004720 }
4721
4722 if (map_to_exists(name, mode, abbr))
4723 rettv->vval.v_number = TRUE;
4724 else
4725 rettv->vval.v_number = FALSE;
4726}
4727
4728/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004729 * "highlightID(name)" function
4730 */
4731 static void
4732f_hlID(typval_T *argvars, typval_T *rettv)
4733{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004734 rettv->vval.v_number = syn_name2id(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004735}
4736
4737/*
4738 * "highlight_exists()" function
4739 */
4740 static void
4741f_hlexists(typval_T *argvars, typval_T *rettv)
4742{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004743 rettv->vval.v_number = highlight_exists(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004744}
4745
4746/*
4747 * "hostname()" function
4748 */
4749 static void
4750f_hostname(typval_T *argvars UNUSED, typval_T *rettv)
4751{
4752 char_u hostname[256];
4753
4754 mch_get_host_name(hostname, 256);
4755 rettv->v_type = VAR_STRING;
4756 rettv->vval.v_string = vim_strsave(hostname);
4757}
4758
4759/*
4760 * iconv() function
4761 */
4762 static void
4763f_iconv(typval_T *argvars UNUSED, typval_T *rettv)
4764{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004765 char_u buf1[NUMBUFLEN];
4766 char_u buf2[NUMBUFLEN];
4767 char_u *from, *to, *str;
4768 vimconv_T vimconv;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004769
4770 rettv->v_type = VAR_STRING;
4771 rettv->vval.v_string = NULL;
4772
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004773 str = tv_get_string(&argvars[0]);
4774 from = enc_canonize(enc_skip(tv_get_string_buf(&argvars[1], buf1)));
4775 to = enc_canonize(enc_skip(tv_get_string_buf(&argvars[2], buf2)));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004776 vimconv.vc_type = CONV_NONE;
4777 convert_setup(&vimconv, from, to);
4778
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004779 // If the encodings are equal, no conversion needed.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004780 if (vimconv.vc_type == CONV_NONE)
4781 rettv->vval.v_string = vim_strsave(str);
4782 else
4783 rettv->vval.v_string = string_convert(&vimconv, str, NULL);
4784
4785 convert_setup(&vimconv, NULL, NULL);
4786 vim_free(from);
4787 vim_free(to);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004788}
4789
4790/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004791 * "index()" function
4792 */
4793 static void
4794f_index(typval_T *argvars, typval_T *rettv)
4795{
4796 list_T *l;
4797 listitem_T *item;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004798 blob_T *b;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004799 long idx = 0;
4800 int ic = FALSE;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004801 int error = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004802
4803 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004804 if (argvars[0].v_type == VAR_BLOB)
4805 {
4806 typval_T tv;
4807 int start = 0;
4808
4809 if (argvars[2].v_type != VAR_UNKNOWN)
4810 {
4811 start = tv_get_number_chk(&argvars[2], &error);
4812 if (error)
4813 return;
4814 }
4815 b = argvars[0].vval.v_blob;
4816 if (b == NULL)
4817 return;
Bram Moolenaar05500ec2019-01-13 19:10:33 +01004818 if (start < 0)
4819 {
4820 start = blob_len(b) + start;
4821 if (start < 0)
4822 start = 0;
4823 }
4824
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004825 for (idx = start; idx < blob_len(b); ++idx)
4826 {
4827 tv.v_type = VAR_NUMBER;
4828 tv.vval.v_number = blob_get(b, idx);
4829 if (tv_equal(&tv, &argvars[1], ic, FALSE))
4830 {
4831 rettv->vval.v_number = idx;
4832 return;
4833 }
4834 }
4835 return;
4836 }
4837 else if (argvars[0].v_type != VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004838 {
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01004839 emsg(_(e_listblobreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004840 return;
4841 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004842
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004843 l = argvars[0].vval.v_list;
4844 if (l != NULL)
4845 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02004846 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004847 item = l->lv_first;
4848 if (argvars[2].v_type != VAR_UNKNOWN)
4849 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004850 // Start at specified item. Use the cached index that list_find()
4851 // sets, so that a negative number also works.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004852 item = list_find(l, (long)tv_get_number_chk(&argvars[2], &error));
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01004853 idx = l->lv_u.mat.lv_idx;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004854 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004855 ic = (int)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004856 if (error)
4857 item = NULL;
4858 }
4859
4860 for ( ; item != NULL; item = item->li_next, ++idx)
4861 if (tv_equal(&item->li_tv, &argvars[1], ic, FALSE))
4862 {
4863 rettv->vval.v_number = idx;
4864 break;
4865 }
4866 }
4867}
4868
4869static int inputsecret_flag = 0;
4870
4871/*
4872 * "input()" function
4873 * Also handles inputsecret() when inputsecret is set.
4874 */
4875 static void
4876f_input(typval_T *argvars, typval_T *rettv)
4877{
4878 get_user_input(argvars, rettv, FALSE, inputsecret_flag);
4879}
4880
4881/*
4882 * "inputdialog()" function
4883 */
4884 static void
4885f_inputdialog(typval_T *argvars, typval_T *rettv)
4886{
4887#if defined(FEAT_GUI_TEXTDIALOG)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004888 // Use a GUI dialog if the GUI is running and 'c' is not in 'guioptions'
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004889 if (gui.in_use && vim_strchr(p_go, GO_CONDIALOG) == NULL)
4890 {
4891 char_u *message;
4892 char_u buf[NUMBUFLEN];
4893 char_u *defstr = (char_u *)"";
4894
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004895 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004896 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004897 && (defstr = tv_get_string_buf_chk(&argvars[1], buf)) != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004898 vim_strncpy(IObuff, defstr, IOSIZE - 1);
4899 else
4900 IObuff[0] = NUL;
4901 if (message != NULL && defstr != NULL
4902 && do_dialog(VIM_QUESTION, NULL, message,
4903 (char_u *)_("&OK\n&Cancel"), 1, IObuff, FALSE) == 1)
4904 rettv->vval.v_string = vim_strsave(IObuff);
4905 else
4906 {
4907 if (message != NULL && defstr != NULL
4908 && argvars[1].v_type != VAR_UNKNOWN
4909 && argvars[2].v_type != VAR_UNKNOWN)
4910 rettv->vval.v_string = vim_strsave(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004911 tv_get_string_buf(&argvars[2], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004912 else
4913 rettv->vval.v_string = NULL;
4914 }
4915 rettv->v_type = VAR_STRING;
4916 }
4917 else
4918#endif
4919 get_user_input(argvars, rettv, TRUE, inputsecret_flag);
4920}
4921
4922/*
4923 * "inputlist()" function
4924 */
4925 static void
4926f_inputlist(typval_T *argvars, typval_T *rettv)
4927{
Bram Moolenaar50985eb2020-01-27 22:09:39 +01004928 list_T *l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004929 listitem_T *li;
4930 int selected;
4931 int mouse_used;
4932
4933#ifdef NO_CONSOLE_INPUT
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004934 // While starting up, there is no place to enter text. When running tests
4935 // with --not-a-term we assume feedkeys() will be used.
Bram Moolenaar91d348a2017-07-29 20:16:03 +02004936 if (no_console_input() && !is_not_a_term())
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004937 return;
4938#endif
4939 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
4940 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01004941 semsg(_(e_listarg), "inputlist()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004942 return;
4943 }
4944
4945 msg_start();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004946 msg_row = Rows - 1; // for when 'cmdheight' > 1
4947 lines_left = Rows; // avoid more prompt
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004948 msg_scroll = TRUE;
4949 msg_clr_eos();
4950
Bram Moolenaar50985eb2020-01-27 22:09:39 +01004951 l = argvars[0].vval.v_list;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02004952 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02004953 FOR_ALL_LIST_ITEMS(l, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004954 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01004955 msg_puts((char *)tv_get_string(&li->li_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004956 msg_putchar('\n');
4957 }
4958
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004959 // Ask for choice.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004960 selected = prompt_for_number(&mouse_used);
4961 if (mouse_used)
4962 selected -= lines_left;
4963
4964 rettv->vval.v_number = selected;
4965}
4966
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004967static garray_T ga_userinput = {0, 0, sizeof(tasave_T), 4, NULL};
4968
4969/*
4970 * "inputrestore()" function
4971 */
4972 static void
4973f_inputrestore(typval_T *argvars UNUSED, typval_T *rettv)
4974{
4975 if (ga_userinput.ga_len > 0)
4976 {
4977 --ga_userinput.ga_len;
4978 restore_typeahead((tasave_T *)(ga_userinput.ga_data)
4979 + ga_userinput.ga_len);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004980 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004981 }
4982 else if (p_verbose > 1)
4983 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01004984 verb_msg(_("called inputrestore() more often than inputsave()"));
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004985 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004986 }
4987}
4988
4989/*
4990 * "inputsave()" function
4991 */
4992 static void
4993f_inputsave(typval_T *argvars UNUSED, typval_T *rettv)
4994{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004995 // Add an entry to the stack of typeahead storage.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004996 if (ga_grow(&ga_userinput, 1) == OK)
4997 {
4998 save_typeahead((tasave_T *)(ga_userinput.ga_data)
4999 + ga_userinput.ga_len);
5000 ++ga_userinput.ga_len;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005001 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005002 }
5003 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005004 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005005}
5006
5007/*
5008 * "inputsecret()" function
5009 */
5010 static void
5011f_inputsecret(typval_T *argvars, typval_T *rettv)
5012{
5013 ++cmdline_star;
5014 ++inputsecret_flag;
5015 f_input(argvars, rettv);
5016 --cmdline_star;
5017 --inputsecret_flag;
5018}
5019
5020/*
Bram Moolenaar67a2deb2019-11-25 00:05:32 +01005021 * "interrupt()" function
5022 */
5023 static void
5024f_interrupt(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5025{
5026 got_int = TRUE;
5027}
5028
5029/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005030 * "invert(expr)" function
5031 */
5032 static void
5033f_invert(typval_T *argvars, typval_T *rettv)
5034{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005035 rettv->vval.v_number = ~tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005036}
5037
5038/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005039 * "islocked()" function
5040 */
5041 static void
5042f_islocked(typval_T *argvars, typval_T *rettv)
5043{
5044 lval_T lv;
5045 char_u *end;
5046 dictitem_T *di;
5047
5048 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005049 end = get_lval(tv_get_string(&argvars[0]), NULL, &lv, FALSE, FALSE,
Bram Moolenaar3a257732017-02-21 20:47:13 +01005050 GLV_NO_AUTOLOAD | GLV_READ_ONLY, FNE_CHECK_START);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005051 if (end != NULL && lv.ll_name != NULL)
5052 {
5053 if (*end != NUL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005054 emsg(_(e_trailing));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005055 else
5056 {
5057 if (lv.ll_tv == NULL)
5058 {
Bram Moolenaar79518e22017-02-17 16:31:35 +01005059 di = find_var(lv.ll_name, NULL, TRUE);
5060 if (di != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005061 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005062 // Consider a variable locked when:
5063 // 1. the variable itself is locked
5064 // 2. the value of the variable is locked.
5065 // 3. the List or Dict value is locked.
Bram Moolenaar79518e22017-02-17 16:31:35 +01005066 rettv->vval.v_number = ((di->di_flags & DI_FLAGS_LOCK)
5067 || tv_islocked(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005068 }
5069 }
5070 else if (lv.ll_range)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005071 emsg(_("E786: Range not allowed"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005072 else if (lv.ll_newkey != NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005073 semsg(_(e_dictkey), lv.ll_newkey);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005074 else if (lv.ll_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005075 // List item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005076 rettv->vval.v_number = tv_islocked(&lv.ll_li->li_tv);
5077 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005078 // Dictionary item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005079 rettv->vval.v_number = tv_islocked(&lv.ll_di->di_tv);
5080 }
5081 }
5082
5083 clear_lval(&lv);
5084}
5085
5086#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
5087/*
Bram Moolenaarfda1bff2019-04-04 13:44:37 +02005088 * "isinf()" function
5089 */
5090 static void
5091f_isinf(typval_T *argvars, typval_T *rettv)
5092{
5093 if (argvars[0].v_type == VAR_FLOAT && isinf(argvars[0].vval.v_float))
5094 rettv->vval.v_number = argvars[0].vval.v_float > 0.0 ? 1 : -1;
5095}
5096
5097/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005098 * "isnan()" function
5099 */
5100 static void
5101f_isnan(typval_T *argvars, typval_T *rettv)
5102{
5103 rettv->vval.v_number = argvars[0].v_type == VAR_FLOAT
5104 && isnan(argvars[0].vval.v_float);
5105}
5106#endif
5107
5108/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005109 * "last_buffer_nr()" function.
5110 */
5111 static void
5112f_last_buffer_nr(typval_T *argvars UNUSED, typval_T *rettv)
5113{
5114 int n = 0;
5115 buf_T *buf;
5116
Bram Moolenaar29323592016-07-24 22:04:11 +02005117 FOR_ALL_BUFFERS(buf)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005118 if (n < buf->b_fnum)
5119 n = buf->b_fnum;
5120
5121 rettv->vval.v_number = n;
5122}
5123
5124/*
5125 * "len()" function
5126 */
5127 static void
5128f_len(typval_T *argvars, typval_T *rettv)
5129{
5130 switch (argvars[0].v_type)
5131 {
5132 case VAR_STRING:
5133 case VAR_NUMBER:
5134 rettv->vval.v_number = (varnumber_T)STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005135 tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005136 break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005137 case VAR_BLOB:
5138 rettv->vval.v_number = blob_len(argvars[0].vval.v_blob);
5139 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005140 case VAR_LIST:
5141 rettv->vval.v_number = list_len(argvars[0].vval.v_list);
5142 break;
5143 case VAR_DICT:
5144 rettv->vval.v_number = dict_len(argvars[0].vval.v_dict);
5145 break;
5146 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02005147 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01005148 case VAR_VOID:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01005149 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005150 case VAR_SPECIAL:
5151 case VAR_FLOAT:
5152 case VAR_FUNC:
5153 case VAR_PARTIAL:
5154 case VAR_JOB:
5155 case VAR_CHANNEL:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005156 emsg(_("E701: Invalid type for len()"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005157 break;
5158 }
5159}
5160
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005161 static void
Bram Moolenaar6d721c72017-01-17 16:56:28 +01005162libcall_common(typval_T *argvars UNUSED, typval_T *rettv, int type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005163{
5164#ifdef FEAT_LIBCALL
5165 char_u *string_in;
5166 char_u **string_result;
5167 int nr_result;
5168#endif
5169
5170 rettv->v_type = type;
5171 if (type != VAR_NUMBER)
5172 rettv->vval.v_string = NULL;
5173
5174 if (check_restricted() || check_secure())
5175 return;
5176
5177#ifdef FEAT_LIBCALL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005178 // The first two args must be strings, otherwise it's meaningless
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005179 if (argvars[0].v_type == VAR_STRING && argvars[1].v_type == VAR_STRING)
5180 {
5181 string_in = NULL;
5182 if (argvars[2].v_type == VAR_STRING)
5183 string_in = argvars[2].vval.v_string;
5184 if (type == VAR_NUMBER)
5185 string_result = NULL;
5186 else
5187 string_result = &rettv->vval.v_string;
5188 if (mch_libcall(argvars[0].vval.v_string,
5189 argvars[1].vval.v_string,
5190 string_in,
5191 argvars[2].vval.v_number,
5192 string_result,
5193 &nr_result) == OK
5194 && type == VAR_NUMBER)
5195 rettv->vval.v_number = nr_result;
5196 }
5197#endif
5198}
5199
5200/*
5201 * "libcall()" function
5202 */
5203 static void
5204f_libcall(typval_T *argvars, typval_T *rettv)
5205{
5206 libcall_common(argvars, rettv, VAR_STRING);
5207}
5208
5209/*
5210 * "libcallnr()" function
5211 */
5212 static void
5213f_libcallnr(typval_T *argvars, typval_T *rettv)
5214{
5215 libcall_common(argvars, rettv, VAR_NUMBER);
5216}
5217
5218/*
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005219 * "line(string, [winid])" function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005220 */
5221 static void
5222f_line(typval_T *argvars, typval_T *rettv)
5223{
5224 linenr_T lnum = 0;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005225 pos_T *fp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005226 int fnum;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005227 int id;
5228 tabpage_T *tp;
5229 win_T *wp;
5230 win_T *save_curwin;
5231 tabpage_T *save_curtab;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005232
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005233 if (argvars[1].v_type != VAR_UNKNOWN)
5234 {
5235 // use window specified in the second argument
5236 id = (int)tv_get_number(&argvars[1]);
5237 wp = win_id2wp_tp(id, &tp);
5238 if (wp != NULL && tp != NULL)
5239 {
5240 if (switch_win_noblock(&save_curwin, &save_curtab, wp, tp, TRUE)
5241 == OK)
5242 {
5243 check_cursor();
5244 fp = var2fpos(&argvars[0], TRUE, &fnum);
5245 }
5246 restore_win_noblock(save_curwin, save_curtab, TRUE);
5247 }
5248 }
5249 else
5250 // use current window
5251 fp = var2fpos(&argvars[0], TRUE, &fnum);
5252
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005253 if (fp != NULL)
5254 lnum = fp->lnum;
5255 rettv->vval.v_number = lnum;
5256}
5257
5258/*
5259 * "line2byte(lnum)" function
5260 */
5261 static void
5262f_line2byte(typval_T *argvars UNUSED, typval_T *rettv)
5263{
5264#ifndef FEAT_BYTEOFF
5265 rettv->vval.v_number = -1;
5266#else
5267 linenr_T lnum;
5268
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005269 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005270 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count + 1)
5271 rettv->vval.v_number = -1;
5272 else
5273 rettv->vval.v_number = ml_find_line_or_offset(curbuf, lnum, NULL);
5274 if (rettv->vval.v_number >= 0)
5275 ++rettv->vval.v_number;
5276#endif
5277}
5278
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005279#ifdef FEAT_FLOAT
5280/*
5281 * "log()" function
5282 */
5283 static void
5284f_log(typval_T *argvars, typval_T *rettv)
5285{
5286 float_T f = 0.0;
5287
5288 rettv->v_type = VAR_FLOAT;
5289 if (get_float_arg(argvars, &f) == OK)
5290 rettv->vval.v_float = log(f);
5291 else
5292 rettv->vval.v_float = 0.0;
5293}
5294
5295/*
5296 * "log10()" function
5297 */
5298 static void
5299f_log10(typval_T *argvars, typval_T *rettv)
5300{
5301 float_T f = 0.0;
5302
5303 rettv->v_type = VAR_FLOAT;
5304 if (get_float_arg(argvars, &f) == OK)
5305 rettv->vval.v_float = log10(f);
5306 else
5307 rettv->vval.v_float = 0.0;
5308}
5309#endif
5310
5311#ifdef FEAT_LUA
5312/*
5313 * "luaeval()" function
5314 */
5315 static void
5316f_luaeval(typval_T *argvars, typval_T *rettv)
5317{
5318 char_u *str;
5319 char_u buf[NUMBUFLEN];
5320
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005321 if (check_restricted() || check_secure())
5322 return;
5323
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005324 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005325 do_luaeval(str, argvars + 1, rettv);
5326}
5327#endif
5328
5329/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005330 * "maparg()" function
5331 */
5332 static void
5333f_maparg(typval_T *argvars, typval_T *rettv)
5334{
5335 get_maparg(argvars, rettv, TRUE);
5336}
5337
5338/*
5339 * "mapcheck()" function
5340 */
5341 static void
5342f_mapcheck(typval_T *argvars, typval_T *rettv)
5343{
5344 get_maparg(argvars, rettv, FALSE);
5345}
5346
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005347typedef enum
5348{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005349 MATCH_END, // matchend()
5350 MATCH_MATCH, // match()
5351 MATCH_STR, // matchstr()
5352 MATCH_LIST, // matchlist()
5353 MATCH_POS // matchstrpos()
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005354} matchtype_T;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005355
5356 static void
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005357find_some_match(typval_T *argvars, typval_T *rettv, matchtype_T type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005358{
5359 char_u *str = NULL;
5360 long len = 0;
5361 char_u *expr = NULL;
5362 char_u *pat;
5363 regmatch_T regmatch;
5364 char_u patbuf[NUMBUFLEN];
5365 char_u strbuf[NUMBUFLEN];
5366 char_u *save_cpo;
5367 long start = 0;
5368 long nth = 1;
5369 colnr_T startcol = 0;
5370 int match = 0;
5371 list_T *l = NULL;
5372 listitem_T *li = NULL;
5373 long idx = 0;
5374 char_u *tofree = NULL;
5375
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005376 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005377 save_cpo = p_cpo;
5378 p_cpo = (char_u *)"";
5379
5380 rettv->vval.v_number = -1;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005381 if (type == MATCH_LIST || type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005382 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005383 // type MATCH_LIST: return empty list when there are no matches.
5384 // type MATCH_POS: return ["", -1, -1, -1]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005385 if (rettv_list_alloc(rettv) == FAIL)
5386 goto theend;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005387 if (type == MATCH_POS
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005388 && (list_append_string(rettv->vval.v_list,
5389 (char_u *)"", 0) == FAIL
5390 || list_append_number(rettv->vval.v_list,
5391 (varnumber_T)-1) == FAIL
5392 || list_append_number(rettv->vval.v_list,
5393 (varnumber_T)-1) == FAIL
5394 || list_append_number(rettv->vval.v_list,
5395 (varnumber_T)-1) == FAIL))
5396 {
5397 list_free(rettv->vval.v_list);
5398 rettv->vval.v_list = NULL;
5399 goto theend;
5400 }
5401 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005402 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005403 {
5404 rettv->v_type = VAR_STRING;
5405 rettv->vval.v_string = NULL;
5406 }
5407
5408 if (argvars[0].v_type == VAR_LIST)
5409 {
5410 if ((l = argvars[0].vval.v_list) == NULL)
5411 goto theend;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02005412 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005413 li = l->lv_first;
5414 }
5415 else
5416 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005417 expr = str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005418 len = (long)STRLEN(str);
5419 }
5420
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005421 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005422 if (pat == NULL)
5423 goto theend;
5424
5425 if (argvars[2].v_type != VAR_UNKNOWN)
5426 {
5427 int error = FALSE;
5428
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005429 start = (long)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005430 if (error)
5431 goto theend;
5432 if (l != NULL)
5433 {
5434 li = list_find(l, start);
5435 if (li == NULL)
5436 goto theend;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01005437 idx = l->lv_u.mat.lv_idx; // use the cached index
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005438 }
5439 else
5440 {
5441 if (start < 0)
5442 start = 0;
5443 if (start > len)
5444 goto theend;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005445 // When "count" argument is there ignore matches before "start",
5446 // otherwise skip part of the string. Differs when pattern is "^"
5447 // or "\<".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005448 if (argvars[3].v_type != VAR_UNKNOWN)
5449 startcol = start;
5450 else
5451 {
5452 str += start;
5453 len -= start;
5454 }
5455 }
5456
5457 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005458 nth = (long)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005459 if (error)
5460 goto theend;
5461 }
5462
5463 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
5464 if (regmatch.regprog != NULL)
5465 {
5466 regmatch.rm_ic = p_ic;
5467
5468 for (;;)
5469 {
5470 if (l != NULL)
5471 {
5472 if (li == NULL)
5473 {
5474 match = FALSE;
5475 break;
5476 }
5477 vim_free(tofree);
5478 expr = str = echo_string(&li->li_tv, &tofree, strbuf, 0);
5479 if (str == NULL)
5480 break;
5481 }
5482
5483 match = vim_regexec_nl(&regmatch, str, (colnr_T)startcol);
5484
5485 if (match && --nth <= 0)
5486 break;
5487 if (l == NULL && !match)
5488 break;
5489
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005490 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005491 if (l != NULL)
5492 {
5493 li = li->li_next;
5494 ++idx;
5495 }
5496 else
5497 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005498 startcol = (colnr_T)(regmatch.startp[0]
5499 + (*mb_ptr2len)(regmatch.startp[0]) - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005500 if (startcol > (colnr_T)len
5501 || str + startcol <= regmatch.startp[0])
5502 {
5503 match = FALSE;
5504 break;
5505 }
5506 }
5507 }
5508
5509 if (match)
5510 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005511 if (type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005512 {
5513 listitem_T *li1 = rettv->vval.v_list->lv_first;
5514 listitem_T *li2 = li1->li_next;
5515 listitem_T *li3 = li2->li_next;
5516 listitem_T *li4 = li3->li_next;
5517
5518 vim_free(li1->li_tv.vval.v_string);
5519 li1->li_tv.vval.v_string = vim_strnsave(regmatch.startp[0],
5520 (int)(regmatch.endp[0] - regmatch.startp[0]));
5521 li3->li_tv.vval.v_number =
5522 (varnumber_T)(regmatch.startp[0] - expr);
5523 li4->li_tv.vval.v_number =
5524 (varnumber_T)(regmatch.endp[0] - expr);
5525 if (l != NULL)
5526 li2->li_tv.vval.v_number = (varnumber_T)idx;
5527 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005528 else if (type == MATCH_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005529 {
5530 int i;
5531
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005532 // return list with matched string and submatches
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005533 for (i = 0; i < NSUBEXP; ++i)
5534 {
5535 if (regmatch.endp[i] == NULL)
5536 {
5537 if (list_append_string(rettv->vval.v_list,
5538 (char_u *)"", 0) == FAIL)
5539 break;
5540 }
5541 else if (list_append_string(rettv->vval.v_list,
5542 regmatch.startp[i],
5543 (int)(regmatch.endp[i] - regmatch.startp[i]))
5544 == FAIL)
5545 break;
5546 }
5547 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005548 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005549 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005550 // return matched string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005551 if (l != NULL)
5552 copy_tv(&li->li_tv, rettv);
5553 else
5554 rettv->vval.v_string = vim_strnsave(regmatch.startp[0],
5555 (int)(regmatch.endp[0] - regmatch.startp[0]));
5556 }
5557 else if (l != NULL)
5558 rettv->vval.v_number = idx;
5559 else
5560 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005561 if (type != MATCH_END)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005562 rettv->vval.v_number =
5563 (varnumber_T)(regmatch.startp[0] - str);
5564 else
5565 rettv->vval.v_number =
5566 (varnumber_T)(regmatch.endp[0] - str);
5567 rettv->vval.v_number += (varnumber_T)(str - expr);
5568 }
5569 }
5570 vim_regfree(regmatch.regprog);
5571 }
5572
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005573theend:
5574 if (type == MATCH_POS && l == NULL && rettv->vval.v_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005575 // matchstrpos() without a list: drop the second item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005576 listitem_remove(rettv->vval.v_list,
5577 rettv->vval.v_list->lv_first->li_next);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005578 vim_free(tofree);
5579 p_cpo = save_cpo;
5580}
5581
5582/*
5583 * "match()" function
5584 */
5585 static void
5586f_match(typval_T *argvars, typval_T *rettv)
5587{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005588 find_some_match(argvars, rettv, MATCH_MATCH);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005589}
5590
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005591/*
5592 * "matchend()" function
5593 */
5594 static void
5595f_matchend(typval_T *argvars, typval_T *rettv)
5596{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005597 find_some_match(argvars, rettv, MATCH_END);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005598}
5599
5600/*
5601 * "matchlist()" function
5602 */
5603 static void
5604f_matchlist(typval_T *argvars, typval_T *rettv)
5605{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005606 find_some_match(argvars, rettv, MATCH_LIST);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005607}
5608
5609/*
5610 * "matchstr()" function
5611 */
5612 static void
5613f_matchstr(typval_T *argvars, typval_T *rettv)
5614{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005615 find_some_match(argvars, rettv, MATCH_STR);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005616}
5617
5618/*
5619 * "matchstrpos()" function
5620 */
5621 static void
5622f_matchstrpos(typval_T *argvars, typval_T *rettv)
5623{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005624 find_some_match(argvars, rettv, MATCH_POS);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005625}
5626
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005627 static void
5628max_min(typval_T *argvars, typval_T *rettv, int domax)
5629{
5630 varnumber_T n = 0;
5631 varnumber_T i;
5632 int error = FALSE;
5633
5634 if (argvars[0].v_type == VAR_LIST)
5635 {
5636 list_T *l;
5637 listitem_T *li;
5638
5639 l = argvars[0].vval.v_list;
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005640 if (l != NULL && l->lv_len > 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005641 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005642 if (l->lv_first == &range_list_item)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005643 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005644 if ((l->lv_u.nonmat.lv_stride > 0) ^ domax)
5645 n = l->lv_u.nonmat.lv_start;
5646 else
5647 n = l->lv_u.nonmat.lv_start + (l->lv_len - 1)
5648 * l->lv_u.nonmat.lv_stride;
5649 }
5650 else
5651 {
5652 li = l->lv_first;
5653 if (li != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005654 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005655 n = tv_get_number_chk(&li->li_tv, &error);
5656 for (;;)
5657 {
5658 li = li->li_next;
5659 if (li == NULL)
5660 break;
5661 i = tv_get_number_chk(&li->li_tv, &error);
5662 if (domax ? i > n : i < n)
5663 n = i;
5664 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005665 }
5666 }
5667 }
5668 }
5669 else if (argvars[0].v_type == VAR_DICT)
5670 {
5671 dict_T *d;
5672 int first = TRUE;
5673 hashitem_T *hi;
5674 int todo;
5675
5676 d = argvars[0].vval.v_dict;
5677 if (d != NULL)
5678 {
5679 todo = (int)d->dv_hashtab.ht_used;
5680 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
5681 {
5682 if (!HASHITEM_EMPTY(hi))
5683 {
5684 --todo;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005685 i = tv_get_number_chk(&HI2DI(hi)->di_tv, &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005686 if (first)
5687 {
5688 n = i;
5689 first = FALSE;
5690 }
5691 else if (domax ? i > n : i < n)
5692 n = i;
5693 }
5694 }
5695 }
5696 }
5697 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005698 semsg(_(e_listdictarg), domax ? "max()" : "min()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005699 rettv->vval.v_number = error ? 0 : n;
5700}
5701
5702/*
5703 * "max()" function
5704 */
5705 static void
5706f_max(typval_T *argvars, typval_T *rettv)
5707{
5708 max_min(argvars, rettv, TRUE);
5709}
5710
5711/*
5712 * "min()" function
5713 */
5714 static void
5715f_min(typval_T *argvars, typval_T *rettv)
5716{
5717 max_min(argvars, rettv, FALSE);
5718}
5719
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005720#if defined(FEAT_MZSCHEME) || defined(PROTO)
5721/*
5722 * "mzeval()" function
5723 */
5724 static void
5725f_mzeval(typval_T *argvars, typval_T *rettv)
5726{
5727 char_u *str;
5728 char_u buf[NUMBUFLEN];
5729
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005730 if (check_restricted() || check_secure())
5731 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005732 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005733 do_mzeval(str, rettv);
5734}
5735
5736 void
5737mzscheme_call_vim(char_u *name, typval_T *args, typval_T *rettv)
5738{
5739 typval_T argvars[3];
5740
5741 argvars[0].v_type = VAR_STRING;
5742 argvars[0].vval.v_string = name;
5743 copy_tv(args, &argvars[1]);
5744 argvars[2].v_type = VAR_UNKNOWN;
5745 f_call(argvars, rettv);
5746 clear_tv(&argvars[1]);
5747}
5748#endif
5749
5750/*
5751 * "nextnonblank()" function
5752 */
5753 static void
5754f_nextnonblank(typval_T *argvars, typval_T *rettv)
5755{
5756 linenr_T lnum;
5757
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005758 for (lnum = tv_get_lnum(argvars); ; ++lnum)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005759 {
5760 if (lnum < 0 || lnum > curbuf->b_ml.ml_line_count)
5761 {
5762 lnum = 0;
5763 break;
5764 }
5765 if (*skipwhite(ml_get(lnum)) != NUL)
5766 break;
5767 }
5768 rettv->vval.v_number = lnum;
5769}
5770
5771/*
5772 * "nr2char()" function
5773 */
5774 static void
5775f_nr2char(typval_T *argvars, typval_T *rettv)
5776{
5777 char_u buf[NUMBUFLEN];
5778
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005779 if (has_mbyte)
5780 {
5781 int utf8 = 0;
5782
5783 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005784 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005785 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01005786 buf[utf_char2bytes((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005787 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005788 buf[(*mb_char2bytes)((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005789 }
5790 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005791 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005792 buf[0] = (char_u)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005793 buf[1] = NUL;
5794 }
5795 rettv->v_type = VAR_STRING;
5796 rettv->vval.v_string = vim_strsave(buf);
5797}
5798
5799/*
5800 * "or(expr, expr)" function
5801 */
5802 static void
5803f_or(typval_T *argvars, typval_T *rettv)
5804{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005805 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
5806 | tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005807}
5808
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005809#ifdef FEAT_PERL
5810/*
5811 * "perleval()" function
5812 */
5813 static void
5814f_perleval(typval_T *argvars, typval_T *rettv)
5815{
5816 char_u *str;
5817 char_u buf[NUMBUFLEN];
5818
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005819 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005820 do_perleval(str, rettv);
5821}
5822#endif
5823
5824#ifdef FEAT_FLOAT
5825/*
5826 * "pow()" function
5827 */
5828 static void
5829f_pow(typval_T *argvars, typval_T *rettv)
5830{
5831 float_T fx = 0.0, fy = 0.0;
5832
5833 rettv->v_type = VAR_FLOAT;
5834 if (get_float_arg(argvars, &fx) == OK
5835 && get_float_arg(&argvars[1], &fy) == OK)
5836 rettv->vval.v_float = pow(fx, fy);
5837 else
5838 rettv->vval.v_float = 0.0;
5839}
5840#endif
5841
5842/*
5843 * "prevnonblank()" function
5844 */
5845 static void
5846f_prevnonblank(typval_T *argvars, typval_T *rettv)
5847{
5848 linenr_T lnum;
5849
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005850 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005851 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count)
5852 lnum = 0;
5853 else
5854 while (lnum >= 1 && *skipwhite(ml_get(lnum)) == NUL)
5855 --lnum;
5856 rettv->vval.v_number = lnum;
5857}
5858
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005859// This dummy va_list is here because:
5860// - passing a NULL pointer doesn't work when va_list isn't a pointer
5861// - locally in the function results in a "used before set" warning
5862// - using va_start() to initialize it gives "function with fixed args" error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005863static va_list ap;
5864
5865/*
5866 * "printf()" function
5867 */
5868 static void
5869f_printf(typval_T *argvars, typval_T *rettv)
5870{
5871 char_u buf[NUMBUFLEN];
5872 int len;
5873 char_u *s;
5874 int saved_did_emsg = did_emsg;
5875 char *fmt;
5876
5877 rettv->v_type = VAR_STRING;
5878 rettv->vval.v_string = NULL;
5879
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005880 // Get the required length, allocate the buffer and do it for real.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005881 did_emsg = FALSE;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005882 fmt = (char *)tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02005883 len = vim_vsnprintf_typval(NULL, 0, fmt, ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005884 if (!did_emsg)
5885 {
5886 s = alloc(len + 1);
5887 if (s != NULL)
5888 {
5889 rettv->vval.v_string = s;
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02005890 (void)vim_vsnprintf_typval((char *)s, len + 1, fmt,
5891 ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005892 }
5893 }
5894 did_emsg |= saved_did_emsg;
5895}
5896
5897/*
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005898 * "pum_getpos()" function
5899 */
5900 static void
5901f_pum_getpos(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5902{
5903 if (rettv_dict_alloc(rettv) != OK)
5904 return;
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005905 pum_set_event_info(rettv->vval.v_dict);
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005906}
5907
5908/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005909 * "pumvisible()" function
5910 */
5911 static void
5912f_pumvisible(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5913{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005914 if (pum_visible())
5915 rettv->vval.v_number = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005916}
5917
5918#ifdef FEAT_PYTHON3
5919/*
5920 * "py3eval()" function
5921 */
5922 static void
5923f_py3eval(typval_T *argvars, typval_T *rettv)
5924{
5925 char_u *str;
5926 char_u buf[NUMBUFLEN];
5927
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005928 if (check_restricted() || check_secure())
5929 return;
5930
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005931 if (p_pyx == 0)
5932 p_pyx = 3;
5933
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005934 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005935 do_py3eval(str, rettv);
5936}
5937#endif
5938
5939#ifdef FEAT_PYTHON
5940/*
5941 * "pyeval()" function
5942 */
5943 static void
5944f_pyeval(typval_T *argvars, typval_T *rettv)
5945{
5946 char_u *str;
5947 char_u buf[NUMBUFLEN];
5948
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005949 if (check_restricted() || check_secure())
5950 return;
5951
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005952 if (p_pyx == 0)
5953 p_pyx = 2;
5954
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005955 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005956 do_pyeval(str, rettv);
5957}
5958#endif
5959
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005960#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
5961/*
5962 * "pyxeval()" function
5963 */
5964 static void
5965f_pyxeval(typval_T *argvars, typval_T *rettv)
5966{
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005967 if (check_restricted() || check_secure())
5968 return;
5969
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005970# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
5971 init_pyxversion();
5972 if (p_pyx == 2)
5973 f_pyeval(argvars, rettv);
5974 else
5975 f_py3eval(argvars, rettv);
5976# elif defined(FEAT_PYTHON)
5977 f_pyeval(argvars, rettv);
5978# elif defined(FEAT_PYTHON3)
5979 f_py3eval(argvars, rettv);
5980# endif
5981}
5982#endif
5983
Bram Moolenaar4f645c52020-02-08 16:40:39 +01005984static UINT32_T srand_seed_for_testing = 0;
5985static int srand_seed_for_testing_is_used = FALSE;
5986
5987 static void
5988f_test_srand_seed(typval_T *argvars, typval_T *rettv UNUSED)
5989{
5990 if (argvars[0].v_type == VAR_UNKNOWN)
5991 srand_seed_for_testing_is_used = FALSE;
5992 else
5993 {
5994 srand_seed_for_testing = (UINT32_T)tv_get_number(&argvars[0]);
5995 srand_seed_for_testing_is_used = TRUE;
5996 }
5997}
5998
5999 static void
6000init_srand(UINT32_T *x)
6001{
6002#ifndef MSWIN
6003 static int dev_urandom_state = NOTDONE; // FAIL or OK once tried
6004#endif
6005
6006 if (srand_seed_for_testing_is_used)
6007 {
6008 *x = srand_seed_for_testing;
6009 return;
6010 }
6011#ifndef MSWIN
6012 if (dev_urandom_state != FAIL)
6013 {
6014 int fd = open("/dev/urandom", O_RDONLY);
6015 struct {
6016 union {
6017 UINT32_T number;
6018 char bytes[sizeof(UINT32_T)];
6019 } contents;
6020 } buf;
6021
6022 // Attempt reading /dev/urandom.
6023 if (fd == -1)
6024 dev_urandom_state = FAIL;
6025 else
6026 {
6027 buf.contents.number = 0;
6028 if (read(fd, buf.contents.bytes, sizeof(UINT32_T))
6029 != sizeof(UINT32_T))
6030 dev_urandom_state = FAIL;
6031 else
6032 {
6033 dev_urandom_state = OK;
6034 *x = buf.contents.number;
6035 }
6036 close(fd);
6037 }
6038 }
6039 if (dev_urandom_state != OK)
6040 // Reading /dev/urandom doesn't work, fall back to time().
6041#endif
6042 *x = vim_time();
6043}
6044
6045#define ROTL(x, k) ((x << k) | (x >> (32 - k)))
6046#define SPLITMIX32(x, z) ( \
6047 z = (x += 0x9e3779b9), \
6048 z = (z ^ (z >> 16)) * 0x85ebca6b, \
6049 z = (z ^ (z >> 13)) * 0xc2b2ae35, \
6050 z ^ (z >> 16) \
6051 )
6052#define SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w) \
6053 result = ROTL(y * 5, 7) * 9; \
6054 t = y << 9; \
6055 z ^= x; \
6056 w ^= y; \
6057 y ^= z, x ^= w; \
6058 z ^= t; \
6059 w = ROTL(w, 11);
6060
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006061/*
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006062 * "rand()" function
6063 */
6064 static void
6065f_rand(typval_T *argvars, typval_T *rettv)
6066{
6067 list_T *l = NULL;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006068 static UINT32_T gx, gy, gz, gw;
6069 static int initialized = FALSE;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006070 listitem_T *lx, *ly, *lz, *lw;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006071 UINT32_T x, y, z, w, t, result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006072
6073 if (argvars[0].v_type == VAR_UNKNOWN)
6074 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006075 // When no argument is given use the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006076 if (initialized == FALSE)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006077 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006078 // Initialize the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006079 init_srand(&x);
6080
6081 gx = SPLITMIX32(x, z);
6082 gy = SPLITMIX32(x, z);
6083 gz = SPLITMIX32(x, z);
6084 gw = SPLITMIX32(x, z);
6085 initialized = TRUE;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006086 }
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006087
6088 SHUFFLE_XOSHIRO128STARSTAR(gx, gy, gz, gw);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006089 }
6090 else if (argvars[0].v_type == VAR_LIST)
6091 {
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006092 l = argvars[0].vval.v_list;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006093 if (l == NULL || list_len(l) != 4)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006094 goto theend;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006095
6096 lx = list_find(l, 0L);
6097 ly = list_find(l, 1L);
6098 lz = list_find(l, 2L);
6099 lw = list_find(l, 3L);
6100 if (lx->li_tv.v_type != VAR_NUMBER) goto theend;
6101 if (ly->li_tv.v_type != VAR_NUMBER) goto theend;
6102 if (lz->li_tv.v_type != VAR_NUMBER) goto theend;
6103 if (lw->li_tv.v_type != VAR_NUMBER) goto theend;
6104 x = (UINT32_T)lx->li_tv.vval.v_number;
6105 y = (UINT32_T)ly->li_tv.vval.v_number;
6106 z = (UINT32_T)lz->li_tv.vval.v_number;
6107 w = (UINT32_T)lw->li_tv.vval.v_number;
6108
6109 SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w);
6110
6111 lx->li_tv.vval.v_number = (varnumber_T)x;
6112 ly->li_tv.vval.v_number = (varnumber_T)y;
6113 lz->li_tv.vval.v_number = (varnumber_T)z;
6114 lw->li_tv.vval.v_number = (varnumber_T)w;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006115 }
6116 else
6117 goto theend;
6118
6119 rettv->v_type = VAR_NUMBER;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006120 rettv->vval.v_number = (varnumber_T)result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006121 return;
6122
6123theend:
6124 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006125 rettv->v_type = VAR_NUMBER;
6126 rettv->vval.v_number = -1;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006127}
6128
6129/*
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006130 * "srand()" function
6131 */
6132 static void
6133f_srand(typval_T *argvars, typval_T *rettv)
6134{
6135 UINT32_T x = 0, z;
6136
6137 if (rettv_list_alloc(rettv) == FAIL)
6138 return;
6139 if (argvars[0].v_type == VAR_UNKNOWN)
6140 {
6141 init_srand(&x);
6142 }
6143 else
6144 {
6145 int error = FALSE;
6146
6147 x = (UINT32_T)tv_get_number_chk(&argvars[0], &error);
6148 if (error)
6149 return;
6150 }
6151
6152 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6153 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6154 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6155 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6156}
6157
6158#undef ROTL
6159#undef SPLITMIX32
6160#undef SHUFFLE_XOSHIRO128STARSTAR
6161
6162/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006163 * "range()" function
6164 */
6165 static void
6166f_range(typval_T *argvars, typval_T *rettv)
6167{
6168 varnumber_T start;
6169 varnumber_T end;
6170 varnumber_T stride = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006171 int error = FALSE;
6172
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006173 start = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006174 if (argvars[1].v_type == VAR_UNKNOWN)
6175 {
6176 end = start - 1;
6177 start = 0;
6178 }
6179 else
6180 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006181 end = tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006182 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006183 stride = tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006184 }
6185
6186 if (error)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006187 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006188 if (stride == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006189 emsg(_("E726: Stride is zero"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006190 else if (stride > 0 ? end + 1 < start : end - 1 > start)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006191 emsg(_("E727: Start past end"));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006192 else if (rettv_list_alloc(rettv) == OK)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006193 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006194 list_T *list = rettv->vval.v_list;
6195
6196 // Create a non-materialized list. This is much more efficient and
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006197 // works with ":for". If used otherwise CHECK_LIST_MATERIALIZE() must
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006198 // be called.
6199 list->lv_first = &range_list_item;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006200 list->lv_u.nonmat.lv_start = start;
6201 list->lv_u.nonmat.lv_end = end;
6202 list->lv_u.nonmat.lv_stride = stride;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01006203 list->lv_len = (end - start) / stride + 1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006204 }
6205}
6206
6207/*
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006208 * Materialize "list".
6209 * Do not call directly, use CHECK_LIST_MATERIALIZE()
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006210 */
6211 void
6212range_list_materialize(list_T *list)
6213{
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006214 varnumber_T start = list->lv_u.nonmat.lv_start;
6215 varnumber_T end = list->lv_u.nonmat.lv_end;
6216 int stride = list->lv_u.nonmat.lv_stride;
6217 varnumber_T i;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006218
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006219 list->lv_first = NULL;
6220 list->lv_u.mat.lv_last = NULL;
6221 list->lv_len = 0;
6222 list->lv_u.mat.lv_idx_item = NULL;
6223 for (i = start; stride > 0 ? i <= end : i >= end; i += stride)
6224 if (list_append_number(list, (varnumber_T)i) == FAIL)
6225 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006226}
6227
Bram Moolenaar0b6d9112018-05-22 20:35:17 +02006228 static void
6229return_register(int regname, typval_T *rettv)
6230{
6231 char_u buf[2] = {0, 0};
6232
6233 buf[0] = (char_u)regname;
6234 rettv->v_type = VAR_STRING;
6235 rettv->vval.v_string = vim_strsave(buf);
6236}
6237
6238/*
6239 * "reg_executing()" function
6240 */
6241 static void
6242f_reg_executing(typval_T *argvars UNUSED, typval_T *rettv)
6243{
6244 return_register(reg_executing, rettv);
6245}
6246
6247/*
6248 * "reg_recording()" function
6249 */
6250 static void
6251f_reg_recording(typval_T *argvars UNUSED, typval_T *rettv)
6252{
6253 return_register(reg_recording, rettv);
6254}
6255
Bram Moolenaar7416f3e2017-03-18 18:10:13 +01006256/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006257 * "rename({from}, {to})" function
6258 */
6259 static void
6260f_rename(typval_T *argvars, typval_T *rettv)
6261{
6262 char_u buf[NUMBUFLEN];
6263
6264 if (check_restricted() || check_secure())
6265 rettv->vval.v_number = -1;
6266 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006267 rettv->vval.v_number = vim_rename(tv_get_string(&argvars[0]),
6268 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006269}
6270
6271/*
6272 * "repeat()" function
6273 */
6274 static void
6275f_repeat(typval_T *argvars, typval_T *rettv)
6276{
6277 char_u *p;
6278 int n;
6279 int slen;
6280 int len;
6281 char_u *r;
6282 int i;
6283
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006284 n = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006285 if (argvars[0].v_type == VAR_LIST)
6286 {
6287 if (rettv_list_alloc(rettv) == OK && argvars[0].vval.v_list != NULL)
6288 while (n-- > 0)
6289 if (list_extend(rettv->vval.v_list,
6290 argvars[0].vval.v_list, NULL) == FAIL)
6291 break;
6292 }
6293 else
6294 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006295 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006296 rettv->v_type = VAR_STRING;
6297 rettv->vval.v_string = NULL;
6298
6299 slen = (int)STRLEN(p);
6300 len = slen * n;
6301 if (len <= 0)
6302 return;
6303
6304 r = alloc(len + 1);
6305 if (r != NULL)
6306 {
6307 for (i = 0; i < n; i++)
6308 mch_memmove(r + i * slen, p, (size_t)slen);
6309 r[len] = NUL;
6310 }
6311
6312 rettv->vval.v_string = r;
6313 }
6314}
6315
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006316#define SP_NOMOVE 0x01 // don't move cursor
6317#define SP_REPEAT 0x02 // repeat to find outer pair
6318#define SP_RETCOUNT 0x04 // return matchcount
6319#define SP_SETPCMARK 0x08 // set previous context mark
6320#define SP_START 0x10 // accept match at start position
6321#define SP_SUBPAT 0x20 // return nr of matching sub-pattern
6322#define SP_END 0x40 // leave cursor at end of match
6323#define SP_COLUMN 0x80 // start at cursor column
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006324
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006325/*
6326 * Get flags for a search function.
6327 * Possibly sets "p_ws".
6328 * Returns BACKWARD, FORWARD or zero (for an error).
6329 */
6330 static int
6331get_search_arg(typval_T *varp, int *flagsp)
6332{
6333 int dir = FORWARD;
6334 char_u *flags;
6335 char_u nbuf[NUMBUFLEN];
6336 int mask;
6337
6338 if (varp->v_type != VAR_UNKNOWN)
6339 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006340 flags = tv_get_string_buf_chk(varp, nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006341 if (flags == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006342 return 0; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006343 while (*flags != NUL)
6344 {
6345 switch (*flags)
6346 {
6347 case 'b': dir = BACKWARD; break;
6348 case 'w': p_ws = TRUE; break;
6349 case 'W': p_ws = FALSE; break;
6350 default: mask = 0;
6351 if (flagsp != NULL)
6352 switch (*flags)
6353 {
6354 case 'c': mask = SP_START; break;
6355 case 'e': mask = SP_END; break;
6356 case 'm': mask = SP_RETCOUNT; break;
6357 case 'n': mask = SP_NOMOVE; break;
6358 case 'p': mask = SP_SUBPAT; break;
6359 case 'r': mask = SP_REPEAT; break;
6360 case 's': mask = SP_SETPCMARK; break;
6361 case 'z': mask = SP_COLUMN; break;
6362 }
6363 if (mask == 0)
6364 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006365 semsg(_(e_invarg2), flags);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006366 dir = 0;
6367 }
6368 else
6369 *flagsp |= mask;
6370 }
6371 if (dir == 0)
6372 break;
6373 ++flags;
6374 }
6375 }
6376 return dir;
6377}
6378
6379/*
6380 * Shared by search() and searchpos() functions.
6381 */
6382 static int
6383search_cmn(typval_T *argvars, pos_T *match_pos, int *flagsp)
6384{
6385 int flags;
6386 char_u *pat;
6387 pos_T pos;
6388 pos_T save_cursor;
6389 int save_p_ws = p_ws;
6390 int dir;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006391 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006392 long lnum_stop = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006393#ifdef FEAT_RELTIME
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006394 proftime_T tm;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006395 long time_limit = 0;
6396#endif
6397 int options = SEARCH_KEEP;
6398 int subpatnum;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006399 searchit_arg_T sia;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006400
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006401 pat = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006402 dir = get_search_arg(&argvars[1], flagsp); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006403 if (dir == 0)
6404 goto theend;
6405 flags = *flagsp;
6406 if (flags & SP_START)
6407 options |= SEARCH_START;
6408 if (flags & SP_END)
6409 options |= SEARCH_END;
6410 if (flags & SP_COLUMN)
6411 options |= SEARCH_COL;
6412
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006413 // Optional arguments: line number to stop searching and timeout.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006414 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN)
6415 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006416 lnum_stop = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006417 if (lnum_stop < 0)
6418 goto theend;
6419#ifdef FEAT_RELTIME
6420 if (argvars[3].v_type != VAR_UNKNOWN)
6421 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006422 time_limit = (long)tv_get_number_chk(&argvars[3], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006423 if (time_limit < 0)
6424 goto theend;
6425 }
6426#endif
6427 }
6428
6429#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006430 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006431 profile_setlimit(time_limit, &tm);
6432#endif
6433
6434 /*
6435 * This function does not accept SP_REPEAT and SP_RETCOUNT flags.
6436 * Check to make sure only those flags are set.
6437 * Also, Only the SP_NOMOVE or the SP_SETPCMARK flag can be set. Both
6438 * flags cannot be set. Check for that condition also.
6439 */
6440 if (((flags & (SP_REPEAT | SP_RETCOUNT)) != 0)
6441 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
6442 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006443 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006444 goto theend;
6445 }
6446
6447 pos = save_cursor = curwin->w_cursor;
Bram Moolenaara80faa82020-04-12 19:37:17 +02006448 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006449 sia.sa_stop_lnum = (linenr_T)lnum_stop;
6450#ifdef FEAT_RELTIME
6451 sia.sa_tm = &tm;
6452#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01006453 subpatnum = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006454 options, RE_SEARCH, &sia);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006455 if (subpatnum != FAIL)
6456 {
6457 if (flags & SP_SUBPAT)
6458 retval = subpatnum;
6459 else
6460 retval = pos.lnum;
6461 if (flags & SP_SETPCMARK)
6462 setpcmark();
6463 curwin->w_cursor = pos;
6464 if (match_pos != NULL)
6465 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006466 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006467 match_pos->lnum = pos.lnum;
6468 match_pos->col = pos.col + 1;
6469 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006470 // "/$" will put the cursor after the end of the line, may need to
6471 // correct that here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006472 check_cursor();
6473 }
6474
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006475 // If 'n' flag is used: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006476 if (flags & SP_NOMOVE)
6477 curwin->w_cursor = save_cursor;
6478 else
6479 curwin->w_set_curswant = TRUE;
6480theend:
6481 p_ws = save_p_ws;
6482
6483 return retval;
6484}
6485
6486#ifdef FEAT_FLOAT
6487
6488/*
6489 * round() is not in C90, use ceil() or floor() instead.
6490 */
6491 float_T
6492vim_round(float_T f)
6493{
6494 return f > 0 ? floor(f + 0.5) : ceil(f - 0.5);
6495}
6496
6497/*
6498 * "round({float})" function
6499 */
6500 static void
6501f_round(typval_T *argvars, typval_T *rettv)
6502{
6503 float_T f = 0.0;
6504
6505 rettv->v_type = VAR_FLOAT;
6506 if (get_float_arg(argvars, &f) == OK)
6507 rettv->vval.v_float = vim_round(f);
6508 else
6509 rettv->vval.v_float = 0.0;
6510}
6511#endif
6512
Bram Moolenaare99be0e2019-03-26 22:51:09 +01006513#ifdef FEAT_RUBY
6514/*
6515 * "rubyeval()" function
6516 */
6517 static void
6518f_rubyeval(typval_T *argvars, typval_T *rettv)
6519{
6520 char_u *str;
6521 char_u buf[NUMBUFLEN];
6522
6523 str = tv_get_string_buf(&argvars[0], buf);
6524 do_rubyeval(str, rettv);
6525}
6526#endif
6527
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006528/*
6529 * "screenattr()" function
6530 */
6531 static void
6532f_screenattr(typval_T *argvars, typval_T *rettv)
6533{
6534 int row;
6535 int col;
6536 int c;
6537
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006538 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6539 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006540 if (row < 0 || row >= screen_Rows
6541 || col < 0 || col >= screen_Columns)
6542 c = -1;
6543 else
6544 c = ScreenAttrs[LineOffset[row] + col];
6545 rettv->vval.v_number = c;
6546}
6547
6548/*
6549 * "screenchar()" function
6550 */
6551 static void
6552f_screenchar(typval_T *argvars, typval_T *rettv)
6553{
6554 int row;
6555 int col;
6556 int off;
6557 int c;
6558
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006559 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6560 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006561 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006562 c = -1;
6563 else
6564 {
6565 off = LineOffset[row] + col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006566 if (enc_utf8 && ScreenLinesUC[off] != 0)
6567 c = ScreenLinesUC[off];
6568 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006569 c = ScreenLines[off];
6570 }
6571 rettv->vval.v_number = c;
6572}
6573
6574/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006575 * "screenchars()" function
6576 */
6577 static void
6578f_screenchars(typval_T *argvars, typval_T *rettv)
6579{
6580 int row;
6581 int col;
6582 int off;
6583 int c;
6584 int i;
6585
6586 if (rettv_list_alloc(rettv) == FAIL)
6587 return;
6588 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6589 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
6590 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
6591 return;
6592
6593 off = LineOffset[row] + col;
6594 if (enc_utf8 && ScreenLinesUC[off] != 0)
6595 c = ScreenLinesUC[off];
6596 else
6597 c = ScreenLines[off];
6598 list_append_number(rettv->vval.v_list, (varnumber_T)c);
6599
6600 if (enc_utf8)
6601
6602 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
6603 list_append_number(rettv->vval.v_list,
6604 (varnumber_T)ScreenLinesC[i][off]);
6605}
6606
6607/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006608 * "screencol()" function
6609 *
6610 * First column is 1 to be consistent with virtcol().
6611 */
6612 static void
6613f_screencol(typval_T *argvars UNUSED, typval_T *rettv)
6614{
6615 rettv->vval.v_number = screen_screencol() + 1;
6616}
6617
6618/*
6619 * "screenrow()" function
6620 */
6621 static void
6622f_screenrow(typval_T *argvars UNUSED, typval_T *rettv)
6623{
6624 rettv->vval.v_number = screen_screenrow() + 1;
6625}
6626
6627/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006628 * "screenstring()" function
6629 */
6630 static void
6631f_screenstring(typval_T *argvars, typval_T *rettv)
6632{
6633 int row;
6634 int col;
6635 int off;
6636 int c;
6637 int i;
6638 char_u buf[MB_MAXBYTES + 1];
6639 int buflen = 0;
6640
6641 rettv->vval.v_string = NULL;
6642 rettv->v_type = VAR_STRING;
6643
6644 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6645 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
6646 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
6647 return;
6648
6649 off = LineOffset[row] + col;
6650 if (enc_utf8 && ScreenLinesUC[off] != 0)
6651 c = ScreenLinesUC[off];
6652 else
6653 c = ScreenLines[off];
6654 buflen += mb_char2bytes(c, buf);
6655
6656 if (enc_utf8)
6657 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
6658 buflen += mb_char2bytes(ScreenLinesC[i][off], buf + buflen);
6659
6660 buf[buflen] = NUL;
6661 rettv->vval.v_string = vim_strsave(buf);
6662}
6663
6664/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006665 * "search()" function
6666 */
6667 static void
6668f_search(typval_T *argvars, typval_T *rettv)
6669{
6670 int flags = 0;
6671
6672 rettv->vval.v_number = search_cmn(argvars, NULL, &flags);
6673}
6674
6675/*
6676 * "searchdecl()" function
6677 */
6678 static void
6679f_searchdecl(typval_T *argvars, typval_T *rettv)
6680{
6681 int locally = 1;
6682 int thisblock = 0;
6683 int error = FALSE;
6684 char_u *name;
6685
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006686 rettv->vval.v_number = 1; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006687
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006688 name = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006689 if (argvars[1].v_type != VAR_UNKNOWN)
6690 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006691 locally = (int)tv_get_number_chk(&argvars[1], &error) == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006692 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006693 thisblock = (int)tv_get_number_chk(&argvars[2], &error) != 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006694 }
6695 if (!error && name != NULL)
6696 rettv->vval.v_number = find_decl(name, (int)STRLEN(name),
6697 locally, thisblock, SEARCH_KEEP) == FAIL;
6698}
6699
6700/*
6701 * Used by searchpair() and searchpairpos()
6702 */
6703 static int
6704searchpair_cmn(typval_T *argvars, pos_T *match_pos)
6705{
6706 char_u *spat, *mpat, *epat;
Bram Moolenaar48570482017-10-30 21:48:41 +01006707 typval_T *skip;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006708 int save_p_ws = p_ws;
6709 int dir;
6710 int flags = 0;
6711 char_u nbuf1[NUMBUFLEN];
6712 char_u nbuf2[NUMBUFLEN];
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006713 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006714 long lnum_stop = 0;
6715 long time_limit = 0;
6716
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006717 // Get the three pattern arguments: start, middle, end. Will result in an
6718 // error if not a valid argument.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006719 spat = tv_get_string_chk(&argvars[0]);
6720 mpat = tv_get_string_buf_chk(&argvars[1], nbuf1);
6721 epat = tv_get_string_buf_chk(&argvars[2], nbuf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006722 if (spat == NULL || mpat == NULL || epat == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006723 goto theend; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006724
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006725 // Handle the optional fourth argument: flags
6726 dir = get_search_arg(&argvars[3], &flags); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006727 if (dir == 0)
6728 goto theend;
6729
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006730 // Don't accept SP_END or SP_SUBPAT.
6731 // Only one of the SP_NOMOVE or SP_SETPCMARK flags can be set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006732 if ((flags & (SP_END | SP_SUBPAT)) != 0
6733 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
6734 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006735 semsg(_(e_invarg2), tv_get_string(&argvars[3]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006736 goto theend;
6737 }
6738
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006739 // Using 'r' implies 'W', otherwise it doesn't work.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006740 if (flags & SP_REPEAT)
6741 p_ws = FALSE;
6742
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006743 // Optional fifth argument: skip expression
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006744 if (argvars[3].v_type == VAR_UNKNOWN
6745 || argvars[4].v_type == VAR_UNKNOWN)
Bram Moolenaar48570482017-10-30 21:48:41 +01006746 skip = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006747 else
6748 {
Bram Moolenaar48570482017-10-30 21:48:41 +01006749 skip = &argvars[4];
6750 if (skip->v_type != VAR_FUNC && skip->v_type != VAR_PARTIAL
6751 && skip->v_type != VAR_STRING)
6752 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006753 // Type error
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006754 semsg(_(e_invarg2), tv_get_string(&argvars[4]));
Bram Moolenaar48570482017-10-30 21:48:41 +01006755 goto theend;
6756 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006757 if (argvars[5].v_type != VAR_UNKNOWN)
6758 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006759 lnum_stop = (long)tv_get_number_chk(&argvars[5], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006760 if (lnum_stop < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006761 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006762 semsg(_(e_invarg2), tv_get_string(&argvars[5]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006763 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006764 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006765#ifdef FEAT_RELTIME
6766 if (argvars[6].v_type != VAR_UNKNOWN)
6767 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006768 time_limit = (long)tv_get_number_chk(&argvars[6], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006769 if (time_limit < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006770 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006771 semsg(_(e_invarg2), tv_get_string(&argvars[6]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006772 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006773 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006774 }
6775#endif
6776 }
6777 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006778
6779 retval = do_searchpair(spat, mpat, epat, dir, skip, flags,
6780 match_pos, lnum_stop, time_limit);
6781
6782theend:
6783 p_ws = save_p_ws;
6784
6785 return retval;
6786}
6787
6788/*
6789 * "searchpair()" function
6790 */
6791 static void
6792f_searchpair(typval_T *argvars, typval_T *rettv)
6793{
6794 rettv->vval.v_number = searchpair_cmn(argvars, NULL);
6795}
6796
6797/*
6798 * "searchpairpos()" function
6799 */
6800 static void
6801f_searchpairpos(typval_T *argvars, typval_T *rettv)
6802{
6803 pos_T match_pos;
6804 int lnum = 0;
6805 int col = 0;
6806
6807 if (rettv_list_alloc(rettv) == FAIL)
6808 return;
6809
6810 if (searchpair_cmn(argvars, &match_pos) > 0)
6811 {
6812 lnum = match_pos.lnum;
6813 col = match_pos.col;
6814 }
6815
6816 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
6817 list_append_number(rettv->vval.v_list, (varnumber_T)col);
6818}
6819
6820/*
6821 * Search for a start/middle/end thing.
6822 * Used by searchpair(), see its documentation for the details.
6823 * Returns 0 or -1 for no match,
6824 */
6825 long
6826do_searchpair(
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006827 char_u *spat, // start pattern
6828 char_u *mpat, // middle pattern
6829 char_u *epat, // end pattern
6830 int dir, // BACKWARD or FORWARD
6831 typval_T *skip, // skip expression
6832 int flags, // SP_SETPCMARK and other SP_ values
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006833 pos_T *match_pos,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006834 linenr_T lnum_stop, // stop at this line if not zero
6835 long time_limit UNUSED) // stop after this many msec
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006836{
6837 char_u *save_cpo;
6838 char_u *pat, *pat2 = NULL, *pat3 = NULL;
6839 long retval = 0;
6840 pos_T pos;
6841 pos_T firstpos;
6842 pos_T foundpos;
6843 pos_T save_cursor;
6844 pos_T save_pos;
6845 int n;
6846 int r;
6847 int nest = 1;
Bram Moolenaar48570482017-10-30 21:48:41 +01006848 int use_skip = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006849 int err;
6850 int options = SEARCH_KEEP;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006851#ifdef FEAT_RELTIME
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006852 proftime_T tm;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006853#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006854
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006855 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006856 save_cpo = p_cpo;
6857 p_cpo = empty_option;
6858
6859#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006860 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006861 profile_setlimit(time_limit, &tm);
6862#endif
6863
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006864 // Make two search patterns: start/end (pat2, for in nested pairs) and
6865 // start/middle/end (pat3, for the top pair).
Bram Moolenaar964b3742019-05-24 18:54:09 +02006866 pat2 = alloc(STRLEN(spat) + STRLEN(epat) + 17);
6867 pat3 = alloc(STRLEN(spat) + STRLEN(mpat) + STRLEN(epat) + 25);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006868 if (pat2 == NULL || pat3 == NULL)
6869 goto theend;
Bram Moolenaar6e450a52017-01-06 20:03:58 +01006870 sprintf((char *)pat2, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)", spat, epat);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006871 if (*mpat == NUL)
6872 STRCPY(pat3, pat2);
6873 else
Bram Moolenaar6e450a52017-01-06 20:03:58 +01006874 sprintf((char *)pat3, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)\\|\\(%s\\m\\)",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006875 spat, epat, mpat);
6876 if (flags & SP_START)
6877 options |= SEARCH_START;
6878
Bram Moolenaar48570482017-10-30 21:48:41 +01006879 if (skip != NULL)
6880 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006881 // Empty string means to not use the skip expression.
Bram Moolenaar48570482017-10-30 21:48:41 +01006882 if (skip->v_type == VAR_STRING || skip->v_type == VAR_FUNC)
6883 use_skip = skip->vval.v_string != NULL
6884 && *skip->vval.v_string != NUL;
6885 }
6886
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006887 save_cursor = curwin->w_cursor;
6888 pos = curwin->w_cursor;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006889 CLEAR_POS(&firstpos);
6890 CLEAR_POS(&foundpos);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006891 pat = pat3;
6892 for (;;)
6893 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006894 searchit_arg_T sia;
6895
Bram Moolenaara80faa82020-04-12 19:37:17 +02006896 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006897 sia.sa_stop_lnum = lnum_stop;
6898#ifdef FEAT_RELTIME
6899 sia.sa_tm = &tm;
6900#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01006901 n = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006902 options, RE_SEARCH, &sia);
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006903 if (n == FAIL || (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos)))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006904 // didn't find it or found the first match again: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006905 break;
6906
6907 if (firstpos.lnum == 0)
6908 firstpos = pos;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006909 if (EQUAL_POS(pos, foundpos))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006910 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006911 // Found the same position again. Can happen with a pattern that
6912 // has "\zs" at the end and searching backwards. Advance one
6913 // character and try again.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006914 if (dir == BACKWARD)
6915 decl(&pos);
6916 else
6917 incl(&pos);
6918 }
6919 foundpos = pos;
6920
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006921 // clear the start flag to avoid getting stuck here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006922 options &= ~SEARCH_START;
6923
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006924 // If the skip pattern matches, ignore this match.
Bram Moolenaar48570482017-10-30 21:48:41 +01006925 if (use_skip)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006926 {
6927 save_pos = curwin->w_cursor;
6928 curwin->w_cursor = pos;
Bram Moolenaar48570482017-10-30 21:48:41 +01006929 err = FALSE;
6930 r = eval_expr_to_bool(skip, &err);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006931 curwin->w_cursor = save_pos;
6932 if (err)
6933 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006934 // Evaluating {skip} caused an error, break here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006935 curwin->w_cursor = save_cursor;
6936 retval = -1;
6937 break;
6938 }
6939 if (r)
6940 continue;
6941 }
6942
6943 if ((dir == BACKWARD && n == 3) || (dir == FORWARD && n == 2))
6944 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006945 // Found end when searching backwards or start when searching
6946 // forward: nested pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006947 ++nest;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006948 pat = pat2; // nested, don't search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006949 }
6950 else
6951 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006952 // Found end when searching forward or start when searching
6953 // backward: end of (nested) pair; or found middle in outer pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006954 if (--nest == 1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006955 pat = pat3; // outer level, search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006956 }
6957
6958 if (nest == 0)
6959 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006960 // Found the match: return matchcount or line number.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006961 if (flags & SP_RETCOUNT)
6962 ++retval;
6963 else
6964 retval = pos.lnum;
6965 if (flags & SP_SETPCMARK)
6966 setpcmark();
6967 curwin->w_cursor = pos;
6968 if (!(flags & SP_REPEAT))
6969 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006970 nest = 1; // search for next unmatched
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006971 }
6972 }
6973
6974 if (match_pos != NULL)
6975 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006976 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006977 match_pos->lnum = curwin->w_cursor.lnum;
6978 match_pos->col = curwin->w_cursor.col + 1;
6979 }
6980
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006981 // If 'n' flag is used or search failed: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006982 if ((flags & SP_NOMOVE) || retval == 0)
6983 curwin->w_cursor = save_cursor;
6984
6985theend:
6986 vim_free(pat2);
6987 vim_free(pat3);
6988 if (p_cpo == empty_option)
6989 p_cpo = save_cpo;
6990 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006991 // Darn, evaluating the {skip} expression changed the value.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006992 free_string_option(save_cpo);
6993
6994 return retval;
6995}
6996
6997/*
6998 * "searchpos()" function
6999 */
7000 static void
7001f_searchpos(typval_T *argvars, typval_T *rettv)
7002{
7003 pos_T match_pos;
7004 int lnum = 0;
7005 int col = 0;
7006 int n;
7007 int flags = 0;
7008
7009 if (rettv_list_alloc(rettv) == FAIL)
7010 return;
7011
7012 n = search_cmn(argvars, &match_pos, &flags);
7013 if (n > 0)
7014 {
7015 lnum = match_pos.lnum;
7016 col = match_pos.col;
7017 }
7018
7019 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
7020 list_append_number(rettv->vval.v_list, (varnumber_T)col);
7021 if (flags & SP_SUBPAT)
7022 list_append_number(rettv->vval.v_list, (varnumber_T)n);
7023}
7024
7025 static void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007026f_setcharsearch(typval_T *argvars, typval_T *rettv UNUSED)
7027{
7028 dict_T *d;
7029 dictitem_T *di;
7030 char_u *csearch;
7031
7032 if (argvars[0].v_type != VAR_DICT)
7033 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007034 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007035 return;
7036 }
7037
7038 if ((d = argvars[0].vval.v_dict) != NULL)
7039 {
Bram Moolenaar8f667172018-12-14 15:38:31 +01007040 csearch = dict_get_string(d, (char_u *)"char", FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007041 if (csearch != NULL)
7042 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007043 if (enc_utf8)
7044 {
7045 int pcc[MAX_MCO];
7046 int c = utfc_ptr2char(csearch, pcc);
7047
7048 set_last_csearch(c, csearch, utfc_ptr2len(csearch));
7049 }
7050 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007051 set_last_csearch(PTR2CHAR(csearch),
Bram Moolenaar1614a142019-10-06 22:00:13 +02007052 csearch, mb_ptr2len(csearch));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007053 }
7054
7055 di = dict_find(d, (char_u *)"forward", -1);
7056 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007057 set_csearch_direction((int)tv_get_number(&di->di_tv)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007058 ? FORWARD : BACKWARD);
7059
7060 di = dict_find(d, (char_u *)"until", -1);
7061 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007062 set_csearch_until(!!tv_get_number(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007063 }
7064}
7065
7066/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02007067 * "setenv()" function
7068 */
7069 static void
7070f_setenv(typval_T *argvars, typval_T *rettv UNUSED)
7071{
7072 char_u namebuf[NUMBUFLEN];
7073 char_u valbuf[NUMBUFLEN];
7074 char_u *name = tv_get_string_buf(&argvars[0], namebuf);
7075
7076 if (argvars[1].v_type == VAR_SPECIAL
7077 && argvars[1].vval.v_number == VVAL_NULL)
7078 vim_unsetenv(name);
7079 else
7080 vim_setenv(name, tv_get_string_buf(&argvars[1], valbuf));
7081}
7082
7083/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007084 * "setfperm({fname}, {mode})" function
7085 */
7086 static void
7087f_setfperm(typval_T *argvars, typval_T *rettv)
7088{
7089 char_u *fname;
7090 char_u modebuf[NUMBUFLEN];
7091 char_u *mode_str;
7092 int i;
7093 int mask;
7094 int mode = 0;
7095
7096 rettv->vval.v_number = 0;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007097 fname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007098 if (fname == NULL)
7099 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007100 mode_str = tv_get_string_buf_chk(&argvars[1], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007101 if (mode_str == NULL)
7102 return;
7103 if (STRLEN(mode_str) != 9)
7104 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007105 semsg(_(e_invarg2), mode_str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007106 return;
7107 }
7108
7109 mask = 1;
7110 for (i = 8; i >= 0; --i)
7111 {
7112 if (mode_str[i] != '-')
7113 mode |= mask;
7114 mask = mask << 1;
7115 }
7116 rettv->vval.v_number = mch_setperm(fname, mode) == OK;
7117}
7118
7119/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007120 * "setpos()" function
7121 */
7122 static void
7123f_setpos(typval_T *argvars, typval_T *rettv)
7124{
7125 pos_T pos;
7126 int fnum;
7127 char_u *name;
7128 colnr_T curswant = -1;
7129
7130 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007131 name = tv_get_string_chk(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007132 if (name != NULL)
7133 {
7134 if (list2fpos(&argvars[1], &pos, &fnum, &curswant) == OK)
7135 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01007136 if (pos.col != MAXCOL && --pos.col < 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007137 pos.col = 0;
7138 if (name[0] == '.' && name[1] == NUL)
7139 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007140 // set cursor; "fnum" is ignored
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007141 curwin->w_cursor = pos;
7142 if (curswant >= 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007143 {
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007144 curwin->w_curswant = curswant - 1;
7145 curwin->w_set_curswant = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007146 }
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007147 check_cursor();
7148 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007149 }
7150 else if (name[0] == '\'' && name[1] != NUL && name[2] == NUL)
7151 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007152 // set mark
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007153 if (setmark_pos(name[1], &pos, fnum) == OK)
7154 rettv->vval.v_number = 0;
7155 }
7156 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007157 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007158 }
7159 }
7160}
7161
7162/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007163 * "setreg()" function
7164 */
7165 static void
7166f_setreg(typval_T *argvars, typval_T *rettv)
7167{
7168 int regname;
7169 char_u *strregname;
7170 char_u *stropt;
7171 char_u *strval;
7172 int append;
7173 char_u yank_type;
7174 long block_len;
7175
7176 block_len = -1;
7177 yank_type = MAUTO;
7178 append = FALSE;
7179
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007180 strregname = tv_get_string_chk(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007181 rettv->vval.v_number = 1; // FAIL is default
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007182
7183 if (strregname == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007184 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007185 regname = *strregname;
7186 if (regname == 0 || regname == '@')
7187 regname = '"';
7188
7189 if (argvars[2].v_type != VAR_UNKNOWN)
7190 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007191 stropt = tv_get_string_chk(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007192 if (stropt == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007193 return; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007194 for (; *stropt != NUL; ++stropt)
7195 switch (*stropt)
7196 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007197 case 'a': case 'A': // append
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007198 append = TRUE;
7199 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007200 case 'v': case 'c': // character-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007201 yank_type = MCHAR;
7202 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007203 case 'V': case 'l': // line-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007204 yank_type = MLINE;
7205 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007206 case 'b': case Ctrl_V: // block-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007207 yank_type = MBLOCK;
7208 if (VIM_ISDIGIT(stropt[1]))
7209 {
7210 ++stropt;
7211 block_len = getdigits(&stropt) - 1;
7212 --stropt;
7213 }
7214 break;
7215 }
7216 }
7217
7218 if (argvars[1].v_type == VAR_LIST)
7219 {
7220 char_u **lstval;
7221 char_u **allocval;
7222 char_u buf[NUMBUFLEN];
7223 char_u **curval;
7224 char_u **curallocval;
7225 list_T *ll = argvars[1].vval.v_list;
7226 listitem_T *li;
7227 int len;
7228
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007229 // If the list is NULL handle like an empty list.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007230 len = ll == NULL ? 0 : ll->lv_len;
7231
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007232 // First half: use for pointers to result lines; second half: use for
7233 // pointers to allocated copies.
Bram Moolenaarc799fe22019-05-28 23:08:19 +02007234 lstval = ALLOC_MULT(char_u *, (len + 1) * 2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007235 if (lstval == NULL)
7236 return;
7237 curval = lstval;
7238 allocval = lstval + len + 2;
7239 curallocval = allocval;
7240
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007241 if (ll != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007242 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007243 CHECK_LIST_MATERIALIZE(ll);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02007244 FOR_ALL_LIST_ITEMS(ll, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007245 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007246 strval = tv_get_string_buf_chk(&li->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007247 if (strval == NULL)
7248 goto free_lstval;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007249 if (strval == buf)
7250 {
7251 // Need to make a copy, next tv_get_string_buf_chk() will
7252 // overwrite the string.
7253 strval = vim_strsave(buf);
7254 if (strval == NULL)
7255 goto free_lstval;
7256 *curallocval++ = strval;
7257 }
7258 *curval++ = strval;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007259 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007260 }
7261 *curval++ = NULL;
7262
7263 write_reg_contents_lst(regname, lstval, -1,
7264 append, yank_type, block_len);
7265free_lstval:
7266 while (curallocval > allocval)
7267 vim_free(*--curallocval);
7268 vim_free(lstval);
7269 }
7270 else
7271 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007272 strval = tv_get_string_chk(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007273 if (strval == NULL)
7274 return;
7275 write_reg_contents_ex(regname, strval, -1,
7276 append, yank_type, block_len);
7277 }
7278 rettv->vval.v_number = 0;
7279}
7280
7281/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007282 * "settagstack()" function
7283 */
7284 static void
7285f_settagstack(typval_T *argvars, typval_T *rettv)
7286{
7287 static char *e_invact2 = N_("E962: Invalid action: '%s'");
7288 win_T *wp;
7289 dict_T *d;
7290 int action = 'r';
7291
7292 rettv->vval.v_number = -1;
7293
7294 // first argument: window number or id
7295 wp = find_win_by_nr_or_id(&argvars[0]);
7296 if (wp == NULL)
7297 return;
7298
7299 // second argument: dict with items to set in the tag stack
7300 if (argvars[1].v_type != VAR_DICT)
7301 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007302 emsg(_(e_dictreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007303 return;
7304 }
7305 d = argvars[1].vval.v_dict;
7306 if (d == NULL)
7307 return;
7308
7309 // third argument: action - 'a' for append and 'r' for replace.
7310 // default is to replace the stack.
7311 if (argvars[2].v_type == VAR_UNKNOWN)
7312 action = 'r';
7313 else if (argvars[2].v_type == VAR_STRING)
7314 {
7315 char_u *actstr;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007316 actstr = tv_get_string_chk(&argvars[2]);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007317 if (actstr == NULL)
7318 return;
Bram Moolenaar271fa082020-01-02 14:02:16 +01007319 if ((*actstr == 'r' || *actstr == 'a' || *actstr == 't')
7320 && actstr[1] == NUL)
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007321 action = *actstr;
7322 else
7323 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007324 semsg(_(e_invact2), actstr);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007325 return;
7326 }
7327 }
7328 else
7329 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007330 emsg(_(e_stringreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007331 return;
7332 }
7333
7334 if (set_tagstack(wp, d, action) == OK)
7335 rettv->vval.v_number = 0;
7336}
7337
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007338#ifdef FEAT_CRYPT
7339/*
7340 * "sha256({string})" function
7341 */
7342 static void
7343f_sha256(typval_T *argvars, typval_T *rettv)
7344{
7345 char_u *p;
7346
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007347 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007348 rettv->vval.v_string = vim_strsave(
7349 sha256_bytes(p, (int)STRLEN(p), NULL, 0));
7350 rettv->v_type = VAR_STRING;
7351}
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007352#endif // FEAT_CRYPT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007353
7354/*
7355 * "shellescape({string})" function
7356 */
7357 static void
7358f_shellescape(typval_T *argvars, typval_T *rettv)
7359{
Bram Moolenaar20615522017-06-05 18:46:26 +02007360 int do_special = non_zero_arg(&argvars[1]);
7361
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007362 rettv->vval.v_string = vim_strsave_shellescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007363 tv_get_string(&argvars[0]), do_special, do_special);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007364 rettv->v_type = VAR_STRING;
7365}
7366
7367/*
7368 * shiftwidth() function
7369 */
7370 static void
7371f_shiftwidth(typval_T *argvars UNUSED, typval_T *rettv)
7372{
Bram Moolenaarf9514162018-11-22 03:08:29 +01007373 rettv->vval.v_number = 0;
7374
7375 if (argvars[0].v_type != VAR_UNKNOWN)
7376 {
7377 long col;
7378
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007379 col = (long)tv_get_number_chk(argvars, NULL);
Bram Moolenaarf9514162018-11-22 03:08:29 +01007380 if (col < 0)
7381 return; // type error; errmsg already given
7382#ifdef FEAT_VARTABS
7383 rettv->vval.v_number = get_sw_value_col(curbuf, col);
7384 return;
7385#endif
7386 }
7387
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007388 rettv->vval.v_number = get_sw_value(curbuf);
7389}
7390
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007391#ifdef FEAT_FLOAT
7392/*
7393 * "sin()" function
7394 */
7395 static void
7396f_sin(typval_T *argvars, typval_T *rettv)
7397{
7398 float_T f = 0.0;
7399
7400 rettv->v_type = VAR_FLOAT;
7401 if (get_float_arg(argvars, &f) == OK)
7402 rettv->vval.v_float = sin(f);
7403 else
7404 rettv->vval.v_float = 0.0;
7405}
7406
7407/*
7408 * "sinh()" function
7409 */
7410 static void
7411f_sinh(typval_T *argvars, typval_T *rettv)
7412{
7413 float_T f = 0.0;
7414
7415 rettv->v_type = VAR_FLOAT;
7416 if (get_float_arg(argvars, &f) == OK)
7417 rettv->vval.v_float = sinh(f);
7418 else
7419 rettv->vval.v_float = 0.0;
7420}
7421#endif
7422
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007423/*
7424 * "soundfold({word})" function
7425 */
7426 static void
7427f_soundfold(typval_T *argvars, typval_T *rettv)
7428{
7429 char_u *s;
7430
7431 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007432 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007433#ifdef FEAT_SPELL
7434 rettv->vval.v_string = eval_soundfold(s);
7435#else
7436 rettv->vval.v_string = vim_strsave(s);
7437#endif
7438}
7439
7440/*
7441 * "spellbadword()" function
7442 */
7443 static void
7444f_spellbadword(typval_T *argvars UNUSED, typval_T *rettv)
7445{
7446 char_u *word = (char_u *)"";
7447 hlf_T attr = HLF_COUNT;
7448 int len = 0;
7449
7450 if (rettv_list_alloc(rettv) == FAIL)
7451 return;
7452
7453#ifdef FEAT_SPELL
7454 if (argvars[0].v_type == VAR_UNKNOWN)
7455 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007456 // Find the start and length of the badly spelled word.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007457 len = spell_move_to(curwin, FORWARD, TRUE, TRUE, &attr);
7458 if (len != 0)
Bram Moolenaarb73fa622017-12-21 20:27:47 +01007459 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007460 word = ml_get_cursor();
Bram Moolenaarb73fa622017-12-21 20:27:47 +01007461 curwin->w_set_curswant = TRUE;
7462 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007463 }
7464 else if (curwin->w_p_spell && *curbuf->b_s.b_p_spl != NUL)
7465 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007466 char_u *str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007467 int capcol = -1;
7468
7469 if (str != NULL)
7470 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007471 // Check the argument for spelling.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007472 while (*str != NUL)
7473 {
7474 len = spell_check(curwin, str, &attr, &capcol, FALSE);
7475 if (attr != HLF_COUNT)
7476 {
7477 word = str;
7478 break;
7479 }
7480 str += len;
Bram Moolenaar66ab9162018-07-20 20:28:48 +02007481 capcol -= len;
Bram Moolenaar0c779e82019-08-09 17:01:02 +02007482 len = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007483 }
7484 }
7485 }
7486#endif
7487
7488 list_append_string(rettv->vval.v_list, word, len);
7489 list_append_string(rettv->vval.v_list, (char_u *)(
7490 attr == HLF_SPB ? "bad" :
7491 attr == HLF_SPR ? "rare" :
7492 attr == HLF_SPL ? "local" :
7493 attr == HLF_SPC ? "caps" :
7494 ""), -1);
7495}
7496
7497/*
7498 * "spellsuggest()" function
7499 */
7500 static void
7501f_spellsuggest(typval_T *argvars UNUSED, typval_T *rettv)
7502{
7503#ifdef FEAT_SPELL
7504 char_u *str;
7505 int typeerr = FALSE;
7506 int maxcount;
7507 garray_T ga;
7508 int i;
7509 listitem_T *li;
7510 int need_capital = FALSE;
7511#endif
7512
7513 if (rettv_list_alloc(rettv) == FAIL)
7514 return;
7515
7516#ifdef FEAT_SPELL
7517 if (curwin->w_p_spell && *curwin->w_s->b_p_spl != NUL)
7518 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007519 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007520 if (argvars[1].v_type != VAR_UNKNOWN)
7521 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007522 maxcount = (int)tv_get_number_chk(&argvars[1], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007523 if (maxcount <= 0)
7524 return;
7525 if (argvars[2].v_type != VAR_UNKNOWN)
7526 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007527 need_capital = (int)tv_get_number_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007528 if (typeerr)
7529 return;
7530 }
7531 }
7532 else
7533 maxcount = 25;
7534
7535 spell_suggest_list(&ga, str, maxcount, need_capital, FALSE);
7536
7537 for (i = 0; i < ga.ga_len; ++i)
7538 {
7539 str = ((char_u **)ga.ga_data)[i];
7540
7541 li = listitem_alloc();
7542 if (li == NULL)
7543 vim_free(str);
7544 else
7545 {
7546 li->li_tv.v_type = VAR_STRING;
7547 li->li_tv.v_lock = 0;
7548 li->li_tv.vval.v_string = str;
7549 list_append(rettv->vval.v_list, li);
7550 }
7551 }
7552 ga_clear(&ga);
7553 }
7554#endif
7555}
7556
7557 static void
7558f_split(typval_T *argvars, typval_T *rettv)
7559{
7560 char_u *str;
7561 char_u *end;
7562 char_u *pat = NULL;
7563 regmatch_T regmatch;
7564 char_u patbuf[NUMBUFLEN];
7565 char_u *save_cpo;
7566 int match;
7567 colnr_T col = 0;
7568 int keepempty = FALSE;
7569 int typeerr = FALSE;
7570
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007571 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007572 save_cpo = p_cpo;
7573 p_cpo = (char_u *)"";
7574
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007575 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007576 if (argvars[1].v_type != VAR_UNKNOWN)
7577 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007578 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007579 if (pat == NULL)
7580 typeerr = TRUE;
7581 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007582 keepempty = (int)tv_get_number_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007583 }
7584 if (pat == NULL || *pat == NUL)
7585 pat = (char_u *)"[\\x01- ]\\+";
7586
7587 if (rettv_list_alloc(rettv) == FAIL)
7588 return;
7589 if (typeerr)
7590 return;
7591
7592 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
7593 if (regmatch.regprog != NULL)
7594 {
7595 regmatch.rm_ic = FALSE;
7596 while (*str != NUL || keepempty)
7597 {
7598 if (*str == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007599 match = FALSE; // empty item at the end
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007600 else
7601 match = vim_regexec_nl(&regmatch, str, col);
7602 if (match)
7603 end = regmatch.startp[0];
7604 else
7605 end = str + STRLEN(str);
7606 if (keepempty || end > str || (rettv->vval.v_list->lv_len > 0
7607 && *str != NUL && match && end < regmatch.endp[0]))
7608 {
7609 if (list_append_string(rettv->vval.v_list, str,
7610 (int)(end - str)) == FAIL)
7611 break;
7612 }
7613 if (!match)
7614 break;
Bram Moolenaar13505972019-01-24 15:04:48 +01007615 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007616 if (regmatch.endp[0] > str)
7617 col = 0;
7618 else
Bram Moolenaar13505972019-01-24 15:04:48 +01007619 // Don't get stuck at the same match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007620 col = (*mb_ptr2len)(regmatch.endp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007621 str = regmatch.endp[0];
7622 }
7623
7624 vim_regfree(regmatch.regprog);
7625 }
7626
7627 p_cpo = save_cpo;
7628}
7629
7630#ifdef FEAT_FLOAT
7631/*
7632 * "sqrt()" function
7633 */
7634 static void
7635f_sqrt(typval_T *argvars, typval_T *rettv)
7636{
7637 float_T f = 0.0;
7638
7639 rettv->v_type = VAR_FLOAT;
7640 if (get_float_arg(argvars, &f) == OK)
7641 rettv->vval.v_float = sqrt(f);
7642 else
7643 rettv->vval.v_float = 0.0;
7644}
Bram Moolenaar0387cae2019-11-29 21:07:58 +01007645#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007646
Bram Moolenaar0387cae2019-11-29 21:07:58 +01007647#ifdef FEAT_FLOAT
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007648/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007649 * "str2float()" function
7650 */
7651 static void
7652f_str2float(typval_T *argvars, typval_T *rettv)
7653{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007654 char_u *p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01007655 int isneg = (*p == '-');
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007656
Bram Moolenaar08243d22017-01-10 16:12:29 +01007657 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007658 p = skipwhite(p + 1);
7659 (void)string2float(p, &rettv->vval.v_float);
Bram Moolenaar08243d22017-01-10 16:12:29 +01007660 if (isneg)
7661 rettv->vval.v_float *= -1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007662 rettv->v_type = VAR_FLOAT;
7663}
7664#endif
7665
7666/*
Bram Moolenaar9d401282019-04-06 13:18:12 +02007667 * "str2list()" function
7668 */
7669 static void
7670f_str2list(typval_T *argvars, typval_T *rettv)
7671{
7672 char_u *p;
7673 int utf8 = FALSE;
7674
7675 if (rettv_list_alloc(rettv) == FAIL)
7676 return;
7677
7678 if (argvars[1].v_type != VAR_UNKNOWN)
7679 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
7680
7681 p = tv_get_string(&argvars[0]);
7682
7683 if (has_mbyte || utf8)
7684 {
7685 int (*ptr2len)(char_u *);
7686 int (*ptr2char)(char_u *);
7687
7688 if (utf8 || enc_utf8)
7689 {
7690 ptr2len = utf_ptr2len;
7691 ptr2char = utf_ptr2char;
7692 }
7693 else
7694 {
7695 ptr2len = mb_ptr2len;
7696 ptr2char = mb_ptr2char;
7697 }
7698
7699 for ( ; *p != NUL; p += (*ptr2len)(p))
7700 list_append_number(rettv->vval.v_list, (*ptr2char)(p));
7701 }
7702 else
7703 for ( ; *p != NUL; ++p)
7704 list_append_number(rettv->vval.v_list, *p);
7705}
7706
7707/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007708 * "str2nr()" function
7709 */
7710 static void
7711f_str2nr(typval_T *argvars, typval_T *rettv)
7712{
7713 int base = 10;
7714 char_u *p;
7715 varnumber_T n;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007716 int what = 0;
Bram Moolenaar08243d22017-01-10 16:12:29 +01007717 int isneg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007718
7719 if (argvars[1].v_type != VAR_UNKNOWN)
7720 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007721 base = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007722 if (base != 2 && base != 8 && base != 10 && base != 16)
7723 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007724 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007725 return;
7726 }
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007727 if (argvars[2].v_type != VAR_UNKNOWN && tv_get_number(&argvars[2]))
7728 what |= STR2NR_QUOTE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007729 }
7730
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007731 p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01007732 isneg = (*p == '-');
7733 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007734 p = skipwhite(p + 1);
7735 switch (base)
7736 {
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007737 case 2: what |= STR2NR_BIN + STR2NR_FORCE; break;
7738 case 8: what |= STR2NR_OCT + STR2NR_FORCE; break;
7739 case 16: what |= STR2NR_HEX + STR2NR_FORCE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007740 }
Bram Moolenaar16e9b852019-05-19 19:59:35 +02007741 vim_str2nr(p, NULL, NULL, what, &n, NULL, 0, FALSE);
7742 // Text after the number is silently ignored.
Bram Moolenaar08243d22017-01-10 16:12:29 +01007743 if (isneg)
7744 rettv->vval.v_number = -n;
7745 else
7746 rettv->vval.v_number = n;
7747
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007748}
7749
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007750/*
7751 * "strgetchar()" function
7752 */
7753 static void
7754f_strgetchar(typval_T *argvars, typval_T *rettv)
7755{
7756 char_u *str;
7757 int len;
7758 int error = FALSE;
7759 int charidx;
Bram Moolenaar13505972019-01-24 15:04:48 +01007760 int byteidx = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007761
7762 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007763 str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007764 if (str == NULL)
7765 return;
7766 len = (int)STRLEN(str);
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007767 charidx = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007768 if (error)
7769 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007770
Bram Moolenaar13505972019-01-24 15:04:48 +01007771 while (charidx >= 0 && byteidx < len)
7772 {
7773 if (charidx == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007774 {
Bram Moolenaar13505972019-01-24 15:04:48 +01007775 rettv->vval.v_number = mb_ptr2char(str + byteidx);
7776 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007777 }
Bram Moolenaar13505972019-01-24 15:04:48 +01007778 --charidx;
7779 byteidx += MB_CPTR2LEN(str + byteidx);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007780 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007781}
7782
7783/*
7784 * "stridx()" function
7785 */
7786 static void
7787f_stridx(typval_T *argvars, typval_T *rettv)
7788{
7789 char_u buf[NUMBUFLEN];
7790 char_u *needle;
7791 char_u *haystack;
7792 char_u *save_haystack;
7793 char_u *pos;
7794 int start_idx;
7795
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007796 needle = tv_get_string_chk(&argvars[1]);
7797 save_haystack = haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007798 rettv->vval.v_number = -1;
7799 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007800 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007801
7802 if (argvars[2].v_type != VAR_UNKNOWN)
7803 {
7804 int error = FALSE;
7805
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007806 start_idx = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007807 if (error || start_idx >= (int)STRLEN(haystack))
7808 return;
7809 if (start_idx >= 0)
7810 haystack += start_idx;
7811 }
7812
7813 pos = (char_u *)strstr((char *)haystack, (char *)needle);
7814 if (pos != NULL)
7815 rettv->vval.v_number = (varnumber_T)(pos - save_haystack);
7816}
7817
7818/*
7819 * "string()" function
7820 */
Bram Moolenaar461a7fc2018-12-22 13:28:07 +01007821 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007822f_string(typval_T *argvars, typval_T *rettv)
7823{
7824 char_u *tofree;
7825 char_u numbuf[NUMBUFLEN];
7826
7827 rettv->v_type = VAR_STRING;
7828 rettv->vval.v_string = tv2string(&argvars[0], &tofree, numbuf,
7829 get_copyID());
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007830 // Make a copy if we have a value but it's not in allocated memory.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007831 if (rettv->vval.v_string != NULL && tofree == NULL)
7832 rettv->vval.v_string = vim_strsave(rettv->vval.v_string);
7833}
7834
7835/*
7836 * "strlen()" function
7837 */
7838 static void
7839f_strlen(typval_T *argvars, typval_T *rettv)
7840{
7841 rettv->vval.v_number = (varnumber_T)(STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007842 tv_get_string(&argvars[0])));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007843}
7844
7845/*
7846 * "strchars()" function
7847 */
7848 static void
7849f_strchars(typval_T *argvars, typval_T *rettv)
7850{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007851 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007852 int skipcc = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007853 varnumber_T len = 0;
7854 int (*func_mb_ptr2char_adv)(char_u **pp);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007855
7856 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007857 skipcc = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007858 if (skipcc < 0 || skipcc > 1)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007859 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007860 else
7861 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007862 func_mb_ptr2char_adv = skipcc ? mb_ptr2char_adv : mb_cptr2char_adv;
7863 while (*s != NUL)
7864 {
7865 func_mb_ptr2char_adv(&s);
7866 ++len;
7867 }
7868 rettv->vval.v_number = len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007869 }
7870}
7871
7872/*
7873 * "strdisplaywidth()" function
7874 */
7875 static void
7876f_strdisplaywidth(typval_T *argvars, typval_T *rettv)
7877{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007878 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007879 int col = 0;
7880
7881 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007882 col = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007883
7884 rettv->vval.v_number = (varnumber_T)(linetabsize_col(col, s) - col);
7885}
7886
7887/*
7888 * "strwidth()" function
7889 */
7890 static void
7891f_strwidth(typval_T *argvars, typval_T *rettv)
7892{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007893 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007894
Bram Moolenaar13505972019-01-24 15:04:48 +01007895 rettv->vval.v_number = (varnumber_T)(mb_string2cells(s, -1));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007896}
7897
7898/*
7899 * "strcharpart()" function
7900 */
7901 static void
7902f_strcharpart(typval_T *argvars, typval_T *rettv)
7903{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007904 char_u *p;
7905 int nchar;
7906 int nbyte = 0;
7907 int charlen;
7908 int len = 0;
7909 int slen;
7910 int error = FALSE;
7911
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007912 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007913 slen = (int)STRLEN(p);
7914
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007915 nchar = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007916 if (!error)
7917 {
7918 if (nchar > 0)
7919 while (nchar > 0 && nbyte < slen)
7920 {
Bram Moolenaard3c907b2016-08-17 21:32:09 +02007921 nbyte += MB_CPTR2LEN(p + nbyte);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007922 --nchar;
7923 }
7924 else
7925 nbyte = nchar;
7926 if (argvars[2].v_type != VAR_UNKNOWN)
7927 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007928 charlen = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007929 while (charlen > 0 && nbyte + len < slen)
7930 {
7931 int off = nbyte + len;
7932
7933 if (off < 0)
7934 len += 1;
7935 else
Bram Moolenaard3c907b2016-08-17 21:32:09 +02007936 len += MB_CPTR2LEN(p + off);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007937 --charlen;
7938 }
7939 }
7940 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007941 len = slen - nbyte; // default: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007942 }
7943
7944 /*
7945 * Only return the overlap between the specified part and the actual
7946 * string.
7947 */
7948 if (nbyte < 0)
7949 {
7950 len += nbyte;
7951 nbyte = 0;
7952 }
7953 else if (nbyte > slen)
7954 nbyte = slen;
7955 if (len < 0)
7956 len = 0;
7957 else if (nbyte + len > slen)
7958 len = slen - nbyte;
7959
7960 rettv->v_type = VAR_STRING;
7961 rettv->vval.v_string = vim_strnsave(p + nbyte, len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007962}
7963
7964/*
7965 * "strpart()" function
7966 */
7967 static void
7968f_strpart(typval_T *argvars, typval_T *rettv)
7969{
7970 char_u *p;
7971 int n;
7972 int len;
7973 int slen;
7974 int error = FALSE;
7975
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007976 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007977 slen = (int)STRLEN(p);
7978
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007979 n = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007980 if (error)
7981 len = 0;
7982 else if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007983 len = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007984 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007985 len = slen - n; // default len: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007986
7987 /*
7988 * Only return the overlap between the specified part and the actual
7989 * string.
7990 */
7991 if (n < 0)
7992 {
7993 len += n;
7994 n = 0;
7995 }
7996 else if (n > slen)
7997 n = slen;
7998 if (len < 0)
7999 len = 0;
8000 else if (n + len > slen)
8001 len = slen - n;
8002
8003 rettv->v_type = VAR_STRING;
8004 rettv->vval.v_string = vim_strnsave(p + n, len);
8005}
8006
8007/*
8008 * "strridx()" function
8009 */
8010 static void
8011f_strridx(typval_T *argvars, typval_T *rettv)
8012{
8013 char_u buf[NUMBUFLEN];
8014 char_u *needle;
8015 char_u *haystack;
8016 char_u *rest;
8017 char_u *lastmatch = NULL;
8018 int haystack_len, end_idx;
8019
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008020 needle = tv_get_string_chk(&argvars[1]);
8021 haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008022
8023 rettv->vval.v_number = -1;
8024 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008025 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008026
8027 haystack_len = (int)STRLEN(haystack);
8028 if (argvars[2].v_type != VAR_UNKNOWN)
8029 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008030 // Third argument: upper limit for index
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008031 end_idx = (int)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008032 if (end_idx < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008033 return; // can never find a match
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008034 }
8035 else
8036 end_idx = haystack_len;
8037
8038 if (*needle == NUL)
8039 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008040 // Empty string matches past the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008041 lastmatch = haystack + end_idx;
8042 }
8043 else
8044 {
8045 for (rest = haystack; *rest != '\0'; ++rest)
8046 {
8047 rest = (char_u *)strstr((char *)rest, (char *)needle);
8048 if (rest == NULL || rest > haystack + end_idx)
8049 break;
8050 lastmatch = rest;
8051 }
8052 }
8053
8054 if (lastmatch == NULL)
8055 rettv->vval.v_number = -1;
8056 else
8057 rettv->vval.v_number = (varnumber_T)(lastmatch - haystack);
8058}
8059
8060/*
8061 * "strtrans()" function
8062 */
8063 static void
8064f_strtrans(typval_T *argvars, typval_T *rettv)
8065{
8066 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008067 rettv->vval.v_string = transstr(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008068}
8069
8070/*
8071 * "submatch()" function
8072 */
8073 static void
8074f_submatch(typval_T *argvars, typval_T *rettv)
8075{
8076 int error = FALSE;
8077 int no;
8078 int retList = 0;
8079
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008080 no = (int)tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008081 if (error)
8082 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02008083 if (no < 0 || no >= NSUBEXP)
8084 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008085 semsg(_("E935: invalid submatch number: %d"), no);
Bram Moolenaar79518e22017-02-17 16:31:35 +01008086 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02008087 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008088 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008089 retList = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008090 if (error)
8091 return;
8092
8093 if (retList == 0)
8094 {
8095 rettv->v_type = VAR_STRING;
8096 rettv->vval.v_string = reg_submatch(no);
8097 }
8098 else
8099 {
8100 rettv->v_type = VAR_LIST;
8101 rettv->vval.v_list = reg_submatch_list(no);
8102 }
8103}
8104
8105/*
8106 * "substitute()" function
8107 */
8108 static void
8109f_substitute(typval_T *argvars, typval_T *rettv)
8110{
8111 char_u patbuf[NUMBUFLEN];
8112 char_u subbuf[NUMBUFLEN];
8113 char_u flagsbuf[NUMBUFLEN];
8114
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008115 char_u *str = tv_get_string_chk(&argvars[0]);
8116 char_u *pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008117 char_u *sub = NULL;
8118 typval_T *expr = NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008119 char_u *flg = tv_get_string_buf_chk(&argvars[3], flagsbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008120
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008121 if (argvars[2].v_type == VAR_FUNC || argvars[2].v_type == VAR_PARTIAL)
8122 expr = &argvars[2];
8123 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008124 sub = tv_get_string_buf_chk(&argvars[2], subbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008125
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008126 rettv->v_type = VAR_STRING;
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008127 if (str == NULL || pat == NULL || (sub == NULL && expr == NULL)
8128 || flg == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008129 rettv->vval.v_string = NULL;
8130 else
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008131 rettv->vval.v_string = do_string_sub(str, pat, sub, expr, flg);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008132}
8133
8134/*
Bram Moolenaar00f123a2018-08-21 20:28:54 +02008135 * "swapinfo(swap_filename)" function
8136 */
8137 static void
8138f_swapinfo(typval_T *argvars, typval_T *rettv)
8139{
8140 if (rettv_dict_alloc(rettv) == OK)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008141 get_b0_dict(tv_get_string(argvars), rettv->vval.v_dict);
Bram Moolenaar00f123a2018-08-21 20:28:54 +02008142}
8143
8144/*
Bram Moolenaar110bd602018-09-16 18:46:59 +02008145 * "swapname(expr)" function
8146 */
8147 static void
8148f_swapname(typval_T *argvars, typval_T *rettv)
8149{
8150 buf_T *buf;
8151
8152 rettv->v_type = VAR_STRING;
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01008153 buf = tv_get_buf(&argvars[0], FALSE);
Bram Moolenaar110bd602018-09-16 18:46:59 +02008154 if (buf == NULL || buf->b_ml.ml_mfp == NULL
8155 || buf->b_ml.ml_mfp->mf_fname == NULL)
8156 rettv->vval.v_string = NULL;
8157 else
8158 rettv->vval.v_string = vim_strsave(buf->b_ml.ml_mfp->mf_fname);
8159}
8160
8161/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008162 * "synID(lnum, col, trans)" function
8163 */
8164 static void
8165f_synID(typval_T *argvars UNUSED, typval_T *rettv)
8166{
8167 int id = 0;
8168#ifdef FEAT_SYN_HL
8169 linenr_T lnum;
8170 colnr_T col;
8171 int trans;
8172 int transerr = FALSE;
8173
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008174 lnum = tv_get_lnum(argvars); // -1 on type error
8175 col = (linenr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008176 trans = (int)tv_get_number_chk(&argvars[2], &transerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008177
8178 if (!transerr && lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8179 && col >= 0 && col < (long)STRLEN(ml_get(lnum)))
8180 id = syn_get_id(curwin, lnum, (colnr_T)col, trans, NULL, FALSE);
8181#endif
8182
8183 rettv->vval.v_number = id;
8184}
8185
8186/*
8187 * "synIDattr(id, what [, mode])" function
8188 */
8189 static void
8190f_synIDattr(typval_T *argvars UNUSED, typval_T *rettv)
8191{
8192 char_u *p = NULL;
8193#ifdef FEAT_SYN_HL
8194 int id;
8195 char_u *what;
8196 char_u *mode;
8197 char_u modebuf[NUMBUFLEN];
8198 int modec;
8199
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008200 id = (int)tv_get_number(&argvars[0]);
8201 what = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008202 if (argvars[2].v_type != VAR_UNKNOWN)
8203 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008204 mode = tv_get_string_buf(&argvars[2], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008205 modec = TOLOWER_ASC(mode[0]);
8206 if (modec != 't' && modec != 'c' && modec != 'g')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008207 modec = 0; // replace invalid with current
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008208 }
8209 else
8210 {
8211#if defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS)
8212 if (USE_24BIT)
8213 modec = 'g';
8214 else
8215#endif
8216 if (t_colors > 1)
8217 modec = 'c';
8218 else
8219 modec = 't';
8220 }
8221
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008222 switch (TOLOWER_ASC(what[0]))
8223 {
8224 case 'b':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008225 if (TOLOWER_ASC(what[1]) == 'g') // bg[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008226 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008227 else // bold
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008228 p = highlight_has_attr(id, HL_BOLD, modec);
8229 break;
8230
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008231 case 'f': // fg[#] or font
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008232 p = highlight_color(id, what, modec);
8233 break;
8234
8235 case 'i':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008236 if (TOLOWER_ASC(what[1]) == 'n') // inverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008237 p = highlight_has_attr(id, HL_INVERSE, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008238 else // italic
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008239 p = highlight_has_attr(id, HL_ITALIC, modec);
8240 break;
8241
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008242 case 'n': // name
Bram Moolenaarc96272e2017-03-26 13:50:09 +02008243 p = get_highlight_name_ext(NULL, id - 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008244 break;
8245
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008246 case 'r': // reverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008247 p = highlight_has_attr(id, HL_INVERSE, modec);
8248 break;
8249
8250 case 's':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008251 if (TOLOWER_ASC(what[1]) == 'p') // sp[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008252 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008253 // strikeout
Bram Moolenaarcf4b00c2017-09-02 18:33:56 +02008254 else if (TOLOWER_ASC(what[1]) == 't' &&
8255 TOLOWER_ASC(what[2]) == 'r')
8256 p = highlight_has_attr(id, HL_STRIKETHROUGH, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008257 else // standout
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008258 p = highlight_has_attr(id, HL_STANDOUT, modec);
8259 break;
8260
8261 case 'u':
8262 if (STRLEN(what) <= 5 || TOLOWER_ASC(what[5]) != 'c')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008263 // underline
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008264 p = highlight_has_attr(id, HL_UNDERLINE, modec);
8265 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008266 // undercurl
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008267 p = highlight_has_attr(id, HL_UNDERCURL, modec);
8268 break;
8269 }
8270
8271 if (p != NULL)
8272 p = vim_strsave(p);
8273#endif
8274 rettv->v_type = VAR_STRING;
8275 rettv->vval.v_string = p;
8276}
8277
8278/*
8279 * "synIDtrans(id)" function
8280 */
8281 static void
8282f_synIDtrans(typval_T *argvars UNUSED, typval_T *rettv)
8283{
8284 int id;
8285
8286#ifdef FEAT_SYN_HL
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008287 id = (int)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008288
8289 if (id > 0)
8290 id = syn_get_final_id(id);
8291 else
8292#endif
8293 id = 0;
8294
8295 rettv->vval.v_number = id;
8296}
8297
8298/*
8299 * "synconcealed(lnum, col)" function
8300 */
8301 static void
8302f_synconcealed(typval_T *argvars UNUSED, typval_T *rettv)
8303{
8304#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
8305 linenr_T lnum;
8306 colnr_T col;
8307 int syntax_flags = 0;
8308 int cchar;
8309 int matchid = 0;
8310 char_u str[NUMBUFLEN];
8311#endif
8312
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02008313 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008314
8315#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008316 lnum = tv_get_lnum(argvars); // -1 on type error
8317 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008318
Bram Moolenaara80faa82020-04-12 19:37:17 +02008319 CLEAR_FIELD(str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008320
8321 if (rettv_list_alloc(rettv) != FAIL)
8322 {
8323 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8324 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
8325 && curwin->w_p_cole > 0)
8326 {
8327 (void)syn_get_id(curwin, lnum, col, FALSE, NULL, FALSE);
8328 syntax_flags = get_syntax_info(&matchid);
8329
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008330 // get the conceal character
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008331 if ((syntax_flags & HL_CONCEAL) && curwin->w_p_cole < 3)
8332 {
8333 cchar = syn_get_sub_char();
Bram Moolenaar4d785892017-06-22 22:00:50 +02008334 if (cchar == NUL && curwin->w_p_cole == 1)
8335 cchar = (lcs_conceal == NUL) ? ' ' : lcs_conceal;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008336 if (cchar != NUL)
8337 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008338 if (has_mbyte)
8339 (*mb_char2bytes)(cchar, str);
8340 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008341 str[0] = cchar;
8342 }
8343 }
8344 }
8345
8346 list_append_number(rettv->vval.v_list,
8347 (syntax_flags & HL_CONCEAL) != 0);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008348 // -1 to auto-determine strlen
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008349 list_append_string(rettv->vval.v_list, str, -1);
8350 list_append_number(rettv->vval.v_list, matchid);
8351 }
8352#endif
8353}
8354
8355/*
8356 * "synstack(lnum, col)" function
8357 */
8358 static void
8359f_synstack(typval_T *argvars UNUSED, typval_T *rettv)
8360{
8361#ifdef FEAT_SYN_HL
8362 linenr_T lnum;
8363 colnr_T col;
8364 int i;
8365 int id;
8366#endif
8367
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02008368 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008369
8370#ifdef FEAT_SYN_HL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008371 lnum = tv_get_lnum(argvars); // -1 on type error
8372 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008373
8374 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8375 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
8376 && rettv_list_alloc(rettv) != FAIL)
8377 {
8378 (void)syn_get_id(curwin, lnum, (colnr_T)col, FALSE, NULL, TRUE);
8379 for (i = 0; ; ++i)
8380 {
8381 id = syn_get_stack_item(i);
8382 if (id < 0)
8383 break;
8384 if (list_append_number(rettv->vval.v_list, id) == FAIL)
8385 break;
8386 }
8387 }
8388#endif
8389}
8390
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008391/*
8392 * "tabpagebuflist()" function
8393 */
8394 static void
8395f_tabpagebuflist(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8396{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008397 tabpage_T *tp;
8398 win_T *wp = NULL;
8399
8400 if (argvars[0].v_type == VAR_UNKNOWN)
8401 wp = firstwin;
8402 else
8403 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008404 tp = find_tabpage((int)tv_get_number(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008405 if (tp != NULL)
8406 wp = (tp == curtab) ? firstwin : tp->tp_firstwin;
8407 }
8408 if (wp != NULL && rettv_list_alloc(rettv) != FAIL)
8409 {
8410 for (; wp != NULL; wp = wp->w_next)
8411 if (list_append_number(rettv->vval.v_list,
8412 wp->w_buffer->b_fnum) == FAIL)
8413 break;
8414 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008415}
8416
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008417/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008418 * "tagfiles()" function
8419 */
8420 static void
8421f_tagfiles(typval_T *argvars UNUSED, typval_T *rettv)
8422{
8423 char_u *fname;
8424 tagname_T tn;
8425 int first;
8426
8427 if (rettv_list_alloc(rettv) == FAIL)
8428 return;
8429 fname = alloc(MAXPATHL);
8430 if (fname == NULL)
8431 return;
8432
8433 for (first = TRUE; ; first = FALSE)
8434 if (get_tagfname(&tn, first, fname) == FAIL
8435 || list_append_string(rettv->vval.v_list, fname, -1) == FAIL)
8436 break;
8437 tagname_free(&tn);
8438 vim_free(fname);
8439}
8440
8441/*
8442 * "taglist()" function
8443 */
8444 static void
8445f_taglist(typval_T *argvars, typval_T *rettv)
8446{
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008447 char_u *fname = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008448 char_u *tag_pattern;
8449
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008450 tag_pattern = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008451
8452 rettv->vval.v_number = FALSE;
8453 if (*tag_pattern == NUL)
8454 return;
8455
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008456 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008457 fname = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008458 if (rettv_list_alloc(rettv) == OK)
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008459 (void)get_tags(rettv->vval.v_list, tag_pattern, fname);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008460}
8461
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008462#ifdef FEAT_FLOAT
8463/*
8464 * "tan()" function
8465 */
8466 static void
8467f_tan(typval_T *argvars, typval_T *rettv)
8468{
8469 float_T f = 0.0;
8470
8471 rettv->v_type = VAR_FLOAT;
8472 if (get_float_arg(argvars, &f) == OK)
8473 rettv->vval.v_float = tan(f);
8474 else
8475 rettv->vval.v_float = 0.0;
8476}
8477
8478/*
8479 * "tanh()" function
8480 */
8481 static void
8482f_tanh(typval_T *argvars, typval_T *rettv)
8483{
8484 float_T f = 0.0;
8485
8486 rettv->v_type = VAR_FLOAT;
8487 if (get_float_arg(argvars, &f) == OK)
8488 rettv->vval.v_float = tanh(f);
8489 else
8490 rettv->vval.v_float = 0.0;
8491}
8492#endif
8493
8494/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008495 * "tolower(string)" function
8496 */
8497 static void
8498f_tolower(typval_T *argvars, typval_T *rettv)
8499{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008500 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008501 rettv->vval.v_string = strlow_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008502}
8503
8504/*
8505 * "toupper(string)" function
8506 */
8507 static void
8508f_toupper(typval_T *argvars, typval_T *rettv)
8509{
8510 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008511 rettv->vval.v_string = strup_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008512}
8513
8514/*
8515 * "tr(string, fromstr, tostr)" function
8516 */
8517 static void
8518f_tr(typval_T *argvars, typval_T *rettv)
8519{
8520 char_u *in_str;
8521 char_u *fromstr;
8522 char_u *tostr;
8523 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008524 int inlen;
8525 int fromlen;
8526 int tolen;
8527 int idx;
8528 char_u *cpstr;
8529 int cplen;
8530 int first = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008531 char_u buf[NUMBUFLEN];
8532 char_u buf2[NUMBUFLEN];
8533 garray_T ga;
8534
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008535 in_str = tv_get_string(&argvars[0]);
8536 fromstr = tv_get_string_buf_chk(&argvars[1], buf);
8537 tostr = tv_get_string_buf_chk(&argvars[2], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008538
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008539 // Default return value: empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008540 rettv->v_type = VAR_STRING;
8541 rettv->vval.v_string = NULL;
8542 if (fromstr == NULL || tostr == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008543 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008544 ga_init2(&ga, (int)sizeof(char), 80);
8545
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008546 if (!has_mbyte)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008547 // not multi-byte: fromstr and tostr must be the same length
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008548 if (STRLEN(fromstr) != STRLEN(tostr))
8549 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008550error:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008551 semsg(_(e_invarg2), fromstr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008552 ga_clear(&ga);
8553 return;
8554 }
8555
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008556 // fromstr and tostr have to contain the same number of chars
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008557 while (*in_str != NUL)
8558 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008559 if (has_mbyte)
8560 {
8561 inlen = (*mb_ptr2len)(in_str);
8562 cpstr = in_str;
8563 cplen = inlen;
8564 idx = 0;
8565 for (p = fromstr; *p != NUL; p += fromlen)
8566 {
8567 fromlen = (*mb_ptr2len)(p);
8568 if (fromlen == inlen && STRNCMP(in_str, p, inlen) == 0)
8569 {
8570 for (p = tostr; *p != NUL; p += tolen)
8571 {
8572 tolen = (*mb_ptr2len)(p);
8573 if (idx-- == 0)
8574 {
8575 cplen = tolen;
8576 cpstr = p;
8577 break;
8578 }
8579 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008580 if (*p == NUL) // tostr is shorter than fromstr
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008581 goto error;
8582 break;
8583 }
8584 ++idx;
8585 }
8586
8587 if (first && cpstr == in_str)
8588 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008589 // Check that fromstr and tostr have the same number of
8590 // (multi-byte) characters. Done only once when a character
8591 // of in_str doesn't appear in fromstr.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008592 first = FALSE;
8593 for (p = tostr; *p != NUL; p += tolen)
8594 {
8595 tolen = (*mb_ptr2len)(p);
8596 --idx;
8597 }
8598 if (idx != 0)
8599 goto error;
8600 }
8601
8602 (void)ga_grow(&ga, cplen);
8603 mch_memmove((char *)ga.ga_data + ga.ga_len, cpstr, (size_t)cplen);
8604 ga.ga_len += cplen;
8605
8606 in_str += inlen;
8607 }
8608 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008609 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008610 // When not using multi-byte chars we can do it faster.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008611 p = vim_strchr(fromstr, *in_str);
8612 if (p != NULL)
8613 ga_append(&ga, tostr[p - fromstr]);
8614 else
8615 ga_append(&ga, *in_str);
8616 ++in_str;
8617 }
8618 }
8619
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008620 // add a terminating NUL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008621 (void)ga_grow(&ga, 1);
8622 ga_append(&ga, NUL);
8623
8624 rettv->vval.v_string = ga.ga_data;
8625}
8626
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008627/*
8628 * "trim({expr})" function
8629 */
8630 static void
8631f_trim(typval_T *argvars, typval_T *rettv)
8632{
8633 char_u buf1[NUMBUFLEN];
8634 char_u buf2[NUMBUFLEN];
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008635 char_u *head = tv_get_string_buf_chk(&argvars[0], buf1);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008636 char_u *mask = NULL;
8637 char_u *tail;
8638 char_u *prev;
8639 char_u *p;
8640 int c1;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008641 int dir = 0;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008642
8643 rettv->v_type = VAR_STRING;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008644 rettv->vval.v_string = NULL;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008645 if (head == NULL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008646 return;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008647
8648 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008649 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008650 mask = tv_get_string_buf_chk(&argvars[1], buf2);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008651
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008652 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008653 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008654 int error = 0;
8655
8656 // leading or trailing characters to trim
8657 dir = (int)tv_get_number_chk(&argvars[2], &error);
8658 if (error)
8659 return;
8660 if (dir < 0 || dir > 2)
8661 {
8662 semsg(_(e_invarg2), tv_get_string(&argvars[2]));
8663 return;
8664 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008665 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008666 }
8667
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008668 if (dir == 0 || dir == 1)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008669 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008670 // Trim leading characters
8671 while (*head != NUL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008672 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008673 c1 = PTR2CHAR(head);
8674 if (mask == NULL)
8675 {
8676 if (c1 > ' ' && c1 != 0xa0)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008677 break;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008678 }
8679 else
8680 {
8681 for (p = mask; *p != NUL; MB_PTR_ADV(p))
8682 if (c1 == PTR2CHAR(p))
8683 break;
8684 if (*p == NUL)
8685 break;
8686 }
8687 MB_PTR_ADV(head);
8688 }
8689 }
8690
8691 tail = head + STRLEN(head);
8692 if (dir == 0 || dir == 2)
8693 {
8694 // Trim trailing characters
8695 for (; tail > head; tail = prev)
8696 {
8697 prev = tail;
8698 MB_PTR_BACK(head, prev);
8699 c1 = PTR2CHAR(prev);
8700 if (mask == NULL)
8701 {
8702 if (c1 > ' ' && c1 != 0xa0)
8703 break;
8704 }
8705 else
8706 {
8707 for (p = mask; *p != NUL; MB_PTR_ADV(p))
8708 if (c1 == PTR2CHAR(p))
8709 break;
8710 if (*p == NUL)
8711 break;
8712 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008713 }
8714 }
8715 rettv->vval.v_string = vim_strnsave(head, (int)(tail - head));
8716}
8717
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008718#ifdef FEAT_FLOAT
8719/*
8720 * "trunc({float})" function
8721 */
8722 static void
8723f_trunc(typval_T *argvars, typval_T *rettv)
8724{
8725 float_T f = 0.0;
8726
8727 rettv->v_type = VAR_FLOAT;
8728 if (get_float_arg(argvars, &f) == OK)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008729 // trunc() is not in C90, use floor() or ceil() instead.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008730 rettv->vval.v_float = f > 0 ? floor(f) : ceil(f);
8731 else
8732 rettv->vval.v_float = 0.0;
8733}
8734#endif
8735
8736/*
8737 * "type(expr)" function
8738 */
8739 static void
8740f_type(typval_T *argvars, typval_T *rettv)
8741{
8742 int n = -1;
8743
8744 switch (argvars[0].v_type)
8745 {
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01008746 case VAR_NUMBER: n = VAR_TYPE_NUMBER; break;
8747 case VAR_STRING: n = VAR_TYPE_STRING; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008748 case VAR_PARTIAL:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01008749 case VAR_FUNC: n = VAR_TYPE_FUNC; break;
8750 case VAR_LIST: n = VAR_TYPE_LIST; break;
8751 case VAR_DICT: n = VAR_TYPE_DICT; break;
8752 case VAR_FLOAT: n = VAR_TYPE_FLOAT; break;
8753 case VAR_BOOL: n = VAR_TYPE_BOOL; break;
8754 case VAR_SPECIAL: n = VAR_TYPE_NONE; break;
Bram Moolenaarf562e722016-07-19 17:25:25 +02008755 case VAR_JOB: n = VAR_TYPE_JOB; break;
8756 case VAR_CHANNEL: n = VAR_TYPE_CHANNEL; break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01008757 case VAR_BLOB: n = VAR_TYPE_BLOB; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008758 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02008759 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01008760 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01008761 internal_error_no_abort("f_type(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008762 n = -1;
8763 break;
8764 }
8765 rettv->vval.v_number = n;
8766}
8767
8768/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008769 * "virtcol(string)" function
8770 */
8771 static void
8772f_virtcol(typval_T *argvars, typval_T *rettv)
8773{
8774 colnr_T vcol = 0;
8775 pos_T *fp;
8776 int fnum = curbuf->b_fnum;
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01008777 int len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008778
8779 fp = var2fpos(&argvars[0], FALSE, &fnum);
8780 if (fp != NULL && fp->lnum <= curbuf->b_ml.ml_line_count
8781 && fnum == curbuf->b_fnum)
8782 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01008783 // Limit the column to a valid value, getvvcol() doesn't check.
8784 if (fp->col < 0)
8785 fp->col = 0;
8786 else
8787 {
8788 len = (int)STRLEN(ml_get(fp->lnum));
8789 if (fp->col > len)
8790 fp->col = len;
8791 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008792 getvvcol(curwin, fp, NULL, NULL, &vcol);
8793 ++vcol;
8794 }
8795
8796 rettv->vval.v_number = vcol;
8797}
8798
8799/*
8800 * "visualmode()" function
8801 */
8802 static void
8803f_visualmode(typval_T *argvars, typval_T *rettv)
8804{
8805 char_u str[2];
8806
8807 rettv->v_type = VAR_STRING;
8808 str[0] = curbuf->b_visual_mode_eval;
8809 str[1] = NUL;
8810 rettv->vval.v_string = vim_strsave(str);
8811
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008812 // A non-zero number or non-empty string argument: reset mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008813 if (non_zero_arg(&argvars[0]))
8814 curbuf->b_visual_mode_eval = NUL;
8815}
8816
8817/*
8818 * "wildmenumode()" function
8819 */
8820 static void
8821f_wildmenumode(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8822{
8823#ifdef FEAT_WILDMENU
8824 if (wild_menu_showing)
8825 rettv->vval.v_number = 1;
8826#endif
8827}
8828
8829/*
Bram Moolenaar0c1e3742019-12-27 13:49:24 +01008830 * "windowsversion()" function
8831 */
8832 static void
8833f_windowsversion(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8834{
8835 rettv->v_type = VAR_STRING;
8836 rettv->vval.v_string = vim_strsave((char_u *)windowsVersion);
8837}
8838
8839/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008840 * "wordcount()" function
8841 */
8842 static void
8843f_wordcount(typval_T *argvars UNUSED, typval_T *rettv)
8844{
8845 if (rettv_dict_alloc(rettv) == FAIL)
8846 return;
8847 cursor_pos_info(rettv->vval.v_dict);
8848}
8849
8850/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008851 * "xor(expr, expr)" function
8852 */
8853 static void
8854f_xor(typval_T *argvars, typval_T *rettv)
8855{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008856 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
8857 ^ tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008858}
8859
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008860#endif // FEAT_EVAL