blob: c499890d9b192f2be93ee1daf7fb336963badafb [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)},
Bram Moolenaar03a9f842020-05-13 13:40:16 +0200707 {"popup_clear", 0, 1, 0, ret_void, PROP_FUNC(f_popup_clear)},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100708 {"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)},
Bram Moolenaaref6b9792020-05-13 16:34:15 +0200718 {"popup_list", 0, 0, 0, ret_list_number, PROP_FUNC(f_popup_list)},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100719 {"popup_locate", 2, 2, 0, ret_number, PROP_FUNC(f_popup_locate)},
720 {"popup_menu", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_menu)},
721 {"popup_move", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_move)},
722 {"popup_notification", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_notification)},
723 {"popup_setoptions", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_setoptions)},
724 {"popup_settext", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_settext)},
725 {"popup_show", 1, 1, FEARG_1, ret_void, PROP_FUNC(f_popup_show)},
726 {"pow", 2, 2, FEARG_1, ret_float, FLOAT_FUNC(f_pow)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100727 {"prevnonblank", 1, 1, FEARG_1, ret_number, f_prevnonblank},
728 {"printf", 1, 19, FEARG_2, ret_string, f_printf},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100729 {"prompt_setcallback", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_prompt_setcallback)},
730 {"prompt_setinterrupt", 2, 2, FEARG_1,ret_void, JOB_FUNC(f_prompt_setinterrupt)},
731 {"prompt_setprompt", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_prompt_setprompt)},
732 {"prop_add", 3, 3, FEARG_1, ret_void, PROP_FUNC(f_prop_add)},
733 {"prop_clear", 1, 3, FEARG_1, ret_void, PROP_FUNC(f_prop_clear)},
734 {"prop_find", 1, 2, FEARG_1, ret_dict_any, PROP_FUNC(f_prop_find)},
735 {"prop_list", 1, 2, FEARG_1, ret_list_dict_any, PROP_FUNC(f_prop_list)},
736 {"prop_remove", 1, 3, FEARG_1, ret_number, PROP_FUNC(f_prop_remove)},
737 {"prop_type_add", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_prop_type_add)},
738 {"prop_type_change", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_prop_type_change)},
739 {"prop_type_delete", 1, 2, FEARG_1, ret_void, PROP_FUNC(f_prop_type_delete)},
740 {"prop_type_get", 1, 2, FEARG_1, ret_dict_any, PROP_FUNC(f_prop_type_get)},
741 {"prop_type_list", 0, 1, FEARG_1, ret_list_string, PROP_FUNC(f_prop_type_list)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100742 {"pum_getpos", 0, 0, 0, ret_dict_number, f_pum_getpos},
743 {"pumvisible", 0, 0, 0, ret_number, f_pumvisible},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100744 {"py3eval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200745#ifdef FEAT_PYTHON3
Bram Moolenaar15c47602020-03-26 22:16:48 +0100746 f_py3eval
747#else
748 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200749#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100750 },
751 {"pyeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200752#ifdef FEAT_PYTHON
Bram Moolenaar15c47602020-03-26 22:16:48 +0100753 f_pyeval
754#else
755 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200756#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100757 },
758 {"pyxeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100759#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
Bram Moolenaar15c47602020-03-26 22:16:48 +0100760 f_pyxeval
761#else
762 NULL
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100763#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100764 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100765 {"rand", 0, 1, FEARG_1, ret_number, f_rand},
766 {"range", 1, 3, FEARG_1, ret_list_number, f_range},
767 {"readdir", 1, 2, FEARG_1, ret_list_string, f_readdir},
768 {"readfile", 1, 3, FEARG_1, ret_any, f_readfile},
769 {"reg_executing", 0, 0, 0, ret_string, f_reg_executing},
770 {"reg_recording", 0, 0, 0, ret_string, f_reg_recording},
771 {"reltime", 0, 2, FEARG_1, ret_list_any, f_reltime},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100772 {"reltimefloat", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_reltimefloat)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100773 {"reltimestr", 1, 1, FEARG_1, ret_string, f_reltimestr},
774 {"remote_expr", 2, 4, FEARG_1, ret_string, f_remote_expr},
775 {"remote_foreground", 1, 1, FEARG_1, ret_string, f_remote_foreground},
776 {"remote_peek", 1, 2, FEARG_1, ret_number, f_remote_peek},
777 {"remote_read", 1, 2, FEARG_1, ret_string, f_remote_read},
778 {"remote_send", 2, 3, FEARG_1, ret_string, f_remote_send},
779 {"remote_startserver", 1, 1, FEARG_1, ret_void, f_remote_startserver},
780 {"remove", 2, 3, FEARG_1, ret_any, f_remove},
781 {"rename", 2, 2, FEARG_1, ret_number, f_rename},
782 {"repeat", 2, 2, FEARG_1, ret_any, f_repeat},
783 {"resolve", 1, 1, FEARG_1, ret_string, f_resolve},
784 {"reverse", 1, 1, FEARG_1, ret_any, f_reverse},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100785 {"round", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_round)},
786 {"rubyeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100787#ifdef FEAT_RUBY
Bram Moolenaar15c47602020-03-26 22:16:48 +0100788 f_rubyeval
789#else
790 NULL
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100791#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100792 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100793 {"screenattr", 2, 2, FEARG_1, ret_number, f_screenattr},
794 {"screenchar", 2, 2, FEARG_1, ret_number, f_screenchar},
795 {"screenchars", 2, 2, FEARG_1, ret_list_number, f_screenchars},
796 {"screencol", 0, 0, 0, ret_number, f_screencol},
797 {"screenpos", 3, 3, FEARG_1, ret_dict_number, f_screenpos},
798 {"screenrow", 0, 0, 0, ret_number, f_screenrow},
799 {"screenstring", 2, 2, FEARG_1, ret_string, f_screenstring},
800 {"search", 1, 4, FEARG_1, ret_number, f_search},
801 {"searchdecl", 1, 3, FEARG_1, ret_number, f_searchdecl},
802 {"searchpair", 3, 7, 0, ret_number, f_searchpair},
803 {"searchpairpos", 3, 7, 0, ret_list_number, f_searchpairpos},
804 {"searchpos", 1, 4, FEARG_1, ret_list_number, f_searchpos},
805 {"server2client", 2, 2, FEARG_1, ret_number, f_server2client},
806 {"serverlist", 0, 0, 0, ret_string, f_serverlist},
807 {"setbufline", 3, 3, FEARG_3, ret_number, f_setbufline},
808 {"setbufvar", 3, 3, FEARG_3, ret_void, f_setbufvar},
809 {"setcharsearch", 1, 1, FEARG_1, ret_void, f_setcharsearch},
810 {"setcmdpos", 1, 1, FEARG_1, ret_number, f_setcmdpos},
811 {"setenv", 2, 2, FEARG_2, ret_void, f_setenv},
812 {"setfperm", 2, 2, FEARG_1, ret_number, f_setfperm},
813 {"setline", 2, 2, FEARG_2, ret_number, f_setline},
814 {"setloclist", 2, 4, FEARG_2, ret_number, f_setloclist},
815 {"setmatches", 1, 2, FEARG_1, ret_number, f_setmatches},
816 {"setpos", 2, 2, FEARG_2, ret_number, f_setpos},
817 {"setqflist", 1, 3, FEARG_1, ret_number, f_setqflist},
818 {"setreg", 2, 3, FEARG_2, ret_number, f_setreg},
819 {"settabvar", 3, 3, FEARG_3, ret_void, f_settabvar},
820 {"settabwinvar", 4, 4, FEARG_4, ret_void, f_settabwinvar},
821 {"settagstack", 2, 3, FEARG_2, ret_number, f_settagstack},
822 {"setwinvar", 3, 3, FEARG_3, ret_void, f_setwinvar},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100823 {"sha256", 1, 1, FEARG_1, ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200824#ifdef FEAT_CRYPT
Bram Moolenaar15c47602020-03-26 22:16:48 +0100825 f_sha256
826#else
827 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200828#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100829 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100830 {"shellescape", 1, 2, FEARG_1, ret_string, f_shellescape},
831 {"shiftwidth", 0, 1, FEARG_1, ret_number, f_shiftwidth},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100832 {"sign_define", 1, 2, FEARG_1, ret_any, SIGN_FUNC(f_sign_define)},
833 {"sign_getdefined", 0, 1, FEARG_1, ret_list_dict_any, SIGN_FUNC(f_sign_getdefined)},
834 {"sign_getplaced", 0, 2, FEARG_1, ret_list_dict_any, SIGN_FUNC(f_sign_getplaced)},
835 {"sign_jump", 3, 3, FEARG_1, ret_number, SIGN_FUNC(f_sign_jump)},
836 {"sign_place", 4, 5, FEARG_1, ret_number, SIGN_FUNC(f_sign_place)},
837 {"sign_placelist", 1, 1, FEARG_1, ret_list_number, SIGN_FUNC(f_sign_placelist)},
838 {"sign_undefine", 0, 1, FEARG_1, ret_number, SIGN_FUNC(f_sign_undefine)},
839 {"sign_unplace", 1, 2, FEARG_1, ret_number, SIGN_FUNC(f_sign_unplace)},
840 {"sign_unplacelist", 1, 2, FEARG_1, ret_list_number, SIGN_FUNC(f_sign_unplacelist)},
Bram Moolenaar7035fd92020-04-08 20:03:52 +0200841 {"simplify", 1, 1, FEARG_1, ret_string, f_simplify},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100842 {"sin", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_sin)},
843 {"sinh", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_sinh)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100844 {"sort", 1, 3, FEARG_1, ret_list_any, f_sort},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100845 {"sound_clear", 0, 0, 0, ret_void, SOUND_FUNC(f_sound_clear)},
846 {"sound_playevent", 1, 2, FEARG_1, ret_number, SOUND_FUNC(f_sound_playevent)},
847 {"sound_playfile", 1, 2, FEARG_1, ret_number, SOUND_FUNC(f_sound_playfile)},
848 {"sound_stop", 1, 1, FEARG_1, ret_void, SOUND_FUNC(f_sound_stop)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100849 {"soundfold", 1, 1, FEARG_1, ret_string, f_soundfold},
850 {"spellbadword", 0, 1, FEARG_1, ret_list_string, f_spellbadword},
851 {"spellsuggest", 1, 3, FEARG_1, ret_list_string, f_spellsuggest},
852 {"split", 1, 3, FEARG_1, ret_list_string, f_split},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100853 {"sqrt", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_sqrt)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100854 {"srand", 0, 1, FEARG_1, ret_list_number, f_srand},
855 {"state", 0, 1, FEARG_1, ret_string, f_state},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100856 {"str2float", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_str2float)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100857 {"str2list", 1, 2, FEARG_1, ret_list_number, f_str2list},
858 {"str2nr", 1, 3, FEARG_1, ret_number, f_str2nr},
859 {"strcharpart", 2, 3, FEARG_1, ret_string, f_strcharpart},
860 {"strchars", 1, 2, FEARG_1, ret_number, f_strchars},
861 {"strdisplaywidth", 1, 2, FEARG_1, ret_number, f_strdisplaywidth},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100862 {"strftime", 1, 2, FEARG_1, ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200863#ifdef HAVE_STRFTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +0100864 f_strftime
865#else
866 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200867#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100868 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100869 {"strgetchar", 2, 2, FEARG_1, ret_number, f_strgetchar},
870 {"stridx", 2, 3, FEARG_1, ret_number, f_stridx},
871 {"string", 1, 1, FEARG_1, ret_string, f_string},
872 {"strlen", 1, 1, FEARG_1, ret_number, f_strlen},
873 {"strpart", 2, 3, FEARG_1, ret_string, f_strpart},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100874 {"strptime", 2, 2, FEARG_1, ret_number,
Bram Moolenaar10455d42019-11-21 15:36:18 +0100875#ifdef HAVE_STRPTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +0100876 f_strptime
877#else
878 NULL
Bram Moolenaar10455d42019-11-21 15:36:18 +0100879#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100880 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100881 {"strridx", 2, 3, FEARG_1, ret_number, f_strridx},
882 {"strtrans", 1, 1, FEARG_1, ret_string, f_strtrans},
883 {"strwidth", 1, 1, FEARG_1, ret_number, f_strwidth},
884 {"submatch", 1, 2, FEARG_1, ret_string, f_submatch},
885 {"substitute", 4, 4, FEARG_1, ret_string, f_substitute},
886 {"swapinfo", 1, 1, FEARG_1, ret_dict_any, f_swapinfo},
887 {"swapname", 1, 1, FEARG_1, ret_string, f_swapname},
888 {"synID", 3, 3, 0, ret_number, f_synID},
889 {"synIDattr", 2, 3, FEARG_1, ret_string, f_synIDattr},
890 {"synIDtrans", 1, 1, FEARG_1, ret_number, f_synIDtrans},
891 {"synconcealed", 2, 2, 0, ret_list_any, f_synconcealed},
892 {"synstack", 2, 2, 0, ret_list_number, f_synstack},
893 {"system", 1, 2, FEARG_1, ret_string, f_system},
894 {"systemlist", 1, 2, FEARG_1, ret_list_string, f_systemlist},
895 {"tabpagebuflist", 0, 1, FEARG_1, ret_list_number, f_tabpagebuflist},
896 {"tabpagenr", 0, 1, 0, ret_number, f_tabpagenr},
897 {"tabpagewinnr", 1, 2, FEARG_1, ret_number, f_tabpagewinnr},
898 {"tagfiles", 0, 0, 0, ret_list_string, f_tagfiles},
899 {"taglist", 1, 2, FEARG_1, ret_list_dict_any, f_taglist},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100900 {"tan", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_tan)},
901 {"tanh", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_tanh)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100902 {"tempname", 0, 0, 0, ret_string, f_tempname},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100903 {"term_dumpdiff", 2, 3, FEARG_1, ret_number, TERM_FUNC(f_term_dumpdiff)},
904 {"term_dumpload", 1, 2, FEARG_1, ret_number, TERM_FUNC(f_term_dumpload)},
905 {"term_dumpwrite", 2, 3, FEARG_2, ret_void, TERM_FUNC(f_term_dumpwrite)},
906 {"term_getaltscreen", 1, 1, FEARG_1, ret_number, TERM_FUNC(f_term_getaltscreen)},
907 {"term_getansicolors", 1, 1, FEARG_1, ret_list_string,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +0100908#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +0100909 f_term_getansicolors
910#else
911 NULL
Bram Moolenaarc6df10e2017-07-29 20:15:08 +0200912#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100913 },
914 {"term_getattr", 2, 2, FEARG_1, ret_number, TERM_FUNC(f_term_getattr)},
915 {"term_getcursor", 1, 1, FEARG_1, ret_list_any, TERM_FUNC(f_term_getcursor)},
916 {"term_getjob", 1, 1, FEARG_1, ret_job, TERM_FUNC(f_term_getjob)},
917 {"term_getline", 2, 2, FEARG_1, ret_string, TERM_FUNC(f_term_getline)},
918 {"term_getscrolled", 1, 1, FEARG_1, ret_number, TERM_FUNC(f_term_getscrolled)},
919 {"term_getsize", 1, 1, FEARG_1, ret_list_number, TERM_FUNC(f_term_getsize)},
920 {"term_getstatus", 1, 1, FEARG_1, ret_string, TERM_FUNC(f_term_getstatus)},
921 {"term_gettitle", 1, 1, FEARG_1, ret_string, TERM_FUNC(f_term_gettitle)},
922 {"term_gettty", 1, 2, FEARG_1, ret_string, TERM_FUNC(f_term_gettty)},
923 {"term_list", 0, 0, 0, ret_list_number, TERM_FUNC(f_term_list)},
924 {"term_scrape", 2, 2, FEARG_1, ret_list_dict_any, TERM_FUNC(f_term_scrape)},
925 {"term_sendkeys", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_sendkeys)},
926 {"term_setansicolors", 2, 2, FEARG_1, ret_void,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +0100927#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +0100928 f_term_setansicolors
929#else
930 NULL
931#endif
932 },
933 {"term_setapi", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_setapi)},
934 {"term_setkill", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_setkill)},
935 {"term_setrestore", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_setrestore)},
936 {"term_setsize", 3, 3, FEARG_1, ret_void, TERM_FUNC(f_term_setsize)},
937 {"term_start", 1, 2, FEARG_1, ret_number, TERM_FUNC(f_term_start)},
938 {"term_wait", 1, 2, FEARG_1, ret_void, TERM_FUNC(f_term_wait)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100939 {"test_alloc_fail", 3, 3, FEARG_1, ret_void, f_test_alloc_fail},
940 {"test_autochdir", 0, 0, 0, ret_void, f_test_autochdir},
941 {"test_feedinput", 1, 1, FEARG_1, ret_void, f_test_feedinput},
942 {"test_garbagecollect_now", 0, 0, 0, ret_void, f_test_garbagecollect_now},
943 {"test_garbagecollect_soon", 0, 0, 0, ret_void, f_test_garbagecollect_soon},
944 {"test_getvalue", 1, 1, FEARG_1, ret_number, f_test_getvalue},
945 {"test_ignore_error", 1, 1, FEARG_1, ret_void, f_test_ignore_error},
946 {"test_null_blob", 0, 0, 0, ret_blob, f_test_null_blob},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100947 {"test_null_channel", 0, 0, 0, ret_channel, JOB_FUNC(f_test_null_channel)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100948 {"test_null_dict", 0, 0, 0, ret_dict_any, f_test_null_dict},
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200949 {"test_null_function", 0, 0, 0, ret_func_any, f_test_null_function},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100950 {"test_null_job", 0, 0, 0, ret_job, JOB_FUNC(f_test_null_job)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100951 {"test_null_list", 0, 0, 0, ret_list_any, f_test_null_list},
Bram Moolenaard77a8522020-04-03 21:59:57 +0200952 {"test_null_partial", 0, 0, 0, ret_func_any, f_test_null_partial},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100953 {"test_null_string", 0, 0, 0, ret_string, f_test_null_string},
954 {"test_option_not_set", 1, 1, FEARG_1,ret_void, f_test_option_not_set},
955 {"test_override", 2, 2, FEARG_2, ret_void, f_test_override},
956 {"test_refcount", 1, 1, FEARG_1, ret_number, f_test_refcount},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100957 {"test_scrollbar", 3, 3, FEARG_2, ret_void,
Bram Moolenaarab186732018-09-14 21:27:06 +0200958#ifdef FEAT_GUI
Bram Moolenaar15c47602020-03-26 22:16:48 +0100959 f_test_scrollbar
960#else
961 NULL
Bram Moolenaarab186732018-09-14 21:27:06 +0200962#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100963 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100964 {"test_setmouse", 2, 2, 0, ret_void, f_test_setmouse},
965 {"test_settime", 1, 1, FEARG_1, ret_void, f_test_settime},
966 {"test_srand_seed", 0, 1, FEARG_1, ret_void, f_test_srand_seed},
967 {"test_unknown", 0, 0, 0, ret_any, f_test_unknown},
968 {"test_void", 0, 0, 0, ret_any, f_test_void},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100969 {"timer_info", 0, 1, FEARG_1, ret_list_dict_any, TIMER_FUNC(f_timer_info)},
970 {"timer_pause", 2, 2, FEARG_1, ret_void, TIMER_FUNC(f_timer_pause)},
971 {"timer_start", 2, 3, FEARG_1, ret_number, TIMER_FUNC(f_timer_start)},
972 {"timer_stop", 1, 1, FEARG_1, ret_void, TIMER_FUNC(f_timer_stop)},
973 {"timer_stopall", 0, 0, 0, ret_void, TIMER_FUNC(f_timer_stopall)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100974 {"tolower", 1, 1, FEARG_1, ret_string, f_tolower},
975 {"toupper", 1, 1, FEARG_1, ret_string, f_toupper},
976 {"tr", 3, 3, FEARG_1, ret_string, f_tr},
977 {"trim", 1, 2, FEARG_1, ret_string, f_trim},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100978 {"trunc", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_trunc)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100979 {"type", 1, 1, FEARG_1, ret_number, f_type},
980 {"undofile", 1, 1, FEARG_1, ret_string, f_undofile},
981 {"undotree", 0, 0, 0, ret_dict_any, f_undotree},
982 {"uniq", 1, 3, FEARG_1, ret_list_any, f_uniq},
983 {"values", 1, 1, FEARG_1, ret_list_any, f_values},
984 {"virtcol", 1, 1, FEARG_1, ret_number, f_virtcol},
985 {"visualmode", 0, 1, 0, ret_string, f_visualmode},
986 {"wildmenumode", 0, 0, 0, ret_number, f_wildmenumode},
987 {"win_execute", 2, 3, FEARG_2, ret_string, f_win_execute},
988 {"win_findbuf", 1, 1, FEARG_1, ret_list_number, f_win_findbuf},
989 {"win_getid", 0, 2, FEARG_1, ret_number, f_win_getid},
990 {"win_gettype", 0, 1, FEARG_1, ret_string, f_win_gettype},
991 {"win_gotoid", 1, 1, FEARG_1, ret_number, f_win_gotoid},
992 {"win_id2tabwin", 1, 1, FEARG_1, ret_list_number, f_win_id2tabwin},
993 {"win_id2win", 1, 1, FEARG_1, ret_number, f_win_id2win},
994 {"win_screenpos", 1, 1, FEARG_1, ret_list_number, f_win_screenpos},
995 {"win_splitmove", 2, 3, FEARG_1, ret_number, f_win_splitmove},
996 {"winbufnr", 1, 1, FEARG_1, ret_number, f_winbufnr},
997 {"wincol", 0, 0, 0, ret_number, f_wincol},
998 {"windowsversion", 0, 0, 0, ret_string, f_windowsversion},
999 {"winheight", 1, 1, FEARG_1, ret_number, f_winheight},
1000 {"winlayout", 0, 1, FEARG_1, ret_list_any, f_winlayout},
1001 {"winline", 0, 0, 0, ret_number, f_winline},
1002 {"winnr", 0, 1, FEARG_1, ret_number, f_winnr},
1003 {"winrestcmd", 0, 0, 0, ret_string, f_winrestcmd},
1004 {"winrestview", 1, 1, FEARG_1, ret_void, f_winrestview},
1005 {"winsaveview", 0, 0, 0, ret_dict_any, f_winsaveview},
1006 {"winwidth", 1, 1, FEARG_1, ret_number, f_winwidth},
1007 {"wordcount", 0, 0, 0, ret_dict_number, f_wordcount},
1008 {"writefile", 2, 3, FEARG_1, ret_number, f_writefile},
1009 {"xor", 2, 2, FEARG_1, ret_number, f_xor},
Bram Moolenaarac92e252019-08-03 21:58:38 +02001010};
1011
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001012/*
1013 * Function given to ExpandGeneric() to obtain the list of internal
1014 * or user defined function names.
1015 */
1016 char_u *
1017get_function_name(expand_T *xp, int idx)
1018{
1019 static int intidx = -1;
1020 char_u *name;
1021
1022 if (idx == 0)
1023 intidx = -1;
1024 if (intidx < 0)
1025 {
1026 name = get_user_func_name(xp, idx);
1027 if (name != NULL)
1028 return name;
1029 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001030 if (++intidx < (int)(sizeof(global_functions) / sizeof(funcentry_T)))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001031 {
Bram Moolenaarac92e252019-08-03 21:58:38 +02001032 STRCPY(IObuff, global_functions[intidx].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001033 STRCAT(IObuff, "(");
Bram Moolenaarac92e252019-08-03 21:58:38 +02001034 if (global_functions[intidx].f_max_argc == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001035 STRCAT(IObuff, ")");
1036 return IObuff;
1037 }
1038
1039 return NULL;
1040}
1041
1042/*
1043 * Function given to ExpandGeneric() to obtain the list of internal or
1044 * user defined variable or function names.
1045 */
1046 char_u *
1047get_expr_name(expand_T *xp, int idx)
1048{
1049 static int intidx = -1;
1050 char_u *name;
1051
1052 if (idx == 0)
1053 intidx = -1;
1054 if (intidx < 0)
1055 {
1056 name = get_function_name(xp, idx);
1057 if (name != NULL)
1058 return name;
1059 }
1060 return get_user_var_name(xp, ++intidx);
1061}
1062
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001063/*
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001064 * Find internal function "name" in table "global_functions".
Bram Moolenaar15c47602020-03-26 22:16:48 +01001065 * Return index, or -1 if not found or "implemented" is TRUE and the function
1066 * is not implemented.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001067 */
Bram Moolenaar15c47602020-03-26 22:16:48 +01001068 static int
1069find_internal_func_opt(char_u *name, int implemented)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001070{
1071 int first = 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001072 int last;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001073 int cmp;
1074 int x;
1075
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001076 last = (int)(sizeof(global_functions) / sizeof(funcentry_T)) - 1;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001077
1078 // Find the function name in the table. Binary search.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001079 while (first <= last)
1080 {
1081 x = first + ((unsigned)(last - first) >> 1);
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001082 cmp = STRCMP(name, global_functions[x].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001083 if (cmp < 0)
1084 last = x - 1;
1085 else if (cmp > 0)
1086 first = x + 1;
Bram Moolenaar15c47602020-03-26 22:16:48 +01001087 else if (implemented && global_functions[x].f_func == NULL)
1088 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001089 else
1090 return x;
1091 }
1092 return -1;
1093}
1094
Bram Moolenaar15c47602020-03-26 22:16:48 +01001095/*
1096 * Find internal function "name" in table "global_functions".
1097 * Return index, or -1 if not found or the function is not implemented.
1098 */
1099 int
1100find_internal_func(char_u *name)
1101{
1102 return find_internal_func_opt(name, TRUE);
1103}
1104
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001105 int
Bram Moolenaarac92e252019-08-03 21:58:38 +02001106has_internal_func(char_u *name)
1107{
Bram Moolenaar15c47602020-03-26 22:16:48 +01001108 return find_internal_func_opt(name, TRUE) >= 0;
1109}
1110
1111 static int
1112has_internal_func_name(char_u *name)
1113{
1114 return find_internal_func_opt(name, FALSE) >= 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001115}
1116
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001117 char *
1118internal_func_name(int idx)
1119{
1120 return global_functions[idx].f_name;
1121}
1122
1123 type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001124internal_func_ret_type(int idx, int argcount, type_T **argtypes)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001125{
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001126 return global_functions[idx].f_retfunc(argcount, argtypes);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001127}
1128
1129/*
1130 * Check the argument count to use for internal function "idx".
1131 * Returns OK or FAIL;
1132 */
1133 int
1134check_internal_func(int idx, int argcount)
1135{
1136 int res;
1137 char *name;
1138
1139 if (argcount < global_functions[idx].f_min_argc)
1140 res = FCERR_TOOFEW;
1141 else if (argcount > global_functions[idx].f_max_argc)
1142 res = FCERR_TOOMANY;
1143 else
1144 return OK;
1145
1146 name = internal_func_name(idx);
1147 if (res == FCERR_TOOMANY)
1148 semsg(_(e_toomanyarg), name);
1149 else
1150 semsg(_(e_toofewarg), name);
1151 return FAIL;
1152}
1153
Bram Moolenaarac92e252019-08-03 21:58:38 +02001154 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001155call_internal_func(
1156 char_u *name,
1157 int argcount,
1158 typval_T *argvars,
1159 typval_T *rettv)
1160{
1161 int i;
1162
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001163 i = find_internal_func(name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001164 if (i < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001165 return FCERR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001166 if (argcount < global_functions[i].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001167 return FCERR_TOOFEW;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001168 if (argcount > global_functions[i].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001169 return FCERR_TOOMANY;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001170 argvars[argcount].v_type = VAR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001171 global_functions[i].f_func(argvars, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001172 return FCERR_NONE;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001173}
1174
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001175 void
1176call_internal_func_by_idx(
1177 int idx,
1178 typval_T *argvars,
1179 typval_T *rettv)
1180{
1181 global_functions[idx].f_func(argvars, rettv);
1182}
1183
Bram Moolenaarac92e252019-08-03 21:58:38 +02001184/*
1185 * Invoke a method for base->method().
1186 */
1187 int
1188call_internal_method(
1189 char_u *name,
1190 int argcount,
1191 typval_T *argvars,
1192 typval_T *rettv,
1193 typval_T *basetv)
1194{
1195 int i;
1196 int fi;
1197 typval_T argv[MAX_FUNC_ARGS + 1];
1198
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001199 fi = find_internal_func(name);
Bram Moolenaar91746392019-08-16 22:22:31 +02001200 if (fi < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001201 return FCERR_UNKNOWN;
Bram Moolenaar91746392019-08-16 22:22:31 +02001202 if (global_functions[fi].f_argtype == 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001203 return FCERR_NOTMETHOD;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001204 if (argcount + 1 < global_functions[fi].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001205 return FCERR_TOOFEW;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001206 if (argcount + 1 > global_functions[fi].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001207 return FCERR_TOOMANY;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001208
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001209 if (global_functions[fi].f_argtype == FEARG_LAST)
Bram Moolenaar25e42232019-08-04 15:04:10 +02001210 {
1211 // base value goes last
1212 for (i = 0; i < argcount; ++i)
1213 argv[i] = argvars[i];
1214 argv[argcount] = *basetv;
1215 }
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001216 else if (global_functions[fi].f_argtype == FEARG_2)
Bram Moolenaar25e42232019-08-04 15:04:10 +02001217 {
1218 // base value goes second
1219 argv[0] = argvars[0];
1220 argv[1] = *basetv;
1221 for (i = 1; i < argcount; ++i)
1222 argv[i + 1] = argvars[i];
1223 }
Bram Moolenaar24278d22019-08-16 21:49:22 +02001224 else if (global_functions[fi].f_argtype == FEARG_3)
1225 {
1226 // base value goes third
1227 argv[0] = argvars[0];
1228 argv[1] = argvars[1];
1229 argv[2] = *basetv;
1230 for (i = 2; i < argcount; ++i)
1231 argv[i + 1] = argvars[i];
1232 }
Bram Moolenaaraad222c2019-09-06 22:46:09 +02001233 else if (global_functions[fi].f_argtype == FEARG_4)
1234 {
1235 // base value goes fourth
1236 argv[0] = argvars[0];
1237 argv[1] = argvars[1];
1238 argv[2] = argvars[2];
1239 argv[3] = *basetv;
1240 for (i = 3; i < argcount; ++i)
1241 argv[i + 1] = argvars[i];
1242 }
Bram Moolenaar25e42232019-08-04 15:04:10 +02001243 else
1244 {
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001245 // FEARG_1: base value goes first
Bram Moolenaar25e42232019-08-04 15:04:10 +02001246 argv[0] = *basetv;
1247 for (i = 0; i < argcount; ++i)
1248 argv[i + 1] = argvars[i];
1249 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001250 argv[argcount + 1].v_type = VAR_UNKNOWN;
1251
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001252 global_functions[fi].f_func(argv, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001253 return FCERR_NONE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001254}
1255
1256/*
1257 * Return TRUE for a non-zero Number and a non-empty String.
1258 */
Bram Moolenaar0e57dd82019-09-16 22:56:03 +02001259 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001260non_zero_arg(typval_T *argvars)
1261{
1262 return ((argvars[0].v_type == VAR_NUMBER
1263 && argvars[0].vval.v_number != 0)
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01001264 || (argvars[0].v_type == VAR_BOOL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001265 && argvars[0].vval.v_number == VVAL_TRUE)
1266 || (argvars[0].v_type == VAR_STRING
1267 && argvars[0].vval.v_string != NULL
1268 && *argvars[0].vval.v_string != NUL));
1269}
1270
1271/*
1272 * Get the lnum from the first argument.
1273 * Also accepts ".", "$", etc., but that only works for the current buffer.
1274 * Returns -1 on error.
1275 */
Bram Moolenaarb60d8512019-06-29 07:59:04 +02001276 linenr_T
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001277tv_get_lnum(typval_T *argvars)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001278{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001279 linenr_T lnum;
1280
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001281 lnum = (linenr_T)tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar1f3165b2019-09-04 13:21:26 +02001282 if (lnum == 0) // no valid number, try using arg like line()
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001283 {
Bram Moolenaar1f3165b2019-09-04 13:21:26 +02001284 int fnum;
1285 pos_T *fp = var2fpos(&argvars[0], TRUE, &fnum);
1286
1287 if (fp != NULL)
1288 lnum = fp->lnum;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001289 }
1290 return lnum;
1291}
1292
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001293/*
1294 * Get the lnum from the first argument.
1295 * Also accepts "$", then "buf" is used.
1296 * Returns 0 on error.
1297 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02001298 linenr_T
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001299tv_get_lnum_buf(typval_T *argvars, buf_T *buf)
1300{
1301 if (argvars[0].v_type == VAR_STRING
1302 && argvars[0].vval.v_string != NULL
1303 && argvars[0].vval.v_string[0] == '$'
1304 && buf != NULL)
1305 return buf->b_ml.ml_line_count;
1306 return (linenr_T)tv_get_number_chk(&argvars[0], NULL);
1307}
1308
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001309#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001310/*
1311 * Get the float value of "argvars[0]" into "f".
1312 * Returns FAIL when the argument is not a Number or Float.
1313 */
1314 static int
1315get_float_arg(typval_T *argvars, float_T *f)
1316{
1317 if (argvars[0].v_type == VAR_FLOAT)
1318 {
1319 *f = argvars[0].vval.v_float;
1320 return OK;
1321 }
1322 if (argvars[0].v_type == VAR_NUMBER)
1323 {
1324 *f = (float_T)argvars[0].vval.v_number;
1325 return OK;
1326 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001327 emsg(_("E808: Number or Float required"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001328 return FAIL;
1329}
1330
1331/*
1332 * "abs(expr)" function
1333 */
1334 static void
1335f_abs(typval_T *argvars, typval_T *rettv)
1336{
1337 if (argvars[0].v_type == VAR_FLOAT)
1338 {
1339 rettv->v_type = VAR_FLOAT;
1340 rettv->vval.v_float = fabs(argvars[0].vval.v_float);
1341 }
1342 else
1343 {
1344 varnumber_T n;
1345 int error = FALSE;
1346
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001347 n = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001348 if (error)
1349 rettv->vval.v_number = -1;
1350 else if (n > 0)
1351 rettv->vval.v_number = n;
1352 else
1353 rettv->vval.v_number = -n;
1354 }
1355}
1356
1357/*
1358 * "acos()" function
1359 */
1360 static void
1361f_acos(typval_T *argvars, typval_T *rettv)
1362{
1363 float_T f = 0.0;
1364
1365 rettv->v_type = VAR_FLOAT;
1366 if (get_float_arg(argvars, &f) == OK)
1367 rettv->vval.v_float = acos(f);
1368 else
1369 rettv->vval.v_float = 0.0;
1370}
1371#endif
1372
1373/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001374 * "and(expr, expr)" function
1375 */
1376 static void
1377f_and(typval_T *argvars, typval_T *rettv)
1378{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001379 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
1380 & tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaarca851592018-06-06 21:04:07 +02001381}
1382
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001383#ifdef FEAT_FLOAT
1384/*
1385 * "asin()" function
1386 */
1387 static void
1388f_asin(typval_T *argvars, typval_T *rettv)
1389{
1390 float_T f = 0.0;
1391
1392 rettv->v_type = VAR_FLOAT;
1393 if (get_float_arg(argvars, &f) == OK)
1394 rettv->vval.v_float = asin(f);
1395 else
1396 rettv->vval.v_float = 0.0;
1397}
1398
1399/*
1400 * "atan()" function
1401 */
1402 static void
1403f_atan(typval_T *argvars, typval_T *rettv)
1404{
1405 float_T f = 0.0;
1406
1407 rettv->v_type = VAR_FLOAT;
1408 if (get_float_arg(argvars, &f) == OK)
1409 rettv->vval.v_float = atan(f);
1410 else
1411 rettv->vval.v_float = 0.0;
1412}
1413
1414/*
1415 * "atan2()" function
1416 */
1417 static void
1418f_atan2(typval_T *argvars, typval_T *rettv)
1419{
1420 float_T fx = 0.0, fy = 0.0;
1421
1422 rettv->v_type = VAR_FLOAT;
1423 if (get_float_arg(argvars, &fx) == OK
1424 && get_float_arg(&argvars[1], &fy) == OK)
1425 rettv->vval.v_float = atan2(fx, fy);
1426 else
1427 rettv->vval.v_float = 0.0;
1428}
1429#endif
1430
1431/*
Bram Moolenaar59716a22017-03-01 20:32:44 +01001432 * "balloon_show()" function
1433 */
1434#ifdef FEAT_BEVAL
1435 static void
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02001436f_balloon_gettext(typval_T *argvars UNUSED, typval_T *rettv)
1437{
1438 rettv->v_type = VAR_STRING;
1439 if (balloonEval != NULL)
1440 {
1441 if (balloonEval->msg == NULL)
1442 rettv->vval.v_string = NULL;
1443 else
1444 rettv->vval.v_string = vim_strsave(balloonEval->msg);
1445 }
1446}
1447
1448 static void
Bram Moolenaar59716a22017-03-01 20:32:44 +01001449f_balloon_show(typval_T *argvars, typval_T *rettv UNUSED)
1450{
Bram Moolenaarcaf64342017-03-02 22:11:33 +01001451 if (balloonEval != NULL)
Bram Moolenaar246fe032017-11-19 19:56:27 +01001452 {
1453 if (argvars[0].v_type == VAR_LIST
1454# ifdef FEAT_GUI
1455 && !gui.in_use
1456# endif
1457 )
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02001458 {
1459 list_T *l = argvars[0].vval.v_list;
1460
1461 // empty list removes the balloon
1462 post_balloon(balloonEval, NULL,
1463 l == NULL || l->lv_len == 0 ? NULL : l);
1464 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01001465 else
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02001466 {
1467 char_u *mesg = tv_get_string_chk(&argvars[0]);
1468
1469 if (mesg != NULL)
1470 // empty string removes the balloon
1471 post_balloon(balloonEval, *mesg == NUL ? NULL : mesg, NULL);
1472 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01001473 }
1474}
1475
Bram Moolenaar669a8282017-11-19 20:13:05 +01001476# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +01001477 static void
1478f_balloon_split(typval_T *argvars, typval_T *rettv UNUSED)
1479{
1480 if (rettv_list_alloc(rettv) == OK)
1481 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001482 char_u *msg = tv_get_string_chk(&argvars[0]);
Bram Moolenaar246fe032017-11-19 19:56:27 +01001483
1484 if (msg != NULL)
1485 {
1486 pumitem_T *array;
1487 int size = split_message(msg, &array);
1488 int i;
1489
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001490 // Skip the first and last item, they are always empty.
Bram Moolenaar246fe032017-11-19 19:56:27 +01001491 for (i = 1; i < size - 1; ++i)
1492 list_append_string(rettv->vval.v_list, array[i].pum_text, -1);
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01001493 while (size > 0)
1494 vim_free(array[--size].pum_text);
Bram Moolenaar246fe032017-11-19 19:56:27 +01001495 vim_free(array);
1496 }
1497 }
Bram Moolenaar59716a22017-03-01 20:32:44 +01001498}
Bram Moolenaar669a8282017-11-19 20:13:05 +01001499# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +01001500#endif
1501
1502/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001503 * Get buffer by number or pattern.
1504 */
Bram Moolenaarc6df10e2017-07-29 20:15:08 +02001505 buf_T *
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01001506tv_get_buf(typval_T *tv, int curtab_only)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001507{
1508 char_u *name = tv->vval.v_string;
1509 buf_T *buf;
1510
1511 if (tv->v_type == VAR_NUMBER)
1512 return buflist_findnr((int)tv->vval.v_number);
1513 if (tv->v_type != VAR_STRING)
1514 return NULL;
1515 if (name == NULL || *name == NUL)
1516 return curbuf;
1517 if (name[0] == '$' && name[1] == NUL)
1518 return lastbuf;
1519
1520 buf = buflist_find_by_name(name, curtab_only);
1521
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001522 // If not found, try expanding the name, like done for bufexists().
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001523 if (buf == NULL)
1524 buf = find_buffer(tv);
1525
1526 return buf;
1527}
1528
1529/*
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01001530 * Get the buffer from "arg" and give an error and return NULL if it is not
1531 * valid.
1532 */
Bram Moolenaara3347722019-05-11 21:14:24 +02001533 buf_T *
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01001534get_buf_arg(typval_T *arg)
1535{
1536 buf_T *buf;
1537
1538 ++emsg_off;
1539 buf = tv_get_buf(arg, FALSE);
1540 --emsg_off;
1541 if (buf == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001542 semsg(_("E158: Invalid buffer name: %s"), tv_get_string(arg));
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01001543 return buf;
1544}
1545
1546/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001547 * "byte2line(byte)" function
1548 */
1549 static void
1550f_byte2line(typval_T *argvars UNUSED, typval_T *rettv)
1551{
1552#ifndef FEAT_BYTEOFF
1553 rettv->vval.v_number = -1;
1554#else
1555 long boff = 0;
1556
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001557 boff = tv_get_number(&argvars[0]) - 1; // boff gets -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001558 if (boff < 0)
1559 rettv->vval.v_number = -1;
1560 else
1561 rettv->vval.v_number = ml_find_line_or_offset(curbuf,
1562 (linenr_T)0, &boff);
1563#endif
1564}
1565
1566 static void
1567byteidx(typval_T *argvars, typval_T *rettv, int comp UNUSED)
1568{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001569 char_u *t;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001570 char_u *str;
1571 varnumber_T idx;
1572
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001573 str = tv_get_string_chk(&argvars[0]);
1574 idx = tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001575 rettv->vval.v_number = -1;
1576 if (str == NULL || idx < 0)
1577 return;
1578
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001579 t = str;
1580 for ( ; idx > 0; idx--)
1581 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001582 if (*t == NUL) // EOL reached
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001583 return;
1584 if (enc_utf8 && comp)
1585 t += utf_ptr2len(t);
1586 else
1587 t += (*mb_ptr2len)(t);
1588 }
1589 rettv->vval.v_number = (varnumber_T)(t - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001590}
1591
1592/*
1593 * "byteidx()" function
1594 */
1595 static void
1596f_byteidx(typval_T *argvars, typval_T *rettv)
1597{
1598 byteidx(argvars, rettv, FALSE);
1599}
1600
1601/*
1602 * "byteidxcomp()" function
1603 */
1604 static void
1605f_byteidxcomp(typval_T *argvars, typval_T *rettv)
1606{
1607 byteidx(argvars, rettv, TRUE);
1608}
1609
1610/*
1611 * "call(func, arglist [, dict])" function
1612 */
1613 static void
1614f_call(typval_T *argvars, typval_T *rettv)
1615{
1616 char_u *func;
1617 partial_T *partial = NULL;
1618 dict_T *selfdict = NULL;
1619
1620 if (argvars[1].v_type != VAR_LIST)
1621 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001622 emsg(_(e_listreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001623 return;
1624 }
1625 if (argvars[1].vval.v_list == NULL)
1626 return;
1627
1628 if (argvars[0].v_type == VAR_FUNC)
1629 func = argvars[0].vval.v_string;
1630 else if (argvars[0].v_type == VAR_PARTIAL)
1631 {
1632 partial = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02001633 func = partial_name(partial);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001634 }
1635 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001636 func = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001637 if (*func == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001638 return; // type error or empty name
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001639
1640 if (argvars[2].v_type != VAR_UNKNOWN)
1641 {
1642 if (argvars[2].v_type != VAR_DICT)
1643 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001644 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001645 return;
1646 }
1647 selfdict = argvars[2].vval.v_dict;
1648 }
1649
1650 (void)func_call(func, &argvars[1], partial, selfdict, rettv);
1651}
1652
1653#ifdef FEAT_FLOAT
1654/*
1655 * "ceil({float})" function
1656 */
1657 static void
1658f_ceil(typval_T *argvars, typval_T *rettv)
1659{
1660 float_T f = 0.0;
1661
1662 rettv->v_type = VAR_FLOAT;
1663 if (get_float_arg(argvars, &f) == OK)
1664 rettv->vval.v_float = ceil(f);
1665 else
1666 rettv->vval.v_float = 0.0;
1667}
1668#endif
1669
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001670/*
1671 * "changenr()" function
1672 */
1673 static void
1674f_changenr(typval_T *argvars UNUSED, typval_T *rettv)
1675{
1676 rettv->vval.v_number = curbuf->b_u_seq_cur;
1677}
1678
1679/*
1680 * "char2nr(string)" function
1681 */
1682 static void
1683f_char2nr(typval_T *argvars, typval_T *rettv)
1684{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001685 if (has_mbyte)
1686 {
1687 int utf8 = 0;
1688
1689 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001690 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001691
1692 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01001693 rettv->vval.v_number = utf_ptr2char(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001694 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001695 rettv->vval.v_number = (*mb_ptr2char)(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001696 }
1697 else
Bram Moolenaar13505972019-01-24 15:04:48 +01001698 rettv->vval.v_number = tv_get_string(&argvars[0])[0];
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001699}
1700
Bram Moolenaar29b7d7a2019-07-22 23:03:57 +02001701 win_T *
Bram Moolenaaraff74912019-03-30 18:11:49 +01001702get_optional_window(typval_T *argvars, int idx)
1703{
1704 win_T *win = curwin;
1705
1706 if (argvars[idx].v_type != VAR_UNKNOWN)
1707 {
1708 win = find_win_by_nr_or_id(&argvars[idx]);
1709 if (win == NULL)
1710 {
1711 emsg(_(e_invalwindow));
1712 return NULL;
1713 }
1714 }
1715 return win;
1716}
1717
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001718/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001719 * "col(string)" function
1720 */
1721 static void
1722f_col(typval_T *argvars, typval_T *rettv)
1723{
1724 colnr_T col = 0;
1725 pos_T *fp;
1726 int fnum = curbuf->b_fnum;
1727
1728 fp = var2fpos(&argvars[0], FALSE, &fnum);
1729 if (fp != NULL && fnum == curbuf->b_fnum)
1730 {
1731 if (fp->col == MAXCOL)
1732 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001733 // '> can be MAXCOL, get the length of the line then
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001734 if (fp->lnum <= curbuf->b_ml.ml_line_count)
1735 col = (colnr_T)STRLEN(ml_get(fp->lnum)) + 1;
1736 else
1737 col = MAXCOL;
1738 }
1739 else
1740 {
1741 col = fp->col + 1;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001742 // col(".") when the cursor is on the NUL at the end of the line
1743 // because of "coladd" can be seen as an extra column.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001744 if (virtual_active() && fp == &curwin->w_cursor)
1745 {
1746 char_u *p = ml_get_cursor();
1747
1748 if (curwin->w_cursor.coladd >= (colnr_T)chartabsize(p,
1749 curwin->w_virtcol - curwin->w_cursor.coladd))
1750 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001751 int l;
1752
1753 if (*p != NUL && p[(l = (*mb_ptr2len)(p))] == NUL)
1754 col += l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001755 }
1756 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001757 }
1758 }
1759 rettv->vval.v_number = col;
1760}
1761
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001762/*
1763 * "confirm(message, buttons[, default [, type]])" function
1764 */
1765 static void
1766f_confirm(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1767{
1768#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1769 char_u *message;
1770 char_u *buttons = NULL;
1771 char_u buf[NUMBUFLEN];
1772 char_u buf2[NUMBUFLEN];
1773 int def = 1;
1774 int type = VIM_GENERIC;
1775 char_u *typestr;
1776 int error = FALSE;
1777
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001778 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001779 if (message == NULL)
1780 error = TRUE;
1781 if (argvars[1].v_type != VAR_UNKNOWN)
1782 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001783 buttons = tv_get_string_buf_chk(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001784 if (buttons == NULL)
1785 error = TRUE;
1786 if (argvars[2].v_type != VAR_UNKNOWN)
1787 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001788 def = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001789 if (argvars[3].v_type != VAR_UNKNOWN)
1790 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001791 typestr = tv_get_string_buf_chk(&argvars[3], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001792 if (typestr == NULL)
1793 error = TRUE;
1794 else
1795 {
1796 switch (TOUPPER_ASC(*typestr))
1797 {
1798 case 'E': type = VIM_ERROR; break;
1799 case 'Q': type = VIM_QUESTION; break;
1800 case 'I': type = VIM_INFO; break;
1801 case 'W': type = VIM_WARNING; break;
1802 case 'G': type = VIM_GENERIC; break;
1803 }
1804 }
1805 }
1806 }
1807 }
1808
1809 if (buttons == NULL || *buttons == NUL)
1810 buttons = (char_u *)_("&Ok");
1811
1812 if (!error)
1813 rettv->vval.v_number = do_dialog(type, NULL, message, buttons,
1814 def, NULL, FALSE);
1815#endif
1816}
1817
1818/*
1819 * "copy()" function
1820 */
1821 static void
1822f_copy(typval_T *argvars, typval_T *rettv)
1823{
1824 item_copy(&argvars[0], rettv, FALSE, 0);
1825}
1826
1827#ifdef FEAT_FLOAT
1828/*
1829 * "cos()" function
1830 */
1831 static void
1832f_cos(typval_T *argvars, typval_T *rettv)
1833{
1834 float_T f = 0.0;
1835
1836 rettv->v_type = VAR_FLOAT;
1837 if (get_float_arg(argvars, &f) == OK)
1838 rettv->vval.v_float = cos(f);
1839 else
1840 rettv->vval.v_float = 0.0;
1841}
1842
1843/*
1844 * "cosh()" function
1845 */
1846 static void
1847f_cosh(typval_T *argvars, typval_T *rettv)
1848{
1849 float_T f = 0.0;
1850
1851 rettv->v_type = VAR_FLOAT;
1852 if (get_float_arg(argvars, &f) == OK)
1853 rettv->vval.v_float = cosh(f);
1854 else
1855 rettv->vval.v_float = 0.0;
1856}
1857#endif
1858
1859/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001860 * "cursor(lnum, col)" function, or
1861 * "cursor(list)"
1862 *
1863 * Moves the cursor to the specified line and column.
1864 * Returns 0 when the position could be set, -1 otherwise.
1865 */
1866 static void
1867f_cursor(typval_T *argvars, typval_T *rettv)
1868{
1869 long line, col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001870 long coladd = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001871 int set_curswant = TRUE;
1872
1873 rettv->vval.v_number = -1;
1874 if (argvars[1].v_type == VAR_UNKNOWN)
1875 {
1876 pos_T pos;
1877 colnr_T curswant = -1;
1878
1879 if (list2fpos(argvars, &pos, NULL, &curswant) == FAIL)
1880 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001881 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001882 return;
1883 }
1884 line = pos.lnum;
1885 col = pos.col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001886 coladd = pos.coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001887 if (curswant >= 0)
1888 {
1889 curwin->w_curswant = curswant - 1;
1890 set_curswant = FALSE;
1891 }
1892 }
1893 else
1894 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001895 line = tv_get_lnum(argvars);
1896 col = (long)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001897 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001898 coladd = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001899 }
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01001900 if (line < 0 || col < 0 || coladd < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001901 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001902 if (line > 0)
1903 curwin->w_cursor.lnum = line;
1904 if (col > 0)
1905 curwin->w_cursor.col = col - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001906 curwin->w_cursor.coladd = coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001907
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001908 // Make sure the cursor is in a valid position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001909 check_cursor();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001910 // Correct cursor for multi-byte character.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001911 if (has_mbyte)
1912 mb_adjust_cursor();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001913
1914 curwin->w_set_curswant = set_curswant;
1915 rettv->vval.v_number = 0;
1916}
1917
Bram Moolenaar4f974752019-02-17 17:44:42 +01001918#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02001919/*
1920 * "debugbreak()" function
1921 */
1922 static void
1923f_debugbreak(typval_T *argvars, typval_T *rettv)
1924{
1925 int pid;
1926
1927 rettv->vval.v_number = FAIL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001928 pid = (int)tv_get_number(&argvars[0]);
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02001929 if (pid == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001930 emsg(_(e_invarg));
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02001931 else
1932 {
1933 HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, 0, pid);
1934
1935 if (hProcess != NULL)
1936 {
1937 DebugBreakProcess(hProcess);
1938 CloseHandle(hProcess);
1939 rettv->vval.v_number = OK;
1940 }
1941 }
1942}
1943#endif
1944
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001945/*
1946 * "deepcopy()" function
1947 */
1948 static void
1949f_deepcopy(typval_T *argvars, typval_T *rettv)
1950{
1951 int noref = 0;
1952 int copyID;
1953
1954 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001955 noref = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001956 if (noref < 0 || noref > 1)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001957 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001958 else
1959 {
1960 copyID = get_copyID();
1961 item_copy(&argvars[0], rettv, TRUE, noref == 0 ? copyID : 0);
1962 }
1963}
1964
1965/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001966 * "did_filetype()" function
1967 */
1968 static void
1969f_did_filetype(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1970{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001971 rettv->vval.v_number = did_filetype;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001972}
1973
1974/*
Bram Moolenaar4132eb52020-02-14 16:53:00 +01001975 * "echoraw({expr})" function
1976 */
1977 static void
1978f_echoraw(typval_T *argvars, typval_T *rettv UNUSED)
1979{
1980 char_u *str = tv_get_string_chk(&argvars[0]);
1981
1982 if (str != NULL && *str != NUL)
1983 {
1984 out_str(str);
1985 out_flush();
1986 }
1987}
1988
1989/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001990 * "empty({expr})" function
1991 */
1992 static void
1993f_empty(typval_T *argvars, typval_T *rettv)
1994{
1995 int n = FALSE;
1996
1997 switch (argvars[0].v_type)
1998 {
1999 case VAR_STRING:
2000 case VAR_FUNC:
2001 n = argvars[0].vval.v_string == NULL
2002 || *argvars[0].vval.v_string == NUL;
2003 break;
2004 case VAR_PARTIAL:
2005 n = FALSE;
2006 break;
2007 case VAR_NUMBER:
2008 n = argvars[0].vval.v_number == 0;
2009 break;
2010 case VAR_FLOAT:
2011#ifdef FEAT_FLOAT
2012 n = argvars[0].vval.v_float == 0.0;
2013 break;
2014#endif
2015 case VAR_LIST:
2016 n = argvars[0].vval.v_list == NULL
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002017 || argvars[0].vval.v_list->lv_len == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002018 break;
2019 case VAR_DICT:
2020 n = argvars[0].vval.v_dict == NULL
2021 || argvars[0].vval.v_dict->dv_hashtab.ht_used == 0;
2022 break;
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002023 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002024 case VAR_SPECIAL:
2025 n = argvars[0].vval.v_number != VVAL_TRUE;
2026 break;
2027
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002028 case VAR_BLOB:
2029 n = argvars[0].vval.v_blob == NULL
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002030 || argvars[0].vval.v_blob->bv_ga.ga_len == 0;
2031 break;
2032
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002033 case VAR_JOB:
2034#ifdef FEAT_JOB_CHANNEL
2035 n = argvars[0].vval.v_job == NULL
2036 || argvars[0].vval.v_job->jv_status != JOB_STARTED;
2037 break;
2038#endif
2039 case VAR_CHANNEL:
2040#ifdef FEAT_JOB_CHANNEL
2041 n = argvars[0].vval.v_channel == NULL
2042 || !channel_is_open(argvars[0].vval.v_channel);
2043 break;
2044#endif
2045 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02002046 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002047 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01002048 internal_error_no_abort("f_empty(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002049 n = TRUE;
2050 break;
2051 }
2052
2053 rettv->vval.v_number = n;
2054}
2055
2056/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02002057 * "environ()" function
2058 */
2059 static void
2060f_environ(typval_T *argvars UNUSED, typval_T *rettv)
2061{
2062#if !defined(AMIGA)
2063 int i = 0;
2064 char_u *entry, *value;
2065# ifdef MSWIN
2066 extern wchar_t **_wenviron;
2067# else
2068 extern char **environ;
2069# endif
2070
2071 if (rettv_dict_alloc(rettv) != OK)
2072 return;
2073
2074# ifdef MSWIN
2075 if (*_wenviron == NULL)
2076 return;
2077# else
2078 if (*environ == NULL)
2079 return;
2080# endif
2081
2082 for (i = 0; ; ++i)
2083 {
2084# ifdef MSWIN
2085 short_u *p;
2086
2087 if ((p = (short_u *)_wenviron[i]) == NULL)
2088 return;
2089 entry = utf16_to_enc(p, NULL);
2090# else
2091 if ((entry = (char_u *)environ[i]) == NULL)
2092 return;
2093 entry = vim_strsave(entry);
2094# endif
2095 if (entry == NULL) // out of memory
2096 return;
2097 if ((value = vim_strchr(entry, '=')) == NULL)
2098 {
2099 vim_free(entry);
2100 continue;
2101 }
2102 *value++ = NUL;
2103 dict_add_string(rettv->vval.v_dict, (char *)entry, value);
2104 vim_free(entry);
2105 }
2106#endif
2107}
2108
2109/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002110 * "escape({string}, {chars})" function
2111 */
2112 static void
2113f_escape(typval_T *argvars, typval_T *rettv)
2114{
2115 char_u buf[NUMBUFLEN];
2116
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002117 rettv->vval.v_string = vim_strsave_escaped(tv_get_string(&argvars[0]),
2118 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002119 rettv->v_type = VAR_STRING;
2120}
2121
2122/*
2123 * "eval()" function
2124 */
2125 static void
2126f_eval(typval_T *argvars, typval_T *rettv)
2127{
2128 char_u *s, *p;
2129
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002130 s = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002131 if (s != NULL)
2132 s = skipwhite(s);
2133
2134 p = s;
2135 if (s == NULL || eval1(&s, rettv, TRUE) == FAIL)
2136 {
2137 if (p != NULL && !aborting())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002138 semsg(_(e_invexpr2), p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002139 need_clr_eos = FALSE;
2140 rettv->v_type = VAR_NUMBER;
2141 rettv->vval.v_number = 0;
2142 }
2143 else if (*s != NUL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002144 emsg(_(e_trailing));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002145}
2146
2147/*
2148 * "eventhandler()" function
2149 */
2150 static void
2151f_eventhandler(typval_T *argvars UNUSED, typval_T *rettv)
2152{
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002153 rettv->vval.v_number = vgetc_busy || input_busy;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002154}
2155
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002156static garray_T redir_execute_ga;
2157
2158/*
2159 * Append "value[value_len]" to the execute() output.
2160 */
2161 void
2162execute_redir_str(char_u *value, int value_len)
2163{
2164 int len;
2165
2166 if (value_len == -1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002167 len = (int)STRLEN(value); // Append the entire string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002168 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002169 len = value_len; // Append only "value_len" characters
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002170 if (ga_grow(&redir_execute_ga, len) == OK)
2171 {
2172 mch_memmove((char *)redir_execute_ga.ga_data
2173 + redir_execute_ga.ga_len, value, len);
2174 redir_execute_ga.ga_len += len;
2175 }
2176}
2177
2178/*
2179 * Get next line from a list.
2180 * Called by do_cmdline() to get the next line.
2181 * Returns allocated string, or NULL for end of function.
2182 */
2183
2184 static char_u *
2185get_list_line(
2186 int c UNUSED,
2187 void *cookie,
Bram Moolenaare96a2492019-06-25 04:12:16 +02002188 int indent UNUSED,
2189 int do_concat UNUSED)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002190{
2191 listitem_T **p = (listitem_T **)cookie;
2192 listitem_T *item = *p;
2193 char_u buf[NUMBUFLEN];
2194 char_u *s;
2195
2196 if (item == NULL)
2197 return NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002198 s = tv_get_string_buf_chk(&item->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002199 *p = item->li_next;
2200 return s == NULL ? NULL : vim_strsave(s);
2201}
2202
2203/*
2204 * "execute()" function
2205 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02002206 void
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002207execute_common(typval_T *argvars, typval_T *rettv, int arg_off)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002208{
2209 char_u *cmd = NULL;
2210 list_T *list = NULL;
2211 int save_msg_silent = msg_silent;
2212 int save_emsg_silent = emsg_silent;
2213 int save_emsg_noredir = emsg_noredir;
2214 int save_redir_execute = redir_execute;
Bram Moolenaar20951482017-12-25 13:44:43 +01002215 int save_redir_off = redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002216 garray_T save_ga;
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002217 int save_msg_col = msg_col;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002218 int echo_output = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002219
2220 rettv->vval.v_string = NULL;
2221 rettv->v_type = VAR_STRING;
2222
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002223 if (argvars[arg_off].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002224 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002225 list = argvars[arg_off].vval.v_list;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002226 if (list == NULL || list->lv_len == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002227 // empty list, no commands, empty output
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002228 return;
2229 ++list->lv_refcount;
2230 }
Bram Moolenaare2a8f072020-01-08 19:32:18 +01002231 else if (argvars[arg_off].v_type == VAR_JOB
2232 || argvars[arg_off].v_type == VAR_CHANNEL)
2233 {
2234 emsg(_(e_inval_string));
2235 return;
2236 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002237 else
2238 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002239 cmd = tv_get_string_chk(&argvars[arg_off]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002240 if (cmd == NULL)
2241 return;
2242 }
2243
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002244 if (argvars[arg_off + 1].v_type != VAR_UNKNOWN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002245 {
2246 char_u buf[NUMBUFLEN];
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002247 char_u *s = tv_get_string_buf_chk(&argvars[arg_off + 1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002248
2249 if (s == NULL)
2250 return;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002251 if (*s == NUL)
2252 echo_output = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002253 if (STRNCMP(s, "silent", 6) == 0)
2254 ++msg_silent;
2255 if (STRCMP(s, "silent!") == 0)
2256 {
2257 emsg_silent = TRUE;
2258 emsg_noredir = TRUE;
2259 }
2260 }
2261 else
2262 ++msg_silent;
2263
2264 if (redir_execute)
2265 save_ga = redir_execute_ga;
2266 ga_init2(&redir_execute_ga, (int)sizeof(char), 500);
2267 redir_execute = TRUE;
Bram Moolenaar20951482017-12-25 13:44:43 +01002268 redir_off = FALSE;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002269 if (!echo_output)
2270 msg_col = 0; // prevent leading spaces
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002271
2272 if (cmd != NULL)
2273 do_cmdline_cmd(cmd);
2274 else
2275 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002276 listitem_T *item;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002277
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002278 range_list_materialize(list);
2279 item = list->lv_first;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002280 do_cmdline(NULL, get_list_line, (void *)&item,
2281 DOCMD_NOWAIT|DOCMD_VERBOSE|DOCMD_REPEAT|DOCMD_KEYTYPED);
2282 --list->lv_refcount;
2283 }
2284
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002285 // Need to append a NUL to the result.
Bram Moolenaard297f352017-01-29 20:31:21 +01002286 if (ga_grow(&redir_execute_ga, 1) == OK)
2287 {
2288 ((char *)redir_execute_ga.ga_data)[redir_execute_ga.ga_len] = NUL;
2289 rettv->vval.v_string = redir_execute_ga.ga_data;
2290 }
2291 else
2292 {
2293 ga_clear(&redir_execute_ga);
2294 rettv->vval.v_string = NULL;
2295 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002296 msg_silent = save_msg_silent;
2297 emsg_silent = save_emsg_silent;
2298 emsg_noredir = save_emsg_noredir;
2299
2300 redir_execute = save_redir_execute;
2301 if (redir_execute)
2302 redir_execute_ga = save_ga;
Bram Moolenaar20951482017-12-25 13:44:43 +01002303 redir_off = save_redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002304
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002305 // "silent reg" or "silent echo x" leaves msg_col somewhere in the line.
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002306 if (echo_output)
2307 // When not working silently: put it in column zero. A following
2308 // "echon" will overwrite the message, unavoidably.
2309 msg_col = 0;
2310 else
2311 // When working silently: Put it back where it was, since nothing
2312 // should have been written.
2313 msg_col = save_msg_col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002314}
2315
2316/*
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002317 * "execute()" function
2318 */
2319 static void
2320f_execute(typval_T *argvars, typval_T *rettv)
2321{
2322 execute_common(argvars, rettv, 0);
2323}
2324
2325/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002326 * "exists()" function
2327 */
2328 static void
2329f_exists(typval_T *argvars, typval_T *rettv)
2330{
2331 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002332 int n = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002333
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002334 p = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002335 if (*p == '$') // environment variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002336 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002337 // first try "normal" environment variables (fast)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002338 if (mch_getenv(p + 1) != NULL)
2339 n = TRUE;
2340 else
2341 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002342 // try expanding things like $VIM and ${HOME}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002343 p = expand_env_save(p);
2344 if (p != NULL && *p != '$')
2345 n = TRUE;
2346 vim_free(p);
2347 }
2348 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002349 else if (*p == '&' || *p == '+') // option
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002350 {
2351 n = (get_option_tv(&p, NULL, TRUE) == OK);
2352 if (*skipwhite(p) != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002353 n = FALSE; // trailing garbage
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002354 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002355 else if (*p == '*') // internal or user defined function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002356 {
Bram Moolenaarb54c3ff2016-07-31 14:11:58 +02002357 n = function_exists(p + 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002358 }
Bram Moolenaar15c47602020-03-26 22:16:48 +01002359 else if (*p == '?') // internal function only
2360 {
2361 n = has_internal_func_name(p + 1);
2362 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002363 else if (*p == ':')
2364 {
2365 n = cmd_exists(p + 1);
2366 }
2367 else if (*p == '#')
2368 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002369 if (p[1] == '#')
2370 n = autocmd_supported(p + 2);
2371 else
2372 n = au_exists(p + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002373 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002374 else // internal variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002375 {
Bram Moolenaarc6f9f732018-02-11 19:06:26 +01002376 n = var_exists(p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002377 }
2378
2379 rettv->vval.v_number = n;
2380}
2381
2382#ifdef FEAT_FLOAT
2383/*
2384 * "exp()" function
2385 */
2386 static void
2387f_exp(typval_T *argvars, typval_T *rettv)
2388{
2389 float_T f = 0.0;
2390
2391 rettv->v_type = VAR_FLOAT;
2392 if (get_float_arg(argvars, &f) == OK)
2393 rettv->vval.v_float = exp(f);
2394 else
2395 rettv->vval.v_float = 0.0;
2396}
2397#endif
2398
2399/*
2400 * "expand()" function
2401 */
2402 static void
2403f_expand(typval_T *argvars, typval_T *rettv)
2404{
2405 char_u *s;
2406 int len;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002407 char *errormsg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002408 int options = WILD_SILENT|WILD_USE_NL|WILD_LIST_NOTFOUND;
2409 expand_T xpc;
2410 int error = FALSE;
2411 char_u *result;
2412
2413 rettv->v_type = VAR_STRING;
2414 if (argvars[1].v_type != VAR_UNKNOWN
2415 && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002416 && tv_get_number_chk(&argvars[2], &error)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002417 && !error)
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02002418 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002419
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002420 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002421 if (*s == '%' || *s == '#' || *s == '<')
2422 {
2423 ++emsg_off;
2424 result = eval_vars(s, s, &len, NULL, &errormsg, NULL);
2425 --emsg_off;
2426 if (rettv->v_type == VAR_LIST)
2427 {
2428 if (rettv_list_alloc(rettv) != FAIL && result != NULL)
2429 list_append_string(rettv->vval.v_list, result, -1);
Bram Moolenaar86173482019-10-01 17:02:16 +02002430 vim_free(result);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002431 }
2432 else
2433 rettv->vval.v_string = result;
2434 }
2435 else
2436 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002437 // When the optional second argument is non-zero, don't remove matches
2438 // for 'wildignore' and don't put matches for 'suffixes' at the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002439 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002440 && tv_get_number_chk(&argvars[1], &error))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002441 options |= WILD_KEEP_ALL;
2442 if (!error)
2443 {
2444 ExpandInit(&xpc);
2445 xpc.xp_context = EXPAND_FILES;
2446 if (p_wic)
2447 options += WILD_ICASE;
2448 if (rettv->v_type == VAR_STRING)
2449 rettv->vval.v_string = ExpandOne(&xpc, s, NULL,
2450 options, WILD_ALL);
2451 else if (rettv_list_alloc(rettv) != FAIL)
2452 {
2453 int i;
2454
2455 ExpandOne(&xpc, s, NULL, options, WILD_ALL_KEEP);
2456 for (i = 0; i < xpc.xp_numfiles; i++)
2457 list_append_string(rettv->vval.v_list, xpc.xp_files[i], -1);
2458 ExpandCleanup(&xpc);
2459 }
2460 }
2461 else
2462 rettv->vval.v_string = NULL;
2463 }
2464}
2465
2466/*
Bram Moolenaar80dad482019-06-09 17:22:31 +02002467 * "expandcmd()" function
2468 * Expand all the special characters in a command string.
2469 */
2470 static void
2471f_expandcmd(typval_T *argvars, typval_T *rettv)
2472{
2473 exarg_T eap;
2474 char_u *cmdstr;
2475 char *errormsg = NULL;
2476
2477 rettv->v_type = VAR_STRING;
2478 cmdstr = vim_strsave(tv_get_string(&argvars[0]));
2479
2480 memset(&eap, 0, sizeof(eap));
2481 eap.cmd = cmdstr;
2482 eap.arg = cmdstr;
Bram Moolenaar8071cb22019-07-12 17:58:01 +02002483 eap.argt |= EX_NOSPC;
Bram Moolenaar80dad482019-06-09 17:22:31 +02002484 eap.usefilter = FALSE;
2485 eap.nextcmd = NULL;
2486 eap.cmdidx = CMD_USER;
2487
2488 expand_filename(&eap, &cmdstr, &errormsg);
2489 if (errormsg != NULL && *errormsg != NUL)
2490 emsg(errormsg);
2491
2492 rettv->vval.v_string = cmdstr;
2493}
2494
2495/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002496 * "feedkeys()" function
2497 */
2498 static void
2499f_feedkeys(typval_T *argvars, typval_T *rettv UNUSED)
2500{
2501 int remap = TRUE;
2502 int insert = FALSE;
2503 char_u *keys, *flags;
2504 char_u nbuf[NUMBUFLEN];
2505 int typed = FALSE;
2506 int execute = FALSE;
2507 int dangerous = FALSE;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002508 int lowlevel = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002509 char_u *keys_esc;
2510
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002511 // This is not allowed in the sandbox. If the commands would still be
2512 // executed in the sandbox it would be OK, but it probably happens later,
2513 // when "sandbox" is no longer set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002514 if (check_secure())
2515 return;
2516
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002517 keys = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002518
2519 if (argvars[1].v_type != VAR_UNKNOWN)
2520 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002521 flags = tv_get_string_buf(&argvars[1], nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002522 for ( ; *flags != NUL; ++flags)
2523 {
2524 switch (*flags)
2525 {
2526 case 'n': remap = FALSE; break;
2527 case 'm': remap = TRUE; break;
2528 case 't': typed = TRUE; break;
2529 case 'i': insert = TRUE; break;
2530 case 'x': execute = TRUE; break;
2531 case '!': dangerous = TRUE; break;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002532 case 'L': lowlevel = TRUE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002533 }
2534 }
2535 }
2536
2537 if (*keys != NUL || execute)
2538 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002539 // Need to escape K_SPECIAL and CSI before putting the string in the
2540 // typeahead buffer.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002541 keys_esc = vim_strsave_escape_csi(keys);
2542 if (keys_esc != NULL)
2543 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002544 if (lowlevel)
2545 {
2546#ifdef USE_INPUT_BUF
Bram Moolenaar9645e2d2020-03-20 20:48:49 +01002547 int idx;
2548 int len = (int)STRLEN(keys);
2549
2550 for (idx = 0; idx < len; ++idx)
2551 {
2552 // if a CTRL-C was typed, set got_int, similar to what
2553 // happens in fill_input_buf()
2554 if (keys[idx] == 3 && ctrl_c_interrupts && typed)
2555 got_int = TRUE;
2556 add_to_input_buf(keys + idx, 1);
2557 }
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002558#else
2559 emsg(_("E980: lowlevel input not supported"));
2560#endif
2561 }
2562 else
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002563 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002564 ins_typebuf(keys_esc, (remap ? REMAP_YES : REMAP_NONE),
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002565 insert ? 0 : typebuf.tb_len, !typed, FALSE);
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002566 if (vgetc_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02002567#ifdef FEAT_TIMERS
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002568 || timer_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02002569#endif
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002570 || input_busy)
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002571 typebuf_was_filled = TRUE;
2572 }
2573 vim_free(keys_esc);
2574
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002575 if (execute)
2576 {
2577 int save_msg_scroll = msg_scroll;
2578
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002579 // Avoid a 1 second delay when the keys start Insert mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002580 msg_scroll = FALSE;
2581
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02002582 if (!dangerous)
2583 ++ex_normal_busy;
Bram Moolenaar905dd902019-04-07 14:21:47 +02002584 exec_normal(TRUE, lowlevel, TRUE);
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02002585 if (!dangerous)
2586 --ex_normal_busy;
2587
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002588 msg_scroll |= save_msg_scroll;
2589 }
2590 }
2591 }
2592}
2593
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002594#ifdef FEAT_FLOAT
2595/*
2596 * "float2nr({float})" function
2597 */
2598 static void
2599f_float2nr(typval_T *argvars, typval_T *rettv)
2600{
2601 float_T f = 0.0;
2602
2603 if (get_float_arg(argvars, &f) == OK)
2604 {
Bram Moolenaar863e80b2017-06-04 20:30:00 +02002605 if (f <= -VARNUM_MAX + DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002606 rettv->vval.v_number = -VARNUM_MAX;
Bram Moolenaar863e80b2017-06-04 20:30:00 +02002607 else if (f >= VARNUM_MAX - DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002608 rettv->vval.v_number = VARNUM_MAX;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002609 else
2610 rettv->vval.v_number = (varnumber_T)f;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002611 }
2612}
2613
2614/*
2615 * "floor({float})" function
2616 */
2617 static void
2618f_floor(typval_T *argvars, typval_T *rettv)
2619{
2620 float_T f = 0.0;
2621
2622 rettv->v_type = VAR_FLOAT;
2623 if (get_float_arg(argvars, &f) == OK)
2624 rettv->vval.v_float = floor(f);
2625 else
2626 rettv->vval.v_float = 0.0;
2627}
2628
2629/*
2630 * "fmod()" function
2631 */
2632 static void
2633f_fmod(typval_T *argvars, typval_T *rettv)
2634{
2635 float_T fx = 0.0, fy = 0.0;
2636
2637 rettv->v_type = VAR_FLOAT;
2638 if (get_float_arg(argvars, &fx) == OK
2639 && get_float_arg(&argvars[1], &fy) == OK)
2640 rettv->vval.v_float = fmod(fx, fy);
2641 else
2642 rettv->vval.v_float = 0.0;
2643}
2644#endif
2645
2646/*
2647 * "fnameescape({string})" function
2648 */
2649 static void
2650f_fnameescape(typval_T *argvars, typval_T *rettv)
2651{
2652 rettv->vval.v_string = vim_strsave_fnameescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002653 tv_get_string(&argvars[0]), FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002654 rettv->v_type = VAR_STRING;
2655}
2656
2657/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002658 * "foreground()" function
2659 */
2660 static void
2661f_foreground(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2662{
2663#ifdef FEAT_GUI
2664 if (gui.in_use)
Bram Moolenaarafde13b2019-04-28 19:46:49 +02002665 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002666 gui_mch_set_foreground();
Bram Moolenaarafde13b2019-04-28 19:46:49 +02002667 return;
2668 }
2669#endif
2670#if defined(MSWIN) && (!defined(FEAT_GUI) || defined(VIMDLL))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002671 win32_set_foreground();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002672#endif
2673}
2674
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002675 static void
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002676common_function(typval_T *argvars, typval_T *rettv, int is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002677{
2678 char_u *s;
2679 char_u *name;
2680 int use_string = FALSE;
2681 partial_T *arg_pt = NULL;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002682 char_u *trans_name = NULL;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002683 int is_global = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002684
2685 if (argvars[0].v_type == VAR_FUNC)
2686 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002687 // function(MyFunc, [arg], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002688 s = argvars[0].vval.v_string;
2689 }
2690 else if (argvars[0].v_type == VAR_PARTIAL
2691 && argvars[0].vval.v_partial != NULL)
2692 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002693 // function(dict.MyFunc, [arg])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002694 arg_pt = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002695 s = partial_name(arg_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002696 }
2697 else
2698 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002699 // function('MyFunc', [arg], dict)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002700 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002701 use_string = TRUE;
2702 }
2703
Bram Moolenaar843b8842016-08-21 14:36:15 +02002704 if ((use_string && vim_strchr(s, AUTOLOAD_CHAR) == NULL) || is_funcref)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002705 {
2706 name = s;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002707 trans_name = trans_function_name(&name, &is_global, FALSE,
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002708 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF, NULL, NULL);
2709 if (*name != NUL)
2710 s = NULL;
2711 }
2712
Bram Moolenaar843b8842016-08-21 14:36:15 +02002713 if (s == NULL || *s == NUL || (use_string && VIM_ISDIGIT(*s))
2714 || (is_funcref && trans_name == NULL))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002715 semsg(_(e_invarg2), use_string ? tv_get_string(&argvars[0]) : s);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002716 // Don't check an autoload name for existence here.
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002717 else if (trans_name != NULL && (is_funcref
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002718 ? find_func(trans_name, is_global, NULL) == NULL
2719 : !translated_function_exists(trans_name, is_global)))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002720 semsg(_("E700: Unknown function: %s"), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002721 else
2722 {
2723 int dict_idx = 0;
2724 int arg_idx = 0;
2725 list_T *list = NULL;
2726
2727 if (STRNCMP(s, "s:", 2) == 0 || STRNCMP(s, "<SID>", 5) == 0)
2728 {
2729 char sid_buf[25];
2730 int off = *s == 's' ? 2 : 5;
2731
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002732 // Expand s: and <SID> into <SNR>nr_, so that the function can
2733 // also be called from another script. Using trans_function_name()
2734 // would also work, but some plugins depend on the name being
2735 // printable text.
Bram Moolenaarf29c1c62018-09-10 21:05:02 +02002736 sprintf(sid_buf, "<SNR>%ld_", (long)current_sctx.sc_sid);
Bram Moolenaar51e14382019-05-25 20:21:28 +02002737 name = alloc(STRLEN(sid_buf) + STRLEN(s + off) + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002738 if (name != NULL)
2739 {
2740 STRCPY(name, sid_buf);
2741 STRCAT(name, s + off);
2742 }
2743 }
2744 else
2745 name = vim_strsave(s);
2746
2747 if (argvars[1].v_type != VAR_UNKNOWN)
2748 {
2749 if (argvars[2].v_type != VAR_UNKNOWN)
2750 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002751 // function(name, [args], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002752 arg_idx = 1;
2753 dict_idx = 2;
2754 }
2755 else if (argvars[1].v_type == VAR_DICT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002756 // function(name, dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002757 dict_idx = 1;
2758 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002759 // function(name, [args])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002760 arg_idx = 1;
2761 if (dict_idx > 0)
2762 {
2763 if (argvars[dict_idx].v_type != VAR_DICT)
2764 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002765 emsg(_("E922: expected a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002766 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002767 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002768 }
2769 if (argvars[dict_idx].vval.v_dict == NULL)
2770 dict_idx = 0;
2771 }
2772 if (arg_idx > 0)
2773 {
2774 if (argvars[arg_idx].v_type != VAR_LIST)
2775 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002776 emsg(_("E923: Second argument of function() must be a list or a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002777 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002778 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002779 }
2780 list = argvars[arg_idx].vval.v_list;
2781 if (list == NULL || list->lv_len == 0)
2782 arg_idx = 0;
Bram Moolenaar4c054e92019-11-10 00:13:50 +01002783 else if (list->lv_len > MAX_FUNC_ARGS)
2784 {
Bram Moolenaar2118a302019-11-22 19:29:45 +01002785 emsg_funcname((char *)e_toomanyarg, s);
Bram Moolenaar4c054e92019-11-10 00:13:50 +01002786 vim_free(name);
2787 goto theend;
2788 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002789 }
2790 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002791 if (dict_idx > 0 || arg_idx > 0 || arg_pt != NULL || is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002792 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02002793 partial_T *pt = ALLOC_CLEAR_ONE(partial_T);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002794
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002795 // result is a VAR_PARTIAL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002796 if (pt == NULL)
2797 vim_free(name);
2798 else
2799 {
2800 if (arg_idx > 0 || (arg_pt != NULL && arg_pt->pt_argc > 0))
2801 {
2802 listitem_T *li;
2803 int i = 0;
2804 int arg_len = 0;
2805 int lv_len = 0;
2806
2807 if (arg_pt != NULL)
2808 arg_len = arg_pt->pt_argc;
2809 if (list != NULL)
2810 lv_len = list->lv_len;
2811 pt->pt_argc = arg_len + lv_len;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02002812 pt->pt_argv = ALLOC_MULT(typval_T, pt->pt_argc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002813 if (pt->pt_argv == NULL)
2814 {
2815 vim_free(pt);
2816 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002817 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002818 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002819 for (i = 0; i < arg_len; i++)
2820 copy_tv(&arg_pt->pt_argv[i], &pt->pt_argv[i]);
2821 if (lv_len > 0)
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002822 {
2823 range_list_materialize(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002824 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002825 copy_tv(&li->li_tv, &pt->pt_argv[i++]);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002826 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002827 }
2828
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002829 // For "function(dict.func, [], dict)" and "func" is a partial
2830 // use "dict". That is backwards compatible.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002831 if (dict_idx > 0)
2832 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002833 // The dict is bound explicitly, pt_auto is FALSE.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002834 pt->pt_dict = argvars[dict_idx].vval.v_dict;
2835 ++pt->pt_dict->dv_refcount;
2836 }
2837 else if (arg_pt != NULL)
2838 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002839 // If the dict was bound automatically the result is also
2840 // bound automatically.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002841 pt->pt_dict = arg_pt->pt_dict;
2842 pt->pt_auto = arg_pt->pt_auto;
2843 if (pt->pt_dict != NULL)
2844 ++pt->pt_dict->dv_refcount;
2845 }
2846
2847 pt->pt_refcount = 1;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002848 if (arg_pt != NULL && arg_pt->pt_func != NULL)
2849 {
2850 pt->pt_func = arg_pt->pt_func;
2851 func_ptr_ref(pt->pt_func);
2852 vim_free(name);
2853 }
2854 else if (is_funcref)
2855 {
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002856 pt->pt_func = find_func(trans_name, is_global, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002857 func_ptr_ref(pt->pt_func);
2858 vim_free(name);
2859 }
2860 else
2861 {
2862 pt->pt_name = name;
2863 func_ref(name);
2864 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002865 }
2866 rettv->v_type = VAR_PARTIAL;
2867 rettv->vval.v_partial = pt;
2868 }
2869 else
2870 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002871 // result is a VAR_FUNC
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002872 rettv->v_type = VAR_FUNC;
2873 rettv->vval.v_string = name;
2874 func_ref(name);
2875 }
2876 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002877theend:
2878 vim_free(trans_name);
2879}
2880
2881/*
2882 * "funcref()" function
2883 */
2884 static void
2885f_funcref(typval_T *argvars, typval_T *rettv)
2886{
2887 common_function(argvars, rettv, TRUE);
2888}
2889
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002890 static type_T *
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002891ret_f_function(int argcount, type_T **argtypes)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002892{
2893 if (argcount == 1 && argtypes[0]->tt_type == VAR_STRING)
2894 return &t_func_any;
Bram Moolenaard77a8522020-04-03 21:59:57 +02002895 return &t_func_void;
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002896}
2897
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002898/*
2899 * "function()" function
2900 */
2901 static void
2902f_function(typval_T *argvars, typval_T *rettv)
2903{
2904 common_function(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002905}
2906
2907/*
2908 * "garbagecollect()" function
2909 */
2910 static void
2911f_garbagecollect(typval_T *argvars, typval_T *rettv UNUSED)
2912{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002913 // This is postponed until we are back at the toplevel, because we may be
2914 // using Lists and Dicts internally. E.g.: ":echo [garbagecollect()]".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002915 want_garbage_collect = TRUE;
2916
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002917 if (argvars[0].v_type != VAR_UNKNOWN && tv_get_number(&argvars[0]) == 1)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002918 garbage_collect_at_exit = TRUE;
2919}
2920
2921/*
2922 * "get()" function
2923 */
2924 static void
2925f_get(typval_T *argvars, typval_T *rettv)
2926{
2927 listitem_T *li;
2928 list_T *l;
2929 dictitem_T *di;
2930 dict_T *d;
2931 typval_T *tv = NULL;
Bram Moolenaarf91aac52019-07-28 13:21:01 +02002932 int what_is_dict = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002933
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002934 if (argvars[0].v_type == VAR_BLOB)
2935 {
2936 int error = FALSE;
2937 int idx = tv_get_number_chk(&argvars[1], &error);
2938
2939 if (!error)
2940 {
2941 rettv->v_type = VAR_NUMBER;
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002942 if (idx < 0)
2943 idx = blob_len(argvars[0].vval.v_blob) + idx;
2944 if (idx < 0 || idx >= blob_len(argvars[0].vval.v_blob))
2945 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002946 else
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002947 {
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002948 rettv->vval.v_number = blob_get(argvars[0].vval.v_blob, idx);
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002949 tv = rettv;
2950 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002951 }
2952 }
2953 else if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002954 {
2955 if ((l = argvars[0].vval.v_list) != NULL)
2956 {
2957 int error = FALSE;
2958
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002959 li = list_find(l, (long)tv_get_number_chk(&argvars[1], &error));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002960 if (!error && li != NULL)
2961 tv = &li->li_tv;
2962 }
2963 }
2964 else if (argvars[0].v_type == VAR_DICT)
2965 {
2966 if ((d = argvars[0].vval.v_dict) != NULL)
2967 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002968 di = dict_find(d, tv_get_string(&argvars[1]), -1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002969 if (di != NULL)
2970 tv = &di->di_tv;
2971 }
2972 }
2973 else if (argvars[0].v_type == VAR_PARTIAL || argvars[0].v_type == VAR_FUNC)
2974 {
2975 partial_T *pt;
2976 partial_T fref_pt;
2977
2978 if (argvars[0].v_type == VAR_PARTIAL)
2979 pt = argvars[0].vval.v_partial;
2980 else
2981 {
Bram Moolenaara80faa82020-04-12 19:37:17 +02002982 CLEAR_FIELD(fref_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002983 fref_pt.pt_name = argvars[0].vval.v_string;
2984 pt = &fref_pt;
2985 }
2986
2987 if (pt != NULL)
2988 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002989 char_u *what = tv_get_string(&argvars[1]);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002990 char_u *n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002991
2992 if (STRCMP(what, "func") == 0 || STRCMP(what, "name") == 0)
2993 {
2994 rettv->v_type = (*what == 'f' ? VAR_FUNC : VAR_STRING);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002995 n = partial_name(pt);
2996 if (n == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002997 rettv->vval.v_string = NULL;
2998 else
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002999 {
3000 rettv->vval.v_string = vim_strsave(n);
3001 if (rettv->v_type == VAR_FUNC)
3002 func_ref(rettv->vval.v_string);
3003 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003004 }
3005 else if (STRCMP(what, "dict") == 0)
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003006 {
3007 what_is_dict = TRUE;
3008 if (pt->pt_dict != NULL)
3009 rettv_dict_set(rettv, pt->pt_dict);
3010 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003011 else if (STRCMP(what, "args") == 0)
3012 {
3013 rettv->v_type = VAR_LIST;
3014 if (rettv_list_alloc(rettv) == OK)
3015 {
3016 int i;
3017
3018 for (i = 0; i < pt->pt_argc; ++i)
3019 list_append_tv(rettv->vval.v_list, &pt->pt_argv[i]);
3020 }
3021 }
3022 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003023 semsg(_(e_invarg2), what);
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003024
3025 // When {what} == "dict" and pt->pt_dict == NULL, evaluate the
3026 // third argument
3027 if (!what_is_dict)
3028 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003029 }
3030 }
3031 else
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01003032 semsg(_(e_listdictblobarg), "get()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003033
3034 if (tv == NULL)
3035 {
3036 if (argvars[2].v_type != VAR_UNKNOWN)
3037 copy_tv(&argvars[2], rettv);
3038 }
3039 else
3040 copy_tv(tv, rettv);
3041}
3042
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003043/*
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003044 * "getchangelist()" function
3045 */
3046 static void
3047f_getchangelist(typval_T *argvars, typval_T *rettv)
3048{
3049#ifdef FEAT_JUMPLIST
3050 buf_T *buf;
3051 int i;
3052 list_T *l;
3053 dict_T *d;
3054#endif
3055
3056 if (rettv_list_alloc(rettv) != OK)
3057 return;
3058
3059#ifdef FEAT_JUMPLIST
Bram Moolenaar4c313b12019-08-24 22:58:31 +02003060 if (argvars[0].v_type == VAR_UNKNOWN)
3061 buf = curbuf;
3062 else
3063 {
3064 (void)tv_get_number(&argvars[0]); // issue errmsg if type error
3065 ++emsg_off;
3066 buf = tv_get_buf(&argvars[0], FALSE);
3067 --emsg_off;
3068 }
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003069 if (buf == NULL)
3070 return;
3071
3072 l = list_alloc();
3073 if (l == NULL)
3074 return;
3075
3076 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3077 return;
3078 /*
3079 * The current window change list index tracks only the position in the
3080 * current buffer change list. For other buffers, use the change list
3081 * length as the current index.
3082 */
3083 list_append_number(rettv->vval.v_list,
3084 (varnumber_T)((buf == curwin->w_buffer)
3085 ? curwin->w_changelistidx : buf->b_changelistlen));
3086
3087 for (i = 0; i < buf->b_changelistlen; ++i)
3088 {
3089 if (buf->b_changelist[i].lnum == 0)
3090 continue;
3091 if ((d = dict_alloc()) == NULL)
3092 return;
3093 if (list_append_dict(l, d) == FAIL)
3094 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003095 dict_add_number(d, "lnum", (long)buf->b_changelist[i].lnum);
3096 dict_add_number(d, "col", (long)buf->b_changelist[i].col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003097 dict_add_number(d, "coladd", (long)buf->b_changelist[i].coladd);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003098 }
3099#endif
3100}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003101
3102/*
3103 * "getcharsearch()" function
3104 */
3105 static void
3106f_getcharsearch(typval_T *argvars UNUSED, typval_T *rettv)
3107{
3108 if (rettv_dict_alloc(rettv) != FAIL)
3109 {
3110 dict_T *dict = rettv->vval.v_dict;
3111
Bram Moolenaare0be1672018-07-08 16:50:37 +02003112 dict_add_string(dict, "char", last_csearch());
3113 dict_add_number(dict, "forward", last_csearch_forward());
3114 dict_add_number(dict, "until", last_csearch_until());
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003115 }
3116}
3117
3118/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02003119 * "getenv()" function
3120 */
3121 static void
3122f_getenv(typval_T *argvars, typval_T *rettv)
3123{
3124 int mustfree = FALSE;
3125 char_u *p = vim_getenv(tv_get_string(&argvars[0]), &mustfree);
3126
3127 if (p == NULL)
3128 {
3129 rettv->v_type = VAR_SPECIAL;
3130 rettv->vval.v_number = VVAL_NULL;
3131 return;
3132 }
3133 if (!mustfree)
3134 p = vim_strsave(p);
3135 rettv->vval.v_string = p;
3136 rettv->v_type = VAR_STRING;
3137}
3138
3139/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003140 * "getfontname()" function
3141 */
3142 static void
3143f_getfontname(typval_T *argvars UNUSED, typval_T *rettv)
3144{
3145 rettv->v_type = VAR_STRING;
3146 rettv->vval.v_string = NULL;
3147#ifdef FEAT_GUI
3148 if (gui.in_use)
3149 {
3150 GuiFont font;
3151 char_u *name = NULL;
3152
3153 if (argvars[0].v_type == VAR_UNKNOWN)
3154 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003155 // Get the "Normal" font. Either the name saved by
3156 // hl_set_font_name() or from the font ID.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003157 font = gui.norm_font;
3158 name = hl_get_font_name();
3159 }
3160 else
3161 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003162 name = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003163 if (STRCMP(name, "*") == 0) // don't use font dialog
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003164 return;
3165 font = gui_mch_get_font(name, FALSE);
3166 if (font == NOFONT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003167 return; // Invalid font name, return empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003168 }
3169 rettv->vval.v_string = gui_mch_get_fontname(font, name);
3170 if (argvars[0].v_type != VAR_UNKNOWN)
3171 gui_mch_free_font(font);
3172 }
3173#endif
3174}
3175
3176/*
Bram Moolenaar4f505882018-02-10 21:06:32 +01003177 * "getjumplist()" function
3178 */
3179 static void
3180f_getjumplist(typval_T *argvars, typval_T *rettv)
3181{
3182#ifdef FEAT_JUMPLIST
3183 win_T *wp;
3184 int i;
3185 list_T *l;
3186 dict_T *d;
3187#endif
3188
3189 if (rettv_list_alloc(rettv) != OK)
3190 return;
3191
3192#ifdef FEAT_JUMPLIST
Bram Moolenaar00aa0692019-04-27 20:37:57 +02003193 wp = find_tabwin(&argvars[0], &argvars[1], NULL);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003194 if (wp == NULL)
3195 return;
3196
Bram Moolenaar57ee2b62019-02-12 22:15:06 +01003197 cleanup_jumplist(wp, TRUE);
3198
Bram Moolenaar4f505882018-02-10 21:06:32 +01003199 l = list_alloc();
3200 if (l == NULL)
3201 return;
3202
3203 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3204 return;
3205 list_append_number(rettv->vval.v_list, (varnumber_T)wp->w_jumplistidx);
3206
3207 for (i = 0; i < wp->w_jumplistlen; ++i)
3208 {
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003209 if (wp->w_jumplist[i].fmark.mark.lnum == 0)
3210 continue;
Bram Moolenaar4f505882018-02-10 21:06:32 +01003211 if ((d = dict_alloc()) == NULL)
3212 return;
3213 if (list_append_dict(l, d) == FAIL)
3214 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003215 dict_add_number(d, "lnum", (long)wp->w_jumplist[i].fmark.mark.lnum);
3216 dict_add_number(d, "col", (long)wp->w_jumplist[i].fmark.mark.col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003217 dict_add_number(d, "coladd", (long)wp->w_jumplist[i].fmark.mark.coladd);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003218 dict_add_number(d, "bufnr", (long)wp->w_jumplist[i].fmark.fnum);
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003219 if (wp->w_jumplist[i].fname != NULL)
Bram Moolenaare0be1672018-07-08 16:50:37 +02003220 dict_add_string(d, "filename", wp->w_jumplist[i].fname);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003221 }
3222#endif
3223}
3224
3225/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003226 * "getpid()" function
3227 */
3228 static void
3229f_getpid(typval_T *argvars UNUSED, typval_T *rettv)
3230{
3231 rettv->vval.v_number = mch_get_pid();
3232}
3233
3234 static void
3235getpos_both(
3236 typval_T *argvars,
3237 typval_T *rettv,
3238 int getcurpos)
3239{
3240 pos_T *fp;
3241 list_T *l;
3242 int fnum = -1;
3243
3244 if (rettv_list_alloc(rettv) == OK)
3245 {
3246 l = rettv->vval.v_list;
3247 if (getcurpos)
3248 fp = &curwin->w_cursor;
3249 else
3250 fp = var2fpos(&argvars[0], TRUE, &fnum);
3251 if (fnum != -1)
3252 list_append_number(l, (varnumber_T)fnum);
3253 else
3254 list_append_number(l, (varnumber_T)0);
3255 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->lnum
3256 : (varnumber_T)0);
3257 list_append_number(l, (fp != NULL)
3258 ? (varnumber_T)(fp->col == MAXCOL ? MAXCOL : fp->col + 1)
3259 : (varnumber_T)0);
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01003260 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->coladd :
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003261 (varnumber_T)0);
3262 if (getcurpos)
3263 {
Bram Moolenaar19a66852019-03-07 11:25:32 +01003264 int save_set_curswant = curwin->w_set_curswant;
3265 colnr_T save_curswant = curwin->w_curswant;
3266 colnr_T save_virtcol = curwin->w_virtcol;
3267
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003268 update_curswant();
3269 list_append_number(l, curwin->w_curswant == MAXCOL ?
3270 (varnumber_T)MAXCOL : (varnumber_T)curwin->w_curswant + 1);
Bram Moolenaar19a66852019-03-07 11:25:32 +01003271
3272 // Do not change "curswant", as it is unexpected that a get
3273 // function has a side effect.
3274 if (save_set_curswant)
3275 {
3276 curwin->w_set_curswant = save_set_curswant;
3277 curwin->w_curswant = save_curswant;
3278 curwin->w_virtcol = save_virtcol;
3279 curwin->w_valid &= ~VALID_VIRTCOL;
3280 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003281 }
3282 }
3283 else
3284 rettv->vval.v_number = FALSE;
3285}
3286
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003287/*
3288 * "getcurpos()" function
3289 */
3290 static void
3291f_getcurpos(typval_T *argvars, typval_T *rettv)
3292{
3293 getpos_both(argvars, rettv, TRUE);
3294}
3295
3296/*
3297 * "getpos(string)" function
3298 */
3299 static void
3300f_getpos(typval_T *argvars, typval_T *rettv)
3301{
3302 getpos_both(argvars, rettv, FALSE);
3303}
3304
3305/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003306 * "getreg()" function
3307 */
3308 static void
3309f_getreg(typval_T *argvars, typval_T *rettv)
3310{
3311 char_u *strregname;
3312 int regname;
3313 int arg2 = FALSE;
3314 int return_list = FALSE;
3315 int error = FALSE;
3316
3317 if (argvars[0].v_type != VAR_UNKNOWN)
3318 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003319 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003320 error = strregname == NULL;
3321 if (argvars[1].v_type != VAR_UNKNOWN)
3322 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003323 arg2 = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003324 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003325 return_list = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003326 }
3327 }
3328 else
3329 strregname = get_vim_var_str(VV_REG);
3330
3331 if (error)
3332 return;
3333
3334 regname = (strregname == NULL ? '"' : *strregname);
3335 if (regname == 0)
3336 regname = '"';
3337
3338 if (return_list)
3339 {
3340 rettv->v_type = VAR_LIST;
3341 rettv->vval.v_list = (list_T *)get_reg_contents(regname,
3342 (arg2 ? GREG_EXPR_SRC : 0) | GREG_LIST);
3343 if (rettv->vval.v_list == NULL)
3344 (void)rettv_list_alloc(rettv);
3345 else
3346 ++rettv->vval.v_list->lv_refcount;
3347 }
3348 else
3349 {
3350 rettv->v_type = VAR_STRING;
3351 rettv->vval.v_string = get_reg_contents(regname,
3352 arg2 ? GREG_EXPR_SRC : 0);
3353 }
3354}
3355
3356/*
3357 * "getregtype()" function
3358 */
3359 static void
3360f_getregtype(typval_T *argvars, typval_T *rettv)
3361{
3362 char_u *strregname;
3363 int regname;
3364 char_u buf[NUMBUFLEN + 2];
3365 long reglen = 0;
3366
3367 if (argvars[0].v_type != VAR_UNKNOWN)
3368 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003369 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003370 if (strregname == NULL) // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003371 {
3372 rettv->v_type = VAR_STRING;
3373 rettv->vval.v_string = NULL;
3374 return;
3375 }
3376 }
3377 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003378 // Default to v:register
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003379 strregname = get_vim_var_str(VV_REG);
3380
3381 regname = (strregname == NULL ? '"' : *strregname);
3382 if (regname == 0)
3383 regname = '"';
3384
3385 buf[0] = NUL;
3386 buf[1] = NUL;
3387 switch (get_reg_type(regname, &reglen))
3388 {
3389 case MLINE: buf[0] = 'V'; break;
3390 case MCHAR: buf[0] = 'v'; break;
3391 case MBLOCK:
3392 buf[0] = Ctrl_V;
3393 sprintf((char *)buf + 1, "%ld", reglen + 1);
3394 break;
3395 }
3396 rettv->v_type = VAR_STRING;
3397 rettv->vval.v_string = vim_strsave(buf);
3398}
3399
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003400/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01003401 * "gettagstack()" function
3402 */
3403 static void
3404f_gettagstack(typval_T *argvars, typval_T *rettv)
3405{
3406 win_T *wp = curwin; // default is current window
3407
3408 if (rettv_dict_alloc(rettv) != OK)
3409 return;
3410
3411 if (argvars[0].v_type != VAR_UNKNOWN)
3412 {
3413 wp = find_win_by_nr_or_id(&argvars[0]);
3414 if (wp == NULL)
3415 return;
3416 }
3417
3418 get_tagstack(wp, rettv->vval.v_dict);
3419}
3420
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003421// for VIM_VERSION_ defines
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003422#include "version.h"
3423
3424/*
3425 * "has()" function
3426 */
Bram Moolenaara259d8d2020-01-31 20:10:50 +01003427 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003428f_has(typval_T *argvars, typval_T *rettv)
3429{
3430 int i;
3431 char_u *name;
Bram Moolenaar79296512020-03-22 16:17:14 +01003432 int x = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003433 int n = FALSE;
Bram Moolenaar79296512020-03-22 16:17:14 +01003434 typedef struct {
3435 char *name;
3436 short present;
3437 } has_item_T;
3438 static has_item_T has_list[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003439 {
Bram Moolenaar79296512020-03-22 16:17:14 +01003440 {"amiga",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003441#ifdef AMIGA
Bram Moolenaar79296512020-03-22 16:17:14 +01003442 1
Bram Moolenaar39536dd2019-01-29 22:58:21 +01003443#else
Bram Moolenaar79296512020-03-22 16:17:14 +01003444 0
Bram Moolenaar39536dd2019-01-29 22:58:21 +01003445#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003446 },
3447 {"arp",
3448#if defined(AMIGA) && defined(FEAT_ARP)
3449 1
3450#else
3451 0
3452#endif
3453 },
3454 {"beos",
3455#ifdef __BEOS__
3456 1
3457#else
3458 0
3459#endif
3460 },
3461 {"haiku",
3462#ifdef __HAIKU__
3463 1
3464#else
3465 0
3466#endif
3467 },
3468 {"bsd",
3469#if defined(BSD) && !defined(MACOS_X)
3470 1
3471#else
3472 0
3473#endif
3474 },
3475 {"hpux",
3476#ifdef hpux
3477 1
3478#else
3479 0
3480#endif
3481 },
3482 {"linux",
3483#ifdef __linux__
3484 1
3485#else
3486 0
3487#endif
3488 },
3489 {"mac", // Mac OS X (and, once, Mac OS Classic)
3490#ifdef MACOS_X
3491 1
3492#else
3493 0
3494#endif
3495 },
3496 {"osx", // Mac OS X
3497#ifdef MACOS_X
3498 1
3499#else
3500 0
3501#endif
3502 },
3503 {"macunix", // Mac OS X, with the darwin feature
3504#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
3505 1
3506#else
3507 0
3508#endif
3509 },
3510 {"osxdarwin", // synonym for macunix
3511#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
3512 1
3513#else
3514 0
3515#endif
3516 },
3517 {"qnx",
3518#ifdef __QNX__
3519 1
3520#else
3521 0
3522#endif
3523 },
3524 {"sun",
3525#ifdef SUN_SYSTEM
3526 1
3527#else
3528 0
3529#endif
3530 },
3531 {"unix",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003532#ifdef UNIX
Bram Moolenaar79296512020-03-22 16:17:14 +01003533 1
3534#else
3535 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003536#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003537 },
3538 {"vms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003539#ifdef VMS
Bram Moolenaar79296512020-03-22 16:17:14 +01003540 1
3541#else
3542 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003543#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003544 },
3545 {"win32",
Bram Moolenaar4f974752019-02-17 17:44:42 +01003546#ifdef MSWIN
Bram Moolenaar79296512020-03-22 16:17:14 +01003547 1
3548#else
3549 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003550#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003551 },
3552 {"win32unix",
Bram Moolenaar1eed5322019-02-26 17:03:54 +01003553#if defined(UNIX) && defined(__CYGWIN__)
Bram Moolenaar79296512020-03-22 16:17:14 +01003554 1
3555#else
3556 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003557#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003558 },
3559 {"win64",
Bram Moolenaar44b443c2019-02-18 22:14:18 +01003560#ifdef _WIN64
Bram Moolenaar79296512020-03-22 16:17:14 +01003561 1
3562#else
3563 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003564#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003565 },
3566 {"ebcdic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003567#ifdef EBCDIC
Bram Moolenaar79296512020-03-22 16:17:14 +01003568 1
3569#else
3570 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003571#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003572 },
3573 {"fname_case",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003574#ifndef CASE_INSENSITIVE_FILENAME
Bram Moolenaar79296512020-03-22 16:17:14 +01003575 1
3576#else
3577 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003578#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003579 },
3580 {"acl",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003581#ifdef HAVE_ACL
Bram Moolenaar79296512020-03-22 16:17:14 +01003582 1
3583#else
3584 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003585#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003586 },
3587 {"arabic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003588#ifdef FEAT_ARABIC
Bram Moolenaar79296512020-03-22 16:17:14 +01003589 1
3590#else
3591 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003592#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003593 },
3594 {"autocmd", 1},
3595 {"autochdir",
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02003596#ifdef FEAT_AUTOCHDIR
Bram Moolenaar79296512020-03-22 16:17:14 +01003597 1
3598#else
3599 0
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02003600#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003601 },
3602 {"autoservername",
Bram Moolenaare42a6d22017-11-12 19:21:51 +01003603#ifdef FEAT_AUTOSERVERNAME
Bram Moolenaar79296512020-03-22 16:17:14 +01003604 1
3605#else
3606 0
Bram Moolenaare42a6d22017-11-12 19:21:51 +01003607#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003608 },
3609 {"balloon_eval",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01003610#ifdef FEAT_BEVAL_GUI
Bram Moolenaar79296512020-03-22 16:17:14 +01003611 1
3612#else
3613 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003614#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003615 },
3616 {"balloon_multiline",
3617#if defined(FEAT_BEVAL_GUI) && !defined(FEAT_GUI_MSWIN)
3618 // MS-Windows requires runtime check, see below
3619 1
3620#else
3621 0
3622#endif
3623 },
3624 {"balloon_eval_term",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01003625#ifdef FEAT_BEVAL_TERM
Bram Moolenaar79296512020-03-22 16:17:14 +01003626 1
3627#else
3628 0
Bram Moolenaar51b0f372017-11-18 18:52:04 +01003629#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003630 },
3631 {"builtin_terms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003632#if defined(SOME_BUILTIN_TCAPS) || defined(ALL_BUILTIN_TCAPS)
Bram Moolenaar79296512020-03-22 16:17:14 +01003633 1
3634#else
3635 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003636#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003637 },
3638 {"all_builtin_terms",
3639#if defined(ALL_BUILTIN_TCAPS)
3640 1
3641#else
3642 0
3643#endif
3644 },
3645 {"browsefilter",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003646#if defined(FEAT_BROWSE) && (defined(USE_FILE_CHOOSER) \
Bram Moolenaar4f974752019-02-17 17:44:42 +01003647 || defined(FEAT_GUI_MSWIN) \
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003648 || defined(FEAT_GUI_MOTIF))
Bram Moolenaar79296512020-03-22 16:17:14 +01003649 1
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003650#else
Bram Moolenaar79296512020-03-22 16:17:14 +01003651 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003652#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003653 },
3654 {"byte_offset",
3655#ifdef FEAT_BYTEOFF
3656 1
3657#else
3658 0
3659#endif
3660 },
3661 {"channel",
3662#ifdef FEAT_JOB_CHANNEL
3663 1
3664#else
3665 0
3666#endif
3667 },
3668 {"cindent",
3669#ifdef FEAT_CINDENT
3670 1
3671#else
3672 0
3673#endif
3674 },
3675 {"clientserver",
3676#ifdef FEAT_CLIENTSERVER
3677 1
3678#else
3679 0
3680#endif
3681 },
3682 {"clipboard",
3683#ifdef FEAT_CLIPBOARD
3684 1
3685#else
3686 0
3687#endif
3688 },
3689 {"cmdline_compl", 1},
3690 {"cmdline_hist", 1},
3691 {"comments", 1},
3692 {"conceal",
3693#ifdef FEAT_CONCEAL
3694 1
3695#else
3696 0
3697#endif
3698 },
3699 {"cryptv",
3700#ifdef FEAT_CRYPT
3701 1
3702#else
3703 0
3704#endif
3705 },
3706 {"crypt-blowfish",
3707#ifdef FEAT_CRYPT
3708 1
3709#else
3710 0
3711#endif
3712 },
3713 {"crypt-blowfish2",
3714#ifdef FEAT_CRYPT
3715 1
3716#else
3717 0
3718#endif
3719 },
3720 {"cscope",
3721#ifdef FEAT_CSCOPE
3722 1
3723#else
3724 0
3725#endif
3726 },
3727 {"cursorbind", 1},
3728 {"cursorshape",
3729#ifdef CURSOR_SHAPE
3730 1
3731#else
3732 0
3733#endif
3734 },
3735 {"debug",
3736#ifdef DEBUG
3737 1
3738#else
3739 0
3740#endif
3741 },
3742 {"dialog_con",
3743#ifdef FEAT_CON_DIALOG
3744 1
3745#else
3746 0
3747#endif
3748 },
3749 {"dialog_gui",
3750#ifdef FEAT_GUI_DIALOG
3751 1
3752#else
3753 0
3754#endif
3755 },
3756 {"diff",
3757#ifdef FEAT_DIFF
3758 1
3759#else
3760 0
3761#endif
3762 },
3763 {"digraphs",
3764#ifdef FEAT_DIGRAPHS
3765 1
3766#else
3767 0
3768#endif
3769 },
3770 {"directx",
3771#ifdef FEAT_DIRECTX
3772 1
3773#else
3774 0
3775#endif
3776 },
3777 {"dnd",
3778#ifdef FEAT_DND
3779 1
3780#else
3781 0
3782#endif
3783 },
3784 {"emacs_tags",
3785#ifdef FEAT_EMACS_TAGS
3786 1
3787#else
3788 0
3789#endif
3790 },
3791 {"eval", 1}, // always present, of course!
3792 {"ex_extra", 1}, // graduated feature
3793 {"extra_search",
3794#ifdef FEAT_SEARCH_EXTRA
3795 1
3796#else
3797 0
3798#endif
3799 },
3800 {"file_in_path",
3801#ifdef FEAT_SEARCHPATH
3802 1
3803#else
3804 0
3805#endif
3806 },
3807 {"filterpipe",
3808#if defined(FEAT_FILTERPIPE) && !defined(VIMDLL)
3809 1
3810#else
3811 0
3812#endif
3813 },
3814 {"find_in_path",
3815#ifdef FEAT_FIND_ID
3816 1
3817#else
3818 0
3819#endif
3820 },
3821 {"float",
3822#ifdef FEAT_FLOAT
3823 1
3824#else
3825 0
3826#endif
3827 },
3828 {"folding",
3829#ifdef FEAT_FOLDING
3830 1
3831#else
3832 0
3833#endif
3834 },
3835 {"footer",
3836#ifdef FEAT_FOOTER
3837 1
3838#else
3839 0
3840#endif
3841 },
3842 {"fork",
3843#if !defined(USE_SYSTEM) && defined(UNIX)
3844 1
3845#else
3846 0
3847#endif
3848 },
3849 {"gettext",
3850#ifdef FEAT_GETTEXT
3851 1
3852#else
3853 0
3854#endif
3855 },
3856 {"gui",
3857#ifdef FEAT_GUI
3858 1
3859#else
3860 0
3861#endif
3862 },
3863 {"gui_neXtaw",
3864#if defined(FEAT_GUI_ATHENA) && defined(FEAT_GUI_NEXTAW)
3865 1
3866#else
3867 0
3868#endif
3869 },
3870 {"gui_athena",
3871#if defined(FEAT_GUI_ATHENA) && !defined(FEAT_GUI_NEXTAW)
3872 1
3873#else
3874 0
3875#endif
3876 },
3877 {"gui_gtk",
3878#ifdef FEAT_GUI_GTK
3879 1
3880#else
3881 0
3882#endif
3883 },
3884 {"gui_gtk2",
3885#if defined(FEAT_GUI_GTK) && !defined(USE_GTK3)
3886 1
3887#else
3888 0
3889#endif
3890 },
3891 {"gui_gtk3",
3892#if defined(FEAT_GUI_GTK) && defined(USE_GTK3)
3893 1
3894#else
3895 0
3896#endif
3897 },
3898 {"gui_gnome",
3899#ifdef FEAT_GUI_GNOME
3900 1
3901#else
3902 0
3903#endif
3904 },
3905 {"gui_haiku",
3906#ifdef FEAT_GUI_HAIKU
3907 1
3908#else
3909 0
3910#endif
3911 },
3912 {"gui_mac",
3913#ifdef FEAT_GUI_MAC
3914 1
3915#else
3916 0
3917#endif
3918 },
3919 {"gui_motif",
3920#ifdef FEAT_GUI_MOTIF
3921 1
3922#else
3923 0
3924#endif
3925 },
3926 {"gui_photon",
3927#ifdef FEAT_GUI_PHOTON
3928 1
3929#else
3930 0
3931#endif
3932 },
3933 {"gui_win32",
3934#ifdef FEAT_GUI_MSWIN
3935 1
3936#else
3937 0
3938#endif
3939 },
3940 {"iconv",
3941#if defined(HAVE_ICONV_H) && defined(USE_ICONV)
3942 1
3943#else
3944 0
3945#endif
3946 },
3947 {"insert_expand", 1},
Bram Moolenaarbfe13cc2020-04-12 17:53:12 +02003948 {"ipv6",
3949#ifdef FEAT_IPV6
3950 1
3951#else
3952 0
3953#endif
3954 },
Bram Moolenaar79296512020-03-22 16:17:14 +01003955 {"job",
3956#ifdef FEAT_JOB_CHANNEL
3957 1
3958#else
3959 0
3960#endif
3961 },
3962 {"jumplist",
3963#ifdef FEAT_JUMPLIST
3964 1
3965#else
3966 0
3967#endif
3968 },
3969 {"keymap",
3970#ifdef FEAT_KEYMAP
3971 1
3972#else
3973 0
3974#endif
3975 },
3976 {"lambda", 1}, // always with FEAT_EVAL, since 7.4.2120 with closure
3977 {"langmap",
3978#ifdef FEAT_LANGMAP
3979 1
3980#else
3981 0
3982#endif
3983 },
3984 {"libcall",
3985#ifdef FEAT_LIBCALL
3986 1
3987#else
3988 0
3989#endif
3990 },
3991 {"linebreak",
3992#ifdef FEAT_LINEBREAK
3993 1
3994#else
3995 0
3996#endif
3997 },
3998 {"lispindent",
3999#ifdef FEAT_LISP
4000 1
4001#else
4002 0
4003#endif
4004 },
4005 {"listcmds", 1},
4006 {"localmap", 1},
4007 {"lua",
4008#if defined(FEAT_LUA) && !defined(DYNAMIC_LUA)
4009 1
4010#else
4011 0
4012#endif
4013 },
4014 {"menu",
4015#ifdef FEAT_MENU
4016 1
4017#else
4018 0
4019#endif
4020 },
4021 {"mksession",
4022#ifdef FEAT_SESSION
4023 1
4024#else
4025 0
4026#endif
4027 },
4028 {"modify_fname", 1},
4029 {"mouse", 1},
4030 {"mouseshape",
4031#ifdef FEAT_MOUSESHAPE
4032 1
4033#else
4034 0
4035#endif
4036 },
4037 {"mouse_dec",
4038#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_DEC)
4039 1
4040#else
4041 0
4042#endif
4043 },
4044 {"mouse_gpm",
4045#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_GPM)
4046 1
4047#else
4048 0
4049#endif
4050 },
4051 {"mouse_jsbterm",
4052#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_JSB)
4053 1
4054#else
4055 0
4056#endif
4057 },
4058 {"mouse_netterm",
4059#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_NET)
4060 1
4061#else
4062 0
4063#endif
4064 },
4065 {"mouse_pterm",
4066#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_PTERM)
4067 1
4068#else
4069 0
4070#endif
4071 },
4072 {"mouse_sgr",
4073#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4074 1
4075#else
4076 0
4077#endif
4078 },
4079 {"mouse_sysmouse",
4080#if (defined(UNIX) || defined(VMS)) && defined(FEAT_SYSMOUSE)
4081 1
4082#else
4083 0
4084#endif
4085 },
4086 {"mouse_urxvt",
4087#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_URXVT)
4088 1
4089#else
4090 0
4091#endif
4092 },
4093 {"mouse_xterm",
4094#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4095 1
4096#else
4097 0
4098#endif
4099 },
4100 {"multi_byte", 1},
4101 {"multi_byte_ime",
4102#ifdef FEAT_MBYTE_IME
4103 1
4104#else
4105 0
4106#endif
4107 },
4108 {"multi_lang",
4109#ifdef FEAT_MULTI_LANG
4110 1
4111#else
4112 0
4113#endif
4114 },
4115 {"mzscheme",
4116#if defined(FEAT_MZSCHEME) && !defined(DYNAMIC_MZSCHEME)
4117 1
4118#else
4119 0
4120#endif
4121 },
4122 {"num64", 1},
4123 {"ole",
4124#ifdef FEAT_OLE
4125 1
4126#else
4127 0
4128#endif
4129 },
4130 {"packages",
4131#ifdef FEAT_EVAL
4132 1
4133#else
4134 0
4135#endif
4136 },
4137 {"path_extra",
4138#ifdef FEAT_PATH_EXTRA
4139 1
4140#else
4141 0
4142#endif
4143 },
4144 {"perl",
4145#if defined(FEAT_PERL) && !defined(DYNAMIC_PERL)
4146 1
4147#else
4148 0
4149#endif
4150 },
4151 {"persistent_undo",
4152#ifdef FEAT_PERSISTENT_UNDO
4153 1
4154#else
4155 0
4156#endif
4157 },
4158 {"python_compiled",
4159#if defined(FEAT_PYTHON)
4160 1
4161#else
4162 0
4163#endif
4164 },
4165 {"python_dynamic",
4166#if defined(FEAT_PYTHON) && defined(DYNAMIC_PYTHON)
4167 1
4168#else
4169 0
4170#endif
4171 },
4172 {"python",
4173#if defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)
4174 1
4175#else
4176 0
4177#endif
4178 },
4179 {"pythonx",
4180#if (defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)) \
4181 || (defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3))
4182 1
4183#else
4184 0
4185#endif
4186 },
4187 {"python3_compiled",
4188#if defined(FEAT_PYTHON3)
4189 1
4190#else
4191 0
4192#endif
4193 },
4194 {"python3_dynamic",
4195#if defined(FEAT_PYTHON3) && defined(DYNAMIC_PYTHON3)
4196 1
4197#else
4198 0
4199#endif
4200 },
4201 {"python3",
4202#if defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3)
4203 1
4204#else
4205 0
4206#endif
4207 },
4208 {"popupwin",
4209#ifdef FEAT_PROP_POPUP
4210 1
4211#else
4212 0
4213#endif
4214 },
4215 {"postscript",
4216#ifdef FEAT_POSTSCRIPT
4217 1
4218#else
4219 0
4220#endif
4221 },
4222 {"printer",
4223#ifdef FEAT_PRINTER
4224 1
4225#else
4226 0
4227#endif
4228 },
4229 {"profile",
4230#ifdef FEAT_PROFILE
4231 1
4232#else
4233 0
4234#endif
4235 },
4236 {"reltime",
4237#ifdef FEAT_RELTIME
4238 1
4239#else
4240 0
4241#endif
4242 },
4243 {"quickfix",
4244#ifdef FEAT_QUICKFIX
4245 1
4246#else
4247 0
4248#endif
4249 },
4250 {"rightleft",
4251#ifdef FEAT_RIGHTLEFT
4252 1
4253#else
4254 0
4255#endif
4256 },
4257 {"ruby",
4258#if defined(FEAT_RUBY) && !defined(DYNAMIC_RUBY)
4259 1
4260#else
4261 0
4262#endif
4263 },
4264 {"scrollbind", 1},
4265 {"showcmd",
4266#ifdef FEAT_CMDL_INFO
4267 1
4268#else
4269 0
4270#endif
4271 },
4272 {"cmdline_info",
4273#ifdef FEAT_CMDL_INFO
4274 1
4275#else
4276 0
4277#endif
4278 },
4279 {"signs",
4280#ifdef FEAT_SIGNS
4281 1
4282#else
4283 0
4284#endif
4285 },
4286 {"smartindent",
4287#ifdef FEAT_SMARTINDENT
4288 1
4289#else
4290 0
4291#endif
4292 },
4293 {"startuptime",
4294#ifdef STARTUPTIME
4295 1
4296#else
4297 0
4298#endif
4299 },
4300 {"statusline",
4301#ifdef FEAT_STL_OPT
4302 1
4303#else
4304 0
4305#endif
4306 },
4307 {"netbeans_intg",
4308#ifdef FEAT_NETBEANS_INTG
4309 1
4310#else
4311 0
4312#endif
4313 },
4314 {"sound",
4315#ifdef FEAT_SOUND
4316 1
4317#else
4318 0
4319#endif
4320 },
4321 {"spell",
4322#ifdef FEAT_SPELL
4323 1
4324#else
4325 0
4326#endif
4327 },
4328 {"syntax",
4329#ifdef FEAT_SYN_HL
4330 1
4331#else
4332 0
4333#endif
4334 },
4335 {"system",
4336#if defined(USE_SYSTEM) || !defined(UNIX)
4337 1
4338#else
4339 0
4340#endif
4341 },
4342 {"tag_binary",
4343#ifdef FEAT_TAG_BINS
4344 1
4345#else
4346 0
4347#endif
4348 },
4349 {"tcl",
4350#if defined(FEAT_TCL) && !defined(DYNAMIC_TCL)
4351 1
4352#else
4353 0
4354#endif
4355 },
4356 {"termguicolors",
4357#ifdef FEAT_TERMGUICOLORS
4358 1
4359#else
4360 0
4361#endif
4362 },
4363 {"terminal",
4364#if defined(FEAT_TERMINAL) && !defined(MSWIN)
4365 1
4366#else
4367 0
4368#endif
4369 },
4370 {"terminfo",
4371#ifdef TERMINFO
4372 1
4373#else
4374 0
4375#endif
4376 },
4377 {"termresponse",
4378#ifdef FEAT_TERMRESPONSE
4379 1
4380#else
4381 0
4382#endif
4383 },
4384 {"textobjects",
4385#ifdef FEAT_TEXTOBJ
4386 1
4387#else
4388 0
4389#endif
4390 },
4391 {"textprop",
4392#ifdef FEAT_PROP_POPUP
4393 1
4394#else
4395 0
4396#endif
4397 },
4398 {"tgetent",
4399#ifdef HAVE_TGETENT
4400 1
4401#else
4402 0
4403#endif
4404 },
4405 {"timers",
4406#ifdef FEAT_TIMERS
4407 1
4408#else
4409 0
4410#endif
4411 },
4412 {"title",
4413#ifdef FEAT_TITLE
4414 1
4415#else
4416 0
4417#endif
4418 },
4419 {"toolbar",
4420#ifdef FEAT_TOOLBAR
4421 1
4422#else
4423 0
4424#endif
4425 },
4426 {"unnamedplus",
4427#if defined(FEAT_CLIPBOARD) && defined(FEAT_X11)
4428 1
4429#else
4430 0
4431#endif
4432 },
4433 {"user-commands", 1}, // was accidentally included in 5.4
4434 {"user_commands", 1},
4435 {"vartabs",
4436#ifdef FEAT_VARTABS
4437 1
4438#else
4439 0
4440#endif
4441 },
4442 {"vertsplit", 1},
4443 {"viminfo",
4444#ifdef FEAT_VIMINFO
4445 1
4446#else
4447 0
4448#endif
4449 },
4450 {"vimscript-1", 1},
4451 {"vimscript-2", 1},
4452 {"vimscript-3", 1},
4453 {"vimscript-4", 1},
4454 {"virtualedit", 1},
4455 {"visual", 1},
4456 {"visualextra", 1},
4457 {"vreplace", 1},
4458 {"vtp",
4459#ifdef FEAT_VTP
4460 1
4461#else
4462 0
4463#endif
4464 },
4465 {"wildignore",
4466#ifdef FEAT_WILDIGN
4467 1
4468#else
4469 0
4470#endif
4471 },
4472 {"wildmenu",
4473#ifdef FEAT_WILDMENU
4474 1
4475#else
4476 0
4477#endif
4478 },
4479 {"windows", 1},
4480 {"winaltkeys",
4481#ifdef FEAT_WAK
4482 1
4483#else
4484 0
4485#endif
4486 },
4487 {"writebackup",
4488#ifdef FEAT_WRITEBACKUP
4489 1
4490#else
4491 0
4492#endif
4493 },
4494 {"xim",
4495#ifdef FEAT_XIM
4496 1
4497#else
4498 0
4499#endif
4500 },
4501 {"xfontset",
4502#ifdef FEAT_XFONTSET
4503 1
4504#else
4505 0
4506#endif
4507 },
4508 {"xpm",
4509#if defined(FEAT_XPM_W32) || defined(HAVE_XPM)
4510 1
4511#else
4512 0
4513#endif
4514 },
4515 {"xpm_w32", // for backward compatibility
4516#ifdef FEAT_XPM_W32
4517 1
4518#else
4519 0
4520#endif
4521 },
4522 {"xsmp",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004523#ifdef USE_XSMP
Bram Moolenaar79296512020-03-22 16:17:14 +01004524 1
4525#else
4526 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004527#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004528 },
4529 {"xsmp_interact",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004530#ifdef USE_XSMP_INTERACT
Bram Moolenaar79296512020-03-22 16:17:14 +01004531 1
4532#else
4533 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004534#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004535 },
4536 {"xterm_clipboard",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004537#ifdef FEAT_XCLIPBOARD
Bram Moolenaar79296512020-03-22 16:17:14 +01004538 1
4539#else
4540 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004541#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004542 },
4543 {"xterm_save",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004544#ifdef FEAT_XTERM_SAVE
Bram Moolenaar79296512020-03-22 16:17:14 +01004545 1
4546#else
4547 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004548#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004549 },
4550 {"X11",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004551#if defined(UNIX) && defined(FEAT_X11)
Bram Moolenaar79296512020-03-22 16:17:14 +01004552 1
4553#else
4554 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004555#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004556 },
4557 {NULL, 0}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004558 };
4559
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004560 name = tv_get_string(&argvars[0]);
Bram Moolenaar79296512020-03-22 16:17:14 +01004561 for (i = 0; has_list[i].name != NULL; ++i)
4562 if (STRICMP(name, has_list[i].name) == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004563 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004564 x = TRUE;
4565 n = has_list[i].present;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004566 break;
4567 }
4568
Bram Moolenaar79296512020-03-22 16:17:14 +01004569 // features also in has_list[] but sometimes enabled at runtime
4570 if (x == TRUE && n == FALSE)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004571 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004572 if (0)
Bram Moolenaar86b9a3e2020-04-07 19:57:29 +02004573 {
4574 // intentionally empty
4575 }
Bram Moolenaar4f974752019-02-17 17:44:42 +01004576#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004577 else if (STRICMP(name, "balloon_multiline") == 0)
4578 n = multiline_balloon_available();
4579#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004580#ifdef VIMDLL
4581 else if (STRICMP(name, "filterpipe") == 0)
4582 n = gui.in_use || gui.starting;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004583#endif
4584#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
4585 else if (STRICMP(name, "iconv") == 0)
4586 n = iconv_enabled(FALSE);
4587#endif
4588#ifdef DYNAMIC_LUA
4589 else if (STRICMP(name, "lua") == 0)
4590 n = lua_enabled(FALSE);
4591#endif
4592#ifdef DYNAMIC_MZSCHEME
4593 else if (STRICMP(name, "mzscheme") == 0)
4594 n = mzscheme_enabled(FALSE);
4595#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004596#ifdef DYNAMIC_PERL
4597 else if (STRICMP(name, "perl") == 0)
4598 n = perl_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004599#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004600#ifdef DYNAMIC_PYTHON
4601 else if (STRICMP(name, "python") == 0)
4602 n = python_enabled(FALSE);
4603#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004604#ifdef DYNAMIC_PYTHON3
4605 else if (STRICMP(name, "python3") == 0)
4606 n = python3_enabled(FALSE);
4607#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01004608#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
4609 else if (STRICMP(name, "pythonx") == 0)
4610 {
4611# if defined(DYNAMIC_PYTHON) && defined(DYNAMIC_PYTHON3)
4612 if (p_pyx == 0)
4613 n = python3_enabled(FALSE) || python_enabled(FALSE);
4614 else if (p_pyx == 3)
4615 n = python3_enabled(FALSE);
4616 else if (p_pyx == 2)
4617 n = python_enabled(FALSE);
4618# elif defined(DYNAMIC_PYTHON)
4619 n = python_enabled(FALSE);
4620# elif defined(DYNAMIC_PYTHON3)
4621 n = python3_enabled(FALSE);
4622# endif
4623 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004624#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004625#ifdef DYNAMIC_RUBY
4626 else if (STRICMP(name, "ruby") == 0)
4627 n = ruby_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004628#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004629#ifdef DYNAMIC_TCL
4630 else if (STRICMP(name, "tcl") == 0)
4631 n = tcl_enabled(FALSE);
Bram Moolenaar4b8366b2019-05-04 17:34:34 +02004632#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +01004633#if defined(FEAT_TERMINAL) && defined(MSWIN)
Bram Moolenaara83e3962017-08-17 14:39:07 +02004634 else if (STRICMP(name, "terminal") == 0)
4635 n = terminal_enabled();
4636#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004637 }
4638
Bram Moolenaar79296512020-03-22 16:17:14 +01004639 // features not in has_list[]
4640 if (x == FALSE)
4641 {
4642 if (STRNICMP(name, "patch", 5) == 0)
4643 {
4644 x = TRUE;
4645 if (name[5] == '-'
4646 && STRLEN(name) >= 11
4647 && vim_isdigit(name[6])
4648 && vim_isdigit(name[8])
4649 && vim_isdigit(name[10]))
4650 {
4651 int major = atoi((char *)name + 6);
4652 int minor = atoi((char *)name + 8);
4653
4654 // Expect "patch-9.9.01234".
4655 n = (major < VIM_VERSION_MAJOR
4656 || (major == VIM_VERSION_MAJOR
4657 && (minor < VIM_VERSION_MINOR
4658 || (minor == VIM_VERSION_MINOR
4659 && has_patch(atoi((char *)name + 10))))));
4660 }
4661 else
4662 n = has_patch(atoi((char *)name + 5));
4663 }
4664 else if (STRICMP(name, "vim_starting") == 0)
4665 {
4666 x = TRUE;
4667 n = (starting != 0);
4668 }
4669 else if (STRICMP(name, "ttyin") == 0)
4670 {
4671 x = TRUE;
4672 n = mch_input_isatty();
4673 }
4674 else if (STRICMP(name, "ttyout") == 0)
4675 {
4676 x = TRUE;
4677 n = stdout_isatty;
4678 }
4679 else if (STRICMP(name, "multi_byte_encoding") == 0)
4680 {
4681 x = TRUE;
4682 n = has_mbyte;
4683 }
4684 else if (STRICMP(name, "gui_running") == 0)
4685 {
4686 x = TRUE;
4687#ifdef FEAT_GUI
4688 n = (gui.in_use || gui.starting);
4689#endif
4690 }
4691 else if (STRICMP(name, "browse") == 0)
4692 {
4693 x = TRUE;
4694#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
4695 n = gui.in_use; // gui_mch_browse() works when GUI is running
4696#endif
4697 }
4698 else if (STRICMP(name, "syntax_items") == 0)
4699 {
4700 x = TRUE;
4701#ifdef FEAT_SYN_HL
4702 n = syntax_present(curwin);
4703#endif
4704 }
4705 else if (STRICMP(name, "vcon") == 0)
4706 {
4707 x = TRUE;
4708#ifdef FEAT_VTP
4709 n = is_term_win32() && has_vtp_working();
4710#endif
4711 }
4712 else if (STRICMP(name, "netbeans_enabled") == 0)
4713 {
4714 x = TRUE;
4715#ifdef FEAT_NETBEANS_INTG
4716 n = netbeans_active();
4717#endif
4718 }
4719 else if (STRICMP(name, "mouse_gpm_enabled") == 0)
4720 {
4721 x = TRUE;
4722#ifdef FEAT_MOUSE_GPM
4723 n = gpm_enabled();
4724#endif
4725 }
4726 else if (STRICMP(name, "conpty") == 0)
4727 {
4728 x = TRUE;
4729#if defined(FEAT_TERMINAL) && defined(MSWIN)
4730 n = use_conpty();
4731#endif
4732 }
4733 else if (STRICMP(name, "clipboard_working") == 0)
4734 {
4735 x = TRUE;
4736#ifdef FEAT_CLIPBOARD
4737 n = clip_star.available;
4738#endif
4739 }
4740 }
4741
4742 if (argvars[1].v_type != VAR_UNKNOWN && tv_get_number(&argvars[1]) != 0)
4743 // return whether feature could ever be enabled
4744 rettv->vval.v_number = x;
4745 else
4746 // return whether feature is enabled
4747 rettv->vval.v_number = n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004748}
4749
4750/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004751 * "haslocaldir()" function
4752 */
4753 static void
4754f_haslocaldir(typval_T *argvars, typval_T *rettv)
4755{
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004756 tabpage_T *tp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004757 win_T *wp = NULL;
4758
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004759 wp = find_tabwin(&argvars[0], &argvars[1], &tp);
4760
4761 // Check for window-local and tab-local directories
4762 if (wp != NULL && wp->w_localdir != NULL)
4763 rettv->vval.v_number = 1;
4764 else if (tp != NULL && tp->tp_localdir != NULL)
4765 rettv->vval.v_number = 2;
4766 else
4767 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004768}
4769
4770/*
4771 * "hasmapto()" function
4772 */
4773 static void
4774f_hasmapto(typval_T *argvars, typval_T *rettv)
4775{
4776 char_u *name;
4777 char_u *mode;
4778 char_u buf[NUMBUFLEN];
4779 int abbr = FALSE;
4780
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004781 name = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004782 if (argvars[1].v_type == VAR_UNKNOWN)
4783 mode = (char_u *)"nvo";
4784 else
4785 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004786 mode = tv_get_string_buf(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004787 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004788 abbr = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004789 }
4790
4791 if (map_to_exists(name, mode, abbr))
4792 rettv->vval.v_number = TRUE;
4793 else
4794 rettv->vval.v_number = FALSE;
4795}
4796
4797/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004798 * "highlightID(name)" function
4799 */
4800 static void
4801f_hlID(typval_T *argvars, typval_T *rettv)
4802{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004803 rettv->vval.v_number = syn_name2id(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004804}
4805
4806/*
4807 * "highlight_exists()" function
4808 */
4809 static void
4810f_hlexists(typval_T *argvars, typval_T *rettv)
4811{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004812 rettv->vval.v_number = highlight_exists(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004813}
4814
4815/*
4816 * "hostname()" function
4817 */
4818 static void
4819f_hostname(typval_T *argvars UNUSED, typval_T *rettv)
4820{
4821 char_u hostname[256];
4822
4823 mch_get_host_name(hostname, 256);
4824 rettv->v_type = VAR_STRING;
4825 rettv->vval.v_string = vim_strsave(hostname);
4826}
4827
4828/*
4829 * iconv() function
4830 */
4831 static void
4832f_iconv(typval_T *argvars UNUSED, typval_T *rettv)
4833{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004834 char_u buf1[NUMBUFLEN];
4835 char_u buf2[NUMBUFLEN];
4836 char_u *from, *to, *str;
4837 vimconv_T vimconv;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004838
4839 rettv->v_type = VAR_STRING;
4840 rettv->vval.v_string = NULL;
4841
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004842 str = tv_get_string(&argvars[0]);
4843 from = enc_canonize(enc_skip(tv_get_string_buf(&argvars[1], buf1)));
4844 to = enc_canonize(enc_skip(tv_get_string_buf(&argvars[2], buf2)));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004845 vimconv.vc_type = CONV_NONE;
4846 convert_setup(&vimconv, from, to);
4847
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004848 // If the encodings are equal, no conversion needed.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004849 if (vimconv.vc_type == CONV_NONE)
4850 rettv->vval.v_string = vim_strsave(str);
4851 else
4852 rettv->vval.v_string = string_convert(&vimconv, str, NULL);
4853
4854 convert_setup(&vimconv, NULL, NULL);
4855 vim_free(from);
4856 vim_free(to);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004857}
4858
4859/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004860 * "index()" function
4861 */
4862 static void
4863f_index(typval_T *argvars, typval_T *rettv)
4864{
4865 list_T *l;
4866 listitem_T *item;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004867 blob_T *b;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004868 long idx = 0;
4869 int ic = FALSE;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004870 int error = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004871
4872 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004873 if (argvars[0].v_type == VAR_BLOB)
4874 {
4875 typval_T tv;
4876 int start = 0;
4877
4878 if (argvars[2].v_type != VAR_UNKNOWN)
4879 {
4880 start = tv_get_number_chk(&argvars[2], &error);
4881 if (error)
4882 return;
4883 }
4884 b = argvars[0].vval.v_blob;
4885 if (b == NULL)
4886 return;
Bram Moolenaar05500ec2019-01-13 19:10:33 +01004887 if (start < 0)
4888 {
4889 start = blob_len(b) + start;
4890 if (start < 0)
4891 start = 0;
4892 }
4893
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004894 for (idx = start; idx < blob_len(b); ++idx)
4895 {
4896 tv.v_type = VAR_NUMBER;
4897 tv.vval.v_number = blob_get(b, idx);
4898 if (tv_equal(&tv, &argvars[1], ic, FALSE))
4899 {
4900 rettv->vval.v_number = idx;
4901 return;
4902 }
4903 }
4904 return;
4905 }
4906 else if (argvars[0].v_type != VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004907 {
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01004908 emsg(_(e_listblobreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004909 return;
4910 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004911
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004912 l = argvars[0].vval.v_list;
4913 if (l != NULL)
4914 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01004915 range_list_materialize(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004916 item = l->lv_first;
4917 if (argvars[2].v_type != VAR_UNKNOWN)
4918 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004919 // Start at specified item. Use the cached index that list_find()
4920 // sets, so that a negative number also works.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004921 item = list_find(l, (long)tv_get_number_chk(&argvars[2], &error));
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01004922 idx = l->lv_u.mat.lv_idx;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004923 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004924 ic = (int)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004925 if (error)
4926 item = NULL;
4927 }
4928
4929 for ( ; item != NULL; item = item->li_next, ++idx)
4930 if (tv_equal(&item->li_tv, &argvars[1], ic, FALSE))
4931 {
4932 rettv->vval.v_number = idx;
4933 break;
4934 }
4935 }
4936}
4937
4938static int inputsecret_flag = 0;
4939
4940/*
4941 * "input()" function
4942 * Also handles inputsecret() when inputsecret is set.
4943 */
4944 static void
4945f_input(typval_T *argvars, typval_T *rettv)
4946{
4947 get_user_input(argvars, rettv, FALSE, inputsecret_flag);
4948}
4949
4950/*
4951 * "inputdialog()" function
4952 */
4953 static void
4954f_inputdialog(typval_T *argvars, typval_T *rettv)
4955{
4956#if defined(FEAT_GUI_TEXTDIALOG)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004957 // Use a GUI dialog if the GUI is running and 'c' is not in 'guioptions'
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004958 if (gui.in_use && vim_strchr(p_go, GO_CONDIALOG) == NULL)
4959 {
4960 char_u *message;
4961 char_u buf[NUMBUFLEN];
4962 char_u *defstr = (char_u *)"";
4963
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004964 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004965 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004966 && (defstr = tv_get_string_buf_chk(&argvars[1], buf)) != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004967 vim_strncpy(IObuff, defstr, IOSIZE - 1);
4968 else
4969 IObuff[0] = NUL;
4970 if (message != NULL && defstr != NULL
4971 && do_dialog(VIM_QUESTION, NULL, message,
4972 (char_u *)_("&OK\n&Cancel"), 1, IObuff, FALSE) == 1)
4973 rettv->vval.v_string = vim_strsave(IObuff);
4974 else
4975 {
4976 if (message != NULL && defstr != NULL
4977 && argvars[1].v_type != VAR_UNKNOWN
4978 && argvars[2].v_type != VAR_UNKNOWN)
4979 rettv->vval.v_string = vim_strsave(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004980 tv_get_string_buf(&argvars[2], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004981 else
4982 rettv->vval.v_string = NULL;
4983 }
4984 rettv->v_type = VAR_STRING;
4985 }
4986 else
4987#endif
4988 get_user_input(argvars, rettv, TRUE, inputsecret_flag);
4989}
4990
4991/*
4992 * "inputlist()" function
4993 */
4994 static void
4995f_inputlist(typval_T *argvars, typval_T *rettv)
4996{
Bram Moolenaar50985eb2020-01-27 22:09:39 +01004997 list_T *l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004998 listitem_T *li;
4999 int selected;
5000 int mouse_used;
5001
5002#ifdef NO_CONSOLE_INPUT
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005003 // While starting up, there is no place to enter text. When running tests
5004 // with --not-a-term we assume feedkeys() will be used.
Bram Moolenaar91d348a2017-07-29 20:16:03 +02005005 if (no_console_input() && !is_not_a_term())
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005006 return;
5007#endif
5008 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
5009 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005010 semsg(_(e_listarg), "inputlist()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005011 return;
5012 }
5013
5014 msg_start();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005015 msg_row = Rows - 1; // for when 'cmdheight' > 1
5016 lines_left = Rows; // avoid more prompt
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005017 msg_scroll = TRUE;
5018 msg_clr_eos();
5019
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005020 l = argvars[0].vval.v_list;
5021 range_list_materialize(l);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02005022 FOR_ALL_LIST_ITEMS(l, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005023 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005024 msg_puts((char *)tv_get_string(&li->li_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005025 msg_putchar('\n');
5026 }
5027
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005028 // Ask for choice.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005029 selected = prompt_for_number(&mouse_used);
5030 if (mouse_used)
5031 selected -= lines_left;
5032
5033 rettv->vval.v_number = selected;
5034}
5035
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005036static garray_T ga_userinput = {0, 0, sizeof(tasave_T), 4, NULL};
5037
5038/*
5039 * "inputrestore()" function
5040 */
5041 static void
5042f_inputrestore(typval_T *argvars UNUSED, typval_T *rettv)
5043{
5044 if (ga_userinput.ga_len > 0)
5045 {
5046 --ga_userinput.ga_len;
5047 restore_typeahead((tasave_T *)(ga_userinput.ga_data)
5048 + ga_userinput.ga_len);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005049 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005050 }
5051 else if (p_verbose > 1)
5052 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005053 verb_msg(_("called inputrestore() more often than inputsave()"));
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005054 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005055 }
5056}
5057
5058/*
5059 * "inputsave()" function
5060 */
5061 static void
5062f_inputsave(typval_T *argvars UNUSED, typval_T *rettv)
5063{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005064 // Add an entry to the stack of typeahead storage.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005065 if (ga_grow(&ga_userinput, 1) == OK)
5066 {
5067 save_typeahead((tasave_T *)(ga_userinput.ga_data)
5068 + ga_userinput.ga_len);
5069 ++ga_userinput.ga_len;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005070 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005071 }
5072 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005073 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005074}
5075
5076/*
5077 * "inputsecret()" function
5078 */
5079 static void
5080f_inputsecret(typval_T *argvars, typval_T *rettv)
5081{
5082 ++cmdline_star;
5083 ++inputsecret_flag;
5084 f_input(argvars, rettv);
5085 --cmdline_star;
5086 --inputsecret_flag;
5087}
5088
5089/*
Bram Moolenaar67a2deb2019-11-25 00:05:32 +01005090 * "interrupt()" function
5091 */
5092 static void
5093f_interrupt(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5094{
5095 got_int = TRUE;
5096}
5097
5098/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005099 * "invert(expr)" function
5100 */
5101 static void
5102f_invert(typval_T *argvars, typval_T *rettv)
5103{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005104 rettv->vval.v_number = ~tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005105}
5106
5107/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005108 * Return TRUE if typeval "tv" is locked: Either that value is locked itself
5109 * or it refers to a List or Dictionary that is locked.
5110 */
5111 static int
5112tv_islocked(typval_T *tv)
5113{
5114 return (tv->v_lock & VAR_LOCKED)
5115 || (tv->v_type == VAR_LIST
5116 && tv->vval.v_list != NULL
5117 && (tv->vval.v_list->lv_lock & VAR_LOCKED))
5118 || (tv->v_type == VAR_DICT
5119 && tv->vval.v_dict != NULL
5120 && (tv->vval.v_dict->dv_lock & VAR_LOCKED));
5121}
5122
5123/*
5124 * "islocked()" function
5125 */
5126 static void
5127f_islocked(typval_T *argvars, typval_T *rettv)
5128{
5129 lval_T lv;
5130 char_u *end;
5131 dictitem_T *di;
5132
5133 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005134 end = get_lval(tv_get_string(&argvars[0]), NULL, &lv, FALSE, FALSE,
Bram Moolenaar3a257732017-02-21 20:47:13 +01005135 GLV_NO_AUTOLOAD | GLV_READ_ONLY, FNE_CHECK_START);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005136 if (end != NULL && lv.ll_name != NULL)
5137 {
5138 if (*end != NUL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005139 emsg(_(e_trailing));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005140 else
5141 {
5142 if (lv.ll_tv == NULL)
5143 {
Bram Moolenaar79518e22017-02-17 16:31:35 +01005144 di = find_var(lv.ll_name, NULL, TRUE);
5145 if (di != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005146 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005147 // Consider a variable locked when:
5148 // 1. the variable itself is locked
5149 // 2. the value of the variable is locked.
5150 // 3. the List or Dict value is locked.
Bram Moolenaar79518e22017-02-17 16:31:35 +01005151 rettv->vval.v_number = ((di->di_flags & DI_FLAGS_LOCK)
5152 || tv_islocked(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005153 }
5154 }
5155 else if (lv.ll_range)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005156 emsg(_("E786: Range not allowed"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005157 else if (lv.ll_newkey != NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005158 semsg(_(e_dictkey), lv.ll_newkey);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005159 else if (lv.ll_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005160 // List item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005161 rettv->vval.v_number = tv_islocked(&lv.ll_li->li_tv);
5162 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005163 // Dictionary item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005164 rettv->vval.v_number = tv_islocked(&lv.ll_di->di_tv);
5165 }
5166 }
5167
5168 clear_lval(&lv);
5169}
5170
5171#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
5172/*
Bram Moolenaarfda1bff2019-04-04 13:44:37 +02005173 * "isinf()" function
5174 */
5175 static void
5176f_isinf(typval_T *argvars, typval_T *rettv)
5177{
5178 if (argvars[0].v_type == VAR_FLOAT && isinf(argvars[0].vval.v_float))
5179 rettv->vval.v_number = argvars[0].vval.v_float > 0.0 ? 1 : -1;
5180}
5181
5182/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005183 * "isnan()" function
5184 */
5185 static void
5186f_isnan(typval_T *argvars, typval_T *rettv)
5187{
5188 rettv->vval.v_number = argvars[0].v_type == VAR_FLOAT
5189 && isnan(argvars[0].vval.v_float);
5190}
5191#endif
5192
5193/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005194 * "last_buffer_nr()" function.
5195 */
5196 static void
5197f_last_buffer_nr(typval_T *argvars UNUSED, typval_T *rettv)
5198{
5199 int n = 0;
5200 buf_T *buf;
5201
Bram Moolenaar29323592016-07-24 22:04:11 +02005202 FOR_ALL_BUFFERS(buf)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005203 if (n < buf->b_fnum)
5204 n = buf->b_fnum;
5205
5206 rettv->vval.v_number = n;
5207}
5208
5209/*
5210 * "len()" function
5211 */
5212 static void
5213f_len(typval_T *argvars, typval_T *rettv)
5214{
5215 switch (argvars[0].v_type)
5216 {
5217 case VAR_STRING:
5218 case VAR_NUMBER:
5219 rettv->vval.v_number = (varnumber_T)STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005220 tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005221 break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005222 case VAR_BLOB:
5223 rettv->vval.v_number = blob_len(argvars[0].vval.v_blob);
5224 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005225 case VAR_LIST:
5226 rettv->vval.v_number = list_len(argvars[0].vval.v_list);
5227 break;
5228 case VAR_DICT:
5229 rettv->vval.v_number = dict_len(argvars[0].vval.v_dict);
5230 break;
5231 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02005232 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01005233 case VAR_VOID:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01005234 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005235 case VAR_SPECIAL:
5236 case VAR_FLOAT:
5237 case VAR_FUNC:
5238 case VAR_PARTIAL:
5239 case VAR_JOB:
5240 case VAR_CHANNEL:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005241 emsg(_("E701: Invalid type for len()"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005242 break;
5243 }
5244}
5245
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005246 static void
Bram Moolenaar6d721c72017-01-17 16:56:28 +01005247libcall_common(typval_T *argvars UNUSED, typval_T *rettv, int type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005248{
5249#ifdef FEAT_LIBCALL
5250 char_u *string_in;
5251 char_u **string_result;
5252 int nr_result;
5253#endif
5254
5255 rettv->v_type = type;
5256 if (type != VAR_NUMBER)
5257 rettv->vval.v_string = NULL;
5258
5259 if (check_restricted() || check_secure())
5260 return;
5261
5262#ifdef FEAT_LIBCALL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005263 // The first two args must be strings, otherwise it's meaningless
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005264 if (argvars[0].v_type == VAR_STRING && argvars[1].v_type == VAR_STRING)
5265 {
5266 string_in = NULL;
5267 if (argvars[2].v_type == VAR_STRING)
5268 string_in = argvars[2].vval.v_string;
5269 if (type == VAR_NUMBER)
5270 string_result = NULL;
5271 else
5272 string_result = &rettv->vval.v_string;
5273 if (mch_libcall(argvars[0].vval.v_string,
5274 argvars[1].vval.v_string,
5275 string_in,
5276 argvars[2].vval.v_number,
5277 string_result,
5278 &nr_result) == OK
5279 && type == VAR_NUMBER)
5280 rettv->vval.v_number = nr_result;
5281 }
5282#endif
5283}
5284
5285/*
5286 * "libcall()" function
5287 */
5288 static void
5289f_libcall(typval_T *argvars, typval_T *rettv)
5290{
5291 libcall_common(argvars, rettv, VAR_STRING);
5292}
5293
5294/*
5295 * "libcallnr()" function
5296 */
5297 static void
5298f_libcallnr(typval_T *argvars, typval_T *rettv)
5299{
5300 libcall_common(argvars, rettv, VAR_NUMBER);
5301}
5302
5303/*
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005304 * "line(string, [winid])" function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005305 */
5306 static void
5307f_line(typval_T *argvars, typval_T *rettv)
5308{
5309 linenr_T lnum = 0;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005310 pos_T *fp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005311 int fnum;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005312 int id;
5313 tabpage_T *tp;
5314 win_T *wp;
5315 win_T *save_curwin;
5316 tabpage_T *save_curtab;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005317
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005318 if (argvars[1].v_type != VAR_UNKNOWN)
5319 {
5320 // use window specified in the second argument
5321 id = (int)tv_get_number(&argvars[1]);
5322 wp = win_id2wp_tp(id, &tp);
5323 if (wp != NULL && tp != NULL)
5324 {
5325 if (switch_win_noblock(&save_curwin, &save_curtab, wp, tp, TRUE)
5326 == OK)
5327 {
5328 check_cursor();
5329 fp = var2fpos(&argvars[0], TRUE, &fnum);
5330 }
5331 restore_win_noblock(save_curwin, save_curtab, TRUE);
5332 }
5333 }
5334 else
5335 // use current window
5336 fp = var2fpos(&argvars[0], TRUE, &fnum);
5337
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005338 if (fp != NULL)
5339 lnum = fp->lnum;
5340 rettv->vval.v_number = lnum;
5341}
5342
5343/*
5344 * "line2byte(lnum)" function
5345 */
5346 static void
5347f_line2byte(typval_T *argvars UNUSED, typval_T *rettv)
5348{
5349#ifndef FEAT_BYTEOFF
5350 rettv->vval.v_number = -1;
5351#else
5352 linenr_T lnum;
5353
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005354 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005355 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count + 1)
5356 rettv->vval.v_number = -1;
5357 else
5358 rettv->vval.v_number = ml_find_line_or_offset(curbuf, lnum, NULL);
5359 if (rettv->vval.v_number >= 0)
5360 ++rettv->vval.v_number;
5361#endif
5362}
5363
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005364#ifdef FEAT_FLOAT
5365/*
5366 * "log()" function
5367 */
5368 static void
5369f_log(typval_T *argvars, typval_T *rettv)
5370{
5371 float_T f = 0.0;
5372
5373 rettv->v_type = VAR_FLOAT;
5374 if (get_float_arg(argvars, &f) == OK)
5375 rettv->vval.v_float = log(f);
5376 else
5377 rettv->vval.v_float = 0.0;
5378}
5379
5380/*
5381 * "log10()" function
5382 */
5383 static void
5384f_log10(typval_T *argvars, typval_T *rettv)
5385{
5386 float_T f = 0.0;
5387
5388 rettv->v_type = VAR_FLOAT;
5389 if (get_float_arg(argvars, &f) == OK)
5390 rettv->vval.v_float = log10(f);
5391 else
5392 rettv->vval.v_float = 0.0;
5393}
5394#endif
5395
5396#ifdef FEAT_LUA
5397/*
5398 * "luaeval()" function
5399 */
5400 static void
5401f_luaeval(typval_T *argvars, typval_T *rettv)
5402{
5403 char_u *str;
5404 char_u buf[NUMBUFLEN];
5405
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005406 if (check_restricted() || check_secure())
5407 return;
5408
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005409 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005410 do_luaeval(str, argvars + 1, rettv);
5411}
5412#endif
5413
5414/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005415 * "maparg()" function
5416 */
5417 static void
5418f_maparg(typval_T *argvars, typval_T *rettv)
5419{
5420 get_maparg(argvars, rettv, TRUE);
5421}
5422
5423/*
5424 * "mapcheck()" function
5425 */
5426 static void
5427f_mapcheck(typval_T *argvars, typval_T *rettv)
5428{
5429 get_maparg(argvars, rettv, FALSE);
5430}
5431
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005432typedef enum
5433{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005434 MATCH_END, // matchend()
5435 MATCH_MATCH, // match()
5436 MATCH_STR, // matchstr()
5437 MATCH_LIST, // matchlist()
5438 MATCH_POS // matchstrpos()
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005439} matchtype_T;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005440
5441 static void
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005442find_some_match(typval_T *argvars, typval_T *rettv, matchtype_T type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005443{
5444 char_u *str = NULL;
5445 long len = 0;
5446 char_u *expr = NULL;
5447 char_u *pat;
5448 regmatch_T regmatch;
5449 char_u patbuf[NUMBUFLEN];
5450 char_u strbuf[NUMBUFLEN];
5451 char_u *save_cpo;
5452 long start = 0;
5453 long nth = 1;
5454 colnr_T startcol = 0;
5455 int match = 0;
5456 list_T *l = NULL;
5457 listitem_T *li = NULL;
5458 long idx = 0;
5459 char_u *tofree = NULL;
5460
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005461 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005462 save_cpo = p_cpo;
5463 p_cpo = (char_u *)"";
5464
5465 rettv->vval.v_number = -1;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005466 if (type == MATCH_LIST || type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005467 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005468 // type MATCH_LIST: return empty list when there are no matches.
5469 // type MATCH_POS: return ["", -1, -1, -1]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005470 if (rettv_list_alloc(rettv) == FAIL)
5471 goto theend;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005472 if (type == MATCH_POS
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005473 && (list_append_string(rettv->vval.v_list,
5474 (char_u *)"", 0) == FAIL
5475 || list_append_number(rettv->vval.v_list,
5476 (varnumber_T)-1) == FAIL
5477 || list_append_number(rettv->vval.v_list,
5478 (varnumber_T)-1) == FAIL
5479 || list_append_number(rettv->vval.v_list,
5480 (varnumber_T)-1) == FAIL))
5481 {
5482 list_free(rettv->vval.v_list);
5483 rettv->vval.v_list = NULL;
5484 goto theend;
5485 }
5486 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005487 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005488 {
5489 rettv->v_type = VAR_STRING;
5490 rettv->vval.v_string = NULL;
5491 }
5492
5493 if (argvars[0].v_type == VAR_LIST)
5494 {
5495 if ((l = argvars[0].vval.v_list) == NULL)
5496 goto theend;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005497 range_list_materialize(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005498 li = l->lv_first;
5499 }
5500 else
5501 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005502 expr = str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005503 len = (long)STRLEN(str);
5504 }
5505
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005506 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005507 if (pat == NULL)
5508 goto theend;
5509
5510 if (argvars[2].v_type != VAR_UNKNOWN)
5511 {
5512 int error = FALSE;
5513
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005514 start = (long)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005515 if (error)
5516 goto theend;
5517 if (l != NULL)
5518 {
5519 li = list_find(l, start);
5520 if (li == NULL)
5521 goto theend;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01005522 idx = l->lv_u.mat.lv_idx; // use the cached index
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005523 }
5524 else
5525 {
5526 if (start < 0)
5527 start = 0;
5528 if (start > len)
5529 goto theend;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005530 // When "count" argument is there ignore matches before "start",
5531 // otherwise skip part of the string. Differs when pattern is "^"
5532 // or "\<".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005533 if (argvars[3].v_type != VAR_UNKNOWN)
5534 startcol = start;
5535 else
5536 {
5537 str += start;
5538 len -= start;
5539 }
5540 }
5541
5542 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005543 nth = (long)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005544 if (error)
5545 goto theend;
5546 }
5547
5548 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
5549 if (regmatch.regprog != NULL)
5550 {
5551 regmatch.rm_ic = p_ic;
5552
5553 for (;;)
5554 {
5555 if (l != NULL)
5556 {
5557 if (li == NULL)
5558 {
5559 match = FALSE;
5560 break;
5561 }
5562 vim_free(tofree);
5563 expr = str = echo_string(&li->li_tv, &tofree, strbuf, 0);
5564 if (str == NULL)
5565 break;
5566 }
5567
5568 match = vim_regexec_nl(&regmatch, str, (colnr_T)startcol);
5569
5570 if (match && --nth <= 0)
5571 break;
5572 if (l == NULL && !match)
5573 break;
5574
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005575 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005576 if (l != NULL)
5577 {
5578 li = li->li_next;
5579 ++idx;
5580 }
5581 else
5582 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005583 startcol = (colnr_T)(regmatch.startp[0]
5584 + (*mb_ptr2len)(regmatch.startp[0]) - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005585 if (startcol > (colnr_T)len
5586 || str + startcol <= regmatch.startp[0])
5587 {
5588 match = FALSE;
5589 break;
5590 }
5591 }
5592 }
5593
5594 if (match)
5595 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005596 if (type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005597 {
5598 listitem_T *li1 = rettv->vval.v_list->lv_first;
5599 listitem_T *li2 = li1->li_next;
5600 listitem_T *li3 = li2->li_next;
5601 listitem_T *li4 = li3->li_next;
5602
5603 vim_free(li1->li_tv.vval.v_string);
5604 li1->li_tv.vval.v_string = vim_strnsave(regmatch.startp[0],
5605 (int)(regmatch.endp[0] - regmatch.startp[0]));
5606 li3->li_tv.vval.v_number =
5607 (varnumber_T)(regmatch.startp[0] - expr);
5608 li4->li_tv.vval.v_number =
5609 (varnumber_T)(regmatch.endp[0] - expr);
5610 if (l != NULL)
5611 li2->li_tv.vval.v_number = (varnumber_T)idx;
5612 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005613 else if (type == MATCH_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005614 {
5615 int i;
5616
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005617 // return list with matched string and submatches
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005618 for (i = 0; i < NSUBEXP; ++i)
5619 {
5620 if (regmatch.endp[i] == NULL)
5621 {
5622 if (list_append_string(rettv->vval.v_list,
5623 (char_u *)"", 0) == FAIL)
5624 break;
5625 }
5626 else if (list_append_string(rettv->vval.v_list,
5627 regmatch.startp[i],
5628 (int)(regmatch.endp[i] - regmatch.startp[i]))
5629 == FAIL)
5630 break;
5631 }
5632 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005633 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005634 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005635 // return matched string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005636 if (l != NULL)
5637 copy_tv(&li->li_tv, rettv);
5638 else
5639 rettv->vval.v_string = vim_strnsave(regmatch.startp[0],
5640 (int)(regmatch.endp[0] - regmatch.startp[0]));
5641 }
5642 else if (l != NULL)
5643 rettv->vval.v_number = idx;
5644 else
5645 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005646 if (type != MATCH_END)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005647 rettv->vval.v_number =
5648 (varnumber_T)(regmatch.startp[0] - str);
5649 else
5650 rettv->vval.v_number =
5651 (varnumber_T)(regmatch.endp[0] - str);
5652 rettv->vval.v_number += (varnumber_T)(str - expr);
5653 }
5654 }
5655 vim_regfree(regmatch.regprog);
5656 }
5657
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005658theend:
5659 if (type == MATCH_POS && l == NULL && rettv->vval.v_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005660 // matchstrpos() without a list: drop the second item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005661 listitem_remove(rettv->vval.v_list,
5662 rettv->vval.v_list->lv_first->li_next);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005663 vim_free(tofree);
5664 p_cpo = save_cpo;
5665}
5666
5667/*
5668 * "match()" function
5669 */
5670 static void
5671f_match(typval_T *argvars, typval_T *rettv)
5672{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005673 find_some_match(argvars, rettv, MATCH_MATCH);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005674}
5675
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005676/*
5677 * "matchend()" function
5678 */
5679 static void
5680f_matchend(typval_T *argvars, typval_T *rettv)
5681{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005682 find_some_match(argvars, rettv, MATCH_END);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005683}
5684
5685/*
5686 * "matchlist()" function
5687 */
5688 static void
5689f_matchlist(typval_T *argvars, typval_T *rettv)
5690{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005691 find_some_match(argvars, rettv, MATCH_LIST);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005692}
5693
5694/*
5695 * "matchstr()" function
5696 */
5697 static void
5698f_matchstr(typval_T *argvars, typval_T *rettv)
5699{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005700 find_some_match(argvars, rettv, MATCH_STR);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005701}
5702
5703/*
5704 * "matchstrpos()" function
5705 */
5706 static void
5707f_matchstrpos(typval_T *argvars, typval_T *rettv)
5708{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005709 find_some_match(argvars, rettv, MATCH_POS);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005710}
5711
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005712 static void
5713max_min(typval_T *argvars, typval_T *rettv, int domax)
5714{
5715 varnumber_T n = 0;
5716 varnumber_T i;
5717 int error = FALSE;
5718
5719 if (argvars[0].v_type == VAR_LIST)
5720 {
5721 list_T *l;
5722 listitem_T *li;
5723
5724 l = argvars[0].vval.v_list;
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005725 if (l != NULL && l->lv_len > 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005726 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005727 if (l->lv_first == &range_list_item)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005728 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005729 if ((l->lv_u.nonmat.lv_stride > 0) ^ domax)
5730 n = l->lv_u.nonmat.lv_start;
5731 else
5732 n = l->lv_u.nonmat.lv_start + (l->lv_len - 1)
5733 * l->lv_u.nonmat.lv_stride;
5734 }
5735 else
5736 {
5737 li = l->lv_first;
5738 if (li != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005739 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005740 n = tv_get_number_chk(&li->li_tv, &error);
5741 for (;;)
5742 {
5743 li = li->li_next;
5744 if (li == NULL)
5745 break;
5746 i = tv_get_number_chk(&li->li_tv, &error);
5747 if (domax ? i > n : i < n)
5748 n = i;
5749 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005750 }
5751 }
5752 }
5753 }
5754 else if (argvars[0].v_type == VAR_DICT)
5755 {
5756 dict_T *d;
5757 int first = TRUE;
5758 hashitem_T *hi;
5759 int todo;
5760
5761 d = argvars[0].vval.v_dict;
5762 if (d != NULL)
5763 {
5764 todo = (int)d->dv_hashtab.ht_used;
5765 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
5766 {
5767 if (!HASHITEM_EMPTY(hi))
5768 {
5769 --todo;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005770 i = tv_get_number_chk(&HI2DI(hi)->di_tv, &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005771 if (first)
5772 {
5773 n = i;
5774 first = FALSE;
5775 }
5776 else if (domax ? i > n : i < n)
5777 n = i;
5778 }
5779 }
5780 }
5781 }
5782 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005783 semsg(_(e_listdictarg), domax ? "max()" : "min()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005784 rettv->vval.v_number = error ? 0 : n;
5785}
5786
5787/*
5788 * "max()" function
5789 */
5790 static void
5791f_max(typval_T *argvars, typval_T *rettv)
5792{
5793 max_min(argvars, rettv, TRUE);
5794}
5795
5796/*
5797 * "min()" function
5798 */
5799 static void
5800f_min(typval_T *argvars, typval_T *rettv)
5801{
5802 max_min(argvars, rettv, FALSE);
5803}
5804
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005805#if defined(FEAT_MZSCHEME) || defined(PROTO)
5806/*
5807 * "mzeval()" function
5808 */
5809 static void
5810f_mzeval(typval_T *argvars, typval_T *rettv)
5811{
5812 char_u *str;
5813 char_u buf[NUMBUFLEN];
5814
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005815 if (check_restricted() || check_secure())
5816 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005817 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005818 do_mzeval(str, rettv);
5819}
5820
5821 void
5822mzscheme_call_vim(char_u *name, typval_T *args, typval_T *rettv)
5823{
5824 typval_T argvars[3];
5825
5826 argvars[0].v_type = VAR_STRING;
5827 argvars[0].vval.v_string = name;
5828 copy_tv(args, &argvars[1]);
5829 argvars[2].v_type = VAR_UNKNOWN;
5830 f_call(argvars, rettv);
5831 clear_tv(&argvars[1]);
5832}
5833#endif
5834
5835/*
5836 * "nextnonblank()" function
5837 */
5838 static void
5839f_nextnonblank(typval_T *argvars, typval_T *rettv)
5840{
5841 linenr_T lnum;
5842
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005843 for (lnum = tv_get_lnum(argvars); ; ++lnum)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005844 {
5845 if (lnum < 0 || lnum > curbuf->b_ml.ml_line_count)
5846 {
5847 lnum = 0;
5848 break;
5849 }
5850 if (*skipwhite(ml_get(lnum)) != NUL)
5851 break;
5852 }
5853 rettv->vval.v_number = lnum;
5854}
5855
5856/*
5857 * "nr2char()" function
5858 */
5859 static void
5860f_nr2char(typval_T *argvars, typval_T *rettv)
5861{
5862 char_u buf[NUMBUFLEN];
5863
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005864 if (has_mbyte)
5865 {
5866 int utf8 = 0;
5867
5868 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005869 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005870 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01005871 buf[utf_char2bytes((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005872 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005873 buf[(*mb_char2bytes)((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005874 }
5875 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005876 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005877 buf[0] = (char_u)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005878 buf[1] = NUL;
5879 }
5880 rettv->v_type = VAR_STRING;
5881 rettv->vval.v_string = vim_strsave(buf);
5882}
5883
5884/*
5885 * "or(expr, expr)" function
5886 */
5887 static void
5888f_or(typval_T *argvars, typval_T *rettv)
5889{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005890 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
5891 | tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005892}
5893
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005894#ifdef FEAT_PERL
5895/*
5896 * "perleval()" function
5897 */
5898 static void
5899f_perleval(typval_T *argvars, typval_T *rettv)
5900{
5901 char_u *str;
5902 char_u buf[NUMBUFLEN];
5903
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005904 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005905 do_perleval(str, rettv);
5906}
5907#endif
5908
5909#ifdef FEAT_FLOAT
5910/*
5911 * "pow()" function
5912 */
5913 static void
5914f_pow(typval_T *argvars, typval_T *rettv)
5915{
5916 float_T fx = 0.0, fy = 0.0;
5917
5918 rettv->v_type = VAR_FLOAT;
5919 if (get_float_arg(argvars, &fx) == OK
5920 && get_float_arg(&argvars[1], &fy) == OK)
5921 rettv->vval.v_float = pow(fx, fy);
5922 else
5923 rettv->vval.v_float = 0.0;
5924}
5925#endif
5926
5927/*
5928 * "prevnonblank()" function
5929 */
5930 static void
5931f_prevnonblank(typval_T *argvars, typval_T *rettv)
5932{
5933 linenr_T lnum;
5934
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005935 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005936 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count)
5937 lnum = 0;
5938 else
5939 while (lnum >= 1 && *skipwhite(ml_get(lnum)) == NUL)
5940 --lnum;
5941 rettv->vval.v_number = lnum;
5942}
5943
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005944// This dummy va_list is here because:
5945// - passing a NULL pointer doesn't work when va_list isn't a pointer
5946// - locally in the function results in a "used before set" warning
5947// - using va_start() to initialize it gives "function with fixed args" error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005948static va_list ap;
5949
5950/*
5951 * "printf()" function
5952 */
5953 static void
5954f_printf(typval_T *argvars, typval_T *rettv)
5955{
5956 char_u buf[NUMBUFLEN];
5957 int len;
5958 char_u *s;
5959 int saved_did_emsg = did_emsg;
5960 char *fmt;
5961
5962 rettv->v_type = VAR_STRING;
5963 rettv->vval.v_string = NULL;
5964
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005965 // Get the required length, allocate the buffer and do it for real.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005966 did_emsg = FALSE;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005967 fmt = (char *)tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02005968 len = vim_vsnprintf_typval(NULL, 0, fmt, ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005969 if (!did_emsg)
5970 {
5971 s = alloc(len + 1);
5972 if (s != NULL)
5973 {
5974 rettv->vval.v_string = s;
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02005975 (void)vim_vsnprintf_typval((char *)s, len + 1, fmt,
5976 ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005977 }
5978 }
5979 did_emsg |= saved_did_emsg;
5980}
5981
5982/*
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005983 * "pum_getpos()" function
5984 */
5985 static void
5986f_pum_getpos(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5987{
5988 if (rettv_dict_alloc(rettv) != OK)
5989 return;
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005990 pum_set_event_info(rettv->vval.v_dict);
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005991}
5992
5993/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005994 * "pumvisible()" function
5995 */
5996 static void
5997f_pumvisible(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5998{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005999 if (pum_visible())
6000 rettv->vval.v_number = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006001}
6002
6003#ifdef FEAT_PYTHON3
6004/*
6005 * "py3eval()" function
6006 */
6007 static void
6008f_py3eval(typval_T *argvars, typval_T *rettv)
6009{
6010 char_u *str;
6011 char_u buf[NUMBUFLEN];
6012
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006013 if (check_restricted() || check_secure())
6014 return;
6015
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006016 if (p_pyx == 0)
6017 p_pyx = 3;
6018
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006019 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006020 do_py3eval(str, rettv);
6021}
6022#endif
6023
6024#ifdef FEAT_PYTHON
6025/*
6026 * "pyeval()" function
6027 */
6028 static void
6029f_pyeval(typval_T *argvars, typval_T *rettv)
6030{
6031 char_u *str;
6032 char_u buf[NUMBUFLEN];
6033
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006034 if (check_restricted() || check_secure())
6035 return;
6036
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006037 if (p_pyx == 0)
6038 p_pyx = 2;
6039
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006040 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006041 do_pyeval(str, rettv);
6042}
6043#endif
6044
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006045#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
6046/*
6047 * "pyxeval()" function
6048 */
6049 static void
6050f_pyxeval(typval_T *argvars, typval_T *rettv)
6051{
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006052 if (check_restricted() || check_secure())
6053 return;
6054
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006055# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
6056 init_pyxversion();
6057 if (p_pyx == 2)
6058 f_pyeval(argvars, rettv);
6059 else
6060 f_py3eval(argvars, rettv);
6061# elif defined(FEAT_PYTHON)
6062 f_pyeval(argvars, rettv);
6063# elif defined(FEAT_PYTHON3)
6064 f_py3eval(argvars, rettv);
6065# endif
6066}
6067#endif
6068
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006069static UINT32_T srand_seed_for_testing = 0;
6070static int srand_seed_for_testing_is_used = FALSE;
6071
6072 static void
6073f_test_srand_seed(typval_T *argvars, typval_T *rettv UNUSED)
6074{
6075 if (argvars[0].v_type == VAR_UNKNOWN)
6076 srand_seed_for_testing_is_used = FALSE;
6077 else
6078 {
6079 srand_seed_for_testing = (UINT32_T)tv_get_number(&argvars[0]);
6080 srand_seed_for_testing_is_used = TRUE;
6081 }
6082}
6083
6084 static void
6085init_srand(UINT32_T *x)
6086{
6087#ifndef MSWIN
6088 static int dev_urandom_state = NOTDONE; // FAIL or OK once tried
6089#endif
6090
6091 if (srand_seed_for_testing_is_used)
6092 {
6093 *x = srand_seed_for_testing;
6094 return;
6095 }
6096#ifndef MSWIN
6097 if (dev_urandom_state != FAIL)
6098 {
6099 int fd = open("/dev/urandom", O_RDONLY);
6100 struct {
6101 union {
6102 UINT32_T number;
6103 char bytes[sizeof(UINT32_T)];
6104 } contents;
6105 } buf;
6106
6107 // Attempt reading /dev/urandom.
6108 if (fd == -1)
6109 dev_urandom_state = FAIL;
6110 else
6111 {
6112 buf.contents.number = 0;
6113 if (read(fd, buf.contents.bytes, sizeof(UINT32_T))
6114 != sizeof(UINT32_T))
6115 dev_urandom_state = FAIL;
6116 else
6117 {
6118 dev_urandom_state = OK;
6119 *x = buf.contents.number;
6120 }
6121 close(fd);
6122 }
6123 }
6124 if (dev_urandom_state != OK)
6125 // Reading /dev/urandom doesn't work, fall back to time().
6126#endif
6127 *x = vim_time();
6128}
6129
6130#define ROTL(x, k) ((x << k) | (x >> (32 - k)))
6131#define SPLITMIX32(x, z) ( \
6132 z = (x += 0x9e3779b9), \
6133 z = (z ^ (z >> 16)) * 0x85ebca6b, \
6134 z = (z ^ (z >> 13)) * 0xc2b2ae35, \
6135 z ^ (z >> 16) \
6136 )
6137#define SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w) \
6138 result = ROTL(y * 5, 7) * 9; \
6139 t = y << 9; \
6140 z ^= x; \
6141 w ^= y; \
6142 y ^= z, x ^= w; \
6143 z ^= t; \
6144 w = ROTL(w, 11);
6145
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006146/*
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006147 * "rand()" function
6148 */
6149 static void
6150f_rand(typval_T *argvars, typval_T *rettv)
6151{
6152 list_T *l = NULL;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006153 static UINT32_T gx, gy, gz, gw;
6154 static int initialized = FALSE;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006155 listitem_T *lx, *ly, *lz, *lw;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006156 UINT32_T x, y, z, w, t, result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006157
6158 if (argvars[0].v_type == VAR_UNKNOWN)
6159 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006160 // When no argument is given use the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006161 if (initialized == FALSE)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006162 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006163 // Initialize the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006164 init_srand(&x);
6165
6166 gx = SPLITMIX32(x, z);
6167 gy = SPLITMIX32(x, z);
6168 gz = SPLITMIX32(x, z);
6169 gw = SPLITMIX32(x, z);
6170 initialized = TRUE;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006171 }
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006172
6173 SHUFFLE_XOSHIRO128STARSTAR(gx, gy, gz, gw);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006174 }
6175 else if (argvars[0].v_type == VAR_LIST)
6176 {
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006177 l = argvars[0].vval.v_list;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006178 if (l == NULL || list_len(l) != 4)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006179 goto theend;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006180
6181 lx = list_find(l, 0L);
6182 ly = list_find(l, 1L);
6183 lz = list_find(l, 2L);
6184 lw = list_find(l, 3L);
6185 if (lx->li_tv.v_type != VAR_NUMBER) goto theend;
6186 if (ly->li_tv.v_type != VAR_NUMBER) goto theend;
6187 if (lz->li_tv.v_type != VAR_NUMBER) goto theend;
6188 if (lw->li_tv.v_type != VAR_NUMBER) goto theend;
6189 x = (UINT32_T)lx->li_tv.vval.v_number;
6190 y = (UINT32_T)ly->li_tv.vval.v_number;
6191 z = (UINT32_T)lz->li_tv.vval.v_number;
6192 w = (UINT32_T)lw->li_tv.vval.v_number;
6193
6194 SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w);
6195
6196 lx->li_tv.vval.v_number = (varnumber_T)x;
6197 ly->li_tv.vval.v_number = (varnumber_T)y;
6198 lz->li_tv.vval.v_number = (varnumber_T)z;
6199 lw->li_tv.vval.v_number = (varnumber_T)w;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006200 }
6201 else
6202 goto theend;
6203
6204 rettv->v_type = VAR_NUMBER;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006205 rettv->vval.v_number = (varnumber_T)result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006206 return;
6207
6208theend:
6209 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006210 rettv->v_type = VAR_NUMBER;
6211 rettv->vval.v_number = -1;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006212}
6213
6214/*
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006215 * "srand()" function
6216 */
6217 static void
6218f_srand(typval_T *argvars, typval_T *rettv)
6219{
6220 UINT32_T x = 0, z;
6221
6222 if (rettv_list_alloc(rettv) == FAIL)
6223 return;
6224 if (argvars[0].v_type == VAR_UNKNOWN)
6225 {
6226 init_srand(&x);
6227 }
6228 else
6229 {
6230 int error = FALSE;
6231
6232 x = (UINT32_T)tv_get_number_chk(&argvars[0], &error);
6233 if (error)
6234 return;
6235 }
6236
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 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6241}
6242
6243#undef ROTL
6244#undef SPLITMIX32
6245#undef SHUFFLE_XOSHIRO128STARSTAR
6246
6247/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006248 * "range()" function
6249 */
6250 static void
6251f_range(typval_T *argvars, typval_T *rettv)
6252{
6253 varnumber_T start;
6254 varnumber_T end;
6255 varnumber_T stride = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006256 int error = FALSE;
6257
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006258 start = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006259 if (argvars[1].v_type == VAR_UNKNOWN)
6260 {
6261 end = start - 1;
6262 start = 0;
6263 }
6264 else
6265 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006266 end = tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006267 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006268 stride = tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006269 }
6270
6271 if (error)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006272 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006273 if (stride == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006274 emsg(_("E726: Stride is zero"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006275 else if (stride > 0 ? end + 1 < start : end - 1 > start)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006276 emsg(_("E727: Start past end"));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006277 else if (rettv_list_alloc(rettv) == OK)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006278 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006279 list_T *list = rettv->vval.v_list;
6280
6281 // Create a non-materialized list. This is much more efficient and
6282 // works with ":for". If used otherwise range_list_materialize() must
6283 // be called.
6284 list->lv_first = &range_list_item;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006285 list->lv_u.nonmat.lv_start = start;
6286 list->lv_u.nonmat.lv_end = end;
6287 list->lv_u.nonmat.lv_stride = stride;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01006288 list->lv_len = (end - start) / stride + 1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006289 }
6290}
6291
6292/*
6293 * If "list" is a non-materialized list then materialize it now.
6294 */
6295 void
6296range_list_materialize(list_T *list)
6297{
6298 if (list->lv_first == &range_list_item)
6299 {
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006300 varnumber_T start = list->lv_u.nonmat.lv_start;
6301 varnumber_T end = list->lv_u.nonmat.lv_end;
6302 int stride = list->lv_u.nonmat.lv_stride;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006303 varnumber_T i;
6304
6305 list->lv_first = NULL;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006306 list->lv_u.mat.lv_last = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006307 list->lv_len = 0;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006308 list->lv_u.mat.lv_idx_item = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006309 for (i = start; stride > 0 ? i <= end : i >= end; i += stride)
6310 if (list_append_number(list, (varnumber_T)i) == FAIL)
6311 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006312 }
6313}
6314
Bram Moolenaar0b6d9112018-05-22 20:35:17 +02006315 static void
6316return_register(int regname, typval_T *rettv)
6317{
6318 char_u buf[2] = {0, 0};
6319
6320 buf[0] = (char_u)regname;
6321 rettv->v_type = VAR_STRING;
6322 rettv->vval.v_string = vim_strsave(buf);
6323}
6324
6325/*
6326 * "reg_executing()" function
6327 */
6328 static void
6329f_reg_executing(typval_T *argvars UNUSED, typval_T *rettv)
6330{
6331 return_register(reg_executing, rettv);
6332}
6333
6334/*
6335 * "reg_recording()" function
6336 */
6337 static void
6338f_reg_recording(typval_T *argvars UNUSED, typval_T *rettv)
6339{
6340 return_register(reg_recording, rettv);
6341}
6342
Bram Moolenaar7416f3e2017-03-18 18:10:13 +01006343/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006344 * "rename({from}, {to})" function
6345 */
6346 static void
6347f_rename(typval_T *argvars, typval_T *rettv)
6348{
6349 char_u buf[NUMBUFLEN];
6350
6351 if (check_restricted() || check_secure())
6352 rettv->vval.v_number = -1;
6353 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006354 rettv->vval.v_number = vim_rename(tv_get_string(&argvars[0]),
6355 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006356}
6357
6358/*
6359 * "repeat()" function
6360 */
6361 static void
6362f_repeat(typval_T *argvars, typval_T *rettv)
6363{
6364 char_u *p;
6365 int n;
6366 int slen;
6367 int len;
6368 char_u *r;
6369 int i;
6370
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006371 n = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006372 if (argvars[0].v_type == VAR_LIST)
6373 {
6374 if (rettv_list_alloc(rettv) == OK && argvars[0].vval.v_list != NULL)
6375 while (n-- > 0)
6376 if (list_extend(rettv->vval.v_list,
6377 argvars[0].vval.v_list, NULL) == FAIL)
6378 break;
6379 }
6380 else
6381 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006382 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006383 rettv->v_type = VAR_STRING;
6384 rettv->vval.v_string = NULL;
6385
6386 slen = (int)STRLEN(p);
6387 len = slen * n;
6388 if (len <= 0)
6389 return;
6390
6391 r = alloc(len + 1);
6392 if (r != NULL)
6393 {
6394 for (i = 0; i < n; i++)
6395 mch_memmove(r + i * slen, p, (size_t)slen);
6396 r[len] = NUL;
6397 }
6398
6399 rettv->vval.v_string = r;
6400 }
6401}
6402
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006403#define SP_NOMOVE 0x01 // don't move cursor
6404#define SP_REPEAT 0x02 // repeat to find outer pair
6405#define SP_RETCOUNT 0x04 // return matchcount
6406#define SP_SETPCMARK 0x08 // set previous context mark
6407#define SP_START 0x10 // accept match at start position
6408#define SP_SUBPAT 0x20 // return nr of matching sub-pattern
6409#define SP_END 0x40 // leave cursor at end of match
6410#define SP_COLUMN 0x80 // start at cursor column
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006411
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006412/*
6413 * Get flags for a search function.
6414 * Possibly sets "p_ws".
6415 * Returns BACKWARD, FORWARD or zero (for an error).
6416 */
6417 static int
6418get_search_arg(typval_T *varp, int *flagsp)
6419{
6420 int dir = FORWARD;
6421 char_u *flags;
6422 char_u nbuf[NUMBUFLEN];
6423 int mask;
6424
6425 if (varp->v_type != VAR_UNKNOWN)
6426 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006427 flags = tv_get_string_buf_chk(varp, nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006428 if (flags == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006429 return 0; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006430 while (*flags != NUL)
6431 {
6432 switch (*flags)
6433 {
6434 case 'b': dir = BACKWARD; break;
6435 case 'w': p_ws = TRUE; break;
6436 case 'W': p_ws = FALSE; break;
6437 default: mask = 0;
6438 if (flagsp != NULL)
6439 switch (*flags)
6440 {
6441 case 'c': mask = SP_START; break;
6442 case 'e': mask = SP_END; break;
6443 case 'm': mask = SP_RETCOUNT; break;
6444 case 'n': mask = SP_NOMOVE; break;
6445 case 'p': mask = SP_SUBPAT; break;
6446 case 'r': mask = SP_REPEAT; break;
6447 case 's': mask = SP_SETPCMARK; break;
6448 case 'z': mask = SP_COLUMN; break;
6449 }
6450 if (mask == 0)
6451 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006452 semsg(_(e_invarg2), flags);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006453 dir = 0;
6454 }
6455 else
6456 *flagsp |= mask;
6457 }
6458 if (dir == 0)
6459 break;
6460 ++flags;
6461 }
6462 }
6463 return dir;
6464}
6465
6466/*
6467 * Shared by search() and searchpos() functions.
6468 */
6469 static int
6470search_cmn(typval_T *argvars, pos_T *match_pos, int *flagsp)
6471{
6472 int flags;
6473 char_u *pat;
6474 pos_T pos;
6475 pos_T save_cursor;
6476 int save_p_ws = p_ws;
6477 int dir;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006478 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006479 long lnum_stop = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006480#ifdef FEAT_RELTIME
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006481 proftime_T tm;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006482 long time_limit = 0;
6483#endif
6484 int options = SEARCH_KEEP;
6485 int subpatnum;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006486 searchit_arg_T sia;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006487
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006488 pat = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006489 dir = get_search_arg(&argvars[1], flagsp); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006490 if (dir == 0)
6491 goto theend;
6492 flags = *flagsp;
6493 if (flags & SP_START)
6494 options |= SEARCH_START;
6495 if (flags & SP_END)
6496 options |= SEARCH_END;
6497 if (flags & SP_COLUMN)
6498 options |= SEARCH_COL;
6499
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006500 // Optional arguments: line number to stop searching and timeout.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006501 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN)
6502 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006503 lnum_stop = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006504 if (lnum_stop < 0)
6505 goto theend;
6506#ifdef FEAT_RELTIME
6507 if (argvars[3].v_type != VAR_UNKNOWN)
6508 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006509 time_limit = (long)tv_get_number_chk(&argvars[3], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006510 if (time_limit < 0)
6511 goto theend;
6512 }
6513#endif
6514 }
6515
6516#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006517 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006518 profile_setlimit(time_limit, &tm);
6519#endif
6520
6521 /*
6522 * This function does not accept SP_REPEAT and SP_RETCOUNT flags.
6523 * Check to make sure only those flags are set.
6524 * Also, Only the SP_NOMOVE or the SP_SETPCMARK flag can be set. Both
6525 * flags cannot be set. Check for that condition also.
6526 */
6527 if (((flags & (SP_REPEAT | SP_RETCOUNT)) != 0)
6528 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
6529 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006530 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006531 goto theend;
6532 }
6533
6534 pos = save_cursor = curwin->w_cursor;
Bram Moolenaara80faa82020-04-12 19:37:17 +02006535 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006536 sia.sa_stop_lnum = (linenr_T)lnum_stop;
6537#ifdef FEAT_RELTIME
6538 sia.sa_tm = &tm;
6539#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01006540 subpatnum = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006541 options, RE_SEARCH, &sia);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006542 if (subpatnum != FAIL)
6543 {
6544 if (flags & SP_SUBPAT)
6545 retval = subpatnum;
6546 else
6547 retval = pos.lnum;
6548 if (flags & SP_SETPCMARK)
6549 setpcmark();
6550 curwin->w_cursor = pos;
6551 if (match_pos != NULL)
6552 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006553 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006554 match_pos->lnum = pos.lnum;
6555 match_pos->col = pos.col + 1;
6556 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006557 // "/$" will put the cursor after the end of the line, may need to
6558 // correct that here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006559 check_cursor();
6560 }
6561
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006562 // If 'n' flag is used: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006563 if (flags & SP_NOMOVE)
6564 curwin->w_cursor = save_cursor;
6565 else
6566 curwin->w_set_curswant = TRUE;
6567theend:
6568 p_ws = save_p_ws;
6569
6570 return retval;
6571}
6572
6573#ifdef FEAT_FLOAT
6574
6575/*
6576 * round() is not in C90, use ceil() or floor() instead.
6577 */
6578 float_T
6579vim_round(float_T f)
6580{
6581 return f > 0 ? floor(f + 0.5) : ceil(f - 0.5);
6582}
6583
6584/*
6585 * "round({float})" function
6586 */
6587 static void
6588f_round(typval_T *argvars, typval_T *rettv)
6589{
6590 float_T f = 0.0;
6591
6592 rettv->v_type = VAR_FLOAT;
6593 if (get_float_arg(argvars, &f) == OK)
6594 rettv->vval.v_float = vim_round(f);
6595 else
6596 rettv->vval.v_float = 0.0;
6597}
6598#endif
6599
Bram Moolenaare99be0e2019-03-26 22:51:09 +01006600#ifdef FEAT_RUBY
6601/*
6602 * "rubyeval()" function
6603 */
6604 static void
6605f_rubyeval(typval_T *argvars, typval_T *rettv)
6606{
6607 char_u *str;
6608 char_u buf[NUMBUFLEN];
6609
6610 str = tv_get_string_buf(&argvars[0], buf);
6611 do_rubyeval(str, rettv);
6612}
6613#endif
6614
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006615/*
6616 * "screenattr()" function
6617 */
6618 static void
6619f_screenattr(typval_T *argvars, typval_T *rettv)
6620{
6621 int row;
6622 int col;
6623 int c;
6624
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006625 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6626 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006627 if (row < 0 || row >= screen_Rows
6628 || col < 0 || col >= screen_Columns)
6629 c = -1;
6630 else
6631 c = ScreenAttrs[LineOffset[row] + col];
6632 rettv->vval.v_number = c;
6633}
6634
6635/*
6636 * "screenchar()" function
6637 */
6638 static void
6639f_screenchar(typval_T *argvars, typval_T *rettv)
6640{
6641 int row;
6642 int col;
6643 int off;
6644 int c;
6645
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006646 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6647 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006648 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006649 c = -1;
6650 else
6651 {
6652 off = LineOffset[row] + col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006653 if (enc_utf8 && ScreenLinesUC[off] != 0)
6654 c = ScreenLinesUC[off];
6655 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006656 c = ScreenLines[off];
6657 }
6658 rettv->vval.v_number = c;
6659}
6660
6661/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006662 * "screenchars()" function
6663 */
6664 static void
6665f_screenchars(typval_T *argvars, typval_T *rettv)
6666{
6667 int row;
6668 int col;
6669 int off;
6670 int c;
6671 int i;
6672
6673 if (rettv_list_alloc(rettv) == FAIL)
6674 return;
6675 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6676 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
6677 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
6678 return;
6679
6680 off = LineOffset[row] + col;
6681 if (enc_utf8 && ScreenLinesUC[off] != 0)
6682 c = ScreenLinesUC[off];
6683 else
6684 c = ScreenLines[off];
6685 list_append_number(rettv->vval.v_list, (varnumber_T)c);
6686
6687 if (enc_utf8)
6688
6689 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
6690 list_append_number(rettv->vval.v_list,
6691 (varnumber_T)ScreenLinesC[i][off]);
6692}
6693
6694/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006695 * "screencol()" function
6696 *
6697 * First column is 1 to be consistent with virtcol().
6698 */
6699 static void
6700f_screencol(typval_T *argvars UNUSED, typval_T *rettv)
6701{
6702 rettv->vval.v_number = screen_screencol() + 1;
6703}
6704
6705/*
6706 * "screenrow()" function
6707 */
6708 static void
6709f_screenrow(typval_T *argvars UNUSED, typval_T *rettv)
6710{
6711 rettv->vval.v_number = screen_screenrow() + 1;
6712}
6713
6714/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006715 * "screenstring()" function
6716 */
6717 static void
6718f_screenstring(typval_T *argvars, typval_T *rettv)
6719{
6720 int row;
6721 int col;
6722 int off;
6723 int c;
6724 int i;
6725 char_u buf[MB_MAXBYTES + 1];
6726 int buflen = 0;
6727
6728 rettv->vval.v_string = NULL;
6729 rettv->v_type = VAR_STRING;
6730
6731 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6732 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
6733 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
6734 return;
6735
6736 off = LineOffset[row] + col;
6737 if (enc_utf8 && ScreenLinesUC[off] != 0)
6738 c = ScreenLinesUC[off];
6739 else
6740 c = ScreenLines[off];
6741 buflen += mb_char2bytes(c, buf);
6742
6743 if (enc_utf8)
6744 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
6745 buflen += mb_char2bytes(ScreenLinesC[i][off], buf + buflen);
6746
6747 buf[buflen] = NUL;
6748 rettv->vval.v_string = vim_strsave(buf);
6749}
6750
6751/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006752 * "search()" function
6753 */
6754 static void
6755f_search(typval_T *argvars, typval_T *rettv)
6756{
6757 int flags = 0;
6758
6759 rettv->vval.v_number = search_cmn(argvars, NULL, &flags);
6760}
6761
6762/*
6763 * "searchdecl()" function
6764 */
6765 static void
6766f_searchdecl(typval_T *argvars, typval_T *rettv)
6767{
6768 int locally = 1;
6769 int thisblock = 0;
6770 int error = FALSE;
6771 char_u *name;
6772
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006773 rettv->vval.v_number = 1; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006774
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006775 name = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006776 if (argvars[1].v_type != VAR_UNKNOWN)
6777 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006778 locally = (int)tv_get_number_chk(&argvars[1], &error) == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006779 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006780 thisblock = (int)tv_get_number_chk(&argvars[2], &error) != 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006781 }
6782 if (!error && name != NULL)
6783 rettv->vval.v_number = find_decl(name, (int)STRLEN(name),
6784 locally, thisblock, SEARCH_KEEP) == FAIL;
6785}
6786
6787/*
6788 * Used by searchpair() and searchpairpos()
6789 */
6790 static int
6791searchpair_cmn(typval_T *argvars, pos_T *match_pos)
6792{
6793 char_u *spat, *mpat, *epat;
Bram Moolenaar48570482017-10-30 21:48:41 +01006794 typval_T *skip;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006795 int save_p_ws = p_ws;
6796 int dir;
6797 int flags = 0;
6798 char_u nbuf1[NUMBUFLEN];
6799 char_u nbuf2[NUMBUFLEN];
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006800 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006801 long lnum_stop = 0;
6802 long time_limit = 0;
6803
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006804 // Get the three pattern arguments: start, middle, end. Will result in an
6805 // error if not a valid argument.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006806 spat = tv_get_string_chk(&argvars[0]);
6807 mpat = tv_get_string_buf_chk(&argvars[1], nbuf1);
6808 epat = tv_get_string_buf_chk(&argvars[2], nbuf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006809 if (spat == NULL || mpat == NULL || epat == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006810 goto theend; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006811
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006812 // Handle the optional fourth argument: flags
6813 dir = get_search_arg(&argvars[3], &flags); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006814 if (dir == 0)
6815 goto theend;
6816
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006817 // Don't accept SP_END or SP_SUBPAT.
6818 // Only one of the SP_NOMOVE or SP_SETPCMARK flags can be set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006819 if ((flags & (SP_END | SP_SUBPAT)) != 0
6820 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
6821 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006822 semsg(_(e_invarg2), tv_get_string(&argvars[3]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006823 goto theend;
6824 }
6825
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006826 // Using 'r' implies 'W', otherwise it doesn't work.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006827 if (flags & SP_REPEAT)
6828 p_ws = FALSE;
6829
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006830 // Optional fifth argument: skip expression
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006831 if (argvars[3].v_type == VAR_UNKNOWN
6832 || argvars[4].v_type == VAR_UNKNOWN)
Bram Moolenaar48570482017-10-30 21:48:41 +01006833 skip = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006834 else
6835 {
Bram Moolenaar48570482017-10-30 21:48:41 +01006836 skip = &argvars[4];
6837 if (skip->v_type != VAR_FUNC && skip->v_type != VAR_PARTIAL
6838 && skip->v_type != VAR_STRING)
6839 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006840 // Type error
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006841 semsg(_(e_invarg2), tv_get_string(&argvars[4]));
Bram Moolenaar48570482017-10-30 21:48:41 +01006842 goto theend;
6843 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006844 if (argvars[5].v_type != VAR_UNKNOWN)
6845 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006846 lnum_stop = (long)tv_get_number_chk(&argvars[5], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006847 if (lnum_stop < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006848 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006849 semsg(_(e_invarg2), tv_get_string(&argvars[5]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006850 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006851 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006852#ifdef FEAT_RELTIME
6853 if (argvars[6].v_type != VAR_UNKNOWN)
6854 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006855 time_limit = (long)tv_get_number_chk(&argvars[6], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006856 if (time_limit < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006857 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006858 semsg(_(e_invarg2), tv_get_string(&argvars[6]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006859 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006860 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006861 }
6862#endif
6863 }
6864 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006865
6866 retval = do_searchpair(spat, mpat, epat, dir, skip, flags,
6867 match_pos, lnum_stop, time_limit);
6868
6869theend:
6870 p_ws = save_p_ws;
6871
6872 return retval;
6873}
6874
6875/*
6876 * "searchpair()" function
6877 */
6878 static void
6879f_searchpair(typval_T *argvars, typval_T *rettv)
6880{
6881 rettv->vval.v_number = searchpair_cmn(argvars, NULL);
6882}
6883
6884/*
6885 * "searchpairpos()" function
6886 */
6887 static void
6888f_searchpairpos(typval_T *argvars, typval_T *rettv)
6889{
6890 pos_T match_pos;
6891 int lnum = 0;
6892 int col = 0;
6893
6894 if (rettv_list_alloc(rettv) == FAIL)
6895 return;
6896
6897 if (searchpair_cmn(argvars, &match_pos) > 0)
6898 {
6899 lnum = match_pos.lnum;
6900 col = match_pos.col;
6901 }
6902
6903 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
6904 list_append_number(rettv->vval.v_list, (varnumber_T)col);
6905}
6906
6907/*
6908 * Search for a start/middle/end thing.
6909 * Used by searchpair(), see its documentation for the details.
6910 * Returns 0 or -1 for no match,
6911 */
6912 long
6913do_searchpair(
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006914 char_u *spat, // start pattern
6915 char_u *mpat, // middle pattern
6916 char_u *epat, // end pattern
6917 int dir, // BACKWARD or FORWARD
6918 typval_T *skip, // skip expression
6919 int flags, // SP_SETPCMARK and other SP_ values
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006920 pos_T *match_pos,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006921 linenr_T lnum_stop, // stop at this line if not zero
6922 long time_limit UNUSED) // stop after this many msec
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006923{
6924 char_u *save_cpo;
6925 char_u *pat, *pat2 = NULL, *pat3 = NULL;
6926 long retval = 0;
6927 pos_T pos;
6928 pos_T firstpos;
6929 pos_T foundpos;
6930 pos_T save_cursor;
6931 pos_T save_pos;
6932 int n;
6933 int r;
6934 int nest = 1;
Bram Moolenaar48570482017-10-30 21:48:41 +01006935 int use_skip = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006936 int err;
6937 int options = SEARCH_KEEP;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006938#ifdef FEAT_RELTIME
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006939 proftime_T tm;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006940#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006941
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006942 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006943 save_cpo = p_cpo;
6944 p_cpo = empty_option;
6945
6946#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006947 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006948 profile_setlimit(time_limit, &tm);
6949#endif
6950
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006951 // Make two search patterns: start/end (pat2, for in nested pairs) and
6952 // start/middle/end (pat3, for the top pair).
Bram Moolenaar964b3742019-05-24 18:54:09 +02006953 pat2 = alloc(STRLEN(spat) + STRLEN(epat) + 17);
6954 pat3 = alloc(STRLEN(spat) + STRLEN(mpat) + STRLEN(epat) + 25);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006955 if (pat2 == NULL || pat3 == NULL)
6956 goto theend;
Bram Moolenaar6e450a52017-01-06 20:03:58 +01006957 sprintf((char *)pat2, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)", spat, epat);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006958 if (*mpat == NUL)
6959 STRCPY(pat3, pat2);
6960 else
Bram Moolenaar6e450a52017-01-06 20:03:58 +01006961 sprintf((char *)pat3, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)\\|\\(%s\\m\\)",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006962 spat, epat, mpat);
6963 if (flags & SP_START)
6964 options |= SEARCH_START;
6965
Bram Moolenaar48570482017-10-30 21:48:41 +01006966 if (skip != NULL)
6967 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006968 // Empty string means to not use the skip expression.
Bram Moolenaar48570482017-10-30 21:48:41 +01006969 if (skip->v_type == VAR_STRING || skip->v_type == VAR_FUNC)
6970 use_skip = skip->vval.v_string != NULL
6971 && *skip->vval.v_string != NUL;
6972 }
6973
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006974 save_cursor = curwin->w_cursor;
6975 pos = curwin->w_cursor;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006976 CLEAR_POS(&firstpos);
6977 CLEAR_POS(&foundpos);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006978 pat = pat3;
6979 for (;;)
6980 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006981 searchit_arg_T sia;
6982
Bram Moolenaara80faa82020-04-12 19:37:17 +02006983 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006984 sia.sa_stop_lnum = lnum_stop;
6985#ifdef FEAT_RELTIME
6986 sia.sa_tm = &tm;
6987#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01006988 n = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006989 options, RE_SEARCH, &sia);
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006990 if (n == FAIL || (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos)))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006991 // didn't find it or found the first match again: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006992 break;
6993
6994 if (firstpos.lnum == 0)
6995 firstpos = pos;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006996 if (EQUAL_POS(pos, foundpos))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006997 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006998 // Found the same position again. Can happen with a pattern that
6999 // has "\zs" at the end and searching backwards. Advance one
7000 // character and try again.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007001 if (dir == BACKWARD)
7002 decl(&pos);
7003 else
7004 incl(&pos);
7005 }
7006 foundpos = pos;
7007
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007008 // clear the start flag to avoid getting stuck here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007009 options &= ~SEARCH_START;
7010
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007011 // If the skip pattern matches, ignore this match.
Bram Moolenaar48570482017-10-30 21:48:41 +01007012 if (use_skip)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007013 {
7014 save_pos = curwin->w_cursor;
7015 curwin->w_cursor = pos;
Bram Moolenaar48570482017-10-30 21:48:41 +01007016 err = FALSE;
7017 r = eval_expr_to_bool(skip, &err);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007018 curwin->w_cursor = save_pos;
7019 if (err)
7020 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007021 // Evaluating {skip} caused an error, break here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007022 curwin->w_cursor = save_cursor;
7023 retval = -1;
7024 break;
7025 }
7026 if (r)
7027 continue;
7028 }
7029
7030 if ((dir == BACKWARD && n == 3) || (dir == FORWARD && n == 2))
7031 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007032 // Found end when searching backwards or start when searching
7033 // forward: nested pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007034 ++nest;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007035 pat = pat2; // nested, don't search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007036 }
7037 else
7038 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007039 // Found end when searching forward or start when searching
7040 // backward: end of (nested) pair; or found middle in outer pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007041 if (--nest == 1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007042 pat = pat3; // outer level, search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007043 }
7044
7045 if (nest == 0)
7046 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007047 // Found the match: return matchcount or line number.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007048 if (flags & SP_RETCOUNT)
7049 ++retval;
7050 else
7051 retval = pos.lnum;
7052 if (flags & SP_SETPCMARK)
7053 setpcmark();
7054 curwin->w_cursor = pos;
7055 if (!(flags & SP_REPEAT))
7056 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007057 nest = 1; // search for next unmatched
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007058 }
7059 }
7060
7061 if (match_pos != NULL)
7062 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007063 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007064 match_pos->lnum = curwin->w_cursor.lnum;
7065 match_pos->col = curwin->w_cursor.col + 1;
7066 }
7067
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007068 // If 'n' flag is used or search failed: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007069 if ((flags & SP_NOMOVE) || retval == 0)
7070 curwin->w_cursor = save_cursor;
7071
7072theend:
7073 vim_free(pat2);
7074 vim_free(pat3);
7075 if (p_cpo == empty_option)
7076 p_cpo = save_cpo;
7077 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007078 // Darn, evaluating the {skip} expression changed the value.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007079 free_string_option(save_cpo);
7080
7081 return retval;
7082}
7083
7084/*
7085 * "searchpos()" function
7086 */
7087 static void
7088f_searchpos(typval_T *argvars, typval_T *rettv)
7089{
7090 pos_T match_pos;
7091 int lnum = 0;
7092 int col = 0;
7093 int n;
7094 int flags = 0;
7095
7096 if (rettv_list_alloc(rettv) == FAIL)
7097 return;
7098
7099 n = search_cmn(argvars, &match_pos, &flags);
7100 if (n > 0)
7101 {
7102 lnum = match_pos.lnum;
7103 col = match_pos.col;
7104 }
7105
7106 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
7107 list_append_number(rettv->vval.v_list, (varnumber_T)col);
7108 if (flags & SP_SUBPAT)
7109 list_append_number(rettv->vval.v_list, (varnumber_T)n);
7110}
7111
7112 static void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007113f_setcharsearch(typval_T *argvars, typval_T *rettv UNUSED)
7114{
7115 dict_T *d;
7116 dictitem_T *di;
7117 char_u *csearch;
7118
7119 if (argvars[0].v_type != VAR_DICT)
7120 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007121 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007122 return;
7123 }
7124
7125 if ((d = argvars[0].vval.v_dict) != NULL)
7126 {
Bram Moolenaar8f667172018-12-14 15:38:31 +01007127 csearch = dict_get_string(d, (char_u *)"char", FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007128 if (csearch != NULL)
7129 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007130 if (enc_utf8)
7131 {
7132 int pcc[MAX_MCO];
7133 int c = utfc_ptr2char(csearch, pcc);
7134
7135 set_last_csearch(c, csearch, utfc_ptr2len(csearch));
7136 }
7137 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007138 set_last_csearch(PTR2CHAR(csearch),
Bram Moolenaar1614a142019-10-06 22:00:13 +02007139 csearch, mb_ptr2len(csearch));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007140 }
7141
7142 di = dict_find(d, (char_u *)"forward", -1);
7143 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007144 set_csearch_direction((int)tv_get_number(&di->di_tv)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007145 ? FORWARD : BACKWARD);
7146
7147 di = dict_find(d, (char_u *)"until", -1);
7148 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007149 set_csearch_until(!!tv_get_number(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007150 }
7151}
7152
7153/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02007154 * "setenv()" function
7155 */
7156 static void
7157f_setenv(typval_T *argvars, typval_T *rettv UNUSED)
7158{
7159 char_u namebuf[NUMBUFLEN];
7160 char_u valbuf[NUMBUFLEN];
7161 char_u *name = tv_get_string_buf(&argvars[0], namebuf);
7162
7163 if (argvars[1].v_type == VAR_SPECIAL
7164 && argvars[1].vval.v_number == VVAL_NULL)
7165 vim_unsetenv(name);
7166 else
7167 vim_setenv(name, tv_get_string_buf(&argvars[1], valbuf));
7168}
7169
7170/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007171 * "setfperm({fname}, {mode})" function
7172 */
7173 static void
7174f_setfperm(typval_T *argvars, typval_T *rettv)
7175{
7176 char_u *fname;
7177 char_u modebuf[NUMBUFLEN];
7178 char_u *mode_str;
7179 int i;
7180 int mask;
7181 int mode = 0;
7182
7183 rettv->vval.v_number = 0;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007184 fname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007185 if (fname == NULL)
7186 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007187 mode_str = tv_get_string_buf_chk(&argvars[1], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007188 if (mode_str == NULL)
7189 return;
7190 if (STRLEN(mode_str) != 9)
7191 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007192 semsg(_(e_invarg2), mode_str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007193 return;
7194 }
7195
7196 mask = 1;
7197 for (i = 8; i >= 0; --i)
7198 {
7199 if (mode_str[i] != '-')
7200 mode |= mask;
7201 mask = mask << 1;
7202 }
7203 rettv->vval.v_number = mch_setperm(fname, mode) == OK;
7204}
7205
7206/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007207 * "setpos()" function
7208 */
7209 static void
7210f_setpos(typval_T *argvars, typval_T *rettv)
7211{
7212 pos_T pos;
7213 int fnum;
7214 char_u *name;
7215 colnr_T curswant = -1;
7216
7217 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007218 name = tv_get_string_chk(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007219 if (name != NULL)
7220 {
7221 if (list2fpos(&argvars[1], &pos, &fnum, &curswant) == OK)
7222 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01007223 if (pos.col != MAXCOL && --pos.col < 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007224 pos.col = 0;
7225 if (name[0] == '.' && name[1] == NUL)
7226 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007227 // set cursor; "fnum" is ignored
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007228 curwin->w_cursor = pos;
7229 if (curswant >= 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007230 {
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007231 curwin->w_curswant = curswant - 1;
7232 curwin->w_set_curswant = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007233 }
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007234 check_cursor();
7235 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007236 }
7237 else if (name[0] == '\'' && name[1] != NUL && name[2] == NUL)
7238 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007239 // set mark
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007240 if (setmark_pos(name[1], &pos, fnum) == OK)
7241 rettv->vval.v_number = 0;
7242 }
7243 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007244 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007245 }
7246 }
7247}
7248
7249/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007250 * "setreg()" function
7251 */
7252 static void
7253f_setreg(typval_T *argvars, typval_T *rettv)
7254{
7255 int regname;
7256 char_u *strregname;
7257 char_u *stropt;
7258 char_u *strval;
7259 int append;
7260 char_u yank_type;
7261 long block_len;
7262
7263 block_len = -1;
7264 yank_type = MAUTO;
7265 append = FALSE;
7266
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007267 strregname = tv_get_string_chk(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007268 rettv->vval.v_number = 1; // FAIL is default
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007269
7270 if (strregname == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007271 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007272 regname = *strregname;
7273 if (regname == 0 || regname == '@')
7274 regname = '"';
7275
7276 if (argvars[2].v_type != VAR_UNKNOWN)
7277 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007278 stropt = tv_get_string_chk(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007279 if (stropt == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007280 return; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007281 for (; *stropt != NUL; ++stropt)
7282 switch (*stropt)
7283 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007284 case 'a': case 'A': // append
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007285 append = TRUE;
7286 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007287 case 'v': case 'c': // character-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007288 yank_type = MCHAR;
7289 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007290 case 'V': case 'l': // line-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007291 yank_type = MLINE;
7292 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007293 case 'b': case Ctrl_V: // block-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007294 yank_type = MBLOCK;
7295 if (VIM_ISDIGIT(stropt[1]))
7296 {
7297 ++stropt;
7298 block_len = getdigits(&stropt) - 1;
7299 --stropt;
7300 }
7301 break;
7302 }
7303 }
7304
7305 if (argvars[1].v_type == VAR_LIST)
7306 {
7307 char_u **lstval;
7308 char_u **allocval;
7309 char_u buf[NUMBUFLEN];
7310 char_u **curval;
7311 char_u **curallocval;
7312 list_T *ll = argvars[1].vval.v_list;
7313 listitem_T *li;
7314 int len;
7315
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007316 // If the list is NULL handle like an empty list.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007317 len = ll == NULL ? 0 : ll->lv_len;
7318
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007319 // First half: use for pointers to result lines; second half: use for
7320 // pointers to allocated copies.
Bram Moolenaarc799fe22019-05-28 23:08:19 +02007321 lstval = ALLOC_MULT(char_u *, (len + 1) * 2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007322 if (lstval == NULL)
7323 return;
7324 curval = lstval;
7325 allocval = lstval + len + 2;
7326 curallocval = allocval;
7327
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007328 if (ll != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007329 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007330 range_list_materialize(ll);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02007331 FOR_ALL_LIST_ITEMS(ll, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007332 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007333 strval = tv_get_string_buf_chk(&li->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007334 if (strval == NULL)
7335 goto free_lstval;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007336 if (strval == buf)
7337 {
7338 // Need to make a copy, next tv_get_string_buf_chk() will
7339 // overwrite the string.
7340 strval = vim_strsave(buf);
7341 if (strval == NULL)
7342 goto free_lstval;
7343 *curallocval++ = strval;
7344 }
7345 *curval++ = strval;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007346 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007347 }
7348 *curval++ = NULL;
7349
7350 write_reg_contents_lst(regname, lstval, -1,
7351 append, yank_type, block_len);
7352free_lstval:
7353 while (curallocval > allocval)
7354 vim_free(*--curallocval);
7355 vim_free(lstval);
7356 }
7357 else
7358 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007359 strval = tv_get_string_chk(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007360 if (strval == NULL)
7361 return;
7362 write_reg_contents_ex(regname, strval, -1,
7363 append, yank_type, block_len);
7364 }
7365 rettv->vval.v_number = 0;
7366}
7367
7368/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007369 * "settagstack()" function
7370 */
7371 static void
7372f_settagstack(typval_T *argvars, typval_T *rettv)
7373{
7374 static char *e_invact2 = N_("E962: Invalid action: '%s'");
7375 win_T *wp;
7376 dict_T *d;
7377 int action = 'r';
7378
7379 rettv->vval.v_number = -1;
7380
7381 // first argument: window number or id
7382 wp = find_win_by_nr_or_id(&argvars[0]);
7383 if (wp == NULL)
7384 return;
7385
7386 // second argument: dict with items to set in the tag stack
7387 if (argvars[1].v_type != VAR_DICT)
7388 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007389 emsg(_(e_dictreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007390 return;
7391 }
7392 d = argvars[1].vval.v_dict;
7393 if (d == NULL)
7394 return;
7395
7396 // third argument: action - 'a' for append and 'r' for replace.
7397 // default is to replace the stack.
7398 if (argvars[2].v_type == VAR_UNKNOWN)
7399 action = 'r';
7400 else if (argvars[2].v_type == VAR_STRING)
7401 {
7402 char_u *actstr;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007403 actstr = tv_get_string_chk(&argvars[2]);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007404 if (actstr == NULL)
7405 return;
Bram Moolenaar271fa082020-01-02 14:02:16 +01007406 if ((*actstr == 'r' || *actstr == 'a' || *actstr == 't')
7407 && actstr[1] == NUL)
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007408 action = *actstr;
7409 else
7410 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007411 semsg(_(e_invact2), actstr);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007412 return;
7413 }
7414 }
7415 else
7416 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007417 emsg(_(e_stringreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007418 return;
7419 }
7420
7421 if (set_tagstack(wp, d, action) == OK)
7422 rettv->vval.v_number = 0;
7423}
7424
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007425#ifdef FEAT_CRYPT
7426/*
7427 * "sha256({string})" function
7428 */
7429 static void
7430f_sha256(typval_T *argvars, typval_T *rettv)
7431{
7432 char_u *p;
7433
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007434 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007435 rettv->vval.v_string = vim_strsave(
7436 sha256_bytes(p, (int)STRLEN(p), NULL, 0));
7437 rettv->v_type = VAR_STRING;
7438}
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007439#endif // FEAT_CRYPT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007440
7441/*
7442 * "shellescape({string})" function
7443 */
7444 static void
7445f_shellescape(typval_T *argvars, typval_T *rettv)
7446{
Bram Moolenaar20615522017-06-05 18:46:26 +02007447 int do_special = non_zero_arg(&argvars[1]);
7448
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007449 rettv->vval.v_string = vim_strsave_shellescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007450 tv_get_string(&argvars[0]), do_special, do_special);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007451 rettv->v_type = VAR_STRING;
7452}
7453
7454/*
7455 * shiftwidth() function
7456 */
7457 static void
7458f_shiftwidth(typval_T *argvars UNUSED, typval_T *rettv)
7459{
Bram Moolenaarf9514162018-11-22 03:08:29 +01007460 rettv->vval.v_number = 0;
7461
7462 if (argvars[0].v_type != VAR_UNKNOWN)
7463 {
7464 long col;
7465
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007466 col = (long)tv_get_number_chk(argvars, NULL);
Bram Moolenaarf9514162018-11-22 03:08:29 +01007467 if (col < 0)
7468 return; // type error; errmsg already given
7469#ifdef FEAT_VARTABS
7470 rettv->vval.v_number = get_sw_value_col(curbuf, col);
7471 return;
7472#endif
7473 }
7474
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007475 rettv->vval.v_number = get_sw_value(curbuf);
7476}
7477
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007478#ifdef FEAT_FLOAT
7479/*
7480 * "sin()" function
7481 */
7482 static void
7483f_sin(typval_T *argvars, typval_T *rettv)
7484{
7485 float_T f = 0.0;
7486
7487 rettv->v_type = VAR_FLOAT;
7488 if (get_float_arg(argvars, &f) == OK)
7489 rettv->vval.v_float = sin(f);
7490 else
7491 rettv->vval.v_float = 0.0;
7492}
7493
7494/*
7495 * "sinh()" function
7496 */
7497 static void
7498f_sinh(typval_T *argvars, typval_T *rettv)
7499{
7500 float_T f = 0.0;
7501
7502 rettv->v_type = VAR_FLOAT;
7503 if (get_float_arg(argvars, &f) == OK)
7504 rettv->vval.v_float = sinh(f);
7505 else
7506 rettv->vval.v_float = 0.0;
7507}
7508#endif
7509
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007510/*
7511 * "soundfold({word})" function
7512 */
7513 static void
7514f_soundfold(typval_T *argvars, typval_T *rettv)
7515{
7516 char_u *s;
7517
7518 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007519 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007520#ifdef FEAT_SPELL
7521 rettv->vval.v_string = eval_soundfold(s);
7522#else
7523 rettv->vval.v_string = vim_strsave(s);
7524#endif
7525}
7526
7527/*
7528 * "spellbadword()" function
7529 */
7530 static void
7531f_spellbadword(typval_T *argvars UNUSED, typval_T *rettv)
7532{
7533 char_u *word = (char_u *)"";
7534 hlf_T attr = HLF_COUNT;
7535 int len = 0;
7536
7537 if (rettv_list_alloc(rettv) == FAIL)
7538 return;
7539
7540#ifdef FEAT_SPELL
7541 if (argvars[0].v_type == VAR_UNKNOWN)
7542 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007543 // Find the start and length of the badly spelled word.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007544 len = spell_move_to(curwin, FORWARD, TRUE, TRUE, &attr);
7545 if (len != 0)
Bram Moolenaarb73fa622017-12-21 20:27:47 +01007546 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007547 word = ml_get_cursor();
Bram Moolenaarb73fa622017-12-21 20:27:47 +01007548 curwin->w_set_curswant = TRUE;
7549 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007550 }
7551 else if (curwin->w_p_spell && *curbuf->b_s.b_p_spl != NUL)
7552 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007553 char_u *str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007554 int capcol = -1;
7555
7556 if (str != NULL)
7557 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007558 // Check the argument for spelling.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007559 while (*str != NUL)
7560 {
7561 len = spell_check(curwin, str, &attr, &capcol, FALSE);
7562 if (attr != HLF_COUNT)
7563 {
7564 word = str;
7565 break;
7566 }
7567 str += len;
Bram Moolenaar66ab9162018-07-20 20:28:48 +02007568 capcol -= len;
Bram Moolenaar0c779e82019-08-09 17:01:02 +02007569 len = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007570 }
7571 }
7572 }
7573#endif
7574
7575 list_append_string(rettv->vval.v_list, word, len);
7576 list_append_string(rettv->vval.v_list, (char_u *)(
7577 attr == HLF_SPB ? "bad" :
7578 attr == HLF_SPR ? "rare" :
7579 attr == HLF_SPL ? "local" :
7580 attr == HLF_SPC ? "caps" :
7581 ""), -1);
7582}
7583
7584/*
7585 * "spellsuggest()" function
7586 */
7587 static void
7588f_spellsuggest(typval_T *argvars UNUSED, typval_T *rettv)
7589{
7590#ifdef FEAT_SPELL
7591 char_u *str;
7592 int typeerr = FALSE;
7593 int maxcount;
7594 garray_T ga;
7595 int i;
7596 listitem_T *li;
7597 int need_capital = FALSE;
7598#endif
7599
7600 if (rettv_list_alloc(rettv) == FAIL)
7601 return;
7602
7603#ifdef FEAT_SPELL
7604 if (curwin->w_p_spell && *curwin->w_s->b_p_spl != NUL)
7605 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007606 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007607 if (argvars[1].v_type != VAR_UNKNOWN)
7608 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007609 maxcount = (int)tv_get_number_chk(&argvars[1], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007610 if (maxcount <= 0)
7611 return;
7612 if (argvars[2].v_type != VAR_UNKNOWN)
7613 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007614 need_capital = (int)tv_get_number_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007615 if (typeerr)
7616 return;
7617 }
7618 }
7619 else
7620 maxcount = 25;
7621
7622 spell_suggest_list(&ga, str, maxcount, need_capital, FALSE);
7623
7624 for (i = 0; i < ga.ga_len; ++i)
7625 {
7626 str = ((char_u **)ga.ga_data)[i];
7627
7628 li = listitem_alloc();
7629 if (li == NULL)
7630 vim_free(str);
7631 else
7632 {
7633 li->li_tv.v_type = VAR_STRING;
7634 li->li_tv.v_lock = 0;
7635 li->li_tv.vval.v_string = str;
7636 list_append(rettv->vval.v_list, li);
7637 }
7638 }
7639 ga_clear(&ga);
7640 }
7641#endif
7642}
7643
7644 static void
7645f_split(typval_T *argvars, typval_T *rettv)
7646{
7647 char_u *str;
7648 char_u *end;
7649 char_u *pat = NULL;
7650 regmatch_T regmatch;
7651 char_u patbuf[NUMBUFLEN];
7652 char_u *save_cpo;
7653 int match;
7654 colnr_T col = 0;
7655 int keepempty = FALSE;
7656 int typeerr = FALSE;
7657
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007658 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007659 save_cpo = p_cpo;
7660 p_cpo = (char_u *)"";
7661
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007662 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007663 if (argvars[1].v_type != VAR_UNKNOWN)
7664 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007665 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007666 if (pat == NULL)
7667 typeerr = TRUE;
7668 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007669 keepempty = (int)tv_get_number_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007670 }
7671 if (pat == NULL || *pat == NUL)
7672 pat = (char_u *)"[\\x01- ]\\+";
7673
7674 if (rettv_list_alloc(rettv) == FAIL)
7675 return;
7676 if (typeerr)
7677 return;
7678
7679 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
7680 if (regmatch.regprog != NULL)
7681 {
7682 regmatch.rm_ic = FALSE;
7683 while (*str != NUL || keepempty)
7684 {
7685 if (*str == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007686 match = FALSE; // empty item at the end
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007687 else
7688 match = vim_regexec_nl(&regmatch, str, col);
7689 if (match)
7690 end = regmatch.startp[0];
7691 else
7692 end = str + STRLEN(str);
7693 if (keepempty || end > str || (rettv->vval.v_list->lv_len > 0
7694 && *str != NUL && match && end < regmatch.endp[0]))
7695 {
7696 if (list_append_string(rettv->vval.v_list, str,
7697 (int)(end - str)) == FAIL)
7698 break;
7699 }
7700 if (!match)
7701 break;
Bram Moolenaar13505972019-01-24 15:04:48 +01007702 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007703 if (regmatch.endp[0] > str)
7704 col = 0;
7705 else
Bram Moolenaar13505972019-01-24 15:04:48 +01007706 // Don't get stuck at the same match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007707 col = (*mb_ptr2len)(regmatch.endp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007708 str = regmatch.endp[0];
7709 }
7710
7711 vim_regfree(regmatch.regprog);
7712 }
7713
7714 p_cpo = save_cpo;
7715}
7716
7717#ifdef FEAT_FLOAT
7718/*
7719 * "sqrt()" function
7720 */
7721 static void
7722f_sqrt(typval_T *argvars, typval_T *rettv)
7723{
7724 float_T f = 0.0;
7725
7726 rettv->v_type = VAR_FLOAT;
7727 if (get_float_arg(argvars, &f) == OK)
7728 rettv->vval.v_float = sqrt(f);
7729 else
7730 rettv->vval.v_float = 0.0;
7731}
Bram Moolenaar0387cae2019-11-29 21:07:58 +01007732#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007733
Bram Moolenaar0387cae2019-11-29 21:07:58 +01007734#ifdef FEAT_FLOAT
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007735/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007736 * "str2float()" function
7737 */
7738 static void
7739f_str2float(typval_T *argvars, typval_T *rettv)
7740{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007741 char_u *p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01007742 int isneg = (*p == '-');
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007743
Bram Moolenaar08243d22017-01-10 16:12:29 +01007744 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007745 p = skipwhite(p + 1);
7746 (void)string2float(p, &rettv->vval.v_float);
Bram Moolenaar08243d22017-01-10 16:12:29 +01007747 if (isneg)
7748 rettv->vval.v_float *= -1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007749 rettv->v_type = VAR_FLOAT;
7750}
7751#endif
7752
7753/*
Bram Moolenaar9d401282019-04-06 13:18:12 +02007754 * "str2list()" function
7755 */
7756 static void
7757f_str2list(typval_T *argvars, typval_T *rettv)
7758{
7759 char_u *p;
7760 int utf8 = FALSE;
7761
7762 if (rettv_list_alloc(rettv) == FAIL)
7763 return;
7764
7765 if (argvars[1].v_type != VAR_UNKNOWN)
7766 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
7767
7768 p = tv_get_string(&argvars[0]);
7769
7770 if (has_mbyte || utf8)
7771 {
7772 int (*ptr2len)(char_u *);
7773 int (*ptr2char)(char_u *);
7774
7775 if (utf8 || enc_utf8)
7776 {
7777 ptr2len = utf_ptr2len;
7778 ptr2char = utf_ptr2char;
7779 }
7780 else
7781 {
7782 ptr2len = mb_ptr2len;
7783 ptr2char = mb_ptr2char;
7784 }
7785
7786 for ( ; *p != NUL; p += (*ptr2len)(p))
7787 list_append_number(rettv->vval.v_list, (*ptr2char)(p));
7788 }
7789 else
7790 for ( ; *p != NUL; ++p)
7791 list_append_number(rettv->vval.v_list, *p);
7792}
7793
7794/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007795 * "str2nr()" function
7796 */
7797 static void
7798f_str2nr(typval_T *argvars, typval_T *rettv)
7799{
7800 int base = 10;
7801 char_u *p;
7802 varnumber_T n;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007803 int what = 0;
Bram Moolenaar08243d22017-01-10 16:12:29 +01007804 int isneg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007805
7806 if (argvars[1].v_type != VAR_UNKNOWN)
7807 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007808 base = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007809 if (base != 2 && base != 8 && base != 10 && base != 16)
7810 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007811 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007812 return;
7813 }
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007814 if (argvars[2].v_type != VAR_UNKNOWN && tv_get_number(&argvars[2]))
7815 what |= STR2NR_QUOTE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007816 }
7817
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007818 p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01007819 isneg = (*p == '-');
7820 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007821 p = skipwhite(p + 1);
7822 switch (base)
7823 {
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007824 case 2: what |= STR2NR_BIN + STR2NR_FORCE; break;
7825 case 8: what |= STR2NR_OCT + STR2NR_FORCE; break;
7826 case 16: what |= STR2NR_HEX + STR2NR_FORCE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007827 }
Bram Moolenaar16e9b852019-05-19 19:59:35 +02007828 vim_str2nr(p, NULL, NULL, what, &n, NULL, 0, FALSE);
7829 // Text after the number is silently ignored.
Bram Moolenaar08243d22017-01-10 16:12:29 +01007830 if (isneg)
7831 rettv->vval.v_number = -n;
7832 else
7833 rettv->vval.v_number = n;
7834
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007835}
7836
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007837/*
7838 * "strgetchar()" function
7839 */
7840 static void
7841f_strgetchar(typval_T *argvars, typval_T *rettv)
7842{
7843 char_u *str;
7844 int len;
7845 int error = FALSE;
7846 int charidx;
Bram Moolenaar13505972019-01-24 15:04:48 +01007847 int byteidx = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007848
7849 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007850 str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007851 if (str == NULL)
7852 return;
7853 len = (int)STRLEN(str);
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007854 charidx = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007855 if (error)
7856 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007857
Bram Moolenaar13505972019-01-24 15:04:48 +01007858 while (charidx >= 0 && byteidx < len)
7859 {
7860 if (charidx == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007861 {
Bram Moolenaar13505972019-01-24 15:04:48 +01007862 rettv->vval.v_number = mb_ptr2char(str + byteidx);
7863 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007864 }
Bram Moolenaar13505972019-01-24 15:04:48 +01007865 --charidx;
7866 byteidx += MB_CPTR2LEN(str + byteidx);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007867 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007868}
7869
7870/*
7871 * "stridx()" function
7872 */
7873 static void
7874f_stridx(typval_T *argvars, typval_T *rettv)
7875{
7876 char_u buf[NUMBUFLEN];
7877 char_u *needle;
7878 char_u *haystack;
7879 char_u *save_haystack;
7880 char_u *pos;
7881 int start_idx;
7882
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007883 needle = tv_get_string_chk(&argvars[1]);
7884 save_haystack = haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007885 rettv->vval.v_number = -1;
7886 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007887 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007888
7889 if (argvars[2].v_type != VAR_UNKNOWN)
7890 {
7891 int error = FALSE;
7892
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007893 start_idx = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007894 if (error || start_idx >= (int)STRLEN(haystack))
7895 return;
7896 if (start_idx >= 0)
7897 haystack += start_idx;
7898 }
7899
7900 pos = (char_u *)strstr((char *)haystack, (char *)needle);
7901 if (pos != NULL)
7902 rettv->vval.v_number = (varnumber_T)(pos - save_haystack);
7903}
7904
7905/*
7906 * "string()" function
7907 */
Bram Moolenaar461a7fc2018-12-22 13:28:07 +01007908 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007909f_string(typval_T *argvars, typval_T *rettv)
7910{
7911 char_u *tofree;
7912 char_u numbuf[NUMBUFLEN];
7913
7914 rettv->v_type = VAR_STRING;
7915 rettv->vval.v_string = tv2string(&argvars[0], &tofree, numbuf,
7916 get_copyID());
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007917 // Make a copy if we have a value but it's not in allocated memory.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007918 if (rettv->vval.v_string != NULL && tofree == NULL)
7919 rettv->vval.v_string = vim_strsave(rettv->vval.v_string);
7920}
7921
7922/*
7923 * "strlen()" function
7924 */
7925 static void
7926f_strlen(typval_T *argvars, typval_T *rettv)
7927{
7928 rettv->vval.v_number = (varnumber_T)(STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007929 tv_get_string(&argvars[0])));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007930}
7931
7932/*
7933 * "strchars()" function
7934 */
7935 static void
7936f_strchars(typval_T *argvars, typval_T *rettv)
7937{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007938 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007939 int skipcc = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007940 varnumber_T len = 0;
7941 int (*func_mb_ptr2char_adv)(char_u **pp);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007942
7943 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007944 skipcc = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007945 if (skipcc < 0 || skipcc > 1)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007946 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007947 else
7948 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007949 func_mb_ptr2char_adv = skipcc ? mb_ptr2char_adv : mb_cptr2char_adv;
7950 while (*s != NUL)
7951 {
7952 func_mb_ptr2char_adv(&s);
7953 ++len;
7954 }
7955 rettv->vval.v_number = len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007956 }
7957}
7958
7959/*
7960 * "strdisplaywidth()" function
7961 */
7962 static void
7963f_strdisplaywidth(typval_T *argvars, typval_T *rettv)
7964{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007965 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007966 int col = 0;
7967
7968 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007969 col = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007970
7971 rettv->vval.v_number = (varnumber_T)(linetabsize_col(col, s) - col);
7972}
7973
7974/*
7975 * "strwidth()" function
7976 */
7977 static void
7978f_strwidth(typval_T *argvars, typval_T *rettv)
7979{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007980 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007981
Bram Moolenaar13505972019-01-24 15:04:48 +01007982 rettv->vval.v_number = (varnumber_T)(mb_string2cells(s, -1));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007983}
7984
7985/*
7986 * "strcharpart()" function
7987 */
7988 static void
7989f_strcharpart(typval_T *argvars, typval_T *rettv)
7990{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007991 char_u *p;
7992 int nchar;
7993 int nbyte = 0;
7994 int charlen;
7995 int len = 0;
7996 int slen;
7997 int error = FALSE;
7998
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007999 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008000 slen = (int)STRLEN(p);
8001
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008002 nchar = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008003 if (!error)
8004 {
8005 if (nchar > 0)
8006 while (nchar > 0 && nbyte < slen)
8007 {
Bram Moolenaard3c907b2016-08-17 21:32:09 +02008008 nbyte += MB_CPTR2LEN(p + nbyte);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008009 --nchar;
8010 }
8011 else
8012 nbyte = nchar;
8013 if (argvars[2].v_type != VAR_UNKNOWN)
8014 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008015 charlen = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008016 while (charlen > 0 && nbyte + len < slen)
8017 {
8018 int off = nbyte + len;
8019
8020 if (off < 0)
8021 len += 1;
8022 else
Bram Moolenaard3c907b2016-08-17 21:32:09 +02008023 len += MB_CPTR2LEN(p + off);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008024 --charlen;
8025 }
8026 }
8027 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008028 len = slen - nbyte; // default: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008029 }
8030
8031 /*
8032 * Only return the overlap between the specified part and the actual
8033 * string.
8034 */
8035 if (nbyte < 0)
8036 {
8037 len += nbyte;
8038 nbyte = 0;
8039 }
8040 else if (nbyte > slen)
8041 nbyte = slen;
8042 if (len < 0)
8043 len = 0;
8044 else if (nbyte + len > slen)
8045 len = slen - nbyte;
8046
8047 rettv->v_type = VAR_STRING;
8048 rettv->vval.v_string = vim_strnsave(p + nbyte, len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008049}
8050
8051/*
8052 * "strpart()" function
8053 */
8054 static void
8055f_strpart(typval_T *argvars, typval_T *rettv)
8056{
8057 char_u *p;
8058 int n;
8059 int len;
8060 int slen;
8061 int error = FALSE;
8062
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008063 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008064 slen = (int)STRLEN(p);
8065
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008066 n = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008067 if (error)
8068 len = 0;
8069 else if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008070 len = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008071 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008072 len = slen - n; // default len: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008073
8074 /*
8075 * Only return the overlap between the specified part and the actual
8076 * string.
8077 */
8078 if (n < 0)
8079 {
8080 len += n;
8081 n = 0;
8082 }
8083 else if (n > slen)
8084 n = slen;
8085 if (len < 0)
8086 len = 0;
8087 else if (n + len > slen)
8088 len = slen - n;
8089
8090 rettv->v_type = VAR_STRING;
8091 rettv->vval.v_string = vim_strnsave(p + n, len);
8092}
8093
8094/*
8095 * "strridx()" function
8096 */
8097 static void
8098f_strridx(typval_T *argvars, typval_T *rettv)
8099{
8100 char_u buf[NUMBUFLEN];
8101 char_u *needle;
8102 char_u *haystack;
8103 char_u *rest;
8104 char_u *lastmatch = NULL;
8105 int haystack_len, end_idx;
8106
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008107 needle = tv_get_string_chk(&argvars[1]);
8108 haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008109
8110 rettv->vval.v_number = -1;
8111 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008112 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008113
8114 haystack_len = (int)STRLEN(haystack);
8115 if (argvars[2].v_type != VAR_UNKNOWN)
8116 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008117 // Third argument: upper limit for index
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008118 end_idx = (int)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008119 if (end_idx < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008120 return; // can never find a match
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008121 }
8122 else
8123 end_idx = haystack_len;
8124
8125 if (*needle == NUL)
8126 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008127 // Empty string matches past the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008128 lastmatch = haystack + end_idx;
8129 }
8130 else
8131 {
8132 for (rest = haystack; *rest != '\0'; ++rest)
8133 {
8134 rest = (char_u *)strstr((char *)rest, (char *)needle);
8135 if (rest == NULL || rest > haystack + end_idx)
8136 break;
8137 lastmatch = rest;
8138 }
8139 }
8140
8141 if (lastmatch == NULL)
8142 rettv->vval.v_number = -1;
8143 else
8144 rettv->vval.v_number = (varnumber_T)(lastmatch - haystack);
8145}
8146
8147/*
8148 * "strtrans()" function
8149 */
8150 static void
8151f_strtrans(typval_T *argvars, typval_T *rettv)
8152{
8153 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008154 rettv->vval.v_string = transstr(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008155}
8156
8157/*
8158 * "submatch()" function
8159 */
8160 static void
8161f_submatch(typval_T *argvars, typval_T *rettv)
8162{
8163 int error = FALSE;
8164 int no;
8165 int retList = 0;
8166
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008167 no = (int)tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008168 if (error)
8169 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02008170 if (no < 0 || no >= NSUBEXP)
8171 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008172 semsg(_("E935: invalid submatch number: %d"), no);
Bram Moolenaar79518e22017-02-17 16:31:35 +01008173 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02008174 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008175 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008176 retList = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008177 if (error)
8178 return;
8179
8180 if (retList == 0)
8181 {
8182 rettv->v_type = VAR_STRING;
8183 rettv->vval.v_string = reg_submatch(no);
8184 }
8185 else
8186 {
8187 rettv->v_type = VAR_LIST;
8188 rettv->vval.v_list = reg_submatch_list(no);
8189 }
8190}
8191
8192/*
8193 * "substitute()" function
8194 */
8195 static void
8196f_substitute(typval_T *argvars, typval_T *rettv)
8197{
8198 char_u patbuf[NUMBUFLEN];
8199 char_u subbuf[NUMBUFLEN];
8200 char_u flagsbuf[NUMBUFLEN];
8201
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008202 char_u *str = tv_get_string_chk(&argvars[0]);
8203 char_u *pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008204 char_u *sub = NULL;
8205 typval_T *expr = NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008206 char_u *flg = tv_get_string_buf_chk(&argvars[3], flagsbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008207
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008208 if (argvars[2].v_type == VAR_FUNC || argvars[2].v_type == VAR_PARTIAL)
8209 expr = &argvars[2];
8210 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008211 sub = tv_get_string_buf_chk(&argvars[2], subbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008212
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008213 rettv->v_type = VAR_STRING;
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008214 if (str == NULL || pat == NULL || (sub == NULL && expr == NULL)
8215 || flg == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008216 rettv->vval.v_string = NULL;
8217 else
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008218 rettv->vval.v_string = do_string_sub(str, pat, sub, expr, flg);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008219}
8220
8221/*
Bram Moolenaar00f123a2018-08-21 20:28:54 +02008222 * "swapinfo(swap_filename)" function
8223 */
8224 static void
8225f_swapinfo(typval_T *argvars, typval_T *rettv)
8226{
8227 if (rettv_dict_alloc(rettv) == OK)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008228 get_b0_dict(tv_get_string(argvars), rettv->vval.v_dict);
Bram Moolenaar00f123a2018-08-21 20:28:54 +02008229}
8230
8231/*
Bram Moolenaar110bd602018-09-16 18:46:59 +02008232 * "swapname(expr)" function
8233 */
8234 static void
8235f_swapname(typval_T *argvars, typval_T *rettv)
8236{
8237 buf_T *buf;
8238
8239 rettv->v_type = VAR_STRING;
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01008240 buf = tv_get_buf(&argvars[0], FALSE);
Bram Moolenaar110bd602018-09-16 18:46:59 +02008241 if (buf == NULL || buf->b_ml.ml_mfp == NULL
8242 || buf->b_ml.ml_mfp->mf_fname == NULL)
8243 rettv->vval.v_string = NULL;
8244 else
8245 rettv->vval.v_string = vim_strsave(buf->b_ml.ml_mfp->mf_fname);
8246}
8247
8248/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008249 * "synID(lnum, col, trans)" function
8250 */
8251 static void
8252f_synID(typval_T *argvars UNUSED, typval_T *rettv)
8253{
8254 int id = 0;
8255#ifdef FEAT_SYN_HL
8256 linenr_T lnum;
8257 colnr_T col;
8258 int trans;
8259 int transerr = FALSE;
8260
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008261 lnum = tv_get_lnum(argvars); // -1 on type error
8262 col = (linenr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008263 trans = (int)tv_get_number_chk(&argvars[2], &transerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008264
8265 if (!transerr && lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8266 && col >= 0 && col < (long)STRLEN(ml_get(lnum)))
8267 id = syn_get_id(curwin, lnum, (colnr_T)col, trans, NULL, FALSE);
8268#endif
8269
8270 rettv->vval.v_number = id;
8271}
8272
8273/*
8274 * "synIDattr(id, what [, mode])" function
8275 */
8276 static void
8277f_synIDattr(typval_T *argvars UNUSED, typval_T *rettv)
8278{
8279 char_u *p = NULL;
8280#ifdef FEAT_SYN_HL
8281 int id;
8282 char_u *what;
8283 char_u *mode;
8284 char_u modebuf[NUMBUFLEN];
8285 int modec;
8286
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008287 id = (int)tv_get_number(&argvars[0]);
8288 what = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008289 if (argvars[2].v_type != VAR_UNKNOWN)
8290 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008291 mode = tv_get_string_buf(&argvars[2], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008292 modec = TOLOWER_ASC(mode[0]);
8293 if (modec != 't' && modec != 'c' && modec != 'g')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008294 modec = 0; // replace invalid with current
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008295 }
8296 else
8297 {
8298#if defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS)
8299 if (USE_24BIT)
8300 modec = 'g';
8301 else
8302#endif
8303 if (t_colors > 1)
8304 modec = 'c';
8305 else
8306 modec = 't';
8307 }
8308
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008309 switch (TOLOWER_ASC(what[0]))
8310 {
8311 case 'b':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008312 if (TOLOWER_ASC(what[1]) == 'g') // bg[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008313 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008314 else // bold
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008315 p = highlight_has_attr(id, HL_BOLD, modec);
8316 break;
8317
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008318 case 'f': // fg[#] or font
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008319 p = highlight_color(id, what, modec);
8320 break;
8321
8322 case 'i':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008323 if (TOLOWER_ASC(what[1]) == 'n') // inverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008324 p = highlight_has_attr(id, HL_INVERSE, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008325 else // italic
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008326 p = highlight_has_attr(id, HL_ITALIC, modec);
8327 break;
8328
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008329 case 'n': // name
Bram Moolenaarc96272e2017-03-26 13:50:09 +02008330 p = get_highlight_name_ext(NULL, id - 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008331 break;
8332
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008333 case 'r': // reverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008334 p = highlight_has_attr(id, HL_INVERSE, modec);
8335 break;
8336
8337 case 's':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008338 if (TOLOWER_ASC(what[1]) == 'p') // sp[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008339 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008340 // strikeout
Bram Moolenaarcf4b00c2017-09-02 18:33:56 +02008341 else if (TOLOWER_ASC(what[1]) == 't' &&
8342 TOLOWER_ASC(what[2]) == 'r')
8343 p = highlight_has_attr(id, HL_STRIKETHROUGH, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008344 else // standout
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008345 p = highlight_has_attr(id, HL_STANDOUT, modec);
8346 break;
8347
8348 case 'u':
8349 if (STRLEN(what) <= 5 || TOLOWER_ASC(what[5]) != 'c')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008350 // underline
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008351 p = highlight_has_attr(id, HL_UNDERLINE, modec);
8352 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008353 // undercurl
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008354 p = highlight_has_attr(id, HL_UNDERCURL, modec);
8355 break;
8356 }
8357
8358 if (p != NULL)
8359 p = vim_strsave(p);
8360#endif
8361 rettv->v_type = VAR_STRING;
8362 rettv->vval.v_string = p;
8363}
8364
8365/*
8366 * "synIDtrans(id)" function
8367 */
8368 static void
8369f_synIDtrans(typval_T *argvars UNUSED, typval_T *rettv)
8370{
8371 int id;
8372
8373#ifdef FEAT_SYN_HL
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008374 id = (int)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008375
8376 if (id > 0)
8377 id = syn_get_final_id(id);
8378 else
8379#endif
8380 id = 0;
8381
8382 rettv->vval.v_number = id;
8383}
8384
8385/*
8386 * "synconcealed(lnum, col)" function
8387 */
8388 static void
8389f_synconcealed(typval_T *argvars UNUSED, typval_T *rettv)
8390{
8391#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
8392 linenr_T lnum;
8393 colnr_T col;
8394 int syntax_flags = 0;
8395 int cchar;
8396 int matchid = 0;
8397 char_u str[NUMBUFLEN];
8398#endif
8399
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02008400 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008401
8402#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008403 lnum = tv_get_lnum(argvars); // -1 on type error
8404 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008405
Bram Moolenaara80faa82020-04-12 19:37:17 +02008406 CLEAR_FIELD(str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008407
8408 if (rettv_list_alloc(rettv) != FAIL)
8409 {
8410 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8411 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
8412 && curwin->w_p_cole > 0)
8413 {
8414 (void)syn_get_id(curwin, lnum, col, FALSE, NULL, FALSE);
8415 syntax_flags = get_syntax_info(&matchid);
8416
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008417 // get the conceal character
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008418 if ((syntax_flags & HL_CONCEAL) && curwin->w_p_cole < 3)
8419 {
8420 cchar = syn_get_sub_char();
Bram Moolenaar4d785892017-06-22 22:00:50 +02008421 if (cchar == NUL && curwin->w_p_cole == 1)
8422 cchar = (lcs_conceal == NUL) ? ' ' : lcs_conceal;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008423 if (cchar != NUL)
8424 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008425 if (has_mbyte)
8426 (*mb_char2bytes)(cchar, str);
8427 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008428 str[0] = cchar;
8429 }
8430 }
8431 }
8432
8433 list_append_number(rettv->vval.v_list,
8434 (syntax_flags & HL_CONCEAL) != 0);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008435 // -1 to auto-determine strlen
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008436 list_append_string(rettv->vval.v_list, str, -1);
8437 list_append_number(rettv->vval.v_list, matchid);
8438 }
8439#endif
8440}
8441
8442/*
8443 * "synstack(lnum, col)" function
8444 */
8445 static void
8446f_synstack(typval_T *argvars UNUSED, typval_T *rettv)
8447{
8448#ifdef FEAT_SYN_HL
8449 linenr_T lnum;
8450 colnr_T col;
8451 int i;
8452 int id;
8453#endif
8454
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02008455 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008456
8457#ifdef FEAT_SYN_HL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008458 lnum = tv_get_lnum(argvars); // -1 on type error
8459 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008460
8461 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8462 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
8463 && rettv_list_alloc(rettv) != FAIL)
8464 {
8465 (void)syn_get_id(curwin, lnum, (colnr_T)col, FALSE, NULL, TRUE);
8466 for (i = 0; ; ++i)
8467 {
8468 id = syn_get_stack_item(i);
8469 if (id < 0)
8470 break;
8471 if (list_append_number(rettv->vval.v_list, id) == FAIL)
8472 break;
8473 }
8474 }
8475#endif
8476}
8477
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008478/*
8479 * "tabpagebuflist()" function
8480 */
8481 static void
8482f_tabpagebuflist(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8483{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008484 tabpage_T *tp;
8485 win_T *wp = NULL;
8486
8487 if (argvars[0].v_type == VAR_UNKNOWN)
8488 wp = firstwin;
8489 else
8490 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008491 tp = find_tabpage((int)tv_get_number(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008492 if (tp != NULL)
8493 wp = (tp == curtab) ? firstwin : tp->tp_firstwin;
8494 }
8495 if (wp != NULL && rettv_list_alloc(rettv) != FAIL)
8496 {
8497 for (; wp != NULL; wp = wp->w_next)
8498 if (list_append_number(rettv->vval.v_list,
8499 wp->w_buffer->b_fnum) == FAIL)
8500 break;
8501 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008502}
8503
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008504/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008505 * "tagfiles()" function
8506 */
8507 static void
8508f_tagfiles(typval_T *argvars UNUSED, typval_T *rettv)
8509{
8510 char_u *fname;
8511 tagname_T tn;
8512 int first;
8513
8514 if (rettv_list_alloc(rettv) == FAIL)
8515 return;
8516 fname = alloc(MAXPATHL);
8517 if (fname == NULL)
8518 return;
8519
8520 for (first = TRUE; ; first = FALSE)
8521 if (get_tagfname(&tn, first, fname) == FAIL
8522 || list_append_string(rettv->vval.v_list, fname, -1) == FAIL)
8523 break;
8524 tagname_free(&tn);
8525 vim_free(fname);
8526}
8527
8528/*
8529 * "taglist()" function
8530 */
8531 static void
8532f_taglist(typval_T *argvars, typval_T *rettv)
8533{
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008534 char_u *fname = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008535 char_u *tag_pattern;
8536
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008537 tag_pattern = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008538
8539 rettv->vval.v_number = FALSE;
8540 if (*tag_pattern == NUL)
8541 return;
8542
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008543 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008544 fname = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008545 if (rettv_list_alloc(rettv) == OK)
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008546 (void)get_tags(rettv->vval.v_list, tag_pattern, fname);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008547}
8548
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008549#ifdef FEAT_FLOAT
8550/*
8551 * "tan()" function
8552 */
8553 static void
8554f_tan(typval_T *argvars, typval_T *rettv)
8555{
8556 float_T f = 0.0;
8557
8558 rettv->v_type = VAR_FLOAT;
8559 if (get_float_arg(argvars, &f) == OK)
8560 rettv->vval.v_float = tan(f);
8561 else
8562 rettv->vval.v_float = 0.0;
8563}
8564
8565/*
8566 * "tanh()" function
8567 */
8568 static void
8569f_tanh(typval_T *argvars, typval_T *rettv)
8570{
8571 float_T f = 0.0;
8572
8573 rettv->v_type = VAR_FLOAT;
8574 if (get_float_arg(argvars, &f) == OK)
8575 rettv->vval.v_float = tanh(f);
8576 else
8577 rettv->vval.v_float = 0.0;
8578}
8579#endif
8580
8581/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008582 * "tolower(string)" function
8583 */
8584 static void
8585f_tolower(typval_T *argvars, typval_T *rettv)
8586{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008587 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008588 rettv->vval.v_string = strlow_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008589}
8590
8591/*
8592 * "toupper(string)" function
8593 */
8594 static void
8595f_toupper(typval_T *argvars, typval_T *rettv)
8596{
8597 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008598 rettv->vval.v_string = strup_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008599}
8600
8601/*
8602 * "tr(string, fromstr, tostr)" function
8603 */
8604 static void
8605f_tr(typval_T *argvars, typval_T *rettv)
8606{
8607 char_u *in_str;
8608 char_u *fromstr;
8609 char_u *tostr;
8610 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008611 int inlen;
8612 int fromlen;
8613 int tolen;
8614 int idx;
8615 char_u *cpstr;
8616 int cplen;
8617 int first = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008618 char_u buf[NUMBUFLEN];
8619 char_u buf2[NUMBUFLEN];
8620 garray_T ga;
8621
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008622 in_str = tv_get_string(&argvars[0]);
8623 fromstr = tv_get_string_buf_chk(&argvars[1], buf);
8624 tostr = tv_get_string_buf_chk(&argvars[2], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008625
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008626 // Default return value: empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008627 rettv->v_type = VAR_STRING;
8628 rettv->vval.v_string = NULL;
8629 if (fromstr == NULL || tostr == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008630 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008631 ga_init2(&ga, (int)sizeof(char), 80);
8632
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008633 if (!has_mbyte)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008634 // not multi-byte: fromstr and tostr must be the same length
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008635 if (STRLEN(fromstr) != STRLEN(tostr))
8636 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008637error:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008638 semsg(_(e_invarg2), fromstr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008639 ga_clear(&ga);
8640 return;
8641 }
8642
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008643 // fromstr and tostr have to contain the same number of chars
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008644 while (*in_str != NUL)
8645 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008646 if (has_mbyte)
8647 {
8648 inlen = (*mb_ptr2len)(in_str);
8649 cpstr = in_str;
8650 cplen = inlen;
8651 idx = 0;
8652 for (p = fromstr; *p != NUL; p += fromlen)
8653 {
8654 fromlen = (*mb_ptr2len)(p);
8655 if (fromlen == inlen && STRNCMP(in_str, p, inlen) == 0)
8656 {
8657 for (p = tostr; *p != NUL; p += tolen)
8658 {
8659 tolen = (*mb_ptr2len)(p);
8660 if (idx-- == 0)
8661 {
8662 cplen = tolen;
8663 cpstr = p;
8664 break;
8665 }
8666 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008667 if (*p == NUL) // tostr is shorter than fromstr
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008668 goto error;
8669 break;
8670 }
8671 ++idx;
8672 }
8673
8674 if (first && cpstr == in_str)
8675 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008676 // Check that fromstr and tostr have the same number of
8677 // (multi-byte) characters. Done only once when a character
8678 // of in_str doesn't appear in fromstr.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008679 first = FALSE;
8680 for (p = tostr; *p != NUL; p += tolen)
8681 {
8682 tolen = (*mb_ptr2len)(p);
8683 --idx;
8684 }
8685 if (idx != 0)
8686 goto error;
8687 }
8688
8689 (void)ga_grow(&ga, cplen);
8690 mch_memmove((char *)ga.ga_data + ga.ga_len, cpstr, (size_t)cplen);
8691 ga.ga_len += cplen;
8692
8693 in_str += inlen;
8694 }
8695 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008696 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008697 // When not using multi-byte chars we can do it faster.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008698 p = vim_strchr(fromstr, *in_str);
8699 if (p != NULL)
8700 ga_append(&ga, tostr[p - fromstr]);
8701 else
8702 ga_append(&ga, *in_str);
8703 ++in_str;
8704 }
8705 }
8706
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008707 // add a terminating NUL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008708 (void)ga_grow(&ga, 1);
8709 ga_append(&ga, NUL);
8710
8711 rettv->vval.v_string = ga.ga_data;
8712}
8713
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008714/*
8715 * "trim({expr})" function
8716 */
8717 static void
8718f_trim(typval_T *argvars, typval_T *rettv)
8719{
8720 char_u buf1[NUMBUFLEN];
8721 char_u buf2[NUMBUFLEN];
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008722 char_u *head = tv_get_string_buf_chk(&argvars[0], buf1);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008723 char_u *mask = NULL;
8724 char_u *tail;
8725 char_u *prev;
8726 char_u *p;
8727 int c1;
8728
8729 rettv->v_type = VAR_STRING;
8730 if (head == NULL)
8731 {
8732 rettv->vval.v_string = NULL;
8733 return;
8734 }
8735
8736 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008737 mask = tv_get_string_buf_chk(&argvars[1], buf2);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008738
8739 while (*head != NUL)
8740 {
8741 c1 = PTR2CHAR(head);
8742 if (mask == NULL)
8743 {
8744 if (c1 > ' ' && c1 != 0xa0)
8745 break;
8746 }
8747 else
8748 {
8749 for (p = mask; *p != NUL; MB_PTR_ADV(p))
8750 if (c1 == PTR2CHAR(p))
8751 break;
8752 if (*p == NUL)
8753 break;
8754 }
8755 MB_PTR_ADV(head);
8756 }
8757
8758 for (tail = head + STRLEN(head); tail > head; tail = prev)
8759 {
8760 prev = tail;
8761 MB_PTR_BACK(head, prev);
8762 c1 = PTR2CHAR(prev);
8763 if (mask == NULL)
8764 {
8765 if (c1 > ' ' && c1 != 0xa0)
8766 break;
8767 }
8768 else
8769 {
8770 for (p = mask; *p != NUL; MB_PTR_ADV(p))
8771 if (c1 == PTR2CHAR(p))
8772 break;
8773 if (*p == NUL)
8774 break;
8775 }
8776 }
8777 rettv->vval.v_string = vim_strnsave(head, (int)(tail - head));
8778}
8779
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008780#ifdef FEAT_FLOAT
8781/*
8782 * "trunc({float})" function
8783 */
8784 static void
8785f_trunc(typval_T *argvars, typval_T *rettv)
8786{
8787 float_T f = 0.0;
8788
8789 rettv->v_type = VAR_FLOAT;
8790 if (get_float_arg(argvars, &f) == OK)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008791 // trunc() is not in C90, use floor() or ceil() instead.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008792 rettv->vval.v_float = f > 0 ? floor(f) : ceil(f);
8793 else
8794 rettv->vval.v_float = 0.0;
8795}
8796#endif
8797
8798/*
8799 * "type(expr)" function
8800 */
8801 static void
8802f_type(typval_T *argvars, typval_T *rettv)
8803{
8804 int n = -1;
8805
8806 switch (argvars[0].v_type)
8807 {
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01008808 case VAR_NUMBER: n = VAR_TYPE_NUMBER; break;
8809 case VAR_STRING: n = VAR_TYPE_STRING; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008810 case VAR_PARTIAL:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01008811 case VAR_FUNC: n = VAR_TYPE_FUNC; break;
8812 case VAR_LIST: n = VAR_TYPE_LIST; break;
8813 case VAR_DICT: n = VAR_TYPE_DICT; break;
8814 case VAR_FLOAT: n = VAR_TYPE_FLOAT; break;
8815 case VAR_BOOL: n = VAR_TYPE_BOOL; break;
8816 case VAR_SPECIAL: n = VAR_TYPE_NONE; break;
Bram Moolenaarf562e722016-07-19 17:25:25 +02008817 case VAR_JOB: n = VAR_TYPE_JOB; break;
8818 case VAR_CHANNEL: n = VAR_TYPE_CHANNEL; break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01008819 case VAR_BLOB: n = VAR_TYPE_BLOB; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008820 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02008821 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01008822 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01008823 internal_error_no_abort("f_type(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008824 n = -1;
8825 break;
8826 }
8827 rettv->vval.v_number = n;
8828}
8829
8830/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008831 * "virtcol(string)" function
8832 */
8833 static void
8834f_virtcol(typval_T *argvars, typval_T *rettv)
8835{
8836 colnr_T vcol = 0;
8837 pos_T *fp;
8838 int fnum = curbuf->b_fnum;
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01008839 int len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008840
8841 fp = var2fpos(&argvars[0], FALSE, &fnum);
8842 if (fp != NULL && fp->lnum <= curbuf->b_ml.ml_line_count
8843 && fnum == curbuf->b_fnum)
8844 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01008845 // Limit the column to a valid value, getvvcol() doesn't check.
8846 if (fp->col < 0)
8847 fp->col = 0;
8848 else
8849 {
8850 len = (int)STRLEN(ml_get(fp->lnum));
8851 if (fp->col > len)
8852 fp->col = len;
8853 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008854 getvvcol(curwin, fp, NULL, NULL, &vcol);
8855 ++vcol;
8856 }
8857
8858 rettv->vval.v_number = vcol;
8859}
8860
8861/*
8862 * "visualmode()" function
8863 */
8864 static void
8865f_visualmode(typval_T *argvars, typval_T *rettv)
8866{
8867 char_u str[2];
8868
8869 rettv->v_type = VAR_STRING;
8870 str[0] = curbuf->b_visual_mode_eval;
8871 str[1] = NUL;
8872 rettv->vval.v_string = vim_strsave(str);
8873
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008874 // A non-zero number or non-empty string argument: reset mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008875 if (non_zero_arg(&argvars[0]))
8876 curbuf->b_visual_mode_eval = NUL;
8877}
8878
8879/*
8880 * "wildmenumode()" function
8881 */
8882 static void
8883f_wildmenumode(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8884{
8885#ifdef FEAT_WILDMENU
8886 if (wild_menu_showing)
8887 rettv->vval.v_number = 1;
8888#endif
8889}
8890
8891/*
Bram Moolenaar0c1e3742019-12-27 13:49:24 +01008892 * "windowsversion()" function
8893 */
8894 static void
8895f_windowsversion(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8896{
8897 rettv->v_type = VAR_STRING;
8898 rettv->vval.v_string = vim_strsave((char_u *)windowsVersion);
8899}
8900
8901/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008902 * "wordcount()" function
8903 */
8904 static void
8905f_wordcount(typval_T *argvars UNUSED, typval_T *rettv)
8906{
8907 if (rettv_dict_alloc(rettv) == FAIL)
8908 return;
8909 cursor_pos_info(rettv->vval.v_dict);
8910}
8911
8912/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008913 * "xor(expr, expr)" function
8914 */
8915 static void
8916f_xor(typval_T *argvars, typval_T *rettv)
8917{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008918 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
8919 ^ tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008920}
8921
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008922#endif // FEAT_EVAL