Allow diff and index commands to be interrupted
[git/dscho.git] / diff.c
blobec51e7dd1832013cda561fa803668b1f2213c96a
1 /*
2 * Copyright (C) 2005 Junio C Hamano
3 */
4 #include <sys/types.h>
5 #include <sys/wait.h>
6 #include <signal.h>
7 #include "cache.h"
8 #include "quote.h"
9 #include "diff.h"
10 #include "diffcore.h"
12 static const char *diff_opts = "-pu";
14 static int use_size_cache;
16 int diff_rename_limit_default = -1;
18 int git_diff_config(const char *var, const char *value)
20 if (!strcmp(var, "diff.renamelimit")) {
21 diff_rename_limit_default = git_config_int(var, value);
22 return 0;
25 return git_default_config(var, value);
28 static char *quote_one(const char *str)
30 int needlen;
31 char *xp;
33 if (!str)
34 return NULL;
35 needlen = quote_c_style(str, NULL, NULL, 0);
36 if (!needlen)
37 return strdup(str);
38 xp = xmalloc(needlen + 1);
39 quote_c_style(str, xp, NULL, 0);
40 return xp;
43 static char *quote_two(const char *one, const char *two)
45 int need_one = quote_c_style(one, NULL, NULL, 1);
46 int need_two = quote_c_style(two, NULL, NULL, 1);
47 char *xp;
49 if (need_one + need_two) {
50 if (!need_one) need_one = strlen(one);
51 if (!need_two) need_one = strlen(two);
53 xp = xmalloc(need_one + need_two + 3);
54 xp[0] = '"';
55 quote_c_style(one, xp + 1, NULL, 1);
56 quote_c_style(two, xp + need_one + 1, NULL, 1);
57 strcpy(xp + need_one + need_two + 1, "\"");
58 return xp;
60 need_one = strlen(one);
61 need_two = strlen(two);
62 xp = xmalloc(need_one + need_two + 1);
63 strcpy(xp, one);
64 strcpy(xp + need_one, two);
65 return xp;
68 static const char *external_diff(void)
70 static const char *external_diff_cmd = NULL;
71 static int done_preparing = 0;
72 const char *env_diff_opts;
74 if (done_preparing)
75 return external_diff_cmd;
78 * Default values above are meant to match the
79 * Linux kernel development style. Examples of
80 * alternative styles you can specify via environment
81 * variables are:
83 * GIT_DIFF_OPTS="-c";
85 external_diff_cmd = getenv("GIT_EXTERNAL_DIFF");
87 /* In case external diff fails... */
88 env_diff_opts = getenv("GIT_DIFF_OPTS");
89 if (env_diff_opts) diff_opts = env_diff_opts;
91 done_preparing = 1;
92 return external_diff_cmd;
95 #define TEMPFILE_PATH_LEN 50
97 static struct diff_tempfile {
98 const char *name; /* filename external diff should read from */
99 char hex[41];
100 char mode[10];
101 char tmp_path[TEMPFILE_PATH_LEN];
102 } diff_temp[2];
104 static int count_lines(const char *filename)
106 FILE *in;
107 int count, ch, completely_empty = 1, nl_just_seen = 0;
108 in = fopen(filename, "r");
109 count = 0;
110 while ((ch = fgetc(in)) != EOF)
111 if (ch == '\n') {
112 count++;
113 nl_just_seen = 1;
114 completely_empty = 0;
116 else {
117 nl_just_seen = 0;
118 completely_empty = 0;
120 fclose(in);
121 if (completely_empty)
122 return 0;
123 if (!nl_just_seen)
124 count++; /* no trailing newline */
125 return count;
128 static void print_line_count(int count)
130 switch (count) {
131 case 0:
132 printf("0,0");
133 break;
134 case 1:
135 printf("1");
136 break;
137 default:
138 printf("1,%d", count);
139 break;
143 static void copy_file(int prefix, const char *filename)
145 FILE *in;
146 int ch, nl_just_seen = 1;
147 in = fopen(filename, "r");
148 while ((ch = fgetc(in)) != EOF) {
149 if (nl_just_seen)
150 putchar(prefix);
151 putchar(ch);
152 if (ch == '\n')
153 nl_just_seen = 1;
154 else
155 nl_just_seen = 0;
157 fclose(in);
158 if (!nl_just_seen)
159 printf("\n\\ No newline at end of file\n");
162 static void emit_rewrite_diff(const char *name_a,
163 const char *name_b,
164 struct diff_tempfile *temp)
166 /* Use temp[i].name as input, name_a and name_b as labels */
167 int lc_a, lc_b;
168 lc_a = count_lines(temp[0].name);
169 lc_b = count_lines(temp[1].name);
170 printf("--- %s\n+++ %s\n@@ -", name_a, name_b);
171 print_line_count(lc_a);
172 printf(" +");
173 print_line_count(lc_b);
174 printf(" @@\n");
175 if (lc_a)
176 copy_file('-', temp[0].name);
177 if (lc_b)
178 copy_file('+', temp[1].name);
181 static void builtin_diff(const char *name_a,
182 const char *name_b,
183 struct diff_tempfile *temp,
184 const char *xfrm_msg,
185 int complete_rewrite)
187 int i, next_at, cmd_size;
188 const char *const diff_cmd = "diff -L%s -L%s";
189 const char *const diff_arg = "-- %s %s||:"; /* "||:" is to return 0 */
190 const char *input_name_sq[2];
191 const char *label_path[2];
192 char *cmd;
194 /* diff_cmd and diff_arg have 4 %s in total which makes
195 * the sum of these strings 8 bytes larger than required.
196 * we use 2 spaces around diff-opts, and we need to count
197 * terminating NUL; we used to subtract 5 here, but we do not
198 * care about small leaks in this subprocess that is about
199 * to exec "diff" anymore.
201 cmd_size = (strlen(diff_cmd) + strlen(diff_opts) + strlen(diff_arg)
202 + 128);
204 for (i = 0; i < 2; i++) {
205 input_name_sq[i] = sq_quote(temp[i].name);
206 if (!strcmp(temp[i].name, "/dev/null"))
207 label_path[i] = "/dev/null";
208 else if (!i)
209 label_path[i] = sq_quote(quote_two("a/", name_a));
210 else
211 label_path[i] = sq_quote(quote_two("b/", name_b));
212 cmd_size += (strlen(label_path[i]) + strlen(input_name_sq[i]));
215 cmd = xmalloc(cmd_size);
217 next_at = 0;
218 next_at += snprintf(cmd+next_at, cmd_size-next_at,
219 diff_cmd, label_path[0], label_path[1]);
220 next_at += snprintf(cmd+next_at, cmd_size-next_at,
221 " %s ", diff_opts);
222 next_at += snprintf(cmd+next_at, cmd_size-next_at,
223 diff_arg, input_name_sq[0], input_name_sq[1]);
225 printf("diff --git %s %s\n",
226 quote_two("a/", name_a), quote_two("b/", name_b));
227 if (label_path[0][0] == '/') {
228 /* dev/null */
229 printf("new file mode %s\n", temp[1].mode);
230 if (xfrm_msg && xfrm_msg[0])
231 puts(xfrm_msg);
233 else if (label_path[1][0] == '/') {
234 printf("deleted file mode %s\n", temp[0].mode);
235 if (xfrm_msg && xfrm_msg[0])
236 puts(xfrm_msg);
238 else {
239 if (strcmp(temp[0].mode, temp[1].mode)) {
240 printf("old mode %s\n", temp[0].mode);
241 printf("new mode %s\n", temp[1].mode);
243 if (xfrm_msg && xfrm_msg[0])
244 puts(xfrm_msg);
245 if (strncmp(temp[0].mode, temp[1].mode, 3))
246 /* we do not run diff between different kind
247 * of objects.
249 exit(0);
250 if (complete_rewrite) {
251 fflush(NULL);
252 emit_rewrite_diff(name_a, name_b, temp);
253 exit(0);
256 fflush(NULL);
257 execlp("/bin/sh","sh", "-c", cmd, NULL);
260 struct diff_filespec *alloc_filespec(const char *path)
262 int namelen = strlen(path);
263 struct diff_filespec *spec = xmalloc(sizeof(*spec) + namelen + 1);
265 memset(spec, 0, sizeof(*spec));
266 spec->path = (char *)(spec + 1);
267 memcpy(spec->path, path, namelen+1);
268 return spec;
271 void fill_filespec(struct diff_filespec *spec, const unsigned char *sha1,
272 unsigned short mode)
274 if (mode) {
275 spec->mode = DIFF_FILE_CANON_MODE(mode);
276 memcpy(spec->sha1, sha1, 20);
277 spec->sha1_valid = !!memcmp(sha1, null_sha1, 20);
282 * Given a name and sha1 pair, if the dircache tells us the file in
283 * the work tree has that object contents, return true, so that
284 * prepare_temp_file() does not have to inflate and extract.
286 static int work_tree_matches(const char *name, const unsigned char *sha1)
288 struct cache_entry *ce;
289 struct stat st;
290 int pos, len;
292 /* We do not read the cache ourselves here, because the
293 * benchmark with my previous version that always reads cache
294 * shows that it makes things worse for diff-tree comparing
295 * two linux-2.6 kernel trees in an already checked out work
296 * tree. This is because most diff-tree comparisons deal with
297 * only a small number of files, while reading the cache is
298 * expensive for a large project, and its cost outweighs the
299 * savings we get by not inflating the object to a temporary
300 * file. Practically, this code only helps when we are used
301 * by diff-cache --cached, which does read the cache before
302 * calling us.
304 if (!active_cache)
305 return 0;
307 len = strlen(name);
308 pos = cache_name_pos(name, len);
309 if (pos < 0)
310 return 0;
311 ce = active_cache[pos];
312 if ((lstat(name, &st) < 0) ||
313 !S_ISREG(st.st_mode) || /* careful! */
314 ce_match_stat(ce, &st) ||
315 memcmp(sha1, ce->sha1, 20))
316 return 0;
317 /* we return 1 only when we can stat, it is a regular file,
318 * stat information matches, and sha1 recorded in the cache
319 * matches. I.e. we know the file in the work tree really is
320 * the same as the <name, sha1> pair.
322 return 1;
325 static struct sha1_size_cache {
326 unsigned char sha1[20];
327 unsigned long size;
328 } **sha1_size_cache;
329 static int sha1_size_cache_nr, sha1_size_cache_alloc;
331 static struct sha1_size_cache *locate_size_cache(unsigned char *sha1,
332 int find_only,
333 unsigned long size)
335 int first, last;
336 struct sha1_size_cache *e;
338 first = 0;
339 last = sha1_size_cache_nr;
340 while (last > first) {
341 int cmp, next = (last + first) >> 1;
342 e = sha1_size_cache[next];
343 cmp = memcmp(e->sha1, sha1, 20);
344 if (!cmp)
345 return e;
346 if (cmp < 0) {
347 last = next;
348 continue;
350 first = next+1;
352 /* not found */
353 if (find_only)
354 return NULL;
355 /* insert to make it at "first" */
356 if (sha1_size_cache_alloc <= sha1_size_cache_nr) {
357 sha1_size_cache_alloc = alloc_nr(sha1_size_cache_alloc);
358 sha1_size_cache = xrealloc(sha1_size_cache,
359 sha1_size_cache_alloc *
360 sizeof(*sha1_size_cache));
362 sha1_size_cache_nr++;
363 if (first < sha1_size_cache_nr)
364 memmove(sha1_size_cache + first + 1, sha1_size_cache + first,
365 (sha1_size_cache_nr - first - 1) *
366 sizeof(*sha1_size_cache));
367 e = xmalloc(sizeof(struct sha1_size_cache));
368 sha1_size_cache[first] = e;
369 memcpy(e->sha1, sha1, 20);
370 e->size = size;
371 return e;
375 * While doing rename detection and pickaxe operation, we may need to
376 * grab the data for the blob (or file) for our own in-core comparison.
377 * diff_filespec has data and size fields for this purpose.
379 int diff_populate_filespec(struct diff_filespec *s, int size_only)
381 int err = 0;
382 if (!DIFF_FILE_VALID(s))
383 die("internal error: asking to populate invalid file.");
384 if (S_ISDIR(s->mode))
385 return -1;
387 if (!use_size_cache)
388 size_only = 0;
390 if (s->data)
391 return err;
392 if (!s->sha1_valid ||
393 work_tree_matches(s->path, s->sha1)) {
394 struct stat st;
395 int fd;
396 if (lstat(s->path, &st) < 0) {
397 if (errno == ENOENT) {
398 err_empty:
399 err = -1;
400 empty:
401 s->data = "";
402 s->size = 0;
403 return err;
406 s->size = st.st_size;
407 if (!s->size)
408 goto empty;
409 if (size_only)
410 return 0;
411 if (S_ISLNK(st.st_mode)) {
412 int ret;
413 s->data = xmalloc(s->size);
414 s->should_free = 1;
415 ret = readlink(s->path, s->data, s->size);
416 if (ret < 0) {
417 free(s->data);
418 goto err_empty;
420 return 0;
422 fd = open(s->path, O_RDONLY);
423 if (fd < 0)
424 goto err_empty;
425 s->data = mmap(NULL, s->size, PROT_READ, MAP_PRIVATE, fd, 0);
426 close(fd);
427 if (s->data == MAP_FAILED)
428 goto err_empty;
429 s->should_munmap = 1;
431 else {
432 char type[20];
433 struct sha1_size_cache *e;
435 if (size_only) {
436 e = locate_size_cache(s->sha1, 1, 0);
437 if (e) {
438 s->size = e->size;
439 return 0;
441 if (!sha1_object_info(s->sha1, type, &s->size))
442 locate_size_cache(s->sha1, 0, s->size);
444 else {
445 s->data = read_sha1_file(s->sha1, type, &s->size);
446 s->should_free = 1;
449 return 0;
452 void diff_free_filespec_data(struct diff_filespec *s)
454 if (s->should_free)
455 free(s->data);
456 else if (s->should_munmap)
457 munmap(s->data, s->size);
458 s->should_free = s->should_munmap = 0;
459 s->data = NULL;
462 static void prep_temp_blob(struct diff_tempfile *temp,
463 void *blob,
464 unsigned long size,
465 const unsigned char *sha1,
466 int mode)
468 int fd;
470 fd = git_mkstemp(temp->tmp_path, TEMPFILE_PATH_LEN, ".diff_XXXXXX");
471 if (fd < 0)
472 die("unable to create temp-file");
473 if (write(fd, blob, size) != size)
474 die("unable to write temp-file");
475 close(fd);
476 temp->name = temp->tmp_path;
477 strcpy(temp->hex, sha1_to_hex(sha1));
478 temp->hex[40] = 0;
479 sprintf(temp->mode, "%06o", mode);
482 static void prepare_temp_file(const char *name,
483 struct diff_tempfile *temp,
484 struct diff_filespec *one)
486 if (!DIFF_FILE_VALID(one)) {
487 not_a_valid_file:
488 /* A '-' entry produces this for file-2, and
489 * a '+' entry produces this for file-1.
491 temp->name = "/dev/null";
492 strcpy(temp->hex, ".");
493 strcpy(temp->mode, ".");
494 return;
497 if (!one->sha1_valid ||
498 work_tree_matches(name, one->sha1)) {
499 struct stat st;
500 if (lstat(name, &st) < 0) {
501 if (errno == ENOENT)
502 goto not_a_valid_file;
503 die("stat(%s): %s", name, strerror(errno));
505 if (S_ISLNK(st.st_mode)) {
506 int ret;
507 char buf[PATH_MAX + 1]; /* ought to be SYMLINK_MAX */
508 if (sizeof(buf) <= st.st_size)
509 die("symlink too long: %s", name);
510 ret = readlink(name, buf, st.st_size);
511 if (ret < 0)
512 die("readlink(%s)", name);
513 prep_temp_blob(temp, buf, st.st_size,
514 (one->sha1_valid ?
515 one->sha1 : null_sha1),
516 (one->sha1_valid ?
517 one->mode : S_IFLNK));
519 else {
520 /* we can borrow from the file in the work tree */
521 temp->name = name;
522 if (!one->sha1_valid)
523 strcpy(temp->hex, sha1_to_hex(null_sha1));
524 else
525 strcpy(temp->hex, sha1_to_hex(one->sha1));
526 /* Even though we may sometimes borrow the
527 * contents from the work tree, we always want
528 * one->mode. mode is trustworthy even when
529 * !(one->sha1_valid), as long as
530 * DIFF_FILE_VALID(one).
532 sprintf(temp->mode, "%06o", one->mode);
534 return;
536 else {
537 if (diff_populate_filespec(one, 0))
538 die("cannot read data blob for %s", one->path);
539 prep_temp_blob(temp, one->data, one->size,
540 one->sha1, one->mode);
544 static void remove_tempfile(void)
546 int i;
548 for (i = 0; i < 2; i++)
549 if (diff_temp[i].name == diff_temp[i].tmp_path) {
550 unlink(diff_temp[i].name);
551 diff_temp[i].name = NULL;
555 static void remove_tempfile_on_signal(int signo)
557 remove_tempfile();
558 signal(SIGINT, SIG_DFL);
559 raise(signo);
562 /* An external diff command takes:
564 * diff-cmd name infile1 infile1-sha1 infile1-mode \
565 * infile2 infile2-sha1 infile2-mode [ rename-to ]
568 static void run_external_diff(const char *pgm,
569 const char *name,
570 const char *other,
571 struct diff_filespec *one,
572 struct diff_filespec *two,
573 const char *xfrm_msg,
574 int complete_rewrite)
576 struct diff_tempfile *temp = diff_temp;
577 pid_t pid;
578 int status;
579 static int atexit_asked = 0;
580 const char *othername;
582 othername = (other? other : name);
583 if (one && two) {
584 prepare_temp_file(name, &temp[0], one);
585 prepare_temp_file(othername, &temp[1], two);
586 if (! atexit_asked &&
587 (temp[0].name == temp[0].tmp_path ||
588 temp[1].name == temp[1].tmp_path)) {
589 atexit_asked = 1;
590 atexit(remove_tempfile);
592 signal(SIGINT, remove_tempfile_on_signal);
595 fflush(NULL);
596 pid = fork();
597 if (pid < 0)
598 die("unable to fork");
599 if (!pid) {
600 if (pgm) {
601 if (one && two) {
602 const char *exec_arg[10];
603 const char **arg = &exec_arg[0];
604 *arg++ = pgm;
605 *arg++ = name;
606 *arg++ = temp[0].name;
607 *arg++ = temp[0].hex;
608 *arg++ = temp[0].mode;
609 *arg++ = temp[1].name;
610 *arg++ = temp[1].hex;
611 *arg++ = temp[1].mode;
612 if (other) {
613 *arg++ = other;
614 *arg++ = xfrm_msg;
616 *arg = NULL;
617 execvp(pgm, (char *const*) exec_arg);
619 else
620 execlp(pgm, pgm, name, NULL);
623 * otherwise we use the built-in one.
625 if (one && two)
626 builtin_diff(name, othername, temp, xfrm_msg,
627 complete_rewrite);
628 else
629 printf("* Unmerged path %s\n", name);
630 exit(0);
632 if (waitpid(pid, &status, 0) < 0 ||
633 !WIFEXITED(status) || WEXITSTATUS(status)) {
634 /* Earlier we did not check the exit status because
635 * diff exits non-zero if files are different, and
636 * we are not interested in knowing that. It was a
637 * mistake which made it harder to quit a diff-*
638 * session that uses the git-apply-patch-script as
639 * the GIT_EXTERNAL_DIFF. A custom GIT_EXTERNAL_DIFF
640 * should also exit non-zero only when it wants to
641 * abort the entire diff-* session.
643 remove_tempfile();
644 fprintf(stderr, "external diff died, stopping at %s.\n", name);
645 exit(1);
647 remove_tempfile();
650 static void diff_fill_sha1_info(struct diff_filespec *one)
652 if (DIFF_FILE_VALID(one)) {
653 if (!one->sha1_valid) {
654 struct stat st;
655 if (lstat(one->path, &st) < 0)
656 die("stat %s", one->path);
657 if (index_path(one->sha1, one->path, &st, 0))
658 die("cannot hash %s\n", one->path);
661 else
662 memset(one->sha1, 0, 20);
665 static void run_diff(struct diff_filepair *p, struct diff_options *o)
667 const char *pgm = external_diff();
668 char msg[PATH_MAX*2+300], *xfrm_msg;
669 struct diff_filespec *one;
670 struct diff_filespec *two;
671 const char *name;
672 const char *other;
673 char *name_munged, *other_munged;
674 int complete_rewrite = 0;
675 int len;
677 if (DIFF_PAIR_UNMERGED(p)) {
678 /* unmerged */
679 run_external_diff(pgm, p->one->path, NULL, NULL, NULL, NULL,
681 return;
684 name = p->one->path;
685 other = (strcmp(name, p->two->path) ? p->two->path : NULL);
686 name_munged = quote_one(name);
687 other_munged = quote_one(other);
688 one = p->one; two = p->two;
690 diff_fill_sha1_info(one);
691 diff_fill_sha1_info(two);
693 len = 0;
694 switch (p->status) {
695 case DIFF_STATUS_COPIED:
696 len += snprintf(msg + len, sizeof(msg) - len,
697 "similarity index %d%%\n"
698 "copy from %s\n"
699 "copy to %s\n",
700 (int)(0.5 + p->score * 100.0/MAX_SCORE),
701 name_munged, other_munged);
702 break;
703 case DIFF_STATUS_RENAMED:
704 len += snprintf(msg + len, sizeof(msg) - len,
705 "similarity index %d%%\n"
706 "rename from %s\n"
707 "rename to %s\n",
708 (int)(0.5 + p->score * 100.0/MAX_SCORE),
709 name_munged, other_munged);
710 break;
711 case DIFF_STATUS_MODIFIED:
712 if (p->score) {
713 len += snprintf(msg + len, sizeof(msg) - len,
714 "dissimilarity index %d%%\n",
715 (int)(0.5 + p->score *
716 100.0/MAX_SCORE));
717 complete_rewrite = 1;
718 break;
720 /* fallthru */
721 default:
722 /* nothing */
726 if (memcmp(one->sha1, two->sha1, 20)) {
727 char one_sha1[41];
728 int abbrev = o->full_index ? 40 : DEFAULT_ABBREV;
729 memcpy(one_sha1, sha1_to_hex(one->sha1), 41);
731 len += snprintf(msg + len, sizeof(msg) - len,
732 "index %.*s..%.*s",
733 abbrev, one_sha1, abbrev,
734 sha1_to_hex(two->sha1));
735 if (one->mode == two->mode)
736 len += snprintf(msg + len, sizeof(msg) - len,
737 " %06o", one->mode);
738 len += snprintf(msg + len, sizeof(msg) - len, "\n");
741 if (len)
742 msg[--len] = 0;
743 xfrm_msg = len ? msg : NULL;
745 if (!pgm &&
746 DIFF_FILE_VALID(one) && DIFF_FILE_VALID(two) &&
747 (S_IFMT & one->mode) != (S_IFMT & two->mode)) {
748 /* a filepair that changes between file and symlink
749 * needs to be split into deletion and creation.
751 struct diff_filespec *null = alloc_filespec(two->path);
752 run_external_diff(NULL, name, other, one, null, xfrm_msg, 0);
753 free(null);
754 null = alloc_filespec(one->path);
755 run_external_diff(NULL, name, other, null, two, xfrm_msg, 0);
756 free(null);
758 else
759 run_external_diff(pgm, name, other, one, two, xfrm_msg,
760 complete_rewrite);
762 free(name_munged);
763 free(other_munged);
766 void diff_setup(struct diff_options *options)
768 memset(options, 0, sizeof(*options));
769 options->output_format = DIFF_FORMAT_RAW;
770 options->line_termination = '\n';
771 options->break_opt = -1;
772 options->rename_limit = -1;
774 options->change = diff_change;
775 options->add_remove = diff_addremove;
778 int diff_setup_done(struct diff_options *options)
780 if ((options->find_copies_harder &&
781 options->detect_rename != DIFF_DETECT_COPY) ||
782 (0 <= options->rename_limit && !options->detect_rename))
783 return -1;
784 if (options->detect_rename && options->rename_limit < 0)
785 options->rename_limit = diff_rename_limit_default;
786 if (options->setup & DIFF_SETUP_USE_CACHE) {
787 if (!active_cache)
788 /* read-cache does not die even when it fails
789 * so it is safe for us to do this here. Also
790 * it does not smudge active_cache or active_nr
791 * when it fails, so we do not have to worry about
792 * cleaning it up ourselves either.
794 read_cache();
796 if (options->setup & DIFF_SETUP_USE_SIZE_CACHE)
797 use_size_cache = 1;
798 if (options->abbrev <= 0 || 40 < options->abbrev)
799 options->abbrev = 40; /* full */
801 return 0;
804 int diff_opt_parse(struct diff_options *options, const char **av, int ac)
806 const char *arg = av[0];
807 if (!strcmp(arg, "-p") || !strcmp(arg, "-u"))
808 options->output_format = DIFF_FORMAT_PATCH;
809 else if (!strcmp(arg, "-z"))
810 options->line_termination = 0;
811 else if (!strncmp(arg, "-l", 2))
812 options->rename_limit = strtoul(arg+2, NULL, 10);
813 else if (!strcmp(arg, "--full-index"))
814 options->full_index = 1;
815 else if (!strcmp(arg, "--name-only"))
816 options->output_format = DIFF_FORMAT_NAME;
817 else if (!strcmp(arg, "--name-status"))
818 options->output_format = DIFF_FORMAT_NAME_STATUS;
819 else if (!strcmp(arg, "-R"))
820 options->reverse_diff = 1;
821 else if (!strncmp(arg, "-S", 2))
822 options->pickaxe = arg + 2;
823 else if (!strcmp(arg, "-s"))
824 options->output_format = DIFF_FORMAT_NO_OUTPUT;
825 else if (!strncmp(arg, "-O", 2))
826 options->orderfile = arg + 2;
827 else if (!strncmp(arg, "--diff-filter=", 14))
828 options->filter = arg + 14;
829 else if (!strcmp(arg, "--pickaxe-all"))
830 options->pickaxe_opts = DIFF_PICKAXE_ALL;
831 else if (!strncmp(arg, "-B", 2)) {
832 if ((options->break_opt =
833 diff_scoreopt_parse(arg)) == -1)
834 return -1;
836 else if (!strncmp(arg, "-M", 2)) {
837 if ((options->rename_score =
838 diff_scoreopt_parse(arg)) == -1)
839 return -1;
840 options->detect_rename = DIFF_DETECT_RENAME;
842 else if (!strncmp(arg, "-C", 2)) {
843 if ((options->rename_score =
844 diff_scoreopt_parse(arg)) == -1)
845 return -1;
846 options->detect_rename = DIFF_DETECT_COPY;
848 else if (!strcmp(arg, "--find-copies-harder"))
849 options->find_copies_harder = 1;
850 else if (!strcmp(arg, "--abbrev"))
851 options->abbrev = DEFAULT_ABBREV;
852 else if (!strncmp(arg, "--abbrev=", 9)) {
853 options->abbrev = strtoul(arg + 9, NULL, 10);
854 if (options->abbrev < MINIMUM_ABBREV)
855 options->abbrev = MINIMUM_ABBREV;
856 else if (40 < options->abbrev)
857 options->abbrev = 40;
859 else
860 return 0;
861 return 1;
864 static int parse_num(const char **cp_p)
866 unsigned long num, scale;
867 int ch, dot;
868 const char *cp = *cp_p;
870 num = 0;
871 scale = 1;
872 dot = 0;
873 for(;;) {
874 ch = *cp;
875 if ( !dot && ch == '.' ) {
876 scale = 1;
877 dot = 1;
878 } else if ( ch == '%' ) {
879 scale = dot ? scale*100 : 100;
880 cp++; /* % is always at the end */
881 break;
882 } else if ( ch >= '0' && ch <= '9' ) {
883 if ( scale < 100000 ) {
884 scale *= 10;
885 num = (num*10) + (ch-'0');
887 } else {
888 break;
890 cp++;
892 *cp_p = cp;
894 /* user says num divided by scale and we say internally that
895 * is MAX_SCORE * num / scale.
897 return (num >= scale) ? MAX_SCORE : (MAX_SCORE * num / scale);
900 int diff_scoreopt_parse(const char *opt)
902 int opt1, opt2, cmd;
904 if (*opt++ != '-')
905 return -1;
906 cmd = *opt++;
907 if (cmd != 'M' && cmd != 'C' && cmd != 'B')
908 return -1; /* that is not a -M, -C nor -B option */
910 opt1 = parse_num(&opt);
911 if (cmd != 'B')
912 opt2 = 0;
913 else {
914 if (*opt == 0)
915 opt2 = 0;
916 else if (*opt != '/')
917 return -1; /* we expect -B80/99 or -B80 */
918 else {
919 opt++;
920 opt2 = parse_num(&opt);
923 if (*opt != 0)
924 return -1;
925 return opt1 | (opt2 << 16);
928 struct diff_queue_struct diff_queued_diff;
930 void diff_q(struct diff_queue_struct *queue, struct diff_filepair *dp)
932 if (queue->alloc <= queue->nr) {
933 queue->alloc = alloc_nr(queue->alloc);
934 queue->queue = xrealloc(queue->queue,
935 sizeof(dp) * queue->alloc);
937 queue->queue[queue->nr++] = dp;
940 struct diff_filepair *diff_queue(struct diff_queue_struct *queue,
941 struct diff_filespec *one,
942 struct diff_filespec *two)
944 struct diff_filepair *dp = xmalloc(sizeof(*dp));
945 dp->one = one;
946 dp->two = two;
947 dp->score = 0;
948 dp->status = 0;
949 dp->source_stays = 0;
950 dp->broken_pair = 0;
951 if (queue)
952 diff_q(queue, dp);
953 return dp;
956 void diff_free_filepair(struct diff_filepair *p)
958 diff_free_filespec_data(p->one);
959 diff_free_filespec_data(p->two);
960 free(p->one);
961 free(p->two);
962 free(p);
965 /* This is different from find_unique_abbrev() in that
966 * it needs to deal with 0{40} SHA1.
968 const char *diff_unique_abbrev(const unsigned char *sha1, int len)
970 int abblen;
971 const char *abbrev;
972 if (len == 40)
973 return sha1_to_hex(sha1);
975 abbrev = find_unique_abbrev(sha1, len);
976 if (!abbrev) {
977 if (!memcmp(sha1, null_sha1, 20)) {
978 char *buf = sha1_to_hex(null_sha1);
979 if (len < 37)
980 strcpy(buf + len, "...");
981 return buf;
983 else
984 return sha1_to_hex(sha1);
986 abblen = strlen(abbrev);
987 if (abblen < 37) {
988 static char hex[41];
989 if (len < abblen && abblen <= len + 2)
990 sprintf(hex, "%s%.*s", abbrev, len+3-abblen, "..");
991 else
992 sprintf(hex, "%s...", abbrev);
993 return hex;
995 return sha1_to_hex(sha1);
998 static void diff_flush_raw(struct diff_filepair *p,
999 int line_termination,
1000 int inter_name_termination,
1001 struct diff_options *options)
1003 int two_paths;
1004 char status[10];
1005 int abbrev = options->abbrev;
1006 const char *path_one, *path_two;
1007 int output_format = options->output_format;
1009 path_one = p->one->path;
1010 path_two = p->two->path;
1011 if (line_termination) {
1012 path_one = quote_one(path_one);
1013 path_two = quote_one(path_two);
1016 if (p->score)
1017 sprintf(status, "%c%03d", p->status,
1018 (int)(0.5 + p->score * 100.0/MAX_SCORE));
1019 else {
1020 status[0] = p->status;
1021 status[1] = 0;
1023 switch (p->status) {
1024 case DIFF_STATUS_COPIED:
1025 case DIFF_STATUS_RENAMED:
1026 two_paths = 1;
1027 break;
1028 case DIFF_STATUS_ADDED:
1029 case DIFF_STATUS_DELETED:
1030 two_paths = 0;
1031 break;
1032 default:
1033 two_paths = 0;
1034 break;
1036 if (output_format != DIFF_FORMAT_NAME_STATUS) {
1037 printf(":%06o %06o %s ",
1038 p->one->mode, p->two->mode,
1039 diff_unique_abbrev(p->one->sha1, abbrev));
1040 printf("%s ",
1041 diff_unique_abbrev(p->two->sha1, abbrev));
1043 printf("%s%c%s", status, inter_name_termination, path_one);
1044 if (two_paths)
1045 printf("%c%s", inter_name_termination, path_two);
1046 putchar(line_termination);
1047 if (path_one != p->one->path)
1048 free((void*)path_one);
1049 if (path_two != p->two->path)
1050 free((void*)path_two);
1053 static void diff_flush_name(struct diff_filepair *p,
1054 int inter_name_termination,
1055 int line_termination)
1057 char *path = p->two->path;
1059 if (line_termination)
1060 path = quote_one(p->two->path);
1061 else
1062 path = p->two->path;
1063 printf("%s%c", path, line_termination);
1064 if (p->two->path != path)
1065 free(path);
1068 int diff_unmodified_pair(struct diff_filepair *p)
1070 /* This function is written stricter than necessary to support
1071 * the currently implemented transformers, but the idea is to
1072 * let transformers to produce diff_filepairs any way they want,
1073 * and filter and clean them up here before producing the output.
1075 struct diff_filespec *one, *two;
1077 if (DIFF_PAIR_UNMERGED(p))
1078 return 0; /* unmerged is interesting */
1080 one = p->one;
1081 two = p->two;
1083 /* deletion, addition, mode or type change
1084 * and rename are all interesting.
1086 if (DIFF_FILE_VALID(one) != DIFF_FILE_VALID(two) ||
1087 DIFF_PAIR_MODE_CHANGED(p) ||
1088 strcmp(one->path, two->path))
1089 return 0;
1091 /* both are valid and point at the same path. that is, we are
1092 * dealing with a change.
1094 if (one->sha1_valid && two->sha1_valid &&
1095 !memcmp(one->sha1, two->sha1, sizeof(one->sha1)))
1096 return 1; /* no change */
1097 if (!one->sha1_valid && !two->sha1_valid)
1098 return 1; /* both look at the same file on the filesystem. */
1099 return 0;
1102 static void diff_flush_patch(struct diff_filepair *p, struct diff_options *o)
1104 if (diff_unmodified_pair(p))
1105 return;
1107 if ((DIFF_FILE_VALID(p->one) && S_ISDIR(p->one->mode)) ||
1108 (DIFF_FILE_VALID(p->two) && S_ISDIR(p->two->mode)))
1109 return; /* no tree diffs in patch format */
1111 run_diff(p, o);
1114 int diff_queue_is_empty(void)
1116 struct diff_queue_struct *q = &diff_queued_diff;
1117 int i;
1118 for (i = 0; i < q->nr; i++)
1119 if (!diff_unmodified_pair(q->queue[i]))
1120 return 0;
1121 return 1;
1124 #if DIFF_DEBUG
1125 void diff_debug_filespec(struct diff_filespec *s, int x, const char *one)
1127 fprintf(stderr, "queue[%d] %s (%s) %s %06o %s\n",
1128 x, one ? one : "",
1129 s->path,
1130 DIFF_FILE_VALID(s) ? "valid" : "invalid",
1131 s->mode,
1132 s->sha1_valid ? sha1_to_hex(s->sha1) : "");
1133 fprintf(stderr, "queue[%d] %s size %lu flags %d\n",
1134 x, one ? one : "",
1135 s->size, s->xfrm_flags);
1138 void diff_debug_filepair(const struct diff_filepair *p, int i)
1140 diff_debug_filespec(p->one, i, "one");
1141 diff_debug_filespec(p->two, i, "two");
1142 fprintf(stderr, "score %d, status %c stays %d broken %d\n",
1143 p->score, p->status ? p->status : '?',
1144 p->source_stays, p->broken_pair);
1147 void diff_debug_queue(const char *msg, struct diff_queue_struct *q)
1149 int i;
1150 if (msg)
1151 fprintf(stderr, "%s\n", msg);
1152 fprintf(stderr, "q->nr = %d\n", q->nr);
1153 for (i = 0; i < q->nr; i++) {
1154 struct diff_filepair *p = q->queue[i];
1155 diff_debug_filepair(p, i);
1158 #endif
1160 static void diff_resolve_rename_copy(void)
1162 int i, j;
1163 struct diff_filepair *p, *pp;
1164 struct diff_queue_struct *q = &diff_queued_diff;
1166 diff_debug_queue("resolve-rename-copy", q);
1168 for (i = 0; i < q->nr; i++) {
1169 p = q->queue[i];
1170 p->status = 0; /* undecided */
1171 if (DIFF_PAIR_UNMERGED(p))
1172 p->status = DIFF_STATUS_UNMERGED;
1173 else if (!DIFF_FILE_VALID(p->one))
1174 p->status = DIFF_STATUS_ADDED;
1175 else if (!DIFF_FILE_VALID(p->two))
1176 p->status = DIFF_STATUS_DELETED;
1177 else if (DIFF_PAIR_TYPE_CHANGED(p))
1178 p->status = DIFF_STATUS_TYPE_CHANGED;
1180 /* from this point on, we are dealing with a pair
1181 * whose both sides are valid and of the same type, i.e.
1182 * either in-place edit or rename/copy edit.
1184 else if (DIFF_PAIR_RENAME(p)) {
1185 if (p->source_stays) {
1186 p->status = DIFF_STATUS_COPIED;
1187 continue;
1189 /* See if there is some other filepair that
1190 * copies from the same source as us. If so
1191 * we are a copy. Otherwise we are either a
1192 * copy if the path stays, or a rename if it
1193 * does not, but we already handled "stays" case.
1195 for (j = i + 1; j < q->nr; j++) {
1196 pp = q->queue[j];
1197 if (strcmp(pp->one->path, p->one->path))
1198 continue; /* not us */
1199 if (!DIFF_PAIR_RENAME(pp))
1200 continue; /* not a rename/copy */
1201 /* pp is a rename/copy from the same source */
1202 p->status = DIFF_STATUS_COPIED;
1203 break;
1205 if (!p->status)
1206 p->status = DIFF_STATUS_RENAMED;
1208 else if (memcmp(p->one->sha1, p->two->sha1, 20) ||
1209 p->one->mode != p->two->mode)
1210 p->status = DIFF_STATUS_MODIFIED;
1211 else {
1212 /* This is a "no-change" entry and should not
1213 * happen anymore, but prepare for broken callers.
1215 error("feeding unmodified %s to diffcore",
1216 p->one->path);
1217 p->status = DIFF_STATUS_UNKNOWN;
1220 diff_debug_queue("resolve-rename-copy done", q);
1223 void diff_flush(struct diff_options *options)
1225 struct diff_queue_struct *q = &diff_queued_diff;
1226 int i;
1227 int inter_name_termination = '\t';
1228 int diff_output_format = options->output_format;
1229 int line_termination = options->line_termination;
1231 if (!line_termination)
1232 inter_name_termination = 0;
1234 for (i = 0; i < q->nr; i++) {
1235 struct diff_filepair *p = q->queue[i];
1236 if ((diff_output_format == DIFF_FORMAT_NO_OUTPUT) ||
1237 (p->status == DIFF_STATUS_UNKNOWN))
1238 continue;
1239 if (p->status == 0)
1240 die("internal error in diff-resolve-rename-copy");
1241 switch (diff_output_format) {
1242 case DIFF_FORMAT_PATCH:
1243 diff_flush_patch(p, options);
1244 break;
1245 case DIFF_FORMAT_RAW:
1246 case DIFF_FORMAT_NAME_STATUS:
1247 diff_flush_raw(p, line_termination,
1248 inter_name_termination,
1249 options);
1250 break;
1251 case DIFF_FORMAT_NAME:
1252 diff_flush_name(p,
1253 inter_name_termination,
1254 line_termination);
1255 break;
1257 diff_free_filepair(q->queue[i]);
1259 free(q->queue);
1260 q->queue = NULL;
1261 q->nr = q->alloc = 0;
1264 static void diffcore_apply_filter(const char *filter)
1266 int i;
1267 struct diff_queue_struct *q = &diff_queued_diff;
1268 struct diff_queue_struct outq;
1269 outq.queue = NULL;
1270 outq.nr = outq.alloc = 0;
1272 if (!filter)
1273 return;
1275 if (strchr(filter, DIFF_STATUS_FILTER_AON)) {
1276 int found;
1277 for (i = found = 0; !found && i < q->nr; i++) {
1278 struct diff_filepair *p = q->queue[i];
1279 if (((p->status == DIFF_STATUS_MODIFIED) &&
1280 ((p->score &&
1281 strchr(filter, DIFF_STATUS_FILTER_BROKEN)) ||
1282 (!p->score &&
1283 strchr(filter, DIFF_STATUS_MODIFIED)))) ||
1284 ((p->status != DIFF_STATUS_MODIFIED) &&
1285 strchr(filter, p->status)))
1286 found++;
1288 if (found)
1289 return;
1291 /* otherwise we will clear the whole queue
1292 * by copying the empty outq at the end of this
1293 * function, but first clear the current entries
1294 * in the queue.
1296 for (i = 0; i < q->nr; i++)
1297 diff_free_filepair(q->queue[i]);
1299 else {
1300 /* Only the matching ones */
1301 for (i = 0; i < q->nr; i++) {
1302 struct diff_filepair *p = q->queue[i];
1304 if (((p->status == DIFF_STATUS_MODIFIED) &&
1305 ((p->score &&
1306 strchr(filter, DIFF_STATUS_FILTER_BROKEN)) ||
1307 (!p->score &&
1308 strchr(filter, DIFF_STATUS_MODIFIED)))) ||
1309 ((p->status != DIFF_STATUS_MODIFIED) &&
1310 strchr(filter, p->status)))
1311 diff_q(&outq, p);
1312 else
1313 diff_free_filepair(p);
1316 free(q->queue);
1317 *q = outq;
1320 void diffcore_std(struct diff_options *options)
1322 if (options->paths && options->paths[0])
1323 diffcore_pathspec(options->paths);
1324 if (options->break_opt != -1)
1325 diffcore_break(options->break_opt);
1326 if (options->detect_rename)
1327 diffcore_rename(options);
1328 if (options->break_opt != -1)
1329 diffcore_merge_broken();
1330 if (options->pickaxe)
1331 diffcore_pickaxe(options->pickaxe, options->pickaxe_opts);
1332 if (options->orderfile)
1333 diffcore_order(options->orderfile);
1334 diff_resolve_rename_copy();
1335 diffcore_apply_filter(options->filter);
1339 void diffcore_std_no_resolve(struct diff_options *options)
1341 if (options->pickaxe)
1342 diffcore_pickaxe(options->pickaxe, options->pickaxe_opts);
1343 if (options->orderfile)
1344 diffcore_order(options->orderfile);
1345 diffcore_apply_filter(options->filter);
1348 void diff_addremove(struct diff_options *options,
1349 int addremove, unsigned mode,
1350 const unsigned char *sha1,
1351 const char *base, const char *path)
1353 char concatpath[PATH_MAX];
1354 struct diff_filespec *one, *two;
1356 /* This may look odd, but it is a preparation for
1357 * feeding "there are unchanged files which should
1358 * not produce diffs, but when you are doing copy
1359 * detection you would need them, so here they are"
1360 * entries to the diff-core. They will be prefixed
1361 * with something like '=' or '*' (I haven't decided
1362 * which but should not make any difference).
1363 * Feeding the same new and old to diff_change()
1364 * also has the same effect.
1365 * Before the final output happens, they are pruned after
1366 * merged into rename/copy pairs as appropriate.
1368 if (options->reverse_diff)
1369 addremove = (addremove == '+' ? '-' :
1370 addremove == '-' ? '+' : addremove);
1372 if (!path) path = "";
1373 sprintf(concatpath, "%s%s", base, path);
1374 one = alloc_filespec(concatpath);
1375 two = alloc_filespec(concatpath);
1377 if (addremove != '+')
1378 fill_filespec(one, sha1, mode);
1379 if (addremove != '-')
1380 fill_filespec(two, sha1, mode);
1382 diff_queue(&diff_queued_diff, one, two);
1385 void diff_change(struct diff_options *options,
1386 unsigned old_mode, unsigned new_mode,
1387 const unsigned char *old_sha1,
1388 const unsigned char *new_sha1,
1389 const char *base, const char *path)
1391 char concatpath[PATH_MAX];
1392 struct diff_filespec *one, *two;
1394 if (options->reverse_diff) {
1395 unsigned tmp;
1396 const unsigned char *tmp_c;
1397 tmp = old_mode; old_mode = new_mode; new_mode = tmp;
1398 tmp_c = old_sha1; old_sha1 = new_sha1; new_sha1 = tmp_c;
1400 if (!path) path = "";
1401 sprintf(concatpath, "%s%s", base, path);
1402 one = alloc_filespec(concatpath);
1403 two = alloc_filespec(concatpath);
1404 fill_filespec(one, old_sha1, old_mode);
1405 fill_filespec(two, new_sha1, new_mode);
1407 diff_queue(&diff_queued_diff, one, two);
1410 void diff_unmerge(struct diff_options *options,
1411 const char *path)
1413 struct diff_filespec *one, *two;
1414 one = alloc_filespec(path);
1415 two = alloc_filespec(path);
1416 diff_queue(&diff_queued_diff, one, two);