blob: 02943c717a87b3a44d5b295e1ce239d77e56af37 [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 Moolenaar7035fd92020-04-08 20:03:52 +0200840 {"simplify", 1, 1, FEARG_1, 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 Moolenaar4c683752020-04-05 21:38:23 +02002045 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002046 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01002047 internal_error_no_abort("f_empty(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002048 n = TRUE;
2049 break;
2050 }
2051
2052 rettv->vval.v_number = n;
2053}
2054
2055/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02002056 * "environ()" function
2057 */
2058 static void
2059f_environ(typval_T *argvars UNUSED, typval_T *rettv)
2060{
2061#if !defined(AMIGA)
2062 int i = 0;
2063 char_u *entry, *value;
2064# ifdef MSWIN
2065 extern wchar_t **_wenviron;
2066# else
2067 extern char **environ;
2068# endif
2069
2070 if (rettv_dict_alloc(rettv) != OK)
2071 return;
2072
2073# ifdef MSWIN
2074 if (*_wenviron == NULL)
2075 return;
2076# else
2077 if (*environ == NULL)
2078 return;
2079# endif
2080
2081 for (i = 0; ; ++i)
2082 {
2083# ifdef MSWIN
2084 short_u *p;
2085
2086 if ((p = (short_u *)_wenviron[i]) == NULL)
2087 return;
2088 entry = utf16_to_enc(p, NULL);
2089# else
2090 if ((entry = (char_u *)environ[i]) == NULL)
2091 return;
2092 entry = vim_strsave(entry);
2093# endif
2094 if (entry == NULL) // out of memory
2095 return;
2096 if ((value = vim_strchr(entry, '=')) == NULL)
2097 {
2098 vim_free(entry);
2099 continue;
2100 }
2101 *value++ = NUL;
2102 dict_add_string(rettv->vval.v_dict, (char *)entry, value);
2103 vim_free(entry);
2104 }
2105#endif
2106}
2107
2108/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002109 * "escape({string}, {chars})" function
2110 */
2111 static void
2112f_escape(typval_T *argvars, typval_T *rettv)
2113{
2114 char_u buf[NUMBUFLEN];
2115
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002116 rettv->vval.v_string = vim_strsave_escaped(tv_get_string(&argvars[0]),
2117 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002118 rettv->v_type = VAR_STRING;
2119}
2120
2121/*
2122 * "eval()" function
2123 */
2124 static void
2125f_eval(typval_T *argvars, typval_T *rettv)
2126{
2127 char_u *s, *p;
2128
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002129 s = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002130 if (s != NULL)
2131 s = skipwhite(s);
2132
2133 p = s;
2134 if (s == NULL || eval1(&s, rettv, TRUE) == FAIL)
2135 {
2136 if (p != NULL && !aborting())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002137 semsg(_(e_invexpr2), p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002138 need_clr_eos = FALSE;
2139 rettv->v_type = VAR_NUMBER;
2140 rettv->vval.v_number = 0;
2141 }
2142 else if (*s != NUL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002143 emsg(_(e_trailing));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002144}
2145
2146/*
2147 * "eventhandler()" function
2148 */
2149 static void
2150f_eventhandler(typval_T *argvars UNUSED, typval_T *rettv)
2151{
2152 rettv->vval.v_number = vgetc_busy;
2153}
2154
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002155static garray_T redir_execute_ga;
2156
2157/*
2158 * Append "value[value_len]" to the execute() output.
2159 */
2160 void
2161execute_redir_str(char_u *value, int value_len)
2162{
2163 int len;
2164
2165 if (value_len == -1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002166 len = (int)STRLEN(value); // Append the entire string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002167 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002168 len = value_len; // Append only "value_len" characters
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002169 if (ga_grow(&redir_execute_ga, len) == OK)
2170 {
2171 mch_memmove((char *)redir_execute_ga.ga_data
2172 + redir_execute_ga.ga_len, value, len);
2173 redir_execute_ga.ga_len += len;
2174 }
2175}
2176
2177/*
2178 * Get next line from a list.
2179 * Called by do_cmdline() to get the next line.
2180 * Returns allocated string, or NULL for end of function.
2181 */
2182
2183 static char_u *
2184get_list_line(
2185 int c UNUSED,
2186 void *cookie,
Bram Moolenaare96a2492019-06-25 04:12:16 +02002187 int indent UNUSED,
2188 int do_concat UNUSED)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002189{
2190 listitem_T **p = (listitem_T **)cookie;
2191 listitem_T *item = *p;
2192 char_u buf[NUMBUFLEN];
2193 char_u *s;
2194
2195 if (item == NULL)
2196 return NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002197 s = tv_get_string_buf_chk(&item->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002198 *p = item->li_next;
2199 return s == NULL ? NULL : vim_strsave(s);
2200}
2201
2202/*
2203 * "execute()" function
2204 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02002205 void
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002206execute_common(typval_T *argvars, typval_T *rettv, int arg_off)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002207{
2208 char_u *cmd = NULL;
2209 list_T *list = NULL;
2210 int save_msg_silent = msg_silent;
2211 int save_emsg_silent = emsg_silent;
2212 int save_emsg_noredir = emsg_noredir;
2213 int save_redir_execute = redir_execute;
Bram Moolenaar20951482017-12-25 13:44:43 +01002214 int save_redir_off = redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002215 garray_T save_ga;
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002216 int save_msg_col = msg_col;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002217 int echo_output = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002218
2219 rettv->vval.v_string = NULL;
2220 rettv->v_type = VAR_STRING;
2221
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002222 if (argvars[arg_off].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002223 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002224 list = argvars[arg_off].vval.v_list;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002225 if (list == NULL || list->lv_len == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002226 // empty list, no commands, empty output
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002227 return;
2228 ++list->lv_refcount;
2229 }
Bram Moolenaare2a8f072020-01-08 19:32:18 +01002230 else if (argvars[arg_off].v_type == VAR_JOB
2231 || argvars[arg_off].v_type == VAR_CHANNEL)
2232 {
2233 emsg(_(e_inval_string));
2234 return;
2235 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002236 else
2237 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002238 cmd = tv_get_string_chk(&argvars[arg_off]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002239 if (cmd == NULL)
2240 return;
2241 }
2242
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002243 if (argvars[arg_off + 1].v_type != VAR_UNKNOWN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002244 {
2245 char_u buf[NUMBUFLEN];
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002246 char_u *s = tv_get_string_buf_chk(&argvars[arg_off + 1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002247
2248 if (s == NULL)
2249 return;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002250 if (*s == NUL)
2251 echo_output = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002252 if (STRNCMP(s, "silent", 6) == 0)
2253 ++msg_silent;
2254 if (STRCMP(s, "silent!") == 0)
2255 {
2256 emsg_silent = TRUE;
2257 emsg_noredir = TRUE;
2258 }
2259 }
2260 else
2261 ++msg_silent;
2262
2263 if (redir_execute)
2264 save_ga = redir_execute_ga;
2265 ga_init2(&redir_execute_ga, (int)sizeof(char), 500);
2266 redir_execute = TRUE;
Bram Moolenaar20951482017-12-25 13:44:43 +01002267 redir_off = FALSE;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002268 if (!echo_output)
2269 msg_col = 0; // prevent leading spaces
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002270
2271 if (cmd != NULL)
2272 do_cmdline_cmd(cmd);
2273 else
2274 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002275 listitem_T *item;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002276
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002277 range_list_materialize(list);
2278 item = list->lv_first;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002279 do_cmdline(NULL, get_list_line, (void *)&item,
2280 DOCMD_NOWAIT|DOCMD_VERBOSE|DOCMD_REPEAT|DOCMD_KEYTYPED);
2281 --list->lv_refcount;
2282 }
2283
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002284 // Need to append a NUL to the result.
Bram Moolenaard297f352017-01-29 20:31:21 +01002285 if (ga_grow(&redir_execute_ga, 1) == OK)
2286 {
2287 ((char *)redir_execute_ga.ga_data)[redir_execute_ga.ga_len] = NUL;
2288 rettv->vval.v_string = redir_execute_ga.ga_data;
2289 }
2290 else
2291 {
2292 ga_clear(&redir_execute_ga);
2293 rettv->vval.v_string = NULL;
2294 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002295 msg_silent = save_msg_silent;
2296 emsg_silent = save_emsg_silent;
2297 emsg_noredir = save_emsg_noredir;
2298
2299 redir_execute = save_redir_execute;
2300 if (redir_execute)
2301 redir_execute_ga = save_ga;
Bram Moolenaar20951482017-12-25 13:44:43 +01002302 redir_off = save_redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002303
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002304 // "silent reg" or "silent echo x" leaves msg_col somewhere in the line.
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002305 if (echo_output)
2306 // When not working silently: put it in column zero. A following
2307 // "echon" will overwrite the message, unavoidably.
2308 msg_col = 0;
2309 else
2310 // When working silently: Put it back where it was, since nothing
2311 // should have been written.
2312 msg_col = save_msg_col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002313}
2314
2315/*
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002316 * "execute()" function
2317 */
2318 static void
2319f_execute(typval_T *argvars, typval_T *rettv)
2320{
2321 execute_common(argvars, rettv, 0);
2322}
2323
2324/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002325 * "exists()" function
2326 */
2327 static void
2328f_exists(typval_T *argvars, typval_T *rettv)
2329{
2330 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002331 int n = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002332
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002333 p = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002334 if (*p == '$') // environment variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002335 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002336 // first try "normal" environment variables (fast)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002337 if (mch_getenv(p + 1) != NULL)
2338 n = TRUE;
2339 else
2340 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002341 // try expanding things like $VIM and ${HOME}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002342 p = expand_env_save(p);
2343 if (p != NULL && *p != '$')
2344 n = TRUE;
2345 vim_free(p);
2346 }
2347 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002348 else if (*p == '&' || *p == '+') // option
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002349 {
2350 n = (get_option_tv(&p, NULL, TRUE) == OK);
2351 if (*skipwhite(p) != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002352 n = FALSE; // trailing garbage
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002353 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002354 else if (*p == '*') // internal or user defined function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002355 {
Bram Moolenaarb54c3ff2016-07-31 14:11:58 +02002356 n = function_exists(p + 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002357 }
Bram Moolenaar15c47602020-03-26 22:16:48 +01002358 else if (*p == '?') // internal function only
2359 {
2360 n = has_internal_func_name(p + 1);
2361 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002362 else if (*p == ':')
2363 {
2364 n = cmd_exists(p + 1);
2365 }
2366 else if (*p == '#')
2367 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002368 if (p[1] == '#')
2369 n = autocmd_supported(p + 2);
2370 else
2371 n = au_exists(p + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002372 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002373 else // internal variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002374 {
Bram Moolenaarc6f9f732018-02-11 19:06:26 +01002375 n = var_exists(p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002376 }
2377
2378 rettv->vval.v_number = n;
2379}
2380
2381#ifdef FEAT_FLOAT
2382/*
2383 * "exp()" function
2384 */
2385 static void
2386f_exp(typval_T *argvars, typval_T *rettv)
2387{
2388 float_T f = 0.0;
2389
2390 rettv->v_type = VAR_FLOAT;
2391 if (get_float_arg(argvars, &f) == OK)
2392 rettv->vval.v_float = exp(f);
2393 else
2394 rettv->vval.v_float = 0.0;
2395}
2396#endif
2397
2398/*
2399 * "expand()" function
2400 */
2401 static void
2402f_expand(typval_T *argvars, typval_T *rettv)
2403{
2404 char_u *s;
2405 int len;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002406 char *errormsg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002407 int options = WILD_SILENT|WILD_USE_NL|WILD_LIST_NOTFOUND;
2408 expand_T xpc;
2409 int error = FALSE;
2410 char_u *result;
2411
2412 rettv->v_type = VAR_STRING;
2413 if (argvars[1].v_type != VAR_UNKNOWN
2414 && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002415 && tv_get_number_chk(&argvars[2], &error)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002416 && !error)
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02002417 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002418
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002419 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002420 if (*s == '%' || *s == '#' || *s == '<')
2421 {
2422 ++emsg_off;
2423 result = eval_vars(s, s, &len, NULL, &errormsg, NULL);
2424 --emsg_off;
2425 if (rettv->v_type == VAR_LIST)
2426 {
2427 if (rettv_list_alloc(rettv) != FAIL && result != NULL)
2428 list_append_string(rettv->vval.v_list, result, -1);
Bram Moolenaar86173482019-10-01 17:02:16 +02002429 vim_free(result);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002430 }
2431 else
2432 rettv->vval.v_string = result;
2433 }
2434 else
2435 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002436 // When the optional second argument is non-zero, don't remove matches
2437 // for 'wildignore' and don't put matches for 'suffixes' at the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002438 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002439 && tv_get_number_chk(&argvars[1], &error))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002440 options |= WILD_KEEP_ALL;
2441 if (!error)
2442 {
2443 ExpandInit(&xpc);
2444 xpc.xp_context = EXPAND_FILES;
2445 if (p_wic)
2446 options += WILD_ICASE;
2447 if (rettv->v_type == VAR_STRING)
2448 rettv->vval.v_string = ExpandOne(&xpc, s, NULL,
2449 options, WILD_ALL);
2450 else if (rettv_list_alloc(rettv) != FAIL)
2451 {
2452 int i;
2453
2454 ExpandOne(&xpc, s, NULL, options, WILD_ALL_KEEP);
2455 for (i = 0; i < xpc.xp_numfiles; i++)
2456 list_append_string(rettv->vval.v_list, xpc.xp_files[i], -1);
2457 ExpandCleanup(&xpc);
2458 }
2459 }
2460 else
2461 rettv->vval.v_string = NULL;
2462 }
2463}
2464
2465/*
Bram Moolenaar80dad482019-06-09 17:22:31 +02002466 * "expandcmd()" function
2467 * Expand all the special characters in a command string.
2468 */
2469 static void
2470f_expandcmd(typval_T *argvars, typval_T *rettv)
2471{
2472 exarg_T eap;
2473 char_u *cmdstr;
2474 char *errormsg = NULL;
2475
2476 rettv->v_type = VAR_STRING;
2477 cmdstr = vim_strsave(tv_get_string(&argvars[0]));
2478
2479 memset(&eap, 0, sizeof(eap));
2480 eap.cmd = cmdstr;
2481 eap.arg = cmdstr;
Bram Moolenaar8071cb22019-07-12 17:58:01 +02002482 eap.argt |= EX_NOSPC;
Bram Moolenaar80dad482019-06-09 17:22:31 +02002483 eap.usefilter = FALSE;
2484 eap.nextcmd = NULL;
2485 eap.cmdidx = CMD_USER;
2486
2487 expand_filename(&eap, &cmdstr, &errormsg);
2488 if (errormsg != NULL && *errormsg != NUL)
2489 emsg(errormsg);
2490
2491 rettv->vval.v_string = cmdstr;
2492}
2493
2494/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002495 * "feedkeys()" function
2496 */
2497 static void
2498f_feedkeys(typval_T *argvars, typval_T *rettv UNUSED)
2499{
2500 int remap = TRUE;
2501 int insert = FALSE;
2502 char_u *keys, *flags;
2503 char_u nbuf[NUMBUFLEN];
2504 int typed = FALSE;
2505 int execute = FALSE;
2506 int dangerous = FALSE;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002507 int lowlevel = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002508 char_u *keys_esc;
2509
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002510 // This is not allowed in the sandbox. If the commands would still be
2511 // executed in the sandbox it would be OK, but it probably happens later,
2512 // when "sandbox" is no longer set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002513 if (check_secure())
2514 return;
2515
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002516 keys = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002517
2518 if (argvars[1].v_type != VAR_UNKNOWN)
2519 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002520 flags = tv_get_string_buf(&argvars[1], nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002521 for ( ; *flags != NUL; ++flags)
2522 {
2523 switch (*flags)
2524 {
2525 case 'n': remap = FALSE; break;
2526 case 'm': remap = TRUE; break;
2527 case 't': typed = TRUE; break;
2528 case 'i': insert = TRUE; break;
2529 case 'x': execute = TRUE; break;
2530 case '!': dangerous = TRUE; break;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002531 case 'L': lowlevel = TRUE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002532 }
2533 }
2534 }
2535
2536 if (*keys != NUL || execute)
2537 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002538 // Need to escape K_SPECIAL and CSI before putting the string in the
2539 // typeahead buffer.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002540 keys_esc = vim_strsave_escape_csi(keys);
2541 if (keys_esc != NULL)
2542 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002543 if (lowlevel)
2544 {
2545#ifdef USE_INPUT_BUF
Bram Moolenaar9645e2d2020-03-20 20:48:49 +01002546 int idx;
2547 int len = (int)STRLEN(keys);
2548
2549 for (idx = 0; idx < len; ++idx)
2550 {
2551 // if a CTRL-C was typed, set got_int, similar to what
2552 // happens in fill_input_buf()
2553 if (keys[idx] == 3 && ctrl_c_interrupts && typed)
2554 got_int = TRUE;
2555 add_to_input_buf(keys + idx, 1);
2556 }
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002557#else
2558 emsg(_("E980: lowlevel input not supported"));
2559#endif
2560 }
2561 else
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002562 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002563 ins_typebuf(keys_esc, (remap ? REMAP_YES : REMAP_NONE),
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002564 insert ? 0 : typebuf.tb_len, !typed, FALSE);
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002565 if (vgetc_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02002566#ifdef FEAT_TIMERS
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002567 || timer_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02002568#endif
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002569 )
2570 typebuf_was_filled = TRUE;
2571 }
2572 vim_free(keys_esc);
2573
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002574 if (execute)
2575 {
2576 int save_msg_scroll = msg_scroll;
2577
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002578 // Avoid a 1 second delay when the keys start Insert mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002579 msg_scroll = FALSE;
2580
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02002581 if (!dangerous)
2582 ++ex_normal_busy;
Bram Moolenaar905dd902019-04-07 14:21:47 +02002583 exec_normal(TRUE, lowlevel, TRUE);
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02002584 if (!dangerous)
2585 --ex_normal_busy;
2586
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002587 msg_scroll |= save_msg_scroll;
2588 }
2589 }
2590 }
2591}
2592
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002593#ifdef FEAT_FLOAT
2594/*
2595 * "float2nr({float})" function
2596 */
2597 static void
2598f_float2nr(typval_T *argvars, typval_T *rettv)
2599{
2600 float_T f = 0.0;
2601
2602 if (get_float_arg(argvars, &f) == OK)
2603 {
Bram Moolenaar863e80b2017-06-04 20:30:00 +02002604 if (f <= -VARNUM_MAX + DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002605 rettv->vval.v_number = -VARNUM_MAX;
Bram Moolenaar863e80b2017-06-04 20:30:00 +02002606 else if (f >= VARNUM_MAX - DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002607 rettv->vval.v_number = VARNUM_MAX;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002608 else
2609 rettv->vval.v_number = (varnumber_T)f;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002610 }
2611}
2612
2613/*
2614 * "floor({float})" function
2615 */
2616 static void
2617f_floor(typval_T *argvars, typval_T *rettv)
2618{
2619 float_T f = 0.0;
2620
2621 rettv->v_type = VAR_FLOAT;
2622 if (get_float_arg(argvars, &f) == OK)
2623 rettv->vval.v_float = floor(f);
2624 else
2625 rettv->vval.v_float = 0.0;
2626}
2627
2628/*
2629 * "fmod()" function
2630 */
2631 static void
2632f_fmod(typval_T *argvars, typval_T *rettv)
2633{
2634 float_T fx = 0.0, fy = 0.0;
2635
2636 rettv->v_type = VAR_FLOAT;
2637 if (get_float_arg(argvars, &fx) == OK
2638 && get_float_arg(&argvars[1], &fy) == OK)
2639 rettv->vval.v_float = fmod(fx, fy);
2640 else
2641 rettv->vval.v_float = 0.0;
2642}
2643#endif
2644
2645/*
2646 * "fnameescape({string})" function
2647 */
2648 static void
2649f_fnameescape(typval_T *argvars, typval_T *rettv)
2650{
2651 rettv->vval.v_string = vim_strsave_fnameescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002652 tv_get_string(&argvars[0]), FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002653 rettv->v_type = VAR_STRING;
2654}
2655
2656/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002657 * "foreground()" function
2658 */
2659 static void
2660f_foreground(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2661{
2662#ifdef FEAT_GUI
2663 if (gui.in_use)
Bram Moolenaarafde13b2019-04-28 19:46:49 +02002664 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002665 gui_mch_set_foreground();
Bram Moolenaarafde13b2019-04-28 19:46:49 +02002666 return;
2667 }
2668#endif
2669#if defined(MSWIN) && (!defined(FEAT_GUI) || defined(VIMDLL))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002670 win32_set_foreground();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002671#endif
2672}
2673
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002674 static void
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002675common_function(typval_T *argvars, typval_T *rettv, int is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002676{
2677 char_u *s;
2678 char_u *name;
2679 int use_string = FALSE;
2680 partial_T *arg_pt = NULL;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002681 char_u *trans_name = NULL;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002682 int is_global = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002683
2684 if (argvars[0].v_type == VAR_FUNC)
2685 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002686 // function(MyFunc, [arg], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002687 s = argvars[0].vval.v_string;
2688 }
2689 else if (argvars[0].v_type == VAR_PARTIAL
2690 && argvars[0].vval.v_partial != NULL)
2691 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002692 // function(dict.MyFunc, [arg])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002693 arg_pt = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002694 s = partial_name(arg_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002695 }
2696 else
2697 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002698 // function('MyFunc', [arg], dict)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002699 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002700 use_string = TRUE;
2701 }
2702
Bram Moolenaar843b8842016-08-21 14:36:15 +02002703 if ((use_string && vim_strchr(s, AUTOLOAD_CHAR) == NULL) || is_funcref)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002704 {
2705 name = s;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002706 trans_name = trans_function_name(&name, &is_global, FALSE,
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002707 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF, NULL, NULL);
2708 if (*name != NUL)
2709 s = NULL;
2710 }
2711
Bram Moolenaar843b8842016-08-21 14:36:15 +02002712 if (s == NULL || *s == NUL || (use_string && VIM_ISDIGIT(*s))
2713 || (is_funcref && trans_name == NULL))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002714 semsg(_(e_invarg2), use_string ? tv_get_string(&argvars[0]) : s);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002715 // Don't check an autoload name for existence here.
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002716 else if (trans_name != NULL && (is_funcref
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002717 ? find_func(trans_name, is_global, NULL) == NULL
2718 : !translated_function_exists(trans_name, is_global)))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002719 semsg(_("E700: Unknown function: %s"), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002720 else
2721 {
2722 int dict_idx = 0;
2723 int arg_idx = 0;
2724 list_T *list = NULL;
2725
2726 if (STRNCMP(s, "s:", 2) == 0 || STRNCMP(s, "<SID>", 5) == 0)
2727 {
2728 char sid_buf[25];
2729 int off = *s == 's' ? 2 : 5;
2730
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002731 // Expand s: and <SID> into <SNR>nr_, so that the function can
2732 // also be called from another script. Using trans_function_name()
2733 // would also work, but some plugins depend on the name being
2734 // printable text.
Bram Moolenaarf29c1c62018-09-10 21:05:02 +02002735 sprintf(sid_buf, "<SNR>%ld_", (long)current_sctx.sc_sid);
Bram Moolenaar51e14382019-05-25 20:21:28 +02002736 name = alloc(STRLEN(sid_buf) + STRLEN(s + off) + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002737 if (name != NULL)
2738 {
2739 STRCPY(name, sid_buf);
2740 STRCAT(name, s + off);
2741 }
2742 }
2743 else
2744 name = vim_strsave(s);
2745
2746 if (argvars[1].v_type != VAR_UNKNOWN)
2747 {
2748 if (argvars[2].v_type != VAR_UNKNOWN)
2749 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002750 // function(name, [args], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002751 arg_idx = 1;
2752 dict_idx = 2;
2753 }
2754 else if (argvars[1].v_type == VAR_DICT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002755 // function(name, dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002756 dict_idx = 1;
2757 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002758 // function(name, [args])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002759 arg_idx = 1;
2760 if (dict_idx > 0)
2761 {
2762 if (argvars[dict_idx].v_type != VAR_DICT)
2763 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002764 emsg(_("E922: expected a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002765 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002766 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002767 }
2768 if (argvars[dict_idx].vval.v_dict == NULL)
2769 dict_idx = 0;
2770 }
2771 if (arg_idx > 0)
2772 {
2773 if (argvars[arg_idx].v_type != VAR_LIST)
2774 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002775 emsg(_("E923: Second argument of function() must be a list or a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002776 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002777 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002778 }
2779 list = argvars[arg_idx].vval.v_list;
2780 if (list == NULL || list->lv_len == 0)
2781 arg_idx = 0;
Bram Moolenaar4c054e92019-11-10 00:13:50 +01002782 else if (list->lv_len > MAX_FUNC_ARGS)
2783 {
Bram Moolenaar2118a302019-11-22 19:29:45 +01002784 emsg_funcname((char *)e_toomanyarg, s);
Bram Moolenaar4c054e92019-11-10 00:13:50 +01002785 vim_free(name);
2786 goto theend;
2787 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002788 }
2789 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002790 if (dict_idx > 0 || arg_idx > 0 || arg_pt != NULL || is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002791 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02002792 partial_T *pt = ALLOC_CLEAR_ONE(partial_T);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002793
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002794 // result is a VAR_PARTIAL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002795 if (pt == NULL)
2796 vim_free(name);
2797 else
2798 {
2799 if (arg_idx > 0 || (arg_pt != NULL && arg_pt->pt_argc > 0))
2800 {
2801 listitem_T *li;
2802 int i = 0;
2803 int arg_len = 0;
2804 int lv_len = 0;
2805
2806 if (arg_pt != NULL)
2807 arg_len = arg_pt->pt_argc;
2808 if (list != NULL)
2809 lv_len = list->lv_len;
2810 pt->pt_argc = arg_len + lv_len;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02002811 pt->pt_argv = ALLOC_MULT(typval_T, pt->pt_argc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002812 if (pt->pt_argv == NULL)
2813 {
2814 vim_free(pt);
2815 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002816 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002817 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002818 for (i = 0; i < arg_len; i++)
2819 copy_tv(&arg_pt->pt_argv[i], &pt->pt_argv[i]);
2820 if (lv_len > 0)
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002821 {
2822 range_list_materialize(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002823 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002824 copy_tv(&li->li_tv, &pt->pt_argv[i++]);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002825 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002826 }
2827
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002828 // For "function(dict.func, [], dict)" and "func" is a partial
2829 // use "dict". That is backwards compatible.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002830 if (dict_idx > 0)
2831 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002832 // The dict is bound explicitly, pt_auto is FALSE.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002833 pt->pt_dict = argvars[dict_idx].vval.v_dict;
2834 ++pt->pt_dict->dv_refcount;
2835 }
2836 else if (arg_pt != NULL)
2837 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002838 // If the dict was bound automatically the result is also
2839 // bound automatically.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002840 pt->pt_dict = arg_pt->pt_dict;
2841 pt->pt_auto = arg_pt->pt_auto;
2842 if (pt->pt_dict != NULL)
2843 ++pt->pt_dict->dv_refcount;
2844 }
2845
2846 pt->pt_refcount = 1;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002847 if (arg_pt != NULL && arg_pt->pt_func != NULL)
2848 {
2849 pt->pt_func = arg_pt->pt_func;
2850 func_ptr_ref(pt->pt_func);
2851 vim_free(name);
2852 }
2853 else if (is_funcref)
2854 {
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002855 pt->pt_func = find_func(trans_name, is_global, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002856 func_ptr_ref(pt->pt_func);
2857 vim_free(name);
2858 }
2859 else
2860 {
2861 pt->pt_name = name;
2862 func_ref(name);
2863 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002864 }
2865 rettv->v_type = VAR_PARTIAL;
2866 rettv->vval.v_partial = pt;
2867 }
2868 else
2869 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002870 // result is a VAR_FUNC
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002871 rettv->v_type = VAR_FUNC;
2872 rettv->vval.v_string = name;
2873 func_ref(name);
2874 }
2875 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002876theend:
2877 vim_free(trans_name);
2878}
2879
2880/*
2881 * "funcref()" function
2882 */
2883 static void
2884f_funcref(typval_T *argvars, typval_T *rettv)
2885{
2886 common_function(argvars, rettv, TRUE);
2887}
2888
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002889 static type_T *
2890ret_f_function(int argcount, type_T **argtypes UNUSED)
2891{
2892 if (argcount == 1 && argtypes[0]->tt_type == VAR_STRING)
2893 return &t_func_any;
Bram Moolenaard77a8522020-04-03 21:59:57 +02002894 return &t_func_void;
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002895}
2896
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002897/*
2898 * "function()" function
2899 */
2900 static void
2901f_function(typval_T *argvars, typval_T *rettv)
2902{
2903 common_function(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002904}
2905
2906/*
2907 * "garbagecollect()" function
2908 */
2909 static void
2910f_garbagecollect(typval_T *argvars, typval_T *rettv UNUSED)
2911{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002912 // This is postponed until we are back at the toplevel, because we may be
2913 // using Lists and Dicts internally. E.g.: ":echo [garbagecollect()]".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002914 want_garbage_collect = TRUE;
2915
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002916 if (argvars[0].v_type != VAR_UNKNOWN && tv_get_number(&argvars[0]) == 1)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002917 garbage_collect_at_exit = TRUE;
2918}
2919
2920/*
2921 * "get()" function
2922 */
2923 static void
2924f_get(typval_T *argvars, typval_T *rettv)
2925{
2926 listitem_T *li;
2927 list_T *l;
2928 dictitem_T *di;
2929 dict_T *d;
2930 typval_T *tv = NULL;
Bram Moolenaarf91aac52019-07-28 13:21:01 +02002931 int what_is_dict = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002932
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002933 if (argvars[0].v_type == VAR_BLOB)
2934 {
2935 int error = FALSE;
2936 int idx = tv_get_number_chk(&argvars[1], &error);
2937
2938 if (!error)
2939 {
2940 rettv->v_type = VAR_NUMBER;
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002941 if (idx < 0)
2942 idx = blob_len(argvars[0].vval.v_blob) + idx;
2943 if (idx < 0 || idx >= blob_len(argvars[0].vval.v_blob))
2944 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002945 else
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002946 {
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002947 rettv->vval.v_number = blob_get(argvars[0].vval.v_blob, idx);
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002948 tv = rettv;
2949 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002950 }
2951 }
2952 else if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002953 {
2954 if ((l = argvars[0].vval.v_list) != NULL)
2955 {
2956 int error = FALSE;
2957
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002958 li = list_find(l, (long)tv_get_number_chk(&argvars[1], &error));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002959 if (!error && li != NULL)
2960 tv = &li->li_tv;
2961 }
2962 }
2963 else if (argvars[0].v_type == VAR_DICT)
2964 {
2965 if ((d = argvars[0].vval.v_dict) != NULL)
2966 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002967 di = dict_find(d, tv_get_string(&argvars[1]), -1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002968 if (di != NULL)
2969 tv = &di->di_tv;
2970 }
2971 }
2972 else if (argvars[0].v_type == VAR_PARTIAL || argvars[0].v_type == VAR_FUNC)
2973 {
2974 partial_T *pt;
2975 partial_T fref_pt;
2976
2977 if (argvars[0].v_type == VAR_PARTIAL)
2978 pt = argvars[0].vval.v_partial;
2979 else
2980 {
Bram Moolenaara80faa82020-04-12 19:37:17 +02002981 CLEAR_FIELD(fref_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002982 fref_pt.pt_name = argvars[0].vval.v_string;
2983 pt = &fref_pt;
2984 }
2985
2986 if (pt != NULL)
2987 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002988 char_u *what = tv_get_string(&argvars[1]);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002989 char_u *n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002990
2991 if (STRCMP(what, "func") == 0 || STRCMP(what, "name") == 0)
2992 {
2993 rettv->v_type = (*what == 'f' ? VAR_FUNC : VAR_STRING);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002994 n = partial_name(pt);
2995 if (n == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002996 rettv->vval.v_string = NULL;
2997 else
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002998 {
2999 rettv->vval.v_string = vim_strsave(n);
3000 if (rettv->v_type == VAR_FUNC)
3001 func_ref(rettv->vval.v_string);
3002 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003003 }
3004 else if (STRCMP(what, "dict") == 0)
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003005 {
3006 what_is_dict = TRUE;
3007 if (pt->pt_dict != NULL)
3008 rettv_dict_set(rettv, pt->pt_dict);
3009 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003010 else if (STRCMP(what, "args") == 0)
3011 {
3012 rettv->v_type = VAR_LIST;
3013 if (rettv_list_alloc(rettv) == OK)
3014 {
3015 int i;
3016
3017 for (i = 0; i < pt->pt_argc; ++i)
3018 list_append_tv(rettv->vval.v_list, &pt->pt_argv[i]);
3019 }
3020 }
3021 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003022 semsg(_(e_invarg2), what);
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003023
3024 // When {what} == "dict" and pt->pt_dict == NULL, evaluate the
3025 // third argument
3026 if (!what_is_dict)
3027 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003028 }
3029 }
3030 else
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01003031 semsg(_(e_listdictblobarg), "get()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003032
3033 if (tv == NULL)
3034 {
3035 if (argvars[2].v_type != VAR_UNKNOWN)
3036 copy_tv(&argvars[2], rettv);
3037 }
3038 else
3039 copy_tv(tv, rettv);
3040}
3041
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003042/*
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003043 * "getchangelist()" function
3044 */
3045 static void
3046f_getchangelist(typval_T *argvars, typval_T *rettv)
3047{
3048#ifdef FEAT_JUMPLIST
3049 buf_T *buf;
3050 int i;
3051 list_T *l;
3052 dict_T *d;
3053#endif
3054
3055 if (rettv_list_alloc(rettv) != OK)
3056 return;
3057
3058#ifdef FEAT_JUMPLIST
Bram Moolenaar4c313b12019-08-24 22:58:31 +02003059 if (argvars[0].v_type == VAR_UNKNOWN)
3060 buf = curbuf;
3061 else
3062 {
3063 (void)tv_get_number(&argvars[0]); // issue errmsg if type error
3064 ++emsg_off;
3065 buf = tv_get_buf(&argvars[0], FALSE);
3066 --emsg_off;
3067 }
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003068 if (buf == NULL)
3069 return;
3070
3071 l = list_alloc();
3072 if (l == NULL)
3073 return;
3074
3075 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3076 return;
3077 /*
3078 * The current window change list index tracks only the position in the
3079 * current buffer change list. For other buffers, use the change list
3080 * length as the current index.
3081 */
3082 list_append_number(rettv->vval.v_list,
3083 (varnumber_T)((buf == curwin->w_buffer)
3084 ? curwin->w_changelistidx : buf->b_changelistlen));
3085
3086 for (i = 0; i < buf->b_changelistlen; ++i)
3087 {
3088 if (buf->b_changelist[i].lnum == 0)
3089 continue;
3090 if ((d = dict_alloc()) == NULL)
3091 return;
3092 if (list_append_dict(l, d) == FAIL)
3093 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003094 dict_add_number(d, "lnum", (long)buf->b_changelist[i].lnum);
3095 dict_add_number(d, "col", (long)buf->b_changelist[i].col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003096 dict_add_number(d, "coladd", (long)buf->b_changelist[i].coladd);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003097 }
3098#endif
3099}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003100
3101/*
3102 * "getcharsearch()" function
3103 */
3104 static void
3105f_getcharsearch(typval_T *argvars UNUSED, typval_T *rettv)
3106{
3107 if (rettv_dict_alloc(rettv) != FAIL)
3108 {
3109 dict_T *dict = rettv->vval.v_dict;
3110
Bram Moolenaare0be1672018-07-08 16:50:37 +02003111 dict_add_string(dict, "char", last_csearch());
3112 dict_add_number(dict, "forward", last_csearch_forward());
3113 dict_add_number(dict, "until", last_csearch_until());
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003114 }
3115}
3116
3117/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02003118 * "getenv()" function
3119 */
3120 static void
3121f_getenv(typval_T *argvars, typval_T *rettv)
3122{
3123 int mustfree = FALSE;
3124 char_u *p = vim_getenv(tv_get_string(&argvars[0]), &mustfree);
3125
3126 if (p == NULL)
3127 {
3128 rettv->v_type = VAR_SPECIAL;
3129 rettv->vval.v_number = VVAL_NULL;
3130 return;
3131 }
3132 if (!mustfree)
3133 p = vim_strsave(p);
3134 rettv->vval.v_string = p;
3135 rettv->v_type = VAR_STRING;
3136}
3137
3138/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003139 * "getfontname()" function
3140 */
3141 static void
3142f_getfontname(typval_T *argvars UNUSED, typval_T *rettv)
3143{
3144 rettv->v_type = VAR_STRING;
3145 rettv->vval.v_string = NULL;
3146#ifdef FEAT_GUI
3147 if (gui.in_use)
3148 {
3149 GuiFont font;
3150 char_u *name = NULL;
3151
3152 if (argvars[0].v_type == VAR_UNKNOWN)
3153 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003154 // Get the "Normal" font. Either the name saved by
3155 // hl_set_font_name() or from the font ID.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003156 font = gui.norm_font;
3157 name = hl_get_font_name();
3158 }
3159 else
3160 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003161 name = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003162 if (STRCMP(name, "*") == 0) // don't use font dialog
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003163 return;
3164 font = gui_mch_get_font(name, FALSE);
3165 if (font == NOFONT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003166 return; // Invalid font name, return empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003167 }
3168 rettv->vval.v_string = gui_mch_get_fontname(font, name);
3169 if (argvars[0].v_type != VAR_UNKNOWN)
3170 gui_mch_free_font(font);
3171 }
3172#endif
3173}
3174
3175/*
Bram Moolenaar4f505882018-02-10 21:06:32 +01003176 * "getjumplist()" function
3177 */
3178 static void
3179f_getjumplist(typval_T *argvars, typval_T *rettv)
3180{
3181#ifdef FEAT_JUMPLIST
3182 win_T *wp;
3183 int i;
3184 list_T *l;
3185 dict_T *d;
3186#endif
3187
3188 if (rettv_list_alloc(rettv) != OK)
3189 return;
3190
3191#ifdef FEAT_JUMPLIST
Bram Moolenaar00aa0692019-04-27 20:37:57 +02003192 wp = find_tabwin(&argvars[0], &argvars[1], NULL);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003193 if (wp == NULL)
3194 return;
3195
Bram Moolenaar57ee2b62019-02-12 22:15:06 +01003196 cleanup_jumplist(wp, TRUE);
3197
Bram Moolenaar4f505882018-02-10 21:06:32 +01003198 l = list_alloc();
3199 if (l == NULL)
3200 return;
3201
3202 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3203 return;
3204 list_append_number(rettv->vval.v_list, (varnumber_T)wp->w_jumplistidx);
3205
3206 for (i = 0; i < wp->w_jumplistlen; ++i)
3207 {
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003208 if (wp->w_jumplist[i].fmark.mark.lnum == 0)
3209 continue;
Bram Moolenaar4f505882018-02-10 21:06:32 +01003210 if ((d = dict_alloc()) == NULL)
3211 return;
3212 if (list_append_dict(l, d) == FAIL)
3213 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003214 dict_add_number(d, "lnum", (long)wp->w_jumplist[i].fmark.mark.lnum);
3215 dict_add_number(d, "col", (long)wp->w_jumplist[i].fmark.mark.col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003216 dict_add_number(d, "coladd", (long)wp->w_jumplist[i].fmark.mark.coladd);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003217 dict_add_number(d, "bufnr", (long)wp->w_jumplist[i].fmark.fnum);
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003218 if (wp->w_jumplist[i].fname != NULL)
Bram Moolenaare0be1672018-07-08 16:50:37 +02003219 dict_add_string(d, "filename", wp->w_jumplist[i].fname);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003220 }
3221#endif
3222}
3223
3224/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003225 * "getpid()" function
3226 */
3227 static void
3228f_getpid(typval_T *argvars UNUSED, typval_T *rettv)
3229{
3230 rettv->vval.v_number = mch_get_pid();
3231}
3232
3233 static void
3234getpos_both(
3235 typval_T *argvars,
3236 typval_T *rettv,
3237 int getcurpos)
3238{
3239 pos_T *fp;
3240 list_T *l;
3241 int fnum = -1;
3242
3243 if (rettv_list_alloc(rettv) == OK)
3244 {
3245 l = rettv->vval.v_list;
3246 if (getcurpos)
3247 fp = &curwin->w_cursor;
3248 else
3249 fp = var2fpos(&argvars[0], TRUE, &fnum);
3250 if (fnum != -1)
3251 list_append_number(l, (varnumber_T)fnum);
3252 else
3253 list_append_number(l, (varnumber_T)0);
3254 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->lnum
3255 : (varnumber_T)0);
3256 list_append_number(l, (fp != NULL)
3257 ? (varnumber_T)(fp->col == MAXCOL ? MAXCOL : fp->col + 1)
3258 : (varnumber_T)0);
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01003259 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->coladd :
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003260 (varnumber_T)0);
3261 if (getcurpos)
3262 {
Bram Moolenaar19a66852019-03-07 11:25:32 +01003263 int save_set_curswant = curwin->w_set_curswant;
3264 colnr_T save_curswant = curwin->w_curswant;
3265 colnr_T save_virtcol = curwin->w_virtcol;
3266
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003267 update_curswant();
3268 list_append_number(l, curwin->w_curswant == MAXCOL ?
3269 (varnumber_T)MAXCOL : (varnumber_T)curwin->w_curswant + 1);
Bram Moolenaar19a66852019-03-07 11:25:32 +01003270
3271 // Do not change "curswant", as it is unexpected that a get
3272 // function has a side effect.
3273 if (save_set_curswant)
3274 {
3275 curwin->w_set_curswant = save_set_curswant;
3276 curwin->w_curswant = save_curswant;
3277 curwin->w_virtcol = save_virtcol;
3278 curwin->w_valid &= ~VALID_VIRTCOL;
3279 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003280 }
3281 }
3282 else
3283 rettv->vval.v_number = FALSE;
3284}
3285
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003286/*
3287 * "getcurpos()" function
3288 */
3289 static void
3290f_getcurpos(typval_T *argvars, typval_T *rettv)
3291{
3292 getpos_both(argvars, rettv, TRUE);
3293}
3294
3295/*
3296 * "getpos(string)" function
3297 */
3298 static void
3299f_getpos(typval_T *argvars, typval_T *rettv)
3300{
3301 getpos_both(argvars, rettv, FALSE);
3302}
3303
3304/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003305 * "getreg()" function
3306 */
3307 static void
3308f_getreg(typval_T *argvars, typval_T *rettv)
3309{
3310 char_u *strregname;
3311 int regname;
3312 int arg2 = FALSE;
3313 int return_list = FALSE;
3314 int error = FALSE;
3315
3316 if (argvars[0].v_type != VAR_UNKNOWN)
3317 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003318 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003319 error = strregname == NULL;
3320 if (argvars[1].v_type != VAR_UNKNOWN)
3321 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003322 arg2 = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003323 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003324 return_list = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003325 }
3326 }
3327 else
3328 strregname = get_vim_var_str(VV_REG);
3329
3330 if (error)
3331 return;
3332
3333 regname = (strregname == NULL ? '"' : *strregname);
3334 if (regname == 0)
3335 regname = '"';
3336
3337 if (return_list)
3338 {
3339 rettv->v_type = VAR_LIST;
3340 rettv->vval.v_list = (list_T *)get_reg_contents(regname,
3341 (arg2 ? GREG_EXPR_SRC : 0) | GREG_LIST);
3342 if (rettv->vval.v_list == NULL)
3343 (void)rettv_list_alloc(rettv);
3344 else
3345 ++rettv->vval.v_list->lv_refcount;
3346 }
3347 else
3348 {
3349 rettv->v_type = VAR_STRING;
3350 rettv->vval.v_string = get_reg_contents(regname,
3351 arg2 ? GREG_EXPR_SRC : 0);
3352 }
3353}
3354
3355/*
3356 * "getregtype()" function
3357 */
3358 static void
3359f_getregtype(typval_T *argvars, typval_T *rettv)
3360{
3361 char_u *strregname;
3362 int regname;
3363 char_u buf[NUMBUFLEN + 2];
3364 long reglen = 0;
3365
3366 if (argvars[0].v_type != VAR_UNKNOWN)
3367 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003368 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003369 if (strregname == NULL) // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003370 {
3371 rettv->v_type = VAR_STRING;
3372 rettv->vval.v_string = NULL;
3373 return;
3374 }
3375 }
3376 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003377 // Default to v:register
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003378 strregname = get_vim_var_str(VV_REG);
3379
3380 regname = (strregname == NULL ? '"' : *strregname);
3381 if (regname == 0)
3382 regname = '"';
3383
3384 buf[0] = NUL;
3385 buf[1] = NUL;
3386 switch (get_reg_type(regname, &reglen))
3387 {
3388 case MLINE: buf[0] = 'V'; break;
3389 case MCHAR: buf[0] = 'v'; break;
3390 case MBLOCK:
3391 buf[0] = Ctrl_V;
3392 sprintf((char *)buf + 1, "%ld", reglen + 1);
3393 break;
3394 }
3395 rettv->v_type = VAR_STRING;
3396 rettv->vval.v_string = vim_strsave(buf);
3397}
3398
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003399/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01003400 * "gettagstack()" function
3401 */
3402 static void
3403f_gettagstack(typval_T *argvars, typval_T *rettv)
3404{
3405 win_T *wp = curwin; // default is current window
3406
3407 if (rettv_dict_alloc(rettv) != OK)
3408 return;
3409
3410 if (argvars[0].v_type != VAR_UNKNOWN)
3411 {
3412 wp = find_win_by_nr_or_id(&argvars[0]);
3413 if (wp == NULL)
3414 return;
3415 }
3416
3417 get_tagstack(wp, rettv->vval.v_dict);
3418}
3419
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003420// for VIM_VERSION_ defines
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003421#include "version.h"
3422
3423/*
3424 * "has()" function
3425 */
Bram Moolenaara259d8d2020-01-31 20:10:50 +01003426 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003427f_has(typval_T *argvars, typval_T *rettv)
3428{
3429 int i;
3430 char_u *name;
Bram Moolenaar79296512020-03-22 16:17:14 +01003431 int x = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003432 int n = FALSE;
Bram Moolenaar79296512020-03-22 16:17:14 +01003433 typedef struct {
3434 char *name;
3435 short present;
3436 } has_item_T;
3437 static has_item_T has_list[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003438 {
Bram Moolenaar79296512020-03-22 16:17:14 +01003439 {"amiga",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003440#ifdef AMIGA
Bram Moolenaar79296512020-03-22 16:17:14 +01003441 1
Bram Moolenaar39536dd2019-01-29 22:58:21 +01003442#else
Bram Moolenaar79296512020-03-22 16:17:14 +01003443 0
Bram Moolenaar39536dd2019-01-29 22:58:21 +01003444#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003445 },
3446 {"arp",
3447#if defined(AMIGA) && defined(FEAT_ARP)
3448 1
3449#else
3450 0
3451#endif
3452 },
3453 {"beos",
3454#ifdef __BEOS__
3455 1
3456#else
3457 0
3458#endif
3459 },
3460 {"haiku",
3461#ifdef __HAIKU__
3462 1
3463#else
3464 0
3465#endif
3466 },
3467 {"bsd",
3468#if defined(BSD) && !defined(MACOS_X)
3469 1
3470#else
3471 0
3472#endif
3473 },
3474 {"hpux",
3475#ifdef hpux
3476 1
3477#else
3478 0
3479#endif
3480 },
3481 {"linux",
3482#ifdef __linux__
3483 1
3484#else
3485 0
3486#endif
3487 },
3488 {"mac", // Mac OS X (and, once, Mac OS Classic)
3489#ifdef MACOS_X
3490 1
3491#else
3492 0
3493#endif
3494 },
3495 {"osx", // Mac OS X
3496#ifdef MACOS_X
3497 1
3498#else
3499 0
3500#endif
3501 },
3502 {"macunix", // Mac OS X, with the darwin feature
3503#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
3504 1
3505#else
3506 0
3507#endif
3508 },
3509 {"osxdarwin", // synonym for macunix
3510#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
3511 1
3512#else
3513 0
3514#endif
3515 },
3516 {"qnx",
3517#ifdef __QNX__
3518 1
3519#else
3520 0
3521#endif
3522 },
3523 {"sun",
3524#ifdef SUN_SYSTEM
3525 1
3526#else
3527 0
3528#endif
3529 },
3530 {"unix",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003531#ifdef UNIX
Bram Moolenaar79296512020-03-22 16:17:14 +01003532 1
3533#else
3534 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003535#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003536 },
3537 {"vms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003538#ifdef VMS
Bram Moolenaar79296512020-03-22 16:17:14 +01003539 1
3540#else
3541 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003542#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003543 },
3544 {"win32",
Bram Moolenaar4f974752019-02-17 17:44:42 +01003545#ifdef MSWIN
Bram Moolenaar79296512020-03-22 16:17:14 +01003546 1
3547#else
3548 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003549#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003550 },
3551 {"win32unix",
Bram Moolenaar1eed5322019-02-26 17:03:54 +01003552#if defined(UNIX) && defined(__CYGWIN__)
Bram Moolenaar79296512020-03-22 16:17:14 +01003553 1
3554#else
3555 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003556#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003557 },
3558 {"win64",
Bram Moolenaar44b443c2019-02-18 22:14:18 +01003559#ifdef _WIN64
Bram Moolenaar79296512020-03-22 16:17:14 +01003560 1
3561#else
3562 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003563#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003564 },
3565 {"ebcdic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003566#ifdef EBCDIC
Bram Moolenaar79296512020-03-22 16:17:14 +01003567 1
3568#else
3569 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003570#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003571 },
3572 {"fname_case",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003573#ifndef CASE_INSENSITIVE_FILENAME
Bram Moolenaar79296512020-03-22 16:17:14 +01003574 1
3575#else
3576 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003577#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003578 },
3579 {"acl",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003580#ifdef HAVE_ACL
Bram Moolenaar79296512020-03-22 16:17:14 +01003581 1
3582#else
3583 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003584#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003585 },
3586 {"arabic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003587#ifdef FEAT_ARABIC
Bram Moolenaar79296512020-03-22 16:17:14 +01003588 1
3589#else
3590 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003591#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003592 },
3593 {"autocmd", 1},
3594 {"autochdir",
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02003595#ifdef FEAT_AUTOCHDIR
Bram Moolenaar79296512020-03-22 16:17:14 +01003596 1
3597#else
3598 0
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02003599#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003600 },
3601 {"autoservername",
Bram Moolenaare42a6d22017-11-12 19:21:51 +01003602#ifdef FEAT_AUTOSERVERNAME
Bram Moolenaar79296512020-03-22 16:17:14 +01003603 1
3604#else
3605 0
Bram Moolenaare42a6d22017-11-12 19:21:51 +01003606#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003607 },
3608 {"balloon_eval",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01003609#ifdef FEAT_BEVAL_GUI
Bram Moolenaar79296512020-03-22 16:17:14 +01003610 1
3611#else
3612 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003613#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003614 },
3615 {"balloon_multiline",
3616#if defined(FEAT_BEVAL_GUI) && !defined(FEAT_GUI_MSWIN)
3617 // MS-Windows requires runtime check, see below
3618 1
3619#else
3620 0
3621#endif
3622 },
3623 {"balloon_eval_term",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01003624#ifdef FEAT_BEVAL_TERM
Bram Moolenaar79296512020-03-22 16:17:14 +01003625 1
3626#else
3627 0
Bram Moolenaar51b0f372017-11-18 18:52:04 +01003628#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003629 },
3630 {"builtin_terms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003631#if defined(SOME_BUILTIN_TCAPS) || defined(ALL_BUILTIN_TCAPS)
Bram Moolenaar79296512020-03-22 16:17:14 +01003632 1
3633#else
3634 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003635#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003636 },
3637 {"all_builtin_terms",
3638#if defined(ALL_BUILTIN_TCAPS)
3639 1
3640#else
3641 0
3642#endif
3643 },
3644 {"browsefilter",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003645#if defined(FEAT_BROWSE) && (defined(USE_FILE_CHOOSER) \
Bram Moolenaar4f974752019-02-17 17:44:42 +01003646 || defined(FEAT_GUI_MSWIN) \
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003647 || defined(FEAT_GUI_MOTIF))
Bram Moolenaar79296512020-03-22 16:17:14 +01003648 1
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003649#else
Bram Moolenaar79296512020-03-22 16:17:14 +01003650 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003651#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003652 },
3653 {"byte_offset",
3654#ifdef FEAT_BYTEOFF
3655 1
3656#else
3657 0
3658#endif
3659 },
3660 {"channel",
3661#ifdef FEAT_JOB_CHANNEL
3662 1
3663#else
3664 0
3665#endif
3666 },
3667 {"cindent",
3668#ifdef FEAT_CINDENT
3669 1
3670#else
3671 0
3672#endif
3673 },
3674 {"clientserver",
3675#ifdef FEAT_CLIENTSERVER
3676 1
3677#else
3678 0
3679#endif
3680 },
3681 {"clipboard",
3682#ifdef FEAT_CLIPBOARD
3683 1
3684#else
3685 0
3686#endif
3687 },
3688 {"cmdline_compl", 1},
3689 {"cmdline_hist", 1},
3690 {"comments", 1},
3691 {"conceal",
3692#ifdef FEAT_CONCEAL
3693 1
3694#else
3695 0
3696#endif
3697 },
3698 {"cryptv",
3699#ifdef FEAT_CRYPT
3700 1
3701#else
3702 0
3703#endif
3704 },
3705 {"crypt-blowfish",
3706#ifdef FEAT_CRYPT
3707 1
3708#else
3709 0
3710#endif
3711 },
3712 {"crypt-blowfish2",
3713#ifdef FEAT_CRYPT
3714 1
3715#else
3716 0
3717#endif
3718 },
3719 {"cscope",
3720#ifdef FEAT_CSCOPE
3721 1
3722#else
3723 0
3724#endif
3725 },
3726 {"cursorbind", 1},
3727 {"cursorshape",
3728#ifdef CURSOR_SHAPE
3729 1
3730#else
3731 0
3732#endif
3733 },
3734 {"debug",
3735#ifdef DEBUG
3736 1
3737#else
3738 0
3739#endif
3740 },
3741 {"dialog_con",
3742#ifdef FEAT_CON_DIALOG
3743 1
3744#else
3745 0
3746#endif
3747 },
3748 {"dialog_gui",
3749#ifdef FEAT_GUI_DIALOG
3750 1
3751#else
3752 0
3753#endif
3754 },
3755 {"diff",
3756#ifdef FEAT_DIFF
3757 1
3758#else
3759 0
3760#endif
3761 },
3762 {"digraphs",
3763#ifdef FEAT_DIGRAPHS
3764 1
3765#else
3766 0
3767#endif
3768 },
3769 {"directx",
3770#ifdef FEAT_DIRECTX
3771 1
3772#else
3773 0
3774#endif
3775 },
3776 {"dnd",
3777#ifdef FEAT_DND
3778 1
3779#else
3780 0
3781#endif
3782 },
3783 {"emacs_tags",
3784#ifdef FEAT_EMACS_TAGS
3785 1
3786#else
3787 0
3788#endif
3789 },
3790 {"eval", 1}, // always present, of course!
3791 {"ex_extra", 1}, // graduated feature
3792 {"extra_search",
3793#ifdef FEAT_SEARCH_EXTRA
3794 1
3795#else
3796 0
3797#endif
3798 },
3799 {"file_in_path",
3800#ifdef FEAT_SEARCHPATH
3801 1
3802#else
3803 0
3804#endif
3805 },
3806 {"filterpipe",
3807#if defined(FEAT_FILTERPIPE) && !defined(VIMDLL)
3808 1
3809#else
3810 0
3811#endif
3812 },
3813 {"find_in_path",
3814#ifdef FEAT_FIND_ID
3815 1
3816#else
3817 0
3818#endif
3819 },
3820 {"float",
3821#ifdef FEAT_FLOAT
3822 1
3823#else
3824 0
3825#endif
3826 },
3827 {"folding",
3828#ifdef FEAT_FOLDING
3829 1
3830#else
3831 0
3832#endif
3833 },
3834 {"footer",
3835#ifdef FEAT_FOOTER
3836 1
3837#else
3838 0
3839#endif
3840 },
3841 {"fork",
3842#if !defined(USE_SYSTEM) && defined(UNIX)
3843 1
3844#else
3845 0
3846#endif
3847 },
3848 {"gettext",
3849#ifdef FEAT_GETTEXT
3850 1
3851#else
3852 0
3853#endif
3854 },
3855 {"gui",
3856#ifdef FEAT_GUI
3857 1
3858#else
3859 0
3860#endif
3861 },
3862 {"gui_neXtaw",
3863#if defined(FEAT_GUI_ATHENA) && defined(FEAT_GUI_NEXTAW)
3864 1
3865#else
3866 0
3867#endif
3868 },
3869 {"gui_athena",
3870#if defined(FEAT_GUI_ATHENA) && !defined(FEAT_GUI_NEXTAW)
3871 1
3872#else
3873 0
3874#endif
3875 },
3876 {"gui_gtk",
3877#ifdef FEAT_GUI_GTK
3878 1
3879#else
3880 0
3881#endif
3882 },
3883 {"gui_gtk2",
3884#if defined(FEAT_GUI_GTK) && !defined(USE_GTK3)
3885 1
3886#else
3887 0
3888#endif
3889 },
3890 {"gui_gtk3",
3891#if defined(FEAT_GUI_GTK) && defined(USE_GTK3)
3892 1
3893#else
3894 0
3895#endif
3896 },
3897 {"gui_gnome",
3898#ifdef FEAT_GUI_GNOME
3899 1
3900#else
3901 0
3902#endif
3903 },
3904 {"gui_haiku",
3905#ifdef FEAT_GUI_HAIKU
3906 1
3907#else
3908 0
3909#endif
3910 },
3911 {"gui_mac",
3912#ifdef FEAT_GUI_MAC
3913 1
3914#else
3915 0
3916#endif
3917 },
3918 {"gui_motif",
3919#ifdef FEAT_GUI_MOTIF
3920 1
3921#else
3922 0
3923#endif
3924 },
3925 {"gui_photon",
3926#ifdef FEAT_GUI_PHOTON
3927 1
3928#else
3929 0
3930#endif
3931 },
3932 {"gui_win32",
3933#ifdef FEAT_GUI_MSWIN
3934 1
3935#else
3936 0
3937#endif
3938 },
3939 {"iconv",
3940#if defined(HAVE_ICONV_H) && defined(USE_ICONV)
3941 1
3942#else
3943 0
3944#endif
3945 },
3946 {"insert_expand", 1},
Bram Moolenaarbfe13cc2020-04-12 17:53:12 +02003947 {"ipv6",
3948#ifdef FEAT_IPV6
3949 1
3950#else
3951 0
3952#endif
3953 },
Bram Moolenaar79296512020-03-22 16:17:14 +01003954 {"job",
3955#ifdef FEAT_JOB_CHANNEL
3956 1
3957#else
3958 0
3959#endif
3960 },
3961 {"jumplist",
3962#ifdef FEAT_JUMPLIST
3963 1
3964#else
3965 0
3966#endif
3967 },
3968 {"keymap",
3969#ifdef FEAT_KEYMAP
3970 1
3971#else
3972 0
3973#endif
3974 },
3975 {"lambda", 1}, // always with FEAT_EVAL, since 7.4.2120 with closure
3976 {"langmap",
3977#ifdef FEAT_LANGMAP
3978 1
3979#else
3980 0
3981#endif
3982 },
3983 {"libcall",
3984#ifdef FEAT_LIBCALL
3985 1
3986#else
3987 0
3988#endif
3989 },
3990 {"linebreak",
3991#ifdef FEAT_LINEBREAK
3992 1
3993#else
3994 0
3995#endif
3996 },
3997 {"lispindent",
3998#ifdef FEAT_LISP
3999 1
4000#else
4001 0
4002#endif
4003 },
4004 {"listcmds", 1},
4005 {"localmap", 1},
4006 {"lua",
4007#if defined(FEAT_LUA) && !defined(DYNAMIC_LUA)
4008 1
4009#else
4010 0
4011#endif
4012 },
4013 {"menu",
4014#ifdef FEAT_MENU
4015 1
4016#else
4017 0
4018#endif
4019 },
4020 {"mksession",
4021#ifdef FEAT_SESSION
4022 1
4023#else
4024 0
4025#endif
4026 },
4027 {"modify_fname", 1},
4028 {"mouse", 1},
4029 {"mouseshape",
4030#ifdef FEAT_MOUSESHAPE
4031 1
4032#else
4033 0
4034#endif
4035 },
4036 {"mouse_dec",
4037#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_DEC)
4038 1
4039#else
4040 0
4041#endif
4042 },
4043 {"mouse_gpm",
4044#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_GPM)
4045 1
4046#else
4047 0
4048#endif
4049 },
4050 {"mouse_jsbterm",
4051#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_JSB)
4052 1
4053#else
4054 0
4055#endif
4056 },
4057 {"mouse_netterm",
4058#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_NET)
4059 1
4060#else
4061 0
4062#endif
4063 },
4064 {"mouse_pterm",
4065#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_PTERM)
4066 1
4067#else
4068 0
4069#endif
4070 },
4071 {"mouse_sgr",
4072#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4073 1
4074#else
4075 0
4076#endif
4077 },
4078 {"mouse_sysmouse",
4079#if (defined(UNIX) || defined(VMS)) && defined(FEAT_SYSMOUSE)
4080 1
4081#else
4082 0
4083#endif
4084 },
4085 {"mouse_urxvt",
4086#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_URXVT)
4087 1
4088#else
4089 0
4090#endif
4091 },
4092 {"mouse_xterm",
4093#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4094 1
4095#else
4096 0
4097#endif
4098 },
4099 {"multi_byte", 1},
4100 {"multi_byte_ime",
4101#ifdef FEAT_MBYTE_IME
4102 1
4103#else
4104 0
4105#endif
4106 },
4107 {"multi_lang",
4108#ifdef FEAT_MULTI_LANG
4109 1
4110#else
4111 0
4112#endif
4113 },
4114 {"mzscheme",
4115#if defined(FEAT_MZSCHEME) && !defined(DYNAMIC_MZSCHEME)
4116 1
4117#else
4118 0
4119#endif
4120 },
4121 {"num64", 1},
4122 {"ole",
4123#ifdef FEAT_OLE
4124 1
4125#else
4126 0
4127#endif
4128 },
4129 {"packages",
4130#ifdef FEAT_EVAL
4131 1
4132#else
4133 0
4134#endif
4135 },
4136 {"path_extra",
4137#ifdef FEAT_PATH_EXTRA
4138 1
4139#else
4140 0
4141#endif
4142 },
4143 {"perl",
4144#if defined(FEAT_PERL) && !defined(DYNAMIC_PERL)
4145 1
4146#else
4147 0
4148#endif
4149 },
4150 {"persistent_undo",
4151#ifdef FEAT_PERSISTENT_UNDO
4152 1
4153#else
4154 0
4155#endif
4156 },
4157 {"python_compiled",
4158#if defined(FEAT_PYTHON)
4159 1
4160#else
4161 0
4162#endif
4163 },
4164 {"python_dynamic",
4165#if defined(FEAT_PYTHON) && defined(DYNAMIC_PYTHON)
4166 1
4167#else
4168 0
4169#endif
4170 },
4171 {"python",
4172#if defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)
4173 1
4174#else
4175 0
4176#endif
4177 },
4178 {"pythonx",
4179#if (defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)) \
4180 || (defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3))
4181 1
4182#else
4183 0
4184#endif
4185 },
4186 {"python3_compiled",
4187#if defined(FEAT_PYTHON3)
4188 1
4189#else
4190 0
4191#endif
4192 },
4193 {"python3_dynamic",
4194#if defined(FEAT_PYTHON3) && defined(DYNAMIC_PYTHON3)
4195 1
4196#else
4197 0
4198#endif
4199 },
4200 {"python3",
4201#if defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3)
4202 1
4203#else
4204 0
4205#endif
4206 },
4207 {"popupwin",
4208#ifdef FEAT_PROP_POPUP
4209 1
4210#else
4211 0
4212#endif
4213 },
4214 {"postscript",
4215#ifdef FEAT_POSTSCRIPT
4216 1
4217#else
4218 0
4219#endif
4220 },
4221 {"printer",
4222#ifdef FEAT_PRINTER
4223 1
4224#else
4225 0
4226#endif
4227 },
4228 {"profile",
4229#ifdef FEAT_PROFILE
4230 1
4231#else
4232 0
4233#endif
4234 },
4235 {"reltime",
4236#ifdef FEAT_RELTIME
4237 1
4238#else
4239 0
4240#endif
4241 },
4242 {"quickfix",
4243#ifdef FEAT_QUICKFIX
4244 1
4245#else
4246 0
4247#endif
4248 },
4249 {"rightleft",
4250#ifdef FEAT_RIGHTLEFT
4251 1
4252#else
4253 0
4254#endif
4255 },
4256 {"ruby",
4257#if defined(FEAT_RUBY) && !defined(DYNAMIC_RUBY)
4258 1
4259#else
4260 0
4261#endif
4262 },
4263 {"scrollbind", 1},
4264 {"showcmd",
4265#ifdef FEAT_CMDL_INFO
4266 1
4267#else
4268 0
4269#endif
4270 },
4271 {"cmdline_info",
4272#ifdef FEAT_CMDL_INFO
4273 1
4274#else
4275 0
4276#endif
4277 },
4278 {"signs",
4279#ifdef FEAT_SIGNS
4280 1
4281#else
4282 0
4283#endif
4284 },
4285 {"smartindent",
4286#ifdef FEAT_SMARTINDENT
4287 1
4288#else
4289 0
4290#endif
4291 },
4292 {"startuptime",
4293#ifdef STARTUPTIME
4294 1
4295#else
4296 0
4297#endif
4298 },
4299 {"statusline",
4300#ifdef FEAT_STL_OPT
4301 1
4302#else
4303 0
4304#endif
4305 },
4306 {"netbeans_intg",
4307#ifdef FEAT_NETBEANS_INTG
4308 1
4309#else
4310 0
4311#endif
4312 },
4313 {"sound",
4314#ifdef FEAT_SOUND
4315 1
4316#else
4317 0
4318#endif
4319 },
4320 {"spell",
4321#ifdef FEAT_SPELL
4322 1
4323#else
4324 0
4325#endif
4326 },
4327 {"syntax",
4328#ifdef FEAT_SYN_HL
4329 1
4330#else
4331 0
4332#endif
4333 },
4334 {"system",
4335#if defined(USE_SYSTEM) || !defined(UNIX)
4336 1
4337#else
4338 0
4339#endif
4340 },
4341 {"tag_binary",
4342#ifdef FEAT_TAG_BINS
4343 1
4344#else
4345 0
4346#endif
4347 },
4348 {"tcl",
4349#if defined(FEAT_TCL) && !defined(DYNAMIC_TCL)
4350 1
4351#else
4352 0
4353#endif
4354 },
4355 {"termguicolors",
4356#ifdef FEAT_TERMGUICOLORS
4357 1
4358#else
4359 0
4360#endif
4361 },
4362 {"terminal",
4363#if defined(FEAT_TERMINAL) && !defined(MSWIN)
4364 1
4365#else
4366 0
4367#endif
4368 },
4369 {"terminfo",
4370#ifdef TERMINFO
4371 1
4372#else
4373 0
4374#endif
4375 },
4376 {"termresponse",
4377#ifdef FEAT_TERMRESPONSE
4378 1
4379#else
4380 0
4381#endif
4382 },
4383 {"textobjects",
4384#ifdef FEAT_TEXTOBJ
4385 1
4386#else
4387 0
4388#endif
4389 },
4390 {"textprop",
4391#ifdef FEAT_PROP_POPUP
4392 1
4393#else
4394 0
4395#endif
4396 },
4397 {"tgetent",
4398#ifdef HAVE_TGETENT
4399 1
4400#else
4401 0
4402#endif
4403 },
4404 {"timers",
4405#ifdef FEAT_TIMERS
4406 1
4407#else
4408 0
4409#endif
4410 },
4411 {"title",
4412#ifdef FEAT_TITLE
4413 1
4414#else
4415 0
4416#endif
4417 },
4418 {"toolbar",
4419#ifdef FEAT_TOOLBAR
4420 1
4421#else
4422 0
4423#endif
4424 },
4425 {"unnamedplus",
4426#if defined(FEAT_CLIPBOARD) && defined(FEAT_X11)
4427 1
4428#else
4429 0
4430#endif
4431 },
4432 {"user-commands", 1}, // was accidentally included in 5.4
4433 {"user_commands", 1},
4434 {"vartabs",
4435#ifdef FEAT_VARTABS
4436 1
4437#else
4438 0
4439#endif
4440 },
4441 {"vertsplit", 1},
4442 {"viminfo",
4443#ifdef FEAT_VIMINFO
4444 1
4445#else
4446 0
4447#endif
4448 },
4449 {"vimscript-1", 1},
4450 {"vimscript-2", 1},
4451 {"vimscript-3", 1},
4452 {"vimscript-4", 1},
4453 {"virtualedit", 1},
4454 {"visual", 1},
4455 {"visualextra", 1},
4456 {"vreplace", 1},
4457 {"vtp",
4458#ifdef FEAT_VTP
4459 1
4460#else
4461 0
4462#endif
4463 },
4464 {"wildignore",
4465#ifdef FEAT_WILDIGN
4466 1
4467#else
4468 0
4469#endif
4470 },
4471 {"wildmenu",
4472#ifdef FEAT_WILDMENU
4473 1
4474#else
4475 0
4476#endif
4477 },
4478 {"windows", 1},
4479 {"winaltkeys",
4480#ifdef FEAT_WAK
4481 1
4482#else
4483 0
4484#endif
4485 },
4486 {"writebackup",
4487#ifdef FEAT_WRITEBACKUP
4488 1
4489#else
4490 0
4491#endif
4492 },
4493 {"xim",
4494#ifdef FEAT_XIM
4495 1
4496#else
4497 0
4498#endif
4499 },
4500 {"xfontset",
4501#ifdef FEAT_XFONTSET
4502 1
4503#else
4504 0
4505#endif
4506 },
4507 {"xpm",
4508#if defined(FEAT_XPM_W32) || defined(HAVE_XPM)
4509 1
4510#else
4511 0
4512#endif
4513 },
4514 {"xpm_w32", // for backward compatibility
4515#ifdef FEAT_XPM_W32
4516 1
4517#else
4518 0
4519#endif
4520 },
4521 {"xsmp",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004522#ifdef USE_XSMP
Bram Moolenaar79296512020-03-22 16:17:14 +01004523 1
4524#else
4525 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004526#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004527 },
4528 {"xsmp_interact",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004529#ifdef USE_XSMP_INTERACT
Bram Moolenaar79296512020-03-22 16:17:14 +01004530 1
4531#else
4532 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004533#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004534 },
4535 {"xterm_clipboard",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004536#ifdef FEAT_XCLIPBOARD
Bram Moolenaar79296512020-03-22 16:17:14 +01004537 1
4538#else
4539 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004540#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004541 },
4542 {"xterm_save",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004543#ifdef FEAT_XTERM_SAVE
Bram Moolenaar79296512020-03-22 16:17:14 +01004544 1
4545#else
4546 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004547#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004548 },
4549 {"X11",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004550#if defined(UNIX) && defined(FEAT_X11)
Bram Moolenaar79296512020-03-22 16:17:14 +01004551 1
4552#else
4553 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004554#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004555 },
4556 {NULL, 0}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004557 };
4558
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004559 name = tv_get_string(&argvars[0]);
Bram Moolenaar79296512020-03-22 16:17:14 +01004560 for (i = 0; has_list[i].name != NULL; ++i)
4561 if (STRICMP(name, has_list[i].name) == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004562 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004563 x = TRUE;
4564 n = has_list[i].present;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004565 break;
4566 }
4567
Bram Moolenaar79296512020-03-22 16:17:14 +01004568 // features also in has_list[] but sometimes enabled at runtime
4569 if (x == TRUE && n == FALSE)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004570 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004571 if (0)
Bram Moolenaar86b9a3e2020-04-07 19:57:29 +02004572 {
4573 // intentionally empty
4574 }
Bram Moolenaar4f974752019-02-17 17:44:42 +01004575#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004576 else if (STRICMP(name, "balloon_multiline") == 0)
4577 n = multiline_balloon_available();
4578#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004579#ifdef VIMDLL
4580 else if (STRICMP(name, "filterpipe") == 0)
4581 n = gui.in_use || gui.starting;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004582#endif
4583#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
4584 else if (STRICMP(name, "iconv") == 0)
4585 n = iconv_enabled(FALSE);
4586#endif
4587#ifdef DYNAMIC_LUA
4588 else if (STRICMP(name, "lua") == 0)
4589 n = lua_enabled(FALSE);
4590#endif
4591#ifdef DYNAMIC_MZSCHEME
4592 else if (STRICMP(name, "mzscheme") == 0)
4593 n = mzscheme_enabled(FALSE);
4594#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004595#ifdef DYNAMIC_PERL
4596 else if (STRICMP(name, "perl") == 0)
4597 n = perl_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004598#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004599#ifdef DYNAMIC_PYTHON
4600 else if (STRICMP(name, "python") == 0)
4601 n = python_enabled(FALSE);
4602#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004603#ifdef DYNAMIC_PYTHON3
4604 else if (STRICMP(name, "python3") == 0)
4605 n = python3_enabled(FALSE);
4606#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01004607#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
4608 else if (STRICMP(name, "pythonx") == 0)
4609 {
4610# if defined(DYNAMIC_PYTHON) && defined(DYNAMIC_PYTHON3)
4611 if (p_pyx == 0)
4612 n = python3_enabled(FALSE) || python_enabled(FALSE);
4613 else if (p_pyx == 3)
4614 n = python3_enabled(FALSE);
4615 else if (p_pyx == 2)
4616 n = python_enabled(FALSE);
4617# elif defined(DYNAMIC_PYTHON)
4618 n = python_enabled(FALSE);
4619# elif defined(DYNAMIC_PYTHON3)
4620 n = python3_enabled(FALSE);
4621# endif
4622 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004623#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004624#ifdef DYNAMIC_RUBY
4625 else if (STRICMP(name, "ruby") == 0)
4626 n = ruby_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004627#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004628#ifdef DYNAMIC_TCL
4629 else if (STRICMP(name, "tcl") == 0)
4630 n = tcl_enabled(FALSE);
Bram Moolenaar4b8366b2019-05-04 17:34:34 +02004631#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +01004632#if defined(FEAT_TERMINAL) && defined(MSWIN)
Bram Moolenaara83e3962017-08-17 14:39:07 +02004633 else if (STRICMP(name, "terminal") == 0)
4634 n = terminal_enabled();
4635#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004636 }
4637
Bram Moolenaar79296512020-03-22 16:17:14 +01004638 // features not in has_list[]
4639 if (x == FALSE)
4640 {
4641 if (STRNICMP(name, "patch", 5) == 0)
4642 {
4643 x = TRUE;
4644 if (name[5] == '-'
4645 && STRLEN(name) >= 11
4646 && vim_isdigit(name[6])
4647 && vim_isdigit(name[8])
4648 && vim_isdigit(name[10]))
4649 {
4650 int major = atoi((char *)name + 6);
4651 int minor = atoi((char *)name + 8);
4652
4653 // Expect "patch-9.9.01234".
4654 n = (major < VIM_VERSION_MAJOR
4655 || (major == VIM_VERSION_MAJOR
4656 && (minor < VIM_VERSION_MINOR
4657 || (minor == VIM_VERSION_MINOR
4658 && has_patch(atoi((char *)name + 10))))));
4659 }
4660 else
4661 n = has_patch(atoi((char *)name + 5));
4662 }
4663 else if (STRICMP(name, "vim_starting") == 0)
4664 {
4665 x = TRUE;
4666 n = (starting != 0);
4667 }
4668 else if (STRICMP(name, "ttyin") == 0)
4669 {
4670 x = TRUE;
4671 n = mch_input_isatty();
4672 }
4673 else if (STRICMP(name, "ttyout") == 0)
4674 {
4675 x = TRUE;
4676 n = stdout_isatty;
4677 }
4678 else if (STRICMP(name, "multi_byte_encoding") == 0)
4679 {
4680 x = TRUE;
4681 n = has_mbyte;
4682 }
4683 else if (STRICMP(name, "gui_running") == 0)
4684 {
4685 x = TRUE;
4686#ifdef FEAT_GUI
4687 n = (gui.in_use || gui.starting);
4688#endif
4689 }
4690 else if (STRICMP(name, "browse") == 0)
4691 {
4692 x = TRUE;
4693#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
4694 n = gui.in_use; // gui_mch_browse() works when GUI is running
4695#endif
4696 }
4697 else if (STRICMP(name, "syntax_items") == 0)
4698 {
4699 x = TRUE;
4700#ifdef FEAT_SYN_HL
4701 n = syntax_present(curwin);
4702#endif
4703 }
4704 else if (STRICMP(name, "vcon") == 0)
4705 {
4706 x = TRUE;
4707#ifdef FEAT_VTP
4708 n = is_term_win32() && has_vtp_working();
4709#endif
4710 }
4711 else if (STRICMP(name, "netbeans_enabled") == 0)
4712 {
4713 x = TRUE;
4714#ifdef FEAT_NETBEANS_INTG
4715 n = netbeans_active();
4716#endif
4717 }
4718 else if (STRICMP(name, "mouse_gpm_enabled") == 0)
4719 {
4720 x = TRUE;
4721#ifdef FEAT_MOUSE_GPM
4722 n = gpm_enabled();
4723#endif
4724 }
4725 else if (STRICMP(name, "conpty") == 0)
4726 {
4727 x = TRUE;
4728#if defined(FEAT_TERMINAL) && defined(MSWIN)
4729 n = use_conpty();
4730#endif
4731 }
4732 else if (STRICMP(name, "clipboard_working") == 0)
4733 {
4734 x = TRUE;
4735#ifdef FEAT_CLIPBOARD
4736 n = clip_star.available;
4737#endif
4738 }
4739 }
4740
4741 if (argvars[1].v_type != VAR_UNKNOWN && tv_get_number(&argvars[1]) != 0)
4742 // return whether feature could ever be enabled
4743 rettv->vval.v_number = x;
4744 else
4745 // return whether feature is enabled
4746 rettv->vval.v_number = n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004747}
4748
4749/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004750 * "haslocaldir()" function
4751 */
4752 static void
4753f_haslocaldir(typval_T *argvars, typval_T *rettv)
4754{
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004755 tabpage_T *tp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004756 win_T *wp = NULL;
4757
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004758 wp = find_tabwin(&argvars[0], &argvars[1], &tp);
4759
4760 // Check for window-local and tab-local directories
4761 if (wp != NULL && wp->w_localdir != NULL)
4762 rettv->vval.v_number = 1;
4763 else if (tp != NULL && tp->tp_localdir != NULL)
4764 rettv->vval.v_number = 2;
4765 else
4766 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004767}
4768
4769/*
4770 * "hasmapto()" function
4771 */
4772 static void
4773f_hasmapto(typval_T *argvars, typval_T *rettv)
4774{
4775 char_u *name;
4776 char_u *mode;
4777 char_u buf[NUMBUFLEN];
4778 int abbr = FALSE;
4779
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004780 name = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004781 if (argvars[1].v_type == VAR_UNKNOWN)
4782 mode = (char_u *)"nvo";
4783 else
4784 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004785 mode = tv_get_string_buf(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004786 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004787 abbr = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004788 }
4789
4790 if (map_to_exists(name, mode, abbr))
4791 rettv->vval.v_number = TRUE;
4792 else
4793 rettv->vval.v_number = FALSE;
4794}
4795
4796/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004797 * "highlightID(name)" function
4798 */
4799 static void
4800f_hlID(typval_T *argvars, typval_T *rettv)
4801{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004802 rettv->vval.v_number = syn_name2id(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004803}
4804
4805/*
4806 * "highlight_exists()" function
4807 */
4808 static void
4809f_hlexists(typval_T *argvars, typval_T *rettv)
4810{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004811 rettv->vval.v_number = highlight_exists(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004812}
4813
4814/*
4815 * "hostname()" function
4816 */
4817 static void
4818f_hostname(typval_T *argvars UNUSED, typval_T *rettv)
4819{
4820 char_u hostname[256];
4821
4822 mch_get_host_name(hostname, 256);
4823 rettv->v_type = VAR_STRING;
4824 rettv->vval.v_string = vim_strsave(hostname);
4825}
4826
4827/*
4828 * iconv() function
4829 */
4830 static void
4831f_iconv(typval_T *argvars UNUSED, typval_T *rettv)
4832{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004833 char_u buf1[NUMBUFLEN];
4834 char_u buf2[NUMBUFLEN];
4835 char_u *from, *to, *str;
4836 vimconv_T vimconv;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004837
4838 rettv->v_type = VAR_STRING;
4839 rettv->vval.v_string = NULL;
4840
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004841 str = tv_get_string(&argvars[0]);
4842 from = enc_canonize(enc_skip(tv_get_string_buf(&argvars[1], buf1)));
4843 to = enc_canonize(enc_skip(tv_get_string_buf(&argvars[2], buf2)));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004844 vimconv.vc_type = CONV_NONE;
4845 convert_setup(&vimconv, from, to);
4846
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004847 // If the encodings are equal, no conversion needed.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004848 if (vimconv.vc_type == CONV_NONE)
4849 rettv->vval.v_string = vim_strsave(str);
4850 else
4851 rettv->vval.v_string = string_convert(&vimconv, str, NULL);
4852
4853 convert_setup(&vimconv, NULL, NULL);
4854 vim_free(from);
4855 vim_free(to);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004856}
4857
4858/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004859 * "index()" function
4860 */
4861 static void
4862f_index(typval_T *argvars, typval_T *rettv)
4863{
4864 list_T *l;
4865 listitem_T *item;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004866 blob_T *b;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004867 long idx = 0;
4868 int ic = FALSE;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004869 int error = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004870
4871 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004872 if (argvars[0].v_type == VAR_BLOB)
4873 {
4874 typval_T tv;
4875 int start = 0;
4876
4877 if (argvars[2].v_type != VAR_UNKNOWN)
4878 {
4879 start = tv_get_number_chk(&argvars[2], &error);
4880 if (error)
4881 return;
4882 }
4883 b = argvars[0].vval.v_blob;
4884 if (b == NULL)
4885 return;
Bram Moolenaar05500ec2019-01-13 19:10:33 +01004886 if (start < 0)
4887 {
4888 start = blob_len(b) + start;
4889 if (start < 0)
4890 start = 0;
4891 }
4892
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004893 for (idx = start; idx < blob_len(b); ++idx)
4894 {
4895 tv.v_type = VAR_NUMBER;
4896 tv.vval.v_number = blob_get(b, idx);
4897 if (tv_equal(&tv, &argvars[1], ic, FALSE))
4898 {
4899 rettv->vval.v_number = idx;
4900 return;
4901 }
4902 }
4903 return;
4904 }
4905 else if (argvars[0].v_type != VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004906 {
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01004907 emsg(_(e_listblobreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004908 return;
4909 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004910
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004911 l = argvars[0].vval.v_list;
4912 if (l != NULL)
4913 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01004914 range_list_materialize(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004915 item = l->lv_first;
4916 if (argvars[2].v_type != VAR_UNKNOWN)
4917 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004918 // Start at specified item. Use the cached index that list_find()
4919 // sets, so that a negative number also works.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004920 item = list_find(l, (long)tv_get_number_chk(&argvars[2], &error));
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01004921 idx = l->lv_u.mat.lv_idx;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004922 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004923 ic = (int)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004924 if (error)
4925 item = NULL;
4926 }
4927
4928 for ( ; item != NULL; item = item->li_next, ++idx)
4929 if (tv_equal(&item->li_tv, &argvars[1], ic, FALSE))
4930 {
4931 rettv->vval.v_number = idx;
4932 break;
4933 }
4934 }
4935}
4936
4937static int inputsecret_flag = 0;
4938
4939/*
4940 * "input()" function
4941 * Also handles inputsecret() when inputsecret is set.
4942 */
4943 static void
4944f_input(typval_T *argvars, typval_T *rettv)
4945{
4946 get_user_input(argvars, rettv, FALSE, inputsecret_flag);
4947}
4948
4949/*
4950 * "inputdialog()" function
4951 */
4952 static void
4953f_inputdialog(typval_T *argvars, typval_T *rettv)
4954{
4955#if defined(FEAT_GUI_TEXTDIALOG)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004956 // Use a GUI dialog if the GUI is running and 'c' is not in 'guioptions'
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004957 if (gui.in_use && vim_strchr(p_go, GO_CONDIALOG) == NULL)
4958 {
4959 char_u *message;
4960 char_u buf[NUMBUFLEN];
4961 char_u *defstr = (char_u *)"";
4962
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004963 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004964 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004965 && (defstr = tv_get_string_buf_chk(&argvars[1], buf)) != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004966 vim_strncpy(IObuff, defstr, IOSIZE - 1);
4967 else
4968 IObuff[0] = NUL;
4969 if (message != NULL && defstr != NULL
4970 && do_dialog(VIM_QUESTION, NULL, message,
4971 (char_u *)_("&OK\n&Cancel"), 1, IObuff, FALSE) == 1)
4972 rettv->vval.v_string = vim_strsave(IObuff);
4973 else
4974 {
4975 if (message != NULL && defstr != NULL
4976 && argvars[1].v_type != VAR_UNKNOWN
4977 && argvars[2].v_type != VAR_UNKNOWN)
4978 rettv->vval.v_string = vim_strsave(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004979 tv_get_string_buf(&argvars[2], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004980 else
4981 rettv->vval.v_string = NULL;
4982 }
4983 rettv->v_type = VAR_STRING;
4984 }
4985 else
4986#endif
4987 get_user_input(argvars, rettv, TRUE, inputsecret_flag);
4988}
4989
4990/*
4991 * "inputlist()" function
4992 */
4993 static void
4994f_inputlist(typval_T *argvars, typval_T *rettv)
4995{
Bram Moolenaar50985eb2020-01-27 22:09:39 +01004996 list_T *l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004997 listitem_T *li;
4998 int selected;
4999 int mouse_used;
5000
5001#ifdef NO_CONSOLE_INPUT
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005002 // While starting up, there is no place to enter text. When running tests
5003 // with --not-a-term we assume feedkeys() will be used.
Bram Moolenaar91d348a2017-07-29 20:16:03 +02005004 if (no_console_input() && !is_not_a_term())
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005005 return;
5006#endif
5007 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
5008 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005009 semsg(_(e_listarg), "inputlist()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005010 return;
5011 }
5012
5013 msg_start();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005014 msg_row = Rows - 1; // for when 'cmdheight' > 1
5015 lines_left = Rows; // avoid more prompt
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005016 msg_scroll = TRUE;
5017 msg_clr_eos();
5018
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005019 l = argvars[0].vval.v_list;
5020 range_list_materialize(l);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02005021 FOR_ALL_LIST_ITEMS(l, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005022 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005023 msg_puts((char *)tv_get_string(&li->li_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005024 msg_putchar('\n');
5025 }
5026
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005027 // Ask for choice.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005028 selected = prompt_for_number(&mouse_used);
5029 if (mouse_used)
5030 selected -= lines_left;
5031
5032 rettv->vval.v_number = selected;
5033}
5034
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005035static garray_T ga_userinput = {0, 0, sizeof(tasave_T), 4, NULL};
5036
5037/*
5038 * "inputrestore()" function
5039 */
5040 static void
5041f_inputrestore(typval_T *argvars UNUSED, typval_T *rettv)
5042{
5043 if (ga_userinput.ga_len > 0)
5044 {
5045 --ga_userinput.ga_len;
5046 restore_typeahead((tasave_T *)(ga_userinput.ga_data)
5047 + ga_userinput.ga_len);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005048 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005049 }
5050 else if (p_verbose > 1)
5051 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005052 verb_msg(_("called inputrestore() more often than inputsave()"));
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005053 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005054 }
5055}
5056
5057/*
5058 * "inputsave()" function
5059 */
5060 static void
5061f_inputsave(typval_T *argvars UNUSED, typval_T *rettv)
5062{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005063 // Add an entry to the stack of typeahead storage.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005064 if (ga_grow(&ga_userinput, 1) == OK)
5065 {
5066 save_typeahead((tasave_T *)(ga_userinput.ga_data)
5067 + ga_userinput.ga_len);
5068 ++ga_userinput.ga_len;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005069 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005070 }
5071 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005072 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005073}
5074
5075/*
5076 * "inputsecret()" function
5077 */
5078 static void
5079f_inputsecret(typval_T *argvars, typval_T *rettv)
5080{
5081 ++cmdline_star;
5082 ++inputsecret_flag;
5083 f_input(argvars, rettv);
5084 --cmdline_star;
5085 --inputsecret_flag;
5086}
5087
5088/*
Bram Moolenaar67a2deb2019-11-25 00:05:32 +01005089 * "interrupt()" function
5090 */
5091 static void
5092f_interrupt(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5093{
5094 got_int = TRUE;
5095}
5096
5097/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005098 * "invert(expr)" function
5099 */
5100 static void
5101f_invert(typval_T *argvars, typval_T *rettv)
5102{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005103 rettv->vval.v_number = ~tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005104}
5105
5106/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005107 * Return TRUE if typeval "tv" is locked: Either that value is locked itself
5108 * or it refers to a List or Dictionary that is locked.
5109 */
5110 static int
5111tv_islocked(typval_T *tv)
5112{
5113 return (tv->v_lock & VAR_LOCKED)
5114 || (tv->v_type == VAR_LIST
5115 && tv->vval.v_list != NULL
5116 && (tv->vval.v_list->lv_lock & VAR_LOCKED))
5117 || (tv->v_type == VAR_DICT
5118 && tv->vval.v_dict != NULL
5119 && (tv->vval.v_dict->dv_lock & VAR_LOCKED));
5120}
5121
5122/*
5123 * "islocked()" function
5124 */
5125 static void
5126f_islocked(typval_T *argvars, typval_T *rettv)
5127{
5128 lval_T lv;
5129 char_u *end;
5130 dictitem_T *di;
5131
5132 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005133 end = get_lval(tv_get_string(&argvars[0]), NULL, &lv, FALSE, FALSE,
Bram Moolenaar3a257732017-02-21 20:47:13 +01005134 GLV_NO_AUTOLOAD | GLV_READ_ONLY, FNE_CHECK_START);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005135 if (end != NULL && lv.ll_name != NULL)
5136 {
5137 if (*end != NUL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005138 emsg(_(e_trailing));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005139 else
5140 {
5141 if (lv.ll_tv == NULL)
5142 {
Bram Moolenaar79518e22017-02-17 16:31:35 +01005143 di = find_var(lv.ll_name, NULL, TRUE);
5144 if (di != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005145 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005146 // Consider a variable locked when:
5147 // 1. the variable itself is locked
5148 // 2. the value of the variable is locked.
5149 // 3. the List or Dict value is locked.
Bram Moolenaar79518e22017-02-17 16:31:35 +01005150 rettv->vval.v_number = ((di->di_flags & DI_FLAGS_LOCK)
5151 || tv_islocked(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005152 }
5153 }
5154 else if (lv.ll_range)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005155 emsg(_("E786: Range not allowed"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005156 else if (lv.ll_newkey != NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005157 semsg(_(e_dictkey), lv.ll_newkey);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005158 else if (lv.ll_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005159 // List item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005160 rettv->vval.v_number = tv_islocked(&lv.ll_li->li_tv);
5161 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005162 // Dictionary item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005163 rettv->vval.v_number = tv_islocked(&lv.ll_di->di_tv);
5164 }
5165 }
5166
5167 clear_lval(&lv);
5168}
5169
5170#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
5171/*
Bram Moolenaarfda1bff2019-04-04 13:44:37 +02005172 * "isinf()" function
5173 */
5174 static void
5175f_isinf(typval_T *argvars, typval_T *rettv)
5176{
5177 if (argvars[0].v_type == VAR_FLOAT && isinf(argvars[0].vval.v_float))
5178 rettv->vval.v_number = argvars[0].vval.v_float > 0.0 ? 1 : -1;
5179}
5180
5181/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005182 * "isnan()" function
5183 */
5184 static void
5185f_isnan(typval_T *argvars, typval_T *rettv)
5186{
5187 rettv->vval.v_number = argvars[0].v_type == VAR_FLOAT
5188 && isnan(argvars[0].vval.v_float);
5189}
5190#endif
5191
5192/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005193 * "last_buffer_nr()" function.
5194 */
5195 static void
5196f_last_buffer_nr(typval_T *argvars UNUSED, typval_T *rettv)
5197{
5198 int n = 0;
5199 buf_T *buf;
5200
Bram Moolenaar29323592016-07-24 22:04:11 +02005201 FOR_ALL_BUFFERS(buf)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005202 if (n < buf->b_fnum)
5203 n = buf->b_fnum;
5204
5205 rettv->vval.v_number = n;
5206}
5207
5208/*
5209 * "len()" function
5210 */
5211 static void
5212f_len(typval_T *argvars, typval_T *rettv)
5213{
5214 switch (argvars[0].v_type)
5215 {
5216 case VAR_STRING:
5217 case VAR_NUMBER:
5218 rettv->vval.v_number = (varnumber_T)STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005219 tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005220 break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005221 case VAR_BLOB:
5222 rettv->vval.v_number = blob_len(argvars[0].vval.v_blob);
5223 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005224 case VAR_LIST:
5225 rettv->vval.v_number = list_len(argvars[0].vval.v_list);
5226 break;
5227 case VAR_DICT:
5228 rettv->vval.v_number = dict_len(argvars[0].vval.v_dict);
5229 break;
5230 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02005231 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01005232 case VAR_VOID:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01005233 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005234 case VAR_SPECIAL:
5235 case VAR_FLOAT:
5236 case VAR_FUNC:
5237 case VAR_PARTIAL:
5238 case VAR_JOB:
5239 case VAR_CHANNEL:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005240 emsg(_("E701: Invalid type for len()"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005241 break;
5242 }
5243}
5244
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005245 static void
Bram Moolenaar6d721c72017-01-17 16:56:28 +01005246libcall_common(typval_T *argvars UNUSED, typval_T *rettv, int type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005247{
5248#ifdef FEAT_LIBCALL
5249 char_u *string_in;
5250 char_u **string_result;
5251 int nr_result;
5252#endif
5253
5254 rettv->v_type = type;
5255 if (type != VAR_NUMBER)
5256 rettv->vval.v_string = NULL;
5257
5258 if (check_restricted() || check_secure())
5259 return;
5260
5261#ifdef FEAT_LIBCALL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005262 // The first two args must be strings, otherwise it's meaningless
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005263 if (argvars[0].v_type == VAR_STRING && argvars[1].v_type == VAR_STRING)
5264 {
5265 string_in = NULL;
5266 if (argvars[2].v_type == VAR_STRING)
5267 string_in = argvars[2].vval.v_string;
5268 if (type == VAR_NUMBER)
5269 string_result = NULL;
5270 else
5271 string_result = &rettv->vval.v_string;
5272 if (mch_libcall(argvars[0].vval.v_string,
5273 argvars[1].vval.v_string,
5274 string_in,
5275 argvars[2].vval.v_number,
5276 string_result,
5277 &nr_result) == OK
5278 && type == VAR_NUMBER)
5279 rettv->vval.v_number = nr_result;
5280 }
5281#endif
5282}
5283
5284/*
5285 * "libcall()" function
5286 */
5287 static void
5288f_libcall(typval_T *argvars, typval_T *rettv)
5289{
5290 libcall_common(argvars, rettv, VAR_STRING);
5291}
5292
5293/*
5294 * "libcallnr()" function
5295 */
5296 static void
5297f_libcallnr(typval_T *argvars, typval_T *rettv)
5298{
5299 libcall_common(argvars, rettv, VAR_NUMBER);
5300}
5301
5302/*
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005303 * "line(string, [winid])" function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005304 */
5305 static void
5306f_line(typval_T *argvars, typval_T *rettv)
5307{
5308 linenr_T lnum = 0;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005309 pos_T *fp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005310 int fnum;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005311 int id;
5312 tabpage_T *tp;
5313 win_T *wp;
5314 win_T *save_curwin;
5315 tabpage_T *save_curtab;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005316
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005317 if (argvars[1].v_type != VAR_UNKNOWN)
5318 {
5319 // use window specified in the second argument
5320 id = (int)tv_get_number(&argvars[1]);
5321 wp = win_id2wp_tp(id, &tp);
5322 if (wp != NULL && tp != NULL)
5323 {
5324 if (switch_win_noblock(&save_curwin, &save_curtab, wp, tp, TRUE)
5325 == OK)
5326 {
5327 check_cursor();
5328 fp = var2fpos(&argvars[0], TRUE, &fnum);
5329 }
5330 restore_win_noblock(save_curwin, save_curtab, TRUE);
5331 }
5332 }
5333 else
5334 // use current window
5335 fp = var2fpos(&argvars[0], TRUE, &fnum);
5336
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005337 if (fp != NULL)
5338 lnum = fp->lnum;
5339 rettv->vval.v_number = lnum;
5340}
5341
5342/*
5343 * "line2byte(lnum)" function
5344 */
5345 static void
5346f_line2byte(typval_T *argvars UNUSED, typval_T *rettv)
5347{
5348#ifndef FEAT_BYTEOFF
5349 rettv->vval.v_number = -1;
5350#else
5351 linenr_T lnum;
5352
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005353 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005354 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count + 1)
5355 rettv->vval.v_number = -1;
5356 else
5357 rettv->vval.v_number = ml_find_line_or_offset(curbuf, lnum, NULL);
5358 if (rettv->vval.v_number >= 0)
5359 ++rettv->vval.v_number;
5360#endif
5361}
5362
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005363#ifdef FEAT_FLOAT
5364/*
5365 * "log()" function
5366 */
5367 static void
5368f_log(typval_T *argvars, typval_T *rettv)
5369{
5370 float_T f = 0.0;
5371
5372 rettv->v_type = VAR_FLOAT;
5373 if (get_float_arg(argvars, &f) == OK)
5374 rettv->vval.v_float = log(f);
5375 else
5376 rettv->vval.v_float = 0.0;
5377}
5378
5379/*
5380 * "log10()" function
5381 */
5382 static void
5383f_log10(typval_T *argvars, typval_T *rettv)
5384{
5385 float_T f = 0.0;
5386
5387 rettv->v_type = VAR_FLOAT;
5388 if (get_float_arg(argvars, &f) == OK)
5389 rettv->vval.v_float = log10(f);
5390 else
5391 rettv->vval.v_float = 0.0;
5392}
5393#endif
5394
5395#ifdef FEAT_LUA
5396/*
5397 * "luaeval()" function
5398 */
5399 static void
5400f_luaeval(typval_T *argvars, typval_T *rettv)
5401{
5402 char_u *str;
5403 char_u buf[NUMBUFLEN];
5404
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005405 if (check_restricted() || check_secure())
5406 return;
5407
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005408 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005409 do_luaeval(str, argvars + 1, rettv);
5410}
5411#endif
5412
5413/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005414 * "maparg()" function
5415 */
5416 static void
5417f_maparg(typval_T *argvars, typval_T *rettv)
5418{
5419 get_maparg(argvars, rettv, TRUE);
5420}
5421
5422/*
5423 * "mapcheck()" function
5424 */
5425 static void
5426f_mapcheck(typval_T *argvars, typval_T *rettv)
5427{
5428 get_maparg(argvars, rettv, FALSE);
5429}
5430
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005431typedef enum
5432{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005433 MATCH_END, // matchend()
5434 MATCH_MATCH, // match()
5435 MATCH_STR, // matchstr()
5436 MATCH_LIST, // matchlist()
5437 MATCH_POS // matchstrpos()
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005438} matchtype_T;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005439
5440 static void
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005441find_some_match(typval_T *argvars, typval_T *rettv, matchtype_T type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005442{
5443 char_u *str = NULL;
5444 long len = 0;
5445 char_u *expr = NULL;
5446 char_u *pat;
5447 regmatch_T regmatch;
5448 char_u patbuf[NUMBUFLEN];
5449 char_u strbuf[NUMBUFLEN];
5450 char_u *save_cpo;
5451 long start = 0;
5452 long nth = 1;
5453 colnr_T startcol = 0;
5454 int match = 0;
5455 list_T *l = NULL;
5456 listitem_T *li = NULL;
5457 long idx = 0;
5458 char_u *tofree = NULL;
5459
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005460 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005461 save_cpo = p_cpo;
5462 p_cpo = (char_u *)"";
5463
5464 rettv->vval.v_number = -1;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005465 if (type == MATCH_LIST || type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005466 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005467 // type MATCH_LIST: return empty list when there are no matches.
5468 // type MATCH_POS: return ["", -1, -1, -1]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005469 if (rettv_list_alloc(rettv) == FAIL)
5470 goto theend;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005471 if (type == MATCH_POS
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005472 && (list_append_string(rettv->vval.v_list,
5473 (char_u *)"", 0) == FAIL
5474 || list_append_number(rettv->vval.v_list,
5475 (varnumber_T)-1) == FAIL
5476 || list_append_number(rettv->vval.v_list,
5477 (varnumber_T)-1) == FAIL
5478 || list_append_number(rettv->vval.v_list,
5479 (varnumber_T)-1) == FAIL))
5480 {
5481 list_free(rettv->vval.v_list);
5482 rettv->vval.v_list = NULL;
5483 goto theend;
5484 }
5485 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005486 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005487 {
5488 rettv->v_type = VAR_STRING;
5489 rettv->vval.v_string = NULL;
5490 }
5491
5492 if (argvars[0].v_type == VAR_LIST)
5493 {
5494 if ((l = argvars[0].vval.v_list) == NULL)
5495 goto theend;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005496 range_list_materialize(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005497 li = l->lv_first;
5498 }
5499 else
5500 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005501 expr = str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005502 len = (long)STRLEN(str);
5503 }
5504
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005505 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005506 if (pat == NULL)
5507 goto theend;
5508
5509 if (argvars[2].v_type != VAR_UNKNOWN)
5510 {
5511 int error = FALSE;
5512
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005513 start = (long)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005514 if (error)
5515 goto theend;
5516 if (l != NULL)
5517 {
5518 li = list_find(l, start);
5519 if (li == NULL)
5520 goto theend;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01005521 idx = l->lv_u.mat.lv_idx; // use the cached index
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005522 }
5523 else
5524 {
5525 if (start < 0)
5526 start = 0;
5527 if (start > len)
5528 goto theend;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005529 // When "count" argument is there ignore matches before "start",
5530 // otherwise skip part of the string. Differs when pattern is "^"
5531 // or "\<".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005532 if (argvars[3].v_type != VAR_UNKNOWN)
5533 startcol = start;
5534 else
5535 {
5536 str += start;
5537 len -= start;
5538 }
5539 }
5540
5541 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005542 nth = (long)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005543 if (error)
5544 goto theend;
5545 }
5546
5547 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
5548 if (regmatch.regprog != NULL)
5549 {
5550 regmatch.rm_ic = p_ic;
5551
5552 for (;;)
5553 {
5554 if (l != NULL)
5555 {
5556 if (li == NULL)
5557 {
5558 match = FALSE;
5559 break;
5560 }
5561 vim_free(tofree);
5562 expr = str = echo_string(&li->li_tv, &tofree, strbuf, 0);
5563 if (str == NULL)
5564 break;
5565 }
5566
5567 match = vim_regexec_nl(&regmatch, str, (colnr_T)startcol);
5568
5569 if (match && --nth <= 0)
5570 break;
5571 if (l == NULL && !match)
5572 break;
5573
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005574 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005575 if (l != NULL)
5576 {
5577 li = li->li_next;
5578 ++idx;
5579 }
5580 else
5581 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005582 startcol = (colnr_T)(regmatch.startp[0]
5583 + (*mb_ptr2len)(regmatch.startp[0]) - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005584 if (startcol > (colnr_T)len
5585 || str + startcol <= regmatch.startp[0])
5586 {
5587 match = FALSE;
5588 break;
5589 }
5590 }
5591 }
5592
5593 if (match)
5594 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005595 if (type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005596 {
5597 listitem_T *li1 = rettv->vval.v_list->lv_first;
5598 listitem_T *li2 = li1->li_next;
5599 listitem_T *li3 = li2->li_next;
5600 listitem_T *li4 = li3->li_next;
5601
5602 vim_free(li1->li_tv.vval.v_string);
5603 li1->li_tv.vval.v_string = vim_strnsave(regmatch.startp[0],
5604 (int)(regmatch.endp[0] - regmatch.startp[0]));
5605 li3->li_tv.vval.v_number =
5606 (varnumber_T)(regmatch.startp[0] - expr);
5607 li4->li_tv.vval.v_number =
5608 (varnumber_T)(regmatch.endp[0] - expr);
5609 if (l != NULL)
5610 li2->li_tv.vval.v_number = (varnumber_T)idx;
5611 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005612 else if (type == MATCH_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005613 {
5614 int i;
5615
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005616 // return list with matched string and submatches
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005617 for (i = 0; i < NSUBEXP; ++i)
5618 {
5619 if (regmatch.endp[i] == NULL)
5620 {
5621 if (list_append_string(rettv->vval.v_list,
5622 (char_u *)"", 0) == FAIL)
5623 break;
5624 }
5625 else if (list_append_string(rettv->vval.v_list,
5626 regmatch.startp[i],
5627 (int)(regmatch.endp[i] - regmatch.startp[i]))
5628 == FAIL)
5629 break;
5630 }
5631 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005632 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005633 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005634 // return matched string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005635 if (l != NULL)
5636 copy_tv(&li->li_tv, rettv);
5637 else
5638 rettv->vval.v_string = vim_strnsave(regmatch.startp[0],
5639 (int)(regmatch.endp[0] - regmatch.startp[0]));
5640 }
5641 else if (l != NULL)
5642 rettv->vval.v_number = idx;
5643 else
5644 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005645 if (type != MATCH_END)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005646 rettv->vval.v_number =
5647 (varnumber_T)(regmatch.startp[0] - str);
5648 else
5649 rettv->vval.v_number =
5650 (varnumber_T)(regmatch.endp[0] - str);
5651 rettv->vval.v_number += (varnumber_T)(str - expr);
5652 }
5653 }
5654 vim_regfree(regmatch.regprog);
5655 }
5656
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005657theend:
5658 if (type == MATCH_POS && l == NULL && rettv->vval.v_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005659 // matchstrpos() without a list: drop the second item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005660 listitem_remove(rettv->vval.v_list,
5661 rettv->vval.v_list->lv_first->li_next);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005662 vim_free(tofree);
5663 p_cpo = save_cpo;
5664}
5665
5666/*
5667 * "match()" function
5668 */
5669 static void
5670f_match(typval_T *argvars, typval_T *rettv)
5671{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005672 find_some_match(argvars, rettv, MATCH_MATCH);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005673}
5674
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005675/*
5676 * "matchend()" function
5677 */
5678 static void
5679f_matchend(typval_T *argvars, typval_T *rettv)
5680{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005681 find_some_match(argvars, rettv, MATCH_END);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005682}
5683
5684/*
5685 * "matchlist()" function
5686 */
5687 static void
5688f_matchlist(typval_T *argvars, typval_T *rettv)
5689{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005690 find_some_match(argvars, rettv, MATCH_LIST);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005691}
5692
5693/*
5694 * "matchstr()" function
5695 */
5696 static void
5697f_matchstr(typval_T *argvars, typval_T *rettv)
5698{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005699 find_some_match(argvars, rettv, MATCH_STR);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005700}
5701
5702/*
5703 * "matchstrpos()" function
5704 */
5705 static void
5706f_matchstrpos(typval_T *argvars, typval_T *rettv)
5707{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005708 find_some_match(argvars, rettv, MATCH_POS);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005709}
5710
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005711 static void
5712max_min(typval_T *argvars, typval_T *rettv, int domax)
5713{
5714 varnumber_T n = 0;
5715 varnumber_T i;
5716 int error = FALSE;
5717
5718 if (argvars[0].v_type == VAR_LIST)
5719 {
5720 list_T *l;
5721 listitem_T *li;
5722
5723 l = argvars[0].vval.v_list;
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005724 if (l != NULL && l->lv_len > 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005725 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005726 if (l->lv_first == &range_list_item)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005727 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005728 if ((l->lv_u.nonmat.lv_stride > 0) ^ domax)
5729 n = l->lv_u.nonmat.lv_start;
5730 else
5731 n = l->lv_u.nonmat.lv_start + (l->lv_len - 1)
5732 * l->lv_u.nonmat.lv_stride;
5733 }
5734 else
5735 {
5736 li = l->lv_first;
5737 if (li != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005738 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005739 n = tv_get_number_chk(&li->li_tv, &error);
5740 for (;;)
5741 {
5742 li = li->li_next;
5743 if (li == NULL)
5744 break;
5745 i = tv_get_number_chk(&li->li_tv, &error);
5746 if (domax ? i > n : i < n)
5747 n = i;
5748 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005749 }
5750 }
5751 }
5752 }
5753 else if (argvars[0].v_type == VAR_DICT)
5754 {
5755 dict_T *d;
5756 int first = TRUE;
5757 hashitem_T *hi;
5758 int todo;
5759
5760 d = argvars[0].vval.v_dict;
5761 if (d != NULL)
5762 {
5763 todo = (int)d->dv_hashtab.ht_used;
5764 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
5765 {
5766 if (!HASHITEM_EMPTY(hi))
5767 {
5768 --todo;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005769 i = tv_get_number_chk(&HI2DI(hi)->di_tv, &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005770 if (first)
5771 {
5772 n = i;
5773 first = FALSE;
5774 }
5775 else if (domax ? i > n : i < n)
5776 n = i;
5777 }
5778 }
5779 }
5780 }
5781 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005782 semsg(_(e_listdictarg), domax ? "max()" : "min()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005783 rettv->vval.v_number = error ? 0 : n;
5784}
5785
5786/*
5787 * "max()" function
5788 */
5789 static void
5790f_max(typval_T *argvars, typval_T *rettv)
5791{
5792 max_min(argvars, rettv, TRUE);
5793}
5794
5795/*
5796 * "min()" function
5797 */
5798 static void
5799f_min(typval_T *argvars, typval_T *rettv)
5800{
5801 max_min(argvars, rettv, FALSE);
5802}
5803
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005804#if defined(FEAT_MZSCHEME) || defined(PROTO)
5805/*
5806 * "mzeval()" function
5807 */
5808 static void
5809f_mzeval(typval_T *argvars, typval_T *rettv)
5810{
5811 char_u *str;
5812 char_u buf[NUMBUFLEN];
5813
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005814 if (check_restricted() || check_secure())
5815 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005816 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005817 do_mzeval(str, rettv);
5818}
5819
5820 void
5821mzscheme_call_vim(char_u *name, typval_T *args, typval_T *rettv)
5822{
5823 typval_T argvars[3];
5824
5825 argvars[0].v_type = VAR_STRING;
5826 argvars[0].vval.v_string = name;
5827 copy_tv(args, &argvars[1]);
5828 argvars[2].v_type = VAR_UNKNOWN;
5829 f_call(argvars, rettv);
5830 clear_tv(&argvars[1]);
5831}
5832#endif
5833
5834/*
5835 * "nextnonblank()" function
5836 */
5837 static void
5838f_nextnonblank(typval_T *argvars, typval_T *rettv)
5839{
5840 linenr_T lnum;
5841
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005842 for (lnum = tv_get_lnum(argvars); ; ++lnum)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005843 {
5844 if (lnum < 0 || lnum > curbuf->b_ml.ml_line_count)
5845 {
5846 lnum = 0;
5847 break;
5848 }
5849 if (*skipwhite(ml_get(lnum)) != NUL)
5850 break;
5851 }
5852 rettv->vval.v_number = lnum;
5853}
5854
5855/*
5856 * "nr2char()" function
5857 */
5858 static void
5859f_nr2char(typval_T *argvars, typval_T *rettv)
5860{
5861 char_u buf[NUMBUFLEN];
5862
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005863 if (has_mbyte)
5864 {
5865 int utf8 = 0;
5866
5867 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005868 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005869 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01005870 buf[utf_char2bytes((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005871 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005872 buf[(*mb_char2bytes)((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005873 }
5874 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005875 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005876 buf[0] = (char_u)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005877 buf[1] = NUL;
5878 }
5879 rettv->v_type = VAR_STRING;
5880 rettv->vval.v_string = vim_strsave(buf);
5881}
5882
5883/*
5884 * "or(expr, expr)" function
5885 */
5886 static void
5887f_or(typval_T *argvars, typval_T *rettv)
5888{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005889 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
5890 | tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005891}
5892
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005893#ifdef FEAT_PERL
5894/*
5895 * "perleval()" function
5896 */
5897 static void
5898f_perleval(typval_T *argvars, typval_T *rettv)
5899{
5900 char_u *str;
5901 char_u buf[NUMBUFLEN];
5902
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005903 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005904 do_perleval(str, rettv);
5905}
5906#endif
5907
5908#ifdef FEAT_FLOAT
5909/*
5910 * "pow()" function
5911 */
5912 static void
5913f_pow(typval_T *argvars, typval_T *rettv)
5914{
5915 float_T fx = 0.0, fy = 0.0;
5916
5917 rettv->v_type = VAR_FLOAT;
5918 if (get_float_arg(argvars, &fx) == OK
5919 && get_float_arg(&argvars[1], &fy) == OK)
5920 rettv->vval.v_float = pow(fx, fy);
5921 else
5922 rettv->vval.v_float = 0.0;
5923}
5924#endif
5925
5926/*
5927 * "prevnonblank()" function
5928 */
5929 static void
5930f_prevnonblank(typval_T *argvars, typval_T *rettv)
5931{
5932 linenr_T lnum;
5933
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005934 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005935 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count)
5936 lnum = 0;
5937 else
5938 while (lnum >= 1 && *skipwhite(ml_get(lnum)) == NUL)
5939 --lnum;
5940 rettv->vval.v_number = lnum;
5941}
5942
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005943// This dummy va_list is here because:
5944// - passing a NULL pointer doesn't work when va_list isn't a pointer
5945// - locally in the function results in a "used before set" warning
5946// - using va_start() to initialize it gives "function with fixed args" error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005947static va_list ap;
5948
5949/*
5950 * "printf()" function
5951 */
5952 static void
5953f_printf(typval_T *argvars, typval_T *rettv)
5954{
5955 char_u buf[NUMBUFLEN];
5956 int len;
5957 char_u *s;
5958 int saved_did_emsg = did_emsg;
5959 char *fmt;
5960
5961 rettv->v_type = VAR_STRING;
5962 rettv->vval.v_string = NULL;
5963
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005964 // Get the required length, allocate the buffer and do it for real.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005965 did_emsg = FALSE;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005966 fmt = (char *)tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02005967 len = vim_vsnprintf_typval(NULL, 0, fmt, ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005968 if (!did_emsg)
5969 {
5970 s = alloc(len + 1);
5971 if (s != NULL)
5972 {
5973 rettv->vval.v_string = s;
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02005974 (void)vim_vsnprintf_typval((char *)s, len + 1, fmt,
5975 ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005976 }
5977 }
5978 did_emsg |= saved_did_emsg;
5979}
5980
5981/*
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005982 * "pum_getpos()" function
5983 */
5984 static void
5985f_pum_getpos(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5986{
5987 if (rettv_dict_alloc(rettv) != OK)
5988 return;
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005989 pum_set_event_info(rettv->vval.v_dict);
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005990}
5991
5992/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005993 * "pumvisible()" function
5994 */
5995 static void
5996f_pumvisible(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5997{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005998 if (pum_visible())
5999 rettv->vval.v_number = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006000}
6001
6002#ifdef FEAT_PYTHON3
6003/*
6004 * "py3eval()" function
6005 */
6006 static void
6007f_py3eval(typval_T *argvars, typval_T *rettv)
6008{
6009 char_u *str;
6010 char_u buf[NUMBUFLEN];
6011
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006012 if (check_restricted() || check_secure())
6013 return;
6014
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006015 if (p_pyx == 0)
6016 p_pyx = 3;
6017
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006018 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006019 do_py3eval(str, rettv);
6020}
6021#endif
6022
6023#ifdef FEAT_PYTHON
6024/*
6025 * "pyeval()" function
6026 */
6027 static void
6028f_pyeval(typval_T *argvars, typval_T *rettv)
6029{
6030 char_u *str;
6031 char_u buf[NUMBUFLEN];
6032
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006033 if (check_restricted() || check_secure())
6034 return;
6035
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006036 if (p_pyx == 0)
6037 p_pyx = 2;
6038
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006039 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006040 do_pyeval(str, rettv);
6041}
6042#endif
6043
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006044#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
6045/*
6046 * "pyxeval()" function
6047 */
6048 static void
6049f_pyxeval(typval_T *argvars, typval_T *rettv)
6050{
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006051 if (check_restricted() || check_secure())
6052 return;
6053
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006054# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
6055 init_pyxversion();
6056 if (p_pyx == 2)
6057 f_pyeval(argvars, rettv);
6058 else
6059 f_py3eval(argvars, rettv);
6060# elif defined(FEAT_PYTHON)
6061 f_pyeval(argvars, rettv);
6062# elif defined(FEAT_PYTHON3)
6063 f_py3eval(argvars, rettv);
6064# endif
6065}
6066#endif
6067
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006068static UINT32_T srand_seed_for_testing = 0;
6069static int srand_seed_for_testing_is_used = FALSE;
6070
6071 static void
6072f_test_srand_seed(typval_T *argvars, typval_T *rettv UNUSED)
6073{
6074 if (argvars[0].v_type == VAR_UNKNOWN)
6075 srand_seed_for_testing_is_used = FALSE;
6076 else
6077 {
6078 srand_seed_for_testing = (UINT32_T)tv_get_number(&argvars[0]);
6079 srand_seed_for_testing_is_used = TRUE;
6080 }
6081}
6082
6083 static void
6084init_srand(UINT32_T *x)
6085{
6086#ifndef MSWIN
6087 static int dev_urandom_state = NOTDONE; // FAIL or OK once tried
6088#endif
6089
6090 if (srand_seed_for_testing_is_used)
6091 {
6092 *x = srand_seed_for_testing;
6093 return;
6094 }
6095#ifndef MSWIN
6096 if (dev_urandom_state != FAIL)
6097 {
6098 int fd = open("/dev/urandom", O_RDONLY);
6099 struct {
6100 union {
6101 UINT32_T number;
6102 char bytes[sizeof(UINT32_T)];
6103 } contents;
6104 } buf;
6105
6106 // Attempt reading /dev/urandom.
6107 if (fd == -1)
6108 dev_urandom_state = FAIL;
6109 else
6110 {
6111 buf.contents.number = 0;
6112 if (read(fd, buf.contents.bytes, sizeof(UINT32_T))
6113 != sizeof(UINT32_T))
6114 dev_urandom_state = FAIL;
6115 else
6116 {
6117 dev_urandom_state = OK;
6118 *x = buf.contents.number;
6119 }
6120 close(fd);
6121 }
6122 }
6123 if (dev_urandom_state != OK)
6124 // Reading /dev/urandom doesn't work, fall back to time().
6125#endif
6126 *x = vim_time();
6127}
6128
6129#define ROTL(x, k) ((x << k) | (x >> (32 - k)))
6130#define SPLITMIX32(x, z) ( \
6131 z = (x += 0x9e3779b9), \
6132 z = (z ^ (z >> 16)) * 0x85ebca6b, \
6133 z = (z ^ (z >> 13)) * 0xc2b2ae35, \
6134 z ^ (z >> 16) \
6135 )
6136#define SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w) \
6137 result = ROTL(y * 5, 7) * 9; \
6138 t = y << 9; \
6139 z ^= x; \
6140 w ^= y; \
6141 y ^= z, x ^= w; \
6142 z ^= t; \
6143 w = ROTL(w, 11);
6144
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006145/*
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006146 * "rand()" function
6147 */
6148 static void
6149f_rand(typval_T *argvars, typval_T *rettv)
6150{
6151 list_T *l = NULL;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006152 static UINT32_T gx, gy, gz, gw;
6153 static int initialized = FALSE;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006154 listitem_T *lx, *ly, *lz, *lw;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006155 UINT32_T x, y, z, w, t, result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006156
6157 if (argvars[0].v_type == VAR_UNKNOWN)
6158 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006159 // When no argument is given use the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006160 if (initialized == FALSE)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006161 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006162 // Initialize the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006163 init_srand(&x);
6164
6165 gx = SPLITMIX32(x, z);
6166 gy = SPLITMIX32(x, z);
6167 gz = SPLITMIX32(x, z);
6168 gw = SPLITMIX32(x, z);
6169 initialized = TRUE;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006170 }
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006171
6172 SHUFFLE_XOSHIRO128STARSTAR(gx, gy, gz, gw);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006173 }
6174 else if (argvars[0].v_type == VAR_LIST)
6175 {
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006176 l = argvars[0].vval.v_list;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006177 if (l == NULL || list_len(l) != 4)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006178 goto theend;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006179
6180 lx = list_find(l, 0L);
6181 ly = list_find(l, 1L);
6182 lz = list_find(l, 2L);
6183 lw = list_find(l, 3L);
6184 if (lx->li_tv.v_type != VAR_NUMBER) goto theend;
6185 if (ly->li_tv.v_type != VAR_NUMBER) goto theend;
6186 if (lz->li_tv.v_type != VAR_NUMBER) goto theend;
6187 if (lw->li_tv.v_type != VAR_NUMBER) goto theend;
6188 x = (UINT32_T)lx->li_tv.vval.v_number;
6189 y = (UINT32_T)ly->li_tv.vval.v_number;
6190 z = (UINT32_T)lz->li_tv.vval.v_number;
6191 w = (UINT32_T)lw->li_tv.vval.v_number;
6192
6193 SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w);
6194
6195 lx->li_tv.vval.v_number = (varnumber_T)x;
6196 ly->li_tv.vval.v_number = (varnumber_T)y;
6197 lz->li_tv.vval.v_number = (varnumber_T)z;
6198 lw->li_tv.vval.v_number = (varnumber_T)w;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006199 }
6200 else
6201 goto theend;
6202
6203 rettv->v_type = VAR_NUMBER;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006204 rettv->vval.v_number = (varnumber_T)result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006205 return;
6206
6207theend:
6208 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006209 rettv->v_type = VAR_NUMBER;
6210 rettv->vval.v_number = -1;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006211}
6212
6213/*
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006214 * "srand()" function
6215 */
6216 static void
6217f_srand(typval_T *argvars, typval_T *rettv)
6218{
6219 UINT32_T x = 0, z;
6220
6221 if (rettv_list_alloc(rettv) == FAIL)
6222 return;
6223 if (argvars[0].v_type == VAR_UNKNOWN)
6224 {
6225 init_srand(&x);
6226 }
6227 else
6228 {
6229 int error = FALSE;
6230
6231 x = (UINT32_T)tv_get_number_chk(&argvars[0], &error);
6232 if (error)
6233 return;
6234 }
6235
6236 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6237 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6238 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6239 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6240}
6241
6242#undef ROTL
6243#undef SPLITMIX32
6244#undef SHUFFLE_XOSHIRO128STARSTAR
6245
6246/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006247 * "range()" function
6248 */
6249 static void
6250f_range(typval_T *argvars, typval_T *rettv)
6251{
6252 varnumber_T start;
6253 varnumber_T end;
6254 varnumber_T stride = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006255 int error = FALSE;
6256
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006257 start = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006258 if (argvars[1].v_type == VAR_UNKNOWN)
6259 {
6260 end = start - 1;
6261 start = 0;
6262 }
6263 else
6264 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006265 end = tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006266 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006267 stride = tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006268 }
6269
6270 if (error)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006271 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006272 if (stride == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006273 emsg(_("E726: Stride is zero"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006274 else if (stride > 0 ? end + 1 < start : end - 1 > start)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006275 emsg(_("E727: Start past end"));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006276 else if (rettv_list_alloc(rettv) == OK)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006277 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006278 list_T *list = rettv->vval.v_list;
6279
6280 // Create a non-materialized list. This is much more efficient and
6281 // works with ":for". If used otherwise range_list_materialize() must
6282 // be called.
6283 list->lv_first = &range_list_item;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006284 list->lv_u.nonmat.lv_start = start;
6285 list->lv_u.nonmat.lv_end = end;
6286 list->lv_u.nonmat.lv_stride = stride;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01006287 list->lv_len = (end - start) / stride + 1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006288 }
6289}
6290
6291/*
6292 * If "list" is a non-materialized list then materialize it now.
6293 */
6294 void
6295range_list_materialize(list_T *list)
6296{
6297 if (list->lv_first == &range_list_item)
6298 {
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006299 varnumber_T start = list->lv_u.nonmat.lv_start;
6300 varnumber_T end = list->lv_u.nonmat.lv_end;
6301 int stride = list->lv_u.nonmat.lv_stride;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006302 varnumber_T i;
6303
6304 list->lv_first = NULL;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006305 list->lv_u.mat.lv_last = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006306 list->lv_len = 0;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006307 list->lv_u.mat.lv_idx_item = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006308 for (i = start; stride > 0 ? i <= end : i >= end; i += stride)
6309 if (list_append_number(list, (varnumber_T)i) == FAIL)
6310 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006311 }
6312}
6313
Bram Moolenaar0b6d9112018-05-22 20:35:17 +02006314 static void
6315return_register(int regname, typval_T *rettv)
6316{
6317 char_u buf[2] = {0, 0};
6318
6319 buf[0] = (char_u)regname;
6320 rettv->v_type = VAR_STRING;
6321 rettv->vval.v_string = vim_strsave(buf);
6322}
6323
6324/*
6325 * "reg_executing()" function
6326 */
6327 static void
6328f_reg_executing(typval_T *argvars UNUSED, typval_T *rettv)
6329{
6330 return_register(reg_executing, rettv);
6331}
6332
6333/*
6334 * "reg_recording()" function
6335 */
6336 static void
6337f_reg_recording(typval_T *argvars UNUSED, typval_T *rettv)
6338{
6339 return_register(reg_recording, rettv);
6340}
6341
Bram Moolenaar7416f3e2017-03-18 18:10:13 +01006342/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006343 * "rename({from}, {to})" function
6344 */
6345 static void
6346f_rename(typval_T *argvars, typval_T *rettv)
6347{
6348 char_u buf[NUMBUFLEN];
6349
6350 if (check_restricted() || check_secure())
6351 rettv->vval.v_number = -1;
6352 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006353 rettv->vval.v_number = vim_rename(tv_get_string(&argvars[0]),
6354 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006355}
6356
6357/*
6358 * "repeat()" function
6359 */
6360 static void
6361f_repeat(typval_T *argvars, typval_T *rettv)
6362{
6363 char_u *p;
6364 int n;
6365 int slen;
6366 int len;
6367 char_u *r;
6368 int i;
6369
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006370 n = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006371 if (argvars[0].v_type == VAR_LIST)
6372 {
6373 if (rettv_list_alloc(rettv) == OK && argvars[0].vval.v_list != NULL)
6374 while (n-- > 0)
6375 if (list_extend(rettv->vval.v_list,
6376 argvars[0].vval.v_list, NULL) == FAIL)
6377 break;
6378 }
6379 else
6380 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006381 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006382 rettv->v_type = VAR_STRING;
6383 rettv->vval.v_string = NULL;
6384
6385 slen = (int)STRLEN(p);
6386 len = slen * n;
6387 if (len <= 0)
6388 return;
6389
6390 r = alloc(len + 1);
6391 if (r != NULL)
6392 {
6393 for (i = 0; i < n; i++)
6394 mch_memmove(r + i * slen, p, (size_t)slen);
6395 r[len] = NUL;
6396 }
6397
6398 rettv->vval.v_string = r;
6399 }
6400}
6401
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006402#define SP_NOMOVE 0x01 // don't move cursor
6403#define SP_REPEAT 0x02 // repeat to find outer pair
6404#define SP_RETCOUNT 0x04 // return matchcount
6405#define SP_SETPCMARK 0x08 // set previous context mark
6406#define SP_START 0x10 // accept match at start position
6407#define SP_SUBPAT 0x20 // return nr of matching sub-pattern
6408#define SP_END 0x40 // leave cursor at end of match
6409#define SP_COLUMN 0x80 // start at cursor column
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006410
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006411/*
6412 * Get flags for a search function.
6413 * Possibly sets "p_ws".
6414 * Returns BACKWARD, FORWARD or zero (for an error).
6415 */
6416 static int
6417get_search_arg(typval_T *varp, int *flagsp)
6418{
6419 int dir = FORWARD;
6420 char_u *flags;
6421 char_u nbuf[NUMBUFLEN];
6422 int mask;
6423
6424 if (varp->v_type != VAR_UNKNOWN)
6425 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006426 flags = tv_get_string_buf_chk(varp, nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006427 if (flags == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006428 return 0; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006429 while (*flags != NUL)
6430 {
6431 switch (*flags)
6432 {
6433 case 'b': dir = BACKWARD; break;
6434 case 'w': p_ws = TRUE; break;
6435 case 'W': p_ws = FALSE; break;
6436 default: mask = 0;
6437 if (flagsp != NULL)
6438 switch (*flags)
6439 {
6440 case 'c': mask = SP_START; break;
6441 case 'e': mask = SP_END; break;
6442 case 'm': mask = SP_RETCOUNT; break;
6443 case 'n': mask = SP_NOMOVE; break;
6444 case 'p': mask = SP_SUBPAT; break;
6445 case 'r': mask = SP_REPEAT; break;
6446 case 's': mask = SP_SETPCMARK; break;
6447 case 'z': mask = SP_COLUMN; break;
6448 }
6449 if (mask == 0)
6450 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006451 semsg(_(e_invarg2), flags);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006452 dir = 0;
6453 }
6454 else
6455 *flagsp |= mask;
6456 }
6457 if (dir == 0)
6458 break;
6459 ++flags;
6460 }
6461 }
6462 return dir;
6463}
6464
6465/*
6466 * Shared by search() and searchpos() functions.
6467 */
6468 static int
6469search_cmn(typval_T *argvars, pos_T *match_pos, int *flagsp)
6470{
6471 int flags;
6472 char_u *pat;
6473 pos_T pos;
6474 pos_T save_cursor;
6475 int save_p_ws = p_ws;
6476 int dir;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006477 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006478 long lnum_stop = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006479#ifdef FEAT_RELTIME
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006480 proftime_T tm;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006481 long time_limit = 0;
6482#endif
6483 int options = SEARCH_KEEP;
6484 int subpatnum;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006485 searchit_arg_T sia;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006486
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006487 pat = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006488 dir = get_search_arg(&argvars[1], flagsp); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006489 if (dir == 0)
6490 goto theend;
6491 flags = *flagsp;
6492 if (flags & SP_START)
6493 options |= SEARCH_START;
6494 if (flags & SP_END)
6495 options |= SEARCH_END;
6496 if (flags & SP_COLUMN)
6497 options |= SEARCH_COL;
6498
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006499 // Optional arguments: line number to stop searching and timeout.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006500 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN)
6501 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006502 lnum_stop = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006503 if (lnum_stop < 0)
6504 goto theend;
6505#ifdef FEAT_RELTIME
6506 if (argvars[3].v_type != VAR_UNKNOWN)
6507 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006508 time_limit = (long)tv_get_number_chk(&argvars[3], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006509 if (time_limit < 0)
6510 goto theend;
6511 }
6512#endif
6513 }
6514
6515#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006516 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006517 profile_setlimit(time_limit, &tm);
6518#endif
6519
6520 /*
6521 * This function does not accept SP_REPEAT and SP_RETCOUNT flags.
6522 * Check to make sure only those flags are set.
6523 * Also, Only the SP_NOMOVE or the SP_SETPCMARK flag can be set. Both
6524 * flags cannot be set. Check for that condition also.
6525 */
6526 if (((flags & (SP_REPEAT | SP_RETCOUNT)) != 0)
6527 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
6528 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006529 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006530 goto theend;
6531 }
6532
6533 pos = save_cursor = curwin->w_cursor;
Bram Moolenaara80faa82020-04-12 19:37:17 +02006534 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006535 sia.sa_stop_lnum = (linenr_T)lnum_stop;
6536#ifdef FEAT_RELTIME
6537 sia.sa_tm = &tm;
6538#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01006539 subpatnum = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006540 options, RE_SEARCH, &sia);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006541 if (subpatnum != FAIL)
6542 {
6543 if (flags & SP_SUBPAT)
6544 retval = subpatnum;
6545 else
6546 retval = pos.lnum;
6547 if (flags & SP_SETPCMARK)
6548 setpcmark();
6549 curwin->w_cursor = pos;
6550 if (match_pos != NULL)
6551 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006552 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006553 match_pos->lnum = pos.lnum;
6554 match_pos->col = pos.col + 1;
6555 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006556 // "/$" will put the cursor after the end of the line, may need to
6557 // correct that here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006558 check_cursor();
6559 }
6560
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006561 // If 'n' flag is used: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006562 if (flags & SP_NOMOVE)
6563 curwin->w_cursor = save_cursor;
6564 else
6565 curwin->w_set_curswant = TRUE;
6566theend:
6567 p_ws = save_p_ws;
6568
6569 return retval;
6570}
6571
6572#ifdef FEAT_FLOAT
6573
6574/*
6575 * round() is not in C90, use ceil() or floor() instead.
6576 */
6577 float_T
6578vim_round(float_T f)
6579{
6580 return f > 0 ? floor(f + 0.5) : ceil(f - 0.5);
6581}
6582
6583/*
6584 * "round({float})" function
6585 */
6586 static void
6587f_round(typval_T *argvars, typval_T *rettv)
6588{
6589 float_T f = 0.0;
6590
6591 rettv->v_type = VAR_FLOAT;
6592 if (get_float_arg(argvars, &f) == OK)
6593 rettv->vval.v_float = vim_round(f);
6594 else
6595 rettv->vval.v_float = 0.0;
6596}
6597#endif
6598
Bram Moolenaare99be0e2019-03-26 22:51:09 +01006599#ifdef FEAT_RUBY
6600/*
6601 * "rubyeval()" function
6602 */
6603 static void
6604f_rubyeval(typval_T *argvars, typval_T *rettv)
6605{
6606 char_u *str;
6607 char_u buf[NUMBUFLEN];
6608
6609 str = tv_get_string_buf(&argvars[0], buf);
6610 do_rubyeval(str, rettv);
6611}
6612#endif
6613
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006614/*
6615 * "screenattr()" function
6616 */
6617 static void
6618f_screenattr(typval_T *argvars, typval_T *rettv)
6619{
6620 int row;
6621 int col;
6622 int c;
6623
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006624 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6625 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006626 if (row < 0 || row >= screen_Rows
6627 || col < 0 || col >= screen_Columns)
6628 c = -1;
6629 else
6630 c = ScreenAttrs[LineOffset[row] + col];
6631 rettv->vval.v_number = c;
6632}
6633
6634/*
6635 * "screenchar()" function
6636 */
6637 static void
6638f_screenchar(typval_T *argvars, typval_T *rettv)
6639{
6640 int row;
6641 int col;
6642 int off;
6643 int c;
6644
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006645 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6646 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006647 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006648 c = -1;
6649 else
6650 {
6651 off = LineOffset[row] + col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006652 if (enc_utf8 && ScreenLinesUC[off] != 0)
6653 c = ScreenLinesUC[off];
6654 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006655 c = ScreenLines[off];
6656 }
6657 rettv->vval.v_number = c;
6658}
6659
6660/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006661 * "screenchars()" function
6662 */
6663 static void
6664f_screenchars(typval_T *argvars, typval_T *rettv)
6665{
6666 int row;
6667 int col;
6668 int off;
6669 int c;
6670 int i;
6671
6672 if (rettv_list_alloc(rettv) == FAIL)
6673 return;
6674 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6675 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
6676 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
6677 return;
6678
6679 off = LineOffset[row] + col;
6680 if (enc_utf8 && ScreenLinesUC[off] != 0)
6681 c = ScreenLinesUC[off];
6682 else
6683 c = ScreenLines[off];
6684 list_append_number(rettv->vval.v_list, (varnumber_T)c);
6685
6686 if (enc_utf8)
6687
6688 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
6689 list_append_number(rettv->vval.v_list,
6690 (varnumber_T)ScreenLinesC[i][off]);
6691}
6692
6693/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006694 * "screencol()" function
6695 *
6696 * First column is 1 to be consistent with virtcol().
6697 */
6698 static void
6699f_screencol(typval_T *argvars UNUSED, typval_T *rettv)
6700{
6701 rettv->vval.v_number = screen_screencol() + 1;
6702}
6703
6704/*
6705 * "screenrow()" function
6706 */
6707 static void
6708f_screenrow(typval_T *argvars UNUSED, typval_T *rettv)
6709{
6710 rettv->vval.v_number = screen_screenrow() + 1;
6711}
6712
6713/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006714 * "screenstring()" function
6715 */
6716 static void
6717f_screenstring(typval_T *argvars, typval_T *rettv)
6718{
6719 int row;
6720 int col;
6721 int off;
6722 int c;
6723 int i;
6724 char_u buf[MB_MAXBYTES + 1];
6725 int buflen = 0;
6726
6727 rettv->vval.v_string = NULL;
6728 rettv->v_type = VAR_STRING;
6729
6730 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6731 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
6732 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
6733 return;
6734
6735 off = LineOffset[row] + col;
6736 if (enc_utf8 && ScreenLinesUC[off] != 0)
6737 c = ScreenLinesUC[off];
6738 else
6739 c = ScreenLines[off];
6740 buflen += mb_char2bytes(c, buf);
6741
6742 if (enc_utf8)
6743 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
6744 buflen += mb_char2bytes(ScreenLinesC[i][off], buf + buflen);
6745
6746 buf[buflen] = NUL;
6747 rettv->vval.v_string = vim_strsave(buf);
6748}
6749
6750/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006751 * "search()" function
6752 */
6753 static void
6754f_search(typval_T *argvars, typval_T *rettv)
6755{
6756 int flags = 0;
6757
6758 rettv->vval.v_number = search_cmn(argvars, NULL, &flags);
6759}
6760
6761/*
6762 * "searchdecl()" function
6763 */
6764 static void
6765f_searchdecl(typval_T *argvars, typval_T *rettv)
6766{
6767 int locally = 1;
6768 int thisblock = 0;
6769 int error = FALSE;
6770 char_u *name;
6771
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006772 rettv->vval.v_number = 1; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006773
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006774 name = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006775 if (argvars[1].v_type != VAR_UNKNOWN)
6776 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006777 locally = (int)tv_get_number_chk(&argvars[1], &error) == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006778 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006779 thisblock = (int)tv_get_number_chk(&argvars[2], &error) != 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006780 }
6781 if (!error && name != NULL)
6782 rettv->vval.v_number = find_decl(name, (int)STRLEN(name),
6783 locally, thisblock, SEARCH_KEEP) == FAIL;
6784}
6785
6786/*
6787 * Used by searchpair() and searchpairpos()
6788 */
6789 static int
6790searchpair_cmn(typval_T *argvars, pos_T *match_pos)
6791{
6792 char_u *spat, *mpat, *epat;
Bram Moolenaar48570482017-10-30 21:48:41 +01006793 typval_T *skip;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006794 int save_p_ws = p_ws;
6795 int dir;
6796 int flags = 0;
6797 char_u nbuf1[NUMBUFLEN];
6798 char_u nbuf2[NUMBUFLEN];
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006799 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006800 long lnum_stop = 0;
6801 long time_limit = 0;
6802
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006803 // Get the three pattern arguments: start, middle, end. Will result in an
6804 // error if not a valid argument.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006805 spat = tv_get_string_chk(&argvars[0]);
6806 mpat = tv_get_string_buf_chk(&argvars[1], nbuf1);
6807 epat = tv_get_string_buf_chk(&argvars[2], nbuf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006808 if (spat == NULL || mpat == NULL || epat == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006809 goto theend; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006810
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006811 // Handle the optional fourth argument: flags
6812 dir = get_search_arg(&argvars[3], &flags); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006813 if (dir == 0)
6814 goto theend;
6815
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006816 // Don't accept SP_END or SP_SUBPAT.
6817 // Only one of the SP_NOMOVE or SP_SETPCMARK flags can be set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006818 if ((flags & (SP_END | SP_SUBPAT)) != 0
6819 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
6820 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006821 semsg(_(e_invarg2), tv_get_string(&argvars[3]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006822 goto theend;
6823 }
6824
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006825 // Using 'r' implies 'W', otherwise it doesn't work.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006826 if (flags & SP_REPEAT)
6827 p_ws = FALSE;
6828
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006829 // Optional fifth argument: skip expression
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006830 if (argvars[3].v_type == VAR_UNKNOWN
6831 || argvars[4].v_type == VAR_UNKNOWN)
Bram Moolenaar48570482017-10-30 21:48:41 +01006832 skip = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006833 else
6834 {
Bram Moolenaar48570482017-10-30 21:48:41 +01006835 skip = &argvars[4];
6836 if (skip->v_type != VAR_FUNC && skip->v_type != VAR_PARTIAL
6837 && skip->v_type != VAR_STRING)
6838 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006839 // Type error
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006840 semsg(_(e_invarg2), tv_get_string(&argvars[4]));
Bram Moolenaar48570482017-10-30 21:48:41 +01006841 goto theend;
6842 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006843 if (argvars[5].v_type != VAR_UNKNOWN)
6844 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006845 lnum_stop = (long)tv_get_number_chk(&argvars[5], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006846 if (lnum_stop < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006847 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006848 semsg(_(e_invarg2), tv_get_string(&argvars[5]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006849 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006850 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006851#ifdef FEAT_RELTIME
6852 if (argvars[6].v_type != VAR_UNKNOWN)
6853 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006854 time_limit = (long)tv_get_number_chk(&argvars[6], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006855 if (time_limit < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006856 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006857 semsg(_(e_invarg2), tv_get_string(&argvars[6]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006858 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006859 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006860 }
6861#endif
6862 }
6863 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006864
6865 retval = do_searchpair(spat, mpat, epat, dir, skip, flags,
6866 match_pos, lnum_stop, time_limit);
6867
6868theend:
6869 p_ws = save_p_ws;
6870
6871 return retval;
6872}
6873
6874/*
6875 * "searchpair()" function
6876 */
6877 static void
6878f_searchpair(typval_T *argvars, typval_T *rettv)
6879{
6880 rettv->vval.v_number = searchpair_cmn(argvars, NULL);
6881}
6882
6883/*
6884 * "searchpairpos()" function
6885 */
6886 static void
6887f_searchpairpos(typval_T *argvars, typval_T *rettv)
6888{
6889 pos_T match_pos;
6890 int lnum = 0;
6891 int col = 0;
6892
6893 if (rettv_list_alloc(rettv) == FAIL)
6894 return;
6895
6896 if (searchpair_cmn(argvars, &match_pos) > 0)
6897 {
6898 lnum = match_pos.lnum;
6899 col = match_pos.col;
6900 }
6901
6902 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
6903 list_append_number(rettv->vval.v_list, (varnumber_T)col);
6904}
6905
6906/*
6907 * Search for a start/middle/end thing.
6908 * Used by searchpair(), see its documentation for the details.
6909 * Returns 0 or -1 for no match,
6910 */
6911 long
6912do_searchpair(
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006913 char_u *spat, // start pattern
6914 char_u *mpat, // middle pattern
6915 char_u *epat, // end pattern
6916 int dir, // BACKWARD or FORWARD
6917 typval_T *skip, // skip expression
6918 int flags, // SP_SETPCMARK and other SP_ values
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006919 pos_T *match_pos,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006920 linenr_T lnum_stop, // stop at this line if not zero
6921 long time_limit UNUSED) // stop after this many msec
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006922{
6923 char_u *save_cpo;
6924 char_u *pat, *pat2 = NULL, *pat3 = NULL;
6925 long retval = 0;
6926 pos_T pos;
6927 pos_T firstpos;
6928 pos_T foundpos;
6929 pos_T save_cursor;
6930 pos_T save_pos;
6931 int n;
6932 int r;
6933 int nest = 1;
Bram Moolenaar48570482017-10-30 21:48:41 +01006934 int use_skip = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006935 int err;
6936 int options = SEARCH_KEEP;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006937#ifdef FEAT_RELTIME
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006938 proftime_T tm;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006939#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006940
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006941 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006942 save_cpo = p_cpo;
6943 p_cpo = empty_option;
6944
6945#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006946 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006947 profile_setlimit(time_limit, &tm);
6948#endif
6949
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006950 // Make two search patterns: start/end (pat2, for in nested pairs) and
6951 // start/middle/end (pat3, for the top pair).
Bram Moolenaar964b3742019-05-24 18:54:09 +02006952 pat2 = alloc(STRLEN(spat) + STRLEN(epat) + 17);
6953 pat3 = alloc(STRLEN(spat) + STRLEN(mpat) + STRLEN(epat) + 25);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006954 if (pat2 == NULL || pat3 == NULL)
6955 goto theend;
Bram Moolenaar6e450a52017-01-06 20:03:58 +01006956 sprintf((char *)pat2, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)", spat, epat);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006957 if (*mpat == NUL)
6958 STRCPY(pat3, pat2);
6959 else
Bram Moolenaar6e450a52017-01-06 20:03:58 +01006960 sprintf((char *)pat3, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)\\|\\(%s\\m\\)",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006961 spat, epat, mpat);
6962 if (flags & SP_START)
6963 options |= SEARCH_START;
6964
Bram Moolenaar48570482017-10-30 21:48:41 +01006965 if (skip != NULL)
6966 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006967 // Empty string means to not use the skip expression.
Bram Moolenaar48570482017-10-30 21:48:41 +01006968 if (skip->v_type == VAR_STRING || skip->v_type == VAR_FUNC)
6969 use_skip = skip->vval.v_string != NULL
6970 && *skip->vval.v_string != NUL;
6971 }
6972
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006973 save_cursor = curwin->w_cursor;
6974 pos = curwin->w_cursor;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006975 CLEAR_POS(&firstpos);
6976 CLEAR_POS(&foundpos);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006977 pat = pat3;
6978 for (;;)
6979 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006980 searchit_arg_T sia;
6981
Bram Moolenaara80faa82020-04-12 19:37:17 +02006982 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006983 sia.sa_stop_lnum = lnum_stop;
6984#ifdef FEAT_RELTIME
6985 sia.sa_tm = &tm;
6986#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01006987 n = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006988 options, RE_SEARCH, &sia);
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006989 if (n == FAIL || (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos)))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006990 // didn't find it or found the first match again: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006991 break;
6992
6993 if (firstpos.lnum == 0)
6994 firstpos = pos;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006995 if (EQUAL_POS(pos, foundpos))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006996 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006997 // Found the same position again. Can happen with a pattern that
6998 // has "\zs" at the end and searching backwards. Advance one
6999 // character and try again.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007000 if (dir == BACKWARD)
7001 decl(&pos);
7002 else
7003 incl(&pos);
7004 }
7005 foundpos = pos;
7006
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007007 // clear the start flag to avoid getting stuck here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007008 options &= ~SEARCH_START;
7009
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007010 // If the skip pattern matches, ignore this match.
Bram Moolenaar48570482017-10-30 21:48:41 +01007011 if (use_skip)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007012 {
7013 save_pos = curwin->w_cursor;
7014 curwin->w_cursor = pos;
Bram Moolenaar48570482017-10-30 21:48:41 +01007015 err = FALSE;
7016 r = eval_expr_to_bool(skip, &err);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007017 curwin->w_cursor = save_pos;
7018 if (err)
7019 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007020 // Evaluating {skip} caused an error, break here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007021 curwin->w_cursor = save_cursor;
7022 retval = -1;
7023 break;
7024 }
7025 if (r)
7026 continue;
7027 }
7028
7029 if ((dir == BACKWARD && n == 3) || (dir == FORWARD && n == 2))
7030 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007031 // Found end when searching backwards or start when searching
7032 // forward: nested pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007033 ++nest;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007034 pat = pat2; // nested, don't search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007035 }
7036 else
7037 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007038 // Found end when searching forward or start when searching
7039 // backward: end of (nested) pair; or found middle in outer pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007040 if (--nest == 1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007041 pat = pat3; // outer level, search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007042 }
7043
7044 if (nest == 0)
7045 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007046 // Found the match: return matchcount or line number.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007047 if (flags & SP_RETCOUNT)
7048 ++retval;
7049 else
7050 retval = pos.lnum;
7051 if (flags & SP_SETPCMARK)
7052 setpcmark();
7053 curwin->w_cursor = pos;
7054 if (!(flags & SP_REPEAT))
7055 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007056 nest = 1; // search for next unmatched
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007057 }
7058 }
7059
7060 if (match_pos != NULL)
7061 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007062 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007063 match_pos->lnum = curwin->w_cursor.lnum;
7064 match_pos->col = curwin->w_cursor.col + 1;
7065 }
7066
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007067 // If 'n' flag is used or search failed: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007068 if ((flags & SP_NOMOVE) || retval == 0)
7069 curwin->w_cursor = save_cursor;
7070
7071theend:
7072 vim_free(pat2);
7073 vim_free(pat3);
7074 if (p_cpo == empty_option)
7075 p_cpo = save_cpo;
7076 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007077 // Darn, evaluating the {skip} expression changed the value.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007078 free_string_option(save_cpo);
7079
7080 return retval;
7081}
7082
7083/*
7084 * "searchpos()" function
7085 */
7086 static void
7087f_searchpos(typval_T *argvars, typval_T *rettv)
7088{
7089 pos_T match_pos;
7090 int lnum = 0;
7091 int col = 0;
7092 int n;
7093 int flags = 0;
7094
7095 if (rettv_list_alloc(rettv) == FAIL)
7096 return;
7097
7098 n = search_cmn(argvars, &match_pos, &flags);
7099 if (n > 0)
7100 {
7101 lnum = match_pos.lnum;
7102 col = match_pos.col;
7103 }
7104
7105 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
7106 list_append_number(rettv->vval.v_list, (varnumber_T)col);
7107 if (flags & SP_SUBPAT)
7108 list_append_number(rettv->vval.v_list, (varnumber_T)n);
7109}
7110
7111 static void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007112f_setcharsearch(typval_T *argvars, typval_T *rettv UNUSED)
7113{
7114 dict_T *d;
7115 dictitem_T *di;
7116 char_u *csearch;
7117
7118 if (argvars[0].v_type != VAR_DICT)
7119 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007120 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007121 return;
7122 }
7123
7124 if ((d = argvars[0].vval.v_dict) != NULL)
7125 {
Bram Moolenaar8f667172018-12-14 15:38:31 +01007126 csearch = dict_get_string(d, (char_u *)"char", FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007127 if (csearch != NULL)
7128 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007129 if (enc_utf8)
7130 {
7131 int pcc[MAX_MCO];
7132 int c = utfc_ptr2char(csearch, pcc);
7133
7134 set_last_csearch(c, csearch, utfc_ptr2len(csearch));
7135 }
7136 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007137 set_last_csearch(PTR2CHAR(csearch),
Bram Moolenaar1614a142019-10-06 22:00:13 +02007138 csearch, mb_ptr2len(csearch));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007139 }
7140
7141 di = dict_find(d, (char_u *)"forward", -1);
7142 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007143 set_csearch_direction((int)tv_get_number(&di->di_tv)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007144 ? FORWARD : BACKWARD);
7145
7146 di = dict_find(d, (char_u *)"until", -1);
7147 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007148 set_csearch_until(!!tv_get_number(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007149 }
7150}
7151
7152/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02007153 * "setenv()" function
7154 */
7155 static void
7156f_setenv(typval_T *argvars, typval_T *rettv UNUSED)
7157{
7158 char_u namebuf[NUMBUFLEN];
7159 char_u valbuf[NUMBUFLEN];
7160 char_u *name = tv_get_string_buf(&argvars[0], namebuf);
7161
7162 if (argvars[1].v_type == VAR_SPECIAL
7163 && argvars[1].vval.v_number == VVAL_NULL)
7164 vim_unsetenv(name);
7165 else
7166 vim_setenv(name, tv_get_string_buf(&argvars[1], valbuf));
7167}
7168
7169/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007170 * "setfperm({fname}, {mode})" function
7171 */
7172 static void
7173f_setfperm(typval_T *argvars, typval_T *rettv)
7174{
7175 char_u *fname;
7176 char_u modebuf[NUMBUFLEN];
7177 char_u *mode_str;
7178 int i;
7179 int mask;
7180 int mode = 0;
7181
7182 rettv->vval.v_number = 0;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007183 fname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007184 if (fname == NULL)
7185 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007186 mode_str = tv_get_string_buf_chk(&argvars[1], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007187 if (mode_str == NULL)
7188 return;
7189 if (STRLEN(mode_str) != 9)
7190 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007191 semsg(_(e_invarg2), mode_str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007192 return;
7193 }
7194
7195 mask = 1;
7196 for (i = 8; i >= 0; --i)
7197 {
7198 if (mode_str[i] != '-')
7199 mode |= mask;
7200 mask = mask << 1;
7201 }
7202 rettv->vval.v_number = mch_setperm(fname, mode) == OK;
7203}
7204
7205/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007206 * "setpos()" function
7207 */
7208 static void
7209f_setpos(typval_T *argvars, typval_T *rettv)
7210{
7211 pos_T pos;
7212 int fnum;
7213 char_u *name;
7214 colnr_T curswant = -1;
7215
7216 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007217 name = tv_get_string_chk(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007218 if (name != NULL)
7219 {
7220 if (list2fpos(&argvars[1], &pos, &fnum, &curswant) == OK)
7221 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01007222 if (pos.col != MAXCOL && --pos.col < 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007223 pos.col = 0;
7224 if (name[0] == '.' && name[1] == NUL)
7225 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007226 // set cursor; "fnum" is ignored
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007227 curwin->w_cursor = pos;
7228 if (curswant >= 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007229 {
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007230 curwin->w_curswant = curswant - 1;
7231 curwin->w_set_curswant = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007232 }
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007233 check_cursor();
7234 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007235 }
7236 else if (name[0] == '\'' && name[1] != NUL && name[2] == NUL)
7237 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007238 // set mark
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007239 if (setmark_pos(name[1], &pos, fnum) == OK)
7240 rettv->vval.v_number = 0;
7241 }
7242 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007243 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007244 }
7245 }
7246}
7247
7248/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007249 * "setreg()" function
7250 */
7251 static void
7252f_setreg(typval_T *argvars, typval_T *rettv)
7253{
7254 int regname;
7255 char_u *strregname;
7256 char_u *stropt;
7257 char_u *strval;
7258 int append;
7259 char_u yank_type;
7260 long block_len;
7261
7262 block_len = -1;
7263 yank_type = MAUTO;
7264 append = FALSE;
7265
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007266 strregname = tv_get_string_chk(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007267 rettv->vval.v_number = 1; // FAIL is default
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007268
7269 if (strregname == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007270 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007271 regname = *strregname;
7272 if (regname == 0 || regname == '@')
7273 regname = '"';
7274
7275 if (argvars[2].v_type != VAR_UNKNOWN)
7276 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007277 stropt = tv_get_string_chk(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007278 if (stropt == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007279 return; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007280 for (; *stropt != NUL; ++stropt)
7281 switch (*stropt)
7282 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007283 case 'a': case 'A': // append
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007284 append = TRUE;
7285 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007286 case 'v': case 'c': // character-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007287 yank_type = MCHAR;
7288 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007289 case 'V': case 'l': // line-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007290 yank_type = MLINE;
7291 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007292 case 'b': case Ctrl_V: // block-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007293 yank_type = MBLOCK;
7294 if (VIM_ISDIGIT(stropt[1]))
7295 {
7296 ++stropt;
7297 block_len = getdigits(&stropt) - 1;
7298 --stropt;
7299 }
7300 break;
7301 }
7302 }
7303
7304 if (argvars[1].v_type == VAR_LIST)
7305 {
7306 char_u **lstval;
7307 char_u **allocval;
7308 char_u buf[NUMBUFLEN];
7309 char_u **curval;
7310 char_u **curallocval;
7311 list_T *ll = argvars[1].vval.v_list;
7312 listitem_T *li;
7313 int len;
7314
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007315 // If the list is NULL handle like an empty list.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007316 len = ll == NULL ? 0 : ll->lv_len;
7317
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007318 // First half: use for pointers to result lines; second half: use for
7319 // pointers to allocated copies.
Bram Moolenaarc799fe22019-05-28 23:08:19 +02007320 lstval = ALLOC_MULT(char_u *, (len + 1) * 2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007321 if (lstval == NULL)
7322 return;
7323 curval = lstval;
7324 allocval = lstval + len + 2;
7325 curallocval = allocval;
7326
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007327 if (ll != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007328 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007329 range_list_materialize(ll);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02007330 FOR_ALL_LIST_ITEMS(ll, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007331 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007332 strval = tv_get_string_buf_chk(&li->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007333 if (strval == NULL)
7334 goto free_lstval;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007335 if (strval == buf)
7336 {
7337 // Need to make a copy, next tv_get_string_buf_chk() will
7338 // overwrite the string.
7339 strval = vim_strsave(buf);
7340 if (strval == NULL)
7341 goto free_lstval;
7342 *curallocval++ = strval;
7343 }
7344 *curval++ = strval;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007345 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007346 }
7347 *curval++ = NULL;
7348
7349 write_reg_contents_lst(regname, lstval, -1,
7350 append, yank_type, block_len);
7351free_lstval:
7352 while (curallocval > allocval)
7353 vim_free(*--curallocval);
7354 vim_free(lstval);
7355 }
7356 else
7357 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007358 strval = tv_get_string_chk(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007359 if (strval == NULL)
7360 return;
7361 write_reg_contents_ex(regname, strval, -1,
7362 append, yank_type, block_len);
7363 }
7364 rettv->vval.v_number = 0;
7365}
7366
7367/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007368 * "settagstack()" function
7369 */
7370 static void
7371f_settagstack(typval_T *argvars, typval_T *rettv)
7372{
7373 static char *e_invact2 = N_("E962: Invalid action: '%s'");
7374 win_T *wp;
7375 dict_T *d;
7376 int action = 'r';
7377
7378 rettv->vval.v_number = -1;
7379
7380 // first argument: window number or id
7381 wp = find_win_by_nr_or_id(&argvars[0]);
7382 if (wp == NULL)
7383 return;
7384
7385 // second argument: dict with items to set in the tag stack
7386 if (argvars[1].v_type != VAR_DICT)
7387 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007388 emsg(_(e_dictreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007389 return;
7390 }
7391 d = argvars[1].vval.v_dict;
7392 if (d == NULL)
7393 return;
7394
7395 // third argument: action - 'a' for append and 'r' for replace.
7396 // default is to replace the stack.
7397 if (argvars[2].v_type == VAR_UNKNOWN)
7398 action = 'r';
7399 else if (argvars[2].v_type == VAR_STRING)
7400 {
7401 char_u *actstr;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007402 actstr = tv_get_string_chk(&argvars[2]);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007403 if (actstr == NULL)
7404 return;
Bram Moolenaar271fa082020-01-02 14:02:16 +01007405 if ((*actstr == 'r' || *actstr == 'a' || *actstr == 't')
7406 && actstr[1] == NUL)
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007407 action = *actstr;
7408 else
7409 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007410 semsg(_(e_invact2), actstr);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007411 return;
7412 }
7413 }
7414 else
7415 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007416 emsg(_(e_stringreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007417 return;
7418 }
7419
7420 if (set_tagstack(wp, d, action) == OK)
7421 rettv->vval.v_number = 0;
7422}
7423
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007424#ifdef FEAT_CRYPT
7425/*
7426 * "sha256({string})" function
7427 */
7428 static void
7429f_sha256(typval_T *argvars, typval_T *rettv)
7430{
7431 char_u *p;
7432
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007433 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007434 rettv->vval.v_string = vim_strsave(
7435 sha256_bytes(p, (int)STRLEN(p), NULL, 0));
7436 rettv->v_type = VAR_STRING;
7437}
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007438#endif // FEAT_CRYPT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007439
7440/*
7441 * "shellescape({string})" function
7442 */
7443 static void
7444f_shellescape(typval_T *argvars, typval_T *rettv)
7445{
Bram Moolenaar20615522017-06-05 18:46:26 +02007446 int do_special = non_zero_arg(&argvars[1]);
7447
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007448 rettv->vval.v_string = vim_strsave_shellescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007449 tv_get_string(&argvars[0]), do_special, do_special);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007450 rettv->v_type = VAR_STRING;
7451}
7452
7453/*
7454 * shiftwidth() function
7455 */
7456 static void
7457f_shiftwidth(typval_T *argvars UNUSED, typval_T *rettv)
7458{
Bram Moolenaarf9514162018-11-22 03:08:29 +01007459 rettv->vval.v_number = 0;
7460
7461 if (argvars[0].v_type != VAR_UNKNOWN)
7462 {
7463 long col;
7464
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007465 col = (long)tv_get_number_chk(argvars, NULL);
Bram Moolenaarf9514162018-11-22 03:08:29 +01007466 if (col < 0)
7467 return; // type error; errmsg already given
7468#ifdef FEAT_VARTABS
7469 rettv->vval.v_number = get_sw_value_col(curbuf, col);
7470 return;
7471#endif
7472 }
7473
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007474 rettv->vval.v_number = get_sw_value(curbuf);
7475}
7476
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007477#ifdef FEAT_FLOAT
7478/*
7479 * "sin()" function
7480 */
7481 static void
7482f_sin(typval_T *argvars, typval_T *rettv)
7483{
7484 float_T f = 0.0;
7485
7486 rettv->v_type = VAR_FLOAT;
7487 if (get_float_arg(argvars, &f) == OK)
7488 rettv->vval.v_float = sin(f);
7489 else
7490 rettv->vval.v_float = 0.0;
7491}
7492
7493/*
7494 * "sinh()" function
7495 */
7496 static void
7497f_sinh(typval_T *argvars, typval_T *rettv)
7498{
7499 float_T f = 0.0;
7500
7501 rettv->v_type = VAR_FLOAT;
7502 if (get_float_arg(argvars, &f) == OK)
7503 rettv->vval.v_float = sinh(f);
7504 else
7505 rettv->vval.v_float = 0.0;
7506}
7507#endif
7508
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007509/*
7510 * "soundfold({word})" function
7511 */
7512 static void
7513f_soundfold(typval_T *argvars, typval_T *rettv)
7514{
7515 char_u *s;
7516
7517 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007518 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007519#ifdef FEAT_SPELL
7520 rettv->vval.v_string = eval_soundfold(s);
7521#else
7522 rettv->vval.v_string = vim_strsave(s);
7523#endif
7524}
7525
7526/*
7527 * "spellbadword()" function
7528 */
7529 static void
7530f_spellbadword(typval_T *argvars UNUSED, typval_T *rettv)
7531{
7532 char_u *word = (char_u *)"";
7533 hlf_T attr = HLF_COUNT;
7534 int len = 0;
7535
7536 if (rettv_list_alloc(rettv) == FAIL)
7537 return;
7538
7539#ifdef FEAT_SPELL
7540 if (argvars[0].v_type == VAR_UNKNOWN)
7541 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007542 // Find the start and length of the badly spelled word.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007543 len = spell_move_to(curwin, FORWARD, TRUE, TRUE, &attr);
7544 if (len != 0)
Bram Moolenaarb73fa622017-12-21 20:27:47 +01007545 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007546 word = ml_get_cursor();
Bram Moolenaarb73fa622017-12-21 20:27:47 +01007547 curwin->w_set_curswant = TRUE;
7548 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007549 }
7550 else if (curwin->w_p_spell && *curbuf->b_s.b_p_spl != NUL)
7551 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007552 char_u *str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007553 int capcol = -1;
7554
7555 if (str != NULL)
7556 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007557 // Check the argument for spelling.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007558 while (*str != NUL)
7559 {
7560 len = spell_check(curwin, str, &attr, &capcol, FALSE);
7561 if (attr != HLF_COUNT)
7562 {
7563 word = str;
7564 break;
7565 }
7566 str += len;
Bram Moolenaar66ab9162018-07-20 20:28:48 +02007567 capcol -= len;
Bram Moolenaar0c779e82019-08-09 17:01:02 +02007568 len = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007569 }
7570 }
7571 }
7572#endif
7573
7574 list_append_string(rettv->vval.v_list, word, len);
7575 list_append_string(rettv->vval.v_list, (char_u *)(
7576 attr == HLF_SPB ? "bad" :
7577 attr == HLF_SPR ? "rare" :
7578 attr == HLF_SPL ? "local" :
7579 attr == HLF_SPC ? "caps" :
7580 ""), -1);
7581}
7582
7583/*
7584 * "spellsuggest()" function
7585 */
7586 static void
7587f_spellsuggest(typval_T *argvars UNUSED, typval_T *rettv)
7588{
7589#ifdef FEAT_SPELL
7590 char_u *str;
7591 int typeerr = FALSE;
7592 int maxcount;
7593 garray_T ga;
7594 int i;
7595 listitem_T *li;
7596 int need_capital = FALSE;
7597#endif
7598
7599 if (rettv_list_alloc(rettv) == FAIL)
7600 return;
7601
7602#ifdef FEAT_SPELL
7603 if (curwin->w_p_spell && *curwin->w_s->b_p_spl != NUL)
7604 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007605 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007606 if (argvars[1].v_type != VAR_UNKNOWN)
7607 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007608 maxcount = (int)tv_get_number_chk(&argvars[1], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007609 if (maxcount <= 0)
7610 return;
7611 if (argvars[2].v_type != VAR_UNKNOWN)
7612 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007613 need_capital = (int)tv_get_number_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007614 if (typeerr)
7615 return;
7616 }
7617 }
7618 else
7619 maxcount = 25;
7620
7621 spell_suggest_list(&ga, str, maxcount, need_capital, FALSE);
7622
7623 for (i = 0; i < ga.ga_len; ++i)
7624 {
7625 str = ((char_u **)ga.ga_data)[i];
7626
7627 li = listitem_alloc();
7628 if (li == NULL)
7629 vim_free(str);
7630 else
7631 {
7632 li->li_tv.v_type = VAR_STRING;
7633 li->li_tv.v_lock = 0;
7634 li->li_tv.vval.v_string = str;
7635 list_append(rettv->vval.v_list, li);
7636 }
7637 }
7638 ga_clear(&ga);
7639 }
7640#endif
7641}
7642
7643 static void
7644f_split(typval_T *argvars, typval_T *rettv)
7645{
7646 char_u *str;
7647 char_u *end;
7648 char_u *pat = NULL;
7649 regmatch_T regmatch;
7650 char_u patbuf[NUMBUFLEN];
7651 char_u *save_cpo;
7652 int match;
7653 colnr_T col = 0;
7654 int keepempty = FALSE;
7655 int typeerr = FALSE;
7656
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007657 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007658 save_cpo = p_cpo;
7659 p_cpo = (char_u *)"";
7660
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007661 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007662 if (argvars[1].v_type != VAR_UNKNOWN)
7663 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007664 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007665 if (pat == NULL)
7666 typeerr = TRUE;
7667 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007668 keepempty = (int)tv_get_number_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007669 }
7670 if (pat == NULL || *pat == NUL)
7671 pat = (char_u *)"[\\x01- ]\\+";
7672
7673 if (rettv_list_alloc(rettv) == FAIL)
7674 return;
7675 if (typeerr)
7676 return;
7677
7678 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
7679 if (regmatch.regprog != NULL)
7680 {
7681 regmatch.rm_ic = FALSE;
7682 while (*str != NUL || keepempty)
7683 {
7684 if (*str == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007685 match = FALSE; // empty item at the end
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007686 else
7687 match = vim_regexec_nl(&regmatch, str, col);
7688 if (match)
7689 end = regmatch.startp[0];
7690 else
7691 end = str + STRLEN(str);
7692 if (keepempty || end > str || (rettv->vval.v_list->lv_len > 0
7693 && *str != NUL && match && end < regmatch.endp[0]))
7694 {
7695 if (list_append_string(rettv->vval.v_list, str,
7696 (int)(end - str)) == FAIL)
7697 break;
7698 }
7699 if (!match)
7700 break;
Bram Moolenaar13505972019-01-24 15:04:48 +01007701 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007702 if (regmatch.endp[0] > str)
7703 col = 0;
7704 else
Bram Moolenaar13505972019-01-24 15:04:48 +01007705 // Don't get stuck at the same match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007706 col = (*mb_ptr2len)(regmatch.endp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007707 str = regmatch.endp[0];
7708 }
7709
7710 vim_regfree(regmatch.regprog);
7711 }
7712
7713 p_cpo = save_cpo;
7714}
7715
7716#ifdef FEAT_FLOAT
7717/*
7718 * "sqrt()" function
7719 */
7720 static void
7721f_sqrt(typval_T *argvars, typval_T *rettv)
7722{
7723 float_T f = 0.0;
7724
7725 rettv->v_type = VAR_FLOAT;
7726 if (get_float_arg(argvars, &f) == OK)
7727 rettv->vval.v_float = sqrt(f);
7728 else
7729 rettv->vval.v_float = 0.0;
7730}
Bram Moolenaar0387cae2019-11-29 21:07:58 +01007731#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007732
Bram Moolenaar0387cae2019-11-29 21:07:58 +01007733#ifdef FEAT_FLOAT
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007734/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007735 * "str2float()" function
7736 */
7737 static void
7738f_str2float(typval_T *argvars, typval_T *rettv)
7739{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007740 char_u *p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01007741 int isneg = (*p == '-');
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007742
Bram Moolenaar08243d22017-01-10 16:12:29 +01007743 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007744 p = skipwhite(p + 1);
7745 (void)string2float(p, &rettv->vval.v_float);
Bram Moolenaar08243d22017-01-10 16:12:29 +01007746 if (isneg)
7747 rettv->vval.v_float *= -1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007748 rettv->v_type = VAR_FLOAT;
7749}
7750#endif
7751
7752/*
Bram Moolenaar9d401282019-04-06 13:18:12 +02007753 * "str2list()" function
7754 */
7755 static void
7756f_str2list(typval_T *argvars, typval_T *rettv)
7757{
7758 char_u *p;
7759 int utf8 = FALSE;
7760
7761 if (rettv_list_alloc(rettv) == FAIL)
7762 return;
7763
7764 if (argvars[1].v_type != VAR_UNKNOWN)
7765 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
7766
7767 p = tv_get_string(&argvars[0]);
7768
7769 if (has_mbyte || utf8)
7770 {
7771 int (*ptr2len)(char_u *);
7772 int (*ptr2char)(char_u *);
7773
7774 if (utf8 || enc_utf8)
7775 {
7776 ptr2len = utf_ptr2len;
7777 ptr2char = utf_ptr2char;
7778 }
7779 else
7780 {
7781 ptr2len = mb_ptr2len;
7782 ptr2char = mb_ptr2char;
7783 }
7784
7785 for ( ; *p != NUL; p += (*ptr2len)(p))
7786 list_append_number(rettv->vval.v_list, (*ptr2char)(p));
7787 }
7788 else
7789 for ( ; *p != NUL; ++p)
7790 list_append_number(rettv->vval.v_list, *p);
7791}
7792
7793/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007794 * "str2nr()" function
7795 */
7796 static void
7797f_str2nr(typval_T *argvars, typval_T *rettv)
7798{
7799 int base = 10;
7800 char_u *p;
7801 varnumber_T n;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007802 int what = 0;
Bram Moolenaar08243d22017-01-10 16:12:29 +01007803 int isneg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007804
7805 if (argvars[1].v_type != VAR_UNKNOWN)
7806 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007807 base = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007808 if (base != 2 && base != 8 && base != 10 && base != 16)
7809 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007810 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007811 return;
7812 }
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007813 if (argvars[2].v_type != VAR_UNKNOWN && tv_get_number(&argvars[2]))
7814 what |= STR2NR_QUOTE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007815 }
7816
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007817 p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01007818 isneg = (*p == '-');
7819 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007820 p = skipwhite(p + 1);
7821 switch (base)
7822 {
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007823 case 2: what |= STR2NR_BIN + STR2NR_FORCE; break;
7824 case 8: what |= STR2NR_OCT + STR2NR_FORCE; break;
7825 case 16: what |= STR2NR_HEX + STR2NR_FORCE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007826 }
Bram Moolenaar16e9b852019-05-19 19:59:35 +02007827 vim_str2nr(p, NULL, NULL, what, &n, NULL, 0, FALSE);
7828 // Text after the number is silently ignored.
Bram Moolenaar08243d22017-01-10 16:12:29 +01007829 if (isneg)
7830 rettv->vval.v_number = -n;
7831 else
7832 rettv->vval.v_number = n;
7833
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007834}
7835
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007836/*
7837 * "strgetchar()" function
7838 */
7839 static void
7840f_strgetchar(typval_T *argvars, typval_T *rettv)
7841{
7842 char_u *str;
7843 int len;
7844 int error = FALSE;
7845 int charidx;
Bram Moolenaar13505972019-01-24 15:04:48 +01007846 int byteidx = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007847
7848 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007849 str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007850 if (str == NULL)
7851 return;
7852 len = (int)STRLEN(str);
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007853 charidx = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007854 if (error)
7855 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007856
Bram Moolenaar13505972019-01-24 15:04:48 +01007857 while (charidx >= 0 && byteidx < len)
7858 {
7859 if (charidx == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007860 {
Bram Moolenaar13505972019-01-24 15:04:48 +01007861 rettv->vval.v_number = mb_ptr2char(str + byteidx);
7862 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007863 }
Bram Moolenaar13505972019-01-24 15:04:48 +01007864 --charidx;
7865 byteidx += MB_CPTR2LEN(str + byteidx);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007866 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007867}
7868
7869/*
7870 * "stridx()" function
7871 */
7872 static void
7873f_stridx(typval_T *argvars, typval_T *rettv)
7874{
7875 char_u buf[NUMBUFLEN];
7876 char_u *needle;
7877 char_u *haystack;
7878 char_u *save_haystack;
7879 char_u *pos;
7880 int start_idx;
7881
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007882 needle = tv_get_string_chk(&argvars[1]);
7883 save_haystack = haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007884 rettv->vval.v_number = -1;
7885 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007886 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007887
7888 if (argvars[2].v_type != VAR_UNKNOWN)
7889 {
7890 int error = FALSE;
7891
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007892 start_idx = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007893 if (error || start_idx >= (int)STRLEN(haystack))
7894 return;
7895 if (start_idx >= 0)
7896 haystack += start_idx;
7897 }
7898
7899 pos = (char_u *)strstr((char *)haystack, (char *)needle);
7900 if (pos != NULL)
7901 rettv->vval.v_number = (varnumber_T)(pos - save_haystack);
7902}
7903
7904/*
7905 * "string()" function
7906 */
Bram Moolenaar461a7fc2018-12-22 13:28:07 +01007907 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007908f_string(typval_T *argvars, typval_T *rettv)
7909{
7910 char_u *tofree;
7911 char_u numbuf[NUMBUFLEN];
7912
7913 rettv->v_type = VAR_STRING;
7914 rettv->vval.v_string = tv2string(&argvars[0], &tofree, numbuf,
7915 get_copyID());
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007916 // Make a copy if we have a value but it's not in allocated memory.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007917 if (rettv->vval.v_string != NULL && tofree == NULL)
7918 rettv->vval.v_string = vim_strsave(rettv->vval.v_string);
7919}
7920
7921/*
7922 * "strlen()" function
7923 */
7924 static void
7925f_strlen(typval_T *argvars, typval_T *rettv)
7926{
7927 rettv->vval.v_number = (varnumber_T)(STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007928 tv_get_string(&argvars[0])));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007929}
7930
7931/*
7932 * "strchars()" function
7933 */
7934 static void
7935f_strchars(typval_T *argvars, typval_T *rettv)
7936{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007937 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007938 int skipcc = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007939 varnumber_T len = 0;
7940 int (*func_mb_ptr2char_adv)(char_u **pp);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007941
7942 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007943 skipcc = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007944 if (skipcc < 0 || skipcc > 1)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007945 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007946 else
7947 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007948 func_mb_ptr2char_adv = skipcc ? mb_ptr2char_adv : mb_cptr2char_adv;
7949 while (*s != NUL)
7950 {
7951 func_mb_ptr2char_adv(&s);
7952 ++len;
7953 }
7954 rettv->vval.v_number = len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007955 }
7956}
7957
7958/*
7959 * "strdisplaywidth()" function
7960 */
7961 static void
7962f_strdisplaywidth(typval_T *argvars, typval_T *rettv)
7963{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007964 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007965 int col = 0;
7966
7967 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007968 col = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007969
7970 rettv->vval.v_number = (varnumber_T)(linetabsize_col(col, s) - col);
7971}
7972
7973/*
7974 * "strwidth()" function
7975 */
7976 static void
7977f_strwidth(typval_T *argvars, typval_T *rettv)
7978{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007979 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007980
Bram Moolenaar13505972019-01-24 15:04:48 +01007981 rettv->vval.v_number = (varnumber_T)(mb_string2cells(s, -1));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007982}
7983
7984/*
7985 * "strcharpart()" function
7986 */
7987 static void
7988f_strcharpart(typval_T *argvars, typval_T *rettv)
7989{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007990 char_u *p;
7991 int nchar;
7992 int nbyte = 0;
7993 int charlen;
7994 int len = 0;
7995 int slen;
7996 int error = FALSE;
7997
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007998 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007999 slen = (int)STRLEN(p);
8000
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008001 nchar = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008002 if (!error)
8003 {
8004 if (nchar > 0)
8005 while (nchar > 0 && nbyte < slen)
8006 {
Bram Moolenaard3c907b2016-08-17 21:32:09 +02008007 nbyte += MB_CPTR2LEN(p + nbyte);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008008 --nchar;
8009 }
8010 else
8011 nbyte = nchar;
8012 if (argvars[2].v_type != VAR_UNKNOWN)
8013 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008014 charlen = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008015 while (charlen > 0 && nbyte + len < slen)
8016 {
8017 int off = nbyte + len;
8018
8019 if (off < 0)
8020 len += 1;
8021 else
Bram Moolenaard3c907b2016-08-17 21:32:09 +02008022 len += MB_CPTR2LEN(p + off);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008023 --charlen;
8024 }
8025 }
8026 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008027 len = slen - nbyte; // default: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008028 }
8029
8030 /*
8031 * Only return the overlap between the specified part and the actual
8032 * string.
8033 */
8034 if (nbyte < 0)
8035 {
8036 len += nbyte;
8037 nbyte = 0;
8038 }
8039 else if (nbyte > slen)
8040 nbyte = slen;
8041 if (len < 0)
8042 len = 0;
8043 else if (nbyte + len > slen)
8044 len = slen - nbyte;
8045
8046 rettv->v_type = VAR_STRING;
8047 rettv->vval.v_string = vim_strnsave(p + nbyte, len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008048}
8049
8050/*
8051 * "strpart()" function
8052 */
8053 static void
8054f_strpart(typval_T *argvars, typval_T *rettv)
8055{
8056 char_u *p;
8057 int n;
8058 int len;
8059 int slen;
8060 int error = FALSE;
8061
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008062 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008063 slen = (int)STRLEN(p);
8064
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008065 n = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008066 if (error)
8067 len = 0;
8068 else if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008069 len = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008070 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008071 len = slen - n; // default len: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008072
8073 /*
8074 * Only return the overlap between the specified part and the actual
8075 * string.
8076 */
8077 if (n < 0)
8078 {
8079 len += n;
8080 n = 0;
8081 }
8082 else if (n > slen)
8083 n = slen;
8084 if (len < 0)
8085 len = 0;
8086 else if (n + len > slen)
8087 len = slen - n;
8088
8089 rettv->v_type = VAR_STRING;
8090 rettv->vval.v_string = vim_strnsave(p + n, len);
8091}
8092
8093/*
8094 * "strridx()" function
8095 */
8096 static void
8097f_strridx(typval_T *argvars, typval_T *rettv)
8098{
8099 char_u buf[NUMBUFLEN];
8100 char_u *needle;
8101 char_u *haystack;
8102 char_u *rest;
8103 char_u *lastmatch = NULL;
8104 int haystack_len, end_idx;
8105
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008106 needle = tv_get_string_chk(&argvars[1]);
8107 haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008108
8109 rettv->vval.v_number = -1;
8110 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008111 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008112
8113 haystack_len = (int)STRLEN(haystack);
8114 if (argvars[2].v_type != VAR_UNKNOWN)
8115 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008116 // Third argument: upper limit for index
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008117 end_idx = (int)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008118 if (end_idx < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008119 return; // can never find a match
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008120 }
8121 else
8122 end_idx = haystack_len;
8123
8124 if (*needle == NUL)
8125 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008126 // Empty string matches past the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008127 lastmatch = haystack + end_idx;
8128 }
8129 else
8130 {
8131 for (rest = haystack; *rest != '\0'; ++rest)
8132 {
8133 rest = (char_u *)strstr((char *)rest, (char *)needle);
8134 if (rest == NULL || rest > haystack + end_idx)
8135 break;
8136 lastmatch = rest;
8137 }
8138 }
8139
8140 if (lastmatch == NULL)
8141 rettv->vval.v_number = -1;
8142 else
8143 rettv->vval.v_number = (varnumber_T)(lastmatch - haystack);
8144}
8145
8146/*
8147 * "strtrans()" function
8148 */
8149 static void
8150f_strtrans(typval_T *argvars, typval_T *rettv)
8151{
8152 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008153 rettv->vval.v_string = transstr(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008154}
8155
8156/*
8157 * "submatch()" function
8158 */
8159 static void
8160f_submatch(typval_T *argvars, typval_T *rettv)
8161{
8162 int error = FALSE;
8163 int no;
8164 int retList = 0;
8165
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008166 no = (int)tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008167 if (error)
8168 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02008169 if (no < 0 || no >= NSUBEXP)
8170 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008171 semsg(_("E935: invalid submatch number: %d"), no);
Bram Moolenaar79518e22017-02-17 16:31:35 +01008172 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02008173 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008174 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008175 retList = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008176 if (error)
8177 return;
8178
8179 if (retList == 0)
8180 {
8181 rettv->v_type = VAR_STRING;
8182 rettv->vval.v_string = reg_submatch(no);
8183 }
8184 else
8185 {
8186 rettv->v_type = VAR_LIST;
8187 rettv->vval.v_list = reg_submatch_list(no);
8188 }
8189}
8190
8191/*
8192 * "substitute()" function
8193 */
8194 static void
8195f_substitute(typval_T *argvars, typval_T *rettv)
8196{
8197 char_u patbuf[NUMBUFLEN];
8198 char_u subbuf[NUMBUFLEN];
8199 char_u flagsbuf[NUMBUFLEN];
8200
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008201 char_u *str = tv_get_string_chk(&argvars[0]);
8202 char_u *pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008203 char_u *sub = NULL;
8204 typval_T *expr = NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008205 char_u *flg = tv_get_string_buf_chk(&argvars[3], flagsbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008206
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008207 if (argvars[2].v_type == VAR_FUNC || argvars[2].v_type == VAR_PARTIAL)
8208 expr = &argvars[2];
8209 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008210 sub = tv_get_string_buf_chk(&argvars[2], subbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008211
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008212 rettv->v_type = VAR_STRING;
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008213 if (str == NULL || pat == NULL || (sub == NULL && expr == NULL)
8214 || flg == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008215 rettv->vval.v_string = NULL;
8216 else
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008217 rettv->vval.v_string = do_string_sub(str, pat, sub, expr, flg);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008218}
8219
8220/*
Bram Moolenaar00f123a2018-08-21 20:28:54 +02008221 * "swapinfo(swap_filename)" function
8222 */
8223 static void
8224f_swapinfo(typval_T *argvars, typval_T *rettv)
8225{
8226 if (rettv_dict_alloc(rettv) == OK)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008227 get_b0_dict(tv_get_string(argvars), rettv->vval.v_dict);
Bram Moolenaar00f123a2018-08-21 20:28:54 +02008228}
8229
8230/*
Bram Moolenaar110bd602018-09-16 18:46:59 +02008231 * "swapname(expr)" function
8232 */
8233 static void
8234f_swapname(typval_T *argvars, typval_T *rettv)
8235{
8236 buf_T *buf;
8237
8238 rettv->v_type = VAR_STRING;
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01008239 buf = tv_get_buf(&argvars[0], FALSE);
Bram Moolenaar110bd602018-09-16 18:46:59 +02008240 if (buf == NULL || buf->b_ml.ml_mfp == NULL
8241 || buf->b_ml.ml_mfp->mf_fname == NULL)
8242 rettv->vval.v_string = NULL;
8243 else
8244 rettv->vval.v_string = vim_strsave(buf->b_ml.ml_mfp->mf_fname);
8245}
8246
8247/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008248 * "synID(lnum, col, trans)" function
8249 */
8250 static void
8251f_synID(typval_T *argvars UNUSED, typval_T *rettv)
8252{
8253 int id = 0;
8254#ifdef FEAT_SYN_HL
8255 linenr_T lnum;
8256 colnr_T col;
8257 int trans;
8258 int transerr = FALSE;
8259
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008260 lnum = tv_get_lnum(argvars); // -1 on type error
8261 col = (linenr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008262 trans = (int)tv_get_number_chk(&argvars[2], &transerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008263
8264 if (!transerr && lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8265 && col >= 0 && col < (long)STRLEN(ml_get(lnum)))
8266 id = syn_get_id(curwin, lnum, (colnr_T)col, trans, NULL, FALSE);
8267#endif
8268
8269 rettv->vval.v_number = id;
8270}
8271
8272/*
8273 * "synIDattr(id, what [, mode])" function
8274 */
8275 static void
8276f_synIDattr(typval_T *argvars UNUSED, typval_T *rettv)
8277{
8278 char_u *p = NULL;
8279#ifdef FEAT_SYN_HL
8280 int id;
8281 char_u *what;
8282 char_u *mode;
8283 char_u modebuf[NUMBUFLEN];
8284 int modec;
8285
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008286 id = (int)tv_get_number(&argvars[0]);
8287 what = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008288 if (argvars[2].v_type != VAR_UNKNOWN)
8289 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008290 mode = tv_get_string_buf(&argvars[2], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008291 modec = TOLOWER_ASC(mode[0]);
8292 if (modec != 't' && modec != 'c' && modec != 'g')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008293 modec = 0; // replace invalid with current
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008294 }
8295 else
8296 {
8297#if defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS)
8298 if (USE_24BIT)
8299 modec = 'g';
8300 else
8301#endif
8302 if (t_colors > 1)
8303 modec = 'c';
8304 else
8305 modec = 't';
8306 }
8307
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008308 switch (TOLOWER_ASC(what[0]))
8309 {
8310 case 'b':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008311 if (TOLOWER_ASC(what[1]) == 'g') // bg[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008312 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008313 else // bold
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008314 p = highlight_has_attr(id, HL_BOLD, modec);
8315 break;
8316
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008317 case 'f': // fg[#] or font
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008318 p = highlight_color(id, what, modec);
8319 break;
8320
8321 case 'i':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008322 if (TOLOWER_ASC(what[1]) == 'n') // inverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008323 p = highlight_has_attr(id, HL_INVERSE, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008324 else // italic
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008325 p = highlight_has_attr(id, HL_ITALIC, modec);
8326 break;
8327
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008328 case 'n': // name
Bram Moolenaarc96272e2017-03-26 13:50:09 +02008329 p = get_highlight_name_ext(NULL, id - 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008330 break;
8331
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008332 case 'r': // reverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008333 p = highlight_has_attr(id, HL_INVERSE, modec);
8334 break;
8335
8336 case 's':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008337 if (TOLOWER_ASC(what[1]) == 'p') // sp[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008338 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008339 // strikeout
Bram Moolenaarcf4b00c2017-09-02 18:33:56 +02008340 else if (TOLOWER_ASC(what[1]) == 't' &&
8341 TOLOWER_ASC(what[2]) == 'r')
8342 p = highlight_has_attr(id, HL_STRIKETHROUGH, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008343 else // standout
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008344 p = highlight_has_attr(id, HL_STANDOUT, modec);
8345 break;
8346
8347 case 'u':
8348 if (STRLEN(what) <= 5 || TOLOWER_ASC(what[5]) != 'c')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008349 // underline
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008350 p = highlight_has_attr(id, HL_UNDERLINE, modec);
8351 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008352 // undercurl
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008353 p = highlight_has_attr(id, HL_UNDERCURL, modec);
8354 break;
8355 }
8356
8357 if (p != NULL)
8358 p = vim_strsave(p);
8359#endif
8360 rettv->v_type = VAR_STRING;
8361 rettv->vval.v_string = p;
8362}
8363
8364/*
8365 * "synIDtrans(id)" function
8366 */
8367 static void
8368f_synIDtrans(typval_T *argvars UNUSED, typval_T *rettv)
8369{
8370 int id;
8371
8372#ifdef FEAT_SYN_HL
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008373 id = (int)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008374
8375 if (id > 0)
8376 id = syn_get_final_id(id);
8377 else
8378#endif
8379 id = 0;
8380
8381 rettv->vval.v_number = id;
8382}
8383
8384/*
8385 * "synconcealed(lnum, col)" function
8386 */
8387 static void
8388f_synconcealed(typval_T *argvars UNUSED, typval_T *rettv)
8389{
8390#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
8391 linenr_T lnum;
8392 colnr_T col;
8393 int syntax_flags = 0;
8394 int cchar;
8395 int matchid = 0;
8396 char_u str[NUMBUFLEN];
8397#endif
8398
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02008399 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008400
8401#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008402 lnum = tv_get_lnum(argvars); // -1 on type error
8403 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008404
Bram Moolenaara80faa82020-04-12 19:37:17 +02008405 CLEAR_FIELD(str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008406
8407 if (rettv_list_alloc(rettv) != FAIL)
8408 {
8409 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8410 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
8411 && curwin->w_p_cole > 0)
8412 {
8413 (void)syn_get_id(curwin, lnum, col, FALSE, NULL, FALSE);
8414 syntax_flags = get_syntax_info(&matchid);
8415
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008416 // get the conceal character
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008417 if ((syntax_flags & HL_CONCEAL) && curwin->w_p_cole < 3)
8418 {
8419 cchar = syn_get_sub_char();
Bram Moolenaar4d785892017-06-22 22:00:50 +02008420 if (cchar == NUL && curwin->w_p_cole == 1)
8421 cchar = (lcs_conceal == NUL) ? ' ' : lcs_conceal;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008422 if (cchar != NUL)
8423 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008424 if (has_mbyte)
8425 (*mb_char2bytes)(cchar, str);
8426 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008427 str[0] = cchar;
8428 }
8429 }
8430 }
8431
8432 list_append_number(rettv->vval.v_list,
8433 (syntax_flags & HL_CONCEAL) != 0);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008434 // -1 to auto-determine strlen
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008435 list_append_string(rettv->vval.v_list, str, -1);
8436 list_append_number(rettv->vval.v_list, matchid);
8437 }
8438#endif
8439}
8440
8441/*
8442 * "synstack(lnum, col)" function
8443 */
8444 static void
8445f_synstack(typval_T *argvars UNUSED, typval_T *rettv)
8446{
8447#ifdef FEAT_SYN_HL
8448 linenr_T lnum;
8449 colnr_T col;
8450 int i;
8451 int id;
8452#endif
8453
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02008454 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008455
8456#ifdef FEAT_SYN_HL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008457 lnum = tv_get_lnum(argvars); // -1 on type error
8458 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008459
8460 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8461 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
8462 && rettv_list_alloc(rettv) != FAIL)
8463 {
8464 (void)syn_get_id(curwin, lnum, (colnr_T)col, FALSE, NULL, TRUE);
8465 for (i = 0; ; ++i)
8466 {
8467 id = syn_get_stack_item(i);
8468 if (id < 0)
8469 break;
8470 if (list_append_number(rettv->vval.v_list, id) == FAIL)
8471 break;
8472 }
8473 }
8474#endif
8475}
8476
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008477/*
8478 * "tabpagebuflist()" function
8479 */
8480 static void
8481f_tabpagebuflist(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8482{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008483 tabpage_T *tp;
8484 win_T *wp = NULL;
8485
8486 if (argvars[0].v_type == VAR_UNKNOWN)
8487 wp = firstwin;
8488 else
8489 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008490 tp = find_tabpage((int)tv_get_number(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008491 if (tp != NULL)
8492 wp = (tp == curtab) ? firstwin : tp->tp_firstwin;
8493 }
8494 if (wp != NULL && rettv_list_alloc(rettv) != FAIL)
8495 {
8496 for (; wp != NULL; wp = wp->w_next)
8497 if (list_append_number(rettv->vval.v_list,
8498 wp->w_buffer->b_fnum) == FAIL)
8499 break;
8500 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008501}
8502
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008503/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008504 * "tagfiles()" function
8505 */
8506 static void
8507f_tagfiles(typval_T *argvars UNUSED, typval_T *rettv)
8508{
8509 char_u *fname;
8510 tagname_T tn;
8511 int first;
8512
8513 if (rettv_list_alloc(rettv) == FAIL)
8514 return;
8515 fname = alloc(MAXPATHL);
8516 if (fname == NULL)
8517 return;
8518
8519 for (first = TRUE; ; first = FALSE)
8520 if (get_tagfname(&tn, first, fname) == FAIL
8521 || list_append_string(rettv->vval.v_list, fname, -1) == FAIL)
8522 break;
8523 tagname_free(&tn);
8524 vim_free(fname);
8525}
8526
8527/*
8528 * "taglist()" function
8529 */
8530 static void
8531f_taglist(typval_T *argvars, typval_T *rettv)
8532{
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008533 char_u *fname = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008534 char_u *tag_pattern;
8535
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008536 tag_pattern = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008537
8538 rettv->vval.v_number = FALSE;
8539 if (*tag_pattern == NUL)
8540 return;
8541
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008542 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008543 fname = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008544 if (rettv_list_alloc(rettv) == OK)
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008545 (void)get_tags(rettv->vval.v_list, tag_pattern, fname);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008546}
8547
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008548#ifdef FEAT_FLOAT
8549/*
8550 * "tan()" function
8551 */
8552 static void
8553f_tan(typval_T *argvars, typval_T *rettv)
8554{
8555 float_T f = 0.0;
8556
8557 rettv->v_type = VAR_FLOAT;
8558 if (get_float_arg(argvars, &f) == OK)
8559 rettv->vval.v_float = tan(f);
8560 else
8561 rettv->vval.v_float = 0.0;
8562}
8563
8564/*
8565 * "tanh()" function
8566 */
8567 static void
8568f_tanh(typval_T *argvars, typval_T *rettv)
8569{
8570 float_T f = 0.0;
8571
8572 rettv->v_type = VAR_FLOAT;
8573 if (get_float_arg(argvars, &f) == OK)
8574 rettv->vval.v_float = tanh(f);
8575 else
8576 rettv->vval.v_float = 0.0;
8577}
8578#endif
8579
8580/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008581 * "tolower(string)" function
8582 */
8583 static void
8584f_tolower(typval_T *argvars, typval_T *rettv)
8585{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008586 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008587 rettv->vval.v_string = strlow_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008588}
8589
8590/*
8591 * "toupper(string)" function
8592 */
8593 static void
8594f_toupper(typval_T *argvars, typval_T *rettv)
8595{
8596 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008597 rettv->vval.v_string = strup_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008598}
8599
8600/*
8601 * "tr(string, fromstr, tostr)" function
8602 */
8603 static void
8604f_tr(typval_T *argvars, typval_T *rettv)
8605{
8606 char_u *in_str;
8607 char_u *fromstr;
8608 char_u *tostr;
8609 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008610 int inlen;
8611 int fromlen;
8612 int tolen;
8613 int idx;
8614 char_u *cpstr;
8615 int cplen;
8616 int first = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008617 char_u buf[NUMBUFLEN];
8618 char_u buf2[NUMBUFLEN];
8619 garray_T ga;
8620
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008621 in_str = tv_get_string(&argvars[0]);
8622 fromstr = tv_get_string_buf_chk(&argvars[1], buf);
8623 tostr = tv_get_string_buf_chk(&argvars[2], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008624
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008625 // Default return value: empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008626 rettv->v_type = VAR_STRING;
8627 rettv->vval.v_string = NULL;
8628 if (fromstr == NULL || tostr == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008629 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008630 ga_init2(&ga, (int)sizeof(char), 80);
8631
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008632 if (!has_mbyte)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008633 // not multi-byte: fromstr and tostr must be the same length
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008634 if (STRLEN(fromstr) != STRLEN(tostr))
8635 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008636error:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008637 semsg(_(e_invarg2), fromstr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008638 ga_clear(&ga);
8639 return;
8640 }
8641
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008642 // fromstr and tostr have to contain the same number of chars
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008643 while (*in_str != NUL)
8644 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008645 if (has_mbyte)
8646 {
8647 inlen = (*mb_ptr2len)(in_str);
8648 cpstr = in_str;
8649 cplen = inlen;
8650 idx = 0;
8651 for (p = fromstr; *p != NUL; p += fromlen)
8652 {
8653 fromlen = (*mb_ptr2len)(p);
8654 if (fromlen == inlen && STRNCMP(in_str, p, inlen) == 0)
8655 {
8656 for (p = tostr; *p != NUL; p += tolen)
8657 {
8658 tolen = (*mb_ptr2len)(p);
8659 if (idx-- == 0)
8660 {
8661 cplen = tolen;
8662 cpstr = p;
8663 break;
8664 }
8665 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008666 if (*p == NUL) // tostr is shorter than fromstr
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008667 goto error;
8668 break;
8669 }
8670 ++idx;
8671 }
8672
8673 if (first && cpstr == in_str)
8674 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008675 // Check that fromstr and tostr have the same number of
8676 // (multi-byte) characters. Done only once when a character
8677 // of in_str doesn't appear in fromstr.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008678 first = FALSE;
8679 for (p = tostr; *p != NUL; p += tolen)
8680 {
8681 tolen = (*mb_ptr2len)(p);
8682 --idx;
8683 }
8684 if (idx != 0)
8685 goto error;
8686 }
8687
8688 (void)ga_grow(&ga, cplen);
8689 mch_memmove((char *)ga.ga_data + ga.ga_len, cpstr, (size_t)cplen);
8690 ga.ga_len += cplen;
8691
8692 in_str += inlen;
8693 }
8694 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008695 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008696 // When not using multi-byte chars we can do it faster.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008697 p = vim_strchr(fromstr, *in_str);
8698 if (p != NULL)
8699 ga_append(&ga, tostr[p - fromstr]);
8700 else
8701 ga_append(&ga, *in_str);
8702 ++in_str;
8703 }
8704 }
8705
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008706 // add a terminating NUL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008707 (void)ga_grow(&ga, 1);
8708 ga_append(&ga, NUL);
8709
8710 rettv->vval.v_string = ga.ga_data;
8711}
8712
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008713/*
8714 * "trim({expr})" function
8715 */
8716 static void
8717f_trim(typval_T *argvars, typval_T *rettv)
8718{
8719 char_u buf1[NUMBUFLEN];
8720 char_u buf2[NUMBUFLEN];
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008721 char_u *head = tv_get_string_buf_chk(&argvars[0], buf1);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008722 char_u *mask = NULL;
8723 char_u *tail;
8724 char_u *prev;
8725 char_u *p;
8726 int c1;
8727
8728 rettv->v_type = VAR_STRING;
8729 if (head == NULL)
8730 {
8731 rettv->vval.v_string = NULL;
8732 return;
8733 }
8734
8735 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008736 mask = tv_get_string_buf_chk(&argvars[1], buf2);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008737
8738 while (*head != NUL)
8739 {
8740 c1 = PTR2CHAR(head);
8741 if (mask == NULL)
8742 {
8743 if (c1 > ' ' && c1 != 0xa0)
8744 break;
8745 }
8746 else
8747 {
8748 for (p = mask; *p != NUL; MB_PTR_ADV(p))
8749 if (c1 == PTR2CHAR(p))
8750 break;
8751 if (*p == NUL)
8752 break;
8753 }
8754 MB_PTR_ADV(head);
8755 }
8756
8757 for (tail = head + STRLEN(head); tail > head; tail = prev)
8758 {
8759 prev = tail;
8760 MB_PTR_BACK(head, prev);
8761 c1 = PTR2CHAR(prev);
8762 if (mask == NULL)
8763 {
8764 if (c1 > ' ' && c1 != 0xa0)
8765 break;
8766 }
8767 else
8768 {
8769 for (p = mask; *p != NUL; MB_PTR_ADV(p))
8770 if (c1 == PTR2CHAR(p))
8771 break;
8772 if (*p == NUL)
8773 break;
8774 }
8775 }
8776 rettv->vval.v_string = vim_strnsave(head, (int)(tail - head));
8777}
8778
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008779#ifdef FEAT_FLOAT
8780/*
8781 * "trunc({float})" function
8782 */
8783 static void
8784f_trunc(typval_T *argvars, typval_T *rettv)
8785{
8786 float_T f = 0.0;
8787
8788 rettv->v_type = VAR_FLOAT;
8789 if (get_float_arg(argvars, &f) == OK)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008790 // trunc() is not in C90, use floor() or ceil() instead.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008791 rettv->vval.v_float = f > 0 ? floor(f) : ceil(f);
8792 else
8793 rettv->vval.v_float = 0.0;
8794}
8795#endif
8796
8797/*
8798 * "type(expr)" function
8799 */
8800 static void
8801f_type(typval_T *argvars, typval_T *rettv)
8802{
8803 int n = -1;
8804
8805 switch (argvars[0].v_type)
8806 {
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01008807 case VAR_NUMBER: n = VAR_TYPE_NUMBER; break;
8808 case VAR_STRING: n = VAR_TYPE_STRING; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008809 case VAR_PARTIAL:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01008810 case VAR_FUNC: n = VAR_TYPE_FUNC; break;
8811 case VAR_LIST: n = VAR_TYPE_LIST; break;
8812 case VAR_DICT: n = VAR_TYPE_DICT; break;
8813 case VAR_FLOAT: n = VAR_TYPE_FLOAT; break;
8814 case VAR_BOOL: n = VAR_TYPE_BOOL; break;
8815 case VAR_SPECIAL: n = VAR_TYPE_NONE; break;
Bram Moolenaarf562e722016-07-19 17:25:25 +02008816 case VAR_JOB: n = VAR_TYPE_JOB; break;
8817 case VAR_CHANNEL: n = VAR_TYPE_CHANNEL; break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01008818 case VAR_BLOB: n = VAR_TYPE_BLOB; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008819 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02008820 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01008821 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01008822 internal_error_no_abort("f_type(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008823 n = -1;
8824 break;
8825 }
8826 rettv->vval.v_number = n;
8827}
8828
8829/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008830 * "virtcol(string)" function
8831 */
8832 static void
8833f_virtcol(typval_T *argvars, typval_T *rettv)
8834{
8835 colnr_T vcol = 0;
8836 pos_T *fp;
8837 int fnum = curbuf->b_fnum;
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01008838 int len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008839
8840 fp = var2fpos(&argvars[0], FALSE, &fnum);
8841 if (fp != NULL && fp->lnum <= curbuf->b_ml.ml_line_count
8842 && fnum == curbuf->b_fnum)
8843 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01008844 // Limit the column to a valid value, getvvcol() doesn't check.
8845 if (fp->col < 0)
8846 fp->col = 0;
8847 else
8848 {
8849 len = (int)STRLEN(ml_get(fp->lnum));
8850 if (fp->col > len)
8851 fp->col = len;
8852 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008853 getvvcol(curwin, fp, NULL, NULL, &vcol);
8854 ++vcol;
8855 }
8856
8857 rettv->vval.v_number = vcol;
8858}
8859
8860/*
8861 * "visualmode()" function
8862 */
8863 static void
8864f_visualmode(typval_T *argvars, typval_T *rettv)
8865{
8866 char_u str[2];
8867
8868 rettv->v_type = VAR_STRING;
8869 str[0] = curbuf->b_visual_mode_eval;
8870 str[1] = NUL;
8871 rettv->vval.v_string = vim_strsave(str);
8872
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008873 // A non-zero number or non-empty string argument: reset mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008874 if (non_zero_arg(&argvars[0]))
8875 curbuf->b_visual_mode_eval = NUL;
8876}
8877
8878/*
8879 * "wildmenumode()" function
8880 */
8881 static void
8882f_wildmenumode(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8883{
8884#ifdef FEAT_WILDMENU
8885 if (wild_menu_showing)
8886 rettv->vval.v_number = 1;
8887#endif
8888}
8889
8890/*
Bram Moolenaar0c1e3742019-12-27 13:49:24 +01008891 * "windowsversion()" function
8892 */
8893 static void
8894f_windowsversion(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8895{
8896 rettv->v_type = VAR_STRING;
8897 rettv->vval.v_string = vim_strsave((char_u *)windowsVersion);
8898}
8899
8900/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008901 * "wordcount()" function
8902 */
8903 static void
8904f_wordcount(typval_T *argvars UNUSED, typval_T *rettv)
8905{
8906 if (rettv_dict_alloc(rettv) == FAIL)
8907 return;
8908 cursor_pos_info(rettv->vval.v_dict);
8909}
8910
8911/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008912 * "xor(expr, expr)" function
8913 */
8914 static void
8915f_xor(typval_T *argvars, typval_T *rettv)
8916{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008917 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
8918 ^ tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008919}
8920
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008921#endif // FEAT_EVAL