blob: 18d545a2ec3736b188c34410a4414ed951642a68 [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9/*
10 * search.c: code for normal mode searching commands
11 */
12
13#include "vim.h"
14
Bram Moolenaar071d4272004-06-13 20:20:40 +000015#ifdef FEAT_EVAL
Bram Moolenaarbaaa7e92016-01-29 22:47:03 +010016static void set_vv_searchforward(void);
17static int first_submatch(regmmatch_T *rp);
Bram Moolenaar071d4272004-06-13 20:20:40 +000018#endif
Bram Moolenaarbaaa7e92016-01-29 22:47:03 +010019static int check_linecomment(char_u *line);
Bram Moolenaar071d4272004-06-13 20:20:40 +000020#ifdef FEAT_FIND_ID
Bram Moolenaarbaaa7e92016-01-29 22:47:03 +010021static void show_pat_in_path(char_u *, int,
22 int, int, FILE *, linenr_T *, long);
Bram Moolenaar071d4272004-06-13 20:20:40 +000023#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +020024
25typedef struct searchstat
26{
27 int cur; // current position of found words
28 int cnt; // total count of found words
29 int exact_match; // TRUE if matched exactly on specified position
30 int incomplete; // 0: search was fully completed
31 // 1: recomputing was timed out
32 // 2: max count exceeded
33 int last_maxcount; // the max count of the last search
34} searchstat_T;
35
36static void cmdline_search_stat(int dirc, pos_T *pos, pos_T *cursor_pos, int show_top_bot_msg, char_u *msgbuf, int recompute, int maxcount, long timeout);
37static void update_search_stat(int dirc, pos_T *pos, pos_T *cursor_pos, searchstat_T *stat, int recompute, int maxcount, long timeout);
38
Bram Moolenaarea6561a2020-06-01 21:32:45 +020039#define SEARCH_STAT_DEF_TIMEOUT 40L
Bram Moolenaare8f5ec02020-06-01 17:28:35 +020040#define SEARCH_STAT_DEF_MAX_COUNT 99
41#define SEARCH_STAT_BUF_LEN 12
Bram Moolenaar071d4272004-06-13 20:20:40 +000042
Bram Moolenaar071d4272004-06-13 20:20:40 +000043/*
44 * This file contains various searching-related routines. These fall into
45 * three groups:
46 * 1. string searches (for /, ?, n, and N)
47 * 2. character searches within a single line (for f, F, t, T, etc)
48 * 3. "other" kinds of searches like the '%' command, and 'word' searches.
49 */
50
51/*
52 * String searches
53 *
54 * The string search functions are divided into two levels:
55 * lowest: searchit(); uses an pos_T for starting position and found match.
56 * Highest: do_search(); uses curwin->w_cursor; calls searchit().
57 *
58 * The last search pattern is remembered for repeating the same search.
59 * This pattern is shared between the :g, :s, ? and / commands.
60 * This is in search_regcomp().
61 *
62 * The actual string matching is done using a heavily modified version of
63 * Henry Spencer's regular expression library. See regexp.c.
64 */
65
Bram Moolenaar071d4272004-06-13 20:20:40 +000066/*
67 * Two search patterns are remembered: One for the :substitute command and
68 * one for other searches. last_idx points to the one that was used the last
69 * time.
70 */
Bram Moolenaarc3328162019-07-23 22:15:25 +020071static spat_T spats[2] =
Bram Moolenaar071d4272004-06-13 20:20:40 +000072{
Bram Moolenaar63d9e732019-12-05 21:10:38 +010073 {NULL, TRUE, FALSE, {'/', 0, 0, 0L}}, // last used search pat
74 {NULL, TRUE, FALSE, {'/', 0, 0, 0L}} // last used substitute pat
Bram Moolenaar071d4272004-06-13 20:20:40 +000075};
76
Bram Moolenaar63d9e732019-12-05 21:10:38 +010077static int last_idx = 0; // index in spats[] for RE_LAST
Bram Moolenaar071d4272004-06-13 20:20:40 +000078
Bram Moolenaar63d9e732019-12-05 21:10:38 +010079static char_u lastc[2] = {NUL, NUL}; // last character searched for
80static int lastcdir = FORWARD; // last direction of character search
81static int last_t_cmd = TRUE; // last search t_cmd
Bram Moolenaardbd24b52015-08-11 14:26:19 +020082static char_u lastc_bytes[MB_MAXBYTES + 1];
Bram Moolenaar63d9e732019-12-05 21:10:38 +010083static int lastc_bytelen = 1; // >1 for multi-byte char
Bram Moolenaardbd24b52015-08-11 14:26:19 +020084
Bram Moolenaar63d9e732019-12-05 21:10:38 +010085// copy of spats[], for keeping the search patterns while executing autocmds
Bram Moolenaarc3328162019-07-23 22:15:25 +020086static spat_T saved_spats[2];
Bram Moolenaar071d4272004-06-13 20:20:40 +000087# ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +010088static int saved_spats_last_idx = 0;
89static int saved_spats_no_hlsearch = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +000090# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +000091
Bram Moolenaar63d9e732019-12-05 21:10:38 +010092static char_u *mr_pattern = NULL; // pattern used by search_regcomp()
Bram Moolenaar071d4272004-06-13 20:20:40 +000093#ifdef FEAT_RIGHTLEFT
Bram Moolenaar63d9e732019-12-05 21:10:38 +010094static int mr_pattern_alloced = FALSE; // mr_pattern was allocated
Bram Moolenaar071d4272004-06-13 20:20:40 +000095#endif
96
97#ifdef FEAT_FIND_ID
98/*
99 * Type used by find_pattern_in_path() to remember which included files have
100 * been searched already.
101 */
102typedef struct SearchedFile
103{
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100104 FILE *fp; // File pointer
105 char_u *name; // Full name of file
106 linenr_T lnum; // Line we were up to in file
107 int matched; // Found a match in this file
Bram Moolenaar071d4272004-06-13 20:20:40 +0000108} SearchedFile;
109#endif
110
111/*
112 * translate search pattern for vim_regcomp()
113 *
114 * pat_save == RE_SEARCH: save pat in spats[RE_SEARCH].pat (normal search cmd)
115 * pat_save == RE_SUBST: save pat in spats[RE_SUBST].pat (:substitute command)
116 * pat_save == RE_BOTH: save pat in both patterns (:global command)
117 * pat_use == RE_SEARCH: use previous search pattern if "pat" is NULL
Bram Moolenaarb8017e72007-05-10 18:59:07 +0000118 * pat_use == RE_SUBST: use previous substitute pattern if "pat" is NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000119 * pat_use == RE_LAST: use last used pattern if "pat" is NULL
120 * options & SEARCH_HIS: put search string in history
121 * options & SEARCH_KEEP: keep previous search pattern
122 *
123 * returns FAIL if failed, OK otherwise.
124 */
125 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100126search_regcomp(
127 char_u *pat,
128 int pat_save,
129 int pat_use,
130 int options,
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100131 regmmatch_T *regmatch) // return: pattern and ignore-case flag
Bram Moolenaar071d4272004-06-13 20:20:40 +0000132{
133 int magic;
134 int i;
135
136 rc_did_emsg = FALSE;
137 magic = p_magic;
138
139 /*
140 * If no pattern given, use a previously defined pattern.
141 */
142 if (pat == NULL || *pat == NUL)
143 {
144 if (pat_use == RE_LAST)
145 i = last_idx;
146 else
147 i = pat_use;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100148 if (spats[i].pat == NULL) // pattern was never defined
Bram Moolenaar071d4272004-06-13 20:20:40 +0000149 {
150 if (pat_use == RE_SUBST)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100151 emsg(_(e_nopresub));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100153 emsg(_(e_noprevre));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000154 rc_did_emsg = TRUE;
155 return FAIL;
156 }
157 pat = spats[i].pat;
158 magic = spats[i].magic;
159 no_smartcase = spats[i].no_scs;
160 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100161 else if (options & SEARCH_HIS) // put new pattern in history
Bram Moolenaar071d4272004-06-13 20:20:40 +0000162 add_to_history(HIST_SEARCH, pat, TRUE, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000163
164#ifdef FEAT_RIGHTLEFT
165 if (mr_pattern_alloced)
166 {
167 vim_free(mr_pattern);
168 mr_pattern_alloced = FALSE;
169 }
170
171 if (curwin->w_p_rl && *curwin->w_p_rlc == 's')
172 {
173 char_u *rev_pattern;
174
175 rev_pattern = reverse_text(pat);
176 if (rev_pattern == NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100177 mr_pattern = pat; // out of memory, keep normal pattern.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000178 else
179 {
180 mr_pattern = rev_pattern;
181 mr_pattern_alloced = TRUE;
182 }
183 }
184 else
185#endif
186 mr_pattern = pat;
187
188 /*
189 * Save the currently used pattern in the appropriate place,
190 * unless the pattern should not be remembered.
191 */
Bram Moolenaar14177b72014-01-14 15:53:51 +0100192 if (!(options & SEARCH_KEEP) && !cmdmod.keeppatterns)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000193 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100194 // search or global command
Bram Moolenaar071d4272004-06-13 20:20:40 +0000195 if (pat_save == RE_SEARCH || pat_save == RE_BOTH)
196 save_re_pat(RE_SEARCH, pat, magic);
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100197 // substitute or global command
Bram Moolenaar071d4272004-06-13 20:20:40 +0000198 if (pat_save == RE_SUBST || pat_save == RE_BOTH)
199 save_re_pat(RE_SUBST, pat, magic);
200 }
201
202 regmatch->rmm_ic = ignorecase(pat);
Bram Moolenaar3b56eb32005-07-11 22:40:32 +0000203 regmatch->rmm_maxcol = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000204 regmatch->regprog = vim_regcomp(pat, magic ? RE_MAGIC : 0);
205 if (regmatch->regprog == NULL)
206 return FAIL;
207 return OK;
208}
209
210/*
211 * Get search pattern used by search_regcomp().
212 */
213 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100214get_search_pat(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000215{
216 return mr_pattern;
217}
218
Bram Moolenaarabc97732007-08-08 20:49:37 +0000219#if defined(FEAT_RIGHTLEFT) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000220/*
221 * Reverse text into allocated memory.
222 * Returns the allocated string, NULL when out of memory.
223 */
Bram Moolenaarabc97732007-08-08 20:49:37 +0000224 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100225reverse_text(char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000226{
227 unsigned len;
228 unsigned s_i, rev_i;
229 char_u *rev;
230
231 /*
232 * Reverse the pattern.
233 */
234 len = (unsigned)STRLEN(s);
235 rev = alloc(len + 1);
236 if (rev != NULL)
237 {
238 rev_i = len;
239 for (s_i = 0; s_i < len; ++s_i)
240 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000241 if (has_mbyte)
242 {
243 int mb_len;
244
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000245 mb_len = (*mb_ptr2len)(s + s_i);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000246 rev_i -= mb_len;
247 mch_memmove(rev + rev_i, s + s_i, mb_len);
248 s_i += mb_len - 1;
249 }
250 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000251 rev[--rev_i] = s[s_i];
252
253 }
254 rev[len] = NUL;
255 }
256 return rev;
257}
258#endif
259
Bram Moolenaarcc2b9d52014-12-13 03:17:11 +0100260 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100261save_re_pat(int idx, char_u *pat, int magic)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000262{
263 if (spats[idx].pat != pat)
264 {
265 vim_free(spats[idx].pat);
266 spats[idx].pat = vim_strsave(pat);
267 spats[idx].magic = magic;
268 spats[idx].no_scs = no_smartcase;
269 last_idx = idx;
270#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100271 // If 'hlsearch' set and search pat changed: need redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000272 if (p_hls)
Bram Moolenaar1c8f93f2006-03-12 22:10:07 +0000273 redraw_all_later(SOME_VALID);
Bram Moolenaar451fc7b2018-04-27 22:53:07 +0200274 set_no_hlsearch(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000275#endif
276 }
277}
278
Bram Moolenaar071d4272004-06-13 20:20:40 +0000279/*
280 * Save the search patterns, so they can be restored later.
281 * Used before/after executing autocommands and user functions.
282 */
283static int save_level = 0;
284
285 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100286save_search_patterns(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000287{
288 if (save_level++ == 0)
289 {
290 saved_spats[0] = spats[0];
291 if (spats[0].pat != NULL)
292 saved_spats[0].pat = vim_strsave(spats[0].pat);
293 saved_spats[1] = spats[1];
294 if (spats[1].pat != NULL)
295 saved_spats[1].pat = vim_strsave(spats[1].pat);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100296#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100297 saved_spats_last_idx = last_idx;
298 saved_spats_no_hlsearch = no_hlsearch;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100299#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000300 }
301}
302
303 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100304restore_search_patterns(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000305{
306 if (--save_level == 0)
307 {
308 vim_free(spats[0].pat);
309 spats[0] = saved_spats[0];
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100310#if defined(FEAT_EVAL)
Bram Moolenaar8c8de832008-06-24 22:58:06 +0000311 set_vv_searchforward();
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100312#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000313 vim_free(spats[1].pat);
314 spats[1] = saved_spats[1];
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100315#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100316 last_idx = saved_spats_last_idx;
317 set_no_hlsearch(saved_spats_no_hlsearch);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100318#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000319 }
320}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000321
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000322#if defined(EXITFREE) || defined(PROTO)
323 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100324free_search_patterns(void)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000325{
326 vim_free(spats[0].pat);
327 vim_free(spats[1].pat);
Bram Moolenaarf2427622009-04-22 16:45:21 +0000328
329# ifdef FEAT_RIGHTLEFT
330 if (mr_pattern_alloced)
331 {
Bram Moolenaarcc448b32010-07-14 16:52:17 +0200332 vim_free(mr_pattern);
333 mr_pattern_alloced = FALSE;
334 mr_pattern = NULL;
Bram Moolenaarf2427622009-04-22 16:45:21 +0000335 }
336# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000337}
338#endif
339
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100340#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100341// copy of spats[RE_SEARCH], for keeping the search patterns while incremental
342// searching
Bram Moolenaarc3328162019-07-23 22:15:25 +0200343static spat_T saved_last_search_spat;
Bram Moolenaared8bc782018-12-01 21:08:21 +0100344static int did_save_last_search_spat = 0;
345static int saved_last_idx = 0;
346static int saved_no_hlsearch = 0;
347
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100348/*
349 * Save and restore the search pattern for incremental highlight search
350 * feature.
351 *
Bram Moolenaarc4568ab2018-11-16 16:21:05 +0100352 * It's similar to but different from save_search_patterns() and
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100353 * restore_search_patterns(), because the search pattern must be restored when
Bram Moolenaarc4568ab2018-11-16 16:21:05 +0100354 * canceling incremental searching even if it's called inside user functions.
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100355 */
356 void
357save_last_search_pattern(void)
358{
Bram Moolenaar442a8532020-06-04 20:56:09 +0200359 if (++did_save_last_search_spat != 1)
360 // nested call, nothing to do
361 return;
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100362
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100363 saved_last_search_spat = spats[RE_SEARCH];
364 if (spats[RE_SEARCH].pat != NULL)
365 saved_last_search_spat.pat = vim_strsave(spats[RE_SEARCH].pat);
366 saved_last_idx = last_idx;
367 saved_no_hlsearch = no_hlsearch;
368}
369
370 void
371restore_last_search_pattern(void)
372{
Bram Moolenaar442a8532020-06-04 20:56:09 +0200373 if (--did_save_last_search_spat > 0)
374 // nested call, nothing to do
375 return;
376 if (did_save_last_search_spat != 0)
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100377 {
Bram Moolenaar442a8532020-06-04 20:56:09 +0200378 iemsg("restore_last_search_pattern() called more often than save_last_search_pattern()");
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100379 return;
380 }
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100381
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100382 vim_free(spats[RE_SEARCH].pat);
383 spats[RE_SEARCH] = saved_last_search_spat;
Bram Moolenaar01a060d2018-11-30 21:57:55 +0100384 saved_last_search_spat.pat = NULL;
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100385# if defined(FEAT_EVAL)
386 set_vv_searchforward();
387# endif
388 last_idx = saved_last_idx;
Bram Moolenaar451fc7b2018-04-27 22:53:07 +0200389 set_no_hlsearch(saved_no_hlsearch);
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100390}
Bram Moolenaard0480092017-11-16 22:20:39 +0100391
392 char_u *
393last_search_pattern(void)
394{
395 return spats[RE_SEARCH].pat;
396}
Bram Moolenaar2e51d9a2017-10-29 16:40:30 +0100397#endif
398
Bram Moolenaar071d4272004-06-13 20:20:40 +0000399/*
400 * Return TRUE when case should be ignored for search pattern "pat".
401 * Uses the 'ignorecase' and 'smartcase' options.
402 */
403 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100404ignorecase(char_u *pat)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000405{
Bram Moolenaar66e29d72016-08-20 16:57:02 +0200406 return ignorecase_opt(pat, p_ic, p_scs);
407}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000408
Bram Moolenaar66e29d72016-08-20 16:57:02 +0200409/*
410 * As ignorecase() put pass the "ic" and "scs" flags.
411 */
412 int
413ignorecase_opt(char_u *pat, int ic_in, int scs)
414{
415 int ic = ic_in;
416
417 if (ic && !no_smartcase && scs
Bram Moolenaare2c453d2019-08-21 14:37:09 +0200418 && !(ctrl_x_mode_not_default() && curbuf->b_p_inf))
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200419 ic = !pat_has_uppercase(pat);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000420 no_smartcase = FALSE;
421
422 return ic;
423}
424
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200425/*
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200426 * Return TRUE if pattern "pat" has an uppercase character.
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200427 */
428 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100429pat_has_uppercase(char_u *pat)
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200430{
431 char_u *p = pat;
432
433 while (*p != NUL)
434 {
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200435 int l;
436
437 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
438 {
439 if (enc_utf8 && utf_isupper(utf_ptr2char(p)))
440 return TRUE;
441 p += l;
442 }
Bram Moolenaar264b74f2019-01-24 17:18:42 +0100443 else if (*p == '\\')
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200444 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100445 if (p[1] == '_' && p[2] != NUL) // skip "\_X"
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200446 p += 3;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100447 else if (p[1] == '%' && p[2] != NUL) // skip "\%X"
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200448 p += 3;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100449 else if (p[1] != NUL) // skip "\X"
Bram Moolenaara9dc3752010-07-11 20:46:53 +0200450 p += 2;
451 else
452 p += 1;
453 }
454 else if (MB_ISUPPER(*p))
455 return TRUE;
456 else
457 ++p;
458 }
459 return FALSE;
460}
461
Bram Moolenaar113e1072019-01-20 15:30:40 +0100462#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000463 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100464last_csearch(void)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200465{
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200466 return lastc_bytes;
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200467}
468
469 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100470last_csearch_forward(void)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200471{
472 return lastcdir == FORWARD;
473}
474
475 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100476last_csearch_until(void)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200477{
478 return last_t_cmd == TRUE;
479}
480
481 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100482set_last_csearch(int c, char_u *s UNUSED, int len UNUSED)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200483{
484 *lastc = c;
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200485 lastc_bytelen = len;
486 if (len)
487 memcpy(lastc_bytes, s, len);
488 else
Bram Moolenaara80faa82020-04-12 19:37:17 +0200489 CLEAR_FIELD(lastc_bytes);
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200490}
Bram Moolenaar113e1072019-01-20 15:30:40 +0100491#endif
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200492
493 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100494set_csearch_direction(int cdir)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200495{
496 lastcdir = cdir;
497}
498
499 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100500set_csearch_until(int t_cmd)
Bram Moolenaardbd24b52015-08-11 14:26:19 +0200501{
502 last_t_cmd = t_cmd;
503}
504
505 char_u *
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100506last_search_pat(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000507{
508 return spats[last_idx].pat;
509}
510
511/*
512 * Reset search direction to forward. For "gd" and "gD" commands.
513 */
514 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100515reset_search_dir(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000516{
517 spats[0].off.dir = '/';
Bram Moolenaar8c8de832008-06-24 22:58:06 +0000518#if defined(FEAT_EVAL)
519 set_vv_searchforward();
520#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000521}
522
523#if defined(FEAT_EVAL) || defined(FEAT_VIMINFO)
524/*
525 * Set the last search pattern. For ":let @/ =" and viminfo.
526 * Also set the saved search pattern, so that this works in an autocommand.
527 */
528 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100529set_last_search_pat(
530 char_u *s,
531 int idx,
532 int magic,
533 int setlast)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000534{
535 vim_free(spats[idx].pat);
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100536 // An empty string means that nothing should be matched.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000537 if (*s == NUL)
538 spats[idx].pat = NULL;
539 else
540 spats[idx].pat = vim_strsave(s);
541 spats[idx].magic = magic;
542 spats[idx].no_scs = FALSE;
543 spats[idx].off.dir = '/';
Bram Moolenaar8c8de832008-06-24 22:58:06 +0000544#if defined(FEAT_EVAL)
545 set_vv_searchforward();
546#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000547 spats[idx].off.line = FALSE;
548 spats[idx].off.end = FALSE;
549 spats[idx].off.off = 0;
550 if (setlast)
551 last_idx = idx;
552 if (save_level)
553 {
554 vim_free(saved_spats[idx].pat);
555 saved_spats[idx] = spats[0];
556 if (spats[idx].pat == NULL)
557 saved_spats[idx].pat = NULL;
558 else
559 saved_spats[idx].pat = vim_strsave(spats[idx].pat);
Bram Moolenaar975880b2019-03-03 14:42:11 +0100560# ifdef FEAT_SEARCH_EXTRA
Bram Moolenaared8bc782018-12-01 21:08:21 +0100561 saved_spats_last_idx = last_idx;
Bram Moolenaar975880b2019-03-03 14:42:11 +0100562# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000563 }
564# ifdef FEAT_SEARCH_EXTRA
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100565 // If 'hlsearch' set and search pat changed: need redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000566 if (p_hls && idx == last_idx && !no_hlsearch)
Bram Moolenaar1c8f93f2006-03-12 22:10:07 +0000567 redraw_all_later(SOME_VALID);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000568# endif
569}
570#endif
571
572#ifdef FEAT_SEARCH_EXTRA
573/*
574 * Get a regexp program for the last used search pattern.
575 * This is used for highlighting all matches in a window.
576 * Values returned in regmatch->regprog and regmatch->rmm_ic.
577 */
578 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100579last_pat_prog(regmmatch_T *regmatch)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000580{
581 if (spats[last_idx].pat == NULL)
582 {
583 regmatch->regprog = NULL;
584 return;
585 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100586 ++emsg_off; // So it doesn't beep if bad expr
Bram Moolenaar071d4272004-06-13 20:20:40 +0000587 (void)search_regcomp((char_u *)"", 0, last_idx, SEARCH_KEEP, regmatch);
588 --emsg_off;
589}
590#endif
591
592/*
Bram Moolenaarf7ff6e82014-03-23 15:13:05 +0100593 * Lowest level search function.
Bram Moolenaar5d24a222018-12-23 19:10:09 +0100594 * Search for 'count'th occurrence of pattern "pat" in direction "dir".
595 * Start at position "pos" and return the found position in "pos".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000596 *
597 * if (options & SEARCH_MSG) == 0 don't give any messages
598 * if (options & SEARCH_MSG) == SEARCH_NFMSG don't give 'notfound' messages
599 * if (options & SEARCH_MSG) == SEARCH_MSG give all messages
600 * if (options & SEARCH_HIS) put search pattern in history
601 * if (options & SEARCH_END) return position at end of match
602 * if (options & SEARCH_START) accept match at pos itself
603 * if (options & SEARCH_KEEP) keep previous search pattern
604 * if (options & SEARCH_FOLD) match only once in a closed fold
605 * if (options & SEARCH_PEEK) check for typed char, cancel search
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100606 * if (options & SEARCH_COL) start at pos->col instead of zero
Bram Moolenaar071d4272004-06-13 20:20:40 +0000607 *
608 * Return FAIL (zero) for failure, non-zero for success.
609 * When FEAT_EVAL is defined, returns the index of the first matching
610 * subpattern plus one; one if there was none.
611 */
612 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100613searchit(
Bram Moolenaar92ea26b2019-10-18 20:53:34 +0200614 win_T *win, // window to search in; can be NULL for a
615 // buffer without a window!
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100616 buf_T *buf,
617 pos_T *pos,
Bram Moolenaar5d24a222018-12-23 19:10:09 +0100618 pos_T *end_pos, // set to end of the match, unless NULL
Bram Moolenaar764b23c2016-01-30 21:10:09 +0100619 int dir,
620 char_u *pat,
621 long count,
622 int options,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +0200623 int pat_use, // which pattern to use when "pat" is empty
624 searchit_arg_T *extra_arg) // optional extra arguments, can be NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000625{
626 int found;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100627 linenr_T lnum; // no init to shut up Apollo cc
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100628 colnr_T col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000629 regmmatch_T regmatch;
630 char_u *ptr;
631 colnr_T matchcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000632 lpos_T endpos;
Bram Moolenaar677ee682005-01-27 14:41:15 +0000633 lpos_T matchpos;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000634 int loop;
635 pos_T start_pos;
636 int at_first_line;
637 int extra_col;
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200638 int start_char_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000639 int match_ok;
640 long nmatched;
641 int submatch = 0;
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100642 int first_match = TRUE;
Bram Moolenaar53989552019-12-23 22:59:18 +0100643 int called_emsg_before = called_emsg;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000644#ifdef FEAT_SEARCH_EXTRA
645 int break_loop = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000646#endif
Bram Moolenaar92ea26b2019-10-18 20:53:34 +0200647 linenr_T stop_lnum = 0; // stop after this line number when != 0
648#ifdef FEAT_RELTIME
649 proftime_T *tm = NULL; // timeout limit or NULL
650 int *timed_out = NULL; // set when timed out or NULL
651#endif
652
653 if (extra_arg != NULL)
654 {
655 stop_lnum = extra_arg->sa_stop_lnum;
656#ifdef FEAT_RELTIME
657 tm = extra_arg->sa_tm;
658 timed_out = &extra_arg->sa_timed_out;
659#endif
660 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000661
662 if (search_regcomp(pat, RE_SEARCH, pat_use,
663 (options & (SEARCH_HIS + SEARCH_KEEP)), &regmatch) == FAIL)
664 {
665 if ((options & SEARCH_MSG) && !rc_did_emsg)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100666 semsg(_("E383: Invalid search string: %s"), mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000667 return FAIL;
668 }
669
Bram Moolenaar280f1262006-01-30 00:14:18 +0000670 /*
671 * find the string
672 */
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100673 do // loop for count
Bram Moolenaar071d4272004-06-13 20:20:40 +0000674 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100675 // When not accepting a match at the start position set "extra_col" to
676 // a non-zero value. Don't do that when starting at MAXCOL, since
677 // MAXCOL + 1 is zero.
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200678 if (pos->col == MAXCOL)
679 start_char_len = 0;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100680 // Watch out for the "col" being MAXCOL - 2, used in a closed fold.
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200681 else if (has_mbyte
682 && pos->lnum >= 1 && pos->lnum <= buf->b_ml.ml_line_count
683 && pos->col < MAXCOL - 2)
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100684 {
Bram Moolenaar82846a02018-02-09 18:09:54 +0100685 ptr = ml_get_buf(buf, pos->lnum, FALSE);
Bram Moolenaar8846ac52018-02-09 19:24:01 +0100686 if ((int)STRLEN(ptr) <= pos->col)
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200687 start_char_len = 1;
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100688 else
Bram Moolenaar82846a02018-02-09 18:09:54 +0100689 start_char_len = (*mb_ptr2len)(ptr + pos->col);
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100690 }
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100691 else
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200692 start_char_len = 1;
693 if (dir == FORWARD)
694 {
695 if (options & SEARCH_START)
696 extra_col = 0;
697 else
698 extra_col = start_char_len;
699 }
700 else
701 {
702 if (options & SEARCH_START)
703 extra_col = start_char_len;
704 else
705 extra_col = 0;
706 }
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100707
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100708 start_pos = *pos; // remember start pos for detecting no match
709 found = 0; // default: not found
710 at_first_line = TRUE; // default: start in first line
711 if (pos->lnum == 0) // correct lnum for when starting in line 0
Bram Moolenaar071d4272004-06-13 20:20:40 +0000712 {
713 pos->lnum = 1;
714 pos->col = 0;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100715 at_first_line = FALSE; // not in first line now
Bram Moolenaar071d4272004-06-13 20:20:40 +0000716 }
717
718 /*
719 * Start searching in current line, unless searching backwards and
720 * we're in column 0.
Bram Moolenaar7a42fa32007-07-10 11:28:55 +0000721 * If we are searching backwards, in column 0, and not including the
722 * current position, gain some efficiency by skipping back a line.
723 * Otherwise begin the search in the current line.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000724 */
Bram Moolenaar7a42fa32007-07-10 11:28:55 +0000725 if (dir == BACKWARD && start_pos.col == 0
726 && (options & SEARCH_START) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000727 {
728 lnum = pos->lnum - 1;
729 at_first_line = FALSE;
730 }
731 else
732 lnum = pos->lnum;
733
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100734 for (loop = 0; loop <= 1; ++loop) // loop twice if 'wrapscan' set
Bram Moolenaar071d4272004-06-13 20:20:40 +0000735 {
736 for ( ; lnum > 0 && lnum <= buf->b_ml.ml_line_count;
737 lnum += dir, at_first_line = FALSE)
738 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100739 // Stop after checking "stop_lnum", if it's set.
Bram Moolenaara23ccb82006-02-27 00:08:02 +0000740 if (stop_lnum != 0 && (dir == FORWARD
741 ? lnum > stop_lnum : lnum < stop_lnum))
742 break;
Bram Moolenaar76929292008-01-06 19:07:36 +0000743#ifdef FEAT_RELTIME
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100744 // Stop after passing the "tm" time limit.
Bram Moolenaar76929292008-01-06 19:07:36 +0000745 if (tm != NULL && profile_passed_limit(tm))
746 break;
747#endif
Bram Moolenaara23ccb82006-02-27 00:08:02 +0000748
Bram Moolenaar071d4272004-06-13 20:20:40 +0000749 /*
Bram Moolenaar677ee682005-01-27 14:41:15 +0000750 * Look for a match somewhere in line "lnum".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000751 */
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100752 col = at_first_line && (options & SEARCH_COL) ? pos->col
753 : (colnr_T)0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000754 nmatched = vim_regexec_multi(&regmatch, win, buf,
Bram Moolenaarad4d8a12015-12-28 19:20:36 +0100755 lnum, col,
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000756#ifdef FEAT_RELTIME
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200757 tm, timed_out
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000758#else
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200759 NULL, NULL
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000760#endif
761 );
Bram Moolenaar795aaa12020-10-02 20:36:01 +0200762 // vim_regexec_multi() may clear "regprog"
763 if (regmatch.regprog == NULL)
764 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100765 // Abort searching on an error (e.g., out of stack).
Bram Moolenaar53989552019-12-23 22:59:18 +0100766 if (called_emsg > called_emsg_before
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200767#ifdef FEAT_RELTIME
768 || (timed_out != NULL && *timed_out)
769#endif
770 )
Bram Moolenaar071d4272004-06-13 20:20:40 +0000771 break;
772 if (nmatched > 0)
773 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100774 // match may actually be in another line when using \zs
Bram Moolenaar677ee682005-01-27 14:41:15 +0000775 matchpos = regmatch.startpos[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000776 endpos = regmatch.endpos[0];
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000777#ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000778 submatch = first_submatch(&regmatch);
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000779#endif
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100780 // "lnum" may be past end of buffer for "\n\zs".
Bram Moolenaar32466aa2006-02-24 23:53:04 +0000781 if (lnum + matchpos.lnum > buf->b_ml.ml_line_count)
782 ptr = (char_u *)"";
783 else
784 ptr = ml_get_buf(buf, lnum + matchpos.lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000785
786 /*
787 * Forward search in the first line: match should be after
788 * the start position. If not, continue at the end of the
789 * match (this is vi compatible) or on the next char.
790 */
791 if (dir == FORWARD && at_first_line)
792 {
793 match_ok = TRUE;
794 /*
Bram Moolenaar677ee682005-01-27 14:41:15 +0000795 * When the match starts in a next line it's certainly
796 * past the start position.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000797 * When match lands on a NUL the cursor will be put
798 * one back afterwards, compare with that position,
799 * otherwise "/$" will get stuck on end of line.
800 */
Bram Moolenaar677ee682005-01-27 14:41:15 +0000801 while (matchpos.lnum == 0
Bram Moolenaara3dfccc2014-11-27 17:29:56 +0100802 && ((options & SEARCH_END) && first_match
Bram Moolenaar677ee682005-01-27 14:41:15 +0000803 ? (nmatched == 1
804 && (int)endpos.col - 1
Bram Moolenaar071d4272004-06-13 20:20:40 +0000805 < (int)start_pos.col + extra_col)
Bram Moolenaar677ee682005-01-27 14:41:15 +0000806 : ((int)matchpos.col
807 - (ptr[matchpos.col] == NUL)
808 < (int)start_pos.col + extra_col)))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000809 {
810 /*
811 * If vi-compatible searching, continue at the end
812 * of the match, otherwise continue one position
813 * forward.
814 */
815 if (vim_strchr(p_cpo, CPO_SEARCH) != NULL)
816 {
817 if (nmatched > 1)
818 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100819 // end is in next line, thus no match in
820 // this line
Bram Moolenaar071d4272004-06-13 20:20:40 +0000821 match_ok = FALSE;
822 break;
823 }
824 matchcol = endpos.col;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100825 // for empty match: advance one char
Bram Moolenaar677ee682005-01-27 14:41:15 +0000826 if (matchcol == matchpos.col
Bram Moolenaar071d4272004-06-13 20:20:40 +0000827 && ptr[matchcol] != NUL)
828 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000829 if (has_mbyte)
830 matchcol +=
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000831 (*mb_ptr2len)(ptr + matchcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000832 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000833 ++matchcol;
834 }
835 }
836 else
837 {
Bram Moolenaar677ee682005-01-27 14:41:15 +0000838 matchcol = matchpos.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000839 if (ptr[matchcol] != NUL)
840 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000841 if (has_mbyte)
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000842 matchcol += (*mb_ptr2len)(ptr
Bram Moolenaar071d4272004-06-13 20:20:40 +0000843 + matchcol);
844 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000845 ++matchcol;
846 }
847 }
Bram Moolenaar7bcb30e2013-04-03 21:14:29 +0200848 if (matchcol == 0 && (options & SEARCH_START))
Bram Moolenaardb333a52013-03-19 15:27:48 +0100849 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000850 if (ptr[matchcol] == NUL
851 || (nmatched = vim_regexec_multi(&regmatch,
Bram Moolenaar677ee682005-01-27 14:41:15 +0000852 win, buf, lnum + matchpos.lnum,
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000853 matchcol,
854#ifdef FEAT_RELTIME
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200855 tm, timed_out
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000856#else
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200857 NULL, NULL
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000858#endif
859 )) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000860 {
861 match_ok = FALSE;
862 break;
863 }
Bram Moolenaar795aaa12020-10-02 20:36:01 +0200864 // vim_regexec_multi() may clear "regprog"
865 if (regmatch.regprog == NULL)
866 break;
Bram Moolenaar677ee682005-01-27 14:41:15 +0000867 matchpos = regmatch.startpos[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000868 endpos = regmatch.endpos[0];
869# ifdef FEAT_EVAL
870 submatch = first_submatch(&regmatch);
871# endif
872
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100873 // Need to get the line pointer again, a
874 // multi-line search may have made it invalid.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000875 ptr = ml_get_buf(buf, lnum + matchpos.lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000876 }
877 if (!match_ok)
878 continue;
879 }
880 if (dir == BACKWARD)
881 {
882 /*
883 * Now, if there are multiple matches on this line,
884 * we have to get the last one. Or the last one before
885 * the cursor, if we're on that line.
886 * When putting the new cursor at the end, compare
887 * relative to the end of the match.
888 */
889 match_ok = FALSE;
890 for (;;)
891 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100892 // Remember a position that is before the start
893 // position, we use it if it's the last match in
894 // the line. Always accept a position after
895 // wrapping around.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000896 if (loop
897 || ((options & SEARCH_END)
898 ? (lnum + regmatch.endpos[0].lnum
899 < start_pos.lnum
900 || (lnum + regmatch.endpos[0].lnum
901 == start_pos.lnum
902 && (int)regmatch.endpos[0].col - 1
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200903 < (int)start_pos.col
904 + extra_col))
Bram Moolenaar677ee682005-01-27 14:41:15 +0000905 : (lnum + regmatch.startpos[0].lnum
906 < start_pos.lnum
907 || (lnum + regmatch.startpos[0].lnum
908 == start_pos.lnum
909 && (int)regmatch.startpos[0].col
Bram Moolenaar5f1e68b2015-07-10 14:43:35 +0200910 < (int)start_pos.col
911 + extra_col))))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000912 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000913 match_ok = TRUE;
Bram Moolenaar677ee682005-01-27 14:41:15 +0000914 matchpos = regmatch.startpos[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000915 endpos = regmatch.endpos[0];
916# ifdef FEAT_EVAL
917 submatch = first_submatch(&regmatch);
918# endif
919 }
920 else
921 break;
922
923 /*
924 * We found a valid match, now check if there is
925 * another one after it.
926 * If vi-compatible searching, continue at the end
927 * of the match, otherwise continue one position
928 * forward.
929 */
930 if (vim_strchr(p_cpo, CPO_SEARCH) != NULL)
931 {
932 if (nmatched > 1)
933 break;
934 matchcol = endpos.col;
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100935 // for empty match: advance one char
Bram Moolenaar677ee682005-01-27 14:41:15 +0000936 if (matchcol == matchpos.col
Bram Moolenaar071d4272004-06-13 20:20:40 +0000937 && ptr[matchcol] != NUL)
938 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000939 if (has_mbyte)
940 matchcol +=
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000941 (*mb_ptr2len)(ptr + matchcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000942 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000943 ++matchcol;
944 }
945 }
946 else
947 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100948 // Stop when the match is in a next line.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000949 if (matchpos.lnum > 0)
950 break;
951 matchcol = matchpos.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000952 if (ptr[matchcol] != NUL)
953 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000954 if (has_mbyte)
955 matchcol +=
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000956 (*mb_ptr2len)(ptr + matchcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000957 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000958 ++matchcol;
959 }
960 }
961 if (ptr[matchcol] == NUL
962 || (nmatched = vim_regexec_multi(&regmatch,
Bram Moolenaar677ee682005-01-27 14:41:15 +0000963 win, buf, lnum + matchpos.lnum,
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000964 matchcol,
965#ifdef FEAT_RELTIME
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200966 tm, timed_out
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000967#else
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +0200968 NULL, NULL
Bram Moolenaar91a4e822008-01-19 14:59:58 +0000969#endif
970 )) == 0)
Bram Moolenaar9d322762018-02-09 16:04:25 +0100971 {
972#ifdef FEAT_RELTIME
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100973 // If the search timed out, we did find a match
974 // but it might be the wrong one, so that's not
975 // OK.
Bram Moolenaar9d322762018-02-09 16:04:25 +0100976 if (timed_out != NULL && *timed_out)
977 match_ok = FALSE;
978#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000979 break;
Bram Moolenaar9d322762018-02-09 16:04:25 +0100980 }
Bram Moolenaar795aaa12020-10-02 20:36:01 +0200981 // vim_regexec_multi() may clear "regprog"
982 if (regmatch.regprog == NULL)
983 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000984
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100985 // Need to get the line pointer again, a
986 // multi-line search may have made it invalid.
Bram Moolenaar677ee682005-01-27 14:41:15 +0000987 ptr = ml_get_buf(buf, lnum + matchpos.lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000988 }
989
990 /*
991 * If there is only a match after the cursor, skip
992 * this match.
993 */
994 if (!match_ok)
995 continue;
996 }
997
Bram Moolenaar63d9e732019-12-05 21:10:38 +0100998 // With the SEARCH_END option move to the last character
999 // of the match. Don't do it for an empty match, end
1000 // should be same as start then.
Bram Moolenaar7bcb30e2013-04-03 21:14:29 +02001001 if ((options & SEARCH_END) && !(options & SEARCH_NOOF)
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001002 && !(matchpos.lnum == endpos.lnum
1003 && matchpos.col == endpos.col))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001004 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001005 // For a match in the first column, set the position
1006 // on the NUL in the previous line.
Bram Moolenaar677ee682005-01-27 14:41:15 +00001007 pos->lnum = lnum + endpos.lnum;
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001008 pos->col = endpos.col;
1009 if (endpos.col == 0)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00001010 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001011 if (pos->lnum > 1) // just in case
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001012 {
1013 --pos->lnum;
1014 pos->col = (colnr_T)STRLEN(ml_get_buf(buf,
1015 pos->lnum, FALSE));
1016 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00001017 }
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001018 else
1019 {
1020 --pos->col;
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001021 if (has_mbyte
1022 && pos->lnum <= buf->b_ml.ml_line_count)
1023 {
1024 ptr = ml_get_buf(buf, pos->lnum, FALSE);
1025 pos->col -= (*mb_head_off)(ptr, ptr + pos->col);
1026 }
Bram Moolenaar5bcbd532008-02-20 12:43:01 +00001027 }
Bram Moolenaar5d24a222018-12-23 19:10:09 +01001028 if (end_pos != NULL)
1029 {
1030 end_pos->lnum = lnum + matchpos.lnum;
1031 end_pos->col = matchpos.col;
1032 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001033 }
1034 else
1035 {
Bram Moolenaar677ee682005-01-27 14:41:15 +00001036 pos->lnum = lnum + matchpos.lnum;
1037 pos->col = matchpos.col;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01001038 if (end_pos != NULL)
1039 {
1040 end_pos->lnum = lnum + endpos.lnum;
1041 end_pos->col = endpos.col;
1042 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001043 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001044 pos->coladd = 0;
Bram Moolenaar5d24a222018-12-23 19:10:09 +01001045 if (end_pos != NULL)
1046 end_pos->coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001047 found = 1;
Bram Moolenaara3dfccc2014-11-27 17:29:56 +01001048 first_match = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001049
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001050 // Set variables used for 'incsearch' highlighting.
Bram Moolenaar677ee682005-01-27 14:41:15 +00001051 search_match_lines = endpos.lnum - matchpos.lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001052 search_match_endcol = endpos.col;
1053 break;
1054 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001055 line_breakcheck(); // stop if ctrl-C typed
Bram Moolenaar071d4272004-06-13 20:20:40 +00001056 if (got_int)
1057 break;
1058
1059#ifdef FEAT_SEARCH_EXTRA
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001060 // Cancel searching if a character was typed. Used for
1061 // 'incsearch'. Don't check too often, that would slowdown
1062 // searching too much.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001063 if ((options & SEARCH_PEEK)
1064 && ((lnum - pos->lnum) & 0x3f) == 0
1065 && char_avail())
1066 {
1067 break_loop = TRUE;
1068 break;
1069 }
1070#endif
1071
1072 if (loop && lnum == start_pos.lnum)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001073 break; // if second loop, stop where started
Bram Moolenaar071d4272004-06-13 20:20:40 +00001074 }
1075 at_first_line = FALSE;
1076
Bram Moolenaar795aaa12020-10-02 20:36:01 +02001077 // vim_regexec_multi() may clear "regprog"
1078 if (regmatch.regprog == NULL)
1079 break;
1080
Bram Moolenaar071d4272004-06-13 20:20:40 +00001081 /*
Bram Moolenaara23ccb82006-02-27 00:08:02 +00001082 * Stop the search if wrapscan isn't set, "stop_lnum" is
1083 * specified, after an interrupt, after a match and after looping
1084 * twice.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001085 */
Bram Moolenaar53989552019-12-23 22:59:18 +01001086 if (!p_ws || stop_lnum != 0 || got_int
1087 || called_emsg > called_emsg_before
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +02001088#ifdef FEAT_RELTIME
1089 || (timed_out != NULL && *timed_out)
Bram Moolenaar78a15312009-05-15 19:33:18 +00001090#endif
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +02001091#ifdef FEAT_SEARCH_EXTRA
1092 || break_loop
1093#endif
1094 || found || loop)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001095 break;
1096
1097 /*
1098 * If 'wrapscan' is set we continue at the other end of the file.
1099 * If 'shortmess' does not contain 's', we give a message.
1100 * This message is also remembered in keep_msg for when the screen
1101 * is redrawn. The keep_msg is cleared whenever another message is
1102 * written.
1103 */
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001104 if (dir == BACKWARD) // start second loop at the other end
Bram Moolenaar071d4272004-06-13 20:20:40 +00001105 lnum = buf->b_ml.ml_line_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001106 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001107 lnum = 1;
Bram Moolenaar92d640f2005-09-05 22:11:52 +00001108 if (!shortmess(SHM_SEARCH) && (options & SEARCH_MSG))
1109 give_warning((char_u *)_(dir == BACKWARD
1110 ? top_bot_msg : bot_top_msg), TRUE);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001111 if (extra_arg != NULL)
1112 extra_arg->sa_wrapped = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001113 }
Bram Moolenaar53989552019-12-23 22:59:18 +01001114 if (got_int || called_emsg > called_emsg_before
Bram Moolenaarfbd0b0a2017-06-17 18:44:21 +02001115#ifdef FEAT_RELTIME
1116 || (timed_out != NULL && *timed_out)
1117#endif
Bram Moolenaar78a15312009-05-15 19:33:18 +00001118#ifdef FEAT_SEARCH_EXTRA
1119 || break_loop
1120#endif
1121 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00001122 break;
1123 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001124 while (--count > 0 && found); // stop after count matches or no match
Bram Moolenaar071d4272004-06-13 20:20:40 +00001125
Bram Moolenaar473de612013-06-08 18:19:48 +02001126 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001127
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001128 if (!found) // did not find it
Bram Moolenaar071d4272004-06-13 20:20:40 +00001129 {
1130 if (got_int)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001131 emsg(_(e_interr));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001132 else if ((options & SEARCH_MSG) == SEARCH_MSG)
1133 {
1134 if (p_ws)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001135 semsg(_(e_patnotf2), mr_pattern);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001136 else if (lnum == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001137 semsg(_("E384: search hit TOP without match for: %s"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00001138 mr_pattern);
1139 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001140 semsg(_("E385: search hit BOTTOM without match for: %s"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00001141 mr_pattern);
1142 }
1143 return FAIL;
1144 }
1145
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001146 // A pattern like "\n\zs" may go past the last line.
Bram Moolenaar32466aa2006-02-24 23:53:04 +00001147 if (pos->lnum > buf->b_ml.ml_line_count)
1148 {
1149 pos->lnum = buf->b_ml.ml_line_count;
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001150 pos->col = (int)STRLEN(ml_get_buf(buf, pos->lnum, FALSE));
Bram Moolenaar32466aa2006-02-24 23:53:04 +00001151 if (pos->col > 0)
1152 --pos->col;
1153 }
1154
Bram Moolenaar071d4272004-06-13 20:20:40 +00001155 return submatch + 1;
1156}
1157
1158#ifdef FEAT_EVAL
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001159 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001160set_search_direction(int cdir)
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001161{
1162 spats[0].off.dir = cdir;
1163}
1164
1165 static void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001166set_vv_searchforward(void)
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001167{
1168 set_vim_var_nr(VV_SEARCHFORWARD, (long)(spats[0].off.dir == '/'));
1169}
1170
Bram Moolenaar071d4272004-06-13 20:20:40 +00001171/*
1172 * Return the number of the first subpat that matched.
Bram Moolenaarad4d8a12015-12-28 19:20:36 +01001173 * Return zero if none of them matched.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001174 */
1175 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001176first_submatch(regmmatch_T *rp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001177{
1178 int submatch;
1179
1180 for (submatch = 1; ; ++submatch)
1181 {
1182 if (rp->startpos[submatch].lnum >= 0)
1183 break;
1184 if (submatch == 9)
1185 {
1186 submatch = 0;
1187 break;
1188 }
1189 }
1190 return submatch;
1191}
1192#endif
1193
1194/*
1195 * Highest level string search function.
Bram Moolenaarb8017e72007-05-10 18:59:07 +00001196 * Search for the 'count'th occurrence of pattern 'pat' in direction 'dirc'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001197 * If 'dirc' is 0: use previous dir.
1198 * If 'pat' is NULL or empty : use previous string.
1199 * If 'options & SEARCH_REV' : go in reverse of previous dir.
1200 * If 'options & SEARCH_ECHO': echo the search command and handle options
1201 * If 'options & SEARCH_MSG' : may give error message
1202 * If 'options & SEARCH_OPT' : interpret optional flags
1203 * If 'options & SEARCH_HIS' : put search pattern in history
1204 * If 'options & SEARCH_NOOF': don't add offset to position
1205 * If 'options & SEARCH_MARK': set previous context mark
1206 * If 'options & SEARCH_KEEP': keep previous search pattern
1207 * If 'options & SEARCH_START': accept match at curpos itself
1208 * If 'options & SEARCH_PEEK': check for typed char, cancel search
1209 *
1210 * Careful: If spats[0].off.line == TRUE and spats[0].off.off == 0 this
1211 * makes the movement linewise without moving the match position.
1212 *
Bram Moolenaarb6c27352015-03-05 19:57:49 +01001213 * Return 0 for failure, 1 for found, 2 for found and line offset added.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001214 */
1215 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001216do_search(
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001217 oparg_T *oap, // can be NULL
1218 int dirc, // '/' or '?'
Bram Moolenaarc036e872020-02-21 21:30:52 +01001219 int search_delim, // the delimiter for the search, e.g. '%' in s%regex%replacement%
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001220 char_u *pat,
1221 long count,
1222 int options,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001223 searchit_arg_T *sia) // optional arguments or NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001224{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001225 pos_T pos; // position of the last match
Bram Moolenaar071d4272004-06-13 20:20:40 +00001226 char_u *searchstr;
Bram Moolenaarc3328162019-07-23 22:15:25 +02001227 soffset_T old_off;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001228 int retval; // Return value
Bram Moolenaar071d4272004-06-13 20:20:40 +00001229 char_u *p;
1230 long c;
1231 char_u *dircp;
1232 char_u *strcopy = NULL;
1233 char_u *ps;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001234 char_u *msgbuf = NULL;
1235 size_t len;
Bram Moolenaar8f46e4c2019-05-24 22:08:15 +02001236 int has_offset = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001237
1238 /*
1239 * A line offset is not remembered, this is vi compatible.
1240 */
1241 if (spats[0].off.line && vim_strchr(p_cpo, CPO_LINEOFF) != NULL)
1242 {
1243 spats[0].off.line = FALSE;
1244 spats[0].off.off = 0;
1245 }
1246
1247 /*
1248 * Save the values for when (options & SEARCH_KEEP) is used.
1249 * (there is no "if ()" around this because gcc wants them initialized)
1250 */
1251 old_off = spats[0].off;
1252
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001253 pos = curwin->w_cursor; // start searching at the cursor position
Bram Moolenaar071d4272004-06-13 20:20:40 +00001254
1255 /*
1256 * Find out the direction of the search.
1257 */
1258 if (dirc == 0)
1259 dirc = spats[0].off.dir;
1260 else
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001261 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001262 spats[0].off.dir = dirc;
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001263#if defined(FEAT_EVAL)
1264 set_vv_searchforward();
1265#endif
1266 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001267 if (options & SEARCH_REV)
1268 {
Bram Moolenaar4f974752019-02-17 17:44:42 +01001269#ifdef MSWIN
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001270 // There is a bug in the Visual C++ 2.2 compiler which means that
1271 // dirc always ends up being '/'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001272 dirc = (dirc == '/') ? '?' : '/';
1273#else
1274 if (dirc == '/')
1275 dirc = '?';
1276 else
1277 dirc = '/';
1278#endif
1279 }
1280
1281#ifdef FEAT_FOLDING
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001282 // If the cursor is in a closed fold, don't find another match in the same
1283 // fold.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001284 if (dirc == '/')
1285 {
1286 if (hasFolding(pos.lnum, NULL, &pos.lnum))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001287 pos.col = MAXCOL - 2; // avoid overflow when adding 1
Bram Moolenaar071d4272004-06-13 20:20:40 +00001288 }
1289 else
1290 {
1291 if (hasFolding(pos.lnum, &pos.lnum, NULL))
1292 pos.col = 0;
1293 }
1294#endif
1295
1296#ifdef FEAT_SEARCH_EXTRA
1297 /*
1298 * Turn 'hlsearch' highlighting back on.
1299 */
1300 if (no_hlsearch && !(options & SEARCH_KEEP))
1301 {
Bram Moolenaar1c8f93f2006-03-12 22:10:07 +00001302 redraw_all_later(SOME_VALID);
Bram Moolenaar451fc7b2018-04-27 22:53:07 +02001303 set_no_hlsearch(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001304 }
1305#endif
1306
1307 /*
1308 * Repeat the search when pattern followed by ';', e.g. "/foo/;?bar".
1309 */
1310 for (;;)
1311 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001312 int show_top_bot_msg = FALSE;
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02001313
Bram Moolenaar071d4272004-06-13 20:20:40 +00001314 searchstr = pat;
1315 dircp = NULL;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001316 // use previous pattern
Bram Moolenaarc036e872020-02-21 21:30:52 +01001317 if (pat == NULL || *pat == NUL || *pat == search_delim)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001318 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001319 if (spats[RE_SEARCH].pat == NULL) // no previous pattern
Bram Moolenaar071d4272004-06-13 20:20:40 +00001320 {
Bram Moolenaarea683da2016-09-09 21:41:34 +02001321 searchstr = spats[RE_SUBST].pat;
1322 if (searchstr == NULL)
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001323 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001324 emsg(_(e_noprevre));
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001325 retval = 0;
1326 goto end_do_search;
1327 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001328 }
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001329 else
1330 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001331 // make search_regcomp() use spats[RE_SEARCH].pat
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01001332 searchstr = (char_u *)"";
1333 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001334 }
1335
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001336 if (pat != NULL && *pat != NUL) // look for (new) offset
Bram Moolenaar071d4272004-06-13 20:20:40 +00001337 {
1338 /*
1339 * Find end of regular expression.
1340 * If there is a matching '/' or '?', toss it.
1341 */
1342 ps = strcopy;
Bram Moolenaare8c4abb2020-04-02 21:13:25 +02001343 p = skip_regexp_ex(pat, search_delim, (int)p_magic, &strcopy, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001344 if (strcopy != ps)
1345 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001346 // made a copy of "pat" to change "\?" to "?"
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001347 searchcmdlen += (int)(STRLEN(pat) - STRLEN(strcopy));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001348 pat = strcopy;
1349 searchstr = strcopy;
1350 }
Bram Moolenaarc036e872020-02-21 21:30:52 +01001351 if (*p == search_delim)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001352 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001353 dircp = p; // remember where we put the NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001354 *p++ = NUL;
1355 }
1356 spats[0].off.line = FALSE;
1357 spats[0].off.end = FALSE;
1358 spats[0].off.off = 0;
1359 /*
1360 * Check for a line offset or a character offset.
1361 * For get_address (echo off) we don't check for a character
1362 * offset, because it is meaningless and the 's' could be a
1363 * substitute command.
1364 */
1365 if (*p == '+' || *p == '-' || VIM_ISDIGIT(*p))
1366 spats[0].off.line = TRUE;
1367 else if ((options & SEARCH_OPT) &&
1368 (*p == 'e' || *p == 's' || *p == 'b'))
1369 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001370 if (*p == 'e') // end
Bram Moolenaar071d4272004-06-13 20:20:40 +00001371 spats[0].off.end = SEARCH_END;
1372 ++p;
1373 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001374 if (VIM_ISDIGIT(*p) || *p == '+' || *p == '-') // got an offset
Bram Moolenaar071d4272004-06-13 20:20:40 +00001375 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001376 // 'nr' or '+nr' or '-nr'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001377 if (VIM_ISDIGIT(*p) || VIM_ISDIGIT(*(p + 1)))
1378 spats[0].off.off = atol((char *)p);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001379 else if (*p == '-') // single '-'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001380 spats[0].off.off = -1;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001381 else // single '+'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001382 spats[0].off.off = 1;
1383 ++p;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001384 while (VIM_ISDIGIT(*p)) // skip number
Bram Moolenaar071d4272004-06-13 20:20:40 +00001385 ++p;
1386 }
1387
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001388 // compute length of search command for get_address()
Bram Moolenaar071d4272004-06-13 20:20:40 +00001389 searchcmdlen += (int)(p - pat);
1390
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001391 pat = p; // put pat after search command
Bram Moolenaar071d4272004-06-13 20:20:40 +00001392 }
1393
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001394 if ((options & SEARCH_ECHO) && messaging() &&
1395 !msg_silent &&
1396 (!cmd_silent || !shortmess(SHM_SEARCHCOUNT)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001397 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001398 char_u *trunc;
Bram Moolenaar984f0312019-05-24 13:11:47 +02001399 char_u off_buf[40];
Bram Moolenaard33a7642019-05-24 17:56:14 +02001400 size_t off_len = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001401
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001402 // Compute msg_row early.
1403 msg_start();
1404
Bram Moolenaar984f0312019-05-24 13:11:47 +02001405 // Get the offset, so we know how long it is.
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001406 if (!cmd_silent &&
1407 (spats[0].off.line || spats[0].off.end || spats[0].off.off))
Bram Moolenaar984f0312019-05-24 13:11:47 +02001408 {
1409 p = off_buf;
1410 *p++ = dirc;
1411 if (spats[0].off.end)
1412 *p++ = 'e';
1413 else if (!spats[0].off.line)
1414 *p++ = 's';
1415 if (spats[0].off.off > 0 || spats[0].off.line)
1416 *p++ = '+';
1417 *p = NUL;
1418 if (spats[0].off.off != 0 || spats[0].off.line)
1419 sprintf((char *)p, "%ld", spats[0].off.off);
1420 off_len = STRLEN(off_buf);
1421 }
1422
Bram Moolenaar071d4272004-06-13 20:20:40 +00001423 if (*searchstr == NUL)
Bram Moolenaar2fb8f682018-12-01 13:14:45 +01001424 p = spats[0].pat;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001425 else
1426 p = searchstr;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001427
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001428 if (!shortmess(SHM_SEARCHCOUNT) || cmd_silent)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001429 {
1430 // Reserve enough space for the search pattern + offset +
Bram Moolenaar984f0312019-05-24 13:11:47 +02001431 // search stat. Use all the space available, so that the
1432 // search state is right aligned. If there is not enough space
1433 // msg_strtrunc() will shorten in the middle.
Bram Moolenaar19e8ac72019-09-03 22:23:38 +02001434 if (msg_scrolled != 0 && !cmd_silent)
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001435 // Use all the columns.
1436 len = (int)(Rows - msg_row) * Columns - 1;
1437 else
1438 // Use up to 'showcmd' column.
1439 len = (int)(Rows - msg_row - 1) * Columns + sc_col - 1;
Bram Moolenaar984f0312019-05-24 13:11:47 +02001440 if (len < STRLEN(p) + off_len + SEARCH_STAT_BUF_LEN + 3)
1441 len = STRLEN(p) + off_len + SEARCH_STAT_BUF_LEN + 3;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001442 }
1443 else
1444 // Reserve enough space for the search pattern + offset.
Bram Moolenaar984f0312019-05-24 13:11:47 +02001445 len = STRLEN(p) + off_len + 3;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001446
Bram Moolenaar880e4d92020-04-11 21:31:28 +02001447 vim_free(msgbuf);
Bram Moolenaar51e14382019-05-25 20:21:28 +02001448 msgbuf = alloc(len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001449 if (msgbuf != NULL)
1450 {
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001451 vim_memset(msgbuf, ' ', len);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001452 msgbuf[len - 1] = NUL;
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001453 // do not fill the msgbuf buffer, if cmd_silent is set, leave it
1454 // empty for the search_stat feature.
1455 if (!cmd_silent)
Bram Moolenaarcafda4f2005-09-06 19:25:11 +00001456 {
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001457 msgbuf[0] = dirc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001458
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001459 if (enc_utf8 && utf_iscomposing(utf_ptr2char(p)))
1460 {
1461 // Use a space to draw the composing char on.
1462 msgbuf[1] = ' ';
1463 mch_memmove(msgbuf + 2, p, STRLEN(p));
1464 }
1465 else
1466 mch_memmove(msgbuf + 1, p, STRLEN(p));
1467 if (off_len > 0)
1468 mch_memmove(msgbuf + STRLEN(p) + 1, off_buf, off_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001469
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001470 trunc = msg_strtrunc(msgbuf, TRUE);
1471 if (trunc != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001472 {
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001473 vim_free(msgbuf);
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001474 msgbuf = trunc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001475 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001476
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001477 #ifdef FEAT_RIGHTLEFT
1478 // The search pattern could be shown on the right in rightleft
1479 // mode, but the 'ruler' and 'showcmd' area use it too, thus
1480 // it would be blanked out again very soon. Show it on the
1481 // left, but do reverse the text.
1482 if (curwin->w_p_rl && *curwin->w_p_rlc == 's')
1483 {
1484 char_u *r;
1485 size_t pat_len;
1486
1487 r = reverse_text(msgbuf);
1488 if (r != NULL)
1489 {
1490 vim_free(msgbuf);
1491 msgbuf = r;
1492 // move reversed text to beginning of buffer
1493 while (*r != NUL && *r == ' ')
1494 r++;
1495 pat_len = msgbuf + STRLEN(msgbuf) - r;
1496 mch_memmove(msgbuf, r, pat_len);
1497 // overwrite old text
1498 if ((size_t)(r - msgbuf) >= pat_len)
1499 vim_memset(r, ' ', pat_len);
1500 else
1501 vim_memset(msgbuf + pat_len, ' ', r - msgbuf);
1502 }
1503 }
1504 #endif
1505 msg_outtrans(msgbuf);
1506 msg_clr_eos();
1507 msg_check();
1508
1509 gotocmdline(FALSE);
1510 out_flush();
1511 msg_nowait = TRUE; // don't wait for this message
1512 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001513 }
1514 }
1515
1516 /*
1517 * If there is a character offset, subtract it from the current
1518 * position, so we don't get stuck at "?pat?e+2" or "/pat/s-2".
Bram Moolenaared203462004-06-16 11:19:22 +00001519 * Skip this if pos.col is near MAXCOL (closed fold).
Bram Moolenaar071d4272004-06-13 20:20:40 +00001520 * This is not done for a line offset, because then we would not be vi
1521 * compatible.
1522 */
Bram Moolenaared203462004-06-16 11:19:22 +00001523 if (!spats[0].off.line && spats[0].off.off && pos.col < MAXCOL - 2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001524 {
1525 if (spats[0].off.off > 0)
1526 {
1527 for (c = spats[0].off.off; c; --c)
1528 if (decl(&pos) == -1)
1529 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001530 if (c) // at start of buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001531 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001532 pos.lnum = 0; // allow lnum == 0 here
Bram Moolenaar071d4272004-06-13 20:20:40 +00001533 pos.col = MAXCOL;
1534 }
1535 }
1536 else
1537 {
1538 for (c = spats[0].off.off; c; ++c)
1539 if (incl(&pos) == -1)
1540 break;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001541 if (c) // at end of buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001542 {
1543 pos.lnum = curbuf->b_ml.ml_line_count + 1;
1544 pos.col = 0;
1545 }
1546 }
1547 }
1548
Bram Moolenaar14184a32019-02-16 15:10:30 +01001549 c = searchit(curwin, curbuf, &pos, NULL,
1550 dirc == '/' ? FORWARD : BACKWARD,
Bram Moolenaar071d4272004-06-13 20:20:40 +00001551 searchstr, count, spats[0].off.end + (options &
1552 (SEARCH_KEEP + SEARCH_PEEK + SEARCH_HIS
1553 + SEARCH_MSG + SEARCH_START
1554 + ((pat != NULL && *pat == ';') ? 0 : SEARCH_NOOF))),
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02001555 RE_LAST, sia);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001556
1557 if (dircp != NULL)
Bram Moolenaarc036e872020-02-21 21:30:52 +01001558 *dircp = search_delim; // restore second '/' or '?' for normal_cmd()
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001559
1560 if (!shortmess(SHM_SEARCH)
1561 && ((dirc == '/' && LT_POS(pos, curwin->w_cursor))
1562 || (dirc == '?' && LT_POS(curwin->w_cursor, pos))))
Bram Moolenaarc7a10b32019-05-06 21:37:18 +02001563 show_top_bot_msg = TRUE;
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001564
Bram Moolenaar071d4272004-06-13 20:20:40 +00001565 if (c == FAIL)
1566 {
1567 retval = 0;
1568 goto end_do_search;
1569 }
1570 if (spats[0].off.end && oap != NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001571 oap->inclusive = TRUE; // 'e' includes last character
Bram Moolenaar071d4272004-06-13 20:20:40 +00001572
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001573 retval = 1; // pattern found
Bram Moolenaar071d4272004-06-13 20:20:40 +00001574
1575 /*
1576 * Add character and/or line offset
1577 */
Bram Moolenaar9160f302006-08-29 15:58:12 +00001578 if (!(options & SEARCH_NOOF) || (pat != NULL && *pat == ';'))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001579 {
Bram Moolenaar8f46e4c2019-05-24 22:08:15 +02001580 pos_T org_pos = pos;
1581
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001582 if (spats[0].off.line) // Add the offset to the line number.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001583 {
1584 c = pos.lnum + spats[0].off.off;
1585 if (c < 1)
1586 pos.lnum = 1;
1587 else if (c > curbuf->b_ml.ml_line_count)
1588 pos.lnum = curbuf->b_ml.ml_line_count;
1589 else
1590 pos.lnum = c;
1591 pos.col = 0;
1592
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001593 retval = 2; // pattern found, line offset added
Bram Moolenaar071d4272004-06-13 20:20:40 +00001594 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001595 else if (pos.col < MAXCOL - 2) // just in case
Bram Moolenaar071d4272004-06-13 20:20:40 +00001596 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001597 // to the right, check for end of file
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001598 c = spats[0].off.off;
1599 if (c > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001600 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001601 while (c-- > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001602 if (incl(&pos) == -1)
1603 break;
1604 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001605 // to the left, check for start of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00001606 else
1607 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00001608 while (c++ < 0)
1609 if (decl(&pos) == -1)
1610 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001611 }
1612 }
Bram Moolenaar8f46e4c2019-05-24 22:08:15 +02001613 if (!EQUAL_POS(pos, org_pos))
1614 has_offset = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001615 }
1616
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001617 // Show [1/15] if 'S' is not in 'shortmess'.
1618 if ((options & SEARCH_ECHO)
1619 && messaging()
Bram Moolenaar359ad1a2019-09-02 21:44:59 +02001620 && !msg_silent
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001621 && c != FAIL
1622 && !shortmess(SHM_SEARCHCOUNT)
1623 && msgbuf != NULL)
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02001624 cmdline_search_stat(dirc, &pos, &curwin->w_cursor,
1625 show_top_bot_msg, msgbuf,
1626 (count != 1 || has_offset
Bram Moolenaar6cb07262020-05-29 22:49:43 +02001627#ifdef FEAT_FOLDING
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02001628 || (!(fdo_flags & FDO_SEARCH)
1629 && hasFolding(curwin->w_cursor.lnum,
1630 NULL, NULL))
Bram Moolenaar6cb07262020-05-29 22:49:43 +02001631#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02001632 ),
1633 SEARCH_STAT_DEF_MAX_COUNT,
1634 SEARCH_STAT_DEF_TIMEOUT);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001635
Bram Moolenaar071d4272004-06-13 20:20:40 +00001636 /*
1637 * The search command can be followed by a ';' to do another search.
1638 * For example: "/pat/;/foo/+3;?bar"
1639 * This is like doing another search command, except:
1640 * - The remembered direction '/' or '?' is from the first search.
1641 * - When an error happens the cursor isn't moved at all.
1642 * Don't do this when called by get_address() (it handles ';' itself).
1643 */
1644 if (!(options & SEARCH_OPT) || pat == NULL || *pat != ';')
1645 break;
1646
1647 dirc = *++pat;
Bram Moolenaarc036e872020-02-21 21:30:52 +01001648 search_delim = dirc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001649 if (dirc != '?' && dirc != '/')
1650 {
1651 retval = 0;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001652 emsg(_("E386: Expected '?' or '/' after ';'"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001653 goto end_do_search;
1654 }
1655 ++pat;
1656 }
1657
1658 if (options & SEARCH_MARK)
1659 setpcmark();
1660 curwin->w_cursor = pos;
1661 curwin->w_set_curswant = TRUE;
1662
1663end_do_search:
Bram Moolenaarac8400d2014-01-14 21:31:34 +01001664 if ((options & SEARCH_KEEP) || cmdmod.keeppatterns)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001665 spats[0].off = old_off;
1666 vim_free(strcopy);
Bram Moolenaar9dfa3132019-05-04 21:08:40 +02001667 vim_free(msgbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001668
1669 return retval;
1670}
1671
Bram Moolenaar071d4272004-06-13 20:20:40 +00001672/*
1673 * search_for_exact_line(buf, pos, dir, pat)
1674 *
1675 * Search for a line starting with the given pattern (ignoring leading
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02001676 * white-space), starting from pos and going in direction "dir". "pos" will
Bram Moolenaar071d4272004-06-13 20:20:40 +00001677 * contain the position of the match found. Blank lines match only if
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02001678 * ADDING is set. If p_ic is set then the pattern must be in lowercase.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001679 * Return OK for success, or FAIL if no line found.
1680 */
1681 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001682search_for_exact_line(
1683 buf_T *buf,
1684 pos_T *pos,
1685 int dir,
1686 char_u *pat)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001687{
1688 linenr_T start = 0;
1689 char_u *ptr;
1690 char_u *p;
1691
1692 if (buf->b_ml.ml_line_count == 0)
1693 return FAIL;
1694 for (;;)
1695 {
1696 pos->lnum += dir;
1697 if (pos->lnum < 1)
1698 {
1699 if (p_ws)
1700 {
1701 pos->lnum = buf->b_ml.ml_line_count;
1702 if (!shortmess(SHM_SEARCH))
1703 give_warning((char_u *)_(top_bot_msg), TRUE);
1704 }
1705 else
1706 {
1707 pos->lnum = 1;
1708 break;
1709 }
1710 }
1711 else if (pos->lnum > buf->b_ml.ml_line_count)
1712 {
1713 if (p_ws)
1714 {
1715 pos->lnum = 1;
1716 if (!shortmess(SHM_SEARCH))
1717 give_warning((char_u *)_(bot_top_msg), TRUE);
1718 }
1719 else
1720 {
1721 pos->lnum = 1;
1722 break;
1723 }
1724 }
1725 if (pos->lnum == start)
1726 break;
1727 if (start == 0)
1728 start = pos->lnum;
1729 ptr = ml_get_buf(buf, pos->lnum, FALSE);
1730 p = skipwhite(ptr);
1731 pos->col = (colnr_T) (p - ptr);
1732
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001733 // when adding lines the matching line may be empty but it is not
1734 // ignored because we are interested in the next line -- Acevedo
Bram Moolenaar4be06f92005-07-29 22:36:03 +00001735 if ((compl_cont_status & CONT_ADDING)
1736 && !(compl_cont_status & CONT_SOL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001737 {
1738 if ((p_ic ? MB_STRICMP(p, pat) : STRCMP(p, pat)) == 0)
1739 return OK;
1740 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001741 else if (*p != NUL) // ignore empty lines
1742 { // expanding lines or words
Bram Moolenaar4be06f92005-07-29 22:36:03 +00001743 if ((p_ic ? MB_STRNICMP(p, pat, compl_length)
1744 : STRNCMP(p, pat, compl_length)) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001745 return OK;
1746 }
1747 }
1748 return FAIL;
1749}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001750
1751/*
1752 * Character Searches
1753 */
1754
1755/*
1756 * Search for a character in a line. If "t_cmd" is FALSE, move to the
1757 * position of the character, otherwise move to just before the char.
1758 * Do this "cap->count1" times.
1759 * Return FAIL or OK.
1760 */
1761 int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001762searchc(cmdarg_T *cap, int t_cmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001763{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001764 int c = cap->nchar; // char to search for
1765 int dir = cap->arg; // TRUE for searching forward
1766 long count = cap->count1; // repeat count
Bram Moolenaar071d4272004-06-13 20:20:40 +00001767 int col;
1768 char_u *p;
1769 int len;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001770 int stop = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001771
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001772 if (c != NUL) // normal search: remember args for repeat
Bram Moolenaar071d4272004-06-13 20:20:40 +00001773 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001774 if (!KeyStuffed) // don't remember when redoing
Bram Moolenaar071d4272004-06-13 20:20:40 +00001775 {
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001776 *lastc = c;
1777 set_csearch_direction(dir);
1778 set_csearch_until(t_cmd);
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001779 lastc_bytelen = (*mb_char2bytes)(c, lastc_bytes);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001780 if (cap->ncharC1 != 0)
1781 {
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001782 lastc_bytelen += (*mb_char2bytes)(cap->ncharC1,
1783 lastc_bytes + lastc_bytelen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001784 if (cap->ncharC2 != 0)
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001785 lastc_bytelen += (*mb_char2bytes)(cap->ncharC2,
1786 lastc_bytes + lastc_bytelen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001787 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001788 }
1789 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001790 else // repeat previous search
Bram Moolenaar071d4272004-06-13 20:20:40 +00001791 {
Bram Moolenaar264b74f2019-01-24 17:18:42 +01001792 if (*lastc == NUL && lastc_bytelen == 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001793 return FAIL;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001794 if (dir) // repeat in opposite direction
Bram Moolenaar071d4272004-06-13 20:20:40 +00001795 dir = -lastcdir;
1796 else
1797 dir = lastcdir;
1798 t_cmd = last_t_cmd;
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001799 c = *lastc;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001800 // For multi-byte re-use last lastc_bytes[] and lastc_bytelen.
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001801
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001802 // Force a move of at least one char, so ";" and "," will move the
1803 // cursor, even if the cursor is right in front of char we are looking
1804 // at.
Bram Moolenaar19fd09a2011-07-15 13:21:30 +02001805 if (vim_strchr(p_cpo, CPO_SCOLON) == NULL && count == 1 && t_cmd)
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001806 stop = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001807 }
1808
Bram Moolenaar60a795a2005-09-16 21:55:43 +00001809 if (dir == BACKWARD)
1810 cap->oap->inclusive = FALSE;
1811 else
1812 cap->oap->inclusive = TRUE;
1813
Bram Moolenaar071d4272004-06-13 20:20:40 +00001814 p = ml_get_curline();
1815 col = curwin->w_cursor.col;
1816 len = (int)STRLEN(p);
1817
1818 while (count--)
1819 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001820 if (has_mbyte)
1821 {
1822 for (;;)
1823 {
1824 if (dir > 0)
1825 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001826 col += (*mb_ptr2len)(p + col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001827 if (col >= len)
1828 return FAIL;
1829 }
1830 else
1831 {
1832 if (col == 0)
1833 return FAIL;
1834 col -= (*mb_head_off)(p, p + col - 1) + 1;
1835 }
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001836 if (lastc_bytelen == 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001837 {
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001838 if (p[col] == c && stop)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001839 break;
1840 }
Bram Moolenaar66727e12017-03-01 22:17:05 +01001841 else if (STRNCMP(p + col, lastc_bytes, lastc_bytelen) == 0
Bram Moolenaarb129a442016-12-01 17:25:20 +01001842 && stop)
Bram Moolenaar66727e12017-03-01 22:17:05 +01001843 break;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001844 stop = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001845 }
1846 }
1847 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001848 {
1849 for (;;)
1850 {
1851 if ((col += dir) < 0 || col >= len)
1852 return FAIL;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001853 if (p[col] == c && stop)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001854 break;
Bram Moolenaar8b3e0332011-06-26 05:36:34 +02001855 stop = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001856 }
1857 }
1858 }
1859
1860 if (t_cmd)
1861 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001862 // backup to before the character (possibly double-byte)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001863 col -= dir;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001864 if (has_mbyte)
1865 {
1866 if (dir < 0)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001867 // Landed on the search char which is lastc_bytelen long
Bram Moolenaardbd24b52015-08-11 14:26:19 +02001868 col += lastc_bytelen - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001869 else
Bram Moolenaar63d9e732019-12-05 21:10:38 +01001870 // To previous char, which may be multi-byte.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001871 col -= (*mb_head_off)(p, p + col);
1872 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001873 }
1874 curwin->w_cursor.col = col;
1875
1876 return OK;
1877}
1878
1879/*
1880 * "Other" Searches
1881 */
1882
1883/*
1884 * findmatch - find the matching paren or brace
1885 *
1886 * Improvement over vi: Braces inside quotes are ignored.
1887 */
1888 pos_T *
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001889findmatch(oparg_T *oap, int initc)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001890{
1891 return findmatchlimit(oap, initc, 0, 0);
1892}
1893
1894/*
1895 * Return TRUE if the character before "linep[col]" equals "ch".
1896 * Return FALSE if "col" is zero.
1897 * Update "*prevcol" to the column of the previous character, unless "prevcol"
1898 * is NULL.
1899 * Handles multibyte string correctly.
1900 */
1901 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001902check_prevcol(
1903 char_u *linep,
1904 int col,
1905 int ch,
1906 int *prevcol)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001907{
1908 --col;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001909 if (col > 0 && has_mbyte)
1910 col -= (*mb_head_off)(linep, linep + col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001911 if (prevcol)
1912 *prevcol = col;
1913 return (col >= 0 && linep[col] == ch) ? TRUE : FALSE;
1914}
1915
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001916/*
1917 * Raw string start is found at linep[startpos.col - 1].
1918 * Return TRUE if the matching end can be found between startpos and endpos.
1919 */
1920 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01001921find_rawstring_end(char_u *linep, pos_T *startpos, pos_T *endpos)
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001922{
1923 char_u *p;
1924 char_u *delim_copy;
1925 size_t delim_len;
1926 linenr_T lnum;
1927 int found = FALSE;
1928
1929 for (p = linep + startpos->col + 1; *p && *p != '('; ++p)
1930 ;
1931 delim_len = (p - linep) - startpos->col - 1;
Bram Moolenaar71ccd032020-06-12 22:59:11 +02001932 delim_copy = vim_strnsave(linep + startpos->col + 1, delim_len);
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001933 if (delim_copy == NULL)
1934 return FALSE;
1935 for (lnum = startpos->lnum; lnum <= endpos->lnum; ++lnum)
1936 {
1937 char_u *line = ml_get(lnum);
1938
1939 for (p = line + (lnum == startpos->lnum
1940 ? startpos->col + 1 : 0); *p; ++p)
1941 {
1942 if (lnum == endpos->lnum && (colnr_T)(p - line) >= endpos->col)
1943 break;
Bram Moolenaar282f9c62020-08-04 21:46:18 +02001944 if (*p == ')' && STRNCMP(delim_copy, p + 1, delim_len) == 0
1945 && p[delim_len + 1] == '"')
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02001946 {
1947 found = TRUE;
1948 break;
1949 }
1950 }
1951 if (found)
1952 break;
1953 }
1954 vim_free(delim_copy);
1955 return found;
1956}
1957
Bram Moolenaar071d4272004-06-13 20:20:40 +00001958/*
Bram Moolenaar556ae8e2019-11-21 22:27:22 +01001959 * Check matchpairs option for "*initc".
1960 * If there is a match set "*initc" to the matching character and "*findc" to
1961 * the opposite character. Set "*backwards" to the direction.
1962 * When "switchit" is TRUE swap the direction.
1963 */
1964 static void
1965find_mps_values(
1966 int *initc,
1967 int *findc,
1968 int *backwards,
1969 int switchit)
1970{
1971 char_u *ptr;
1972
1973 ptr = curbuf->b_p_mps;
1974 while (*ptr != NUL)
1975 {
1976 if (has_mbyte)
1977 {
1978 char_u *prev;
1979
1980 if (mb_ptr2char(ptr) == *initc)
1981 {
1982 if (switchit)
1983 {
1984 *findc = *initc;
1985 *initc = mb_ptr2char(ptr + mb_ptr2len(ptr) + 1);
1986 *backwards = TRUE;
1987 }
1988 else
1989 {
1990 *findc = mb_ptr2char(ptr + mb_ptr2len(ptr) + 1);
1991 *backwards = FALSE;
1992 }
1993 return;
1994 }
1995 prev = ptr;
1996 ptr += mb_ptr2len(ptr) + 1;
1997 if (mb_ptr2char(ptr) == *initc)
1998 {
1999 if (switchit)
2000 {
2001 *findc = *initc;
2002 *initc = mb_ptr2char(prev);
2003 *backwards = FALSE;
2004 }
2005 else
2006 {
2007 *findc = mb_ptr2char(prev);
2008 *backwards = TRUE;
2009 }
2010 return;
2011 }
2012 ptr += mb_ptr2len(ptr);
2013 }
2014 else
2015 {
2016 if (*ptr == *initc)
2017 {
2018 if (switchit)
2019 {
2020 *backwards = TRUE;
2021 *findc = *initc;
2022 *initc = ptr[2];
2023 }
2024 else
2025 {
2026 *backwards = FALSE;
2027 *findc = ptr[2];
2028 }
2029 return;
2030 }
2031 ptr += 2;
2032 if (*ptr == *initc)
2033 {
2034 if (switchit)
2035 {
2036 *backwards = FALSE;
2037 *findc = *initc;
2038 *initc = ptr[-2];
2039 }
2040 else
2041 {
2042 *backwards = TRUE;
2043 *findc = ptr[-2];
2044 }
2045 return;
2046 }
2047 ++ptr;
2048 }
2049 if (*ptr == ',')
2050 ++ptr;
2051 }
2052}
2053
2054/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002055 * findmatchlimit -- find the matching paren or brace, if it exists within
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002056 * maxtravel lines of the cursor. A maxtravel of 0 means search until falling
2057 * off the edge of the file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002058 *
2059 * "initc" is the character to find a match for. NUL means to find the
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002060 * character at or after the cursor. Special values:
2061 * '*' look for C-style comment / *
2062 * '/' look for C-style comment / *, ignoring comment-end
2063 * '#' look for preprocessor directives
2064 * 'R' look for raw string start: R"delim(text)delim" (only backwards)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002065 *
2066 * flags: FM_BACKWARD search backwards (when initc is '/', '*' or '#')
2067 * FM_FORWARD search forwards (when initc is '/', '*' or '#')
2068 * FM_BLOCKSTOP stop at start/end of block ({ or } in column 0)
2069 * FM_SKIPCOMM skip comments (not implemented yet!)
Bram Moolenaarf75a9632005-09-13 21:20:47 +00002070 *
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002071 * "oap" is only used to set oap->motion_type for a linewise motion, it can be
Bram Moolenaarf75a9632005-09-13 21:20:47 +00002072 * NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00002073 */
2074
2075 pos_T *
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002076findmatchlimit(
2077 oparg_T *oap,
2078 int initc,
2079 int flags,
2080 int maxtravel)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002081{
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002082 static pos_T pos; // current search position
2083 int findc = 0; // matching brace
Bram Moolenaar071d4272004-06-13 20:20:40 +00002084 int c;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002085 int count = 0; // cumulative number of braces
2086 int backwards = FALSE; // init for gcc
2087 int raw_string = FALSE; // search for raw string
2088 int inquote = FALSE; // TRUE when inside quotes
2089 char_u *linep; // pointer to current line
Bram Moolenaar071d4272004-06-13 20:20:40 +00002090 char_u *ptr;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002091 int do_quotes; // check for quotes in current line
2092 int at_start; // do_quotes value at start position
2093 int hash_dir = 0; // Direction searched for # things
2094 int comment_dir = 0; // Direction searched for comments
2095 pos_T match_pos; // Where last slash-star was found
2096 int start_in_quotes; // start position is in quotes
2097 int traveled = 0; // how far we've searched so far
2098 int ignore_cend = FALSE; // ignore comment end
2099 int cpo_match; // vi compatible matching
2100 int cpo_bsl; // don't recognize backslashes
2101 int match_escaped = 0; // search for escaped match
2102 int dir; // Direction to search
2103 int comment_col = MAXCOL; // start of / / comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002104#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002105 int lispcomm = FALSE; // inside of Lisp-style comment
2106 int lisp = curbuf->b_p_lisp; // engage Lisp-specific hacks ;)
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002107#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002108
2109 pos = curwin->w_cursor;
Bram Moolenaarc56c4592013-08-14 17:45:29 +02002110 pos.coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002111 linep = ml_get(pos.lnum);
2112
2113 cpo_match = (vim_strchr(p_cpo, CPO_MATCH) != NULL);
2114 cpo_bsl = (vim_strchr(p_cpo, CPO_MATCHBSL) != NULL);
2115
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002116 // Direction to search when initc is '/', '*' or '#'
Bram Moolenaar071d4272004-06-13 20:20:40 +00002117 if (flags & FM_BACKWARD)
2118 dir = BACKWARD;
2119 else if (flags & FM_FORWARD)
2120 dir = FORWARD;
2121 else
2122 dir = 0;
2123
2124 /*
2125 * if initc given, look in the table for the matching character
2126 * '/' and '*' are special cases: look for start or end of comment.
2127 * When '/' is used, we ignore running backwards into an star-slash, for
2128 * "[*" command, we just want to find any comment.
2129 */
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002130 if (initc == '/' || initc == '*' || initc == 'R')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002131 {
2132 comment_dir = dir;
2133 if (initc == '/')
2134 ignore_cend = TRUE;
2135 backwards = (dir == FORWARD) ? FALSE : TRUE;
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002136 raw_string = (initc == 'R');
Bram Moolenaar071d4272004-06-13 20:20:40 +00002137 initc = NUL;
2138 }
2139 else if (initc != '#' && initc != NUL)
2140 {
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002141 find_mps_values(&initc, &findc, &backwards, TRUE);
2142 if (findc == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002143 return NULL;
2144 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002145 else
2146 {
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002147 /*
2148 * Either initc is '#', or no initc was given and we need to look
2149 * under the cursor.
2150 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002151 if (initc == '#')
2152 {
2153 hash_dir = dir;
2154 }
2155 else
2156 {
2157 /*
2158 * initc was not given, must look for something to match under
2159 * or near the cursor.
2160 * Only check for special things when 'cpo' doesn't have '%'.
2161 */
2162 if (!cpo_match)
2163 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002164 // Are we before or at #if, #else etc.?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002165 ptr = skipwhite(linep);
2166 if (*ptr == '#' && pos.col <= (colnr_T)(ptr - linep))
2167 {
2168 ptr = skipwhite(ptr + 1);
2169 if ( STRNCMP(ptr, "if", 2) == 0
2170 || STRNCMP(ptr, "endif", 5) == 0
2171 || STRNCMP(ptr, "el", 2) == 0)
2172 hash_dir = 1;
2173 }
2174
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002175 // Are we on a comment?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002176 else if (linep[pos.col] == '/')
2177 {
2178 if (linep[pos.col + 1] == '*')
2179 {
2180 comment_dir = FORWARD;
2181 backwards = FALSE;
2182 pos.col++;
2183 }
2184 else if (pos.col > 0 && linep[pos.col - 1] == '*')
2185 {
2186 comment_dir = BACKWARD;
2187 backwards = TRUE;
2188 pos.col--;
2189 }
2190 }
2191 else if (linep[pos.col] == '*')
2192 {
2193 if (linep[pos.col + 1] == '/')
2194 {
2195 comment_dir = BACKWARD;
2196 backwards = TRUE;
2197 }
2198 else if (pos.col > 0 && linep[pos.col - 1] == '/')
2199 {
2200 comment_dir = FORWARD;
2201 backwards = FALSE;
2202 }
2203 }
2204 }
2205
2206 /*
2207 * If we are not on a comment or the # at the start of a line, then
2208 * look for brace anywhere on this line after the cursor.
2209 */
2210 if (!hash_dir && !comment_dir)
2211 {
2212 /*
2213 * Find the brace under or after the cursor.
2214 * If beyond the end of the line, use the last character in
2215 * the line.
2216 */
2217 if (linep[pos.col] == NUL && pos.col)
2218 --pos.col;
2219 for (;;)
2220 {
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002221 initc = PTR2CHAR(linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002222 if (initc == NUL)
2223 break;
2224
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002225 find_mps_values(&initc, &findc, &backwards, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002226 if (findc)
2227 break;
Bram Moolenaar1614a142019-10-06 22:00:13 +02002228 pos.col += mb_ptr2len(linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002229 }
2230 if (!findc)
2231 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002232 // no brace in the line, maybe use " #if" then
Bram Moolenaar071d4272004-06-13 20:20:40 +00002233 if (!cpo_match && *skipwhite(linep) == '#')
2234 hash_dir = 1;
2235 else
2236 return NULL;
2237 }
2238 else if (!cpo_bsl)
2239 {
2240 int col, bslcnt = 0;
2241
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002242 // Set "match_escaped" if there are an odd number of
2243 // backslashes.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002244 for (col = pos.col; check_prevcol(linep, col, '\\', &col);)
2245 bslcnt++;
2246 match_escaped = (bslcnt & 1);
2247 }
2248 }
2249 }
2250 if (hash_dir)
2251 {
2252 /*
2253 * Look for matching #if, #else, #elif, or #endif
2254 */
2255 if (oap != NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002256 oap->motion_type = MLINE; // Linewise for this case only
Bram Moolenaar071d4272004-06-13 20:20:40 +00002257 if (initc != '#')
2258 {
2259 ptr = skipwhite(skipwhite(linep) + 1);
2260 if (STRNCMP(ptr, "if", 2) == 0 || STRNCMP(ptr, "el", 2) == 0)
2261 hash_dir = 1;
2262 else if (STRNCMP(ptr, "endif", 5) == 0)
2263 hash_dir = -1;
2264 else
2265 return NULL;
2266 }
2267 pos.col = 0;
2268 while (!got_int)
2269 {
2270 if (hash_dir > 0)
2271 {
2272 if (pos.lnum == curbuf->b_ml.ml_line_count)
2273 break;
2274 }
2275 else if (pos.lnum == 1)
2276 break;
2277 pos.lnum += hash_dir;
2278 linep = ml_get(pos.lnum);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002279 line_breakcheck(); // check for CTRL-C typed
Bram Moolenaar071d4272004-06-13 20:20:40 +00002280 ptr = skipwhite(linep);
2281 if (*ptr != '#')
2282 continue;
2283 pos.col = (colnr_T) (ptr - linep);
2284 ptr = skipwhite(ptr + 1);
2285 if (hash_dir > 0)
2286 {
2287 if (STRNCMP(ptr, "if", 2) == 0)
2288 count++;
2289 else if (STRNCMP(ptr, "el", 2) == 0)
2290 {
2291 if (count == 0)
2292 return &pos;
2293 }
2294 else if (STRNCMP(ptr, "endif", 5) == 0)
2295 {
2296 if (count == 0)
2297 return &pos;
2298 count--;
2299 }
2300 }
2301 else
2302 {
2303 if (STRNCMP(ptr, "if", 2) == 0)
2304 {
2305 if (count == 0)
2306 return &pos;
2307 count--;
2308 }
2309 else if (initc == '#' && STRNCMP(ptr, "el", 2) == 0)
2310 {
2311 if (count == 0)
2312 return &pos;
2313 }
2314 else if (STRNCMP(ptr, "endif", 5) == 0)
2315 count++;
2316 }
2317 }
2318 return NULL;
2319 }
2320 }
2321
2322#ifdef FEAT_RIGHTLEFT
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002323 // This is just guessing: when 'rightleft' is set, search for a matching
2324 // paren/brace in the other direction.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002325 if (curwin->w_p_rl && vim_strchr((char_u *)"()[]{}<>", initc) != NULL)
2326 backwards = !backwards;
2327#endif
2328
2329 do_quotes = -1;
2330 start_in_quotes = MAYBE;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01002331 CLEAR_POS(&match_pos);
Bram Moolenaarfd2ac762006-03-01 22:09:21 +00002332
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002333 // backward search: Check if this line contains a single-line comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002334 if ((backwards && comment_dir)
2335#ifdef FEAT_LISP
2336 || lisp
2337#endif
2338 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002339 comment_col = check_linecomment(linep);
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002340#ifdef FEAT_LISP
2341 if (lisp && comment_col != MAXCOL && pos.col > (colnr_T)comment_col)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002342 lispcomm = TRUE; // find match inside this comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002343#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002344 while (!got_int)
2345 {
2346 /*
2347 * Go to the next position, forward or backward. We could use
2348 * inc() and dec() here, but that is much slower
2349 */
2350 if (backwards)
2351 {
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002352#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002353 // char to match is inside of comment, don't search outside
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002354 if (lispcomm && pos.col < (colnr_T)comment_col)
2355 break;
2356#endif
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002357 if (pos.col == 0) // at start of line, go to prev. one
Bram Moolenaar071d4272004-06-13 20:20:40 +00002358 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002359 if (pos.lnum == 1) // start of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00002360 break;
2361 --pos.lnum;
2362
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002363 if (maxtravel > 0 && ++traveled > maxtravel)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002364 break;
2365
2366 linep = ml_get(pos.lnum);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002367 pos.col = (colnr_T)STRLEN(linep); // pos.col on trailing NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00002368 do_quotes = -1;
2369 line_breakcheck();
2370
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002371 // Check if this line contains a single-line comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002372 if (comment_dir
2373#ifdef FEAT_LISP
2374 || lisp
2375#endif
2376 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002377 comment_col = check_linecomment(linep);
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002378#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002379 // skip comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002380 if (lisp && comment_col != MAXCOL)
2381 pos.col = comment_col;
2382#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002383 }
2384 else
2385 {
2386 --pos.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002387 if (has_mbyte)
2388 pos.col -= (*mb_head_off)(linep, linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002389 }
2390 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002391 else // forward search
Bram Moolenaar071d4272004-06-13 20:20:40 +00002392 {
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002393 if (linep[pos.col] == NUL
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002394 // at end of line, go to next one
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002395#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002396 // don't search for match in comment
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002397 || (lisp && comment_col != MAXCOL
2398 && pos.col == (colnr_T)comment_col)
2399#endif
2400 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002401 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002402 if (pos.lnum == curbuf->b_ml.ml_line_count // end of file
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002403#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002404 // line is exhausted and comment with it,
2405 // don't search for match in code
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002406 || lispcomm
2407#endif
2408 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002409 break;
2410 ++pos.lnum;
2411
2412 if (maxtravel && traveled++ > maxtravel)
2413 break;
2414
2415 linep = ml_get(pos.lnum);
2416 pos.col = 0;
2417 do_quotes = -1;
2418 line_breakcheck();
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002419#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002420 if (lisp) // find comment pos in new line
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002421 comment_col = check_linecomment(linep);
2422#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002423 }
2424 else
2425 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002426 if (has_mbyte)
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00002427 pos.col += (*mb_ptr2len)(linep + pos.col);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002428 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00002429 ++pos.col;
2430 }
2431 }
2432
2433 /*
2434 * If FM_BLOCKSTOP given, stop at a '{' or '}' in column 0.
2435 */
2436 if (pos.col == 0 && (flags & FM_BLOCKSTOP) &&
2437 (linep[0] == '{' || linep[0] == '}'))
2438 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002439 if (linep[0] == findc && count == 0) // match!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002440 return &pos;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002441 break; // out of scope
Bram Moolenaar071d4272004-06-13 20:20:40 +00002442 }
2443
2444 if (comment_dir)
2445 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002446 // Note: comments do not nest, and we ignore quotes in them
2447 // TODO: ignore comment brackets inside strings
Bram Moolenaar071d4272004-06-13 20:20:40 +00002448 if (comment_dir == FORWARD)
2449 {
2450 if (linep[pos.col] == '*' && linep[pos.col + 1] == '/')
2451 {
2452 pos.col++;
2453 return &pos;
2454 }
2455 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002456 else // Searching backwards
Bram Moolenaar071d4272004-06-13 20:20:40 +00002457 {
2458 /*
2459 * A comment may contain / * or / /, it may also start or end
Bram Moolenaarf8c53d32017-11-12 15:36:38 +01002460 * with / * /. Ignore a / * after / / and after *.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002461 */
2462 if (pos.col == 0)
2463 continue;
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002464 else if (raw_string)
2465 {
2466 if (linep[pos.col - 1] == 'R'
2467 && linep[pos.col] == '"'
2468 && vim_strchr(linep + pos.col + 1, '(') != NULL)
2469 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002470 // Possible start of raw string. Now that we have the
2471 // delimiter we can check if it ends before where we
2472 // started searching, or before the previously found
2473 // raw string start.
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002474 if (!find_rawstring_end(linep, &pos,
2475 count > 0 ? &match_pos : &curwin->w_cursor))
2476 {
2477 count++;
2478 match_pos = pos;
2479 match_pos.col--;
2480 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002481 linep = ml_get(pos.lnum); // may have been released
Bram Moolenaarf7bb86d2015-07-28 21:17:36 +02002482 }
2483 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002484 else if ( linep[pos.col - 1] == '/'
2485 && linep[pos.col] == '*'
Bram Moolenaarf8c53d32017-11-12 15:36:38 +01002486 && (pos.col == 1 || linep[pos.col - 2] != '*')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002487 && (int)pos.col < comment_col)
2488 {
2489 count++;
2490 match_pos = pos;
2491 match_pos.col--;
2492 }
2493 else if (linep[pos.col - 1] == '*' && linep[pos.col] == '/')
2494 {
2495 if (count > 0)
2496 pos = match_pos;
2497 else if (pos.col > 1 && linep[pos.col - 2] == '/'
2498 && (int)pos.col <= comment_col)
2499 pos.col -= 2;
2500 else if (ignore_cend)
2501 continue;
2502 else
2503 return NULL;
2504 return &pos;
2505 }
2506 }
2507 continue;
2508 }
2509
2510 /*
2511 * If smart matching ('cpoptions' does not contain '%'), braces inside
2512 * of quotes are ignored, but only if there is an even number of
2513 * quotes in the line.
2514 */
2515 if (cpo_match)
2516 do_quotes = 0;
2517 else if (do_quotes == -1)
2518 {
2519 /*
2520 * Count the number of quotes in the line, skipping \" and '"'.
2521 * Watch out for "\\".
2522 */
2523 at_start = do_quotes;
2524 for (ptr = linep; *ptr; ++ptr)
2525 {
2526 if (ptr == linep + pos.col + backwards)
2527 at_start = (do_quotes & 1);
2528 if (*ptr == '"'
2529 && (ptr == linep || ptr[-1] != '\'' || ptr[1] != '\''))
2530 ++do_quotes;
2531 if (*ptr == '\\' && ptr[1] != NUL)
2532 ++ptr;
2533 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002534 do_quotes &= 1; // result is 1 with even number of quotes
Bram Moolenaar071d4272004-06-13 20:20:40 +00002535
2536 /*
2537 * If we find an uneven count, check current line and previous
2538 * one for a '\' at the end.
2539 */
2540 if (!do_quotes)
2541 {
2542 inquote = FALSE;
2543 if (ptr[-1] == '\\')
2544 {
2545 do_quotes = 1;
2546 if (start_in_quotes == MAYBE)
2547 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002548 // Do we need to use at_start here?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002549 inquote = TRUE;
2550 start_in_quotes = TRUE;
2551 }
2552 else if (backwards)
2553 inquote = TRUE;
2554 }
2555 if (pos.lnum > 1)
2556 {
2557 ptr = ml_get(pos.lnum - 1);
2558 if (*ptr && *(ptr + STRLEN(ptr) - 1) == '\\')
2559 {
2560 do_quotes = 1;
2561 if (start_in_quotes == MAYBE)
2562 {
2563 inquote = at_start;
2564 if (inquote)
2565 start_in_quotes = TRUE;
2566 }
2567 else if (!backwards)
2568 inquote = TRUE;
2569 }
Bram Moolenaaraec11792007-07-10 11:09:36 +00002570
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002571 // ml_get() only keeps one line, need to get linep again
Bram Moolenaaraec11792007-07-10 11:09:36 +00002572 linep = ml_get(pos.lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002573 }
2574 }
2575 }
2576 if (start_in_quotes == MAYBE)
2577 start_in_quotes = FALSE;
2578
2579 /*
2580 * If 'smartmatch' is set:
2581 * Things inside quotes are ignored by setting 'inquote'. If we
2582 * find a quote without a preceding '\' invert 'inquote'. At the
2583 * end of a line not ending in '\' we reset 'inquote'.
2584 *
2585 * In lines with an uneven number of quotes (without preceding '\')
2586 * we do not know which part to ignore. Therefore we only set
2587 * inquote if the number of quotes in a line is even, unless this
2588 * line or the previous one ends in a '\'. Complicated, isn't it?
2589 */
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002590 c = PTR2CHAR(linep + pos.col);
2591 switch (c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002592 {
2593 case NUL:
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002594 // at end of line without trailing backslash, reset inquote
Bram Moolenaar071d4272004-06-13 20:20:40 +00002595 if (pos.col == 0 || linep[pos.col - 1] != '\\')
2596 {
2597 inquote = FALSE;
2598 start_in_quotes = FALSE;
2599 }
2600 break;
2601
2602 case '"':
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002603 // a quote that is preceded with an odd number of backslashes is
2604 // ignored
Bram Moolenaar071d4272004-06-13 20:20:40 +00002605 if (do_quotes)
2606 {
2607 int col;
2608
2609 for (col = pos.col - 1; col >= 0; --col)
2610 if (linep[col] != '\\')
2611 break;
2612 if ((((int)pos.col - 1 - col) & 1) == 0)
2613 {
2614 inquote = !inquote;
2615 start_in_quotes = FALSE;
2616 }
2617 }
2618 break;
2619
2620 /*
2621 * If smart matching ('cpoptions' does not contain '%'):
2622 * Skip things in single quotes: 'x' or '\x'. Be careful for single
2623 * single quotes, eg jon's. Things like '\233' or '\x3f' are not
2624 * skipped, there is never a brace in them.
2625 * Ignore this when finding matches for `'.
2626 */
2627 case '\'':
2628 if (!cpo_match && initc != '\'' && findc != '\'')
2629 {
2630 if (backwards)
2631 {
2632 if (pos.col > 1)
2633 {
2634 if (linep[pos.col - 2] == '\'')
2635 {
2636 pos.col -= 2;
2637 break;
2638 }
2639 else if (linep[pos.col - 2] == '\\' &&
2640 pos.col > 2 && linep[pos.col - 3] == '\'')
2641 {
2642 pos.col -= 3;
2643 break;
2644 }
2645 }
2646 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002647 else if (linep[pos.col + 1]) // forward search
Bram Moolenaar071d4272004-06-13 20:20:40 +00002648 {
2649 if (linep[pos.col + 1] == '\\' &&
2650 linep[pos.col + 2] && linep[pos.col + 3] == '\'')
2651 {
2652 pos.col += 3;
2653 break;
2654 }
2655 else if (linep[pos.col + 2] == '\'')
2656 {
2657 pos.col += 2;
2658 break;
2659 }
2660 }
2661 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002662 // FALLTHROUGH
Bram Moolenaar071d4272004-06-13 20:20:40 +00002663
2664 default:
2665#ifdef FEAT_LISP
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002666 /*
2667 * For Lisp skip over backslashed (), {} and [].
2668 * (actually, we skip #\( et al)
2669 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002670 if (curbuf->b_p_lisp
2671 && vim_strchr((char_u *)"(){}[]", c) != NULL
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002672 && pos.col > 1
2673 && check_prevcol(linep, pos.col, '\\', NULL)
2674 && check_prevcol(linep, pos.col - 1, '#', NULL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002675 break;
2676#endif
2677
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002678 // Check for match outside of quotes, and inside of
2679 // quotes when the start is also inside of quotes.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002680 if ((!inquote || start_in_quotes == TRUE)
2681 && (c == initc || c == findc))
2682 {
2683 int col, bslcnt = 0;
2684
2685 if (!cpo_bsl)
2686 {
2687 for (col = pos.col; check_prevcol(linep, col, '\\', &col);)
2688 bslcnt++;
2689 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002690 // Only accept a match when 'M' is in 'cpo' or when escaping
2691 // is what we expect.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002692 if (cpo_bsl || (bslcnt & 1) == match_escaped)
2693 {
2694 if (c == initc)
2695 count++;
2696 else
2697 {
2698 if (count == 0)
2699 return &pos;
2700 count--;
2701 }
2702 }
2703 }
2704 }
2705 }
2706
2707 if (comment_dir == BACKWARD && count > 0)
2708 {
2709 pos = match_pos;
2710 return &pos;
2711 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002712 return (pos_T *)NULL; // never found it
Bram Moolenaar071d4272004-06-13 20:20:40 +00002713}
2714
2715/*
2716 * Check if line[] contains a / / comment.
2717 * Return MAXCOL if not, otherwise return the column.
2718 * TODO: skip strings.
2719 */
2720 static int
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002721check_linecomment(char_u *line)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002722{
2723 char_u *p;
2724
2725 p = line;
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002726#ifdef FEAT_LISP
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002727 // skip Lispish one-line comments
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002728 if (curbuf->b_p_lisp)
2729 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002730 if (vim_strchr(p, ';') != NULL) // there may be comments
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002731 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002732 int in_str = FALSE; // inside of string
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002733
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002734 p = line; // scan from start
Bram Moolenaar520470a2005-06-16 21:59:56 +00002735 while ((p = vim_strpbrk(p, (char_u *)"\";")) != NULL)
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002736 {
2737 if (*p == '"')
2738 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002739 if (in_str)
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002740 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002741 if (*(p - 1) != '\\') // skip escaped quote
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002742 in_str = FALSE;
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002743 }
2744 else if (p == line || ((p - line) >= 2
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002745 // skip #\" form
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002746 && *(p - 1) != '\\' && *(p - 2) != '#'))
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002747 in_str = TRUE;
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002748 }
Bram Moolenaar70b2a562012-01-10 22:26:17 +01002749 else if (!in_str && ((p - line) < 2
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002750 || (*(p - 1) != '\\' && *(p - 2) != '#')))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002751 break; // found!
Bram Moolenaar325b7a22004-07-05 15:58:32 +00002752 ++p;
2753 }
2754 }
2755 else
2756 p = NULL;
2757 }
2758 else
2759#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002760 while ((p = vim_strchr(p, '/')) != NULL)
2761 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002762 // accept a double /, unless it's preceded with * and followed by *,
2763 // because * / / * is an end and start of a C comment
Bram Moolenaar78d4aba2008-01-01 14:43:35 +00002764 if (p[1] == '/' && (p == line || p[-1] != '*' || p[2] != '*'))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002765 break;
2766 ++p;
2767 }
2768
2769 if (p == NULL)
2770 return MAXCOL;
2771 return (int)(p - line);
2772}
2773
2774/*
2775 * Move cursor briefly to character matching the one under the cursor.
2776 * Used for Insert mode and "r" command.
2777 * Show the match only if it is visible on the screen.
2778 * If there isn't a match, then beep.
2779 */
2780 void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01002781showmatch(
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002782 int c) // char to show match for
Bram Moolenaar071d4272004-06-13 20:20:40 +00002783{
2784 pos_T *lpos, save_cursor;
2785 pos_T mpos;
2786 colnr_T vcol;
2787 long save_so;
2788 long save_siso;
2789#ifdef CURSOR_SHAPE
2790 int save_state;
2791#endif
2792 colnr_T save_dollar_vcol;
2793 char_u *p;
Bram Moolenaar375e3392019-01-31 18:26:10 +01002794 long *so = curwin->w_p_so >= 0 ? &curwin->w_p_so : &p_so;
2795 long *siso = curwin->w_p_siso >= 0 ? &curwin->w_p_siso : &p_siso;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002796
2797 /*
2798 * Only show match for chars in the 'matchpairs' option.
2799 */
Bram Moolenaar63d9e732019-12-05 21:10:38 +01002800 // 'matchpairs' is "x:y,x:y"
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002801 for (p = curbuf->b_p_mps; *p != NUL; ++p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002802 {
2803#ifdef FEAT_RIGHTLEFT
Bram Moolenaar187d3ac2013-02-20 18:39:13 +01002804 if (PTR2CHAR(p) == c && (curwin->w_p_rl ^ p_ri))
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002805 break;
Bram Moolenaar187d3ac2013-02-20 18:39:13 +01002806#endif
Bram Moolenaar1614a142019-10-06 22:00:13 +02002807 p += mb_ptr2len(p) + 1;
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002808 if (PTR2CHAR(p) == c
Bram Moolenaar071d4272004-06-13 20:20:40 +00002809#ifdef FEAT_RIGHTLEFT
2810 && !(curwin->w_p_rl ^ p_ri)
2811#endif
2812 )
2813 break;
Bram Moolenaar1614a142019-10-06 22:00:13 +02002814 p += mb_ptr2len(p);
Bram Moolenaar8c7694a2013-01-17 17:02:05 +01002815 if (*p == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002816 return;
2817 }
2818
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.
Bram Moolenaarb4b0a082011-02-25 18:38:36 +01003319 * If p_ic && (compl_cont_status & CONT_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
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003377 // when CONT_SOL is set compare "ptr" with the beginning of the line
3378 // is faster than quote_meta/regcomp/regexec "ptr" -- Acevedo
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003379 && !(compl_cont_status & CONT_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);
3387 regmatch.regprog = vim_regcomp(pat, p_magic ? RE_MAGIC : 0);
3388 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 Moolenaar482aaeb2005-09-29 18:26:07 +00003395 incl_regmatch.regprog = vim_regcomp(inc_opt, p_magic ? RE_MAGIC : 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003396 if (incl_regmatch.regprog == NULL)
3397 goto fpip_end;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003398 incl_regmatch.rm_ic = FALSE; // don't ignore case in incl. pat.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003399 }
3400 if (type == FIND_DEFINE && (*curbuf->b_p_def != NUL || *p_def != NUL))
3401 {
3402 def_regmatch.regprog = vim_regcomp(*curbuf->b_p_def == NUL
3403 ? p_def : curbuf->b_p_def, p_magic ? RE_MAGIC : 0);
3404 if (def_regmatch.regprog == NULL)
3405 goto fpip_end;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003406 def_regmatch.rm_ic = FALSE; // don't ignore case in define pat.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003407 }
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003408 files = lalloc_clear(max_path_depth * sizeof(SearchedFile), TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003409 if (files == NULL)
3410 goto fpip_end;
3411 old_files = max_path_depth;
3412 depth = depth_displayed = -1;
3413
3414 lnum = start_lnum;
3415 if (end_lnum > curbuf->b_ml.ml_line_count)
3416 end_lnum = curbuf->b_ml.ml_line_count;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003417 if (lnum > end_lnum) // do at least one line
Bram Moolenaar071d4272004-06-13 20:20:40 +00003418 lnum = end_lnum;
3419 line = ml_get(lnum);
3420
3421 for (;;)
3422 {
3423 if (incl_regmatch.regprog != NULL
3424 && vim_regexec(&incl_regmatch, line, (colnr_T)0))
3425 {
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003426 char_u *p_fname = (curr_fname == curbuf->b_fname)
3427 ? curbuf->b_ffname : curr_fname;
3428
3429 if (inc_opt != NULL && strstr((char *)inc_opt, "\\zs") != NULL)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003430 // Use text from '\zs' to '\ze' (or end) of 'include'.
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003431 new_fname = find_file_name_in_path(incl_regmatch.startp[0],
Bram Moolenaarc84e3c12013-07-03 22:28:36 +02003432 (int)(incl_regmatch.endp[0] - incl_regmatch.startp[0]),
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003433 FNAME_EXP|FNAME_INCL|FNAME_REL, 1L, p_fname);
3434 else
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003435 // Use text after match with 'include'.
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00003436 new_fname = file_name_in_line(incl_regmatch.endp[0], 0,
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003437 FNAME_EXP|FNAME_INCL|FNAME_REL, 1L, p_fname, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003438 already_searched = FALSE;
3439 if (new_fname != NULL)
3440 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003441 // Check whether we have already searched in this file
Bram Moolenaar071d4272004-06-13 20:20:40 +00003442 for (i = 0;; i++)
3443 {
3444 if (i == depth + 1)
3445 i = old_files;
3446 if (i == max_path_depth)
3447 break;
Bram Moolenaar99499b12019-05-23 21:35:48 +02003448 if (fullpathcmp(new_fname, files[i].name, TRUE, TRUE)
3449 & FPC_SAME)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003450 {
3451 if (type != CHECK_PATH &&
3452 action == ACTION_SHOW_ALL && files[i].matched)
3453 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003454 msg_putchar('\n'); // cursor below last one
3455 if (!got_int) // don't display if 'q'
3456 // typed at "--more--"
3457 // message
Bram Moolenaar071d4272004-06-13 20:20:40 +00003458 {
3459 msg_home_replace_hl(new_fname);
Bram Moolenaar32526b32019-01-19 17:43:09 +01003460 msg_puts(_(" (includes previously listed match)"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003461 prev_fname = NULL;
3462 }
3463 }
Bram Moolenaard23a8232018-02-10 18:45:26 +01003464 VIM_CLEAR(new_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003465 already_searched = TRUE;
3466 break;
3467 }
3468 }
3469 }
3470
3471 if (type == CHECK_PATH && (action == ACTION_SHOW_ALL
3472 || (new_fname == NULL && !already_searched)))
3473 {
3474 if (did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003475 msg_putchar('\n'); // cursor below last one
Bram Moolenaar071d4272004-06-13 20:20:40 +00003476 else
3477 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003478 gotocmdline(TRUE); // cursor at status line
Bram Moolenaar32526b32019-01-19 17:43:09 +01003479 msg_puts_title(_("--- Included files "));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003480 if (action != ACTION_SHOW_ALL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003481 msg_puts_title(_("not found "));
3482 msg_puts_title(_("in path ---\n"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003483 }
3484 did_show = TRUE;
3485 while (depth_displayed < depth && !got_int)
3486 {
3487 ++depth_displayed;
3488 for (i = 0; i < depth_displayed; i++)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003489 msg_puts(" ");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003490 msg_home_replace(files[depth_displayed].name);
Bram Moolenaar32526b32019-01-19 17:43:09 +01003491 msg_puts(" -->\n");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003492 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003493 if (!got_int) // don't display if 'q' typed
3494 // for "--more--" message
Bram Moolenaar071d4272004-06-13 20:20:40 +00003495 {
3496 for (i = 0; i <= depth_displayed; i++)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003497 msg_puts(" ");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003498 if (new_fname != NULL)
3499 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003500 // using "new_fname" is more reliable, e.g., when
3501 // 'includeexpr' is set.
Bram Moolenaar8820b482017-03-16 17:23:31 +01003502 msg_outtrans_attr(new_fname, HL_ATTR(HLF_D));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003503 }
3504 else
3505 {
3506 /*
3507 * Isolate the file name.
3508 * Include the surrounding "" or <> if present.
3509 */
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003510 if (inc_opt != NULL
3511 && strstr((char *)inc_opt, "\\zs") != NULL)
3512 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003513 // pattern contains \zs, use the match
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003514 p = incl_regmatch.startp[0];
3515 i = (int)(incl_regmatch.endp[0]
3516 - incl_regmatch.startp[0]);
3517 }
3518 else
3519 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003520 // find the file name after the end of the match
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003521 for (p = incl_regmatch.endp[0];
3522 *p && !vim_isfilec(*p); p++)
3523 ;
3524 for (i = 0; vim_isfilec(p[i]); i++)
3525 ;
3526 }
3527
Bram Moolenaar071d4272004-06-13 20:20:40 +00003528 if (i == 0)
3529 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003530 // Nothing found, use the rest of the line.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003531 p = incl_regmatch.endp[0];
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00003532 i = (int)STRLEN(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003533 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003534 // Avoid checking before the start of the line, can
3535 // happen if \zs appears in the regexp.
Bram Moolenaar058bdcf2012-07-25 13:46:30 +02003536 else if (p > line)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003537 {
3538 if (p[-1] == '"' || p[-1] == '<')
3539 {
3540 --p;
3541 ++i;
3542 }
3543 if (p[i] == '"' || p[i] == '>')
3544 ++i;
3545 }
3546 save_char = p[i];
3547 p[i] = NUL;
Bram Moolenaar8820b482017-03-16 17:23:31 +01003548 msg_outtrans_attr(p, HL_ATTR(HLF_D));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003549 p[i] = save_char;
3550 }
3551
3552 if (new_fname == NULL && action == ACTION_SHOW_ALL)
3553 {
3554 if (already_searched)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003555 msg_puts(_(" (Already listed)"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003556 else
Bram Moolenaar32526b32019-01-19 17:43:09 +01003557 msg_puts(_(" NOT FOUND"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003558 }
3559 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003560 out_flush(); // output each line directly
Bram Moolenaar071d4272004-06-13 20:20:40 +00003561 }
3562
3563 if (new_fname != NULL)
3564 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003565 // Push the new file onto the file stack
Bram Moolenaar071d4272004-06-13 20:20:40 +00003566 if (depth + 1 == old_files)
3567 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003568 bigger = ALLOC_MULT(SearchedFile, max_path_depth * 2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003569 if (bigger != NULL)
3570 {
3571 for (i = 0; i <= depth; i++)
3572 bigger[i] = files[i];
3573 for (i = depth + 1; i < old_files + max_path_depth; i++)
3574 {
3575 bigger[i].fp = NULL;
3576 bigger[i].name = NULL;
3577 bigger[i].lnum = 0;
3578 bigger[i].matched = FALSE;
3579 }
3580 for (i = old_files; i < max_path_depth; i++)
3581 bigger[i + max_path_depth] = files[i];
3582 old_files += max_path_depth;
3583 max_path_depth *= 2;
3584 vim_free(files);
3585 files = bigger;
3586 }
3587 }
3588 if ((files[depth + 1].fp = mch_fopen((char *)new_fname, "r"))
3589 == NULL)
3590 vim_free(new_fname);
3591 else
3592 {
3593 if (++depth == old_files)
3594 {
3595 /*
3596 * lalloc() for 'bigger' must have failed above. We
3597 * will forget one of our already visited files now.
3598 */
3599 vim_free(files[old_files].name);
3600 ++old_files;
3601 }
3602 files[depth].name = curr_fname = new_fname;
3603 files[depth].lnum = 0;
3604 files[depth].matched = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003605 if (action == ACTION_EXPAND)
3606 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003607 msg_hist_off = TRUE; // reset in msg_trunc_attr()
Bram Moolenaar555b2802005-05-19 21:08:39 +00003608 vim_snprintf((char*)IObuff, IOSIZE,
3609 _("Scanning included file: %s"),
3610 (char *)new_fname);
Bram Moolenaar32526b32019-01-19 17:43:09 +01003611 msg_trunc_attr((char *)IObuff, TRUE, HL_ATTR(HLF_R));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003612 }
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003613 else if (p_verbose >= 5)
Bram Moolenaar87b5ca52006-03-04 21:55:31 +00003614 {
3615 verbose_enter();
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003616 smsg(_("Searching included file %s"),
Bram Moolenaar87b5ca52006-03-04 21:55:31 +00003617 (char *)new_fname);
3618 verbose_leave();
3619 }
3620
Bram Moolenaar071d4272004-06-13 20:20:40 +00003621 }
3622 }
3623 }
3624 else
3625 {
3626 /*
3627 * Check if the line is a define (type == FIND_DEFINE)
3628 */
3629 p = line;
3630search_line:
3631 define_matched = FALSE;
3632 if (def_regmatch.regprog != NULL
3633 && vim_regexec(&def_regmatch, line, (colnr_T)0))
3634 {
3635 /*
3636 * Pattern must be first identifier after 'define', so skip
3637 * to that position before checking for match of pattern. Also
3638 * don't let it match beyond the end of this identifier.
3639 */
3640 p = def_regmatch.endp[0];
3641 while (*p && !vim_iswordc(*p))
3642 p++;
3643 define_matched = TRUE;
3644 }
3645
3646 /*
3647 * Look for a match. Don't do this if we are looking for a
3648 * define and this line didn't match define_prog above.
3649 */
3650 if (def_regmatch.regprog == NULL || define_matched)
3651 {
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003652 if (define_matched || (compl_cont_status & CONT_SOL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003653 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003654 // compare the first "len" chars from "ptr"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003655 startp = skipwhite(p);
3656 if (p_ic)
3657 matched = !MB_STRNICMP(startp, ptr, len);
3658 else
3659 matched = !STRNCMP(startp, ptr, len);
3660 if (matched && define_matched && whole
3661 && vim_iswordc(startp[len]))
3662 matched = FALSE;
3663 }
3664 else if (regmatch.regprog != NULL
3665 && vim_regexec(&regmatch, line, (colnr_T)(p - line)))
3666 {
3667 matched = TRUE;
3668 startp = regmatch.startp[0];
3669 /*
3670 * Check if the line is not a comment line (unless we are
3671 * looking for a define). A line starting with "# define"
3672 * is not considered to be a comment line.
3673 */
3674 if (!define_matched && skip_comments)
3675 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003676 if ((*line != '#' ||
3677 STRNCMP(skipwhite(line + 1), "define", 6) != 0)
Bram Moolenaar81340392012-06-06 16:12:59 +02003678 && get_leader_len(line, NULL, FALSE, TRUE))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003679 matched = FALSE;
3680
3681 /*
3682 * Also check for a "/ *" or "/ /" before the match.
3683 * Skips lines like "int backwards; / * normal index
3684 * * /" when looking for "normal".
3685 * Note: Doesn't skip "/ *" in comments.
3686 */
3687 p = skipwhite(line);
3688 if (matched
3689 || (p[0] == '/' && p[1] == '*') || p[0] == '*')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003690 for (p = line; *p && p < startp; ++p)
3691 {
3692 if (matched
3693 && p[0] == '/'
3694 && (p[1] == '*' || p[1] == '/'))
3695 {
3696 matched = FALSE;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003697 // After "//" all text is comment
Bram Moolenaar071d4272004-06-13 20:20:40 +00003698 if (p[1] == '/')
3699 break;
3700 ++p;
3701 }
3702 else if (!matched && p[0] == '*' && p[1] == '/')
3703 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003704 // Can find match after "* /".
Bram Moolenaar071d4272004-06-13 20:20:40 +00003705 matched = TRUE;
3706 ++p;
3707 }
3708 }
3709 }
3710 }
3711 }
3712 }
3713 if (matched)
3714 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003715 if (action == ACTION_EXPAND)
3716 {
Bram Moolenaard9eefe32019-04-06 14:22:21 +02003717 int cont_s_ipos = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003718 int add_r;
3719 char_u *aux;
3720
3721 if (depth == -1 && lnum == curwin->w_cursor.lnum)
3722 break;
3723 found = TRUE;
3724 aux = p = startp;
Bram Moolenaar4be06f92005-07-29 22:36:03 +00003725 if (compl_cont_status & CONT_ADDING)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003726 {
Bram Moolenaar4be06f92005-07-29 22:36:03 +00003727 p += compl_length;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003728 if (vim_iswordp(p))
3729 goto exit_matched;
3730 p = find_word_start(p);
3731 }
3732 p = find_word_end(p);
3733 i = (int)(p - aux);
3734
Bram Moolenaar4be06f92005-07-29 22:36:03 +00003735 if ((compl_cont_status & CONT_ADDING) && i == compl_length)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003736 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003737 // IOSIZE > compl_length, so the STRNCPY works
Bram Moolenaar071d4272004-06-13 20:20:40 +00003738 STRNCPY(IObuff, aux, i);
Bram Moolenaar89d40322006-08-29 15:30:07 +00003739
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003740 // Get the next line: when "depth" < 0 from the current
3741 // buffer, otherwise from the included file. Jump to
3742 // exit_matched when past the last line.
Bram Moolenaar89d40322006-08-29 15:30:07 +00003743 if (depth < 0)
3744 {
3745 if (lnum >= end_lnum)
3746 goto exit_matched;
3747 line = ml_get(++lnum);
3748 }
3749 else if (vim_fgets(line = file_line,
3750 LSIZE, files[depth].fp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003751 goto exit_matched;
3752
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003753 // we read a line, set "already" to check this "line" later
3754 // if depth >= 0 we'll increase files[depth].lnum far
3755 // bellow -- Acevedo
Bram Moolenaar071d4272004-06-13 20:20:40 +00003756 already = aux = p = skipwhite(line);
3757 p = find_word_start(p);
3758 p = find_word_end(p);
3759 if (p > aux)
3760 {
3761 if (*aux != ')' && IObuff[i-1] != TAB)
3762 {
3763 if (IObuff[i-1] != ' ')
3764 IObuff[i++] = ' ';
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003765 // IObuf =~ "\(\k\|\i\).* ", thus i >= 2
Bram Moolenaar071d4272004-06-13 20:20:40 +00003766 if (p_js
3767 && (IObuff[i-2] == '.'
3768 || (vim_strchr(p_cpo, CPO_JOINSP) == NULL
3769 && (IObuff[i-2] == '?'
3770 || IObuff[i-2] == '!'))))
3771 IObuff[i++] = ' ';
3772 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003773 // copy as much as possible of the new word
Bram Moolenaar071d4272004-06-13 20:20:40 +00003774 if (p - aux >= IOSIZE - i)
3775 p = aux + IOSIZE - i - 1;
3776 STRNCPY(IObuff + i, aux, p - aux);
3777 i += (int)(p - aux);
Bram Moolenaard9eefe32019-04-06 14:22:21 +02003778 cont_s_ipos = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003779 }
3780 IObuff[i] = NUL;
3781 aux = IObuff;
3782
Bram Moolenaar4be06f92005-07-29 22:36:03 +00003783 if (i == compl_length)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003784 goto exit_matched;
3785 }
3786
Bram Moolenaare8c3a142006-08-29 14:30:35 +00003787 add_r = ins_compl_add_infercase(aux, i, p_ic,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003788 curr_fname == curbuf->b_fname ? NULL : curr_fname,
Bram Moolenaard9eefe32019-04-06 14:22:21 +02003789 dir, cont_s_ipos);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003790 if (add_r == OK)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003791 // if dir was BACKWARD then honor it just once
Bram Moolenaar071d4272004-06-13 20:20:40 +00003792 dir = FORWARD;
Bram Moolenaar572cb562005-08-05 21:35:02 +00003793 else if (add_r == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003794 break;
3795 }
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003796 else if (action == ACTION_SHOW_ALL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003797 {
3798 found = TRUE;
3799 if (!did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003800 gotocmdline(TRUE); // cursor at status line
Bram Moolenaar071d4272004-06-13 20:20:40 +00003801 if (curr_fname != prev_fname)
3802 {
3803 if (did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003804 msg_putchar('\n'); // cursor below last one
3805 if (!got_int) // don't display if 'q' typed
3806 // at "--more--" message
Bram Moolenaar071d4272004-06-13 20:20:40 +00003807 msg_home_replace_hl(curr_fname);
3808 prev_fname = curr_fname;
3809 }
3810 did_show = TRUE;
3811 if (!got_int)
3812 show_pat_in_path(line, type, TRUE, action,
3813 (depth == -1) ? NULL : files[depth].fp,
3814 (depth == -1) ? &lnum : &files[depth].lnum,
3815 match_count++);
3816
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003817 // Set matched flag for this file and all the ones that
3818 // include it
Bram Moolenaar071d4272004-06-13 20:20:40 +00003819 for (i = 0; i <= depth; ++i)
3820 files[i].matched = TRUE;
3821 }
3822 else if (--count <= 0)
3823 {
3824 found = TRUE;
3825 if (depth == -1 && lnum == curwin->w_cursor.lnum
Bram Moolenaar4033c552017-09-16 20:54:51 +02003826#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003827 && g_do_tagpreview == 0
3828#endif
3829 )
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003830 emsg(_("E387: Match is on current line"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003831 else if (action == ACTION_SHOW)
3832 {
3833 show_pat_in_path(line, type, did_show, action,
3834 (depth == -1) ? NULL : files[depth].fp,
3835 (depth == -1) ? &lnum : &files[depth].lnum, 1L);
3836 did_show = TRUE;
3837 }
3838 else
3839 {
3840#ifdef FEAT_GUI
3841 need_mouse_correct = TRUE;
3842#endif
Bram Moolenaar4033c552017-09-16 20:54:51 +02003843#if defined(FEAT_QUICKFIX)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003844 // ":psearch" uses the preview window
Bram Moolenaar071d4272004-06-13 20:20:40 +00003845 if (g_do_tagpreview != 0)
3846 {
3847 curwin_save = curwin;
Bram Moolenaar576a4a62019-08-18 15:25:17 +02003848 prepare_tagpreview(TRUE, TRUE, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003849 }
3850#endif
3851 if (action == ACTION_SPLIT)
3852 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003853 if (win_split(0, 0) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003854 break;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02003855 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003856 }
3857 if (depth == -1)
3858 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003859 // match in current file
Bram Moolenaar4033c552017-09-16 20:54:51 +02003860#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003861 if (g_do_tagpreview != 0)
3862 {
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02003863 if (!GETFILE_SUCCESS(getfile(
Bram Moolenaarc31f9ae2017-07-23 22:02:02 +02003864 curwin_save->w_buffer->b_fnum, NULL,
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02003865 NULL, TRUE, lnum, FALSE)))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003866 break; // failed to jump to file
Bram Moolenaar071d4272004-06-13 20:20:40 +00003867 }
3868 else
3869#endif
3870 setpcmark();
3871 curwin->w_cursor.lnum = lnum;
Bram Moolenaarc31f9ae2017-07-23 22:02:02 +02003872 check_cursor();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003873 }
3874 else
3875 {
Bram Moolenaar8ad80de2017-06-05 16:01:59 +02003876 if (!GETFILE_SUCCESS(getfile(
3877 0, files[depth].name, NULL, TRUE,
3878 files[depth].lnum, FALSE)))
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003879 break; // failed to jump to file
3880 // autocommands may have changed the lnum, we don't
3881 // want that here
Bram Moolenaar071d4272004-06-13 20:20:40 +00003882 curwin->w_cursor.lnum = files[depth].lnum;
3883 }
3884 }
3885 if (action != ACTION_SHOW)
3886 {
Bram Moolenaarfe81d452009-04-22 14:44:41 +00003887 curwin->w_cursor.col = (colnr_T)(startp - line);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003888 curwin->w_set_curswant = TRUE;
3889 }
3890
Bram Moolenaar4033c552017-09-16 20:54:51 +02003891#if defined(FEAT_QUICKFIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003892 if (g_do_tagpreview != 0
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00003893 && curwin != curwin_save && win_valid(curwin_save))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003894 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003895 // Return cursor to where we were
Bram Moolenaar071d4272004-06-13 20:20:40 +00003896 validate_cursor();
3897 redraw_later(VALID);
3898 win_enter(curwin_save, TRUE);
3899 }
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +01003900# ifdef FEAT_PROP_POPUP
Bram Moolenaar1b6d9c42019-08-05 21:52:04 +02003901 else if (WIN_IS_POPUP(curwin))
3902 // can't keep focus in popup window
3903 win_enter(firstwin, TRUE);
3904# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003905#endif
3906 break;
3907 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003908exit_matched:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003909 matched = FALSE;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003910 // look for other matches in the rest of the line if we
3911 // are not at the end of it already
Bram Moolenaar071d4272004-06-13 20:20:40 +00003912 if (def_regmatch.regprog == NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003913 && action == ACTION_EXPAND
Bram Moolenaar4be06f92005-07-29 22:36:03 +00003914 && !(compl_cont_status & CONT_SOL)
Bram Moolenaarfe81d452009-04-22 14:44:41 +00003915 && *startp != NUL
Bram Moolenaar1614a142019-10-06 22:00:13 +02003916 && *(p = startp + mb_ptr2len(startp)) != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003917 goto search_line;
3918 }
3919 line_breakcheck();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003920 if (action == ACTION_EXPAND)
Bram Moolenaar472e8592016-10-15 17:06:47 +02003921 ins_compl_check_keys(30, FALSE);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003922 if (got_int || ins_compl_interrupted())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003923 break;
3924
3925 /*
3926 * Read the next line. When reading an included file and encountering
3927 * end-of-file, close the file and continue in the file that included
3928 * it.
3929 */
3930 while (depth >= 0 && !already
3931 && vim_fgets(line = file_line, LSIZE, files[depth].fp))
3932 {
3933 fclose(files[depth].fp);
3934 --old_files;
3935 files[old_files].name = files[depth].name;
3936 files[old_files].matched = files[depth].matched;
3937 --depth;
3938 curr_fname = (depth == -1) ? curbuf->b_fname
3939 : files[depth].name;
3940 if (depth < depth_displayed)
3941 depth_displayed = depth;
3942 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003943 if (depth >= 0) // we could read the line
Bram Moolenaarc84e3c12013-07-03 22:28:36 +02003944 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003945 files[depth].lnum++;
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003946 // Remove any CR and LF from the line.
Bram Moolenaarc84e3c12013-07-03 22:28:36 +02003947 i = (int)STRLEN(line);
3948 if (i > 0 && line[i - 1] == '\n')
3949 line[--i] = NUL;
3950 if (i > 0 && line[i - 1] == '\r')
3951 line[--i] = NUL;
3952 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003953 else if (!already)
3954 {
3955 if (++lnum > end_lnum)
3956 break;
3957 line = ml_get(lnum);
3958 }
3959 already = NULL;
3960 }
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003961 // End of big for (;;) loop.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003962
Bram Moolenaar63d9e732019-12-05 21:10:38 +01003963 // Close any files that are still open.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003964 for (i = 0; i <= depth; i++)
3965 {
3966 fclose(files[i].fp);
3967 vim_free(files[i].name);
3968 }
3969 for (i = old_files; i < max_path_depth; i++)
3970 vim_free(files[i].name);
3971 vim_free(files);
3972
3973 if (type == CHECK_PATH)
3974 {
3975 if (!did_show)
3976 {
3977 if (action != ACTION_SHOW_ALL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01003978 msg(_("All included files were found"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003979 else
Bram Moolenaar32526b32019-01-19 17:43:09 +01003980 msg(_("No included files"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003981 }
3982 }
Bram Moolenaare2c453d2019-08-21 14:37:09 +02003983 else if (!found && action != ACTION_EXPAND)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003984 {
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003985 if (got_int || ins_compl_interrupted())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003986 emsg(_(e_interr));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003987 else if (type == FIND_DEFINE)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003988 emsg(_("E388: Couldn't find definition"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003989 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003990 emsg(_("E389: Couldn't find pattern"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003991 }
3992 if (action == ACTION_SHOW || action == ACTION_SHOW_ALL)
3993 msg_end();
3994
3995fpip_end:
3996 vim_free(file_line);
Bram Moolenaar473de612013-06-08 18:19:48 +02003997 vim_regfree(regmatch.regprog);
3998 vim_regfree(incl_regmatch.regprog);
3999 vim_regfree(def_regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004000}
4001
4002 static void
Bram Moolenaar764b23c2016-01-30 21:10:09 +01004003show_pat_in_path(
4004 char_u *line,
4005 int type,
4006 int did_show,
4007 int action,
4008 FILE *fp,
4009 linenr_T *lnum,
4010 long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004011{
4012 char_u *p;
4013
4014 if (did_show)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004015 msg_putchar('\n'); // cursor below last one
Bram Moolenaar91170f82006-05-05 21:15:17 +00004016 else if (!msg_silent)
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004017 gotocmdline(TRUE); // cursor at status line
4018 if (got_int) // 'q' typed at "--more--" message
Bram Moolenaar071d4272004-06-13 20:20:40 +00004019 return;
4020 for (;;)
4021 {
4022 p = line + STRLEN(line) - 1;
4023 if (fp != NULL)
4024 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004025 // We used fgets(), so get rid of newline at end
Bram Moolenaar071d4272004-06-13 20:20:40 +00004026 if (p >= line && *p == '\n')
4027 --p;
4028 if (p >= line && *p == '\r')
4029 --p;
4030 *(p + 1) = NUL;
4031 }
4032 if (action == ACTION_SHOW_ALL)
4033 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004034 sprintf((char *)IObuff, "%3ld: ", count); // show match nr
Bram Moolenaar32526b32019-01-19 17:43:09 +01004035 msg_puts((char *)IObuff);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004036 sprintf((char *)IObuff, "%4ld", *lnum); // show line nr
4037 // Highlight line numbers
Bram Moolenaar32526b32019-01-19 17:43:09 +01004038 msg_puts_attr((char *)IObuff, HL_ATTR(HLF_N));
4039 msg_puts(" ");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004040 }
Bram Moolenaar26a60b42005-02-22 08:49:11 +00004041 msg_prt_line(line, FALSE);
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004042 out_flush(); // show one line at a time
Bram Moolenaar071d4272004-06-13 20:20:40 +00004043
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004044 // Definition continues until line that doesn't end with '\'
Bram Moolenaar071d4272004-06-13 20:20:40 +00004045 if (got_int || type != FIND_DEFINE || p < line || *p != '\\')
4046 break;
4047
4048 if (fp != NULL)
4049 {
Bram Moolenaar63d9e732019-12-05 21:10:38 +01004050 if (vim_fgets(line, LSIZE, fp)) // end of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00004051 break;
4052 ++*lnum;
4053 }
4054 else
4055 {
4056 if (++*lnum > curbuf->b_ml.ml_line_count)
4057 break;
4058 line = ml_get(*lnum);
4059 }
4060 msg_putchar('\n');
4061 }
4062}
4063#endif
4064
4065#ifdef FEAT_VIMINFO
Bram Moolenaar6bd1d772019-10-09 22:01:25 +02004066/*
4067 * Return the last used search pattern at "idx".
4068 */
Bram Moolenaarc3328162019-07-23 22:15:25 +02004069 spat_T *
4070get_spat(int idx)
4071{
4072 return &spats[idx];
4073}
4074
Bram Moolenaar6bd1d772019-10-09 22:01:25 +02004075/*
4076 * Return the last used search pattern index.
4077 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004078 int
Bram Moolenaarc3328162019-07-23 22:15:25 +02004079get_spat_last_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004080{
Bram Moolenaarc3328162019-07-23 22:15:25 +02004081 return last_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004082}
Bram Moolenaar071d4272004-06-13 20:20:40 +00004083#endif
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004084
4085#ifdef FEAT_EVAL
4086/*
4087 * "searchcount()" function
4088 */
4089 void
4090f_searchcount(typval_T *argvars, typval_T *rettv)
4091{
4092 pos_T pos = curwin->w_cursor;
4093 char_u *pattern = NULL;
4094 int maxcount = SEARCH_STAT_DEF_MAX_COUNT;
4095 long timeout = SEARCH_STAT_DEF_TIMEOUT;
Bram Moolenaar4140c4f2020-09-05 23:16:00 +02004096 int recompute = TRUE;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004097 searchstat_T stat;
4098
4099 if (rettv_dict_alloc(rettv) == FAIL)
4100 return;
4101
4102 if (shortmess(SHM_SEARCHCOUNT)) // 'shortmess' contains 'S' flag
4103 recompute = TRUE;
4104
4105 if (argvars[0].v_type != VAR_UNKNOWN)
4106 {
Bram Moolenaar14681622020-06-03 22:57:39 +02004107 dict_T *dict;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004108 dictitem_T *di;
4109 listitem_T *li;
4110 int error = FALSE;
4111
Bram Moolenaar14681622020-06-03 22:57:39 +02004112 if (argvars[0].v_type != VAR_DICT || argvars[0].vval.v_dict == NULL)
4113 {
4114 emsg(_(e_dictreq));
4115 return;
4116 }
4117 dict = argvars[0].vval.v_dict;
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004118 di = dict_find(dict, (char_u *)"timeout", -1);
4119 if (di != NULL)
4120 {
4121 timeout = (long)tv_get_number_chk(&di->di_tv, &error);
4122 if (error)
4123 return;
4124 }
4125 di = dict_find(dict, (char_u *)"maxcount", -1);
4126 if (di != NULL)
4127 {
4128 maxcount = (int)tv_get_number_chk(&di->di_tv, &error);
4129 if (error)
4130 return;
4131 }
Bram Moolenaar597aaac2020-09-05 21:21:16 +02004132 recompute = dict_get_bool(dict, (char_u *)"recompute", recompute);
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004133 di = dict_find(dict, (char_u *)"pattern", -1);
4134 if (di != NULL)
4135 {
4136 pattern = tv_get_string_chk(&di->di_tv);
4137 if (pattern == NULL)
4138 return;
4139 }
4140 di = dict_find(dict, (char_u *)"pos", -1);
4141 if (di != NULL)
4142 {
4143 if (di->di_tv.v_type != VAR_LIST)
4144 {
4145 semsg(_(e_invarg2), "pos");
4146 return;
4147 }
4148 if (list_len(di->di_tv.vval.v_list) != 3)
4149 {
4150 semsg(_(e_invarg2), "List format should be [lnum, col, off]");
4151 return;
4152 }
4153 li = list_find(di->di_tv.vval.v_list, 0L);
4154 if (li != NULL)
4155 {
4156 pos.lnum = tv_get_number_chk(&li->li_tv, &error);
4157 if (error)
4158 return;
4159 }
4160 li = list_find(di->di_tv.vval.v_list, 1L);
4161 if (li != NULL)
4162 {
4163 pos.col = tv_get_number_chk(&li->li_tv, &error) - 1;
4164 if (error)
4165 return;
4166 }
4167 li = list_find(di->di_tv.vval.v_list, 2L);
4168 if (li != NULL)
4169 {
4170 pos.coladd = tv_get_number_chk(&li->li_tv, &error);
4171 if (error)
4172 return;
4173 }
4174 }
4175 }
4176
4177 save_last_search_pattern();
4178 if (pattern != NULL)
4179 {
4180 if (*pattern == NUL)
4181 goto the_end;
Bram Moolenaar109aece2020-06-01 19:08:54 +02004182 vim_free(spats[last_idx].pat);
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004183 spats[last_idx].pat = vim_strsave(pattern);
4184 }
4185 if (spats[last_idx].pat == NULL || *spats[last_idx].pat == NUL)
4186 goto the_end; // the previous pattern was never defined
4187
4188 update_search_stat(0, &pos, &pos, &stat, recompute, maxcount, timeout);
4189
4190 dict_add_number(rettv->vval.v_dict, "current", stat.cur);
4191 dict_add_number(rettv->vval.v_dict, "total", stat.cnt);
4192 dict_add_number(rettv->vval.v_dict, "exact_match", stat.exact_match);
4193 dict_add_number(rettv->vval.v_dict, "incomplete", stat.incomplete);
4194 dict_add_number(rettv->vval.v_dict, "maxcount", stat.last_maxcount);
4195
4196the_end:
4197 restore_last_search_pattern();
4198}
Bram Moolenaar635414d2020-09-11 22:25:15 +02004199
4200/*
4201 * Fuzzy string matching
4202 *
4203 * Ported from the lib_fts library authored by Forrest Smith.
4204 * https://github.com/forrestthewoods/lib_fts/tree/master/code
4205 *
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004206 * The following blog describes the fuzzy matching algorithm:
Bram Moolenaar635414d2020-09-11 22:25:15 +02004207 * https://www.forrestthewoods.com/blog/reverse_engineering_sublime_texts_fuzzy_match/
4208 *
4209 * Each matching string is assigned a score. The following factors are checked:
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004210 * - Matched letter
4211 * - Unmatched letter
4212 * - Consecutively matched letters
4213 * - Proximity to start
4214 * - Letter following a separator (space, underscore)
4215 * - Uppercase letter following lowercase (aka CamelCase)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004216 *
4217 * Matched letters are good. Unmatched letters are bad. Matching near the start
4218 * is good. Matching the first letter in the middle of a phrase is good.
4219 * Matching the uppercase letters in camel case entries is good.
4220 *
4221 * The score assigned for each factor is explained below.
4222 * File paths are different from file names. File extensions may be ignorable.
4223 * Single words care about consecutive matches but not separators or camel
4224 * case.
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004225 * Score starts at 100
Bram Moolenaar635414d2020-09-11 22:25:15 +02004226 * Matched letter: +0 points
4227 * Unmatched letter: -1 point
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004228 * Consecutive match bonus: +15 points
4229 * First letter bonus: +15 points
4230 * Separator bonus: +30 points
4231 * Camel case bonus: +30 points
4232 * Unmatched leading letter: -5 points (max: -15)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004233 *
4234 * There is some nuance to this. Scores don’t have an intrinsic meaning. The
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004235 * score range isn’t 0 to 100. It’s roughly [50, 150]. Longer words have a
Bram Moolenaar635414d2020-09-11 22:25:15 +02004236 * lower minimum score due to unmatched letter penalty. Longer search patterns
4237 * have a higher maximum score due to match bonuses.
4238 *
4239 * Separator and camel case bonus is worth a LOT. Consecutive matches are worth
4240 * quite a bit.
4241 *
4242 * There is a penalty if you DON’T match the first three letters. Which
4243 * effectively rewards matching near the start. However there’s no difference
4244 * in matching between the middle and end.
4245 *
4246 * There is not an explicit bonus for an exact match. Unmatched letters receive
4247 * a penalty. So shorter strings and closer matches are worth more.
4248 */
4249typedef struct
4250{
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004251 int idx; // used for stable sort
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004252 listitem_T *item;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004253 int score;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004254 list_T *lmatchpos;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004255} fuzzyItem_T;
4256
Bram Moolenaare9f9f162020-10-20 19:01:30 +02004257// bonus for adjacent matches; this is higher than SEPARATOR_BONUS so that
4258// matching a whole word is preferred.
4259#define SEQUENTIAL_BONUS 40
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004260// bonus if match occurs after a separator
4261#define SEPARATOR_BONUS 30
4262// bonus if match is uppercase and prev is lower
4263#define CAMEL_BONUS 30
4264// bonus if the first letter is matched
4265#define FIRST_LETTER_BONUS 15
4266// penalty applied for every letter in str before the first match
4267#define LEADING_LETTER_PENALTY -5
4268// maximum penalty for leading letters
4269#define MAX_LEADING_LETTER_PENALTY -15
4270// penalty for every letter that doesn't match
4271#define UNMATCHED_LETTER_PENALTY -1
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004272// penalty for gap in matching positions (-2 * k)
4273#define GAP_PENALTY -2
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004274// Score for a string that doesn't fuzzy match the pattern
4275#define SCORE_NONE -9999
4276
4277#define FUZZY_MATCH_RECURSION_LIMIT 10
4278// Maximum number of characters that can be fuzzy matched
4279#define MAXMATCHES 256
4280
4281typedef int_u matchidx_T;
4282
4283/*
4284 * Compute a score for a fuzzy matched string. The matching character locations
4285 * are in 'matches'.
4286 */
4287 static int
4288fuzzy_match_compute_score(
4289 char_u *str,
4290 int strSz,
4291 matchidx_T *matches,
4292 int numMatches)
4293{
4294 int score;
4295 int penalty;
4296 int unmatched;
4297 int i;
4298 char_u *p = str;
4299 matchidx_T sidx = 0;
4300
4301 // Initialize score
4302 score = 100;
4303
4304 // Apply leading letter penalty
4305 penalty = LEADING_LETTER_PENALTY * matches[0];
4306 if (penalty < MAX_LEADING_LETTER_PENALTY)
4307 penalty = MAX_LEADING_LETTER_PENALTY;
4308 score += penalty;
4309
4310 // Apply unmatched penalty
4311 unmatched = strSz - numMatches;
4312 score += UNMATCHED_LETTER_PENALTY * unmatched;
4313
4314 // Apply ordering bonuses
4315 for (i = 0; i < numMatches; ++i)
4316 {
4317 matchidx_T currIdx = matches[i];
4318
4319 if (i > 0)
4320 {
4321 matchidx_T prevIdx = matches[i - 1];
4322
4323 // Sequential
4324 if (currIdx == (prevIdx + 1))
4325 score += SEQUENTIAL_BONUS;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004326 else
4327 score += GAP_PENALTY * (currIdx - prevIdx);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004328 }
4329
4330 // Check for bonuses based on neighbor character value
4331 if (currIdx > 0)
4332 {
4333 // Camel case
Bram Moolenaarc53e9c52020-09-22 22:08:32 +02004334 int neighbor = ' ';
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004335 int curr;
4336 int neighborSeparator;
4337
4338 if (has_mbyte)
4339 {
4340 while (sidx < currIdx)
4341 {
4342 neighbor = (*mb_ptr2char)(p);
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004343 MB_PTR_ADV(p);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004344 sidx++;
4345 }
4346 curr = (*mb_ptr2char)(p);
4347 }
4348 else
4349 {
4350 neighbor = str[currIdx - 1];
4351 curr = str[currIdx];
4352 }
4353
4354 if (vim_islower(neighbor) && vim_isupper(curr))
4355 score += CAMEL_BONUS;
4356
4357 // Separator
4358 neighborSeparator = neighbor == '_' || neighbor == ' ';
4359 if (neighborSeparator)
4360 score += SEPARATOR_BONUS;
4361 }
4362 else
4363 {
4364 // First letter
4365 score += FIRST_LETTER_BONUS;
4366 }
4367 }
4368 return score;
4369}
4370
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004371/*
4372 * Perform a recursive search for fuzzy matching 'fuzpat' in 'str'.
4373 * Return the number of matching characters.
4374 */
Bram Moolenaar635414d2020-09-11 22:25:15 +02004375 static int
4376fuzzy_match_recursive(
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004377 char_u *fuzpat,
4378 char_u *str,
4379 matchidx_T strIdx,
4380 int *outScore,
4381 char_u *strBegin,
4382 int strLen,
4383 matchidx_T *srcMatches,
4384 matchidx_T *matches,
4385 int maxMatches,
4386 int nextMatch,
4387 int *recursionCount)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004388{
4389 // Recursion params
4390 int recursiveMatch = FALSE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004391 matchidx_T bestRecursiveMatches[MAXMATCHES];
Bram Moolenaar635414d2020-09-11 22:25:15 +02004392 int bestRecursiveScore = 0;
4393 int first_match;
4394 int matched;
4395
4396 // Count recursions
4397 ++*recursionCount;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004398 if (*recursionCount >= FUZZY_MATCH_RECURSION_LIMIT)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004399 return 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004400
4401 // Detect end of strings
4402 if (*fuzpat == '\0' || *str == '\0')
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004403 return 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004404
4405 // Loop through fuzpat and str looking for a match
4406 first_match = TRUE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004407 while (*fuzpat != NUL && *str != NUL)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004408 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004409 int c1;
4410 int c2;
4411
4412 c1 = PTR2CHAR(fuzpat);
4413 c2 = PTR2CHAR(str);
4414
Bram Moolenaar635414d2020-09-11 22:25:15 +02004415 // Found match
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004416 if (vim_tolower(c1) == vim_tolower(c2))
Bram Moolenaar635414d2020-09-11 22:25:15 +02004417 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004418 matchidx_T recursiveMatches[MAXMATCHES];
Bram Moolenaar635414d2020-09-11 22:25:15 +02004419 int recursiveScore = 0;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004420 char_u *next_char;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004421
4422 // Supplied matches buffer was too short
4423 if (nextMatch >= maxMatches)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004424 return 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004425
4426 // "Copy-on-Write" srcMatches into matches
4427 if (first_match && srcMatches)
4428 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004429 memcpy(matches, srcMatches, nextMatch * sizeof(srcMatches[0]));
Bram Moolenaar635414d2020-09-11 22:25:15 +02004430 first_match = FALSE;
4431 }
4432
4433 // Recursive call that "skips" this match
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004434 if (has_mbyte)
4435 next_char = str + (*mb_ptr2len)(str);
4436 else
4437 next_char = str + 1;
4438 if (fuzzy_match_recursive(fuzpat, next_char, strIdx + 1,
4439 &recursiveScore, strBegin, strLen, matches,
4440 recursiveMatches,
4441 sizeof(recursiveMatches)/sizeof(recursiveMatches[0]),
4442 nextMatch, recursionCount))
Bram Moolenaar635414d2020-09-11 22:25:15 +02004443 {
4444 // Pick best recursive score
4445 if (!recursiveMatch || recursiveScore > bestRecursiveScore)
4446 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004447 memcpy(bestRecursiveMatches, recursiveMatches,
4448 MAXMATCHES * sizeof(recursiveMatches[0]));
Bram Moolenaar635414d2020-09-11 22:25:15 +02004449 bestRecursiveScore = recursiveScore;
4450 }
4451 recursiveMatch = TRUE;
4452 }
4453
4454 // Advance
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004455 matches[nextMatch++] = strIdx;
4456 if (has_mbyte)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004457 MB_PTR_ADV(fuzpat);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004458 else
4459 ++fuzpat;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004460 }
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004461 if (has_mbyte)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004462 MB_PTR_ADV(str);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004463 else
4464 ++str;
4465 strIdx++;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004466 }
4467
4468 // Determine if full fuzpat was matched
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004469 matched = *fuzpat == NUL ? TRUE : FALSE;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004470
4471 // Calculate score
4472 if (matched)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004473 *outScore = fuzzy_match_compute_score(strBegin, strLen, matches,
4474 nextMatch);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004475
4476 // Return best result
4477 if (recursiveMatch && (!matched || bestRecursiveScore > *outScore))
4478 {
4479 // Recursive score is better than "this"
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004480 memcpy(matches, bestRecursiveMatches, maxMatches * sizeof(matches[0]));
Bram Moolenaar635414d2020-09-11 22:25:15 +02004481 *outScore = bestRecursiveScore;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004482 return nextMatch;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004483 }
4484 else if (matched)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004485 return nextMatch; // "this" score is better than recursive
Bram Moolenaar635414d2020-09-11 22:25:15 +02004486
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004487 return 0; // no match
Bram Moolenaar635414d2020-09-11 22:25:15 +02004488}
4489
4490/*
4491 * fuzzy_match()
4492 *
4493 * Performs exhaustive search via recursion to find all possible matches and
4494 * match with highest score.
4495 * Scores values have no intrinsic meaning. Possible score range is not
4496 * normalized and varies with pattern.
4497 * Recursion is limited internally (default=10) to prevent degenerate cases
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004498 * (pat_arg="aaaaaa" str="aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa").
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004499 * Uses char_u for match indices. Therefore patterns are limited to MAXMATCHES
Bram Moolenaar635414d2020-09-11 22:25:15 +02004500 * characters.
4501 *
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004502 * Returns TRUE if 'pat_arg' matches 'str'. Also returns the match score in
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004503 * 'outScore' and the matching character positions in 'matches'.
Bram Moolenaar635414d2020-09-11 22:25:15 +02004504 */
4505 static int
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004506fuzzy_match(
4507 char_u *str,
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004508 char_u *pat_arg,
4509 int matchseq,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004510 int *outScore,
4511 matchidx_T *matches,
4512 int maxMatches)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004513{
Bram Moolenaar635414d2020-09-11 22:25:15 +02004514 int recursionCount = 0;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004515 int len = MB_CHARLEN(str);
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004516 char_u *save_pat;
4517 char_u *pat;
4518 char_u *p;
4519 int complete = FALSE;
4520 int score = 0;
4521 int numMatches = 0;
4522 int matchCount;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004523
4524 *outScore = 0;
4525
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004526 save_pat = vim_strsave(pat_arg);
4527 if (save_pat == NULL)
4528 return FALSE;
4529 pat = save_pat;
4530 p = pat;
4531
4532 // Try matching each word in 'pat_arg' in 'str'
4533 while (TRUE)
4534 {
4535 if (matchseq)
4536 complete = TRUE;
4537 else
4538 {
4539 // Extract one word from the pattern (separated by space)
4540 p = skipwhite(p);
4541 if (*p == NUL)
4542 break;
4543 pat = p;
4544 while (*p != NUL && !VIM_ISWHITE(PTR2CHAR(p)))
4545 {
4546 if (has_mbyte)
4547 MB_PTR_ADV(p);
4548 else
4549 ++p;
4550 }
4551 if (*p == NUL) // processed all the words
4552 complete = TRUE;
4553 *p = NUL;
4554 }
4555
4556 score = 0;
4557 recursionCount = 0;
4558 matchCount = fuzzy_match_recursive(pat, str, 0, &score, str, len, NULL,
4559 matches + numMatches, maxMatches - numMatches,
4560 0, &recursionCount);
4561 if (matchCount == 0)
4562 {
4563 numMatches = 0;
4564 break;
4565 }
4566
4567 // Accumulate the match score and the number of matches
4568 *outScore += score;
4569 numMatches += matchCount;
4570
4571 if (complete)
4572 break;
4573
4574 // try matching the next word
4575 ++p;
4576 }
4577
4578 vim_free(save_pat);
4579 return numMatches != 0;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004580}
4581
4582/*
4583 * Sort the fuzzy matches in the descending order of the match score.
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004584 * For items with same score, retain the order using the index (stable sort)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004585 */
4586 static int
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004587fuzzy_match_item_compare(const void *s1, const void *s2)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004588{
4589 int v1 = ((fuzzyItem_T *)s1)->score;
4590 int v2 = ((fuzzyItem_T *)s2)->score;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004591 int idx1 = ((fuzzyItem_T *)s1)->idx;
4592 int idx2 = ((fuzzyItem_T *)s2)->idx;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004593
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004594 return v1 == v2 ? (idx1 - idx2) : v1 > v2 ? -1 : 1;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004595}
4596
4597/*
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004598 * Fuzzy search the string 'str' in a list of 'items' and return the matching
4599 * strings in 'fmatchlist'.
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004600 * If 'matchseq' is TRUE, then for multi-word search strings, match all the
4601 * words in sequence.
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004602 * If 'items' is a list of strings, then search for 'str' in the list.
4603 * If 'items' is a list of dicts, then either use 'key' to lookup the string
4604 * for each item or use 'item_cb' Funcref function to get the string.
4605 * If 'retmatchpos' is TRUE, then return a list of positions where 'str'
4606 * matches for each item.
Bram Moolenaar635414d2020-09-11 22:25:15 +02004607 */
4608 static void
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004609fuzzy_match_in_list(
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004610 list_T *items,
4611 char_u *str,
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004612 int matchseq,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004613 char_u *key,
4614 callback_T *item_cb,
4615 int retmatchpos,
4616 list_T *fmatchlist)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004617{
4618 long len;
4619 fuzzyItem_T *ptrs;
4620 listitem_T *li;
4621 long i = 0;
4622 int found_match = FALSE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004623 matchidx_T matches[MAXMATCHES];
Bram Moolenaar635414d2020-09-11 22:25:15 +02004624
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004625 len = list_len(items);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004626 if (len == 0)
4627 return;
4628
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004629 ptrs = ALLOC_CLEAR_MULT(fuzzyItem_T, len);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004630 if (ptrs == NULL)
4631 return;
4632
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004633 // For all the string items in items, get the fuzzy matching score
4634 FOR_ALL_LIST_ITEMS(items, li)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004635 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004636 int score;
4637 char_u *itemstr;
4638 typval_T rettv;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004639
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004640 ptrs[i].idx = i;
Bram Moolenaar635414d2020-09-11 22:25:15 +02004641 ptrs[i].item = li;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004642 ptrs[i].score = SCORE_NONE;
4643 itemstr = NULL;
4644 rettv.v_type = VAR_UNKNOWN;
4645 if (li->li_tv.v_type == VAR_STRING) // list of strings
4646 itemstr = li->li_tv.vval.v_string;
4647 else if (li->li_tv.v_type == VAR_DICT &&
4648 (key != NULL || item_cb->cb_name != NULL))
4649 {
4650 // For a dict, either use the specified key to lookup the string or
4651 // use the specified callback function to get the string.
4652 if (key != NULL)
4653 itemstr = dict_get_string(li->li_tv.vval.v_dict, key, FALSE);
4654 else
Bram Moolenaar635414d2020-09-11 22:25:15 +02004655 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004656 typval_T argv[2];
4657
4658 // Invoke the supplied callback (if any) to get the dict item
4659 li->li_tv.vval.v_dict->dv_refcount++;
4660 argv[0].v_type = VAR_DICT;
4661 argv[0].vval.v_dict = li->li_tv.vval.v_dict;
4662 argv[1].v_type = VAR_UNKNOWN;
4663 if (call_callback(item_cb, -1, &rettv, 1, argv) != FAIL)
4664 {
4665 if (rettv.v_type == VAR_STRING)
4666 itemstr = rettv.vval.v_string;
4667 }
4668 dict_unref(li->li_tv.vval.v_dict);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004669 }
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004670 }
4671
4672 if (itemstr != NULL
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004673 && fuzzy_match(itemstr, str, matchseq, &score, matches,
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004674 sizeof(matches) / sizeof(matches[0])))
4675 {
4676 // Copy the list of matching positions in itemstr to a list, if
4677 // 'retmatchpos' is set.
4678 if (retmatchpos)
4679 {
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004680 int j = 0;
4681 char_u *p;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004682
4683 ptrs[i].lmatchpos = list_alloc();
4684 if (ptrs[i].lmatchpos == NULL)
4685 goto done;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004686
4687 p = str;
4688 while (*p != NUL)
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004689 {
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004690 if (!VIM_ISWHITE(PTR2CHAR(p)))
4691 {
4692 if (list_append_number(ptrs[i].lmatchpos,
4693 matches[j]) == FAIL)
4694 goto done;
4695 j++;
4696 }
4697 if (has_mbyte)
4698 MB_PTR_ADV(p);
4699 else
4700 ++p;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004701 }
4702 }
4703 ptrs[i].score = score;
4704 found_match = TRUE;
4705 }
Bram Moolenaar635414d2020-09-11 22:25:15 +02004706 ++i;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004707 clear_tv(&rettv);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004708 }
4709
4710 if (found_match)
4711 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004712 list_T *l;
4713
Bram Moolenaar635414d2020-09-11 22:25:15 +02004714 // Sort the list by the descending order of the match score
4715 qsort((void *)ptrs, (size_t)len, sizeof(fuzzyItem_T),
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004716 fuzzy_match_item_compare);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004717
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004718 // For matchfuzzy(), return a list of matched strings.
4719 // ['str1', 'str2', 'str3']
4720 // For matchfuzzypos(), return a list with two items.
4721 // The first item is a list of matched strings. The second item
4722 // is a list of lists where each list item is a list of matched
4723 // character positions.
4724 // [['str1', 'str2', 'str3'], [[1, 3], [1, 3], [1, 3]]]
4725 if (retmatchpos)
4726 {
4727 li = list_find(fmatchlist, 0);
4728 if (li == NULL || li->li_tv.vval.v_list == NULL)
4729 goto done;
4730 l = li->li_tv.vval.v_list;
4731 }
4732 else
4733 l = fmatchlist;
4734
4735 // Copy the matching strings with a valid score to the return list
Bram Moolenaar635414d2020-09-11 22:25:15 +02004736 for (i = 0; i < len; i++)
4737 {
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004738 if (ptrs[i].score == SCORE_NONE)
Bram Moolenaar635414d2020-09-11 22:25:15 +02004739 break;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004740 list_append_tv(l, &ptrs[i].item->li_tv);
4741 }
4742
4743 // next copy the list of matching positions
4744 if (retmatchpos)
4745 {
4746 li = list_find(fmatchlist, -1);
4747 if (li == NULL || li->li_tv.vval.v_list == NULL)
4748 goto done;
4749 l = li->li_tv.vval.v_list;
4750
4751 for (i = 0; i < len; i++)
4752 {
4753 if (ptrs[i].score == SCORE_NONE)
4754 break;
4755 if (ptrs[i].lmatchpos != NULL &&
4756 list_append_list(l, ptrs[i].lmatchpos) == FAIL)
4757 goto done;
4758 }
Bram Moolenaar635414d2020-09-11 22:25:15 +02004759 }
4760 }
4761
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004762done:
Bram Moolenaar635414d2020-09-11 22:25:15 +02004763 vim_free(ptrs);
4764}
4765
4766/*
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004767 * Do fuzzy matching. Returns the list of matched strings in 'rettv'.
4768 * If 'retmatchpos' is TRUE, also returns the matching character positions.
4769 */
4770 static void
4771do_fuzzymatch(typval_T *argvars, typval_T *rettv, int retmatchpos)
4772{
4773 callback_T cb;
4774 char_u *key = NULL;
4775 int ret;
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004776 int matchseq = FALSE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004777
4778 CLEAR_POINTER(&cb);
4779
4780 // validate and get the arguments
4781 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
4782 {
4783 semsg(_(e_listarg), retmatchpos ? "matchfuzzypos()" : "matchfuzzy()");
4784 return;
4785 }
4786 if (argvars[1].v_type != VAR_STRING
4787 || argvars[1].vval.v_string == NULL)
4788 {
4789 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
4790 return;
4791 }
4792
4793 if (argvars[2].v_type != VAR_UNKNOWN)
4794 {
4795 dict_T *d;
4796 dictitem_T *di;
4797
4798 if (argvars[2].v_type != VAR_DICT || argvars[2].vval.v_dict == NULL)
4799 {
4800 emsg(_(e_dictreq));
4801 return;
4802 }
4803
4804 // To search a dict, either a callback function or a key can be
4805 // specified.
4806 d = argvars[2].vval.v_dict;
4807 if ((di = dict_find(d, (char_u *)"key", -1)) != NULL)
4808 {
4809 if (di->di_tv.v_type != VAR_STRING
4810 || di->di_tv.vval.v_string == NULL
4811 || *di->di_tv.vval.v_string == NUL)
4812 {
4813 semsg(_(e_invarg2), tv_get_string(&di->di_tv));
4814 return;
4815 }
4816 key = tv_get_string(&di->di_tv);
4817 }
4818 else if ((di = dict_find(d, (char_u *)"text_cb", -1)) != NULL)
4819 {
4820 cb = get_callback(&di->di_tv);
4821 if (cb.cb_name == NULL)
4822 {
4823 semsg(_(e_invargval), "text_cb");
4824 return;
4825 }
4826 }
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004827 if ((di = dict_find(d, (char_u *)"matchseq", -1)) != NULL)
4828 matchseq = TRUE;
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004829 }
4830
4831 // get the fuzzy matches
4832 ret = rettv_list_alloc(rettv);
4833 if (ret != OK)
4834 goto done;
4835 if (retmatchpos)
4836 {
4837 list_T *l;
4838
4839 // For matchfuzzypos(), a list with two items are returned. First item
4840 // is a list of matching strings and the second item is a list of
4841 // lists with matching positions within each string.
4842 l = list_alloc();
4843 if (l == NULL)
4844 goto done;
4845 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4846 goto done;
4847 l = list_alloc();
4848 if (l == NULL)
4849 goto done;
4850 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4851 goto done;
4852 }
4853
Bram Moolenaar8ded5b62020-10-23 16:50:30 +02004854 fuzzy_match_in_list(argvars[0].vval.v_list, tv_get_string(&argvars[1]),
4855 matchseq, key, &cb, retmatchpos, rettv->vval.v_list);
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004856
4857done:
4858 free_callback(&cb);
4859}
4860
4861/*
Bram Moolenaar635414d2020-09-11 22:25:15 +02004862 * "matchfuzzy()" function
4863 */
4864 void
4865f_matchfuzzy(typval_T *argvars, typval_T *rettv)
4866{
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02004867 do_fuzzymatch(argvars, rettv, FALSE);
4868}
4869
4870/*
4871 * "matchfuzzypos()" function
4872 */
4873 void
4874f_matchfuzzypos(typval_T *argvars, typval_T *rettv)
4875{
4876 do_fuzzymatch(argvars, rettv, TRUE);
Bram Moolenaar635414d2020-09-11 22:25:15 +02004877}
4878
Bram Moolenaare8f5ec02020-06-01 17:28:35 +02004879#endif