blob: 641d439a486e0339f2c40e027d0c65cd925a4038 [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9/*
10 * search.c: code for normal mode searching commands
11 */
12
13#include "vim.h"
14
Bram Moolenaar071d4272004-06-13 20:20:40 +000015#ifdef FEAT_EVAL
Bram Moolenaarbaaa7e92016-01-29 22:47:03 +010016static void set_vv_searchforward(void);
17static int first_submatch(regmmatch_T *rp);
Bram Moolenaar071d4272004-06-13 20:20:40 +000018#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +000019#ifdef FEAT_FIND_ID
Bram Moolenaarbaaa7e92016-01-29 22:47:03 +010020static void show_pat_in_path(char_u *, int,
21 int, int, FILE *, linenr_T *, long);
Bram Moolenaar071d4272004-06-13 20:20:40 +000022#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +020023
24typedef struct searchstat
25{
26 int cur; // current position of found words
27 int cnt; // total count of found words
28 int exact_match; // TRUE if matched exactly on specified position
29 int incomplete; // 0: search was fully completed
30 // 1: recomputing was timed out
31 // 2: max count exceeded
32 int last_maxcount; // the max count of the last search
33} searchstat_T;
34
35static void cmdline_search_stat(int dirc, pos_T *pos, pos_T *cursor_pos, int show_top_bot_msg, char_u *msgbuf, int recompute, int maxcount, long timeout);
36static void update_search_stat(int dirc, pos_T *pos, pos_T *cursor_pos, searchstat_T *stat, int recompute, int maxcount, long timeout);
37
Bram Moolenaarea6561a2020-06-01 21:32:45 +020038#define SEARCH_STAT_DEF_TIMEOUT 40L
Bram Moolenaare8f5ec02020-06-01 17:28:35 +020039#define SEARCH_STAT_DEF_MAX_COUNT 99
40#define SEARCH_STAT_BUF_LEN 12
Bram Moolenaar071d4272004-06-13 20:20:40 +000041
Bram Moolenaar071d4272004-06-13 20:20:40 +000042/*
43 * This file contains various searching-related routines. These fall into
44 * three groups:
45 * 1. string searches (for /, ?, n, and N)
46 * 2. character searches within a single line (for f, F, t, T, etc)
47 * 3. "other" kinds of searches like the '%' command, and 'word' searches.
48 */
49
50/*
51 * String searches
52 *
53 * The string search functions are divided into two levels:
54 * lowest: searchit(); uses an pos_T for starting position and found match.
55 * Highest: do_search(); uses curwin->w_cursor; calls searchit().
56 *
57 * The last search pattern is remembered for repeating the same search.
58 * This pattern is shared between the :g, :s, ? and / commands.
59 * This is in search_regcomp().
60 *
61 * The actual string matching is done using a heavily modified version of
62 * Henry Spencer's regular expression library. See regexp.c.
63 */
64
Bram Moolenaar071d4272004-06-13 20:20:40 +000065/*
66 * Two search patterns are remembered: One for the :substitute command and
67 * one for other searches. last_idx points to the one that was used the last
68 * time.
69 */
Bram Moolenaarc3328162019-07-23 22:15:25 +020070static spat_T spats[2] =
Bram Moolenaar071d4272004-06-13 20:20:40 +000071{
Bram Moolenaar63d9e732019-12-05 21:10:38 +010072 {NULL, TRUE, FALSE, {'/', 0, 0, 0L}}, // last used search pat
73 {NULL, TRUE, FALSE, {'/', 0, 0, 0L}} // last used substitute pat
Bram Moolenaar071d4272004-06-13 20:20:40 +000074};
75
Bram Moolenaar63d9e732019-12-05 21:10:38 +010076static int last_idx = 0; // index in spats[] for RE_LAST
Bram Moolenaar071d4272004-06-13 20:20:40 +000077
Bram Moolenaar63d9e732019-12-05 21:10:38 +010078static char_u lastc[2] = {NUL, NUL}; // last character searched for
79static int lastcdir = FORWARD; // last direction of character search
80static int last_t_cmd = TRUE; // last search t_cmd
Bram Moolenaardbd24b52015-08-11 14:26:19 +020081static char_u lastc_bytes[MB_MAXBYTES + 1];
Bram Moolenaar63d9e732019-12-05 21:10:38 +010082static int lastc_bytelen = 1; // >1 for multi-byte char
Bram Moolenaardbd24b52015-08-11 14:26:19 +020083
Bram Moolenaar63d9e732019-12-05 21:10:38 +010084// copy of spats[], for keeping the search patterns while executing autocmds
Bram Moolenaarc3328162019-07-23 22:15:25 +020085static spat_T saved_spats[2];
Bram Moolenaara2cff1d2021-10-15 12:51:29 +010086static char_u *saved_mr_pattern = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +000087# ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +010088static int saved_spats_last_idx = 0;
89static int saved_spats_no_hlsearch = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +000090# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +000091
Bram Moolenaara2cff1d2021-10-15 12:51:29 +010092// allocated copy of pattern used by search_regcomp()
93static char_u *mr_pattern = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +000094
95#ifdef FEAT_FIND_ID
96/*
97 * Type used by find_pattern_in_path() to remember which included files have
98 * been searched already.
99 */
100typedef struct SearchedFile
101{
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100102 FILE *fp; // File pointer
103 char_u *name; // Full name of file
104 linenr_T lnum; // Line we were up to in file
105 int matched; // Found a match in this file
Bram Moolenaar071d4272004-06-13 20:20:40 +0000106} SearchedFile;
107#endif
108
109/*
110 * translate search pattern for vim_regcomp()
111 *
112 * pat_save == RE_SEARCH: save pat in spats[RE_SEARCH].pat (normal search cmd)
113 * pat_save == RE_SUBST: save pat in spats[RE_SUBST].pat (:substitute command)
114 * pat_save == RE_BOTH: save pat in both patterns (:global command)
115 * pat_use == RE_SEARCH: use previous search pattern if "pat" is NULL
Bram Moolenaarb8017e72007-05-10 18:59:07 +0000116 * pat_use == RE_SUBST: use previous substitute pattern if "pat" is NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000117 * pat_use == RE_LAST: use last used pattern if "pat" is NULL
118 * options & SEARCH_HIS: put search string in history
119 * options & SEARCH_KEEP: keep previous search pattern
120 *
121 * returns FAIL if failed, OK otherwise.
122 */
123 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100124search_regcomp(
125 char_u *pat,
126 int pat_save,
127 int pat_use,
128 int options,
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100129 regmmatch_T *regmatch) // return: pattern and ignore-case flag
Bram Moolenaar071d4272004-06-13 20:20:40 +0000130{
131 int magic;
132 int i;
133
134 rc_did_emsg = FALSE;
Bram Moolenaarf4e20992020-12-21 19:59:08 +0100135 magic = magic_isset();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000136
137 /*
138 * If no pattern given, use a previously defined pattern.
139 */
140 if (pat == NULL || *pat == NUL)
141 {
142 if (pat_use == RE_LAST)
143 i = last_idx;
144 else
145 i = pat_use;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100146 if (spats[i].pat == NULL) // pattern was never defined
Bram Moolenaar071d4272004-06-13 20:20:40 +0000147 {
148 if (pat_use == RE_SUBST)
Bram Moolenaare29a27f2021-07-20 21:07:36 +0200149 emsg(_(e_no_previous_substitute_regular_expression));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000150 else
Bram Moolenaare29a27f2021-07-20 21:07:36 +0200151 emsg(_(e_no_previous_regular_expression));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152 rc_did_emsg = TRUE;
153 return FAIL;
154 }
155 pat = spats[i].pat;
156 magic = spats[i].magic;
157 no_smartcase = spats[i].no_scs;
158 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100159 else if (options & SEARCH_HIS) // put new pattern in history
Bram Moolenaar071d4272004-06-13 20:20:40 +0000160 add_to_history(HIST_SEARCH, pat, TRUE, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000161
Bram Moolenaara2cff1d2021-10-15 12:51:29 +0100162 vim_free(mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000163#ifdef FEAT_RIGHTLEFT
Bram Moolenaar071d4272004-06-13 20:20:40 +0000164 if (curwin->w_p_rl && *curwin->w_p_rlc == 's')
Bram Moolenaara2cff1d2021-10-15 12:51:29 +0100165 mr_pattern = reverse_text(pat);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000166 else
167#endif
Bram Moolenaara2cff1d2021-10-15 12:51:29 +0100168 mr_pattern = vim_strsave(pat);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000169
170 /*
171 * Save the currently used pattern in the appropriate place,
172 * unless the pattern should not be remembered.
173 */
Bram Moolenaare1004402020-10-24 20:49:43 +0200174 if (!(options & SEARCH_KEEP)
175 && (cmdmod.cmod_flags & CMOD_KEEPPATTERNS) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000176 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100177 // search or global command
Bram Moolenaar071d4272004-06-13 20:20:40 +0000178 if (pat_save == RE_SEARCH || pat_save == RE_BOTH)
179 save_re_pat(RE_SEARCH, pat, magic);
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100180 // substitute or global command
Bram Moolenaar071d4272004-06-13 20:20:40 +0000181 if (pat_save == RE_SUBST || pat_save == RE_BOTH)
182 save_re_pat(RE_SUBST, pat, magic);
183 }
184
185 regmatch->rmm_ic = ignorecase(pat);
Bram Moolenaar3b56eb32005-07-11 22:40:32 +0000186 regmatch->rmm_maxcol = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000187 regmatch->regprog = vim_regcomp(pat, magic ? RE_MAGIC : 0);
188 if (regmatch->regprog == NULL)
189 return FAIL;
190 return OK;
191}
192
193/*
194 * Get search pattern used by search_regcomp().
195 */
196 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100197get_search_pat(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000198{
199 return mr_pattern;
200}
201
Bram Moolenaarabc97732007-08-08 20:49:37 +0000202#if defined(FEAT_RIGHTLEFT) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000203/*
204 * Reverse text into allocated memory.
205 * Returns the allocated string, NULL when out of memory.
206 */
Bram Moolenaarabc97732007-08-08 20:49:37 +0000207 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100208reverse_text(char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000209{
210 unsigned len;
211 unsigned s_i, rev_i;
212 char_u *rev;
213
214 /*
215 * Reverse the pattern.
216 */
217 len = (unsigned)STRLEN(s);
218 rev = alloc(len + 1);
219 if (rev != NULL)
220 {
221 rev_i = len;
222 for (s_i = 0; s_i < len; ++s_i)
223 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000224 if (has_mbyte)
225 {
226 int mb_len;
227
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000228 mb_len = (*mb_ptr2len)(s + s_i);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000229 rev_i -= mb_len;
230 mch_memmove(rev + rev_i, s + s_i, mb_len);
231 s_i += mb_len - 1;
232 }
233 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000234 rev[--rev_i] = s[s_i];
235
236 }
237 rev[len] = NUL;
238 }
239 return rev;
240}
241#endif
242
Bram Moolenaarcc2b9d52014-12-13 03:17:11 +0100243 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100244save_re_pat(int idx, char_u *pat, int magic)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000245{
246 if (spats[idx].pat != pat)
247 {
248 vim_free(spats[idx].pat);
249 spats[idx].pat = vim_strsave(pat);
250 spats[idx].magic = magic;
251 spats[idx].no_scs = no_smartcase;
252 last_idx = idx;
253#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100254 // If 'hlsearch' set and search pat changed: need redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000255 if (p_hls)
Bram Moolenaar1c8f93f2006-03-12 22:10:07 +0000256 redraw_all_later(SOME_VALID);
Bram Moolenaar451fc7b2018-04-27 22:53:07 +0200257 set_no_hlsearch(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000258#endif
259 }
260}
261
Bram Moolenaar071d4272004-06-13 20:20:40 +0000262/*
263 * Save the search patterns, so they can be restored later.
264 * Used before/after executing autocommands and user functions.
265 */
266static int save_level = 0;
267
268 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100269save_search_patterns(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000270{
271 if (save_level++ == 0)
272 {
273 saved_spats[0] = spats[0];
274 if (spats[0].pat != NULL)
275 saved_spats[0].pat = vim_strsave(spats[0].pat);
276 saved_spats[1] = spats[1];
277 if (spats[1].pat != NULL)
278 saved_spats[1].pat = vim_strsave(spats[1].pat);
Bram Moolenaara2cff1d2021-10-15 12:51:29 +0100279 if (mr_pattern == NULL)
280 saved_mr_pattern = NULL;
281 else
282 saved_mr_pattern = vim_strsave(mr_pattern);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100283#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100284 saved_spats_last_idx = last_idx;
285 saved_spats_no_hlsearch = no_hlsearch;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100286#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000287 }
288}
289
290 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100291restore_search_patterns(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000292{
293 if (--save_level == 0)
294 {
295 vim_free(spats[0].pat);
296 spats[0] = saved_spats[0];
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100297#if defined(FEAT_EVAL)
Bram Moolenaar8c8de832008-06-24 22:58:06 +0000298 set_vv_searchforward();
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100299#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000300 vim_free(spats[1].pat);
301 spats[1] = saved_spats[1];
Bram Moolenaara2cff1d2021-10-15 12:51:29 +0100302 vim_free(mr_pattern);
303 mr_pattern = saved_mr_pattern;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100304#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100305 last_idx = saved_spats_last_idx;
306 set_no_hlsearch(saved_spats_no_hlsearch);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100307#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000308 }
309}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000310
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000311#if defined(EXITFREE) || defined(PROTO)
312 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100313free_search_patterns(void)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000314{
315 vim_free(spats[0].pat);
316 vim_free(spats[1].pat);
Bram Moolenaara2cff1d2021-10-15 12:51:29 +0100317 VIM_CLEAR(mr_pattern);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000318}
319#endif
320
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100321#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100322// copy of spats[RE_SEARCH], for keeping the search patterns while incremental
323// searching
Bram Moolenaarc3328162019-07-23 22:15:25 +0200324static spat_T saved_last_search_spat;
Bram Moolenaared8bc782018-12-01 21:08:21 +0100325static int did_save_last_search_spat = 0;
326static int saved_last_idx = 0;
327static int saved_no_hlsearch = 0;
328
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100329/*
330 * Save and restore the search pattern for incremental highlight search
331 * feature.
332 *
Bram Moolenaarc4568ab2018-11-16 16:21:05 +0100333 * It's similar to but different from save_search_patterns() and
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100334 * restore_search_patterns(), because the search pattern must be restored when
Bram Moolenaarc4568ab2018-11-16 16:21:05 +0100335 * canceling incremental searching even if it's called inside user functions.
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100336 */
337 void
338save_last_search_pattern(void)
339{
Bram Moolenaar442a8532020-06-04 20:56:09 +0200340 if (++did_save_last_search_spat != 1)
341 // nested call, nothing to do
342 return;
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100343
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100344 saved_last_search_spat = spats[RE_SEARCH];
345 if (spats[RE_SEARCH].pat != NULL)
346 saved_last_search_spat.pat = vim_strsave(spats[RE_SEARCH].pat);
347 saved_last_idx = last_idx;
348 saved_no_hlsearch = no_hlsearch;
349}
350
351 void
352restore_last_search_pattern(void)
353{
Bram Moolenaar442a8532020-06-04 20:56:09 +0200354 if (--did_save_last_search_spat > 0)
355 // nested call, nothing to do
356 return;
357 if (did_save_last_search_spat != 0)
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100358 {
Bram Moolenaar442a8532020-06-04 20:56:09 +0200359 iemsg("restore_last_search_pattern() called more often than save_last_search_pattern()");
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100360 return;
361 }
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100362
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100363 vim_free(spats[RE_SEARCH].pat);
364 spats[RE_SEARCH] = saved_last_search_spat;
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100365 saved_last_search_spat.pat = NULL;
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100366# if defined(FEAT_EVAL)
367 set_vv_searchforward();
368# endif
369 last_idx = saved_last_idx;
Bram Moolenaar451fc7b2018-04-27 22:53:07 +0200370 set_no_hlsearch(saved_no_hlsearch);
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100371}
Bram Moolenaard0480092017-11-16 22:20:39 +0100372
373 char_u *
374last_search_pattern(void)
375{
376 return spats[RE_SEARCH].pat;
377}
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100378#endif
379
Bram Moolenaar071d4272004-06-13 20:20:40 +0000380/*
381 * Return TRUE when case should be ignored for search pattern "pat".
382 * Uses the 'ignorecase' and 'smartcase' options.
383 */
384 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100385ignorecase(char_u *pat)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000386{
Bram Moolenaar66e29d72016-08-20 16:57:02 +0200387 return ignorecase_opt(pat, p_ic, p_scs);
388}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000389
Bram Moolenaar66e29d72016-08-20 16:57:02 +0200390/*
391 * As ignorecase() put pass the "ic" and "scs" flags.
392 */
393 int
394ignorecase_opt(char_u *pat, int ic_in, int scs)
395{
396 int ic = ic_in;
397
398 if (ic && !no_smartcase && scs
Bram Moolenaare2c453d2019-08-21 14:37:09 +0200399 && !(ctrl_x_mode_not_default() && curbuf->b_p_inf))
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200400 ic = !pat_has_uppercase(pat);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000401 no_smartcase = FALSE;
402
403 return ic;
404}
405
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200406/*
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200407 * Return TRUE if pattern "pat" has an uppercase character.
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200408 */
409 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100410pat_has_uppercase(char_u *pat)
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200411{
412 char_u *p = pat;
Christian Brabandt78ba9332021-08-01 12:44:37 +0200413 magic_T magic_val = MAGIC_ON;
414
415 // get the magicness of the pattern
416 (void)skip_regexp_ex(pat, NUL, magic_isset(), NULL, NULL, &magic_val);
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200417
418 while (*p != NUL)
419 {
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200420 int l;
421
422 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
423 {
424 if (enc_utf8 && utf_isupper(utf_ptr2char(p)))
425 return TRUE;
426 p += l;
427 }
Christian Brabandtbc67e5a2021-08-05 15:24:59 +0200428 else if (*p == '\\' && magic_val <= MAGIC_ON)
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200429 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100430 if (p[1] == '_' && p[2] != NUL) // skip "\_X"
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200431 p += 3;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100432 else if (p[1] == '%' && p[2] != NUL) // skip "\%X"
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200433 p += 3;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100434 else if (p[1] != NUL) // skip "\X"
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200435 p += 2;
436 else
437 p += 1;
438 }
Christian Brabandt78ba9332021-08-01 12:44:37 +0200439 else if ((*p == '%' || *p == '_') && magic_val == MAGIC_ALL)
440 {
441 if (p[1] != NUL) // skip "_X" and %X
442 p += 2;
Christian Brabandtbc67e5a2021-08-05 15:24:59 +0200443 else
444 p++;
Christian Brabandt78ba9332021-08-01 12:44:37 +0200445 }
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200446 else if (MB_ISUPPER(*p))
447 return TRUE;
448 else
449 ++p;
450 }
451 return FALSE;
452}
453
Bram Moolenaar113e1072019-01-20 15:30:40 +0100454#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000455 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100456last_csearch(void)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200457{
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200458 return lastc_bytes;
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200459}
460
461 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100462last_csearch_forward(void)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200463{
464 return lastcdir == FORWARD;
465}
466
467 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100468last_csearch_until(void)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200469{
470 return last_t_cmd == TRUE;
471}
472
473 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100474set_last_csearch(int c, char_u *s UNUSED, int len UNUSED)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200475{
476 *lastc = c;
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200477 lastc_bytelen = len;
478 if (len)
479 memcpy(lastc_bytes, s, len);
480 else
Bram Moolenaara80faa82020-04-12 19:37:17 +0200481 CLEAR_FIELD(lastc_bytes);
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200482}
Bram Moolenaar113e1072019-01-20 15:30:40 +0100483#endif
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200484
485 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100486set_csearch_direction(int cdir)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200487{
488 lastcdir = cdir;
489}
490
491 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100492set_csearch_until(int t_cmd)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200493{
494 last_t_cmd = t_cmd;
495}
496
497 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100498last_search_pat(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000499{
500 return spats[last_idx].pat;
501}
502
503/*
504 * Reset search direction to forward. For "gd" and "gD" commands.
505 */
506 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100507reset_search_dir(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000508{
509 spats[0].off.dir = '/';
Bram Moolenaar8c8de832008-06-24 22:58:06 +0000510#if defined(FEAT_EVAL)
511 set_vv_searchforward();
512#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000513}
514
515#if defined(FEAT_EVAL) || defined(FEAT_VIMINFO)
516/*
517 * Set the last search pattern. For ":let @/ =" and viminfo.
518 * Also set the saved search pattern, so that this works in an autocommand.
519 */
520 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100521set_last_search_pat(
522 char_u *s,
523 int idx,
524 int magic,
525 int setlast)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000526{
527 vim_free(spats[idx].pat);
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100528 // An empty string means that nothing should be matched.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000529 if (*s == NUL)
530 spats[idx].pat = NULL;
531 else
532 spats[idx].pat = vim_strsave(s);
533 spats[idx].magic = magic;
534 spats[idx].no_scs = FALSE;
535 spats[idx].off.dir = '/';
Bram Moolenaar8c8de832008-06-24 22:58:06 +0000536#if defined(FEAT_EVAL)
537 set_vv_searchforward();
538#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000539 spats[idx].off.line = FALSE;
540 spats[idx].off.end = FALSE;
541 spats[idx].off.off = 0;
542 if (setlast)
543 last_idx = idx;
544 if (save_level)
545 {
546 vim_free(saved_spats[idx].pat);
547 saved_spats[idx] = spats[0];
548 if (spats[idx].pat == NULL)
549 saved_spats[idx].pat = NULL;
550 else
551 saved_spats[idx].pat = vim_strsave(spats[idx].pat);
Bram Moolenaar975880b2019-03-03 14:42:11 +0100552# ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100553 saved_spats_last_idx = last_idx;
Bram Moolenaar975880b2019-03-03 14:42:11 +0100554# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000555 }
556# ifdef FEAT_SEARCH_EXTRA
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100557 // If 'hlsearch' set and search pat changed: need redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000558 if (p_hls && idx == last_idx && !no_hlsearch)
Bram Moolenaar1c8f93f2006-03-12 22:10:07 +0000559 redraw_all_later(SOME_VALID);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000560# endif
561}
562#endif
563
564#ifdef FEAT_SEARCH_EXTRA
565/*
566 * Get a regexp program for the last used search pattern.
567 * This is used for highlighting all matches in a window.
568 * Values returned in regmatch->regprog and regmatch->rmm_ic.
569 */
570 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100571last_pat_prog(regmmatch_T *regmatch)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000572{
573 if (spats[last_idx].pat == NULL)
574 {
575 regmatch->regprog = NULL;
576 return;
577 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100578 ++emsg_off; // So it doesn't beep if bad expr
Bram Moolenaar071d4272004-06-13 20:20:40 +0000579 (void)search_regcomp((char_u *)"", 0, last_idx, SEARCH_KEEP, regmatch);
580 --emsg_off;
581}
582#endif
583
584/*
Bram Moolenaarf7ff6e82014-03-23 15:13:05 +0100585 * Lowest level search function.
Bram Moolenaar5d24a222018-12-23 19:10:09 +0100586 * Search for 'count'th occurrence of pattern "pat" in direction "dir".
587 * Start at position "pos" and return the found position in "pos".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000588 *
589 * if (options & SEARCH_MSG) == 0 don't give any messages
590 * if (options & SEARCH_MSG) == SEARCH_NFMSG don't give 'notfound' messages
591 * if (options & SEARCH_MSG) == SEARCH_MSG give all messages
592 * if (options & SEARCH_HIS) put search pattern in history
593 * if (options & SEARCH_END) return position at end of match
594 * if (options & SEARCH_START) accept match at pos itself
595 * if (options & SEARCH_KEEP) keep previous search pattern
596 * if (options & SEARCH_FOLD) match only once in a closed fold
597 * if (options & SEARCH_PEEK) check for typed char, cancel search
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100598 * if (options & SEARCH_COL) start at pos->col instead of zero
Bram Moolenaar071d4272004-06-13 20:20:40 +0000599 *
600 * Return FAIL (zero) for failure, non-zero for success.
601 * When FEAT_EVAL is defined, returns the index of the first matching
602 * subpattern plus one; one if there was none.
603 */
604 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100605searchit(
Bram Moolenaar92ea26b2019-10-18 20:53:34 +0200606 win_T *win, // window to search in; can be NULL for a
607 // buffer without a window!
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100608 buf_T *buf,
609 pos_T *pos,
Bram Moolenaar5d24a222018-12-23 19:10:09 +0100610 pos_T *end_pos, // set to end of the match, unless NULL
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100611 int dir,
612 char_u *pat,
613 long count,
614 int options,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +0200615 int pat_use, // which pattern to use when "pat" is empty
616 searchit_arg_T *extra_arg) // optional extra arguments, can be NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000617{
618 int found;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100619 linenr_T lnum; // no init to shut up Apollo cc
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100620 colnr_T col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000621 regmmatch_T regmatch;
622 char_u *ptr;
623 colnr_T matchcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000624 lpos_T endpos;
Bram Moolenaar677ee682005-01-27 14:41:15 +0000625 lpos_T matchpos;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000626 int loop;
627 pos_T start_pos;
628 int at_first_line;
629 int extra_col;
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200630 int start_char_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000631 int match_ok;
632 long nmatched;
633 int submatch = 0;
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100634 int first_match = TRUE;
Bram Moolenaar53989552019-12-23 22:59:18 +0100635 int called_emsg_before = called_emsg;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000636#ifdef FEAT_SEARCH_EXTRA
637 int break_loop = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000638#endif
Bram Moolenaar92ea26b2019-10-18 20:53:34 +0200639 linenr_T stop_lnum = 0; // stop after this line number when != 0
640#ifdef FEAT_RELTIME
641 proftime_T *tm = NULL; // timeout limit or NULL
642 int *timed_out = NULL; // set when timed out or NULL
643#endif
644
645 if (extra_arg != NULL)
646 {
647 stop_lnum = extra_arg->sa_stop_lnum;
648#ifdef FEAT_RELTIME
649 tm = extra_arg->sa_tm;
650 timed_out = &extra_arg->sa_timed_out;
651#endif
652 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000653
654 if (search_regcomp(pat, RE_SEARCH, pat_use,
655 (options & (SEARCH_HIS + SEARCH_KEEP)), &regmatch) == FAIL)
656 {
657 if ((options & SEARCH_MSG) && !rc_did_emsg)
Bram Moolenaarac78dd42022-01-02 19:25:26 +0000658 semsg(_(e_invalid_search_string_str), mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000659 return FAIL;
660 }
661
Bram Moolenaar280f1262006-01-30 00:14:18 +0000662 /*
663 * find the string
664 */
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100665 do // loop for count
Bram Moolenaar071d4272004-06-13 20:20:40 +0000666 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100667 // When not accepting a match at the start position set "extra_col" to
668 // a non-zero value. Don't do that when starting at MAXCOL, since
669 // MAXCOL + 1 is zero.
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200670 if (pos->col == MAXCOL)
671 start_char_len = 0;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100672 // Watch out for the "col" being MAXCOL - 2, used in a closed fold.
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200673 else if (has_mbyte
674 && pos->lnum >= 1 && pos->lnum <= buf->b_ml.ml_line_count
675 && pos->col < MAXCOL - 2)
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100676 {
Bram Moolenaar82846a02018-02-09 18:09:54 +0100677 ptr = ml_get_buf(buf, pos->lnum, FALSE);
Bram Moolenaar8846ac52018-02-09 19:24:01 +0100678 if ((int)STRLEN(ptr) <= pos->col)
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200679 start_char_len = 1;
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100680 else
Bram Moolenaar82846a02018-02-09 18:09:54 +0100681 start_char_len = (*mb_ptr2len)(ptr + pos->col);
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100682 }
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100683 else
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200684 start_char_len = 1;
685 if (dir == FORWARD)
686 {
687 if (options & SEARCH_START)
688 extra_col = 0;
689 else
690 extra_col = start_char_len;
691 }
692 else
693 {
694 if (options & SEARCH_START)
695 extra_col = start_char_len;
696 else
697 extra_col = 0;
698 }
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100699
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100700 start_pos = *pos; // remember start pos for detecting no match
701 found = 0; // default: not found
702 at_first_line = TRUE; // default: start in first line
703 if (pos->lnum == 0) // correct lnum for when starting in line 0
Bram Moolenaar071d4272004-06-13 20:20:40 +0000704 {
705 pos->lnum = 1;
706 pos->col = 0;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100707 at_first_line = FALSE; // not in first line now
Bram Moolenaar071d4272004-06-13 20:20:40 +0000708 }
709
710 /*
711 * Start searching in current line, unless searching backwards and
712 * we're in column 0.
Bram Moolenaar7a42fa32007-07-10 11:28:55 +0000713 * If we are searching backwards, in column 0, and not including the
714 * current position, gain some efficiency by skipping back a line.
715 * Otherwise begin the search in the current line.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000716 */
Bram Moolenaar7a42fa32007-07-10 11:28:55 +0000717 if (dir == BACKWARD && start_pos.col == 0
718 && (options & SEARCH_START) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000719 {
720 lnum = pos->lnum - 1;
721 at_first_line = FALSE;
722 }
723 else
724 lnum = pos->lnum;
725
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100726 for (loop = 0; loop <= 1; ++loop) // loop twice if 'wrapscan' set
Bram Moolenaar071d4272004-06-13 20:20:40 +0000727 {
728 for ( ; lnum > 0 && lnum <= buf->b_ml.ml_line_count;
729 lnum += dir, at_first_line = FALSE)
730 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100731 // Stop after checking "stop_lnum", if it's set.
Bram Moolenaara23ccb82006-02-27 00:08:02 +0000732 if (stop_lnum != 0 && (dir == FORWARD
733 ? lnum > stop_lnum : lnum < stop_lnum))
734 break;
Bram Moolenaar76929292008-01-06 19:07:36 +0000735#ifdef FEAT_RELTIME
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100736 // Stop after passing the "tm" time limit.
Bram Moolenaar76929292008-01-06 19:07:36 +0000737 if (tm != NULL && profile_passed_limit(tm))
738 break;
739#endif
Bram Moolenaara23ccb82006-02-27 00:08:02 +0000740
Bram Moolenaar071d4272004-06-13 20:20:40 +0000741 /*
Bram Moolenaar677ee682005-01-27 14:41:15 +0000742 * Look for a match somewhere in line "lnum".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000743 */
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100744 col = at_first_line && (options & SEARCH_COL) ? pos->col
745 : (colnr_T)0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000746 nmatched = vim_regexec_multi(&regmatch, win, buf,
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100747 lnum, col,
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000748#ifdef FEAT_RELTIME
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200749 tm, timed_out
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000750#else
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200751 NULL, NULL
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000752#endif
753 );
Bram Moolenaar795aaa12020-10-02 20:36:01 +0200754 // vim_regexec_multi() may clear "regprog"
755 if (regmatch.regprog == NULL)
756 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100757 // Abort searching on an error (e.g., out of stack).
Bram Moolenaar53989552019-12-23 22:59:18 +0100758 if (called_emsg > called_emsg_before
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200759#ifdef FEAT_RELTIME
760 || (timed_out != NULL && *timed_out)
761#endif
762 )
Bram Moolenaar071d4272004-06-13 20:20:40 +0000763 break;
764 if (nmatched > 0)
765 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100766 // match may actually be in another line when using \zs
Bram Moolenaar677ee682005-01-27 14:41:15 +0000767 matchpos = regmatch.startpos[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000768 endpos = regmatch.endpos[0];
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000769#ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000770 submatch = first_submatch(&regmatch);
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000771#endif
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100772 // "lnum" may be past end of buffer for "\n\zs".
Bram Moolenaar32466aa2006-02-24 23:53:04 +0000773 if (lnum + matchpos.lnum > buf->b_ml.ml_line_count)
774 ptr = (char_u *)"";
775 else
776 ptr = ml_get_buf(buf, lnum + matchpos.lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000777
778 /*
779 * Forward search in the first line: match should be after
780 * the start position. If not, continue at the end of the
781 * match (this is vi compatible) or on the next char.
782 */
783 if (dir == FORWARD && at_first_line)
784 {
785 match_ok = TRUE;
786 /*
Bram Moolenaar677ee682005-01-27 14:41:15 +0000787 * When the match starts in a next line it's certainly
788 * past the start position.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000789 * When match lands on a NUL the cursor will be put
790 * one back afterwards, compare with that position,
791 * otherwise "/$" will get stuck on end of line.
792 */
Bram Moolenaar677ee682005-01-27 14:41:15 +0000793 while (matchpos.lnum == 0
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100794 && ((options & SEARCH_END) && first_match
Bram Moolenaar677ee682005-01-27 14:41:15 +0000795 ? (nmatched == 1
796 && (int)endpos.col - 1
Bram Moolenaar071d4272004-06-13 20:20:40 +0000797 < (int)start_pos.col + extra_col)
Bram Moolenaar677ee682005-01-27 14:41:15 +0000798 : ((int)matchpos.col
799 - (ptr[matchpos.col] == NUL)
800 < (int)start_pos.col + extra_col)))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000801 {
802 /*
803 * If vi-compatible searching, continue at the end
804 * of the match, otherwise continue one position
805 * forward.
806 */
807 if (vim_strchr(p_cpo, CPO_SEARCH) != NULL)
808 {
809 if (nmatched > 1)
810 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100811 // end is in next line, thus no match in
812 // this line
Bram Moolenaar071d4272004-06-13 20:20:40 +0000813 match_ok = FALSE;
814 break;
815 }
816 matchcol = endpos.col;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100817 // for empty match: advance one char
Bram Moolenaar677ee682005-01-27 14:41:15 +0000818 if (matchcol == matchpos.col
Bram Moolenaar071d4272004-06-13 20:20:40 +0000819 && ptr[matchcol] != NUL)
820 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000821 if (has_mbyte)
822 matchcol +=
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000823 (*mb_ptr2len)(ptr + matchcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000824 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000825 ++matchcol;
826 }
827 }
828 else
829 {
Bram Moolenaar677ee682005-01-27 14:41:15 +0000830 matchcol = matchpos.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000831 if (ptr[matchcol] != NUL)
832 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000833 if (has_mbyte)
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000834 matchcol += (*mb_ptr2len)(ptr
Bram Moolenaar071d4272004-06-13 20:20:40 +0000835 + matchcol);
836 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000837 ++matchcol;
838 }
839 }
Bram Moolenaar7bcb30e2013-04-03 21:14:29 +0200840 if (matchcol == 0 && (options & SEARCH_START))
Bram Moolenaardb333a52013-03-19 15:27:48 +0100841 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000842 if (ptr[matchcol] == NUL
843 || (nmatched = vim_regexec_multi(&regmatch,
Bram Moolenaar677ee682005-01-27 14:41:15 +0000844 win, buf, lnum + matchpos.lnum,
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000845 matchcol,
846#ifdef FEAT_RELTIME
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200847 tm, timed_out
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000848#else
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200849 NULL, NULL
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000850#endif
851 )) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000852 {
853 match_ok = FALSE;
854 break;
855 }
Bram Moolenaar795aaa12020-10-02 20:36:01 +0200856 // vim_regexec_multi() may clear "regprog"
857 if (regmatch.regprog == NULL)
858 break;
Bram Moolenaar677ee682005-01-27 14:41:15 +0000859 matchpos = regmatch.startpos[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000860 endpos = regmatch.endpos[0];
861# ifdef FEAT_EVAL
862 submatch = first_submatch(&regmatch);
863# endif
864
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100865 // Need to get the line pointer again, a
866 // multi-line search may have made it invalid.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000867 ptr = ml_get_buf(buf, lnum + matchpos.lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000868 }
869 if (!match_ok)
870 continue;
871 }
872 if (dir == BACKWARD)
873 {
874 /*
875 * Now, if there are multiple matches on this line,
876 * we have to get the last one. Or the last one before
877 * the cursor, if we're on that line.
878 * When putting the new cursor at the end, compare
879 * relative to the end of the match.
880 */
881 match_ok = FALSE;
882 for (;;)
883 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100884 // Remember a position that is before the start
885 // position, we use it if it's the last match in
886 // the line. Always accept a position after
887 // wrapping around.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000888 if (loop
889 || ((options & SEARCH_END)
890 ? (lnum + regmatch.endpos[0].lnum
891 < start_pos.lnum
892 || (lnum + regmatch.endpos[0].lnum
893 == start_pos.lnum
894 && (int)regmatch.endpos[0].col - 1
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200895 < (int)start_pos.col
896 + extra_col))
Bram Moolenaar677ee682005-01-27 14:41:15 +0000897 : (lnum + regmatch.startpos[0].lnum
898 < start_pos.lnum
899 || (lnum + regmatch.startpos[0].lnum
900 == start_pos.lnum
901 && (int)regmatch.startpos[0].col
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200902 < (int)start_pos.col
903 + extra_col))))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000904 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000905 match_ok = TRUE;
Bram Moolenaar677ee682005-01-27 14:41:15 +0000906 matchpos = regmatch.startpos[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000907 endpos = regmatch.endpos[0];
908# ifdef FEAT_EVAL
909 submatch = first_submatch(&regmatch);
910# endif
911 }
912 else
913 break;
914
915 /*
916 * We found a valid match, now check if there is
917 * another one after it.
918 * If vi-compatible searching, continue at the end
919 * of the match, otherwise continue one position
920 * forward.
921 */
922 if (vim_strchr(p_cpo, CPO_SEARCH) != NULL)
923 {
924 if (nmatched > 1)
925 break;
926 matchcol = endpos.col;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100927 // for empty match: advance one char
Bram Moolenaar677ee682005-01-27 14:41:15 +0000928 if (matchcol == matchpos.col
Bram Moolenaar071d4272004-06-13 20:20:40 +0000929 && ptr[matchcol] != NUL)
930 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000931 if (has_mbyte)
932 matchcol +=
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000933 (*mb_ptr2len)(ptr + matchcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000934 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000935 ++matchcol;
936 }
937 }
938 else
939 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100940 // Stop when the match is in a next line.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000941 if (matchpos.lnum > 0)
942 break;
943 matchcol = matchpos.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000944 if (ptr[matchcol] != NUL)
945 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000946 if (has_mbyte)
947 matchcol +=
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000948 (*mb_ptr2len)(ptr + matchcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000949 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000950 ++matchcol;
951 }
952 }
953 if (ptr[matchcol] == NUL
954 || (nmatched = vim_regexec_multi(&regmatch,
Bram Moolenaar677ee682005-01-27 14:41:15 +0000955 win, buf, lnum + matchpos.lnum,
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000956 matchcol,
957#ifdef FEAT_RELTIME
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200958 tm, timed_out
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000959#else
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200960 NULL, NULL
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000961#endif
962 )) == 0)
Bram Moolenaar9d322762018-02-09 16:04:25 +0100963 {
964#ifdef FEAT_RELTIME
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100965 // If the search timed out, we did find a match
966 // but it might be the wrong one, so that's not
967 // OK.
Bram Moolenaar9d322762018-02-09 16:04:25 +0100968 if (timed_out != NULL && *timed_out)
969 match_ok = FALSE;
970#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000971 break;
Bram Moolenaar9d322762018-02-09 16:04:25 +0100972 }
Bram Moolenaar795aaa12020-10-02 20:36:01 +0200973 // vim_regexec_multi() may clear "regprog"
974 if (regmatch.regprog == NULL)
975 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000976
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100977 // Need to get the line pointer again, a
978 // multi-line search may have made it invalid.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000979 ptr = ml_get_buf(buf, lnum + matchpos.lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000980 }
981
982 /*
983 * If there is only a match after the cursor, skip
984 * this match.
985 */
986 if (!match_ok)
987 continue;
988 }
989
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100990 // With the SEARCH_END option move to the last character
991 // of the match. Don't do it for an empty match, end
992 // should be same as start then.
Bram Moolenaar7bcb30e2013-04-03 21:14:29 +0200993 if ((options & SEARCH_END) && !(options & SEARCH_NOOF)
Bram Moolenaar5bcbd532008-02-20 12:43:01 +0000994 && !(matchpos.lnum == endpos.lnum
995 && matchpos.col == endpos.col))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000996 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100997 // For a match in the first column, set the position
998 // on the NUL in the previous line.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000999 pos->lnum = lnum + endpos.lnum;
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001000 pos->col = endpos.col;
1001 if (endpos.col == 0)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00001002 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001003 if (pos->lnum > 1) // just in case
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001004 {
1005 --pos->lnum;
1006 pos->col = (colnr_T)STRLEN(ml_get_buf(buf,
1007 pos->lnum, FALSE));
1008 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00001009 }
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001010 else
1011 {
1012 --pos->col;
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001013 if (has_mbyte
1014 && pos->lnum <= buf->b_ml.ml_line_count)
1015 {
1016 ptr = ml_get_buf(buf, pos->lnum, FALSE);
1017 pos->col -= (*mb_head_off)(ptr, ptr + pos->col);
1018 }
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001019 }
Bram Moolenaar5d24a222018-12-23 19:10:09 +01001020 if (end_pos != NULL)
1021 {
1022 end_pos->lnum = lnum + matchpos.lnum;
1023 end_pos->col = matchpos.col;
1024 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001025 }
1026 else
1027 {
Bram Moolenaar677ee682005-01-27 14:41:15 +00001028 pos->lnum = lnum + matchpos.lnum;
1029 pos->col = matchpos.col;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01001030 if (end_pos != NULL)
1031 {
1032 end_pos->lnum = lnum + endpos.lnum;
1033 end_pos->col = endpos.col;
1034 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001035 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001036 pos->coladd = 0;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01001037 if (end_pos != NULL)
1038 end_pos->coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001039 found = 1;
Bram Moolenaara3dfccc2014-11-27 17:29:56 +01001040 first_match = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001041
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001042 // Set variables used for 'incsearch' highlighting.
Bram Moolenaar677ee682005-01-27 14:41:15 +00001043 search_match_lines = endpos.lnum - matchpos.lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001044 search_match_endcol = endpos.col;
1045 break;
1046 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001047 line_breakcheck(); // stop if ctrl-C typed
Bram Moolenaar071d4272004-06-13 20:20:40 +00001048 if (got_int)
1049 break;
1050
1051#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001052 // Cancel searching if a character was typed. Used for
1053 // 'incsearch'. Don't check too often, that would slowdown
1054 // searching too much.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001055 if ((options & SEARCH_PEEK)
1056 && ((lnum - pos->lnum) & 0x3f) == 0
1057 && char_avail())
1058 {
1059 break_loop = TRUE;
1060 break;
1061 }
1062#endif
1063
1064 if (loop && lnum == start_pos.lnum)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001065 break; // if second loop, stop where started
Bram Moolenaar071d4272004-06-13 20:20:40 +00001066 }
1067 at_first_line = FALSE;
1068
Bram Moolenaar795aaa12020-10-02 20:36:01 +02001069 // vim_regexec_multi() may clear "regprog"
1070 if (regmatch.regprog == NULL)
1071 break;
1072
Bram Moolenaar071d4272004-06-13 20:20:40 +00001073 /*
Bram Moolenaara23ccb82006-02-27 00:08:02 +00001074 * Stop the search if wrapscan isn't set, "stop_lnum" is
1075 * specified, after an interrupt, after a match and after looping
1076 * twice.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001077 */
Bram Moolenaar53989552019-12-23 22:59:18 +01001078 if (!p_ws || stop_lnum != 0 || got_int
1079 || called_emsg > called_emsg_before
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +02001080#ifdef FEAT_RELTIME
1081 || (timed_out != NULL && *timed_out)
Bram Moolenaar78a15312009-05-15 19:33:18 +00001082#endif
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +02001083#ifdef FEAT_SEARCH_EXTRA
1084 || break_loop
1085#endif
1086 || found || loop)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001087 break;
1088
1089 /*
1090 * If 'wrapscan' is set we continue at the other end of the file.
1091 * If 'shortmess' does not contain 's', we give a message.
1092 * This message is also remembered in keep_msg for when the screen
1093 * is redrawn. The keep_msg is cleared whenever another message is
1094 * written.
1095 */
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001096 if (dir == BACKWARD) // start second loop at the other end
Bram Moolenaar071d4272004-06-13 20:20:40 +00001097 lnum = buf->b_ml.ml_line_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001098 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001099 lnum = 1;
Bram Moolenaar92d640f2005-09-05 22:11:52 +00001100 if (!shortmess(SHM_SEARCH) && (options & SEARCH_MSG))
1101 give_warning((char_u *)_(dir == BACKWARD
1102 ? top_bot_msg : bot_top_msg), TRUE);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001103 if (extra_arg != NULL)
1104 extra_arg->sa_wrapped = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001105 }
Bram Moolenaar53989552019-12-23 22:59:18 +01001106 if (got_int || called_emsg > called_emsg_before
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +02001107#ifdef FEAT_RELTIME
1108 || (timed_out != NULL && *timed_out)
1109#endif
Bram Moolenaar78a15312009-05-15 19:33:18 +00001110#ifdef FEAT_SEARCH_EXTRA
1111 || break_loop
1112#endif
1113 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00001114 break;
1115 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001116 while (--count > 0 && found); // stop after count matches or no match
Bram Moolenaar071d4272004-06-13 20:20:40 +00001117
Bram Moolenaar473de612013-06-08 18:19:48 +02001118 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001119
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001120 if (!found) // did not find it
Bram Moolenaar071d4272004-06-13 20:20:40 +00001121 {
1122 if (got_int)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001123 emsg(_(e_interrupted));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001124 else if ((options & SEARCH_MSG) == SEARCH_MSG)
1125 {
1126 if (p_ws)
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00001127 semsg(_(e_pattern_not_found_str), mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001128 else if (lnum == 0)
Bram Moolenaarac78dd42022-01-02 19:25:26 +00001129 semsg(_(e_search_hit_top_without_match_for_str), mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001130 else
Bram Moolenaarac78dd42022-01-02 19:25:26 +00001131 semsg(_(e_search_hit_bottom_without_match_for_str), mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001132 }
1133 return FAIL;
1134 }
1135
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001136 // A pattern like "\n\zs" may go past the last line.
Bram Moolenaar32466aa2006-02-24 23:53:04 +00001137 if (pos->lnum > buf->b_ml.ml_line_count)
1138 {
1139 pos->lnum = buf->b_ml.ml_line_count;
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001140 pos->col = (int)STRLEN(ml_get_buf(buf, pos->lnum, FALSE));
Bram Moolenaar32466aa2006-02-24 23:53:04 +00001141 if (pos->col > 0)
1142 --pos->col;
1143 }
1144
Bram Moolenaar071d4272004-06-13 20:20:40 +00001145 return submatch + 1;
1146}
1147
1148#ifdef FEAT_EVAL
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001149 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001150set_search_direction(int cdir)
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001151{
1152 spats[0].off.dir = cdir;
1153}
1154
1155 static void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001156set_vv_searchforward(void)
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001157{
1158 set_vim_var_nr(VV_SEARCHFORWARD, (long)(spats[0].off.dir == '/'));
1159}
1160
Bram Moolenaar071d4272004-06-13 20:20:40 +00001161/*
1162 * Return the number of the first subpat that matched.
Bram Moolenaarad4d8a12015-12-28 19:20:36 +01001163 * Return zero if none of them matched.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001164 */
1165 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001166first_submatch(regmmatch_T *rp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001167{
1168 int submatch;
1169
1170 for (submatch = 1; ; ++submatch)
1171 {
1172 if (rp->startpos[submatch].lnum >= 0)
1173 break;
1174 if (submatch == 9)
1175 {
1176 submatch = 0;
1177 break;
1178 }
1179 }
1180 return submatch;
1181}
1182#endif
1183
1184/*
1185 * Highest level string search function.
Bram Moolenaarb8017e72007-05-10 18:59:07 +00001186 * Search for the 'count'th occurrence of pattern 'pat' in direction 'dirc'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001187 * If 'dirc' is 0: use previous dir.
1188 * If 'pat' is NULL or empty : use previous string.
1189 * If 'options & SEARCH_REV' : go in reverse of previous dir.
1190 * If 'options & SEARCH_ECHO': echo the search command and handle options
1191 * If 'options & SEARCH_MSG' : may give error message
1192 * If 'options & SEARCH_OPT' : interpret optional flags
1193 * If 'options & SEARCH_HIS' : put search pattern in history
1194 * If 'options & SEARCH_NOOF': don't add offset to position
1195 * If 'options & SEARCH_MARK': set previous context mark
1196 * If 'options & SEARCH_KEEP': keep previous search pattern
1197 * If 'options & SEARCH_START': accept match at curpos itself
1198 * If 'options & SEARCH_PEEK': check for typed char, cancel search
1199 *
1200 * Careful: If spats[0].off.line == TRUE and spats[0].off.off == 0 this
1201 * makes the movement linewise without moving the match position.
1202 *
Bram Moolenaarb6c27352015-03-05 19:57:49 +01001203 * Return 0 for failure, 1 for found, 2 for found and line offset added.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001204 */
1205 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001206do_search(
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001207 oparg_T *oap, // can be NULL
1208 int dirc, // '/' or '?'
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001209 int search_delim, // the delimiter for the search, e.g. '%' in
1210 // s%regex%replacement%
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001211 char_u *pat,
1212 long count,
1213 int options,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001214 searchit_arg_T *sia) // optional arguments or NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001215{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001216 pos_T pos; // position of the last match
Bram Moolenaar071d4272004-06-13 20:20:40 +00001217 char_u *searchstr;
Bram Moolenaarc3328162019-07-23 22:15:25 +02001218 soffset_T old_off;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001219 int retval; // Return value
Bram Moolenaar071d4272004-06-13 20:20:40 +00001220 char_u *p;
1221 long c;
1222 char_u *dircp;
1223 char_u *strcopy = NULL;
1224 char_u *ps;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001225 char_u *msgbuf = NULL;
1226 size_t len;
Bram Moolenaar8f46e4c2019-05-24 22:08:15 +02001227 int has_offset = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001228
1229 /*
1230 * A line offset is not remembered, this is vi compatible.
1231 */
1232 if (spats[0].off.line && vim_strchr(p_cpo, CPO_LINEOFF) != NULL)
1233 {
1234 spats[0].off.line = FALSE;
1235 spats[0].off.off = 0;
1236 }
1237
1238 /*
1239 * Save the values for when (options & SEARCH_KEEP) is used.
1240 * (there is no "if ()" around this because gcc wants them initialized)
1241 */
1242 old_off = spats[0].off;
1243
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001244 pos = curwin->w_cursor; // start searching at the cursor position
Bram Moolenaar071d4272004-06-13 20:20:40 +00001245
1246 /*
1247 * Find out the direction of the search.
1248 */
1249 if (dirc == 0)
1250 dirc = spats[0].off.dir;
1251 else
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001252 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001253 spats[0].off.dir = dirc;
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001254#if defined(FEAT_EVAL)
1255 set_vv_searchforward();
1256#endif
1257 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001258 if (options & SEARCH_REV)
1259 {
Bram Moolenaar4f974752019-02-17 17:44:42 +01001260#ifdef MSWIN
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001261 // There is a bug in the Visual C++ 2.2 compiler which means that
1262 // dirc always ends up being '/'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001263 dirc = (dirc == '/') ? '?' : '/';
1264#else
1265 if (dirc == '/')
1266 dirc = '?';
1267 else
1268 dirc = '/';
1269#endif
1270 }
1271
1272#ifdef FEAT_FOLDING
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001273 // If the cursor is in a closed fold, don't find another match in the same
1274 // fold.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001275 if (dirc == '/')
1276 {
1277 if (hasFolding(pos.lnum, NULL, &pos.lnum))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001278 pos.col = MAXCOL - 2; // avoid overflow when adding 1
Bram Moolenaar071d4272004-06-13 20:20:40 +00001279 }
1280 else
1281 {
1282 if (hasFolding(pos.lnum, &pos.lnum, NULL))
1283 pos.col = 0;
1284 }
1285#endif
1286
1287#ifdef FEAT_SEARCH_EXTRA
1288 /*
1289 * Turn 'hlsearch' highlighting back on.
1290 */
1291 if (no_hlsearch && !(options & SEARCH_KEEP))
1292 {
Bram Moolenaar1c8f93f2006-03-12 22:10:07 +00001293 redraw_all_later(SOME_VALID);
Bram Moolenaar451fc7b2018-04-27 22:53:07 +02001294 set_no_hlsearch(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001295 }
1296#endif
1297
1298 /*
1299 * Repeat the search when pattern followed by ';', e.g. "/foo/;?bar".
1300 */
1301 for (;;)
1302 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001303 int show_top_bot_msg = FALSE;
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02001304
Bram Moolenaar071d4272004-06-13 20:20:40 +00001305 searchstr = pat;
1306 dircp = NULL;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001307 // use previous pattern
Bram Moolenaarc036e872020-02-21 21:30:52 +01001308 if (pat == NULL || *pat == NUL || *pat == search_delim)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001309 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001310 if (spats[RE_SEARCH].pat == NULL) // no previous pattern
Bram Moolenaar071d4272004-06-13 20:20:40 +00001311 {
Bram Moolenaarea683da2016-09-09 21:41:34 +02001312 searchstr = spats[RE_SUBST].pat;
1313 if (searchstr == NULL)
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001314 {
Bram Moolenaare29a27f2021-07-20 21:07:36 +02001315 emsg(_(e_no_previous_regular_expression));
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001316 retval = 0;
1317 goto end_do_search;
1318 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001319 }
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001320 else
1321 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001322 // make search_regcomp() use spats[RE_SEARCH].pat
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001323 searchstr = (char_u *)"";
1324 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001325 }
1326
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001327 if (pat != NULL && *pat != NUL) // look for (new) offset
Bram Moolenaar071d4272004-06-13 20:20:40 +00001328 {
1329 /*
1330 * Find end of regular expression.
1331 * If there is a matching '/' or '?', toss it.
1332 */
1333 ps = strcopy;
Bram Moolenaarf4e20992020-12-21 19:59:08 +01001334 p = skip_regexp_ex(pat, search_delim, magic_isset(),
Bram Moolenaard93a7fc2021-01-04 12:42:13 +01001335 &strcopy, NULL, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001336 if (strcopy != ps)
1337 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001338 // made a copy of "pat" to change "\?" to "?"
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001339 searchcmdlen += (int)(STRLEN(pat) - STRLEN(strcopy));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001340 pat = strcopy;
1341 searchstr = strcopy;
1342 }
Bram Moolenaarc036e872020-02-21 21:30:52 +01001343 if (*p == search_delim)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001344 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001345 dircp = p; // remember where we put the NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001346 *p++ = NUL;
1347 }
1348 spats[0].off.line = FALSE;
1349 spats[0].off.end = FALSE;
1350 spats[0].off.off = 0;
1351 /*
1352 * Check for a line offset or a character offset.
1353 * For get_address (echo off) we don't check for a character
1354 * offset, because it is meaningless and the 's' could be a
1355 * substitute command.
1356 */
1357 if (*p == '+' || *p == '-' || VIM_ISDIGIT(*p))
1358 spats[0].off.line = TRUE;
1359 else if ((options & SEARCH_OPT) &&
1360 (*p == 'e' || *p == 's' || *p == 'b'))
1361 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001362 if (*p == 'e') // end
Bram Moolenaar071d4272004-06-13 20:20:40 +00001363 spats[0].off.end = SEARCH_END;
1364 ++p;
1365 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001366 if (VIM_ISDIGIT(*p) || *p == '+' || *p == '-') // got an offset
Bram Moolenaar071d4272004-06-13 20:20:40 +00001367 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001368 // 'nr' or '+nr' or '-nr'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001369 if (VIM_ISDIGIT(*p) || VIM_ISDIGIT(*(p + 1)))
1370 spats[0].off.off = atol((char *)p);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001371 else if (*p == '-') // single '-'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001372 spats[0].off.off = -1;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001373 else // single '+'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001374 spats[0].off.off = 1;
1375 ++p;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001376 while (VIM_ISDIGIT(*p)) // skip number
Bram Moolenaar071d4272004-06-13 20:20:40 +00001377 ++p;
1378 }
1379
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001380 // compute length of search command for get_address()
Bram Moolenaar071d4272004-06-13 20:20:40 +00001381 searchcmdlen += (int)(p - pat);
1382
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001383 pat = p; // put pat after search command
Bram Moolenaar071d4272004-06-13 20:20:40 +00001384 }
1385
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001386 if ((options & SEARCH_ECHO) && messaging() &&
1387 !msg_silent &&
1388 (!cmd_silent || !shortmess(SHM_SEARCHCOUNT)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001389 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001390 char_u *trunc;
Bram Moolenaar984f0312019-05-24 13:11:47 +02001391 char_u off_buf[40];
Bram Moolenaard33a7642019-05-24 17:56:14 +02001392 size_t off_len = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001393
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001394 // Compute msg_row early.
1395 msg_start();
1396
Bram Moolenaar984f0312019-05-24 13:11:47 +02001397 // Get the offset, so we know how long it is.
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001398 if (!cmd_silent &&
1399 (spats[0].off.line || spats[0].off.end || spats[0].off.off))
Bram Moolenaar984f0312019-05-24 13:11:47 +02001400 {
1401 p = off_buf;
1402 *p++ = dirc;
1403 if (spats[0].off.end)
1404 *p++ = 'e';
1405 else if (!spats[0].off.line)
1406 *p++ = 's';
1407 if (spats[0].off.off > 0 || spats[0].off.line)
1408 *p++ = '+';
1409 *p = NUL;
1410 if (spats[0].off.off != 0 || spats[0].off.line)
1411 sprintf((char *)p, "%ld", spats[0].off.off);
1412 off_len = STRLEN(off_buf);
1413 }
1414
Bram Moolenaar071d4272004-06-13 20:20:40 +00001415 if (*searchstr == NUL)
Bram Moolenaar2fb8f682018-12-01 13:14:45 +01001416 p = spats[0].pat;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001417 else
1418 p = searchstr;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001419
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001420 if (!shortmess(SHM_SEARCHCOUNT) || cmd_silent)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001421 {
1422 // Reserve enough space for the search pattern + offset +
Bram Moolenaar984f0312019-05-24 13:11:47 +02001423 // search stat. Use all the space available, so that the
1424 // search state is right aligned. If there is not enough space
1425 // msg_strtrunc() will shorten in the middle.
Bram Moolenaar19e8ac72019-09-03 22:23:38 +02001426 if (msg_scrolled != 0 && !cmd_silent)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001427 // Use all the columns.
1428 len = (int)(Rows - msg_row) * Columns - 1;
1429 else
1430 // Use up to 'showcmd' column.
1431 len = (int)(Rows - msg_row - 1) * Columns + sc_col - 1;
Bram Moolenaar984f0312019-05-24 13:11:47 +02001432 if (len < STRLEN(p) + off_len + SEARCH_STAT_BUF_LEN + 3)
1433 len = STRLEN(p) + off_len + SEARCH_STAT_BUF_LEN + 3;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001434 }
1435 else
1436 // Reserve enough space for the search pattern + offset.
Bram Moolenaar984f0312019-05-24 13:11:47 +02001437 len = STRLEN(p) + off_len + 3;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001438
Bram Moolenaar880e4d92020-04-11 21:31:28 +02001439 vim_free(msgbuf);
Bram Moolenaar51e14382019-05-25 20:21:28 +02001440 msgbuf = alloc(len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001441 if (msgbuf != NULL)
1442 {
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001443 vim_memset(msgbuf, ' ', len);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001444 msgbuf[len - 1] = NUL;
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001445 // do not fill the msgbuf buffer, if cmd_silent is set, leave it
1446 // empty for the search_stat feature.
1447 if (!cmd_silent)
Bram Moolenaarcafda4f2005-09-06 19:25:11 +00001448 {
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001449 msgbuf[0] = dirc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001450
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001451 if (enc_utf8 && utf_iscomposing(utf_ptr2char(p)))
1452 {
1453 // Use a space to draw the composing char on.
1454 msgbuf[1] = ' ';
1455 mch_memmove(msgbuf + 2, p, STRLEN(p));
1456 }
1457 else
1458 mch_memmove(msgbuf + 1, p, STRLEN(p));
1459 if (off_len > 0)
1460 mch_memmove(msgbuf + STRLEN(p) + 1, off_buf, off_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001461
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001462 trunc = msg_strtrunc(msgbuf, TRUE);
1463 if (trunc != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001464 {
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001465 vim_free(msgbuf);
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001466 msgbuf = trunc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001467 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001468
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001469#ifdef FEAT_RIGHTLEFT
1470 // The search pattern could be shown on the right in
1471 // rightleft mode, but the 'ruler' and 'showcmd' area use
1472 // it too, thus it would be blanked out again very soon.
1473 // Show it on the left, but do reverse the text.
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001474 if (curwin->w_p_rl && *curwin->w_p_rlc == 's')
1475 {
1476 char_u *r;
1477 size_t pat_len;
1478
1479 r = reverse_text(msgbuf);
1480 if (r != NULL)
1481 {
1482 vim_free(msgbuf);
1483 msgbuf = r;
1484 // move reversed text to beginning of buffer
1485 while (*r != NUL && *r == ' ')
1486 r++;
1487 pat_len = msgbuf + STRLEN(msgbuf) - r;
1488 mch_memmove(msgbuf, r, pat_len);
1489 // overwrite old text
1490 if ((size_t)(r - msgbuf) >= pat_len)
1491 vim_memset(r, ' ', pat_len);
1492 else
1493 vim_memset(msgbuf + pat_len, ' ', r - msgbuf);
1494 }
1495 }
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001496#endif
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001497 msg_outtrans(msgbuf);
1498 msg_clr_eos();
1499 msg_check();
1500
1501 gotocmdline(FALSE);
1502 out_flush();
1503 msg_nowait = TRUE; // don't wait for this message
1504 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001505 }
1506 }
1507
1508 /*
1509 * If there is a character offset, subtract it from the current
1510 * position, so we don't get stuck at "?pat?e+2" or "/pat/s-2".
Bram Moolenaared203462004-06-16 11:19:22 +00001511 * Skip this if pos.col is near MAXCOL (closed fold).
Bram Moolenaar071d4272004-06-13 20:20:40 +00001512 * This is not done for a line offset, because then we would not be vi
1513 * compatible.
1514 */
Bram Moolenaared203462004-06-16 11:19:22 +00001515 if (!spats[0].off.line && spats[0].off.off && pos.col < MAXCOL - 2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001516 {
1517 if (spats[0].off.off > 0)
1518 {
1519 for (c = spats[0].off.off; c; --c)
1520 if (decl(&pos) == -1)
1521 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001522 if (c) // at start of buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001523 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001524 pos.lnum = 0; // allow lnum == 0 here
Bram Moolenaar071d4272004-06-13 20:20:40 +00001525 pos.col = MAXCOL;
1526 }
1527 }
1528 else
1529 {
1530 for (c = spats[0].off.off; c; ++c)
1531 if (incl(&pos) == -1)
1532 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001533 if (c) // at end of buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001534 {
1535 pos.lnum = curbuf->b_ml.ml_line_count + 1;
1536 pos.col = 0;
1537 }
1538 }
1539 }
1540
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001541 /*
1542 * The actual search.
1543 */
Bram Moolenaar14184a32019-02-16 15:10:30 +01001544 c = searchit(curwin, curbuf, &pos, NULL,
1545 dirc == '/' ? FORWARD : BACKWARD,
Bram Moolenaar071d4272004-06-13 20:20:40 +00001546 searchstr, count, spats[0].off.end + (options &
1547 (SEARCH_KEEP + SEARCH_PEEK + SEARCH_HIS
1548 + SEARCH_MSG + SEARCH_START
1549 + ((pat != NULL && *pat == ';') ? 0 : SEARCH_NOOF))),
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001550 RE_LAST, sia);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001551
1552 if (dircp != NULL)
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001553 *dircp = search_delim; // restore second '/' or '?' for normal_cmd()
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001554
1555 if (!shortmess(SHM_SEARCH)
1556 && ((dirc == '/' && LT_POS(pos, curwin->w_cursor))
1557 || (dirc == '?' && LT_POS(curwin->w_cursor, pos))))
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02001558 show_top_bot_msg = TRUE;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001559
Bram Moolenaar071d4272004-06-13 20:20:40 +00001560 if (c == FAIL)
1561 {
1562 retval = 0;
1563 goto end_do_search;
1564 }
1565 if (spats[0].off.end && oap != NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001566 oap->inclusive = TRUE; // 'e' includes last character
Bram Moolenaar071d4272004-06-13 20:20:40 +00001567
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001568 retval = 1; // pattern found
Bram Moolenaar071d4272004-06-13 20:20:40 +00001569
1570 /*
1571 * Add character and/or line offset
1572 */
Bram Moolenaar9160f302006-08-29 15:58:12 +00001573 if (!(options & SEARCH_NOOF) || (pat != NULL && *pat == ';'))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001574 {
Bram Moolenaar8f46e4c2019-05-24 22:08:15 +02001575 pos_T org_pos = pos;
1576
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001577 if (spats[0].off.line) // Add the offset to the line number.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001578 {
1579 c = pos.lnum + spats[0].off.off;
1580 if (c < 1)
1581 pos.lnum = 1;
1582 else if (c > curbuf->b_ml.ml_line_count)
1583 pos.lnum = curbuf->b_ml.ml_line_count;
1584 else
1585 pos.lnum = c;
1586 pos.col = 0;
1587
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001588 retval = 2; // pattern found, line offset added
Bram Moolenaar071d4272004-06-13 20:20:40 +00001589 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001590 else if (pos.col < MAXCOL - 2) // just in case
Bram Moolenaar071d4272004-06-13 20:20:40 +00001591 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001592 // to the right, check for end of file
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001593 c = spats[0].off.off;
1594 if (c > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001595 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001596 while (c-- > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001597 if (incl(&pos) == -1)
1598 break;
1599 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001600 // to the left, check for start of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00001601 else
1602 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001603 while (c++ < 0)
1604 if (decl(&pos) == -1)
1605 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001606 }
1607 }
Bram Moolenaar8f46e4c2019-05-24 22:08:15 +02001608 if (!EQUAL_POS(pos, org_pos))
1609 has_offset = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001610 }
1611
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001612 // Show [1/15] if 'S' is not in 'shortmess'.
1613 if ((options & SEARCH_ECHO)
1614 && messaging()
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001615 && !msg_silent
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001616 && c != FAIL
1617 && !shortmess(SHM_SEARCHCOUNT)
1618 && msgbuf != NULL)
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02001619 cmdline_search_stat(dirc, &pos, &curwin->w_cursor,
1620 show_top_bot_msg, msgbuf,
1621 (count != 1 || has_offset
Bram Moolenaar6cb07262020-05-29 22:49:43 +02001622#ifdef FEAT_FOLDING
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02001623 || (!(fdo_flags & FDO_SEARCH)
1624 && hasFolding(curwin->w_cursor.lnum,
1625 NULL, NULL))
Bram Moolenaar6cb07262020-05-29 22:49:43 +02001626#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02001627 ),
1628 SEARCH_STAT_DEF_MAX_COUNT,
1629 SEARCH_STAT_DEF_TIMEOUT);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001630
Bram Moolenaar071d4272004-06-13 20:20:40 +00001631 /*
1632 * The search command can be followed by a ';' to do another search.
1633 * For example: "/pat/;/foo/+3;?bar"
1634 * This is like doing another search command, except:
1635 * - The remembered direction '/' or '?' is from the first search.
1636 * - When an error happens the cursor isn't moved at all.
1637 * Don't do this when called by get_address() (it handles ';' itself).
1638 */
1639 if (!(options & SEARCH_OPT) || pat == NULL || *pat != ';')
1640 break;
1641
1642 dirc = *++pat;
Bram Moolenaarc036e872020-02-21 21:30:52 +01001643 search_delim = dirc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001644 if (dirc != '?' && dirc != '/')
1645 {
1646 retval = 0;
Bram Moolenaarac78dd42022-01-02 19:25:26 +00001647 emsg(_(e_expected_question_or_slash_after_semicolon));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001648 goto end_do_search;
1649 }
1650 ++pat;
1651 }
1652
1653 if (options & SEARCH_MARK)
1654 setpcmark();
1655 curwin->w_cursor = pos;
1656 curwin->w_set_curswant = TRUE;
1657
1658end_do_search:
Bram Moolenaare1004402020-10-24 20:49:43 +02001659 if ((options & SEARCH_KEEP) || (cmdmod.cmod_flags & CMOD_KEEPPATTERNS))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001660 spats[0].off = old_off;
1661 vim_free(strcopy);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001662 vim_free(msgbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001663
1664 return retval;
1665}
1666
Bram Moolenaar071d4272004-06-13 20:20:40 +00001667/*
1668 * search_for_exact_line(buf, pos, dir, pat)
1669 *
1670 * Search for a line starting with the given pattern (ignoring leading
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02001671 * white-space), starting from pos and going in direction "dir". "pos" will
Bram Moolenaar071d4272004-06-13 20:20:40 +00001672 * contain the position of the match found. Blank lines match only if
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02001673 * ADDING is set. If p_ic is set then the pattern must be in lowercase.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001674 * Return OK for success, or FAIL if no line found.
1675 */
1676 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001677search_for_exact_line(
1678 buf_T *buf,
1679 pos_T *pos,
1680 int dir,
1681 char_u *pat)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001682{
1683 linenr_T start = 0;
1684 char_u *ptr;
1685 char_u *p;
1686
1687 if (buf->b_ml.ml_line_count == 0)
1688 return FAIL;
1689 for (;;)
1690 {
1691 pos->lnum += dir;
1692 if (pos->lnum < 1)
1693 {
1694 if (p_ws)
1695 {
1696 pos->lnum = buf->b_ml.ml_line_count;
1697 if (!shortmess(SHM_SEARCH))
1698 give_warning((char_u *)_(top_bot_msg), TRUE);
1699 }
1700 else
1701 {
1702 pos->lnum = 1;
1703 break;
1704 }
1705 }
1706 else if (pos->lnum > buf->b_ml.ml_line_count)
1707 {
1708 if (p_ws)
1709 {
1710 pos->lnum = 1;
1711 if (!shortmess(SHM_SEARCH))
1712 give_warning((char_u *)_(bot_top_msg), TRUE);
1713 }
1714 else
1715 {
1716 pos->lnum = 1;
1717 break;
1718 }
1719 }
1720 if (pos->lnum == start)
1721 break;
1722 if (start == 0)
1723 start = pos->lnum;
1724 ptr = ml_get_buf(buf, pos->lnum, FALSE);
1725 p = skipwhite(ptr);
1726 pos->col = (colnr_T) (p - ptr);
1727
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001728 // when adding lines the matching line may be empty but it is not
1729 // ignored because we are interested in the next line -- Acevedo
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00001730 if (compl_status_adding() && !compl_status_sol())
Bram Moolenaar071d4272004-06-13 20:20:40 +00001731 {
1732 if ((p_ic ? MB_STRICMP(p, pat) : STRCMP(p, pat)) == 0)
1733 return OK;
1734 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001735 else if (*p != NUL) // ignore empty lines
1736 { // expanding lines or words
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00001737 if ((p_ic ? MB_STRNICMP(p, pat, ins_compl_len())
1738 : STRNCMP(p, pat, ins_compl_len())) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001739 return OK;
1740 }
1741 }
1742 return FAIL;
1743}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001744
1745/*
1746 * Character Searches
1747 */
1748
1749/*
1750 * Search for a character in a line. If "t_cmd" is FALSE, move to the
1751 * position of the character, otherwise move to just before the char.
1752 * Do this "cap->count1" times.
1753 * Return FAIL or OK.
1754 */
1755 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001756searchc(cmdarg_T *cap, int t_cmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001757{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001758 int c = cap->nchar; // char to search for
1759 int dir = cap->arg; // TRUE for searching forward
1760 long count = cap->count1; // repeat count
Bram Moolenaar071d4272004-06-13 20:20:40 +00001761 int col;
1762 char_u *p;
1763 int len;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001764 int stop = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001765
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001766 if (c != NUL) // normal search: remember args for repeat
Bram Moolenaar071d4272004-06-13 20:20:40 +00001767 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001768 if (!KeyStuffed) // don't remember when redoing
Bram Moolenaar071d4272004-06-13 20:20:40 +00001769 {
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001770 *lastc = c;
1771 set_csearch_direction(dir);
1772 set_csearch_until(t_cmd);
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001773 lastc_bytelen = (*mb_char2bytes)(c, lastc_bytes);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001774 if (cap->ncharC1 != 0)
1775 {
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001776 lastc_bytelen += (*mb_char2bytes)(cap->ncharC1,
1777 lastc_bytes + lastc_bytelen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001778 if (cap->ncharC2 != 0)
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001779 lastc_bytelen += (*mb_char2bytes)(cap->ncharC2,
1780 lastc_bytes + lastc_bytelen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001781 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001782 }
1783 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001784 else // repeat previous search
Bram Moolenaar071d4272004-06-13 20:20:40 +00001785 {
Bram Moolenaar264b74f2019-01-24 17:18:42 +01001786 if (*lastc == NUL && lastc_bytelen == 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001787 return FAIL;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001788 if (dir) // repeat in opposite direction
Bram Moolenaar071d4272004-06-13 20:20:40 +00001789 dir = -lastcdir;
1790 else
1791 dir = lastcdir;
1792 t_cmd = last_t_cmd;
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001793 c = *lastc;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001794 // For multi-byte re-use last lastc_bytes[] and lastc_bytelen.
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001795
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001796 // Force a move of at least one char, so ";" and "," will move the
1797 // cursor, even if the cursor is right in front of char we are looking
1798 // at.
Bram Moolenaar19fd09a2011-07-15 13:21:30 +02001799 if (vim_strchr(p_cpo, CPO_SCOLON) == NULL && count == 1 && t_cmd)
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001800 stop = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001801 }
1802
Bram Moolenaar60a795a2005-09-16 21:55:43 +00001803 if (dir == BACKWARD)
1804 cap->oap->inclusive = FALSE;
1805 else
1806 cap->oap->inclusive = TRUE;
1807
Bram Moolenaar071d4272004-06-13 20:20:40 +00001808 p = ml_get_curline();
1809 col = curwin->w_cursor.col;
1810 len = (int)STRLEN(p);
1811
1812 while (count--)
1813 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001814 if (has_mbyte)
1815 {
1816 for (;;)
1817 {
1818 if (dir > 0)
1819 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001820 col += (*mb_ptr2len)(p + col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001821 if (col >= len)
1822 return FAIL;
1823 }
1824 else
1825 {
1826 if (col == 0)
1827 return FAIL;
1828 col -= (*mb_head_off)(p, p + col - 1) + 1;
1829 }
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001830 if (lastc_bytelen == 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001831 {
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001832 if (p[col] == c && stop)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001833 break;
1834 }
Bram Moolenaar66727e12017-03-01 22:17:05 +01001835 else if (STRNCMP(p + col, lastc_bytes, lastc_bytelen) == 0
Bram Moolenaarb129a442016-12-01 17:25:20 +01001836 && stop)
Bram Moolenaar66727e12017-03-01 22:17:05 +01001837 break;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001838 stop = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001839 }
1840 }
1841 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001842 {
1843 for (;;)
1844 {
1845 if ((col += dir) < 0 || col >= len)
1846 return FAIL;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001847 if (p[col] == c && stop)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001848 break;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001849 stop = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001850 }
1851 }
1852 }
1853
1854 if (t_cmd)
1855 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001856 // backup to before the character (possibly double-byte)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001857 col -= dir;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001858 if (has_mbyte)
1859 {
1860 if (dir < 0)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001861 // Landed on the search char which is lastc_bytelen long
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001862 col += lastc_bytelen - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001863 else
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001864 // To previous char, which may be multi-byte.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001865 col -= (*mb_head_off)(p, p + col);
1866 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001867 }
1868 curwin->w_cursor.col = col;
1869
1870 return OK;
1871}
1872
1873/*
1874 * "Other" Searches
1875 */
1876
1877/*
1878 * findmatch - find the matching paren or brace
1879 *
1880 * Improvement over vi: Braces inside quotes are ignored.
1881 */
1882 pos_T *
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001883findmatch(oparg_T *oap, int initc)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001884{
1885 return findmatchlimit(oap, initc, 0, 0);
1886}
1887
1888/*
1889 * Return TRUE if the character before "linep[col]" equals "ch".
1890 * Return FALSE if "col" is zero.
1891 * Update "*prevcol" to the column of the previous character, unless "prevcol"
1892 * is NULL.
1893 * Handles multibyte string correctly.
1894 */
1895 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001896check_prevcol(
1897 char_u *linep,
1898 int col,
1899 int ch,
1900 int *prevcol)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001901{
1902 --col;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001903 if (col > 0 && has_mbyte)
1904 col -= (*mb_head_off)(linep, linep + col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001905 if (prevcol)
1906 *prevcol = col;
1907 return (col >= 0 && linep[col] == ch) ? TRUE : FALSE;
1908}
1909
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001910/*
1911 * Raw string start is found at linep[startpos.col - 1].
1912 * Return TRUE if the matching end can be found between startpos and endpos.
1913 */
1914 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001915find_rawstring_end(char_u *linep, pos_T *startpos, pos_T *endpos)
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001916{
1917 char_u *p;
1918 char_u *delim_copy;
1919 size_t delim_len;
1920 linenr_T lnum;
1921 int found = FALSE;
1922
1923 for (p = linep + startpos->col + 1; *p && *p != '('; ++p)
1924 ;
1925 delim_len = (p - linep) - startpos->col - 1;
Bram Moolenaar71ccd032020-06-12 22:59:11 +02001926 delim_copy = vim_strnsave(linep + startpos->col + 1, delim_len);
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001927 if (delim_copy == NULL)
1928 return FALSE;
1929 for (lnum = startpos->lnum; lnum <= endpos->lnum; ++lnum)
1930 {
1931 char_u *line = ml_get(lnum);
1932
1933 for (p = line + (lnum == startpos->lnum
1934 ? startpos->col + 1 : 0); *p; ++p)
1935 {
1936 if (lnum == endpos->lnum && (colnr_T)(p - line) >= endpos->col)
1937 break;
Bram Moolenaar282f9c62020-08-04 21:46:18 +02001938 if (*p == ')' && STRNCMP(delim_copy, p + 1, delim_len) == 0
1939 && p[delim_len + 1] == '"')
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001940 {
1941 found = TRUE;
1942 break;
1943 }
1944 }
1945 if (found)
1946 break;
1947 }
1948 vim_free(delim_copy);
1949 return found;
1950}
1951
Bram Moolenaar071d4272004-06-13 20:20:40 +00001952/*
Bram Moolenaar556ae8e2019-11-21 22:27:22 +01001953 * Check matchpairs option for "*initc".
1954 * If there is a match set "*initc" to the matching character and "*findc" to
1955 * the opposite character. Set "*backwards" to the direction.
1956 * When "switchit" is TRUE swap the direction.
1957 */
1958 static void
1959find_mps_values(
1960 int *initc,
1961 int *findc,
1962 int *backwards,
1963 int switchit)
1964{
1965 char_u *ptr;
1966
1967 ptr = curbuf->b_p_mps;
1968 while (*ptr != NUL)
1969 {
1970 if (has_mbyte)
1971 {
1972 char_u *prev;
1973
1974 if (mb_ptr2char(ptr) == *initc)
1975 {
1976 if (switchit)
1977 {
1978 *findc = *initc;
1979 *initc = mb_ptr2char(ptr + mb_ptr2len(ptr) + 1);
1980 *backwards = TRUE;
1981 }
1982 else
1983 {
1984 *findc = mb_ptr2char(ptr + mb_ptr2len(ptr) + 1);
1985 *backwards = FALSE;
1986 }
1987 return;
1988 }
1989 prev = ptr;
1990 ptr += mb_ptr2len(ptr) + 1;
1991 if (mb_ptr2char(ptr) == *initc)
1992 {
1993 if (switchit)
1994 {
1995 *findc = *initc;
1996 *initc = mb_ptr2char(prev);
1997 *backwards = FALSE;
1998 }
1999 else
2000 {
2001 *findc = mb_ptr2char(prev);
2002 *backwards = TRUE;
2003 }
2004 return;
2005 }
2006 ptr += mb_ptr2len(ptr);
2007 }
2008 else
2009 {
2010 if (*ptr == *initc)
2011 {
2012 if (switchit)
2013 {
2014 *backwards = TRUE;
2015 *findc = *initc;
2016 *initc = ptr[2];
2017 }
2018 else
2019 {
2020 *backwards = FALSE;
2021 *findc = ptr[2];
2022 }
2023 return;
2024 }
2025 ptr += 2;
2026 if (*ptr == *initc)
2027 {
2028 if (switchit)
2029 {
2030 *backwards = FALSE;
2031 *findc = *initc;
2032 *initc = ptr[-2];
2033 }
2034 else
2035 {
2036 *backwards = TRUE;
2037 *findc = ptr[-2];
2038 }
2039 return;
2040 }
2041 ++ptr;
2042 }
2043 if (*ptr == ',')
2044 ++ptr;
2045 }
2046}
2047
2048/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002049 * findmatchlimit -- find the matching paren or brace, if it exists within
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002050 * maxtravel lines of the cursor. A maxtravel of 0 means search until falling
2051 * off the edge of the file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002052 *
2053 * "initc" is the character to find a match for. NUL means to find the
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002054 * character at or after the cursor. Special values:
2055 * '*' look for C-style comment / *
2056 * '/' look for C-style comment / *, ignoring comment-end
2057 * '#' look for preprocessor directives
2058 * 'R' look for raw string start: R"delim(text)delim" (only backwards)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002059 *
2060 * flags: FM_BACKWARD search backwards (when initc is '/', '*' or '#')
2061 * FM_FORWARD search forwards (when initc is '/', '*' or '#')
2062 * FM_BLOCKSTOP stop at start/end of block ({ or } in column 0)
2063 * FM_SKIPCOMM skip comments (not implemented yet!)
Bram Moolenaarf75a9632005-09-13 21:20:47 +00002064 *
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002065 * "oap" is only used to set oap->motion_type for a linewise motion, it can be
Bram Moolenaarf75a9632005-09-13 21:20:47 +00002066 * NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00002067 */
2068
2069 pos_T *
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002070findmatchlimit(
2071 oparg_T *oap,
2072 int initc,
2073 int flags,
2074 int maxtravel)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002075{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002076 static pos_T pos; // current search position
2077 int findc = 0; // matching brace
Bram Moolenaar071d4272004-06-13 20:20:40 +00002078 int c;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002079 int count = 0; // cumulative number of braces
2080 int backwards = FALSE; // init for gcc
2081 int raw_string = FALSE; // search for raw string
2082 int inquote = FALSE; // TRUE when inside quotes
2083 char_u *linep; // pointer to current line
Bram Moolenaar071d4272004-06-13 20:20:40 +00002084 char_u *ptr;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002085 int do_quotes; // check for quotes in current line
2086 int at_start; // do_quotes value at start position
2087 int hash_dir = 0; // Direction searched for # things
2088 int comment_dir = 0; // Direction searched for comments
2089 pos_T match_pos; // Where last slash-star was found
2090 int start_in_quotes; // start position is in quotes
2091 int traveled = 0; // how far we've searched so far
2092 int ignore_cend = FALSE; // ignore comment end
2093 int cpo_match; // vi compatible matching
2094 int cpo_bsl; // don't recognize backslashes
2095 int match_escaped = 0; // search for escaped match
2096 int dir; // Direction to search
2097 int comment_col = MAXCOL; // start of / / comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002098#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002099 int lispcomm = FALSE; // inside of Lisp-style comment
2100 int lisp = curbuf->b_p_lisp; // engage Lisp-specific hacks ;)
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002101#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002102
2103 pos = curwin->w_cursor;
Bram Moolenaarc56c4592013-08-14 17:45:29 +02002104 pos.coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002105 linep = ml_get(pos.lnum);
2106
2107 cpo_match = (vim_strchr(p_cpo, CPO_MATCH) != NULL);
2108 cpo_bsl = (vim_strchr(p_cpo, CPO_MATCHBSL) != NULL);
2109
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002110 // Direction to search when initc is '/', '*' or '#'
Bram Moolenaar071d4272004-06-13 20:20:40 +00002111 if (flags & FM_BACKWARD)
2112 dir = BACKWARD;
2113 else if (flags & FM_FORWARD)
2114 dir = FORWARD;
2115 else
2116 dir = 0;
2117
2118 /*
2119 * if initc given, look in the table for the matching character
2120 * '/' and '*' are special cases: look for start or end of comment.
2121 * When '/' is used, we ignore running backwards into an star-slash, for
2122 * "[*" command, we just want to find any comment.
2123 */
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002124 if (initc == '/' || initc == '*' || initc == 'R')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002125 {
2126 comment_dir = dir;
2127 if (initc == '/')
2128 ignore_cend = TRUE;
2129 backwards = (dir == FORWARD) ? FALSE : TRUE;
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002130 raw_string = (initc == 'R');
Bram Moolenaar071d4272004-06-13 20:20:40 +00002131 initc = NUL;
2132 }
2133 else if (initc != '#' && initc != NUL)
2134 {
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002135 find_mps_values(&initc, &findc, &backwards, TRUE);
Connor Lane Smithb9115da2021-07-31 13:31:42 +02002136 if (dir)
2137 backwards = (dir == FORWARD) ? FALSE : TRUE;
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002138 if (findc == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002139 return NULL;
2140 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002141 else
2142 {
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002143 /*
2144 * Either initc is '#', or no initc was given and we need to look
2145 * under the cursor.
2146 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002147 if (initc == '#')
2148 {
2149 hash_dir = dir;
2150 }
2151 else
2152 {
2153 /*
2154 * initc was not given, must look for something to match under
2155 * or near the cursor.
2156 * Only check for special things when 'cpo' doesn't have '%'.
2157 */
2158 if (!cpo_match)
2159 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002160 // Are we before or at #if, #else etc.?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002161 ptr = skipwhite(linep);
2162 if (*ptr == '#' && pos.col <= (colnr_T)(ptr - linep))
2163 {
2164 ptr = skipwhite(ptr + 1);
2165 if ( STRNCMP(ptr, "if", 2) == 0
2166 || STRNCMP(ptr, "endif", 5) == 0
2167 || STRNCMP(ptr, "el", 2) == 0)
2168 hash_dir = 1;
2169 }
2170
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002171 // Are we on a comment?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002172 else if (linep[pos.col] == '/')
2173 {
2174 if (linep[pos.col + 1] == '*')
2175 {
2176 comment_dir = FORWARD;
2177 backwards = FALSE;
2178 pos.col++;
2179 }
2180 else if (pos.col > 0 && linep[pos.col - 1] == '*')
2181 {
2182 comment_dir = BACKWARD;
2183 backwards = TRUE;
2184 pos.col--;
2185 }
2186 }
2187 else if (linep[pos.col] == '*')
2188 {
2189 if (linep[pos.col + 1] == '/')
2190 {
2191 comment_dir = BACKWARD;
2192 backwards = TRUE;
2193 }
2194 else if (pos.col > 0 && linep[pos.col - 1] == '/')
2195 {
2196 comment_dir = FORWARD;
2197 backwards = FALSE;
2198 }
2199 }
2200 }
2201
2202 /*
2203 * If we are not on a comment or the # at the start of a line, then
2204 * look for brace anywhere on this line after the cursor.
2205 */
2206 if (!hash_dir && !comment_dir)
2207 {
2208 /*
2209 * Find the brace under or after the cursor.
2210 * If beyond the end of the line, use the last character in
2211 * the line.
2212 */
2213 if (linep[pos.col] == NUL && pos.col)
2214 --pos.col;
2215 for (;;)
2216 {
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002217 initc = PTR2CHAR(linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002218 if (initc == NUL)
2219 break;
2220
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002221 find_mps_values(&initc, &findc, &backwards, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002222 if (findc)
2223 break;
Bram Moolenaar1614a142019-10-06 22:00:13 +02002224 pos.col += mb_ptr2len(linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002225 }
2226 if (!findc)
2227 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002228 // no brace in the line, maybe use " #if" then
Bram Moolenaar071d4272004-06-13 20:20:40 +00002229 if (!cpo_match && *skipwhite(linep) == '#')
2230 hash_dir = 1;
2231 else
2232 return NULL;
2233 }
2234 else if (!cpo_bsl)
2235 {
2236 int col, bslcnt = 0;
2237
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002238 // Set "match_escaped" if there are an odd number of
2239 // backslashes.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002240 for (col = pos.col; check_prevcol(linep, col, '\\', &col);)
2241 bslcnt++;
2242 match_escaped = (bslcnt & 1);
2243 }
2244 }
2245 }
2246 if (hash_dir)
2247 {
2248 /*
2249 * Look for matching #if, #else, #elif, or #endif
2250 */
2251 if (oap != NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002252 oap->motion_type = MLINE; // Linewise for this case only
Bram Moolenaar071d4272004-06-13 20:20:40 +00002253 if (initc != '#')
2254 {
2255 ptr = skipwhite(skipwhite(linep) + 1);
2256 if (STRNCMP(ptr, "if", 2) == 0 || STRNCMP(ptr, "el", 2) == 0)
2257 hash_dir = 1;
2258 else if (STRNCMP(ptr, "endif", 5) == 0)
2259 hash_dir = -1;
2260 else
2261 return NULL;
2262 }
2263 pos.col = 0;
2264 while (!got_int)
2265 {
2266 if (hash_dir > 0)
2267 {
2268 if (pos.lnum == curbuf->b_ml.ml_line_count)
2269 break;
2270 }
2271 else if (pos.lnum == 1)
2272 break;
2273 pos.lnum += hash_dir;
2274 linep = ml_get(pos.lnum);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002275 line_breakcheck(); // check for CTRL-C typed
Bram Moolenaar071d4272004-06-13 20:20:40 +00002276 ptr = skipwhite(linep);
2277 if (*ptr != '#')
2278 continue;
2279 pos.col = (colnr_T) (ptr - linep);
2280 ptr = skipwhite(ptr + 1);
2281 if (hash_dir > 0)
2282 {
2283 if (STRNCMP(ptr, "if", 2) == 0)
2284 count++;
2285 else if (STRNCMP(ptr, "el", 2) == 0)
2286 {
2287 if (count == 0)
2288 return &pos;
2289 }
2290 else if (STRNCMP(ptr, "endif", 5) == 0)
2291 {
2292 if (count == 0)
2293 return &pos;
2294 count--;
2295 }
2296 }
2297 else
2298 {
2299 if (STRNCMP(ptr, "if", 2) == 0)
2300 {
2301 if (count == 0)
2302 return &pos;
2303 count--;
2304 }
2305 else if (initc == '#' && STRNCMP(ptr, "el", 2) == 0)
2306 {
2307 if (count == 0)
2308 return &pos;
2309 }
2310 else if (STRNCMP(ptr, "endif", 5) == 0)
2311 count++;
2312 }
2313 }
2314 return NULL;
2315 }
2316 }
2317
2318#ifdef FEAT_RIGHTLEFT
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002319 // This is just guessing: when 'rightleft' is set, search for a matching
2320 // paren/brace in the other direction.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002321 if (curwin->w_p_rl && vim_strchr((char_u *)"()[]{}<>", initc) != NULL)
2322 backwards = !backwards;
2323#endif
2324
2325 do_quotes = -1;
2326 start_in_quotes = MAYBE;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01002327 CLEAR_POS(&match_pos);
Bram Moolenaarfd2ac762006-03-01 22:09:21 +00002328
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002329 // backward search: Check if this line contains a single-line comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002330 if ((backwards && comment_dir)
2331#ifdef FEAT_LISP
2332 || lisp
2333#endif
2334 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002335 comment_col = check_linecomment(linep);
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002336#ifdef FEAT_LISP
2337 if (lisp && comment_col != MAXCOL && pos.col > (colnr_T)comment_col)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002338 lispcomm = TRUE; // find match inside this comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002339#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002340 while (!got_int)
2341 {
2342 /*
2343 * Go to the next position, forward or backward. We could use
2344 * inc() and dec() here, but that is much slower
2345 */
2346 if (backwards)
2347 {
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002348#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002349 // char to match is inside of comment, don't search outside
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002350 if (lispcomm && pos.col < (colnr_T)comment_col)
2351 break;
2352#endif
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002353 if (pos.col == 0) // at start of line, go to prev. one
Bram Moolenaar071d4272004-06-13 20:20:40 +00002354 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002355 if (pos.lnum == 1) // start of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00002356 break;
2357 --pos.lnum;
2358
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002359 if (maxtravel > 0 && ++traveled > maxtravel)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002360 break;
2361
2362 linep = ml_get(pos.lnum);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002363 pos.col = (colnr_T)STRLEN(linep); // pos.col on trailing NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00002364 do_quotes = -1;
2365 line_breakcheck();
2366
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002367 // Check if this line contains a single-line comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002368 if (comment_dir
2369#ifdef FEAT_LISP
2370 || lisp
2371#endif
2372 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002373 comment_col = check_linecomment(linep);
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002374#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002375 // skip comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002376 if (lisp && comment_col != MAXCOL)
2377 pos.col = comment_col;
2378#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002379 }
2380 else
2381 {
2382 --pos.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002383 if (has_mbyte)
2384 pos.col -= (*mb_head_off)(linep, linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002385 }
2386 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002387 else // forward search
Bram Moolenaar071d4272004-06-13 20:20:40 +00002388 {
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002389 if (linep[pos.col] == NUL
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002390 // at end of line, go to next one
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002391#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002392 // don't search for match in comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002393 || (lisp && comment_col != MAXCOL
2394 && pos.col == (colnr_T)comment_col)
2395#endif
2396 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002397 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002398 if (pos.lnum == curbuf->b_ml.ml_line_count // end of file
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002399#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002400 // line is exhausted and comment with it,
2401 // don't search for match in code
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002402 || lispcomm
2403#endif
2404 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002405 break;
2406 ++pos.lnum;
2407
2408 if (maxtravel && traveled++ > maxtravel)
2409 break;
2410
2411 linep = ml_get(pos.lnum);
2412 pos.col = 0;
2413 do_quotes = -1;
2414 line_breakcheck();
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002415#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002416 if (lisp) // find comment pos in new line
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002417 comment_col = check_linecomment(linep);
2418#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002419 }
2420 else
2421 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002422 if (has_mbyte)
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00002423 pos.col += (*mb_ptr2len)(linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002424 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00002425 ++pos.col;
2426 }
2427 }
2428
2429 /*
2430 * If FM_BLOCKSTOP given, stop at a '{' or '}' in column 0.
2431 */
2432 if (pos.col == 0 && (flags & FM_BLOCKSTOP) &&
2433 (linep[0] == '{' || linep[0] == '}'))
2434 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002435 if (linep[0] == findc && count == 0) // match!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002436 return &pos;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002437 break; // out of scope
Bram Moolenaar071d4272004-06-13 20:20:40 +00002438 }
2439
2440 if (comment_dir)
2441 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002442 // Note: comments do not nest, and we ignore quotes in them
2443 // TODO: ignore comment brackets inside strings
Bram Moolenaar071d4272004-06-13 20:20:40 +00002444 if (comment_dir == FORWARD)
2445 {
2446 if (linep[pos.col] == '*' && linep[pos.col + 1] == '/')
2447 {
2448 pos.col++;
2449 return &pos;
2450 }
2451 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002452 else // Searching backwards
Bram Moolenaar071d4272004-06-13 20:20:40 +00002453 {
2454 /*
2455 * A comment may contain / * or / /, it may also start or end
Bram Moolenaarf8c53d32017-11-12 15:36:38 +01002456 * with / * /. Ignore a / * after / / and after *.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002457 */
2458 if (pos.col == 0)
2459 continue;
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002460 else if (raw_string)
2461 {
2462 if (linep[pos.col - 1] == 'R'
2463 && linep[pos.col] == '"'
2464 && vim_strchr(linep + pos.col + 1, '(') != NULL)
2465 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002466 // Possible start of raw string. Now that we have the
2467 // delimiter we can check if it ends before where we
2468 // started searching, or before the previously found
2469 // raw string start.
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002470 if (!find_rawstring_end(linep, &pos,
2471 count > 0 ? &match_pos : &curwin->w_cursor))
2472 {
2473 count++;
2474 match_pos = pos;
2475 match_pos.col--;
2476 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002477 linep = ml_get(pos.lnum); // may have been released
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002478 }
2479 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002480 else if ( linep[pos.col - 1] == '/'
2481 && linep[pos.col] == '*'
Bram Moolenaarf8c53d32017-11-12 15:36:38 +01002482 && (pos.col == 1 || linep[pos.col - 2] != '*')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002483 && (int)pos.col < comment_col)
2484 {
2485 count++;
2486 match_pos = pos;
2487 match_pos.col--;
2488 }
2489 else if (linep[pos.col - 1] == '*' && linep[pos.col] == '/')
2490 {
2491 if (count > 0)
2492 pos = match_pos;
2493 else if (pos.col > 1 && linep[pos.col - 2] == '/'
2494 && (int)pos.col <= comment_col)
2495 pos.col -= 2;
2496 else if (ignore_cend)
2497 continue;
2498 else
2499 return NULL;
2500 return &pos;
2501 }
2502 }
2503 continue;
2504 }
2505
2506 /*
2507 * If smart matching ('cpoptions' does not contain '%'), braces inside
2508 * of quotes are ignored, but only if there is an even number of
2509 * quotes in the line.
2510 */
2511 if (cpo_match)
2512 do_quotes = 0;
2513 else if (do_quotes == -1)
2514 {
2515 /*
2516 * Count the number of quotes in the line, skipping \" and '"'.
2517 * Watch out for "\\".
2518 */
2519 at_start = do_quotes;
2520 for (ptr = linep; *ptr; ++ptr)
2521 {
2522 if (ptr == linep + pos.col + backwards)
2523 at_start = (do_quotes & 1);
2524 if (*ptr == '"'
2525 && (ptr == linep || ptr[-1] != '\'' || ptr[1] != '\''))
2526 ++do_quotes;
2527 if (*ptr == '\\' && ptr[1] != NUL)
2528 ++ptr;
2529 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002530 do_quotes &= 1; // result is 1 with even number of quotes
Bram Moolenaar071d4272004-06-13 20:20:40 +00002531
2532 /*
2533 * If we find an uneven count, check current line and previous
2534 * one for a '\' at the end.
2535 */
2536 if (!do_quotes)
2537 {
2538 inquote = FALSE;
2539 if (ptr[-1] == '\\')
2540 {
2541 do_quotes = 1;
2542 if (start_in_quotes == MAYBE)
2543 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002544 // Do we need to use at_start here?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002545 inquote = TRUE;
2546 start_in_quotes = TRUE;
2547 }
2548 else if (backwards)
2549 inquote = TRUE;
2550 }
2551 if (pos.lnum > 1)
2552 {
2553 ptr = ml_get(pos.lnum - 1);
2554 if (*ptr && *(ptr + STRLEN(ptr) - 1) == '\\')
2555 {
2556 do_quotes = 1;
2557 if (start_in_quotes == MAYBE)
2558 {
2559 inquote = at_start;
2560 if (inquote)
2561 start_in_quotes = TRUE;
2562 }
2563 else if (!backwards)
2564 inquote = TRUE;
2565 }
Bram Moolenaaraec11792007-07-10 11:09:36 +00002566
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002567 // ml_get() only keeps one line, need to get linep again
Bram Moolenaaraec11792007-07-10 11:09:36 +00002568 linep = ml_get(pos.lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002569 }
2570 }
2571 }
2572 if (start_in_quotes == MAYBE)
2573 start_in_quotes = FALSE;
2574
2575 /*
2576 * If 'smartmatch' is set:
2577 * Things inside quotes are ignored by setting 'inquote'. If we
2578 * find a quote without a preceding '\' invert 'inquote'. At the
2579 * end of a line not ending in '\' we reset 'inquote'.
2580 *
2581 * In lines with an uneven number of quotes (without preceding '\')
2582 * we do not know which part to ignore. Therefore we only set
2583 * inquote if the number of quotes in a line is even, unless this
2584 * line or the previous one ends in a '\'. Complicated, isn't it?
2585 */
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002586 c = PTR2CHAR(linep + pos.col);
2587 switch (c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002588 {
2589 case NUL:
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002590 // at end of line without trailing backslash, reset inquote
Bram Moolenaar071d4272004-06-13 20:20:40 +00002591 if (pos.col == 0 || linep[pos.col - 1] != '\\')
2592 {
2593 inquote = FALSE;
2594 start_in_quotes = FALSE;
2595 }
2596 break;
2597
2598 case '"':
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002599 // a quote that is preceded with an odd number of backslashes is
2600 // ignored
Bram Moolenaar071d4272004-06-13 20:20:40 +00002601 if (do_quotes)
2602 {
2603 int col;
2604
2605 for (col = pos.col - 1; col >= 0; --col)
2606 if (linep[col] != '\\')
2607 break;
2608 if ((((int)pos.col - 1 - col) & 1) == 0)
2609 {
2610 inquote = !inquote;
2611 start_in_quotes = FALSE;
2612 }
2613 }
2614 break;
2615
2616 /*
2617 * If smart matching ('cpoptions' does not contain '%'):
2618 * Skip things in single quotes: 'x' or '\x'. Be careful for single
2619 * single quotes, eg jon's. Things like '\233' or '\x3f' are not
2620 * skipped, there is never a brace in them.
2621 * Ignore this when finding matches for `'.
2622 */
2623 case '\'':
2624 if (!cpo_match && initc != '\'' && findc != '\'')
2625 {
2626 if (backwards)
2627 {
2628 if (pos.col > 1)
2629 {
2630 if (linep[pos.col - 2] == '\'')
2631 {
2632 pos.col -= 2;
2633 break;
2634 }
2635 else if (linep[pos.col - 2] == '\\' &&
2636 pos.col > 2 && linep[pos.col - 3] == '\'')
2637 {
2638 pos.col -= 3;
2639 break;
2640 }
2641 }
2642 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002643 else if (linep[pos.col + 1]) // forward search
Bram Moolenaar071d4272004-06-13 20:20:40 +00002644 {
2645 if (linep[pos.col + 1] == '\\' &&
2646 linep[pos.col + 2] && linep[pos.col + 3] == '\'')
2647 {
2648 pos.col += 3;
2649 break;
2650 }
2651 else if (linep[pos.col + 2] == '\'')
2652 {
2653 pos.col += 2;
2654 break;
2655 }
2656 }
2657 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002658 // FALLTHROUGH
Bram Moolenaar071d4272004-06-13 20:20:40 +00002659
2660 default:
2661#ifdef FEAT_LISP
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002662 /*
2663 * For Lisp skip over backslashed (), {} and [].
2664 * (actually, we skip #\( et al)
2665 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002666 if (curbuf->b_p_lisp
2667 && vim_strchr((char_u *)"(){}[]", c) != NULL
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002668 && pos.col > 1
2669 && check_prevcol(linep, pos.col, '\\', NULL)
2670 && check_prevcol(linep, pos.col - 1, '#', NULL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002671 break;
2672#endif
2673
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002674 // Check for match outside of quotes, and inside of
2675 // quotes when the start is also inside of quotes.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002676 if ((!inquote || start_in_quotes == TRUE)
2677 && (c == initc || c == findc))
2678 {
2679 int col, bslcnt = 0;
2680
2681 if (!cpo_bsl)
2682 {
2683 for (col = pos.col; check_prevcol(linep, col, '\\', &col);)
2684 bslcnt++;
2685 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002686 // Only accept a match when 'M' is in 'cpo' or when escaping
2687 // is what we expect.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002688 if (cpo_bsl || (bslcnt & 1) == match_escaped)
2689 {
2690 if (c == initc)
2691 count++;
2692 else
2693 {
2694 if (count == 0)
2695 return &pos;
2696 count--;
2697 }
2698 }
2699 }
2700 }
2701 }
2702
2703 if (comment_dir == BACKWARD && count > 0)
2704 {
2705 pos = match_pos;
2706 return &pos;
2707 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002708 return (pos_T *)NULL; // never found it
Bram Moolenaar071d4272004-06-13 20:20:40 +00002709}
2710
2711/*
2712 * Check if line[] contains a / / comment.
2713 * Return MAXCOL if not, otherwise return the column.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002714 */
Bram Moolenaar6e371ec2021-12-12 14:16:39 +00002715 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002716check_linecomment(char_u *line)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002717{
2718 char_u *p;
2719
2720 p = line;
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002721#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002722 // skip Lispish one-line comments
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002723 if (curbuf->b_p_lisp)
2724 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002725 if (vim_strchr(p, ';') != NULL) // there may be comments
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002726 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002727 int in_str = FALSE; // inside of string
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002728
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002729 p = line; // scan from start
Bram Moolenaar520470a2005-06-16 21:59:56 +00002730 while ((p = vim_strpbrk(p, (char_u *)"\";")) != NULL)
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002731 {
2732 if (*p == '"')
2733 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002734 if (in_str)
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002735 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002736 if (*(p - 1) != '\\') // skip escaped quote
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002737 in_str = FALSE;
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002738 }
2739 else if (p == line || ((p - line) >= 2
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002740 // skip #\" form
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002741 && *(p - 1) != '\\' && *(p - 2) != '#'))
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002742 in_str = TRUE;
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002743 }
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002744 else if (!in_str && ((p - line) < 2
Bram Moolenaarba263672021-12-29 18:09:13 +00002745 || (*(p - 1) != '\\' && *(p - 2) != '#'))
2746 && !is_pos_in_string(line, (colnr_T)(p - line)))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002747 break; // found!
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002748 ++p;
2749 }
2750 }
2751 else
2752 p = NULL;
2753 }
2754 else
2755#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002756 while ((p = vim_strchr(p, '/')) != NULL)
2757 {
Bram Moolenaarba263672021-12-29 18:09:13 +00002758 // Accept a double /, unless it's preceded with * and followed by *,
2759 // because * / / * is an end and start of a C comment.
2760 // Only accept the position if it is not inside a string.
2761 if (p[1] == '/' && (p == line || p[-1] != '*' || p[2] != '*')
2762 && !is_pos_in_string(line, (colnr_T)(p - line)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002763 break;
2764 ++p;
2765 }
2766
2767 if (p == NULL)
2768 return MAXCOL;
2769 return (int)(p - line);
2770}
2771
2772/*
2773 * Move cursor briefly to character matching the one under the cursor.
2774 * Used for Insert mode and "r" command.
2775 * Show the match only if it is visible on the screen.
2776 * If there isn't a match, then beep.
2777 */
2778 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002779showmatch(
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002780 int c) // char to show match for
Bram Moolenaar071d4272004-06-13 20:20:40 +00002781{
2782 pos_T *lpos, save_cursor;
2783 pos_T mpos;
2784 colnr_T vcol;
2785 long save_so;
2786 long save_siso;
2787#ifdef CURSOR_SHAPE
2788 int save_state;
2789#endif
2790 colnr_T save_dollar_vcol;
2791 char_u *p;
Bram Moolenaar375e3392019-01-31 18:26:10 +01002792 long *so = curwin->w_p_so >= 0 ? &curwin->w_p_so : &p_so;
2793 long *siso = curwin->w_p_siso >= 0 ? &curwin->w_p_siso : &p_siso;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002794
2795 /*
2796 * Only show match for chars in the 'matchpairs' option.
2797 */
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002798 // 'matchpairs' is "x:y,x:y"
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002799 for (p = curbuf->b_p_mps; *p != NUL; ++p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002800 {
2801#ifdef FEAT_RIGHTLEFT
Bram Moolenaar187d3ac2013-02-20 18:39:13 +01002802 if (PTR2CHAR(p) == c && (curwin->w_p_rl ^ p_ri))
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002803 break;
Bram Moolenaar187d3ac2013-02-20 18:39:13 +01002804#endif
Bram Moolenaar1614a142019-10-06 22:00:13 +02002805 p += mb_ptr2len(p) + 1;
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002806 if (PTR2CHAR(p) == c
Bram Moolenaar071d4272004-06-13 20:20:40 +00002807#ifdef FEAT_RIGHTLEFT
2808 && !(curwin->w_p_rl ^ p_ri)
2809#endif
2810 )
2811 break;
Bram Moolenaar1614a142019-10-06 22:00:13 +02002812 p += mb_ptr2len(p);
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002813 if (*p == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002814 return;
2815 }
Bram Moolenaar5b8cabf2021-04-02 18:55:57 +02002816 if (*p == NUL)
2817 return;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002818
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002819 if ((lpos = findmatch(NULL, NUL)) == NULL) // no match, so beep
Bram Moolenaar165bc692015-07-21 17:53:25 +02002820 vim_beep(BO_MATCH);
Bram Moolenaar187d3ac2013-02-20 18:39:13 +01002821 else if (lpos->lnum >= curwin->w_topline && lpos->lnum < curwin->w_botline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002822 {
2823 if (!curwin->w_p_wrap)
2824 getvcol(curwin, lpos, NULL, &vcol, NULL);
2825 if (curwin->w_p_wrap || (vcol >= curwin->w_leftcol
Bram Moolenaar02631462017-09-22 15:20:32 +02002826 && vcol < curwin->w_leftcol + curwin->w_width))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002827 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002828 mpos = *lpos; // save the pos, update_screen() may change it
Bram Moolenaar071d4272004-06-13 20:20:40 +00002829 save_cursor = curwin->w_cursor;
Bram Moolenaar375e3392019-01-31 18:26:10 +01002830 save_so = *so;
2831 save_siso = *siso;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002832 // Handle "$" in 'cpo': If the ')' is typed on top of the "$",
2833 // stop displaying the "$".
Bram Moolenaar76b9b362012-02-04 23:35:00 +01002834 if (dollar_vcol >= 0 && dollar_vcol == curwin->w_virtcol)
2835 dollar_vcol = -1;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002836 ++curwin->w_virtcol; // do display ')' just before "$"
2837 update_screen(VALID); // show the new char first
Bram Moolenaar071d4272004-06-13 20:20:40 +00002838
2839 save_dollar_vcol = dollar_vcol;
2840#ifdef CURSOR_SHAPE
2841 save_state = State;
2842 State = SHOWMATCH;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002843 ui_cursor_shape(); // may show different cursor shape
Bram Moolenaar071d4272004-06-13 20:20:40 +00002844#endif
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002845 curwin->w_cursor = mpos; // move to matching char
2846 *so = 0; // don't use 'scrolloff' here
2847 *siso = 0; // don't use 'sidescrolloff' here
Bram Moolenaar071d4272004-06-13 20:20:40 +00002848 showruler(FALSE);
2849 setcursor();
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002850 cursor_on(); // make sure that the cursor is shown
Bram Moolenaara338adc2018-01-31 20:51:47 +01002851 out_flush_cursor(TRUE, FALSE);
2852
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002853 // Restore dollar_vcol(), because setcursor() may call curs_rows()
2854 // which resets it if the matching position is in a previous line
2855 // and has a higher column number.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002856 dollar_vcol = save_dollar_vcol;
2857
2858 /*
2859 * brief pause, unless 'm' is present in 'cpo' and a character is
2860 * available.
2861 */
2862 if (vim_strchr(p_cpo, CPO_SHOWMATCH) != NULL)
Bram Moolenaareda1da02019-11-17 17:06:33 +01002863 ui_delay(p_mat * 100L + 8, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002864 else if (!char_avail())
Bram Moolenaareda1da02019-11-17 17:06:33 +01002865 ui_delay(p_mat * 100L + 9, FALSE);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002866 curwin->w_cursor = save_cursor; // restore cursor position
Bram Moolenaar375e3392019-01-31 18:26:10 +01002867 *so = save_so;
2868 *siso = save_siso;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002869#ifdef CURSOR_SHAPE
2870 State = save_state;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002871 ui_cursor_shape(); // may show different cursor shape
Bram Moolenaar071d4272004-06-13 20:20:40 +00002872#endif
2873 }
2874 }
2875}
2876
2877/*
Bram Moolenaar453c1922019-10-26 14:42:09 +02002878 * Check if the pattern is zero-width.
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002879 * If move is TRUE, check from the beginning of the buffer, else from position
2880 * "cur".
2881 * "direction" is FORWARD or BACKWARD.
2882 * Returns TRUE, FALSE or -1 for failure.
2883 */
2884 static int
2885is_zero_width(char_u *pattern, int move, pos_T *cur, int direction)
2886{
2887 regmmatch_T regmatch;
2888 int nmatched = 0;
2889 int result = -1;
2890 pos_T pos;
Bram Moolenaar53989552019-12-23 22:59:18 +01002891 int called_emsg_before = called_emsg;
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002892 int flag = 0;
2893
2894 if (pattern == NULL)
2895 pattern = spats[last_idx].pat;
2896
2897 if (search_regcomp(pattern, RE_SEARCH, RE_SEARCH,
2898 SEARCH_KEEP, &regmatch) == FAIL)
2899 return -1;
2900
2901 // init startcol correctly
2902 regmatch.startpos[0].col = -1;
2903 // move to match
2904 if (move)
2905 {
2906 CLEAR_POS(&pos);
2907 }
2908 else
2909 {
2910 pos = *cur;
2911 // accept a match at the cursor position
2912 flag = SEARCH_START;
2913 }
2914
2915 if (searchit(curwin, curbuf, &pos, NULL, direction, pattern, 1,
2916 SEARCH_KEEP + flag, RE_SEARCH, NULL) != FAIL)
2917 {
2918 // Zero-width pattern should match somewhere, then we can check if
2919 // start and end are in the same position.
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002920 do
2921 {
2922 regmatch.startpos[0].col++;
2923 nmatched = vim_regexec_multi(&regmatch, curwin, curbuf,
2924 pos.lnum, regmatch.startpos[0].col, NULL, NULL);
2925 if (nmatched != 0)
2926 break;
Bram Moolenaar795aaa12020-10-02 20:36:01 +02002927 } while (regmatch.regprog != NULL
2928 && direction == FORWARD ? regmatch.startpos[0].col < pos.col
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002929 : regmatch.startpos[0].col > pos.col);
2930
Bram Moolenaar53989552019-12-23 22:59:18 +01002931 if (called_emsg == called_emsg_before)
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002932 {
2933 result = (nmatched != 0
2934 && regmatch.startpos[0].lnum == regmatch.endpos[0].lnum
2935 && regmatch.startpos[0].col == regmatch.endpos[0].col);
2936 }
2937 }
2938
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002939 vim_regfree(regmatch.regprog);
2940 return result;
2941}
2942
Bram Moolenaardde0efe2012-08-23 15:53:05 +02002943
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002944/*
2945 * Find next search match under cursor, cursor at end.
2946 * Used while an operator is pending, and in Visual mode.
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002947 */
2948 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002949current_search(
2950 long count,
Bram Moolenaar5d24a222018-12-23 19:10:09 +01002951 int forward) // TRUE for forward, FALSE for backward
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002952{
Bram Moolenaar5d24a222018-12-23 19:10:09 +01002953 pos_T start_pos; // start position of the pattern match
2954 pos_T end_pos; // end position of the pattern match
2955 pos_T orig_pos; // position of the cursor at beginning
2956 pos_T pos; // position after the pattern
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002957 int i;
2958 int dir;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01002959 int result; // result of various function calls
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002960 char_u old_p_ws = p_ws;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002961 int flags = 0;
Bram Moolenaarde9149e2013-07-17 19:22:13 +02002962 pos_T save_VIsual = VIsual;
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002963 int zero_width;
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02002964 int skip_first_backward;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002965
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002966 // Correct cursor when 'selection' is exclusive
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01002967 if (VIsual_active && *p_sel == 'e' && LT_POS(VIsual, curwin->w_cursor))
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002968 dec_cursor();
2969
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02002970 // When searching forward and the cursor is at the start of the Visual
2971 // area, skip the first search backward, otherwise it doesn't move.
2972 skip_first_backward = forward && VIsual_active
2973 && LT_POS(curwin->w_cursor, VIsual);
2974
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002975 orig_pos = pos = curwin->w_cursor;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002976 if (VIsual_active)
2977 {
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002978 if (forward)
2979 incl(&pos);
2980 else
2981 decl(&pos);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002982 }
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002983
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002984 // Is the pattern is zero-width?, this time, don't care about the direction
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002985 zero_width = is_zero_width(spats[last_idx].pat, TRUE, &curwin->w_cursor,
Bram Moolenaar22ab5472017-09-26 12:28:45 +02002986 FORWARD);
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02002987 if (zero_width == -1)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002988 return FAIL; // pattern not found
Bram Moolenaarba6ba362012-08-08 15:27:57 +02002989
Bram Moolenaarba6ba362012-08-08 15:27:57 +02002990 /*
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002991 * The trick is to first search backwards and then search forward again,
2992 * so that a match at the current cursor position will be correctly
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02002993 * captured. When "forward" is false do it the other way around.
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002994 */
2995 for (i = 0; i < 2; i++)
2996 {
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02002997 if (forward)
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02002998 {
2999 if (i == 0 && skip_first_backward)
3000 continue;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003001 dir = i;
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02003002 }
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003003 else
3004 dir = !i;
Bram Moolenaarba6ba362012-08-08 15:27:57 +02003005
3006 flags = 0;
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02003007 if (!dir && !zero_width)
Bram Moolenaarba6ba362012-08-08 15:27:57 +02003008 flags = SEARCH_END;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01003009 end_pos = pos;
Bram Moolenaarba6ba362012-08-08 15:27:57 +02003010
Bram Moolenaar82cf7f62019-11-02 23:22:47 +01003011 // wrapping should not occur in the first round
3012 if (i == 0)
3013 p_ws = FALSE;
3014
Bram Moolenaar5d24a222018-12-23 19:10:09 +01003015 result = searchit(curwin, curbuf, &pos, &end_pos,
3016 (dir ? FORWARD : BACKWARD),
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003017 spats[last_idx].pat, (long) (i ? count : 1),
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02003018 SEARCH_KEEP | flags, RE_SEARCH, NULL);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003019
Bram Moolenaar82cf7f62019-11-02 23:22:47 +01003020 p_ws = old_p_ws;
3021
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003022 // First search may fail, but then start searching from the
3023 // beginning of the file (cursor might be on the search match)
3024 // except when Visual mode is active, so that extending the visual
3025 // selection works.
3026 if (i == 1 && !result) // not found, abort
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003027 {
3028 curwin->w_cursor = orig_pos;
3029 if (VIsual_active)
3030 VIsual = save_VIsual;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003031 return FAIL;
3032 }
Bram Moolenaar5d24a222018-12-23 19:10:09 +01003033 else if (i == 0 && !result)
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003034 {
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01003035 if (forward)
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003036 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003037 // try again from start of buffer
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01003038 CLEAR_POS(&pos);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003039 }
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01003040 else
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003041 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003042 // try again from end of buffer
3043 // searching backwards, so set pos to last line and col
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003044 pos.lnum = curwin->w_buffer->b_ml.ml_line_count;
Bram Moolenaar09168a72012-08-02 21:24:42 +02003045 pos.col = (colnr_T)STRLEN(
3046 ml_get(curwin->w_buffer->b_ml.ml_line_count));
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003047 }
3048 }
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003049 }
3050
3051 start_pos = pos;
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003052
3053 if (!VIsual_active)
3054 VIsual = start_pos;
3055
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02003056 // put the cursor after the match
Bram Moolenaar5d24a222018-12-23 19:10:09 +01003057 curwin->w_cursor = end_pos;
Bram Moolenaar453c1922019-10-26 14:42:09 +02003058 if (LT_POS(VIsual, end_pos) && forward)
Bram Moolenaarc07b7f72020-10-11 20:44:15 +02003059 {
3060 if (skip_first_backward)
3061 // put the cursor on the start of the match
3062 curwin->w_cursor = pos;
3063 else
3064 // put the cursor on last character of match
3065 dec_cursor();
3066 }
Bram Moolenaar28f224b2020-10-10 16:45:25 +02003067 else if (VIsual_active && LT_POS(curwin->w_cursor, VIsual) && forward)
Bram Moolenaaredaad6e2019-10-24 15:23:37 +02003068 curwin->w_cursor = pos; // put the cursor on the start of the match
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003069 VIsual_active = TRUE;
3070 VIsual_mode = 'v';
3071
Bram Moolenaarb7633612019-02-10 21:48:25 +01003072 if (*p_sel == 'e')
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003073 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003074 // Correction for exclusive selection depends on the direction.
Bram Moolenaarb7633612019-02-10 21:48:25 +01003075 if (forward && LTOREQ_POS(VIsual, curwin->w_cursor))
3076 inc_cursor();
3077 else if (!forward && LTOREQ_POS(curwin->w_cursor, VIsual))
3078 inc(&VIsual);
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003079 }
3080
3081#ifdef FEAT_FOLDING
3082 if (fdo_flags & FDO_SEARCH && KeyTyped)
3083 foldOpenCursor();
3084#endif
3085
3086 may_start_select('c');
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003087 setmouse();
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003088#ifdef FEAT_CLIPBOARD
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003089 // Make sure the clipboard gets updated. Needed because start and
3090 // end are still the same, and the selection needs to be owned
Bram Moolenaar8a0f3c72012-07-29 12:55:32 +02003091 clip_star.vmode = NUL;
3092#endif
3093 redraw_curbuf_later(INVERTED);
3094 showmode();
3095
3096 return OK;
3097}
Bram Moolenaardde0efe2012-08-23 15:53:05 +02003098
Bram Moolenaar071d4272004-06-13 20:20:40 +00003099#if defined(FEAT_LISP) || defined(FEAT_CINDENT) || defined(FEAT_TEXTOBJ) \
3100 || defined(PROTO)
3101/*
3102 * return TRUE if line 'lnum' is empty or has white chars only.
3103 */
3104 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01003105linewhite(linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003106{
3107 char_u *p;
3108
3109 p = skipwhite(ml_get(lnum));
3110 return (*p == NUL);
3111}
3112#endif
3113
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003114/*
3115 * Add the search count "[3/19]" to "msgbuf".
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003116 * See update_search_stat() for other arguments.
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003117 */
3118 static void
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003119cmdline_search_stat(
3120 int dirc,
3121 pos_T *pos,
3122 pos_T *cursor_pos,
3123 int show_top_bot_msg,
3124 char_u *msgbuf,
3125 int recompute,
3126 int maxcount,
3127 long timeout)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003128{
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003129 searchstat_T stat;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003130
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003131 update_search_stat(dirc, pos, cursor_pos, &stat, recompute, maxcount,
3132 timeout);
3133 if (stat.cur > 0)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003134 {
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003135 char t[SEARCH_STAT_BUF_LEN];
Bram Moolenaare2ad8262019-05-24 13:22:22 +02003136 size_t len;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003137
3138#ifdef FEAT_RIGHTLEFT
3139 if (curwin->w_p_rl && *curwin->w_p_rlc == 's')
3140 {
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003141 if (stat.incomplete == 1)
Bram Moolenaarb6cb26f2019-05-07 21:34:37 +02003142 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[?/??]");
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003143 else if (stat.cnt > maxcount && stat.cur > maxcount)
3144 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[>%d/>%d]",
3145 maxcount, maxcount);
3146 else if (stat.cnt > maxcount)
3147 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[>%d/%d]",
3148 maxcount, stat.cur);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003149 else
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003150 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[%d/%d]",
3151 stat.cnt, stat.cur);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003152 }
3153 else
3154#endif
3155 {
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003156 if (stat.incomplete == 1)
Bram Moolenaarb6cb26f2019-05-07 21:34:37 +02003157 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[?/??]");
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003158 else if (stat.cnt > maxcount && stat.cur > maxcount)
3159 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[>%d/>%d]",
3160 maxcount, maxcount);
3161 else if (stat.cnt > maxcount)
3162 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[%d/>%d]",
3163 stat.cur, maxcount);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003164 else
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003165 vim_snprintf(t, SEARCH_STAT_BUF_LEN, "[%d/%d]",
3166 stat.cur, stat.cnt);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003167 }
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02003168
3169 len = STRLEN(t);
Bram Moolenaardc6855a2019-05-18 19:26:29 +02003170 if (show_top_bot_msg && len + 2 < SEARCH_STAT_BUF_LEN)
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02003171 {
Bram Moolenaar16b58ae2019-09-06 20:40:21 +02003172 mch_memmove(t + 2, t, len);
3173 t[0] = 'W';
3174 t[1] = ' ';
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02003175 len += 2;
3176 }
3177
3178 mch_memmove(msgbuf + STRLEN(msgbuf) - len, t, len);
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003179 if (dirc == '?' && stat.cur == maxcount + 1)
3180 stat.cur = -1;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003181
Bram Moolenaar984f0312019-05-24 13:11:47 +02003182 // keep the message even after redraw, but don't put in history
3183 msg_hist_off = TRUE;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003184 give_warning(msgbuf, FALSE);
Bram Moolenaar984f0312019-05-24 13:11:47 +02003185 msg_hist_off = FALSE;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003186 }
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003187}
3188
3189/*
3190 * Add the search count information to "stat".
3191 * "stat" must not be NULL.
3192 * When "recompute" is TRUE always recompute the numbers.
3193 * dirc == 0: don't find the next/previous match (only set the result to "stat")
3194 * dirc == '/': find the next match
3195 * dirc == '?': find the previous match
3196 */
3197 static void
3198update_search_stat(
3199 int dirc,
3200 pos_T *pos,
3201 pos_T *cursor_pos,
3202 searchstat_T *stat,
3203 int recompute,
3204 int maxcount,
Bram Moolenaarf9ca08e2020-06-01 18:56:03 +02003205 long timeout UNUSED)
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003206{
3207 int save_ws = p_ws;
3208 int wraparound = FALSE;
3209 pos_T p = (*pos);
Bram Moolenaar14681622020-06-03 22:57:39 +02003210 static pos_T lastpos = {0, 0, 0};
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003211 static int cur = 0;
3212 static int cnt = 0;
3213 static int exact_match = FALSE;
3214 static int incomplete = 0;
3215 static int last_maxcount = SEARCH_STAT_DEF_MAX_COUNT;
3216 static int chgtick = 0;
3217 static char_u *lastpat = NULL;
3218 static buf_T *lbuf = NULL;
3219#ifdef FEAT_RELTIME
3220 proftime_T start;
3221#endif
3222
3223 vim_memset(stat, 0, sizeof(searchstat_T));
3224
3225 if (dirc == 0 && !recompute && !EMPTY_POS(lastpos))
3226 {
3227 stat->cur = cur;
3228 stat->cnt = cnt;
3229 stat->exact_match = exact_match;
3230 stat->incomplete = incomplete;
3231 stat->last_maxcount = last_maxcount;
3232 return;
3233 }
3234 last_maxcount = maxcount;
3235
3236 wraparound = ((dirc == '?' && LT_POS(lastpos, p))
3237 || (dirc == '/' && LT_POS(p, lastpos)));
3238
3239 // If anything relevant changed the count has to be recomputed.
3240 // MB_STRNICMP ignores case, but we should not ignore case.
3241 // Unfortunately, there is no MB_STRNICMP function.
3242 // XXX: above comment should be "no MB_STRCMP function" ?
3243 if (!(chgtick == CHANGEDTICK(curbuf)
3244 && MB_STRNICMP(lastpat, spats[last_idx].pat, STRLEN(lastpat)) == 0
3245 && STRLEN(lastpat) == STRLEN(spats[last_idx].pat)
3246 && EQUAL_POS(lastpos, *cursor_pos)
3247 && lbuf == curbuf) || wraparound || cur < 0
3248 || (maxcount > 0 && cur > maxcount) || recompute)
3249 {
3250 cur = 0;
3251 cnt = 0;
3252 exact_match = FALSE;
3253 incomplete = 0;
3254 CLEAR_POS(&lastpos);
3255 lbuf = curbuf;
3256 }
3257
3258 if (EQUAL_POS(lastpos, *cursor_pos) && !wraparound
3259 && (dirc == 0 || dirc == '/' ? cur < cnt : cur > 0))
3260 cur += dirc == 0 ? 0 : dirc == '/' ? 1 : -1;
3261 else
3262 {
3263 int done_search = FALSE;
3264 pos_T endpos = {0, 0, 0};
3265
3266 p_ws = FALSE;
3267#ifdef FEAT_RELTIME
3268 if (timeout > 0)
3269 profile_setlimit(timeout, &start);
3270#endif
3271 while (!got_int && searchit(curwin, curbuf, &lastpos, &endpos,
3272 FORWARD, NULL, 1, SEARCH_KEEP, RE_LAST, NULL) != FAIL)
3273 {
3274 done_search = TRUE;
3275#ifdef FEAT_RELTIME
3276 // Stop after passing the time limit.
3277 if (timeout > 0 && profile_passed_limit(&start))
3278 {
3279 incomplete = 1;
3280 break;
3281 }
3282#endif
3283 cnt++;
3284 if (LTOREQ_POS(lastpos, p))
3285 {
3286 cur = cnt;
Bram Moolenaar57f75a52020-06-02 22:06:21 +02003287 if (LT_POS(p, endpos))
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02003288 exact_match = TRUE;
3289 }
3290 fast_breakcheck();
3291 if (maxcount > 0 && cnt > maxcount)
3292 {
3293 incomplete = 2; // max count exceeded
3294 break;
3295 }
3296 }
3297 if (got_int)
3298 cur = -1; // abort
3299 if (done_search)
3300 {
3301 vim_free(lastpat);
3302 lastpat = vim_strsave(spats[last_idx].pat);
3303 chgtick = CHANGEDTICK(curbuf);
3304 lbuf = curbuf;
3305 lastpos = p;
3306 }
3307 }
3308 stat->cur = cur;
3309 stat->cnt = cnt;
3310 stat->exact_match = exact_match;
3311 stat->incomplete = incomplete;
3312 stat->last_maxcount = last_maxcount;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02003313 p_ws = save_ws;
3314}
3315
Bram Moolenaar071d4272004-06-13 20:20:40 +00003316#if defined(FEAT_FIND_ID) || defined(PROTO)
3317/*
3318 * Find identifiers or defines in included files.
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003319 * If p_ic && compl_status_sol() then ptr must be in lowercase.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003320 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003321 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01003322find_pattern_in_path(
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003323 char_u *ptr, // pointer to search pattern
3324 int dir UNUSED, // direction of expansion
3325 int len, // length of search pattern
3326 int whole, // match whole words only
3327 int skip_comments, // don't match inside comments
3328 int type, // Type of search; are we looking for a type?
3329 // a macro?
Bram Moolenaar764b23c2016-01-30 21:10:09 +01003330 long count,
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003331 int action, // What to do when we find it
3332 linenr_T start_lnum, // first line to start searching
3333 linenr_T end_lnum) // last line for searching
Bram Moolenaar071d4272004-06-13 20:20:40 +00003334{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003335 SearchedFile *files; // Stack of included files
3336 SearchedFile *bigger; // When we need more space
Bram Moolenaar071d4272004-06-13 20:20:40 +00003337 int max_path_depth = 50;
3338 long match_count = 1;
3339
3340 char_u *pat;
3341 char_u *new_fname;
3342 char_u *curr_fname = curbuf->b_fname;
3343 char_u *prev_fname = NULL;
3344 linenr_T lnum;
3345 int depth;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003346 int depth_displayed; // For type==CHECK_PATH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003347 int old_files;
3348 int already_searched;
3349 char_u *file_line;
3350 char_u *line;
3351 char_u *p;
3352 char_u save_char;
3353 int define_matched;
3354 regmatch_T regmatch;
3355 regmatch_T incl_regmatch;
3356 regmatch_T def_regmatch;
3357 int matched = FALSE;
3358 int did_show = FALSE;
3359 int found = FALSE;
3360 int i;
3361 char_u *already = NULL;
3362 char_u *startp = NULL;
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003363 char_u *inc_opt = NULL;
Bram Moolenaar4033c552017-09-16 20:54:51 +02003364#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003365 win_T *curwin_save = NULL;
3366#endif
3367
3368 regmatch.regprog = NULL;
3369 incl_regmatch.regprog = NULL;
3370 def_regmatch.regprog = NULL;
3371
3372 file_line = alloc(LSIZE);
3373 if (file_line == NULL)
3374 return;
3375
Bram Moolenaar071d4272004-06-13 20:20:40 +00003376 if (type != CHECK_PATH && type != FIND_DEFINE
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003377 // when CONT_SOL is set compare "ptr" with the beginning of the
3378 // line is faster than quote_meta/regcomp/regexec "ptr" -- Acevedo
3379 && !compl_status_sol())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003380 {
3381 pat = alloc(len + 5);
3382 if (pat == NULL)
3383 goto fpip_end;
3384 sprintf((char *)pat, whole ? "\\<%.*s\\>" : "%.*s", len, ptr);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003385 // ignore case according to p_ic, p_scs and pat
Bram Moolenaar071d4272004-06-13 20:20:40 +00003386 regmatch.rm_ic = ignorecase(pat);
Bram Moolenaarf4e20992020-12-21 19:59:08 +01003387 regmatch.regprog = vim_regcomp(pat, magic_isset() ? RE_MAGIC : 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003388 vim_free(pat);
3389 if (regmatch.regprog == NULL)
3390 goto fpip_end;
3391 }
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003392 inc_opt = (*curbuf->b_p_inc == NUL) ? p_inc : curbuf->b_p_inc;
3393 if (*inc_opt != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003394 {
Bram Moolenaarf4e20992020-12-21 19:59:08 +01003395 incl_regmatch.regprog = vim_regcomp(inc_opt,
3396 magic_isset() ? RE_MAGIC : 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003397 if (incl_regmatch.regprog == NULL)
3398 goto fpip_end;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003399 incl_regmatch.rm_ic = FALSE; // don't ignore case in incl. pat.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003400 }
3401 if (type == FIND_DEFINE && (*curbuf->b_p_def != NUL || *p_def != NUL))
3402 {
3403 def_regmatch.regprog = vim_regcomp(*curbuf->b_p_def == NUL
Bram Moolenaarf4e20992020-12-21 19:59:08 +01003404 ? p_def : curbuf->b_p_def,
3405 magic_isset() ? RE_MAGIC : 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003406 if (def_regmatch.regprog == NULL)
3407 goto fpip_end;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003408 def_regmatch.rm_ic = FALSE; // don't ignore case in define pat.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003409 }
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003410 files = lalloc_clear(max_path_depth * sizeof(SearchedFile), TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003411 if (files == NULL)
3412 goto fpip_end;
3413 old_files = max_path_depth;
3414 depth = depth_displayed = -1;
3415
3416 lnum = start_lnum;
3417 if (end_lnum > curbuf->b_ml.ml_line_count)
3418 end_lnum = curbuf->b_ml.ml_line_count;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003419 if (lnum > end_lnum) // do at least one line
Bram Moolenaar071d4272004-06-13 20:20:40 +00003420 lnum = end_lnum;
3421 line = ml_get(lnum);
3422
3423 for (;;)
3424 {
3425 if (incl_regmatch.regprog != NULL
3426 && vim_regexec(&incl_regmatch, line, (colnr_T)0))
3427 {
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003428 char_u *p_fname = (curr_fname == curbuf->b_fname)
3429 ? curbuf->b_ffname : curr_fname;
3430
3431 if (inc_opt != NULL && strstr((char *)inc_opt, "\\zs") != NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003432 // Use text from '\zs' to '\ze' (or end) of 'include'.
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003433 new_fname = find_file_name_in_path(incl_regmatch.startp[0],
Bram Moolenaarc84e3c12013-07-03 22:28:36 +02003434 (int)(incl_regmatch.endp[0] - incl_regmatch.startp[0]),
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003435 FNAME_EXP|FNAME_INCL|FNAME_REL, 1L, p_fname);
3436 else
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003437 // Use text after match with 'include'.
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003438 new_fname = file_name_in_line(incl_regmatch.endp[0], 0,
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003439 FNAME_EXP|FNAME_INCL|FNAME_REL, 1L, p_fname, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003440 already_searched = FALSE;
3441 if (new_fname != NULL)
3442 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003443 // Check whether we have already searched in this file
Bram Moolenaar071d4272004-06-13 20:20:40 +00003444 for (i = 0;; i++)
3445 {
3446 if (i == depth + 1)
3447 i = old_files;
3448 if (i == max_path_depth)
3449 break;
Bram Moolenaar99499b12019-05-23 21:35:48 +02003450 if (fullpathcmp(new_fname, files[i].name, TRUE, TRUE)
3451 & FPC_SAME)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003452 {
3453 if (type != CHECK_PATH &&
3454 action == ACTION_SHOW_ALL && files[i].matched)
3455 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003456 msg_putchar('\n'); // cursor below last one
3457 if (!got_int) // don't display if 'q'
3458 // typed at "--more--"
3459 // message
Bram Moolenaar071d4272004-06-13 20:20:40 +00003460 {
3461 msg_home_replace_hl(new_fname);
Bram Moolenaar32526b32019-01-19 17:43:09 +01003462 msg_puts(_(" (includes previously listed match)"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003463 prev_fname = NULL;
3464 }
3465 }
Bram Moolenaard23a8232018-02-10 18:45:26 +01003466 VIM_CLEAR(new_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003467 already_searched = TRUE;
3468 break;
3469 }
3470 }
3471 }
3472
3473 if (type == CHECK_PATH && (action == ACTION_SHOW_ALL
3474 || (new_fname == NULL && !already_searched)))
3475 {
3476 if (did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003477 msg_putchar('\n'); // cursor below last one
Bram Moolenaar071d4272004-06-13 20:20:40 +00003478 else
3479 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003480 gotocmdline(TRUE); // cursor at status line
Bram Moolenaar32526b32019-01-19 17:43:09 +01003481 msg_puts_title(_("--- Included files "));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003482 if (action != ACTION_SHOW_ALL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003483 msg_puts_title(_("not found "));
3484 msg_puts_title(_("in path ---\n"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003485 }
3486 did_show = TRUE;
3487 while (depth_displayed < depth && !got_int)
3488 {
3489 ++depth_displayed;
3490 for (i = 0; i < depth_displayed; i++)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003491 msg_puts(" ");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003492 msg_home_replace(files[depth_displayed].name);
Bram Moolenaar32526b32019-01-19 17:43:09 +01003493 msg_puts(" -->\n");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003494 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003495 if (!got_int) // don't display if 'q' typed
3496 // for "--more--" message
Bram Moolenaar071d4272004-06-13 20:20:40 +00003497 {
3498 for (i = 0; i <= depth_displayed; i++)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003499 msg_puts(" ");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003500 if (new_fname != NULL)
3501 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003502 // using "new_fname" is more reliable, e.g., when
3503 // 'includeexpr' is set.
Bram Moolenaar8820b482017-03-16 17:23:31 +01003504 msg_outtrans_attr(new_fname, HL_ATTR(HLF_D));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003505 }
3506 else
3507 {
3508 /*
3509 * Isolate the file name.
3510 * Include the surrounding "" or <> if present.
3511 */
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003512 if (inc_opt != NULL
3513 && strstr((char *)inc_opt, "\\zs") != NULL)
3514 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003515 // pattern contains \zs, use the match
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003516 p = incl_regmatch.startp[0];
3517 i = (int)(incl_regmatch.endp[0]
3518 - incl_regmatch.startp[0]);
3519 }
3520 else
3521 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003522 // find the file name after the end of the match
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003523 for (p = incl_regmatch.endp[0];
3524 *p && !vim_isfilec(*p); p++)
3525 ;
3526 for (i = 0; vim_isfilec(p[i]); i++)
3527 ;
3528 }
3529
Bram Moolenaar071d4272004-06-13 20:20:40 +00003530 if (i == 0)
3531 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003532 // Nothing found, use the rest of the line.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003533 p = incl_regmatch.endp[0];
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00003534 i = (int)STRLEN(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003535 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003536 // Avoid checking before the start of the line, can
3537 // happen if \zs appears in the regexp.
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003538 else if (p > line)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003539 {
3540 if (p[-1] == '"' || p[-1] == '<')
3541 {
3542 --p;
3543 ++i;
3544 }
3545 if (p[i] == '"' || p[i] == '>')
3546 ++i;
3547 }
3548 save_char = p[i];
3549 p[i] = NUL;
Bram Moolenaar8820b482017-03-16 17:23:31 +01003550 msg_outtrans_attr(p, HL_ATTR(HLF_D));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003551 p[i] = save_char;
3552 }
3553
3554 if (new_fname == NULL && action == ACTION_SHOW_ALL)
3555 {
3556 if (already_searched)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003557 msg_puts(_(" (Already listed)"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003558 else
Bram Moolenaar32526b32019-01-19 17:43:09 +01003559 msg_puts(_(" NOT FOUND"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003560 }
3561 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003562 out_flush(); // output each line directly
Bram Moolenaar071d4272004-06-13 20:20:40 +00003563 }
3564
3565 if (new_fname != NULL)
3566 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003567 // Push the new file onto the file stack
Bram Moolenaar071d4272004-06-13 20:20:40 +00003568 if (depth + 1 == old_files)
3569 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003570 bigger = ALLOC_MULT(SearchedFile, max_path_depth * 2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003571 if (bigger != NULL)
3572 {
3573 for (i = 0; i <= depth; i++)
3574 bigger[i] = files[i];
3575 for (i = depth + 1; i < old_files + max_path_depth; i++)
3576 {
3577 bigger[i].fp = NULL;
3578 bigger[i].name = NULL;
3579 bigger[i].lnum = 0;
3580 bigger[i].matched = FALSE;
3581 }
3582 for (i = old_files; i < max_path_depth; i++)
3583 bigger[i + max_path_depth] = files[i];
3584 old_files += max_path_depth;
3585 max_path_depth *= 2;
3586 vim_free(files);
3587 files = bigger;
3588 }
3589 }
3590 if ((files[depth + 1].fp = mch_fopen((char *)new_fname, "r"))
3591 == NULL)
3592 vim_free(new_fname);
3593 else
3594 {
3595 if (++depth == old_files)
3596 {
3597 /*
3598 * lalloc() for 'bigger' must have failed above. We
3599 * will forget one of our already visited files now.
3600 */
3601 vim_free(files[old_files].name);
3602 ++old_files;
3603 }
3604 files[depth].name = curr_fname = new_fname;
3605 files[depth].lnum = 0;
3606 files[depth].matched = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003607 if (action == ACTION_EXPAND)
3608 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003609 msg_hist_off = TRUE; // reset in msg_trunc_attr()
Bram Moolenaar555b2802005-05-19 21:08:39 +00003610 vim_snprintf((char*)IObuff, IOSIZE,
3611 _("Scanning included file: %s"),
3612 (char *)new_fname);
Bram Moolenaar32526b32019-01-19 17:43:09 +01003613 msg_trunc_attr((char *)IObuff, TRUE, HL_ATTR(HLF_R));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003614 }
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003615 else if (p_verbose >= 5)
Bram Moolenaar87b5ca52006-03-04 21:55:31 +00003616 {
3617 verbose_enter();
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003618 smsg(_("Searching included file %s"),
Bram Moolenaar87b5ca52006-03-04 21:55:31 +00003619 (char *)new_fname);
3620 verbose_leave();
3621 }
3622
Bram Moolenaar071d4272004-06-13 20:20:40 +00003623 }
3624 }
3625 }
3626 else
3627 {
3628 /*
3629 * Check if the line is a define (type == FIND_DEFINE)
3630 */
3631 p = line;
3632search_line:
3633 define_matched = FALSE;
3634 if (def_regmatch.regprog != NULL
3635 && vim_regexec(&def_regmatch, line, (colnr_T)0))
3636 {
3637 /*
3638 * Pattern must be first identifier after 'define', so skip
3639 * to that position before checking for match of pattern. Also
3640 * don't let it match beyond the end of this identifier.
3641 */
3642 p = def_regmatch.endp[0];
3643 while (*p && !vim_iswordc(*p))
3644 p++;
3645 define_matched = TRUE;
3646 }
3647
3648 /*
3649 * Look for a match. Don't do this if we are looking for a
3650 * define and this line didn't match define_prog above.
3651 */
3652 if (def_regmatch.regprog == NULL || define_matched)
3653 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003654 if (define_matched || compl_status_sol())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003655 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003656 // compare the first "len" chars from "ptr"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003657 startp = skipwhite(p);
3658 if (p_ic)
3659 matched = !MB_STRNICMP(startp, ptr, len);
3660 else
3661 matched = !STRNCMP(startp, ptr, len);
3662 if (matched && define_matched && whole
3663 && vim_iswordc(startp[len]))
3664 matched = FALSE;
3665 }
3666 else if (regmatch.regprog != NULL
3667 && vim_regexec(&regmatch, line, (colnr_T)(p - line)))
3668 {
3669 matched = TRUE;
3670 startp = regmatch.startp[0];
3671 /*
3672 * Check if the line is not a comment line (unless we are
3673 * looking for a define). A line starting with "# define"
3674 * is not considered to be a comment line.
3675 */
3676 if (!define_matched && skip_comments)
3677 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003678 if ((*line != '#' ||
3679 STRNCMP(skipwhite(line + 1), "define", 6) != 0)
Bram Moolenaar81340392012-06-06 16:12:59 +02003680 && get_leader_len(line, NULL, FALSE, TRUE))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003681 matched = FALSE;
3682
3683 /*
3684 * Also check for a "/ *" or "/ /" before the match.
3685 * Skips lines like "int backwards; / * normal index
3686 * * /" when looking for "normal".
3687 * Note: Doesn't skip "/ *" in comments.
3688 */
3689 p = skipwhite(line);
3690 if (matched
3691 || (p[0] == '/' && p[1] == '*') || p[0] == '*')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003692 for (p = line; *p && p < startp; ++p)
3693 {
3694 if (matched
3695 && p[0] == '/'
3696 && (p[1] == '*' || p[1] == '/'))
3697 {
3698 matched = FALSE;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003699 // After "//" all text is comment
Bram Moolenaar071d4272004-06-13 20:20:40 +00003700 if (p[1] == '/')
3701 break;
3702 ++p;
3703 }
3704 else if (!matched && p[0] == '*' && p[1] == '/')
3705 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003706 // Can find match after "* /".
Bram Moolenaar071d4272004-06-13 20:20:40 +00003707 matched = TRUE;
3708 ++p;
3709 }
3710 }
3711 }
3712 }
3713 }
3714 }
3715 if (matched)
3716 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003717 if (action == ACTION_EXPAND)
3718 {
Bram Moolenaard9eefe32019-04-06 14:22:21 +02003719 int cont_s_ipos = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003720 int add_r;
3721 char_u *aux;
3722
3723 if (depth == -1 && lnum == curwin->w_cursor.lnum)
3724 break;
3725 found = TRUE;
3726 aux = p = startp;
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003727 if (compl_status_adding())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003728 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003729 p += ins_compl_len();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003730 if (vim_iswordp(p))
3731 goto exit_matched;
3732 p = find_word_start(p);
3733 }
3734 p = find_word_end(p);
3735 i = (int)(p - aux);
3736
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003737 if (compl_status_adding() && i == ins_compl_len())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003738 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003739 // IOSIZE > compl_length, so the STRNCPY works
Bram Moolenaar071d4272004-06-13 20:20:40 +00003740 STRNCPY(IObuff, aux, i);
Bram Moolenaar89d40322006-08-29 15:30:07 +00003741
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003742 // Get the next line: when "depth" < 0 from the current
3743 // buffer, otherwise from the included file. Jump to
3744 // exit_matched when past the last line.
Bram Moolenaar89d40322006-08-29 15:30:07 +00003745 if (depth < 0)
3746 {
3747 if (lnum >= end_lnum)
3748 goto exit_matched;
3749 line = ml_get(++lnum);
3750 }
3751 else if (vim_fgets(line = file_line,
3752 LSIZE, files[depth].fp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003753 goto exit_matched;
3754
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003755 // we read a line, set "already" to check this "line" later
3756 // if depth >= 0 we'll increase files[depth].lnum far
Bram Moolenaar8e7d6222020-12-18 19:49:56 +01003757 // below -- Acevedo
Bram Moolenaar071d4272004-06-13 20:20:40 +00003758 already = aux = p = skipwhite(line);
3759 p = find_word_start(p);
3760 p = find_word_end(p);
3761 if (p > aux)
3762 {
3763 if (*aux != ')' && IObuff[i-1] != TAB)
3764 {
3765 if (IObuff[i-1] != ' ')
3766 IObuff[i++] = ' ';
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003767 // IObuf =~ "\(\k\|\i\).* ", thus i >= 2
Bram Moolenaar071d4272004-06-13 20:20:40 +00003768 if (p_js
3769 && (IObuff[i-2] == '.'
3770 || (vim_strchr(p_cpo, CPO_JOINSP) == NULL
3771 && (IObuff[i-2] == '?'
3772 || IObuff[i-2] == '!'))))
3773 IObuff[i++] = ' ';
3774 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003775 // copy as much as possible of the new word
Bram Moolenaar071d4272004-06-13 20:20:40 +00003776 if (p - aux >= IOSIZE - i)
3777 p = aux + IOSIZE - i - 1;
3778 STRNCPY(IObuff + i, aux, p - aux);
3779 i += (int)(p - aux);
Bram Moolenaard9eefe32019-04-06 14:22:21 +02003780 cont_s_ipos = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003781 }
3782 IObuff[i] = NUL;
3783 aux = IObuff;
3784
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003785 if (i == ins_compl_len())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003786 goto exit_matched;
3787 }
3788
Bram Moolenaare8c3a142006-08-29 14:30:35 +00003789 add_r = ins_compl_add_infercase(aux, i, p_ic,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003790 curr_fname == curbuf->b_fname ? NULL : curr_fname,
Bram Moolenaard9eefe32019-04-06 14:22:21 +02003791 dir, cont_s_ipos);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003792 if (add_r == OK)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003793 // if dir was BACKWARD then honor it just once
Bram Moolenaar071d4272004-06-13 20:20:40 +00003794 dir = FORWARD;
Bram Moolenaar572cb562005-08-05 21:35:02 +00003795 else if (add_r == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003796 break;
3797 }
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003798 else if (action == ACTION_SHOW_ALL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003799 {
3800 found = TRUE;
3801 if (!did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003802 gotocmdline(TRUE); // cursor at status line
Bram Moolenaar071d4272004-06-13 20:20:40 +00003803 if (curr_fname != prev_fname)
3804 {
3805 if (did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003806 msg_putchar('\n'); // cursor below last one
3807 if (!got_int) // don't display if 'q' typed
3808 // at "--more--" message
Bram Moolenaar071d4272004-06-13 20:20:40 +00003809 msg_home_replace_hl(curr_fname);
3810 prev_fname = curr_fname;
3811 }
3812 did_show = TRUE;
3813 if (!got_int)
3814 show_pat_in_path(line, type, TRUE, action,
3815 (depth == -1) ? NULL : files[depth].fp,
3816 (depth == -1) ? &lnum : &files[depth].lnum,
3817 match_count++);
3818
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003819 // Set matched flag for this file and all the ones that
3820 // include it
Bram Moolenaar071d4272004-06-13 20:20:40 +00003821 for (i = 0; i <= depth; ++i)
3822 files[i].matched = TRUE;
3823 }
3824 else if (--count <= 0)
3825 {
3826 found = TRUE;
3827 if (depth == -1 && lnum == curwin->w_cursor.lnum
Bram Moolenaar4033c552017-09-16 20:54:51 +02003828#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003829 && g_do_tagpreview == 0
3830#endif
3831 )
Bram Moolenaarac78dd42022-01-02 19:25:26 +00003832 emsg(_(e_match_is_on_current_line));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003833 else if (action == ACTION_SHOW)
3834 {
3835 show_pat_in_path(line, type, did_show, action,
3836 (depth == -1) ? NULL : files[depth].fp,
3837 (depth == -1) ? &lnum : &files[depth].lnum, 1L);
3838 did_show = TRUE;
3839 }
3840 else
3841 {
3842#ifdef FEAT_GUI
3843 need_mouse_correct = TRUE;
3844#endif
Bram Moolenaar4033c552017-09-16 20:54:51 +02003845#if defined(FEAT_QUICKFIX)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003846 // ":psearch" uses the preview window
Bram Moolenaar071d4272004-06-13 20:20:40 +00003847 if (g_do_tagpreview != 0)
3848 {
3849 curwin_save = curwin;
Bram Moolenaar576a4a62019-08-18 15:25:17 +02003850 prepare_tagpreview(TRUE, TRUE, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003851 }
3852#endif
3853 if (action == ACTION_SPLIT)
3854 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003855 if (win_split(0, 0) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003856 break;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02003857 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003858 }
3859 if (depth == -1)
3860 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003861 // match in current file
Bram Moolenaar4033c552017-09-16 20:54:51 +02003862#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003863 if (g_do_tagpreview != 0)
3864 {
Bram Moolenaar92bb83e2021-02-03 23:04:46 +01003865 if (!win_valid(curwin_save))
3866 break;
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02003867 if (!GETFILE_SUCCESS(getfile(
Bram Moolenaarc31f9ae2017-07-23 22:02:02 +02003868 curwin_save->w_buffer->b_fnum, NULL,
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02003869 NULL, TRUE, lnum, FALSE)))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003870 break; // failed to jump to file
Bram Moolenaar071d4272004-06-13 20:20:40 +00003871 }
3872 else
3873#endif
3874 setpcmark();
3875 curwin->w_cursor.lnum = lnum;
Bram Moolenaarc31f9ae2017-07-23 22:02:02 +02003876 check_cursor();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003877 }
3878 else
3879 {
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02003880 if (!GETFILE_SUCCESS(getfile(
3881 0, files[depth].name, NULL, TRUE,
3882 files[depth].lnum, FALSE)))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003883 break; // failed to jump to file
3884 // autocommands may have changed the lnum, we don't
3885 // want that here
Bram Moolenaar071d4272004-06-13 20:20:40 +00003886 curwin->w_cursor.lnum = files[depth].lnum;
3887 }
3888 }
3889 if (action != ACTION_SHOW)
3890 {
Bram Moolenaarfe81d452009-04-22 14:44:41 +00003891 curwin->w_cursor.col = (colnr_T)(startp - line);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003892 curwin->w_set_curswant = TRUE;
3893 }
3894
Bram Moolenaar4033c552017-09-16 20:54:51 +02003895#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003896 if (g_do_tagpreview != 0
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00003897 && curwin != curwin_save && win_valid(curwin_save))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003898 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003899 // Return cursor to where we were
Bram Moolenaar071d4272004-06-13 20:20:40 +00003900 validate_cursor();
3901 redraw_later(VALID);
3902 win_enter(curwin_save, TRUE);
3903 }
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +01003904# ifdef FEAT_PROP_POPUP
Bram Moolenaar1b6d9c42019-08-05 21:52:04 +02003905 else if (WIN_IS_POPUP(curwin))
3906 // can't keep focus in popup window
3907 win_enter(firstwin, TRUE);
3908# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003909#endif
3910 break;
3911 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003912exit_matched:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003913 matched = FALSE;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003914 // look for other matches in the rest of the line if we
3915 // are not at the end of it already
Bram Moolenaar071d4272004-06-13 20:20:40 +00003916 if (def_regmatch.regprog == NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003917 && action == ACTION_EXPAND
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003918 && !compl_status_sol()
Bram Moolenaarfe81d452009-04-22 14:44:41 +00003919 && *startp != NUL
Bram Moolenaar1614a142019-10-06 22:00:13 +02003920 && *(p = startp + mb_ptr2len(startp)) != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003921 goto search_line;
3922 }
3923 line_breakcheck();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003924 if (action == ACTION_EXPAND)
Bram Moolenaar472e8592016-10-15 17:06:47 +02003925 ins_compl_check_keys(30, FALSE);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003926 if (got_int || ins_compl_interrupted())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003927 break;
3928
3929 /*
3930 * Read the next line. When reading an included file and encountering
3931 * end-of-file, close the file and continue in the file that included
3932 * it.
3933 */
3934 while (depth >= 0 && !already
3935 && vim_fgets(line = file_line, LSIZE, files[depth].fp))
3936 {
3937 fclose(files[depth].fp);
3938 --old_files;
3939 files[old_files].name = files[depth].name;
3940 files[old_files].matched = files[depth].matched;
3941 --depth;
3942 curr_fname = (depth == -1) ? curbuf->b_fname
3943 : files[depth].name;
3944 if (depth < depth_displayed)
3945 depth_displayed = depth;
3946 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003947 if (depth >= 0) // we could read the line
Bram Moolenaarc84e3c12013-07-03 22:28:36 +02003948 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003949 files[depth].lnum++;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003950 // Remove any CR and LF from the line.
Bram Moolenaarc84e3c12013-07-03 22:28:36 +02003951 i = (int)STRLEN(line);
3952 if (i > 0 && line[i - 1] == '\n')
3953 line[--i] = NUL;
3954 if (i > 0 && line[i - 1] == '\r')
3955 line[--i] = NUL;
3956 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003957 else if (!already)
3958 {
3959 if (++lnum > end_lnum)
3960 break;
3961 line = ml_get(lnum);
3962 }
3963 already = NULL;
3964 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003965 // End of big for (;;) loop.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003966
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003967 // Close any files that are still open.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003968 for (i = 0; i <= depth; i++)
3969 {
3970 fclose(files[i].fp);
3971 vim_free(files[i].name);
3972 }
3973 for (i = old_files; i < max_path_depth; i++)
3974 vim_free(files[i].name);
3975 vim_free(files);
3976
3977 if (type == CHECK_PATH)
3978 {
3979 if (!did_show)
3980 {
3981 if (action != ACTION_SHOW_ALL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003982 msg(_("All included files were found"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003983 else
Bram Moolenaar32526b32019-01-19 17:43:09 +01003984 msg(_("No included files"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003985 }
3986 }
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003987 else if (!found && action != ACTION_EXPAND)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003988 {
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003989 if (got_int || ins_compl_interrupted())
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00003990 emsg(_(e_interrupted));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003991 else if (type == FIND_DEFINE)
Bram Moolenaarac78dd42022-01-02 19:25:26 +00003992 emsg(_(e_couldnt_find_definition));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003993 else
Bram Moolenaarac78dd42022-01-02 19:25:26 +00003994 emsg(_(e_couldnt_find_pattern));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003995 }
3996 if (action == ACTION_SHOW || action == ACTION_SHOW_ALL)
3997 msg_end();
3998
3999fpip_end:
4000 vim_free(file_line);
Bram Moolenaar473de612013-06-08 18:19:48 +02004001 vim_regfree(regmatch.regprog);
4002 vim_regfree(incl_regmatch.regprog);
4003 vim_regfree(def_regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004004}
4005
4006 static void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01004007show_pat_in_path(
4008 char_u *line,
4009 int type,
4010 int did_show,
4011 int action,
4012 FILE *fp,
4013 linenr_T *lnum,
4014 long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004015{
4016 char_u *p;
4017
4018 if (did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004019 msg_putchar('\n'); // cursor below last one
Bram Moolenaar91170f82006-05-05 21:15:17 +00004020 else if (!msg_silent)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004021 gotocmdline(TRUE); // cursor at status line
4022 if (got_int) // 'q' typed at "--more--" message
Bram Moolenaar071d4272004-06-13 20:20:40 +00004023 return;
4024 for (;;)
4025 {
4026 p = line + STRLEN(line) - 1;
4027 if (fp != NULL)
4028 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004029 // We used fgets(), so get rid of newline at end
Bram Moolenaar071d4272004-06-13 20:20:40 +00004030 if (p >= line && *p == '\n')
4031 --p;
4032 if (p >= line && *p == '\r')
4033 --p;
4034 *(p + 1) = NUL;
4035 }
4036 if (action == ACTION_SHOW_ALL)
4037 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004038 sprintf((char *)IObuff, "%3ld: ", count); // show match nr
Bram Moolenaar32526b32019-01-19 17:43:09 +01004039 msg_puts((char *)IObuff);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004040 sprintf((char *)IObuff, "%4ld", *lnum); // show line nr
4041 // Highlight line numbers
Bram Moolenaar32526b32019-01-19 17:43:09 +01004042 msg_puts_attr((char *)IObuff, HL_ATTR(HLF_N));
4043 msg_puts(" ");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004044 }
Bram Moolenaar26a60b42005-02-22 08:49:11 +00004045 msg_prt_line(line, FALSE);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004046 out_flush(); // show one line at a time
Bram Moolenaar071d4272004-06-13 20:20:40 +00004047
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004048 // Definition continues until line that doesn't end with '\'
Bram Moolenaar071d4272004-06-13 20:20:40 +00004049 if (got_int || type != FIND_DEFINE || p < line || *p != '\\')
4050 break;
4051
4052 if (fp != NULL)
4053 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004054 if (vim_fgets(line, LSIZE, fp)) // end of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00004055 break;
4056 ++*lnum;
4057 }
4058 else
4059 {
4060 if (++*lnum > curbuf->b_ml.ml_line_count)
4061 break;
4062 line = ml_get(*lnum);
4063 }
4064 msg_putchar('\n');
4065 }
4066}
4067#endif
4068
4069#ifdef FEAT_VIMINFO
Bram Moolenaar6bd1d772019-10-09 22:01:25 +02004070/*
4071 * Return the last used search pattern at "idx".
4072 */
Bram Moolenaarc3328162019-07-23 22:15:25 +02004073 spat_T *
4074get_spat(int idx)
4075{
4076 return &spats[idx];
4077}
4078
Bram Moolenaar6bd1d772019-10-09 22:01:25 +02004079/*
4080 * Return the last used search pattern index.
4081 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004082 int
Bram Moolenaarc3328162019-07-23 22:15:25 +02004083get_spat_last_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004084{
Bram Moolenaarc3328162019-07-23 22:15:25 +02004085 return last_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004086}
Bram Moolenaar071d4272004-06-13 20:20:40 +00004087#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004088
4089#ifdef FEAT_EVAL
4090/*
4091 * "searchcount()" function
4092 */
4093 void
4094f_searchcount(typval_T *argvars, typval_T *rettv)
4095{
4096 pos_T pos = curwin->w_cursor;
4097 char_u *pattern = NULL;
4098 int maxcount = SEARCH_STAT_DEF_MAX_COUNT;
4099 long timeout = SEARCH_STAT_DEF_TIMEOUT;
Bram Moolenaar4140c4f2020-09-05 23:16:00 +02004100 int recompute = TRUE;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004101 searchstat_T stat;
4102
4103 if (rettv_dict_alloc(rettv) == FAIL)
4104 return;
4105
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02004106 if (in_vim9script() && check_for_opt_dict_arg(argvars, 0) == FAIL)
4107 return;
4108
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004109 if (shortmess(SHM_SEARCHCOUNT)) // 'shortmess' contains 'S' flag
4110 recompute = TRUE;
4111
4112 if (argvars[0].v_type != VAR_UNKNOWN)
4113 {
Bram Moolenaar14681622020-06-03 22:57:39 +02004114 dict_T *dict;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004115 dictitem_T *di;
4116 listitem_T *li;
4117 int error = FALSE;
4118
Bram Moolenaar14681622020-06-03 22:57:39 +02004119 if (argvars[0].v_type != VAR_DICT || argvars[0].vval.v_dict == NULL)
4120 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00004121 emsg(_(e_dictionary_required));
Bram Moolenaar14681622020-06-03 22:57:39 +02004122 return;
4123 }
4124 dict = argvars[0].vval.v_dict;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004125 di = dict_find(dict, (char_u *)"timeout", -1);
4126 if (di != NULL)
4127 {
4128 timeout = (long)tv_get_number_chk(&di->di_tv, &error);
4129 if (error)
4130 return;
4131 }
4132 di = dict_find(dict, (char_u *)"maxcount", -1);
4133 if (di != NULL)
4134 {
4135 maxcount = (int)tv_get_number_chk(&di->di_tv, &error);
4136 if (error)
4137 return;
4138 }
Bram Moolenaar597aaac2020-09-05 21:21:16 +02004139 recompute = dict_get_bool(dict, (char_u *)"recompute", recompute);
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004140 di = dict_find(dict, (char_u *)"pattern", -1);
4141 if (di != NULL)
4142 {
4143 pattern = tv_get_string_chk(&di->di_tv);
4144 if (pattern == NULL)
4145 return;
4146 }
4147 di = dict_find(dict, (char_u *)"pos", -1);
4148 if (di != NULL)
4149 {
4150 if (di->di_tv.v_type != VAR_LIST)
4151 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00004152 semsg(_(e_invalid_argument_str), "pos");
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004153 return;
4154 }
4155 if (list_len(di->di_tv.vval.v_list) != 3)
4156 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00004157 semsg(_(e_invalid_argument_str), "List format should be [lnum, col, off]");
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004158 return;
4159 }
4160 li = list_find(di->di_tv.vval.v_list, 0L);
4161 if (li != NULL)
4162 {
4163 pos.lnum = tv_get_number_chk(&li->li_tv, &error);
4164 if (error)
4165 return;
4166 }
4167 li = list_find(di->di_tv.vval.v_list, 1L);
4168 if (li != NULL)
4169 {
4170 pos.col = tv_get_number_chk(&li->li_tv, &error) - 1;
4171 if (error)
4172 return;
4173 }
4174 li = list_find(di->di_tv.vval.v_list, 2L);
4175 if (li != NULL)
4176 {
4177 pos.coladd = tv_get_number_chk(&li->li_tv, &error);
4178 if (error)
4179 return;
4180 }
4181 }
4182 }
4183
4184 save_last_search_pattern();
4185 if (pattern != NULL)
4186 {
4187 if (*pattern == NUL)
4188 goto the_end;
Bram Moolenaar109aece2020-06-01 19:08:54 +02004189 vim_free(spats[last_idx].pat);
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004190 spats[last_idx].pat = vim_strsave(pattern);
4191 }
4192 if (spats[last_idx].pat == NULL || *spats[last_idx].pat == NUL)
4193 goto the_end; // the previous pattern was never defined
4194
4195 update_search_stat(0, &pos, &pos, &stat, recompute, maxcount, timeout);
4196
4197 dict_add_number(rettv->vval.v_dict, "current", stat.cur);
4198 dict_add_number(rettv->vval.v_dict, "total", stat.cnt);
4199 dict_add_number(rettv->vval.v_dict, "exact_match", stat.exact_match);
4200 dict_add_number(rettv->vval.v_dict, "incomplete", stat.incomplete);
4201 dict_add_number(rettv->vval.v_dict, "maxcount", stat.last_maxcount);
4202
4203the_end:
4204 restore_last_search_pattern();
4205}
Bram Moolenaar635414d2020-09-11 22:25:15 +02004206
4207/*
4208 * Fuzzy string matching
4209 *
4210 * Ported from the lib_fts library authored by Forrest Smith.
4211 * https://github.com/forrestthewoods/lib_fts/tree/master/code
4212 *
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004213 * The following blog describes the fuzzy matching algorithm:
Bram Moolenaar635414d2020-09-11 22:25:15 +02004214 * https://www.forrestthewoods.com/blog/reverse_engineering_sublime_texts_fuzzy_match/
4215 *
4216 * Each matching string is assigned a score. The following factors are checked:
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004217 * - Matched letter
4218 * - Unmatched letter
4219 * - Consecutively matched letters
4220 * - Proximity to start
4221 * - Letter following a separator (space, underscore)
4222 * - Uppercase letter following lowercase (aka CamelCase)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004223 *
4224 * Matched letters are good. Unmatched letters are bad. Matching near the start
4225 * is good. Matching the first letter in the middle of a phrase is good.
4226 * Matching the uppercase letters in camel case entries is good.
4227 *
4228 * The score assigned for each factor is explained below.
4229 * File paths are different from file names. File extensions may be ignorable.
4230 * Single words care about consecutive matches but not separators or camel
4231 * case.
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004232 * Score starts at 100
Bram Moolenaar635414d2020-09-11 22:25:15 +02004233 * Matched letter: +0 points
4234 * Unmatched letter: -1 point
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004235 * Consecutive match bonus: +15 points
4236 * First letter bonus: +15 points
4237 * Separator bonus: +30 points
4238 * Camel case bonus: +30 points
4239 * Unmatched leading letter: -5 points (max: -15)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004240 *
4241 * There is some nuance to this. Scores don’t have an intrinsic meaning. The
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004242 * score range isn’t 0 to 100. It’s roughly [50, 150]. Longer words have a
Bram Moolenaar635414d2020-09-11 22:25:15 +02004243 * lower minimum score due to unmatched letter penalty. Longer search patterns
4244 * have a higher maximum score due to match bonuses.
4245 *
4246 * Separator and camel case bonus is worth a LOT. Consecutive matches are worth
4247 * quite a bit.
4248 *
4249 * There is a penalty if you DON’T match the first three letters. Which
4250 * effectively rewards matching near the start. However there’s no difference
4251 * in matching between the middle and end.
4252 *
4253 * There is not an explicit bonus for an exact match. Unmatched letters receive
4254 * a penalty. So shorter strings and closer matches are worth more.
4255 */
4256typedef struct
4257{
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004258 int idx; // used for stable sort
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004259 listitem_T *item;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004260 int score;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004261 list_T *lmatchpos;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004262} fuzzyItem_T;
4263
Bram Moolenaare9f9f162020-10-20 19:01:30 +02004264// bonus for adjacent matches; this is higher than SEPARATOR_BONUS so that
4265// matching a whole word is preferred.
4266#define SEQUENTIAL_BONUS 40
Bram Moolenaardcdd42a2020-10-29 18:58:01 +01004267// bonus if match occurs after a path separator
4268#define PATH_SEPARATOR_BONUS 30
4269// bonus if match occurs after a word separator
4270#define WORD_SEPARATOR_BONUS 25
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004271// bonus if match is uppercase and prev is lower
4272#define CAMEL_BONUS 30
4273// bonus if the first letter is matched
4274#define FIRST_LETTER_BONUS 15
4275// penalty applied for every letter in str before the first match
4276#define LEADING_LETTER_PENALTY -5
4277// maximum penalty for leading letters
4278#define MAX_LEADING_LETTER_PENALTY -15
4279// penalty for every letter that doesn't match
4280#define UNMATCHED_LETTER_PENALTY -1
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004281// penalty for gap in matching positions (-2 * k)
4282#define GAP_PENALTY -2
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004283// Score for a string that doesn't fuzzy match the pattern
4284#define SCORE_NONE -9999
4285
4286#define FUZZY_MATCH_RECURSION_LIMIT 10
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004287
4288/*
4289 * Compute a score for a fuzzy matched string. The matching character locations
4290 * are in 'matches'.
4291 */
4292 static int
4293fuzzy_match_compute_score(
4294 char_u *str,
4295 int strSz,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004296 int_u *matches,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004297 int numMatches)
4298{
4299 int score;
4300 int penalty;
4301 int unmatched;
4302 int i;
4303 char_u *p = str;
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004304 int_u sidx = 0;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004305
4306 // Initialize score
4307 score = 100;
4308
4309 // Apply leading letter penalty
4310 penalty = LEADING_LETTER_PENALTY * matches[0];
4311 if (penalty < MAX_LEADING_LETTER_PENALTY)
4312 penalty = MAX_LEADING_LETTER_PENALTY;
4313 score += penalty;
4314
4315 // Apply unmatched penalty
4316 unmatched = strSz - numMatches;
4317 score += UNMATCHED_LETTER_PENALTY * unmatched;
4318
4319 // Apply ordering bonuses
4320 for (i = 0; i < numMatches; ++i)
4321 {
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004322 int_u currIdx = matches[i];
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004323
4324 if (i > 0)
4325 {
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004326 int_u prevIdx = matches[i - 1];
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004327
4328 // Sequential
4329 if (currIdx == (prevIdx + 1))
4330 score += SEQUENTIAL_BONUS;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004331 else
4332 score += GAP_PENALTY * (currIdx - prevIdx);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004333 }
4334
4335 // Check for bonuses based on neighbor character value
4336 if (currIdx > 0)
4337 {
4338 // Camel case
Bram Moolenaarc53e9c52020-09-22 22:08:32 +02004339 int neighbor = ' ';
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004340 int curr;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004341
4342 if (has_mbyte)
4343 {
4344 while (sidx < currIdx)
4345 {
4346 neighbor = (*mb_ptr2char)(p);
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004347 MB_PTR_ADV(p);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004348 sidx++;
4349 }
4350 curr = (*mb_ptr2char)(p);
4351 }
4352 else
4353 {
4354 neighbor = str[currIdx - 1];
4355 curr = str[currIdx];
4356 }
4357
4358 if (vim_islower(neighbor) && vim_isupper(curr))
4359 score += CAMEL_BONUS;
4360
Bram Moolenaardcdd42a2020-10-29 18:58:01 +01004361 // Bonus if the match follows a separator character
4362 if (neighbor == '/' || neighbor == '\\')
4363 score += PATH_SEPARATOR_BONUS;
4364 else if (neighbor == ' ' || neighbor == '_')
4365 score += WORD_SEPARATOR_BONUS;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004366 }
4367 else
4368 {
4369 // First letter
4370 score += FIRST_LETTER_BONUS;
4371 }
4372 }
4373 return score;
4374}
4375
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004376/*
4377 * Perform a recursive search for fuzzy matching 'fuzpat' in 'str'.
4378 * Return the number of matching characters.
4379 */
Bram Moolenaar635414d2020-09-11 22:25:15 +02004380 static int
4381fuzzy_match_recursive(
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004382 char_u *fuzpat,
4383 char_u *str,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004384 int_u strIdx,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004385 int *outScore,
4386 char_u *strBegin,
4387 int strLen,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004388 int_u *srcMatches,
4389 int_u *matches,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004390 int maxMatches,
4391 int nextMatch,
4392 int *recursionCount)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004393{
4394 // Recursion params
4395 int recursiveMatch = FALSE;
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004396 int_u bestRecursiveMatches[MAX_FUZZY_MATCHES];
Bram Moolenaar635414d2020-09-11 22:25:15 +02004397 int bestRecursiveScore = 0;
4398 int first_match;
4399 int matched;
4400
4401 // Count recursions
4402 ++*recursionCount;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004403 if (*recursionCount >= FUZZY_MATCH_RECURSION_LIMIT)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004404 return 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004405
4406 // Detect end of strings
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004407 if (*fuzpat == NUL || *str == NUL)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004408 return 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004409
4410 // Loop through fuzpat and str looking for a match
4411 first_match = TRUE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004412 while (*fuzpat != NUL && *str != NUL)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004413 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004414 int c1;
4415 int c2;
4416
4417 c1 = PTR2CHAR(fuzpat);
4418 c2 = PTR2CHAR(str);
4419
Bram Moolenaar635414d2020-09-11 22:25:15 +02004420 // Found match
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004421 if (vim_tolower(c1) == vim_tolower(c2))
Bram Moolenaar635414d2020-09-11 22:25:15 +02004422 {
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004423 int_u recursiveMatches[MAX_FUZZY_MATCHES];
Bram Moolenaar635414d2020-09-11 22:25:15 +02004424 int recursiveScore = 0;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004425 char_u *next_char;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004426
4427 // Supplied matches buffer was too short
4428 if (nextMatch >= maxMatches)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004429 return 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004430
4431 // "Copy-on-Write" srcMatches into matches
4432 if (first_match && srcMatches)
4433 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004434 memcpy(matches, srcMatches, nextMatch * sizeof(srcMatches[0]));
Bram Moolenaar635414d2020-09-11 22:25:15 +02004435 first_match = FALSE;
4436 }
4437
4438 // Recursive call that "skips" this match
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004439 if (has_mbyte)
4440 next_char = str + (*mb_ptr2len)(str);
4441 else
4442 next_char = str + 1;
4443 if (fuzzy_match_recursive(fuzpat, next_char, strIdx + 1,
4444 &recursiveScore, strBegin, strLen, matches,
4445 recursiveMatches,
K.Takataeeec2542021-06-02 13:28:16 +02004446 ARRAY_LENGTH(recursiveMatches),
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004447 nextMatch, recursionCount))
Bram Moolenaar635414d2020-09-11 22:25:15 +02004448 {
4449 // Pick best recursive score
4450 if (!recursiveMatch || recursiveScore > bestRecursiveScore)
4451 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004452 memcpy(bestRecursiveMatches, recursiveMatches,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004453 MAX_FUZZY_MATCHES * sizeof(recursiveMatches[0]));
Bram Moolenaar635414d2020-09-11 22:25:15 +02004454 bestRecursiveScore = recursiveScore;
4455 }
4456 recursiveMatch = TRUE;
4457 }
4458
4459 // Advance
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004460 matches[nextMatch++] = strIdx;
4461 if (has_mbyte)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004462 MB_PTR_ADV(fuzpat);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004463 else
4464 ++fuzpat;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004465 }
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004466 if (has_mbyte)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004467 MB_PTR_ADV(str);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004468 else
4469 ++str;
4470 strIdx++;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004471 }
4472
4473 // Determine if full fuzpat was matched
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004474 matched = *fuzpat == NUL ? TRUE : FALSE;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004475
4476 // Calculate score
4477 if (matched)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004478 *outScore = fuzzy_match_compute_score(strBegin, strLen, matches,
4479 nextMatch);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004480
4481 // Return best result
4482 if (recursiveMatch && (!matched || bestRecursiveScore > *outScore))
4483 {
4484 // Recursive score is better than "this"
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004485 memcpy(matches, bestRecursiveMatches, maxMatches * sizeof(matches[0]));
Bram Moolenaar635414d2020-09-11 22:25:15 +02004486 *outScore = bestRecursiveScore;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004487 return nextMatch;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004488 }
4489 else if (matched)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004490 return nextMatch; // "this" score is better than recursive
Bram Moolenaar635414d2020-09-11 22:25:15 +02004491
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004492 return 0; // no match
Bram Moolenaar635414d2020-09-11 22:25:15 +02004493}
4494
4495/*
4496 * fuzzy_match()
4497 *
4498 * Performs exhaustive search via recursion to find all possible matches and
4499 * match with highest score.
4500 * Scores values have no intrinsic meaning. Possible score range is not
4501 * normalized and varies with pattern.
4502 * Recursion is limited internally (default=10) to prevent degenerate cases
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004503 * (pat_arg="aaaaaa" str="aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa").
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004504 * Uses char_u for match indices. Therefore patterns are limited to
4505 * MAX_FUZZY_MATCHES characters.
Bram Moolenaar635414d2020-09-11 22:25:15 +02004506 *
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004507 * Returns TRUE if 'pat_arg' matches 'str'. Also returns the match score in
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004508 * 'outScore' and the matching character positions in 'matches'.
Bram Moolenaar635414d2020-09-11 22:25:15 +02004509 */
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004510 int
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004511fuzzy_match(
4512 char_u *str,
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004513 char_u *pat_arg,
4514 int matchseq,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004515 int *outScore,
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004516 int_u *matches,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004517 int maxMatches)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004518{
Bram Moolenaar635414d2020-09-11 22:25:15 +02004519 int recursionCount = 0;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004520 int len = MB_CHARLEN(str);
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004521 char_u *save_pat;
4522 char_u *pat;
4523 char_u *p;
4524 int complete = FALSE;
4525 int score = 0;
4526 int numMatches = 0;
4527 int matchCount;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004528
4529 *outScore = 0;
4530
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004531 save_pat = vim_strsave(pat_arg);
4532 if (save_pat == NULL)
4533 return FALSE;
4534 pat = save_pat;
4535 p = pat;
4536
4537 // Try matching each word in 'pat_arg' in 'str'
4538 while (TRUE)
4539 {
4540 if (matchseq)
4541 complete = TRUE;
4542 else
4543 {
4544 // Extract one word from the pattern (separated by space)
4545 p = skipwhite(p);
4546 if (*p == NUL)
4547 break;
4548 pat = p;
4549 while (*p != NUL && !VIM_ISWHITE(PTR2CHAR(p)))
4550 {
4551 if (has_mbyte)
4552 MB_PTR_ADV(p);
4553 else
4554 ++p;
4555 }
4556 if (*p == NUL) // processed all the words
4557 complete = TRUE;
4558 *p = NUL;
4559 }
4560
4561 score = 0;
4562 recursionCount = 0;
4563 matchCount = fuzzy_match_recursive(pat, str, 0, &score, str, len, NULL,
4564 matches + numMatches, maxMatches - numMatches,
4565 0, &recursionCount);
4566 if (matchCount == 0)
4567 {
4568 numMatches = 0;
4569 break;
4570 }
4571
4572 // Accumulate the match score and the number of matches
4573 *outScore += score;
4574 numMatches += matchCount;
4575
4576 if (complete)
4577 break;
4578
4579 // try matching the next word
4580 ++p;
4581 }
4582
4583 vim_free(save_pat);
4584 return numMatches != 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004585}
4586
4587/*
4588 * Sort the fuzzy matches in the descending order of the match score.
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004589 * For items with same score, retain the order using the index (stable sort)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004590 */
4591 static int
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004592fuzzy_match_item_compare(const void *s1, const void *s2)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004593{
4594 int v1 = ((fuzzyItem_T *)s1)->score;
4595 int v2 = ((fuzzyItem_T *)s2)->score;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004596 int idx1 = ((fuzzyItem_T *)s1)->idx;
4597 int idx2 = ((fuzzyItem_T *)s2)->idx;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004598
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004599 return v1 == v2 ? (idx1 - idx2) : v1 > v2 ? -1 : 1;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004600}
4601
4602/*
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004603 * Fuzzy search the string 'str' in a list of 'items' and return the matching
4604 * strings in 'fmatchlist'.
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004605 * If 'matchseq' is TRUE, then for multi-word search strings, match all the
4606 * words in sequence.
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004607 * If 'items' is a list of strings, then search for 'str' in the list.
4608 * If 'items' is a list of dicts, then either use 'key' to lookup the string
4609 * for each item or use 'item_cb' Funcref function to get the string.
4610 * If 'retmatchpos' is TRUE, then return a list of positions where 'str'
4611 * matches for each item.
Bram Moolenaar635414d2020-09-11 22:25:15 +02004612 */
4613 static void
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004614fuzzy_match_in_list(
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004615 list_T *items,
4616 char_u *str,
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004617 int matchseq,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004618 char_u *key,
4619 callback_T *item_cb,
4620 int retmatchpos,
4621 list_T *fmatchlist)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004622{
4623 long len;
4624 fuzzyItem_T *ptrs;
4625 listitem_T *li;
4626 long i = 0;
4627 int found_match = FALSE;
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004628 int_u matches[MAX_FUZZY_MATCHES];
Bram Moolenaar635414d2020-09-11 22:25:15 +02004629
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004630 len = list_len(items);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004631 if (len == 0)
4632 return;
4633
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004634 ptrs = ALLOC_CLEAR_MULT(fuzzyItem_T, len);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004635 if (ptrs == NULL)
4636 return;
4637
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004638 // For all the string items in items, get the fuzzy matching score
4639 FOR_ALL_LIST_ITEMS(items, li)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004640 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004641 int score;
4642 char_u *itemstr;
4643 typval_T rettv;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004644
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004645 ptrs[i].idx = i;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004646 ptrs[i].item = li;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004647 ptrs[i].score = SCORE_NONE;
4648 itemstr = NULL;
4649 rettv.v_type = VAR_UNKNOWN;
4650 if (li->li_tv.v_type == VAR_STRING) // list of strings
4651 itemstr = li->li_tv.vval.v_string;
4652 else if (li->li_tv.v_type == VAR_DICT &&
4653 (key != NULL || item_cb->cb_name != NULL))
4654 {
4655 // For a dict, either use the specified key to lookup the string or
4656 // use the specified callback function to get the string.
4657 if (key != NULL)
4658 itemstr = dict_get_string(li->li_tv.vval.v_dict, key, FALSE);
4659 else
Bram Moolenaar635414d2020-09-11 22:25:15 +02004660 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004661 typval_T argv[2];
4662
4663 // Invoke the supplied callback (if any) to get the dict item
4664 li->li_tv.vval.v_dict->dv_refcount++;
4665 argv[0].v_type = VAR_DICT;
4666 argv[0].vval.v_dict = li->li_tv.vval.v_dict;
4667 argv[1].v_type = VAR_UNKNOWN;
4668 if (call_callback(item_cb, -1, &rettv, 1, argv) != FAIL)
4669 {
4670 if (rettv.v_type == VAR_STRING)
4671 itemstr = rettv.vval.v_string;
4672 }
4673 dict_unref(li->li_tv.vval.v_dict);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004674 }
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004675 }
4676
4677 if (itemstr != NULL
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004678 && fuzzy_match(itemstr, str, matchseq, &score, matches,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004679 sizeof(matches) / sizeof(matches[0])))
4680 {
4681 // Copy the list of matching positions in itemstr to a list, if
4682 // 'retmatchpos' is set.
4683 if (retmatchpos)
4684 {
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004685 int j = 0;
4686 char_u *p;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004687
4688 ptrs[i].lmatchpos = list_alloc();
4689 if (ptrs[i].lmatchpos == NULL)
4690 goto done;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004691
4692 p = str;
4693 while (*p != NUL)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004694 {
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004695 if (!VIM_ISWHITE(PTR2CHAR(p)))
4696 {
4697 if (list_append_number(ptrs[i].lmatchpos,
4698 matches[j]) == FAIL)
4699 goto done;
4700 j++;
4701 }
4702 if (has_mbyte)
4703 MB_PTR_ADV(p);
4704 else
4705 ++p;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004706 }
4707 }
4708 ptrs[i].score = score;
4709 found_match = TRUE;
4710 }
Bram Moolenaar635414d2020-09-11 22:25:15 +02004711 ++i;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004712 clear_tv(&rettv);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004713 }
4714
4715 if (found_match)
4716 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004717 list_T *l;
4718
Bram Moolenaar635414d2020-09-11 22:25:15 +02004719 // Sort the list by the descending order of the match score
4720 qsort((void *)ptrs, (size_t)len, sizeof(fuzzyItem_T),
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004721 fuzzy_match_item_compare);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004722
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004723 // For matchfuzzy(), return a list of matched strings.
4724 // ['str1', 'str2', 'str3']
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004725 // For matchfuzzypos(), return a list with three items.
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004726 // The first item is a list of matched strings. The second item
4727 // is a list of lists where each list item is a list of matched
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004728 // character positions. The third item is a list of matching scores.
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004729 // [['str1', 'str2', 'str3'], [[1, 3], [1, 3], [1, 3]]]
4730 if (retmatchpos)
4731 {
4732 li = list_find(fmatchlist, 0);
4733 if (li == NULL || li->li_tv.vval.v_list == NULL)
4734 goto done;
4735 l = li->li_tv.vval.v_list;
4736 }
4737 else
4738 l = fmatchlist;
4739
4740 // Copy the matching strings with a valid score to the return list
Bram Moolenaar635414d2020-09-11 22:25:15 +02004741 for (i = 0; i < len; i++)
4742 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004743 if (ptrs[i].score == SCORE_NONE)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004744 break;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004745 list_append_tv(l, &ptrs[i].item->li_tv);
4746 }
4747
4748 // next copy the list of matching positions
4749 if (retmatchpos)
4750 {
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004751 li = list_find(fmatchlist, -2);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004752 if (li == NULL || li->li_tv.vval.v_list == NULL)
4753 goto done;
4754 l = li->li_tv.vval.v_list;
4755
4756 for (i = 0; i < len; i++)
4757 {
4758 if (ptrs[i].score == SCORE_NONE)
4759 break;
4760 if (ptrs[i].lmatchpos != NULL &&
4761 list_append_list(l, ptrs[i].lmatchpos) == FAIL)
4762 goto done;
4763 }
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004764
4765 // copy the matching scores
4766 li = list_find(fmatchlist, -1);
4767 if (li == NULL || li->li_tv.vval.v_list == NULL)
4768 goto done;
4769 l = li->li_tv.vval.v_list;
4770 for (i = 0; i < len; i++)
4771 {
4772 if (ptrs[i].score == SCORE_NONE)
4773 break;
4774 if (list_append_number(l, ptrs[i].score) == FAIL)
4775 goto done;
4776 }
Bram Moolenaar635414d2020-09-11 22:25:15 +02004777 }
4778 }
4779
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004780done:
Bram Moolenaar635414d2020-09-11 22:25:15 +02004781 vim_free(ptrs);
4782}
4783
4784/*
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004785 * Do fuzzy matching. Returns the list of matched strings in 'rettv'.
4786 * If 'retmatchpos' is TRUE, also returns the matching character positions.
4787 */
4788 static void
4789do_fuzzymatch(typval_T *argvars, typval_T *rettv, int retmatchpos)
4790{
4791 callback_T cb;
4792 char_u *key = NULL;
4793 int ret;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004794 int matchseq = FALSE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004795
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02004796 if (in_vim9script()
4797 && (check_for_list_arg(argvars, 0) == FAIL
4798 || check_for_string_arg(argvars, 1) == FAIL
4799 || check_for_opt_dict_arg(argvars, 2) == FAIL))
4800 return;
4801
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004802 CLEAR_POINTER(&cb);
4803
4804 // validate and get the arguments
4805 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
4806 {
Bram Moolenaar3a846e62022-01-01 16:21:00 +00004807 semsg(_(e_argument_of_str_must_be_list),
4808 retmatchpos ? "matchfuzzypos()" : "matchfuzzy()");
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004809 return;
4810 }
4811 if (argvars[1].v_type != VAR_STRING
4812 || argvars[1].vval.v_string == NULL)
4813 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00004814 semsg(_(e_invalid_argument_str), tv_get_string(&argvars[1]));
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004815 return;
4816 }
4817
4818 if (argvars[2].v_type != VAR_UNKNOWN)
4819 {
4820 dict_T *d;
4821 dictitem_T *di;
4822
4823 if (argvars[2].v_type != VAR_DICT || argvars[2].vval.v_dict == NULL)
4824 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00004825 emsg(_(e_dictionary_required));
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004826 return;
4827 }
4828
4829 // To search a dict, either a callback function or a key can be
4830 // specified.
4831 d = argvars[2].vval.v_dict;
4832 if ((di = dict_find(d, (char_u *)"key", -1)) != NULL)
4833 {
4834 if (di->di_tv.v_type != VAR_STRING
4835 || di->di_tv.vval.v_string == NULL
4836 || *di->di_tv.vval.v_string == NUL)
4837 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00004838 semsg(_(e_invalid_argument_str), tv_get_string(&di->di_tv));
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004839 return;
4840 }
4841 key = tv_get_string(&di->di_tv);
4842 }
4843 else if ((di = dict_find(d, (char_u *)"text_cb", -1)) != NULL)
4844 {
4845 cb = get_callback(&di->di_tv);
4846 if (cb.cb_name == NULL)
4847 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00004848 semsg(_(e_invalid_value_for_argument_str), "text_cb");
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004849 return;
4850 }
4851 }
Yegappan Lakshmananbb01a1e2021-04-26 21:17:52 +02004852 if (dict_find(d, (char_u *)"matchseq", -1) != NULL)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004853 matchseq = TRUE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004854 }
4855
4856 // get the fuzzy matches
4857 ret = rettv_list_alloc(rettv);
4858 if (ret != OK)
4859 goto done;
4860 if (retmatchpos)
4861 {
4862 list_T *l;
4863
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +01004864 // For matchfuzzypos(), a list with three items are returned. First
4865 // item is a list of matching strings, the second item is a list of
4866 // lists with matching positions within each string and the third item
4867 // is the list of scores of the matches.
4868 l = list_alloc();
4869 if (l == NULL)
4870 goto done;
4871 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4872 goto done;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004873 l = list_alloc();
4874 if (l == NULL)
4875 goto done;
4876 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4877 goto done;
4878 l = list_alloc();
4879 if (l == NULL)
4880 goto done;
4881 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4882 goto done;
4883 }
4884
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004885 fuzzy_match_in_list(argvars[0].vval.v_list, tv_get_string(&argvars[1]),
4886 matchseq, key, &cb, retmatchpos, rettv->vval.v_list);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004887
4888done:
4889 free_callback(&cb);
4890}
4891
4892/*
Bram Moolenaar635414d2020-09-11 22:25:15 +02004893 * "matchfuzzy()" function
4894 */
4895 void
4896f_matchfuzzy(typval_T *argvars, typval_T *rettv)
4897{
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004898 do_fuzzymatch(argvars, rettv, FALSE);
4899}
4900
4901/*
4902 * "matchfuzzypos()" function
4903 */
4904 void
4905f_matchfuzzypos(typval_T *argvars, typval_T *rettv)
4906{
4907 do_fuzzymatch(argvars, rettv, TRUE);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004908}
4909
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004910#endif