blob: 920c68c5aae1e2cfd5bc1ff29e019afdc8aa4984 [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 Moolenaarbaaa7e92016-01-29 22:47:03 +010019static int check_linecomment(char_u *line);
Bram Moolenaar071d4272004-06-13 20:20:40 +000020#ifdef FEAT_FIND_ID
Bram Moolenaarbaaa7e92016-01-29 22:47:03 +010021static void show_pat_in_path(char_u *, int,
22 int, int, FILE *, linenr_T *, long);
Bram Moolenaar071d4272004-06-13 20:20:40 +000023#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +020024
25typedef struct searchstat
26{
27 int cur; // current position of found words
28 int cnt; // total count of found words
29 int exact_match; // TRUE if matched exactly on specified position
30 int incomplete; // 0: search was fully completed
31 // 1: recomputing was timed out
32 // 2: max count exceeded
33 int last_maxcount; // the max count of the last search
34} searchstat_T;
35
36static 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);
37static void update_search_stat(int dirc, pos_T *pos, pos_T *cursor_pos, searchstat_T *stat, int recompute, int maxcount, long timeout);
38
Bram Moolenaarea6561a2020-06-01 21:32:45 +020039#define SEARCH_STAT_DEF_TIMEOUT 40L
Bram Moolenaare8f5ec02020-06-01 17:28:35 +020040#define SEARCH_STAT_DEF_MAX_COUNT 99
41#define SEARCH_STAT_BUF_LEN 12
Bram Moolenaar071d4272004-06-13 20:20:40 +000042
Bram Moolenaar071d4272004-06-13 20:20:40 +000043/*
44 * This file contains various searching-related routines. These fall into
45 * three groups:
46 * 1. string searches (for /, ?, n, and N)
47 * 2. character searches within a single line (for f, F, t, T, etc)
48 * 3. "other" kinds of searches like the '%' command, and 'word' searches.
49 */
50
51/*
52 * String searches
53 *
54 * The string search functions are divided into two levels:
55 * lowest: searchit(); uses an pos_T for starting position and found match.
56 * Highest: do_search(); uses curwin->w_cursor; calls searchit().
57 *
58 * The last search pattern is remembered for repeating the same search.
59 * This pattern is shared between the :g, :s, ? and / commands.
60 * This is in search_regcomp().
61 *
62 * The actual string matching is done using a heavily modified version of
63 * Henry Spencer's regular expression library. See regexp.c.
64 */
65
Bram Moolenaar071d4272004-06-13 20:20:40 +000066/*
67 * Two search patterns are remembered: One for the :substitute command and
68 * one for other searches. last_idx points to the one that was used the last
69 * time.
70 */
Bram Moolenaarc3328162019-07-23 22:15:25 +020071static spat_T spats[2] =
Bram Moolenaar071d4272004-06-13 20:20:40 +000072{
Bram Moolenaar63d9e732019-12-05 21:10:38 +010073 {NULL, TRUE, FALSE, {'/', 0, 0, 0L}}, // last used search pat
74 {NULL, TRUE, FALSE, {'/', 0, 0, 0L}} // last used substitute pat
Bram Moolenaar071d4272004-06-13 20:20:40 +000075};
76
Bram Moolenaar63d9e732019-12-05 21:10:38 +010077static int last_idx = 0; // index in spats[] for RE_LAST
Bram Moolenaar071d4272004-06-13 20:20:40 +000078
Bram Moolenaar63d9e732019-12-05 21:10:38 +010079static char_u lastc[2] = {NUL, NUL}; // last character searched for
80static int lastcdir = FORWARD; // last direction of character search
81static int last_t_cmd = TRUE; // last search t_cmd
Bram Moolenaardbd24b52015-08-11 14:26:19 +020082static char_u lastc_bytes[MB_MAXBYTES + 1];
Bram Moolenaar63d9e732019-12-05 21:10:38 +010083static int lastc_bytelen = 1; // >1 for multi-byte char
Bram Moolenaardbd24b52015-08-11 14:26:19 +020084
Bram Moolenaar63d9e732019-12-05 21:10:38 +010085// copy of spats[], for keeping the search patterns while executing autocmds
Bram Moolenaarc3328162019-07-23 22:15:25 +020086static spat_T saved_spats[2];
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 Moolenaar63d9e732019-12-05 21:10:38 +010092static char_u *mr_pattern = NULL; // pattern used by search_regcomp()
Bram Moolenaar071d4272004-06-13 20:20:40 +000093#ifdef FEAT_RIGHTLEFT
Bram Moolenaar63d9e732019-12-05 21:10:38 +010094static int mr_pattern_alloced = FALSE; // mr_pattern was allocated
Bram Moolenaar071d4272004-06-13 20:20:40 +000095#endif
96
97#ifdef FEAT_FIND_ID
98/*
99 * Type used by find_pattern_in_path() to remember which included files have
100 * been searched already.
101 */
102typedef struct SearchedFile
103{
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100104 FILE *fp; // File pointer
105 char_u *name; // Full name of file
106 linenr_T lnum; // Line we were up to in file
107 int matched; // Found a match in this file
Bram Moolenaar071d4272004-06-13 20:20:40 +0000108} SearchedFile;
109#endif
110
111/*
112 * translate search pattern for vim_regcomp()
113 *
114 * pat_save == RE_SEARCH: save pat in spats[RE_SEARCH].pat (normal search cmd)
115 * pat_save == RE_SUBST: save pat in spats[RE_SUBST].pat (:substitute command)
116 * pat_save == RE_BOTH: save pat in both patterns (:global command)
117 * pat_use == RE_SEARCH: use previous search pattern if "pat" is NULL
Bram Moolenaarb8017e72007-05-10 18:59:07 +0000118 * pat_use == RE_SUBST: use previous substitute pattern if "pat" is NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000119 * pat_use == RE_LAST: use last used pattern if "pat" is NULL
120 * options & SEARCH_HIS: put search string in history
121 * options & SEARCH_KEEP: keep previous search pattern
122 *
123 * returns FAIL if failed, OK otherwise.
124 */
125 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100126search_regcomp(
127 char_u *pat,
128 int pat_save,
129 int pat_use,
130 int options,
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100131 regmmatch_T *regmatch) // return: pattern and ignore-case flag
Bram Moolenaar071d4272004-06-13 20:20:40 +0000132{
133 int magic;
134 int i;
135
136 rc_did_emsg = FALSE;
Bram Moolenaarf4e20992020-12-21 19:59:08 +0100137 magic = magic_isset();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000138
139 /*
140 * If no pattern given, use a previously defined pattern.
141 */
142 if (pat == NULL || *pat == NUL)
143 {
144 if (pat_use == RE_LAST)
145 i = last_idx;
146 else
147 i = pat_use;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100148 if (spats[i].pat == NULL) // pattern was never defined
Bram Moolenaar071d4272004-06-13 20:20:40 +0000149 {
150 if (pat_use == RE_SUBST)
Bram Moolenaare29a27f2021-07-20 21:07:36 +0200151 emsg(_(e_no_previous_substitute_regular_expression));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152 else
Bram Moolenaare29a27f2021-07-20 21:07:36 +0200153 emsg(_(e_no_previous_regular_expression));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000154 rc_did_emsg = TRUE;
155 return FAIL;
156 }
157 pat = spats[i].pat;
158 magic = spats[i].magic;
159 no_smartcase = spats[i].no_scs;
160 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100161 else if (options & SEARCH_HIS) // put new pattern in history
Bram Moolenaar071d4272004-06-13 20:20:40 +0000162 add_to_history(HIST_SEARCH, pat, TRUE, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000163
164#ifdef FEAT_RIGHTLEFT
165 if (mr_pattern_alloced)
166 {
167 vim_free(mr_pattern);
168 mr_pattern_alloced = FALSE;
169 }
170
171 if (curwin->w_p_rl && *curwin->w_p_rlc == 's')
172 {
173 char_u *rev_pattern;
174
175 rev_pattern = reverse_text(pat);
176 if (rev_pattern == NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100177 mr_pattern = pat; // out of memory, keep normal pattern.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000178 else
179 {
180 mr_pattern = rev_pattern;
181 mr_pattern_alloced = TRUE;
182 }
183 }
184 else
185#endif
186 mr_pattern = pat;
187
188 /*
189 * Save the currently used pattern in the appropriate place,
190 * unless the pattern should not be remembered.
191 */
Bram Moolenaare1004402020-10-24 20:49:43 +0200192 if (!(options & SEARCH_KEEP)
193 && (cmdmod.cmod_flags & CMOD_KEEPPATTERNS) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000194 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100195 // search or global command
Bram Moolenaar071d4272004-06-13 20:20:40 +0000196 if (pat_save == RE_SEARCH || pat_save == RE_BOTH)
197 save_re_pat(RE_SEARCH, pat, magic);
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100198 // substitute or global command
Bram Moolenaar071d4272004-06-13 20:20:40 +0000199 if (pat_save == RE_SUBST || pat_save == RE_BOTH)
200 save_re_pat(RE_SUBST, pat, magic);
201 }
202
203 regmatch->rmm_ic = ignorecase(pat);
Bram Moolenaar3b56eb32005-07-11 22:40:32 +0000204 regmatch->rmm_maxcol = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000205 regmatch->regprog = vim_regcomp(pat, magic ? RE_MAGIC : 0);
206 if (regmatch->regprog == NULL)
207 return FAIL;
208 return OK;
209}
210
211/*
212 * Get search pattern used by search_regcomp().
213 */
214 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100215get_search_pat(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000216{
217 return mr_pattern;
218}
219
Bram Moolenaarabc97732007-08-08 20:49:37 +0000220#if defined(FEAT_RIGHTLEFT) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000221/*
222 * Reverse text into allocated memory.
223 * Returns the allocated string, NULL when out of memory.
224 */
Bram Moolenaarabc97732007-08-08 20:49:37 +0000225 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100226reverse_text(char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000227{
228 unsigned len;
229 unsigned s_i, rev_i;
230 char_u *rev;
231
232 /*
233 * Reverse the pattern.
234 */
235 len = (unsigned)STRLEN(s);
236 rev = alloc(len + 1);
237 if (rev != NULL)
238 {
239 rev_i = len;
240 for (s_i = 0; s_i < len; ++s_i)
241 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000242 if (has_mbyte)
243 {
244 int mb_len;
245
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000246 mb_len = (*mb_ptr2len)(s + s_i);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000247 rev_i -= mb_len;
248 mch_memmove(rev + rev_i, s + s_i, mb_len);
249 s_i += mb_len - 1;
250 }
251 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000252 rev[--rev_i] = s[s_i];
253
254 }
255 rev[len] = NUL;
256 }
257 return rev;
258}
259#endif
260
Bram Moolenaarcc2b9d52014-12-13 03:17:11 +0100261 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100262save_re_pat(int idx, char_u *pat, int magic)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000263{
264 if (spats[idx].pat != pat)
265 {
266 vim_free(spats[idx].pat);
267 spats[idx].pat = vim_strsave(pat);
268 spats[idx].magic = magic;
269 spats[idx].no_scs = no_smartcase;
270 last_idx = idx;
271#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100272 // If 'hlsearch' set and search pat changed: need redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000273 if (p_hls)
Bram Moolenaar1c8f93f2006-03-12 22:10:07 +0000274 redraw_all_later(SOME_VALID);
Bram Moolenaar451fc7b2018-04-27 22:53:07 +0200275 set_no_hlsearch(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000276#endif
277 }
278}
279
Bram Moolenaar071d4272004-06-13 20:20:40 +0000280/*
281 * Save the search patterns, so they can be restored later.
282 * Used before/after executing autocommands and user functions.
283 */
284static int save_level = 0;
285
286 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100287save_search_patterns(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000288{
289 if (save_level++ == 0)
290 {
291 saved_spats[0] = spats[0];
292 if (spats[0].pat != NULL)
293 saved_spats[0].pat = vim_strsave(spats[0].pat);
294 saved_spats[1] = spats[1];
295 if (spats[1].pat != NULL)
296 saved_spats[1].pat = vim_strsave(spats[1].pat);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100297#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100298 saved_spats_last_idx = last_idx;
299 saved_spats_no_hlsearch = no_hlsearch;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100300#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000301 }
302}
303
304 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100305restore_search_patterns(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000306{
307 if (--save_level == 0)
308 {
309 vim_free(spats[0].pat);
310 spats[0] = saved_spats[0];
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100311#if defined(FEAT_EVAL)
Bram Moolenaar8c8de832008-06-24 22:58:06 +0000312 set_vv_searchforward();
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100313#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000314 vim_free(spats[1].pat);
315 spats[1] = saved_spats[1];
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100316#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100317 last_idx = saved_spats_last_idx;
318 set_no_hlsearch(saved_spats_no_hlsearch);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100319#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000320 }
321}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000322
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000323#if defined(EXITFREE) || defined(PROTO)
324 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100325free_search_patterns(void)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000326{
327 vim_free(spats[0].pat);
328 vim_free(spats[1].pat);
Bram Moolenaarf2427622009-04-22 16:45:21 +0000329
330# ifdef FEAT_RIGHTLEFT
331 if (mr_pattern_alloced)
332 {
Bram Moolenaarcc448b32010-07-14 16:52:17 +0200333 vim_free(mr_pattern);
334 mr_pattern_alloced = FALSE;
335 mr_pattern = NULL;
Bram Moolenaarf2427622009-04-22 16:45:21 +0000336 }
337# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000338}
339#endif
340
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100341#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100342// copy of spats[RE_SEARCH], for keeping the search patterns while incremental
343// searching
Bram Moolenaarc3328162019-07-23 22:15:25 +0200344static spat_T saved_last_search_spat;
Bram Moolenaared8bc782018-12-01 21:08:21 +0100345static int did_save_last_search_spat = 0;
346static int saved_last_idx = 0;
347static int saved_no_hlsearch = 0;
348
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100349/*
350 * Save and restore the search pattern for incremental highlight search
351 * feature.
352 *
Bram Moolenaarc4568ab2018-11-16 16:21:05 +0100353 * It's similar to but different from save_search_patterns() and
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100354 * restore_search_patterns(), because the search pattern must be restored when
Bram Moolenaarc4568ab2018-11-16 16:21:05 +0100355 * canceling incremental searching even if it's called inside user functions.
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100356 */
357 void
358save_last_search_pattern(void)
359{
Bram Moolenaar442a8532020-06-04 20:56:09 +0200360 if (++did_save_last_search_spat != 1)
361 // nested call, nothing to do
362 return;
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100363
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100364 saved_last_search_spat = spats[RE_SEARCH];
365 if (spats[RE_SEARCH].pat != NULL)
366 saved_last_search_spat.pat = vim_strsave(spats[RE_SEARCH].pat);
367 saved_last_idx = last_idx;
368 saved_no_hlsearch = no_hlsearch;
369}
370
371 void
372restore_last_search_pattern(void)
373{
Bram Moolenaar442a8532020-06-04 20:56:09 +0200374 if (--did_save_last_search_spat > 0)
375 // nested call, nothing to do
376 return;
377 if (did_save_last_search_spat != 0)
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100378 {
Bram Moolenaar442a8532020-06-04 20:56:09 +0200379 iemsg("restore_last_search_pattern() called more often than save_last_search_pattern()");
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100380 return;
381 }
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100382
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100383 vim_free(spats[RE_SEARCH].pat);
384 spats[RE_SEARCH] = saved_last_search_spat;
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100385 saved_last_search_spat.pat = NULL;
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100386# if defined(FEAT_EVAL)
387 set_vv_searchforward();
388# endif
389 last_idx = saved_last_idx;
Bram Moolenaar451fc7b2018-04-27 22:53:07 +0200390 set_no_hlsearch(saved_no_hlsearch);
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100391}
Bram Moolenaard0480092017-11-16 22:20:39 +0100392
393 char_u *
394last_search_pattern(void)
395{
396 return spats[RE_SEARCH].pat;
397}
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100398#endif
399
Bram Moolenaar071d4272004-06-13 20:20:40 +0000400/*
401 * Return TRUE when case should be ignored for search pattern "pat".
402 * Uses the 'ignorecase' and 'smartcase' options.
403 */
404 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100405ignorecase(char_u *pat)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000406{
Bram Moolenaar66e29d72016-08-20 16:57:02 +0200407 return ignorecase_opt(pat, p_ic, p_scs);
408}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000409
Bram Moolenaar66e29d72016-08-20 16:57:02 +0200410/*
411 * As ignorecase() put pass the "ic" and "scs" flags.
412 */
413 int
414ignorecase_opt(char_u *pat, int ic_in, int scs)
415{
416 int ic = ic_in;
417
418 if (ic && !no_smartcase && scs
Bram Moolenaare2c453d2019-08-21 14:37:09 +0200419 && !(ctrl_x_mode_not_default() && curbuf->b_p_inf))
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200420 ic = !pat_has_uppercase(pat);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000421 no_smartcase = FALSE;
422
423 return ic;
424}
425
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200426/*
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200427 * Return TRUE if pattern "pat" has an uppercase character.
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200428 */
429 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100430pat_has_uppercase(char_u *pat)
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200431{
432 char_u *p = pat;
433
434 while (*p != NUL)
435 {
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200436 int l;
437
438 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
439 {
440 if (enc_utf8 && utf_isupper(utf_ptr2char(p)))
441 return TRUE;
442 p += l;
443 }
Bram Moolenaar264b74f2019-01-24 17:18:42 +0100444 else if (*p == '\\')
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200445 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100446 if (p[1] == '_' && p[2] != NUL) // skip "\_X"
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200447 p += 3;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100448 else if (p[1] == '%' && p[2] != NUL) // skip "\%X"
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200449 p += 3;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100450 else if (p[1] != NUL) // skip "\X"
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200451 p += 2;
452 else
453 p += 1;
454 }
455 else if (MB_ISUPPER(*p))
456 return TRUE;
457 else
458 ++p;
459 }
460 return FALSE;
461}
462
Bram Moolenaar113e1072019-01-20 15:30:40 +0100463#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000464 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100465last_csearch(void)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200466{
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200467 return lastc_bytes;
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200468}
469
470 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100471last_csearch_forward(void)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200472{
473 return lastcdir == FORWARD;
474}
475
476 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100477last_csearch_until(void)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200478{
479 return last_t_cmd == TRUE;
480}
481
482 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100483set_last_csearch(int c, char_u *s UNUSED, int len UNUSED)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200484{
485 *lastc = c;
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200486 lastc_bytelen = len;
487 if (len)
488 memcpy(lastc_bytes, s, len);
489 else
Bram Moolenaara80faa82020-04-12 19:37:17 +0200490 CLEAR_FIELD(lastc_bytes);
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200491}
Bram Moolenaar113e1072019-01-20 15:30:40 +0100492#endif
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200493
494 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100495set_csearch_direction(int cdir)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200496{
497 lastcdir = cdir;
498}
499
500 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100501set_csearch_until(int t_cmd)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200502{
503 last_t_cmd = t_cmd;
504}
505
506 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100507last_search_pat(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000508{
509 return spats[last_idx].pat;
510}
511
512/*
513 * Reset search direction to forward. For "gd" and "gD" commands.
514 */
515 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100516reset_search_dir(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000517{
518 spats[0].off.dir = '/';
Bram Moolenaar8c8de832008-06-24 22:58:06 +0000519#if defined(FEAT_EVAL)
520 set_vv_searchforward();
521#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000522}
523
524#if defined(FEAT_EVAL) || defined(FEAT_VIMINFO)
525/*
526 * Set the last search pattern. For ":let @/ =" and viminfo.
527 * Also set the saved search pattern, so that this works in an autocommand.
528 */
529 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100530set_last_search_pat(
531 char_u *s,
532 int idx,
533 int magic,
534 int setlast)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000535{
536 vim_free(spats[idx].pat);
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100537 // An empty string means that nothing should be matched.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000538 if (*s == NUL)
539 spats[idx].pat = NULL;
540 else
541 spats[idx].pat = vim_strsave(s);
542 spats[idx].magic = magic;
543 spats[idx].no_scs = FALSE;
544 spats[idx].off.dir = '/';
Bram Moolenaar8c8de832008-06-24 22:58:06 +0000545#if defined(FEAT_EVAL)
546 set_vv_searchforward();
547#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000548 spats[idx].off.line = FALSE;
549 spats[idx].off.end = FALSE;
550 spats[idx].off.off = 0;
551 if (setlast)
552 last_idx = idx;
553 if (save_level)
554 {
555 vim_free(saved_spats[idx].pat);
556 saved_spats[idx] = spats[0];
557 if (spats[idx].pat == NULL)
558 saved_spats[idx].pat = NULL;
559 else
560 saved_spats[idx].pat = vim_strsave(spats[idx].pat);
Bram Moolenaar975880b2019-03-03 14:42:11 +0100561# ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100562 saved_spats_last_idx = last_idx;
Bram Moolenaar975880b2019-03-03 14:42:11 +0100563# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000564 }
565# ifdef FEAT_SEARCH_EXTRA
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100566 // If 'hlsearch' set and search pat changed: need redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000567 if (p_hls && idx == last_idx && !no_hlsearch)
Bram Moolenaar1c8f93f2006-03-12 22:10:07 +0000568 redraw_all_later(SOME_VALID);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000569# endif
570}
571#endif
572
573#ifdef FEAT_SEARCH_EXTRA
574/*
575 * Get a regexp program for the last used search pattern.
576 * This is used for highlighting all matches in a window.
577 * Values returned in regmatch->regprog and regmatch->rmm_ic.
578 */
579 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100580last_pat_prog(regmmatch_T *regmatch)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000581{
582 if (spats[last_idx].pat == NULL)
583 {
584 regmatch->regprog = NULL;
585 return;
586 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100587 ++emsg_off; // So it doesn't beep if bad expr
Bram Moolenaar071d4272004-06-13 20:20:40 +0000588 (void)search_regcomp((char_u *)"", 0, last_idx, SEARCH_KEEP, regmatch);
589 --emsg_off;
590}
591#endif
592
593/*
Bram Moolenaarf7ff6e82014-03-23 15:13:05 +0100594 * Lowest level search function.
Bram Moolenaar5d24a222018-12-23 19:10:09 +0100595 * Search for 'count'th occurrence of pattern "pat" in direction "dir".
596 * Start at position "pos" and return the found position in "pos".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000597 *
598 * if (options & SEARCH_MSG) == 0 don't give any messages
599 * if (options & SEARCH_MSG) == SEARCH_NFMSG don't give 'notfound' messages
600 * if (options & SEARCH_MSG) == SEARCH_MSG give all messages
601 * if (options & SEARCH_HIS) put search pattern in history
602 * if (options & SEARCH_END) return position at end of match
603 * if (options & SEARCH_START) accept match at pos itself
604 * if (options & SEARCH_KEEP) keep previous search pattern
605 * if (options & SEARCH_FOLD) match only once in a closed fold
606 * if (options & SEARCH_PEEK) check for typed char, cancel search
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100607 * if (options & SEARCH_COL) start at pos->col instead of zero
Bram Moolenaar071d4272004-06-13 20:20:40 +0000608 *
609 * Return FAIL (zero) for failure, non-zero for success.
610 * When FEAT_EVAL is defined, returns the index of the first matching
611 * subpattern plus one; one if there was none.
612 */
613 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100614searchit(
Bram Moolenaar92ea26b2019-10-18 20:53:34 +0200615 win_T *win, // window to search in; can be NULL for a
616 // buffer without a window!
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100617 buf_T *buf,
618 pos_T *pos,
Bram Moolenaar5d24a222018-12-23 19:10:09 +0100619 pos_T *end_pos, // set to end of the match, unless NULL
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100620 int dir,
621 char_u *pat,
622 long count,
623 int options,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +0200624 int pat_use, // which pattern to use when "pat" is empty
625 searchit_arg_T *extra_arg) // optional extra arguments, can be NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000626{
627 int found;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100628 linenr_T lnum; // no init to shut up Apollo cc
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100629 colnr_T col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000630 regmmatch_T regmatch;
631 char_u *ptr;
632 colnr_T matchcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000633 lpos_T endpos;
Bram Moolenaar677ee682005-01-27 14:41:15 +0000634 lpos_T matchpos;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000635 int loop;
636 pos_T start_pos;
637 int at_first_line;
638 int extra_col;
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200639 int start_char_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000640 int match_ok;
641 long nmatched;
642 int submatch = 0;
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100643 int first_match = TRUE;
Bram Moolenaar53989552019-12-23 22:59:18 +0100644 int called_emsg_before = called_emsg;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000645#ifdef FEAT_SEARCH_EXTRA
646 int break_loop = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000647#endif
Bram Moolenaar92ea26b2019-10-18 20:53:34 +0200648 linenr_T stop_lnum = 0; // stop after this line number when != 0
649#ifdef FEAT_RELTIME
650 proftime_T *tm = NULL; // timeout limit or NULL
651 int *timed_out = NULL; // set when timed out or NULL
652#endif
653
654 if (extra_arg != NULL)
655 {
656 stop_lnum = extra_arg->sa_stop_lnum;
657#ifdef FEAT_RELTIME
658 tm = extra_arg->sa_tm;
659 timed_out = &extra_arg->sa_timed_out;
660#endif
661 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000662
663 if (search_regcomp(pat, RE_SEARCH, pat_use,
664 (options & (SEARCH_HIS + SEARCH_KEEP)), &regmatch) == FAIL)
665 {
666 if ((options & SEARCH_MSG) && !rc_did_emsg)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100667 semsg(_("E383: Invalid search string: %s"), mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000668 return FAIL;
669 }
670
Bram Moolenaar280f1262006-01-30 00:14:18 +0000671 /*
672 * find the string
673 */
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100674 do // loop for count
Bram Moolenaar071d4272004-06-13 20:20:40 +0000675 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100676 // When not accepting a match at the start position set "extra_col" to
677 // a non-zero value. Don't do that when starting at MAXCOL, since
678 // MAXCOL + 1 is zero.
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200679 if (pos->col == MAXCOL)
680 start_char_len = 0;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100681 // Watch out for the "col" being MAXCOL - 2, used in a closed fold.
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200682 else if (has_mbyte
683 && pos->lnum >= 1 && pos->lnum <= buf->b_ml.ml_line_count
684 && pos->col < MAXCOL - 2)
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100685 {
Bram Moolenaar82846a02018-02-09 18:09:54 +0100686 ptr = ml_get_buf(buf, pos->lnum, FALSE);
Bram Moolenaar8846ac52018-02-09 19:24:01 +0100687 if ((int)STRLEN(ptr) <= pos->col)
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200688 start_char_len = 1;
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100689 else
Bram Moolenaar82846a02018-02-09 18:09:54 +0100690 start_char_len = (*mb_ptr2len)(ptr + pos->col);
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100691 }
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100692 else
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200693 start_char_len = 1;
694 if (dir == FORWARD)
695 {
696 if (options & SEARCH_START)
697 extra_col = 0;
698 else
699 extra_col = start_char_len;
700 }
701 else
702 {
703 if (options & SEARCH_START)
704 extra_col = start_char_len;
705 else
706 extra_col = 0;
707 }
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100708
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100709 start_pos = *pos; // remember start pos for detecting no match
710 found = 0; // default: not found
711 at_first_line = TRUE; // default: start in first line
712 if (pos->lnum == 0) // correct lnum for when starting in line 0
Bram Moolenaar071d4272004-06-13 20:20:40 +0000713 {
714 pos->lnum = 1;
715 pos->col = 0;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100716 at_first_line = FALSE; // not in first line now
Bram Moolenaar071d4272004-06-13 20:20:40 +0000717 }
718
719 /*
720 * Start searching in current line, unless searching backwards and
721 * we're in column 0.
Bram Moolenaar7a42fa32007-07-10 11:28:55 +0000722 * If we are searching backwards, in column 0, and not including the
723 * current position, gain some efficiency by skipping back a line.
724 * Otherwise begin the search in the current line.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000725 */
Bram Moolenaar7a42fa32007-07-10 11:28:55 +0000726 if (dir == BACKWARD && start_pos.col == 0
727 && (options & SEARCH_START) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000728 {
729 lnum = pos->lnum - 1;
730 at_first_line = FALSE;
731 }
732 else
733 lnum = pos->lnum;
734
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100735 for (loop = 0; loop <= 1; ++loop) // loop twice if 'wrapscan' set
Bram Moolenaar071d4272004-06-13 20:20:40 +0000736 {
737 for ( ; lnum > 0 && lnum <= buf->b_ml.ml_line_count;
738 lnum += dir, at_first_line = FALSE)
739 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100740 // Stop after checking "stop_lnum", if it's set.
Bram Moolenaara23ccb82006-02-27 00:08:02 +0000741 if (stop_lnum != 0 && (dir == FORWARD
742 ? lnum > stop_lnum : lnum < stop_lnum))
743 break;
Bram Moolenaar76929292008-01-06 19:07:36 +0000744#ifdef FEAT_RELTIME
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100745 // Stop after passing the "tm" time limit.
Bram Moolenaar76929292008-01-06 19:07:36 +0000746 if (tm != NULL && profile_passed_limit(tm))
747 break;
748#endif
Bram Moolenaara23ccb82006-02-27 00:08:02 +0000749
Bram Moolenaar071d4272004-06-13 20:20:40 +0000750 /*
Bram Moolenaar677ee682005-01-27 14:41:15 +0000751 * Look for a match somewhere in line "lnum".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000752 */
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100753 col = at_first_line && (options & SEARCH_COL) ? pos->col
754 : (colnr_T)0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000755 nmatched = vim_regexec_multi(&regmatch, win, buf,
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100756 lnum, col,
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000757#ifdef FEAT_RELTIME
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200758 tm, timed_out
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000759#else
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200760 NULL, NULL
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000761#endif
762 );
Bram Moolenaar795aaa12020-10-02 20:36:01 +0200763 // vim_regexec_multi() may clear "regprog"
764 if (regmatch.regprog == NULL)
765 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100766 // Abort searching on an error (e.g., out of stack).
Bram Moolenaar53989552019-12-23 22:59:18 +0100767 if (called_emsg > called_emsg_before
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200768#ifdef FEAT_RELTIME
769 || (timed_out != NULL && *timed_out)
770#endif
771 )
Bram Moolenaar071d4272004-06-13 20:20:40 +0000772 break;
773 if (nmatched > 0)
774 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100775 // match may actually be in another line when using \zs
Bram Moolenaar677ee682005-01-27 14:41:15 +0000776 matchpos = regmatch.startpos[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000777 endpos = regmatch.endpos[0];
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000778#ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000779 submatch = first_submatch(&regmatch);
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000780#endif
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100781 // "lnum" may be past end of buffer for "\n\zs".
Bram Moolenaar32466aa2006-02-24 23:53:04 +0000782 if (lnum + matchpos.lnum > buf->b_ml.ml_line_count)
783 ptr = (char_u *)"";
784 else
785 ptr = ml_get_buf(buf, lnum + matchpos.lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000786
787 /*
788 * Forward search in the first line: match should be after
789 * the start position. If not, continue at the end of the
790 * match (this is vi compatible) or on the next char.
791 */
792 if (dir == FORWARD && at_first_line)
793 {
794 match_ok = TRUE;
795 /*
Bram Moolenaar677ee682005-01-27 14:41:15 +0000796 * When the match starts in a next line it's certainly
797 * past the start position.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000798 * When match lands on a NUL the cursor will be put
799 * one back afterwards, compare with that position,
800 * otherwise "/$" will get stuck on end of line.
801 */
Bram Moolenaar677ee682005-01-27 14:41:15 +0000802 while (matchpos.lnum == 0
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100803 && ((options & SEARCH_END) && first_match
Bram Moolenaar677ee682005-01-27 14:41:15 +0000804 ? (nmatched == 1
805 && (int)endpos.col - 1
Bram Moolenaar071d4272004-06-13 20:20:40 +0000806 < (int)start_pos.col + extra_col)
Bram Moolenaar677ee682005-01-27 14:41:15 +0000807 : ((int)matchpos.col
808 - (ptr[matchpos.col] == NUL)
809 < (int)start_pos.col + extra_col)))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000810 {
811 /*
812 * If vi-compatible searching, continue at the end
813 * of the match, otherwise continue one position
814 * forward.
815 */
816 if (vim_strchr(p_cpo, CPO_SEARCH) != NULL)
817 {
818 if (nmatched > 1)
819 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100820 // end is in next line, thus no match in
821 // this line
Bram Moolenaar071d4272004-06-13 20:20:40 +0000822 match_ok = FALSE;
823 break;
824 }
825 matchcol = endpos.col;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100826 // for empty match: advance one char
Bram Moolenaar677ee682005-01-27 14:41:15 +0000827 if (matchcol == matchpos.col
Bram Moolenaar071d4272004-06-13 20:20:40 +0000828 && ptr[matchcol] != NUL)
829 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000830 if (has_mbyte)
831 matchcol +=
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000832 (*mb_ptr2len)(ptr + matchcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000833 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000834 ++matchcol;
835 }
836 }
837 else
838 {
Bram Moolenaar677ee682005-01-27 14:41:15 +0000839 matchcol = matchpos.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000840 if (ptr[matchcol] != NUL)
841 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000842 if (has_mbyte)
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000843 matchcol += (*mb_ptr2len)(ptr
Bram Moolenaar071d4272004-06-13 20:20:40 +0000844 + matchcol);
845 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000846 ++matchcol;
847 }
848 }
Bram Moolenaar7bcb30e2013-04-03 21:14:29 +0200849 if (matchcol == 0 && (options & SEARCH_START))
Bram Moolenaardb333a52013-03-19 15:27:48 +0100850 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000851 if (ptr[matchcol] == NUL
852 || (nmatched = vim_regexec_multi(&regmatch,
Bram Moolenaar677ee682005-01-27 14:41:15 +0000853 win, buf, lnum + matchpos.lnum,
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000854 matchcol,
855#ifdef FEAT_RELTIME
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200856 tm, timed_out
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000857#else
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200858 NULL, NULL
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000859#endif
860 )) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000861 {
862 match_ok = FALSE;
863 break;
864 }
Bram Moolenaar795aaa12020-10-02 20:36:01 +0200865 // vim_regexec_multi() may clear "regprog"
866 if (regmatch.regprog == NULL)
867 break;
Bram Moolenaar677ee682005-01-27 14:41:15 +0000868 matchpos = regmatch.startpos[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000869 endpos = regmatch.endpos[0];
870# ifdef FEAT_EVAL
871 submatch = first_submatch(&regmatch);
872# endif
873
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100874 // Need to get the line pointer again, a
875 // multi-line search may have made it invalid.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000876 ptr = ml_get_buf(buf, lnum + matchpos.lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000877 }
878 if (!match_ok)
879 continue;
880 }
881 if (dir == BACKWARD)
882 {
883 /*
884 * Now, if there are multiple matches on this line,
885 * we have to get the last one. Or the last one before
886 * the cursor, if we're on that line.
887 * When putting the new cursor at the end, compare
888 * relative to the end of the match.
889 */
890 match_ok = FALSE;
891 for (;;)
892 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100893 // Remember a position that is before the start
894 // position, we use it if it's the last match in
895 // the line. Always accept a position after
896 // wrapping around.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000897 if (loop
898 || ((options & SEARCH_END)
899 ? (lnum + regmatch.endpos[0].lnum
900 < start_pos.lnum
901 || (lnum + regmatch.endpos[0].lnum
902 == start_pos.lnum
903 && (int)regmatch.endpos[0].col - 1
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200904 < (int)start_pos.col
905 + extra_col))
Bram Moolenaar677ee682005-01-27 14:41:15 +0000906 : (lnum + regmatch.startpos[0].lnum
907 < start_pos.lnum
908 || (lnum + regmatch.startpos[0].lnum
909 == start_pos.lnum
910 && (int)regmatch.startpos[0].col
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200911 < (int)start_pos.col
912 + extra_col))))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000913 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000914 match_ok = TRUE;
Bram Moolenaar677ee682005-01-27 14:41:15 +0000915 matchpos = regmatch.startpos[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000916 endpos = regmatch.endpos[0];
917# ifdef FEAT_EVAL
918 submatch = first_submatch(&regmatch);
919# endif
920 }
921 else
922 break;
923
924 /*
925 * We found a valid match, now check if there is
926 * another one after it.
927 * If vi-compatible searching, continue at the end
928 * of the match, otherwise continue one position
929 * forward.
930 */
931 if (vim_strchr(p_cpo, CPO_SEARCH) != NULL)
932 {
933 if (nmatched > 1)
934 break;
935 matchcol = endpos.col;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100936 // for empty match: advance one char
Bram Moolenaar677ee682005-01-27 14:41:15 +0000937 if (matchcol == matchpos.col
Bram Moolenaar071d4272004-06-13 20:20:40 +0000938 && ptr[matchcol] != NUL)
939 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000940 if (has_mbyte)
941 matchcol +=
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000942 (*mb_ptr2len)(ptr + matchcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000943 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000944 ++matchcol;
945 }
946 }
947 else
948 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100949 // Stop when the match is in a next line.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000950 if (matchpos.lnum > 0)
951 break;
952 matchcol = matchpos.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000953 if (ptr[matchcol] != NUL)
954 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000955 if (has_mbyte)
956 matchcol +=
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000957 (*mb_ptr2len)(ptr + matchcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000958 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000959 ++matchcol;
960 }
961 }
962 if (ptr[matchcol] == NUL
963 || (nmatched = vim_regexec_multi(&regmatch,
Bram Moolenaar677ee682005-01-27 14:41:15 +0000964 win, buf, lnum + matchpos.lnum,
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000965 matchcol,
966#ifdef FEAT_RELTIME
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200967 tm, timed_out
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000968#else
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200969 NULL, NULL
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000970#endif
971 )) == 0)
Bram Moolenaar9d322762018-02-09 16:04:25 +0100972 {
973#ifdef FEAT_RELTIME
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100974 // If the search timed out, we did find a match
975 // but it might be the wrong one, so that's not
976 // OK.
Bram Moolenaar9d322762018-02-09 16:04:25 +0100977 if (timed_out != NULL && *timed_out)
978 match_ok = FALSE;
979#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000980 break;
Bram Moolenaar9d322762018-02-09 16:04:25 +0100981 }
Bram Moolenaar795aaa12020-10-02 20:36:01 +0200982 // vim_regexec_multi() may clear "regprog"
983 if (regmatch.regprog == NULL)
984 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000985
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100986 // Need to get the line pointer again, a
987 // multi-line search may have made it invalid.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000988 ptr = ml_get_buf(buf, lnum + matchpos.lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000989 }
990
991 /*
992 * If there is only a match after the cursor, skip
993 * this match.
994 */
995 if (!match_ok)
996 continue;
997 }
998
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100999 // With the SEARCH_END option move to the last character
1000 // of the match. Don't do it for an empty match, end
1001 // should be same as start then.
Bram Moolenaar7bcb30e2013-04-03 21:14:29 +02001002 if ((options & SEARCH_END) && !(options & SEARCH_NOOF)
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001003 && !(matchpos.lnum == endpos.lnum
1004 && matchpos.col == endpos.col))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001005 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001006 // For a match in the first column, set the position
1007 // on the NUL in the previous line.
Bram Moolenaar677ee682005-01-27 14:41:15 +00001008 pos->lnum = lnum + endpos.lnum;
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001009 pos->col = endpos.col;
1010 if (endpos.col == 0)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00001011 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001012 if (pos->lnum > 1) // just in case
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001013 {
1014 --pos->lnum;
1015 pos->col = (colnr_T)STRLEN(ml_get_buf(buf,
1016 pos->lnum, FALSE));
1017 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00001018 }
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001019 else
1020 {
1021 --pos->col;
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001022 if (has_mbyte
1023 && pos->lnum <= buf->b_ml.ml_line_count)
1024 {
1025 ptr = ml_get_buf(buf, pos->lnum, FALSE);
1026 pos->col -= (*mb_head_off)(ptr, ptr + pos->col);
1027 }
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001028 }
Bram Moolenaar5d24a222018-12-23 19:10:09 +01001029 if (end_pos != NULL)
1030 {
1031 end_pos->lnum = lnum + matchpos.lnum;
1032 end_pos->col = matchpos.col;
1033 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001034 }
1035 else
1036 {
Bram Moolenaar677ee682005-01-27 14:41:15 +00001037 pos->lnum = lnum + matchpos.lnum;
1038 pos->col = matchpos.col;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01001039 if (end_pos != NULL)
1040 {
1041 end_pos->lnum = lnum + endpos.lnum;
1042 end_pos->col = endpos.col;
1043 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001044 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001045 pos->coladd = 0;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01001046 if (end_pos != NULL)
1047 end_pos->coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001048 found = 1;
Bram Moolenaara3dfccc2014-11-27 17:29:56 +01001049 first_match = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001050
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001051 // Set variables used for 'incsearch' highlighting.
Bram Moolenaar677ee682005-01-27 14:41:15 +00001052 search_match_lines = endpos.lnum - matchpos.lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001053 search_match_endcol = endpos.col;
1054 break;
1055 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001056 line_breakcheck(); // stop if ctrl-C typed
Bram Moolenaar071d4272004-06-13 20:20:40 +00001057 if (got_int)
1058 break;
1059
1060#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001061 // Cancel searching if a character was typed. Used for
1062 // 'incsearch'. Don't check too often, that would slowdown
1063 // searching too much.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001064 if ((options & SEARCH_PEEK)
1065 && ((lnum - pos->lnum) & 0x3f) == 0
1066 && char_avail())
1067 {
1068 break_loop = TRUE;
1069 break;
1070 }
1071#endif
1072
1073 if (loop && lnum == start_pos.lnum)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001074 break; // if second loop, stop where started
Bram Moolenaar071d4272004-06-13 20:20:40 +00001075 }
1076 at_first_line = FALSE;
1077
Bram Moolenaar795aaa12020-10-02 20:36:01 +02001078 // vim_regexec_multi() may clear "regprog"
1079 if (regmatch.regprog == NULL)
1080 break;
1081
Bram Moolenaar071d4272004-06-13 20:20:40 +00001082 /*
Bram Moolenaara23ccb82006-02-27 00:08:02 +00001083 * Stop the search if wrapscan isn't set, "stop_lnum" is
1084 * specified, after an interrupt, after a match and after looping
1085 * twice.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001086 */
Bram Moolenaar53989552019-12-23 22:59:18 +01001087 if (!p_ws || stop_lnum != 0 || got_int
1088 || called_emsg > called_emsg_before
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +02001089#ifdef FEAT_RELTIME
1090 || (timed_out != NULL && *timed_out)
Bram Moolenaar78a15312009-05-15 19:33:18 +00001091#endif
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +02001092#ifdef FEAT_SEARCH_EXTRA
1093 || break_loop
1094#endif
1095 || found || loop)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001096 break;
1097
1098 /*
1099 * If 'wrapscan' is set we continue at the other end of the file.
1100 * If 'shortmess' does not contain 's', we give a message.
1101 * This message is also remembered in keep_msg for when the screen
1102 * is redrawn. The keep_msg is cleared whenever another message is
1103 * written.
1104 */
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001105 if (dir == BACKWARD) // start second loop at the other end
Bram Moolenaar071d4272004-06-13 20:20:40 +00001106 lnum = buf->b_ml.ml_line_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001107 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001108 lnum = 1;
Bram Moolenaar92d640f2005-09-05 22:11:52 +00001109 if (!shortmess(SHM_SEARCH) && (options & SEARCH_MSG))
1110 give_warning((char_u *)_(dir == BACKWARD
1111 ? top_bot_msg : bot_top_msg), TRUE);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001112 if (extra_arg != NULL)
1113 extra_arg->sa_wrapped = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001114 }
Bram Moolenaar53989552019-12-23 22:59:18 +01001115 if (got_int || called_emsg > called_emsg_before
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +02001116#ifdef FEAT_RELTIME
1117 || (timed_out != NULL && *timed_out)
1118#endif
Bram Moolenaar78a15312009-05-15 19:33:18 +00001119#ifdef FEAT_SEARCH_EXTRA
1120 || break_loop
1121#endif
1122 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00001123 break;
1124 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001125 while (--count > 0 && found); // stop after count matches or no match
Bram Moolenaar071d4272004-06-13 20:20:40 +00001126
Bram Moolenaar473de612013-06-08 18:19:48 +02001127 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001128
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001129 if (!found) // did not find it
Bram Moolenaar071d4272004-06-13 20:20:40 +00001130 {
1131 if (got_int)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001132 emsg(_(e_interr));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001133 else if ((options & SEARCH_MSG) == SEARCH_MSG)
1134 {
1135 if (p_ws)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001136 semsg(_(e_patnotf2), mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001137 else if (lnum == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001138 semsg(_("E384: search hit TOP without match for: %s"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00001139 mr_pattern);
1140 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001141 semsg(_("E385: search hit BOTTOM without match for: %s"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00001142 mr_pattern);
1143 }
1144 return FAIL;
1145 }
1146
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001147 // A pattern like "\n\zs" may go past the last line.
Bram Moolenaar32466aa2006-02-24 23:53:04 +00001148 if (pos->lnum > buf->b_ml.ml_line_count)
1149 {
1150 pos->lnum = buf->b_ml.ml_line_count;
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001151 pos->col = (int)STRLEN(ml_get_buf(buf, pos->lnum, FALSE));
Bram Moolenaar32466aa2006-02-24 23:53:04 +00001152 if (pos->col > 0)
1153 --pos->col;
1154 }
1155
Bram Moolenaar071d4272004-06-13 20:20:40 +00001156 return submatch + 1;
1157}
1158
1159#ifdef FEAT_EVAL
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001160 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001161set_search_direction(int cdir)
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001162{
1163 spats[0].off.dir = cdir;
1164}
1165
1166 static void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001167set_vv_searchforward(void)
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001168{
1169 set_vim_var_nr(VV_SEARCHFORWARD, (long)(spats[0].off.dir == '/'));
1170}
1171
Bram Moolenaar071d4272004-06-13 20:20:40 +00001172/*
1173 * Return the number of the first subpat that matched.
Bram Moolenaarad4d8a12015-12-28 19:20:36 +01001174 * Return zero if none of them matched.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001175 */
1176 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001177first_submatch(regmmatch_T *rp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001178{
1179 int submatch;
1180
1181 for (submatch = 1; ; ++submatch)
1182 {
1183 if (rp->startpos[submatch].lnum >= 0)
1184 break;
1185 if (submatch == 9)
1186 {
1187 submatch = 0;
1188 break;
1189 }
1190 }
1191 return submatch;
1192}
1193#endif
1194
1195/*
1196 * Highest level string search function.
Bram Moolenaarb8017e72007-05-10 18:59:07 +00001197 * Search for the 'count'th occurrence of pattern 'pat' in direction 'dirc'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001198 * If 'dirc' is 0: use previous dir.
1199 * If 'pat' is NULL or empty : use previous string.
1200 * If 'options & SEARCH_REV' : go in reverse of previous dir.
1201 * If 'options & SEARCH_ECHO': echo the search command and handle options
1202 * If 'options & SEARCH_MSG' : may give error message
1203 * If 'options & SEARCH_OPT' : interpret optional flags
1204 * If 'options & SEARCH_HIS' : put search pattern in history
1205 * If 'options & SEARCH_NOOF': don't add offset to position
1206 * If 'options & SEARCH_MARK': set previous context mark
1207 * If 'options & SEARCH_KEEP': keep previous search pattern
1208 * If 'options & SEARCH_START': accept match at curpos itself
1209 * If 'options & SEARCH_PEEK': check for typed char, cancel search
1210 *
1211 * Careful: If spats[0].off.line == TRUE and spats[0].off.off == 0 this
1212 * makes the movement linewise without moving the match position.
1213 *
Bram Moolenaarb6c27352015-03-05 19:57:49 +01001214 * Return 0 for failure, 1 for found, 2 for found and line offset added.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001215 */
1216 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001217do_search(
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001218 oparg_T *oap, // can be NULL
1219 int dirc, // '/' or '?'
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001220 int search_delim, // the delimiter for the search, e.g. '%' in
1221 // s%regex%replacement%
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001222 char_u *pat,
1223 long count,
1224 int options,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001225 searchit_arg_T *sia) // optional arguments or NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001226{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001227 pos_T pos; // position of the last match
Bram Moolenaar071d4272004-06-13 20:20:40 +00001228 char_u *searchstr;
Bram Moolenaarc3328162019-07-23 22:15:25 +02001229 soffset_T old_off;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001230 int retval; // Return value
Bram Moolenaar071d4272004-06-13 20:20:40 +00001231 char_u *p;
1232 long c;
1233 char_u *dircp;
1234 char_u *strcopy = NULL;
1235 char_u *ps;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001236 char_u *msgbuf = NULL;
1237 size_t len;
Bram Moolenaar8f46e4c2019-05-24 22:08:15 +02001238 int has_offset = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001239
1240 /*
1241 * A line offset is not remembered, this is vi compatible.
1242 */
1243 if (spats[0].off.line && vim_strchr(p_cpo, CPO_LINEOFF) != NULL)
1244 {
1245 spats[0].off.line = FALSE;
1246 spats[0].off.off = 0;
1247 }
1248
1249 /*
1250 * Save the values for when (options & SEARCH_KEEP) is used.
1251 * (there is no "if ()" around this because gcc wants them initialized)
1252 */
1253 old_off = spats[0].off;
1254
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001255 pos = curwin->w_cursor; // start searching at the cursor position
Bram Moolenaar071d4272004-06-13 20:20:40 +00001256
1257 /*
1258 * Find out the direction of the search.
1259 */
1260 if (dirc == 0)
1261 dirc = spats[0].off.dir;
1262 else
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001263 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001264 spats[0].off.dir = dirc;
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001265#if defined(FEAT_EVAL)
1266 set_vv_searchforward();
1267#endif
1268 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001269 if (options & SEARCH_REV)
1270 {
Bram Moolenaar4f974752019-02-17 17:44:42 +01001271#ifdef MSWIN
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001272 // There is a bug in the Visual C++ 2.2 compiler which means that
1273 // dirc always ends up being '/'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001274 dirc = (dirc == '/') ? '?' : '/';
1275#else
1276 if (dirc == '/')
1277 dirc = '?';
1278 else
1279 dirc = '/';
1280#endif
1281 }
1282
1283#ifdef FEAT_FOLDING
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001284 // If the cursor is in a closed fold, don't find another match in the same
1285 // fold.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001286 if (dirc == '/')
1287 {
1288 if (hasFolding(pos.lnum, NULL, &pos.lnum))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001289 pos.col = MAXCOL - 2; // avoid overflow when adding 1
Bram Moolenaar071d4272004-06-13 20:20:40 +00001290 }
1291 else
1292 {
1293 if (hasFolding(pos.lnum, &pos.lnum, NULL))
1294 pos.col = 0;
1295 }
1296#endif
1297
1298#ifdef FEAT_SEARCH_EXTRA
1299 /*
1300 * Turn 'hlsearch' highlighting back on.
1301 */
1302 if (no_hlsearch && !(options & SEARCH_KEEP))
1303 {
Bram Moolenaar1c8f93f2006-03-12 22:10:07 +00001304 redraw_all_later(SOME_VALID);
Bram Moolenaar451fc7b2018-04-27 22:53:07 +02001305 set_no_hlsearch(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001306 }
1307#endif
1308
1309 /*
1310 * Repeat the search when pattern followed by ';', e.g. "/foo/;?bar".
1311 */
1312 for (;;)
1313 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001314 int show_top_bot_msg = FALSE;
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02001315
Bram Moolenaar071d4272004-06-13 20:20:40 +00001316 searchstr = pat;
1317 dircp = NULL;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001318 // use previous pattern
Bram Moolenaarc036e872020-02-21 21:30:52 +01001319 if (pat == NULL || *pat == NUL || *pat == search_delim)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001320 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001321 if (spats[RE_SEARCH].pat == NULL) // no previous pattern
Bram Moolenaar071d4272004-06-13 20:20:40 +00001322 {
Bram Moolenaarea683da2016-09-09 21:41:34 +02001323 searchstr = spats[RE_SUBST].pat;
1324 if (searchstr == NULL)
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001325 {
Bram Moolenaare29a27f2021-07-20 21:07:36 +02001326 emsg(_(e_no_previous_regular_expression));
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001327 retval = 0;
1328 goto end_do_search;
1329 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001330 }
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001331 else
1332 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001333 // make search_regcomp() use spats[RE_SEARCH].pat
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001334 searchstr = (char_u *)"";
1335 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001336 }
1337
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001338 if (pat != NULL && *pat != NUL) // look for (new) offset
Bram Moolenaar071d4272004-06-13 20:20:40 +00001339 {
1340 /*
1341 * Find end of regular expression.
1342 * If there is a matching '/' or '?', toss it.
1343 */
1344 ps = strcopy;
Bram Moolenaarf4e20992020-12-21 19:59:08 +01001345 p = skip_regexp_ex(pat, search_delim, magic_isset(),
Bram Moolenaard93a7fc2021-01-04 12:42:13 +01001346 &strcopy, NULL, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001347 if (strcopy != ps)
1348 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001349 // made a copy of "pat" to change "\?" to "?"
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001350 searchcmdlen += (int)(STRLEN(pat) - STRLEN(strcopy));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001351 pat = strcopy;
1352 searchstr = strcopy;
1353 }
Bram Moolenaarc036e872020-02-21 21:30:52 +01001354 if (*p == search_delim)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001355 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001356 dircp = p; // remember where we put the NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001357 *p++ = NUL;
1358 }
1359 spats[0].off.line = FALSE;
1360 spats[0].off.end = FALSE;
1361 spats[0].off.off = 0;
1362 /*
1363 * Check for a line offset or a character offset.
1364 * For get_address (echo off) we don't check for a character
1365 * offset, because it is meaningless and the 's' could be a
1366 * substitute command.
1367 */
1368 if (*p == '+' || *p == '-' || VIM_ISDIGIT(*p))
1369 spats[0].off.line = TRUE;
1370 else if ((options & SEARCH_OPT) &&
1371 (*p == 'e' || *p == 's' || *p == 'b'))
1372 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001373 if (*p == 'e') // end
Bram Moolenaar071d4272004-06-13 20:20:40 +00001374 spats[0].off.end = SEARCH_END;
1375 ++p;
1376 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001377 if (VIM_ISDIGIT(*p) || *p == '+' || *p == '-') // got an offset
Bram Moolenaar071d4272004-06-13 20:20:40 +00001378 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001379 // 'nr' or '+nr' or '-nr'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001380 if (VIM_ISDIGIT(*p) || VIM_ISDIGIT(*(p + 1)))
1381 spats[0].off.off = atol((char *)p);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001382 else if (*p == '-') // single '-'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001383 spats[0].off.off = -1;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001384 else // single '+'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001385 spats[0].off.off = 1;
1386 ++p;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001387 while (VIM_ISDIGIT(*p)) // skip number
Bram Moolenaar071d4272004-06-13 20:20:40 +00001388 ++p;
1389 }
1390
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001391 // compute length of search command for get_address()
Bram Moolenaar071d4272004-06-13 20:20:40 +00001392 searchcmdlen += (int)(p - pat);
1393
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001394 pat = p; // put pat after search command
Bram Moolenaar071d4272004-06-13 20:20:40 +00001395 }
1396
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001397 if ((options & SEARCH_ECHO) && messaging() &&
1398 !msg_silent &&
1399 (!cmd_silent || !shortmess(SHM_SEARCHCOUNT)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001400 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001401 char_u *trunc;
Bram Moolenaar984f0312019-05-24 13:11:47 +02001402 char_u off_buf[40];
Bram Moolenaard33a7642019-05-24 17:56:14 +02001403 size_t off_len = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001404
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001405 // Compute msg_row early.
1406 msg_start();
1407
Bram Moolenaar984f0312019-05-24 13:11:47 +02001408 // Get the offset, so we know how long it is.
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001409 if (!cmd_silent &&
1410 (spats[0].off.line || spats[0].off.end || spats[0].off.off))
Bram Moolenaar984f0312019-05-24 13:11:47 +02001411 {
1412 p = off_buf;
1413 *p++ = dirc;
1414 if (spats[0].off.end)
1415 *p++ = 'e';
1416 else if (!spats[0].off.line)
1417 *p++ = 's';
1418 if (spats[0].off.off > 0 || spats[0].off.line)
1419 *p++ = '+';
1420 *p = NUL;
1421 if (spats[0].off.off != 0 || spats[0].off.line)
1422 sprintf((char *)p, "%ld", spats[0].off.off);
1423 off_len = STRLEN(off_buf);
1424 }
1425
Bram Moolenaar071d4272004-06-13 20:20:40 +00001426 if (*searchstr == NUL)
Bram Moolenaar2fb8f682018-12-01 13:14:45 +01001427 p = spats[0].pat;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001428 else
1429 p = searchstr;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001430
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001431 if (!shortmess(SHM_SEARCHCOUNT) || cmd_silent)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001432 {
1433 // Reserve enough space for the search pattern + offset +
Bram Moolenaar984f0312019-05-24 13:11:47 +02001434 // search stat. Use all the space available, so that the
1435 // search state is right aligned. If there is not enough space
1436 // msg_strtrunc() will shorten in the middle.
Bram Moolenaar19e8ac72019-09-03 22:23:38 +02001437 if (msg_scrolled != 0 && !cmd_silent)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001438 // Use all the columns.
1439 len = (int)(Rows - msg_row) * Columns - 1;
1440 else
1441 // Use up to 'showcmd' column.
1442 len = (int)(Rows - msg_row - 1) * Columns + sc_col - 1;
Bram Moolenaar984f0312019-05-24 13:11:47 +02001443 if (len < STRLEN(p) + off_len + SEARCH_STAT_BUF_LEN + 3)
1444 len = STRLEN(p) + off_len + SEARCH_STAT_BUF_LEN + 3;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001445 }
1446 else
1447 // Reserve enough space for the search pattern + offset.
Bram Moolenaar984f0312019-05-24 13:11:47 +02001448 len = STRLEN(p) + off_len + 3;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001449
Bram Moolenaar880e4d92020-04-11 21:31:28 +02001450 vim_free(msgbuf);
Bram Moolenaar51e14382019-05-25 20:21:28 +02001451 msgbuf = alloc(len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001452 if (msgbuf != NULL)
1453 {
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001454 vim_memset(msgbuf, ' ', len);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001455 msgbuf[len - 1] = NUL;
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001456 // do not fill the msgbuf buffer, if cmd_silent is set, leave it
1457 // empty for the search_stat feature.
1458 if (!cmd_silent)
Bram Moolenaarcafda4f2005-09-06 19:25:11 +00001459 {
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001460 msgbuf[0] = dirc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001461
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001462 if (enc_utf8 && utf_iscomposing(utf_ptr2char(p)))
1463 {
1464 // Use a space to draw the composing char on.
1465 msgbuf[1] = ' ';
1466 mch_memmove(msgbuf + 2, p, STRLEN(p));
1467 }
1468 else
1469 mch_memmove(msgbuf + 1, p, STRLEN(p));
1470 if (off_len > 0)
1471 mch_memmove(msgbuf + STRLEN(p) + 1, off_buf, off_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001472
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001473 trunc = msg_strtrunc(msgbuf, TRUE);
1474 if (trunc != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001475 {
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001476 vim_free(msgbuf);
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001477 msgbuf = trunc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001478 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001479
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001480#ifdef FEAT_RIGHTLEFT
1481 // The search pattern could be shown on the right in
1482 // rightleft mode, but the 'ruler' and 'showcmd' area use
1483 // it too, thus it would be blanked out again very soon.
1484 // Show it on the left, but do reverse the text.
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001485 if (curwin->w_p_rl && *curwin->w_p_rlc == 's')
1486 {
1487 char_u *r;
1488 size_t pat_len;
1489
1490 r = reverse_text(msgbuf);
1491 if (r != NULL)
1492 {
1493 vim_free(msgbuf);
1494 msgbuf = r;
1495 // move reversed text to beginning of buffer
1496 while (*r != NUL && *r == ' ')
1497 r++;
1498 pat_len = msgbuf + STRLEN(msgbuf) - r;
1499 mch_memmove(msgbuf, r, pat_len);
1500 // overwrite old text
1501 if ((size_t)(r - msgbuf) >= pat_len)
1502 vim_memset(r, ' ', pat_len);
1503 else
1504 vim_memset(msgbuf + pat_len, ' ', r - msgbuf);
1505 }
1506 }
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001507#endif
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001508 msg_outtrans(msgbuf);
1509 msg_clr_eos();
1510 msg_check();
1511
1512 gotocmdline(FALSE);
1513 out_flush();
1514 msg_nowait = TRUE; // don't wait for this message
1515 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001516 }
1517 }
1518
1519 /*
1520 * If there is a character offset, subtract it from the current
1521 * position, so we don't get stuck at "?pat?e+2" or "/pat/s-2".
Bram Moolenaared203462004-06-16 11:19:22 +00001522 * Skip this if pos.col is near MAXCOL (closed fold).
Bram Moolenaar071d4272004-06-13 20:20:40 +00001523 * This is not done for a line offset, because then we would not be vi
1524 * compatible.
1525 */
Bram Moolenaared203462004-06-16 11:19:22 +00001526 if (!spats[0].off.line && spats[0].off.off && pos.col < MAXCOL - 2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001527 {
1528 if (spats[0].off.off > 0)
1529 {
1530 for (c = spats[0].off.off; c; --c)
1531 if (decl(&pos) == -1)
1532 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001533 if (c) // at start of buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001534 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001535 pos.lnum = 0; // allow lnum == 0 here
Bram Moolenaar071d4272004-06-13 20:20:40 +00001536 pos.col = MAXCOL;
1537 }
1538 }
1539 else
1540 {
1541 for (c = spats[0].off.off; c; ++c)
1542 if (incl(&pos) == -1)
1543 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001544 if (c) // at end of buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001545 {
1546 pos.lnum = curbuf->b_ml.ml_line_count + 1;
1547 pos.col = 0;
1548 }
1549 }
1550 }
1551
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001552 /*
1553 * The actual search.
1554 */
Bram Moolenaar14184a32019-02-16 15:10:30 +01001555 c = searchit(curwin, curbuf, &pos, NULL,
1556 dirc == '/' ? FORWARD : BACKWARD,
Bram Moolenaar071d4272004-06-13 20:20:40 +00001557 searchstr, count, spats[0].off.end + (options &
1558 (SEARCH_KEEP + SEARCH_PEEK + SEARCH_HIS
1559 + SEARCH_MSG + SEARCH_START
1560 + ((pat != NULL && *pat == ';') ? 0 : SEARCH_NOOF))),
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001561 RE_LAST, sia);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001562
1563 if (dircp != NULL)
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001564 *dircp = search_delim; // restore second '/' or '?' for normal_cmd()
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001565
1566 if (!shortmess(SHM_SEARCH)
1567 && ((dirc == '/' && LT_POS(pos, curwin->w_cursor))
1568 || (dirc == '?' && LT_POS(curwin->w_cursor, pos))))
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02001569 show_top_bot_msg = TRUE;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001570
Bram Moolenaar071d4272004-06-13 20:20:40 +00001571 if (c == FAIL)
1572 {
1573 retval = 0;
1574 goto end_do_search;
1575 }
1576 if (spats[0].off.end && oap != NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001577 oap->inclusive = TRUE; // 'e' includes last character
Bram Moolenaar071d4272004-06-13 20:20:40 +00001578
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001579 retval = 1; // pattern found
Bram Moolenaar071d4272004-06-13 20:20:40 +00001580
1581 /*
1582 * Add character and/or line offset
1583 */
Bram Moolenaar9160f302006-08-29 15:58:12 +00001584 if (!(options & SEARCH_NOOF) || (pat != NULL && *pat == ';'))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001585 {
Bram Moolenaar8f46e4c2019-05-24 22:08:15 +02001586 pos_T org_pos = pos;
1587
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001588 if (spats[0].off.line) // Add the offset to the line number.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001589 {
1590 c = pos.lnum + spats[0].off.off;
1591 if (c < 1)
1592 pos.lnum = 1;
1593 else if (c > curbuf->b_ml.ml_line_count)
1594 pos.lnum = curbuf->b_ml.ml_line_count;
1595 else
1596 pos.lnum = c;
1597 pos.col = 0;
1598
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001599 retval = 2; // pattern found, line offset added
Bram Moolenaar071d4272004-06-13 20:20:40 +00001600 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001601 else if (pos.col < MAXCOL - 2) // just in case
Bram Moolenaar071d4272004-06-13 20:20:40 +00001602 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001603 // to the right, check for end of file
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001604 c = spats[0].off.off;
1605 if (c > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001606 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001607 while (c-- > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001608 if (incl(&pos) == -1)
1609 break;
1610 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001611 // to the left, check for start of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00001612 else
1613 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001614 while (c++ < 0)
1615 if (decl(&pos) == -1)
1616 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001617 }
1618 }
Bram Moolenaar8f46e4c2019-05-24 22:08:15 +02001619 if (!EQUAL_POS(pos, org_pos))
1620 has_offset = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001621 }
1622
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001623 // Show [1/15] if 'S' is not in 'shortmess'.
1624 if ((options & SEARCH_ECHO)
1625 && messaging()
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001626 && !msg_silent
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001627 && c != FAIL
1628 && !shortmess(SHM_SEARCHCOUNT)
1629 && msgbuf != NULL)
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02001630 cmdline_search_stat(dirc, &pos, &curwin->w_cursor,
1631 show_top_bot_msg, msgbuf,
1632 (count != 1 || has_offset
Bram Moolenaar6cb07262020-05-29 22:49:43 +02001633#ifdef FEAT_FOLDING
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02001634 || (!(fdo_flags & FDO_SEARCH)
1635 && hasFolding(curwin->w_cursor.lnum,
1636 NULL, NULL))
Bram Moolenaar6cb07262020-05-29 22:49:43 +02001637#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02001638 ),
1639 SEARCH_STAT_DEF_MAX_COUNT,
1640 SEARCH_STAT_DEF_TIMEOUT);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001641
Bram Moolenaar071d4272004-06-13 20:20:40 +00001642 /*
1643 * The search command can be followed by a ';' to do another search.
1644 * For example: "/pat/;/foo/+3;?bar"
1645 * This is like doing another search command, except:
1646 * - The remembered direction '/' or '?' is from the first search.
1647 * - When an error happens the cursor isn't moved at all.
1648 * Don't do this when called by get_address() (it handles ';' itself).
1649 */
1650 if (!(options & SEARCH_OPT) || pat == NULL || *pat != ';')
1651 break;
1652
1653 dirc = *++pat;
Bram Moolenaarc036e872020-02-21 21:30:52 +01001654 search_delim = dirc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001655 if (dirc != '?' && dirc != '/')
1656 {
1657 retval = 0;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001658 emsg(_("E386: Expected '?' or '/' after ';'"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001659 goto end_do_search;
1660 }
1661 ++pat;
1662 }
1663
1664 if (options & SEARCH_MARK)
1665 setpcmark();
1666 curwin->w_cursor = pos;
1667 curwin->w_set_curswant = TRUE;
1668
1669end_do_search:
Bram Moolenaare1004402020-10-24 20:49:43 +02001670 if ((options & SEARCH_KEEP) || (cmdmod.cmod_flags & CMOD_KEEPPATTERNS))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001671 spats[0].off = old_off;
1672 vim_free(strcopy);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001673 vim_free(msgbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001674
1675 return retval;
1676}
1677
Bram Moolenaar071d4272004-06-13 20:20:40 +00001678/*
1679 * search_for_exact_line(buf, pos, dir, pat)
1680 *
1681 * Search for a line starting with the given pattern (ignoring leading
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02001682 * white-space), starting from pos and going in direction "dir". "pos" will
Bram Moolenaar071d4272004-06-13 20:20:40 +00001683 * contain the position of the match found. Blank lines match only if
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02001684 * ADDING is set. If p_ic is set then the pattern must be in lowercase.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001685 * Return OK for success, or FAIL if no line found.
1686 */
1687 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001688search_for_exact_line(
1689 buf_T *buf,
1690 pos_T *pos,
1691 int dir,
1692 char_u *pat)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001693{
1694 linenr_T start = 0;
1695 char_u *ptr;
1696 char_u *p;
1697
1698 if (buf->b_ml.ml_line_count == 0)
1699 return FAIL;
1700 for (;;)
1701 {
1702 pos->lnum += dir;
1703 if (pos->lnum < 1)
1704 {
1705 if (p_ws)
1706 {
1707 pos->lnum = buf->b_ml.ml_line_count;
1708 if (!shortmess(SHM_SEARCH))
1709 give_warning((char_u *)_(top_bot_msg), TRUE);
1710 }
1711 else
1712 {
1713 pos->lnum = 1;
1714 break;
1715 }
1716 }
1717 else if (pos->lnum > buf->b_ml.ml_line_count)
1718 {
1719 if (p_ws)
1720 {
1721 pos->lnum = 1;
1722 if (!shortmess(SHM_SEARCH))
1723 give_warning((char_u *)_(bot_top_msg), TRUE);
1724 }
1725 else
1726 {
1727 pos->lnum = 1;
1728 break;
1729 }
1730 }
1731 if (pos->lnum == start)
1732 break;
1733 if (start == 0)
1734 start = pos->lnum;
1735 ptr = ml_get_buf(buf, pos->lnum, FALSE);
1736 p = skipwhite(ptr);
1737 pos->col = (colnr_T) (p - ptr);
1738
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001739 // when adding lines the matching line may be empty but it is not
1740 // ignored because we are interested in the next line -- Acevedo
Bram Moolenaar4be06f92005-07-29 22:36:03 +00001741 if ((compl_cont_status & CONT_ADDING)
1742 && !(compl_cont_status & CONT_SOL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001743 {
1744 if ((p_ic ? MB_STRICMP(p, pat) : STRCMP(p, pat)) == 0)
1745 return OK;
1746 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001747 else if (*p != NUL) // ignore empty lines
1748 { // expanding lines or words
Bram Moolenaar4be06f92005-07-29 22:36:03 +00001749 if ((p_ic ? MB_STRNICMP(p, pat, compl_length)
1750 : STRNCMP(p, pat, compl_length)) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001751 return OK;
1752 }
1753 }
1754 return FAIL;
1755}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001756
1757/*
1758 * Character Searches
1759 */
1760
1761/*
1762 * Search for a character in a line. If "t_cmd" is FALSE, move to the
1763 * position of the character, otherwise move to just before the char.
1764 * Do this "cap->count1" times.
1765 * Return FAIL or OK.
1766 */
1767 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001768searchc(cmdarg_T *cap, int t_cmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001769{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001770 int c = cap->nchar; // char to search for
1771 int dir = cap->arg; // TRUE for searching forward
1772 long count = cap->count1; // repeat count
Bram Moolenaar071d4272004-06-13 20:20:40 +00001773 int col;
1774 char_u *p;
1775 int len;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001776 int stop = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001777
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001778 if (c != NUL) // normal search: remember args for repeat
Bram Moolenaar071d4272004-06-13 20:20:40 +00001779 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001780 if (!KeyStuffed) // don't remember when redoing
Bram Moolenaar071d4272004-06-13 20:20:40 +00001781 {
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001782 *lastc = c;
1783 set_csearch_direction(dir);
1784 set_csearch_until(t_cmd);
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001785 lastc_bytelen = (*mb_char2bytes)(c, lastc_bytes);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001786 if (cap->ncharC1 != 0)
1787 {
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001788 lastc_bytelen += (*mb_char2bytes)(cap->ncharC1,
1789 lastc_bytes + lastc_bytelen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001790 if (cap->ncharC2 != 0)
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001791 lastc_bytelen += (*mb_char2bytes)(cap->ncharC2,
1792 lastc_bytes + lastc_bytelen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001793 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001794 }
1795 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001796 else // repeat previous search
Bram Moolenaar071d4272004-06-13 20:20:40 +00001797 {
Bram Moolenaar264b74f2019-01-24 17:18:42 +01001798 if (*lastc == NUL && lastc_bytelen == 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001799 return FAIL;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001800 if (dir) // repeat in opposite direction
Bram Moolenaar071d4272004-06-13 20:20:40 +00001801 dir = -lastcdir;
1802 else
1803 dir = lastcdir;
1804 t_cmd = last_t_cmd;
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001805 c = *lastc;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001806 // For multi-byte re-use last lastc_bytes[] and lastc_bytelen.
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001807
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001808 // Force a move of at least one char, so ";" and "," will move the
1809 // cursor, even if the cursor is right in front of char we are looking
1810 // at.
Bram Moolenaar19fd09a2011-07-15 13:21:30 +02001811 if (vim_strchr(p_cpo, CPO_SCOLON) == NULL && count == 1 && t_cmd)
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001812 stop = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001813 }
1814
Bram Moolenaar60a795a2005-09-16 21:55:43 +00001815 if (dir == BACKWARD)
1816 cap->oap->inclusive = FALSE;
1817 else
1818 cap->oap->inclusive = TRUE;
1819
Bram Moolenaar071d4272004-06-13 20:20:40 +00001820 p = ml_get_curline();
1821 col = curwin->w_cursor.col;
1822 len = (int)STRLEN(p);
1823
1824 while (count--)
1825 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001826 if (has_mbyte)
1827 {
1828 for (;;)
1829 {
1830 if (dir > 0)
1831 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001832 col += (*mb_ptr2len)(p + col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001833 if (col >= len)
1834 return FAIL;
1835 }
1836 else
1837 {
1838 if (col == 0)
1839 return FAIL;
1840 col -= (*mb_head_off)(p, p + col - 1) + 1;
1841 }
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001842 if (lastc_bytelen == 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001843 {
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001844 if (p[col] == c && stop)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001845 break;
1846 }
Bram Moolenaar66727e12017-03-01 22:17:05 +01001847 else if (STRNCMP(p + col, lastc_bytes, lastc_bytelen) == 0
Bram Moolenaarb129a442016-12-01 17:25:20 +01001848 && stop)
Bram Moolenaar66727e12017-03-01 22:17:05 +01001849 break;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001850 stop = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001851 }
1852 }
1853 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001854 {
1855 for (;;)
1856 {
1857 if ((col += dir) < 0 || col >= len)
1858 return FAIL;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001859 if (p[col] == c && stop)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001860 break;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001861 stop = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001862 }
1863 }
1864 }
1865
1866 if (t_cmd)
1867 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001868 // backup to before the character (possibly double-byte)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001869 col -= dir;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001870 if (has_mbyte)
1871 {
1872 if (dir < 0)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001873 // Landed on the search char which is lastc_bytelen long
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001874 col += lastc_bytelen - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001875 else
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001876 // To previous char, which may be multi-byte.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001877 col -= (*mb_head_off)(p, p + col);
1878 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001879 }
1880 curwin->w_cursor.col = col;
1881
1882 return OK;
1883}
1884
1885/*
1886 * "Other" Searches
1887 */
1888
1889/*
1890 * findmatch - find the matching paren or brace
1891 *
1892 * Improvement over vi: Braces inside quotes are ignored.
1893 */
1894 pos_T *
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001895findmatch(oparg_T *oap, int initc)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001896{
1897 return findmatchlimit(oap, initc, 0, 0);
1898}
1899
1900/*
1901 * Return TRUE if the character before "linep[col]" equals "ch".
1902 * Return FALSE if "col" is zero.
1903 * Update "*prevcol" to the column of the previous character, unless "prevcol"
1904 * is NULL.
1905 * Handles multibyte string correctly.
1906 */
1907 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001908check_prevcol(
1909 char_u *linep,
1910 int col,
1911 int ch,
1912 int *prevcol)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001913{
1914 --col;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001915 if (col > 0 && has_mbyte)
1916 col -= (*mb_head_off)(linep, linep + col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001917 if (prevcol)
1918 *prevcol = col;
1919 return (col >= 0 && linep[col] == ch) ? TRUE : FALSE;
1920}
1921
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001922/*
1923 * Raw string start is found at linep[startpos.col - 1].
1924 * Return TRUE if the matching end can be found between startpos and endpos.
1925 */
1926 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001927find_rawstring_end(char_u *linep, pos_T *startpos, pos_T *endpos)
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001928{
1929 char_u *p;
1930 char_u *delim_copy;
1931 size_t delim_len;
1932 linenr_T lnum;
1933 int found = FALSE;
1934
1935 for (p = linep + startpos->col + 1; *p && *p != '('; ++p)
1936 ;
1937 delim_len = (p - linep) - startpos->col - 1;
Bram Moolenaar71ccd032020-06-12 22:59:11 +02001938 delim_copy = vim_strnsave(linep + startpos->col + 1, delim_len);
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001939 if (delim_copy == NULL)
1940 return FALSE;
1941 for (lnum = startpos->lnum; lnum <= endpos->lnum; ++lnum)
1942 {
1943 char_u *line = ml_get(lnum);
1944
1945 for (p = line + (lnum == startpos->lnum
1946 ? startpos->col + 1 : 0); *p; ++p)
1947 {
1948 if (lnum == endpos->lnum && (colnr_T)(p - line) >= endpos->col)
1949 break;
Bram Moolenaar282f9c62020-08-04 21:46:18 +02001950 if (*p == ')' && STRNCMP(delim_copy, p + 1, delim_len) == 0
1951 && p[delim_len + 1] == '"')
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001952 {
1953 found = TRUE;
1954 break;
1955 }
1956 }
1957 if (found)
1958 break;
1959 }
1960 vim_free(delim_copy);
1961 return found;
1962}
1963
Bram Moolenaar071d4272004-06-13 20:20:40 +00001964/*
Bram Moolenaar556ae8e2019-11-21 22:27:22 +01001965 * Check matchpairs option for "*initc".
1966 * If there is a match set "*initc" to the matching character and "*findc" to
1967 * the opposite character. Set "*backwards" to the direction.
1968 * When "switchit" is TRUE swap the direction.
1969 */
1970 static void
1971find_mps_values(
1972 int *initc,
1973 int *findc,
1974 int *backwards,
1975 int switchit)
1976{
1977 char_u *ptr;
1978
1979 ptr = curbuf->b_p_mps;
1980 while (*ptr != NUL)
1981 {
1982 if (has_mbyte)
1983 {
1984 char_u *prev;
1985
1986 if (mb_ptr2char(ptr) == *initc)
1987 {
1988 if (switchit)
1989 {
1990 *findc = *initc;
1991 *initc = mb_ptr2char(ptr + mb_ptr2len(ptr) + 1);
1992 *backwards = TRUE;
1993 }
1994 else
1995 {
1996 *findc = mb_ptr2char(ptr + mb_ptr2len(ptr) + 1);
1997 *backwards = FALSE;
1998 }
1999 return;
2000 }
2001 prev = ptr;
2002 ptr += mb_ptr2len(ptr) + 1;
2003 if (mb_ptr2char(ptr) == *initc)
2004 {
2005 if (switchit)
2006 {
2007 *findc = *initc;
2008 *initc = mb_ptr2char(prev);
2009 *backwards = FALSE;
2010 }
2011 else
2012 {
2013 *findc = mb_ptr2char(prev);
2014 *backwards = TRUE;
2015 }
2016 return;
2017 }
2018 ptr += mb_ptr2len(ptr);
2019 }
2020 else
2021 {
2022 if (*ptr == *initc)
2023 {
2024 if (switchit)
2025 {
2026 *backwards = TRUE;
2027 *findc = *initc;
2028 *initc = ptr[2];
2029 }
2030 else
2031 {
2032 *backwards = FALSE;
2033 *findc = ptr[2];
2034 }
2035 return;
2036 }
2037 ptr += 2;
2038 if (*ptr == *initc)
2039 {
2040 if (switchit)
2041 {
2042 *backwards = FALSE;
2043 *findc = *initc;
2044 *initc = ptr[-2];
2045 }
2046 else
2047 {
2048 *backwards = TRUE;
2049 *findc = ptr[-2];
2050 }
2051 return;
2052 }
2053 ++ptr;
2054 }
2055 if (*ptr == ',')
2056 ++ptr;
2057 }
2058}
2059
2060/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002061 * findmatchlimit -- find the matching paren or brace, if it exists within
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002062 * maxtravel lines of the cursor. A maxtravel of 0 means search until falling
2063 * off the edge of the file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002064 *
2065 * "initc" is the character to find a match for. NUL means to find the
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002066 * character at or after the cursor. Special values:
2067 * '*' look for C-style comment / *
2068 * '/' look for C-style comment / *, ignoring comment-end
2069 * '#' look for preprocessor directives
2070 * 'R' look for raw string start: R"delim(text)delim" (only backwards)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002071 *
2072 * flags: FM_BACKWARD search backwards (when initc is '/', '*' or '#')
2073 * FM_FORWARD search forwards (when initc is '/', '*' or '#')
2074 * FM_BLOCKSTOP stop at start/end of block ({ or } in column 0)
2075 * FM_SKIPCOMM skip comments (not implemented yet!)
Bram Moolenaarf75a9632005-09-13 21:20:47 +00002076 *
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002077 * "oap" is only used to set oap->motion_type for a linewise motion, it can be
Bram Moolenaarf75a9632005-09-13 21:20:47 +00002078 * NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00002079 */
2080
2081 pos_T *
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002082findmatchlimit(
2083 oparg_T *oap,
2084 int initc,
2085 int flags,
2086 int maxtravel)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002087{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002088 static pos_T pos; // current search position
2089 int findc = 0; // matching brace
Bram Moolenaar071d4272004-06-13 20:20:40 +00002090 int c;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002091 int count = 0; // cumulative number of braces
2092 int backwards = FALSE; // init for gcc
2093 int raw_string = FALSE; // search for raw string
2094 int inquote = FALSE; // TRUE when inside quotes
2095 char_u *linep; // pointer to current line
Bram Moolenaar071d4272004-06-13 20:20:40 +00002096 char_u *ptr;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002097 int do_quotes; // check for quotes in current line
2098 int at_start; // do_quotes value at start position
2099 int hash_dir = 0; // Direction searched for # things
2100 int comment_dir = 0; // Direction searched for comments
2101 pos_T match_pos; // Where last slash-star was found
2102 int start_in_quotes; // start position is in quotes
2103 int traveled = 0; // how far we've searched so far
2104 int ignore_cend = FALSE; // ignore comment end
2105 int cpo_match; // vi compatible matching
2106 int cpo_bsl; // don't recognize backslashes
2107 int match_escaped = 0; // search for escaped match
2108 int dir; // Direction to search
2109 int comment_col = MAXCOL; // start of / / comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002110#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002111 int lispcomm = FALSE; // inside of Lisp-style comment
2112 int lisp = curbuf->b_p_lisp; // engage Lisp-specific hacks ;)
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002113#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002114
2115 pos = curwin->w_cursor;
Bram Moolenaarc56c4592013-08-14 17:45:29 +02002116 pos.coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002117 linep = ml_get(pos.lnum);
2118
2119 cpo_match = (vim_strchr(p_cpo, CPO_MATCH) != NULL);
2120 cpo_bsl = (vim_strchr(p_cpo, CPO_MATCHBSL) != NULL);
2121
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002122 // Direction to search when initc is '/', '*' or '#'
Bram Moolenaar071d4272004-06-13 20:20:40 +00002123 if (flags & FM_BACKWARD)
2124 dir = BACKWARD;
2125 else if (flags & FM_FORWARD)
2126 dir = FORWARD;
2127 else
2128 dir = 0;
2129
2130 /*
2131 * if initc given, look in the table for the matching character
2132 * '/' and '*' are special cases: look for start or end of comment.
2133 * When '/' is used, we ignore running backwards into an star-slash, for
2134 * "[*" command, we just want to find any comment.
2135 */
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002136 if (initc == '/' || initc == '*' || initc == 'R')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002137 {
2138 comment_dir = dir;
2139 if (initc == '/')
2140 ignore_cend = TRUE;
2141 backwards = (dir == FORWARD) ? FALSE : TRUE;
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002142 raw_string = (initc == 'R');
Bram Moolenaar071d4272004-06-13 20:20:40 +00002143 initc = NUL;
2144 }
2145 else if (initc != '#' && initc != NUL)
2146 {
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002147 find_mps_values(&initc, &findc, &backwards, TRUE);
Connor Lane Smithb9115da2021-07-31 13:31:42 +02002148 if (dir)
2149 backwards = (dir == FORWARD) ? FALSE : TRUE;
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002150 if (findc == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002151 return NULL;
2152 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002153 else
2154 {
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002155 /*
2156 * Either initc is '#', or no initc was given and we need to look
2157 * under the cursor.
2158 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002159 if (initc == '#')
2160 {
2161 hash_dir = dir;
2162 }
2163 else
2164 {
2165 /*
2166 * initc was not given, must look for something to match under
2167 * or near the cursor.
2168 * Only check for special things when 'cpo' doesn't have '%'.
2169 */
2170 if (!cpo_match)
2171 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002172 // Are we before or at #if, #else etc.?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002173 ptr = skipwhite(linep);
2174 if (*ptr == '#' && pos.col <= (colnr_T)(ptr - linep))
2175 {
2176 ptr = skipwhite(ptr + 1);
2177 if ( STRNCMP(ptr, "if", 2) == 0
2178 || STRNCMP(ptr, "endif", 5) == 0
2179 || STRNCMP(ptr, "el", 2) == 0)
2180 hash_dir = 1;
2181 }
2182
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002183 // Are we on a comment?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002184 else if (linep[pos.col] == '/')
2185 {
2186 if (linep[pos.col + 1] == '*')
2187 {
2188 comment_dir = FORWARD;
2189 backwards = FALSE;
2190 pos.col++;
2191 }
2192 else if (pos.col > 0 && linep[pos.col - 1] == '*')
2193 {
2194 comment_dir = BACKWARD;
2195 backwards = TRUE;
2196 pos.col--;
2197 }
2198 }
2199 else if (linep[pos.col] == '*')
2200 {
2201 if (linep[pos.col + 1] == '/')
2202 {
2203 comment_dir = BACKWARD;
2204 backwards = TRUE;
2205 }
2206 else if (pos.col > 0 && linep[pos.col - 1] == '/')
2207 {
2208 comment_dir = FORWARD;
2209 backwards = FALSE;
2210 }
2211 }
2212 }
2213
2214 /*
2215 * If we are not on a comment or the # at the start of a line, then
2216 * look for brace anywhere on this line after the cursor.
2217 */
2218 if (!hash_dir && !comment_dir)
2219 {
2220 /*
2221 * Find the brace under or after the cursor.
2222 * If beyond the end of the line, use the last character in
2223 * the line.
2224 */
2225 if (linep[pos.col] == NUL && pos.col)
2226 --pos.col;
2227 for (;;)
2228 {
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002229 initc = PTR2CHAR(linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002230 if (initc == NUL)
2231 break;
2232
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002233 find_mps_values(&initc, &findc, &backwards, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002234 if (findc)
2235 break;
Bram Moolenaar1614a142019-10-06 22:00:13 +02002236 pos.col += mb_ptr2len(linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002237 }
2238 if (!findc)
2239 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002240 // no brace in the line, maybe use " #if" then
Bram Moolenaar071d4272004-06-13 20:20:40 +00002241 if (!cpo_match && *skipwhite(linep) == '#')
2242 hash_dir = 1;
2243 else
2244 return NULL;
2245 }
2246 else if (!cpo_bsl)
2247 {
2248 int col, bslcnt = 0;
2249
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002250 // Set "match_escaped" if there are an odd number of
2251 // backslashes.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002252 for (col = pos.col; check_prevcol(linep, col, '\\', &col);)
2253 bslcnt++;
2254 match_escaped = (bslcnt & 1);
2255 }
2256 }
2257 }
2258 if (hash_dir)
2259 {
2260 /*
2261 * Look for matching #if, #else, #elif, or #endif
2262 */
2263 if (oap != NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002264 oap->motion_type = MLINE; // Linewise for this case only
Bram Moolenaar071d4272004-06-13 20:20:40 +00002265 if (initc != '#')
2266 {
2267 ptr = skipwhite(skipwhite(linep) + 1);
2268 if (STRNCMP(ptr, "if", 2) == 0 || STRNCMP(ptr, "el", 2) == 0)
2269 hash_dir = 1;
2270 else if (STRNCMP(ptr, "endif", 5) == 0)
2271 hash_dir = -1;
2272 else
2273 return NULL;
2274 }
2275 pos.col = 0;
2276 while (!got_int)
2277 {
2278 if (hash_dir > 0)
2279 {
2280 if (pos.lnum == curbuf->b_ml.ml_line_count)
2281 break;
2282 }
2283 else if (pos.lnum == 1)
2284 break;
2285 pos.lnum += hash_dir;
2286 linep = ml_get(pos.lnum);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002287 line_breakcheck(); // check for CTRL-C typed
Bram Moolenaar071d4272004-06-13 20:20:40 +00002288 ptr = skipwhite(linep);
2289 if (*ptr != '#')
2290 continue;
2291 pos.col = (colnr_T) (ptr - linep);
2292 ptr = skipwhite(ptr + 1);
2293 if (hash_dir > 0)
2294 {
2295 if (STRNCMP(ptr, "if", 2) == 0)
2296 count++;
2297 else if (STRNCMP(ptr, "el", 2) == 0)
2298 {
2299 if (count == 0)
2300 return &pos;
2301 }
2302 else if (STRNCMP(ptr, "endif", 5) == 0)
2303 {
2304 if (count == 0)
2305 return &pos;
2306 count--;
2307 }
2308 }
2309 else
2310 {
2311 if (STRNCMP(ptr, "if", 2) == 0)
2312 {
2313 if (count == 0)
2314 return &pos;
2315 count--;
2316 }
2317 else if (initc == '#' && STRNCMP(ptr, "el", 2) == 0)
2318 {
2319 if (count == 0)
2320 return &pos;
2321 }
2322 else if (STRNCMP(ptr, "endif", 5) == 0)
2323 count++;
2324 }
2325 }
2326 return NULL;
2327 }
2328 }
2329
2330#ifdef FEAT_RIGHTLEFT
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002331 // This is just guessing: when 'rightleft' is set, search for a matching
2332 // paren/brace in the other direction.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002333 if (curwin->w_p_rl && vim_strchr((char_u *)"()[]{}<>", initc) != NULL)
2334 backwards = !backwards;
2335#endif
2336
2337 do_quotes = -1;
2338 start_in_quotes = MAYBE;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01002339 CLEAR_POS(&match_pos);
Bram Moolenaarfd2ac762006-03-01 22:09:21 +00002340
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002341 // backward search: Check if this line contains a single-line comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002342 if ((backwards && comment_dir)
2343#ifdef FEAT_LISP
2344 || lisp
2345#endif
2346 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002347 comment_col = check_linecomment(linep);
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002348#ifdef FEAT_LISP
2349 if (lisp && comment_col != MAXCOL && pos.col > (colnr_T)comment_col)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002350 lispcomm = TRUE; // find match inside this comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002351#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002352 while (!got_int)
2353 {
2354 /*
2355 * Go to the next position, forward or backward. We could use
2356 * inc() and dec() here, but that is much slower
2357 */
2358 if (backwards)
2359 {
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002360#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002361 // char to match is inside of comment, don't search outside
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002362 if (lispcomm && pos.col < (colnr_T)comment_col)
2363 break;
2364#endif
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002365 if (pos.col == 0) // at start of line, go to prev. one
Bram Moolenaar071d4272004-06-13 20:20:40 +00002366 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002367 if (pos.lnum == 1) // start of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00002368 break;
2369 --pos.lnum;
2370
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002371 if (maxtravel > 0 && ++traveled > maxtravel)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002372 break;
2373
2374 linep = ml_get(pos.lnum);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002375 pos.col = (colnr_T)STRLEN(linep); // pos.col on trailing NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00002376 do_quotes = -1;
2377 line_breakcheck();
2378
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002379 // Check if this line contains a single-line comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002380 if (comment_dir
2381#ifdef FEAT_LISP
2382 || lisp
2383#endif
2384 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002385 comment_col = check_linecomment(linep);
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002386#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002387 // skip comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002388 if (lisp && comment_col != MAXCOL)
2389 pos.col = comment_col;
2390#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002391 }
2392 else
2393 {
2394 --pos.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002395 if (has_mbyte)
2396 pos.col -= (*mb_head_off)(linep, linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002397 }
2398 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002399 else // forward search
Bram Moolenaar071d4272004-06-13 20:20:40 +00002400 {
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002401 if (linep[pos.col] == NUL
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002402 // at end of line, go to next one
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002403#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002404 // don't search for match in comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002405 || (lisp && comment_col != MAXCOL
2406 && pos.col == (colnr_T)comment_col)
2407#endif
2408 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002409 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002410 if (pos.lnum == curbuf->b_ml.ml_line_count // end of file
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002411#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002412 // line is exhausted and comment with it,
2413 // don't search for match in code
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002414 || lispcomm
2415#endif
2416 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002417 break;
2418 ++pos.lnum;
2419
2420 if (maxtravel && traveled++ > maxtravel)
2421 break;
2422
2423 linep = ml_get(pos.lnum);
2424 pos.col = 0;
2425 do_quotes = -1;
2426 line_breakcheck();
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002427#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002428 if (lisp) // find comment pos in new line
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002429 comment_col = check_linecomment(linep);
2430#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002431 }
2432 else
2433 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002434 if (has_mbyte)
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00002435 pos.col += (*mb_ptr2len)(linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002436 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00002437 ++pos.col;
2438 }
2439 }
2440
2441 /*
2442 * If FM_BLOCKSTOP given, stop at a '{' or '}' in column 0.
2443 */
2444 if (pos.col == 0 && (flags & FM_BLOCKSTOP) &&
2445 (linep[0] == '{' || linep[0] == '}'))
2446 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002447 if (linep[0] == findc && count == 0) // match!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002448 return &pos;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002449 break; // out of scope
Bram Moolenaar071d4272004-06-13 20:20:40 +00002450 }
2451
2452 if (comment_dir)
2453 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002454 // Note: comments do not nest, and we ignore quotes in them
2455 // TODO: ignore comment brackets inside strings
Bram Moolenaar071d4272004-06-13 20:20:40 +00002456 if (comment_dir == FORWARD)
2457 {
2458 if (linep[pos.col] == '*' && linep[pos.col + 1] == '/')
2459 {
2460 pos.col++;
2461 return &pos;
2462 }
2463 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002464 else // Searching backwards
Bram Moolenaar071d4272004-06-13 20:20:40 +00002465 {
2466 /*
2467 * A comment may contain / * or / /, it may also start or end
Bram Moolenaarf8c53d32017-11-12 15:36:38 +01002468 * with / * /. Ignore a / * after / / and after *.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002469 */
2470 if (pos.col == 0)
2471 continue;
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002472 else if (raw_string)
2473 {
2474 if (linep[pos.col - 1] == 'R'
2475 && linep[pos.col] == '"'
2476 && vim_strchr(linep + pos.col + 1, '(') != NULL)
2477 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002478 // Possible start of raw string. Now that we have the
2479 // delimiter we can check if it ends before where we
2480 // started searching, or before the previously found
2481 // raw string start.
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002482 if (!find_rawstring_end(linep, &pos,
2483 count > 0 ? &match_pos : &curwin->w_cursor))
2484 {
2485 count++;
2486 match_pos = pos;
2487 match_pos.col--;
2488 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002489 linep = ml_get(pos.lnum); // may have been released
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002490 }
2491 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002492 else if ( linep[pos.col - 1] == '/'
2493 && linep[pos.col] == '*'
Bram Moolenaarf8c53d32017-11-12 15:36:38 +01002494 && (pos.col == 1 || linep[pos.col - 2] != '*')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002495 && (int)pos.col < comment_col)
2496 {
2497 count++;
2498 match_pos = pos;
2499 match_pos.col--;
2500 }
2501 else if (linep[pos.col - 1] == '*' && linep[pos.col] == '/')
2502 {
2503 if (count > 0)
2504 pos = match_pos;
2505 else if (pos.col > 1 && linep[pos.col - 2] == '/'
2506 && (int)pos.col <= comment_col)
2507 pos.col -= 2;
2508 else if (ignore_cend)
2509 continue;
2510 else
2511 return NULL;
2512 return &pos;
2513 }
2514 }
2515 continue;
2516 }
2517
2518 /*
2519 * If smart matching ('cpoptions' does not contain '%'), braces inside
2520 * of quotes are ignored, but only if there is an even number of
2521 * quotes in the line.
2522 */
2523 if (cpo_match)
2524 do_quotes = 0;
2525 else if (do_quotes == -1)
2526 {
2527 /*
2528 * Count the number of quotes in the line, skipping \" and '"'.
2529 * Watch out for "\\".
2530 */
2531 at_start = do_quotes;
2532 for (ptr = linep; *ptr; ++ptr)
2533 {
2534 if (ptr == linep + pos.col + backwards)
2535 at_start = (do_quotes & 1);
2536 if (*ptr == '"'
2537 && (ptr == linep || ptr[-1] != '\'' || ptr[1] != '\''))
2538 ++do_quotes;
2539 if (*ptr == '\\' && ptr[1] != NUL)
2540 ++ptr;
2541 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002542 do_quotes &= 1; // result is 1 with even number of quotes
Bram Moolenaar071d4272004-06-13 20:20:40 +00002543
2544 /*
2545 * If we find an uneven count, check current line and previous
2546 * one for a '\' at the end.
2547 */
2548 if (!do_quotes)
2549 {
2550 inquote = FALSE;
2551 if (ptr[-1] == '\\')
2552 {
2553 do_quotes = 1;
2554 if (start_in_quotes == MAYBE)
2555 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002556 // Do we need to use at_start here?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002557 inquote = TRUE;
2558 start_in_quotes = TRUE;
2559 }
2560 else if (backwards)
2561 inquote = TRUE;
2562 }
2563 if (pos.lnum > 1)
2564 {
2565 ptr = ml_get(pos.lnum - 1);
2566 if (*ptr && *(ptr + STRLEN(ptr) - 1) == '\\')
2567 {
2568 do_quotes = 1;
2569 if (start_in_quotes == MAYBE)
2570 {
2571 inquote = at_start;
2572 if (inquote)
2573 start_in_quotes = TRUE;
2574 }
2575 else if (!backwards)
2576 inquote = TRUE;
2577 }
Bram Moolenaaraec11792007-07-10 11:09:36 +00002578
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002579 // ml_get() only keeps one line, need to get linep again
Bram Moolenaaraec11792007-07-10 11:09:36 +00002580 linep = ml_get(pos.lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002581 }
2582 }
2583 }
2584 if (start_in_quotes == MAYBE)
2585 start_in_quotes = FALSE;
2586
2587 /*
2588 * If 'smartmatch' is set:
2589 * Things inside quotes are ignored by setting 'inquote'. If we
2590 * find a quote without a preceding '\' invert 'inquote'. At the
2591 * end of a line not ending in '\' we reset 'inquote'.
2592 *
2593 * In lines with an uneven number of quotes (without preceding '\')
2594 * we do not know which part to ignore. Therefore we only set
2595 * inquote if the number of quotes in a line is even, unless this
2596 * line or the previous one ends in a '\'. Complicated, isn't it?
2597 */
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002598 c = PTR2CHAR(linep + pos.col);
2599 switch (c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002600 {
2601 case NUL:
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002602 // at end of line without trailing backslash, reset inquote
Bram Moolenaar071d4272004-06-13 20:20:40 +00002603 if (pos.col == 0 || linep[pos.col - 1] != '\\')
2604 {
2605 inquote = FALSE;
2606 start_in_quotes = FALSE;
2607 }
2608 break;
2609
2610 case '"':
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002611 // a quote that is preceded with an odd number of backslashes is
2612 // ignored
Bram Moolenaar071d4272004-06-13 20:20:40 +00002613 if (do_quotes)
2614 {
2615 int col;
2616
2617 for (col = pos.col - 1; col >= 0; --col)
2618 if (linep[col] != '\\')
2619 break;
2620 if ((((int)pos.col - 1 - col) & 1) == 0)
2621 {
2622 inquote = !inquote;
2623 start_in_quotes = FALSE;
2624 }
2625 }
2626 break;
2627
2628 /*
2629 * If smart matching ('cpoptions' does not contain '%'):
2630 * Skip things in single quotes: 'x' or '\x'. Be careful for single
2631 * single quotes, eg jon's. Things like '\233' or '\x3f' are not
2632 * skipped, there is never a brace in them.
2633 * Ignore this when finding matches for `'.
2634 */
2635 case '\'':
2636 if (!cpo_match && initc != '\'' && findc != '\'')
2637 {
2638 if (backwards)
2639 {
2640 if (pos.col > 1)
2641 {
2642 if (linep[pos.col - 2] == '\'')
2643 {
2644 pos.col -= 2;
2645 break;
2646 }
2647 else if (linep[pos.col - 2] == '\\' &&
2648 pos.col > 2 && linep[pos.col - 3] == '\'')
2649 {
2650 pos.col -= 3;
2651 break;
2652 }
2653 }
2654 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002655 else if (linep[pos.col + 1]) // forward search
Bram Moolenaar071d4272004-06-13 20:20:40 +00002656 {
2657 if (linep[pos.col + 1] == '\\' &&
2658 linep[pos.col + 2] && linep[pos.col + 3] == '\'')
2659 {
2660 pos.col += 3;
2661 break;
2662 }
2663 else if (linep[pos.col + 2] == '\'')
2664 {
2665 pos.col += 2;
2666 break;
2667 }
2668 }
2669 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002670 // FALLTHROUGH
Bram Moolenaar071d4272004-06-13 20:20:40 +00002671
2672 default:
2673#ifdef FEAT_LISP
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002674 /*
2675 * For Lisp skip over backslashed (), {} and [].
2676 * (actually, we skip #\( et al)
2677 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002678 if (curbuf->b_p_lisp
2679 && vim_strchr((char_u *)"(){}[]", c) != NULL
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002680 && pos.col > 1
2681 && check_prevcol(linep, pos.col, '\\', NULL)
2682 && check_prevcol(linep, pos.col - 1, '#', NULL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002683 break;
2684#endif
2685
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002686 // Check for match outside of quotes, and inside of
2687 // quotes when the start is also inside of quotes.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002688 if ((!inquote || start_in_quotes == TRUE)
2689 && (c == initc || c == findc))
2690 {
2691 int col, bslcnt = 0;
2692
2693 if (!cpo_bsl)
2694 {
2695 for (col = pos.col; check_prevcol(linep, col, '\\', &col);)
2696 bslcnt++;
2697 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002698 // Only accept a match when 'M' is in 'cpo' or when escaping
2699 // is what we expect.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002700 if (cpo_bsl || (bslcnt & 1) == match_escaped)
2701 {
2702 if (c == initc)
2703 count++;
2704 else
2705 {
2706 if (count == 0)
2707 return &pos;
2708 count--;
2709 }
2710 }
2711 }
2712 }
2713 }
2714
2715 if (comment_dir == BACKWARD && count > 0)
2716 {
2717 pos = match_pos;
2718 return &pos;
2719 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002720 return (pos_T *)NULL; // never found it
Bram Moolenaar071d4272004-06-13 20:20:40 +00002721}
2722
2723/*
2724 * Check if line[] contains a / / comment.
2725 * Return MAXCOL if not, otherwise return the column.
2726 * TODO: skip strings.
2727 */
2728 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002729check_linecomment(char_u *line)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002730{
2731 char_u *p;
2732
2733 p = line;
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002734#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002735 // skip Lispish one-line comments
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002736 if (curbuf->b_p_lisp)
2737 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002738 if (vim_strchr(p, ';') != NULL) // there may be comments
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002739 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002740 int in_str = FALSE; // inside of string
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002741
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002742 p = line; // scan from start
Bram Moolenaar520470a2005-06-16 21:59:56 +00002743 while ((p = vim_strpbrk(p, (char_u *)"\";")) != NULL)
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002744 {
2745 if (*p == '"')
2746 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002747 if (in_str)
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002748 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002749 if (*(p - 1) != '\\') // skip escaped quote
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002750 in_str = FALSE;
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002751 }
2752 else if (p == line || ((p - line) >= 2
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002753 // skip #\" form
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002754 && *(p - 1) != '\\' && *(p - 2) != '#'))
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002755 in_str = TRUE;
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002756 }
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002757 else if (!in_str && ((p - line) < 2
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002758 || (*(p - 1) != '\\' && *(p - 2) != '#')))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002759 break; // found!
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002760 ++p;
2761 }
2762 }
2763 else
2764 p = NULL;
2765 }
2766 else
2767#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002768 while ((p = vim_strchr(p, '/')) != NULL)
2769 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002770 // accept a double /, unless it's preceded with * and followed by *,
2771 // because * / / * is an end and start of a C comment
Bram Moolenaar78d4aba2008-01-01 14:43:35 +00002772 if (p[1] == '/' && (p == line || p[-1] != '*' || p[2] != '*'))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002773 break;
2774 ++p;
2775 }
2776
2777 if (p == NULL)
2778 return MAXCOL;
2779 return (int)(p - line);
2780}
2781
2782/*
2783 * Move cursor briefly to character matching the one under the cursor.
2784 * Used for Insert mode and "r" command.
2785 * Show the match only if it is visible on the screen.
2786 * If there isn't a match, then beep.
2787 */
2788 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002789showmatch(
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002790 int c) // char to show match for
Bram Moolenaar071d4272004-06-13 20:20:40 +00002791{
2792 pos_T *lpos, save_cursor;
2793 pos_T mpos;
2794 colnr_T vcol;
2795 long save_so;
2796 long save_siso;
2797#ifdef CURSOR_SHAPE
2798 int save_state;
2799#endif
2800 colnr_T save_dollar_vcol;
2801 char_u *p;
Bram Moolenaar375e3392019-01-31 18:26:10 +01002802 long *so = curwin->w_p_so >= 0 ? &curwin->w_p_so : &p_so;
2803 long *siso = curwin->w_p_siso >= 0 ? &curwin->w_p_siso : &p_siso;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002804
2805 /*
2806 * Only show match for chars in the 'matchpairs' option.
2807 */
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002808 // 'matchpairs' is "x:y,x:y"
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002809 for (p = curbuf->b_p_mps; *p != NUL; ++p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002810 {
2811#ifdef FEAT_RIGHTLEFT
Bram Moolenaar187d3ac2013-02-20 18:39:13 +01002812 if (PTR2CHAR(p) == c && (curwin->w_p_rl ^ p_ri))
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002813 break;
Bram Moolenaar187d3ac2013-02-20 18:39:13 +01002814#endif
Bram Moolenaar1614a142019-10-06 22:00:13 +02002815 p += mb_ptr2len(p) + 1;
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002816 if (PTR2CHAR(p) == c
Bram Moolenaar071d4272004-06-13 20:20:40 +00002817#ifdef FEAT_RIGHTLEFT
2818 && !(curwin->w_p_rl ^ p_ri)
2819#endif
2820 )
2821 break;
Bram Moolenaar1614a142019-10-06 22:00:13 +02002822 p += mb_ptr2len(p);
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002823 if (*p == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002824 return;
2825 }
Bram Moolenaar5b8cabf2021-04-02 18:55:57 +02002826 if (*p == NUL)
2827 return;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002828
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002829 if ((lpos = findmatch(NULL, NUL)) == NULL) // no match, so beep
Bram Moolenaar165bc692015-07-21 17:53:25 +02002830 vim_beep(BO_MATCH);
Bram Moolenaar187d3ac2013-02-20 18:39:13 +01002831 else if (lpos->lnum >= curwin->w_topline && lpos->lnum < curwin->w_botline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002832 {
2833 if (!curwin->w_p_wrap)
2834 getvcol(curwin, lpos, NULL, &vcol, NULL);
2835 if (curwin->w_p_wrap || (vcol >= curwin->w_leftcol
Bram Moolenaar02631462017-09-22 15:20:32 +02002836 && vcol < curwin->w_leftcol + curwin->w_width))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002837 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002838 mpos = *lpos; // save the pos, update_screen() may change it
Bram Moolenaar071d4272004-06-13 20:20:40 +00002839 save_cursor = curwin->w_cursor;
Bram Moolenaar375e3392019-01-31 18:26:10 +01002840 save_so = *so;
2841 save_siso = *siso;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002842 // Handle "$" in 'cpo': If the ')' is typed on top of the "$",
2843 // stop displaying the "$".
Bram Moolenaar76b9b362012-02-04 23:35:00 +01002844 if (dollar_vcol >= 0 && dollar_vcol == curwin->w_virtcol)
2845 dollar_vcol = -1;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002846 ++curwin->w_virtcol; // do display ')' just before "$"
2847 update_screen(VALID); // show the new char first
Bram Moolenaar071d4272004-06-13 20:20:40 +00002848
2849 save_dollar_vcol = dollar_vcol;
2850#ifdef CURSOR_SHAPE
2851 save_state = State;
2852 State = SHOWMATCH;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002853 ui_cursor_shape(); // may show different cursor shape
Bram Moolenaar071d4272004-06-13 20:20:40 +00002854#endif
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002855 curwin->w_cursor = mpos; // move to matching char
2856 *so = 0; // don't use 'scrolloff' here
2857 *siso = 0; // don't use 'sidescrolloff' here
Bram Moolenaar071d4272004-06-13 20:20:40 +00002858 showruler(FALSE);
2859 setcursor();
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002860 cursor_on(); // make sure that the cursor is shown
Bram Moolenaara338adc2018-01-31 20:51:47 +01002861 out_flush_cursor(TRUE, FALSE);
2862
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002863 // Restore dollar_vcol(), because setcursor() may call curs_rows()
2864 // which resets it if the matching position is in a previous line
2865 // and has a higher column number.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002866 dollar_vcol = save_dollar_vcol;
2867
2868 /*
2869 * brief pause, unless 'm' is present in 'cpo' and a character is
2870 * available.
2871 */
2872 if (vim_strchr(p_cpo, CPO_SHOWMATCH) != NULL)
Bram Moolenaareda1da02019-11-17 17:06:33 +01002873 ui_delay(p_mat * 100L + 8, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002874 else if (!char_avail())
Bram Moolenaareda1da02019-11-17 17:06:33 +01002875 ui_delay(p_mat * 100L + 9, FALSE);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002876 curwin->w_cursor = save_cursor; // restore cursor position
Bram Moolenaar375e3392019-01-31 18:26:10 +01002877 *so = save_so;
2878 *siso = save_siso;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002879#ifdef CURSOR_SHAPE
2880 State = save_state;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002881 ui_cursor_shape(); // may show different cursor shape
Bram Moolenaar071d4272004-06-13 20:20:40 +00002882#endif
2883 }
2884 }
2885}
2886
2887/*
Bram Moolenaar453c1922019-10-26 14:42:09 +02002888 * Check if the pattern is zero-width.
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002889 * If move is TRUE, check from the beginning of the buffer, else from position
2890 * "cur".
2891 * "direction" is FORWARD or BACKWARD.
2892 * Returns TRUE, FALSE or -1 for failure.
2893 */
2894 static int
2895is_zero_width(char_u *pattern, int move, pos_T *cur, int direction)
2896{
2897 regmmatch_T regmatch;
2898 int nmatched = 0;
2899 int result = -1;
2900 pos_T pos;
Bram Moolenaar53989552019-12-23 22:59:18 +01002901 int called_emsg_before = called_emsg;
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002902 int flag = 0;
2903
2904 if (pattern == NULL)
2905 pattern = spats[last_idx].pat;
2906
2907 if (search_regcomp(pattern, RE_SEARCH, RE_SEARCH,
2908 SEARCH_KEEP, &regmatch) == FAIL)
2909 return -1;
2910
2911 // init startcol correctly
2912 regmatch.startpos[0].col = -1;
2913 // move to match
2914 if (move)
2915 {
2916 CLEAR_POS(&pos);
2917 }
2918 else
2919 {
2920 pos = *cur;
2921 // accept a match at the cursor position
2922 flag = SEARCH_START;
2923 }
2924
2925 if (searchit(curwin, curbuf, &pos, NULL, direction, pattern, 1,
2926 SEARCH_KEEP + flag, RE_SEARCH, NULL) != FAIL)
2927 {
2928 // Zero-width pattern should match somewhere, then we can check if
2929 // start and end are in the same position.
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002930 do
2931 {
2932 regmatch.startpos[0].col++;
2933 nmatched = vim_regexec_multi(&regmatch, curwin, curbuf,
2934 pos.lnum, regmatch.startpos[0].col, NULL, NULL);
2935 if (nmatched != 0)
2936 break;
Bram Moolenaar795aaa12020-10-02 20:36:01 +02002937 } while (regmatch.regprog != NULL
2938 && direction == FORWARD ? regmatch.startpos[0].col < pos.col
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002939 : regmatch.startpos[0].col > pos.col);
2940
Bram Moolenaar53989552019-12-23 22:59:18 +01002941 if (called_emsg == called_emsg_before)
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002942 {
2943 result = (nmatched != 0
2944 && regmatch.startpos[0].lnum == regmatch.endpos[0].lnum
2945 && regmatch.startpos[0].col == regmatch.endpos[0].col);
2946 }
2947 }
2948
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002949 vim_regfree(regmatch.regprog);
2950 return result;
2951}
2952
Bram Moolenaardde0efe2012-08-23 15:53:05 +02002953
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002954/*
2955 * Find next search match under cursor, cursor at end.
2956 * Used while an operator is pending, and in Visual mode.
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002957 */
2958 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002959current_search(
2960 long count,
Bram Moolenaar5d24a222018-12-23 19:10:09 +01002961 int forward) // TRUE for forward, FALSE for backward
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002962{
Bram Moolenaar5d24a222018-12-23 19:10:09 +01002963 pos_T start_pos; // start position of the pattern match
2964 pos_T end_pos; // end position of the pattern match
2965 pos_T orig_pos; // position of the cursor at beginning
2966 pos_T pos; // position after the pattern
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002967 int i;
2968 int dir;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01002969 int result; // result of various function calls
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002970 char_u old_p_ws = p_ws;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002971 int flags = 0;
Bram Moolenaarde9149e2013-07-17 19:22:13 +02002972 pos_T save_VIsual = VIsual;
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002973 int zero_width;
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02002974 int skip_first_backward;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002975
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002976 // Correct cursor when 'selection' is exclusive
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01002977 if (VIsual_active && *p_sel == 'e' && LT_POS(VIsual, curwin->w_cursor))
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002978 dec_cursor();
2979
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02002980 // When searching forward and the cursor is at the start of the Visual
2981 // area, skip the first search backward, otherwise it doesn't move.
2982 skip_first_backward = forward && VIsual_active
2983 && LT_POS(curwin->w_cursor, VIsual);
2984
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002985 orig_pos = pos = curwin->w_cursor;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002986 if (VIsual_active)
2987 {
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002988 if (forward)
2989 incl(&pos);
2990 else
2991 decl(&pos);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002992 }
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002993
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002994 // Is the pattern is zero-width?, this time, don't care about the direction
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002995 zero_width = is_zero_width(spats[last_idx].pat, TRUE, &curwin->w_cursor,
Bram Moolenaar22ab5472017-09-26 12:28:45 +02002996 FORWARD);
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002997 if (zero_width == -1)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002998 return FAIL; // pattern not found
Bram Moolenaarba6ba362012-08-08 15:27:57 +02002999
Bram Moolenaarba6ba362012-08-08 15:27:57 +02003000 /*
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003001 * The trick is to first search backwards and then search forward again,
3002 * so that a match at the current cursor position will be correctly
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02003003 * captured. When "forward" is false do it the other way around.
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003004 */
3005 for (i = 0; i < 2; i++)
3006 {
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003007 if (forward)
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02003008 {
3009 if (i == 0 && skip_first_backward)
3010 continue;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003011 dir = i;
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02003012 }
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003013 else
3014 dir = !i;
Bram Moolenaarba6ba362012-08-08 15:27:57 +02003015
3016 flags = 0;
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02003017 if (!dir && !zero_width)
Bram Moolenaarba6ba362012-08-08 15:27:57 +02003018 flags = SEARCH_END;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01003019 end_pos = pos;
Bram Moolenaarba6ba362012-08-08 15:27:57 +02003020
Bram Moolenaar82cf7f62019-11-02 23:22:47 +01003021 // wrapping should not occur in the first round
3022 if (i == 0)
3023 p_ws = FALSE;
3024
Bram Moolenaar5d24a222018-12-23 19:10:09 +01003025 result = searchit(curwin, curbuf, &pos, &end_pos,
3026 (dir ? FORWARD : BACKWARD),
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003027 spats[last_idx].pat, (long) (i ? count : 1),
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02003028 SEARCH_KEEP | flags, RE_SEARCH, NULL);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003029
Bram Moolenaar82cf7f62019-11-02 23:22:47 +01003030 p_ws = old_p_ws;
3031
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003032 // First search may fail, but then start searching from the
3033 // beginning of the file (cursor might be on the search match)
3034 // except when Visual mode is active, so that extending the visual
3035 // selection works.
3036 if (i == 1 && !result) // not found, abort
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003037 {
3038 curwin->w_cursor = orig_pos;
3039 if (VIsual_active)
3040 VIsual = save_VIsual;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003041 return FAIL;
3042 }
Bram Moolenaar5d24a222018-12-23 19:10:09 +01003043 else if (i == 0 && !result)
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003044 {
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01003045 if (forward)
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003046 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003047 // try again from start of buffer
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01003048 CLEAR_POS(&pos);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003049 }
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01003050 else
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003051 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003052 // try again from end of buffer
3053 // searching backwards, so set pos to last line and col
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003054 pos.lnum = curwin->w_buffer->b_ml.ml_line_count;
Bram Moolenaar09168a72012-08-02 21:24:42 +02003055 pos.col = (colnr_T)STRLEN(
3056 ml_get(curwin->w_buffer->b_ml.ml_line_count));
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003057 }
3058 }
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003059 }
3060
3061 start_pos = pos;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003062
3063 if (!VIsual_active)
3064 VIsual = start_pos;
3065
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02003066 // put the cursor after the match
Bram Moolenaar5d24a222018-12-23 19:10:09 +01003067 curwin->w_cursor = end_pos;
Bram Moolenaar453c1922019-10-26 14:42:09 +02003068 if (LT_POS(VIsual, end_pos) && forward)
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02003069 {
3070 if (skip_first_backward)
3071 // put the cursor on the start of the match
3072 curwin->w_cursor = pos;
3073 else
3074 // put the cursor on last character of match
3075 dec_cursor();
3076 }
Bram Moolenaar28f224b2020-10-10 16:45:25 +02003077 else if (VIsual_active && LT_POS(curwin->w_cursor, VIsual) && forward)
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02003078 curwin->w_cursor = pos; // put the cursor on the start of the match
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003079 VIsual_active = TRUE;
3080 VIsual_mode = 'v';
3081
Bram Moolenaarb7633612019-02-10 21:48:25 +01003082 if (*p_sel == 'e')
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003083 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003084 // Correction for exclusive selection depends on the direction.
Bram Moolenaarb7633612019-02-10 21:48:25 +01003085 if (forward && LTOREQ_POS(VIsual, curwin->w_cursor))
3086 inc_cursor();
3087 else if (!forward && LTOREQ_POS(curwin->w_cursor, VIsual))
3088 inc(&VIsual);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003089 }
3090
3091#ifdef FEAT_FOLDING
3092 if (fdo_flags & FDO_SEARCH && KeyTyped)
3093 foldOpenCursor();
3094#endif
3095
3096 may_start_select('c');
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003097 setmouse();
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003098#ifdef FEAT_CLIPBOARD
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003099 // Make sure the clipboard gets updated. Needed because start and
3100 // end are still the same, and the selection needs to be owned
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003101 clip_star.vmode = NUL;
3102#endif
3103 redraw_curbuf_later(INVERTED);
3104 showmode();
3105
3106 return OK;
3107}
Bram Moolenaardde0efe2012-08-23 15:53:05 +02003108
Bram Moolenaar071d4272004-06-13 20:20:40 +00003109#if defined(FEAT_LISP) || defined(FEAT_CINDENT) || defined(FEAT_TEXTOBJ) \
3110 || defined(PROTO)
3111/*
3112 * return TRUE if line 'lnum' is empty or has white chars only.
3113 */
3114 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01003115linewhite(linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003116{
3117 char_u *p;
3118
3119 p = skipwhite(ml_get(lnum));
3120 return (*p == NUL);
3121}
3122#endif
3123
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003124/*
3125 * Add the search count "[3/19]" to "msgbuf".
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003126 * See update_search_stat() for other arguments.
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003127 */
3128 static void
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003129cmdline_search_stat(
3130 int dirc,
3131 pos_T *pos,
3132 pos_T *cursor_pos,
3133 int show_top_bot_msg,
3134 char_u *msgbuf,
3135 int recompute,
3136 int maxcount,
3137 long timeout)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003138{
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003139 searchstat_T stat;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003140
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003141 update_search_stat(dirc, pos, cursor_pos, &stat, recompute, maxcount,
3142 timeout);
3143 if (stat.cur > 0)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003144 {
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003145 char t[SEARCH_STAT_BUF_LEN];
Bram Moolenaare2ad8262019-05-24 13:22:22 +02003146 size_t len;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003147
3148#ifdef FEAT_RIGHTLEFT
3149 if (curwin->w_p_rl && *curwin->w_p_rlc == 's')
3150 {
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003151 if (stat.incomplete == 1)
Bram Moolenaarb6cb26f2019-05-07 21:34:37 +02003152 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[?/??]");
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003153 else if (stat.cnt > maxcount && stat.cur > maxcount)
3154 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[>%d/>%d]",
3155 maxcount, maxcount);
3156 else if (stat.cnt > maxcount)
3157 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[>%d/%d]",
3158 maxcount, stat.cur);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003159 else
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003160 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[%d/%d]",
3161 stat.cnt, stat.cur);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003162 }
3163 else
3164#endif
3165 {
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003166 if (stat.incomplete == 1)
Bram Moolenaarb6cb26f2019-05-07 21:34:37 +02003167 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[?/??]");
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003168 else if (stat.cnt > maxcount && stat.cur > maxcount)
3169 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[>%d/>%d]",
3170 maxcount, maxcount);
3171 else if (stat.cnt > maxcount)
3172 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[%d/>%d]",
3173 stat.cur, maxcount);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003174 else
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003175 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[%d/%d]",
3176 stat.cur, stat.cnt);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003177 }
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02003178
3179 len = STRLEN(t);
Bram Moolenaardc6855a2019-05-18 19:26:29 +02003180 if (show_top_bot_msg && len + 2 < SEARCH_STAT_BUF_LEN)
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02003181 {
Bram Moolenaar16b58ae2019-09-06 20:40:21 +02003182 mch_memmove(t + 2, t, len);
3183 t[0] = 'W';
3184 t[1] = ' ';
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02003185 len += 2;
3186 }
3187
3188 mch_memmove(msgbuf + STRLEN(msgbuf) - len, t, len);
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003189 if (dirc == '?' && stat.cur == maxcount + 1)
3190 stat.cur = -1;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003191
Bram Moolenaar984f0312019-05-24 13:11:47 +02003192 // keep the message even after redraw, but don't put in history
3193 msg_hist_off = TRUE;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003194 give_warning(msgbuf, FALSE);
Bram Moolenaar984f0312019-05-24 13:11:47 +02003195 msg_hist_off = FALSE;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003196 }
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003197}
3198
3199/*
3200 * Add the search count information to "stat".
3201 * "stat" must not be NULL.
3202 * When "recompute" is TRUE always recompute the numbers.
3203 * dirc == 0: don't find the next/previous match (only set the result to "stat")
3204 * dirc == '/': find the next match
3205 * dirc == '?': find the previous match
3206 */
3207 static void
3208update_search_stat(
3209 int dirc,
3210 pos_T *pos,
3211 pos_T *cursor_pos,
3212 searchstat_T *stat,
3213 int recompute,
3214 int maxcount,
Bram Moolenaarf9ca08e2020-06-01 18:56:03 +02003215 long timeout UNUSED)
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003216{
3217 int save_ws = p_ws;
3218 int wraparound = FALSE;
3219 pos_T p = (*pos);
Bram Moolenaar14681622020-06-03 22:57:39 +02003220 static pos_T lastpos = {0, 0, 0};
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003221 static int cur = 0;
3222 static int cnt = 0;
3223 static int exact_match = FALSE;
3224 static int incomplete = 0;
3225 static int last_maxcount = SEARCH_STAT_DEF_MAX_COUNT;
3226 static int chgtick = 0;
3227 static char_u *lastpat = NULL;
3228 static buf_T *lbuf = NULL;
3229#ifdef FEAT_RELTIME
3230 proftime_T start;
3231#endif
3232
3233 vim_memset(stat, 0, sizeof(searchstat_T));
3234
3235 if (dirc == 0 && !recompute && !EMPTY_POS(lastpos))
3236 {
3237 stat->cur = cur;
3238 stat->cnt = cnt;
3239 stat->exact_match = exact_match;
3240 stat->incomplete = incomplete;
3241 stat->last_maxcount = last_maxcount;
3242 return;
3243 }
3244 last_maxcount = maxcount;
3245
3246 wraparound = ((dirc == '?' && LT_POS(lastpos, p))
3247 || (dirc == '/' && LT_POS(p, lastpos)));
3248
3249 // If anything relevant changed the count has to be recomputed.
3250 // MB_STRNICMP ignores case, but we should not ignore case.
3251 // Unfortunately, there is no MB_STRNICMP function.
3252 // XXX: above comment should be "no MB_STRCMP function" ?
3253 if (!(chgtick == CHANGEDTICK(curbuf)
3254 && MB_STRNICMP(lastpat, spats[last_idx].pat, STRLEN(lastpat)) == 0
3255 && STRLEN(lastpat) == STRLEN(spats[last_idx].pat)
3256 && EQUAL_POS(lastpos, *cursor_pos)
3257 && lbuf == curbuf) || wraparound || cur < 0
3258 || (maxcount > 0 && cur > maxcount) || recompute)
3259 {
3260 cur = 0;
3261 cnt = 0;
3262 exact_match = FALSE;
3263 incomplete = 0;
3264 CLEAR_POS(&lastpos);
3265 lbuf = curbuf;
3266 }
3267
3268 if (EQUAL_POS(lastpos, *cursor_pos) && !wraparound
3269 && (dirc == 0 || dirc == '/' ? cur < cnt : cur > 0))
3270 cur += dirc == 0 ? 0 : dirc == '/' ? 1 : -1;
3271 else
3272 {
3273 int done_search = FALSE;
3274 pos_T endpos = {0, 0, 0};
3275
3276 p_ws = FALSE;
3277#ifdef FEAT_RELTIME
3278 if (timeout > 0)
3279 profile_setlimit(timeout, &start);
3280#endif
3281 while (!got_int && searchit(curwin, curbuf, &lastpos, &endpos,
3282 FORWARD, NULL, 1, SEARCH_KEEP, RE_LAST, NULL) != FAIL)
3283 {
3284 done_search = TRUE;
3285#ifdef FEAT_RELTIME
3286 // Stop after passing the time limit.
3287 if (timeout > 0 && profile_passed_limit(&start))
3288 {
3289 incomplete = 1;
3290 break;
3291 }
3292#endif
3293 cnt++;
3294 if (LTOREQ_POS(lastpos, p))
3295 {
3296 cur = cnt;
Bram Moolenaar57f75a52020-06-02 22:06:21 +02003297 if (LT_POS(p, endpos))
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003298 exact_match = TRUE;
3299 }
3300 fast_breakcheck();
3301 if (maxcount > 0 && cnt > maxcount)
3302 {
3303 incomplete = 2; // max count exceeded
3304 break;
3305 }
3306 }
3307 if (got_int)
3308 cur = -1; // abort
3309 if (done_search)
3310 {
3311 vim_free(lastpat);
3312 lastpat = vim_strsave(spats[last_idx].pat);
3313 chgtick = CHANGEDTICK(curbuf);
3314 lbuf = curbuf;
3315 lastpos = p;
3316 }
3317 }
3318 stat->cur = cur;
3319 stat->cnt = cnt;
3320 stat->exact_match = exact_match;
3321 stat->incomplete = incomplete;
3322 stat->last_maxcount = last_maxcount;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003323 p_ws = save_ws;
3324}
3325
Bram Moolenaar071d4272004-06-13 20:20:40 +00003326#if defined(FEAT_FIND_ID) || defined(PROTO)
3327/*
3328 * Find identifiers or defines in included files.
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01003329 * If p_ic && (compl_cont_status & CONT_SOL) then ptr must be in lowercase.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003330 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003331 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01003332find_pattern_in_path(
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003333 char_u *ptr, // pointer to search pattern
3334 int dir UNUSED, // direction of expansion
3335 int len, // length of search pattern
3336 int whole, // match whole words only
3337 int skip_comments, // don't match inside comments
3338 int type, // Type of search; are we looking for a type?
3339 // a macro?
Bram Moolenaar764b23c2016-01-30 21:10:09 +01003340 long count,
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003341 int action, // What to do when we find it
3342 linenr_T start_lnum, // first line to start searching
3343 linenr_T end_lnum) // last line for searching
Bram Moolenaar071d4272004-06-13 20:20:40 +00003344{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003345 SearchedFile *files; // Stack of included files
3346 SearchedFile *bigger; // When we need more space
Bram Moolenaar071d4272004-06-13 20:20:40 +00003347 int max_path_depth = 50;
3348 long match_count = 1;
3349
3350 char_u *pat;
3351 char_u *new_fname;
3352 char_u *curr_fname = curbuf->b_fname;
3353 char_u *prev_fname = NULL;
3354 linenr_T lnum;
3355 int depth;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003356 int depth_displayed; // For type==CHECK_PATH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003357 int old_files;
3358 int already_searched;
3359 char_u *file_line;
3360 char_u *line;
3361 char_u *p;
3362 char_u save_char;
3363 int define_matched;
3364 regmatch_T regmatch;
3365 regmatch_T incl_regmatch;
3366 regmatch_T def_regmatch;
3367 int matched = FALSE;
3368 int did_show = FALSE;
3369 int found = FALSE;
3370 int i;
3371 char_u *already = NULL;
3372 char_u *startp = NULL;
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003373 char_u *inc_opt = NULL;
Bram Moolenaar4033c552017-09-16 20:54:51 +02003374#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003375 win_T *curwin_save = NULL;
3376#endif
3377
3378 regmatch.regprog = NULL;
3379 incl_regmatch.regprog = NULL;
3380 def_regmatch.regprog = NULL;
3381
3382 file_line = alloc(LSIZE);
3383 if (file_line == NULL)
3384 return;
3385
Bram Moolenaar071d4272004-06-13 20:20:40 +00003386 if (type != CHECK_PATH && type != FIND_DEFINE
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003387 // when CONT_SOL is set compare "ptr" with the beginning of the line
3388 // is faster than quote_meta/regcomp/regexec "ptr" -- Acevedo
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003389 && !(compl_cont_status & CONT_SOL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003390 {
3391 pat = alloc(len + 5);
3392 if (pat == NULL)
3393 goto fpip_end;
3394 sprintf((char *)pat, whole ? "\\<%.*s\\>" : "%.*s", len, ptr);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003395 // ignore case according to p_ic, p_scs and pat
Bram Moolenaar071d4272004-06-13 20:20:40 +00003396 regmatch.rm_ic = ignorecase(pat);
Bram Moolenaarf4e20992020-12-21 19:59:08 +01003397 regmatch.regprog = vim_regcomp(pat, magic_isset() ? RE_MAGIC : 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003398 vim_free(pat);
3399 if (regmatch.regprog == NULL)
3400 goto fpip_end;
3401 }
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003402 inc_opt = (*curbuf->b_p_inc == NUL) ? p_inc : curbuf->b_p_inc;
3403 if (*inc_opt != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003404 {
Bram Moolenaarf4e20992020-12-21 19:59:08 +01003405 incl_regmatch.regprog = vim_regcomp(inc_opt,
3406 magic_isset() ? RE_MAGIC : 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003407 if (incl_regmatch.regprog == NULL)
3408 goto fpip_end;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003409 incl_regmatch.rm_ic = FALSE; // don't ignore case in incl. pat.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003410 }
3411 if (type == FIND_DEFINE && (*curbuf->b_p_def != NUL || *p_def != NUL))
3412 {
3413 def_regmatch.regprog = vim_regcomp(*curbuf->b_p_def == NUL
Bram Moolenaarf4e20992020-12-21 19:59:08 +01003414 ? p_def : curbuf->b_p_def,
3415 magic_isset() ? RE_MAGIC : 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003416 if (def_regmatch.regprog == NULL)
3417 goto fpip_end;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003418 def_regmatch.rm_ic = FALSE; // don't ignore case in define pat.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003419 }
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003420 files = lalloc_clear(max_path_depth * sizeof(SearchedFile), TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003421 if (files == NULL)
3422 goto fpip_end;
3423 old_files = max_path_depth;
3424 depth = depth_displayed = -1;
3425
3426 lnum = start_lnum;
3427 if (end_lnum > curbuf->b_ml.ml_line_count)
3428 end_lnum = curbuf->b_ml.ml_line_count;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003429 if (lnum > end_lnum) // do at least one line
Bram Moolenaar071d4272004-06-13 20:20:40 +00003430 lnum = end_lnum;
3431 line = ml_get(lnum);
3432
3433 for (;;)
3434 {
3435 if (incl_regmatch.regprog != NULL
3436 && vim_regexec(&incl_regmatch, line, (colnr_T)0))
3437 {
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003438 char_u *p_fname = (curr_fname == curbuf->b_fname)
3439 ? curbuf->b_ffname : curr_fname;
3440
3441 if (inc_opt != NULL && strstr((char *)inc_opt, "\\zs") != NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003442 // Use text from '\zs' to '\ze' (or end) of 'include'.
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003443 new_fname = find_file_name_in_path(incl_regmatch.startp[0],
Bram Moolenaarc84e3c12013-07-03 22:28:36 +02003444 (int)(incl_regmatch.endp[0] - incl_regmatch.startp[0]),
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003445 FNAME_EXP|FNAME_INCL|FNAME_REL, 1L, p_fname);
3446 else
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003447 // Use text after match with 'include'.
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003448 new_fname = file_name_in_line(incl_regmatch.endp[0], 0,
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003449 FNAME_EXP|FNAME_INCL|FNAME_REL, 1L, p_fname, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003450 already_searched = FALSE;
3451 if (new_fname != NULL)
3452 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003453 // Check whether we have already searched in this file
Bram Moolenaar071d4272004-06-13 20:20:40 +00003454 for (i = 0;; i++)
3455 {
3456 if (i == depth + 1)
3457 i = old_files;
3458 if (i == max_path_depth)
3459 break;
Bram Moolenaar99499b12019-05-23 21:35:48 +02003460 if (fullpathcmp(new_fname, files[i].name, TRUE, TRUE)
3461 & FPC_SAME)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003462 {
3463 if (type != CHECK_PATH &&
3464 action == ACTION_SHOW_ALL && files[i].matched)
3465 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003466 msg_putchar('\n'); // cursor below last one
3467 if (!got_int) // don't display if 'q'
3468 // typed at "--more--"
3469 // message
Bram Moolenaar071d4272004-06-13 20:20:40 +00003470 {
3471 msg_home_replace_hl(new_fname);
Bram Moolenaar32526b32019-01-19 17:43:09 +01003472 msg_puts(_(" (includes previously listed match)"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003473 prev_fname = NULL;
3474 }
3475 }
Bram Moolenaard23a8232018-02-10 18:45:26 +01003476 VIM_CLEAR(new_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003477 already_searched = TRUE;
3478 break;
3479 }
3480 }
3481 }
3482
3483 if (type == CHECK_PATH && (action == ACTION_SHOW_ALL
3484 || (new_fname == NULL && !already_searched)))
3485 {
3486 if (did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003487 msg_putchar('\n'); // cursor below last one
Bram Moolenaar071d4272004-06-13 20:20:40 +00003488 else
3489 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003490 gotocmdline(TRUE); // cursor at status line
Bram Moolenaar32526b32019-01-19 17:43:09 +01003491 msg_puts_title(_("--- Included files "));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003492 if (action != ACTION_SHOW_ALL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003493 msg_puts_title(_("not found "));
3494 msg_puts_title(_("in path ---\n"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003495 }
3496 did_show = TRUE;
3497 while (depth_displayed < depth && !got_int)
3498 {
3499 ++depth_displayed;
3500 for (i = 0; i < depth_displayed; i++)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003501 msg_puts(" ");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003502 msg_home_replace(files[depth_displayed].name);
Bram Moolenaar32526b32019-01-19 17:43:09 +01003503 msg_puts(" -->\n");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003504 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003505 if (!got_int) // don't display if 'q' typed
3506 // for "--more--" message
Bram Moolenaar071d4272004-06-13 20:20:40 +00003507 {
3508 for (i = 0; i <= depth_displayed; i++)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003509 msg_puts(" ");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003510 if (new_fname != NULL)
3511 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003512 // using "new_fname" is more reliable, e.g., when
3513 // 'includeexpr' is set.
Bram Moolenaar8820b482017-03-16 17:23:31 +01003514 msg_outtrans_attr(new_fname, HL_ATTR(HLF_D));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003515 }
3516 else
3517 {
3518 /*
3519 * Isolate the file name.
3520 * Include the surrounding "" or <> if present.
3521 */
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003522 if (inc_opt != NULL
3523 && strstr((char *)inc_opt, "\\zs") != NULL)
3524 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003525 // pattern contains \zs, use the match
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003526 p = incl_regmatch.startp[0];
3527 i = (int)(incl_regmatch.endp[0]
3528 - incl_regmatch.startp[0]);
3529 }
3530 else
3531 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003532 // find the file name after the end of the match
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003533 for (p = incl_regmatch.endp[0];
3534 *p && !vim_isfilec(*p); p++)
3535 ;
3536 for (i = 0; vim_isfilec(p[i]); i++)
3537 ;
3538 }
3539
Bram Moolenaar071d4272004-06-13 20:20:40 +00003540 if (i == 0)
3541 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003542 // Nothing found, use the rest of the line.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003543 p = incl_regmatch.endp[0];
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00003544 i = (int)STRLEN(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003545 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003546 // Avoid checking before the start of the line, can
3547 // happen if \zs appears in the regexp.
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003548 else if (p > line)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003549 {
3550 if (p[-1] == '"' || p[-1] == '<')
3551 {
3552 --p;
3553 ++i;
3554 }
3555 if (p[i] == '"' || p[i] == '>')
3556 ++i;
3557 }
3558 save_char = p[i];
3559 p[i] = NUL;
Bram Moolenaar8820b482017-03-16 17:23:31 +01003560 msg_outtrans_attr(p, HL_ATTR(HLF_D));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003561 p[i] = save_char;
3562 }
3563
3564 if (new_fname == NULL && action == ACTION_SHOW_ALL)
3565 {
3566 if (already_searched)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003567 msg_puts(_(" (Already listed)"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003568 else
Bram Moolenaar32526b32019-01-19 17:43:09 +01003569 msg_puts(_(" NOT FOUND"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003570 }
3571 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003572 out_flush(); // output each line directly
Bram Moolenaar071d4272004-06-13 20:20:40 +00003573 }
3574
3575 if (new_fname != NULL)
3576 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003577 // Push the new file onto the file stack
Bram Moolenaar071d4272004-06-13 20:20:40 +00003578 if (depth + 1 == old_files)
3579 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003580 bigger = ALLOC_MULT(SearchedFile, max_path_depth * 2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003581 if (bigger != NULL)
3582 {
3583 for (i = 0; i <= depth; i++)
3584 bigger[i] = files[i];
3585 for (i = depth + 1; i < old_files + max_path_depth; i++)
3586 {
3587 bigger[i].fp = NULL;
3588 bigger[i].name = NULL;
3589 bigger[i].lnum = 0;
3590 bigger[i].matched = FALSE;
3591 }
3592 for (i = old_files; i < max_path_depth; i++)
3593 bigger[i + max_path_depth] = files[i];
3594 old_files += max_path_depth;
3595 max_path_depth *= 2;
3596 vim_free(files);
3597 files = bigger;
3598 }
3599 }
3600 if ((files[depth + 1].fp = mch_fopen((char *)new_fname, "r"))
3601 == NULL)
3602 vim_free(new_fname);
3603 else
3604 {
3605 if (++depth == old_files)
3606 {
3607 /*
3608 * lalloc() for 'bigger' must have failed above. We
3609 * will forget one of our already visited files now.
3610 */
3611 vim_free(files[old_files].name);
3612 ++old_files;
3613 }
3614 files[depth].name = curr_fname = new_fname;
3615 files[depth].lnum = 0;
3616 files[depth].matched = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003617 if (action == ACTION_EXPAND)
3618 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003619 msg_hist_off = TRUE; // reset in msg_trunc_attr()
Bram Moolenaar555b2802005-05-19 21:08:39 +00003620 vim_snprintf((char*)IObuff, IOSIZE,
3621 _("Scanning included file: %s"),
3622 (char *)new_fname);
Bram Moolenaar32526b32019-01-19 17:43:09 +01003623 msg_trunc_attr((char *)IObuff, TRUE, HL_ATTR(HLF_R));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003624 }
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003625 else if (p_verbose >= 5)
Bram Moolenaar87b5ca52006-03-04 21:55:31 +00003626 {
3627 verbose_enter();
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003628 smsg(_("Searching included file %s"),
Bram Moolenaar87b5ca52006-03-04 21:55:31 +00003629 (char *)new_fname);
3630 verbose_leave();
3631 }
3632
Bram Moolenaar071d4272004-06-13 20:20:40 +00003633 }
3634 }
3635 }
3636 else
3637 {
3638 /*
3639 * Check if the line is a define (type == FIND_DEFINE)
3640 */
3641 p = line;
3642search_line:
3643 define_matched = FALSE;
3644 if (def_regmatch.regprog != NULL
3645 && vim_regexec(&def_regmatch, line, (colnr_T)0))
3646 {
3647 /*
3648 * Pattern must be first identifier after 'define', so skip
3649 * to that position before checking for match of pattern. Also
3650 * don't let it match beyond the end of this identifier.
3651 */
3652 p = def_regmatch.endp[0];
3653 while (*p && !vim_iswordc(*p))
3654 p++;
3655 define_matched = TRUE;
3656 }
3657
3658 /*
3659 * Look for a match. Don't do this if we are looking for a
3660 * define and this line didn't match define_prog above.
3661 */
3662 if (def_regmatch.regprog == NULL || define_matched)
3663 {
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003664 if (define_matched || (compl_cont_status & CONT_SOL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003665 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003666 // compare the first "len" chars from "ptr"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003667 startp = skipwhite(p);
3668 if (p_ic)
3669 matched = !MB_STRNICMP(startp, ptr, len);
3670 else
3671 matched = !STRNCMP(startp, ptr, len);
3672 if (matched && define_matched && whole
3673 && vim_iswordc(startp[len]))
3674 matched = FALSE;
3675 }
3676 else if (regmatch.regprog != NULL
3677 && vim_regexec(&regmatch, line, (colnr_T)(p - line)))
3678 {
3679 matched = TRUE;
3680 startp = regmatch.startp[0];
3681 /*
3682 * Check if the line is not a comment line (unless we are
3683 * looking for a define). A line starting with "# define"
3684 * is not considered to be a comment line.
3685 */
3686 if (!define_matched && skip_comments)
3687 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003688 if ((*line != '#' ||
3689 STRNCMP(skipwhite(line + 1), "define", 6) != 0)
Bram Moolenaar81340392012-06-06 16:12:59 +02003690 && get_leader_len(line, NULL, FALSE, TRUE))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003691 matched = FALSE;
3692
3693 /*
3694 * Also check for a "/ *" or "/ /" before the match.
3695 * Skips lines like "int backwards; / * normal index
3696 * * /" when looking for "normal".
3697 * Note: Doesn't skip "/ *" in comments.
3698 */
3699 p = skipwhite(line);
3700 if (matched
3701 || (p[0] == '/' && p[1] == '*') || p[0] == '*')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003702 for (p = line; *p && p < startp; ++p)
3703 {
3704 if (matched
3705 && p[0] == '/'
3706 && (p[1] == '*' || p[1] == '/'))
3707 {
3708 matched = FALSE;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003709 // After "//" all text is comment
Bram Moolenaar071d4272004-06-13 20:20:40 +00003710 if (p[1] == '/')
3711 break;
3712 ++p;
3713 }
3714 else if (!matched && p[0] == '*' && p[1] == '/')
3715 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003716 // Can find match after "* /".
Bram Moolenaar071d4272004-06-13 20:20:40 +00003717 matched = TRUE;
3718 ++p;
3719 }
3720 }
3721 }
3722 }
3723 }
3724 }
3725 if (matched)
3726 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003727 if (action == ACTION_EXPAND)
3728 {
Bram Moolenaard9eefe32019-04-06 14:22:21 +02003729 int cont_s_ipos = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003730 int add_r;
3731 char_u *aux;
3732
3733 if (depth == -1 && lnum == curwin->w_cursor.lnum)
3734 break;
3735 found = TRUE;
3736 aux = p = startp;
Bram Moolenaar4be06f92005-07-29 22:36:03 +00003737 if (compl_cont_status & CONT_ADDING)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003738 {
Bram Moolenaar4be06f92005-07-29 22:36:03 +00003739 p += compl_length;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003740 if (vim_iswordp(p))
3741 goto exit_matched;
3742 p = find_word_start(p);
3743 }
3744 p = find_word_end(p);
3745 i = (int)(p - aux);
3746
Bram Moolenaar4be06f92005-07-29 22:36:03 +00003747 if ((compl_cont_status & CONT_ADDING) && i == compl_length)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003748 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003749 // IOSIZE > compl_length, so the STRNCPY works
Bram Moolenaar071d4272004-06-13 20:20:40 +00003750 STRNCPY(IObuff, aux, i);
Bram Moolenaar89d40322006-08-29 15:30:07 +00003751
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003752 // Get the next line: when "depth" < 0 from the current
3753 // buffer, otherwise from the included file. Jump to
3754 // exit_matched when past the last line.
Bram Moolenaar89d40322006-08-29 15:30:07 +00003755 if (depth < 0)
3756 {
3757 if (lnum >= end_lnum)
3758 goto exit_matched;
3759 line = ml_get(++lnum);
3760 }
3761 else if (vim_fgets(line = file_line,
3762 LSIZE, files[depth].fp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003763 goto exit_matched;
3764
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003765 // we read a line, set "already" to check this "line" later
3766 // if depth >= 0 we'll increase files[depth].lnum far
Bram Moolenaar8e7d6222020-12-18 19:49:56 +01003767 // below -- Acevedo
Bram Moolenaar071d4272004-06-13 20:20:40 +00003768 already = aux = p = skipwhite(line);
3769 p = find_word_start(p);
3770 p = find_word_end(p);
3771 if (p > aux)
3772 {
3773 if (*aux != ')' && IObuff[i-1] != TAB)
3774 {
3775 if (IObuff[i-1] != ' ')
3776 IObuff[i++] = ' ';
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003777 // IObuf =~ "\(\k\|\i\).* ", thus i >= 2
Bram Moolenaar071d4272004-06-13 20:20:40 +00003778 if (p_js
3779 && (IObuff[i-2] == '.'
3780 || (vim_strchr(p_cpo, CPO_JOINSP) == NULL
3781 && (IObuff[i-2] == '?'
3782 || IObuff[i-2] == '!'))))
3783 IObuff[i++] = ' ';
3784 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003785 // copy as much as possible of the new word
Bram Moolenaar071d4272004-06-13 20:20:40 +00003786 if (p - aux >= IOSIZE - i)
3787 p = aux + IOSIZE - i - 1;
3788 STRNCPY(IObuff + i, aux, p - aux);
3789 i += (int)(p - aux);
Bram Moolenaard9eefe32019-04-06 14:22:21 +02003790 cont_s_ipos = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003791 }
3792 IObuff[i] = NUL;
3793 aux = IObuff;
3794
Bram Moolenaar4be06f92005-07-29 22:36:03 +00003795 if (i == compl_length)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003796 goto exit_matched;
3797 }
3798
Bram Moolenaare8c3a142006-08-29 14:30:35 +00003799 add_r = ins_compl_add_infercase(aux, i, p_ic,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003800 curr_fname == curbuf->b_fname ? NULL : curr_fname,
Bram Moolenaard9eefe32019-04-06 14:22:21 +02003801 dir, cont_s_ipos);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003802 if (add_r == OK)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003803 // if dir was BACKWARD then honor it just once
Bram Moolenaar071d4272004-06-13 20:20:40 +00003804 dir = FORWARD;
Bram Moolenaar572cb562005-08-05 21:35:02 +00003805 else if (add_r == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003806 break;
3807 }
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003808 else if (action == ACTION_SHOW_ALL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003809 {
3810 found = TRUE;
3811 if (!did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003812 gotocmdline(TRUE); // cursor at status line
Bram Moolenaar071d4272004-06-13 20:20:40 +00003813 if (curr_fname != prev_fname)
3814 {
3815 if (did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003816 msg_putchar('\n'); // cursor below last one
3817 if (!got_int) // don't display if 'q' typed
3818 // at "--more--" message
Bram Moolenaar071d4272004-06-13 20:20:40 +00003819 msg_home_replace_hl(curr_fname);
3820 prev_fname = curr_fname;
3821 }
3822 did_show = TRUE;
3823 if (!got_int)
3824 show_pat_in_path(line, type, TRUE, action,
3825 (depth == -1) ? NULL : files[depth].fp,
3826 (depth == -1) ? &lnum : &files[depth].lnum,
3827 match_count++);
3828
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003829 // Set matched flag for this file and all the ones that
3830 // include it
Bram Moolenaar071d4272004-06-13 20:20:40 +00003831 for (i = 0; i <= depth; ++i)
3832 files[i].matched = TRUE;
3833 }
3834 else if (--count <= 0)
3835 {
3836 found = TRUE;
3837 if (depth == -1 && lnum == curwin->w_cursor.lnum
Bram Moolenaar4033c552017-09-16 20:54:51 +02003838#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003839 && g_do_tagpreview == 0
3840#endif
3841 )
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003842 emsg(_("E387: Match is on current line"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003843 else if (action == ACTION_SHOW)
3844 {
3845 show_pat_in_path(line, type, did_show, action,
3846 (depth == -1) ? NULL : files[depth].fp,
3847 (depth == -1) ? &lnum : &files[depth].lnum, 1L);
3848 did_show = TRUE;
3849 }
3850 else
3851 {
3852#ifdef FEAT_GUI
3853 need_mouse_correct = TRUE;
3854#endif
Bram Moolenaar4033c552017-09-16 20:54:51 +02003855#if defined(FEAT_QUICKFIX)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003856 // ":psearch" uses the preview window
Bram Moolenaar071d4272004-06-13 20:20:40 +00003857 if (g_do_tagpreview != 0)
3858 {
3859 curwin_save = curwin;
Bram Moolenaar576a4a62019-08-18 15:25:17 +02003860 prepare_tagpreview(TRUE, TRUE, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003861 }
3862#endif
3863 if (action == ACTION_SPLIT)
3864 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003865 if (win_split(0, 0) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003866 break;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02003867 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003868 }
3869 if (depth == -1)
3870 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003871 // match in current file
Bram Moolenaar4033c552017-09-16 20:54:51 +02003872#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003873 if (g_do_tagpreview != 0)
3874 {
Bram Moolenaar92bb83e2021-02-03 23:04:46 +01003875 if (!win_valid(curwin_save))
3876 break;
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02003877 if (!GETFILE_SUCCESS(getfile(
Bram Moolenaarc31f9ae2017-07-23 22:02:02 +02003878 curwin_save->w_buffer->b_fnum, NULL,
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02003879 NULL, TRUE, lnum, FALSE)))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003880 break; // failed to jump to file
Bram Moolenaar071d4272004-06-13 20:20:40 +00003881 }
3882 else
3883#endif
3884 setpcmark();
3885 curwin->w_cursor.lnum = lnum;
Bram Moolenaarc31f9ae2017-07-23 22:02:02 +02003886 check_cursor();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003887 }
3888 else
3889 {
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02003890 if (!GETFILE_SUCCESS(getfile(
3891 0, files[depth].name, NULL, TRUE,
3892 files[depth].lnum, FALSE)))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003893 break; // failed to jump to file
3894 // autocommands may have changed the lnum, we don't
3895 // want that here
Bram Moolenaar071d4272004-06-13 20:20:40 +00003896 curwin->w_cursor.lnum = files[depth].lnum;
3897 }
3898 }
3899 if (action != ACTION_SHOW)
3900 {
Bram Moolenaarfe81d452009-04-22 14:44:41 +00003901 curwin->w_cursor.col = (colnr_T)(startp - line);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003902 curwin->w_set_curswant = TRUE;
3903 }
3904
Bram Moolenaar4033c552017-09-16 20:54:51 +02003905#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003906 if (g_do_tagpreview != 0
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00003907 && curwin != curwin_save && win_valid(curwin_save))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003908 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003909 // Return cursor to where we were
Bram Moolenaar071d4272004-06-13 20:20:40 +00003910 validate_cursor();
3911 redraw_later(VALID);
3912 win_enter(curwin_save, TRUE);
3913 }
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +01003914# ifdef FEAT_PROP_POPUP
Bram Moolenaar1b6d9c42019-08-05 21:52:04 +02003915 else if (WIN_IS_POPUP(curwin))
3916 // can't keep focus in popup window
3917 win_enter(firstwin, TRUE);
3918# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003919#endif
3920 break;
3921 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003922exit_matched:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003923 matched = FALSE;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003924 // look for other matches in the rest of the line if we
3925 // are not at the end of it already
Bram Moolenaar071d4272004-06-13 20:20:40 +00003926 if (def_regmatch.regprog == NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003927 && action == ACTION_EXPAND
Bram Moolenaar4be06f92005-07-29 22:36:03 +00003928 && !(compl_cont_status & CONT_SOL)
Bram Moolenaarfe81d452009-04-22 14:44:41 +00003929 && *startp != NUL
Bram Moolenaar1614a142019-10-06 22:00:13 +02003930 && *(p = startp + mb_ptr2len(startp)) != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003931 goto search_line;
3932 }
3933 line_breakcheck();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003934 if (action == ACTION_EXPAND)
Bram Moolenaar472e8592016-10-15 17:06:47 +02003935 ins_compl_check_keys(30, FALSE);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003936 if (got_int || ins_compl_interrupted())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003937 break;
3938
3939 /*
3940 * Read the next line. When reading an included file and encountering
3941 * end-of-file, close the file and continue in the file that included
3942 * it.
3943 */
3944 while (depth >= 0 && !already
3945 && vim_fgets(line = file_line, LSIZE, files[depth].fp))
3946 {
3947 fclose(files[depth].fp);
3948 --old_files;
3949 files[old_files].name = files[depth].name;
3950 files[old_files].matched = files[depth].matched;
3951 --depth;
3952 curr_fname = (depth == -1) ? curbuf->b_fname
3953 : files[depth].name;
3954 if (depth < depth_displayed)
3955 depth_displayed = depth;
3956 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003957 if (depth >= 0) // we could read the line
Bram Moolenaarc84e3c12013-07-03 22:28:36 +02003958 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003959 files[depth].lnum++;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003960 // Remove any CR and LF from the line.
Bram Moolenaarc84e3c12013-07-03 22:28:36 +02003961 i = (int)STRLEN(line);
3962 if (i > 0 && line[i - 1] == '\n')
3963 line[--i] = NUL;
3964 if (i > 0 && line[i - 1] == '\r')
3965 line[--i] = NUL;
3966 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003967 else if (!already)
3968 {
3969 if (++lnum > end_lnum)
3970 break;
3971 line = ml_get(lnum);
3972 }
3973 already = NULL;
3974 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003975 // End of big for (;;) loop.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003976
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003977 // Close any files that are still open.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003978 for (i = 0; i <= depth; i++)
3979 {
3980 fclose(files[i].fp);
3981 vim_free(files[i].name);
3982 }
3983 for (i = old_files; i < max_path_depth; i++)
3984 vim_free(files[i].name);
3985 vim_free(files);
3986
3987 if (type == CHECK_PATH)
3988 {
3989 if (!did_show)
3990 {
3991 if (action != ACTION_SHOW_ALL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003992 msg(_("All included files were found"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003993 else
Bram Moolenaar32526b32019-01-19 17:43:09 +01003994 msg(_("No included files"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003995 }
3996 }
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003997 else if (!found && action != ACTION_EXPAND)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003998 {
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003999 if (got_int || ins_compl_interrupted())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01004000 emsg(_(e_interr));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004001 else if (type == FIND_DEFINE)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01004002 emsg(_("E388: Couldn't find definition"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004003 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01004004 emsg(_("E389: Couldn't find pattern"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004005 }
4006 if (action == ACTION_SHOW || action == ACTION_SHOW_ALL)
4007 msg_end();
4008
4009fpip_end:
4010 vim_free(file_line);
Bram Moolenaar473de612013-06-08 18:19:48 +02004011 vim_regfree(regmatch.regprog);
4012 vim_regfree(incl_regmatch.regprog);
4013 vim_regfree(def_regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004014}
4015
4016 static void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01004017show_pat_in_path(
4018 char_u *line,
4019 int type,
4020 int did_show,
4021 int action,
4022 FILE *fp,
4023 linenr_T *lnum,
4024 long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004025{
4026 char_u *p;
4027
4028 if (did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004029 msg_putchar('\n'); // cursor below last one
Bram Moolenaar91170f82006-05-05 21:15:17 +00004030 else if (!msg_silent)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004031 gotocmdline(TRUE); // cursor at status line
4032 if (got_int) // 'q' typed at "--more--" message
Bram Moolenaar071d4272004-06-13 20:20:40 +00004033 return;
4034 for (;;)
4035 {
4036 p = line + STRLEN(line) - 1;
4037 if (fp != NULL)
4038 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004039 // We used fgets(), so get rid of newline at end
Bram Moolenaar071d4272004-06-13 20:20:40 +00004040 if (p >= line && *p == '\n')
4041 --p;
4042 if (p >= line && *p == '\r')
4043 --p;
4044 *(p + 1) = NUL;
4045 }
4046 if (action == ACTION_SHOW_ALL)
4047 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004048 sprintf((char *)IObuff, "%3ld: ", count); // show match nr
Bram Moolenaar32526b32019-01-19 17:43:09 +01004049 msg_puts((char *)IObuff);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004050 sprintf((char *)IObuff, "%4ld", *lnum); // show line nr
4051 // Highlight line numbers
Bram Moolenaar32526b32019-01-19 17:43:09 +01004052 msg_puts_attr((char *)IObuff, HL_ATTR(HLF_N));
4053 msg_puts(" ");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004054 }
Bram Moolenaar26a60b42005-02-22 08:49:11 +00004055 msg_prt_line(line, FALSE);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004056 out_flush(); // show one line at a time
Bram Moolenaar071d4272004-06-13 20:20:40 +00004057
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004058 // Definition continues until line that doesn't end with '\'
Bram Moolenaar071d4272004-06-13 20:20:40 +00004059 if (got_int || type != FIND_DEFINE || p < line || *p != '\\')
4060 break;
4061
4062 if (fp != NULL)
4063 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004064 if (vim_fgets(line, LSIZE, fp)) // end of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00004065 break;
4066 ++*lnum;
4067 }
4068 else
4069 {
4070 if (++*lnum > curbuf->b_ml.ml_line_count)
4071 break;
4072 line = ml_get(*lnum);
4073 }
4074 msg_putchar('\n');
4075 }
4076}
4077#endif
4078
4079#ifdef FEAT_VIMINFO
Bram Moolenaar6bd1d772019-10-09 22:01:25 +02004080/*
4081 * Return the last used search pattern at "idx".
4082 */
Bram Moolenaarc3328162019-07-23 22:15:25 +02004083 spat_T *
4084get_spat(int idx)
4085{
4086 return &spats[idx];
4087}
4088
Bram Moolenaar6bd1d772019-10-09 22:01:25 +02004089/*
4090 * Return the last used search pattern index.
4091 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004092 int
Bram Moolenaarc3328162019-07-23 22:15:25 +02004093get_spat_last_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004094{
Bram Moolenaarc3328162019-07-23 22:15:25 +02004095 return last_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004096}
Bram Moolenaar071d4272004-06-13 20:20:40 +00004097#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004098
4099#ifdef FEAT_EVAL
4100/*
4101 * "searchcount()" function
4102 */
4103 void
4104f_searchcount(typval_T *argvars, typval_T *rettv)
4105{
4106 pos_T pos = curwin->w_cursor;
4107 char_u *pattern = NULL;
4108 int maxcount = SEARCH_STAT_DEF_MAX_COUNT;
4109 long timeout = SEARCH_STAT_DEF_TIMEOUT;
Bram Moolenaar4140c4f2020-09-05 23:16:00 +02004110 int recompute = TRUE;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004111 searchstat_T stat;
4112
4113 if (rettv_dict_alloc(rettv) == FAIL)
4114 return;
4115
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02004116 if (in_vim9script() && check_for_opt_dict_arg(argvars, 0) == FAIL)
4117 return;
4118
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004119 if (shortmess(SHM_SEARCHCOUNT)) // 'shortmess' contains 'S' flag
4120 recompute = TRUE;
4121
4122 if (argvars[0].v_type != VAR_UNKNOWN)
4123 {
Bram Moolenaar14681622020-06-03 22:57:39 +02004124 dict_T *dict;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004125 dictitem_T *di;
4126 listitem_T *li;
4127 int error = FALSE;
4128
Bram Moolenaar14681622020-06-03 22:57:39 +02004129 if (argvars[0].v_type != VAR_DICT || argvars[0].vval.v_dict == NULL)
4130 {
4131 emsg(_(e_dictreq));
4132 return;
4133 }
4134 dict = argvars[0].vval.v_dict;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004135 di = dict_find(dict, (char_u *)"timeout", -1);
4136 if (di != NULL)
4137 {
4138 timeout = (long)tv_get_number_chk(&di->di_tv, &error);
4139 if (error)
4140 return;
4141 }
4142 di = dict_find(dict, (char_u *)"maxcount", -1);
4143 if (di != NULL)
4144 {
4145 maxcount = (int)tv_get_number_chk(&di->di_tv, &error);
4146 if (error)
4147 return;
4148 }
Bram Moolenaar597aaac2020-09-05 21:21:16 +02004149 recompute = dict_get_bool(dict, (char_u *)"recompute", recompute);
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004150 di = dict_find(dict, (char_u *)"pattern", -1);
4151 if (di != NULL)
4152 {
4153 pattern = tv_get_string_chk(&di->di_tv);
4154 if (pattern == NULL)
4155 return;
4156 }
4157 di = dict_find(dict, (char_u *)"pos", -1);
4158 if (di != NULL)
4159 {
4160 if (di->di_tv.v_type != VAR_LIST)
4161 {
4162 semsg(_(e_invarg2), "pos");
4163 return;
4164 }
4165 if (list_len(di->di_tv.vval.v_list) != 3)
4166 {
4167 semsg(_(e_invarg2), "List format should be [lnum, col, off]");
4168 return;
4169 }
4170 li = list_find(di->di_tv.vval.v_list, 0L);
4171 if (li != NULL)
4172 {
4173 pos.lnum = tv_get_number_chk(&li->li_tv, &error);
4174 if (error)
4175 return;
4176 }
4177 li = list_find(di->di_tv.vval.v_list, 1L);
4178 if (li != NULL)
4179 {
4180 pos.col = tv_get_number_chk(&li->li_tv, &error) - 1;
4181 if (error)
4182 return;
4183 }
4184 li = list_find(di->di_tv.vval.v_list, 2L);
4185 if (li != NULL)
4186 {
4187 pos.coladd = tv_get_number_chk(&li->li_tv, &error);
4188 if (error)
4189 return;
4190 }
4191 }
4192 }
4193
4194 save_last_search_pattern();
4195 if (pattern != NULL)
4196 {
4197 if (*pattern == NUL)
4198 goto the_end;
Bram Moolenaar109aece2020-06-01 19:08:54 +02004199 vim_free(spats[last_idx].pat);
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004200 spats[last_idx].pat = vim_strsave(pattern);
4201 }
4202 if (spats[last_idx].pat == NULL || *spats[last_idx].pat == NUL)
4203 goto the_end; // the previous pattern was never defined
4204
4205 update_search_stat(0, &pos, &pos, &stat, recompute, maxcount, timeout);
4206
4207 dict_add_number(rettv->vval.v_dict, "current", stat.cur);
4208 dict_add_number(rettv->vval.v_dict, "total", stat.cnt);
4209 dict_add_number(rettv->vval.v_dict, "exact_match", stat.exact_match);
4210 dict_add_number(rettv->vval.v_dict, "incomplete", stat.incomplete);
4211 dict_add_number(rettv->vval.v_dict, "maxcount", stat.last_maxcount);
4212
4213the_end:
4214 restore_last_search_pattern();
4215}
Bram Moolenaar635414d2020-09-11 22:25:15 +02004216
4217/*
4218 * Fuzzy string matching
4219 *
4220 * Ported from the lib_fts library authored by Forrest Smith.
4221 * https://github.com/forrestthewoods/lib_fts/tree/master/code
4222 *
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004223 * The following blog describes the fuzzy matching algorithm:
Bram Moolenaar635414d2020-09-11 22:25:15 +02004224 * https://www.forrestthewoods.com/blog/reverse_engineering_sublime_texts_fuzzy_match/
4225 *
4226 * Each matching string is assigned a score. The following factors are checked:
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004227 * - Matched letter
4228 * - Unmatched letter
4229 * - Consecutively matched letters
4230 * - Proximity to start
4231 * - Letter following a separator (space, underscore)
4232 * - Uppercase letter following lowercase (aka CamelCase)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004233 *
4234 * Matched letters are good. Unmatched letters are bad. Matching near the start
4235 * is good. Matching the first letter in the middle of a phrase is good.
4236 * Matching the uppercase letters in camel case entries is good.
4237 *
4238 * The score assigned for each factor is explained below.
4239 * File paths are different from file names. File extensions may be ignorable.
4240 * Single words care about consecutive matches but not separators or camel
4241 * case.
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004242 * Score starts at 100
Bram Moolenaar635414d2020-09-11 22:25:15 +02004243 * Matched letter: +0 points
4244 * Unmatched letter: -1 point
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004245 * Consecutive match bonus: +15 points
4246 * First letter bonus: +15 points
4247 * Separator bonus: +30 points
4248 * Camel case bonus: +30 points
4249 * Unmatched leading letter: -5 points (max: -15)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004250 *
4251 * There is some nuance to this. Scores don’t have an intrinsic meaning. The
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004252 * score range isn’t 0 to 100. It’s roughly [50, 150]. Longer words have a
Bram Moolenaar635414d2020-09-11 22:25:15 +02004253 * lower minimum score due to unmatched letter penalty. Longer search patterns
4254 * have a higher maximum score due to match bonuses.
4255 *
4256 * Separator and camel case bonus is worth a LOT. Consecutive matches are worth
4257 * quite a bit.
4258 *
4259 * There is a penalty if you DON’T match the first three letters. Which
4260 * effectively rewards matching near the start. However there’s no difference
4261 * in matching between the middle and end.
4262 *
4263 * There is not an explicit bonus for an exact match. Unmatched letters receive
4264 * a penalty. So shorter strings and closer matches are worth more.
4265 */
4266typedef struct
4267{
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004268 int idx; // used for stable sort
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004269 listitem_T *item;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004270 int score;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004271 list_T *lmatchpos;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004272} fuzzyItem_T;
4273
Bram Moolenaare9f9f162020-10-20 19:01:30 +02004274// bonus for adjacent matches; this is higher than SEPARATOR_BONUS so that
4275// matching a whole word is preferred.
4276#define SEQUENTIAL_BONUS 40
Bram Moolenaardcdd42a2020-10-29 18:58:01 +01004277// bonus if match occurs after a path separator
4278#define PATH_SEPARATOR_BONUS 30
4279// bonus if match occurs after a word separator
4280#define WORD_SEPARATOR_BONUS 25
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004281// bonus if match is uppercase and prev is lower
4282#define CAMEL_BONUS 30
4283// bonus if the first letter is matched
4284#define FIRST_LETTER_BONUS 15
4285// penalty applied for every letter in str before the first match
4286#define LEADING_LETTER_PENALTY -5
4287// maximum penalty for leading letters
4288#define MAX_LEADING_LETTER_PENALTY -15
4289// penalty for every letter that doesn't match
4290#define UNMATCHED_LETTER_PENALTY -1
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004291// penalty for gap in matching positions (-2 * k)
4292#define GAP_PENALTY -2
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004293// Score for a string that doesn't fuzzy match the pattern
4294#define SCORE_NONE -9999
4295
4296#define FUZZY_MATCH_RECURSION_LIMIT 10
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004297
4298/*
4299 * Compute a score for a fuzzy matched string. The matching character locations
4300 * are in 'matches'.
4301 */
4302 static int
4303fuzzy_match_compute_score(
4304 char_u *str,
4305 int strSz,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004306 int_u *matches,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004307 int numMatches)
4308{
4309 int score;
4310 int penalty;
4311 int unmatched;
4312 int i;
4313 char_u *p = str;
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004314 int_u sidx = 0;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004315
4316 // Initialize score
4317 score = 100;
4318
4319 // Apply leading letter penalty
4320 penalty = LEADING_LETTER_PENALTY * matches[0];
4321 if (penalty < MAX_LEADING_LETTER_PENALTY)
4322 penalty = MAX_LEADING_LETTER_PENALTY;
4323 score += penalty;
4324
4325 // Apply unmatched penalty
4326 unmatched = strSz - numMatches;
4327 score += UNMATCHED_LETTER_PENALTY * unmatched;
4328
4329 // Apply ordering bonuses
4330 for (i = 0; i < numMatches; ++i)
4331 {
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004332 int_u currIdx = matches[i];
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004333
4334 if (i > 0)
4335 {
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004336 int_u prevIdx = matches[i - 1];
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004337
4338 // Sequential
4339 if (currIdx == (prevIdx + 1))
4340 score += SEQUENTIAL_BONUS;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004341 else
4342 score += GAP_PENALTY * (currIdx - prevIdx);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004343 }
4344
4345 // Check for bonuses based on neighbor character value
4346 if (currIdx > 0)
4347 {
4348 // Camel case
Bram Moolenaarc53e9c52020-09-22 22:08:32 +02004349 int neighbor = ' ';
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004350 int curr;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004351
4352 if (has_mbyte)
4353 {
4354 while (sidx < currIdx)
4355 {
4356 neighbor = (*mb_ptr2char)(p);
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004357 MB_PTR_ADV(p);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004358 sidx++;
4359 }
4360 curr = (*mb_ptr2char)(p);
4361 }
4362 else
4363 {
4364 neighbor = str[currIdx - 1];
4365 curr = str[currIdx];
4366 }
4367
4368 if (vim_islower(neighbor) && vim_isupper(curr))
4369 score += CAMEL_BONUS;
4370
Bram Moolenaardcdd42a2020-10-29 18:58:01 +01004371 // Bonus if the match follows a separator character
4372 if (neighbor == '/' || neighbor == '\\')
4373 score += PATH_SEPARATOR_BONUS;
4374 else if (neighbor == ' ' || neighbor == '_')
4375 score += WORD_SEPARATOR_BONUS;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004376 }
4377 else
4378 {
4379 // First letter
4380 score += FIRST_LETTER_BONUS;
4381 }
4382 }
4383 return score;
4384}
4385
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004386/*
4387 * Perform a recursive search for fuzzy matching 'fuzpat' in 'str'.
4388 * Return the number of matching characters.
4389 */
Bram Moolenaar635414d2020-09-11 22:25:15 +02004390 static int
4391fuzzy_match_recursive(
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004392 char_u *fuzpat,
4393 char_u *str,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004394 int_u strIdx,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004395 int *outScore,
4396 char_u *strBegin,
4397 int strLen,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004398 int_u *srcMatches,
4399 int_u *matches,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004400 int maxMatches,
4401 int nextMatch,
4402 int *recursionCount)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004403{
4404 // Recursion params
4405 int recursiveMatch = FALSE;
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004406 int_u bestRecursiveMatches[MAX_FUZZY_MATCHES];
Bram Moolenaar635414d2020-09-11 22:25:15 +02004407 int bestRecursiveScore = 0;
4408 int first_match;
4409 int matched;
4410
4411 // Count recursions
4412 ++*recursionCount;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004413 if (*recursionCount >= FUZZY_MATCH_RECURSION_LIMIT)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004414 return 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004415
4416 // Detect end of strings
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004417 if (*fuzpat == NUL || *str == NUL)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004418 return 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004419
4420 // Loop through fuzpat and str looking for a match
4421 first_match = TRUE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004422 while (*fuzpat != NUL && *str != NUL)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004423 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004424 int c1;
4425 int c2;
4426
4427 c1 = PTR2CHAR(fuzpat);
4428 c2 = PTR2CHAR(str);
4429
Bram Moolenaar635414d2020-09-11 22:25:15 +02004430 // Found match
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004431 if (vim_tolower(c1) == vim_tolower(c2))
Bram Moolenaar635414d2020-09-11 22:25:15 +02004432 {
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004433 int_u recursiveMatches[MAX_FUZZY_MATCHES];
Bram Moolenaar635414d2020-09-11 22:25:15 +02004434 int recursiveScore = 0;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004435 char_u *next_char;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004436
4437 // Supplied matches buffer was too short
4438 if (nextMatch >= maxMatches)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004439 return 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004440
4441 // "Copy-on-Write" srcMatches into matches
4442 if (first_match && srcMatches)
4443 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004444 memcpy(matches, srcMatches, nextMatch * sizeof(srcMatches[0]));
Bram Moolenaar635414d2020-09-11 22:25:15 +02004445 first_match = FALSE;
4446 }
4447
4448 // Recursive call that "skips" this match
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004449 if (has_mbyte)
4450 next_char = str + (*mb_ptr2len)(str);
4451 else
4452 next_char = str + 1;
4453 if (fuzzy_match_recursive(fuzpat, next_char, strIdx + 1,
4454 &recursiveScore, strBegin, strLen, matches,
4455 recursiveMatches,
K.Takataeeec2542021-06-02 13:28:16 +02004456 ARRAY_LENGTH(recursiveMatches),
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004457 nextMatch, recursionCount))
Bram Moolenaar635414d2020-09-11 22:25:15 +02004458 {
4459 // Pick best recursive score
4460 if (!recursiveMatch || recursiveScore > bestRecursiveScore)
4461 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004462 memcpy(bestRecursiveMatches, recursiveMatches,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004463 MAX_FUZZY_MATCHES * sizeof(recursiveMatches[0]));
Bram Moolenaar635414d2020-09-11 22:25:15 +02004464 bestRecursiveScore = recursiveScore;
4465 }
4466 recursiveMatch = TRUE;
4467 }
4468
4469 // Advance
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004470 matches[nextMatch++] = strIdx;
4471 if (has_mbyte)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004472 MB_PTR_ADV(fuzpat);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004473 else
4474 ++fuzpat;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004475 }
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004476 if (has_mbyte)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004477 MB_PTR_ADV(str);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004478 else
4479 ++str;
4480 strIdx++;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004481 }
4482
4483 // Determine if full fuzpat was matched
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004484 matched = *fuzpat == NUL ? TRUE : FALSE;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004485
4486 // Calculate score
4487 if (matched)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004488 *outScore = fuzzy_match_compute_score(strBegin, strLen, matches,
4489 nextMatch);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004490
4491 // Return best result
4492 if (recursiveMatch && (!matched || bestRecursiveScore > *outScore))
4493 {
4494 // Recursive score is better than "this"
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004495 memcpy(matches, bestRecursiveMatches, maxMatches * sizeof(matches[0]));
Bram Moolenaar635414d2020-09-11 22:25:15 +02004496 *outScore = bestRecursiveScore;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004497 return nextMatch;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004498 }
4499 else if (matched)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004500 return nextMatch; // "this" score is better than recursive
Bram Moolenaar635414d2020-09-11 22:25:15 +02004501
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004502 return 0; // no match
Bram Moolenaar635414d2020-09-11 22:25:15 +02004503}
4504
4505/*
4506 * fuzzy_match()
4507 *
4508 * Performs exhaustive search via recursion to find all possible matches and
4509 * match with highest score.
4510 * Scores values have no intrinsic meaning. Possible score range is not
4511 * normalized and varies with pattern.
4512 * Recursion is limited internally (default=10) to prevent degenerate cases
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004513 * (pat_arg="aaaaaa" str="aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa").
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004514 * Uses char_u for match indices. Therefore patterns are limited to
4515 * MAX_FUZZY_MATCHES characters.
Bram Moolenaar635414d2020-09-11 22:25:15 +02004516 *
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004517 * Returns TRUE if 'pat_arg' matches 'str'. Also returns the match score in
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004518 * 'outScore' and the matching character positions in 'matches'.
Bram Moolenaar635414d2020-09-11 22:25:15 +02004519 */
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004520 int
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004521fuzzy_match(
4522 char_u *str,
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004523 char_u *pat_arg,
4524 int matchseq,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004525 int *outScore,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004526 int_u *matches,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004527 int maxMatches)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004528{
Bram Moolenaar635414d2020-09-11 22:25:15 +02004529 int recursionCount = 0;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004530 int len = MB_CHARLEN(str);
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004531 char_u *save_pat;
4532 char_u *pat;
4533 char_u *p;
4534 int complete = FALSE;
4535 int score = 0;
4536 int numMatches = 0;
4537 int matchCount;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004538
4539 *outScore = 0;
4540
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004541 save_pat = vim_strsave(pat_arg);
4542 if (save_pat == NULL)
4543 return FALSE;
4544 pat = save_pat;
4545 p = pat;
4546
4547 // Try matching each word in 'pat_arg' in 'str'
4548 while (TRUE)
4549 {
4550 if (matchseq)
4551 complete = TRUE;
4552 else
4553 {
4554 // Extract one word from the pattern (separated by space)
4555 p = skipwhite(p);
4556 if (*p == NUL)
4557 break;
4558 pat = p;
4559 while (*p != NUL && !VIM_ISWHITE(PTR2CHAR(p)))
4560 {
4561 if (has_mbyte)
4562 MB_PTR_ADV(p);
4563 else
4564 ++p;
4565 }
4566 if (*p == NUL) // processed all the words
4567 complete = TRUE;
4568 *p = NUL;
4569 }
4570
4571 score = 0;
4572 recursionCount = 0;
4573 matchCount = fuzzy_match_recursive(pat, str, 0, &score, str, len, NULL,
4574 matches + numMatches, maxMatches - numMatches,
4575 0, &recursionCount);
4576 if (matchCount == 0)
4577 {
4578 numMatches = 0;
4579 break;
4580 }
4581
4582 // Accumulate the match score and the number of matches
4583 *outScore += score;
4584 numMatches += matchCount;
4585
4586 if (complete)
4587 break;
4588
4589 // try matching the next word
4590 ++p;
4591 }
4592
4593 vim_free(save_pat);
4594 return numMatches != 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004595}
4596
4597/*
4598 * Sort the fuzzy matches in the descending order of the match score.
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004599 * For items with same score, retain the order using the index (stable sort)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004600 */
4601 static int
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004602fuzzy_match_item_compare(const void *s1, const void *s2)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004603{
4604 int v1 = ((fuzzyItem_T *)s1)->score;
4605 int v2 = ((fuzzyItem_T *)s2)->score;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004606 int idx1 = ((fuzzyItem_T *)s1)->idx;
4607 int idx2 = ((fuzzyItem_T *)s2)->idx;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004608
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004609 return v1 == v2 ? (idx1 - idx2) : v1 > v2 ? -1 : 1;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004610}
4611
4612/*
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004613 * Fuzzy search the string 'str' in a list of 'items' and return the matching
4614 * strings in 'fmatchlist'.
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004615 * If 'matchseq' is TRUE, then for multi-word search strings, match all the
4616 * words in sequence.
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004617 * If 'items' is a list of strings, then search for 'str' in the list.
4618 * If 'items' is a list of dicts, then either use 'key' to lookup the string
4619 * for each item or use 'item_cb' Funcref function to get the string.
4620 * If 'retmatchpos' is TRUE, then return a list of positions where 'str'
4621 * matches for each item.
Bram Moolenaar635414d2020-09-11 22:25:15 +02004622 */
4623 static void
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004624fuzzy_match_in_list(
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004625 list_T *items,
4626 char_u *str,
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004627 int matchseq,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004628 char_u *key,
4629 callback_T *item_cb,
4630 int retmatchpos,
4631 list_T *fmatchlist)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004632{
4633 long len;
4634 fuzzyItem_T *ptrs;
4635 listitem_T *li;
4636 long i = 0;
4637 int found_match = FALSE;
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004638 int_u matches[MAX_FUZZY_MATCHES];
Bram Moolenaar635414d2020-09-11 22:25:15 +02004639
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004640 len = list_len(items);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004641 if (len == 0)
4642 return;
4643
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004644 ptrs = ALLOC_CLEAR_MULT(fuzzyItem_T, len);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004645 if (ptrs == NULL)
4646 return;
4647
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004648 // For all the string items in items, get the fuzzy matching score
4649 FOR_ALL_LIST_ITEMS(items, li)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004650 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004651 int score;
4652 char_u *itemstr;
4653 typval_T rettv;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004654
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004655 ptrs[i].idx = i;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004656 ptrs[i].item = li;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004657 ptrs[i].score = SCORE_NONE;
4658 itemstr = NULL;
4659 rettv.v_type = VAR_UNKNOWN;
4660 if (li->li_tv.v_type == VAR_STRING) // list of strings
4661 itemstr = li->li_tv.vval.v_string;
4662 else if (li->li_tv.v_type == VAR_DICT &&
4663 (key != NULL || item_cb->cb_name != NULL))
4664 {
4665 // For a dict, either use the specified key to lookup the string or
4666 // use the specified callback function to get the string.
4667 if (key != NULL)
4668 itemstr = dict_get_string(li->li_tv.vval.v_dict, key, FALSE);
4669 else
Bram Moolenaar635414d2020-09-11 22:25:15 +02004670 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004671 typval_T argv[2];
4672
4673 // Invoke the supplied callback (if any) to get the dict item
4674 li->li_tv.vval.v_dict->dv_refcount++;
4675 argv[0].v_type = VAR_DICT;
4676 argv[0].vval.v_dict = li->li_tv.vval.v_dict;
4677 argv[1].v_type = VAR_UNKNOWN;
4678 if (call_callback(item_cb, -1, &rettv, 1, argv) != FAIL)
4679 {
4680 if (rettv.v_type == VAR_STRING)
4681 itemstr = rettv.vval.v_string;
4682 }
4683 dict_unref(li->li_tv.vval.v_dict);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004684 }
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004685 }
4686
4687 if (itemstr != NULL
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004688 && fuzzy_match(itemstr, str, matchseq, &score, matches,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004689 sizeof(matches) / sizeof(matches[0])))
4690 {
4691 // Copy the list of matching positions in itemstr to a list, if
4692 // 'retmatchpos' is set.
4693 if (retmatchpos)
4694 {
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004695 int j = 0;
4696 char_u *p;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004697
4698 ptrs[i].lmatchpos = list_alloc();
4699 if (ptrs[i].lmatchpos == NULL)
4700 goto done;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004701
4702 p = str;
4703 while (*p != NUL)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004704 {
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004705 if (!VIM_ISWHITE(PTR2CHAR(p)))
4706 {
4707 if (list_append_number(ptrs[i].lmatchpos,
4708 matches[j]) == FAIL)
4709 goto done;
4710 j++;
4711 }
4712 if (has_mbyte)
4713 MB_PTR_ADV(p);
4714 else
4715 ++p;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004716 }
4717 }
4718 ptrs[i].score = score;
4719 found_match = TRUE;
4720 }
Bram Moolenaar635414d2020-09-11 22:25:15 +02004721 ++i;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004722 clear_tv(&rettv);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004723 }
4724
4725 if (found_match)
4726 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004727 list_T *l;
4728
Bram Moolenaar635414d2020-09-11 22:25:15 +02004729 // Sort the list by the descending order of the match score
4730 qsort((void *)ptrs, (size_t)len, sizeof(fuzzyItem_T),
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004731 fuzzy_match_item_compare);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004732
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004733 // For matchfuzzy(), return a list of matched strings.
4734 // ['str1', 'str2', 'str3']
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004735 // For matchfuzzypos(), return a list with three items.
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004736 // The first item is a list of matched strings. The second item
4737 // is a list of lists where each list item is a list of matched
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004738 // character positions. The third item is a list of matching scores.
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004739 // [['str1', 'str2', 'str3'], [[1, 3], [1, 3], [1, 3]]]
4740 if (retmatchpos)
4741 {
4742 li = list_find(fmatchlist, 0);
4743 if (li == NULL || li->li_tv.vval.v_list == NULL)
4744 goto done;
4745 l = li->li_tv.vval.v_list;
4746 }
4747 else
4748 l = fmatchlist;
4749
4750 // Copy the matching strings with a valid score to the return list
Bram Moolenaar635414d2020-09-11 22:25:15 +02004751 for (i = 0; i < len; i++)
4752 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004753 if (ptrs[i].score == SCORE_NONE)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004754 break;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004755 list_append_tv(l, &ptrs[i].item->li_tv);
4756 }
4757
4758 // next copy the list of matching positions
4759 if (retmatchpos)
4760 {
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004761 li = list_find(fmatchlist, -2);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004762 if (li == NULL || li->li_tv.vval.v_list == NULL)
4763 goto done;
4764 l = li->li_tv.vval.v_list;
4765
4766 for (i = 0; i < len; i++)
4767 {
4768 if (ptrs[i].score == SCORE_NONE)
4769 break;
4770 if (ptrs[i].lmatchpos != NULL &&
4771 list_append_list(l, ptrs[i].lmatchpos) == FAIL)
4772 goto done;
4773 }
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004774
4775 // copy the matching scores
4776 li = list_find(fmatchlist, -1);
4777 if (li == NULL || li->li_tv.vval.v_list == NULL)
4778 goto done;
4779 l = li->li_tv.vval.v_list;
4780 for (i = 0; i < len; i++)
4781 {
4782 if (ptrs[i].score == SCORE_NONE)
4783 break;
4784 if (list_append_number(l, ptrs[i].score) == FAIL)
4785 goto done;
4786 }
Bram Moolenaar635414d2020-09-11 22:25:15 +02004787 }
4788 }
4789
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004790done:
Bram Moolenaar635414d2020-09-11 22:25:15 +02004791 vim_free(ptrs);
4792}
4793
4794/*
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004795 * Do fuzzy matching. Returns the list of matched strings in 'rettv'.
4796 * If 'retmatchpos' is TRUE, also returns the matching character positions.
4797 */
4798 static void
4799do_fuzzymatch(typval_T *argvars, typval_T *rettv, int retmatchpos)
4800{
4801 callback_T cb;
4802 char_u *key = NULL;
4803 int ret;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004804 int matchseq = FALSE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004805
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02004806 if (in_vim9script()
4807 && (check_for_list_arg(argvars, 0) == FAIL
4808 || check_for_string_arg(argvars, 1) == FAIL
4809 || check_for_opt_dict_arg(argvars, 2) == FAIL))
4810 return;
4811
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004812 CLEAR_POINTER(&cb);
4813
4814 // validate and get the arguments
4815 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
4816 {
4817 semsg(_(e_listarg), retmatchpos ? "matchfuzzypos()" : "matchfuzzy()");
4818 return;
4819 }
4820 if (argvars[1].v_type != VAR_STRING
4821 || argvars[1].vval.v_string == NULL)
4822 {
4823 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
4824 return;
4825 }
4826
4827 if (argvars[2].v_type != VAR_UNKNOWN)
4828 {
4829 dict_T *d;
4830 dictitem_T *di;
4831
4832 if (argvars[2].v_type != VAR_DICT || argvars[2].vval.v_dict == NULL)
4833 {
4834 emsg(_(e_dictreq));
4835 return;
4836 }
4837
4838 // To search a dict, either a callback function or a key can be
4839 // specified.
4840 d = argvars[2].vval.v_dict;
4841 if ((di = dict_find(d, (char_u *)"key", -1)) != NULL)
4842 {
4843 if (di->di_tv.v_type != VAR_STRING
4844 || di->di_tv.vval.v_string == NULL
4845 || *di->di_tv.vval.v_string == NUL)
4846 {
4847 semsg(_(e_invarg2), tv_get_string(&di->di_tv));
4848 return;
4849 }
4850 key = tv_get_string(&di->di_tv);
4851 }
4852 else if ((di = dict_find(d, (char_u *)"text_cb", -1)) != NULL)
4853 {
4854 cb = get_callback(&di->di_tv);
4855 if (cb.cb_name == NULL)
4856 {
4857 semsg(_(e_invargval), "text_cb");
4858 return;
4859 }
4860 }
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004861 if (dict_find(d, (char_u *)"matchseq", -1) != NULL)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004862 matchseq = TRUE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004863 }
4864
4865 // get the fuzzy matches
4866 ret = rettv_list_alloc(rettv);
4867 if (ret != OK)
4868 goto done;
4869 if (retmatchpos)
4870 {
4871 list_T *l;
4872
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004873 // For matchfuzzypos(), a list with three items are returned. First
4874 // item is a list of matching strings, the second item is a list of
4875 // lists with matching positions within each string and the third item
4876 // is the list of scores of the matches.
4877 l = list_alloc();
4878 if (l == NULL)
4879 goto done;
4880 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4881 goto done;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004882 l = list_alloc();
4883 if (l == NULL)
4884 goto done;
4885 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4886 goto done;
4887 l = list_alloc();
4888 if (l == NULL)
4889 goto done;
4890 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4891 goto done;
4892 }
4893
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004894 fuzzy_match_in_list(argvars[0].vval.v_list, tv_get_string(&argvars[1]),
4895 matchseq, key, &cb, retmatchpos, rettv->vval.v_list);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004896
4897done:
4898 free_callback(&cb);
4899}
4900
4901/*
Bram Moolenaar635414d2020-09-11 22:25:15 +02004902 * "matchfuzzy()" function
4903 */
4904 void
4905f_matchfuzzy(typval_T *argvars, typval_T *rettv)
4906{
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004907 do_fuzzymatch(argvars, rettv, FALSE);
4908}
4909
4910/*
4911 * "matchfuzzypos()" function
4912 */
4913 void
4914f_matchfuzzypos(typval_T *argvars, typval_T *rettv)
4915{
4916 do_fuzzymatch(argvars, rettv, TRUE);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004917}
4918
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004919#endif