blob: 9f1a80ad86a7616fc3a6c2ba973e5502ce4ac15c [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
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 * search.c: code for normal mode searching commands
11 */
12
13#include "vim.h"
14
Bram Moolenaar071d4272004-06-13 20:20:40 +000015#ifdef FEAT_EVAL
Bram Moolenaarbaaa7e92016-01-29 22:47:03 +010016static void set_vv_searchforward(void);
17static int first_submatch(regmmatch_T *rp);
Bram Moolenaar071d4272004-06-13 20:20:40 +000018#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +000019#ifdef FEAT_FIND_ID
Bram Moolenaarbaaa7e92016-01-29 22:47:03 +010020static void show_pat_in_path(char_u *, int,
21 int, int, FILE *, linenr_T *, long);
Bram Moolenaar071d4272004-06-13 20:20:40 +000022#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +020023
24typedef struct searchstat
25{
26 int cur; // current position of found words
27 int cnt; // total count of found words
28 int exact_match; // TRUE if matched exactly on specified position
29 int incomplete; // 0: search was fully completed
30 // 1: recomputing was timed out
31 // 2: max count exceeded
32 int last_maxcount; // the max count of the last search
33} searchstat_T;
34
35static void cmdline_search_stat(int dirc, pos_T *pos, pos_T *cursor_pos, int show_top_bot_msg, char_u *msgbuf, int recompute, int maxcount, long timeout);
36static void update_search_stat(int dirc, pos_T *pos, pos_T *cursor_pos, searchstat_T *stat, int recompute, int maxcount, long timeout);
37
Bram Moolenaarea6561a2020-06-01 21:32:45 +020038#define SEARCH_STAT_DEF_TIMEOUT 40L
Bram Moolenaare8f5ec02020-06-01 17:28:35 +020039#define SEARCH_STAT_DEF_MAX_COUNT 99
40#define SEARCH_STAT_BUF_LEN 12
Bram Moolenaar071d4272004-06-13 20:20:40 +000041
Bram Moolenaar071d4272004-06-13 20:20:40 +000042/*
43 * This file contains various searching-related routines. These fall into
44 * three groups:
45 * 1. string searches (for /, ?, n, and N)
46 * 2. character searches within a single line (for f, F, t, T, etc)
47 * 3. "other" kinds of searches like the '%' command, and 'word' searches.
48 */
49
50/*
51 * String searches
52 *
53 * The string search functions are divided into two levels:
54 * lowest: searchit(); uses an pos_T for starting position and found match.
55 * Highest: do_search(); uses curwin->w_cursor; calls searchit().
56 *
57 * The last search pattern is remembered for repeating the same search.
58 * This pattern is shared between the :g, :s, ? and / commands.
59 * This is in search_regcomp().
60 *
61 * The actual string matching is done using a heavily modified version of
62 * Henry Spencer's regular expression library. See regexp.c.
63 */
64
Bram Moolenaar071d4272004-06-13 20:20:40 +000065/*
66 * Two search patterns are remembered: One for the :substitute command and
67 * one for other searches. last_idx points to the one that was used the last
68 * time.
69 */
Bram Moolenaarc3328162019-07-23 22:15:25 +020070static spat_T spats[2] =
Bram Moolenaar071d4272004-06-13 20:20:40 +000071{
Bram Moolenaar63d9e732019-12-05 21:10:38 +010072 {NULL, TRUE, FALSE, {'/', 0, 0, 0L}}, // last used search pat
73 {NULL, TRUE, FALSE, {'/', 0, 0, 0L}} // last used substitute pat
Bram Moolenaar071d4272004-06-13 20:20:40 +000074};
75
Bram Moolenaar63d9e732019-12-05 21:10:38 +010076static int last_idx = 0; // index in spats[] for RE_LAST
Bram Moolenaar071d4272004-06-13 20:20:40 +000077
Bram Moolenaar63d9e732019-12-05 21:10:38 +010078static char_u lastc[2] = {NUL, NUL}; // last character searched for
79static int lastcdir = FORWARD; // last direction of character search
80static int last_t_cmd = TRUE; // last search t_cmd
Bram Moolenaardbd24b52015-08-11 14:26:19 +020081static char_u lastc_bytes[MB_MAXBYTES + 1];
Bram Moolenaar63d9e732019-12-05 21:10:38 +010082static int lastc_bytelen = 1; // >1 for multi-byte char
Bram Moolenaardbd24b52015-08-11 14:26:19 +020083
Bram Moolenaar63d9e732019-12-05 21:10:38 +010084// copy of spats[], for keeping the search patterns while executing autocmds
Bram Moolenaarc3328162019-07-23 22:15:25 +020085static spat_T saved_spats[2];
Bram Moolenaara2cff1d2021-10-15 12:51:29 +010086static char_u *saved_mr_pattern = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +000087# ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +010088static int saved_spats_last_idx = 0;
89static int saved_spats_no_hlsearch = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +000090# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +000091
Bram Moolenaara2cff1d2021-10-15 12:51:29 +010092// allocated copy of pattern used by search_regcomp()
93static char_u *mr_pattern = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +000094
95#ifdef FEAT_FIND_ID
96/*
97 * Type used by find_pattern_in_path() to remember which included files have
98 * been searched already.
99 */
100typedef struct SearchedFile
101{
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100102 FILE *fp; // File pointer
103 char_u *name; // Full name of file
104 linenr_T lnum; // Line we were up to in file
105 int matched; // Found a match in this file
Bram Moolenaar071d4272004-06-13 20:20:40 +0000106} SearchedFile;
107#endif
108
109/*
110 * translate search pattern for vim_regcomp()
111 *
112 * pat_save == RE_SEARCH: save pat in spats[RE_SEARCH].pat (normal search cmd)
113 * pat_save == RE_SUBST: save pat in spats[RE_SUBST].pat (:substitute command)
114 * pat_save == RE_BOTH: save pat in both patterns (:global command)
115 * pat_use == RE_SEARCH: use previous search pattern if "pat" is NULL
Bram Moolenaarb8017e72007-05-10 18:59:07 +0000116 * pat_use == RE_SUBST: use previous substitute pattern if "pat" is NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000117 * pat_use == RE_LAST: use last used pattern if "pat" is NULL
118 * options & SEARCH_HIS: put search string in history
119 * options & SEARCH_KEEP: keep previous search pattern
120 *
121 * returns FAIL if failed, OK otherwise.
122 */
123 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100124search_regcomp(
125 char_u *pat,
126 int pat_save,
127 int pat_use,
128 int options,
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100129 regmmatch_T *regmatch) // return: pattern and ignore-case flag
Bram Moolenaar071d4272004-06-13 20:20:40 +0000130{
131 int magic;
132 int i;
133
134 rc_did_emsg = FALSE;
Bram Moolenaarf4e20992020-12-21 19:59:08 +0100135 magic = magic_isset();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000136
137 /*
138 * If no pattern given, use a previously defined pattern.
139 */
140 if (pat == NULL || *pat == NUL)
141 {
142 if (pat_use == RE_LAST)
143 i = last_idx;
144 else
145 i = pat_use;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100146 if (spats[i].pat == NULL) // pattern was never defined
Bram Moolenaar071d4272004-06-13 20:20:40 +0000147 {
148 if (pat_use == RE_SUBST)
Bram Moolenaare29a27f2021-07-20 21:07:36 +0200149 emsg(_(e_no_previous_substitute_regular_expression));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000150 else
Bram Moolenaare29a27f2021-07-20 21:07:36 +0200151 emsg(_(e_no_previous_regular_expression));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152 rc_did_emsg = TRUE;
153 return FAIL;
154 }
155 pat = spats[i].pat;
156 magic = spats[i].magic;
157 no_smartcase = spats[i].no_scs;
158 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100159 else if (options & SEARCH_HIS) // put new pattern in history
Bram Moolenaar071d4272004-06-13 20:20:40 +0000160 add_to_history(HIST_SEARCH, pat, TRUE, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000161
Bram Moolenaara2cff1d2021-10-15 12:51:29 +0100162 vim_free(mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000163#ifdef FEAT_RIGHTLEFT
Bram Moolenaar071d4272004-06-13 20:20:40 +0000164 if (curwin->w_p_rl && *curwin->w_p_rlc == 's')
Bram Moolenaara2cff1d2021-10-15 12:51:29 +0100165 mr_pattern = reverse_text(pat);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000166 else
167#endif
Bram Moolenaara2cff1d2021-10-15 12:51:29 +0100168 mr_pattern = vim_strsave(pat);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000169
170 /*
171 * Save the currently used pattern in the appropriate place,
172 * unless the pattern should not be remembered.
173 */
Bram Moolenaare1004402020-10-24 20:49:43 +0200174 if (!(options & SEARCH_KEEP)
175 && (cmdmod.cmod_flags & CMOD_KEEPPATTERNS) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000176 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100177 // search or global command
Bram Moolenaar071d4272004-06-13 20:20:40 +0000178 if (pat_save == RE_SEARCH || pat_save == RE_BOTH)
179 save_re_pat(RE_SEARCH, pat, magic);
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100180 // substitute or global command
Bram Moolenaar071d4272004-06-13 20:20:40 +0000181 if (pat_save == RE_SUBST || pat_save == RE_BOTH)
182 save_re_pat(RE_SUBST, pat, magic);
183 }
184
185 regmatch->rmm_ic = ignorecase(pat);
Bram Moolenaar3b56eb32005-07-11 22:40:32 +0000186 regmatch->rmm_maxcol = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000187 regmatch->regprog = vim_regcomp(pat, magic ? RE_MAGIC : 0);
188 if (regmatch->regprog == NULL)
189 return FAIL;
190 return OK;
191}
192
193/*
194 * Get search pattern used by search_regcomp().
195 */
196 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100197get_search_pat(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000198{
199 return mr_pattern;
200}
201
Bram Moolenaarabc97732007-08-08 20:49:37 +0000202#if defined(FEAT_RIGHTLEFT) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000203/*
204 * Reverse text into allocated memory.
205 * Returns the allocated string, NULL when out of memory.
206 */
Bram Moolenaarabc97732007-08-08 20:49:37 +0000207 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100208reverse_text(char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000209{
210 unsigned len;
211 unsigned s_i, rev_i;
212 char_u *rev;
213
214 /*
215 * Reverse the pattern.
216 */
217 len = (unsigned)STRLEN(s);
218 rev = alloc(len + 1);
219 if (rev != NULL)
220 {
221 rev_i = len;
222 for (s_i = 0; s_i < len; ++s_i)
223 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000224 if (has_mbyte)
225 {
226 int mb_len;
227
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000228 mb_len = (*mb_ptr2len)(s + s_i);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000229 rev_i -= mb_len;
230 mch_memmove(rev + rev_i, s + s_i, mb_len);
231 s_i += mb_len - 1;
232 }
233 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000234 rev[--rev_i] = s[s_i];
235
236 }
237 rev[len] = NUL;
238 }
239 return rev;
240}
241#endif
242
Bram Moolenaarcc2b9d52014-12-13 03:17:11 +0100243 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100244save_re_pat(int idx, char_u *pat, int magic)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000245{
246 if (spats[idx].pat != pat)
247 {
248 vim_free(spats[idx].pat);
249 spats[idx].pat = vim_strsave(pat);
250 spats[idx].magic = magic;
251 spats[idx].no_scs = no_smartcase;
252 last_idx = idx;
253#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100254 // If 'hlsearch' set and search pat changed: need redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000255 if (p_hls)
Bram Moolenaara4d158b2022-08-14 14:17:45 +0100256 redraw_all_later(UPD_SOME_VALID);
Bram Moolenaar451fc7b2018-04-27 22:53:07 +0200257 set_no_hlsearch(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000258#endif
259 }
260}
261
Bram Moolenaar071d4272004-06-13 20:20:40 +0000262/*
263 * Save the search patterns, so they can be restored later.
264 * Used before/after executing autocommands and user functions.
265 */
266static int save_level = 0;
267
268 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100269save_search_patterns(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000270{
271 if (save_level++ == 0)
272 {
273 saved_spats[0] = spats[0];
274 if (spats[0].pat != NULL)
275 saved_spats[0].pat = vim_strsave(spats[0].pat);
276 saved_spats[1] = spats[1];
277 if (spats[1].pat != NULL)
278 saved_spats[1].pat = vim_strsave(spats[1].pat);
Bram Moolenaara2cff1d2021-10-15 12:51:29 +0100279 if (mr_pattern == NULL)
280 saved_mr_pattern = NULL;
281 else
282 saved_mr_pattern = vim_strsave(mr_pattern);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100283#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100284 saved_spats_last_idx = last_idx;
285 saved_spats_no_hlsearch = no_hlsearch;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100286#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000287 }
288}
289
290 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100291restore_search_patterns(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000292{
293 if (--save_level == 0)
294 {
295 vim_free(spats[0].pat);
296 spats[0] = saved_spats[0];
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100297#if defined(FEAT_EVAL)
Bram Moolenaar8c8de832008-06-24 22:58:06 +0000298 set_vv_searchforward();
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100299#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000300 vim_free(spats[1].pat);
301 spats[1] = saved_spats[1];
Bram Moolenaara2cff1d2021-10-15 12:51:29 +0100302 vim_free(mr_pattern);
303 mr_pattern = saved_mr_pattern;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100304#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100305 last_idx = saved_spats_last_idx;
306 set_no_hlsearch(saved_spats_no_hlsearch);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100307#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000308 }
309}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000310
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000311#if defined(EXITFREE) || defined(PROTO)
312 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100313free_search_patterns(void)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000314{
315 vim_free(spats[0].pat);
316 vim_free(spats[1].pat);
Bram Moolenaara2cff1d2021-10-15 12:51:29 +0100317 VIM_CLEAR(mr_pattern);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000318}
319#endif
320
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100321#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100322// copy of spats[RE_SEARCH], for keeping the search patterns while incremental
323// searching
Bram Moolenaarc3328162019-07-23 22:15:25 +0200324static spat_T saved_last_search_spat;
Bram Moolenaared8bc782018-12-01 21:08:21 +0100325static int did_save_last_search_spat = 0;
326static int saved_last_idx = 0;
327static int saved_no_hlsearch = 0;
Christian Brabandt6dd74242022-02-14 12:44:32 +0000328static int saved_search_match_endcol;
329static int saved_search_match_lines;
Bram Moolenaared8bc782018-12-01 21:08:21 +0100330
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100331/*
332 * Save and restore the search pattern for incremental highlight search
333 * feature.
334 *
Bram Moolenaarc4568ab2018-11-16 16:21:05 +0100335 * It's similar to but different from save_search_patterns() and
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100336 * restore_search_patterns(), because the search pattern must be restored when
Bram Moolenaarc4568ab2018-11-16 16:21:05 +0100337 * canceling incremental searching even if it's called inside user functions.
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100338 */
339 void
340save_last_search_pattern(void)
341{
Bram Moolenaar442a8532020-06-04 20:56:09 +0200342 if (++did_save_last_search_spat != 1)
343 // nested call, nothing to do
344 return;
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100345
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100346 saved_last_search_spat = spats[RE_SEARCH];
347 if (spats[RE_SEARCH].pat != NULL)
348 saved_last_search_spat.pat = vim_strsave(spats[RE_SEARCH].pat);
349 saved_last_idx = last_idx;
350 saved_no_hlsearch = no_hlsearch;
351}
352
353 void
354restore_last_search_pattern(void)
355{
Bram Moolenaar442a8532020-06-04 20:56:09 +0200356 if (--did_save_last_search_spat > 0)
357 // nested call, nothing to do
358 return;
359 if (did_save_last_search_spat != 0)
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100360 {
Bram Moolenaar442a8532020-06-04 20:56:09 +0200361 iemsg("restore_last_search_pattern() called more often than save_last_search_pattern()");
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100362 return;
363 }
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100364
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100365 vim_free(spats[RE_SEARCH].pat);
366 spats[RE_SEARCH] = saved_last_search_spat;
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100367 saved_last_search_spat.pat = NULL;
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100368# if defined(FEAT_EVAL)
369 set_vv_searchforward();
370# endif
371 last_idx = saved_last_idx;
Bram Moolenaar451fc7b2018-04-27 22:53:07 +0200372 set_no_hlsearch(saved_no_hlsearch);
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100373}
Bram Moolenaard0480092017-11-16 22:20:39 +0100374
Christian Brabandt6dd74242022-02-14 12:44:32 +0000375/*
376 * Save and restore the incsearch highlighting variables.
377 * This is required so that calling searchcount() at does not invalidate the
378 * incsearch highlighting.
379 */
380 static void
381save_incsearch_state(void)
382{
383 saved_search_match_endcol = search_match_endcol;
384 saved_search_match_lines = search_match_lines;
385}
386
387 static void
388restore_incsearch_state(void)
389{
390 search_match_endcol = saved_search_match_endcol;
391 search_match_lines = saved_search_match_lines;
392}
393
Bram Moolenaard0480092017-11-16 22:20:39 +0100394 char_u *
395last_search_pattern(void)
396{
397 return spats[RE_SEARCH].pat;
398}
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100399#endif
400
Bram Moolenaar071d4272004-06-13 20:20:40 +0000401/*
402 * Return TRUE when case should be ignored for search pattern "pat".
403 * Uses the 'ignorecase' and 'smartcase' options.
404 */
405 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100406ignorecase(char_u *pat)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000407{
Bram Moolenaar66e29d72016-08-20 16:57:02 +0200408 return ignorecase_opt(pat, p_ic, p_scs);
409}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000410
Bram Moolenaar66e29d72016-08-20 16:57:02 +0200411/*
412 * As ignorecase() put pass the "ic" and "scs" flags.
413 */
414 int
415ignorecase_opt(char_u *pat, int ic_in, int scs)
416{
417 int ic = ic_in;
418
419 if (ic && !no_smartcase && scs
Bram Moolenaare2c453d2019-08-21 14:37:09 +0200420 && !(ctrl_x_mode_not_default() && curbuf->b_p_inf))
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200421 ic = !pat_has_uppercase(pat);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000422 no_smartcase = FALSE;
423
424 return ic;
425}
426
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200427/*
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200428 * Return TRUE if pattern "pat" has an uppercase character.
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200429 */
430 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100431pat_has_uppercase(char_u *pat)
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200432{
433 char_u *p = pat;
Christian Brabandt78ba9332021-08-01 12:44:37 +0200434 magic_T magic_val = MAGIC_ON;
435
436 // get the magicness of the pattern
437 (void)skip_regexp_ex(pat, NUL, magic_isset(), NULL, NULL, &magic_val);
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200438
439 while (*p != NUL)
440 {
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200441 int l;
442
443 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
444 {
445 if (enc_utf8 && utf_isupper(utf_ptr2char(p)))
446 return TRUE;
447 p += l;
448 }
Christian Brabandtbc67e5a2021-08-05 15:24:59 +0200449 else if (*p == '\\' && magic_val <= MAGIC_ON)
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200450 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100451 if (p[1] == '_' && p[2] != NUL) // skip "\_X"
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200452 p += 3;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100453 else if (p[1] == '%' && p[2] != NUL) // skip "\%X"
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200454 p += 3;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100455 else if (p[1] != NUL) // skip "\X"
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200456 p += 2;
457 else
458 p += 1;
459 }
Christian Brabandt78ba9332021-08-01 12:44:37 +0200460 else if ((*p == '%' || *p == '_') && magic_val == MAGIC_ALL)
461 {
462 if (p[1] != NUL) // skip "_X" and %X
463 p += 2;
Christian Brabandtbc67e5a2021-08-05 15:24:59 +0200464 else
465 p++;
Christian Brabandt78ba9332021-08-01 12:44:37 +0200466 }
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200467 else if (MB_ISUPPER(*p))
468 return TRUE;
469 else
470 ++p;
471 }
472 return FALSE;
473}
474
Bram Moolenaar113e1072019-01-20 15:30:40 +0100475#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000476 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100477last_csearch(void)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200478{
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200479 return lastc_bytes;
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200480}
481
482 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100483last_csearch_forward(void)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200484{
485 return lastcdir == FORWARD;
486}
487
488 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100489last_csearch_until(void)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200490{
491 return last_t_cmd == TRUE;
492}
493
494 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100495set_last_csearch(int c, char_u *s UNUSED, int len UNUSED)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200496{
497 *lastc = c;
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200498 lastc_bytelen = len;
499 if (len)
500 memcpy(lastc_bytes, s, len);
501 else
Bram Moolenaara80faa82020-04-12 19:37:17 +0200502 CLEAR_FIELD(lastc_bytes);
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200503}
Bram Moolenaar113e1072019-01-20 15:30:40 +0100504#endif
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200505
506 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100507set_csearch_direction(int cdir)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200508{
509 lastcdir = cdir;
510}
511
512 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100513set_csearch_until(int t_cmd)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200514{
515 last_t_cmd = t_cmd;
516}
517
518 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100519last_search_pat(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000520{
521 return spats[last_idx].pat;
522}
523
524/*
525 * Reset search direction to forward. For "gd" and "gD" commands.
526 */
527 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100528reset_search_dir(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000529{
530 spats[0].off.dir = '/';
Bram Moolenaar8c8de832008-06-24 22:58:06 +0000531#if defined(FEAT_EVAL)
532 set_vv_searchforward();
533#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000534}
535
536#if defined(FEAT_EVAL) || defined(FEAT_VIMINFO)
537/*
538 * Set the last search pattern. For ":let @/ =" and viminfo.
539 * Also set the saved search pattern, so that this works in an autocommand.
540 */
541 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100542set_last_search_pat(
543 char_u *s,
544 int idx,
545 int magic,
546 int setlast)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000547{
548 vim_free(spats[idx].pat);
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100549 // An empty string means that nothing should be matched.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000550 if (*s == NUL)
551 spats[idx].pat = NULL;
552 else
553 spats[idx].pat = vim_strsave(s);
554 spats[idx].magic = magic;
555 spats[idx].no_scs = FALSE;
556 spats[idx].off.dir = '/';
Bram Moolenaar8c8de832008-06-24 22:58:06 +0000557#if defined(FEAT_EVAL)
558 set_vv_searchforward();
559#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000560 spats[idx].off.line = FALSE;
561 spats[idx].off.end = FALSE;
562 spats[idx].off.off = 0;
563 if (setlast)
564 last_idx = idx;
565 if (save_level)
566 {
567 vim_free(saved_spats[idx].pat);
568 saved_spats[idx] = spats[0];
569 if (spats[idx].pat == NULL)
570 saved_spats[idx].pat = NULL;
571 else
572 saved_spats[idx].pat = vim_strsave(spats[idx].pat);
Bram Moolenaar975880b2019-03-03 14:42:11 +0100573# ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100574 saved_spats_last_idx = last_idx;
Bram Moolenaar975880b2019-03-03 14:42:11 +0100575# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000576 }
577# ifdef FEAT_SEARCH_EXTRA
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100578 // If 'hlsearch' set and search pat changed: need redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000579 if (p_hls && idx == last_idx && !no_hlsearch)
Bram Moolenaara4d158b2022-08-14 14:17:45 +0100580 redraw_all_later(UPD_SOME_VALID);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000581# endif
582}
583#endif
584
585#ifdef FEAT_SEARCH_EXTRA
586/*
587 * Get a regexp program for the last used search pattern.
588 * This is used for highlighting all matches in a window.
589 * Values returned in regmatch->regprog and regmatch->rmm_ic.
590 */
591 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100592last_pat_prog(regmmatch_T *regmatch)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000593{
594 if (spats[last_idx].pat == NULL)
595 {
596 regmatch->regprog = NULL;
597 return;
598 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100599 ++emsg_off; // So it doesn't beep if bad expr
Bram Moolenaar071d4272004-06-13 20:20:40 +0000600 (void)search_regcomp((char_u *)"", 0, last_idx, SEARCH_KEEP, regmatch);
601 --emsg_off;
602}
603#endif
604
605/*
Bram Moolenaarf7ff6e82014-03-23 15:13:05 +0100606 * Lowest level search function.
Bram Moolenaar5d24a222018-12-23 19:10:09 +0100607 * Search for 'count'th occurrence of pattern "pat" in direction "dir".
608 * Start at position "pos" and return the found position in "pos".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000609 *
610 * if (options & SEARCH_MSG) == 0 don't give any messages
611 * if (options & SEARCH_MSG) == SEARCH_NFMSG don't give 'notfound' messages
612 * if (options & SEARCH_MSG) == SEARCH_MSG give all messages
613 * if (options & SEARCH_HIS) put search pattern in history
614 * if (options & SEARCH_END) return position at end of match
615 * if (options & SEARCH_START) accept match at pos itself
616 * if (options & SEARCH_KEEP) keep previous search pattern
617 * if (options & SEARCH_FOLD) match only once in a closed fold
618 * if (options & SEARCH_PEEK) check for typed char, cancel search
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100619 * if (options & SEARCH_COL) start at pos->col instead of zero
Bram Moolenaar071d4272004-06-13 20:20:40 +0000620 *
621 * Return FAIL (zero) for failure, non-zero for success.
622 * When FEAT_EVAL is defined, returns the index of the first matching
623 * subpattern plus one; one if there was none.
624 */
625 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100626searchit(
Bram Moolenaar92ea26b2019-10-18 20:53:34 +0200627 win_T *win, // window to search in; can be NULL for a
628 // buffer without a window!
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100629 buf_T *buf,
630 pos_T *pos,
Bram Moolenaar5d24a222018-12-23 19:10:09 +0100631 pos_T *end_pos, // set to end of the match, unless NULL
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100632 int dir,
633 char_u *pat,
634 long count,
635 int options,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +0200636 int pat_use, // which pattern to use when "pat" is empty
637 searchit_arg_T *extra_arg) // optional extra arguments, can be NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000638{
639 int found;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100640 linenr_T lnum; // no init to shut up Apollo cc
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100641 colnr_T col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000642 regmmatch_T regmatch;
643 char_u *ptr;
644 colnr_T matchcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000645 lpos_T endpos;
Bram Moolenaar677ee682005-01-27 14:41:15 +0000646 lpos_T matchpos;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000647 int loop;
648 pos_T start_pos;
649 int at_first_line;
650 int extra_col;
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200651 int start_char_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000652 int match_ok;
653 long nmatched;
654 int submatch = 0;
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100655 int first_match = TRUE;
Bram Moolenaar53989552019-12-23 22:59:18 +0100656 int called_emsg_before = called_emsg;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000657#ifdef FEAT_SEARCH_EXTRA
658 int break_loop = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000659#endif
Bram Moolenaar92ea26b2019-10-18 20:53:34 +0200660 linenr_T stop_lnum = 0; // stop after this line number when != 0
Paul Ollis65745772022-06-05 16:55:54 +0100661 int unused_timeout_flag = FALSE;
662 int *timed_out = &unused_timeout_flag; // set when timed out.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000663
664 if (search_regcomp(pat, RE_SEARCH, pat_use,
665 (options & (SEARCH_HIS + SEARCH_KEEP)), &regmatch) == FAIL)
666 {
667 if ((options & SEARCH_MSG) && !rc_did_emsg)
Bram Moolenaarac78dd42022-01-02 19:25:26 +0000668 semsg(_(e_invalid_search_string_str), mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000669 return FAIL;
670 }
671
Paul Ollis65745772022-06-05 16:55:54 +0100672 if (extra_arg != NULL)
673 {
674 stop_lnum = extra_arg->sa_stop_lnum;
675#ifdef FEAT_RELTIME
676 if (extra_arg->sa_tm > 0)
Paul Ollis65745772022-06-05 16:55:54 +0100677 init_regexp_timeout(extra_arg->sa_tm);
Bram Moolenaar5ea38d12022-06-16 21:20:48 +0100678 // Also set the pointer when sa_tm is zero, the caller may have set the
679 // timeout.
680 timed_out = &extra_arg->sa_timed_out;
Paul Ollis65745772022-06-05 16:55:54 +0100681#endif
682 }
683
Bram Moolenaar280f1262006-01-30 00:14:18 +0000684 /*
685 * find the string
686 */
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100687 do // loop for count
Bram Moolenaar071d4272004-06-13 20:20:40 +0000688 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100689 // When not accepting a match at the start position set "extra_col" to
690 // a non-zero value. Don't do that when starting at MAXCOL, since
691 // MAXCOL + 1 is zero.
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200692 if (pos->col == MAXCOL)
693 start_char_len = 0;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100694 // Watch out for the "col" being MAXCOL - 2, used in a closed fold.
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200695 else if (has_mbyte
696 && pos->lnum >= 1 && pos->lnum <= buf->b_ml.ml_line_count
697 && pos->col < MAXCOL - 2)
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100698 {
Bram Moolenaar82846a02018-02-09 18:09:54 +0100699 ptr = ml_get_buf(buf, pos->lnum, FALSE);
Bram Moolenaar8846ac52018-02-09 19:24:01 +0100700 if ((int)STRLEN(ptr) <= pos->col)
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200701 start_char_len = 1;
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100702 else
Bram Moolenaar82846a02018-02-09 18:09:54 +0100703 start_char_len = (*mb_ptr2len)(ptr + pos->col);
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100704 }
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100705 else
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200706 start_char_len = 1;
707 if (dir == FORWARD)
708 {
709 if (options & SEARCH_START)
710 extra_col = 0;
711 else
712 extra_col = start_char_len;
713 }
714 else
715 {
716 if (options & SEARCH_START)
717 extra_col = start_char_len;
718 else
719 extra_col = 0;
720 }
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100721
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100722 start_pos = *pos; // remember start pos for detecting no match
723 found = 0; // default: not found
724 at_first_line = TRUE; // default: start in first line
725 if (pos->lnum == 0) // correct lnum for when starting in line 0
Bram Moolenaar071d4272004-06-13 20:20:40 +0000726 {
727 pos->lnum = 1;
728 pos->col = 0;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100729 at_first_line = FALSE; // not in first line now
Bram Moolenaar071d4272004-06-13 20:20:40 +0000730 }
731
732 /*
733 * Start searching in current line, unless searching backwards and
734 * we're in column 0.
Bram Moolenaar7a42fa32007-07-10 11:28:55 +0000735 * If we are searching backwards, in column 0, and not including the
736 * current position, gain some efficiency by skipping back a line.
737 * Otherwise begin the search in the current line.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000738 */
Bram Moolenaar7a42fa32007-07-10 11:28:55 +0000739 if (dir == BACKWARD && start_pos.col == 0
740 && (options & SEARCH_START) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000741 {
742 lnum = pos->lnum - 1;
743 at_first_line = FALSE;
744 }
745 else
746 lnum = pos->lnum;
747
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100748 for (loop = 0; loop <= 1; ++loop) // loop twice if 'wrapscan' set
Bram Moolenaar071d4272004-06-13 20:20:40 +0000749 {
750 for ( ; lnum > 0 && lnum <= buf->b_ml.ml_line_count;
751 lnum += dir, at_first_line = FALSE)
752 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100753 // Stop after checking "stop_lnum", if it's set.
Bram Moolenaara23ccb82006-02-27 00:08:02 +0000754 if (stop_lnum != 0 && (dir == FORWARD
755 ? lnum > stop_lnum : lnum < stop_lnum))
756 break;
Paul Ollis65745772022-06-05 16:55:54 +0100757 // Stop after passing the time limit.
758 if (*timed_out)
Bram Moolenaar76929292008-01-06 19:07:36 +0000759 break;
Bram Moolenaara23ccb82006-02-27 00:08:02 +0000760
Bram Moolenaar071d4272004-06-13 20:20:40 +0000761 /*
Bram Moolenaar677ee682005-01-27 14:41:15 +0000762 * Look for a match somewhere in line "lnum".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000763 */
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100764 col = at_first_line && (options & SEARCH_COL) ? pos->col
765 : (colnr_T)0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000766 nmatched = vim_regexec_multi(&regmatch, win, buf,
Paul Ollis65745772022-06-05 16:55:54 +0100767 lnum, col, timed_out);
Bram Moolenaar795aaa12020-10-02 20:36:01 +0200768 // vim_regexec_multi() may clear "regprog"
769 if (regmatch.regprog == NULL)
770 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100771 // Abort searching on an error (e.g., out of stack).
Paul Ollis65745772022-06-05 16:55:54 +0100772 if (called_emsg > called_emsg_before || *timed_out)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000773 break;
774 if (nmatched > 0)
775 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100776 // match may actually be in another line when using \zs
Bram Moolenaar677ee682005-01-27 14:41:15 +0000777 matchpos = regmatch.startpos[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000778 endpos = regmatch.endpos[0];
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000779#ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000780 submatch = first_submatch(&regmatch);
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000781#endif
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100782 // "lnum" may be past end of buffer for "\n\zs".
Bram Moolenaar32466aa2006-02-24 23:53:04 +0000783 if (lnum + matchpos.lnum > buf->b_ml.ml_line_count)
784 ptr = (char_u *)"";
785 else
786 ptr = ml_get_buf(buf, lnum + matchpos.lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000787
788 /*
789 * Forward search in the first line: match should be after
790 * the start position. If not, continue at the end of the
791 * match (this is vi compatible) or on the next char.
792 */
793 if (dir == FORWARD && at_first_line)
794 {
795 match_ok = TRUE;
796 /*
Bram Moolenaar677ee682005-01-27 14:41:15 +0000797 * When the match starts in a next line it's certainly
798 * past the start position.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000799 * When match lands on a NUL the cursor will be put
800 * one back afterwards, compare with that position,
801 * otherwise "/$" will get stuck on end of line.
802 */
Bram Moolenaar677ee682005-01-27 14:41:15 +0000803 while (matchpos.lnum == 0
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100804 && ((options & SEARCH_END) && first_match
Bram Moolenaar677ee682005-01-27 14:41:15 +0000805 ? (nmatched == 1
806 && (int)endpos.col - 1
Bram Moolenaar071d4272004-06-13 20:20:40 +0000807 < (int)start_pos.col + extra_col)
Bram Moolenaar677ee682005-01-27 14:41:15 +0000808 : ((int)matchpos.col
809 - (ptr[matchpos.col] == NUL)
810 < (int)start_pos.col + extra_col)))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000811 {
812 /*
813 * If vi-compatible searching, continue at the end
814 * of the match, otherwise continue one position
815 * forward.
816 */
817 if (vim_strchr(p_cpo, CPO_SEARCH) != NULL)
818 {
819 if (nmatched > 1)
820 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100821 // end is in next line, thus no match in
822 // this line
Bram Moolenaar071d4272004-06-13 20:20:40 +0000823 match_ok = FALSE;
824 break;
825 }
826 matchcol = endpos.col;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100827 // for empty match: advance one char
Bram Moolenaar677ee682005-01-27 14:41:15 +0000828 if (matchcol == matchpos.col
Bram Moolenaar071d4272004-06-13 20:20:40 +0000829 && ptr[matchcol] != NUL)
830 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000831 if (has_mbyte)
832 matchcol +=
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000833 (*mb_ptr2len)(ptr + matchcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000834 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000835 ++matchcol;
836 }
837 }
838 else
839 {
Bram Moolenaar677ee682005-01-27 14:41:15 +0000840 matchcol = matchpos.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000841 if (ptr[matchcol] != NUL)
842 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000843 if (has_mbyte)
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000844 matchcol += (*mb_ptr2len)(ptr
Bram Moolenaar071d4272004-06-13 20:20:40 +0000845 + matchcol);
846 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000847 ++matchcol;
848 }
849 }
Bram Moolenaar7bcb30e2013-04-03 21:14:29 +0200850 if (matchcol == 0 && (options & SEARCH_START))
Bram Moolenaardb333a52013-03-19 15:27:48 +0100851 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000852 if (ptr[matchcol] == NUL
853 || (nmatched = vim_regexec_multi(&regmatch,
Bram Moolenaar677ee682005-01-27 14:41:15 +0000854 win, buf, lnum + matchpos.lnum,
Paul Ollis65745772022-06-05 16:55:54 +0100855 matchcol, timed_out)) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000856 {
857 match_ok = FALSE;
858 break;
859 }
Bram Moolenaar795aaa12020-10-02 20:36:01 +0200860 // vim_regexec_multi() may clear "regprog"
861 if (regmatch.regprog == NULL)
862 break;
Bram Moolenaar677ee682005-01-27 14:41:15 +0000863 matchpos = regmatch.startpos[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000864 endpos = regmatch.endpos[0];
865# ifdef FEAT_EVAL
866 submatch = first_submatch(&regmatch);
867# endif
868
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100869 // Need to get the line pointer again, a
870 // multi-line search may have made it invalid.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000871 ptr = ml_get_buf(buf, lnum + matchpos.lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000872 }
873 if (!match_ok)
874 continue;
875 }
876 if (dir == BACKWARD)
877 {
878 /*
879 * Now, if there are multiple matches on this line,
880 * we have to get the last one. Or the last one before
881 * the cursor, if we're on that line.
882 * When putting the new cursor at the end, compare
883 * relative to the end of the match.
884 */
885 match_ok = FALSE;
886 for (;;)
887 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100888 // Remember a position that is before the start
889 // position, we use it if it's the last match in
890 // the line. Always accept a position after
891 // wrapping around.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000892 if (loop
893 || ((options & SEARCH_END)
894 ? (lnum + regmatch.endpos[0].lnum
895 < start_pos.lnum
896 || (lnum + regmatch.endpos[0].lnum
897 == start_pos.lnum
898 && (int)regmatch.endpos[0].col - 1
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200899 < (int)start_pos.col
900 + extra_col))
Bram Moolenaar677ee682005-01-27 14:41:15 +0000901 : (lnum + regmatch.startpos[0].lnum
902 < start_pos.lnum
903 || (lnum + regmatch.startpos[0].lnum
904 == start_pos.lnum
905 && (int)regmatch.startpos[0].col
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200906 < (int)start_pos.col
907 + extra_col))))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000908 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000909 match_ok = TRUE;
Bram Moolenaar677ee682005-01-27 14:41:15 +0000910 matchpos = regmatch.startpos[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000911 endpos = regmatch.endpos[0];
912# ifdef FEAT_EVAL
913 submatch = first_submatch(&regmatch);
914# endif
915 }
916 else
917 break;
918
919 /*
920 * We found a valid match, now check if there is
921 * another one after it.
922 * If vi-compatible searching, continue at the end
923 * of the match, otherwise continue one position
924 * forward.
925 */
926 if (vim_strchr(p_cpo, CPO_SEARCH) != NULL)
927 {
928 if (nmatched > 1)
929 break;
930 matchcol = endpos.col;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100931 // for empty match: advance one char
Bram Moolenaar677ee682005-01-27 14:41:15 +0000932 if (matchcol == matchpos.col
Bram Moolenaar071d4272004-06-13 20:20:40 +0000933 && ptr[matchcol] != NUL)
934 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000935 if (has_mbyte)
936 matchcol +=
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000937 (*mb_ptr2len)(ptr + matchcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000938 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000939 ++matchcol;
940 }
941 }
942 else
943 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100944 // Stop when the match is in a next line.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000945 if (matchpos.lnum > 0)
946 break;
947 matchcol = matchpos.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000948 if (ptr[matchcol] != NUL)
949 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000950 if (has_mbyte)
951 matchcol +=
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000952 (*mb_ptr2len)(ptr + matchcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000953 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000954 ++matchcol;
955 }
956 }
957 if (ptr[matchcol] == NUL
958 || (nmatched = vim_regexec_multi(&regmatch,
Bram Moolenaar677ee682005-01-27 14:41:15 +0000959 win, buf, lnum + matchpos.lnum,
Paul Ollis65745772022-06-05 16:55:54 +0100960 matchcol, timed_out)) == 0)
Bram Moolenaar9d322762018-02-09 16:04:25 +0100961 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100962 // If the search timed out, we did find a match
963 // but it might be the wrong one, so that's not
964 // OK.
Paul Ollis65745772022-06-05 16:55:54 +0100965 if (*timed_out)
Bram Moolenaar9d322762018-02-09 16:04:25 +0100966 match_ok = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000967 break;
Bram Moolenaar9d322762018-02-09 16:04:25 +0100968 }
Bram Moolenaar795aaa12020-10-02 20:36:01 +0200969 // vim_regexec_multi() may clear "regprog"
970 if (regmatch.regprog == NULL)
971 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000972
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100973 // Need to get the line pointer again, a
974 // multi-line search may have made it invalid.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000975 ptr = ml_get_buf(buf, lnum + matchpos.lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000976 }
977
978 /*
979 * If there is only a match after the cursor, skip
980 * this match.
981 */
982 if (!match_ok)
983 continue;
984 }
985
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100986 // With the SEARCH_END option move to the last character
987 // of the match. Don't do it for an empty match, end
988 // should be same as start then.
Bram Moolenaar7bcb30e2013-04-03 21:14:29 +0200989 if ((options & SEARCH_END) && !(options & SEARCH_NOOF)
Bram Moolenaar5bcbd532008-02-20 12:43:01 +0000990 && !(matchpos.lnum == endpos.lnum
991 && matchpos.col == endpos.col))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000992 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100993 // For a match in the first column, set the position
994 // on the NUL in the previous line.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000995 pos->lnum = lnum + endpos.lnum;
Bram Moolenaar5bcbd532008-02-20 12:43:01 +0000996 pos->col = endpos.col;
997 if (endpos.col == 0)
Bram Moolenaar910f66f2006-04-05 20:41:53 +0000998 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100999 if (pos->lnum > 1) // just in case
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001000 {
1001 --pos->lnum;
1002 pos->col = (colnr_T)STRLEN(ml_get_buf(buf,
1003 pos->lnum, FALSE));
1004 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00001005 }
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001006 else
1007 {
1008 --pos->col;
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001009 if (has_mbyte
1010 && pos->lnum <= buf->b_ml.ml_line_count)
1011 {
1012 ptr = ml_get_buf(buf, pos->lnum, FALSE);
1013 pos->col -= (*mb_head_off)(ptr, ptr + pos->col);
1014 }
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001015 }
Bram Moolenaar5d24a222018-12-23 19:10:09 +01001016 if (end_pos != NULL)
1017 {
1018 end_pos->lnum = lnum + matchpos.lnum;
1019 end_pos->col = matchpos.col;
1020 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001021 }
1022 else
1023 {
Bram Moolenaar677ee682005-01-27 14:41:15 +00001024 pos->lnum = lnum + matchpos.lnum;
1025 pos->col = matchpos.col;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01001026 if (end_pos != NULL)
1027 {
1028 end_pos->lnum = lnum + endpos.lnum;
1029 end_pos->col = endpos.col;
1030 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001031 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001032 pos->coladd = 0;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01001033 if (end_pos != NULL)
1034 end_pos->coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001035 found = 1;
Bram Moolenaara3dfccc2014-11-27 17:29:56 +01001036 first_match = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001037
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001038 // Set variables used for 'incsearch' highlighting.
Bram Moolenaar677ee682005-01-27 14:41:15 +00001039 search_match_lines = endpos.lnum - matchpos.lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001040 search_match_endcol = endpos.col;
1041 break;
1042 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001043 line_breakcheck(); // stop if ctrl-C typed
Bram Moolenaar071d4272004-06-13 20:20:40 +00001044 if (got_int)
1045 break;
1046
1047#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001048 // Cancel searching if a character was typed. Used for
1049 // 'incsearch'. Don't check too often, that would slowdown
1050 // searching too much.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001051 if ((options & SEARCH_PEEK)
1052 && ((lnum - pos->lnum) & 0x3f) == 0
1053 && char_avail())
1054 {
1055 break_loop = TRUE;
1056 break;
1057 }
1058#endif
1059
1060 if (loop && lnum == start_pos.lnum)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001061 break; // if second loop, stop where started
Bram Moolenaar071d4272004-06-13 20:20:40 +00001062 }
1063 at_first_line = FALSE;
1064
Bram Moolenaar795aaa12020-10-02 20:36:01 +02001065 // vim_regexec_multi() may clear "regprog"
1066 if (regmatch.regprog == NULL)
1067 break;
1068
Bram Moolenaar071d4272004-06-13 20:20:40 +00001069 /*
Bram Moolenaara23ccb82006-02-27 00:08:02 +00001070 * Stop the search if wrapscan isn't set, "stop_lnum" is
1071 * specified, after an interrupt, after a match and after looping
1072 * twice.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001073 */
Bram Moolenaar53989552019-12-23 22:59:18 +01001074 if (!p_ws || stop_lnum != 0 || got_int
Paul Ollis65745772022-06-05 16:55:54 +01001075 || called_emsg > called_emsg_before || *timed_out
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +02001076#ifdef FEAT_SEARCH_EXTRA
1077 || break_loop
1078#endif
1079 || found || loop)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001080 break;
1081
1082 /*
1083 * If 'wrapscan' is set we continue at the other end of the file.
1084 * If 'shortmess' does not contain 's', we give a message.
1085 * This message is also remembered in keep_msg for when the screen
1086 * is redrawn. The keep_msg is cleared whenever another message is
1087 * written.
1088 */
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001089 if (dir == BACKWARD) // start second loop at the other end
Bram Moolenaar071d4272004-06-13 20:20:40 +00001090 lnum = buf->b_ml.ml_line_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001091 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001092 lnum = 1;
Bram Moolenaar92d640f2005-09-05 22:11:52 +00001093 if (!shortmess(SHM_SEARCH) && (options & SEARCH_MSG))
1094 give_warning((char_u *)_(dir == BACKWARD
1095 ? top_bot_msg : bot_top_msg), TRUE);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001096 if (extra_arg != NULL)
1097 extra_arg->sa_wrapped = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001098 }
Paul Ollis65745772022-06-05 16:55:54 +01001099 if (got_int || called_emsg > called_emsg_before || *timed_out
Bram Moolenaar78a15312009-05-15 19:33:18 +00001100#ifdef FEAT_SEARCH_EXTRA
1101 || break_loop
1102#endif
1103 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00001104 break;
1105 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001106 while (--count > 0 && found); // stop after count matches or no match
Bram Moolenaar071d4272004-06-13 20:20:40 +00001107
Bram Moolenaar5ea38d12022-06-16 21:20:48 +01001108#ifdef FEAT_RELTIME
1109 if (extra_arg != NULL && extra_arg->sa_tm > 0)
1110 disable_regexp_timeout();
1111#endif
Bram Moolenaar473de612013-06-08 18:19:48 +02001112 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001113
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001114 if (!found) // did not find it
Bram Moolenaar071d4272004-06-13 20:20:40 +00001115 {
1116 if (got_int)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001117 emsg(_(e_interrupted));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001118 else if ((options & SEARCH_MSG) == SEARCH_MSG)
1119 {
1120 if (p_ws)
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00001121 semsg(_(e_pattern_not_found_str), mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001122 else if (lnum == 0)
Bram Moolenaarac78dd42022-01-02 19:25:26 +00001123 semsg(_(e_search_hit_top_without_match_for_str), mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001124 else
Bram Moolenaarac78dd42022-01-02 19:25:26 +00001125 semsg(_(e_search_hit_bottom_without_match_for_str), mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001126 }
1127 return FAIL;
1128 }
1129
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001130 // A pattern like "\n\zs" may go past the last line.
Bram Moolenaar32466aa2006-02-24 23:53:04 +00001131 if (pos->lnum > buf->b_ml.ml_line_count)
1132 {
1133 pos->lnum = buf->b_ml.ml_line_count;
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001134 pos->col = (int)STRLEN(ml_get_buf(buf, pos->lnum, FALSE));
Bram Moolenaar32466aa2006-02-24 23:53:04 +00001135 if (pos->col > 0)
1136 --pos->col;
1137 }
1138
Bram Moolenaar071d4272004-06-13 20:20:40 +00001139 return submatch + 1;
1140}
1141
Yegappan Lakshmanan38b85cb2022-02-24 13:28:41 +00001142#if defined(FEAT_EVAL) || defined(FEAT_PROTO)
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001143 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001144set_search_direction(int cdir)
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001145{
1146 spats[0].off.dir = cdir;
1147}
1148
1149 static void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001150set_vv_searchforward(void)
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001151{
1152 set_vim_var_nr(VV_SEARCHFORWARD, (long)(spats[0].off.dir == '/'));
1153}
1154
Bram Moolenaar071d4272004-06-13 20:20:40 +00001155/*
1156 * Return the number of the first subpat that matched.
Bram Moolenaarad4d8a12015-12-28 19:20:36 +01001157 * Return zero if none of them matched.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001158 */
1159 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001160first_submatch(regmmatch_T *rp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001161{
1162 int submatch;
1163
1164 for (submatch = 1; ; ++submatch)
1165 {
1166 if (rp->startpos[submatch].lnum >= 0)
1167 break;
1168 if (submatch == 9)
1169 {
1170 submatch = 0;
1171 break;
1172 }
1173 }
1174 return submatch;
1175}
1176#endif
1177
1178/*
1179 * Highest level string search function.
Bram Moolenaarb8017e72007-05-10 18:59:07 +00001180 * Search for the 'count'th occurrence of pattern 'pat' in direction 'dirc'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001181 * If 'dirc' is 0: use previous dir.
1182 * If 'pat' is NULL or empty : use previous string.
1183 * If 'options & SEARCH_REV' : go in reverse of previous dir.
1184 * If 'options & SEARCH_ECHO': echo the search command and handle options
1185 * If 'options & SEARCH_MSG' : may give error message
1186 * If 'options & SEARCH_OPT' : interpret optional flags
1187 * If 'options & SEARCH_HIS' : put search pattern in history
1188 * If 'options & SEARCH_NOOF': don't add offset to position
1189 * If 'options & SEARCH_MARK': set previous context mark
1190 * If 'options & SEARCH_KEEP': keep previous search pattern
1191 * If 'options & SEARCH_START': accept match at curpos itself
1192 * If 'options & SEARCH_PEEK': check for typed char, cancel search
1193 *
1194 * Careful: If spats[0].off.line == TRUE and spats[0].off.off == 0 this
1195 * makes the movement linewise without moving the match position.
1196 *
Bram Moolenaarb6c27352015-03-05 19:57:49 +01001197 * Return 0 for failure, 1 for found, 2 for found and line offset added.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001198 */
1199 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001200do_search(
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001201 oparg_T *oap, // can be NULL
1202 int dirc, // '/' or '?'
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001203 int search_delim, // the delimiter for the search, e.g. '%' in
1204 // s%regex%replacement%
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001205 char_u *pat,
1206 long count,
1207 int options,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001208 searchit_arg_T *sia) // optional arguments or NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001209{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001210 pos_T pos; // position of the last match
Bram Moolenaar071d4272004-06-13 20:20:40 +00001211 char_u *searchstr;
Bram Moolenaarc3328162019-07-23 22:15:25 +02001212 soffset_T old_off;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001213 int retval; // Return value
Bram Moolenaar071d4272004-06-13 20:20:40 +00001214 char_u *p;
1215 long c;
1216 char_u *dircp;
1217 char_u *strcopy = NULL;
1218 char_u *ps;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001219 char_u *msgbuf = NULL;
1220 size_t len;
Bram Moolenaar8f46e4c2019-05-24 22:08:15 +02001221 int has_offset = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001222
1223 /*
1224 * A line offset is not remembered, this is vi compatible.
1225 */
1226 if (spats[0].off.line && vim_strchr(p_cpo, CPO_LINEOFF) != NULL)
1227 {
1228 spats[0].off.line = FALSE;
1229 spats[0].off.off = 0;
1230 }
1231
1232 /*
1233 * Save the values for when (options & SEARCH_KEEP) is used.
1234 * (there is no "if ()" around this because gcc wants them initialized)
1235 */
1236 old_off = spats[0].off;
1237
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001238 pos = curwin->w_cursor; // start searching at the cursor position
Bram Moolenaar071d4272004-06-13 20:20:40 +00001239
1240 /*
1241 * Find out the direction of the search.
1242 */
1243 if (dirc == 0)
1244 dirc = spats[0].off.dir;
1245 else
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001246 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001247 spats[0].off.dir = dirc;
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001248#if defined(FEAT_EVAL)
1249 set_vv_searchforward();
1250#endif
1251 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001252 if (options & SEARCH_REV)
1253 {
Bram Moolenaar4f974752019-02-17 17:44:42 +01001254#ifdef MSWIN
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001255 // There is a bug in the Visual C++ 2.2 compiler which means that
1256 // dirc always ends up being '/'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001257 dirc = (dirc == '/') ? '?' : '/';
1258#else
1259 if (dirc == '/')
1260 dirc = '?';
1261 else
1262 dirc = '/';
1263#endif
1264 }
1265
1266#ifdef FEAT_FOLDING
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001267 // If the cursor is in a closed fold, don't find another match in the same
1268 // fold.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001269 if (dirc == '/')
1270 {
1271 if (hasFolding(pos.lnum, NULL, &pos.lnum))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001272 pos.col = MAXCOL - 2; // avoid overflow when adding 1
Bram Moolenaar071d4272004-06-13 20:20:40 +00001273 }
1274 else
1275 {
1276 if (hasFolding(pos.lnum, &pos.lnum, NULL))
1277 pos.col = 0;
1278 }
1279#endif
1280
1281#ifdef FEAT_SEARCH_EXTRA
1282 /*
1283 * Turn 'hlsearch' highlighting back on.
1284 */
1285 if (no_hlsearch && !(options & SEARCH_KEEP))
1286 {
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001287 redraw_all_later(UPD_SOME_VALID);
Bram Moolenaar451fc7b2018-04-27 22:53:07 +02001288 set_no_hlsearch(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001289 }
1290#endif
1291
1292 /*
1293 * Repeat the search when pattern followed by ';', e.g. "/foo/;?bar".
1294 */
1295 for (;;)
1296 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001297 int show_top_bot_msg = FALSE;
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02001298
Bram Moolenaar071d4272004-06-13 20:20:40 +00001299 searchstr = pat;
1300 dircp = NULL;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001301 // use previous pattern
Bram Moolenaarc036e872020-02-21 21:30:52 +01001302 if (pat == NULL || *pat == NUL || *pat == search_delim)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001303 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001304 if (spats[RE_SEARCH].pat == NULL) // no previous pattern
Bram Moolenaar071d4272004-06-13 20:20:40 +00001305 {
Bram Moolenaarea683da2016-09-09 21:41:34 +02001306 searchstr = spats[RE_SUBST].pat;
1307 if (searchstr == NULL)
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001308 {
Bram Moolenaare29a27f2021-07-20 21:07:36 +02001309 emsg(_(e_no_previous_regular_expression));
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001310 retval = 0;
1311 goto end_do_search;
1312 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001313 }
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001314 else
1315 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001316 // make search_regcomp() use spats[RE_SEARCH].pat
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001317 searchstr = (char_u *)"";
1318 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001319 }
1320
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001321 if (pat != NULL && *pat != NUL) // look for (new) offset
Bram Moolenaar071d4272004-06-13 20:20:40 +00001322 {
1323 /*
1324 * Find end of regular expression.
1325 * If there is a matching '/' or '?', toss it.
1326 */
1327 ps = strcopy;
Bram Moolenaarf4e20992020-12-21 19:59:08 +01001328 p = skip_regexp_ex(pat, search_delim, magic_isset(),
Bram Moolenaard93a7fc2021-01-04 12:42:13 +01001329 &strcopy, NULL, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001330 if (strcopy != ps)
1331 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001332 // made a copy of "pat" to change "\?" to "?"
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001333 searchcmdlen += (int)(STRLEN(pat) - STRLEN(strcopy));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001334 pat = strcopy;
1335 searchstr = strcopy;
1336 }
Bram Moolenaarc036e872020-02-21 21:30:52 +01001337 if (*p == search_delim)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001338 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001339 dircp = p; // remember where we put the NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001340 *p++ = NUL;
1341 }
1342 spats[0].off.line = FALSE;
1343 spats[0].off.end = FALSE;
1344 spats[0].off.off = 0;
1345 /*
1346 * Check for a line offset or a character offset.
1347 * For get_address (echo off) we don't check for a character
1348 * offset, because it is meaningless and the 's' could be a
1349 * substitute command.
1350 */
1351 if (*p == '+' || *p == '-' || VIM_ISDIGIT(*p))
1352 spats[0].off.line = TRUE;
Dominique Pelle7765f5c2022-04-10 11:26:53 +01001353 else if ((options & SEARCH_OPT)
1354 && (*p == 'e' || *p == 's' || *p == 'b'))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001355 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001356 if (*p == 'e') // end
Bram Moolenaar071d4272004-06-13 20:20:40 +00001357 spats[0].off.end = SEARCH_END;
1358 ++p;
1359 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001360 if (VIM_ISDIGIT(*p) || *p == '+' || *p == '-') // got an offset
Bram Moolenaar071d4272004-06-13 20:20:40 +00001361 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001362 // 'nr' or '+nr' or '-nr'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001363 if (VIM_ISDIGIT(*p) || VIM_ISDIGIT(*(p + 1)))
1364 spats[0].off.off = atol((char *)p);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001365 else if (*p == '-') // single '-'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001366 spats[0].off.off = -1;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001367 else // single '+'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001368 spats[0].off.off = 1;
1369 ++p;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001370 while (VIM_ISDIGIT(*p)) // skip number
Bram Moolenaar071d4272004-06-13 20:20:40 +00001371 ++p;
1372 }
1373
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001374 // compute length of search command for get_address()
Bram Moolenaar071d4272004-06-13 20:20:40 +00001375 searchcmdlen += (int)(p - pat);
1376
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001377 pat = p; // put pat after search command
Bram Moolenaar071d4272004-06-13 20:20:40 +00001378 }
1379
Dominique Pelle7765f5c2022-04-10 11:26:53 +01001380 if ((options & SEARCH_ECHO) && messaging()
1381 && !msg_silent
1382 && (!cmd_silent || !shortmess(SHM_SEARCHCOUNT)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001383 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001384 char_u *trunc;
Bram Moolenaar984f0312019-05-24 13:11:47 +02001385 char_u off_buf[40];
Bram Moolenaard33a7642019-05-24 17:56:14 +02001386 size_t off_len = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001387
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001388 // Compute msg_row early.
1389 msg_start();
1390
Bram Moolenaar984f0312019-05-24 13:11:47 +02001391 // Get the offset, so we know how long it is.
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001392 if (!cmd_silent &&
1393 (spats[0].off.line || spats[0].off.end || spats[0].off.off))
Bram Moolenaar984f0312019-05-24 13:11:47 +02001394 {
1395 p = off_buf;
1396 *p++ = dirc;
1397 if (spats[0].off.end)
1398 *p++ = 'e';
1399 else if (!spats[0].off.line)
1400 *p++ = 's';
1401 if (spats[0].off.off > 0 || spats[0].off.line)
1402 *p++ = '+';
1403 *p = NUL;
1404 if (spats[0].off.off != 0 || spats[0].off.line)
1405 sprintf((char *)p, "%ld", spats[0].off.off);
1406 off_len = STRLEN(off_buf);
1407 }
1408
Bram Moolenaar071d4272004-06-13 20:20:40 +00001409 if (*searchstr == NUL)
Bram Moolenaar2fb8f682018-12-01 13:14:45 +01001410 p = spats[0].pat;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001411 else
1412 p = searchstr;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001413
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001414 if (!shortmess(SHM_SEARCHCOUNT) || cmd_silent)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001415 {
1416 // Reserve enough space for the search pattern + offset +
Bram Moolenaar984f0312019-05-24 13:11:47 +02001417 // search stat. Use all the space available, so that the
1418 // search state is right aligned. If there is not enough space
1419 // msg_strtrunc() will shorten in the middle.
Bram Moolenaar19e8ac72019-09-03 22:23:38 +02001420 if (msg_scrolled != 0 && !cmd_silent)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001421 // Use all the columns.
1422 len = (int)(Rows - msg_row) * Columns - 1;
1423 else
1424 // Use up to 'showcmd' column.
1425 len = (int)(Rows - msg_row - 1) * Columns + sc_col - 1;
Bram Moolenaar984f0312019-05-24 13:11:47 +02001426 if (len < STRLEN(p) + off_len + SEARCH_STAT_BUF_LEN + 3)
1427 len = STRLEN(p) + off_len + SEARCH_STAT_BUF_LEN + 3;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001428 }
1429 else
1430 // Reserve enough space for the search pattern + offset.
Bram Moolenaar984f0312019-05-24 13:11:47 +02001431 len = STRLEN(p) + off_len + 3;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001432
Bram Moolenaar880e4d92020-04-11 21:31:28 +02001433 vim_free(msgbuf);
Bram Moolenaar51e14382019-05-25 20:21:28 +02001434 msgbuf = alloc(len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001435 if (msgbuf != NULL)
1436 {
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001437 vim_memset(msgbuf, ' ', len);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001438 msgbuf[len - 1] = NUL;
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001439 // do not fill the msgbuf buffer, if cmd_silent is set, leave it
1440 // empty for the search_stat feature.
1441 if (!cmd_silent)
Bram Moolenaarcafda4f2005-09-06 19:25:11 +00001442 {
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001443 msgbuf[0] = dirc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001444
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001445 if (enc_utf8 && utf_iscomposing(utf_ptr2char(p)))
1446 {
1447 // Use a space to draw the composing char on.
1448 msgbuf[1] = ' ';
1449 mch_memmove(msgbuf + 2, p, STRLEN(p));
1450 }
1451 else
1452 mch_memmove(msgbuf + 1, p, STRLEN(p));
1453 if (off_len > 0)
1454 mch_memmove(msgbuf + STRLEN(p) + 1, off_buf, off_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001455
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001456 trunc = msg_strtrunc(msgbuf, TRUE);
1457 if (trunc != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001458 {
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001459 vim_free(msgbuf);
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001460 msgbuf = trunc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001461 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001462
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001463#ifdef FEAT_RIGHTLEFT
1464 // The search pattern could be shown on the right in
1465 // rightleft mode, but the 'ruler' and 'showcmd' area use
1466 // it too, thus it would be blanked out again very soon.
1467 // Show it on the left, but do reverse the text.
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001468 if (curwin->w_p_rl && *curwin->w_p_rlc == 's')
1469 {
1470 char_u *r;
1471 size_t pat_len;
1472
1473 r = reverse_text(msgbuf);
1474 if (r != NULL)
1475 {
1476 vim_free(msgbuf);
1477 msgbuf = r;
1478 // move reversed text to beginning of buffer
1479 while (*r != NUL && *r == ' ')
1480 r++;
1481 pat_len = msgbuf + STRLEN(msgbuf) - r;
1482 mch_memmove(msgbuf, r, pat_len);
1483 // overwrite old text
1484 if ((size_t)(r - msgbuf) >= pat_len)
1485 vim_memset(r, ' ', pat_len);
1486 else
1487 vim_memset(msgbuf + pat_len, ' ', r - msgbuf);
1488 }
1489 }
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001490#endif
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001491 msg_outtrans(msgbuf);
1492 msg_clr_eos();
1493 msg_check();
1494
1495 gotocmdline(FALSE);
1496 out_flush();
1497 msg_nowait = TRUE; // don't wait for this message
1498 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001499 }
1500 }
1501
1502 /*
1503 * If there is a character offset, subtract it from the current
1504 * position, so we don't get stuck at "?pat?e+2" or "/pat/s-2".
Bram Moolenaared203462004-06-16 11:19:22 +00001505 * Skip this if pos.col is near MAXCOL (closed fold).
Bram Moolenaar071d4272004-06-13 20:20:40 +00001506 * This is not done for a line offset, because then we would not be vi
1507 * compatible.
1508 */
Bram Moolenaared203462004-06-16 11:19:22 +00001509 if (!spats[0].off.line && spats[0].off.off && pos.col < MAXCOL - 2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001510 {
1511 if (spats[0].off.off > 0)
1512 {
1513 for (c = spats[0].off.off; c; --c)
1514 if (decl(&pos) == -1)
1515 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001516 if (c) // at start of buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001517 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001518 pos.lnum = 0; // allow lnum == 0 here
Bram Moolenaar071d4272004-06-13 20:20:40 +00001519 pos.col = MAXCOL;
1520 }
1521 }
1522 else
1523 {
1524 for (c = spats[0].off.off; c; ++c)
1525 if (incl(&pos) == -1)
1526 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001527 if (c) // at end of buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001528 {
1529 pos.lnum = curbuf->b_ml.ml_line_count + 1;
1530 pos.col = 0;
1531 }
1532 }
1533 }
1534
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001535 /*
1536 * The actual search.
1537 */
Bram Moolenaar14184a32019-02-16 15:10:30 +01001538 c = searchit(curwin, curbuf, &pos, NULL,
1539 dirc == '/' ? FORWARD : BACKWARD,
Bram Moolenaar071d4272004-06-13 20:20:40 +00001540 searchstr, count, spats[0].off.end + (options &
1541 (SEARCH_KEEP + SEARCH_PEEK + SEARCH_HIS
1542 + SEARCH_MSG + SEARCH_START
1543 + ((pat != NULL && *pat == ';') ? 0 : SEARCH_NOOF))),
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001544 RE_LAST, sia);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001545
1546 if (dircp != NULL)
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001547 *dircp = search_delim; // restore second '/' or '?' for normal_cmd()
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001548
1549 if (!shortmess(SHM_SEARCH)
1550 && ((dirc == '/' && LT_POS(pos, curwin->w_cursor))
1551 || (dirc == '?' && LT_POS(curwin->w_cursor, pos))))
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02001552 show_top_bot_msg = TRUE;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001553
Bram Moolenaar071d4272004-06-13 20:20:40 +00001554 if (c == FAIL)
1555 {
1556 retval = 0;
1557 goto end_do_search;
1558 }
1559 if (spats[0].off.end && oap != NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001560 oap->inclusive = TRUE; // 'e' includes last character
Bram Moolenaar071d4272004-06-13 20:20:40 +00001561
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001562 retval = 1; // pattern found
Bram Moolenaar071d4272004-06-13 20:20:40 +00001563
1564 /*
1565 * Add character and/or line offset
1566 */
Bram Moolenaar9160f302006-08-29 15:58:12 +00001567 if (!(options & SEARCH_NOOF) || (pat != NULL && *pat == ';'))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001568 {
Bram Moolenaar8f46e4c2019-05-24 22:08:15 +02001569 pos_T org_pos = pos;
1570
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001571 if (spats[0].off.line) // Add the offset to the line number.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001572 {
1573 c = pos.lnum + spats[0].off.off;
1574 if (c < 1)
1575 pos.lnum = 1;
1576 else if (c > curbuf->b_ml.ml_line_count)
1577 pos.lnum = curbuf->b_ml.ml_line_count;
1578 else
1579 pos.lnum = c;
1580 pos.col = 0;
1581
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001582 retval = 2; // pattern found, line offset added
Bram Moolenaar071d4272004-06-13 20:20:40 +00001583 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001584 else if (pos.col < MAXCOL - 2) // just in case
Bram Moolenaar071d4272004-06-13 20:20:40 +00001585 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001586 // to the right, check for end of file
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001587 c = spats[0].off.off;
1588 if (c > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001589 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001590 while (c-- > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001591 if (incl(&pos) == -1)
1592 break;
1593 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001594 // to the left, check for start of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00001595 else
1596 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001597 while (c++ < 0)
1598 if (decl(&pos) == -1)
1599 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001600 }
1601 }
Bram Moolenaar8f46e4c2019-05-24 22:08:15 +02001602 if (!EQUAL_POS(pos, org_pos))
1603 has_offset = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001604 }
1605
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001606 // Show [1/15] if 'S' is not in 'shortmess'.
1607 if ((options & SEARCH_ECHO)
1608 && messaging()
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001609 && !msg_silent
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001610 && c != FAIL
1611 && !shortmess(SHM_SEARCHCOUNT)
1612 && msgbuf != NULL)
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02001613 cmdline_search_stat(dirc, &pos, &curwin->w_cursor,
1614 show_top_bot_msg, msgbuf,
1615 (count != 1 || has_offset
Bram Moolenaar6cb07262020-05-29 22:49:43 +02001616#ifdef FEAT_FOLDING
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02001617 || (!(fdo_flags & FDO_SEARCH)
1618 && hasFolding(curwin->w_cursor.lnum,
1619 NULL, NULL))
Bram Moolenaar6cb07262020-05-29 22:49:43 +02001620#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02001621 ),
1622 SEARCH_STAT_DEF_MAX_COUNT,
1623 SEARCH_STAT_DEF_TIMEOUT);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001624
Bram Moolenaar071d4272004-06-13 20:20:40 +00001625 /*
1626 * The search command can be followed by a ';' to do another search.
1627 * For example: "/pat/;/foo/+3;?bar"
1628 * This is like doing another search command, except:
1629 * - The remembered direction '/' or '?' is from the first search.
1630 * - When an error happens the cursor isn't moved at all.
1631 * Don't do this when called by get_address() (it handles ';' itself).
1632 */
1633 if (!(options & SEARCH_OPT) || pat == NULL || *pat != ';')
1634 break;
1635
1636 dirc = *++pat;
Bram Moolenaarc036e872020-02-21 21:30:52 +01001637 search_delim = dirc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001638 if (dirc != '?' && dirc != '/')
1639 {
1640 retval = 0;
Bram Moolenaarac78dd42022-01-02 19:25:26 +00001641 emsg(_(e_expected_question_or_slash_after_semicolon));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001642 goto end_do_search;
1643 }
1644 ++pat;
1645 }
1646
1647 if (options & SEARCH_MARK)
1648 setpcmark();
1649 curwin->w_cursor = pos;
1650 curwin->w_set_curswant = TRUE;
1651
1652end_do_search:
Bram Moolenaare1004402020-10-24 20:49:43 +02001653 if ((options & SEARCH_KEEP) || (cmdmod.cmod_flags & CMOD_KEEPPATTERNS))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001654 spats[0].off = old_off;
1655 vim_free(strcopy);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001656 vim_free(msgbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001657
1658 return retval;
1659}
1660
Bram Moolenaar071d4272004-06-13 20:20:40 +00001661/*
1662 * search_for_exact_line(buf, pos, dir, pat)
1663 *
1664 * Search for a line starting with the given pattern (ignoring leading
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02001665 * white-space), starting from pos and going in direction "dir". "pos" will
Bram Moolenaar071d4272004-06-13 20:20:40 +00001666 * contain the position of the match found. Blank lines match only if
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02001667 * ADDING is set. If p_ic is set then the pattern must be in lowercase.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001668 * Return OK for success, or FAIL if no line found.
1669 */
1670 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001671search_for_exact_line(
1672 buf_T *buf,
1673 pos_T *pos,
1674 int dir,
1675 char_u *pat)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001676{
1677 linenr_T start = 0;
1678 char_u *ptr;
1679 char_u *p;
1680
1681 if (buf->b_ml.ml_line_count == 0)
1682 return FAIL;
1683 for (;;)
1684 {
1685 pos->lnum += dir;
1686 if (pos->lnum < 1)
1687 {
1688 if (p_ws)
1689 {
1690 pos->lnum = buf->b_ml.ml_line_count;
1691 if (!shortmess(SHM_SEARCH))
1692 give_warning((char_u *)_(top_bot_msg), TRUE);
1693 }
1694 else
1695 {
1696 pos->lnum = 1;
1697 break;
1698 }
1699 }
1700 else if (pos->lnum > buf->b_ml.ml_line_count)
1701 {
1702 if (p_ws)
1703 {
1704 pos->lnum = 1;
1705 if (!shortmess(SHM_SEARCH))
1706 give_warning((char_u *)_(bot_top_msg), TRUE);
1707 }
1708 else
1709 {
1710 pos->lnum = 1;
1711 break;
1712 }
1713 }
1714 if (pos->lnum == start)
1715 break;
1716 if (start == 0)
1717 start = pos->lnum;
1718 ptr = ml_get_buf(buf, pos->lnum, FALSE);
1719 p = skipwhite(ptr);
1720 pos->col = (colnr_T) (p - ptr);
1721
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001722 // when adding lines the matching line may be empty but it is not
1723 // ignored because we are interested in the next line -- Acevedo
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00001724 if (compl_status_adding() && !compl_status_sol())
Bram Moolenaar071d4272004-06-13 20:20:40 +00001725 {
1726 if ((p_ic ? MB_STRICMP(p, pat) : STRCMP(p, pat)) == 0)
1727 return OK;
1728 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001729 else if (*p != NUL) // ignore empty lines
1730 { // expanding lines or words
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00001731 if ((p_ic ? MB_STRNICMP(p, pat, ins_compl_len())
1732 : STRNCMP(p, pat, ins_compl_len())) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001733 return OK;
1734 }
1735 }
1736 return FAIL;
1737}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001738
1739/*
1740 * Character Searches
1741 */
1742
1743/*
1744 * Search for a character in a line. If "t_cmd" is FALSE, move to the
1745 * position of the character, otherwise move to just before the char.
1746 * Do this "cap->count1" times.
1747 * Return FAIL or OK.
1748 */
1749 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001750searchc(cmdarg_T *cap, int t_cmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001751{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001752 int c = cap->nchar; // char to search for
1753 int dir = cap->arg; // TRUE for searching forward
1754 long count = cap->count1; // repeat count
Bram Moolenaar071d4272004-06-13 20:20:40 +00001755 int col;
1756 char_u *p;
1757 int len;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001758 int stop = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001759
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001760 if (c != NUL) // normal search: remember args for repeat
Bram Moolenaar071d4272004-06-13 20:20:40 +00001761 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001762 if (!KeyStuffed) // don't remember when redoing
Bram Moolenaar071d4272004-06-13 20:20:40 +00001763 {
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001764 *lastc = c;
1765 set_csearch_direction(dir);
1766 set_csearch_until(t_cmd);
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001767 lastc_bytelen = (*mb_char2bytes)(c, lastc_bytes);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001768 if (cap->ncharC1 != 0)
1769 {
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001770 lastc_bytelen += (*mb_char2bytes)(cap->ncharC1,
1771 lastc_bytes + lastc_bytelen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001772 if (cap->ncharC2 != 0)
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001773 lastc_bytelen += (*mb_char2bytes)(cap->ncharC2,
1774 lastc_bytes + lastc_bytelen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001775 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001776 }
1777 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001778 else // repeat previous search
Bram Moolenaar071d4272004-06-13 20:20:40 +00001779 {
Bram Moolenaar264b74f2019-01-24 17:18:42 +01001780 if (*lastc == NUL && lastc_bytelen == 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001781 return FAIL;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001782 if (dir) // repeat in opposite direction
Bram Moolenaar071d4272004-06-13 20:20:40 +00001783 dir = -lastcdir;
1784 else
1785 dir = lastcdir;
1786 t_cmd = last_t_cmd;
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001787 c = *lastc;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001788 // For multi-byte re-use last lastc_bytes[] and lastc_bytelen.
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001789
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001790 // Force a move of at least one char, so ";" and "," will move the
1791 // cursor, even if the cursor is right in front of char we are looking
1792 // at.
Bram Moolenaar19fd09a2011-07-15 13:21:30 +02001793 if (vim_strchr(p_cpo, CPO_SCOLON) == NULL && count == 1 && t_cmd)
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001794 stop = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001795 }
1796
Bram Moolenaar60a795a2005-09-16 21:55:43 +00001797 if (dir == BACKWARD)
1798 cap->oap->inclusive = FALSE;
1799 else
1800 cap->oap->inclusive = TRUE;
1801
Bram Moolenaar071d4272004-06-13 20:20:40 +00001802 p = ml_get_curline();
1803 col = curwin->w_cursor.col;
1804 len = (int)STRLEN(p);
1805
1806 while (count--)
1807 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001808 if (has_mbyte)
1809 {
1810 for (;;)
1811 {
1812 if (dir > 0)
1813 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001814 col += (*mb_ptr2len)(p + col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001815 if (col >= len)
1816 return FAIL;
1817 }
1818 else
1819 {
1820 if (col == 0)
1821 return FAIL;
1822 col -= (*mb_head_off)(p, p + col - 1) + 1;
1823 }
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001824 if (lastc_bytelen == 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001825 {
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001826 if (p[col] == c && stop)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001827 break;
1828 }
Bram Moolenaar66727e12017-03-01 22:17:05 +01001829 else if (STRNCMP(p + col, lastc_bytes, lastc_bytelen) == 0
Bram Moolenaarb129a442016-12-01 17:25:20 +01001830 && stop)
Bram Moolenaar66727e12017-03-01 22:17:05 +01001831 break;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001832 stop = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001833 }
1834 }
1835 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001836 {
1837 for (;;)
1838 {
1839 if ((col += dir) < 0 || col >= len)
1840 return FAIL;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001841 if (p[col] == c && stop)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001842 break;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001843 stop = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001844 }
1845 }
1846 }
1847
1848 if (t_cmd)
1849 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001850 // backup to before the character (possibly double-byte)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001851 col -= dir;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001852 if (has_mbyte)
1853 {
1854 if (dir < 0)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001855 // Landed on the search char which is lastc_bytelen long
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001856 col += lastc_bytelen - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001857 else
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001858 // To previous char, which may be multi-byte.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001859 col -= (*mb_head_off)(p, p + col);
1860 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001861 }
1862 curwin->w_cursor.col = col;
1863
1864 return OK;
1865}
1866
1867/*
1868 * "Other" Searches
1869 */
1870
1871/*
1872 * findmatch - find the matching paren or brace
1873 *
1874 * Improvement over vi: Braces inside quotes are ignored.
1875 */
1876 pos_T *
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001877findmatch(oparg_T *oap, int initc)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001878{
1879 return findmatchlimit(oap, initc, 0, 0);
1880}
1881
1882/*
1883 * Return TRUE if the character before "linep[col]" equals "ch".
1884 * Return FALSE if "col" is zero.
1885 * Update "*prevcol" to the column of the previous character, unless "prevcol"
1886 * is NULL.
1887 * Handles multibyte string correctly.
1888 */
1889 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001890check_prevcol(
1891 char_u *linep,
1892 int col,
1893 int ch,
1894 int *prevcol)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001895{
1896 --col;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001897 if (col > 0 && has_mbyte)
1898 col -= (*mb_head_off)(linep, linep + col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001899 if (prevcol)
1900 *prevcol = col;
1901 return (col >= 0 && linep[col] == ch) ? TRUE : FALSE;
1902}
1903
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001904/*
1905 * Raw string start is found at linep[startpos.col - 1].
1906 * Return TRUE if the matching end can be found between startpos and endpos.
1907 */
1908 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001909find_rawstring_end(char_u *linep, pos_T *startpos, pos_T *endpos)
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001910{
1911 char_u *p;
1912 char_u *delim_copy;
1913 size_t delim_len;
1914 linenr_T lnum;
1915 int found = FALSE;
1916
1917 for (p = linep + startpos->col + 1; *p && *p != '('; ++p)
1918 ;
1919 delim_len = (p - linep) - startpos->col - 1;
Bram Moolenaar71ccd032020-06-12 22:59:11 +02001920 delim_copy = vim_strnsave(linep + startpos->col + 1, delim_len);
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001921 if (delim_copy == NULL)
1922 return FALSE;
1923 for (lnum = startpos->lnum; lnum <= endpos->lnum; ++lnum)
1924 {
1925 char_u *line = ml_get(lnum);
1926
1927 for (p = line + (lnum == startpos->lnum
1928 ? startpos->col + 1 : 0); *p; ++p)
1929 {
1930 if (lnum == endpos->lnum && (colnr_T)(p - line) >= endpos->col)
1931 break;
Bram Moolenaar282f9c62020-08-04 21:46:18 +02001932 if (*p == ')' && STRNCMP(delim_copy, p + 1, delim_len) == 0
1933 && p[delim_len + 1] == '"')
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001934 {
1935 found = TRUE;
1936 break;
1937 }
1938 }
1939 if (found)
1940 break;
1941 }
1942 vim_free(delim_copy);
1943 return found;
1944}
1945
Bram Moolenaar071d4272004-06-13 20:20:40 +00001946/*
Bram Moolenaar556ae8e2019-11-21 22:27:22 +01001947 * Check matchpairs option for "*initc".
1948 * If there is a match set "*initc" to the matching character and "*findc" to
1949 * the opposite character. Set "*backwards" to the direction.
1950 * When "switchit" is TRUE swap the direction.
1951 */
1952 static void
1953find_mps_values(
1954 int *initc,
1955 int *findc,
1956 int *backwards,
1957 int switchit)
1958{
1959 char_u *ptr;
1960
1961 ptr = curbuf->b_p_mps;
1962 while (*ptr != NUL)
1963 {
1964 if (has_mbyte)
1965 {
1966 char_u *prev;
1967
1968 if (mb_ptr2char(ptr) == *initc)
1969 {
1970 if (switchit)
1971 {
1972 *findc = *initc;
1973 *initc = mb_ptr2char(ptr + mb_ptr2len(ptr) + 1);
1974 *backwards = TRUE;
1975 }
1976 else
1977 {
1978 *findc = mb_ptr2char(ptr + mb_ptr2len(ptr) + 1);
1979 *backwards = FALSE;
1980 }
1981 return;
1982 }
1983 prev = ptr;
1984 ptr += mb_ptr2len(ptr) + 1;
1985 if (mb_ptr2char(ptr) == *initc)
1986 {
1987 if (switchit)
1988 {
1989 *findc = *initc;
1990 *initc = mb_ptr2char(prev);
1991 *backwards = FALSE;
1992 }
1993 else
1994 {
1995 *findc = mb_ptr2char(prev);
1996 *backwards = TRUE;
1997 }
1998 return;
1999 }
2000 ptr += mb_ptr2len(ptr);
2001 }
2002 else
2003 {
2004 if (*ptr == *initc)
2005 {
2006 if (switchit)
2007 {
2008 *backwards = TRUE;
2009 *findc = *initc;
2010 *initc = ptr[2];
2011 }
2012 else
2013 {
2014 *backwards = FALSE;
2015 *findc = ptr[2];
2016 }
2017 return;
2018 }
2019 ptr += 2;
2020 if (*ptr == *initc)
2021 {
2022 if (switchit)
2023 {
2024 *backwards = FALSE;
2025 *findc = *initc;
2026 *initc = ptr[-2];
2027 }
2028 else
2029 {
2030 *backwards = TRUE;
2031 *findc = ptr[-2];
2032 }
2033 return;
2034 }
2035 ++ptr;
2036 }
2037 if (*ptr == ',')
2038 ++ptr;
2039 }
2040}
2041
2042/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002043 * findmatchlimit -- find the matching paren or brace, if it exists within
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002044 * maxtravel lines of the cursor. A maxtravel of 0 means search until falling
2045 * off the edge of the file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002046 *
2047 * "initc" is the character to find a match for. NUL means to find the
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002048 * character at or after the cursor. Special values:
2049 * '*' look for C-style comment / *
2050 * '/' look for C-style comment / *, ignoring comment-end
2051 * '#' look for preprocessor directives
2052 * 'R' look for raw string start: R"delim(text)delim" (only backwards)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002053 *
2054 * flags: FM_BACKWARD search backwards (when initc is '/', '*' or '#')
2055 * FM_FORWARD search forwards (when initc is '/', '*' or '#')
2056 * FM_BLOCKSTOP stop at start/end of block ({ or } in column 0)
2057 * FM_SKIPCOMM skip comments (not implemented yet!)
Bram Moolenaarf75a9632005-09-13 21:20:47 +00002058 *
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002059 * "oap" is only used to set oap->motion_type for a linewise motion, it can be
Bram Moolenaarf75a9632005-09-13 21:20:47 +00002060 * NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00002061 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002062 pos_T *
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002063findmatchlimit(
2064 oparg_T *oap,
2065 int initc,
2066 int flags,
2067 int maxtravel)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002068{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002069 static pos_T pos; // current search position
2070 int findc = 0; // matching brace
Bram Moolenaar071d4272004-06-13 20:20:40 +00002071 int c;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002072 int count = 0; // cumulative number of braces
2073 int backwards = FALSE; // init for gcc
2074 int raw_string = FALSE; // search for raw string
2075 int inquote = FALSE; // TRUE when inside quotes
2076 char_u *linep; // pointer to current line
Bram Moolenaar071d4272004-06-13 20:20:40 +00002077 char_u *ptr;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002078 int do_quotes; // check for quotes in current line
2079 int at_start; // do_quotes value at start position
2080 int hash_dir = 0; // Direction searched for # things
2081 int comment_dir = 0; // Direction searched for comments
2082 pos_T match_pos; // Where last slash-star was found
2083 int start_in_quotes; // start position is in quotes
2084 int traveled = 0; // how far we've searched so far
2085 int ignore_cend = FALSE; // ignore comment end
2086 int cpo_match; // vi compatible matching
2087 int cpo_bsl; // don't recognize backslashes
2088 int match_escaped = 0; // search for escaped match
2089 int dir; // Direction to search
2090 int comment_col = MAXCOL; // start of / / comment
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002091 int lispcomm = FALSE; // inside of Lisp-style comment
2092 int lisp = curbuf->b_p_lisp; // engage Lisp-specific hacks ;)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002093
2094 pos = curwin->w_cursor;
Bram Moolenaarc56c4592013-08-14 17:45:29 +02002095 pos.coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002096 linep = ml_get(pos.lnum);
2097
2098 cpo_match = (vim_strchr(p_cpo, CPO_MATCH) != NULL);
2099 cpo_bsl = (vim_strchr(p_cpo, CPO_MATCHBSL) != NULL);
2100
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002101 // Direction to search when initc is '/', '*' or '#'
Bram Moolenaar071d4272004-06-13 20:20:40 +00002102 if (flags & FM_BACKWARD)
2103 dir = BACKWARD;
2104 else if (flags & FM_FORWARD)
2105 dir = FORWARD;
2106 else
2107 dir = 0;
2108
2109 /*
2110 * if initc given, look in the table for the matching character
2111 * '/' and '*' are special cases: look for start or end of comment.
2112 * When '/' is used, we ignore running backwards into an star-slash, for
2113 * "[*" command, we just want to find any comment.
2114 */
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002115 if (initc == '/' || initc == '*' || initc == 'R')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002116 {
2117 comment_dir = dir;
2118 if (initc == '/')
2119 ignore_cend = TRUE;
2120 backwards = (dir == FORWARD) ? FALSE : TRUE;
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002121 raw_string = (initc == 'R');
Bram Moolenaar071d4272004-06-13 20:20:40 +00002122 initc = NUL;
2123 }
2124 else if (initc != '#' && initc != NUL)
2125 {
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002126 find_mps_values(&initc, &findc, &backwards, TRUE);
Connor Lane Smithb9115da2021-07-31 13:31:42 +02002127 if (dir)
2128 backwards = (dir == FORWARD) ? FALSE : TRUE;
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002129 if (findc == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002130 return NULL;
2131 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002132 else
2133 {
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002134 /*
2135 * Either initc is '#', or no initc was given and we need to look
2136 * under the cursor.
2137 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002138 if (initc == '#')
2139 {
2140 hash_dir = dir;
2141 }
2142 else
2143 {
2144 /*
2145 * initc was not given, must look for something to match under
2146 * or near the cursor.
2147 * Only check for special things when 'cpo' doesn't have '%'.
2148 */
2149 if (!cpo_match)
2150 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002151 // Are we before or at #if, #else etc.?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002152 ptr = skipwhite(linep);
2153 if (*ptr == '#' && pos.col <= (colnr_T)(ptr - linep))
2154 {
2155 ptr = skipwhite(ptr + 1);
2156 if ( STRNCMP(ptr, "if", 2) == 0
2157 || STRNCMP(ptr, "endif", 5) == 0
2158 || STRNCMP(ptr, "el", 2) == 0)
2159 hash_dir = 1;
2160 }
2161
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002162 // Are we on a comment?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002163 else if (linep[pos.col] == '/')
2164 {
2165 if (linep[pos.col + 1] == '*')
2166 {
2167 comment_dir = FORWARD;
2168 backwards = FALSE;
2169 pos.col++;
2170 }
2171 else if (pos.col > 0 && linep[pos.col - 1] == '*')
2172 {
2173 comment_dir = BACKWARD;
2174 backwards = TRUE;
2175 pos.col--;
2176 }
2177 }
2178 else if (linep[pos.col] == '*')
2179 {
2180 if (linep[pos.col + 1] == '/')
2181 {
2182 comment_dir = BACKWARD;
2183 backwards = TRUE;
2184 }
2185 else if (pos.col > 0 && linep[pos.col - 1] == '/')
2186 {
2187 comment_dir = FORWARD;
2188 backwards = FALSE;
2189 }
2190 }
2191 }
2192
2193 /*
2194 * If we are not on a comment or the # at the start of a line, then
2195 * look for brace anywhere on this line after the cursor.
2196 */
2197 if (!hash_dir && !comment_dir)
2198 {
2199 /*
2200 * Find the brace under or after the cursor.
2201 * If beyond the end of the line, use the last character in
2202 * the line.
2203 */
2204 if (linep[pos.col] == NUL && pos.col)
2205 --pos.col;
2206 for (;;)
2207 {
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002208 initc = PTR2CHAR(linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002209 if (initc == NUL)
2210 break;
2211
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002212 find_mps_values(&initc, &findc, &backwards, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002213 if (findc)
2214 break;
Bram Moolenaar1614a142019-10-06 22:00:13 +02002215 pos.col += mb_ptr2len(linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002216 }
2217 if (!findc)
2218 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002219 // no brace in the line, maybe use " #if" then
Bram Moolenaar071d4272004-06-13 20:20:40 +00002220 if (!cpo_match && *skipwhite(linep) == '#')
2221 hash_dir = 1;
2222 else
2223 return NULL;
2224 }
2225 else if (!cpo_bsl)
2226 {
2227 int col, bslcnt = 0;
2228
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002229 // Set "match_escaped" if there are an odd number of
2230 // backslashes.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002231 for (col = pos.col; check_prevcol(linep, col, '\\', &col);)
2232 bslcnt++;
2233 match_escaped = (bslcnt & 1);
2234 }
2235 }
2236 }
2237 if (hash_dir)
2238 {
2239 /*
2240 * Look for matching #if, #else, #elif, or #endif
2241 */
2242 if (oap != NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002243 oap->motion_type = MLINE; // Linewise for this case only
Bram Moolenaar071d4272004-06-13 20:20:40 +00002244 if (initc != '#')
2245 {
2246 ptr = skipwhite(skipwhite(linep) + 1);
2247 if (STRNCMP(ptr, "if", 2) == 0 || STRNCMP(ptr, "el", 2) == 0)
2248 hash_dir = 1;
2249 else if (STRNCMP(ptr, "endif", 5) == 0)
2250 hash_dir = -1;
2251 else
2252 return NULL;
2253 }
2254 pos.col = 0;
2255 while (!got_int)
2256 {
2257 if (hash_dir > 0)
2258 {
2259 if (pos.lnum == curbuf->b_ml.ml_line_count)
2260 break;
2261 }
2262 else if (pos.lnum == 1)
2263 break;
2264 pos.lnum += hash_dir;
2265 linep = ml_get(pos.lnum);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002266 line_breakcheck(); // check for CTRL-C typed
Bram Moolenaar071d4272004-06-13 20:20:40 +00002267 ptr = skipwhite(linep);
2268 if (*ptr != '#')
2269 continue;
2270 pos.col = (colnr_T) (ptr - linep);
2271 ptr = skipwhite(ptr + 1);
2272 if (hash_dir > 0)
2273 {
2274 if (STRNCMP(ptr, "if", 2) == 0)
2275 count++;
2276 else if (STRNCMP(ptr, "el", 2) == 0)
2277 {
2278 if (count == 0)
2279 return &pos;
2280 }
2281 else if (STRNCMP(ptr, "endif", 5) == 0)
2282 {
2283 if (count == 0)
2284 return &pos;
2285 count--;
2286 }
2287 }
2288 else
2289 {
2290 if (STRNCMP(ptr, "if", 2) == 0)
2291 {
2292 if (count == 0)
2293 return &pos;
2294 count--;
2295 }
2296 else if (initc == '#' && STRNCMP(ptr, "el", 2) == 0)
2297 {
2298 if (count == 0)
2299 return &pos;
2300 }
2301 else if (STRNCMP(ptr, "endif", 5) == 0)
2302 count++;
2303 }
2304 }
2305 return NULL;
2306 }
2307 }
2308
2309#ifdef FEAT_RIGHTLEFT
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002310 // This is just guessing: when 'rightleft' is set, search for a matching
2311 // paren/brace in the other direction.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002312 if (curwin->w_p_rl && vim_strchr((char_u *)"()[]{}<>", initc) != NULL)
2313 backwards = !backwards;
2314#endif
2315
2316 do_quotes = -1;
2317 start_in_quotes = MAYBE;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01002318 CLEAR_POS(&match_pos);
Bram Moolenaarfd2ac762006-03-01 22:09:21 +00002319
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002320 // backward search: Check if this line contains a single-line comment
Bram Moolenaar8e145b82022-05-21 20:17:31 +01002321 if ((backwards && comment_dir) || lisp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002322 comment_col = check_linecomment(linep);
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002323 if (lisp && comment_col != MAXCOL && pos.col > (colnr_T)comment_col)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002324 lispcomm = TRUE; // find match inside this comment
Bram Moolenaar8e145b82022-05-21 20:17:31 +01002325
Bram Moolenaar071d4272004-06-13 20:20:40 +00002326 while (!got_int)
2327 {
2328 /*
2329 * Go to the next position, forward or backward. We could use
2330 * inc() and dec() here, but that is much slower
2331 */
2332 if (backwards)
2333 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002334 // char to match is inside of comment, don't search outside
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002335 if (lispcomm && pos.col < (colnr_T)comment_col)
2336 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002337 if (pos.col == 0) // at start of line, go to prev. one
Bram Moolenaar071d4272004-06-13 20:20:40 +00002338 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002339 if (pos.lnum == 1) // start of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00002340 break;
2341 --pos.lnum;
2342
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002343 if (maxtravel > 0 && ++traveled > maxtravel)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002344 break;
2345
2346 linep = ml_get(pos.lnum);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002347 pos.col = (colnr_T)STRLEN(linep); // pos.col on trailing NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00002348 do_quotes = -1;
2349 line_breakcheck();
2350
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002351 // Check if this line contains a single-line comment
Bram Moolenaar8e145b82022-05-21 20:17:31 +01002352 if (comment_dir || lisp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002353 comment_col = check_linecomment(linep);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002354 // skip comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002355 if (lisp && comment_col != MAXCOL)
2356 pos.col = comment_col;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002357 }
2358 else
2359 {
2360 --pos.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002361 if (has_mbyte)
2362 pos.col -= (*mb_head_off)(linep, linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002363 }
2364 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002365 else // forward search
Bram Moolenaar071d4272004-06-13 20:20:40 +00002366 {
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002367 if (linep[pos.col] == NUL
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002368 // at end of line, go to next one
Bram Moolenaar8e145b82022-05-21 20:17:31 +01002369 // For lisp don't search for match in comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002370 || (lisp && comment_col != MAXCOL
Bram Moolenaar8e145b82022-05-21 20:17:31 +01002371 && pos.col == (colnr_T)comment_col))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002372 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002373 if (pos.lnum == curbuf->b_ml.ml_line_count // end of file
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002374 // line is exhausted and comment with it,
2375 // don't search for match in code
Bram Moolenaar8e145b82022-05-21 20:17:31 +01002376 || lispcomm)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002377 break;
2378 ++pos.lnum;
2379
2380 if (maxtravel && traveled++ > maxtravel)
2381 break;
2382
2383 linep = ml_get(pos.lnum);
2384 pos.col = 0;
2385 do_quotes = -1;
2386 line_breakcheck();
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002387 if (lisp) // find comment pos in new line
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002388 comment_col = check_linecomment(linep);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002389 }
2390 else
2391 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002392 if (has_mbyte)
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00002393 pos.col += (*mb_ptr2len)(linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002394 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00002395 ++pos.col;
2396 }
2397 }
2398
2399 /*
2400 * If FM_BLOCKSTOP given, stop at a '{' or '}' in column 0.
2401 */
Dominique Pelle7765f5c2022-04-10 11:26:53 +01002402 if (pos.col == 0 && (flags & FM_BLOCKSTOP)
2403 && (linep[0] == '{' || linep[0] == '}'))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002404 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002405 if (linep[0] == findc && count == 0) // match!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002406 return &pos;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002407 break; // out of scope
Bram Moolenaar071d4272004-06-13 20:20:40 +00002408 }
2409
2410 if (comment_dir)
2411 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002412 // Note: comments do not nest, and we ignore quotes in them
2413 // TODO: ignore comment brackets inside strings
Bram Moolenaar071d4272004-06-13 20:20:40 +00002414 if (comment_dir == FORWARD)
2415 {
2416 if (linep[pos.col] == '*' && linep[pos.col + 1] == '/')
2417 {
2418 pos.col++;
2419 return &pos;
2420 }
2421 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002422 else // Searching backwards
Bram Moolenaar071d4272004-06-13 20:20:40 +00002423 {
2424 /*
2425 * A comment may contain / * or / /, it may also start or end
Bram Moolenaarf8c53d32017-11-12 15:36:38 +01002426 * with / * /. Ignore a / * after / / and after *.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002427 */
2428 if (pos.col == 0)
2429 continue;
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002430 else if (raw_string)
2431 {
2432 if (linep[pos.col - 1] == 'R'
2433 && linep[pos.col] == '"'
2434 && vim_strchr(linep + pos.col + 1, '(') != NULL)
2435 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002436 // Possible start of raw string. Now that we have the
2437 // delimiter we can check if it ends before where we
2438 // started searching, or before the previously found
2439 // raw string start.
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002440 if (!find_rawstring_end(linep, &pos,
2441 count > 0 ? &match_pos : &curwin->w_cursor))
2442 {
2443 count++;
2444 match_pos = pos;
2445 match_pos.col--;
2446 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002447 linep = ml_get(pos.lnum); // may have been released
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002448 }
2449 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002450 else if ( linep[pos.col - 1] == '/'
2451 && linep[pos.col] == '*'
Bram Moolenaarf8c53d32017-11-12 15:36:38 +01002452 && (pos.col == 1 || linep[pos.col - 2] != '*')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002453 && (int)pos.col < comment_col)
2454 {
2455 count++;
2456 match_pos = pos;
2457 match_pos.col--;
2458 }
2459 else if (linep[pos.col - 1] == '*' && linep[pos.col] == '/')
2460 {
2461 if (count > 0)
2462 pos = match_pos;
2463 else if (pos.col > 1 && linep[pos.col - 2] == '/'
2464 && (int)pos.col <= comment_col)
2465 pos.col -= 2;
2466 else if (ignore_cend)
2467 continue;
2468 else
2469 return NULL;
2470 return &pos;
2471 }
2472 }
2473 continue;
2474 }
2475
2476 /*
2477 * If smart matching ('cpoptions' does not contain '%'), braces inside
2478 * of quotes are ignored, but only if there is an even number of
2479 * quotes in the line.
2480 */
2481 if (cpo_match)
2482 do_quotes = 0;
2483 else if (do_quotes == -1)
2484 {
2485 /*
2486 * Count the number of quotes in the line, skipping \" and '"'.
2487 * Watch out for "\\".
2488 */
2489 at_start = do_quotes;
2490 for (ptr = linep; *ptr; ++ptr)
2491 {
2492 if (ptr == linep + pos.col + backwards)
2493 at_start = (do_quotes & 1);
2494 if (*ptr == '"'
2495 && (ptr == linep || ptr[-1] != '\'' || ptr[1] != '\''))
2496 ++do_quotes;
2497 if (*ptr == '\\' && ptr[1] != NUL)
2498 ++ptr;
2499 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002500 do_quotes &= 1; // result is 1 with even number of quotes
Bram Moolenaar071d4272004-06-13 20:20:40 +00002501
2502 /*
2503 * If we find an uneven count, check current line and previous
2504 * one for a '\' at the end.
2505 */
2506 if (!do_quotes)
2507 {
2508 inquote = FALSE;
2509 if (ptr[-1] == '\\')
2510 {
2511 do_quotes = 1;
2512 if (start_in_quotes == MAYBE)
2513 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002514 // Do we need to use at_start here?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002515 inquote = TRUE;
2516 start_in_quotes = TRUE;
2517 }
2518 else if (backwards)
2519 inquote = TRUE;
2520 }
2521 if (pos.lnum > 1)
2522 {
2523 ptr = ml_get(pos.lnum - 1);
2524 if (*ptr && *(ptr + STRLEN(ptr) - 1) == '\\')
2525 {
2526 do_quotes = 1;
2527 if (start_in_quotes == MAYBE)
2528 {
2529 inquote = at_start;
2530 if (inquote)
2531 start_in_quotes = TRUE;
2532 }
2533 else if (!backwards)
2534 inquote = TRUE;
2535 }
Bram Moolenaaraec11792007-07-10 11:09:36 +00002536
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002537 // ml_get() only keeps one line, need to get linep again
Bram Moolenaaraec11792007-07-10 11:09:36 +00002538 linep = ml_get(pos.lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002539 }
2540 }
2541 }
2542 if (start_in_quotes == MAYBE)
2543 start_in_quotes = FALSE;
2544
2545 /*
2546 * If 'smartmatch' is set:
2547 * Things inside quotes are ignored by setting 'inquote'. If we
2548 * find a quote without a preceding '\' invert 'inquote'. At the
2549 * end of a line not ending in '\' we reset 'inquote'.
2550 *
2551 * In lines with an uneven number of quotes (without preceding '\')
2552 * we do not know which part to ignore. Therefore we only set
2553 * inquote if the number of quotes in a line is even, unless this
2554 * line or the previous one ends in a '\'. Complicated, isn't it?
2555 */
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002556 c = PTR2CHAR(linep + pos.col);
2557 switch (c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002558 {
2559 case NUL:
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002560 // at end of line without trailing backslash, reset inquote
Bram Moolenaar071d4272004-06-13 20:20:40 +00002561 if (pos.col == 0 || linep[pos.col - 1] != '\\')
2562 {
2563 inquote = FALSE;
2564 start_in_quotes = FALSE;
2565 }
2566 break;
2567
2568 case '"':
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002569 // a quote that is preceded with an odd number of backslashes is
2570 // ignored
Bram Moolenaar071d4272004-06-13 20:20:40 +00002571 if (do_quotes)
2572 {
2573 int col;
2574
2575 for (col = pos.col - 1; col >= 0; --col)
2576 if (linep[col] != '\\')
2577 break;
2578 if ((((int)pos.col - 1 - col) & 1) == 0)
2579 {
2580 inquote = !inquote;
2581 start_in_quotes = FALSE;
2582 }
2583 }
2584 break;
2585
2586 /*
2587 * If smart matching ('cpoptions' does not contain '%'):
2588 * Skip things in single quotes: 'x' or '\x'. Be careful for single
2589 * single quotes, eg jon's. Things like '\233' or '\x3f' are not
2590 * skipped, there is never a brace in them.
2591 * Ignore this when finding matches for `'.
2592 */
2593 case '\'':
2594 if (!cpo_match && initc != '\'' && findc != '\'')
2595 {
2596 if (backwards)
2597 {
2598 if (pos.col > 1)
2599 {
2600 if (linep[pos.col - 2] == '\'')
2601 {
2602 pos.col -= 2;
2603 break;
2604 }
Dominique Pelle7765f5c2022-04-10 11:26:53 +01002605 else if (linep[pos.col - 2] == '\\'
2606 && pos.col > 2 && linep[pos.col - 3] == '\'')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002607 {
2608 pos.col -= 3;
2609 break;
2610 }
2611 }
2612 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002613 else if (linep[pos.col + 1]) // forward search
Bram Moolenaar071d4272004-06-13 20:20:40 +00002614 {
Dominique Pelle7765f5c2022-04-10 11:26:53 +01002615 if (linep[pos.col + 1] == '\\'
2616 && linep[pos.col + 2] && linep[pos.col + 3] == '\'')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002617 {
2618 pos.col += 3;
2619 break;
2620 }
2621 else if (linep[pos.col + 2] == '\'')
2622 {
2623 pos.col += 2;
2624 break;
2625 }
2626 }
2627 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002628 // FALLTHROUGH
Bram Moolenaar071d4272004-06-13 20:20:40 +00002629
2630 default:
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002631 /*
2632 * For Lisp skip over backslashed (), {} and [].
2633 * (actually, we skip #\( et al)
2634 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002635 if (curbuf->b_p_lisp
2636 && vim_strchr((char_u *)"(){}[]", c) != NULL
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002637 && pos.col > 1
2638 && check_prevcol(linep, pos.col, '\\', NULL)
2639 && check_prevcol(linep, pos.col - 1, '#', NULL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002640 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002641
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002642 // Check for match outside of quotes, and inside of
2643 // quotes when the start is also inside of quotes.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002644 if ((!inquote || start_in_quotes == TRUE)
2645 && (c == initc || c == findc))
2646 {
2647 int col, bslcnt = 0;
2648
2649 if (!cpo_bsl)
2650 {
2651 for (col = pos.col; check_prevcol(linep, col, '\\', &col);)
2652 bslcnt++;
2653 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002654 // Only accept a match when 'M' is in 'cpo' or when escaping
2655 // is what we expect.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002656 if (cpo_bsl || (bslcnt & 1) == match_escaped)
2657 {
2658 if (c == initc)
2659 count++;
2660 else
2661 {
2662 if (count == 0)
2663 return &pos;
2664 count--;
2665 }
2666 }
2667 }
2668 }
2669 }
2670
2671 if (comment_dir == BACKWARD && count > 0)
2672 {
2673 pos = match_pos;
2674 return &pos;
2675 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002676 return (pos_T *)NULL; // never found it
Bram Moolenaar071d4272004-06-13 20:20:40 +00002677}
2678
2679/*
2680 * Check if line[] contains a / / comment.
2681 * Return MAXCOL if not, otherwise return the column.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002682 */
Bram Moolenaar6e371ec2021-12-12 14:16:39 +00002683 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002684check_linecomment(char_u *line)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002685{
2686 char_u *p;
2687
2688 p = line;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002689 // skip Lispish one-line comments
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002690 if (curbuf->b_p_lisp)
2691 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002692 if (vim_strchr(p, ';') != NULL) // there may be comments
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002693 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002694 int in_str = FALSE; // inside of string
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002695
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002696 p = line; // scan from start
Bram Moolenaar520470a2005-06-16 21:59:56 +00002697 while ((p = vim_strpbrk(p, (char_u *)"\";")) != NULL)
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002698 {
2699 if (*p == '"')
2700 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002701 if (in_str)
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002702 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002703 if (*(p - 1) != '\\') // skip escaped quote
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002704 in_str = FALSE;
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002705 }
2706 else if (p == line || ((p - line) >= 2
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002707 // skip #\" form
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002708 && *(p - 1) != '\\' && *(p - 2) != '#'))
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002709 in_str = TRUE;
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002710 }
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002711 else if (!in_str && ((p - line) < 2
Bram Moolenaarba263672021-12-29 18:09:13 +00002712 || (*(p - 1) != '\\' && *(p - 2) != '#'))
2713 && !is_pos_in_string(line, (colnr_T)(p - line)))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002714 break; // found!
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002715 ++p;
2716 }
2717 }
2718 else
2719 p = NULL;
2720 }
2721 else
Bram Moolenaar8e145b82022-05-21 20:17:31 +01002722 while ((p = vim_strchr(p, '/')) != NULL)
2723 {
2724 // Accept a double /, unless it's preceded with * and followed by
2725 // *, because * / / * is an end and start of a C comment. Only
2726 // accept the position if it is not inside a string.
2727 if (p[1] == '/' && (p == line || p[-1] != '*' || p[2] != '*')
Bram Moolenaarba263672021-12-29 18:09:13 +00002728 && !is_pos_in_string(line, (colnr_T)(p - line)))
Bram Moolenaar8e145b82022-05-21 20:17:31 +01002729 break;
2730 ++p;
2731 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002732
2733 if (p == NULL)
2734 return MAXCOL;
2735 return (int)(p - line);
2736}
2737
2738/*
2739 * Move cursor briefly to character matching the one under the cursor.
2740 * Used for Insert mode and "r" command.
2741 * Show the match only if it is visible on the screen.
2742 * If there isn't a match, then beep.
2743 */
2744 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002745showmatch(
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002746 int c) // char to show match for
Bram Moolenaar071d4272004-06-13 20:20:40 +00002747{
2748 pos_T *lpos, save_cursor;
2749 pos_T mpos;
2750 colnr_T vcol;
2751 long save_so;
2752 long save_siso;
2753#ifdef CURSOR_SHAPE
2754 int save_state;
2755#endif
2756 colnr_T save_dollar_vcol;
2757 char_u *p;
Bram Moolenaar6ed545e2022-05-09 20:09:23 +01002758 long *so = curwin->w_p_so >= 0 ? &curwin->w_p_so : &p_so;
2759 long *siso = curwin->w_p_siso >= 0 ? &curwin->w_p_siso : &p_siso;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002760
2761 /*
2762 * Only show match for chars in the 'matchpairs' option.
2763 */
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002764 // 'matchpairs' is "x:y,x:y"
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002765 for (p = curbuf->b_p_mps; *p != NUL; ++p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002766 {
2767#ifdef FEAT_RIGHTLEFT
Bram Moolenaar187d3ac2013-02-20 18:39:13 +01002768 if (PTR2CHAR(p) == c && (curwin->w_p_rl ^ p_ri))
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002769 break;
Bram Moolenaar187d3ac2013-02-20 18:39:13 +01002770#endif
Bram Moolenaar1614a142019-10-06 22:00:13 +02002771 p += mb_ptr2len(p) + 1;
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002772 if (PTR2CHAR(p) == c
Bram Moolenaar071d4272004-06-13 20:20:40 +00002773#ifdef FEAT_RIGHTLEFT
2774 && !(curwin->w_p_rl ^ p_ri)
2775#endif
2776 )
2777 break;
Bram Moolenaar1614a142019-10-06 22:00:13 +02002778 p += mb_ptr2len(p);
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002779 if (*p == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002780 return;
2781 }
Bram Moolenaar5b8cabf2021-04-02 18:55:57 +02002782 if (*p == NUL)
2783 return;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002784
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002785 if ((lpos = findmatch(NULL, NUL)) == NULL) // no match, so beep
Bram Moolenaar165bc692015-07-21 17:53:25 +02002786 vim_beep(BO_MATCH);
Bram Moolenaar187d3ac2013-02-20 18:39:13 +01002787 else if (lpos->lnum >= curwin->w_topline && lpos->lnum < curwin->w_botline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002788 {
2789 if (!curwin->w_p_wrap)
2790 getvcol(curwin, lpos, NULL, &vcol, NULL);
2791 if (curwin->w_p_wrap || (vcol >= curwin->w_leftcol
Bram Moolenaar02631462017-09-22 15:20:32 +02002792 && vcol < curwin->w_leftcol + curwin->w_width))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002793 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002794 mpos = *lpos; // save the pos, update_screen() may change it
Bram Moolenaar071d4272004-06-13 20:20:40 +00002795 save_cursor = curwin->w_cursor;
Bram Moolenaar375e3392019-01-31 18:26:10 +01002796 save_so = *so;
2797 save_siso = *siso;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002798 // Handle "$" in 'cpo': If the ')' is typed on top of the "$",
2799 // stop displaying the "$".
Bram Moolenaar76b9b362012-02-04 23:35:00 +01002800 if (dollar_vcol >= 0 && dollar_vcol == curwin->w_virtcol)
2801 dollar_vcol = -1;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002802 ++curwin->w_virtcol; // do display ')' just before "$"
Bram Moolenaara4d158b2022-08-14 14:17:45 +01002803 update_screen(UPD_VALID); // show the new char first
Bram Moolenaar071d4272004-06-13 20:20:40 +00002804
2805 save_dollar_vcol = dollar_vcol;
2806#ifdef CURSOR_SHAPE
2807 save_state = State;
Bram Moolenaar24959102022-05-07 20:01:16 +01002808 State = MODE_SHOWMATCH;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002809 ui_cursor_shape(); // may show different cursor shape
Bram Moolenaar071d4272004-06-13 20:20:40 +00002810#endif
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002811 curwin->w_cursor = mpos; // move to matching char
2812 *so = 0; // don't use 'scrolloff' here
2813 *siso = 0; // don't use 'sidescrolloff' here
Bram Moolenaar071d4272004-06-13 20:20:40 +00002814 showruler(FALSE);
2815 setcursor();
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002816 cursor_on(); // make sure that the cursor is shown
Bram Moolenaara338adc2018-01-31 20:51:47 +01002817 out_flush_cursor(TRUE, FALSE);
2818
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002819 // Restore dollar_vcol(), because setcursor() may call curs_rows()
2820 // which resets it if the matching position is in a previous line
2821 // and has a higher column number.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002822 dollar_vcol = save_dollar_vcol;
2823
2824 /*
2825 * brief pause, unless 'm' is present in 'cpo' and a character is
2826 * available.
2827 */
2828 if (vim_strchr(p_cpo, CPO_SHOWMATCH) != NULL)
Bram Moolenaareda1da02019-11-17 17:06:33 +01002829 ui_delay(p_mat * 100L + 8, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002830 else if (!char_avail())
Bram Moolenaareda1da02019-11-17 17:06:33 +01002831 ui_delay(p_mat * 100L + 9, FALSE);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002832 curwin->w_cursor = save_cursor; // restore cursor position
Bram Moolenaar375e3392019-01-31 18:26:10 +01002833 *so = save_so;
2834 *siso = save_siso;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002835#ifdef CURSOR_SHAPE
2836 State = save_state;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002837 ui_cursor_shape(); // may show different cursor shape
Bram Moolenaar071d4272004-06-13 20:20:40 +00002838#endif
2839 }
2840 }
2841}
2842
2843/*
Bram Moolenaar453c1922019-10-26 14:42:09 +02002844 * Check if the pattern is zero-width.
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002845 * If move is TRUE, check from the beginning of the buffer, else from position
2846 * "cur".
2847 * "direction" is FORWARD or BACKWARD.
2848 * Returns TRUE, FALSE or -1 for failure.
2849 */
2850 static int
2851is_zero_width(char_u *pattern, int move, pos_T *cur, int direction)
2852{
2853 regmmatch_T regmatch;
2854 int nmatched = 0;
2855 int result = -1;
2856 pos_T pos;
Bram Moolenaar53989552019-12-23 22:59:18 +01002857 int called_emsg_before = called_emsg;
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002858 int flag = 0;
2859
2860 if (pattern == NULL)
2861 pattern = spats[last_idx].pat;
2862
2863 if (search_regcomp(pattern, RE_SEARCH, RE_SEARCH,
2864 SEARCH_KEEP, &regmatch) == FAIL)
2865 return -1;
2866
2867 // init startcol correctly
2868 regmatch.startpos[0].col = -1;
2869 // move to match
2870 if (move)
2871 {
2872 CLEAR_POS(&pos);
2873 }
2874 else
2875 {
2876 pos = *cur;
2877 // accept a match at the cursor position
2878 flag = SEARCH_START;
2879 }
2880
2881 if (searchit(curwin, curbuf, &pos, NULL, direction, pattern, 1,
2882 SEARCH_KEEP + flag, RE_SEARCH, NULL) != FAIL)
2883 {
2884 // Zero-width pattern should match somewhere, then we can check if
2885 // start and end are in the same position.
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002886 do
2887 {
2888 regmatch.startpos[0].col++;
2889 nmatched = vim_regexec_multi(&regmatch, curwin, curbuf,
Paul Ollis65745772022-06-05 16:55:54 +01002890 pos.lnum, regmatch.startpos[0].col, NULL);
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002891 if (nmatched != 0)
2892 break;
Bram Moolenaar795aaa12020-10-02 20:36:01 +02002893 } while (regmatch.regprog != NULL
2894 && direction == FORWARD ? regmatch.startpos[0].col < pos.col
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002895 : regmatch.startpos[0].col > pos.col);
2896
Bram Moolenaar53989552019-12-23 22:59:18 +01002897 if (called_emsg == called_emsg_before)
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002898 {
2899 result = (nmatched != 0
2900 && regmatch.startpos[0].lnum == regmatch.endpos[0].lnum
2901 && regmatch.startpos[0].col == regmatch.endpos[0].col);
2902 }
2903 }
2904
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002905 vim_regfree(regmatch.regprog);
2906 return result;
2907}
2908
Bram Moolenaardde0efe2012-08-23 15:53:05 +02002909
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002910/*
2911 * Find next search match under cursor, cursor at end.
2912 * Used while an operator is pending, and in Visual mode.
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002913 */
2914 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002915current_search(
2916 long count,
Bram Moolenaar5d24a222018-12-23 19:10:09 +01002917 int forward) // TRUE for forward, FALSE for backward
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002918{
Bram Moolenaar5d24a222018-12-23 19:10:09 +01002919 pos_T start_pos; // start position of the pattern match
2920 pos_T end_pos; // end position of the pattern match
2921 pos_T orig_pos; // position of the cursor at beginning
2922 pos_T pos; // position after the pattern
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002923 int i;
2924 int dir;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01002925 int result; // result of various function calls
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002926 char_u old_p_ws = p_ws;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002927 int flags = 0;
Bram Moolenaarde9149e2013-07-17 19:22:13 +02002928 pos_T save_VIsual = VIsual;
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002929 int zero_width;
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02002930 int skip_first_backward;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002931
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002932 // Correct cursor when 'selection' is exclusive
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01002933 if (VIsual_active && *p_sel == 'e' && LT_POS(VIsual, curwin->w_cursor))
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002934 dec_cursor();
2935
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02002936 // When searching forward and the cursor is at the start of the Visual
2937 // area, skip the first search backward, otherwise it doesn't move.
2938 skip_first_backward = forward && VIsual_active
2939 && LT_POS(curwin->w_cursor, VIsual);
2940
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002941 orig_pos = pos = curwin->w_cursor;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002942 if (VIsual_active)
2943 {
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002944 if (forward)
2945 incl(&pos);
2946 else
2947 decl(&pos);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002948 }
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002949
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002950 // Is the pattern is zero-width?, this time, don't care about the direction
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002951 zero_width = is_zero_width(spats[last_idx].pat, TRUE, &curwin->w_cursor,
Bram Moolenaar22ab5472017-09-26 12:28:45 +02002952 FORWARD);
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002953 if (zero_width == -1)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002954 return FAIL; // pattern not found
Bram Moolenaarba6ba362012-08-08 15:27:57 +02002955
Bram Moolenaarba6ba362012-08-08 15:27:57 +02002956 /*
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002957 * The trick is to first search backwards and then search forward again,
2958 * so that a match at the current cursor position will be correctly
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02002959 * captured. When "forward" is false do it the other way around.
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002960 */
2961 for (i = 0; i < 2; i++)
2962 {
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002963 if (forward)
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02002964 {
2965 if (i == 0 && skip_first_backward)
2966 continue;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002967 dir = i;
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02002968 }
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002969 else
2970 dir = !i;
Bram Moolenaarba6ba362012-08-08 15:27:57 +02002971
2972 flags = 0;
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002973 if (!dir && !zero_width)
Bram Moolenaarba6ba362012-08-08 15:27:57 +02002974 flags = SEARCH_END;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01002975 end_pos = pos;
Bram Moolenaarba6ba362012-08-08 15:27:57 +02002976
Bram Moolenaar82cf7f62019-11-02 23:22:47 +01002977 // wrapping should not occur in the first round
2978 if (i == 0)
2979 p_ws = FALSE;
2980
Bram Moolenaar5d24a222018-12-23 19:10:09 +01002981 result = searchit(curwin, curbuf, &pos, &end_pos,
2982 (dir ? FORWARD : BACKWARD),
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002983 spats[last_idx].pat, (long) (i ? count : 1),
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02002984 SEARCH_KEEP | flags, RE_SEARCH, NULL);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002985
Bram Moolenaar82cf7f62019-11-02 23:22:47 +01002986 p_ws = old_p_ws;
2987
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002988 // First search may fail, but then start searching from the
2989 // beginning of the file (cursor might be on the search match)
2990 // except when Visual mode is active, so that extending the visual
2991 // selection works.
2992 if (i == 1 && !result) // not found, abort
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002993 {
2994 curwin->w_cursor = orig_pos;
2995 if (VIsual_active)
2996 VIsual = save_VIsual;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002997 return FAIL;
2998 }
Bram Moolenaar5d24a222018-12-23 19:10:09 +01002999 else if (i == 0 && !result)
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003000 {
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01003001 if (forward)
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003002 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003003 // try again from start of buffer
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01003004 CLEAR_POS(&pos);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003005 }
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01003006 else
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003007 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003008 // try again from end of buffer
3009 // searching backwards, so set pos to last line and col
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003010 pos.lnum = curwin->w_buffer->b_ml.ml_line_count;
Bram Moolenaar09168a72012-08-02 21:24:42 +02003011 pos.col = (colnr_T)STRLEN(
3012 ml_get(curwin->w_buffer->b_ml.ml_line_count));
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003013 }
3014 }
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003015 }
3016
3017 start_pos = pos;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003018
3019 if (!VIsual_active)
3020 VIsual = start_pos;
3021
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02003022 // put the cursor after the match
Bram Moolenaar5d24a222018-12-23 19:10:09 +01003023 curwin->w_cursor = end_pos;
Bram Moolenaar453c1922019-10-26 14:42:09 +02003024 if (LT_POS(VIsual, end_pos) && forward)
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02003025 {
3026 if (skip_first_backward)
3027 // put the cursor on the start of the match
3028 curwin->w_cursor = pos;
3029 else
3030 // put the cursor on last character of match
3031 dec_cursor();
3032 }
Bram Moolenaar28f224b2020-10-10 16:45:25 +02003033 else if (VIsual_active && LT_POS(curwin->w_cursor, VIsual) && forward)
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02003034 curwin->w_cursor = pos; // put the cursor on the start of the match
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003035 VIsual_active = TRUE;
3036 VIsual_mode = 'v';
3037
Bram Moolenaarb7633612019-02-10 21:48:25 +01003038 if (*p_sel == 'e')
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003039 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003040 // Correction for exclusive selection depends on the direction.
Bram Moolenaarb7633612019-02-10 21:48:25 +01003041 if (forward && LTOREQ_POS(VIsual, curwin->w_cursor))
3042 inc_cursor();
3043 else if (!forward && LTOREQ_POS(curwin->w_cursor, VIsual))
3044 inc(&VIsual);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003045 }
3046
3047#ifdef FEAT_FOLDING
3048 if (fdo_flags & FDO_SEARCH && KeyTyped)
3049 foldOpenCursor();
3050#endif
3051
3052 may_start_select('c');
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003053 setmouse();
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003054#ifdef FEAT_CLIPBOARD
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003055 // Make sure the clipboard gets updated. Needed because start and
3056 // end are still the same, and the selection needs to be owned
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003057 clip_star.vmode = NUL;
3058#endif
Bram Moolenaara4d158b2022-08-14 14:17:45 +01003059 redraw_curbuf_later(UPD_INVERTED);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003060 showmode();
3061
3062 return OK;
3063}
Bram Moolenaardde0efe2012-08-23 15:53:05 +02003064
Bram Moolenaar071d4272004-06-13 20:20:40 +00003065/*
3066 * return TRUE if line 'lnum' is empty or has white chars only.
3067 */
3068 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01003069linewhite(linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003070{
3071 char_u *p;
3072
3073 p = skipwhite(ml_get(lnum));
3074 return (*p == NUL);
3075}
Bram Moolenaar071d4272004-06-13 20:20:40 +00003076
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003077/*
3078 * Add the search count "[3/19]" to "msgbuf".
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003079 * See update_search_stat() for other arguments.
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003080 */
3081 static void
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003082cmdline_search_stat(
3083 int dirc,
3084 pos_T *pos,
3085 pos_T *cursor_pos,
3086 int show_top_bot_msg,
3087 char_u *msgbuf,
3088 int recompute,
3089 int maxcount,
3090 long timeout)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003091{
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003092 searchstat_T stat;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003093
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003094 update_search_stat(dirc, pos, cursor_pos, &stat, recompute, maxcount,
3095 timeout);
3096 if (stat.cur > 0)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003097 {
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003098 char t[SEARCH_STAT_BUF_LEN];
Bram Moolenaare2ad8262019-05-24 13:22:22 +02003099 size_t len;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003100
3101#ifdef FEAT_RIGHTLEFT
3102 if (curwin->w_p_rl && *curwin->w_p_rlc == 's')
3103 {
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003104 if (stat.incomplete == 1)
Bram Moolenaarb6cb26f2019-05-07 21:34:37 +02003105 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[?/??]");
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003106 else if (stat.cnt > maxcount && stat.cur > maxcount)
3107 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[>%d/>%d]",
3108 maxcount, maxcount);
3109 else if (stat.cnt > maxcount)
3110 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[>%d/%d]",
3111 maxcount, stat.cur);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003112 else
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003113 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[%d/%d]",
3114 stat.cnt, stat.cur);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003115 }
3116 else
3117#endif
3118 {
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003119 if (stat.incomplete == 1)
Bram Moolenaarb6cb26f2019-05-07 21:34:37 +02003120 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[?/??]");
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003121 else if (stat.cnt > maxcount && stat.cur > maxcount)
3122 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[>%d/>%d]",
3123 maxcount, maxcount);
3124 else if (stat.cnt > maxcount)
3125 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[%d/>%d]",
3126 stat.cur, maxcount);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003127 else
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003128 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[%d/%d]",
3129 stat.cur, stat.cnt);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003130 }
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02003131
3132 len = STRLEN(t);
Bram Moolenaardc6855a2019-05-18 19:26:29 +02003133 if (show_top_bot_msg && len + 2 < SEARCH_STAT_BUF_LEN)
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02003134 {
Bram Moolenaar16b58ae2019-09-06 20:40:21 +02003135 mch_memmove(t + 2, t, len);
3136 t[0] = 'W';
3137 t[1] = ' ';
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02003138 len += 2;
3139 }
3140
3141 mch_memmove(msgbuf + STRLEN(msgbuf) - len, t, len);
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003142 if (dirc == '?' && stat.cur == maxcount + 1)
3143 stat.cur = -1;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003144
Bram Moolenaar984f0312019-05-24 13:11:47 +02003145 // keep the message even after redraw, but don't put in history
3146 msg_hist_off = TRUE;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003147 give_warning(msgbuf, FALSE);
Bram Moolenaar984f0312019-05-24 13:11:47 +02003148 msg_hist_off = FALSE;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003149 }
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003150}
3151
3152/*
3153 * Add the search count information to "stat".
3154 * "stat" must not be NULL.
3155 * When "recompute" is TRUE always recompute the numbers.
3156 * dirc == 0: don't find the next/previous match (only set the result to "stat")
3157 * dirc == '/': find the next match
3158 * dirc == '?': find the previous match
3159 */
3160 static void
3161update_search_stat(
3162 int dirc,
3163 pos_T *pos,
3164 pos_T *cursor_pos,
3165 searchstat_T *stat,
3166 int recompute,
3167 int maxcount,
Bram Moolenaarf9ca08e2020-06-01 18:56:03 +02003168 long timeout UNUSED)
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003169{
3170 int save_ws = p_ws;
3171 int wraparound = FALSE;
3172 pos_T p = (*pos);
Bram Moolenaar14681622020-06-03 22:57:39 +02003173 static pos_T lastpos = {0, 0, 0};
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003174 static int cur = 0;
3175 static int cnt = 0;
3176 static int exact_match = FALSE;
3177 static int incomplete = 0;
3178 static int last_maxcount = SEARCH_STAT_DEF_MAX_COUNT;
3179 static int chgtick = 0;
3180 static char_u *lastpat = NULL;
3181 static buf_T *lbuf = NULL;
3182#ifdef FEAT_RELTIME
3183 proftime_T start;
3184#endif
3185
3186 vim_memset(stat, 0, sizeof(searchstat_T));
3187
3188 if (dirc == 0 && !recompute && !EMPTY_POS(lastpos))
3189 {
3190 stat->cur = cur;
3191 stat->cnt = cnt;
3192 stat->exact_match = exact_match;
3193 stat->incomplete = incomplete;
3194 stat->last_maxcount = last_maxcount;
3195 return;
3196 }
3197 last_maxcount = maxcount;
3198
3199 wraparound = ((dirc == '?' && LT_POS(lastpos, p))
3200 || (dirc == '/' && LT_POS(p, lastpos)));
3201
3202 // If anything relevant changed the count has to be recomputed.
3203 // MB_STRNICMP ignores case, but we should not ignore case.
3204 // Unfortunately, there is no MB_STRNICMP function.
3205 // XXX: above comment should be "no MB_STRCMP function" ?
3206 if (!(chgtick == CHANGEDTICK(curbuf)
3207 && MB_STRNICMP(lastpat, spats[last_idx].pat, STRLEN(lastpat)) == 0
3208 && STRLEN(lastpat) == STRLEN(spats[last_idx].pat)
3209 && EQUAL_POS(lastpos, *cursor_pos)
3210 && lbuf == curbuf) || wraparound || cur < 0
3211 || (maxcount > 0 && cur > maxcount) || recompute)
3212 {
3213 cur = 0;
3214 cnt = 0;
3215 exact_match = FALSE;
3216 incomplete = 0;
3217 CLEAR_POS(&lastpos);
3218 lbuf = curbuf;
3219 }
3220
3221 if (EQUAL_POS(lastpos, *cursor_pos) && !wraparound
3222 && (dirc == 0 || dirc == '/' ? cur < cnt : cur > 0))
3223 cur += dirc == 0 ? 0 : dirc == '/' ? 1 : -1;
3224 else
3225 {
3226 int done_search = FALSE;
3227 pos_T endpos = {0, 0, 0};
3228
3229 p_ws = FALSE;
3230#ifdef FEAT_RELTIME
3231 if (timeout > 0)
3232 profile_setlimit(timeout, &start);
3233#endif
3234 while (!got_int && searchit(curwin, curbuf, &lastpos, &endpos,
3235 FORWARD, NULL, 1, SEARCH_KEEP, RE_LAST, NULL) != FAIL)
3236 {
3237 done_search = TRUE;
3238#ifdef FEAT_RELTIME
3239 // Stop after passing the time limit.
3240 if (timeout > 0 && profile_passed_limit(&start))
3241 {
3242 incomplete = 1;
3243 break;
3244 }
3245#endif
3246 cnt++;
3247 if (LTOREQ_POS(lastpos, p))
3248 {
3249 cur = cnt;
Bram Moolenaar57f75a52020-06-02 22:06:21 +02003250 if (LT_POS(p, endpos))
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003251 exact_match = TRUE;
3252 }
3253 fast_breakcheck();
3254 if (maxcount > 0 && cnt > maxcount)
3255 {
3256 incomplete = 2; // max count exceeded
3257 break;
3258 }
3259 }
3260 if (got_int)
3261 cur = -1; // abort
3262 if (done_search)
3263 {
3264 vim_free(lastpat);
3265 lastpat = vim_strsave(spats[last_idx].pat);
3266 chgtick = CHANGEDTICK(curbuf);
3267 lbuf = curbuf;
3268 lastpos = p;
3269 }
3270 }
3271 stat->cur = cur;
3272 stat->cnt = cnt;
3273 stat->exact_match = exact_match;
3274 stat->incomplete = incomplete;
3275 stat->last_maxcount = last_maxcount;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003276 p_ws = save_ws;
3277}
3278
Bram Moolenaar071d4272004-06-13 20:20:40 +00003279#if defined(FEAT_FIND_ID) || defined(PROTO)
Bram Moolenaar409510c2022-06-01 15:23:13 +01003280
3281/*
3282 * Get line "lnum" and copy it into "buf[LSIZE]".
3283 * The copy is made because the regexp may make the line invalid when using a
3284 * mark.
3285 */
3286 static char_u *
3287get_line_and_copy(linenr_T lnum, char_u *buf)
3288{
3289 char_u *line = ml_get(lnum);
3290
3291 vim_strncpy(buf, line, LSIZE - 1);
3292 return buf;
3293}
3294
Bram Moolenaar071d4272004-06-13 20:20:40 +00003295/*
3296 * Find identifiers or defines in included files.
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003297 * If p_ic && compl_status_sol() then ptr must be in lowercase.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003298 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003299 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01003300find_pattern_in_path(
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003301 char_u *ptr, // pointer to search pattern
3302 int dir UNUSED, // direction of expansion
3303 int len, // length of search pattern
3304 int whole, // match whole words only
3305 int skip_comments, // don't match inside comments
3306 int type, // Type of search; are we looking for a type?
3307 // a macro?
Bram Moolenaar764b23c2016-01-30 21:10:09 +01003308 long count,
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003309 int action, // What to do when we find it
3310 linenr_T start_lnum, // first line to start searching
3311 linenr_T end_lnum) // last line for searching
Bram Moolenaar071d4272004-06-13 20:20:40 +00003312{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003313 SearchedFile *files; // Stack of included files
3314 SearchedFile *bigger; // When we need more space
Bram Moolenaar071d4272004-06-13 20:20:40 +00003315 int max_path_depth = 50;
3316 long match_count = 1;
3317
3318 char_u *pat;
3319 char_u *new_fname;
3320 char_u *curr_fname = curbuf->b_fname;
3321 char_u *prev_fname = NULL;
3322 linenr_T lnum;
3323 int depth;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003324 int depth_displayed; // For type==CHECK_PATH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003325 int old_files;
3326 int already_searched;
3327 char_u *file_line;
3328 char_u *line;
3329 char_u *p;
3330 char_u save_char;
3331 int define_matched;
3332 regmatch_T regmatch;
3333 regmatch_T incl_regmatch;
3334 regmatch_T def_regmatch;
3335 int matched = FALSE;
3336 int did_show = FALSE;
3337 int found = FALSE;
3338 int i;
3339 char_u *already = NULL;
3340 char_u *startp = NULL;
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003341 char_u *inc_opt = NULL;
Bram Moolenaar4033c552017-09-16 20:54:51 +02003342#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003343 win_T *curwin_save = NULL;
3344#endif
3345
3346 regmatch.regprog = NULL;
3347 incl_regmatch.regprog = NULL;
3348 def_regmatch.regprog = NULL;
3349
3350 file_line = alloc(LSIZE);
3351 if (file_line == NULL)
3352 return;
3353
Bram Moolenaar071d4272004-06-13 20:20:40 +00003354 if (type != CHECK_PATH && type != FIND_DEFINE
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003355 // when CONT_SOL is set compare "ptr" with the beginning of the
3356 // line is faster than quote_meta/regcomp/regexec "ptr" -- Acevedo
3357 && !compl_status_sol())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003358 {
3359 pat = alloc(len + 5);
3360 if (pat == NULL)
3361 goto fpip_end;
3362 sprintf((char *)pat, whole ? "\\<%.*s\\>" : "%.*s", len, ptr);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003363 // ignore case according to p_ic, p_scs and pat
Bram Moolenaar071d4272004-06-13 20:20:40 +00003364 regmatch.rm_ic = ignorecase(pat);
Bram Moolenaarf4e20992020-12-21 19:59:08 +01003365 regmatch.regprog = vim_regcomp(pat, magic_isset() ? RE_MAGIC : 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003366 vim_free(pat);
3367 if (regmatch.regprog == NULL)
3368 goto fpip_end;
3369 }
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003370 inc_opt = (*curbuf->b_p_inc == NUL) ? p_inc : curbuf->b_p_inc;
3371 if (*inc_opt != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003372 {
Bram Moolenaarf4e20992020-12-21 19:59:08 +01003373 incl_regmatch.regprog = vim_regcomp(inc_opt,
3374 magic_isset() ? RE_MAGIC : 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003375 if (incl_regmatch.regprog == NULL)
3376 goto fpip_end;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003377 incl_regmatch.rm_ic = FALSE; // don't ignore case in incl. pat.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003378 }
3379 if (type == FIND_DEFINE && (*curbuf->b_p_def != NUL || *p_def != NUL))
3380 {
3381 def_regmatch.regprog = vim_regcomp(*curbuf->b_p_def == NUL
Bram Moolenaarf4e20992020-12-21 19:59:08 +01003382 ? p_def : curbuf->b_p_def,
3383 magic_isset() ? RE_MAGIC : 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003384 if (def_regmatch.regprog == NULL)
3385 goto fpip_end;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003386 def_regmatch.rm_ic = FALSE; // don't ignore case in define pat.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003387 }
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003388 files = lalloc_clear(max_path_depth * sizeof(SearchedFile), TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003389 if (files == NULL)
3390 goto fpip_end;
3391 old_files = max_path_depth;
3392 depth = depth_displayed = -1;
3393
3394 lnum = start_lnum;
3395 if (end_lnum > curbuf->b_ml.ml_line_count)
3396 end_lnum = curbuf->b_ml.ml_line_count;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003397 if (lnum > end_lnum) // do at least one line
Bram Moolenaar071d4272004-06-13 20:20:40 +00003398 lnum = end_lnum;
Bram Moolenaar409510c2022-06-01 15:23:13 +01003399 line = get_line_and_copy(lnum, file_line);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003400
3401 for (;;)
3402 {
3403 if (incl_regmatch.regprog != NULL
3404 && vim_regexec(&incl_regmatch, line, (colnr_T)0))
3405 {
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003406 char_u *p_fname = (curr_fname == curbuf->b_fname)
3407 ? curbuf->b_ffname : curr_fname;
3408
3409 if (inc_opt != NULL && strstr((char *)inc_opt, "\\zs") != NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003410 // Use text from '\zs' to '\ze' (or end) of 'include'.
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003411 new_fname = find_file_name_in_path(incl_regmatch.startp[0],
Bram Moolenaarc84e3c12013-07-03 22:28:36 +02003412 (int)(incl_regmatch.endp[0] - incl_regmatch.startp[0]),
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003413 FNAME_EXP|FNAME_INCL|FNAME_REL, 1L, p_fname);
3414 else
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003415 // Use text after match with 'include'.
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003416 new_fname = file_name_in_line(incl_regmatch.endp[0], 0,
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003417 FNAME_EXP|FNAME_INCL|FNAME_REL, 1L, p_fname, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003418 already_searched = FALSE;
3419 if (new_fname != NULL)
3420 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003421 // Check whether we have already searched in this file
Bram Moolenaar071d4272004-06-13 20:20:40 +00003422 for (i = 0;; i++)
3423 {
3424 if (i == depth + 1)
3425 i = old_files;
3426 if (i == max_path_depth)
3427 break;
Bram Moolenaar99499b12019-05-23 21:35:48 +02003428 if (fullpathcmp(new_fname, files[i].name, TRUE, TRUE)
3429 & FPC_SAME)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003430 {
Dominique Pelle7765f5c2022-04-10 11:26:53 +01003431 if (type != CHECK_PATH
3432 && action == ACTION_SHOW_ALL
3433 && files[i].matched)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003434 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003435 msg_putchar('\n'); // cursor below last one
3436 if (!got_int) // don't display if 'q'
3437 // typed at "--more--"
3438 // message
Bram Moolenaar071d4272004-06-13 20:20:40 +00003439 {
3440 msg_home_replace_hl(new_fname);
Bram Moolenaar32526b32019-01-19 17:43:09 +01003441 msg_puts(_(" (includes previously listed match)"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003442 prev_fname = NULL;
3443 }
3444 }
Bram Moolenaard23a8232018-02-10 18:45:26 +01003445 VIM_CLEAR(new_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003446 already_searched = TRUE;
3447 break;
3448 }
3449 }
3450 }
3451
3452 if (type == CHECK_PATH && (action == ACTION_SHOW_ALL
3453 || (new_fname == NULL && !already_searched)))
3454 {
3455 if (did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003456 msg_putchar('\n'); // cursor below last one
Bram Moolenaar071d4272004-06-13 20:20:40 +00003457 else
3458 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003459 gotocmdline(TRUE); // cursor at status line
Bram Moolenaar32526b32019-01-19 17:43:09 +01003460 msg_puts_title(_("--- Included files "));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003461 if (action != ACTION_SHOW_ALL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003462 msg_puts_title(_("not found "));
3463 msg_puts_title(_("in path ---\n"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003464 }
3465 did_show = TRUE;
3466 while (depth_displayed < depth && !got_int)
3467 {
3468 ++depth_displayed;
3469 for (i = 0; i < depth_displayed; i++)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003470 msg_puts(" ");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003471 msg_home_replace(files[depth_displayed].name);
Bram Moolenaar32526b32019-01-19 17:43:09 +01003472 msg_puts(" -->\n");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003473 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003474 if (!got_int) // don't display if 'q' typed
3475 // for "--more--" message
Bram Moolenaar071d4272004-06-13 20:20:40 +00003476 {
3477 for (i = 0; i <= depth_displayed; i++)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003478 msg_puts(" ");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003479 if (new_fname != NULL)
3480 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003481 // using "new_fname" is more reliable, e.g., when
3482 // 'includeexpr' is set.
Bram Moolenaar8820b482017-03-16 17:23:31 +01003483 msg_outtrans_attr(new_fname, HL_ATTR(HLF_D));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003484 }
3485 else
3486 {
3487 /*
3488 * Isolate the file name.
3489 * Include the surrounding "" or <> if present.
3490 */
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003491 if (inc_opt != NULL
3492 && strstr((char *)inc_opt, "\\zs") != NULL)
3493 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003494 // pattern contains \zs, use the match
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003495 p = incl_regmatch.startp[0];
3496 i = (int)(incl_regmatch.endp[0]
3497 - incl_regmatch.startp[0]);
3498 }
3499 else
3500 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003501 // find the file name after the end of the match
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003502 for (p = incl_regmatch.endp[0];
3503 *p && !vim_isfilec(*p); p++)
3504 ;
3505 for (i = 0; vim_isfilec(p[i]); i++)
3506 ;
3507 }
3508
Bram Moolenaar071d4272004-06-13 20:20:40 +00003509 if (i == 0)
3510 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003511 // Nothing found, use the rest of the line.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003512 p = incl_regmatch.endp[0];
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00003513 i = (int)STRLEN(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003514 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003515 // Avoid checking before the start of the line, can
3516 // happen if \zs appears in the regexp.
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003517 else if (p > line)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003518 {
3519 if (p[-1] == '"' || p[-1] == '<')
3520 {
3521 --p;
3522 ++i;
3523 }
3524 if (p[i] == '"' || p[i] == '>')
3525 ++i;
3526 }
3527 save_char = p[i];
3528 p[i] = NUL;
Bram Moolenaar8820b482017-03-16 17:23:31 +01003529 msg_outtrans_attr(p, HL_ATTR(HLF_D));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003530 p[i] = save_char;
3531 }
3532
3533 if (new_fname == NULL && action == ACTION_SHOW_ALL)
3534 {
3535 if (already_searched)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003536 msg_puts(_(" (Already listed)"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003537 else
Bram Moolenaar32526b32019-01-19 17:43:09 +01003538 msg_puts(_(" NOT FOUND"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003539 }
3540 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003541 out_flush(); // output each line directly
Bram Moolenaar071d4272004-06-13 20:20:40 +00003542 }
3543
3544 if (new_fname != NULL)
3545 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003546 // Push the new file onto the file stack
Bram Moolenaar071d4272004-06-13 20:20:40 +00003547 if (depth + 1 == old_files)
3548 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003549 bigger = ALLOC_MULT(SearchedFile, max_path_depth * 2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003550 if (bigger != NULL)
3551 {
3552 for (i = 0; i <= depth; i++)
3553 bigger[i] = files[i];
3554 for (i = depth + 1; i < old_files + max_path_depth; i++)
3555 {
3556 bigger[i].fp = NULL;
3557 bigger[i].name = NULL;
3558 bigger[i].lnum = 0;
3559 bigger[i].matched = FALSE;
3560 }
3561 for (i = old_files; i < max_path_depth; i++)
3562 bigger[i + max_path_depth] = files[i];
3563 old_files += max_path_depth;
3564 max_path_depth *= 2;
3565 vim_free(files);
3566 files = bigger;
3567 }
3568 }
3569 if ((files[depth + 1].fp = mch_fopen((char *)new_fname, "r"))
3570 == NULL)
3571 vim_free(new_fname);
3572 else
3573 {
3574 if (++depth == old_files)
3575 {
3576 /*
3577 * lalloc() for 'bigger' must have failed above. We
3578 * will forget one of our already visited files now.
3579 */
3580 vim_free(files[old_files].name);
3581 ++old_files;
3582 }
3583 files[depth].name = curr_fname = new_fname;
3584 files[depth].lnum = 0;
3585 files[depth].matched = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003586 if (action == ACTION_EXPAND)
3587 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003588 msg_hist_off = TRUE; // reset in msg_trunc_attr()
Bram Moolenaar555b2802005-05-19 21:08:39 +00003589 vim_snprintf((char*)IObuff, IOSIZE,
3590 _("Scanning included file: %s"),
3591 (char *)new_fname);
Bram Moolenaar32526b32019-01-19 17:43:09 +01003592 msg_trunc_attr((char *)IObuff, TRUE, HL_ATTR(HLF_R));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003593 }
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003594 else if (p_verbose >= 5)
Bram Moolenaar87b5ca52006-03-04 21:55:31 +00003595 {
3596 verbose_enter();
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003597 smsg(_("Searching included file %s"),
Bram Moolenaar87b5ca52006-03-04 21:55:31 +00003598 (char *)new_fname);
3599 verbose_leave();
3600 }
3601
Bram Moolenaar071d4272004-06-13 20:20:40 +00003602 }
3603 }
3604 }
3605 else
3606 {
3607 /*
3608 * Check if the line is a define (type == FIND_DEFINE)
3609 */
3610 p = line;
3611search_line:
3612 define_matched = FALSE;
3613 if (def_regmatch.regprog != NULL
3614 && vim_regexec(&def_regmatch, line, (colnr_T)0))
3615 {
3616 /*
3617 * Pattern must be first identifier after 'define', so skip
3618 * to that position before checking for match of pattern. Also
3619 * don't let it match beyond the end of this identifier.
3620 */
3621 p = def_regmatch.endp[0];
3622 while (*p && !vim_iswordc(*p))
3623 p++;
3624 define_matched = TRUE;
3625 }
3626
3627 /*
3628 * Look for a match. Don't do this if we are looking for a
3629 * define and this line didn't match define_prog above.
3630 */
3631 if (def_regmatch.regprog == NULL || define_matched)
3632 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003633 if (define_matched || compl_status_sol())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003634 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003635 // compare the first "len" chars from "ptr"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003636 startp = skipwhite(p);
3637 if (p_ic)
3638 matched = !MB_STRNICMP(startp, ptr, len);
3639 else
3640 matched = !STRNCMP(startp, ptr, len);
3641 if (matched && define_matched && whole
3642 && vim_iswordc(startp[len]))
3643 matched = FALSE;
3644 }
3645 else if (regmatch.regprog != NULL
3646 && vim_regexec(&regmatch, line, (colnr_T)(p - line)))
3647 {
3648 matched = TRUE;
3649 startp = regmatch.startp[0];
3650 /*
3651 * Check if the line is not a comment line (unless we are
3652 * looking for a define). A line starting with "# define"
3653 * is not considered to be a comment line.
3654 */
3655 if (!define_matched && skip_comments)
3656 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003657 if ((*line != '#' ||
3658 STRNCMP(skipwhite(line + 1), "define", 6) != 0)
Bram Moolenaar81340392012-06-06 16:12:59 +02003659 && get_leader_len(line, NULL, FALSE, TRUE))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003660 matched = FALSE;
3661
3662 /*
3663 * Also check for a "/ *" or "/ /" before the match.
3664 * Skips lines like "int backwards; / * normal index
3665 * * /" when looking for "normal".
3666 * Note: Doesn't skip "/ *" in comments.
3667 */
3668 p = skipwhite(line);
3669 if (matched
3670 || (p[0] == '/' && p[1] == '*') || p[0] == '*')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003671 for (p = line; *p && p < startp; ++p)
3672 {
3673 if (matched
3674 && p[0] == '/'
3675 && (p[1] == '*' || p[1] == '/'))
3676 {
3677 matched = FALSE;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003678 // After "//" all text is comment
Bram Moolenaar071d4272004-06-13 20:20:40 +00003679 if (p[1] == '/')
3680 break;
3681 ++p;
3682 }
3683 else if (!matched && p[0] == '*' && p[1] == '/')
3684 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003685 // Can find match after "* /".
Bram Moolenaar071d4272004-06-13 20:20:40 +00003686 matched = TRUE;
3687 ++p;
3688 }
3689 }
3690 }
3691 }
3692 }
3693 }
3694 if (matched)
3695 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003696 if (action == ACTION_EXPAND)
3697 {
Bram Moolenaard9eefe32019-04-06 14:22:21 +02003698 int cont_s_ipos = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003699 int add_r;
3700 char_u *aux;
3701
3702 if (depth == -1 && lnum == curwin->w_cursor.lnum)
3703 break;
3704 found = TRUE;
3705 aux = p = startp;
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003706 if (compl_status_adding())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003707 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003708 p += ins_compl_len();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003709 if (vim_iswordp(p))
3710 goto exit_matched;
3711 p = find_word_start(p);
3712 }
3713 p = find_word_end(p);
3714 i = (int)(p - aux);
3715
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003716 if (compl_status_adding() && i == ins_compl_len())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003717 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003718 // IOSIZE > compl_length, so the STRNCPY works
Bram Moolenaar071d4272004-06-13 20:20:40 +00003719 STRNCPY(IObuff, aux, i);
Bram Moolenaar89d40322006-08-29 15:30:07 +00003720
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003721 // Get the next line: when "depth" < 0 from the current
3722 // buffer, otherwise from the included file. Jump to
3723 // exit_matched when past the last line.
Bram Moolenaar89d40322006-08-29 15:30:07 +00003724 if (depth < 0)
3725 {
3726 if (lnum >= end_lnum)
3727 goto exit_matched;
Bram Moolenaar409510c2022-06-01 15:23:13 +01003728 line = get_line_and_copy(++lnum, file_line);
Bram Moolenaar89d40322006-08-29 15:30:07 +00003729 }
3730 else if (vim_fgets(line = file_line,
3731 LSIZE, files[depth].fp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003732 goto exit_matched;
3733
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003734 // we read a line, set "already" to check this "line" later
3735 // if depth >= 0 we'll increase files[depth].lnum far
Bram Moolenaar8e7d6222020-12-18 19:49:56 +01003736 // below -- Acevedo
Bram Moolenaar071d4272004-06-13 20:20:40 +00003737 already = aux = p = skipwhite(line);
3738 p = find_word_start(p);
3739 p = find_word_end(p);
3740 if (p > aux)
3741 {
3742 if (*aux != ')' && IObuff[i-1] != TAB)
3743 {
3744 if (IObuff[i-1] != ' ')
3745 IObuff[i++] = ' ';
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003746 // IObuf =~ "\(\k\|\i\).* ", thus i >= 2
Bram Moolenaar071d4272004-06-13 20:20:40 +00003747 if (p_js
3748 && (IObuff[i-2] == '.'
3749 || (vim_strchr(p_cpo, CPO_JOINSP) == NULL
3750 && (IObuff[i-2] == '?'
3751 || IObuff[i-2] == '!'))))
3752 IObuff[i++] = ' ';
3753 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003754 // copy as much as possible of the new word
Bram Moolenaar071d4272004-06-13 20:20:40 +00003755 if (p - aux >= IOSIZE - i)
3756 p = aux + IOSIZE - i - 1;
3757 STRNCPY(IObuff + i, aux, p - aux);
3758 i += (int)(p - aux);
Bram Moolenaard9eefe32019-04-06 14:22:21 +02003759 cont_s_ipos = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003760 }
3761 IObuff[i] = NUL;
3762 aux = IObuff;
3763
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003764 if (i == ins_compl_len())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003765 goto exit_matched;
3766 }
3767
Bram Moolenaare8c3a142006-08-29 14:30:35 +00003768 add_r = ins_compl_add_infercase(aux, i, p_ic,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003769 curr_fname == curbuf->b_fname ? NULL : curr_fname,
Bram Moolenaard9eefe32019-04-06 14:22:21 +02003770 dir, cont_s_ipos);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003771 if (add_r == OK)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003772 // if dir was BACKWARD then honor it just once
Bram Moolenaar071d4272004-06-13 20:20:40 +00003773 dir = FORWARD;
Bram Moolenaar572cb562005-08-05 21:35:02 +00003774 else if (add_r == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003775 break;
3776 }
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003777 else if (action == ACTION_SHOW_ALL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003778 {
3779 found = TRUE;
3780 if (!did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003781 gotocmdline(TRUE); // cursor at status line
Bram Moolenaar071d4272004-06-13 20:20:40 +00003782 if (curr_fname != prev_fname)
3783 {
3784 if (did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003785 msg_putchar('\n'); // cursor below last one
3786 if (!got_int) // don't display if 'q' typed
3787 // at "--more--" message
Bram Moolenaar071d4272004-06-13 20:20:40 +00003788 msg_home_replace_hl(curr_fname);
3789 prev_fname = curr_fname;
3790 }
3791 did_show = TRUE;
3792 if (!got_int)
3793 show_pat_in_path(line, type, TRUE, action,
3794 (depth == -1) ? NULL : files[depth].fp,
3795 (depth == -1) ? &lnum : &files[depth].lnum,
3796 match_count++);
3797
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003798 // Set matched flag for this file and all the ones that
3799 // include it
Bram Moolenaar071d4272004-06-13 20:20:40 +00003800 for (i = 0; i <= depth; ++i)
3801 files[i].matched = TRUE;
3802 }
3803 else if (--count <= 0)
3804 {
3805 found = TRUE;
3806 if (depth == -1 && lnum == curwin->w_cursor.lnum
Bram Moolenaar4033c552017-09-16 20:54:51 +02003807#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003808 && g_do_tagpreview == 0
3809#endif
3810 )
Bram Moolenaarac78dd42022-01-02 19:25:26 +00003811 emsg(_(e_match_is_on_current_line));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003812 else if (action == ACTION_SHOW)
3813 {
3814 show_pat_in_path(line, type, did_show, action,
3815 (depth == -1) ? NULL : files[depth].fp,
3816 (depth == -1) ? &lnum : &files[depth].lnum, 1L);
3817 did_show = TRUE;
3818 }
3819 else
3820 {
3821#ifdef FEAT_GUI
3822 need_mouse_correct = TRUE;
3823#endif
Bram Moolenaar4033c552017-09-16 20:54:51 +02003824#if defined(FEAT_QUICKFIX)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003825 // ":psearch" uses the preview window
Bram Moolenaar071d4272004-06-13 20:20:40 +00003826 if (g_do_tagpreview != 0)
3827 {
3828 curwin_save = curwin;
Bram Moolenaar576a4a62019-08-18 15:25:17 +02003829 prepare_tagpreview(TRUE, TRUE, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003830 }
3831#endif
3832 if (action == ACTION_SPLIT)
3833 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003834 if (win_split(0, 0) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003835 break;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02003836 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003837 }
3838 if (depth == -1)
3839 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003840 // match in current file
Bram Moolenaar4033c552017-09-16 20:54:51 +02003841#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003842 if (g_do_tagpreview != 0)
3843 {
Bram Moolenaar92bb83e2021-02-03 23:04:46 +01003844 if (!win_valid(curwin_save))
3845 break;
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02003846 if (!GETFILE_SUCCESS(getfile(
Bram Moolenaarc31f9ae2017-07-23 22:02:02 +02003847 curwin_save->w_buffer->b_fnum, NULL,
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02003848 NULL, TRUE, lnum, FALSE)))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003849 break; // failed to jump to file
Bram Moolenaar071d4272004-06-13 20:20:40 +00003850 }
3851 else
3852#endif
3853 setpcmark();
3854 curwin->w_cursor.lnum = lnum;
Bram Moolenaarc31f9ae2017-07-23 22:02:02 +02003855 check_cursor();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003856 }
3857 else
3858 {
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02003859 if (!GETFILE_SUCCESS(getfile(
3860 0, files[depth].name, NULL, TRUE,
3861 files[depth].lnum, FALSE)))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003862 break; // failed to jump to file
3863 // autocommands may have changed the lnum, we don't
3864 // want that here
Bram Moolenaar071d4272004-06-13 20:20:40 +00003865 curwin->w_cursor.lnum = files[depth].lnum;
3866 }
3867 }
3868 if (action != ACTION_SHOW)
3869 {
Bram Moolenaarfe81d452009-04-22 14:44:41 +00003870 curwin->w_cursor.col = (colnr_T)(startp - line);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003871 curwin->w_set_curswant = TRUE;
3872 }
3873
Bram Moolenaar4033c552017-09-16 20:54:51 +02003874#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003875 if (g_do_tagpreview != 0
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00003876 && curwin != curwin_save && win_valid(curwin_save))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003877 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003878 // Return cursor to where we were
Bram Moolenaar071d4272004-06-13 20:20:40 +00003879 validate_cursor();
Bram Moolenaara4d158b2022-08-14 14:17:45 +01003880 redraw_later(UPD_VALID);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003881 win_enter(curwin_save, TRUE);
3882 }
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +01003883# ifdef FEAT_PROP_POPUP
Bram Moolenaar1b6d9c42019-08-05 21:52:04 +02003884 else if (WIN_IS_POPUP(curwin))
3885 // can't keep focus in popup window
3886 win_enter(firstwin, TRUE);
3887# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003888#endif
3889 break;
3890 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003891exit_matched:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003892 matched = FALSE;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003893 // look for other matches in the rest of the line if we
3894 // are not at the end of it already
Bram Moolenaar071d4272004-06-13 20:20:40 +00003895 if (def_regmatch.regprog == NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003896 && action == ACTION_EXPAND
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003897 && !compl_status_sol()
Bram Moolenaarfe81d452009-04-22 14:44:41 +00003898 && *startp != NUL
Bram Moolenaar1614a142019-10-06 22:00:13 +02003899 && *(p = startp + mb_ptr2len(startp)) != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003900 goto search_line;
3901 }
3902 line_breakcheck();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003903 if (action == ACTION_EXPAND)
Bram Moolenaar472e8592016-10-15 17:06:47 +02003904 ins_compl_check_keys(30, FALSE);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003905 if (got_int || ins_compl_interrupted())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003906 break;
3907
3908 /*
3909 * Read the next line. When reading an included file and encountering
3910 * end-of-file, close the file and continue in the file that included
3911 * it.
3912 */
3913 while (depth >= 0 && !already
3914 && vim_fgets(line = file_line, LSIZE, files[depth].fp))
3915 {
3916 fclose(files[depth].fp);
3917 --old_files;
3918 files[old_files].name = files[depth].name;
3919 files[old_files].matched = files[depth].matched;
3920 --depth;
3921 curr_fname = (depth == -1) ? curbuf->b_fname
3922 : files[depth].name;
3923 if (depth < depth_displayed)
3924 depth_displayed = depth;
3925 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003926 if (depth >= 0) // we could read the line
Bram Moolenaarc84e3c12013-07-03 22:28:36 +02003927 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003928 files[depth].lnum++;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003929 // Remove any CR and LF from the line.
Bram Moolenaarc84e3c12013-07-03 22:28:36 +02003930 i = (int)STRLEN(line);
3931 if (i > 0 && line[i - 1] == '\n')
3932 line[--i] = NUL;
3933 if (i > 0 && line[i - 1] == '\r')
3934 line[--i] = NUL;
3935 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003936 else if (!already)
3937 {
3938 if (++lnum > end_lnum)
3939 break;
Bram Moolenaar409510c2022-06-01 15:23:13 +01003940 line = get_line_and_copy(lnum, file_line);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003941 }
3942 already = NULL;
3943 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003944 // End of big for (;;) loop.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003945
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003946 // Close any files that are still open.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003947 for (i = 0; i <= depth; i++)
3948 {
3949 fclose(files[i].fp);
3950 vim_free(files[i].name);
3951 }
3952 for (i = old_files; i < max_path_depth; i++)
3953 vim_free(files[i].name);
3954 vim_free(files);
3955
3956 if (type == CHECK_PATH)
3957 {
3958 if (!did_show)
3959 {
3960 if (action != ACTION_SHOW_ALL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003961 msg(_("All included files were found"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003962 else
Bram Moolenaar32526b32019-01-19 17:43:09 +01003963 msg(_("No included files"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003964 }
3965 }
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003966 else if (!found && action != ACTION_EXPAND)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003967 {
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003968 if (got_int || ins_compl_interrupted())
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00003969 emsg(_(e_interrupted));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003970 else if (type == FIND_DEFINE)
Bram Moolenaarac78dd42022-01-02 19:25:26 +00003971 emsg(_(e_couldnt_find_definition));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003972 else
Bram Moolenaarac78dd42022-01-02 19:25:26 +00003973 emsg(_(e_couldnt_find_pattern));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003974 }
3975 if (action == ACTION_SHOW || action == ACTION_SHOW_ALL)
3976 msg_end();
3977
3978fpip_end:
3979 vim_free(file_line);
Bram Moolenaar473de612013-06-08 18:19:48 +02003980 vim_regfree(regmatch.regprog);
3981 vim_regfree(incl_regmatch.regprog);
3982 vim_regfree(def_regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003983}
3984
3985 static void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01003986show_pat_in_path(
3987 char_u *line,
3988 int type,
3989 int did_show,
3990 int action,
3991 FILE *fp,
3992 linenr_T *lnum,
3993 long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003994{
3995 char_u *p;
3996
3997 if (did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003998 msg_putchar('\n'); // cursor below last one
Bram Moolenaar91170f82006-05-05 21:15:17 +00003999 else if (!msg_silent)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004000 gotocmdline(TRUE); // cursor at status line
4001 if (got_int) // 'q' typed at "--more--" message
Bram Moolenaar071d4272004-06-13 20:20:40 +00004002 return;
4003 for (;;)
4004 {
4005 p = line + STRLEN(line) - 1;
4006 if (fp != NULL)
4007 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004008 // We used fgets(), so get rid of newline at end
Bram Moolenaar071d4272004-06-13 20:20:40 +00004009 if (p >= line && *p == '\n')
4010 --p;
4011 if (p >= line && *p == '\r')
4012 --p;
4013 *(p + 1) = NUL;
4014 }
4015 if (action == ACTION_SHOW_ALL)
4016 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004017 sprintf((char *)IObuff, "%3ld: ", count); // show match nr
Bram Moolenaar32526b32019-01-19 17:43:09 +01004018 msg_puts((char *)IObuff);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004019 sprintf((char *)IObuff, "%4ld", *lnum); // show line nr
4020 // Highlight line numbers
Bram Moolenaar32526b32019-01-19 17:43:09 +01004021 msg_puts_attr((char *)IObuff, HL_ATTR(HLF_N));
4022 msg_puts(" ");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004023 }
Bram Moolenaar26a60b42005-02-22 08:49:11 +00004024 msg_prt_line(line, FALSE);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004025 out_flush(); // show one line at a time
Bram Moolenaar071d4272004-06-13 20:20:40 +00004026
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004027 // Definition continues until line that doesn't end with '\'
Bram Moolenaar071d4272004-06-13 20:20:40 +00004028 if (got_int || type != FIND_DEFINE || p < line || *p != '\\')
4029 break;
4030
4031 if (fp != NULL)
4032 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004033 if (vim_fgets(line, LSIZE, fp)) // end of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00004034 break;
4035 ++*lnum;
4036 }
4037 else
4038 {
4039 if (++*lnum > curbuf->b_ml.ml_line_count)
4040 break;
4041 line = ml_get(*lnum);
4042 }
4043 msg_putchar('\n');
4044 }
4045}
4046#endif
4047
4048#ifdef FEAT_VIMINFO
Bram Moolenaar6bd1d772019-10-09 22:01:25 +02004049/*
4050 * Return the last used search pattern at "idx".
4051 */
Bram Moolenaarc3328162019-07-23 22:15:25 +02004052 spat_T *
4053get_spat(int idx)
4054{
4055 return &spats[idx];
4056}
4057
Bram Moolenaar6bd1d772019-10-09 22:01:25 +02004058/*
4059 * Return the last used search pattern index.
4060 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004061 int
Bram Moolenaarc3328162019-07-23 22:15:25 +02004062get_spat_last_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004063{
Bram Moolenaarc3328162019-07-23 22:15:25 +02004064 return last_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004065}
Bram Moolenaar071d4272004-06-13 20:20:40 +00004066#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004067
Yegappan Lakshmanan38b85cb2022-02-24 13:28:41 +00004068#if defined(FEAT_EVAL) || defined(FEAT_PROTO)
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004069/*
4070 * "searchcount()" function
4071 */
4072 void
4073f_searchcount(typval_T *argvars, typval_T *rettv)
4074{
4075 pos_T pos = curwin->w_cursor;
4076 char_u *pattern = NULL;
4077 int maxcount = SEARCH_STAT_DEF_MAX_COUNT;
4078 long timeout = SEARCH_STAT_DEF_TIMEOUT;
Bram Moolenaar4140c4f2020-09-05 23:16:00 +02004079 int recompute = TRUE;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004080 searchstat_T stat;
4081
4082 if (rettv_dict_alloc(rettv) == FAIL)
4083 return;
4084
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02004085 if (in_vim9script() && check_for_opt_dict_arg(argvars, 0) == FAIL)
4086 return;
4087
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004088 if (shortmess(SHM_SEARCHCOUNT)) // 'shortmess' contains 'S' flag
4089 recompute = TRUE;
4090
4091 if (argvars[0].v_type != VAR_UNKNOWN)
4092 {
Bram Moolenaar14681622020-06-03 22:57:39 +02004093 dict_T *dict;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004094 dictitem_T *di;
4095 listitem_T *li;
4096 int error = FALSE;
4097
Bram Moolenaar14681622020-06-03 22:57:39 +02004098 if (argvars[0].v_type != VAR_DICT || argvars[0].vval.v_dict == NULL)
4099 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00004100 emsg(_(e_dictionary_required));
Bram Moolenaar14681622020-06-03 22:57:39 +02004101 return;
4102 }
4103 dict = argvars[0].vval.v_dict;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004104 di = dict_find(dict, (char_u *)"timeout", -1);
4105 if (di != NULL)
4106 {
4107 timeout = (long)tv_get_number_chk(&di->di_tv, &error);
4108 if (error)
4109 return;
4110 }
4111 di = dict_find(dict, (char_u *)"maxcount", -1);
4112 if (di != NULL)
4113 {
4114 maxcount = (int)tv_get_number_chk(&di->di_tv, &error);
4115 if (error)
4116 return;
4117 }
Bram Moolenaard61efa52022-07-23 09:52:04 +01004118 recompute = dict_get_bool(dict, "recompute", recompute);
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004119 di = dict_find(dict, (char_u *)"pattern", -1);
4120 if (di != NULL)
4121 {
4122 pattern = tv_get_string_chk(&di->di_tv);
4123 if (pattern == NULL)
4124 return;
4125 }
4126 di = dict_find(dict, (char_u *)"pos", -1);
4127 if (di != NULL)
4128 {
4129 if (di->di_tv.v_type != VAR_LIST)
4130 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00004131 semsg(_(e_invalid_argument_str), "pos");
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004132 return;
4133 }
4134 if (list_len(di->di_tv.vval.v_list) != 3)
4135 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00004136 semsg(_(e_invalid_argument_str), "List format should be [lnum, col, off]");
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004137 return;
4138 }
4139 li = list_find(di->di_tv.vval.v_list, 0L);
4140 if (li != NULL)
4141 {
4142 pos.lnum = tv_get_number_chk(&li->li_tv, &error);
4143 if (error)
4144 return;
4145 }
4146 li = list_find(di->di_tv.vval.v_list, 1L);
4147 if (li != NULL)
4148 {
Bram Moolenaar6ed545e2022-05-09 20:09:23 +01004149 pos.col = tv_get_number_chk(&li->li_tv, &error) - 1;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004150 if (error)
4151 return;
4152 }
4153 li = list_find(di->di_tv.vval.v_list, 2L);
4154 if (li != NULL)
4155 {
Bram Moolenaar6ed545e2022-05-09 20:09:23 +01004156 pos.coladd = tv_get_number_chk(&li->li_tv, &error);
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004157 if (error)
4158 return;
4159 }
4160 }
4161 }
4162
4163 save_last_search_pattern();
Christian Brabandt6dd74242022-02-14 12:44:32 +00004164#ifdef FEAT_SEARCH_EXTRA
4165 save_incsearch_state();
4166#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004167 if (pattern != NULL)
4168 {
4169 if (*pattern == NUL)
4170 goto the_end;
Bram Moolenaar109aece2020-06-01 19:08:54 +02004171 vim_free(spats[last_idx].pat);
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004172 spats[last_idx].pat = vim_strsave(pattern);
4173 }
4174 if (spats[last_idx].pat == NULL || *spats[last_idx].pat == NUL)
4175 goto the_end; // the previous pattern was never defined
4176
4177 update_search_stat(0, &pos, &pos, &stat, recompute, maxcount, timeout);
4178
4179 dict_add_number(rettv->vval.v_dict, "current", stat.cur);
4180 dict_add_number(rettv->vval.v_dict, "total", stat.cnt);
4181 dict_add_number(rettv->vval.v_dict, "exact_match", stat.exact_match);
4182 dict_add_number(rettv->vval.v_dict, "incomplete", stat.incomplete);
4183 dict_add_number(rettv->vval.v_dict, "maxcount", stat.last_maxcount);
4184
4185the_end:
4186 restore_last_search_pattern();
Christian Brabandt6dd74242022-02-14 12:44:32 +00004187#ifdef FEAT_SEARCH_EXTRA
4188 restore_incsearch_state();
4189#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004190}
Yegappan Lakshmanan38b85cb2022-02-24 13:28:41 +00004191#endif
Bram Moolenaar635414d2020-09-11 22:25:15 +02004192
4193/*
4194 * Fuzzy string matching
4195 *
4196 * Ported from the lib_fts library authored by Forrest Smith.
4197 * https://github.com/forrestthewoods/lib_fts/tree/master/code
4198 *
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004199 * The following blog describes the fuzzy matching algorithm:
Bram Moolenaar635414d2020-09-11 22:25:15 +02004200 * https://www.forrestthewoods.com/blog/reverse_engineering_sublime_texts_fuzzy_match/
4201 *
4202 * Each matching string is assigned a score. The following factors are checked:
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004203 * - Matched letter
4204 * - Unmatched letter
4205 * - Consecutively matched letters
4206 * - Proximity to start
4207 * - Letter following a separator (space, underscore)
4208 * - Uppercase letter following lowercase (aka CamelCase)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004209 *
4210 * Matched letters are good. Unmatched letters are bad. Matching near the start
4211 * is good. Matching the first letter in the middle of a phrase is good.
4212 * Matching the uppercase letters in camel case entries is good.
4213 *
4214 * The score assigned for each factor is explained below.
4215 * File paths are different from file names. File extensions may be ignorable.
4216 * Single words care about consecutive matches but not separators or camel
4217 * case.
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004218 * Score starts at 100
Bram Moolenaar635414d2020-09-11 22:25:15 +02004219 * Matched letter: +0 points
4220 * Unmatched letter: -1 point
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004221 * Consecutive match bonus: +15 points
4222 * First letter bonus: +15 points
4223 * Separator bonus: +30 points
4224 * Camel case bonus: +30 points
4225 * Unmatched leading letter: -5 points (max: -15)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004226 *
4227 * There is some nuance to this. Scores don’t have an intrinsic meaning. The
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004228 * score range isn’t 0 to 100. It’s roughly [50, 150]. Longer words have a
Bram Moolenaar635414d2020-09-11 22:25:15 +02004229 * lower minimum score due to unmatched letter penalty. Longer search patterns
4230 * have a higher maximum score due to match bonuses.
4231 *
4232 * Separator and camel case bonus is worth a LOT. Consecutive matches are worth
4233 * quite a bit.
4234 *
4235 * There is a penalty if you DON’T match the first three letters. Which
4236 * effectively rewards matching near the start. However there’s no difference
4237 * in matching between the middle and end.
4238 *
4239 * There is not an explicit bonus for an exact match. Unmatched letters receive
4240 * a penalty. So shorter strings and closer matches are worth more.
4241 */
4242typedef struct
4243{
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004244 int idx; // used for stable sort
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004245 listitem_T *item;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004246 int score;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004247 list_T *lmatchpos;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004248} fuzzyItem_T;
4249
Bram Moolenaare9f9f162020-10-20 19:01:30 +02004250// bonus for adjacent matches; this is higher than SEPARATOR_BONUS so that
4251// matching a whole word is preferred.
4252#define SEQUENTIAL_BONUS 40
Bram Moolenaardcdd42a2020-10-29 18:58:01 +01004253// bonus if match occurs after a path separator
4254#define PATH_SEPARATOR_BONUS 30
4255// bonus if match occurs after a word separator
4256#define WORD_SEPARATOR_BONUS 25
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004257// bonus if match is uppercase and prev is lower
4258#define CAMEL_BONUS 30
4259// bonus if the first letter is matched
4260#define FIRST_LETTER_BONUS 15
4261// penalty applied for every letter in str before the first match
kylo252ae6f1d82022-02-16 19:24:07 +00004262#define LEADING_LETTER_PENALTY (-5)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004263// maximum penalty for leading letters
kylo252ae6f1d82022-02-16 19:24:07 +00004264#define MAX_LEADING_LETTER_PENALTY (-15)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004265// penalty for every letter that doesn't match
kylo252ae6f1d82022-02-16 19:24:07 +00004266#define UNMATCHED_LETTER_PENALTY (-1)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004267// penalty for gap in matching positions (-2 * k)
kylo252ae6f1d82022-02-16 19:24:07 +00004268#define GAP_PENALTY (-2)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004269// Score for a string that doesn't fuzzy match the pattern
kylo252ae6f1d82022-02-16 19:24:07 +00004270#define SCORE_NONE (-9999)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004271
4272#define FUZZY_MATCH_RECURSION_LIMIT 10
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004273
4274/*
4275 * Compute a score for a fuzzy matched string. The matching character locations
4276 * are in 'matches'.
4277 */
4278 static int
4279fuzzy_match_compute_score(
4280 char_u *str,
4281 int strSz,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004282 int_u *matches,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004283 int numMatches)
4284{
4285 int score;
4286 int penalty;
4287 int unmatched;
4288 int i;
4289 char_u *p = str;
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004290 int_u sidx = 0;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004291
4292 // Initialize score
4293 score = 100;
4294
4295 // Apply leading letter penalty
4296 penalty = LEADING_LETTER_PENALTY * matches[0];
4297 if (penalty < MAX_LEADING_LETTER_PENALTY)
4298 penalty = MAX_LEADING_LETTER_PENALTY;
4299 score += penalty;
4300
4301 // Apply unmatched penalty
4302 unmatched = strSz - numMatches;
4303 score += UNMATCHED_LETTER_PENALTY * unmatched;
4304
4305 // Apply ordering bonuses
4306 for (i = 0; i < numMatches; ++i)
4307 {
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004308 int_u currIdx = matches[i];
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004309
4310 if (i > 0)
4311 {
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004312 int_u prevIdx = matches[i - 1];
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004313
4314 // Sequential
4315 if (currIdx == (prevIdx + 1))
4316 score += SEQUENTIAL_BONUS;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004317 else
4318 score += GAP_PENALTY * (currIdx - prevIdx);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004319 }
4320
4321 // Check for bonuses based on neighbor character value
4322 if (currIdx > 0)
4323 {
4324 // Camel case
Bram Moolenaarc53e9c52020-09-22 22:08:32 +02004325 int neighbor = ' ';
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004326 int curr;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004327
4328 if (has_mbyte)
4329 {
4330 while (sidx < currIdx)
4331 {
4332 neighbor = (*mb_ptr2char)(p);
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004333 MB_PTR_ADV(p);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004334 sidx++;
4335 }
4336 curr = (*mb_ptr2char)(p);
4337 }
4338 else
4339 {
4340 neighbor = str[currIdx - 1];
4341 curr = str[currIdx];
4342 }
4343
4344 if (vim_islower(neighbor) && vim_isupper(curr))
4345 score += CAMEL_BONUS;
4346
Bram Moolenaardcdd42a2020-10-29 18:58:01 +01004347 // Bonus if the match follows a separator character
4348 if (neighbor == '/' || neighbor == '\\')
4349 score += PATH_SEPARATOR_BONUS;
4350 else if (neighbor == ' ' || neighbor == '_')
4351 score += WORD_SEPARATOR_BONUS;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004352 }
4353 else
4354 {
4355 // First letter
4356 score += FIRST_LETTER_BONUS;
4357 }
4358 }
4359 return score;
4360}
4361
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004362/*
4363 * Perform a recursive search for fuzzy matching 'fuzpat' in 'str'.
4364 * Return the number of matching characters.
4365 */
Bram Moolenaar635414d2020-09-11 22:25:15 +02004366 static int
4367fuzzy_match_recursive(
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004368 char_u *fuzpat,
4369 char_u *str,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004370 int_u strIdx,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004371 int *outScore,
4372 char_u *strBegin,
4373 int strLen,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004374 int_u *srcMatches,
4375 int_u *matches,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004376 int maxMatches,
4377 int nextMatch,
4378 int *recursionCount)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004379{
4380 // Recursion params
4381 int recursiveMatch = FALSE;
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004382 int_u bestRecursiveMatches[MAX_FUZZY_MATCHES];
Bram Moolenaar635414d2020-09-11 22:25:15 +02004383 int bestRecursiveScore = 0;
4384 int first_match;
4385 int matched;
4386
4387 // Count recursions
4388 ++*recursionCount;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004389 if (*recursionCount >= FUZZY_MATCH_RECURSION_LIMIT)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004390 return 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004391
4392 // Detect end of strings
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004393 if (*fuzpat == NUL || *str == NUL)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004394 return 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004395
4396 // Loop through fuzpat and str looking for a match
4397 first_match = TRUE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004398 while (*fuzpat != NUL && *str != NUL)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004399 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004400 int c1;
4401 int c2;
4402
4403 c1 = PTR2CHAR(fuzpat);
4404 c2 = PTR2CHAR(str);
4405
Bram Moolenaar635414d2020-09-11 22:25:15 +02004406 // Found match
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004407 if (vim_tolower(c1) == vim_tolower(c2))
Bram Moolenaar635414d2020-09-11 22:25:15 +02004408 {
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004409 int_u recursiveMatches[MAX_FUZZY_MATCHES];
Bram Moolenaar635414d2020-09-11 22:25:15 +02004410 int recursiveScore = 0;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004411 char_u *next_char;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004412
4413 // Supplied matches buffer was too short
4414 if (nextMatch >= maxMatches)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004415 return 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004416
4417 // "Copy-on-Write" srcMatches into matches
4418 if (first_match && srcMatches)
4419 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004420 memcpy(matches, srcMatches, nextMatch * sizeof(srcMatches[0]));
Bram Moolenaar635414d2020-09-11 22:25:15 +02004421 first_match = FALSE;
4422 }
4423
4424 // Recursive call that "skips" this match
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004425 if (has_mbyte)
4426 next_char = str + (*mb_ptr2len)(str);
4427 else
4428 next_char = str + 1;
4429 if (fuzzy_match_recursive(fuzpat, next_char, strIdx + 1,
4430 &recursiveScore, strBegin, strLen, matches,
4431 recursiveMatches,
K.Takataeeec2542021-06-02 13:28:16 +02004432 ARRAY_LENGTH(recursiveMatches),
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004433 nextMatch, recursionCount))
Bram Moolenaar635414d2020-09-11 22:25:15 +02004434 {
4435 // Pick best recursive score
4436 if (!recursiveMatch || recursiveScore > bestRecursiveScore)
4437 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004438 memcpy(bestRecursiveMatches, recursiveMatches,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004439 MAX_FUZZY_MATCHES * sizeof(recursiveMatches[0]));
Bram Moolenaar635414d2020-09-11 22:25:15 +02004440 bestRecursiveScore = recursiveScore;
4441 }
4442 recursiveMatch = TRUE;
4443 }
4444
4445 // Advance
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004446 matches[nextMatch++] = strIdx;
4447 if (has_mbyte)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004448 MB_PTR_ADV(fuzpat);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004449 else
4450 ++fuzpat;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004451 }
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004452 if (has_mbyte)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004453 MB_PTR_ADV(str);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004454 else
4455 ++str;
4456 strIdx++;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004457 }
4458
4459 // Determine if full fuzpat was matched
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004460 matched = *fuzpat == NUL ? TRUE : FALSE;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004461
4462 // Calculate score
4463 if (matched)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004464 *outScore = fuzzy_match_compute_score(strBegin, strLen, matches,
4465 nextMatch);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004466
4467 // Return best result
4468 if (recursiveMatch && (!matched || bestRecursiveScore > *outScore))
4469 {
4470 // Recursive score is better than "this"
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004471 memcpy(matches, bestRecursiveMatches, maxMatches * sizeof(matches[0]));
Bram Moolenaar635414d2020-09-11 22:25:15 +02004472 *outScore = bestRecursiveScore;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004473 return nextMatch;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004474 }
4475 else if (matched)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004476 return nextMatch; // "this" score is better than recursive
Bram Moolenaar635414d2020-09-11 22:25:15 +02004477
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004478 return 0; // no match
Bram Moolenaar635414d2020-09-11 22:25:15 +02004479}
4480
4481/*
4482 * fuzzy_match()
4483 *
4484 * Performs exhaustive search via recursion to find all possible matches and
4485 * match with highest score.
4486 * Scores values have no intrinsic meaning. Possible score range is not
4487 * normalized and varies with pattern.
4488 * Recursion is limited internally (default=10) to prevent degenerate cases
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004489 * (pat_arg="aaaaaa" str="aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa").
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004490 * Uses char_u for match indices. Therefore patterns are limited to
4491 * MAX_FUZZY_MATCHES characters.
Bram Moolenaar635414d2020-09-11 22:25:15 +02004492 *
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004493 * Returns TRUE if 'pat_arg' matches 'str'. Also returns the match score in
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004494 * 'outScore' and the matching character positions in 'matches'.
Bram Moolenaar635414d2020-09-11 22:25:15 +02004495 */
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004496 int
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004497fuzzy_match(
4498 char_u *str,
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004499 char_u *pat_arg,
4500 int matchseq,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004501 int *outScore,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004502 int_u *matches,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004503 int maxMatches)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004504{
Bram Moolenaar635414d2020-09-11 22:25:15 +02004505 int recursionCount = 0;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004506 int len = MB_CHARLEN(str);
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004507 char_u *save_pat;
4508 char_u *pat;
4509 char_u *p;
4510 int complete = FALSE;
4511 int score = 0;
4512 int numMatches = 0;
4513 int matchCount;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004514
4515 *outScore = 0;
4516
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004517 save_pat = vim_strsave(pat_arg);
4518 if (save_pat == NULL)
4519 return FALSE;
4520 pat = save_pat;
4521 p = pat;
4522
4523 // Try matching each word in 'pat_arg' in 'str'
4524 while (TRUE)
4525 {
4526 if (matchseq)
4527 complete = TRUE;
4528 else
4529 {
4530 // Extract one word from the pattern (separated by space)
4531 p = skipwhite(p);
4532 if (*p == NUL)
4533 break;
4534 pat = p;
4535 while (*p != NUL && !VIM_ISWHITE(PTR2CHAR(p)))
4536 {
4537 if (has_mbyte)
4538 MB_PTR_ADV(p);
4539 else
4540 ++p;
4541 }
4542 if (*p == NUL) // processed all the words
4543 complete = TRUE;
4544 *p = NUL;
4545 }
4546
4547 score = 0;
4548 recursionCount = 0;
4549 matchCount = fuzzy_match_recursive(pat, str, 0, &score, str, len, NULL,
4550 matches + numMatches, maxMatches - numMatches,
4551 0, &recursionCount);
4552 if (matchCount == 0)
4553 {
4554 numMatches = 0;
4555 break;
4556 }
4557
4558 // Accumulate the match score and the number of matches
4559 *outScore += score;
4560 numMatches += matchCount;
4561
4562 if (complete)
4563 break;
4564
4565 // try matching the next word
4566 ++p;
4567 }
4568
4569 vim_free(save_pat);
4570 return numMatches != 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004571}
4572
Yegappan Lakshmanan38b85cb2022-02-24 13:28:41 +00004573#if defined(FEAT_EVAL) || defined(FEAT_PROTO)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004574/*
4575 * Sort the fuzzy matches in the descending order of the match score.
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004576 * For items with same score, retain the order using the index (stable sort)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004577 */
4578 static int
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004579fuzzy_match_item_compare(const void *s1, const void *s2)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004580{
4581 int v1 = ((fuzzyItem_T *)s1)->score;
4582 int v2 = ((fuzzyItem_T *)s2)->score;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004583 int idx1 = ((fuzzyItem_T *)s1)->idx;
4584 int idx2 = ((fuzzyItem_T *)s2)->idx;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004585
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004586 return v1 == v2 ? (idx1 - idx2) : v1 > v2 ? -1 : 1;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004587}
4588
4589/*
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004590 * Fuzzy search the string 'str' in a list of 'items' and return the matching
4591 * strings in 'fmatchlist'.
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004592 * If 'matchseq' is TRUE, then for multi-word search strings, match all the
4593 * words in sequence.
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004594 * If 'items' is a list of strings, then search for 'str' in the list.
4595 * If 'items' is a list of dicts, then either use 'key' to lookup the string
4596 * for each item or use 'item_cb' Funcref function to get the string.
4597 * If 'retmatchpos' is TRUE, then return a list of positions where 'str'
4598 * matches for each item.
Bram Moolenaar635414d2020-09-11 22:25:15 +02004599 */
4600 static void
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004601fuzzy_match_in_list(
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004602 list_T *l,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004603 char_u *str,
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004604 int matchseq,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004605 char_u *key,
4606 callback_T *item_cb,
4607 int retmatchpos,
Yasuhiro Matsumoto9029a6e2022-04-16 12:35:35 +01004608 list_T *fmatchlist,
4609 long max_matches)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004610{
4611 long len;
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004612 fuzzyItem_T *items;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004613 listitem_T *li;
4614 long i = 0;
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004615 long match_count = 0;
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004616 int_u matches[MAX_FUZZY_MATCHES];
Bram Moolenaar635414d2020-09-11 22:25:15 +02004617
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004618 len = list_len(l);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004619 if (len == 0)
4620 return;
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004621 if (max_matches > 0 && len > max_matches)
4622 len = max_matches;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004623
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004624 items = ALLOC_CLEAR_MULT(fuzzyItem_T, len);
4625 if (items == NULL)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004626 return;
4627
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004628 // For all the string items in items, get the fuzzy matching score
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004629 FOR_ALL_LIST_ITEMS(l, li)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004630 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004631 int score;
4632 char_u *itemstr;
4633 typval_T rettv;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004634
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004635 if (max_matches > 0 && match_count >= max_matches)
4636 break;
Yasuhiro Matsumoto9029a6e2022-04-16 12:35:35 +01004637
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004638 itemstr = NULL;
4639 rettv.v_type = VAR_UNKNOWN;
4640 if (li->li_tv.v_type == VAR_STRING) // list of strings
4641 itemstr = li->li_tv.vval.v_string;
Dominique Pelle7765f5c2022-04-10 11:26:53 +01004642 else if (li->li_tv.v_type == VAR_DICT
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004643 && (key != NULL || item_cb->cb_name != NULL))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004644 {
4645 // For a dict, either use the specified key to lookup the string or
4646 // use the specified callback function to get the string.
4647 if (key != NULL)
Bram Moolenaard61efa52022-07-23 09:52:04 +01004648 itemstr = dict_get_string(li->li_tv.vval.v_dict,
4649 (char *)key, FALSE);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004650 else
Bram Moolenaar635414d2020-09-11 22:25:15 +02004651 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004652 typval_T argv[2];
4653
4654 // Invoke the supplied callback (if any) to get the dict item
4655 li->li_tv.vval.v_dict->dv_refcount++;
4656 argv[0].v_type = VAR_DICT;
4657 argv[0].vval.v_dict = li->li_tv.vval.v_dict;
4658 argv[1].v_type = VAR_UNKNOWN;
4659 if (call_callback(item_cb, -1, &rettv, 1, argv) != FAIL)
4660 {
4661 if (rettv.v_type == VAR_STRING)
4662 itemstr = rettv.vval.v_string;
4663 }
4664 dict_unref(li->li_tv.vval.v_dict);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004665 }
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004666 }
4667
4668 if (itemstr != NULL
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004669 && fuzzy_match(itemstr, str, matchseq, &score, matches,
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004670 MAX_FUZZY_MATCHES))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004671 {
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004672 items[match_count].idx = match_count;
4673 items[match_count].item = li;
4674 items[match_count].score = score;
4675
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004676 // Copy the list of matching positions in itemstr to a list, if
4677 // 'retmatchpos' is set.
4678 if (retmatchpos)
4679 {
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004680 int j = 0;
4681 char_u *p;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004682
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004683 items[match_count].lmatchpos = list_alloc();
4684 if (items[match_count].lmatchpos == NULL)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004685 goto done;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004686
4687 p = str;
4688 while (*p != NUL)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004689 {
zeertzjq9af2bc02022-05-11 14:15:37 +01004690 if (!VIM_ISWHITE(PTR2CHAR(p)) || matchseq)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004691 {
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004692 if (list_append_number(items[match_count].lmatchpos,
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004693 matches[j]) == FAIL)
4694 goto done;
4695 j++;
4696 }
4697 if (has_mbyte)
4698 MB_PTR_ADV(p);
4699 else
4700 ++p;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004701 }
4702 }
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004703 ++match_count;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004704 }
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004705 clear_tv(&rettv);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004706 }
4707
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004708 if (match_count > 0)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004709 {
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004710 list_T *retlist;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004711
Bram Moolenaar635414d2020-09-11 22:25:15 +02004712 // Sort the list by the descending order of the match score
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004713 qsort((void *)items, (size_t)match_count, sizeof(fuzzyItem_T),
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004714 fuzzy_match_item_compare);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004715
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004716 // For matchfuzzy(), return a list of matched strings.
4717 // ['str1', 'str2', 'str3']
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004718 // For matchfuzzypos(), return a list with three items.
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004719 // The first item is a list of matched strings. The second item
4720 // is a list of lists where each list item is a list of matched
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004721 // character positions. The third item is a list of matching scores.
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004722 // [['str1', 'str2', 'str3'], [[1, 3], [1, 3], [1, 3]]]
4723 if (retmatchpos)
4724 {
4725 li = list_find(fmatchlist, 0);
4726 if (li == NULL || li->li_tv.vval.v_list == NULL)
4727 goto done;
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004728 retlist = li->li_tv.vval.v_list;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004729 }
4730 else
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004731 retlist = fmatchlist;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004732
4733 // Copy the matching strings with a valid score to the return list
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004734 for (i = 0; i < match_count; i++)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004735 {
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004736 if (items[i].score == SCORE_NONE)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004737 break;
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004738 list_append_tv(retlist, &items[i].item->li_tv);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004739 }
4740
4741 // next copy the list of matching positions
4742 if (retmatchpos)
4743 {
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004744 li = list_find(fmatchlist, -2);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004745 if (li == NULL || li->li_tv.vval.v_list == NULL)
4746 goto done;
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004747 retlist = li->li_tv.vval.v_list;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004748
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004749 for (i = 0; i < match_count; i++)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004750 {
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004751 if (items[i].score == SCORE_NONE)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004752 break;
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004753 if (items[i].lmatchpos != NULL
4754 && list_append_list(retlist, items[i].lmatchpos)
4755 == FAIL)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004756 goto done;
4757 }
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004758
4759 // copy the matching scores
4760 li = list_find(fmatchlist, -1);
4761 if (li == NULL || li->li_tv.vval.v_list == NULL)
4762 goto done;
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004763 retlist = li->li_tv.vval.v_list;
4764 for (i = 0; i < match_count; i++)
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004765 {
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004766 if (items[i].score == SCORE_NONE)
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004767 break;
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004768 if (list_append_number(retlist, items[i].score) == FAIL)
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004769 goto done;
4770 }
Bram Moolenaar635414d2020-09-11 22:25:15 +02004771 }
4772 }
4773
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004774done:
Yegappan Lakshmanan047a7012022-04-16 20:42:40 +01004775 vim_free(items);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004776}
4777
4778/*
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004779 * Do fuzzy matching. Returns the list of matched strings in 'rettv'.
4780 * If 'retmatchpos' is TRUE, also returns the matching character positions.
4781 */
4782 static void
4783do_fuzzymatch(typval_T *argvars, typval_T *rettv, int retmatchpos)
4784{
4785 callback_T cb;
4786 char_u *key = NULL;
4787 int ret;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004788 int matchseq = FALSE;
Yasuhiro Matsumoto9029a6e2022-04-16 12:35:35 +01004789 long max_matches = 0;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004790
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02004791 if (in_vim9script()
4792 && (check_for_list_arg(argvars, 0) == FAIL
4793 || check_for_string_arg(argvars, 1) == FAIL
4794 || check_for_opt_dict_arg(argvars, 2) == FAIL))
4795 return;
4796
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004797 CLEAR_POINTER(&cb);
4798
4799 // validate and get the arguments
4800 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
4801 {
Bram Moolenaar3a846e62022-01-01 16:21:00 +00004802 semsg(_(e_argument_of_str_must_be_list),
4803 retmatchpos ? "matchfuzzypos()" : "matchfuzzy()");
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004804 return;
4805 }
4806 if (argvars[1].v_type != VAR_STRING
4807 || argvars[1].vval.v_string == NULL)
4808 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00004809 semsg(_(e_invalid_argument_str), tv_get_string(&argvars[1]));
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004810 return;
4811 }
4812
4813 if (argvars[2].v_type != VAR_UNKNOWN)
4814 {
4815 dict_T *d;
4816 dictitem_T *di;
4817
4818 if (argvars[2].v_type != VAR_DICT || argvars[2].vval.v_dict == NULL)
4819 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00004820 emsg(_(e_dictionary_required));
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004821 return;
4822 }
4823
4824 // To search a dict, either a callback function or a key can be
4825 // specified.
4826 d = argvars[2].vval.v_dict;
4827 if ((di = dict_find(d, (char_u *)"key", -1)) != NULL)
4828 {
4829 if (di->di_tv.v_type != VAR_STRING
4830 || di->di_tv.vval.v_string == NULL
4831 || *di->di_tv.vval.v_string == NUL)
4832 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00004833 semsg(_(e_invalid_argument_str), tv_get_string(&di->di_tv));
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004834 return;
4835 }
4836 key = tv_get_string(&di->di_tv);
4837 }
4838 else if ((di = dict_find(d, (char_u *)"text_cb", -1)) != NULL)
4839 {
4840 cb = get_callback(&di->di_tv);
4841 if (cb.cb_name == NULL)
4842 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00004843 semsg(_(e_invalid_value_for_argument_str), "text_cb");
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004844 return;
4845 }
4846 }
Kazuyuki Miyagi47f1a552022-06-17 18:30:03 +01004847
4848 if ((di = dict_find(d, (char_u *)"limit", -1)) != NULL)
Yasuhiro Matsumoto9029a6e2022-04-16 12:35:35 +01004849 {
4850 if (di->di_tv.v_type != VAR_NUMBER)
4851 {
4852 semsg(_(e_invalid_argument_str), tv_get_string(&di->di_tv));
4853 return;
4854 }
4855 max_matches = (long)tv_get_number_chk(&di->di_tv, NULL);
4856 }
4857
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01004858 if (dict_has_key(d, "matchseq"))
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004859 matchseq = TRUE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004860 }
4861
4862 // get the fuzzy matches
4863 ret = rettv_list_alloc(rettv);
Bram Moolenaar5ea38d12022-06-16 21:20:48 +01004864 if (ret == FAIL)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004865 goto done;
4866 if (retmatchpos)
4867 {
4868 list_T *l;
4869
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004870 // For matchfuzzypos(), a list with three items are returned. First
4871 // item is a list of matching strings, the second item is a list of
4872 // lists with matching positions within each string and the third item
4873 // is the list of scores of the matches.
4874 l = list_alloc();
4875 if (l == NULL)
4876 goto done;
4877 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4878 goto done;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004879 l = list_alloc();
4880 if (l == NULL)
4881 goto done;
4882 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4883 goto done;
4884 l = list_alloc();
4885 if (l == NULL)
4886 goto done;
4887 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4888 goto done;
4889 }
4890
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004891 fuzzy_match_in_list(argvars[0].vval.v_list, tv_get_string(&argvars[1]),
Yasuhiro Matsumoto9029a6e2022-04-16 12:35:35 +01004892 matchseq, key, &cb, retmatchpos, rettv->vval.v_list, max_matches);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004893
4894done:
4895 free_callback(&cb);
4896}
4897
4898/*
Bram Moolenaar635414d2020-09-11 22:25:15 +02004899 * "matchfuzzy()" function
4900 */
4901 void
4902f_matchfuzzy(typval_T *argvars, typval_T *rettv)
4903{
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004904 do_fuzzymatch(argvars, rettv, FALSE);
4905}
4906
4907/*
4908 * "matchfuzzypos()" function
4909 */
4910 void
4911f_matchfuzzypos(typval_T *argvars, typval_T *rettv)
4912{
4913 do_fuzzymatch(argvars, rettv, TRUE);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004914}
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004915#endif
Yegappan Lakshmanan38b85cb2022-02-24 13:28:41 +00004916
4917/*
4918 * Same as fuzzy_match_item_compare() except for use with a string match
4919 */
4920 static int
4921fuzzy_match_str_compare(const void *s1, const void *s2)
4922{
4923 int v1 = ((fuzmatch_str_T *)s1)->score;
4924 int v2 = ((fuzmatch_str_T *)s2)->score;
4925 int idx1 = ((fuzmatch_str_T *)s1)->idx;
4926 int idx2 = ((fuzmatch_str_T *)s2)->idx;
4927
4928 return v1 == v2 ? (idx1 - idx2) : v1 > v2 ? -1 : 1;
4929}
4930
4931/*
4932 * Sort fuzzy matches by score
4933 */
4934 static void
4935fuzzy_match_str_sort(fuzmatch_str_T *fm, int sz)
4936{
4937 // Sort the list by the descending order of the match score
4938 qsort((void *)fm, (size_t)sz, sizeof(fuzmatch_str_T),
4939 fuzzy_match_str_compare);
4940}
4941
4942/*
4943 * Same as fuzzy_match_item_compare() except for use with a function name
4944 * string match. <SNR> functions should be sorted to the end.
4945 */
4946 static int
4947fuzzy_match_func_compare(const void *s1, const void *s2)
4948{
4949 int v1 = ((fuzmatch_str_T *)s1)->score;
4950 int v2 = ((fuzmatch_str_T *)s2)->score;
4951 int idx1 = ((fuzmatch_str_T *)s1)->idx;
4952 int idx2 = ((fuzmatch_str_T *)s2)->idx;
4953 char_u *str1 = ((fuzmatch_str_T *)s1)->str;
4954 char_u *str2 = ((fuzmatch_str_T *)s2)->str;
4955
4956 if (*str1 != '<' && *str2 == '<') return -1;
4957 if (*str1 == '<' && *str2 != '<') return 1;
4958 return v1 == v2 ? (idx1 - idx2) : v1 > v2 ? -1 : 1;
4959}
4960
4961/*
4962 * Sort fuzzy matches of function names by score.
4963 * <SNR> functions should be sorted to the end.
4964 */
4965 static void
4966fuzzy_match_func_sort(fuzmatch_str_T *fm, int sz)
4967{
4968 // Sort the list by the descending order of the match score
4969 qsort((void *)fm, (size_t)sz, sizeof(fuzmatch_str_T),
4970 fuzzy_match_func_compare);
4971}
4972
4973/*
4974 * Fuzzy match 'pat' in 'str'. Returns 0 if there is no match. Otherwise,
4975 * returns the match score.
4976 */
4977 int
4978fuzzy_match_str(char_u *str, char_u *pat)
4979{
4980 int score = 0;
Yegappan Lakshmanan5ec633b2022-02-25 15:24:24 +00004981 int_u matchpos[MAX_FUZZY_MATCHES];
Yegappan Lakshmanan38b85cb2022-02-24 13:28:41 +00004982
4983 if (str == NULL || pat == NULL)
4984 return 0;
4985
Yegappan Lakshmanan6caeda22022-02-27 12:07:30 +00004986 fuzzy_match(str, pat, TRUE, &score, matchpos,
Yegappan Lakshmanan38b85cb2022-02-24 13:28:41 +00004987 sizeof(matchpos) / sizeof(matchpos[0]));
4988
4989 return score;
4990}
4991
4992/*
Bram Moolenaarc6e0a5e2022-04-10 18:09:06 +01004993 * Free an array of fuzzy string matches "fuzmatch[count]".
4994 */
4995 void
4996fuzmatch_str_free(fuzmatch_str_T *fuzmatch, int count)
4997{
4998 int i;
4999
5000 if (fuzmatch == NULL)
5001 return;
5002 for (i = 0; i < count; ++i)
5003 vim_free(fuzmatch[i].str);
5004 vim_free(fuzmatch);
5005}
5006
5007/*
Yegappan Lakshmanan38b85cb2022-02-24 13:28:41 +00005008 * Copy a list of fuzzy matches into a string list after sorting the matches by
5009 * the fuzzy score. Frees the memory allocated for 'fuzmatch'.
5010 * Returns OK on success and FAIL on memory allocation failure.
5011 */
5012 int
5013fuzzymatches_to_strmatches(
5014 fuzmatch_str_T *fuzmatch,
5015 char_u ***matches,
5016 int count,
5017 int funcsort)
5018{
5019 int i;
5020
5021 if (count <= 0)
5022 return OK;
5023
5024 *matches = ALLOC_MULT(char_u *, count);
5025 if (*matches == NULL)
5026 {
Bram Moolenaarc6e0a5e2022-04-10 18:09:06 +01005027 fuzmatch_str_free(fuzmatch, count);
Yegappan Lakshmanan38b85cb2022-02-24 13:28:41 +00005028 return FAIL;
5029 }
5030
5031 // Sort the list by the descending order of the match score
5032 if (funcsort)
5033 fuzzy_match_func_sort((void *)fuzmatch, (size_t)count);
5034 else
5035 fuzzy_match_str_sort((void *)fuzmatch, (size_t)count);
5036
5037 for (i = 0; i < count; i++)
5038 (*matches)[i] = fuzmatch[i].str;
5039 vim_free(fuzmatch);
5040
5041 return OK;
5042}