blob: 5a5daaa5af0cb731307257b29ccfafcdfa6eb829 [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},
582 {"getmatches", 0, 1, 0, ret_list_dict_any, f_getmatches},
583 {"getmousepos", 0, 0, 0, ret_dict_number, f_getmousepos},
584 {"getpid", 0, 0, 0, ret_number, f_getpid},
585 {"getpos", 1, 1, FEARG_1, ret_list_number, f_getpos},
586 {"getqflist", 0, 1, 0, ret_list_dict_any, f_getqflist},
587 {"getreg", 0, 3, FEARG_1, ret_string, f_getreg},
588 {"getregtype", 0, 1, FEARG_1, ret_string, f_getregtype},
589 {"gettabinfo", 0, 1, FEARG_1, ret_list_dict_any, f_gettabinfo},
590 {"gettabvar", 2, 3, FEARG_1, ret_any, f_gettabvar},
591 {"gettabwinvar", 3, 4, FEARG_1, ret_any, f_gettabwinvar},
592 {"gettagstack", 0, 1, FEARG_1, ret_dict_any, f_gettagstack},
593 {"getwininfo", 0, 1, FEARG_1, ret_list_dict_any, f_getwininfo},
594 {"getwinpos", 0, 1, FEARG_1, ret_list_number, f_getwinpos},
595 {"getwinposx", 0, 0, 0, ret_number, f_getwinposx},
596 {"getwinposy", 0, 0, 0, ret_number, f_getwinposy},
597 {"getwinvar", 2, 3, FEARG_1, ret_any, f_getwinvar},
598 {"glob", 1, 4, FEARG_1, ret_any, f_glob},
599 {"glob2regpat", 1, 1, FEARG_1, ret_string, f_glob2regpat},
600 {"globpath", 2, 5, FEARG_2, ret_any, f_globpath},
Bram Moolenaar79296512020-03-22 16:17:14 +0100601 {"has", 1, 2, 0, ret_number, f_has},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100602 {"has_key", 2, 2, FEARG_1, ret_number, f_has_key},
603 {"haslocaldir", 0, 2, FEARG_1, ret_number, f_haslocaldir},
604 {"hasmapto", 1, 3, FEARG_1, ret_number, f_hasmapto},
605 {"highlightID", 1, 1, FEARG_1, ret_number, f_hlID}, // obsolete
606 {"highlight_exists",1, 1, FEARG_1, ret_number, f_hlexists}, // obsolete
607 {"histadd", 2, 2, FEARG_2, ret_number, f_histadd},
608 {"histdel", 1, 2, FEARG_1, ret_number, f_histdel},
609 {"histget", 1, 2, FEARG_1, ret_string, f_histget},
610 {"histnr", 1, 1, FEARG_1, ret_number, f_histnr},
611 {"hlID", 1, 1, FEARG_1, ret_number, f_hlID},
612 {"hlexists", 1, 1, FEARG_1, ret_number, f_hlexists},
613 {"hostname", 0, 0, 0, ret_string, f_hostname},
614 {"iconv", 3, 3, FEARG_1, ret_string, f_iconv},
615 {"indent", 1, 1, FEARG_1, ret_number, f_indent},
616 {"index", 2, 4, FEARG_1, ret_number, f_index},
617 {"input", 1, 3, FEARG_1, ret_string, f_input},
618 {"inputdialog", 1, 3, FEARG_1, ret_string, f_inputdialog},
619 {"inputlist", 1, 1, FEARG_1, ret_number, f_inputlist},
620 {"inputrestore", 0, 0, 0, ret_number, f_inputrestore},
621 {"inputsave", 0, 0, 0, ret_number, f_inputsave},
622 {"inputsecret", 1, 2, FEARG_1, ret_string, f_inputsecret},
623 {"insert", 2, 3, FEARG_1, ret_any, f_insert},
624 {"interrupt", 0, 0, 0, ret_void, f_interrupt},
625 {"invert", 1, 1, FEARG_1, ret_number, f_invert},
626 {"isdirectory", 1, 1, FEARG_1, ret_number, f_isdirectory},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100627 {"isinf", 1, 1, FEARG_1, ret_number, MATH_FUNC(f_isinf)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100628 {"islocked", 1, 1, FEARG_1, ret_number, f_islocked},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100629 {"isnan", 1, 1, FEARG_1, ret_number, MATH_FUNC(f_isnan)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100630 {"items", 1, 1, FEARG_1, ret_list_any, f_items},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100631 {"job_getchannel", 1, 1, FEARG_1, ret_channel, JOB_FUNC(f_job_getchannel)},
632 {"job_info", 0, 1, FEARG_1, ret_dict_any, JOB_FUNC(f_job_info)},
633 {"job_setoptions", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_job_setoptions)},
634 {"job_start", 1, 2, FEARG_1, ret_job, JOB_FUNC(f_job_start)},
635 {"job_status", 1, 1, FEARG_1, ret_string, JOB_FUNC(f_job_status)},
636 {"job_stop", 1, 2, FEARG_1, ret_number, JOB_FUNC(f_job_stop)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100637 {"join", 1, 2, FEARG_1, ret_string, f_join},
638 {"js_decode", 1, 1, FEARG_1, ret_any, f_js_decode},
639 {"js_encode", 1, 1, FEARG_1, ret_string, f_js_encode},
640 {"json_decode", 1, 1, FEARG_1, ret_any, f_json_decode},
641 {"json_encode", 1, 1, FEARG_1, ret_string, f_json_encode},
642 {"keys", 1, 1, FEARG_1, ret_list_any, f_keys},
643 {"last_buffer_nr", 0, 0, 0, ret_number, f_last_buffer_nr}, // obsolete
644 {"len", 1, 1, FEARG_1, ret_number, f_len},
645 {"libcall", 3, 3, FEARG_3, ret_string, f_libcall},
646 {"libcallnr", 3, 3, FEARG_3, ret_number, f_libcallnr},
647 {"line", 1, 2, FEARG_1, ret_number, f_line},
648 {"line2byte", 1, 1, FEARG_1, ret_number, f_line2byte},
649 {"lispindent", 1, 1, FEARG_1, ret_number, f_lispindent},
650 {"list2str", 1, 2, FEARG_1, ret_string, f_list2str},
651 {"listener_add", 1, 2, FEARG_2, ret_number, f_listener_add},
652 {"listener_flush", 0, 1, FEARG_1, ret_void, f_listener_flush},
653 {"listener_remove", 1, 1, FEARG_1, ret_number, f_listener_remove},
654 {"localtime", 0, 0, 0, ret_number, f_localtime},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100655 {"log", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_log)},
656 {"log10", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_log10)},
657 {"luaeval", 1, 2, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200658#ifdef FEAT_LUA
Bram Moolenaar15c47602020-03-26 22:16:48 +0100659 f_luaeval
660#else
661 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200662#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100663 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100664 {"map", 2, 2, FEARG_1, ret_any, f_map},
665 {"maparg", 1, 4, FEARG_1, ret_string, f_maparg},
666 {"mapcheck", 1, 3, FEARG_1, ret_string, f_mapcheck},
667 {"match", 2, 4, FEARG_1, ret_any, f_match},
668 {"matchadd", 2, 5, FEARG_1, ret_number, f_matchadd},
669 {"matchaddpos", 2, 5, FEARG_1, ret_number, f_matchaddpos},
670 {"matcharg", 1, 1, FEARG_1, ret_list_string, f_matcharg},
671 {"matchdelete", 1, 2, FEARG_1, ret_number, f_matchdelete},
672 {"matchend", 2, 4, FEARG_1, ret_number, f_matchend},
673 {"matchlist", 2, 4, FEARG_1, ret_list_string, f_matchlist},
674 {"matchstr", 2, 4, FEARG_1, ret_string, f_matchstr},
675 {"matchstrpos", 2, 4, FEARG_1, ret_list_any, f_matchstrpos},
676 {"max", 1, 1, FEARG_1, ret_any, f_max},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100677 {"menu_info", 1, 2, FEARG_1, ret_dict_any,
Bram Moolenaara2cbdea2020-03-16 21:08:31 +0100678#ifdef FEAT_MENU
Bram Moolenaar15c47602020-03-26 22:16:48 +0100679 f_menu_info
680#else
681 NULL
Bram Moolenaara2cbdea2020-03-16 21:08:31 +0100682#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100683 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100684 {"min", 1, 1, FEARG_1, ret_any, f_min},
685 {"mkdir", 1, 3, FEARG_1, ret_number, f_mkdir},
686 {"mode", 0, 1, FEARG_1, ret_string, f_mode},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100687 {"mzeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200688#ifdef FEAT_MZSCHEME
Bram Moolenaar15c47602020-03-26 22:16:48 +0100689 f_mzeval
690#else
691 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200692#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100693 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100694 {"nextnonblank", 1, 1, FEARG_1, ret_number, f_nextnonblank},
695 {"nr2char", 1, 2, FEARG_1, ret_string, f_nr2char},
696 {"or", 2, 2, FEARG_1, ret_number, f_or},
697 {"pathshorten", 1, 1, FEARG_1, ret_string, f_pathshorten},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100698 {"perleval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200699#ifdef FEAT_PERL
Bram Moolenaar15c47602020-03-26 22:16:48 +0100700 f_perleval
701#else
702 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200703#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100704 },
705 {"popup_atcursor", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_atcursor)},
706 {"popup_beval", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_beval)},
707 {"popup_clear", 0, 0, 0, ret_void, PROP_FUNC(f_popup_clear)},
708 {"popup_close", 1, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_close)},
709 {"popup_create", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_create)},
710 {"popup_dialog", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_dialog)},
711 {"popup_filter_menu", 2, 2, 0, ret_number, PROP_FUNC(f_popup_filter_menu)},
712 {"popup_filter_yesno", 2, 2, 0, ret_number, PROP_FUNC(f_popup_filter_yesno)},
713 {"popup_findinfo", 0, 0, 0, ret_number, PROP_FUNC(f_popup_findinfo)},
714 {"popup_findpreview", 0, 0, 0, ret_number, PROP_FUNC(f_popup_findpreview)},
715 {"popup_getoptions", 1, 1, FEARG_1, ret_dict_any, PROP_FUNC(f_popup_getoptions)},
716 {"popup_getpos", 1, 1, FEARG_1, ret_dict_any, PROP_FUNC(f_popup_getpos)},
717 {"popup_hide", 1, 1, FEARG_1, ret_void, PROP_FUNC(f_popup_hide)},
718 {"popup_locate", 2, 2, 0, ret_number, PROP_FUNC(f_popup_locate)},
719 {"popup_menu", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_menu)},
720 {"popup_move", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_move)},
721 {"popup_notification", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_notification)},
722 {"popup_setoptions", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_setoptions)},
723 {"popup_settext", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_settext)},
724 {"popup_show", 1, 1, FEARG_1, ret_void, PROP_FUNC(f_popup_show)},
725 {"pow", 2, 2, FEARG_1, ret_float, FLOAT_FUNC(f_pow)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100726 {"prevnonblank", 1, 1, FEARG_1, ret_number, f_prevnonblank},
727 {"printf", 1, 19, FEARG_2, ret_string, f_printf},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100728 {"prompt_setcallback", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_prompt_setcallback)},
729 {"prompt_setinterrupt", 2, 2, FEARG_1,ret_void, JOB_FUNC(f_prompt_setinterrupt)},
730 {"prompt_setprompt", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_prompt_setprompt)},
731 {"prop_add", 3, 3, FEARG_1, ret_void, PROP_FUNC(f_prop_add)},
732 {"prop_clear", 1, 3, FEARG_1, ret_void, PROP_FUNC(f_prop_clear)},
733 {"prop_find", 1, 2, FEARG_1, ret_dict_any, PROP_FUNC(f_prop_find)},
734 {"prop_list", 1, 2, FEARG_1, ret_list_dict_any, PROP_FUNC(f_prop_list)},
735 {"prop_remove", 1, 3, FEARG_1, ret_number, PROP_FUNC(f_prop_remove)},
736 {"prop_type_add", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_prop_type_add)},
737 {"prop_type_change", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_prop_type_change)},
738 {"prop_type_delete", 1, 2, FEARG_1, ret_void, PROP_FUNC(f_prop_type_delete)},
739 {"prop_type_get", 1, 2, FEARG_1, ret_dict_any, PROP_FUNC(f_prop_type_get)},
740 {"prop_type_list", 0, 1, FEARG_1, ret_list_string, PROP_FUNC(f_prop_type_list)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100741 {"pum_getpos", 0, 0, 0, ret_dict_number, f_pum_getpos},
742 {"pumvisible", 0, 0, 0, ret_number, f_pumvisible},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100743 {"py3eval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200744#ifdef FEAT_PYTHON3
Bram Moolenaar15c47602020-03-26 22:16:48 +0100745 f_py3eval
746#else
747 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200748#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100749 },
750 {"pyeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200751#ifdef FEAT_PYTHON
Bram Moolenaar15c47602020-03-26 22:16:48 +0100752 f_pyeval
753#else
754 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200755#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100756 },
757 {"pyxeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100758#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
Bram Moolenaar15c47602020-03-26 22:16:48 +0100759 f_pyxeval
760#else
761 NULL
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100762#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100763 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100764 {"rand", 0, 1, FEARG_1, ret_number, f_rand},
765 {"range", 1, 3, FEARG_1, ret_list_number, f_range},
766 {"readdir", 1, 2, FEARG_1, ret_list_string, f_readdir},
767 {"readfile", 1, 3, FEARG_1, ret_any, f_readfile},
768 {"reg_executing", 0, 0, 0, ret_string, f_reg_executing},
769 {"reg_recording", 0, 0, 0, ret_string, f_reg_recording},
770 {"reltime", 0, 2, FEARG_1, ret_list_any, f_reltime},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100771 {"reltimefloat", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_reltimefloat)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100772 {"reltimestr", 1, 1, FEARG_1, ret_string, f_reltimestr},
773 {"remote_expr", 2, 4, FEARG_1, ret_string, f_remote_expr},
774 {"remote_foreground", 1, 1, FEARG_1, ret_string, f_remote_foreground},
775 {"remote_peek", 1, 2, FEARG_1, ret_number, f_remote_peek},
776 {"remote_read", 1, 2, FEARG_1, ret_string, f_remote_read},
777 {"remote_send", 2, 3, FEARG_1, ret_string, f_remote_send},
778 {"remote_startserver", 1, 1, FEARG_1, ret_void, f_remote_startserver},
779 {"remove", 2, 3, FEARG_1, ret_any, f_remove},
780 {"rename", 2, 2, FEARG_1, ret_number, f_rename},
781 {"repeat", 2, 2, FEARG_1, ret_any, f_repeat},
782 {"resolve", 1, 1, FEARG_1, ret_string, f_resolve},
783 {"reverse", 1, 1, FEARG_1, ret_any, f_reverse},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100784 {"round", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_round)},
785 {"rubyeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100786#ifdef FEAT_RUBY
Bram Moolenaar15c47602020-03-26 22:16:48 +0100787 f_rubyeval
788#else
789 NULL
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100790#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100791 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100792 {"screenattr", 2, 2, FEARG_1, ret_number, f_screenattr},
793 {"screenchar", 2, 2, FEARG_1, ret_number, f_screenchar},
794 {"screenchars", 2, 2, FEARG_1, ret_list_number, f_screenchars},
795 {"screencol", 0, 0, 0, ret_number, f_screencol},
796 {"screenpos", 3, 3, FEARG_1, ret_dict_number, f_screenpos},
797 {"screenrow", 0, 0, 0, ret_number, f_screenrow},
798 {"screenstring", 2, 2, FEARG_1, ret_string, f_screenstring},
799 {"search", 1, 4, FEARG_1, ret_number, f_search},
800 {"searchdecl", 1, 3, FEARG_1, ret_number, f_searchdecl},
801 {"searchpair", 3, 7, 0, ret_number, f_searchpair},
802 {"searchpairpos", 3, 7, 0, ret_list_number, f_searchpairpos},
803 {"searchpos", 1, 4, FEARG_1, ret_list_number, f_searchpos},
804 {"server2client", 2, 2, FEARG_1, ret_number, f_server2client},
805 {"serverlist", 0, 0, 0, ret_string, f_serverlist},
806 {"setbufline", 3, 3, FEARG_3, ret_number, f_setbufline},
807 {"setbufvar", 3, 3, FEARG_3, ret_void, f_setbufvar},
808 {"setcharsearch", 1, 1, FEARG_1, ret_void, f_setcharsearch},
809 {"setcmdpos", 1, 1, FEARG_1, ret_number, f_setcmdpos},
810 {"setenv", 2, 2, FEARG_2, ret_void, f_setenv},
811 {"setfperm", 2, 2, FEARG_1, ret_number, f_setfperm},
812 {"setline", 2, 2, FEARG_2, ret_number, f_setline},
813 {"setloclist", 2, 4, FEARG_2, ret_number, f_setloclist},
814 {"setmatches", 1, 2, FEARG_1, ret_number, f_setmatches},
815 {"setpos", 2, 2, FEARG_2, ret_number, f_setpos},
816 {"setqflist", 1, 3, FEARG_1, ret_number, f_setqflist},
817 {"setreg", 2, 3, FEARG_2, ret_number, f_setreg},
818 {"settabvar", 3, 3, FEARG_3, ret_void, f_settabvar},
819 {"settabwinvar", 4, 4, FEARG_4, ret_void, f_settabwinvar},
820 {"settagstack", 2, 3, FEARG_2, ret_number, f_settagstack},
821 {"setwinvar", 3, 3, FEARG_3, ret_void, f_setwinvar},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100822 {"sha256", 1, 1, FEARG_1, ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200823#ifdef FEAT_CRYPT
Bram Moolenaar15c47602020-03-26 22:16:48 +0100824 f_sha256
825#else
826 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200827#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100828 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100829 {"shellescape", 1, 2, FEARG_1, ret_string, f_shellescape},
830 {"shiftwidth", 0, 1, FEARG_1, ret_number, f_shiftwidth},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100831 {"sign_define", 1, 2, FEARG_1, ret_any, SIGN_FUNC(f_sign_define)},
832 {"sign_getdefined", 0, 1, FEARG_1, ret_list_dict_any, SIGN_FUNC(f_sign_getdefined)},
833 {"sign_getplaced", 0, 2, FEARG_1, ret_list_dict_any, SIGN_FUNC(f_sign_getplaced)},
834 {"sign_jump", 3, 3, FEARG_1, ret_number, SIGN_FUNC(f_sign_jump)},
835 {"sign_place", 4, 5, FEARG_1, ret_number, SIGN_FUNC(f_sign_place)},
836 {"sign_placelist", 1, 1, FEARG_1, ret_list_number, SIGN_FUNC(f_sign_placelist)},
837 {"sign_undefine", 0, 1, FEARG_1, ret_number, SIGN_FUNC(f_sign_undefine)},
838 {"sign_unplace", 1, 2, FEARG_1, ret_number, SIGN_FUNC(f_sign_unplace)},
839 {"sign_unplacelist", 1, 2, FEARG_1, ret_list_number, SIGN_FUNC(f_sign_unplacelist)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100840 {"simplify", 1, 1, 0, ret_string, f_simplify},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100841 {"sin", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_sin)},
842 {"sinh", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_sinh)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100843 {"sort", 1, 3, FEARG_1, ret_list_any, f_sort},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100844 {"sound_clear", 0, 0, 0, ret_void, SOUND_FUNC(f_sound_clear)},
845 {"sound_playevent", 1, 2, FEARG_1, ret_number, SOUND_FUNC(f_sound_playevent)},
846 {"sound_playfile", 1, 2, FEARG_1, ret_number, SOUND_FUNC(f_sound_playfile)},
847 {"sound_stop", 1, 1, FEARG_1, ret_void, SOUND_FUNC(f_sound_stop)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100848 {"soundfold", 1, 1, FEARG_1, ret_string, f_soundfold},
849 {"spellbadword", 0, 1, FEARG_1, ret_list_string, f_spellbadword},
850 {"spellsuggest", 1, 3, FEARG_1, ret_list_string, f_spellsuggest},
851 {"split", 1, 3, FEARG_1, ret_list_string, f_split},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100852 {"sqrt", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_sqrt)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100853 {"srand", 0, 1, FEARG_1, ret_list_number, f_srand},
854 {"state", 0, 1, FEARG_1, ret_string, f_state},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100855 {"str2float", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_str2float)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100856 {"str2list", 1, 2, FEARG_1, ret_list_number, f_str2list},
857 {"str2nr", 1, 3, FEARG_1, ret_number, f_str2nr},
858 {"strcharpart", 2, 3, FEARG_1, ret_string, f_strcharpart},
859 {"strchars", 1, 2, FEARG_1, ret_number, f_strchars},
860 {"strdisplaywidth", 1, 2, FEARG_1, ret_number, f_strdisplaywidth},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100861 {"strftime", 1, 2, FEARG_1, ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200862#ifdef HAVE_STRFTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +0100863 f_strftime
864#else
865 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200866#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100867 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100868 {"strgetchar", 2, 2, FEARG_1, ret_number, f_strgetchar},
869 {"stridx", 2, 3, FEARG_1, ret_number, f_stridx},
870 {"string", 1, 1, FEARG_1, ret_string, f_string},
871 {"strlen", 1, 1, FEARG_1, ret_number, f_strlen},
872 {"strpart", 2, 3, FEARG_1, ret_string, f_strpart},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100873 {"strptime", 2, 2, FEARG_1, ret_number,
Bram Moolenaar10455d42019-11-21 15:36:18 +0100874#ifdef HAVE_STRPTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +0100875 f_strptime
876#else
877 NULL
Bram Moolenaar10455d42019-11-21 15:36:18 +0100878#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100879 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100880 {"strridx", 2, 3, FEARG_1, ret_number, f_strridx},
881 {"strtrans", 1, 1, FEARG_1, ret_string, f_strtrans},
882 {"strwidth", 1, 1, FEARG_1, ret_number, f_strwidth},
883 {"submatch", 1, 2, FEARG_1, ret_string, f_submatch},
884 {"substitute", 4, 4, FEARG_1, ret_string, f_substitute},
885 {"swapinfo", 1, 1, FEARG_1, ret_dict_any, f_swapinfo},
886 {"swapname", 1, 1, FEARG_1, ret_string, f_swapname},
887 {"synID", 3, 3, 0, ret_number, f_synID},
888 {"synIDattr", 2, 3, FEARG_1, ret_string, f_synIDattr},
889 {"synIDtrans", 1, 1, FEARG_1, ret_number, f_synIDtrans},
890 {"synconcealed", 2, 2, 0, ret_list_any, f_synconcealed},
891 {"synstack", 2, 2, 0, ret_list_number, f_synstack},
892 {"system", 1, 2, FEARG_1, ret_string, f_system},
893 {"systemlist", 1, 2, FEARG_1, ret_list_string, f_systemlist},
894 {"tabpagebuflist", 0, 1, FEARG_1, ret_list_number, f_tabpagebuflist},
895 {"tabpagenr", 0, 1, 0, ret_number, f_tabpagenr},
896 {"tabpagewinnr", 1, 2, FEARG_1, ret_number, f_tabpagewinnr},
897 {"tagfiles", 0, 0, 0, ret_list_string, f_tagfiles},
898 {"taglist", 1, 2, FEARG_1, ret_list_dict_any, f_taglist},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100899 {"tan", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_tan)},
900 {"tanh", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_tanh)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100901 {"tempname", 0, 0, 0, ret_string, f_tempname},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100902 {"term_dumpdiff", 2, 3, FEARG_1, ret_number, TERM_FUNC(f_term_dumpdiff)},
903 {"term_dumpload", 1, 2, FEARG_1, ret_number, TERM_FUNC(f_term_dumpload)},
904 {"term_dumpwrite", 2, 3, FEARG_2, ret_void, TERM_FUNC(f_term_dumpwrite)},
905 {"term_getaltscreen", 1, 1, FEARG_1, ret_number, TERM_FUNC(f_term_getaltscreen)},
906 {"term_getansicolors", 1, 1, FEARG_1, ret_list_string,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +0100907#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +0100908 f_term_getansicolors
909#else
910 NULL
Bram Moolenaarc6df10e2017-07-29 20:15:08 +0200911#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100912 },
913 {"term_getattr", 2, 2, FEARG_1, ret_number, TERM_FUNC(f_term_getattr)},
914 {"term_getcursor", 1, 1, FEARG_1, ret_list_any, TERM_FUNC(f_term_getcursor)},
915 {"term_getjob", 1, 1, FEARG_1, ret_job, TERM_FUNC(f_term_getjob)},
916 {"term_getline", 2, 2, FEARG_1, ret_string, TERM_FUNC(f_term_getline)},
917 {"term_getscrolled", 1, 1, FEARG_1, ret_number, TERM_FUNC(f_term_getscrolled)},
918 {"term_getsize", 1, 1, FEARG_1, ret_list_number, TERM_FUNC(f_term_getsize)},
919 {"term_getstatus", 1, 1, FEARG_1, ret_string, TERM_FUNC(f_term_getstatus)},
920 {"term_gettitle", 1, 1, FEARG_1, ret_string, TERM_FUNC(f_term_gettitle)},
921 {"term_gettty", 1, 2, FEARG_1, ret_string, TERM_FUNC(f_term_gettty)},
922 {"term_list", 0, 0, 0, ret_list_number, TERM_FUNC(f_term_list)},
923 {"term_scrape", 2, 2, FEARG_1, ret_list_dict_any, TERM_FUNC(f_term_scrape)},
924 {"term_sendkeys", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_sendkeys)},
925 {"term_setansicolors", 2, 2, FEARG_1, ret_void,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +0100926#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +0100927 f_term_setansicolors
928#else
929 NULL
930#endif
931 },
932 {"term_setapi", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_setapi)},
933 {"term_setkill", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_setkill)},
934 {"term_setrestore", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_setrestore)},
935 {"term_setsize", 3, 3, FEARG_1, ret_void, TERM_FUNC(f_term_setsize)},
936 {"term_start", 1, 2, FEARG_1, ret_number, TERM_FUNC(f_term_start)},
937 {"term_wait", 1, 2, FEARG_1, ret_void, TERM_FUNC(f_term_wait)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100938 {"test_alloc_fail", 3, 3, FEARG_1, ret_void, f_test_alloc_fail},
939 {"test_autochdir", 0, 0, 0, ret_void, f_test_autochdir},
940 {"test_feedinput", 1, 1, FEARG_1, ret_void, f_test_feedinput},
941 {"test_garbagecollect_now", 0, 0, 0, ret_void, f_test_garbagecollect_now},
942 {"test_garbagecollect_soon", 0, 0, 0, ret_void, f_test_garbagecollect_soon},
943 {"test_getvalue", 1, 1, FEARG_1, ret_number, f_test_getvalue},
944 {"test_ignore_error", 1, 1, FEARG_1, ret_void, f_test_ignore_error},
945 {"test_null_blob", 0, 0, 0, ret_blob, f_test_null_blob},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100946 {"test_null_channel", 0, 0, 0, ret_channel, JOB_FUNC(f_test_null_channel)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100947 {"test_null_dict", 0, 0, 0, ret_dict_any, f_test_null_dict},
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200948 {"test_null_function", 0, 0, 0, ret_func_any, f_test_null_function},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100949 {"test_null_job", 0, 0, 0, ret_job, JOB_FUNC(f_test_null_job)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100950 {"test_null_list", 0, 0, 0, ret_list_any, f_test_null_list},
Bram Moolenaard77a8522020-04-03 21:59:57 +0200951 {"test_null_partial", 0, 0, 0, ret_func_any, f_test_null_partial},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100952 {"test_null_string", 0, 0, 0, ret_string, f_test_null_string},
953 {"test_option_not_set", 1, 1, FEARG_1,ret_void, f_test_option_not_set},
954 {"test_override", 2, 2, FEARG_2, ret_void, f_test_override},
955 {"test_refcount", 1, 1, FEARG_1, ret_number, f_test_refcount},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100956 {"test_scrollbar", 3, 3, FEARG_2, ret_void,
Bram Moolenaarab186732018-09-14 21:27:06 +0200957#ifdef FEAT_GUI
Bram Moolenaar15c47602020-03-26 22:16:48 +0100958 f_test_scrollbar
959#else
960 NULL
Bram Moolenaarab186732018-09-14 21:27:06 +0200961#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100962 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100963 {"test_setmouse", 2, 2, 0, ret_void, f_test_setmouse},
964 {"test_settime", 1, 1, FEARG_1, ret_void, f_test_settime},
965 {"test_srand_seed", 0, 1, FEARG_1, ret_void, f_test_srand_seed},
966 {"test_unknown", 0, 0, 0, ret_any, f_test_unknown},
967 {"test_void", 0, 0, 0, ret_any, f_test_void},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100968 {"timer_info", 0, 1, FEARG_1, ret_list_dict_any, TIMER_FUNC(f_timer_info)},
969 {"timer_pause", 2, 2, FEARG_1, ret_void, TIMER_FUNC(f_timer_pause)},
970 {"timer_start", 2, 3, FEARG_1, ret_number, TIMER_FUNC(f_timer_start)},
971 {"timer_stop", 1, 1, FEARG_1, ret_void, TIMER_FUNC(f_timer_stop)},
972 {"timer_stopall", 0, 0, 0, ret_void, TIMER_FUNC(f_timer_stopall)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100973 {"tolower", 1, 1, FEARG_1, ret_string, f_tolower},
974 {"toupper", 1, 1, FEARG_1, ret_string, f_toupper},
975 {"tr", 3, 3, FEARG_1, ret_string, f_tr},
976 {"trim", 1, 2, FEARG_1, ret_string, f_trim},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100977 {"trunc", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_trunc)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100978 {"type", 1, 1, FEARG_1, ret_number, f_type},
979 {"undofile", 1, 1, FEARG_1, ret_string, f_undofile},
980 {"undotree", 0, 0, 0, ret_dict_any, f_undotree},
981 {"uniq", 1, 3, FEARG_1, ret_list_any, f_uniq},
982 {"values", 1, 1, FEARG_1, ret_list_any, f_values},
983 {"virtcol", 1, 1, FEARG_1, ret_number, f_virtcol},
984 {"visualmode", 0, 1, 0, ret_string, f_visualmode},
985 {"wildmenumode", 0, 0, 0, ret_number, f_wildmenumode},
986 {"win_execute", 2, 3, FEARG_2, ret_string, f_win_execute},
987 {"win_findbuf", 1, 1, FEARG_1, ret_list_number, f_win_findbuf},
988 {"win_getid", 0, 2, FEARG_1, ret_number, f_win_getid},
989 {"win_gettype", 0, 1, FEARG_1, ret_string, f_win_gettype},
990 {"win_gotoid", 1, 1, FEARG_1, ret_number, f_win_gotoid},
991 {"win_id2tabwin", 1, 1, FEARG_1, ret_list_number, f_win_id2tabwin},
992 {"win_id2win", 1, 1, FEARG_1, ret_number, f_win_id2win},
993 {"win_screenpos", 1, 1, FEARG_1, ret_list_number, f_win_screenpos},
994 {"win_splitmove", 2, 3, FEARG_1, ret_number, f_win_splitmove},
995 {"winbufnr", 1, 1, FEARG_1, ret_number, f_winbufnr},
996 {"wincol", 0, 0, 0, ret_number, f_wincol},
997 {"windowsversion", 0, 0, 0, ret_string, f_windowsversion},
998 {"winheight", 1, 1, FEARG_1, ret_number, f_winheight},
999 {"winlayout", 0, 1, FEARG_1, ret_list_any, f_winlayout},
1000 {"winline", 0, 0, 0, ret_number, f_winline},
1001 {"winnr", 0, 1, FEARG_1, ret_number, f_winnr},
1002 {"winrestcmd", 0, 0, 0, ret_string, f_winrestcmd},
1003 {"winrestview", 1, 1, FEARG_1, ret_void, f_winrestview},
1004 {"winsaveview", 0, 0, 0, ret_dict_any, f_winsaveview},
1005 {"winwidth", 1, 1, FEARG_1, ret_number, f_winwidth},
1006 {"wordcount", 0, 0, 0, ret_dict_number, f_wordcount},
1007 {"writefile", 2, 3, FEARG_1, ret_number, f_writefile},
1008 {"xor", 2, 2, FEARG_1, ret_number, f_xor},
Bram Moolenaarac92e252019-08-03 21:58:38 +02001009};
1010
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001011/*
1012 * Function given to ExpandGeneric() to obtain the list of internal
1013 * or user defined function names.
1014 */
1015 char_u *
1016get_function_name(expand_T *xp, int idx)
1017{
1018 static int intidx = -1;
1019 char_u *name;
1020
1021 if (idx == 0)
1022 intidx = -1;
1023 if (intidx < 0)
1024 {
1025 name = get_user_func_name(xp, idx);
1026 if (name != NULL)
1027 return name;
1028 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001029 if (++intidx < (int)(sizeof(global_functions) / sizeof(funcentry_T)))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001030 {
Bram Moolenaarac92e252019-08-03 21:58:38 +02001031 STRCPY(IObuff, global_functions[intidx].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001032 STRCAT(IObuff, "(");
Bram Moolenaarac92e252019-08-03 21:58:38 +02001033 if (global_functions[intidx].f_max_argc == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001034 STRCAT(IObuff, ")");
1035 return IObuff;
1036 }
1037
1038 return NULL;
1039}
1040
1041/*
1042 * Function given to ExpandGeneric() to obtain the list of internal or
1043 * user defined variable or function names.
1044 */
1045 char_u *
1046get_expr_name(expand_T *xp, int idx)
1047{
1048 static int intidx = -1;
1049 char_u *name;
1050
1051 if (idx == 0)
1052 intidx = -1;
1053 if (intidx < 0)
1054 {
1055 name = get_function_name(xp, idx);
1056 if (name != NULL)
1057 return name;
1058 }
1059 return get_user_var_name(xp, ++intidx);
1060}
1061
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001062/*
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001063 * Find internal function "name" in table "global_functions".
Bram Moolenaar15c47602020-03-26 22:16:48 +01001064 * Return index, or -1 if not found or "implemented" is TRUE and the function
1065 * is not implemented.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001066 */
Bram Moolenaar15c47602020-03-26 22:16:48 +01001067 static int
1068find_internal_func_opt(char_u *name, int implemented)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001069{
1070 int first = 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001071 int last;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001072 int cmp;
1073 int x;
1074
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001075 last = (int)(sizeof(global_functions) / sizeof(funcentry_T)) - 1;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001076
1077 // Find the function name in the table. Binary search.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001078 while (first <= last)
1079 {
1080 x = first + ((unsigned)(last - first) >> 1);
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001081 cmp = STRCMP(name, global_functions[x].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001082 if (cmp < 0)
1083 last = x - 1;
1084 else if (cmp > 0)
1085 first = x + 1;
Bram Moolenaar15c47602020-03-26 22:16:48 +01001086 else if (implemented && global_functions[x].f_func == NULL)
1087 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001088 else
1089 return x;
1090 }
1091 return -1;
1092}
1093
Bram Moolenaar15c47602020-03-26 22:16:48 +01001094/*
1095 * Find internal function "name" in table "global_functions".
1096 * Return index, or -1 if not found or the function is not implemented.
1097 */
1098 int
1099find_internal_func(char_u *name)
1100{
1101 return find_internal_func_opt(name, TRUE);
1102}
1103
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001104 int
Bram Moolenaarac92e252019-08-03 21:58:38 +02001105has_internal_func(char_u *name)
1106{
Bram Moolenaar15c47602020-03-26 22:16:48 +01001107 return find_internal_func_opt(name, TRUE) >= 0;
1108}
1109
1110 static int
1111has_internal_func_name(char_u *name)
1112{
1113 return find_internal_func_opt(name, FALSE) >= 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001114}
1115
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001116 char *
1117internal_func_name(int idx)
1118{
1119 return global_functions[idx].f_name;
1120}
1121
1122 type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001123internal_func_ret_type(int idx, int argcount, type_T **argtypes)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001124{
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001125 return global_functions[idx].f_retfunc(argcount, argtypes);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001126}
1127
1128/*
1129 * Check the argument count to use for internal function "idx".
1130 * Returns OK or FAIL;
1131 */
1132 int
1133check_internal_func(int idx, int argcount)
1134{
1135 int res;
1136 char *name;
1137
1138 if (argcount < global_functions[idx].f_min_argc)
1139 res = FCERR_TOOFEW;
1140 else if (argcount > global_functions[idx].f_max_argc)
1141 res = FCERR_TOOMANY;
1142 else
1143 return OK;
1144
1145 name = internal_func_name(idx);
1146 if (res == FCERR_TOOMANY)
1147 semsg(_(e_toomanyarg), name);
1148 else
1149 semsg(_(e_toofewarg), name);
1150 return FAIL;
1151}
1152
Bram Moolenaarac92e252019-08-03 21:58:38 +02001153 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001154call_internal_func(
1155 char_u *name,
1156 int argcount,
1157 typval_T *argvars,
1158 typval_T *rettv)
1159{
1160 int i;
1161
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001162 i = find_internal_func(name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001163 if (i < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001164 return FCERR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001165 if (argcount < global_functions[i].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001166 return FCERR_TOOFEW;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001167 if (argcount > global_functions[i].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001168 return FCERR_TOOMANY;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001169 argvars[argcount].v_type = VAR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001170 global_functions[i].f_func(argvars, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001171 return FCERR_NONE;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001172}
1173
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001174 void
1175call_internal_func_by_idx(
1176 int idx,
1177 typval_T *argvars,
1178 typval_T *rettv)
1179{
1180 global_functions[idx].f_func(argvars, rettv);
1181}
1182
Bram Moolenaarac92e252019-08-03 21:58:38 +02001183/*
1184 * Invoke a method for base->method().
1185 */
1186 int
1187call_internal_method(
1188 char_u *name,
1189 int argcount,
1190 typval_T *argvars,
1191 typval_T *rettv,
1192 typval_T *basetv)
1193{
1194 int i;
1195 int fi;
1196 typval_T argv[MAX_FUNC_ARGS + 1];
1197
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001198 fi = find_internal_func(name);
Bram Moolenaar91746392019-08-16 22:22:31 +02001199 if (fi < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001200 return FCERR_UNKNOWN;
Bram Moolenaar91746392019-08-16 22:22:31 +02001201 if (global_functions[fi].f_argtype == 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001202 return FCERR_NOTMETHOD;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001203 if (argcount + 1 < global_functions[fi].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001204 return FCERR_TOOFEW;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001205 if (argcount + 1 > global_functions[fi].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001206 return FCERR_TOOMANY;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001207
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001208 if (global_functions[fi].f_argtype == FEARG_LAST)
Bram Moolenaar25e42232019-08-04 15:04:10 +02001209 {
1210 // base value goes last
1211 for (i = 0; i < argcount; ++i)
1212 argv[i] = argvars[i];
1213 argv[argcount] = *basetv;
1214 }
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001215 else if (global_functions[fi].f_argtype == FEARG_2)
Bram Moolenaar25e42232019-08-04 15:04:10 +02001216 {
1217 // base value goes second
1218 argv[0] = argvars[0];
1219 argv[1] = *basetv;
1220 for (i = 1; i < argcount; ++i)
1221 argv[i + 1] = argvars[i];
1222 }
Bram Moolenaar24278d22019-08-16 21:49:22 +02001223 else if (global_functions[fi].f_argtype == FEARG_3)
1224 {
1225 // base value goes third
1226 argv[0] = argvars[0];
1227 argv[1] = argvars[1];
1228 argv[2] = *basetv;
1229 for (i = 2; i < argcount; ++i)
1230 argv[i + 1] = argvars[i];
1231 }
Bram Moolenaaraad222c2019-09-06 22:46:09 +02001232 else if (global_functions[fi].f_argtype == FEARG_4)
1233 {
1234 // base value goes fourth
1235 argv[0] = argvars[0];
1236 argv[1] = argvars[1];
1237 argv[2] = argvars[2];
1238 argv[3] = *basetv;
1239 for (i = 3; i < argcount; ++i)
1240 argv[i + 1] = argvars[i];
1241 }
Bram Moolenaar25e42232019-08-04 15:04:10 +02001242 else
1243 {
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001244 // FEARG_1: base value goes first
Bram Moolenaar25e42232019-08-04 15:04:10 +02001245 argv[0] = *basetv;
1246 for (i = 0; i < argcount; ++i)
1247 argv[i + 1] = argvars[i];
1248 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001249 argv[argcount + 1].v_type = VAR_UNKNOWN;
1250
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001251 global_functions[fi].f_func(argv, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001252 return FCERR_NONE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001253}
1254
1255/*
1256 * Return TRUE for a non-zero Number and a non-empty String.
1257 */
Bram Moolenaar0e57dd82019-09-16 22:56:03 +02001258 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001259non_zero_arg(typval_T *argvars)
1260{
1261 return ((argvars[0].v_type == VAR_NUMBER
1262 && argvars[0].vval.v_number != 0)
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01001263 || (argvars[0].v_type == VAR_BOOL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001264 && argvars[0].vval.v_number == VVAL_TRUE)
1265 || (argvars[0].v_type == VAR_STRING
1266 && argvars[0].vval.v_string != NULL
1267 && *argvars[0].vval.v_string != NUL));
1268}
1269
1270/*
1271 * Get the lnum from the first argument.
1272 * Also accepts ".", "$", etc., but that only works for the current buffer.
1273 * Returns -1 on error.
1274 */
Bram Moolenaarb60d8512019-06-29 07:59:04 +02001275 linenr_T
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001276tv_get_lnum(typval_T *argvars)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001277{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001278 linenr_T lnum;
1279
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001280 lnum = (linenr_T)tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar1f3165b2019-09-04 13:21:26 +02001281 if (lnum == 0) // no valid number, try using arg like line()
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001282 {
Bram Moolenaar1f3165b2019-09-04 13:21:26 +02001283 int fnum;
1284 pos_T *fp = var2fpos(&argvars[0], TRUE, &fnum);
1285
1286 if (fp != NULL)
1287 lnum = fp->lnum;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001288 }
1289 return lnum;
1290}
1291
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001292/*
1293 * Get the lnum from the first argument.
1294 * Also accepts "$", then "buf" is used.
1295 * Returns 0 on error.
1296 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02001297 linenr_T
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001298tv_get_lnum_buf(typval_T *argvars, buf_T *buf)
1299{
1300 if (argvars[0].v_type == VAR_STRING
1301 && argvars[0].vval.v_string != NULL
1302 && argvars[0].vval.v_string[0] == '$'
1303 && buf != NULL)
1304 return buf->b_ml.ml_line_count;
1305 return (linenr_T)tv_get_number_chk(&argvars[0], NULL);
1306}
1307
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001308#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001309/*
1310 * Get the float value of "argvars[0]" into "f".
1311 * Returns FAIL when the argument is not a Number or Float.
1312 */
1313 static int
1314get_float_arg(typval_T *argvars, float_T *f)
1315{
1316 if (argvars[0].v_type == VAR_FLOAT)
1317 {
1318 *f = argvars[0].vval.v_float;
1319 return OK;
1320 }
1321 if (argvars[0].v_type == VAR_NUMBER)
1322 {
1323 *f = (float_T)argvars[0].vval.v_number;
1324 return OK;
1325 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001326 emsg(_("E808: Number or Float required"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001327 return FAIL;
1328}
1329
1330/*
1331 * "abs(expr)" function
1332 */
1333 static void
1334f_abs(typval_T *argvars, typval_T *rettv)
1335{
1336 if (argvars[0].v_type == VAR_FLOAT)
1337 {
1338 rettv->v_type = VAR_FLOAT;
1339 rettv->vval.v_float = fabs(argvars[0].vval.v_float);
1340 }
1341 else
1342 {
1343 varnumber_T n;
1344 int error = FALSE;
1345
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001346 n = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001347 if (error)
1348 rettv->vval.v_number = -1;
1349 else if (n > 0)
1350 rettv->vval.v_number = n;
1351 else
1352 rettv->vval.v_number = -n;
1353 }
1354}
1355
1356/*
1357 * "acos()" function
1358 */
1359 static void
1360f_acos(typval_T *argvars, typval_T *rettv)
1361{
1362 float_T f = 0.0;
1363
1364 rettv->v_type = VAR_FLOAT;
1365 if (get_float_arg(argvars, &f) == OK)
1366 rettv->vval.v_float = acos(f);
1367 else
1368 rettv->vval.v_float = 0.0;
1369}
1370#endif
1371
1372/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001373 * "and(expr, expr)" function
1374 */
1375 static void
1376f_and(typval_T *argvars, typval_T *rettv)
1377{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001378 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
1379 & tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaarca851592018-06-06 21:04:07 +02001380}
1381
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001382#ifdef FEAT_FLOAT
1383/*
1384 * "asin()" function
1385 */
1386 static void
1387f_asin(typval_T *argvars, typval_T *rettv)
1388{
1389 float_T f = 0.0;
1390
1391 rettv->v_type = VAR_FLOAT;
1392 if (get_float_arg(argvars, &f) == OK)
1393 rettv->vval.v_float = asin(f);
1394 else
1395 rettv->vval.v_float = 0.0;
1396}
1397
1398/*
1399 * "atan()" function
1400 */
1401 static void
1402f_atan(typval_T *argvars, typval_T *rettv)
1403{
1404 float_T f = 0.0;
1405
1406 rettv->v_type = VAR_FLOAT;
1407 if (get_float_arg(argvars, &f) == OK)
1408 rettv->vval.v_float = atan(f);
1409 else
1410 rettv->vval.v_float = 0.0;
1411}
1412
1413/*
1414 * "atan2()" function
1415 */
1416 static void
1417f_atan2(typval_T *argvars, typval_T *rettv)
1418{
1419 float_T fx = 0.0, fy = 0.0;
1420
1421 rettv->v_type = VAR_FLOAT;
1422 if (get_float_arg(argvars, &fx) == OK
1423 && get_float_arg(&argvars[1], &fy) == OK)
1424 rettv->vval.v_float = atan2(fx, fy);
1425 else
1426 rettv->vval.v_float = 0.0;
1427}
1428#endif
1429
1430/*
Bram Moolenaar59716a22017-03-01 20:32:44 +01001431 * "balloon_show()" function
1432 */
1433#ifdef FEAT_BEVAL
1434 static void
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02001435f_balloon_gettext(typval_T *argvars UNUSED, typval_T *rettv)
1436{
1437 rettv->v_type = VAR_STRING;
1438 if (balloonEval != NULL)
1439 {
1440 if (balloonEval->msg == NULL)
1441 rettv->vval.v_string = NULL;
1442 else
1443 rettv->vval.v_string = vim_strsave(balloonEval->msg);
1444 }
1445}
1446
1447 static void
Bram Moolenaar59716a22017-03-01 20:32:44 +01001448f_balloon_show(typval_T *argvars, typval_T *rettv UNUSED)
1449{
Bram Moolenaarcaf64342017-03-02 22:11:33 +01001450 if (balloonEval != NULL)
Bram Moolenaar246fe032017-11-19 19:56:27 +01001451 {
1452 if (argvars[0].v_type == VAR_LIST
1453# ifdef FEAT_GUI
1454 && !gui.in_use
1455# endif
1456 )
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02001457 {
1458 list_T *l = argvars[0].vval.v_list;
1459
1460 // empty list removes the balloon
1461 post_balloon(balloonEval, NULL,
1462 l == NULL || l->lv_len == 0 ? NULL : l);
1463 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01001464 else
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02001465 {
1466 char_u *mesg = tv_get_string_chk(&argvars[0]);
1467
1468 if (mesg != NULL)
1469 // empty string removes the balloon
1470 post_balloon(balloonEval, *mesg == NUL ? NULL : mesg, NULL);
1471 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01001472 }
1473}
1474
Bram Moolenaar669a8282017-11-19 20:13:05 +01001475# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +01001476 static void
1477f_balloon_split(typval_T *argvars, typval_T *rettv UNUSED)
1478{
1479 if (rettv_list_alloc(rettv) == OK)
1480 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001481 char_u *msg = tv_get_string_chk(&argvars[0]);
Bram Moolenaar246fe032017-11-19 19:56:27 +01001482
1483 if (msg != NULL)
1484 {
1485 pumitem_T *array;
1486 int size = split_message(msg, &array);
1487 int i;
1488
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001489 // Skip the first and last item, they are always empty.
Bram Moolenaar246fe032017-11-19 19:56:27 +01001490 for (i = 1; i < size - 1; ++i)
1491 list_append_string(rettv->vval.v_list, array[i].pum_text, -1);
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01001492 while (size > 0)
1493 vim_free(array[--size].pum_text);
Bram Moolenaar246fe032017-11-19 19:56:27 +01001494 vim_free(array);
1495 }
1496 }
Bram Moolenaar59716a22017-03-01 20:32:44 +01001497}
Bram Moolenaar669a8282017-11-19 20:13:05 +01001498# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +01001499#endif
1500
1501/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001502 * Get buffer by number or pattern.
1503 */
Bram Moolenaarc6df10e2017-07-29 20:15:08 +02001504 buf_T *
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01001505tv_get_buf(typval_T *tv, int curtab_only)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001506{
1507 char_u *name = tv->vval.v_string;
1508 buf_T *buf;
1509
1510 if (tv->v_type == VAR_NUMBER)
1511 return buflist_findnr((int)tv->vval.v_number);
1512 if (tv->v_type != VAR_STRING)
1513 return NULL;
1514 if (name == NULL || *name == NUL)
1515 return curbuf;
1516 if (name[0] == '$' && name[1] == NUL)
1517 return lastbuf;
1518
1519 buf = buflist_find_by_name(name, curtab_only);
1520
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001521 // If not found, try expanding the name, like done for bufexists().
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001522 if (buf == NULL)
1523 buf = find_buffer(tv);
1524
1525 return buf;
1526}
1527
1528/*
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01001529 * Get the buffer from "arg" and give an error and return NULL if it is not
1530 * valid.
1531 */
Bram Moolenaara3347722019-05-11 21:14:24 +02001532 buf_T *
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01001533get_buf_arg(typval_T *arg)
1534{
1535 buf_T *buf;
1536
1537 ++emsg_off;
1538 buf = tv_get_buf(arg, FALSE);
1539 --emsg_off;
1540 if (buf == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001541 semsg(_("E158: Invalid buffer name: %s"), tv_get_string(arg));
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01001542 return buf;
1543}
1544
1545/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001546 * "byte2line(byte)" function
1547 */
1548 static void
1549f_byte2line(typval_T *argvars UNUSED, typval_T *rettv)
1550{
1551#ifndef FEAT_BYTEOFF
1552 rettv->vval.v_number = -1;
1553#else
1554 long boff = 0;
1555
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001556 boff = tv_get_number(&argvars[0]) - 1; // boff gets -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001557 if (boff < 0)
1558 rettv->vval.v_number = -1;
1559 else
1560 rettv->vval.v_number = ml_find_line_or_offset(curbuf,
1561 (linenr_T)0, &boff);
1562#endif
1563}
1564
1565 static void
1566byteidx(typval_T *argvars, typval_T *rettv, int comp UNUSED)
1567{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001568 char_u *t;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001569 char_u *str;
1570 varnumber_T idx;
1571
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001572 str = tv_get_string_chk(&argvars[0]);
1573 idx = tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001574 rettv->vval.v_number = -1;
1575 if (str == NULL || idx < 0)
1576 return;
1577
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001578 t = str;
1579 for ( ; idx > 0; idx--)
1580 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001581 if (*t == NUL) // EOL reached
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001582 return;
1583 if (enc_utf8 && comp)
1584 t += utf_ptr2len(t);
1585 else
1586 t += (*mb_ptr2len)(t);
1587 }
1588 rettv->vval.v_number = (varnumber_T)(t - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001589}
1590
1591/*
1592 * "byteidx()" function
1593 */
1594 static void
1595f_byteidx(typval_T *argvars, typval_T *rettv)
1596{
1597 byteidx(argvars, rettv, FALSE);
1598}
1599
1600/*
1601 * "byteidxcomp()" function
1602 */
1603 static void
1604f_byteidxcomp(typval_T *argvars, typval_T *rettv)
1605{
1606 byteidx(argvars, rettv, TRUE);
1607}
1608
1609/*
1610 * "call(func, arglist [, dict])" function
1611 */
1612 static void
1613f_call(typval_T *argvars, typval_T *rettv)
1614{
1615 char_u *func;
1616 partial_T *partial = NULL;
1617 dict_T *selfdict = NULL;
1618
1619 if (argvars[1].v_type != VAR_LIST)
1620 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001621 emsg(_(e_listreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001622 return;
1623 }
1624 if (argvars[1].vval.v_list == NULL)
1625 return;
1626
1627 if (argvars[0].v_type == VAR_FUNC)
1628 func = argvars[0].vval.v_string;
1629 else if (argvars[0].v_type == VAR_PARTIAL)
1630 {
1631 partial = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02001632 func = partial_name(partial);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001633 }
1634 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001635 func = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001636 if (*func == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001637 return; // type error or empty name
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001638
1639 if (argvars[2].v_type != VAR_UNKNOWN)
1640 {
1641 if (argvars[2].v_type != VAR_DICT)
1642 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001643 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001644 return;
1645 }
1646 selfdict = argvars[2].vval.v_dict;
1647 }
1648
1649 (void)func_call(func, &argvars[1], partial, selfdict, rettv);
1650}
1651
1652#ifdef FEAT_FLOAT
1653/*
1654 * "ceil({float})" function
1655 */
1656 static void
1657f_ceil(typval_T *argvars, typval_T *rettv)
1658{
1659 float_T f = 0.0;
1660
1661 rettv->v_type = VAR_FLOAT;
1662 if (get_float_arg(argvars, &f) == OK)
1663 rettv->vval.v_float = ceil(f);
1664 else
1665 rettv->vval.v_float = 0.0;
1666}
1667#endif
1668
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001669/*
1670 * "changenr()" function
1671 */
1672 static void
1673f_changenr(typval_T *argvars UNUSED, typval_T *rettv)
1674{
1675 rettv->vval.v_number = curbuf->b_u_seq_cur;
1676}
1677
1678/*
1679 * "char2nr(string)" function
1680 */
1681 static void
1682f_char2nr(typval_T *argvars, typval_T *rettv)
1683{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001684 if (has_mbyte)
1685 {
1686 int utf8 = 0;
1687
1688 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001689 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001690
1691 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01001692 rettv->vval.v_number = utf_ptr2char(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001693 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001694 rettv->vval.v_number = (*mb_ptr2char)(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001695 }
1696 else
Bram Moolenaar13505972019-01-24 15:04:48 +01001697 rettv->vval.v_number = tv_get_string(&argvars[0])[0];
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001698}
1699
Bram Moolenaar29b7d7a2019-07-22 23:03:57 +02001700 win_T *
Bram Moolenaaraff74912019-03-30 18:11:49 +01001701get_optional_window(typval_T *argvars, int idx)
1702{
1703 win_T *win = curwin;
1704
1705 if (argvars[idx].v_type != VAR_UNKNOWN)
1706 {
1707 win = find_win_by_nr_or_id(&argvars[idx]);
1708 if (win == NULL)
1709 {
1710 emsg(_(e_invalwindow));
1711 return NULL;
1712 }
1713 }
1714 return win;
1715}
1716
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001717/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001718 * "col(string)" function
1719 */
1720 static void
1721f_col(typval_T *argvars, typval_T *rettv)
1722{
1723 colnr_T col = 0;
1724 pos_T *fp;
1725 int fnum = curbuf->b_fnum;
1726
1727 fp = var2fpos(&argvars[0], FALSE, &fnum);
1728 if (fp != NULL && fnum == curbuf->b_fnum)
1729 {
1730 if (fp->col == MAXCOL)
1731 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001732 // '> can be MAXCOL, get the length of the line then
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001733 if (fp->lnum <= curbuf->b_ml.ml_line_count)
1734 col = (colnr_T)STRLEN(ml_get(fp->lnum)) + 1;
1735 else
1736 col = MAXCOL;
1737 }
1738 else
1739 {
1740 col = fp->col + 1;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001741 // col(".") when the cursor is on the NUL at the end of the line
1742 // because of "coladd" can be seen as an extra column.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001743 if (virtual_active() && fp == &curwin->w_cursor)
1744 {
1745 char_u *p = ml_get_cursor();
1746
1747 if (curwin->w_cursor.coladd >= (colnr_T)chartabsize(p,
1748 curwin->w_virtcol - curwin->w_cursor.coladd))
1749 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001750 int l;
1751
1752 if (*p != NUL && p[(l = (*mb_ptr2len)(p))] == NUL)
1753 col += l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001754 }
1755 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001756 }
1757 }
1758 rettv->vval.v_number = col;
1759}
1760
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001761/*
1762 * "confirm(message, buttons[, default [, type]])" function
1763 */
1764 static void
1765f_confirm(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1766{
1767#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1768 char_u *message;
1769 char_u *buttons = NULL;
1770 char_u buf[NUMBUFLEN];
1771 char_u buf2[NUMBUFLEN];
1772 int def = 1;
1773 int type = VIM_GENERIC;
1774 char_u *typestr;
1775 int error = FALSE;
1776
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001777 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001778 if (message == NULL)
1779 error = TRUE;
1780 if (argvars[1].v_type != VAR_UNKNOWN)
1781 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001782 buttons = tv_get_string_buf_chk(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001783 if (buttons == NULL)
1784 error = TRUE;
1785 if (argvars[2].v_type != VAR_UNKNOWN)
1786 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001787 def = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001788 if (argvars[3].v_type != VAR_UNKNOWN)
1789 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001790 typestr = tv_get_string_buf_chk(&argvars[3], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001791 if (typestr == NULL)
1792 error = TRUE;
1793 else
1794 {
1795 switch (TOUPPER_ASC(*typestr))
1796 {
1797 case 'E': type = VIM_ERROR; break;
1798 case 'Q': type = VIM_QUESTION; break;
1799 case 'I': type = VIM_INFO; break;
1800 case 'W': type = VIM_WARNING; break;
1801 case 'G': type = VIM_GENERIC; break;
1802 }
1803 }
1804 }
1805 }
1806 }
1807
1808 if (buttons == NULL || *buttons == NUL)
1809 buttons = (char_u *)_("&Ok");
1810
1811 if (!error)
1812 rettv->vval.v_number = do_dialog(type, NULL, message, buttons,
1813 def, NULL, FALSE);
1814#endif
1815}
1816
1817/*
1818 * "copy()" function
1819 */
1820 static void
1821f_copy(typval_T *argvars, typval_T *rettv)
1822{
1823 item_copy(&argvars[0], rettv, FALSE, 0);
1824}
1825
1826#ifdef FEAT_FLOAT
1827/*
1828 * "cos()" function
1829 */
1830 static void
1831f_cos(typval_T *argvars, typval_T *rettv)
1832{
1833 float_T f = 0.0;
1834
1835 rettv->v_type = VAR_FLOAT;
1836 if (get_float_arg(argvars, &f) == OK)
1837 rettv->vval.v_float = cos(f);
1838 else
1839 rettv->vval.v_float = 0.0;
1840}
1841
1842/*
1843 * "cosh()" function
1844 */
1845 static void
1846f_cosh(typval_T *argvars, typval_T *rettv)
1847{
1848 float_T f = 0.0;
1849
1850 rettv->v_type = VAR_FLOAT;
1851 if (get_float_arg(argvars, &f) == OK)
1852 rettv->vval.v_float = cosh(f);
1853 else
1854 rettv->vval.v_float = 0.0;
1855}
1856#endif
1857
1858/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001859 * "cursor(lnum, col)" function, or
1860 * "cursor(list)"
1861 *
1862 * Moves the cursor to the specified line and column.
1863 * Returns 0 when the position could be set, -1 otherwise.
1864 */
1865 static void
1866f_cursor(typval_T *argvars, typval_T *rettv)
1867{
1868 long line, col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001869 long coladd = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001870 int set_curswant = TRUE;
1871
1872 rettv->vval.v_number = -1;
1873 if (argvars[1].v_type == VAR_UNKNOWN)
1874 {
1875 pos_T pos;
1876 colnr_T curswant = -1;
1877
1878 if (list2fpos(argvars, &pos, NULL, &curswant) == FAIL)
1879 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001880 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001881 return;
1882 }
1883 line = pos.lnum;
1884 col = pos.col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001885 coladd = pos.coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001886 if (curswant >= 0)
1887 {
1888 curwin->w_curswant = curswant - 1;
1889 set_curswant = FALSE;
1890 }
1891 }
1892 else
1893 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001894 line = tv_get_lnum(argvars);
1895 col = (long)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001896 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001897 coladd = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001898 }
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01001899 if (line < 0 || col < 0 || coladd < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001900 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001901 if (line > 0)
1902 curwin->w_cursor.lnum = line;
1903 if (col > 0)
1904 curwin->w_cursor.col = col - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001905 curwin->w_cursor.coladd = coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001906
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001907 // Make sure the cursor is in a valid position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001908 check_cursor();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001909 // Correct cursor for multi-byte character.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001910 if (has_mbyte)
1911 mb_adjust_cursor();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001912
1913 curwin->w_set_curswant = set_curswant;
1914 rettv->vval.v_number = 0;
1915}
1916
Bram Moolenaar4f974752019-02-17 17:44:42 +01001917#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02001918/*
1919 * "debugbreak()" function
1920 */
1921 static void
1922f_debugbreak(typval_T *argvars, typval_T *rettv)
1923{
1924 int pid;
1925
1926 rettv->vval.v_number = FAIL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001927 pid = (int)tv_get_number(&argvars[0]);
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02001928 if (pid == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001929 emsg(_(e_invarg));
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02001930 else
1931 {
1932 HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, 0, pid);
1933
1934 if (hProcess != NULL)
1935 {
1936 DebugBreakProcess(hProcess);
1937 CloseHandle(hProcess);
1938 rettv->vval.v_number = OK;
1939 }
1940 }
1941}
1942#endif
1943
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001944/*
1945 * "deepcopy()" function
1946 */
1947 static void
1948f_deepcopy(typval_T *argvars, typval_T *rettv)
1949{
1950 int noref = 0;
1951 int copyID;
1952
1953 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001954 noref = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001955 if (noref < 0 || noref > 1)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001956 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001957 else
1958 {
1959 copyID = get_copyID();
1960 item_copy(&argvars[0], rettv, TRUE, noref == 0 ? copyID : 0);
1961 }
1962}
1963
1964/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001965 * "did_filetype()" function
1966 */
1967 static void
1968f_did_filetype(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1969{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001970 rettv->vval.v_number = did_filetype;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001971}
1972
1973/*
Bram Moolenaar4132eb52020-02-14 16:53:00 +01001974 * "echoraw({expr})" function
1975 */
1976 static void
1977f_echoraw(typval_T *argvars, typval_T *rettv UNUSED)
1978{
1979 char_u *str = tv_get_string_chk(&argvars[0]);
1980
1981 if (str != NULL && *str != NUL)
1982 {
1983 out_str(str);
1984 out_flush();
1985 }
1986}
1987
1988/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001989 * "empty({expr})" function
1990 */
1991 static void
1992f_empty(typval_T *argvars, typval_T *rettv)
1993{
1994 int n = FALSE;
1995
1996 switch (argvars[0].v_type)
1997 {
1998 case VAR_STRING:
1999 case VAR_FUNC:
2000 n = argvars[0].vval.v_string == NULL
2001 || *argvars[0].vval.v_string == NUL;
2002 break;
2003 case VAR_PARTIAL:
2004 n = FALSE;
2005 break;
2006 case VAR_NUMBER:
2007 n = argvars[0].vval.v_number == 0;
2008 break;
2009 case VAR_FLOAT:
2010#ifdef FEAT_FLOAT
2011 n = argvars[0].vval.v_float == 0.0;
2012 break;
2013#endif
2014 case VAR_LIST:
2015 n = argvars[0].vval.v_list == NULL
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002016 || argvars[0].vval.v_list->lv_len == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002017 break;
2018 case VAR_DICT:
2019 n = argvars[0].vval.v_dict == NULL
2020 || argvars[0].vval.v_dict->dv_hashtab.ht_used == 0;
2021 break;
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002022 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002023 case VAR_SPECIAL:
2024 n = argvars[0].vval.v_number != VVAL_TRUE;
2025 break;
2026
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002027 case VAR_BLOB:
2028 n = argvars[0].vval.v_blob == NULL
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002029 || argvars[0].vval.v_blob->bv_ga.ga_len == 0;
2030 break;
2031
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002032 case VAR_JOB:
2033#ifdef FEAT_JOB_CHANNEL
2034 n = argvars[0].vval.v_job == NULL
2035 || argvars[0].vval.v_job->jv_status != JOB_STARTED;
2036 break;
2037#endif
2038 case VAR_CHANNEL:
2039#ifdef FEAT_JOB_CHANNEL
2040 n = argvars[0].vval.v_channel == NULL
2041 || !channel_is_open(argvars[0].vval.v_channel);
2042 break;
2043#endif
2044 case VAR_UNKNOWN:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002045 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01002046 internal_error_no_abort("f_empty(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002047 n = TRUE;
2048 break;
2049 }
2050
2051 rettv->vval.v_number = n;
2052}
2053
2054/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02002055 * "environ()" function
2056 */
2057 static void
2058f_environ(typval_T *argvars UNUSED, typval_T *rettv)
2059{
2060#if !defined(AMIGA)
2061 int i = 0;
2062 char_u *entry, *value;
2063# ifdef MSWIN
2064 extern wchar_t **_wenviron;
2065# else
2066 extern char **environ;
2067# endif
2068
2069 if (rettv_dict_alloc(rettv) != OK)
2070 return;
2071
2072# ifdef MSWIN
2073 if (*_wenviron == NULL)
2074 return;
2075# else
2076 if (*environ == NULL)
2077 return;
2078# endif
2079
2080 for (i = 0; ; ++i)
2081 {
2082# ifdef MSWIN
2083 short_u *p;
2084
2085 if ((p = (short_u *)_wenviron[i]) == NULL)
2086 return;
2087 entry = utf16_to_enc(p, NULL);
2088# else
2089 if ((entry = (char_u *)environ[i]) == NULL)
2090 return;
2091 entry = vim_strsave(entry);
2092# endif
2093 if (entry == NULL) // out of memory
2094 return;
2095 if ((value = vim_strchr(entry, '=')) == NULL)
2096 {
2097 vim_free(entry);
2098 continue;
2099 }
2100 *value++ = NUL;
2101 dict_add_string(rettv->vval.v_dict, (char *)entry, value);
2102 vim_free(entry);
2103 }
2104#endif
2105}
2106
2107/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002108 * "escape({string}, {chars})" function
2109 */
2110 static void
2111f_escape(typval_T *argvars, typval_T *rettv)
2112{
2113 char_u buf[NUMBUFLEN];
2114
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002115 rettv->vval.v_string = vim_strsave_escaped(tv_get_string(&argvars[0]),
2116 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002117 rettv->v_type = VAR_STRING;
2118}
2119
2120/*
2121 * "eval()" function
2122 */
2123 static void
2124f_eval(typval_T *argvars, typval_T *rettv)
2125{
2126 char_u *s, *p;
2127
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002128 s = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002129 if (s != NULL)
2130 s = skipwhite(s);
2131
2132 p = s;
2133 if (s == NULL || eval1(&s, rettv, TRUE) == FAIL)
2134 {
2135 if (p != NULL && !aborting())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002136 semsg(_(e_invexpr2), p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002137 need_clr_eos = FALSE;
2138 rettv->v_type = VAR_NUMBER;
2139 rettv->vval.v_number = 0;
2140 }
2141 else if (*s != NUL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002142 emsg(_(e_trailing));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002143}
2144
2145/*
2146 * "eventhandler()" function
2147 */
2148 static void
2149f_eventhandler(typval_T *argvars UNUSED, typval_T *rettv)
2150{
2151 rettv->vval.v_number = vgetc_busy;
2152}
2153
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002154static garray_T redir_execute_ga;
2155
2156/*
2157 * Append "value[value_len]" to the execute() output.
2158 */
2159 void
2160execute_redir_str(char_u *value, int value_len)
2161{
2162 int len;
2163
2164 if (value_len == -1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002165 len = (int)STRLEN(value); // Append the entire string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002166 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002167 len = value_len; // Append only "value_len" characters
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002168 if (ga_grow(&redir_execute_ga, len) == OK)
2169 {
2170 mch_memmove((char *)redir_execute_ga.ga_data
2171 + redir_execute_ga.ga_len, value, len);
2172 redir_execute_ga.ga_len += len;
2173 }
2174}
2175
2176/*
2177 * Get next line from a list.
2178 * Called by do_cmdline() to get the next line.
2179 * Returns allocated string, or NULL for end of function.
2180 */
2181
2182 static char_u *
2183get_list_line(
2184 int c UNUSED,
2185 void *cookie,
Bram Moolenaare96a2492019-06-25 04:12:16 +02002186 int indent UNUSED,
2187 int do_concat UNUSED)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002188{
2189 listitem_T **p = (listitem_T **)cookie;
2190 listitem_T *item = *p;
2191 char_u buf[NUMBUFLEN];
2192 char_u *s;
2193
2194 if (item == NULL)
2195 return NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002196 s = tv_get_string_buf_chk(&item->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002197 *p = item->li_next;
2198 return s == NULL ? NULL : vim_strsave(s);
2199}
2200
2201/*
2202 * "execute()" function
2203 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02002204 void
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002205execute_common(typval_T *argvars, typval_T *rettv, int arg_off)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002206{
2207 char_u *cmd = NULL;
2208 list_T *list = NULL;
2209 int save_msg_silent = msg_silent;
2210 int save_emsg_silent = emsg_silent;
2211 int save_emsg_noredir = emsg_noredir;
2212 int save_redir_execute = redir_execute;
Bram Moolenaar20951482017-12-25 13:44:43 +01002213 int save_redir_off = redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002214 garray_T save_ga;
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002215 int save_msg_col = msg_col;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002216 int echo_output = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002217
2218 rettv->vval.v_string = NULL;
2219 rettv->v_type = VAR_STRING;
2220
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002221 if (argvars[arg_off].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002222 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002223 list = argvars[arg_off].vval.v_list;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002224 if (list == NULL || list->lv_len == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002225 // empty list, no commands, empty output
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002226 return;
2227 ++list->lv_refcount;
2228 }
Bram Moolenaare2a8f072020-01-08 19:32:18 +01002229 else if (argvars[arg_off].v_type == VAR_JOB
2230 || argvars[arg_off].v_type == VAR_CHANNEL)
2231 {
2232 emsg(_(e_inval_string));
2233 return;
2234 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002235 else
2236 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002237 cmd = tv_get_string_chk(&argvars[arg_off]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002238 if (cmd == NULL)
2239 return;
2240 }
2241
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002242 if (argvars[arg_off + 1].v_type != VAR_UNKNOWN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002243 {
2244 char_u buf[NUMBUFLEN];
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002245 char_u *s = tv_get_string_buf_chk(&argvars[arg_off + 1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002246
2247 if (s == NULL)
2248 return;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002249 if (*s == NUL)
2250 echo_output = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002251 if (STRNCMP(s, "silent", 6) == 0)
2252 ++msg_silent;
2253 if (STRCMP(s, "silent!") == 0)
2254 {
2255 emsg_silent = TRUE;
2256 emsg_noredir = TRUE;
2257 }
2258 }
2259 else
2260 ++msg_silent;
2261
2262 if (redir_execute)
2263 save_ga = redir_execute_ga;
2264 ga_init2(&redir_execute_ga, (int)sizeof(char), 500);
2265 redir_execute = TRUE;
Bram Moolenaar20951482017-12-25 13:44:43 +01002266 redir_off = FALSE;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002267 if (!echo_output)
2268 msg_col = 0; // prevent leading spaces
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002269
2270 if (cmd != NULL)
2271 do_cmdline_cmd(cmd);
2272 else
2273 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002274 listitem_T *item;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002275
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002276 range_list_materialize(list);
2277 item = list->lv_first;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002278 do_cmdline(NULL, get_list_line, (void *)&item,
2279 DOCMD_NOWAIT|DOCMD_VERBOSE|DOCMD_REPEAT|DOCMD_KEYTYPED);
2280 --list->lv_refcount;
2281 }
2282
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002283 // Need to append a NUL to the result.
Bram Moolenaard297f352017-01-29 20:31:21 +01002284 if (ga_grow(&redir_execute_ga, 1) == OK)
2285 {
2286 ((char *)redir_execute_ga.ga_data)[redir_execute_ga.ga_len] = NUL;
2287 rettv->vval.v_string = redir_execute_ga.ga_data;
2288 }
2289 else
2290 {
2291 ga_clear(&redir_execute_ga);
2292 rettv->vval.v_string = NULL;
2293 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002294 msg_silent = save_msg_silent;
2295 emsg_silent = save_emsg_silent;
2296 emsg_noredir = save_emsg_noredir;
2297
2298 redir_execute = save_redir_execute;
2299 if (redir_execute)
2300 redir_execute_ga = save_ga;
Bram Moolenaar20951482017-12-25 13:44:43 +01002301 redir_off = save_redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002302
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002303 // "silent reg" or "silent echo x" leaves msg_col somewhere in the line.
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002304 if (echo_output)
2305 // When not working silently: put it in column zero. A following
2306 // "echon" will overwrite the message, unavoidably.
2307 msg_col = 0;
2308 else
2309 // When working silently: Put it back where it was, since nothing
2310 // should have been written.
2311 msg_col = save_msg_col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002312}
2313
2314/*
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002315 * "execute()" function
2316 */
2317 static void
2318f_execute(typval_T *argvars, typval_T *rettv)
2319{
2320 execute_common(argvars, rettv, 0);
2321}
2322
2323/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002324 * "exists()" function
2325 */
2326 static void
2327f_exists(typval_T *argvars, typval_T *rettv)
2328{
2329 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002330 int n = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002331
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002332 p = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002333 if (*p == '$') // environment variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002334 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002335 // first try "normal" environment variables (fast)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002336 if (mch_getenv(p + 1) != NULL)
2337 n = TRUE;
2338 else
2339 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002340 // try expanding things like $VIM and ${HOME}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002341 p = expand_env_save(p);
2342 if (p != NULL && *p != '$')
2343 n = TRUE;
2344 vim_free(p);
2345 }
2346 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002347 else if (*p == '&' || *p == '+') // option
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002348 {
2349 n = (get_option_tv(&p, NULL, TRUE) == OK);
2350 if (*skipwhite(p) != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002351 n = FALSE; // trailing garbage
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002352 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002353 else if (*p == '*') // internal or user defined function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002354 {
Bram Moolenaarb54c3ff2016-07-31 14:11:58 +02002355 n = function_exists(p + 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002356 }
Bram Moolenaar15c47602020-03-26 22:16:48 +01002357 else if (*p == '?') // internal function only
2358 {
2359 n = has_internal_func_name(p + 1);
2360 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002361 else if (*p == ':')
2362 {
2363 n = cmd_exists(p + 1);
2364 }
2365 else if (*p == '#')
2366 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002367 if (p[1] == '#')
2368 n = autocmd_supported(p + 2);
2369 else
2370 n = au_exists(p + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002371 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002372 else // internal variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002373 {
Bram Moolenaarc6f9f732018-02-11 19:06:26 +01002374 n = var_exists(p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002375 }
2376
2377 rettv->vval.v_number = n;
2378}
2379
2380#ifdef FEAT_FLOAT
2381/*
2382 * "exp()" function
2383 */
2384 static void
2385f_exp(typval_T *argvars, typval_T *rettv)
2386{
2387 float_T f = 0.0;
2388
2389 rettv->v_type = VAR_FLOAT;
2390 if (get_float_arg(argvars, &f) == OK)
2391 rettv->vval.v_float = exp(f);
2392 else
2393 rettv->vval.v_float = 0.0;
2394}
2395#endif
2396
2397/*
2398 * "expand()" function
2399 */
2400 static void
2401f_expand(typval_T *argvars, typval_T *rettv)
2402{
2403 char_u *s;
2404 int len;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002405 char *errormsg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002406 int options = WILD_SILENT|WILD_USE_NL|WILD_LIST_NOTFOUND;
2407 expand_T xpc;
2408 int error = FALSE;
2409 char_u *result;
2410
2411 rettv->v_type = VAR_STRING;
2412 if (argvars[1].v_type != VAR_UNKNOWN
2413 && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002414 && tv_get_number_chk(&argvars[2], &error)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002415 && !error)
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02002416 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002417
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002418 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002419 if (*s == '%' || *s == '#' || *s == '<')
2420 {
2421 ++emsg_off;
2422 result = eval_vars(s, s, &len, NULL, &errormsg, NULL);
2423 --emsg_off;
2424 if (rettv->v_type == VAR_LIST)
2425 {
2426 if (rettv_list_alloc(rettv) != FAIL && result != NULL)
2427 list_append_string(rettv->vval.v_list, result, -1);
Bram Moolenaar86173482019-10-01 17:02:16 +02002428 vim_free(result);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002429 }
2430 else
2431 rettv->vval.v_string = result;
2432 }
2433 else
2434 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002435 // When the optional second argument is non-zero, don't remove matches
2436 // for 'wildignore' and don't put matches for 'suffixes' at the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002437 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002438 && tv_get_number_chk(&argvars[1], &error))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002439 options |= WILD_KEEP_ALL;
2440 if (!error)
2441 {
2442 ExpandInit(&xpc);
2443 xpc.xp_context = EXPAND_FILES;
2444 if (p_wic)
2445 options += WILD_ICASE;
2446 if (rettv->v_type == VAR_STRING)
2447 rettv->vval.v_string = ExpandOne(&xpc, s, NULL,
2448 options, WILD_ALL);
2449 else if (rettv_list_alloc(rettv) != FAIL)
2450 {
2451 int i;
2452
2453 ExpandOne(&xpc, s, NULL, options, WILD_ALL_KEEP);
2454 for (i = 0; i < xpc.xp_numfiles; i++)
2455 list_append_string(rettv->vval.v_list, xpc.xp_files[i], -1);
2456 ExpandCleanup(&xpc);
2457 }
2458 }
2459 else
2460 rettv->vval.v_string = NULL;
2461 }
2462}
2463
2464/*
Bram Moolenaar80dad482019-06-09 17:22:31 +02002465 * "expandcmd()" function
2466 * Expand all the special characters in a command string.
2467 */
2468 static void
2469f_expandcmd(typval_T *argvars, typval_T *rettv)
2470{
2471 exarg_T eap;
2472 char_u *cmdstr;
2473 char *errormsg = NULL;
2474
2475 rettv->v_type = VAR_STRING;
2476 cmdstr = vim_strsave(tv_get_string(&argvars[0]));
2477
2478 memset(&eap, 0, sizeof(eap));
2479 eap.cmd = cmdstr;
2480 eap.arg = cmdstr;
Bram Moolenaar8071cb22019-07-12 17:58:01 +02002481 eap.argt |= EX_NOSPC;
Bram Moolenaar80dad482019-06-09 17:22:31 +02002482 eap.usefilter = FALSE;
2483 eap.nextcmd = NULL;
2484 eap.cmdidx = CMD_USER;
2485
2486 expand_filename(&eap, &cmdstr, &errormsg);
2487 if (errormsg != NULL && *errormsg != NUL)
2488 emsg(errormsg);
2489
2490 rettv->vval.v_string = cmdstr;
2491}
2492
2493/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002494 * "feedkeys()" function
2495 */
2496 static void
2497f_feedkeys(typval_T *argvars, typval_T *rettv UNUSED)
2498{
2499 int remap = TRUE;
2500 int insert = FALSE;
2501 char_u *keys, *flags;
2502 char_u nbuf[NUMBUFLEN];
2503 int typed = FALSE;
2504 int execute = FALSE;
2505 int dangerous = FALSE;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002506 int lowlevel = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002507 char_u *keys_esc;
2508
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002509 // This is not allowed in the sandbox. If the commands would still be
2510 // executed in the sandbox it would be OK, but it probably happens later,
2511 // when "sandbox" is no longer set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002512 if (check_secure())
2513 return;
2514
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002515 keys = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002516
2517 if (argvars[1].v_type != VAR_UNKNOWN)
2518 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002519 flags = tv_get_string_buf(&argvars[1], nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002520 for ( ; *flags != NUL; ++flags)
2521 {
2522 switch (*flags)
2523 {
2524 case 'n': remap = FALSE; break;
2525 case 'm': remap = TRUE; break;
2526 case 't': typed = TRUE; break;
2527 case 'i': insert = TRUE; break;
2528 case 'x': execute = TRUE; break;
2529 case '!': dangerous = TRUE; break;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002530 case 'L': lowlevel = TRUE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002531 }
2532 }
2533 }
2534
2535 if (*keys != NUL || execute)
2536 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002537 // Need to escape K_SPECIAL and CSI before putting the string in the
2538 // typeahead buffer.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002539 keys_esc = vim_strsave_escape_csi(keys);
2540 if (keys_esc != NULL)
2541 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002542 if (lowlevel)
2543 {
2544#ifdef USE_INPUT_BUF
Bram Moolenaar9645e2d2020-03-20 20:48:49 +01002545 int idx;
2546 int len = (int)STRLEN(keys);
2547
2548 for (idx = 0; idx < len; ++idx)
2549 {
2550 // if a CTRL-C was typed, set got_int, similar to what
2551 // happens in fill_input_buf()
2552 if (keys[idx] == 3 && ctrl_c_interrupts && typed)
2553 got_int = TRUE;
2554 add_to_input_buf(keys + idx, 1);
2555 }
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002556#else
2557 emsg(_("E980: lowlevel input not supported"));
2558#endif
2559 }
2560 else
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002561 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002562 ins_typebuf(keys_esc, (remap ? REMAP_YES : REMAP_NONE),
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002563 insert ? 0 : typebuf.tb_len, !typed, FALSE);
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002564 if (vgetc_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02002565#ifdef FEAT_TIMERS
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002566 || timer_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02002567#endif
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002568 )
2569 typebuf_was_filled = TRUE;
2570 }
2571 vim_free(keys_esc);
2572
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002573 if (execute)
2574 {
2575 int save_msg_scroll = msg_scroll;
2576
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002577 // Avoid a 1 second delay when the keys start Insert mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002578 msg_scroll = FALSE;
2579
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02002580 if (!dangerous)
2581 ++ex_normal_busy;
Bram Moolenaar905dd902019-04-07 14:21:47 +02002582 exec_normal(TRUE, lowlevel, TRUE);
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02002583 if (!dangerous)
2584 --ex_normal_busy;
2585
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002586 msg_scroll |= save_msg_scroll;
2587 }
2588 }
2589 }
2590}
2591
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002592#ifdef FEAT_FLOAT
2593/*
2594 * "float2nr({float})" function
2595 */
2596 static void
2597f_float2nr(typval_T *argvars, typval_T *rettv)
2598{
2599 float_T f = 0.0;
2600
2601 if (get_float_arg(argvars, &f) == OK)
2602 {
Bram Moolenaar863e80b2017-06-04 20:30:00 +02002603 if (f <= -VARNUM_MAX + DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002604 rettv->vval.v_number = -VARNUM_MAX;
Bram Moolenaar863e80b2017-06-04 20:30:00 +02002605 else if (f >= VARNUM_MAX - DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002606 rettv->vval.v_number = VARNUM_MAX;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002607 else
2608 rettv->vval.v_number = (varnumber_T)f;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002609 }
2610}
2611
2612/*
2613 * "floor({float})" function
2614 */
2615 static void
2616f_floor(typval_T *argvars, typval_T *rettv)
2617{
2618 float_T f = 0.0;
2619
2620 rettv->v_type = VAR_FLOAT;
2621 if (get_float_arg(argvars, &f) == OK)
2622 rettv->vval.v_float = floor(f);
2623 else
2624 rettv->vval.v_float = 0.0;
2625}
2626
2627/*
2628 * "fmod()" function
2629 */
2630 static void
2631f_fmod(typval_T *argvars, typval_T *rettv)
2632{
2633 float_T fx = 0.0, fy = 0.0;
2634
2635 rettv->v_type = VAR_FLOAT;
2636 if (get_float_arg(argvars, &fx) == OK
2637 && get_float_arg(&argvars[1], &fy) == OK)
2638 rettv->vval.v_float = fmod(fx, fy);
2639 else
2640 rettv->vval.v_float = 0.0;
2641}
2642#endif
2643
2644/*
2645 * "fnameescape({string})" function
2646 */
2647 static void
2648f_fnameescape(typval_T *argvars, typval_T *rettv)
2649{
2650 rettv->vval.v_string = vim_strsave_fnameescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002651 tv_get_string(&argvars[0]), FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002652 rettv->v_type = VAR_STRING;
2653}
2654
2655/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002656 * "foreground()" function
2657 */
2658 static void
2659f_foreground(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2660{
2661#ifdef FEAT_GUI
2662 if (gui.in_use)
Bram Moolenaarafde13b2019-04-28 19:46:49 +02002663 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002664 gui_mch_set_foreground();
Bram Moolenaarafde13b2019-04-28 19:46:49 +02002665 return;
2666 }
2667#endif
2668#if defined(MSWIN) && (!defined(FEAT_GUI) || defined(VIMDLL))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002669 win32_set_foreground();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002670#endif
2671}
2672
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002673 static void
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002674common_function(typval_T *argvars, typval_T *rettv, int is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002675{
2676 char_u *s;
2677 char_u *name;
2678 int use_string = FALSE;
2679 partial_T *arg_pt = NULL;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002680 char_u *trans_name = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002681
2682 if (argvars[0].v_type == VAR_FUNC)
2683 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002684 // function(MyFunc, [arg], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002685 s = argvars[0].vval.v_string;
2686 }
2687 else if (argvars[0].v_type == VAR_PARTIAL
2688 && argvars[0].vval.v_partial != NULL)
2689 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002690 // function(dict.MyFunc, [arg])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002691 arg_pt = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002692 s = partial_name(arg_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002693 }
2694 else
2695 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002696 // function('MyFunc', [arg], dict)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002697 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002698 use_string = TRUE;
2699 }
2700
Bram Moolenaar843b8842016-08-21 14:36:15 +02002701 if ((use_string && vim_strchr(s, AUTOLOAD_CHAR) == NULL) || is_funcref)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002702 {
2703 name = s;
2704 trans_name = trans_function_name(&name, FALSE,
2705 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF, NULL, NULL);
2706 if (*name != NUL)
2707 s = NULL;
2708 }
2709
Bram Moolenaar843b8842016-08-21 14:36:15 +02002710 if (s == NULL || *s == NUL || (use_string && VIM_ISDIGIT(*s))
2711 || (is_funcref && trans_name == NULL))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002712 semsg(_(e_invarg2), use_string ? tv_get_string(&argvars[0]) : s);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002713 // Don't check an autoload name for existence here.
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002714 else if (trans_name != NULL && (is_funcref
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002715 ? find_func(trans_name, NULL) == NULL
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002716 : !translated_function_exists(trans_name)))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002717 semsg(_("E700: Unknown function: %s"), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002718 else
2719 {
2720 int dict_idx = 0;
2721 int arg_idx = 0;
2722 list_T *list = NULL;
2723
2724 if (STRNCMP(s, "s:", 2) == 0 || STRNCMP(s, "<SID>", 5) == 0)
2725 {
2726 char sid_buf[25];
2727 int off = *s == 's' ? 2 : 5;
2728
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002729 // Expand s: and <SID> into <SNR>nr_, so that the function can
2730 // also be called from another script. Using trans_function_name()
2731 // would also work, but some plugins depend on the name being
2732 // printable text.
Bram Moolenaarf29c1c62018-09-10 21:05:02 +02002733 sprintf(sid_buf, "<SNR>%ld_", (long)current_sctx.sc_sid);
Bram Moolenaar51e14382019-05-25 20:21:28 +02002734 name = alloc(STRLEN(sid_buf) + STRLEN(s + off) + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002735 if (name != NULL)
2736 {
2737 STRCPY(name, sid_buf);
2738 STRCAT(name, s + off);
2739 }
2740 }
2741 else
2742 name = vim_strsave(s);
2743
2744 if (argvars[1].v_type != VAR_UNKNOWN)
2745 {
2746 if (argvars[2].v_type != VAR_UNKNOWN)
2747 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002748 // function(name, [args], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002749 arg_idx = 1;
2750 dict_idx = 2;
2751 }
2752 else if (argvars[1].v_type == VAR_DICT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002753 // function(name, dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002754 dict_idx = 1;
2755 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002756 // function(name, [args])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002757 arg_idx = 1;
2758 if (dict_idx > 0)
2759 {
2760 if (argvars[dict_idx].v_type != VAR_DICT)
2761 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002762 emsg(_("E922: expected a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002763 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002764 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002765 }
2766 if (argvars[dict_idx].vval.v_dict == NULL)
2767 dict_idx = 0;
2768 }
2769 if (arg_idx > 0)
2770 {
2771 if (argvars[arg_idx].v_type != VAR_LIST)
2772 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002773 emsg(_("E923: Second argument of function() must be a list or a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002774 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002775 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002776 }
2777 list = argvars[arg_idx].vval.v_list;
2778 if (list == NULL || list->lv_len == 0)
2779 arg_idx = 0;
Bram Moolenaar4c054e92019-11-10 00:13:50 +01002780 else if (list->lv_len > MAX_FUNC_ARGS)
2781 {
Bram Moolenaar2118a302019-11-22 19:29:45 +01002782 emsg_funcname((char *)e_toomanyarg, s);
Bram Moolenaar4c054e92019-11-10 00:13:50 +01002783 vim_free(name);
2784 goto theend;
2785 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002786 }
2787 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002788 if (dict_idx > 0 || arg_idx > 0 || arg_pt != NULL || is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002789 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02002790 partial_T *pt = ALLOC_CLEAR_ONE(partial_T);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002791
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002792 // result is a VAR_PARTIAL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002793 if (pt == NULL)
2794 vim_free(name);
2795 else
2796 {
2797 if (arg_idx > 0 || (arg_pt != NULL && arg_pt->pt_argc > 0))
2798 {
2799 listitem_T *li;
2800 int i = 0;
2801 int arg_len = 0;
2802 int lv_len = 0;
2803
2804 if (arg_pt != NULL)
2805 arg_len = arg_pt->pt_argc;
2806 if (list != NULL)
2807 lv_len = list->lv_len;
2808 pt->pt_argc = arg_len + lv_len;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02002809 pt->pt_argv = ALLOC_MULT(typval_T, pt->pt_argc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002810 if (pt->pt_argv == NULL)
2811 {
2812 vim_free(pt);
2813 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002814 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002815 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002816 for (i = 0; i < arg_len; i++)
2817 copy_tv(&arg_pt->pt_argv[i], &pt->pt_argv[i]);
2818 if (lv_len > 0)
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002819 {
2820 range_list_materialize(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002821 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002822 copy_tv(&li->li_tv, &pt->pt_argv[i++]);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002823 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002824 }
2825
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002826 // For "function(dict.func, [], dict)" and "func" is a partial
2827 // use "dict". That is backwards compatible.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002828 if (dict_idx > 0)
2829 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002830 // The dict is bound explicitly, pt_auto is FALSE.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002831 pt->pt_dict = argvars[dict_idx].vval.v_dict;
2832 ++pt->pt_dict->dv_refcount;
2833 }
2834 else if (arg_pt != NULL)
2835 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002836 // If the dict was bound automatically the result is also
2837 // bound automatically.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002838 pt->pt_dict = arg_pt->pt_dict;
2839 pt->pt_auto = arg_pt->pt_auto;
2840 if (pt->pt_dict != NULL)
2841 ++pt->pt_dict->dv_refcount;
2842 }
2843
2844 pt->pt_refcount = 1;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002845 if (arg_pt != NULL && arg_pt->pt_func != NULL)
2846 {
2847 pt->pt_func = arg_pt->pt_func;
2848 func_ptr_ref(pt->pt_func);
2849 vim_free(name);
2850 }
2851 else if (is_funcref)
2852 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002853 pt->pt_func = find_func(trans_name, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002854 func_ptr_ref(pt->pt_func);
2855 vim_free(name);
2856 }
2857 else
2858 {
2859 pt->pt_name = name;
2860 func_ref(name);
2861 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002862 }
2863 rettv->v_type = VAR_PARTIAL;
2864 rettv->vval.v_partial = pt;
2865 }
2866 else
2867 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002868 // result is a VAR_FUNC
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002869 rettv->v_type = VAR_FUNC;
2870 rettv->vval.v_string = name;
2871 func_ref(name);
2872 }
2873 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002874theend:
2875 vim_free(trans_name);
2876}
2877
2878/*
2879 * "funcref()" function
2880 */
2881 static void
2882f_funcref(typval_T *argvars, typval_T *rettv)
2883{
2884 common_function(argvars, rettv, TRUE);
2885}
2886
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002887 static type_T *
2888ret_f_function(int argcount, type_T **argtypes UNUSED)
2889{
2890 if (argcount == 1 && argtypes[0]->tt_type == VAR_STRING)
2891 return &t_func_any;
Bram Moolenaard77a8522020-04-03 21:59:57 +02002892 return &t_func_void;
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002893}
2894
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002895/*
2896 * "function()" function
2897 */
2898 static void
2899f_function(typval_T *argvars, typval_T *rettv)
2900{
2901 common_function(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002902}
2903
2904/*
2905 * "garbagecollect()" function
2906 */
2907 static void
2908f_garbagecollect(typval_T *argvars, typval_T *rettv UNUSED)
2909{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002910 // This is postponed until we are back at the toplevel, because we may be
2911 // using Lists and Dicts internally. E.g.: ":echo [garbagecollect()]".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002912 want_garbage_collect = TRUE;
2913
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002914 if (argvars[0].v_type != VAR_UNKNOWN && tv_get_number(&argvars[0]) == 1)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002915 garbage_collect_at_exit = TRUE;
2916}
2917
2918/*
2919 * "get()" function
2920 */
2921 static void
2922f_get(typval_T *argvars, typval_T *rettv)
2923{
2924 listitem_T *li;
2925 list_T *l;
2926 dictitem_T *di;
2927 dict_T *d;
2928 typval_T *tv = NULL;
Bram Moolenaarf91aac52019-07-28 13:21:01 +02002929 int what_is_dict = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002930
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002931 if (argvars[0].v_type == VAR_BLOB)
2932 {
2933 int error = FALSE;
2934 int idx = tv_get_number_chk(&argvars[1], &error);
2935
2936 if (!error)
2937 {
2938 rettv->v_type = VAR_NUMBER;
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002939 if (idx < 0)
2940 idx = blob_len(argvars[0].vval.v_blob) + idx;
2941 if (idx < 0 || idx >= blob_len(argvars[0].vval.v_blob))
2942 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002943 else
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002944 {
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002945 rettv->vval.v_number = blob_get(argvars[0].vval.v_blob, idx);
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002946 tv = rettv;
2947 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002948 }
2949 }
2950 else if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002951 {
2952 if ((l = argvars[0].vval.v_list) != NULL)
2953 {
2954 int error = FALSE;
2955
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002956 li = list_find(l, (long)tv_get_number_chk(&argvars[1], &error));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002957 if (!error && li != NULL)
2958 tv = &li->li_tv;
2959 }
2960 }
2961 else if (argvars[0].v_type == VAR_DICT)
2962 {
2963 if ((d = argvars[0].vval.v_dict) != NULL)
2964 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002965 di = dict_find(d, tv_get_string(&argvars[1]), -1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002966 if (di != NULL)
2967 tv = &di->di_tv;
2968 }
2969 }
2970 else if (argvars[0].v_type == VAR_PARTIAL || argvars[0].v_type == VAR_FUNC)
2971 {
2972 partial_T *pt;
2973 partial_T fref_pt;
2974
2975 if (argvars[0].v_type == VAR_PARTIAL)
2976 pt = argvars[0].vval.v_partial;
2977 else
2978 {
2979 vim_memset(&fref_pt, 0, sizeof(fref_pt));
2980 fref_pt.pt_name = argvars[0].vval.v_string;
2981 pt = &fref_pt;
2982 }
2983
2984 if (pt != NULL)
2985 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002986 char_u *what = tv_get_string(&argvars[1]);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002987 char_u *n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002988
2989 if (STRCMP(what, "func") == 0 || STRCMP(what, "name") == 0)
2990 {
2991 rettv->v_type = (*what == 'f' ? VAR_FUNC : VAR_STRING);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002992 n = partial_name(pt);
2993 if (n == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002994 rettv->vval.v_string = NULL;
2995 else
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002996 {
2997 rettv->vval.v_string = vim_strsave(n);
2998 if (rettv->v_type == VAR_FUNC)
2999 func_ref(rettv->vval.v_string);
3000 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003001 }
3002 else if (STRCMP(what, "dict") == 0)
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003003 {
3004 what_is_dict = TRUE;
3005 if (pt->pt_dict != NULL)
3006 rettv_dict_set(rettv, pt->pt_dict);
3007 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003008 else if (STRCMP(what, "args") == 0)
3009 {
3010 rettv->v_type = VAR_LIST;
3011 if (rettv_list_alloc(rettv) == OK)
3012 {
3013 int i;
3014
3015 for (i = 0; i < pt->pt_argc; ++i)
3016 list_append_tv(rettv->vval.v_list, &pt->pt_argv[i]);
3017 }
3018 }
3019 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003020 semsg(_(e_invarg2), what);
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003021
3022 // When {what} == "dict" and pt->pt_dict == NULL, evaluate the
3023 // third argument
3024 if (!what_is_dict)
3025 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003026 }
3027 }
3028 else
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01003029 semsg(_(e_listdictblobarg), "get()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003030
3031 if (tv == NULL)
3032 {
3033 if (argvars[2].v_type != VAR_UNKNOWN)
3034 copy_tv(&argvars[2], rettv);
3035 }
3036 else
3037 copy_tv(tv, rettv);
3038}
3039
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003040/*
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003041 * "getchangelist()" function
3042 */
3043 static void
3044f_getchangelist(typval_T *argvars, typval_T *rettv)
3045{
3046#ifdef FEAT_JUMPLIST
3047 buf_T *buf;
3048 int i;
3049 list_T *l;
3050 dict_T *d;
3051#endif
3052
3053 if (rettv_list_alloc(rettv) != OK)
3054 return;
3055
3056#ifdef FEAT_JUMPLIST
Bram Moolenaar4c313b12019-08-24 22:58:31 +02003057 if (argvars[0].v_type == VAR_UNKNOWN)
3058 buf = curbuf;
3059 else
3060 {
3061 (void)tv_get_number(&argvars[0]); // issue errmsg if type error
3062 ++emsg_off;
3063 buf = tv_get_buf(&argvars[0], FALSE);
3064 --emsg_off;
3065 }
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003066 if (buf == NULL)
3067 return;
3068
3069 l = list_alloc();
3070 if (l == NULL)
3071 return;
3072
3073 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3074 return;
3075 /*
3076 * The current window change list index tracks only the position in the
3077 * current buffer change list. For other buffers, use the change list
3078 * length as the current index.
3079 */
3080 list_append_number(rettv->vval.v_list,
3081 (varnumber_T)((buf == curwin->w_buffer)
3082 ? curwin->w_changelistidx : buf->b_changelistlen));
3083
3084 for (i = 0; i < buf->b_changelistlen; ++i)
3085 {
3086 if (buf->b_changelist[i].lnum == 0)
3087 continue;
3088 if ((d = dict_alloc()) == NULL)
3089 return;
3090 if (list_append_dict(l, d) == FAIL)
3091 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003092 dict_add_number(d, "lnum", (long)buf->b_changelist[i].lnum);
3093 dict_add_number(d, "col", (long)buf->b_changelist[i].col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003094 dict_add_number(d, "coladd", (long)buf->b_changelist[i].coladd);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003095 }
3096#endif
3097}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003098
3099/*
3100 * "getcharsearch()" function
3101 */
3102 static void
3103f_getcharsearch(typval_T *argvars UNUSED, typval_T *rettv)
3104{
3105 if (rettv_dict_alloc(rettv) != FAIL)
3106 {
3107 dict_T *dict = rettv->vval.v_dict;
3108
Bram Moolenaare0be1672018-07-08 16:50:37 +02003109 dict_add_string(dict, "char", last_csearch());
3110 dict_add_number(dict, "forward", last_csearch_forward());
3111 dict_add_number(dict, "until", last_csearch_until());
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003112 }
3113}
3114
3115/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02003116 * "getenv()" function
3117 */
3118 static void
3119f_getenv(typval_T *argvars, typval_T *rettv)
3120{
3121 int mustfree = FALSE;
3122 char_u *p = vim_getenv(tv_get_string(&argvars[0]), &mustfree);
3123
3124 if (p == NULL)
3125 {
3126 rettv->v_type = VAR_SPECIAL;
3127 rettv->vval.v_number = VVAL_NULL;
3128 return;
3129 }
3130 if (!mustfree)
3131 p = vim_strsave(p);
3132 rettv->vval.v_string = p;
3133 rettv->v_type = VAR_STRING;
3134}
3135
3136/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003137 * "getfontname()" function
3138 */
3139 static void
3140f_getfontname(typval_T *argvars UNUSED, typval_T *rettv)
3141{
3142 rettv->v_type = VAR_STRING;
3143 rettv->vval.v_string = NULL;
3144#ifdef FEAT_GUI
3145 if (gui.in_use)
3146 {
3147 GuiFont font;
3148 char_u *name = NULL;
3149
3150 if (argvars[0].v_type == VAR_UNKNOWN)
3151 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003152 // Get the "Normal" font. Either the name saved by
3153 // hl_set_font_name() or from the font ID.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003154 font = gui.norm_font;
3155 name = hl_get_font_name();
3156 }
3157 else
3158 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003159 name = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003160 if (STRCMP(name, "*") == 0) // don't use font dialog
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003161 return;
3162 font = gui_mch_get_font(name, FALSE);
3163 if (font == NOFONT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003164 return; // Invalid font name, return empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003165 }
3166 rettv->vval.v_string = gui_mch_get_fontname(font, name);
3167 if (argvars[0].v_type != VAR_UNKNOWN)
3168 gui_mch_free_font(font);
3169 }
3170#endif
3171}
3172
3173/*
Bram Moolenaar4f505882018-02-10 21:06:32 +01003174 * "getjumplist()" function
3175 */
3176 static void
3177f_getjumplist(typval_T *argvars, typval_T *rettv)
3178{
3179#ifdef FEAT_JUMPLIST
3180 win_T *wp;
3181 int i;
3182 list_T *l;
3183 dict_T *d;
3184#endif
3185
3186 if (rettv_list_alloc(rettv) != OK)
3187 return;
3188
3189#ifdef FEAT_JUMPLIST
Bram Moolenaar00aa0692019-04-27 20:37:57 +02003190 wp = find_tabwin(&argvars[0], &argvars[1], NULL);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003191 if (wp == NULL)
3192 return;
3193
Bram Moolenaar57ee2b62019-02-12 22:15:06 +01003194 cleanup_jumplist(wp, TRUE);
3195
Bram Moolenaar4f505882018-02-10 21:06:32 +01003196 l = list_alloc();
3197 if (l == NULL)
3198 return;
3199
3200 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3201 return;
3202 list_append_number(rettv->vval.v_list, (varnumber_T)wp->w_jumplistidx);
3203
3204 for (i = 0; i < wp->w_jumplistlen; ++i)
3205 {
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003206 if (wp->w_jumplist[i].fmark.mark.lnum == 0)
3207 continue;
Bram Moolenaar4f505882018-02-10 21:06:32 +01003208 if ((d = dict_alloc()) == NULL)
3209 return;
3210 if (list_append_dict(l, d) == FAIL)
3211 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003212 dict_add_number(d, "lnum", (long)wp->w_jumplist[i].fmark.mark.lnum);
3213 dict_add_number(d, "col", (long)wp->w_jumplist[i].fmark.mark.col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003214 dict_add_number(d, "coladd", (long)wp->w_jumplist[i].fmark.mark.coladd);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003215 dict_add_number(d, "bufnr", (long)wp->w_jumplist[i].fmark.fnum);
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003216 if (wp->w_jumplist[i].fname != NULL)
Bram Moolenaare0be1672018-07-08 16:50:37 +02003217 dict_add_string(d, "filename", wp->w_jumplist[i].fname);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003218 }
3219#endif
3220}
3221
3222/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003223 * "getpid()" function
3224 */
3225 static void
3226f_getpid(typval_T *argvars UNUSED, typval_T *rettv)
3227{
3228 rettv->vval.v_number = mch_get_pid();
3229}
3230
3231 static void
3232getpos_both(
3233 typval_T *argvars,
3234 typval_T *rettv,
3235 int getcurpos)
3236{
3237 pos_T *fp;
3238 list_T *l;
3239 int fnum = -1;
3240
3241 if (rettv_list_alloc(rettv) == OK)
3242 {
3243 l = rettv->vval.v_list;
3244 if (getcurpos)
3245 fp = &curwin->w_cursor;
3246 else
3247 fp = var2fpos(&argvars[0], TRUE, &fnum);
3248 if (fnum != -1)
3249 list_append_number(l, (varnumber_T)fnum);
3250 else
3251 list_append_number(l, (varnumber_T)0);
3252 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->lnum
3253 : (varnumber_T)0);
3254 list_append_number(l, (fp != NULL)
3255 ? (varnumber_T)(fp->col == MAXCOL ? MAXCOL : fp->col + 1)
3256 : (varnumber_T)0);
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01003257 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->coladd :
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003258 (varnumber_T)0);
3259 if (getcurpos)
3260 {
Bram Moolenaar19a66852019-03-07 11:25:32 +01003261 int save_set_curswant = curwin->w_set_curswant;
3262 colnr_T save_curswant = curwin->w_curswant;
3263 colnr_T save_virtcol = curwin->w_virtcol;
3264
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003265 update_curswant();
3266 list_append_number(l, curwin->w_curswant == MAXCOL ?
3267 (varnumber_T)MAXCOL : (varnumber_T)curwin->w_curswant + 1);
Bram Moolenaar19a66852019-03-07 11:25:32 +01003268
3269 // Do not change "curswant", as it is unexpected that a get
3270 // function has a side effect.
3271 if (save_set_curswant)
3272 {
3273 curwin->w_set_curswant = save_set_curswant;
3274 curwin->w_curswant = save_curswant;
3275 curwin->w_virtcol = save_virtcol;
3276 curwin->w_valid &= ~VALID_VIRTCOL;
3277 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003278 }
3279 }
3280 else
3281 rettv->vval.v_number = FALSE;
3282}
3283
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003284/*
3285 * "getcurpos()" function
3286 */
3287 static void
3288f_getcurpos(typval_T *argvars, typval_T *rettv)
3289{
3290 getpos_both(argvars, rettv, TRUE);
3291}
3292
3293/*
3294 * "getpos(string)" function
3295 */
3296 static void
3297f_getpos(typval_T *argvars, typval_T *rettv)
3298{
3299 getpos_both(argvars, rettv, FALSE);
3300}
3301
3302/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003303 * "getreg()" function
3304 */
3305 static void
3306f_getreg(typval_T *argvars, typval_T *rettv)
3307{
3308 char_u *strregname;
3309 int regname;
3310 int arg2 = FALSE;
3311 int return_list = FALSE;
3312 int error = FALSE;
3313
3314 if (argvars[0].v_type != VAR_UNKNOWN)
3315 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003316 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003317 error = strregname == NULL;
3318 if (argvars[1].v_type != VAR_UNKNOWN)
3319 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003320 arg2 = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003321 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003322 return_list = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003323 }
3324 }
3325 else
3326 strregname = get_vim_var_str(VV_REG);
3327
3328 if (error)
3329 return;
3330
3331 regname = (strregname == NULL ? '"' : *strregname);
3332 if (regname == 0)
3333 regname = '"';
3334
3335 if (return_list)
3336 {
3337 rettv->v_type = VAR_LIST;
3338 rettv->vval.v_list = (list_T *)get_reg_contents(regname,
3339 (arg2 ? GREG_EXPR_SRC : 0) | GREG_LIST);
3340 if (rettv->vval.v_list == NULL)
3341 (void)rettv_list_alloc(rettv);
3342 else
3343 ++rettv->vval.v_list->lv_refcount;
3344 }
3345 else
3346 {
3347 rettv->v_type = VAR_STRING;
3348 rettv->vval.v_string = get_reg_contents(regname,
3349 arg2 ? GREG_EXPR_SRC : 0);
3350 }
3351}
3352
3353/*
3354 * "getregtype()" function
3355 */
3356 static void
3357f_getregtype(typval_T *argvars, typval_T *rettv)
3358{
3359 char_u *strregname;
3360 int regname;
3361 char_u buf[NUMBUFLEN + 2];
3362 long reglen = 0;
3363
3364 if (argvars[0].v_type != VAR_UNKNOWN)
3365 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003366 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003367 if (strregname == NULL) // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003368 {
3369 rettv->v_type = VAR_STRING;
3370 rettv->vval.v_string = NULL;
3371 return;
3372 }
3373 }
3374 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003375 // Default to v:register
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003376 strregname = get_vim_var_str(VV_REG);
3377
3378 regname = (strregname == NULL ? '"' : *strregname);
3379 if (regname == 0)
3380 regname = '"';
3381
3382 buf[0] = NUL;
3383 buf[1] = NUL;
3384 switch (get_reg_type(regname, &reglen))
3385 {
3386 case MLINE: buf[0] = 'V'; break;
3387 case MCHAR: buf[0] = 'v'; break;
3388 case MBLOCK:
3389 buf[0] = Ctrl_V;
3390 sprintf((char *)buf + 1, "%ld", reglen + 1);
3391 break;
3392 }
3393 rettv->v_type = VAR_STRING;
3394 rettv->vval.v_string = vim_strsave(buf);
3395}
3396
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003397/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01003398 * "gettagstack()" function
3399 */
3400 static void
3401f_gettagstack(typval_T *argvars, typval_T *rettv)
3402{
3403 win_T *wp = curwin; // default is current window
3404
3405 if (rettv_dict_alloc(rettv) != OK)
3406 return;
3407
3408 if (argvars[0].v_type != VAR_UNKNOWN)
3409 {
3410 wp = find_win_by_nr_or_id(&argvars[0]);
3411 if (wp == NULL)
3412 return;
3413 }
3414
3415 get_tagstack(wp, rettv->vval.v_dict);
3416}
3417
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003418// for VIM_VERSION_ defines
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003419#include "version.h"
3420
3421/*
3422 * "has()" function
3423 */
Bram Moolenaara259d8d2020-01-31 20:10:50 +01003424 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003425f_has(typval_T *argvars, typval_T *rettv)
3426{
3427 int i;
3428 char_u *name;
Bram Moolenaar79296512020-03-22 16:17:14 +01003429 int x = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003430 int n = FALSE;
Bram Moolenaar79296512020-03-22 16:17:14 +01003431 typedef struct {
3432 char *name;
3433 short present;
3434 } has_item_T;
3435 static has_item_T has_list[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003436 {
Bram Moolenaar79296512020-03-22 16:17:14 +01003437 {"amiga",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003438#ifdef AMIGA
Bram Moolenaar79296512020-03-22 16:17:14 +01003439 1
Bram Moolenaar39536dd2019-01-29 22:58:21 +01003440#else
Bram Moolenaar79296512020-03-22 16:17:14 +01003441 0
Bram Moolenaar39536dd2019-01-29 22:58:21 +01003442#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003443 },
3444 {"arp",
3445#if defined(AMIGA) && defined(FEAT_ARP)
3446 1
3447#else
3448 0
3449#endif
3450 },
3451 {"beos",
3452#ifdef __BEOS__
3453 1
3454#else
3455 0
3456#endif
3457 },
3458 {"haiku",
3459#ifdef __HAIKU__
3460 1
3461#else
3462 0
3463#endif
3464 },
3465 {"bsd",
3466#if defined(BSD) && !defined(MACOS_X)
3467 1
3468#else
3469 0
3470#endif
3471 },
3472 {"hpux",
3473#ifdef hpux
3474 1
3475#else
3476 0
3477#endif
3478 },
3479 {"linux",
3480#ifdef __linux__
3481 1
3482#else
3483 0
3484#endif
3485 },
3486 {"mac", // Mac OS X (and, once, Mac OS Classic)
3487#ifdef MACOS_X
3488 1
3489#else
3490 0
3491#endif
3492 },
3493 {"osx", // Mac OS X
3494#ifdef MACOS_X
3495 1
3496#else
3497 0
3498#endif
3499 },
3500 {"macunix", // Mac OS X, with the darwin feature
3501#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
3502 1
3503#else
3504 0
3505#endif
3506 },
3507 {"osxdarwin", // synonym for macunix
3508#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
3509 1
3510#else
3511 0
3512#endif
3513 },
3514 {"qnx",
3515#ifdef __QNX__
3516 1
3517#else
3518 0
3519#endif
3520 },
3521 {"sun",
3522#ifdef SUN_SYSTEM
3523 1
3524#else
3525 0
3526#endif
3527 },
3528 {"unix",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003529#ifdef UNIX
Bram Moolenaar79296512020-03-22 16:17:14 +01003530 1
3531#else
3532 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003533#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003534 },
3535 {"vms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003536#ifdef VMS
Bram Moolenaar79296512020-03-22 16:17:14 +01003537 1
3538#else
3539 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003540#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003541 },
3542 {"win32",
Bram Moolenaar4f974752019-02-17 17:44:42 +01003543#ifdef MSWIN
Bram Moolenaar79296512020-03-22 16:17:14 +01003544 1
3545#else
3546 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003547#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003548 },
3549 {"win32unix",
Bram Moolenaar1eed5322019-02-26 17:03:54 +01003550#if defined(UNIX) && defined(__CYGWIN__)
Bram Moolenaar79296512020-03-22 16:17:14 +01003551 1
3552#else
3553 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003554#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003555 },
3556 {"win64",
Bram Moolenaar44b443c2019-02-18 22:14:18 +01003557#ifdef _WIN64
Bram Moolenaar79296512020-03-22 16:17:14 +01003558 1
3559#else
3560 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003561#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003562 },
3563 {"ebcdic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003564#ifdef EBCDIC
Bram Moolenaar79296512020-03-22 16:17:14 +01003565 1
3566#else
3567 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003568#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003569 },
3570 {"fname_case",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003571#ifndef CASE_INSENSITIVE_FILENAME
Bram Moolenaar79296512020-03-22 16:17:14 +01003572 1
3573#else
3574 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003575#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003576 },
3577 {"acl",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003578#ifdef HAVE_ACL
Bram Moolenaar79296512020-03-22 16:17:14 +01003579 1
3580#else
3581 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003582#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003583 },
3584 {"arabic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003585#ifdef FEAT_ARABIC
Bram Moolenaar79296512020-03-22 16:17:14 +01003586 1
3587#else
3588 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003589#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003590 },
3591 {"autocmd", 1},
3592 {"autochdir",
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02003593#ifdef FEAT_AUTOCHDIR
Bram Moolenaar79296512020-03-22 16:17:14 +01003594 1
3595#else
3596 0
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02003597#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003598 },
3599 {"autoservername",
Bram Moolenaare42a6d22017-11-12 19:21:51 +01003600#ifdef FEAT_AUTOSERVERNAME
Bram Moolenaar79296512020-03-22 16:17:14 +01003601 1
3602#else
3603 0
Bram Moolenaare42a6d22017-11-12 19:21:51 +01003604#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003605 },
3606 {"balloon_eval",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01003607#ifdef FEAT_BEVAL_GUI
Bram Moolenaar79296512020-03-22 16:17:14 +01003608 1
3609#else
3610 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003611#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003612 },
3613 {"balloon_multiline",
3614#if defined(FEAT_BEVAL_GUI) && !defined(FEAT_GUI_MSWIN)
3615 // MS-Windows requires runtime check, see below
3616 1
3617#else
3618 0
3619#endif
3620 },
3621 {"balloon_eval_term",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01003622#ifdef FEAT_BEVAL_TERM
Bram Moolenaar79296512020-03-22 16:17:14 +01003623 1
3624#else
3625 0
Bram Moolenaar51b0f372017-11-18 18:52:04 +01003626#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003627 },
3628 {"builtin_terms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003629#if defined(SOME_BUILTIN_TCAPS) || defined(ALL_BUILTIN_TCAPS)
Bram Moolenaar79296512020-03-22 16:17:14 +01003630 1
3631#else
3632 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003633#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003634 },
3635 {"all_builtin_terms",
3636#if defined(ALL_BUILTIN_TCAPS)
3637 1
3638#else
3639 0
3640#endif
3641 },
3642 {"browsefilter",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003643#if defined(FEAT_BROWSE) && (defined(USE_FILE_CHOOSER) \
Bram Moolenaar4f974752019-02-17 17:44:42 +01003644 || defined(FEAT_GUI_MSWIN) \
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003645 || defined(FEAT_GUI_MOTIF))
Bram Moolenaar79296512020-03-22 16:17:14 +01003646 1
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003647#else
Bram Moolenaar79296512020-03-22 16:17:14 +01003648 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003649#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003650 },
3651 {"byte_offset",
3652#ifdef FEAT_BYTEOFF
3653 1
3654#else
3655 0
3656#endif
3657 },
3658 {"channel",
3659#ifdef FEAT_JOB_CHANNEL
3660 1
3661#else
3662 0
3663#endif
3664 },
3665 {"cindent",
3666#ifdef FEAT_CINDENT
3667 1
3668#else
3669 0
3670#endif
3671 },
3672 {"clientserver",
3673#ifdef FEAT_CLIENTSERVER
3674 1
3675#else
3676 0
3677#endif
3678 },
3679 {"clipboard",
3680#ifdef FEAT_CLIPBOARD
3681 1
3682#else
3683 0
3684#endif
3685 },
3686 {"cmdline_compl", 1},
3687 {"cmdline_hist", 1},
3688 {"comments", 1},
3689 {"conceal",
3690#ifdef FEAT_CONCEAL
3691 1
3692#else
3693 0
3694#endif
3695 },
3696 {"cryptv",
3697#ifdef FEAT_CRYPT
3698 1
3699#else
3700 0
3701#endif
3702 },
3703 {"crypt-blowfish",
3704#ifdef FEAT_CRYPT
3705 1
3706#else
3707 0
3708#endif
3709 },
3710 {"crypt-blowfish2",
3711#ifdef FEAT_CRYPT
3712 1
3713#else
3714 0
3715#endif
3716 },
3717 {"cscope",
3718#ifdef FEAT_CSCOPE
3719 1
3720#else
3721 0
3722#endif
3723 },
3724 {"cursorbind", 1},
3725 {"cursorshape",
3726#ifdef CURSOR_SHAPE
3727 1
3728#else
3729 0
3730#endif
3731 },
3732 {"debug",
3733#ifdef DEBUG
3734 1
3735#else
3736 0
3737#endif
3738 },
3739 {"dialog_con",
3740#ifdef FEAT_CON_DIALOG
3741 1
3742#else
3743 0
3744#endif
3745 },
3746 {"dialog_gui",
3747#ifdef FEAT_GUI_DIALOG
3748 1
3749#else
3750 0
3751#endif
3752 },
3753 {"diff",
3754#ifdef FEAT_DIFF
3755 1
3756#else
3757 0
3758#endif
3759 },
3760 {"digraphs",
3761#ifdef FEAT_DIGRAPHS
3762 1
3763#else
3764 0
3765#endif
3766 },
3767 {"directx",
3768#ifdef FEAT_DIRECTX
3769 1
3770#else
3771 0
3772#endif
3773 },
3774 {"dnd",
3775#ifdef FEAT_DND
3776 1
3777#else
3778 0
3779#endif
3780 },
3781 {"emacs_tags",
3782#ifdef FEAT_EMACS_TAGS
3783 1
3784#else
3785 0
3786#endif
3787 },
3788 {"eval", 1}, // always present, of course!
3789 {"ex_extra", 1}, // graduated feature
3790 {"extra_search",
3791#ifdef FEAT_SEARCH_EXTRA
3792 1
3793#else
3794 0
3795#endif
3796 },
3797 {"file_in_path",
3798#ifdef FEAT_SEARCHPATH
3799 1
3800#else
3801 0
3802#endif
3803 },
3804 {"filterpipe",
3805#if defined(FEAT_FILTERPIPE) && !defined(VIMDLL)
3806 1
3807#else
3808 0
3809#endif
3810 },
3811 {"find_in_path",
3812#ifdef FEAT_FIND_ID
3813 1
3814#else
3815 0
3816#endif
3817 },
3818 {"float",
3819#ifdef FEAT_FLOAT
3820 1
3821#else
3822 0
3823#endif
3824 },
3825 {"folding",
3826#ifdef FEAT_FOLDING
3827 1
3828#else
3829 0
3830#endif
3831 },
3832 {"footer",
3833#ifdef FEAT_FOOTER
3834 1
3835#else
3836 0
3837#endif
3838 },
3839 {"fork",
3840#if !defined(USE_SYSTEM) && defined(UNIX)
3841 1
3842#else
3843 0
3844#endif
3845 },
3846 {"gettext",
3847#ifdef FEAT_GETTEXT
3848 1
3849#else
3850 0
3851#endif
3852 },
3853 {"gui",
3854#ifdef FEAT_GUI
3855 1
3856#else
3857 0
3858#endif
3859 },
3860 {"gui_neXtaw",
3861#if defined(FEAT_GUI_ATHENA) && defined(FEAT_GUI_NEXTAW)
3862 1
3863#else
3864 0
3865#endif
3866 },
3867 {"gui_athena",
3868#if defined(FEAT_GUI_ATHENA) && !defined(FEAT_GUI_NEXTAW)
3869 1
3870#else
3871 0
3872#endif
3873 },
3874 {"gui_gtk",
3875#ifdef FEAT_GUI_GTK
3876 1
3877#else
3878 0
3879#endif
3880 },
3881 {"gui_gtk2",
3882#if defined(FEAT_GUI_GTK) && !defined(USE_GTK3)
3883 1
3884#else
3885 0
3886#endif
3887 },
3888 {"gui_gtk3",
3889#if defined(FEAT_GUI_GTK) && defined(USE_GTK3)
3890 1
3891#else
3892 0
3893#endif
3894 },
3895 {"gui_gnome",
3896#ifdef FEAT_GUI_GNOME
3897 1
3898#else
3899 0
3900#endif
3901 },
3902 {"gui_haiku",
3903#ifdef FEAT_GUI_HAIKU
3904 1
3905#else
3906 0
3907#endif
3908 },
3909 {"gui_mac",
3910#ifdef FEAT_GUI_MAC
3911 1
3912#else
3913 0
3914#endif
3915 },
3916 {"gui_motif",
3917#ifdef FEAT_GUI_MOTIF
3918 1
3919#else
3920 0
3921#endif
3922 },
3923 {"gui_photon",
3924#ifdef FEAT_GUI_PHOTON
3925 1
3926#else
3927 0
3928#endif
3929 },
3930 {"gui_win32",
3931#ifdef FEAT_GUI_MSWIN
3932 1
3933#else
3934 0
3935#endif
3936 },
3937 {"iconv",
3938#if defined(HAVE_ICONV_H) && defined(USE_ICONV)
3939 1
3940#else
3941 0
3942#endif
3943 },
3944 {"insert_expand", 1},
3945 {"job",
3946#ifdef FEAT_JOB_CHANNEL
3947 1
3948#else
3949 0
3950#endif
3951 },
3952 {"jumplist",
3953#ifdef FEAT_JUMPLIST
3954 1
3955#else
3956 0
3957#endif
3958 },
3959 {"keymap",
3960#ifdef FEAT_KEYMAP
3961 1
3962#else
3963 0
3964#endif
3965 },
3966 {"lambda", 1}, // always with FEAT_EVAL, since 7.4.2120 with closure
3967 {"langmap",
3968#ifdef FEAT_LANGMAP
3969 1
3970#else
3971 0
3972#endif
3973 },
3974 {"libcall",
3975#ifdef FEAT_LIBCALL
3976 1
3977#else
3978 0
3979#endif
3980 },
3981 {"linebreak",
3982#ifdef FEAT_LINEBREAK
3983 1
3984#else
3985 0
3986#endif
3987 },
3988 {"lispindent",
3989#ifdef FEAT_LISP
3990 1
3991#else
3992 0
3993#endif
3994 },
3995 {"listcmds", 1},
3996 {"localmap", 1},
3997 {"lua",
3998#if defined(FEAT_LUA) && !defined(DYNAMIC_LUA)
3999 1
4000#else
4001 0
4002#endif
4003 },
4004 {"menu",
4005#ifdef FEAT_MENU
4006 1
4007#else
4008 0
4009#endif
4010 },
4011 {"mksession",
4012#ifdef FEAT_SESSION
4013 1
4014#else
4015 0
4016#endif
4017 },
4018 {"modify_fname", 1},
4019 {"mouse", 1},
4020 {"mouseshape",
4021#ifdef FEAT_MOUSESHAPE
4022 1
4023#else
4024 0
4025#endif
4026 },
4027 {"mouse_dec",
4028#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_DEC)
4029 1
4030#else
4031 0
4032#endif
4033 },
4034 {"mouse_gpm",
4035#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_GPM)
4036 1
4037#else
4038 0
4039#endif
4040 },
4041 {"mouse_jsbterm",
4042#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_JSB)
4043 1
4044#else
4045 0
4046#endif
4047 },
4048 {"mouse_netterm",
4049#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_NET)
4050 1
4051#else
4052 0
4053#endif
4054 },
4055 {"mouse_pterm",
4056#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_PTERM)
4057 1
4058#else
4059 0
4060#endif
4061 },
4062 {"mouse_sgr",
4063#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4064 1
4065#else
4066 0
4067#endif
4068 },
4069 {"mouse_sysmouse",
4070#if (defined(UNIX) || defined(VMS)) && defined(FEAT_SYSMOUSE)
4071 1
4072#else
4073 0
4074#endif
4075 },
4076 {"mouse_urxvt",
4077#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_URXVT)
4078 1
4079#else
4080 0
4081#endif
4082 },
4083 {"mouse_xterm",
4084#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4085 1
4086#else
4087 0
4088#endif
4089 },
4090 {"multi_byte", 1},
4091 {"multi_byte_ime",
4092#ifdef FEAT_MBYTE_IME
4093 1
4094#else
4095 0
4096#endif
4097 },
4098 {"multi_lang",
4099#ifdef FEAT_MULTI_LANG
4100 1
4101#else
4102 0
4103#endif
4104 },
4105 {"mzscheme",
4106#if defined(FEAT_MZSCHEME) && !defined(DYNAMIC_MZSCHEME)
4107 1
4108#else
4109 0
4110#endif
4111 },
4112 {"num64", 1},
4113 {"ole",
4114#ifdef FEAT_OLE
4115 1
4116#else
4117 0
4118#endif
4119 },
4120 {"packages",
4121#ifdef FEAT_EVAL
4122 1
4123#else
4124 0
4125#endif
4126 },
4127 {"path_extra",
4128#ifdef FEAT_PATH_EXTRA
4129 1
4130#else
4131 0
4132#endif
4133 },
4134 {"perl",
4135#if defined(FEAT_PERL) && !defined(DYNAMIC_PERL)
4136 1
4137#else
4138 0
4139#endif
4140 },
4141 {"persistent_undo",
4142#ifdef FEAT_PERSISTENT_UNDO
4143 1
4144#else
4145 0
4146#endif
4147 },
4148 {"python_compiled",
4149#if defined(FEAT_PYTHON)
4150 1
4151#else
4152 0
4153#endif
4154 },
4155 {"python_dynamic",
4156#if defined(FEAT_PYTHON) && defined(DYNAMIC_PYTHON)
4157 1
4158#else
4159 0
4160#endif
4161 },
4162 {"python",
4163#if defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)
4164 1
4165#else
4166 0
4167#endif
4168 },
4169 {"pythonx",
4170#if (defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)) \
4171 || (defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3))
4172 1
4173#else
4174 0
4175#endif
4176 },
4177 {"python3_compiled",
4178#if defined(FEAT_PYTHON3)
4179 1
4180#else
4181 0
4182#endif
4183 },
4184 {"python3_dynamic",
4185#if defined(FEAT_PYTHON3) && defined(DYNAMIC_PYTHON3)
4186 1
4187#else
4188 0
4189#endif
4190 },
4191 {"python3",
4192#if defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3)
4193 1
4194#else
4195 0
4196#endif
4197 },
4198 {"popupwin",
4199#ifdef FEAT_PROP_POPUP
4200 1
4201#else
4202 0
4203#endif
4204 },
4205 {"postscript",
4206#ifdef FEAT_POSTSCRIPT
4207 1
4208#else
4209 0
4210#endif
4211 },
4212 {"printer",
4213#ifdef FEAT_PRINTER
4214 1
4215#else
4216 0
4217#endif
4218 },
4219 {"profile",
4220#ifdef FEAT_PROFILE
4221 1
4222#else
4223 0
4224#endif
4225 },
4226 {"reltime",
4227#ifdef FEAT_RELTIME
4228 1
4229#else
4230 0
4231#endif
4232 },
4233 {"quickfix",
4234#ifdef FEAT_QUICKFIX
4235 1
4236#else
4237 0
4238#endif
4239 },
4240 {"rightleft",
4241#ifdef FEAT_RIGHTLEFT
4242 1
4243#else
4244 0
4245#endif
4246 },
4247 {"ruby",
4248#if defined(FEAT_RUBY) && !defined(DYNAMIC_RUBY)
4249 1
4250#else
4251 0
4252#endif
4253 },
4254 {"scrollbind", 1},
4255 {"showcmd",
4256#ifdef FEAT_CMDL_INFO
4257 1
4258#else
4259 0
4260#endif
4261 },
4262 {"cmdline_info",
4263#ifdef FEAT_CMDL_INFO
4264 1
4265#else
4266 0
4267#endif
4268 },
4269 {"signs",
4270#ifdef FEAT_SIGNS
4271 1
4272#else
4273 0
4274#endif
4275 },
4276 {"smartindent",
4277#ifdef FEAT_SMARTINDENT
4278 1
4279#else
4280 0
4281#endif
4282 },
4283 {"startuptime",
4284#ifdef STARTUPTIME
4285 1
4286#else
4287 0
4288#endif
4289 },
4290 {"statusline",
4291#ifdef FEAT_STL_OPT
4292 1
4293#else
4294 0
4295#endif
4296 },
4297 {"netbeans_intg",
4298#ifdef FEAT_NETBEANS_INTG
4299 1
4300#else
4301 0
4302#endif
4303 },
4304 {"sound",
4305#ifdef FEAT_SOUND
4306 1
4307#else
4308 0
4309#endif
4310 },
4311 {"spell",
4312#ifdef FEAT_SPELL
4313 1
4314#else
4315 0
4316#endif
4317 },
4318 {"syntax",
4319#ifdef FEAT_SYN_HL
4320 1
4321#else
4322 0
4323#endif
4324 },
4325 {"system",
4326#if defined(USE_SYSTEM) || !defined(UNIX)
4327 1
4328#else
4329 0
4330#endif
4331 },
4332 {"tag_binary",
4333#ifdef FEAT_TAG_BINS
4334 1
4335#else
4336 0
4337#endif
4338 },
4339 {"tcl",
4340#if defined(FEAT_TCL) && !defined(DYNAMIC_TCL)
4341 1
4342#else
4343 0
4344#endif
4345 },
4346 {"termguicolors",
4347#ifdef FEAT_TERMGUICOLORS
4348 1
4349#else
4350 0
4351#endif
4352 },
4353 {"terminal",
4354#if defined(FEAT_TERMINAL) && !defined(MSWIN)
4355 1
4356#else
4357 0
4358#endif
4359 },
4360 {"terminfo",
4361#ifdef TERMINFO
4362 1
4363#else
4364 0
4365#endif
4366 },
4367 {"termresponse",
4368#ifdef FEAT_TERMRESPONSE
4369 1
4370#else
4371 0
4372#endif
4373 },
4374 {"textobjects",
4375#ifdef FEAT_TEXTOBJ
4376 1
4377#else
4378 0
4379#endif
4380 },
4381 {"textprop",
4382#ifdef FEAT_PROP_POPUP
4383 1
4384#else
4385 0
4386#endif
4387 },
4388 {"tgetent",
4389#ifdef HAVE_TGETENT
4390 1
4391#else
4392 0
4393#endif
4394 },
4395 {"timers",
4396#ifdef FEAT_TIMERS
4397 1
4398#else
4399 0
4400#endif
4401 },
4402 {"title",
4403#ifdef FEAT_TITLE
4404 1
4405#else
4406 0
4407#endif
4408 },
4409 {"toolbar",
4410#ifdef FEAT_TOOLBAR
4411 1
4412#else
4413 0
4414#endif
4415 },
4416 {"unnamedplus",
4417#if defined(FEAT_CLIPBOARD) && defined(FEAT_X11)
4418 1
4419#else
4420 0
4421#endif
4422 },
4423 {"user-commands", 1}, // was accidentally included in 5.4
4424 {"user_commands", 1},
4425 {"vartabs",
4426#ifdef FEAT_VARTABS
4427 1
4428#else
4429 0
4430#endif
4431 },
4432 {"vertsplit", 1},
4433 {"viminfo",
4434#ifdef FEAT_VIMINFO
4435 1
4436#else
4437 0
4438#endif
4439 },
4440 {"vimscript-1", 1},
4441 {"vimscript-2", 1},
4442 {"vimscript-3", 1},
4443 {"vimscript-4", 1},
4444 {"virtualedit", 1},
4445 {"visual", 1},
4446 {"visualextra", 1},
4447 {"vreplace", 1},
4448 {"vtp",
4449#ifdef FEAT_VTP
4450 1
4451#else
4452 0
4453#endif
4454 },
4455 {"wildignore",
4456#ifdef FEAT_WILDIGN
4457 1
4458#else
4459 0
4460#endif
4461 },
4462 {"wildmenu",
4463#ifdef FEAT_WILDMENU
4464 1
4465#else
4466 0
4467#endif
4468 },
4469 {"windows", 1},
4470 {"winaltkeys",
4471#ifdef FEAT_WAK
4472 1
4473#else
4474 0
4475#endif
4476 },
4477 {"writebackup",
4478#ifdef FEAT_WRITEBACKUP
4479 1
4480#else
4481 0
4482#endif
4483 },
4484 {"xim",
4485#ifdef FEAT_XIM
4486 1
4487#else
4488 0
4489#endif
4490 },
4491 {"xfontset",
4492#ifdef FEAT_XFONTSET
4493 1
4494#else
4495 0
4496#endif
4497 },
4498 {"xpm",
4499#if defined(FEAT_XPM_W32) || defined(HAVE_XPM)
4500 1
4501#else
4502 0
4503#endif
4504 },
4505 {"xpm_w32", // for backward compatibility
4506#ifdef FEAT_XPM_W32
4507 1
4508#else
4509 0
4510#endif
4511 },
4512 {"xsmp",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004513#ifdef USE_XSMP
Bram Moolenaar79296512020-03-22 16:17:14 +01004514 1
4515#else
4516 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004517#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004518 },
4519 {"xsmp_interact",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004520#ifdef USE_XSMP_INTERACT
Bram Moolenaar79296512020-03-22 16:17:14 +01004521 1
4522#else
4523 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004524#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004525 },
4526 {"xterm_clipboard",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004527#ifdef FEAT_XCLIPBOARD
Bram Moolenaar79296512020-03-22 16:17:14 +01004528 1
4529#else
4530 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004531#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004532 },
4533 {"xterm_save",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004534#ifdef FEAT_XTERM_SAVE
Bram Moolenaar79296512020-03-22 16:17:14 +01004535 1
4536#else
4537 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004538#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004539 },
4540 {"X11",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004541#if defined(UNIX) && defined(FEAT_X11)
Bram Moolenaar79296512020-03-22 16:17:14 +01004542 1
4543#else
4544 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004545#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004546 },
4547 {NULL, 0}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004548 };
4549
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004550 name = tv_get_string(&argvars[0]);
Bram Moolenaar79296512020-03-22 16:17:14 +01004551 for (i = 0; has_list[i].name != NULL; ++i)
4552 if (STRICMP(name, has_list[i].name) == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004553 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004554 x = TRUE;
4555 n = has_list[i].present;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004556 break;
4557 }
4558
Bram Moolenaar79296512020-03-22 16:17:14 +01004559 // features also in has_list[] but sometimes enabled at runtime
4560 if (x == TRUE && n == FALSE)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004561 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004562 if (0)
4563 ;
Bram Moolenaar4f974752019-02-17 17:44:42 +01004564#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004565 else if (STRICMP(name, "balloon_multiline") == 0)
4566 n = multiline_balloon_available();
4567#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004568#ifdef VIMDLL
4569 else if (STRICMP(name, "filterpipe") == 0)
4570 n = gui.in_use || gui.starting;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004571#endif
4572#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
4573 else if (STRICMP(name, "iconv") == 0)
4574 n = iconv_enabled(FALSE);
4575#endif
4576#ifdef DYNAMIC_LUA
4577 else if (STRICMP(name, "lua") == 0)
4578 n = lua_enabled(FALSE);
4579#endif
4580#ifdef DYNAMIC_MZSCHEME
4581 else if (STRICMP(name, "mzscheme") == 0)
4582 n = mzscheme_enabled(FALSE);
4583#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004584#ifdef DYNAMIC_PERL
4585 else if (STRICMP(name, "perl") == 0)
4586 n = perl_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004587#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004588#ifdef DYNAMIC_PYTHON
4589 else if (STRICMP(name, "python") == 0)
4590 n = python_enabled(FALSE);
4591#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004592#ifdef DYNAMIC_PYTHON3
4593 else if (STRICMP(name, "python3") == 0)
4594 n = python3_enabled(FALSE);
4595#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01004596#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
4597 else if (STRICMP(name, "pythonx") == 0)
4598 {
4599# if defined(DYNAMIC_PYTHON) && defined(DYNAMIC_PYTHON3)
4600 if (p_pyx == 0)
4601 n = python3_enabled(FALSE) || python_enabled(FALSE);
4602 else if (p_pyx == 3)
4603 n = python3_enabled(FALSE);
4604 else if (p_pyx == 2)
4605 n = python_enabled(FALSE);
4606# elif defined(DYNAMIC_PYTHON)
4607 n = python_enabled(FALSE);
4608# elif defined(DYNAMIC_PYTHON3)
4609 n = python3_enabled(FALSE);
4610# endif
4611 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004612#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004613#ifdef DYNAMIC_RUBY
4614 else if (STRICMP(name, "ruby") == 0)
4615 n = ruby_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004616#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004617#ifdef DYNAMIC_TCL
4618 else if (STRICMP(name, "tcl") == 0)
4619 n = tcl_enabled(FALSE);
Bram Moolenaar4b8366b2019-05-04 17:34:34 +02004620#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +01004621#if defined(FEAT_TERMINAL) && defined(MSWIN)
Bram Moolenaara83e3962017-08-17 14:39:07 +02004622 else if (STRICMP(name, "terminal") == 0)
4623 n = terminal_enabled();
4624#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004625 }
4626
Bram Moolenaar79296512020-03-22 16:17:14 +01004627 // features not in has_list[]
4628 if (x == FALSE)
4629 {
4630 if (STRNICMP(name, "patch", 5) == 0)
4631 {
4632 x = TRUE;
4633 if (name[5] == '-'
4634 && STRLEN(name) >= 11
4635 && vim_isdigit(name[6])
4636 && vim_isdigit(name[8])
4637 && vim_isdigit(name[10]))
4638 {
4639 int major = atoi((char *)name + 6);
4640 int minor = atoi((char *)name + 8);
4641
4642 // Expect "patch-9.9.01234".
4643 n = (major < VIM_VERSION_MAJOR
4644 || (major == VIM_VERSION_MAJOR
4645 && (minor < VIM_VERSION_MINOR
4646 || (minor == VIM_VERSION_MINOR
4647 && has_patch(atoi((char *)name + 10))))));
4648 }
4649 else
4650 n = has_patch(atoi((char *)name + 5));
4651 }
4652 else if (STRICMP(name, "vim_starting") == 0)
4653 {
4654 x = TRUE;
4655 n = (starting != 0);
4656 }
4657 else if (STRICMP(name, "ttyin") == 0)
4658 {
4659 x = TRUE;
4660 n = mch_input_isatty();
4661 }
4662 else if (STRICMP(name, "ttyout") == 0)
4663 {
4664 x = TRUE;
4665 n = stdout_isatty;
4666 }
4667 else if (STRICMP(name, "multi_byte_encoding") == 0)
4668 {
4669 x = TRUE;
4670 n = has_mbyte;
4671 }
4672 else if (STRICMP(name, "gui_running") == 0)
4673 {
4674 x = TRUE;
4675#ifdef FEAT_GUI
4676 n = (gui.in_use || gui.starting);
4677#endif
4678 }
4679 else if (STRICMP(name, "browse") == 0)
4680 {
4681 x = TRUE;
4682#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
4683 n = gui.in_use; // gui_mch_browse() works when GUI is running
4684#endif
4685 }
4686 else if (STRICMP(name, "syntax_items") == 0)
4687 {
4688 x = TRUE;
4689#ifdef FEAT_SYN_HL
4690 n = syntax_present(curwin);
4691#endif
4692 }
4693 else if (STRICMP(name, "vcon") == 0)
4694 {
4695 x = TRUE;
4696#ifdef FEAT_VTP
4697 n = is_term_win32() && has_vtp_working();
4698#endif
4699 }
4700 else if (STRICMP(name, "netbeans_enabled") == 0)
4701 {
4702 x = TRUE;
4703#ifdef FEAT_NETBEANS_INTG
4704 n = netbeans_active();
4705#endif
4706 }
4707 else if (STRICMP(name, "mouse_gpm_enabled") == 0)
4708 {
4709 x = TRUE;
4710#ifdef FEAT_MOUSE_GPM
4711 n = gpm_enabled();
4712#endif
4713 }
4714 else if (STRICMP(name, "conpty") == 0)
4715 {
4716 x = TRUE;
4717#if defined(FEAT_TERMINAL) && defined(MSWIN)
4718 n = use_conpty();
4719#endif
4720 }
4721 else if (STRICMP(name, "clipboard_working") == 0)
4722 {
4723 x = TRUE;
4724#ifdef FEAT_CLIPBOARD
4725 n = clip_star.available;
4726#endif
4727 }
4728 }
4729
4730 if (argvars[1].v_type != VAR_UNKNOWN && tv_get_number(&argvars[1]) != 0)
4731 // return whether feature could ever be enabled
4732 rettv->vval.v_number = x;
4733 else
4734 // return whether feature is enabled
4735 rettv->vval.v_number = n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004736}
4737
4738/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004739 * "haslocaldir()" function
4740 */
4741 static void
4742f_haslocaldir(typval_T *argvars, typval_T *rettv)
4743{
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004744 tabpage_T *tp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004745 win_T *wp = NULL;
4746
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004747 wp = find_tabwin(&argvars[0], &argvars[1], &tp);
4748
4749 // Check for window-local and tab-local directories
4750 if (wp != NULL && wp->w_localdir != NULL)
4751 rettv->vval.v_number = 1;
4752 else if (tp != NULL && tp->tp_localdir != NULL)
4753 rettv->vval.v_number = 2;
4754 else
4755 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004756}
4757
4758/*
4759 * "hasmapto()" function
4760 */
4761 static void
4762f_hasmapto(typval_T *argvars, typval_T *rettv)
4763{
4764 char_u *name;
4765 char_u *mode;
4766 char_u buf[NUMBUFLEN];
4767 int abbr = FALSE;
4768
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004769 name = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004770 if (argvars[1].v_type == VAR_UNKNOWN)
4771 mode = (char_u *)"nvo";
4772 else
4773 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004774 mode = tv_get_string_buf(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004775 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004776 abbr = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004777 }
4778
4779 if (map_to_exists(name, mode, abbr))
4780 rettv->vval.v_number = TRUE;
4781 else
4782 rettv->vval.v_number = FALSE;
4783}
4784
4785/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004786 * "highlightID(name)" function
4787 */
4788 static void
4789f_hlID(typval_T *argvars, typval_T *rettv)
4790{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004791 rettv->vval.v_number = syn_name2id(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004792}
4793
4794/*
4795 * "highlight_exists()" function
4796 */
4797 static void
4798f_hlexists(typval_T *argvars, typval_T *rettv)
4799{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004800 rettv->vval.v_number = highlight_exists(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004801}
4802
4803/*
4804 * "hostname()" function
4805 */
4806 static void
4807f_hostname(typval_T *argvars UNUSED, typval_T *rettv)
4808{
4809 char_u hostname[256];
4810
4811 mch_get_host_name(hostname, 256);
4812 rettv->v_type = VAR_STRING;
4813 rettv->vval.v_string = vim_strsave(hostname);
4814}
4815
4816/*
4817 * iconv() function
4818 */
4819 static void
4820f_iconv(typval_T *argvars UNUSED, typval_T *rettv)
4821{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004822 char_u buf1[NUMBUFLEN];
4823 char_u buf2[NUMBUFLEN];
4824 char_u *from, *to, *str;
4825 vimconv_T vimconv;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004826
4827 rettv->v_type = VAR_STRING;
4828 rettv->vval.v_string = NULL;
4829
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004830 str = tv_get_string(&argvars[0]);
4831 from = enc_canonize(enc_skip(tv_get_string_buf(&argvars[1], buf1)));
4832 to = enc_canonize(enc_skip(tv_get_string_buf(&argvars[2], buf2)));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004833 vimconv.vc_type = CONV_NONE;
4834 convert_setup(&vimconv, from, to);
4835
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004836 // If the encodings are equal, no conversion needed.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004837 if (vimconv.vc_type == CONV_NONE)
4838 rettv->vval.v_string = vim_strsave(str);
4839 else
4840 rettv->vval.v_string = string_convert(&vimconv, str, NULL);
4841
4842 convert_setup(&vimconv, NULL, NULL);
4843 vim_free(from);
4844 vim_free(to);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004845}
4846
4847/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004848 * "index()" function
4849 */
4850 static void
4851f_index(typval_T *argvars, typval_T *rettv)
4852{
4853 list_T *l;
4854 listitem_T *item;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004855 blob_T *b;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004856 long idx = 0;
4857 int ic = FALSE;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004858 int error = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004859
4860 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004861 if (argvars[0].v_type == VAR_BLOB)
4862 {
4863 typval_T tv;
4864 int start = 0;
4865
4866 if (argvars[2].v_type != VAR_UNKNOWN)
4867 {
4868 start = tv_get_number_chk(&argvars[2], &error);
4869 if (error)
4870 return;
4871 }
4872 b = argvars[0].vval.v_blob;
4873 if (b == NULL)
4874 return;
Bram Moolenaar05500ec2019-01-13 19:10:33 +01004875 if (start < 0)
4876 {
4877 start = blob_len(b) + start;
4878 if (start < 0)
4879 start = 0;
4880 }
4881
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004882 for (idx = start; idx < blob_len(b); ++idx)
4883 {
4884 tv.v_type = VAR_NUMBER;
4885 tv.vval.v_number = blob_get(b, idx);
4886 if (tv_equal(&tv, &argvars[1], ic, FALSE))
4887 {
4888 rettv->vval.v_number = idx;
4889 return;
4890 }
4891 }
4892 return;
4893 }
4894 else if (argvars[0].v_type != VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004895 {
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01004896 emsg(_(e_listblobreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004897 return;
4898 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004899
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004900 l = argvars[0].vval.v_list;
4901 if (l != NULL)
4902 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01004903 range_list_materialize(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004904 item = l->lv_first;
4905 if (argvars[2].v_type != VAR_UNKNOWN)
4906 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004907 // Start at specified item. Use the cached index that list_find()
4908 // sets, so that a negative number also works.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004909 item = list_find(l, (long)tv_get_number_chk(&argvars[2], &error));
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01004910 idx = l->lv_u.mat.lv_idx;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004911 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004912 ic = (int)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004913 if (error)
4914 item = NULL;
4915 }
4916
4917 for ( ; item != NULL; item = item->li_next, ++idx)
4918 if (tv_equal(&item->li_tv, &argvars[1], ic, FALSE))
4919 {
4920 rettv->vval.v_number = idx;
4921 break;
4922 }
4923 }
4924}
4925
4926static int inputsecret_flag = 0;
4927
4928/*
4929 * "input()" function
4930 * Also handles inputsecret() when inputsecret is set.
4931 */
4932 static void
4933f_input(typval_T *argvars, typval_T *rettv)
4934{
4935 get_user_input(argvars, rettv, FALSE, inputsecret_flag);
4936}
4937
4938/*
4939 * "inputdialog()" function
4940 */
4941 static void
4942f_inputdialog(typval_T *argvars, typval_T *rettv)
4943{
4944#if defined(FEAT_GUI_TEXTDIALOG)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004945 // Use a GUI dialog if the GUI is running and 'c' is not in 'guioptions'
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004946 if (gui.in_use && vim_strchr(p_go, GO_CONDIALOG) == NULL)
4947 {
4948 char_u *message;
4949 char_u buf[NUMBUFLEN];
4950 char_u *defstr = (char_u *)"";
4951
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004952 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004953 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004954 && (defstr = tv_get_string_buf_chk(&argvars[1], buf)) != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004955 vim_strncpy(IObuff, defstr, IOSIZE - 1);
4956 else
4957 IObuff[0] = NUL;
4958 if (message != NULL && defstr != NULL
4959 && do_dialog(VIM_QUESTION, NULL, message,
4960 (char_u *)_("&OK\n&Cancel"), 1, IObuff, FALSE) == 1)
4961 rettv->vval.v_string = vim_strsave(IObuff);
4962 else
4963 {
4964 if (message != NULL && defstr != NULL
4965 && argvars[1].v_type != VAR_UNKNOWN
4966 && argvars[2].v_type != VAR_UNKNOWN)
4967 rettv->vval.v_string = vim_strsave(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004968 tv_get_string_buf(&argvars[2], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004969 else
4970 rettv->vval.v_string = NULL;
4971 }
4972 rettv->v_type = VAR_STRING;
4973 }
4974 else
4975#endif
4976 get_user_input(argvars, rettv, TRUE, inputsecret_flag);
4977}
4978
4979/*
4980 * "inputlist()" function
4981 */
4982 static void
4983f_inputlist(typval_T *argvars, typval_T *rettv)
4984{
Bram Moolenaar50985eb2020-01-27 22:09:39 +01004985 list_T *l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004986 listitem_T *li;
4987 int selected;
4988 int mouse_used;
4989
4990#ifdef NO_CONSOLE_INPUT
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004991 // While starting up, there is no place to enter text. When running tests
4992 // with --not-a-term we assume feedkeys() will be used.
Bram Moolenaar91d348a2017-07-29 20:16:03 +02004993 if (no_console_input() && !is_not_a_term())
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004994 return;
4995#endif
4996 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
4997 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01004998 semsg(_(e_listarg), "inputlist()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004999 return;
5000 }
5001
5002 msg_start();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005003 msg_row = Rows - 1; // for when 'cmdheight' > 1
5004 lines_left = Rows; // avoid more prompt
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005005 msg_scroll = TRUE;
5006 msg_clr_eos();
5007
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005008 l = argvars[0].vval.v_list;
5009 range_list_materialize(l);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02005010 FOR_ALL_LIST_ITEMS(argvars[0].vval.v_list, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005011 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005012 msg_puts((char *)tv_get_string(&li->li_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005013 msg_putchar('\n');
5014 }
5015
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005016 // Ask for choice.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005017 selected = prompt_for_number(&mouse_used);
5018 if (mouse_used)
5019 selected -= lines_left;
5020
5021 rettv->vval.v_number = selected;
5022}
5023
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005024static garray_T ga_userinput = {0, 0, sizeof(tasave_T), 4, NULL};
5025
5026/*
5027 * "inputrestore()" function
5028 */
5029 static void
5030f_inputrestore(typval_T *argvars UNUSED, typval_T *rettv)
5031{
5032 if (ga_userinput.ga_len > 0)
5033 {
5034 --ga_userinput.ga_len;
5035 restore_typeahead((tasave_T *)(ga_userinput.ga_data)
5036 + ga_userinput.ga_len);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005037 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005038 }
5039 else if (p_verbose > 1)
5040 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005041 verb_msg(_("called inputrestore() more often than inputsave()"));
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005042 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005043 }
5044}
5045
5046/*
5047 * "inputsave()" function
5048 */
5049 static void
5050f_inputsave(typval_T *argvars UNUSED, typval_T *rettv)
5051{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005052 // Add an entry to the stack of typeahead storage.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005053 if (ga_grow(&ga_userinput, 1) == OK)
5054 {
5055 save_typeahead((tasave_T *)(ga_userinput.ga_data)
5056 + ga_userinput.ga_len);
5057 ++ga_userinput.ga_len;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005058 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005059 }
5060 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005061 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005062}
5063
5064/*
5065 * "inputsecret()" function
5066 */
5067 static void
5068f_inputsecret(typval_T *argvars, typval_T *rettv)
5069{
5070 ++cmdline_star;
5071 ++inputsecret_flag;
5072 f_input(argvars, rettv);
5073 --cmdline_star;
5074 --inputsecret_flag;
5075}
5076
5077/*
Bram Moolenaar67a2deb2019-11-25 00:05:32 +01005078 * "interrupt()" function
5079 */
5080 static void
5081f_interrupt(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5082{
5083 got_int = TRUE;
5084}
5085
5086/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005087 * "invert(expr)" function
5088 */
5089 static void
5090f_invert(typval_T *argvars, typval_T *rettv)
5091{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005092 rettv->vval.v_number = ~tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005093}
5094
5095/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005096 * Return TRUE if typeval "tv" is locked: Either that value is locked itself
5097 * or it refers to a List or Dictionary that is locked.
5098 */
5099 static int
5100tv_islocked(typval_T *tv)
5101{
5102 return (tv->v_lock & VAR_LOCKED)
5103 || (tv->v_type == VAR_LIST
5104 && tv->vval.v_list != NULL
5105 && (tv->vval.v_list->lv_lock & VAR_LOCKED))
5106 || (tv->v_type == VAR_DICT
5107 && tv->vval.v_dict != NULL
5108 && (tv->vval.v_dict->dv_lock & VAR_LOCKED));
5109}
5110
5111/*
5112 * "islocked()" function
5113 */
5114 static void
5115f_islocked(typval_T *argvars, typval_T *rettv)
5116{
5117 lval_T lv;
5118 char_u *end;
5119 dictitem_T *di;
5120
5121 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005122 end = get_lval(tv_get_string(&argvars[0]), NULL, &lv, FALSE, FALSE,
Bram Moolenaar3a257732017-02-21 20:47:13 +01005123 GLV_NO_AUTOLOAD | GLV_READ_ONLY, FNE_CHECK_START);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005124 if (end != NULL && lv.ll_name != NULL)
5125 {
5126 if (*end != NUL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005127 emsg(_(e_trailing));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005128 else
5129 {
5130 if (lv.ll_tv == NULL)
5131 {
Bram Moolenaar79518e22017-02-17 16:31:35 +01005132 di = find_var(lv.ll_name, NULL, TRUE);
5133 if (di != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005134 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005135 // Consider a variable locked when:
5136 // 1. the variable itself is locked
5137 // 2. the value of the variable is locked.
5138 // 3. the List or Dict value is locked.
Bram Moolenaar79518e22017-02-17 16:31:35 +01005139 rettv->vval.v_number = ((di->di_flags & DI_FLAGS_LOCK)
5140 || tv_islocked(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005141 }
5142 }
5143 else if (lv.ll_range)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005144 emsg(_("E786: Range not allowed"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005145 else if (lv.ll_newkey != NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005146 semsg(_(e_dictkey), lv.ll_newkey);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005147 else if (lv.ll_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005148 // List item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005149 rettv->vval.v_number = tv_islocked(&lv.ll_li->li_tv);
5150 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005151 // Dictionary item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005152 rettv->vval.v_number = tv_islocked(&lv.ll_di->di_tv);
5153 }
5154 }
5155
5156 clear_lval(&lv);
5157}
5158
5159#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
5160/*
Bram Moolenaarfda1bff2019-04-04 13:44:37 +02005161 * "isinf()" function
5162 */
5163 static void
5164f_isinf(typval_T *argvars, typval_T *rettv)
5165{
5166 if (argvars[0].v_type == VAR_FLOAT && isinf(argvars[0].vval.v_float))
5167 rettv->vval.v_number = argvars[0].vval.v_float > 0.0 ? 1 : -1;
5168}
5169
5170/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005171 * "isnan()" function
5172 */
5173 static void
5174f_isnan(typval_T *argvars, typval_T *rettv)
5175{
5176 rettv->vval.v_number = argvars[0].v_type == VAR_FLOAT
5177 && isnan(argvars[0].vval.v_float);
5178}
5179#endif
5180
5181/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005182 * "last_buffer_nr()" function.
5183 */
5184 static void
5185f_last_buffer_nr(typval_T *argvars UNUSED, typval_T *rettv)
5186{
5187 int n = 0;
5188 buf_T *buf;
5189
Bram Moolenaar29323592016-07-24 22:04:11 +02005190 FOR_ALL_BUFFERS(buf)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005191 if (n < buf->b_fnum)
5192 n = buf->b_fnum;
5193
5194 rettv->vval.v_number = n;
5195}
5196
5197/*
5198 * "len()" function
5199 */
5200 static void
5201f_len(typval_T *argvars, typval_T *rettv)
5202{
5203 switch (argvars[0].v_type)
5204 {
5205 case VAR_STRING:
5206 case VAR_NUMBER:
5207 rettv->vval.v_number = (varnumber_T)STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005208 tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005209 break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005210 case VAR_BLOB:
5211 rettv->vval.v_number = blob_len(argvars[0].vval.v_blob);
5212 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005213 case VAR_LIST:
5214 rettv->vval.v_number = list_len(argvars[0].vval.v_list);
5215 break;
5216 case VAR_DICT:
5217 rettv->vval.v_number = dict_len(argvars[0].vval.v_dict);
5218 break;
5219 case VAR_UNKNOWN:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01005220 case VAR_VOID:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01005221 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005222 case VAR_SPECIAL:
5223 case VAR_FLOAT:
5224 case VAR_FUNC:
5225 case VAR_PARTIAL:
5226 case VAR_JOB:
5227 case VAR_CHANNEL:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005228 emsg(_("E701: Invalid type for len()"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005229 break;
5230 }
5231}
5232
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005233 static void
Bram Moolenaar6d721c72017-01-17 16:56:28 +01005234libcall_common(typval_T *argvars UNUSED, typval_T *rettv, int type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005235{
5236#ifdef FEAT_LIBCALL
5237 char_u *string_in;
5238 char_u **string_result;
5239 int nr_result;
5240#endif
5241
5242 rettv->v_type = type;
5243 if (type != VAR_NUMBER)
5244 rettv->vval.v_string = NULL;
5245
5246 if (check_restricted() || check_secure())
5247 return;
5248
5249#ifdef FEAT_LIBCALL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005250 // The first two args must be strings, otherwise it's meaningless
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005251 if (argvars[0].v_type == VAR_STRING && argvars[1].v_type == VAR_STRING)
5252 {
5253 string_in = NULL;
5254 if (argvars[2].v_type == VAR_STRING)
5255 string_in = argvars[2].vval.v_string;
5256 if (type == VAR_NUMBER)
5257 string_result = NULL;
5258 else
5259 string_result = &rettv->vval.v_string;
5260 if (mch_libcall(argvars[0].vval.v_string,
5261 argvars[1].vval.v_string,
5262 string_in,
5263 argvars[2].vval.v_number,
5264 string_result,
5265 &nr_result) == OK
5266 && type == VAR_NUMBER)
5267 rettv->vval.v_number = nr_result;
5268 }
5269#endif
5270}
5271
5272/*
5273 * "libcall()" function
5274 */
5275 static void
5276f_libcall(typval_T *argvars, typval_T *rettv)
5277{
5278 libcall_common(argvars, rettv, VAR_STRING);
5279}
5280
5281/*
5282 * "libcallnr()" function
5283 */
5284 static void
5285f_libcallnr(typval_T *argvars, typval_T *rettv)
5286{
5287 libcall_common(argvars, rettv, VAR_NUMBER);
5288}
5289
5290/*
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005291 * "line(string, [winid])" function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005292 */
5293 static void
5294f_line(typval_T *argvars, typval_T *rettv)
5295{
5296 linenr_T lnum = 0;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005297 pos_T *fp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005298 int fnum;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005299 int id;
5300 tabpage_T *tp;
5301 win_T *wp;
5302 win_T *save_curwin;
5303 tabpage_T *save_curtab;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005304
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005305 if (argvars[1].v_type != VAR_UNKNOWN)
5306 {
5307 // use window specified in the second argument
5308 id = (int)tv_get_number(&argvars[1]);
5309 wp = win_id2wp_tp(id, &tp);
5310 if (wp != NULL && tp != NULL)
5311 {
5312 if (switch_win_noblock(&save_curwin, &save_curtab, wp, tp, TRUE)
5313 == OK)
5314 {
5315 check_cursor();
5316 fp = var2fpos(&argvars[0], TRUE, &fnum);
5317 }
5318 restore_win_noblock(save_curwin, save_curtab, TRUE);
5319 }
5320 }
5321 else
5322 // use current window
5323 fp = var2fpos(&argvars[0], TRUE, &fnum);
5324
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005325 if (fp != NULL)
5326 lnum = fp->lnum;
5327 rettv->vval.v_number = lnum;
5328}
5329
5330/*
5331 * "line2byte(lnum)" function
5332 */
5333 static void
5334f_line2byte(typval_T *argvars UNUSED, typval_T *rettv)
5335{
5336#ifndef FEAT_BYTEOFF
5337 rettv->vval.v_number = -1;
5338#else
5339 linenr_T lnum;
5340
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005341 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005342 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count + 1)
5343 rettv->vval.v_number = -1;
5344 else
5345 rettv->vval.v_number = ml_find_line_or_offset(curbuf, lnum, NULL);
5346 if (rettv->vval.v_number >= 0)
5347 ++rettv->vval.v_number;
5348#endif
5349}
5350
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005351#ifdef FEAT_FLOAT
5352/*
5353 * "log()" function
5354 */
5355 static void
5356f_log(typval_T *argvars, typval_T *rettv)
5357{
5358 float_T f = 0.0;
5359
5360 rettv->v_type = VAR_FLOAT;
5361 if (get_float_arg(argvars, &f) == OK)
5362 rettv->vval.v_float = log(f);
5363 else
5364 rettv->vval.v_float = 0.0;
5365}
5366
5367/*
5368 * "log10()" function
5369 */
5370 static void
5371f_log10(typval_T *argvars, typval_T *rettv)
5372{
5373 float_T f = 0.0;
5374
5375 rettv->v_type = VAR_FLOAT;
5376 if (get_float_arg(argvars, &f) == OK)
5377 rettv->vval.v_float = log10(f);
5378 else
5379 rettv->vval.v_float = 0.0;
5380}
5381#endif
5382
5383#ifdef FEAT_LUA
5384/*
5385 * "luaeval()" function
5386 */
5387 static void
5388f_luaeval(typval_T *argvars, typval_T *rettv)
5389{
5390 char_u *str;
5391 char_u buf[NUMBUFLEN];
5392
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005393 if (check_restricted() || check_secure())
5394 return;
5395
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005396 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005397 do_luaeval(str, argvars + 1, rettv);
5398}
5399#endif
5400
5401/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005402 * "maparg()" function
5403 */
5404 static void
5405f_maparg(typval_T *argvars, typval_T *rettv)
5406{
5407 get_maparg(argvars, rettv, TRUE);
5408}
5409
5410/*
5411 * "mapcheck()" function
5412 */
5413 static void
5414f_mapcheck(typval_T *argvars, typval_T *rettv)
5415{
5416 get_maparg(argvars, rettv, FALSE);
5417}
5418
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005419typedef enum
5420{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005421 MATCH_END, // matchend()
5422 MATCH_MATCH, // match()
5423 MATCH_STR, // matchstr()
5424 MATCH_LIST, // matchlist()
5425 MATCH_POS // matchstrpos()
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005426} matchtype_T;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005427
5428 static void
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005429find_some_match(typval_T *argvars, typval_T *rettv, matchtype_T type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005430{
5431 char_u *str = NULL;
5432 long len = 0;
5433 char_u *expr = NULL;
5434 char_u *pat;
5435 regmatch_T regmatch;
5436 char_u patbuf[NUMBUFLEN];
5437 char_u strbuf[NUMBUFLEN];
5438 char_u *save_cpo;
5439 long start = 0;
5440 long nth = 1;
5441 colnr_T startcol = 0;
5442 int match = 0;
5443 list_T *l = NULL;
5444 listitem_T *li = NULL;
5445 long idx = 0;
5446 char_u *tofree = NULL;
5447
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005448 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005449 save_cpo = p_cpo;
5450 p_cpo = (char_u *)"";
5451
5452 rettv->vval.v_number = -1;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005453 if (type == MATCH_LIST || type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005454 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005455 // type MATCH_LIST: return empty list when there are no matches.
5456 // type MATCH_POS: return ["", -1, -1, -1]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005457 if (rettv_list_alloc(rettv) == FAIL)
5458 goto theend;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005459 if (type == MATCH_POS
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005460 && (list_append_string(rettv->vval.v_list,
5461 (char_u *)"", 0) == FAIL
5462 || list_append_number(rettv->vval.v_list,
5463 (varnumber_T)-1) == FAIL
5464 || list_append_number(rettv->vval.v_list,
5465 (varnumber_T)-1) == FAIL
5466 || list_append_number(rettv->vval.v_list,
5467 (varnumber_T)-1) == FAIL))
5468 {
5469 list_free(rettv->vval.v_list);
5470 rettv->vval.v_list = NULL;
5471 goto theend;
5472 }
5473 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005474 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005475 {
5476 rettv->v_type = VAR_STRING;
5477 rettv->vval.v_string = NULL;
5478 }
5479
5480 if (argvars[0].v_type == VAR_LIST)
5481 {
5482 if ((l = argvars[0].vval.v_list) == NULL)
5483 goto theend;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005484 range_list_materialize(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005485 li = l->lv_first;
5486 }
5487 else
5488 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005489 expr = str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005490 len = (long)STRLEN(str);
5491 }
5492
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005493 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005494 if (pat == NULL)
5495 goto theend;
5496
5497 if (argvars[2].v_type != VAR_UNKNOWN)
5498 {
5499 int error = FALSE;
5500
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005501 start = (long)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005502 if (error)
5503 goto theend;
5504 if (l != NULL)
5505 {
5506 li = list_find(l, start);
5507 if (li == NULL)
5508 goto theend;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01005509 idx = l->lv_u.mat.lv_idx; // use the cached index
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005510 }
5511 else
5512 {
5513 if (start < 0)
5514 start = 0;
5515 if (start > len)
5516 goto theend;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005517 // When "count" argument is there ignore matches before "start",
5518 // otherwise skip part of the string. Differs when pattern is "^"
5519 // or "\<".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005520 if (argvars[3].v_type != VAR_UNKNOWN)
5521 startcol = start;
5522 else
5523 {
5524 str += start;
5525 len -= start;
5526 }
5527 }
5528
5529 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005530 nth = (long)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005531 if (error)
5532 goto theend;
5533 }
5534
5535 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
5536 if (regmatch.regprog != NULL)
5537 {
5538 regmatch.rm_ic = p_ic;
5539
5540 for (;;)
5541 {
5542 if (l != NULL)
5543 {
5544 if (li == NULL)
5545 {
5546 match = FALSE;
5547 break;
5548 }
5549 vim_free(tofree);
5550 expr = str = echo_string(&li->li_tv, &tofree, strbuf, 0);
5551 if (str == NULL)
5552 break;
5553 }
5554
5555 match = vim_regexec_nl(&regmatch, str, (colnr_T)startcol);
5556
5557 if (match && --nth <= 0)
5558 break;
5559 if (l == NULL && !match)
5560 break;
5561
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005562 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005563 if (l != NULL)
5564 {
5565 li = li->li_next;
5566 ++idx;
5567 }
5568 else
5569 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005570 startcol = (colnr_T)(regmatch.startp[0]
5571 + (*mb_ptr2len)(regmatch.startp[0]) - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005572 if (startcol > (colnr_T)len
5573 || str + startcol <= regmatch.startp[0])
5574 {
5575 match = FALSE;
5576 break;
5577 }
5578 }
5579 }
5580
5581 if (match)
5582 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005583 if (type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005584 {
5585 listitem_T *li1 = rettv->vval.v_list->lv_first;
5586 listitem_T *li2 = li1->li_next;
5587 listitem_T *li3 = li2->li_next;
5588 listitem_T *li4 = li3->li_next;
5589
5590 vim_free(li1->li_tv.vval.v_string);
5591 li1->li_tv.vval.v_string = vim_strnsave(regmatch.startp[0],
5592 (int)(regmatch.endp[0] - regmatch.startp[0]));
5593 li3->li_tv.vval.v_number =
5594 (varnumber_T)(regmatch.startp[0] - expr);
5595 li4->li_tv.vval.v_number =
5596 (varnumber_T)(regmatch.endp[0] - expr);
5597 if (l != NULL)
5598 li2->li_tv.vval.v_number = (varnumber_T)idx;
5599 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005600 else if (type == MATCH_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005601 {
5602 int i;
5603
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005604 // return list with matched string and submatches
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005605 for (i = 0; i < NSUBEXP; ++i)
5606 {
5607 if (regmatch.endp[i] == NULL)
5608 {
5609 if (list_append_string(rettv->vval.v_list,
5610 (char_u *)"", 0) == FAIL)
5611 break;
5612 }
5613 else if (list_append_string(rettv->vval.v_list,
5614 regmatch.startp[i],
5615 (int)(regmatch.endp[i] - regmatch.startp[i]))
5616 == FAIL)
5617 break;
5618 }
5619 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005620 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005621 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005622 // return matched string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005623 if (l != NULL)
5624 copy_tv(&li->li_tv, rettv);
5625 else
5626 rettv->vval.v_string = vim_strnsave(regmatch.startp[0],
5627 (int)(regmatch.endp[0] - regmatch.startp[0]));
5628 }
5629 else if (l != NULL)
5630 rettv->vval.v_number = idx;
5631 else
5632 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005633 if (type != MATCH_END)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005634 rettv->vval.v_number =
5635 (varnumber_T)(regmatch.startp[0] - str);
5636 else
5637 rettv->vval.v_number =
5638 (varnumber_T)(regmatch.endp[0] - str);
5639 rettv->vval.v_number += (varnumber_T)(str - expr);
5640 }
5641 }
5642 vim_regfree(regmatch.regprog);
5643 }
5644
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005645theend:
5646 if (type == MATCH_POS && l == NULL && rettv->vval.v_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005647 // matchstrpos() without a list: drop the second item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005648 listitem_remove(rettv->vval.v_list,
5649 rettv->vval.v_list->lv_first->li_next);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005650 vim_free(tofree);
5651 p_cpo = save_cpo;
5652}
5653
5654/*
5655 * "match()" function
5656 */
5657 static void
5658f_match(typval_T *argvars, typval_T *rettv)
5659{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005660 find_some_match(argvars, rettv, MATCH_MATCH);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005661}
5662
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005663/*
5664 * "matchend()" function
5665 */
5666 static void
5667f_matchend(typval_T *argvars, typval_T *rettv)
5668{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005669 find_some_match(argvars, rettv, MATCH_END);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005670}
5671
5672/*
5673 * "matchlist()" function
5674 */
5675 static void
5676f_matchlist(typval_T *argvars, typval_T *rettv)
5677{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005678 find_some_match(argvars, rettv, MATCH_LIST);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005679}
5680
5681/*
5682 * "matchstr()" function
5683 */
5684 static void
5685f_matchstr(typval_T *argvars, typval_T *rettv)
5686{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005687 find_some_match(argvars, rettv, MATCH_STR);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005688}
5689
5690/*
5691 * "matchstrpos()" function
5692 */
5693 static void
5694f_matchstrpos(typval_T *argvars, typval_T *rettv)
5695{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005696 find_some_match(argvars, rettv, MATCH_POS);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005697}
5698
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005699 static void
5700max_min(typval_T *argvars, typval_T *rettv, int domax)
5701{
5702 varnumber_T n = 0;
5703 varnumber_T i;
5704 int error = FALSE;
5705
5706 if (argvars[0].v_type == VAR_LIST)
5707 {
5708 list_T *l;
5709 listitem_T *li;
5710
5711 l = argvars[0].vval.v_list;
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005712 if (l != NULL && l->lv_len > 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005713 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005714 if (l->lv_first == &range_list_item)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005715 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005716 if ((l->lv_u.nonmat.lv_stride > 0) ^ domax)
5717 n = l->lv_u.nonmat.lv_start;
5718 else
5719 n = l->lv_u.nonmat.lv_start + (l->lv_len - 1)
5720 * l->lv_u.nonmat.lv_stride;
5721 }
5722 else
5723 {
5724 li = l->lv_first;
5725 if (li != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005726 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005727 n = tv_get_number_chk(&li->li_tv, &error);
5728 for (;;)
5729 {
5730 li = li->li_next;
5731 if (li == NULL)
5732 break;
5733 i = tv_get_number_chk(&li->li_tv, &error);
5734 if (domax ? i > n : i < n)
5735 n = i;
5736 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005737 }
5738 }
5739 }
5740 }
5741 else if (argvars[0].v_type == VAR_DICT)
5742 {
5743 dict_T *d;
5744 int first = TRUE;
5745 hashitem_T *hi;
5746 int todo;
5747
5748 d = argvars[0].vval.v_dict;
5749 if (d != NULL)
5750 {
5751 todo = (int)d->dv_hashtab.ht_used;
5752 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
5753 {
5754 if (!HASHITEM_EMPTY(hi))
5755 {
5756 --todo;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005757 i = tv_get_number_chk(&HI2DI(hi)->di_tv, &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005758 if (first)
5759 {
5760 n = i;
5761 first = FALSE;
5762 }
5763 else if (domax ? i > n : i < n)
5764 n = i;
5765 }
5766 }
5767 }
5768 }
5769 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005770 semsg(_(e_listdictarg), domax ? "max()" : "min()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005771 rettv->vval.v_number = error ? 0 : n;
5772}
5773
5774/*
5775 * "max()" function
5776 */
5777 static void
5778f_max(typval_T *argvars, typval_T *rettv)
5779{
5780 max_min(argvars, rettv, TRUE);
5781}
5782
5783/*
5784 * "min()" function
5785 */
5786 static void
5787f_min(typval_T *argvars, typval_T *rettv)
5788{
5789 max_min(argvars, rettv, FALSE);
5790}
5791
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005792#if defined(FEAT_MZSCHEME) || defined(PROTO)
5793/*
5794 * "mzeval()" function
5795 */
5796 static void
5797f_mzeval(typval_T *argvars, typval_T *rettv)
5798{
5799 char_u *str;
5800 char_u buf[NUMBUFLEN];
5801
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005802 if (check_restricted() || check_secure())
5803 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005804 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005805 do_mzeval(str, rettv);
5806}
5807
5808 void
5809mzscheme_call_vim(char_u *name, typval_T *args, typval_T *rettv)
5810{
5811 typval_T argvars[3];
5812
5813 argvars[0].v_type = VAR_STRING;
5814 argvars[0].vval.v_string = name;
5815 copy_tv(args, &argvars[1]);
5816 argvars[2].v_type = VAR_UNKNOWN;
5817 f_call(argvars, rettv);
5818 clear_tv(&argvars[1]);
5819}
5820#endif
5821
5822/*
5823 * "nextnonblank()" function
5824 */
5825 static void
5826f_nextnonblank(typval_T *argvars, typval_T *rettv)
5827{
5828 linenr_T lnum;
5829
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005830 for (lnum = tv_get_lnum(argvars); ; ++lnum)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005831 {
5832 if (lnum < 0 || lnum > curbuf->b_ml.ml_line_count)
5833 {
5834 lnum = 0;
5835 break;
5836 }
5837 if (*skipwhite(ml_get(lnum)) != NUL)
5838 break;
5839 }
5840 rettv->vval.v_number = lnum;
5841}
5842
5843/*
5844 * "nr2char()" function
5845 */
5846 static void
5847f_nr2char(typval_T *argvars, typval_T *rettv)
5848{
5849 char_u buf[NUMBUFLEN];
5850
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005851 if (has_mbyte)
5852 {
5853 int utf8 = 0;
5854
5855 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005856 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005857 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01005858 buf[utf_char2bytes((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005859 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005860 buf[(*mb_char2bytes)((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005861 }
5862 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005863 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005864 buf[0] = (char_u)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005865 buf[1] = NUL;
5866 }
5867 rettv->v_type = VAR_STRING;
5868 rettv->vval.v_string = vim_strsave(buf);
5869}
5870
5871/*
5872 * "or(expr, expr)" function
5873 */
5874 static void
5875f_or(typval_T *argvars, typval_T *rettv)
5876{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005877 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
5878 | tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005879}
5880
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005881#ifdef FEAT_PERL
5882/*
5883 * "perleval()" function
5884 */
5885 static void
5886f_perleval(typval_T *argvars, typval_T *rettv)
5887{
5888 char_u *str;
5889 char_u buf[NUMBUFLEN];
5890
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005891 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005892 do_perleval(str, rettv);
5893}
5894#endif
5895
5896#ifdef FEAT_FLOAT
5897/*
5898 * "pow()" function
5899 */
5900 static void
5901f_pow(typval_T *argvars, typval_T *rettv)
5902{
5903 float_T fx = 0.0, fy = 0.0;
5904
5905 rettv->v_type = VAR_FLOAT;
5906 if (get_float_arg(argvars, &fx) == OK
5907 && get_float_arg(&argvars[1], &fy) == OK)
5908 rettv->vval.v_float = pow(fx, fy);
5909 else
5910 rettv->vval.v_float = 0.0;
5911}
5912#endif
5913
5914/*
5915 * "prevnonblank()" function
5916 */
5917 static void
5918f_prevnonblank(typval_T *argvars, typval_T *rettv)
5919{
5920 linenr_T lnum;
5921
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005922 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005923 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count)
5924 lnum = 0;
5925 else
5926 while (lnum >= 1 && *skipwhite(ml_get(lnum)) == NUL)
5927 --lnum;
5928 rettv->vval.v_number = lnum;
5929}
5930
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005931// This dummy va_list is here because:
5932// - passing a NULL pointer doesn't work when va_list isn't a pointer
5933// - locally in the function results in a "used before set" warning
5934// - using va_start() to initialize it gives "function with fixed args" error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005935static va_list ap;
5936
5937/*
5938 * "printf()" function
5939 */
5940 static void
5941f_printf(typval_T *argvars, typval_T *rettv)
5942{
5943 char_u buf[NUMBUFLEN];
5944 int len;
5945 char_u *s;
5946 int saved_did_emsg = did_emsg;
5947 char *fmt;
5948
5949 rettv->v_type = VAR_STRING;
5950 rettv->vval.v_string = NULL;
5951
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005952 // Get the required length, allocate the buffer and do it for real.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005953 did_emsg = FALSE;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005954 fmt = (char *)tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02005955 len = vim_vsnprintf_typval(NULL, 0, fmt, ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005956 if (!did_emsg)
5957 {
5958 s = alloc(len + 1);
5959 if (s != NULL)
5960 {
5961 rettv->vval.v_string = s;
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02005962 (void)vim_vsnprintf_typval((char *)s, len + 1, fmt,
5963 ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005964 }
5965 }
5966 did_emsg |= saved_did_emsg;
5967}
5968
5969/*
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005970 * "pum_getpos()" function
5971 */
5972 static void
5973f_pum_getpos(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5974{
5975 if (rettv_dict_alloc(rettv) != OK)
5976 return;
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005977 pum_set_event_info(rettv->vval.v_dict);
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005978}
5979
5980/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005981 * "pumvisible()" function
5982 */
5983 static void
5984f_pumvisible(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5985{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005986 if (pum_visible())
5987 rettv->vval.v_number = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005988}
5989
5990#ifdef FEAT_PYTHON3
5991/*
5992 * "py3eval()" function
5993 */
5994 static void
5995f_py3eval(typval_T *argvars, typval_T *rettv)
5996{
5997 char_u *str;
5998 char_u buf[NUMBUFLEN];
5999
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006000 if (check_restricted() || check_secure())
6001 return;
6002
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006003 if (p_pyx == 0)
6004 p_pyx = 3;
6005
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006006 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006007 do_py3eval(str, rettv);
6008}
6009#endif
6010
6011#ifdef FEAT_PYTHON
6012/*
6013 * "pyeval()" function
6014 */
6015 static void
6016f_pyeval(typval_T *argvars, typval_T *rettv)
6017{
6018 char_u *str;
6019 char_u buf[NUMBUFLEN];
6020
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006021 if (check_restricted() || check_secure())
6022 return;
6023
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006024 if (p_pyx == 0)
6025 p_pyx = 2;
6026
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006027 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006028 do_pyeval(str, rettv);
6029}
6030#endif
6031
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006032#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
6033/*
6034 * "pyxeval()" function
6035 */
6036 static void
6037f_pyxeval(typval_T *argvars, typval_T *rettv)
6038{
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006039 if (check_restricted() || check_secure())
6040 return;
6041
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006042# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
6043 init_pyxversion();
6044 if (p_pyx == 2)
6045 f_pyeval(argvars, rettv);
6046 else
6047 f_py3eval(argvars, rettv);
6048# elif defined(FEAT_PYTHON)
6049 f_pyeval(argvars, rettv);
6050# elif defined(FEAT_PYTHON3)
6051 f_py3eval(argvars, rettv);
6052# endif
6053}
6054#endif
6055
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006056static UINT32_T srand_seed_for_testing = 0;
6057static int srand_seed_for_testing_is_used = FALSE;
6058
6059 static void
6060f_test_srand_seed(typval_T *argvars, typval_T *rettv UNUSED)
6061{
6062 if (argvars[0].v_type == VAR_UNKNOWN)
6063 srand_seed_for_testing_is_used = FALSE;
6064 else
6065 {
6066 srand_seed_for_testing = (UINT32_T)tv_get_number(&argvars[0]);
6067 srand_seed_for_testing_is_used = TRUE;
6068 }
6069}
6070
6071 static void
6072init_srand(UINT32_T *x)
6073{
6074#ifndef MSWIN
6075 static int dev_urandom_state = NOTDONE; // FAIL or OK once tried
6076#endif
6077
6078 if (srand_seed_for_testing_is_used)
6079 {
6080 *x = srand_seed_for_testing;
6081 return;
6082 }
6083#ifndef MSWIN
6084 if (dev_urandom_state != FAIL)
6085 {
6086 int fd = open("/dev/urandom", O_RDONLY);
6087 struct {
6088 union {
6089 UINT32_T number;
6090 char bytes[sizeof(UINT32_T)];
6091 } contents;
6092 } buf;
6093
6094 // Attempt reading /dev/urandom.
6095 if (fd == -1)
6096 dev_urandom_state = FAIL;
6097 else
6098 {
6099 buf.contents.number = 0;
6100 if (read(fd, buf.contents.bytes, sizeof(UINT32_T))
6101 != sizeof(UINT32_T))
6102 dev_urandom_state = FAIL;
6103 else
6104 {
6105 dev_urandom_state = OK;
6106 *x = buf.contents.number;
6107 }
6108 close(fd);
6109 }
6110 }
6111 if (dev_urandom_state != OK)
6112 // Reading /dev/urandom doesn't work, fall back to time().
6113#endif
6114 *x = vim_time();
6115}
6116
6117#define ROTL(x, k) ((x << k) | (x >> (32 - k)))
6118#define SPLITMIX32(x, z) ( \
6119 z = (x += 0x9e3779b9), \
6120 z = (z ^ (z >> 16)) * 0x85ebca6b, \
6121 z = (z ^ (z >> 13)) * 0xc2b2ae35, \
6122 z ^ (z >> 16) \
6123 )
6124#define SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w) \
6125 result = ROTL(y * 5, 7) * 9; \
6126 t = y << 9; \
6127 z ^= x; \
6128 w ^= y; \
6129 y ^= z, x ^= w; \
6130 z ^= t; \
6131 w = ROTL(w, 11);
6132
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006133/*
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006134 * "rand()" function
6135 */
6136 static void
6137f_rand(typval_T *argvars, typval_T *rettv)
6138{
6139 list_T *l = NULL;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006140 static UINT32_T gx, gy, gz, gw;
6141 static int initialized = FALSE;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006142 listitem_T *lx, *ly, *lz, *lw;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006143 UINT32_T x, y, z, w, t, result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006144
6145 if (argvars[0].v_type == VAR_UNKNOWN)
6146 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006147 // When no argument is given use the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006148 if (initialized == FALSE)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006149 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006150 // Initialize the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006151 init_srand(&x);
6152
6153 gx = SPLITMIX32(x, z);
6154 gy = SPLITMIX32(x, z);
6155 gz = SPLITMIX32(x, z);
6156 gw = SPLITMIX32(x, z);
6157 initialized = TRUE;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006158 }
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006159
6160 SHUFFLE_XOSHIRO128STARSTAR(gx, gy, gz, gw);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006161 }
6162 else if (argvars[0].v_type == VAR_LIST)
6163 {
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006164 l = argvars[0].vval.v_list;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006165 if (l == NULL || list_len(l) != 4)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006166 goto theend;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006167
6168 lx = list_find(l, 0L);
6169 ly = list_find(l, 1L);
6170 lz = list_find(l, 2L);
6171 lw = list_find(l, 3L);
6172 if (lx->li_tv.v_type != VAR_NUMBER) goto theend;
6173 if (ly->li_tv.v_type != VAR_NUMBER) goto theend;
6174 if (lz->li_tv.v_type != VAR_NUMBER) goto theend;
6175 if (lw->li_tv.v_type != VAR_NUMBER) goto theend;
6176 x = (UINT32_T)lx->li_tv.vval.v_number;
6177 y = (UINT32_T)ly->li_tv.vval.v_number;
6178 z = (UINT32_T)lz->li_tv.vval.v_number;
6179 w = (UINT32_T)lw->li_tv.vval.v_number;
6180
6181 SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w);
6182
6183 lx->li_tv.vval.v_number = (varnumber_T)x;
6184 ly->li_tv.vval.v_number = (varnumber_T)y;
6185 lz->li_tv.vval.v_number = (varnumber_T)z;
6186 lw->li_tv.vval.v_number = (varnumber_T)w;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006187 }
6188 else
6189 goto theend;
6190
6191 rettv->v_type = VAR_NUMBER;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006192 rettv->vval.v_number = (varnumber_T)result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006193 return;
6194
6195theend:
6196 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006197 rettv->v_type = VAR_NUMBER;
6198 rettv->vval.v_number = -1;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006199}
6200
6201/*
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006202 * "srand()" function
6203 */
6204 static void
6205f_srand(typval_T *argvars, typval_T *rettv)
6206{
6207 UINT32_T x = 0, z;
6208
6209 if (rettv_list_alloc(rettv) == FAIL)
6210 return;
6211 if (argvars[0].v_type == VAR_UNKNOWN)
6212 {
6213 init_srand(&x);
6214 }
6215 else
6216 {
6217 int error = FALSE;
6218
6219 x = (UINT32_T)tv_get_number_chk(&argvars[0], &error);
6220 if (error)
6221 return;
6222 }
6223
6224 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6225 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6226 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6227 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6228}
6229
6230#undef ROTL
6231#undef SPLITMIX32
6232#undef SHUFFLE_XOSHIRO128STARSTAR
6233
6234/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006235 * "range()" function
6236 */
6237 static void
6238f_range(typval_T *argvars, typval_T *rettv)
6239{
6240 varnumber_T start;
6241 varnumber_T end;
6242 varnumber_T stride = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006243 int error = FALSE;
6244
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006245 start = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006246 if (argvars[1].v_type == VAR_UNKNOWN)
6247 {
6248 end = start - 1;
6249 start = 0;
6250 }
6251 else
6252 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006253 end = tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006254 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006255 stride = tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006256 }
6257
6258 if (error)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006259 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006260 if (stride == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006261 emsg(_("E726: Stride is zero"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006262 else if (stride > 0 ? end + 1 < start : end - 1 > start)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006263 emsg(_("E727: Start past end"));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006264 else if (rettv_list_alloc(rettv) == OK)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006265 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006266 list_T *list = rettv->vval.v_list;
6267
6268 // Create a non-materialized list. This is much more efficient and
6269 // works with ":for". If used otherwise range_list_materialize() must
6270 // be called.
6271 list->lv_first = &range_list_item;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006272 list->lv_u.nonmat.lv_start = start;
6273 list->lv_u.nonmat.lv_end = end;
6274 list->lv_u.nonmat.lv_stride = stride;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01006275 list->lv_len = (end - start) / stride + 1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006276 }
6277}
6278
6279/*
6280 * If "list" is a non-materialized list then materialize it now.
6281 */
6282 void
6283range_list_materialize(list_T *list)
6284{
6285 if (list->lv_first == &range_list_item)
6286 {
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006287 varnumber_T start = list->lv_u.nonmat.lv_start;
6288 varnumber_T end = list->lv_u.nonmat.lv_end;
6289 int stride = list->lv_u.nonmat.lv_stride;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006290 varnumber_T i;
6291
6292 list->lv_first = NULL;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006293 list->lv_u.mat.lv_last = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006294 list->lv_len = 0;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006295 list->lv_u.mat.lv_idx_item = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006296 for (i = start; stride > 0 ? i <= end : i >= end; i += stride)
6297 if (list_append_number(list, (varnumber_T)i) == FAIL)
6298 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006299 }
6300}
6301
Bram Moolenaar0b6d9112018-05-22 20:35:17 +02006302 static void
6303return_register(int regname, typval_T *rettv)
6304{
6305 char_u buf[2] = {0, 0};
6306
6307 buf[0] = (char_u)regname;
6308 rettv->v_type = VAR_STRING;
6309 rettv->vval.v_string = vim_strsave(buf);
6310}
6311
6312/*
6313 * "reg_executing()" function
6314 */
6315 static void
6316f_reg_executing(typval_T *argvars UNUSED, typval_T *rettv)
6317{
6318 return_register(reg_executing, rettv);
6319}
6320
6321/*
6322 * "reg_recording()" function
6323 */
6324 static void
6325f_reg_recording(typval_T *argvars UNUSED, typval_T *rettv)
6326{
6327 return_register(reg_recording, rettv);
6328}
6329
Bram Moolenaar7416f3e2017-03-18 18:10:13 +01006330/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006331 * "rename({from}, {to})" function
6332 */
6333 static void
6334f_rename(typval_T *argvars, typval_T *rettv)
6335{
6336 char_u buf[NUMBUFLEN];
6337
6338 if (check_restricted() || check_secure())
6339 rettv->vval.v_number = -1;
6340 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006341 rettv->vval.v_number = vim_rename(tv_get_string(&argvars[0]),
6342 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006343}
6344
6345/*
6346 * "repeat()" function
6347 */
6348 static void
6349f_repeat(typval_T *argvars, typval_T *rettv)
6350{
6351 char_u *p;
6352 int n;
6353 int slen;
6354 int len;
6355 char_u *r;
6356 int i;
6357
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006358 n = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006359 if (argvars[0].v_type == VAR_LIST)
6360 {
6361 if (rettv_list_alloc(rettv) == OK && argvars[0].vval.v_list != NULL)
6362 while (n-- > 0)
6363 if (list_extend(rettv->vval.v_list,
6364 argvars[0].vval.v_list, NULL) == FAIL)
6365 break;
6366 }
6367 else
6368 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006369 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006370 rettv->v_type = VAR_STRING;
6371 rettv->vval.v_string = NULL;
6372
6373 slen = (int)STRLEN(p);
6374 len = slen * n;
6375 if (len <= 0)
6376 return;
6377
6378 r = alloc(len + 1);
6379 if (r != NULL)
6380 {
6381 for (i = 0; i < n; i++)
6382 mch_memmove(r + i * slen, p, (size_t)slen);
6383 r[len] = NUL;
6384 }
6385
6386 rettv->vval.v_string = r;
6387 }
6388}
6389
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006390#define SP_NOMOVE 0x01 // don't move cursor
6391#define SP_REPEAT 0x02 // repeat to find outer pair
6392#define SP_RETCOUNT 0x04 // return matchcount
6393#define SP_SETPCMARK 0x08 // set previous context mark
6394#define SP_START 0x10 // accept match at start position
6395#define SP_SUBPAT 0x20 // return nr of matching sub-pattern
6396#define SP_END 0x40 // leave cursor at end of match
6397#define SP_COLUMN 0x80 // start at cursor column
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006398
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006399/*
6400 * Get flags for a search function.
6401 * Possibly sets "p_ws".
6402 * Returns BACKWARD, FORWARD or zero (for an error).
6403 */
6404 static int
6405get_search_arg(typval_T *varp, int *flagsp)
6406{
6407 int dir = FORWARD;
6408 char_u *flags;
6409 char_u nbuf[NUMBUFLEN];
6410 int mask;
6411
6412 if (varp->v_type != VAR_UNKNOWN)
6413 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006414 flags = tv_get_string_buf_chk(varp, nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006415 if (flags == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006416 return 0; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006417 while (*flags != NUL)
6418 {
6419 switch (*flags)
6420 {
6421 case 'b': dir = BACKWARD; break;
6422 case 'w': p_ws = TRUE; break;
6423 case 'W': p_ws = FALSE; break;
6424 default: mask = 0;
6425 if (flagsp != NULL)
6426 switch (*flags)
6427 {
6428 case 'c': mask = SP_START; break;
6429 case 'e': mask = SP_END; break;
6430 case 'm': mask = SP_RETCOUNT; break;
6431 case 'n': mask = SP_NOMOVE; break;
6432 case 'p': mask = SP_SUBPAT; break;
6433 case 'r': mask = SP_REPEAT; break;
6434 case 's': mask = SP_SETPCMARK; break;
6435 case 'z': mask = SP_COLUMN; break;
6436 }
6437 if (mask == 0)
6438 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006439 semsg(_(e_invarg2), flags);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006440 dir = 0;
6441 }
6442 else
6443 *flagsp |= mask;
6444 }
6445 if (dir == 0)
6446 break;
6447 ++flags;
6448 }
6449 }
6450 return dir;
6451}
6452
6453/*
6454 * Shared by search() and searchpos() functions.
6455 */
6456 static int
6457search_cmn(typval_T *argvars, pos_T *match_pos, int *flagsp)
6458{
6459 int flags;
6460 char_u *pat;
6461 pos_T pos;
6462 pos_T save_cursor;
6463 int save_p_ws = p_ws;
6464 int dir;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006465 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006466 long lnum_stop = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006467#ifdef FEAT_RELTIME
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006468 proftime_T tm;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006469 long time_limit = 0;
6470#endif
6471 int options = SEARCH_KEEP;
6472 int subpatnum;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006473 searchit_arg_T sia;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006474
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006475 pat = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006476 dir = get_search_arg(&argvars[1], flagsp); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006477 if (dir == 0)
6478 goto theend;
6479 flags = *flagsp;
6480 if (flags & SP_START)
6481 options |= SEARCH_START;
6482 if (flags & SP_END)
6483 options |= SEARCH_END;
6484 if (flags & SP_COLUMN)
6485 options |= SEARCH_COL;
6486
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006487 // Optional arguments: line number to stop searching and timeout.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006488 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN)
6489 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006490 lnum_stop = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006491 if (lnum_stop < 0)
6492 goto theend;
6493#ifdef FEAT_RELTIME
6494 if (argvars[3].v_type != VAR_UNKNOWN)
6495 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006496 time_limit = (long)tv_get_number_chk(&argvars[3], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006497 if (time_limit < 0)
6498 goto theend;
6499 }
6500#endif
6501 }
6502
6503#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006504 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006505 profile_setlimit(time_limit, &tm);
6506#endif
6507
6508 /*
6509 * This function does not accept SP_REPEAT and SP_RETCOUNT flags.
6510 * Check to make sure only those flags are set.
6511 * Also, Only the SP_NOMOVE or the SP_SETPCMARK flag can be set. Both
6512 * flags cannot be set. Check for that condition also.
6513 */
6514 if (((flags & (SP_REPEAT | SP_RETCOUNT)) != 0)
6515 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
6516 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006517 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006518 goto theend;
6519 }
6520
6521 pos = save_cursor = curwin->w_cursor;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006522 vim_memset(&sia, 0, sizeof(sia));
6523 sia.sa_stop_lnum = (linenr_T)lnum_stop;
6524#ifdef FEAT_RELTIME
6525 sia.sa_tm = &tm;
6526#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01006527 subpatnum = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006528 options, RE_SEARCH, &sia);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006529 if (subpatnum != FAIL)
6530 {
6531 if (flags & SP_SUBPAT)
6532 retval = subpatnum;
6533 else
6534 retval = pos.lnum;
6535 if (flags & SP_SETPCMARK)
6536 setpcmark();
6537 curwin->w_cursor = pos;
6538 if (match_pos != NULL)
6539 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006540 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006541 match_pos->lnum = pos.lnum;
6542 match_pos->col = pos.col + 1;
6543 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006544 // "/$" will put the cursor after the end of the line, may need to
6545 // correct that here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006546 check_cursor();
6547 }
6548
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006549 // If 'n' flag is used: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006550 if (flags & SP_NOMOVE)
6551 curwin->w_cursor = save_cursor;
6552 else
6553 curwin->w_set_curswant = TRUE;
6554theend:
6555 p_ws = save_p_ws;
6556
6557 return retval;
6558}
6559
6560#ifdef FEAT_FLOAT
6561
6562/*
6563 * round() is not in C90, use ceil() or floor() instead.
6564 */
6565 float_T
6566vim_round(float_T f)
6567{
6568 return f > 0 ? floor(f + 0.5) : ceil(f - 0.5);
6569}
6570
6571/*
6572 * "round({float})" function
6573 */
6574 static void
6575f_round(typval_T *argvars, typval_T *rettv)
6576{
6577 float_T f = 0.0;
6578
6579 rettv->v_type = VAR_FLOAT;
6580 if (get_float_arg(argvars, &f) == OK)
6581 rettv->vval.v_float = vim_round(f);
6582 else
6583 rettv->vval.v_float = 0.0;
6584}
6585#endif
6586
Bram Moolenaare99be0e2019-03-26 22:51:09 +01006587#ifdef FEAT_RUBY
6588/*
6589 * "rubyeval()" function
6590 */
6591 static void
6592f_rubyeval(typval_T *argvars, typval_T *rettv)
6593{
6594 char_u *str;
6595 char_u buf[NUMBUFLEN];
6596
6597 str = tv_get_string_buf(&argvars[0], buf);
6598 do_rubyeval(str, rettv);
6599}
6600#endif
6601
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006602/*
6603 * "screenattr()" function
6604 */
6605 static void
6606f_screenattr(typval_T *argvars, typval_T *rettv)
6607{
6608 int row;
6609 int col;
6610 int c;
6611
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006612 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6613 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006614 if (row < 0 || row >= screen_Rows
6615 || col < 0 || col >= screen_Columns)
6616 c = -1;
6617 else
6618 c = ScreenAttrs[LineOffset[row] + col];
6619 rettv->vval.v_number = c;
6620}
6621
6622/*
6623 * "screenchar()" function
6624 */
6625 static void
6626f_screenchar(typval_T *argvars, typval_T *rettv)
6627{
6628 int row;
6629 int col;
6630 int off;
6631 int c;
6632
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006633 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6634 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006635 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006636 c = -1;
6637 else
6638 {
6639 off = LineOffset[row] + col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006640 if (enc_utf8 && ScreenLinesUC[off] != 0)
6641 c = ScreenLinesUC[off];
6642 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006643 c = ScreenLines[off];
6644 }
6645 rettv->vval.v_number = c;
6646}
6647
6648/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006649 * "screenchars()" function
6650 */
6651 static void
6652f_screenchars(typval_T *argvars, typval_T *rettv)
6653{
6654 int row;
6655 int col;
6656 int off;
6657 int c;
6658 int i;
6659
6660 if (rettv_list_alloc(rettv) == FAIL)
6661 return;
6662 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6663 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
6664 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
6665 return;
6666
6667 off = LineOffset[row] + col;
6668 if (enc_utf8 && ScreenLinesUC[off] != 0)
6669 c = ScreenLinesUC[off];
6670 else
6671 c = ScreenLines[off];
6672 list_append_number(rettv->vval.v_list, (varnumber_T)c);
6673
6674 if (enc_utf8)
6675
6676 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
6677 list_append_number(rettv->vval.v_list,
6678 (varnumber_T)ScreenLinesC[i][off]);
6679}
6680
6681/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006682 * "screencol()" function
6683 *
6684 * First column is 1 to be consistent with virtcol().
6685 */
6686 static void
6687f_screencol(typval_T *argvars UNUSED, typval_T *rettv)
6688{
6689 rettv->vval.v_number = screen_screencol() + 1;
6690}
6691
6692/*
6693 * "screenrow()" function
6694 */
6695 static void
6696f_screenrow(typval_T *argvars UNUSED, typval_T *rettv)
6697{
6698 rettv->vval.v_number = screen_screenrow() + 1;
6699}
6700
6701/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006702 * "screenstring()" function
6703 */
6704 static void
6705f_screenstring(typval_T *argvars, typval_T *rettv)
6706{
6707 int row;
6708 int col;
6709 int off;
6710 int c;
6711 int i;
6712 char_u buf[MB_MAXBYTES + 1];
6713 int buflen = 0;
6714
6715 rettv->vval.v_string = NULL;
6716 rettv->v_type = VAR_STRING;
6717
6718 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6719 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
6720 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
6721 return;
6722
6723 off = LineOffset[row] + col;
6724 if (enc_utf8 && ScreenLinesUC[off] != 0)
6725 c = ScreenLinesUC[off];
6726 else
6727 c = ScreenLines[off];
6728 buflen += mb_char2bytes(c, buf);
6729
6730 if (enc_utf8)
6731 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
6732 buflen += mb_char2bytes(ScreenLinesC[i][off], buf + buflen);
6733
6734 buf[buflen] = NUL;
6735 rettv->vval.v_string = vim_strsave(buf);
6736}
6737
6738/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006739 * "search()" function
6740 */
6741 static void
6742f_search(typval_T *argvars, typval_T *rettv)
6743{
6744 int flags = 0;
6745
6746 rettv->vval.v_number = search_cmn(argvars, NULL, &flags);
6747}
6748
6749/*
6750 * "searchdecl()" function
6751 */
6752 static void
6753f_searchdecl(typval_T *argvars, typval_T *rettv)
6754{
6755 int locally = 1;
6756 int thisblock = 0;
6757 int error = FALSE;
6758 char_u *name;
6759
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006760 rettv->vval.v_number = 1; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006761
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006762 name = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006763 if (argvars[1].v_type != VAR_UNKNOWN)
6764 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006765 locally = (int)tv_get_number_chk(&argvars[1], &error) == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006766 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006767 thisblock = (int)tv_get_number_chk(&argvars[2], &error) != 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006768 }
6769 if (!error && name != NULL)
6770 rettv->vval.v_number = find_decl(name, (int)STRLEN(name),
6771 locally, thisblock, SEARCH_KEEP) == FAIL;
6772}
6773
6774/*
6775 * Used by searchpair() and searchpairpos()
6776 */
6777 static int
6778searchpair_cmn(typval_T *argvars, pos_T *match_pos)
6779{
6780 char_u *spat, *mpat, *epat;
Bram Moolenaar48570482017-10-30 21:48:41 +01006781 typval_T *skip;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006782 int save_p_ws = p_ws;
6783 int dir;
6784 int flags = 0;
6785 char_u nbuf1[NUMBUFLEN];
6786 char_u nbuf2[NUMBUFLEN];
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006787 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006788 long lnum_stop = 0;
6789 long time_limit = 0;
6790
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006791 // Get the three pattern arguments: start, middle, end. Will result in an
6792 // error if not a valid argument.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006793 spat = tv_get_string_chk(&argvars[0]);
6794 mpat = tv_get_string_buf_chk(&argvars[1], nbuf1);
6795 epat = tv_get_string_buf_chk(&argvars[2], nbuf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006796 if (spat == NULL || mpat == NULL || epat == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006797 goto theend; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006798
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006799 // Handle the optional fourth argument: flags
6800 dir = get_search_arg(&argvars[3], &flags); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006801 if (dir == 0)
6802 goto theend;
6803
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006804 // Don't accept SP_END or SP_SUBPAT.
6805 // Only one of the SP_NOMOVE or SP_SETPCMARK flags can be set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006806 if ((flags & (SP_END | SP_SUBPAT)) != 0
6807 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
6808 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006809 semsg(_(e_invarg2), tv_get_string(&argvars[3]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006810 goto theend;
6811 }
6812
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006813 // Using 'r' implies 'W', otherwise it doesn't work.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006814 if (flags & SP_REPEAT)
6815 p_ws = FALSE;
6816
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006817 // Optional fifth argument: skip expression
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006818 if (argvars[3].v_type == VAR_UNKNOWN
6819 || argvars[4].v_type == VAR_UNKNOWN)
Bram Moolenaar48570482017-10-30 21:48:41 +01006820 skip = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006821 else
6822 {
Bram Moolenaar48570482017-10-30 21:48:41 +01006823 skip = &argvars[4];
6824 if (skip->v_type != VAR_FUNC && skip->v_type != VAR_PARTIAL
6825 && skip->v_type != VAR_STRING)
6826 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006827 // Type error
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006828 semsg(_(e_invarg2), tv_get_string(&argvars[4]));
Bram Moolenaar48570482017-10-30 21:48:41 +01006829 goto theend;
6830 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006831 if (argvars[5].v_type != VAR_UNKNOWN)
6832 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006833 lnum_stop = (long)tv_get_number_chk(&argvars[5], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006834 if (lnum_stop < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006835 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006836 semsg(_(e_invarg2), tv_get_string(&argvars[5]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006837 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006838 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006839#ifdef FEAT_RELTIME
6840 if (argvars[6].v_type != VAR_UNKNOWN)
6841 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006842 time_limit = (long)tv_get_number_chk(&argvars[6], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006843 if (time_limit < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006844 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006845 semsg(_(e_invarg2), tv_get_string(&argvars[6]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006846 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006847 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006848 }
6849#endif
6850 }
6851 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006852
6853 retval = do_searchpair(spat, mpat, epat, dir, skip, flags,
6854 match_pos, lnum_stop, time_limit);
6855
6856theend:
6857 p_ws = save_p_ws;
6858
6859 return retval;
6860}
6861
6862/*
6863 * "searchpair()" function
6864 */
6865 static void
6866f_searchpair(typval_T *argvars, typval_T *rettv)
6867{
6868 rettv->vval.v_number = searchpair_cmn(argvars, NULL);
6869}
6870
6871/*
6872 * "searchpairpos()" function
6873 */
6874 static void
6875f_searchpairpos(typval_T *argvars, typval_T *rettv)
6876{
6877 pos_T match_pos;
6878 int lnum = 0;
6879 int col = 0;
6880
6881 if (rettv_list_alloc(rettv) == FAIL)
6882 return;
6883
6884 if (searchpair_cmn(argvars, &match_pos) > 0)
6885 {
6886 lnum = match_pos.lnum;
6887 col = match_pos.col;
6888 }
6889
6890 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
6891 list_append_number(rettv->vval.v_list, (varnumber_T)col);
6892}
6893
6894/*
6895 * Search for a start/middle/end thing.
6896 * Used by searchpair(), see its documentation for the details.
6897 * Returns 0 or -1 for no match,
6898 */
6899 long
6900do_searchpair(
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006901 char_u *spat, // start pattern
6902 char_u *mpat, // middle pattern
6903 char_u *epat, // end pattern
6904 int dir, // BACKWARD or FORWARD
6905 typval_T *skip, // skip expression
6906 int flags, // SP_SETPCMARK and other SP_ values
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006907 pos_T *match_pos,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006908 linenr_T lnum_stop, // stop at this line if not zero
6909 long time_limit UNUSED) // stop after this many msec
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006910{
6911 char_u *save_cpo;
6912 char_u *pat, *pat2 = NULL, *pat3 = NULL;
6913 long retval = 0;
6914 pos_T pos;
6915 pos_T firstpos;
6916 pos_T foundpos;
6917 pos_T save_cursor;
6918 pos_T save_pos;
6919 int n;
6920 int r;
6921 int nest = 1;
Bram Moolenaar48570482017-10-30 21:48:41 +01006922 int use_skip = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006923 int err;
6924 int options = SEARCH_KEEP;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006925#ifdef FEAT_RELTIME
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006926 proftime_T tm;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006927#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006928
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006929 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006930 save_cpo = p_cpo;
6931 p_cpo = empty_option;
6932
6933#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006934 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006935 profile_setlimit(time_limit, &tm);
6936#endif
6937
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006938 // Make two search patterns: start/end (pat2, for in nested pairs) and
6939 // start/middle/end (pat3, for the top pair).
Bram Moolenaar964b3742019-05-24 18:54:09 +02006940 pat2 = alloc(STRLEN(spat) + STRLEN(epat) + 17);
6941 pat3 = alloc(STRLEN(spat) + STRLEN(mpat) + STRLEN(epat) + 25);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006942 if (pat2 == NULL || pat3 == NULL)
6943 goto theend;
Bram Moolenaar6e450a52017-01-06 20:03:58 +01006944 sprintf((char *)pat2, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)", spat, epat);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006945 if (*mpat == NUL)
6946 STRCPY(pat3, pat2);
6947 else
Bram Moolenaar6e450a52017-01-06 20:03:58 +01006948 sprintf((char *)pat3, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)\\|\\(%s\\m\\)",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006949 spat, epat, mpat);
6950 if (flags & SP_START)
6951 options |= SEARCH_START;
6952
Bram Moolenaar48570482017-10-30 21:48:41 +01006953 if (skip != NULL)
6954 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006955 // Empty string means to not use the skip expression.
Bram Moolenaar48570482017-10-30 21:48:41 +01006956 if (skip->v_type == VAR_STRING || skip->v_type == VAR_FUNC)
6957 use_skip = skip->vval.v_string != NULL
6958 && *skip->vval.v_string != NUL;
6959 }
6960
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006961 save_cursor = curwin->w_cursor;
6962 pos = curwin->w_cursor;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006963 CLEAR_POS(&firstpos);
6964 CLEAR_POS(&foundpos);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006965 pat = pat3;
6966 for (;;)
6967 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006968 searchit_arg_T sia;
6969
6970 vim_memset(&sia, 0, sizeof(sia));
6971 sia.sa_stop_lnum = lnum_stop;
6972#ifdef FEAT_RELTIME
6973 sia.sa_tm = &tm;
6974#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01006975 n = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006976 options, RE_SEARCH, &sia);
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006977 if (n == FAIL || (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos)))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006978 // didn't find it or found the first match again: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006979 break;
6980
6981 if (firstpos.lnum == 0)
6982 firstpos = pos;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006983 if (EQUAL_POS(pos, foundpos))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006984 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006985 // Found the same position again. Can happen with a pattern that
6986 // has "\zs" at the end and searching backwards. Advance one
6987 // character and try again.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006988 if (dir == BACKWARD)
6989 decl(&pos);
6990 else
6991 incl(&pos);
6992 }
6993 foundpos = pos;
6994
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006995 // clear the start flag to avoid getting stuck here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006996 options &= ~SEARCH_START;
6997
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006998 // If the skip pattern matches, ignore this match.
Bram Moolenaar48570482017-10-30 21:48:41 +01006999 if (use_skip)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007000 {
7001 save_pos = curwin->w_cursor;
7002 curwin->w_cursor = pos;
Bram Moolenaar48570482017-10-30 21:48:41 +01007003 err = FALSE;
7004 r = eval_expr_to_bool(skip, &err);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007005 curwin->w_cursor = save_pos;
7006 if (err)
7007 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007008 // Evaluating {skip} caused an error, break here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007009 curwin->w_cursor = save_cursor;
7010 retval = -1;
7011 break;
7012 }
7013 if (r)
7014 continue;
7015 }
7016
7017 if ((dir == BACKWARD && n == 3) || (dir == FORWARD && n == 2))
7018 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007019 // Found end when searching backwards or start when searching
7020 // forward: nested pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007021 ++nest;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007022 pat = pat2; // nested, don't search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007023 }
7024 else
7025 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007026 // Found end when searching forward or start when searching
7027 // backward: end of (nested) pair; or found middle in outer pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007028 if (--nest == 1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007029 pat = pat3; // outer level, search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007030 }
7031
7032 if (nest == 0)
7033 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007034 // Found the match: return matchcount or line number.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007035 if (flags & SP_RETCOUNT)
7036 ++retval;
7037 else
7038 retval = pos.lnum;
7039 if (flags & SP_SETPCMARK)
7040 setpcmark();
7041 curwin->w_cursor = pos;
7042 if (!(flags & SP_REPEAT))
7043 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007044 nest = 1; // search for next unmatched
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007045 }
7046 }
7047
7048 if (match_pos != NULL)
7049 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007050 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007051 match_pos->lnum = curwin->w_cursor.lnum;
7052 match_pos->col = curwin->w_cursor.col + 1;
7053 }
7054
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007055 // If 'n' flag is used or search failed: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007056 if ((flags & SP_NOMOVE) || retval == 0)
7057 curwin->w_cursor = save_cursor;
7058
7059theend:
7060 vim_free(pat2);
7061 vim_free(pat3);
7062 if (p_cpo == empty_option)
7063 p_cpo = save_cpo;
7064 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007065 // Darn, evaluating the {skip} expression changed the value.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007066 free_string_option(save_cpo);
7067
7068 return retval;
7069}
7070
7071/*
7072 * "searchpos()" function
7073 */
7074 static void
7075f_searchpos(typval_T *argvars, typval_T *rettv)
7076{
7077 pos_T match_pos;
7078 int lnum = 0;
7079 int col = 0;
7080 int n;
7081 int flags = 0;
7082
7083 if (rettv_list_alloc(rettv) == FAIL)
7084 return;
7085
7086 n = search_cmn(argvars, &match_pos, &flags);
7087 if (n > 0)
7088 {
7089 lnum = match_pos.lnum;
7090 col = match_pos.col;
7091 }
7092
7093 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
7094 list_append_number(rettv->vval.v_list, (varnumber_T)col);
7095 if (flags & SP_SUBPAT)
7096 list_append_number(rettv->vval.v_list, (varnumber_T)n);
7097}
7098
7099 static void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007100f_setcharsearch(typval_T *argvars, typval_T *rettv UNUSED)
7101{
7102 dict_T *d;
7103 dictitem_T *di;
7104 char_u *csearch;
7105
7106 if (argvars[0].v_type != VAR_DICT)
7107 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007108 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007109 return;
7110 }
7111
7112 if ((d = argvars[0].vval.v_dict) != NULL)
7113 {
Bram Moolenaar8f667172018-12-14 15:38:31 +01007114 csearch = dict_get_string(d, (char_u *)"char", FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007115 if (csearch != NULL)
7116 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007117 if (enc_utf8)
7118 {
7119 int pcc[MAX_MCO];
7120 int c = utfc_ptr2char(csearch, pcc);
7121
7122 set_last_csearch(c, csearch, utfc_ptr2len(csearch));
7123 }
7124 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007125 set_last_csearch(PTR2CHAR(csearch),
Bram Moolenaar1614a142019-10-06 22:00:13 +02007126 csearch, mb_ptr2len(csearch));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007127 }
7128
7129 di = dict_find(d, (char_u *)"forward", -1);
7130 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007131 set_csearch_direction((int)tv_get_number(&di->di_tv)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007132 ? FORWARD : BACKWARD);
7133
7134 di = dict_find(d, (char_u *)"until", -1);
7135 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007136 set_csearch_until(!!tv_get_number(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007137 }
7138}
7139
7140/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02007141 * "setenv()" function
7142 */
7143 static void
7144f_setenv(typval_T *argvars, typval_T *rettv UNUSED)
7145{
7146 char_u namebuf[NUMBUFLEN];
7147 char_u valbuf[NUMBUFLEN];
7148 char_u *name = tv_get_string_buf(&argvars[0], namebuf);
7149
7150 if (argvars[1].v_type == VAR_SPECIAL
7151 && argvars[1].vval.v_number == VVAL_NULL)
7152 vim_unsetenv(name);
7153 else
7154 vim_setenv(name, tv_get_string_buf(&argvars[1], valbuf));
7155}
7156
7157/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007158 * "setfperm({fname}, {mode})" function
7159 */
7160 static void
7161f_setfperm(typval_T *argvars, typval_T *rettv)
7162{
7163 char_u *fname;
7164 char_u modebuf[NUMBUFLEN];
7165 char_u *mode_str;
7166 int i;
7167 int mask;
7168 int mode = 0;
7169
7170 rettv->vval.v_number = 0;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007171 fname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007172 if (fname == NULL)
7173 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007174 mode_str = tv_get_string_buf_chk(&argvars[1], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007175 if (mode_str == NULL)
7176 return;
7177 if (STRLEN(mode_str) != 9)
7178 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007179 semsg(_(e_invarg2), mode_str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007180 return;
7181 }
7182
7183 mask = 1;
7184 for (i = 8; i >= 0; --i)
7185 {
7186 if (mode_str[i] != '-')
7187 mode |= mask;
7188 mask = mask << 1;
7189 }
7190 rettv->vval.v_number = mch_setperm(fname, mode) == OK;
7191}
7192
7193/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007194 * "setpos()" function
7195 */
7196 static void
7197f_setpos(typval_T *argvars, typval_T *rettv)
7198{
7199 pos_T pos;
7200 int fnum;
7201 char_u *name;
7202 colnr_T curswant = -1;
7203
7204 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007205 name = tv_get_string_chk(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007206 if (name != NULL)
7207 {
7208 if (list2fpos(&argvars[1], &pos, &fnum, &curswant) == OK)
7209 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01007210 if (pos.col != MAXCOL && --pos.col < 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007211 pos.col = 0;
7212 if (name[0] == '.' && name[1] == NUL)
7213 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007214 // set cursor; "fnum" is ignored
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007215 curwin->w_cursor = pos;
7216 if (curswant >= 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007217 {
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007218 curwin->w_curswant = curswant - 1;
7219 curwin->w_set_curswant = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007220 }
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007221 check_cursor();
7222 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007223 }
7224 else if (name[0] == '\'' && name[1] != NUL && name[2] == NUL)
7225 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007226 // set mark
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007227 if (setmark_pos(name[1], &pos, fnum) == OK)
7228 rettv->vval.v_number = 0;
7229 }
7230 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007231 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007232 }
7233 }
7234}
7235
7236/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007237 * "setreg()" function
7238 */
7239 static void
7240f_setreg(typval_T *argvars, typval_T *rettv)
7241{
7242 int regname;
7243 char_u *strregname;
7244 char_u *stropt;
7245 char_u *strval;
7246 int append;
7247 char_u yank_type;
7248 long block_len;
7249
7250 block_len = -1;
7251 yank_type = MAUTO;
7252 append = FALSE;
7253
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007254 strregname = tv_get_string_chk(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007255 rettv->vval.v_number = 1; // FAIL is default
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007256
7257 if (strregname == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007258 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007259 regname = *strregname;
7260 if (regname == 0 || regname == '@')
7261 regname = '"';
7262
7263 if (argvars[2].v_type != VAR_UNKNOWN)
7264 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007265 stropt = tv_get_string_chk(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007266 if (stropt == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007267 return; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007268 for (; *stropt != NUL; ++stropt)
7269 switch (*stropt)
7270 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007271 case 'a': case 'A': // append
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007272 append = TRUE;
7273 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007274 case 'v': case 'c': // character-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007275 yank_type = MCHAR;
7276 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007277 case 'V': case 'l': // line-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007278 yank_type = MLINE;
7279 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007280 case 'b': case Ctrl_V: // block-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007281 yank_type = MBLOCK;
7282 if (VIM_ISDIGIT(stropt[1]))
7283 {
7284 ++stropt;
7285 block_len = getdigits(&stropt) - 1;
7286 --stropt;
7287 }
7288 break;
7289 }
7290 }
7291
7292 if (argvars[1].v_type == VAR_LIST)
7293 {
7294 char_u **lstval;
7295 char_u **allocval;
7296 char_u buf[NUMBUFLEN];
7297 char_u **curval;
7298 char_u **curallocval;
7299 list_T *ll = argvars[1].vval.v_list;
7300 listitem_T *li;
7301 int len;
7302
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007303 // If the list is NULL handle like an empty list.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007304 len = ll == NULL ? 0 : ll->lv_len;
7305
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007306 // First half: use for pointers to result lines; second half: use for
7307 // pointers to allocated copies.
Bram Moolenaarc799fe22019-05-28 23:08:19 +02007308 lstval = ALLOC_MULT(char_u *, (len + 1) * 2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007309 if (lstval == NULL)
7310 return;
7311 curval = lstval;
7312 allocval = lstval + len + 2;
7313 curallocval = allocval;
7314
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007315 if (ll != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007316 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007317 range_list_materialize(ll);
7318 for (li = ll->lv_first; li != NULL; li = li->li_next)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007319 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007320 strval = tv_get_string_buf_chk(&li->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007321 if (strval == NULL)
7322 goto free_lstval;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007323 if (strval == buf)
7324 {
7325 // Need to make a copy, next tv_get_string_buf_chk() will
7326 // overwrite the string.
7327 strval = vim_strsave(buf);
7328 if (strval == NULL)
7329 goto free_lstval;
7330 *curallocval++ = strval;
7331 }
7332 *curval++ = strval;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007333 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007334 }
7335 *curval++ = NULL;
7336
7337 write_reg_contents_lst(regname, lstval, -1,
7338 append, yank_type, block_len);
7339free_lstval:
7340 while (curallocval > allocval)
7341 vim_free(*--curallocval);
7342 vim_free(lstval);
7343 }
7344 else
7345 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007346 strval = tv_get_string_chk(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007347 if (strval == NULL)
7348 return;
7349 write_reg_contents_ex(regname, strval, -1,
7350 append, yank_type, block_len);
7351 }
7352 rettv->vval.v_number = 0;
7353}
7354
7355/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007356 * "settagstack()" function
7357 */
7358 static void
7359f_settagstack(typval_T *argvars, typval_T *rettv)
7360{
7361 static char *e_invact2 = N_("E962: Invalid action: '%s'");
7362 win_T *wp;
7363 dict_T *d;
7364 int action = 'r';
7365
7366 rettv->vval.v_number = -1;
7367
7368 // first argument: window number or id
7369 wp = find_win_by_nr_or_id(&argvars[0]);
7370 if (wp == NULL)
7371 return;
7372
7373 // second argument: dict with items to set in the tag stack
7374 if (argvars[1].v_type != VAR_DICT)
7375 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007376 emsg(_(e_dictreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007377 return;
7378 }
7379 d = argvars[1].vval.v_dict;
7380 if (d == NULL)
7381 return;
7382
7383 // third argument: action - 'a' for append and 'r' for replace.
7384 // default is to replace the stack.
7385 if (argvars[2].v_type == VAR_UNKNOWN)
7386 action = 'r';
7387 else if (argvars[2].v_type == VAR_STRING)
7388 {
7389 char_u *actstr;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007390 actstr = tv_get_string_chk(&argvars[2]);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007391 if (actstr == NULL)
7392 return;
Bram Moolenaar271fa082020-01-02 14:02:16 +01007393 if ((*actstr == 'r' || *actstr == 'a' || *actstr == 't')
7394 && actstr[1] == NUL)
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007395 action = *actstr;
7396 else
7397 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007398 semsg(_(e_invact2), actstr);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007399 return;
7400 }
7401 }
7402 else
7403 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007404 emsg(_(e_stringreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007405 return;
7406 }
7407
7408 if (set_tagstack(wp, d, action) == OK)
7409 rettv->vval.v_number = 0;
7410}
7411
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007412#ifdef FEAT_CRYPT
7413/*
7414 * "sha256({string})" function
7415 */
7416 static void
7417f_sha256(typval_T *argvars, typval_T *rettv)
7418{
7419 char_u *p;
7420
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007421 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007422 rettv->vval.v_string = vim_strsave(
7423 sha256_bytes(p, (int)STRLEN(p), NULL, 0));
7424 rettv->v_type = VAR_STRING;
7425}
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007426#endif // FEAT_CRYPT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007427
7428/*
7429 * "shellescape({string})" function
7430 */
7431 static void
7432f_shellescape(typval_T *argvars, typval_T *rettv)
7433{
Bram Moolenaar20615522017-06-05 18:46:26 +02007434 int do_special = non_zero_arg(&argvars[1]);
7435
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007436 rettv->vval.v_string = vim_strsave_shellescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007437 tv_get_string(&argvars[0]), do_special, do_special);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007438 rettv->v_type = VAR_STRING;
7439}
7440
7441/*
7442 * shiftwidth() function
7443 */
7444 static void
7445f_shiftwidth(typval_T *argvars UNUSED, typval_T *rettv)
7446{
Bram Moolenaarf9514162018-11-22 03:08:29 +01007447 rettv->vval.v_number = 0;
7448
7449 if (argvars[0].v_type != VAR_UNKNOWN)
7450 {
7451 long col;
7452
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007453 col = (long)tv_get_number_chk(argvars, NULL);
Bram Moolenaarf9514162018-11-22 03:08:29 +01007454 if (col < 0)
7455 return; // type error; errmsg already given
7456#ifdef FEAT_VARTABS
7457 rettv->vval.v_number = get_sw_value_col(curbuf, col);
7458 return;
7459#endif
7460 }
7461
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007462 rettv->vval.v_number = get_sw_value(curbuf);
7463}
7464
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007465#ifdef FEAT_FLOAT
7466/*
7467 * "sin()" function
7468 */
7469 static void
7470f_sin(typval_T *argvars, typval_T *rettv)
7471{
7472 float_T f = 0.0;
7473
7474 rettv->v_type = VAR_FLOAT;
7475 if (get_float_arg(argvars, &f) == OK)
7476 rettv->vval.v_float = sin(f);
7477 else
7478 rettv->vval.v_float = 0.0;
7479}
7480
7481/*
7482 * "sinh()" function
7483 */
7484 static void
7485f_sinh(typval_T *argvars, typval_T *rettv)
7486{
7487 float_T f = 0.0;
7488
7489 rettv->v_type = VAR_FLOAT;
7490 if (get_float_arg(argvars, &f) == OK)
7491 rettv->vval.v_float = sinh(f);
7492 else
7493 rettv->vval.v_float = 0.0;
7494}
7495#endif
7496
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007497/*
7498 * "soundfold({word})" function
7499 */
7500 static void
7501f_soundfold(typval_T *argvars, typval_T *rettv)
7502{
7503 char_u *s;
7504
7505 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007506 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007507#ifdef FEAT_SPELL
7508 rettv->vval.v_string = eval_soundfold(s);
7509#else
7510 rettv->vval.v_string = vim_strsave(s);
7511#endif
7512}
7513
7514/*
7515 * "spellbadword()" function
7516 */
7517 static void
7518f_spellbadword(typval_T *argvars UNUSED, typval_T *rettv)
7519{
7520 char_u *word = (char_u *)"";
7521 hlf_T attr = HLF_COUNT;
7522 int len = 0;
7523
7524 if (rettv_list_alloc(rettv) == FAIL)
7525 return;
7526
7527#ifdef FEAT_SPELL
7528 if (argvars[0].v_type == VAR_UNKNOWN)
7529 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007530 // Find the start and length of the badly spelled word.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007531 len = spell_move_to(curwin, FORWARD, TRUE, TRUE, &attr);
7532 if (len != 0)
Bram Moolenaarb73fa622017-12-21 20:27:47 +01007533 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007534 word = ml_get_cursor();
Bram Moolenaarb73fa622017-12-21 20:27:47 +01007535 curwin->w_set_curswant = TRUE;
7536 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007537 }
7538 else if (curwin->w_p_spell && *curbuf->b_s.b_p_spl != NUL)
7539 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007540 char_u *str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007541 int capcol = -1;
7542
7543 if (str != NULL)
7544 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007545 // Check the argument for spelling.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007546 while (*str != NUL)
7547 {
7548 len = spell_check(curwin, str, &attr, &capcol, FALSE);
7549 if (attr != HLF_COUNT)
7550 {
7551 word = str;
7552 break;
7553 }
7554 str += len;
Bram Moolenaar66ab9162018-07-20 20:28:48 +02007555 capcol -= len;
Bram Moolenaar0c779e82019-08-09 17:01:02 +02007556 len = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007557 }
7558 }
7559 }
7560#endif
7561
7562 list_append_string(rettv->vval.v_list, word, len);
7563 list_append_string(rettv->vval.v_list, (char_u *)(
7564 attr == HLF_SPB ? "bad" :
7565 attr == HLF_SPR ? "rare" :
7566 attr == HLF_SPL ? "local" :
7567 attr == HLF_SPC ? "caps" :
7568 ""), -1);
7569}
7570
7571/*
7572 * "spellsuggest()" function
7573 */
7574 static void
7575f_spellsuggest(typval_T *argvars UNUSED, typval_T *rettv)
7576{
7577#ifdef FEAT_SPELL
7578 char_u *str;
7579 int typeerr = FALSE;
7580 int maxcount;
7581 garray_T ga;
7582 int i;
7583 listitem_T *li;
7584 int need_capital = FALSE;
7585#endif
7586
7587 if (rettv_list_alloc(rettv) == FAIL)
7588 return;
7589
7590#ifdef FEAT_SPELL
7591 if (curwin->w_p_spell && *curwin->w_s->b_p_spl != NUL)
7592 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007593 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007594 if (argvars[1].v_type != VAR_UNKNOWN)
7595 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007596 maxcount = (int)tv_get_number_chk(&argvars[1], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007597 if (maxcount <= 0)
7598 return;
7599 if (argvars[2].v_type != VAR_UNKNOWN)
7600 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007601 need_capital = (int)tv_get_number_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007602 if (typeerr)
7603 return;
7604 }
7605 }
7606 else
7607 maxcount = 25;
7608
7609 spell_suggest_list(&ga, str, maxcount, need_capital, FALSE);
7610
7611 for (i = 0; i < ga.ga_len; ++i)
7612 {
7613 str = ((char_u **)ga.ga_data)[i];
7614
7615 li = listitem_alloc();
7616 if (li == NULL)
7617 vim_free(str);
7618 else
7619 {
7620 li->li_tv.v_type = VAR_STRING;
7621 li->li_tv.v_lock = 0;
7622 li->li_tv.vval.v_string = str;
7623 list_append(rettv->vval.v_list, li);
7624 }
7625 }
7626 ga_clear(&ga);
7627 }
7628#endif
7629}
7630
7631 static void
7632f_split(typval_T *argvars, typval_T *rettv)
7633{
7634 char_u *str;
7635 char_u *end;
7636 char_u *pat = NULL;
7637 regmatch_T regmatch;
7638 char_u patbuf[NUMBUFLEN];
7639 char_u *save_cpo;
7640 int match;
7641 colnr_T col = 0;
7642 int keepempty = FALSE;
7643 int typeerr = FALSE;
7644
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007645 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007646 save_cpo = p_cpo;
7647 p_cpo = (char_u *)"";
7648
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007649 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007650 if (argvars[1].v_type != VAR_UNKNOWN)
7651 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007652 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007653 if (pat == NULL)
7654 typeerr = TRUE;
7655 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007656 keepempty = (int)tv_get_number_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007657 }
7658 if (pat == NULL || *pat == NUL)
7659 pat = (char_u *)"[\\x01- ]\\+";
7660
7661 if (rettv_list_alloc(rettv) == FAIL)
7662 return;
7663 if (typeerr)
7664 return;
7665
7666 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
7667 if (regmatch.regprog != NULL)
7668 {
7669 regmatch.rm_ic = FALSE;
7670 while (*str != NUL || keepempty)
7671 {
7672 if (*str == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007673 match = FALSE; // empty item at the end
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007674 else
7675 match = vim_regexec_nl(&regmatch, str, col);
7676 if (match)
7677 end = regmatch.startp[0];
7678 else
7679 end = str + STRLEN(str);
7680 if (keepempty || end > str || (rettv->vval.v_list->lv_len > 0
7681 && *str != NUL && match && end < regmatch.endp[0]))
7682 {
7683 if (list_append_string(rettv->vval.v_list, str,
7684 (int)(end - str)) == FAIL)
7685 break;
7686 }
7687 if (!match)
7688 break;
Bram Moolenaar13505972019-01-24 15:04:48 +01007689 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007690 if (regmatch.endp[0] > str)
7691 col = 0;
7692 else
Bram Moolenaar13505972019-01-24 15:04:48 +01007693 // Don't get stuck at the same match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007694 col = (*mb_ptr2len)(regmatch.endp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007695 str = regmatch.endp[0];
7696 }
7697
7698 vim_regfree(regmatch.regprog);
7699 }
7700
7701 p_cpo = save_cpo;
7702}
7703
7704#ifdef FEAT_FLOAT
7705/*
7706 * "sqrt()" function
7707 */
7708 static void
7709f_sqrt(typval_T *argvars, typval_T *rettv)
7710{
7711 float_T f = 0.0;
7712
7713 rettv->v_type = VAR_FLOAT;
7714 if (get_float_arg(argvars, &f) == OK)
7715 rettv->vval.v_float = sqrt(f);
7716 else
7717 rettv->vval.v_float = 0.0;
7718}
Bram Moolenaar0387cae2019-11-29 21:07:58 +01007719#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007720
Bram Moolenaar0387cae2019-11-29 21:07:58 +01007721#ifdef FEAT_FLOAT
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007722/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007723 * "str2float()" function
7724 */
7725 static void
7726f_str2float(typval_T *argvars, typval_T *rettv)
7727{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007728 char_u *p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01007729 int isneg = (*p == '-');
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007730
Bram Moolenaar08243d22017-01-10 16:12:29 +01007731 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007732 p = skipwhite(p + 1);
7733 (void)string2float(p, &rettv->vval.v_float);
Bram Moolenaar08243d22017-01-10 16:12:29 +01007734 if (isneg)
7735 rettv->vval.v_float *= -1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007736 rettv->v_type = VAR_FLOAT;
7737}
7738#endif
7739
7740/*
Bram Moolenaar9d401282019-04-06 13:18:12 +02007741 * "str2list()" function
7742 */
7743 static void
7744f_str2list(typval_T *argvars, typval_T *rettv)
7745{
7746 char_u *p;
7747 int utf8 = FALSE;
7748
7749 if (rettv_list_alloc(rettv) == FAIL)
7750 return;
7751
7752 if (argvars[1].v_type != VAR_UNKNOWN)
7753 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
7754
7755 p = tv_get_string(&argvars[0]);
7756
7757 if (has_mbyte || utf8)
7758 {
7759 int (*ptr2len)(char_u *);
7760 int (*ptr2char)(char_u *);
7761
7762 if (utf8 || enc_utf8)
7763 {
7764 ptr2len = utf_ptr2len;
7765 ptr2char = utf_ptr2char;
7766 }
7767 else
7768 {
7769 ptr2len = mb_ptr2len;
7770 ptr2char = mb_ptr2char;
7771 }
7772
7773 for ( ; *p != NUL; p += (*ptr2len)(p))
7774 list_append_number(rettv->vval.v_list, (*ptr2char)(p));
7775 }
7776 else
7777 for ( ; *p != NUL; ++p)
7778 list_append_number(rettv->vval.v_list, *p);
7779}
7780
7781/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007782 * "str2nr()" function
7783 */
7784 static void
7785f_str2nr(typval_T *argvars, typval_T *rettv)
7786{
7787 int base = 10;
7788 char_u *p;
7789 varnumber_T n;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007790 int what = 0;
Bram Moolenaar08243d22017-01-10 16:12:29 +01007791 int isneg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007792
7793 if (argvars[1].v_type != VAR_UNKNOWN)
7794 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007795 base = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007796 if (base != 2 && base != 8 && base != 10 && base != 16)
7797 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007798 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007799 return;
7800 }
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007801 if (argvars[2].v_type != VAR_UNKNOWN && tv_get_number(&argvars[2]))
7802 what |= STR2NR_QUOTE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007803 }
7804
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007805 p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01007806 isneg = (*p == '-');
7807 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007808 p = skipwhite(p + 1);
7809 switch (base)
7810 {
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007811 case 2: what |= STR2NR_BIN + STR2NR_FORCE; break;
7812 case 8: what |= STR2NR_OCT + STR2NR_FORCE; break;
7813 case 16: what |= STR2NR_HEX + STR2NR_FORCE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007814 }
Bram Moolenaar16e9b852019-05-19 19:59:35 +02007815 vim_str2nr(p, NULL, NULL, what, &n, NULL, 0, FALSE);
7816 // Text after the number is silently ignored.
Bram Moolenaar08243d22017-01-10 16:12:29 +01007817 if (isneg)
7818 rettv->vval.v_number = -n;
7819 else
7820 rettv->vval.v_number = n;
7821
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007822}
7823
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007824/*
7825 * "strgetchar()" function
7826 */
7827 static void
7828f_strgetchar(typval_T *argvars, typval_T *rettv)
7829{
7830 char_u *str;
7831 int len;
7832 int error = FALSE;
7833 int charidx;
Bram Moolenaar13505972019-01-24 15:04:48 +01007834 int byteidx = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007835
7836 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007837 str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007838 if (str == NULL)
7839 return;
7840 len = (int)STRLEN(str);
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007841 charidx = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007842 if (error)
7843 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007844
Bram Moolenaar13505972019-01-24 15:04:48 +01007845 while (charidx >= 0 && byteidx < len)
7846 {
7847 if (charidx == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007848 {
Bram Moolenaar13505972019-01-24 15:04:48 +01007849 rettv->vval.v_number = mb_ptr2char(str + byteidx);
7850 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007851 }
Bram Moolenaar13505972019-01-24 15:04:48 +01007852 --charidx;
7853 byteidx += MB_CPTR2LEN(str + byteidx);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007854 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007855}
7856
7857/*
7858 * "stridx()" function
7859 */
7860 static void
7861f_stridx(typval_T *argvars, typval_T *rettv)
7862{
7863 char_u buf[NUMBUFLEN];
7864 char_u *needle;
7865 char_u *haystack;
7866 char_u *save_haystack;
7867 char_u *pos;
7868 int start_idx;
7869
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007870 needle = tv_get_string_chk(&argvars[1]);
7871 save_haystack = haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007872 rettv->vval.v_number = -1;
7873 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007874 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007875
7876 if (argvars[2].v_type != VAR_UNKNOWN)
7877 {
7878 int error = FALSE;
7879
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007880 start_idx = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007881 if (error || start_idx >= (int)STRLEN(haystack))
7882 return;
7883 if (start_idx >= 0)
7884 haystack += start_idx;
7885 }
7886
7887 pos = (char_u *)strstr((char *)haystack, (char *)needle);
7888 if (pos != NULL)
7889 rettv->vval.v_number = (varnumber_T)(pos - save_haystack);
7890}
7891
7892/*
7893 * "string()" function
7894 */
Bram Moolenaar461a7fc2018-12-22 13:28:07 +01007895 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007896f_string(typval_T *argvars, typval_T *rettv)
7897{
7898 char_u *tofree;
7899 char_u numbuf[NUMBUFLEN];
7900
7901 rettv->v_type = VAR_STRING;
7902 rettv->vval.v_string = tv2string(&argvars[0], &tofree, numbuf,
7903 get_copyID());
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007904 // Make a copy if we have a value but it's not in allocated memory.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007905 if (rettv->vval.v_string != NULL && tofree == NULL)
7906 rettv->vval.v_string = vim_strsave(rettv->vval.v_string);
7907}
7908
7909/*
7910 * "strlen()" function
7911 */
7912 static void
7913f_strlen(typval_T *argvars, typval_T *rettv)
7914{
7915 rettv->vval.v_number = (varnumber_T)(STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007916 tv_get_string(&argvars[0])));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007917}
7918
7919/*
7920 * "strchars()" function
7921 */
7922 static void
7923f_strchars(typval_T *argvars, typval_T *rettv)
7924{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007925 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007926 int skipcc = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007927 varnumber_T len = 0;
7928 int (*func_mb_ptr2char_adv)(char_u **pp);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007929
7930 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007931 skipcc = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007932 if (skipcc < 0 || skipcc > 1)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007933 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007934 else
7935 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007936 func_mb_ptr2char_adv = skipcc ? mb_ptr2char_adv : mb_cptr2char_adv;
7937 while (*s != NUL)
7938 {
7939 func_mb_ptr2char_adv(&s);
7940 ++len;
7941 }
7942 rettv->vval.v_number = len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007943 }
7944}
7945
7946/*
7947 * "strdisplaywidth()" function
7948 */
7949 static void
7950f_strdisplaywidth(typval_T *argvars, typval_T *rettv)
7951{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007952 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007953 int col = 0;
7954
7955 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007956 col = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007957
7958 rettv->vval.v_number = (varnumber_T)(linetabsize_col(col, s) - col);
7959}
7960
7961/*
7962 * "strwidth()" function
7963 */
7964 static void
7965f_strwidth(typval_T *argvars, typval_T *rettv)
7966{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007967 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007968
Bram Moolenaar13505972019-01-24 15:04:48 +01007969 rettv->vval.v_number = (varnumber_T)(mb_string2cells(s, -1));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007970}
7971
7972/*
7973 * "strcharpart()" function
7974 */
7975 static void
7976f_strcharpart(typval_T *argvars, typval_T *rettv)
7977{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007978 char_u *p;
7979 int nchar;
7980 int nbyte = 0;
7981 int charlen;
7982 int len = 0;
7983 int slen;
7984 int error = FALSE;
7985
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007986 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007987 slen = (int)STRLEN(p);
7988
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007989 nchar = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007990 if (!error)
7991 {
7992 if (nchar > 0)
7993 while (nchar > 0 && nbyte < slen)
7994 {
Bram Moolenaard3c907b2016-08-17 21:32:09 +02007995 nbyte += MB_CPTR2LEN(p + nbyte);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007996 --nchar;
7997 }
7998 else
7999 nbyte = nchar;
8000 if (argvars[2].v_type != VAR_UNKNOWN)
8001 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008002 charlen = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008003 while (charlen > 0 && nbyte + len < slen)
8004 {
8005 int off = nbyte + len;
8006
8007 if (off < 0)
8008 len += 1;
8009 else
Bram Moolenaard3c907b2016-08-17 21:32:09 +02008010 len += MB_CPTR2LEN(p + off);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008011 --charlen;
8012 }
8013 }
8014 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008015 len = slen - nbyte; // default: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008016 }
8017
8018 /*
8019 * Only return the overlap between the specified part and the actual
8020 * string.
8021 */
8022 if (nbyte < 0)
8023 {
8024 len += nbyte;
8025 nbyte = 0;
8026 }
8027 else if (nbyte > slen)
8028 nbyte = slen;
8029 if (len < 0)
8030 len = 0;
8031 else if (nbyte + len > slen)
8032 len = slen - nbyte;
8033
8034 rettv->v_type = VAR_STRING;
8035 rettv->vval.v_string = vim_strnsave(p + nbyte, len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008036}
8037
8038/*
8039 * "strpart()" function
8040 */
8041 static void
8042f_strpart(typval_T *argvars, typval_T *rettv)
8043{
8044 char_u *p;
8045 int n;
8046 int len;
8047 int slen;
8048 int error = FALSE;
8049
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008050 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008051 slen = (int)STRLEN(p);
8052
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008053 n = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008054 if (error)
8055 len = 0;
8056 else if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008057 len = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008058 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008059 len = slen - n; // default len: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008060
8061 /*
8062 * Only return the overlap between the specified part and the actual
8063 * string.
8064 */
8065 if (n < 0)
8066 {
8067 len += n;
8068 n = 0;
8069 }
8070 else if (n > slen)
8071 n = slen;
8072 if (len < 0)
8073 len = 0;
8074 else if (n + len > slen)
8075 len = slen - n;
8076
8077 rettv->v_type = VAR_STRING;
8078 rettv->vval.v_string = vim_strnsave(p + n, len);
8079}
8080
8081/*
8082 * "strridx()" function
8083 */
8084 static void
8085f_strridx(typval_T *argvars, typval_T *rettv)
8086{
8087 char_u buf[NUMBUFLEN];
8088 char_u *needle;
8089 char_u *haystack;
8090 char_u *rest;
8091 char_u *lastmatch = NULL;
8092 int haystack_len, end_idx;
8093
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008094 needle = tv_get_string_chk(&argvars[1]);
8095 haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008096
8097 rettv->vval.v_number = -1;
8098 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008099 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008100
8101 haystack_len = (int)STRLEN(haystack);
8102 if (argvars[2].v_type != VAR_UNKNOWN)
8103 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008104 // Third argument: upper limit for index
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008105 end_idx = (int)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008106 if (end_idx < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008107 return; // can never find a match
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008108 }
8109 else
8110 end_idx = haystack_len;
8111
8112 if (*needle == NUL)
8113 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008114 // Empty string matches past the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008115 lastmatch = haystack + end_idx;
8116 }
8117 else
8118 {
8119 for (rest = haystack; *rest != '\0'; ++rest)
8120 {
8121 rest = (char_u *)strstr((char *)rest, (char *)needle);
8122 if (rest == NULL || rest > haystack + end_idx)
8123 break;
8124 lastmatch = rest;
8125 }
8126 }
8127
8128 if (lastmatch == NULL)
8129 rettv->vval.v_number = -1;
8130 else
8131 rettv->vval.v_number = (varnumber_T)(lastmatch - haystack);
8132}
8133
8134/*
8135 * "strtrans()" function
8136 */
8137 static void
8138f_strtrans(typval_T *argvars, typval_T *rettv)
8139{
8140 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008141 rettv->vval.v_string = transstr(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008142}
8143
8144/*
8145 * "submatch()" function
8146 */
8147 static void
8148f_submatch(typval_T *argvars, typval_T *rettv)
8149{
8150 int error = FALSE;
8151 int no;
8152 int retList = 0;
8153
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008154 no = (int)tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008155 if (error)
8156 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02008157 if (no < 0 || no >= NSUBEXP)
8158 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008159 semsg(_("E935: invalid submatch number: %d"), no);
Bram Moolenaar79518e22017-02-17 16:31:35 +01008160 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02008161 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008162 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008163 retList = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008164 if (error)
8165 return;
8166
8167 if (retList == 0)
8168 {
8169 rettv->v_type = VAR_STRING;
8170 rettv->vval.v_string = reg_submatch(no);
8171 }
8172 else
8173 {
8174 rettv->v_type = VAR_LIST;
8175 rettv->vval.v_list = reg_submatch_list(no);
8176 }
8177}
8178
8179/*
8180 * "substitute()" function
8181 */
8182 static void
8183f_substitute(typval_T *argvars, typval_T *rettv)
8184{
8185 char_u patbuf[NUMBUFLEN];
8186 char_u subbuf[NUMBUFLEN];
8187 char_u flagsbuf[NUMBUFLEN];
8188
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008189 char_u *str = tv_get_string_chk(&argvars[0]);
8190 char_u *pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008191 char_u *sub = NULL;
8192 typval_T *expr = NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008193 char_u *flg = tv_get_string_buf_chk(&argvars[3], flagsbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008194
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008195 if (argvars[2].v_type == VAR_FUNC || argvars[2].v_type == VAR_PARTIAL)
8196 expr = &argvars[2];
8197 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008198 sub = tv_get_string_buf_chk(&argvars[2], subbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008199
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008200 rettv->v_type = VAR_STRING;
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008201 if (str == NULL || pat == NULL || (sub == NULL && expr == NULL)
8202 || flg == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008203 rettv->vval.v_string = NULL;
8204 else
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008205 rettv->vval.v_string = do_string_sub(str, pat, sub, expr, flg);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008206}
8207
8208/*
Bram Moolenaar00f123a2018-08-21 20:28:54 +02008209 * "swapinfo(swap_filename)" function
8210 */
8211 static void
8212f_swapinfo(typval_T *argvars, typval_T *rettv)
8213{
8214 if (rettv_dict_alloc(rettv) == OK)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008215 get_b0_dict(tv_get_string(argvars), rettv->vval.v_dict);
Bram Moolenaar00f123a2018-08-21 20:28:54 +02008216}
8217
8218/*
Bram Moolenaar110bd602018-09-16 18:46:59 +02008219 * "swapname(expr)" function
8220 */
8221 static void
8222f_swapname(typval_T *argvars, typval_T *rettv)
8223{
8224 buf_T *buf;
8225
8226 rettv->v_type = VAR_STRING;
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01008227 buf = tv_get_buf(&argvars[0], FALSE);
Bram Moolenaar110bd602018-09-16 18:46:59 +02008228 if (buf == NULL || buf->b_ml.ml_mfp == NULL
8229 || buf->b_ml.ml_mfp->mf_fname == NULL)
8230 rettv->vval.v_string = NULL;
8231 else
8232 rettv->vval.v_string = vim_strsave(buf->b_ml.ml_mfp->mf_fname);
8233}
8234
8235/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008236 * "synID(lnum, col, trans)" function
8237 */
8238 static void
8239f_synID(typval_T *argvars UNUSED, typval_T *rettv)
8240{
8241 int id = 0;
8242#ifdef FEAT_SYN_HL
8243 linenr_T lnum;
8244 colnr_T col;
8245 int trans;
8246 int transerr = FALSE;
8247
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008248 lnum = tv_get_lnum(argvars); // -1 on type error
8249 col = (linenr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008250 trans = (int)tv_get_number_chk(&argvars[2], &transerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008251
8252 if (!transerr && lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8253 && col >= 0 && col < (long)STRLEN(ml_get(lnum)))
8254 id = syn_get_id(curwin, lnum, (colnr_T)col, trans, NULL, FALSE);
8255#endif
8256
8257 rettv->vval.v_number = id;
8258}
8259
8260/*
8261 * "synIDattr(id, what [, mode])" function
8262 */
8263 static void
8264f_synIDattr(typval_T *argvars UNUSED, typval_T *rettv)
8265{
8266 char_u *p = NULL;
8267#ifdef FEAT_SYN_HL
8268 int id;
8269 char_u *what;
8270 char_u *mode;
8271 char_u modebuf[NUMBUFLEN];
8272 int modec;
8273
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008274 id = (int)tv_get_number(&argvars[0]);
8275 what = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008276 if (argvars[2].v_type != VAR_UNKNOWN)
8277 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008278 mode = tv_get_string_buf(&argvars[2], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008279 modec = TOLOWER_ASC(mode[0]);
8280 if (modec != 't' && modec != 'c' && modec != 'g')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008281 modec = 0; // replace invalid with current
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008282 }
8283 else
8284 {
8285#if defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS)
8286 if (USE_24BIT)
8287 modec = 'g';
8288 else
8289#endif
8290 if (t_colors > 1)
8291 modec = 'c';
8292 else
8293 modec = 't';
8294 }
8295
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008296 switch (TOLOWER_ASC(what[0]))
8297 {
8298 case 'b':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008299 if (TOLOWER_ASC(what[1]) == 'g') // bg[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008300 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008301 else // bold
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008302 p = highlight_has_attr(id, HL_BOLD, modec);
8303 break;
8304
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008305 case 'f': // fg[#] or font
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008306 p = highlight_color(id, what, modec);
8307 break;
8308
8309 case 'i':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008310 if (TOLOWER_ASC(what[1]) == 'n') // inverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008311 p = highlight_has_attr(id, HL_INVERSE, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008312 else // italic
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008313 p = highlight_has_attr(id, HL_ITALIC, modec);
8314 break;
8315
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008316 case 'n': // name
Bram Moolenaarc96272e2017-03-26 13:50:09 +02008317 p = get_highlight_name_ext(NULL, id - 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008318 break;
8319
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008320 case 'r': // reverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008321 p = highlight_has_attr(id, HL_INVERSE, modec);
8322 break;
8323
8324 case 's':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008325 if (TOLOWER_ASC(what[1]) == 'p') // sp[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008326 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008327 // strikeout
Bram Moolenaarcf4b00c2017-09-02 18:33:56 +02008328 else if (TOLOWER_ASC(what[1]) == 't' &&
8329 TOLOWER_ASC(what[2]) == 'r')
8330 p = highlight_has_attr(id, HL_STRIKETHROUGH, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008331 else // standout
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008332 p = highlight_has_attr(id, HL_STANDOUT, modec);
8333 break;
8334
8335 case 'u':
8336 if (STRLEN(what) <= 5 || TOLOWER_ASC(what[5]) != 'c')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008337 // underline
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008338 p = highlight_has_attr(id, HL_UNDERLINE, modec);
8339 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008340 // undercurl
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008341 p = highlight_has_attr(id, HL_UNDERCURL, modec);
8342 break;
8343 }
8344
8345 if (p != NULL)
8346 p = vim_strsave(p);
8347#endif
8348 rettv->v_type = VAR_STRING;
8349 rettv->vval.v_string = p;
8350}
8351
8352/*
8353 * "synIDtrans(id)" function
8354 */
8355 static void
8356f_synIDtrans(typval_T *argvars UNUSED, typval_T *rettv)
8357{
8358 int id;
8359
8360#ifdef FEAT_SYN_HL
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008361 id = (int)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008362
8363 if (id > 0)
8364 id = syn_get_final_id(id);
8365 else
8366#endif
8367 id = 0;
8368
8369 rettv->vval.v_number = id;
8370}
8371
8372/*
8373 * "synconcealed(lnum, col)" function
8374 */
8375 static void
8376f_synconcealed(typval_T *argvars UNUSED, typval_T *rettv)
8377{
8378#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
8379 linenr_T lnum;
8380 colnr_T col;
8381 int syntax_flags = 0;
8382 int cchar;
8383 int matchid = 0;
8384 char_u str[NUMBUFLEN];
8385#endif
8386
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02008387 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008388
8389#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008390 lnum = tv_get_lnum(argvars); // -1 on type error
8391 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008392
8393 vim_memset(str, NUL, sizeof(str));
8394
8395 if (rettv_list_alloc(rettv) != FAIL)
8396 {
8397 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8398 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
8399 && curwin->w_p_cole > 0)
8400 {
8401 (void)syn_get_id(curwin, lnum, col, FALSE, NULL, FALSE);
8402 syntax_flags = get_syntax_info(&matchid);
8403
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008404 // get the conceal character
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008405 if ((syntax_flags & HL_CONCEAL) && curwin->w_p_cole < 3)
8406 {
8407 cchar = syn_get_sub_char();
Bram Moolenaar4d785892017-06-22 22:00:50 +02008408 if (cchar == NUL && curwin->w_p_cole == 1)
8409 cchar = (lcs_conceal == NUL) ? ' ' : lcs_conceal;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008410 if (cchar != NUL)
8411 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008412 if (has_mbyte)
8413 (*mb_char2bytes)(cchar, str);
8414 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008415 str[0] = cchar;
8416 }
8417 }
8418 }
8419
8420 list_append_number(rettv->vval.v_list,
8421 (syntax_flags & HL_CONCEAL) != 0);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008422 // -1 to auto-determine strlen
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008423 list_append_string(rettv->vval.v_list, str, -1);
8424 list_append_number(rettv->vval.v_list, matchid);
8425 }
8426#endif
8427}
8428
8429/*
8430 * "synstack(lnum, col)" function
8431 */
8432 static void
8433f_synstack(typval_T *argvars UNUSED, typval_T *rettv)
8434{
8435#ifdef FEAT_SYN_HL
8436 linenr_T lnum;
8437 colnr_T col;
8438 int i;
8439 int id;
8440#endif
8441
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02008442 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008443
8444#ifdef FEAT_SYN_HL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008445 lnum = tv_get_lnum(argvars); // -1 on type error
8446 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008447
8448 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8449 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
8450 && rettv_list_alloc(rettv) != FAIL)
8451 {
8452 (void)syn_get_id(curwin, lnum, (colnr_T)col, FALSE, NULL, TRUE);
8453 for (i = 0; ; ++i)
8454 {
8455 id = syn_get_stack_item(i);
8456 if (id < 0)
8457 break;
8458 if (list_append_number(rettv->vval.v_list, id) == FAIL)
8459 break;
8460 }
8461 }
8462#endif
8463}
8464
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008465/*
8466 * "tabpagebuflist()" function
8467 */
8468 static void
8469f_tabpagebuflist(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8470{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008471 tabpage_T *tp;
8472 win_T *wp = NULL;
8473
8474 if (argvars[0].v_type == VAR_UNKNOWN)
8475 wp = firstwin;
8476 else
8477 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008478 tp = find_tabpage((int)tv_get_number(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008479 if (tp != NULL)
8480 wp = (tp == curtab) ? firstwin : tp->tp_firstwin;
8481 }
8482 if (wp != NULL && rettv_list_alloc(rettv) != FAIL)
8483 {
8484 for (; wp != NULL; wp = wp->w_next)
8485 if (list_append_number(rettv->vval.v_list,
8486 wp->w_buffer->b_fnum) == FAIL)
8487 break;
8488 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008489}
8490
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008491/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008492 * "tagfiles()" function
8493 */
8494 static void
8495f_tagfiles(typval_T *argvars UNUSED, typval_T *rettv)
8496{
8497 char_u *fname;
8498 tagname_T tn;
8499 int first;
8500
8501 if (rettv_list_alloc(rettv) == FAIL)
8502 return;
8503 fname = alloc(MAXPATHL);
8504 if (fname == NULL)
8505 return;
8506
8507 for (first = TRUE; ; first = FALSE)
8508 if (get_tagfname(&tn, first, fname) == FAIL
8509 || list_append_string(rettv->vval.v_list, fname, -1) == FAIL)
8510 break;
8511 tagname_free(&tn);
8512 vim_free(fname);
8513}
8514
8515/*
8516 * "taglist()" function
8517 */
8518 static void
8519f_taglist(typval_T *argvars, typval_T *rettv)
8520{
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008521 char_u *fname = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008522 char_u *tag_pattern;
8523
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008524 tag_pattern = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008525
8526 rettv->vval.v_number = FALSE;
8527 if (*tag_pattern == NUL)
8528 return;
8529
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008530 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008531 fname = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008532 if (rettv_list_alloc(rettv) == OK)
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008533 (void)get_tags(rettv->vval.v_list, tag_pattern, fname);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008534}
8535
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008536#ifdef FEAT_FLOAT
8537/*
8538 * "tan()" function
8539 */
8540 static void
8541f_tan(typval_T *argvars, typval_T *rettv)
8542{
8543 float_T f = 0.0;
8544
8545 rettv->v_type = VAR_FLOAT;
8546 if (get_float_arg(argvars, &f) == OK)
8547 rettv->vval.v_float = tan(f);
8548 else
8549 rettv->vval.v_float = 0.0;
8550}
8551
8552/*
8553 * "tanh()" function
8554 */
8555 static void
8556f_tanh(typval_T *argvars, typval_T *rettv)
8557{
8558 float_T f = 0.0;
8559
8560 rettv->v_type = VAR_FLOAT;
8561 if (get_float_arg(argvars, &f) == OK)
8562 rettv->vval.v_float = tanh(f);
8563 else
8564 rettv->vval.v_float = 0.0;
8565}
8566#endif
8567
8568/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008569 * "tolower(string)" function
8570 */
8571 static void
8572f_tolower(typval_T *argvars, typval_T *rettv)
8573{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008574 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008575 rettv->vval.v_string = strlow_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008576}
8577
8578/*
8579 * "toupper(string)" function
8580 */
8581 static void
8582f_toupper(typval_T *argvars, typval_T *rettv)
8583{
8584 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008585 rettv->vval.v_string = strup_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008586}
8587
8588/*
8589 * "tr(string, fromstr, tostr)" function
8590 */
8591 static void
8592f_tr(typval_T *argvars, typval_T *rettv)
8593{
8594 char_u *in_str;
8595 char_u *fromstr;
8596 char_u *tostr;
8597 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008598 int inlen;
8599 int fromlen;
8600 int tolen;
8601 int idx;
8602 char_u *cpstr;
8603 int cplen;
8604 int first = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008605 char_u buf[NUMBUFLEN];
8606 char_u buf2[NUMBUFLEN];
8607 garray_T ga;
8608
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008609 in_str = tv_get_string(&argvars[0]);
8610 fromstr = tv_get_string_buf_chk(&argvars[1], buf);
8611 tostr = tv_get_string_buf_chk(&argvars[2], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008612
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008613 // Default return value: empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008614 rettv->v_type = VAR_STRING;
8615 rettv->vval.v_string = NULL;
8616 if (fromstr == NULL || tostr == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008617 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008618 ga_init2(&ga, (int)sizeof(char), 80);
8619
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008620 if (!has_mbyte)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008621 // not multi-byte: fromstr and tostr must be the same length
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008622 if (STRLEN(fromstr) != STRLEN(tostr))
8623 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008624error:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008625 semsg(_(e_invarg2), fromstr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008626 ga_clear(&ga);
8627 return;
8628 }
8629
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008630 // fromstr and tostr have to contain the same number of chars
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008631 while (*in_str != NUL)
8632 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008633 if (has_mbyte)
8634 {
8635 inlen = (*mb_ptr2len)(in_str);
8636 cpstr = in_str;
8637 cplen = inlen;
8638 idx = 0;
8639 for (p = fromstr; *p != NUL; p += fromlen)
8640 {
8641 fromlen = (*mb_ptr2len)(p);
8642 if (fromlen == inlen && STRNCMP(in_str, p, inlen) == 0)
8643 {
8644 for (p = tostr; *p != NUL; p += tolen)
8645 {
8646 tolen = (*mb_ptr2len)(p);
8647 if (idx-- == 0)
8648 {
8649 cplen = tolen;
8650 cpstr = p;
8651 break;
8652 }
8653 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008654 if (*p == NUL) // tostr is shorter than fromstr
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008655 goto error;
8656 break;
8657 }
8658 ++idx;
8659 }
8660
8661 if (first && cpstr == in_str)
8662 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008663 // Check that fromstr and tostr have the same number of
8664 // (multi-byte) characters. Done only once when a character
8665 // of in_str doesn't appear in fromstr.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008666 first = FALSE;
8667 for (p = tostr; *p != NUL; p += tolen)
8668 {
8669 tolen = (*mb_ptr2len)(p);
8670 --idx;
8671 }
8672 if (idx != 0)
8673 goto error;
8674 }
8675
8676 (void)ga_grow(&ga, cplen);
8677 mch_memmove((char *)ga.ga_data + ga.ga_len, cpstr, (size_t)cplen);
8678 ga.ga_len += cplen;
8679
8680 in_str += inlen;
8681 }
8682 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008683 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008684 // When not using multi-byte chars we can do it faster.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008685 p = vim_strchr(fromstr, *in_str);
8686 if (p != NULL)
8687 ga_append(&ga, tostr[p - fromstr]);
8688 else
8689 ga_append(&ga, *in_str);
8690 ++in_str;
8691 }
8692 }
8693
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008694 // add a terminating NUL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008695 (void)ga_grow(&ga, 1);
8696 ga_append(&ga, NUL);
8697
8698 rettv->vval.v_string = ga.ga_data;
8699}
8700
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008701/*
8702 * "trim({expr})" function
8703 */
8704 static void
8705f_trim(typval_T *argvars, typval_T *rettv)
8706{
8707 char_u buf1[NUMBUFLEN];
8708 char_u buf2[NUMBUFLEN];
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008709 char_u *head = tv_get_string_buf_chk(&argvars[0], buf1);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008710 char_u *mask = NULL;
8711 char_u *tail;
8712 char_u *prev;
8713 char_u *p;
8714 int c1;
8715
8716 rettv->v_type = VAR_STRING;
8717 if (head == NULL)
8718 {
8719 rettv->vval.v_string = NULL;
8720 return;
8721 }
8722
8723 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008724 mask = tv_get_string_buf_chk(&argvars[1], buf2);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008725
8726 while (*head != NUL)
8727 {
8728 c1 = PTR2CHAR(head);
8729 if (mask == NULL)
8730 {
8731 if (c1 > ' ' && c1 != 0xa0)
8732 break;
8733 }
8734 else
8735 {
8736 for (p = mask; *p != NUL; MB_PTR_ADV(p))
8737 if (c1 == PTR2CHAR(p))
8738 break;
8739 if (*p == NUL)
8740 break;
8741 }
8742 MB_PTR_ADV(head);
8743 }
8744
8745 for (tail = head + STRLEN(head); tail > head; tail = prev)
8746 {
8747 prev = tail;
8748 MB_PTR_BACK(head, prev);
8749 c1 = PTR2CHAR(prev);
8750 if (mask == NULL)
8751 {
8752 if (c1 > ' ' && c1 != 0xa0)
8753 break;
8754 }
8755 else
8756 {
8757 for (p = mask; *p != NUL; MB_PTR_ADV(p))
8758 if (c1 == PTR2CHAR(p))
8759 break;
8760 if (*p == NUL)
8761 break;
8762 }
8763 }
8764 rettv->vval.v_string = vim_strnsave(head, (int)(tail - head));
8765}
8766
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008767#ifdef FEAT_FLOAT
8768/*
8769 * "trunc({float})" function
8770 */
8771 static void
8772f_trunc(typval_T *argvars, typval_T *rettv)
8773{
8774 float_T f = 0.0;
8775
8776 rettv->v_type = VAR_FLOAT;
8777 if (get_float_arg(argvars, &f) == OK)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008778 // trunc() is not in C90, use floor() or ceil() instead.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008779 rettv->vval.v_float = f > 0 ? floor(f) : ceil(f);
8780 else
8781 rettv->vval.v_float = 0.0;
8782}
8783#endif
8784
8785/*
8786 * "type(expr)" function
8787 */
8788 static void
8789f_type(typval_T *argvars, typval_T *rettv)
8790{
8791 int n = -1;
8792
8793 switch (argvars[0].v_type)
8794 {
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01008795 case VAR_NUMBER: n = VAR_TYPE_NUMBER; break;
8796 case VAR_STRING: n = VAR_TYPE_STRING; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008797 case VAR_PARTIAL:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01008798 case VAR_FUNC: n = VAR_TYPE_FUNC; break;
8799 case VAR_LIST: n = VAR_TYPE_LIST; break;
8800 case VAR_DICT: n = VAR_TYPE_DICT; break;
8801 case VAR_FLOAT: n = VAR_TYPE_FLOAT; break;
8802 case VAR_BOOL: n = VAR_TYPE_BOOL; break;
8803 case VAR_SPECIAL: n = VAR_TYPE_NONE; break;
Bram Moolenaarf562e722016-07-19 17:25:25 +02008804 case VAR_JOB: n = VAR_TYPE_JOB; break;
8805 case VAR_CHANNEL: n = VAR_TYPE_CHANNEL; break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01008806 case VAR_BLOB: n = VAR_TYPE_BLOB; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008807 case VAR_UNKNOWN:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01008808 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01008809 internal_error_no_abort("f_type(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008810 n = -1;
8811 break;
8812 }
8813 rettv->vval.v_number = n;
8814}
8815
8816/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008817 * "virtcol(string)" function
8818 */
8819 static void
8820f_virtcol(typval_T *argvars, typval_T *rettv)
8821{
8822 colnr_T vcol = 0;
8823 pos_T *fp;
8824 int fnum = curbuf->b_fnum;
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01008825 int len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008826
8827 fp = var2fpos(&argvars[0], FALSE, &fnum);
8828 if (fp != NULL && fp->lnum <= curbuf->b_ml.ml_line_count
8829 && fnum == curbuf->b_fnum)
8830 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01008831 // Limit the column to a valid value, getvvcol() doesn't check.
8832 if (fp->col < 0)
8833 fp->col = 0;
8834 else
8835 {
8836 len = (int)STRLEN(ml_get(fp->lnum));
8837 if (fp->col > len)
8838 fp->col = len;
8839 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008840 getvvcol(curwin, fp, NULL, NULL, &vcol);
8841 ++vcol;
8842 }
8843
8844 rettv->vval.v_number = vcol;
8845}
8846
8847/*
8848 * "visualmode()" function
8849 */
8850 static void
8851f_visualmode(typval_T *argvars, typval_T *rettv)
8852{
8853 char_u str[2];
8854
8855 rettv->v_type = VAR_STRING;
8856 str[0] = curbuf->b_visual_mode_eval;
8857 str[1] = NUL;
8858 rettv->vval.v_string = vim_strsave(str);
8859
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008860 // A non-zero number or non-empty string argument: reset mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008861 if (non_zero_arg(&argvars[0]))
8862 curbuf->b_visual_mode_eval = NUL;
8863}
8864
8865/*
8866 * "wildmenumode()" function
8867 */
8868 static void
8869f_wildmenumode(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8870{
8871#ifdef FEAT_WILDMENU
8872 if (wild_menu_showing)
8873 rettv->vval.v_number = 1;
8874#endif
8875}
8876
8877/*
Bram Moolenaar0c1e3742019-12-27 13:49:24 +01008878 * "windowsversion()" function
8879 */
8880 static void
8881f_windowsversion(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8882{
8883 rettv->v_type = VAR_STRING;
8884 rettv->vval.v_string = vim_strsave((char_u *)windowsVersion);
8885}
8886
8887/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008888 * "wordcount()" function
8889 */
8890 static void
8891f_wordcount(typval_T *argvars UNUSED, typval_T *rettv)
8892{
8893 if (rettv_dict_alloc(rettv) == FAIL)
8894 return;
8895 cursor_pos_info(rettv->vval.v_dict);
8896}
8897
8898/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008899 * "xor(expr, expr)" function
8900 */
8901 static void
8902f_xor(typval_T *argvars, typval_T *rettv)
8903{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008904 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
8905 ^ tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008906}
8907
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008908#endif // FEAT_EVAL