]> git.ipfire.org Git - thirdparty/git.git/blame - grep.c
Documentation: object_id_len goes up to 31
[thirdparty/git.git] / grep.c
CommitLineData
83b5d2f5 1#include "cache.h"
b2141fc1 2#include "config.h"
83b5d2f5 3#include "grep.h"
cbd53a21 4#include "object-store.h"
60ecac98 5#include "userdiff.h"
6bfce93e 6#include "xdiff-interface.h"
335ec3bf
JK
7#include "diff.h"
8#include "diffcore.h"
5c1ebcca 9#include "commit.h"
793dc676 10#include "quote.h"
3ac68a93 11#include "help.h"
83b5d2f5 12
07a7d656 13static int grep_source_load(struct grep_source *gs);
acd00ea0
NTND
14static int grep_source_is_binary(struct grep_source *gs,
15 struct index_state *istate);
07a7d656 16
96313423
17static void std_output(struct grep_opt *opt, const void *buf, size_t size)
18{
19 fwrite(buf, size, 1, stdout);
20}
21
22static struct grep_opt grep_defaults = {
23 .relative = 1,
24 .pathname = 1,
25 .max_depth = -1,
26 .pattern_type_option = GREP_PATTERN_TYPE_UNSPECIFIED,
27 .colors = {
28 [GREP_COLOR_CONTEXT] = "",
b83f99c3 29 [GREP_COLOR_FILENAME] = GIT_COLOR_MAGENTA,
96313423 30 [GREP_COLOR_FUNCTION] = "",
b83f99c3
LH
31 [GREP_COLOR_LINENO] = GIT_COLOR_GREEN,
32 [GREP_COLOR_COLUMNNO] = GIT_COLOR_GREEN,
96313423
33 [GREP_COLOR_MATCH_CONTEXT] = GIT_COLOR_BOLD_RED,
34 [GREP_COLOR_MATCH_SELECTED] = GIT_COLOR_BOLD_RED,
35 [GREP_COLOR_SELECTED] = "",
36 [GREP_COLOR_SEP] = GIT_COLOR_CYAN,
37 },
38 .only_matching = 0,
39 .color = -1,
40 .output = std_output,
41};
7687a054 42
fa151dc5
NTND
43static const char *color_grep_slots[] = {
44 [GREP_COLOR_CONTEXT] = "context",
45 [GREP_COLOR_FILENAME] = "filename",
46 [GREP_COLOR_FUNCTION] = "function",
47 [GREP_COLOR_LINENO] = "lineNumber",
d036d667 48 [GREP_COLOR_COLUMNNO] = "column",
fa151dc5
NTND
49 [GREP_COLOR_MATCH_CONTEXT] = "matchContext",
50 [GREP_COLOR_MATCH_SELECTED] = "matchSelected",
51 [GREP_COLOR_SELECTED] = "selected",
52 [GREP_COLOR_SEP] = "separator",
53};
54
7687a054
JH
55static int parse_pattern_type_arg(const char *opt, const char *arg)
56{
57 if (!strcmp(arg, "default"))
58 return GREP_PATTERN_TYPE_UNSPECIFIED;
59 else if (!strcmp(arg, "basic"))
60 return GREP_PATTERN_TYPE_BRE;
61 else if (!strcmp(arg, "extended"))
62 return GREP_PATTERN_TYPE_ERE;
63 else if (!strcmp(arg, "fixed"))
64 return GREP_PATTERN_TYPE_FIXED;
65 else if (!strcmp(arg, "perl"))
66 return GREP_PATTERN_TYPE_PCRE;
67 die("bad %s argument: %s", opt, arg);
68}
69
3ac68a93
NTND
70define_list_config_array_extra(color_grep_slots, {"match"});
71
7687a054
JH
72/*
73 * Read the configuration file once and store it in
74 * the grep_defaults template.
75 */
76int grep_config(const char *var, const char *value, void *cb)
77{
78 struct grep_opt *opt = &grep_defaults;
fa151dc5 79 const char *slot;
7687a054
JH
80
81 if (userdiff_config(var, value) < 0)
82 return -1;
83
6ba9bb76
84 /*
85 * The instance of grep_opt that we set up here is copied by
86 * grep_init() to be used by each individual invocation.
87 * When populating a new field of this structure here, be
88 * sure to think about ownership -- e.g., you might need to
89 * override the shallow copy in grep_init() with a deep copy.
90 */
91
7687a054 92 if (!strcmp(var, "grep.extendedregexp")) {
c7e38551 93 opt->extended_regexp_option = git_config_bool(var, value);
7687a054
JH
94 return 0;
95 }
96
97 if (!strcmp(var, "grep.patterntype")) {
98 opt->pattern_type_option = parse_pattern_type_arg(var, value);
99 return 0;
100 }
101
102 if (!strcmp(var, "grep.linenumber")) {
103 opt->linenum = git_config_bool(var, value);
104 return 0;
105 }
6653fec3
TB
106 if (!strcmp(var, "grep.column")) {
107 opt->columnnum = git_config_bool(var, value);
108 return 0;
109 }
7687a054 110
6453f7b3
AS
111 if (!strcmp(var, "grep.fullname")) {
112 opt->relative = !git_config_bool(var, value);
113 return 0;
114 }
115
7687a054
JH
116 if (!strcmp(var, "color.grep"))
117 opt->color = git_config_colorbool(var, value);
fa151dc5
NTND
118 if (!strcmp(var, "color.grep.match")) {
119 if (grep_config("color.grep.matchcontext", value, cb) < 0)
120 return -1;
121 if (grep_config("color.grep.matchselected", value, cb) < 0)
122 return -1;
123 } else if (skip_prefix(var, "color.grep.", &slot)) {
124 int i = LOOKUP_CONFIG(color_grep_slots, slot);
125 char *color;
126
127 if (i < 0)
128 return -1;
129 color = opt->colors[i];
7687a054
JH
130 if (!value)
131 return config_error_nonbool(var);
f6c5a296 132 return color_parse(value, color);
7687a054
JH
133 }
134 return 0;
135}
136
137/*
138 * Initialize one instance of grep_opt and copy the
139 * default values from the template we read the configuration
140 * information in an earlier call to git_config(grep_config).
141 */
38bbc2ea 142void grep_init(struct grep_opt *opt, struct repository *repo, const char *prefix)
7687a054 143{
6ba9bb76
144 *opt = grep_defaults;
145
38bbc2ea 146 opt->repo = repo;
7687a054
JH
147 opt->prefix = prefix;
148 opt->prefix_length = (prefix && *prefix) ? strlen(prefix) : 0;
149 opt->pattern_tail = &opt->pattern_list;
150 opt->header_tail = &opt->header_list;
7687a054 151}
07a7d656 152
8465541e 153static void grep_set_pattern_type_option(enum grep_pattern_type pattern_type, struct grep_opt *opt)
c5c31d33 154{
07a3d411
ÆAB
155 /*
156 * When committing to the pattern type by setting the relevant
157 * fields in grep_opt it's generally not necessary to zero out
158 * the fields we're not choosing, since they won't have been
159 * set by anything. The extended_regexp_option field is the
160 * only exception to this.
161 *
162 * This is because in the process of parsing grep.patternType
163 * & grep.extendedRegexp we set opt->pattern_type_option and
164 * opt->extended_regexp_option, respectively. We then
165 * internally use opt->extended_regexp_option to see if we're
166 * compiling an ERE. It must be unset if that's not actually
167 * the case.
168 */
169 if (pattern_type != GREP_PATTERN_TYPE_ERE &&
170 opt->extended_regexp_option)
171 opt->extended_regexp_option = 0;
172
c5c31d33
JH
173 switch (pattern_type) {
174 case GREP_PATTERN_TYPE_UNSPECIFIED:
175 /* fall through */
176
177 case GREP_PATTERN_TYPE_BRE:
c5c31d33
JH
178 break;
179
180 case GREP_PATTERN_TYPE_ERE:
07a3d411 181 opt->extended_regexp_option = 1;
c5c31d33
JH
182 break;
183
184 case GREP_PATTERN_TYPE_FIXED:
185 opt->fixed = 1;
c5c31d33
JH
186 break;
187
188 case GREP_PATTERN_TYPE_PCRE:
94da9193 189 opt->pcre2 = 1;
c5c31d33
JH
190 break;
191 }
192}
193
8465541e
JH
194void grep_commit_pattern_type(enum grep_pattern_type pattern_type, struct grep_opt *opt)
195{
196 if (pattern_type != GREP_PATTERN_TYPE_UNSPECIFIED)
197 grep_set_pattern_type_option(pattern_type, opt);
198 else if (opt->pattern_type_option != GREP_PATTERN_TYPE_UNSPECIFIED)
199 grep_set_pattern_type_option(opt->pattern_type_option, opt);
200 else if (opt->extended_regexp_option)
07a3d411
ÆAB
201 /*
202 * This branch *must* happen after setting from the
203 * opt->pattern_type_option above, we don't want
204 * grep.extendedRegexp to override grep.patternType!
205 */
8465541e
JH
206 grep_set_pattern_type_option(GREP_PATTERN_TYPE_ERE, opt);
207}
208
fc456751
RS
209static struct grep_pat *create_grep_pat(const char *pat, size_t patlen,
210 const char *origin, int no,
211 enum grep_pat_token t,
212 enum grep_header_field field)
a4d7d2c6
JH
213{
214 struct grep_pat *p = xcalloc(1, sizeof(*p));
526a858a 215 p->pattern = xmemdupz(pat, patlen);
fc456751
RS
216 p->patternlen = patlen;
217 p->origin = origin;
218 p->no = no;
219 p->token = t;
a4d7d2c6 220 p->field = field;
fc456751
RS
221 return p;
222}
223
2b3873ff
RS
224static void do_append_grep_pat(struct grep_pat ***tail, struct grep_pat *p)
225{
226 **tail = p;
227 *tail = &p->next;
a4d7d2c6 228 p->next = NULL;
526a858a
RS
229
230 switch (p->token) {
231 case GREP_PATTERN: /* atom */
232 case GREP_PATTERN_HEAD:
233 case GREP_PATTERN_BODY:
234 for (;;) {
235 struct grep_pat *new_pat;
236 size_t len = 0;
237 char *cp = p->pattern + p->patternlen, *nl = NULL;
238 while (++len <= p->patternlen) {
239 if (*(--cp) == '\n') {
240 nl = cp;
241 break;
242 }
243 }
244 if (!nl)
245 break;
246 new_pat = create_grep_pat(nl + 1, len - 1, p->origin,
247 p->no, p->token, p->field);
248 new_pat->next = p->next;
249 if (!p->next)
250 *tail = &new_pat->next;
251 p->next = new_pat;
252 *nl = '\0';
253 p->patternlen -= len;
254 }
255 break;
256 default:
257 break;
258 }
2b3873ff
RS
259}
260
fc456751
RS
261void append_header_grep_pattern(struct grep_opt *opt,
262 enum grep_header_field field, const char *pat)
263{
264 struct grep_pat *p = create_grep_pat(pat, strlen(pat), "header", 0,
265 GREP_PATTERN_HEAD, field);
baa6378f
JH
266 if (field == GREP_HEADER_REFLOG)
267 opt->use_reflog_filter = 1;
2b3873ff 268 do_append_grep_pat(&opt->header_tail, p);
a4d7d2c6
JH
269}
270
83b5d2f5
JH
271void append_grep_pattern(struct grep_opt *opt, const char *pat,
272 const char *origin, int no, enum grep_pat_token t)
ed40a095
RS
273{
274 append_grep_pat(opt, pat, strlen(pat), origin, no, t);
275}
276
277void append_grep_pat(struct grep_opt *opt, const char *pat, size_t patlen,
278 const char *origin, int no, enum grep_pat_token t)
83b5d2f5 279{
fc456751 280 struct grep_pat *p = create_grep_pat(pat, patlen, origin, no, t, 0);
2b3873ff 281 do_append_grep_pat(&opt->pattern_tail, p);
83b5d2f5
JH
282}
283
5b594f45
FK
284struct grep_opt *grep_opt_dup(const struct grep_opt *opt)
285{
286 struct grep_pat *pat;
287 struct grep_opt *ret = xmalloc(sizeof(struct grep_opt));
288 *ret = *opt;
289
290 ret->pattern_list = NULL;
291 ret->pattern_tail = &ret->pattern_list;
292
293 for(pat = opt->pattern_list; pat != NULL; pat = pat->next)
294 {
295 if(pat->token == GREP_PATTERN_HEAD)
296 append_header_grep_pattern(ret, pat->field,
297 pat->pattern);
298 else
ed40a095
RS
299 append_grep_pat(ret, pat->pattern, pat->patternlen,
300 pat->origin, pat->no, pat->token);
5b594f45
FK
301 }
302
303 return ret;
304}
305
a30c148a
MK
306static NORETURN void compile_regexp_failed(const struct grep_pat *p,
307 const char *error)
308{
309 char where[1024];
310
311 if (p->no)
19bdd3e7 312 xsnprintf(where, sizeof(where), "In '%s' at %d, ", p->origin, p->no);
a30c148a 313 else if (p->origin)
19bdd3e7 314 xsnprintf(where, sizeof(where), "%s, ", p->origin);
a30c148a
MK
315 else
316 where[0] = 0;
317
318 die("%s'%s': %s", where, p->pattern, error);
319}
320
543f1c0c
ÆAB
321static int is_fixed(const char *s, size_t len)
322{
323 size_t i;
324
325 for (i = 0; i < len; i++) {
326 if (is_regex_special(s[i]))
327 return 0;
328 }
329
330 return 1;
331}
332
94da9193 333#ifdef USE_LIBPCRE2
c1760352
ÆAB
334#define GREP_PCRE2_DEBUG_MALLOC 0
335
336static void *pcre2_malloc(PCRE2_SIZE size, MAYBE_UNUSED void *memory_data)
337{
338 void *pointer = malloc(size);
339#if GREP_PCRE2_DEBUG_MALLOC
340 static int count = 1;
341 fprintf(stderr, "PCRE2:%p -> #%02d: alloc(%lu)\n", pointer, count++, size);
342#endif
343 return pointer;
344}
345
346static void pcre2_free(void *pointer, MAYBE_UNUSED void *memory_data)
347{
348#if GREP_PCRE2_DEBUG_MALLOC
349 static int count = 1;
350 if (pointer)
351 fprintf(stderr, "PCRE2:%p -> #%02d: free()\n", pointer, count++);
352#endif
353 free(pointer);
354}
355
94da9193
ÆAB
356static void compile_pcre2_pattern(struct grep_pat *p, const struct grep_opt *opt)
357{
358 int error;
359 PCRE2_UCHAR errbuf[256];
360 PCRE2_SIZE erroffset;
361 int options = PCRE2_MULTILINE;
94da9193 362 int jitret;
a25b9085
ÆAB
363 int patinforet;
364 size_t jitsizearg;
32e3e8bc 365 int literal = !opt->ignore_case && (p->fixed || p->is_fixed);
94da9193 366
cbe81e65
ÆAB
367 /*
368 * Call pcre2_general_context_create() before calling any
369 * other pcre2_*(). It sets up our malloc()/free() functions
370 * with which everything else is allocated.
371 */
372 p->pcre2_general_context = pcre2_general_context_create(
373 pcre2_malloc, pcre2_free, NULL);
374 if (!p->pcre2_general_context)
375 die("Couldn't allocate PCRE2 general context");
94da9193
ÆAB
376
377 if (opt->ignore_case) {
44570188 378 if (!opt->ignore_locale && has_non_ascii(p->pattern)) {
cbe81e65
ÆAB
379 p->pcre2_tables = pcre2_maketables(p->pcre2_general_context);
380 p->pcre2_compile_context = pcre2_compile_context_create(p->pcre2_general_context);
10da030a
CMAB
381 pcre2_set_character_tables(p->pcre2_compile_context,
382 p->pcre2_tables);
94da9193
ÆAB
383 }
384 options |= PCRE2_CASELESS;
385 }
32e3e8bc 386 if (!opt->ignore_locale && is_utf8_locale() && !literal)
95ca1f98
ÆAB
387 options |= (PCRE2_UTF | PCRE2_MATCH_INVALID_UTF);
388
797c3599 389#ifdef GIT_PCRE2_VERSION_10_36_OR_HIGHER
95ca1f98 390 /* Work around https://bugs.exim.org/show_bug.cgi?id=2642 fixed in 10.36 */
797c3599
ÆAB
391 if (PCRE2_MATCH_INVALID_UTF && options & (PCRE2_UTF | PCRE2_CASELESS))
392 options |= PCRE2_NO_START_OPTIMIZE;
393#endif
94da9193
ÆAB
394
395 p->pcre2_pattern = pcre2_compile((PCRE2_SPTR)p->pattern,
396 p->patternlen, options, &error, &erroffset,
397 p->pcre2_compile_context);
398
399 if (p->pcre2_pattern) {
cbe81e65 400 p->pcre2_match_data = pcre2_match_data_create_from_pattern(p->pcre2_pattern, p->pcre2_general_context);
94da9193
ÆAB
401 if (!p->pcre2_match_data)
402 die("Couldn't allocate PCRE2 match data");
403 } else {
404 pcre2_get_error_message(error, errbuf, sizeof(errbuf));
405 compile_regexp_failed(p, (const char *)&errbuf);
406 }
407
408 pcre2_config(PCRE2_CONFIG_JIT, &p->pcre2_jit_on);
04bef50c 409 if (p->pcre2_jit_on) {
94da9193
ÆAB
410 jitret = pcre2_jit_compile(p->pcre2_pattern, PCRE2_JIT_COMPLETE);
411 if (jitret)
412 die("Couldn't JIT the PCRE2 pattern '%s', got '%d'\n", p->pattern, jitret);
a25b9085
ÆAB
413
414 /*
415 * The pcre2_config(PCRE2_CONFIG_JIT, ...) call just
416 * tells us whether the library itself supports JIT,
417 * but to see whether we're going to be actually using
418 * JIT we need to extract PCRE2_INFO_JITSIZE from the
419 * pattern *after* we do pcre2_jit_compile() above.
420 *
421 * This is because if the pattern contains the
422 * (*NO_JIT) verb (see pcre2syntax(3))
423 * pcre2_jit_compile() will exit early with 0. If we
424 * then proceed to call pcre2_jit_match() further down
425 * the line instead of pcre2_match() we'll either
426 * segfault (pre PCRE 10.31) or run into a fatal error
427 * (post PCRE2 10.31)
428 */
429 patinforet = pcre2_pattern_info(p->pcre2_pattern, PCRE2_INFO_JITSIZE, &jitsizearg);
430 if (patinforet)
431 BUG("pcre2_pattern_info() failed: %d", patinforet);
432 if (jitsizearg == 0) {
433 p->pcre2_jit_on = 0;
434 return;
435 }
94da9193
ÆAB
436 }
437}
438
439static int pcre2match(struct grep_pat *p, const char *line, const char *eol,
440 regmatch_t *match, int eflags)
441{
442 int ret, flags = 0;
443 PCRE2_SIZE *ovector;
444 PCRE2_UCHAR errbuf[256];
445
446 if (eflags & REG_NOTBOL)
447 flags |= PCRE2_NOTBOL;
448
449 if (p->pcre2_jit_on)
450 ret = pcre2_jit_match(p->pcre2_pattern, (unsigned char *)line,
451 eol - line, 0, flags, p->pcre2_match_data,
452 NULL);
453 else
454 ret = pcre2_match(p->pcre2_pattern, (unsigned char *)line,
455 eol - line, 0, flags, p->pcre2_match_data,
456 NULL);
457
458 if (ret < 0 && ret != PCRE2_ERROR_NOMATCH) {
459 pcre2_get_error_message(ret, errbuf, sizeof(errbuf));
460 die("%s failed with error code %d: %s",
461 (p->pcre2_jit_on ? "pcre2_jit_match" : "pcre2_match"), ret,
462 errbuf);
463 }
464 if (ret > 0) {
465 ovector = pcre2_get_ovector_pointer(p->pcre2_match_data);
466 ret = 0;
467 match->rm_so = (int)ovector[0];
468 match->rm_eo = (int)ovector[1];
469 }
470
471 return ret;
472}
473
474static void free_pcre2_pattern(struct grep_pat *p)
475{
476 pcre2_compile_context_free(p->pcre2_compile_context);
477 pcre2_code_free(p->pcre2_pattern);
478 pcre2_match_data_free(p->pcre2_match_data);
b76bf27f 479#ifdef GIT_PCRE2_VERSION_10_34_OR_HIGHER
cbe81e65 480 pcre2_maketables_free(p->pcre2_general_context, p->pcre2_tables);
b76bf27f 481#else
10da030a 482 free((void *)p->pcre2_tables);
b76bf27f 483#endif
cbe81e65 484 pcre2_general_context_free(p->pcre2_general_context);
94da9193
ÆAB
485}
486#else /* !USE_LIBPCRE2 */
487static void compile_pcre2_pattern(struct grep_pat *p, const struct grep_opt *opt)
488{
94da9193
ÆAB
489 die("cannot use Perl-compatible regexes when not compiled with USE_LIBPCRE");
490}
491
492static int pcre2match(struct grep_pat *p, const char *line, const char *eol,
493 regmatch_t *match, int eflags)
494{
495 return 1;
496}
497
498static void free_pcre2_pattern(struct grep_pat *p)
499{
500}
94da9193 501
793dc676
NTND
502static void compile_fixed_regexp(struct grep_pat *p, struct grep_opt *opt)
503{
504 struct strbuf sb = STRBUF_INIT;
505 int err;
1ceababc 506 int regflags = 0;
793dc676
NTND
507
508 basic_regex_quote_buf(&sb, p->pattern);
793dc676
NTND
509 if (opt->ignore_case)
510 regflags |= REG_ICASE;
511 err = regcomp(&p->regexp, sb.buf, regflags);
793dc676
NTND
512 strbuf_release(&sb);
513 if (err) {
514 char errbuf[1024];
515 regerror(err, &p->regexp, errbuf, sizeof(errbuf));
793dc676
NTND
516 compile_regexp_failed(p, errbuf);
517 }
518}
b65abcaf 519#endif /* !USE_LIBPCRE2 */
793dc676 520
83b5d2f5
JH
521static void compile_regexp(struct grep_pat *p, struct grep_opt *opt)
522{
c822255c 523 int err;
07a3d411 524 int regflags = REG_NEWLINE;
c822255c 525
d7eb527d 526 p->word_regexp = opt->word_regexp;
5183bf67 527 p->ignore_case = opt->ignore_case;
48de2a76 528 p->fixed = opt->fixed;
d7eb527d 529
45d1f37c
ÆAB
530 if (memchr(p->pattern, 0, p->patternlen) && !opt->pcre2)
531 die(_("given pattern contains NULL byte (via -f <file>). This is only supported with -P under PCRE v2"));
532
09872f64 533 p->is_fixed = is_fixed(p->pattern, p->patternlen);
8a599983
ÆAB
534#ifdef USE_LIBPCRE2
535 if (!p->fixed && !p->is_fixed) {
536 const char *no_jit = "(*NO_JIT)";
537 const int no_jit_len = strlen(no_jit);
538 if (starts_with(p->pattern, no_jit) &&
539 is_fixed(p->pattern + no_jit_len,
540 p->patternlen - no_jit_len))
541 p->is_fixed = 1;
542 }
543#endif
09872f64 544 if (p->fixed || p->is_fixed) {
b65abcaf 545#ifdef USE_LIBPCRE2
09872f64 546 if (p->is_fixed) {
b65abcaf
ÆAB
547 compile_pcre2_pattern(p, opt);
548 } else {
549 /*
550 * E.g. t7811-grep-open.sh relies on the
551 * pattern being restored.
552 */
553 char *old_pattern = p->pattern;
554 size_t old_patternlen = p->patternlen;
555 struct strbuf sb = STRBUF_INIT;
556
557 /*
558 * There is the PCRE2_LITERAL flag, but it's
559 * only in PCRE v2 10.30 and later. Needing to
560 * ifdef our way around that and dealing with
561 * it + PCRE2_MULTILINE being an error is more
562 * complex than just quoting this ourselves.
563 */
564 strbuf_add(&sb, "\\Q", 2);
565 strbuf_add(&sb, p->pattern, p->patternlen);
566 strbuf_add(&sb, "\\E", 2);
567
568 p->pattern = sb.buf;
569 p->patternlen = sb.len;
570 compile_pcre2_pattern(p, opt);
571 p->pattern = old_pattern;
572 p->patternlen = old_patternlen;
573 strbuf_release(&sb);
574 }
575#else /* !USE_LIBPCRE2 */
793dc676 576 compile_fixed_regexp(p, opt);
b65abcaf 577#endif /* !USE_LIBPCRE2 */
793dc676 578 return;
9eceddee 579 }
c822255c 580
94da9193
ÆAB
581 if (opt->pcre2) {
582 compile_pcre2_pattern(p, opt);
583 return;
584 }
585
07a3d411
ÆAB
586 if (p->ignore_case)
587 regflags |= REG_ICASE;
588 if (opt->extended_regexp_option)
589 regflags |= REG_EXTENDED;
590 err = regcomp(&p->regexp, p->pattern, regflags);
83b5d2f5
JH
591 if (err) {
592 char errbuf[1024];
83b5d2f5 593 regerror(err, &p->regexp, errbuf, 1024);
a30c148a 594 compile_regexp_failed(p, errbuf);
83b5d2f5
JH
595 }
596}
597
e2b15427
RS
598static struct grep_expr *grep_not_expr(struct grep_expr *expr)
599{
600 struct grep_expr *z = xcalloc(1, sizeof(*z));
601 z->node = GREP_NODE_NOT;
602 z->u.unary = expr;
603 return z;
604}
605
f2d27598
TB
606static struct grep_expr *grep_binexp(enum grep_expr_node kind,
607 struct grep_expr *left,
608 struct grep_expr *right)
9dbf00ba
RS
609{
610 struct grep_expr *z = xcalloc(1, sizeof(*z));
f2d27598 611 z->node = kind;
9dbf00ba
RS
612 z->u.binary.left = left;
613 z->u.binary.right = right;
614 return z;
615}
616
f2d27598
TB
617static struct grep_expr *grep_or_expr(struct grep_expr *left, struct grep_expr *right)
618{
619 return grep_binexp(GREP_NODE_OR, left, right);
620}
621
0a6adc26
TB
622static struct grep_expr *grep_and_expr(struct grep_expr *left, struct grep_expr *right)
623{
624 return grep_binexp(GREP_NODE_AND, left, right);
625}
626
0ab7befa 627static struct grep_expr *compile_pattern_or(struct grep_pat **);
83b5d2f5
JH
628static struct grep_expr *compile_pattern_atom(struct grep_pat **list)
629{
630 struct grep_pat *p;
631 struct grep_expr *x;
632
633 p = *list;
c922b01f
LT
634 if (!p)
635 return NULL;
83b5d2f5
JH
636 switch (p->token) {
637 case GREP_PATTERN: /* atom */
480c1ca6
JH
638 case GREP_PATTERN_HEAD:
639 case GREP_PATTERN_BODY:
ca56dadb 640 CALLOC_ARRAY(x, 1);
83b5d2f5
JH
641 x->node = GREP_NODE_ATOM;
642 x->u.atom = p;
643 *list = p->next;
644 return x;
645 case GREP_OPEN_PAREN:
646 *list = p->next;
0ab7befa 647 x = compile_pattern_or(list);
83b5d2f5
JH
648 if (!*list || (*list)->token != GREP_CLOSE_PAREN)
649 die("unmatched parenthesis");
650 *list = (*list)->next;
651 return x;
652 default:
653 return NULL;
654 }
655}
656
657static struct grep_expr *compile_pattern_not(struct grep_pat **list)
658{
659 struct grep_pat *p;
660 struct grep_expr *x;
661
662 p = *list;
c922b01f
LT
663 if (!p)
664 return NULL;
83b5d2f5
JH
665 switch (p->token) {
666 case GREP_NOT:
667 if (!p->next)
668 die("--not not followed by pattern expression");
669 *list = p->next;
e2b15427
RS
670 x = compile_pattern_not(list);
671 if (!x)
83b5d2f5 672 die("--not followed by non pattern expression");
e2b15427 673 return grep_not_expr(x);
83b5d2f5
JH
674 default:
675 return compile_pattern_atom(list);
676 }
677}
678
679static struct grep_expr *compile_pattern_and(struct grep_pat **list)
680{
681 struct grep_pat *p;
0a6adc26 682 struct grep_expr *x, *y;
83b5d2f5
JH
683
684 x = compile_pattern_not(list);
685 p = *list;
686 if (p && p->token == GREP_AND) {
fe7fe62d
RS
687 if (!x)
688 die("--and not preceded by pattern expression");
83b5d2f5
JH
689 if (!p->next)
690 die("--and not followed by pattern expression");
691 *list = p->next;
692 y = compile_pattern_and(list);
693 if (!y)
694 die("--and not followed by pattern expression");
0a6adc26 695 return grep_and_expr(x, y);
83b5d2f5
JH
696 }
697 return x;
698}
699
700static struct grep_expr *compile_pattern_or(struct grep_pat **list)
701{
702 struct grep_pat *p;
9dbf00ba 703 struct grep_expr *x, *y;
83b5d2f5
JH
704
705 x = compile_pattern_and(list);
706 p = *list;
707 if (x && p && p->token != GREP_CLOSE_PAREN) {
708 y = compile_pattern_or(list);
709 if (!y)
710 die("not a pattern expression %s", p->pattern);
9dbf00ba 711 return grep_or_expr(x, y);
83b5d2f5
JH
712 }
713 return x;
714}
715
716static struct grep_expr *compile_pattern_expr(struct grep_pat **list)
717{
718 return compile_pattern_or(list);
719}
720
5aaeb733
JH
721static struct grep_expr *grep_true_expr(void)
722{
723 struct grep_expr *z = xcalloc(1, sizeof(*z));
724 z->node = GREP_NODE_TRUE;
725 return z;
726}
727
95ce9ce2 728static struct grep_expr *prep_header_patterns(struct grep_opt *opt)
83b5d2f5
JH
729{
730 struct grep_pat *p;
95ce9ce2 731 struct grep_expr *header_expr;
5aaeb733
JH
732 struct grep_expr *(header_group[GREP_HEADER_FIELD_MAX]);
733 enum grep_header_field fld;
83b5d2f5 734
95ce9ce2
JH
735 if (!opt->header_list)
736 return NULL;
2385f246 737
95ce9ce2
JH
738 for (p = opt->header_list; p; p = p->next) {
739 if (p->token != GREP_PATTERN_HEAD)
033abf97 740 BUG("a non-header pattern in grep header list.");
3ce3ffb8
AP
741 if (p->field < GREP_HEADER_FIELD_MIN ||
742 GREP_HEADER_FIELD_MAX <= p->field)
033abf97 743 BUG("unknown header field %d", p->field);
95ce9ce2 744 compile_regexp(p, opt);
80235ba7 745 }
5aaeb733
JH
746
747 for (fld = 0; fld < GREP_HEADER_FIELD_MAX; fld++)
748 header_group[fld] = NULL;
749
750 for (p = opt->header_list; p; p = p->next) {
751 struct grep_expr *h;
752 struct grep_pat *pp = p;
753
754 h = compile_pattern_atom(&pp);
755 if (!h || pp != p->next)
033abf97 756 BUG("malformed header expr");
5aaeb733
JH
757 if (!header_group[p->field]) {
758 header_group[p->field] = h;
759 continue;
760 }
761 header_group[p->field] = grep_or_expr(h, header_group[p->field]);
762 }
763
764 header_expr = NULL;
765
766 for (fld = 0; fld < GREP_HEADER_FIELD_MAX; fld++) {
767 if (!header_group[fld])
768 continue;
769 if (!header_expr)
770 header_expr = grep_true_expr();
771 header_expr = grep_or_expr(header_group[fld], header_expr);
772 }
95ce9ce2
JH
773 return header_expr;
774}
775
13e4fc7e
JH
776static struct grep_expr *grep_splice_or(struct grep_expr *x, struct grep_expr *y)
777{
778 struct grep_expr *z = x;
779
780 while (x) {
781 assert(x->node == GREP_NODE_OR);
782 if (x->u.binary.right &&
783 x->u.binary.right->node == GREP_NODE_TRUE) {
784 x->u.binary.right = y;
785 break;
786 }
787 x = x->u.binary.right;
788 }
789 return z;
790}
791
15c96497 792void compile_grep_patterns(struct grep_opt *opt)
95ce9ce2
JH
793{
794 struct grep_pat *p;
795 struct grep_expr *header_expr = prep_header_patterns(opt);
0ab7befa 796
83b5d2f5 797 for (p = opt->pattern_list; p; p = p->next) {
480c1ca6
JH
798 switch (p->token) {
799 case GREP_PATTERN: /* atom */
800 case GREP_PATTERN_HEAD:
801 case GREP_PATTERN_BODY:
c822255c 802 compile_regexp(p, opt);
480c1ca6
JH
803 break;
804 default:
83b5d2f5 805 opt->extended = 1;
480c1ca6
JH
806 break;
807 }
83b5d2f5
JH
808 }
809
794c0002 810 if (opt->all_match || opt->no_body_match || header_expr)
80235ba7 811 opt->extended = 1;
15c96497 812 else if (!opt->extended)
83b5d2f5
JH
813 return;
814
83b5d2f5 815 p = opt->pattern_list;
ba150a3f
MB
816 if (p)
817 opt->pattern_expression = compile_pattern_expr(&p);
83b5d2f5
JH
818 if (p)
819 die("incomplete pattern expression: %s", p->pattern);
80235ba7 820
794c0002
RS
821 if (opt->no_body_match && opt->pattern_expression)
822 opt->pattern_expression = grep_not_expr(opt->pattern_expression);
823
80235ba7
JH
824 if (!header_expr)
825 return;
826
5aaeb733 827 if (!opt->pattern_expression)
80235ba7 828 opt->pattern_expression = header_expr;
13e4fc7e
JH
829 else if (opt->all_match)
830 opt->pattern_expression = grep_splice_or(header_expr,
831 opt->pattern_expression);
5aaeb733
JH
832 else
833 opt->pattern_expression = grep_or_expr(opt->pattern_expression,
834 header_expr);
80235ba7 835 opt->all_match = 1;
83b5d2f5
JH
836}
837
b48fb5b6
JH
838static void free_pattern_expr(struct grep_expr *x)
839{
840 switch (x->node) {
5aaeb733 841 case GREP_NODE_TRUE:
b48fb5b6
JH
842 case GREP_NODE_ATOM:
843 break;
844 case GREP_NODE_NOT:
845 free_pattern_expr(x->u.unary);
846 break;
847 case GREP_NODE_AND:
848 case GREP_NODE_OR:
849 free_pattern_expr(x->u.binary.left);
850 free_pattern_expr(x->u.binary.right);
851 break;
852 }
853 free(x);
854}
855
856void free_grep_patterns(struct grep_opt *opt)
857{
858 struct grep_pat *p, *n;
859
860 for (p = opt->pattern_list; p; p = n) {
861 n = p->next;
862 switch (p->token) {
863 case GREP_PATTERN: /* atom */
864 case GREP_PATTERN_HEAD:
865 case GREP_PATTERN_BODY:
7599730b 866 if (p->pcre2_pattern)
94da9193 867 free_pcre2_pattern(p);
63e7e9d8
MK
868 else
869 regfree(&p->regexp);
526a858a 870 free(p->pattern);
b48fb5b6
JH
871 break;
872 default:
873 break;
874 }
875 free(p);
876 }
877
878 if (!opt->extended)
879 return;
880 free_pattern_expr(opt->pattern_expression);
881}
882
1a845fbc 883static const char *end_of_line(const char *cp, unsigned long *left)
83b5d2f5
JH
884{
885 unsigned long l = *left;
886 while (l && *cp != '\n') {
887 l--;
888 cp++;
889 }
890 *left = l;
891 return cp;
892}
893
894static int word_char(char ch)
895{
896 return isalnum(ch) || ch == '_';
897}
898
55f638bd
ML
899static void output_color(struct grep_opt *opt, const void *data, size_t size,
900 const char *color)
901{
daa0c3d9 902 if (want_color(opt->color) && color && color[0]) {
55f638bd
ML
903 opt->output(opt, color, strlen(color));
904 opt->output(opt, data, size);
905 opt->output(opt, GIT_COLOR_RESET, strlen(GIT_COLOR_RESET));
906 } else
907 opt->output(opt, data, size);
908}
909
910static void output_sep(struct grep_opt *opt, char sign)
911{
912 if (opt->null_following_name)
913 opt->output(opt, "\0", 1);
914 else
fa151dc5 915 output_color(opt, &sign, 1, opt->colors[GREP_COLOR_SEP]);
55f638bd
ML
916}
917
83caecca
RZ
918static void show_name(struct grep_opt *opt, const char *name)
919{
fa151dc5 920 output_color(opt, name, strlen(name), opt->colors[GREP_COLOR_FILENAME]);
5b594f45 921 opt->output(opt, opt->null_following_name ? "\0" : "\n", 1);
83caecca
RZ
922}
923
1a845fbc
JK
924static int patmatch(struct grep_pat *p,
925 const char *line, const char *eol,
97e77784
MK
926 regmatch_t *match, int eflags)
927{
928 int hit;
929
7599730b 930 if (p->pcre2_pattern)
94da9193 931 hit = !pcre2match(p, line, eol, match, eflags);
97e77784 932 else
b7d36ffc
JS
933 hit = !regexec_buf(&p->regexp, line, eol - line, 1, match,
934 eflags);
97e77784
MK
935
936 return hit;
937}
938
1a845fbc 939static void strip_timestamp(const char *bol, const char **eol_p)
a4d7d2c6 940{
1a845fbc 941 const char *eol = *eol_p;
a4d7d2c6
JH
942
943 while (bol < --eol) {
944 if (*eol != '>')
945 continue;
946 *eol_p = ++eol;
cc8e26ee 947 break;
a4d7d2c6 948 }
a4d7d2c6
JH
949}
950
951static struct {
952 const char *field;
953 size_t len;
954} header_field[] = {
955 { "author ", 7 },
956 { "committer ", 10 },
72fd13f7 957 { "reflog ", 7 },
a4d7d2c6
JH
958};
959
3f566c4e
HM
960static int headerless_match_one_pattern(struct grep_pat *p,
961 const char *bol, const char *eol,
962 enum grep_context ctx,
963 regmatch_t *pmatch, int eflags)
83b5d2f5
JH
964{
965 int hit = 0;
e701fadb 966 const char *start = bol;
83b5d2f5 967
480c1ca6
JH
968 if ((p->token != GREP_PATTERN) &&
969 ((p->token == GREP_PATTERN_HEAD) != (ctx == GREP_CONTEXT_HEAD)))
970 return 0;
971
83b5d2f5 972 again:
97e77784 973 hit = patmatch(p, bol, eol, pmatch, eflags);
83b5d2f5 974
d7eb527d 975 if (hit && p->word_regexp) {
83b5d2f5 976 if ((pmatch[0].rm_so < 0) ||
84201eae 977 (eol - bol) < pmatch[0].rm_so ||
83b5d2f5
JH
978 (pmatch[0].rm_eo < 0) ||
979 (eol - bol) < pmatch[0].rm_eo)
980 die("regexp returned nonsense");
981
982 /* Match beginning must be either beginning of the
983 * line, or at word boundary (i.e. the last char must
984 * not be a word char). Similarly, match end must be
985 * either end of the line, or at word boundary
986 * (i.e. the next char must not be a word char).
987 */
fb62eb7f 988 if ( ((pmatch[0].rm_so == 0) ||
83b5d2f5
JH
989 !word_char(bol[pmatch[0].rm_so-1])) &&
990 ((pmatch[0].rm_eo == (eol-bol)) ||
991 !word_char(bol[pmatch[0].rm_eo])) )
992 ;
993 else
994 hit = 0;
995
84201eae
RS
996 /* Words consist of at least one character. */
997 if (pmatch->rm_so == pmatch->rm_eo)
998 hit = 0;
999
83b5d2f5
JH
1000 if (!hit && pmatch[0].rm_so + bol + 1 < eol) {
1001 /* There could be more than one match on the
1002 * line, and the first match might not be
1003 * strict word match. But later ones could be!
fb62eb7f
RS
1004 * Forward to the next possible start, i.e. the
1005 * next position following a non-word char.
83b5d2f5
JH
1006 */
1007 bol = pmatch[0].rm_so + bol + 1;
fb62eb7f
RS
1008 while (word_char(bol[-1]) && bol < eol)
1009 bol++;
dbb6a4ad 1010 eflags |= REG_NOTBOL;
fb62eb7f
RS
1011 if (bol < eol)
1012 goto again;
83b5d2f5
JH
1013 }
1014 }
e701fadb
RS
1015 if (hit) {
1016 pmatch[0].rm_so += bol - start;
1017 pmatch[0].rm_eo += bol - start;
1018 }
83b5d2f5
JH
1019 return hit;
1020}
1021
3f566c4e
HM
1022static int match_one_pattern(struct grep_pat *p,
1023 const char *bol, const char *eol,
1024 enum grep_context ctx, regmatch_t *pmatch,
1025 int eflags)
1026{
1027 const char *field;
1028 size_t len;
1029
1030 if (p->token == GREP_PATTERN_HEAD) {
1031 assert(p->field < ARRAY_SIZE(header_field));
1032 field = header_field[p->field].field;
1033 len = header_field[p->field].len;
1034 if (strncmp(bol, field, len))
1035 return 0;
1036 bol += len;
1037
1038 switch (p->field) {
1039 case GREP_HEADER_AUTHOR:
1040 case GREP_HEADER_COMMITTER:
1041 strip_timestamp(bol, &eol);
1042 break;
1043 default:
1044 break;
1045 }
1046 }
1047
1048 return headerless_match_one_pattern(p, bol, eol, ctx, pmatch, eflags);
1049}
1050
1051
1a845fbc
JK
1052static int match_expr_eval(struct grep_opt *opt, struct grep_expr *x,
1053 const char *bol, const char *eol,
1054 enum grep_context ctx, ssize_t *col,
68d686e6 1055 ssize_t *icol, int collect_hits)
83b5d2f5 1056{
0ab7befa
JH
1057 int h = 0;
1058
c922b01f
LT
1059 if (!x)
1060 die("Not a valid grep expression");
83b5d2f5 1061 switch (x->node) {
5aaeb733
JH
1062 case GREP_NODE_TRUE:
1063 h = 1;
1064 break;
83b5d2f5 1065 case GREP_NODE_ATOM:
68d686e6
TB
1066 {
1067 regmatch_t tmp;
1068 h = match_one_pattern(x->u.atom, bol, eol, ctx,
1069 &tmp, 0);
1070 if (h && (*col < 0 || tmp.rm_so < *col))
1071 *col = tmp.rm_so;
1072 }
794c0002
RS
1073 if (x->u.atom->token == GREP_PATTERN_BODY)
1074 opt->body_hit |= h;
83b5d2f5
JH
1075 break;
1076 case GREP_NODE_NOT:
68d686e6
TB
1077 /*
1078 * Upon visiting a GREP_NODE_NOT, col and icol become swapped.
1079 */
1080 h = !match_expr_eval(opt, x->u.unary, bol, eol, ctx, icol, col,
1081 0);
0ab7befa 1082 break;
83b5d2f5 1083 case GREP_NODE_AND:
017c0fcf 1084 h = match_expr_eval(opt, x->u.binary.left, bol, eol, ctx, col,
68d686e6 1085 icol, 0);
017c0fcf
TB
1086 if (h || opt->columnnum) {
1087 /*
1088 * Don't short-circuit AND when given --column, since a
1089 * NOT earlier in the tree may turn this into an OR. In
1090 * this case, see the below comment.
1091 */
1092 h &= match_expr_eval(opt, x->u.binary.right, bol, eol,
1093 ctx, col, icol, 0);
1094 }
0ab7befa 1095 break;
83b5d2f5 1096 case GREP_NODE_OR:
017c0fcf
TB
1097 if (!(collect_hits || opt->columnnum)) {
1098 /*
1099 * Don't short-circuit OR when given --column (or
1100 * collecting hits) to ensure we don't skip a later
1101 * child that would produce an earlier match.
1102 */
68d686e6
TB
1103 return (match_expr_eval(opt, x->u.binary.left, bol, eol,
1104 ctx, col, icol, 0) ||
1105 match_expr_eval(opt, x->u.binary.right, bol,
1106 eol, ctx, col, icol, 0));
017c0fcf 1107 }
68d686e6
TB
1108 h = match_expr_eval(opt, x->u.binary.left, bol, eol, ctx, col,
1109 icol, 0);
017c0fcf
TB
1110 if (collect_hits)
1111 x->u.binary.left->hit |= h;
68d686e6 1112 h |= match_expr_eval(opt, x->u.binary.right, bol, eol, ctx, col,
017c0fcf 1113 icol, collect_hits);
0ab7befa
JH
1114 break;
1115 default:
d7530708 1116 die("Unexpected node type (internal error) %d", x->node);
83b5d2f5 1117 }
0ab7befa
JH
1118 if (collect_hits)
1119 x->hit |= h;
1120 return h;
83b5d2f5
JH
1121}
1122
1a845fbc
JK
1123static int match_expr(struct grep_opt *opt,
1124 const char *bol, const char *eol,
68d686e6
TB
1125 enum grep_context ctx, ssize_t *col,
1126 ssize_t *icol, int collect_hits)
83b5d2f5
JH
1127{
1128 struct grep_expr *x = opt->pattern_expression;
68d686e6 1129 return match_expr_eval(opt, x, bol, eol, ctx, col, icol, collect_hits);
83b5d2f5
JH
1130}
1131
1a845fbc
JK
1132static int match_line(struct grep_opt *opt,
1133 const char *bol, const char *eol,
68d686e6 1134 ssize_t *col, ssize_t *icol,
0ab7befa 1135 enum grep_context ctx, int collect_hits)
83b5d2f5
JH
1136{
1137 struct grep_pat *p;
017c0fcf 1138 int hit = 0;
79212772 1139
83b5d2f5 1140 if (opt->extended)
68d686e6
TB
1141 return match_expr(opt, bol, eol, ctx, col, icol,
1142 collect_hits);
0ab7befa
JH
1143
1144 /* we do not call with collect_hits without being extended */
83b5d2f5 1145 for (p = opt->pattern_list; p; p = p->next) {
68d686e6
TB
1146 regmatch_t tmp;
1147 if (match_one_pattern(p, bol, eol, ctx, &tmp, 0)) {
017c0fcf
TB
1148 hit |= 1;
1149 if (!opt->columnnum) {
1150 /*
1151 * Without --column, any single match on a line
1152 * is enough to know that it needs to be
1153 * printed. With --column, scan _all_ patterns
1154 * to find the earliest.
1155 */
1156 break;
1157 }
1158 if (*col < 0 || tmp.rm_so < *col)
1159 *col = tmp.rm_so;
68d686e6 1160 }
83b5d2f5 1161 }
017c0fcf 1162 return hit;
83b5d2f5
JH
1163}
1164
1a845fbc
JK
1165static int match_next_pattern(struct grep_pat *p,
1166 const char *bol, const char *eol,
7e8f59d5
RS
1167 enum grep_context ctx,
1168 regmatch_t *pmatch, int eflags)
1169{
1170 regmatch_t match;
1171
3f566c4e 1172 if (!headerless_match_one_pattern(p, bol, eol, ctx, &match, eflags))
7e8f59d5
RS
1173 return 0;
1174 if (match.rm_so < 0 || match.rm_eo < 0)
1175 return 0;
1176 if (pmatch->rm_so >= 0 && pmatch->rm_eo >= 0) {
1177 if (match.rm_so > pmatch->rm_so)
1178 return 1;
1179 if (match.rm_so == pmatch->rm_so && match.rm_eo < pmatch->rm_eo)
1180 return 1;
1181 }
1182 pmatch->rm_so = match.rm_so;
1183 pmatch->rm_eo = match.rm_eo;
1184 return 1;
1185}
1186
3f566c4e
HM
1187int grep_next_match(struct grep_opt *opt,
1188 const char *bol, const char *eol,
1189 enum grep_context ctx, regmatch_t *pmatch,
1190 enum grep_header_field field, int eflags)
7e8f59d5
RS
1191{
1192 struct grep_pat *p;
1193 int hit = 0;
1194
1195 pmatch->rm_so = pmatch->rm_eo = -1;
1196 if (bol < eol) {
3f566c4e
HM
1197 for (p = ((ctx == GREP_CONTEXT_HEAD)
1198 ? opt->header_list : opt->pattern_list);
1199 p; p = p->next) {
7e8f59d5 1200 switch (p->token) {
7e8f59d5 1201 case GREP_PATTERN_HEAD:
3f566c4e
HM
1202 if ((field != GREP_HEADER_FIELD_MAX) &&
1203 (p->field != field))
1204 continue;
1205 /* fall thru */
1206 case GREP_PATTERN: /* atom */
7e8f59d5
RS
1207 case GREP_PATTERN_BODY:
1208 hit |= match_next_pattern(p, bol, eol, ctx,
1209 pmatch, eflags);
1210 break;
1211 default:
1212 break;
1213 }
1214 }
1215 }
1216 return hit;
1217}
1218
c707ded3
TB
1219static void show_line_header(struct grep_opt *opt, const char *name,
1220 unsigned lno, ssize_t cno, char sign)
7e8f59d5 1221{
1d84f72e 1222 if (opt->heading && opt->last_shown == 0) {
fa151dc5 1223 output_color(opt, name, strlen(name), opt->colors[GREP_COLOR_FILENAME]);
1d84f72e
RS
1224 opt->output(opt, "\n", 1);
1225 }
5dd06d38
RS
1226 opt->last_shown = lno;
1227
1d84f72e 1228 if (!opt->heading && opt->pathname) {
fa151dc5 1229 output_color(opt, name, strlen(name), opt->colors[GREP_COLOR_FILENAME]);
55f638bd 1230 output_sep(opt, sign);
5b594f45
FK
1231 }
1232 if (opt->linenum) {
1233 char buf[32];
1a168e5c 1234 xsnprintf(buf, sizeof(buf), "%d", lno);
fa151dc5 1235 output_color(opt, buf, strlen(buf), opt->colors[GREP_COLOR_LINENO]);
55f638bd 1236 output_sep(opt, sign);
5b594f45 1237 }
89252cd0
TB
1238 /*
1239 * Treat 'cno' as the 1-indexed offset from the start of a non-context
1240 * line to its first match. Otherwise, 'cno' is 0 indicating that we are
1241 * being called with a context line.
1242 */
1243 if (opt->columnnum && cno) {
1244 char buf[32];
1245 xsnprintf(buf, sizeof(buf), "%"PRIuMAX, (uintmax_t)cno);
d036d667 1246 output_color(opt, buf, strlen(buf), opt->colors[GREP_COLOR_COLUMNNO]);
89252cd0
TB
1247 output_sep(opt, sign);
1248 }
c707ded3
TB
1249}
1250
1a845fbc
JK
1251static void show_line(struct grep_opt *opt,
1252 const char *bol, const char *eol,
c707ded3
TB
1253 const char *name, unsigned lno, ssize_t cno, char sign)
1254{
1255 int rest = eol - bol;
9d8db06e
TB
1256 const char *match_color = NULL;
1257 const char *line_color = NULL;
c707ded3
TB
1258
1259 if (opt->file_break && opt->last_shown == 0) {
1260 if (opt->show_hunk_mark)
1261 opt->output(opt, "\n", 1);
1262 } else if (opt->pre_context || opt->post_context || opt->funcbody) {
1263 if (opt->last_shown == 0) {
1264 if (opt->show_hunk_mark) {
87ece7ce 1265 output_color(opt, "--", 2, opt->colors[GREP_COLOR_SEP]);
c707ded3
TB
1266 opt->output(opt, "\n", 1);
1267 }
1268 } else if (lno > opt->last_shown + 1) {
87ece7ce 1269 output_color(opt, "--", 2, opt->colors[GREP_COLOR_SEP]);
c707ded3
TB
1270 opt->output(opt, "\n", 1);
1271 }
1272 }
9d8db06e
TB
1273 if (!opt->only_matching) {
1274 /*
1275 * In case the line we're being called with contains more than
1276 * one match, leave printing each header to the loop below.
1277 */
1278 show_line_header(opt, name, lno, cno, sign);
1279 }
1280 if (opt->color || opt->only_matching) {
7e8f59d5
RS
1281 regmatch_t match;
1282 enum grep_context ctx = GREP_CONTEXT_BODY;
7e8f59d5
RS
1283 int eflags = 0;
1284
9d8db06e
TB
1285 if (opt->color) {
1286 if (sign == ':')
87ece7ce 1287 match_color = opt->colors[GREP_COLOR_MATCH_SELECTED];
9d8db06e 1288 else
87ece7ce 1289 match_color = opt->colors[GREP_COLOR_MATCH_CONTEXT];
9d8db06e 1290 if (sign == ':')
87ece7ce 1291 line_color = opt->colors[GREP_COLOR_SELECTED];
9d8db06e 1292 else if (sign == '-')
87ece7ce 1293 line_color = opt->colors[GREP_COLOR_CONTEXT];
9d8db06e 1294 else if (sign == '=')
87ece7ce 1295 line_color = opt->colors[GREP_COLOR_FUNCTION];
9d8db06e 1296 }
3f566c4e
HM
1297 while (grep_next_match(opt, bol, eol, ctx, &match,
1298 GREP_HEADER_FIELD_MAX, eflags)) {
1f5b9cc4
RS
1299 if (match.rm_so == match.rm_eo)
1300 break;
5b594f45 1301
9d8db06e
TB
1302 if (opt->only_matching)
1303 show_line_header(opt, name, lno, cno, sign);
1304 else
1305 output_color(opt, bol, match.rm_so, line_color);
55f638bd 1306 output_color(opt, bol + match.rm_so,
79a77109 1307 match.rm_eo - match.rm_so, match_color);
9d8db06e
TB
1308 if (opt->only_matching)
1309 opt->output(opt, "\n", 1);
7e8f59d5 1310 bol += match.rm_eo;
9d8db06e 1311 cno += match.rm_eo;
7e8f59d5
RS
1312 rest -= match.rm_eo;
1313 eflags = REG_NOTBOL;
1314 }
7e8f59d5 1315 }
9d8db06e
TB
1316 if (!opt->only_matching) {
1317 output_color(opt, bol, rest, line_color);
1318 opt->output(opt, "\n", 1);
1319 }
7e8f59d5
RS
1320}
1321
78db6ea9
JK
1322int grep_use_locks;
1323
0579f91d
TR
1324/*
1325 * This lock protects access to the gitattributes machinery, which is
1326 * not thread-safe.
1327 */
1328pthread_mutex_t grep_attr_mutex;
1329
78db6ea9 1330static inline void grep_attr_lock(void)
0579f91d 1331{
78db6ea9 1332 if (grep_use_locks)
0579f91d
TR
1333 pthread_mutex_lock(&grep_attr_mutex);
1334}
1335
78db6ea9 1336static inline void grep_attr_unlock(void)
0579f91d 1337{
78db6ea9 1338 if (grep_use_locks)
0579f91d
TR
1339 pthread_mutex_unlock(&grep_attr_mutex);
1340}
b3aeb285 1341
1a845fbc
JK
1342static int match_funcname(struct grep_opt *opt, struct grep_source *gs,
1343 const char *bol, const char *eol)
2944e4e6 1344{
60ecac98 1345 xdemitconf_t *xecfg = opt->priv;
0579f91d 1346 if (xecfg && !xecfg->find_func) {
acd00ea0 1347 grep_source_load_driver(gs, opt->repo->index);
94ad9d9e
JK
1348 if (gs->driver->funcname.pattern) {
1349 const struct userdiff_funcname *pe = &gs->driver->funcname;
0579f91d
TR
1350 xdiff_set_find_func(xecfg, pe->pattern, pe->cflags);
1351 } else {
1352 xecfg = opt->priv = NULL;
1353 }
1354 }
1355
1356 if (xecfg) {
60ecac98
RS
1357 char buf[1];
1358 return xecfg->find_func(bol, eol - bol, buf, 1,
1359 xecfg->find_func_priv) >= 0;
1360 }
1361
2944e4e6
RS
1362 if (bol == eol)
1363 return 0;
1364 if (isalpha(*bol) || *bol == '_' || *bol == '$')
1365 return 1;
1366 return 0;
1367}
1368
e1327023 1369static void show_funcname_line(struct grep_opt *opt, struct grep_source *gs,
1a845fbc 1370 const char *bol, unsigned lno)
2944e4e6 1371{
e1327023 1372 while (bol > gs->buf) {
1a845fbc 1373 const char *eol = --bol;
2944e4e6 1374
e1327023 1375 while (bol > gs->buf && bol[-1] != '\n')
2944e4e6
RS
1376 bol--;
1377 lno--;
1378
1379 if (lno <= opt->last_shown)
1380 break;
1381
e1327023 1382 if (match_funcname(opt, gs, bol, eol)) {
89252cd0 1383 show_line(opt, bol, eol, gs->name, lno, 0, '=');
2944e4e6
RS
1384 break;
1385 }
1386 }
1387}
1388
a5dc20b0
RS
1389static int is_empty_line(const char *bol, const char *eol);
1390
e1327023 1391static void show_pre_context(struct grep_opt *opt, struct grep_source *gs,
1a845fbc 1392 const char *bol, const char *end, unsigned lno)
49de3216 1393{
6653a01b 1394 unsigned cur = lno, from = 1, funcname_lno = 0, orig_from;
a5dc20b0 1395 int funcname_needed = !!opt->funcname, comment_needed = 0;
ba8ea749 1396
49de3216
RS
1397 if (opt->pre_context < lno)
1398 from = lno - opt->pre_context;
1399 if (from <= opt->last_shown)
1400 from = opt->last_shown + 1;
6653a01b 1401 orig_from = from;
a5dc20b0
RS
1402 if (opt->funcbody) {
1403 if (match_funcname(opt, gs, bol, end))
1404 comment_needed = 1;
1405 else
1406 funcname_needed = 1;
6653a01b
RS
1407 from = opt->last_shown + 1;
1408 }
49de3216
RS
1409
1410 /* Rewind. */
6653a01b 1411 while (bol > gs->buf && cur > from) {
1a845fbc
JK
1412 const char *next_bol = bol;
1413 const char *eol = --bol;
2944e4e6 1414
e1327023 1415 while (bol > gs->buf && bol[-1] != '\n')
49de3216
RS
1416 bol--;
1417 cur--;
a5dc20b0
RS
1418 if (comment_needed && (is_empty_line(bol, eol) ||
1419 match_funcname(opt, gs, bol, eol))) {
1420 comment_needed = 0;
1421 from = orig_from;
1422 if (cur < from) {
1423 cur++;
1424 bol = next_bol;
1425 break;
1426 }
1427 }
e1327023 1428 if (funcname_needed && match_funcname(opt, gs, bol, eol)) {
2944e4e6
RS
1429 funcname_lno = cur;
1430 funcname_needed = 0;
a5dc20b0
RS
1431 if (opt->funcbody)
1432 comment_needed = 1;
1433 else
1434 from = orig_from;
2944e4e6 1435 }
49de3216
RS
1436 }
1437
2944e4e6
RS
1438 /* We need to look even further back to find a function signature. */
1439 if (opt->funcname && funcname_needed)
e1327023 1440 show_funcname_line(opt, gs, bol, cur);
2944e4e6 1441
49de3216
RS
1442 /* Back forward. */
1443 while (cur < lno) {
1a845fbc 1444 const char *eol = bol, sign = (cur == funcname_lno) ? '=' : '-';
49de3216
RS
1445
1446 while (*eol != '\n')
1447 eol++;
89252cd0 1448 show_line(opt, bol, eol, gs->name, cur, 0, sign);
49de3216
RS
1449 bol = eol + 1;
1450 cur++;
1451 }
1452}
1453
a26345b6
JH
1454static int should_lookahead(struct grep_opt *opt)
1455{
1456 struct grep_pat *p;
1457
1458 if (opt->extended)
1459 return 0; /* punt for too complex stuff */
1460 if (opt->invert)
1461 return 0;
1462 for (p = opt->pattern_list; p; p = p->next) {
1463 if (p->token != GREP_PATTERN)
1464 return 0; /* punt for "header only" and stuff */
1465 }
1466 return 1;
1467}
1468
1469static int look_ahead(struct grep_opt *opt,
1470 unsigned long *left_p,
1471 unsigned *lno_p,
1a845fbc 1472 const char **bol_p)
a26345b6
JH
1473{
1474 unsigned lno = *lno_p;
1a845fbc 1475 const char *bol = *bol_p;
a26345b6 1476 struct grep_pat *p;
1a845fbc 1477 const char *sp, *last_bol;
a26345b6
JH
1478 regoff_t earliest = -1;
1479
1480 for (p = opt->pattern_list; p; p = p->next) {
1481 int hit;
1482 regmatch_t m;
1483
97e77784 1484 hit = patmatch(p, bol, bol + *left_p, &m, 0);
a26345b6
JH
1485 if (!hit || m.rm_so < 0 || m.rm_eo < 0)
1486 continue;
1487 if (earliest < 0 || m.rm_so < earliest)
1488 earliest = m.rm_so;
1489 }
1490
1491 if (earliest < 0) {
1492 *bol_p = bol + *left_p;
1493 *left_p = 0;
1494 return 1;
1495 }
1496 for (sp = bol + earliest; bol < sp && sp[-1] != '\n'; sp--)
1497 ; /* find the beginning of the line */
1498 last_bol = sp;
1499
1500 for (sp = bol; sp < last_bol; sp++) {
1501 if (*sp == '\n')
1502 lno++;
1503 }
1504 *left_p -= last_bol - bol;
1505 *bol_p = last_bol;
1506 *lno_p = lno;
1507 return 0;
1508}
1509
38bbc2ea
NTND
1510static int fill_textconv_grep(struct repository *r,
1511 struct userdiff_driver *driver,
335ec3bf
JK
1512 struct grep_source *gs)
1513{
1514 struct diff_filespec *df;
1515 char *buf;
1516 size_t size;
1517
1518 if (!driver || !driver->textconv)
1519 return grep_source_load(gs);
1520
1521 /*
1522 * The textconv interface is intimately tied to diff_filespecs, so we
1523 * have to pretend to be one. If we could unify the grep_source
1524 * and diff_filespec structs, this mess could just go away.
1525 */
1526 df = alloc_filespec(gs->path);
1527 switch (gs->type) {
1c41c82b 1528 case GREP_SOURCE_OID:
335ec3bf
JK
1529 fill_filespec(df, gs->identifier, 1, 0100644);
1530 break;
1531 case GREP_SOURCE_FILE:
14228447 1532 fill_filespec(df, null_oid(), 0, 0100644);
335ec3bf
JK
1533 break;
1534 default:
033abf97 1535 BUG("attempt to textconv something without a path?");
335ec3bf
JK
1536 }
1537
1538 /*
1d1729ca
MT
1539 * fill_textconv is not remotely thread-safe; it modifies the global
1540 * diff tempfile structure, writes to the_repo's odb and might
1541 * internally call thread-unsafe functions such as the
1542 * prepare_packed_git() lazy-initializator. Because of the last two, we
1543 * must ensure mutual exclusion between this call and the object reading
1544 * API, thus we use obj_read_lock() here.
1545 *
1546 * TODO: allowing text conversion to run in parallel with object
1547 * reading operations might increase performance in the multithreaded
1548 * non-worktreee git-grep with --textconv.
335ec3bf 1549 */
1d1729ca 1550 obj_read_lock();
38bbc2ea 1551 size = fill_textconv(r, driver, df, &buf);
1d1729ca 1552 obj_read_unlock();
335ec3bf
JK
1553 free_filespec(df);
1554
1555 /*
1556 * The normal fill_textconv usage by the diff machinery would just keep
1557 * the textconv'd buf separate from the diff_filespec. But much of the
1558 * grep code passes around a grep_source and assumes that its "buf"
1559 * pointer is the beginning of the thing we are searching. So let's
1560 * install our textconv'd version into the grep_source, taking care not
1561 * to leak any existing buffer.
1562 */
1563 grep_source_clear_data(gs);
1564 gs->buf = buf;
1565 gs->size = size;
1566
1567 return 0;
1568}
1569
4aa2c475
RS
1570static int is_empty_line(const char *bol, const char *eol)
1571{
1572 while (bol < eol && isspace(*bol))
1573 bol++;
1574 return bol == eol;
1575}
1576
e1327023 1577static int grep_source_1(struct grep_opt *opt, struct grep_source *gs, int collect_hits)
83b5d2f5 1578{
1a845fbc
JK
1579 const char *bol;
1580 const char *peek_bol = NULL;
e1327023 1581 unsigned long left;
83b5d2f5 1582 unsigned lno = 1;
83b5d2f5 1583 unsigned last_hit = 0;
83b5d2f5 1584 int binary_match_only = 0;
83b5d2f5 1585 unsigned count = 0;
a26345b6 1586 int try_lookahead = 0;
ba8ea749 1587 int show_function = 0;
335ec3bf 1588 struct userdiff_driver *textconv = NULL;
480c1ca6 1589 enum grep_context ctx = GREP_CONTEXT_HEAD;
60ecac98 1590 xdemitconf_t xecfg;
83b5d2f5 1591
de99eb0c
ES
1592 if (!opt->status_only && gs->name == NULL)
1593 BUG("grep call which could print a name requires "
1594 "grep_source.name be non-NULL");
1595
5b594f45
FK
1596 if (!opt->output)
1597 opt->output = std_output;
1598
ba8ea749
RS
1599 if (opt->pre_context || opt->post_context || opt->file_break ||
1600 opt->funcbody) {
08303c36
RS
1601 /* Show hunk marks, except for the first file. */
1602 if (opt->last_shown)
1603 opt->show_hunk_mark = 1;
1604 /*
1605 * If we're using threads then we can't easily identify
1606 * the first file. Always put hunk marks in that case
1607 * and skip the very first one later in work_done().
1608 */
1609 if (opt->output != std_output)
1610 opt->show_hunk_mark = 1;
1611 }
431d6e7b
RS
1612 opt->last_shown = 0;
1613
335ec3bf 1614 if (opt->allow_textconv) {
acd00ea0 1615 grep_source_load_driver(gs, opt->repo->index);
335ec3bf
JK
1616 /*
1617 * We might set up the shared textconv cache data here, which
1d1729ca
MT
1618 * is not thread-safe. Also, get_oid_with_context() and
1619 * parse_object() might be internally called. As they are not
84544f2e 1620 * currently thread-safe and might be racy with object reading,
1d1729ca 1621 * obj_read_lock() must be called.
335ec3bf
JK
1622 */
1623 grep_attr_lock();
1d1729ca 1624 obj_read_lock();
bd7ad45b 1625 textconv = userdiff_get_textconv(opt->repo, gs->driver);
1d1729ca 1626 obj_read_unlock();
335ec3bf
JK
1627 grep_attr_unlock();
1628 }
1629
1630 /*
1631 * We know the result of a textconv is text, so we only have to care
1632 * about binary handling if we are not using it.
1633 */
1634 if (!textconv) {
1635 switch (opt->binary) {
1636 case GREP_BINARY_DEFAULT:
acd00ea0 1637 if (grep_source_is_binary(gs, opt->repo->index))
335ec3bf
JK
1638 binary_match_only = 1;
1639 break;
1640 case GREP_BINARY_NOMATCH:
acd00ea0 1641 if (grep_source_is_binary(gs, opt->repo->index))
335ec3bf
JK
1642 return 0; /* Assume unmatch */
1643 break;
1644 case GREP_BINARY_TEXT:
1645 break;
1646 default:
033abf97 1647 BUG("unknown binary handling mode");
335ec3bf 1648 }
83b5d2f5
JH
1649 }
1650
60ecac98 1651 memset(&xecfg, 0, sizeof(xecfg));
0579f91d
TR
1652 opt->priv = &xecfg;
1653
a26345b6 1654 try_lookahead = should_lookahead(opt);
60ecac98 1655
38bbc2ea 1656 if (fill_textconv_grep(opt->repo, textconv, gs) < 0)
08265798
JK
1657 return 0;
1658
e1327023
JK
1659 bol = gs->buf;
1660 left = gs->size;
83b5d2f5 1661 while (left) {
1a845fbc 1662 const char *eol;
0ab7befa 1663 int hit;
89252cd0 1664 ssize_t cno;
68d686e6 1665 ssize_t col = -1, icol = -1;
83b5d2f5 1666
a26345b6 1667 /*
8997da38 1668 * look_ahead() skips quickly to the line that possibly
a26345b6
JH
1669 * has the next hit; don't call it if we need to do
1670 * something more than just skipping the current line
1671 * in response to an unmatch for the current line. E.g.
1672 * inside a post-context window, we will show the current
1673 * line as a context around the previous hit when it
1674 * doesn't hit.
1675 */
1676 if (try_lookahead
1677 && !(last_hit
ba8ea749
RS
1678 && (show_function ||
1679 lno <= last_hit + opt->post_context))
a26345b6
JH
1680 && look_ahead(opt, &left, &lno, &bol))
1681 break;
83b5d2f5 1682 eol = end_of_line(bol, &left);
83b5d2f5 1683
480c1ca6
JH
1684 if ((ctx == GREP_CONTEXT_HEAD) && (eol == bol))
1685 ctx = GREP_CONTEXT_BODY;
1686
68d686e6 1687 hit = match_line(opt, bol, eol, &col, &icol, ctx, collect_hits);
83b5d2f5 1688
0ab7befa
JH
1689 if (collect_hits)
1690 goto next_line;
1691
83b5d2f5
JH
1692 /* "grep -v -e foo -e bla" should list lines
1693 * that do not have either, so inversion should
1694 * be done outside.
1695 */
1696 if (opt->invert)
1697 hit = !hit;
1698 if (opt->unmatch_name_only) {
1699 if (hit)
1700 return 0;
1701 goto next_line;
1702 }
1703 if (hit) {
1704 count++;
1705 if (opt->status_only)
1706 return 1;
321ffcc0 1707 if (opt->name_only) {
e1327023 1708 show_name(opt, gs->name);
321ffcc0
RS
1709 return 1;
1710 }
c30c10cf
RS
1711 if (opt->count)
1712 goto next_line;
83b5d2f5 1713 if (binary_match_only) {
5b594f45 1714 opt->output(opt, "Binary file ", 12);
e1327023 1715 output_color(opt, gs->name, strlen(gs->name),
fa151dc5 1716 opt->colors[GREP_COLOR_FILENAME]);
5b594f45 1717 opt->output(opt, " matches\n", 9);
83b5d2f5
JH
1718 return 1;
1719 }
83b5d2f5
JH
1720 /* Hit at this line. If we haven't shown the
1721 * pre-context lines, we would need to show them.
83b5d2f5 1722 */
ba8ea749 1723 if (opt->pre_context || opt->funcbody)
e1327023 1724 show_pre_context(opt, gs, bol, eol, lno);
2944e4e6 1725 else if (opt->funcname)
e1327023 1726 show_funcname_line(opt, gs, bol, lno);
89252cd0
TB
1727 cno = opt->invert ? icol : col;
1728 if (cno < 0) {
1729 /*
1730 * A negative cno indicates that there was no
1731 * match on the line. We are thus inverted and
1732 * being asked to show all lines that _don't_
1733 * match a given expression. Therefore, set cno
1734 * to 0 to suggest the whole line matches.
1735 */
1736 cno = 0;
1737 }
1738 show_line(opt, bol, eol, gs->name, lno, cno + 1, ':');
5dd06d38 1739 last_hit = lno;
ba8ea749
RS
1740 if (opt->funcbody)
1741 show_function = 1;
1742 goto next_line;
83b5d2f5 1743 }
4aa2c475
RS
1744 if (show_function && (!peek_bol || peek_bol < bol)) {
1745 unsigned long peek_left = left;
1a845fbc 1746 const char *peek_eol = eol;
4aa2c475
RS
1747
1748 /*
1749 * Trailing empty lines are not interesting.
1750 * Peek past them to see if they belong to the
1751 * body of the current function.
1752 */
1753 peek_bol = bol;
1754 while (is_empty_line(peek_bol, peek_eol)) {
1755 peek_bol = peek_eol + 1;
1756 peek_eol = end_of_line(peek_bol, &peek_left);
1757 }
1758
1759 if (match_funcname(opt, gs, peek_bol, peek_eol))
1760 show_function = 0;
1761 }
ba8ea749
RS
1762 if (show_function ||
1763 (last_hit && lno <= last_hit + opt->post_context)) {
83b5d2f5
JH
1764 /* If the last hit is within the post context,
1765 * we need to show this line.
1766 */
89252cd0 1767 show_line(opt, bol, eol, gs->name, lno, col + 1, '-');
83b5d2f5 1768 }
83b5d2f5
JH
1769
1770 next_line:
1771 bol = eol + 1;
1772 if (!left)
1773 break;
1774 left--;
1775 lno++;
1776 }
1777
0ab7befa
JH
1778 if (collect_hits)
1779 return 0;
b48fb5b6 1780
83b5d2f5 1781 if (opt->status_only)
e1f68c66 1782 return opt->unmatch_name_only;
83b5d2f5
JH
1783 if (opt->unmatch_name_only) {
1784 /* We did not see any hit, so we want to show this */
e1327023 1785 show_name(opt, gs->name);
83b5d2f5
JH
1786 return 1;
1787 }
1788
60ecac98
RS
1789 xdiff_clear_find_func(&xecfg);
1790 opt->priv = NULL;
1791
83b5d2f5
JH
1792 /* NEEDSWORK:
1793 * The real "grep -c foo *.c" gives many "bar.c:0" lines,
1794 * which feels mostly useless but sometimes useful. Maybe
1795 * make it another option? For now suppress them.
1796 */
5b594f45
FK
1797 if (opt->count && count) {
1798 char buf[32];
f76d947a
RS
1799 if (opt->pathname) {
1800 output_color(opt, gs->name, strlen(gs->name),
fa151dc5 1801 opt->colors[GREP_COLOR_FILENAME]);
f76d947a
RS
1802 output_sep(opt, ':');
1803 }
1a168e5c 1804 xsnprintf(buf, sizeof(buf), "%u\n", count);
5b594f45 1805 opt->output(opt, buf, strlen(buf));
c30c10cf 1806 return 1;
5b594f45 1807 }
83b5d2f5
JH
1808 return !!last_hit;
1809}
1810
0ab7befa
JH
1811static void clr_hit_marker(struct grep_expr *x)
1812{
1813 /* All-hit markers are meaningful only at the very top level
1814 * OR node.
1815 */
1816 while (1) {
1817 x->hit = 0;
1818 if (x->node != GREP_NODE_OR)
1819 return;
1820 x->u.binary.left->hit = 0;
1821 x = x->u.binary.right;
1822 }
1823}
1824
1825static int chk_hit_marker(struct grep_expr *x)
1826{
1827 /* Top level nodes have hit markers. See if they all are hits */
1828 while (1) {
1829 if (x->node != GREP_NODE_OR)
1830 return x->hit;
1831 if (!x->u.binary.left->hit)
1832 return 0;
1833 x = x->u.binary.right;
1834 }
1835}
1836
e1327023 1837int grep_source(struct grep_opt *opt, struct grep_source *gs)
0ab7befa
JH
1838{
1839 /*
1840 * we do not have to do the two-pass grep when we do not check
1841 * buffer-wide "all-match".
1842 */
794c0002 1843 if (!opt->all_match && !opt->no_body_match)
e1327023 1844 return grep_source_1(opt, gs, 0);
0ab7befa
JH
1845
1846 /* Otherwise the toplevel "or" terms hit a bit differently.
1847 * We first clear hit markers from them.
1848 */
1849 clr_hit_marker(opt->pattern_expression);
794c0002 1850 opt->body_hit = 0;
e1327023 1851 grep_source_1(opt, gs, 1);
0ab7befa 1852
794c0002
RS
1853 if (opt->all_match && !chk_hit_marker(opt->pattern_expression))
1854 return 0;
1855 if (opt->no_body_match && opt->body_hit)
0ab7befa
JH
1856 return 0;
1857
e1327023
JK
1858 return grep_source_1(opt, gs, 0);
1859}
1860
1e668716
JK
1861static void grep_source_init_buf(struct grep_source *gs,
1862 const char *buf,
50d92b5f
JT
1863 unsigned long size)
1864{
1865 gs->type = GREP_SOURCE_BUF;
1866 gs->name = NULL;
1867 gs->path = NULL;
1868 gs->buf = buf;
1869 gs->size = size;
1870 gs->driver = NULL;
1871 gs->identifier = NULL;
1872}
1873
1e668716 1874int grep_buffer(struct grep_opt *opt, const char *buf, unsigned long size)
e1327023
JK
1875{
1876 struct grep_source gs;
1877 int r;
1878
50d92b5f 1879 grep_source_init_buf(&gs, buf, size);
e1327023
JK
1880
1881 r = grep_source(opt, &gs);
1882
1883 grep_source_clear(&gs);
1884 return r;
1885}
1886
50d92b5f
JT
1887void grep_source_init_file(struct grep_source *gs, const char *name,
1888 const char *path)
e1327023 1889{
50d92b5f 1890 gs->type = GREP_SOURCE_FILE;
8c53f071
JK
1891 gs->name = xstrdup_or_null(name);
1892 gs->path = xstrdup_or_null(path);
e1327023
JK
1893 gs->buf = NULL;
1894 gs->size = 0;
94ad9d9e 1895 gs->driver = NULL;
50d92b5f
JT
1896 gs->identifier = xstrdup(path);
1897}
e1327023 1898
50d92b5f 1899void grep_source_init_oid(struct grep_source *gs, const char *name,
0693806b
JT
1900 const char *path, const struct object_id *oid,
1901 struct repository *repo)
50d92b5f
JT
1902{
1903 gs->type = GREP_SOURCE_OID;
1904 gs->name = xstrdup_or_null(name);
1905 gs->path = xstrdup_or_null(path);
1906 gs->buf = NULL;
1907 gs->size = 0;
1908 gs->driver = NULL;
1909 gs->identifier = oiddup(oid);
0693806b 1910 gs->repo = repo;
e1327023
JK
1911}
1912
1913void grep_source_clear(struct grep_source *gs)
1914{
88ce3ef6
ÆAB
1915 FREE_AND_NULL(gs->name);
1916 FREE_AND_NULL(gs->path);
1917 FREE_AND_NULL(gs->identifier);
e1327023
JK
1918 grep_source_clear_data(gs);
1919}
1920
1921void grep_source_clear_data(struct grep_source *gs)
1922{
1923 switch (gs->type) {
1924 case GREP_SOURCE_FILE:
1c41c82b 1925 case GREP_SOURCE_OID:
1e668716
JK
1926 /* these types own the buffer */
1927 free((char *)gs->buf);
1928 gs->buf = NULL;
e1327023
JK
1929 gs->size = 0;
1930 break;
1931 case GREP_SOURCE_BUF:
1932 /* leave user-provided buf intact */
1933 break;
1934 }
1935}
1936
1c41c82b 1937static int grep_source_load_oid(struct grep_source *gs)
e1327023
JK
1938{
1939 enum object_type type;
1940
0693806b
JT
1941 gs->buf = repo_read_object_file(gs->repo, gs->identifier, &type,
1942 &gs->size);
e1327023
JK
1943 if (!gs->buf)
1944 return error(_("'%s': unable to read %s"),
1945 gs->name,
1c41c82b 1946 oid_to_hex(gs->identifier));
e1327023
JK
1947 return 0;
1948}
1949
1950static int grep_source_load_file(struct grep_source *gs)
1951{
1952 const char *filename = gs->identifier;
1953 struct stat st;
1954 char *data;
1955 size_t size;
1956 int i;
1957
1958 if (lstat(filename, &st) < 0) {
1959 err_ret:
1960 if (errno != ENOENT)
7645d8f1 1961 error_errno(_("failed to stat '%s'"), filename);
e1327023
JK
1962 return -1;
1963 }
1964 if (!S_ISREG(st.st_mode))
1965 return -1;
1966 size = xsize_t(st.st_size);
1967 i = open(filename, O_RDONLY);
1968 if (i < 0)
1969 goto err_ret;
3733e694 1970 data = xmallocz(size);
e1327023 1971 if (st.st_size != read_in_full(i, data, size)) {
7645d8f1 1972 error_errno(_("'%s': short read"), filename);
e1327023
JK
1973 close(i);
1974 free(data);
1975 return -1;
1976 }
1977 close(i);
e1327023
JK
1978
1979 gs->buf = data;
1980 gs->size = size;
1981 return 0;
1982}
1983
3083301e 1984static int grep_source_load(struct grep_source *gs)
e1327023
JK
1985{
1986 if (gs->buf)
1987 return 0;
1988
1989 switch (gs->type) {
1990 case GREP_SOURCE_FILE:
1991 return grep_source_load_file(gs);
1c41c82b
BW
1992 case GREP_SOURCE_OID:
1993 return grep_source_load_oid(gs);
e1327023
JK
1994 case GREP_SOURCE_BUF:
1995 return gs->buf ? 0 : -1;
1996 }
033abf97 1997 BUG("invalid grep_source type to load");
0ab7befa 1998}
94ad9d9e 1999
acd00ea0
NTND
2000void grep_source_load_driver(struct grep_source *gs,
2001 struct index_state *istate)
94ad9d9e
JK
2002{
2003 if (gs->driver)
2004 return;
2005
2006 grep_attr_lock();
1d1729ca 2007 if (gs->path)
acd00ea0 2008 gs->driver = userdiff_find_by_path(istate, gs->path);
94ad9d9e
JK
2009 if (!gs->driver)
2010 gs->driver = userdiff_find_by_name("default");
2011 grep_attr_unlock();
2012}
41b59bfc 2013
acd00ea0
NTND
2014static int grep_source_is_binary(struct grep_source *gs,
2015 struct index_state *istate)
41b59bfc 2016{
acd00ea0 2017 grep_source_load_driver(gs, istate);
41b59bfc
JK
2018 if (gs->driver->binary != -1)
2019 return gs->driver->binary;
2020
2021 if (!grep_source_load(gs))
2022 return buffer_is_binary(gs->buf, gs->size);
2023
2024 return 0;
2025}