diff.con commit Merge master.kernel.org:/pub/scm/gitk/gitk (07ee0d7)
   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"
  11
  12static const char *diff_opts = "-pu";
  13static unsigned char null_sha1[20] = { 0, };
  14
  15static int reverse_diff;
  16static int use_size_cache;
  17
  18static const char *external_diff(void)
  19{
  20        static const char *external_diff_cmd = NULL;
  21        static int done_preparing = 0;
  22
  23        if (done_preparing)
  24                return external_diff_cmd;
  25
  26        /*
  27         * Default values above are meant to match the
  28         * Linux kernel development style.  Examples of
  29         * alternative styles you can specify via environment
  30         * variables are:
  31         *
  32         * GIT_DIFF_OPTS="-c";
  33         */
  34        if (gitenv("GIT_EXTERNAL_DIFF"))
  35                external_diff_cmd = gitenv("GIT_EXTERNAL_DIFF");
  36
  37        /* In case external diff fails... */
  38        diff_opts = gitenv("GIT_DIFF_OPTS") ? : diff_opts;
  39
  40        done_preparing = 1;
  41        return external_diff_cmd;
  42}
  43
  44static struct diff_tempfile {
  45        const char *name; /* filename external diff should read from */
  46        char hex[41];
  47        char mode[10];
  48        char tmp_path[50];
  49} diff_temp[2];
  50
  51static int count_lines(const char *filename)
  52{
  53        FILE *in;
  54        int count, ch, completely_empty = 1, nl_just_seen = 0;
  55        in = fopen(filename, "r");
  56        count = 0;
  57        while ((ch = fgetc(in)) != EOF)
  58                if (ch == '\n') {
  59                        count++;
  60                        nl_just_seen = 1;
  61                        completely_empty = 0;
  62                }
  63                else {
  64                        nl_just_seen = 0;
  65                        completely_empty = 0;
  66                }
  67        fclose(in);
  68        if (completely_empty)
  69                return 0;
  70        if (!nl_just_seen)
  71                count++; /* no trailing newline */
  72        return count;
  73}
  74
  75static void print_line_count(int count)
  76{
  77        switch (count) {
  78        case 0:
  79                printf("0,0");
  80                break;
  81        case 1:
  82                printf("1");
  83                break;
  84        default:
  85                printf("1,%d", count);
  86                break;
  87        }
  88}
  89
  90static void copy_file(int prefix, const char *filename)
  91{
  92        FILE *in;
  93        int ch, nl_just_seen = 1;
  94        in = fopen(filename, "r");
  95        while ((ch = fgetc(in)) != EOF) {
  96                if (nl_just_seen)
  97                        putchar(prefix);
  98                putchar(ch);
  99                if (ch == '\n')
 100                        nl_just_seen = 1;
 101                else
 102                        nl_just_seen = 0;
 103        }
 104        fclose(in);
 105        if (!nl_just_seen)
 106                printf("\n\\ No newline at end of file\n");
 107}
 108
 109static void emit_rewrite_diff(const char *name_a,
 110                              const char *name_b,
 111                              struct diff_tempfile *temp)
 112{
 113        /* Use temp[i].name as input, name_a and name_b as labels */
 114        int lc_a, lc_b;
 115        lc_a = count_lines(temp[0].name);
 116        lc_b = count_lines(temp[1].name);
 117        printf("--- %s\n+++ %s\n@@ -", name_a, name_b);
 118        print_line_count(lc_a);
 119        printf(" +");
 120        print_line_count(lc_b);
 121        printf(" @@\n");
 122        if (lc_a)
 123                copy_file('-', temp[0].name);
 124        if (lc_b)
 125                copy_file('+', temp[1].name);
 126}
 127
 128static void builtin_diff(const char *name_a,
 129                         const char *name_b,
 130                         struct diff_tempfile *temp,
 131                         const char *xfrm_msg,
 132                         int complete_rewrite)
 133{
 134        int i, next_at, cmd_size;
 135        const char *diff_cmd = "diff -L%s%s -L%s%s";
 136        const char *diff_arg  = "%s %s||:"; /* "||:" is to return 0 */
 137        const char *input_name_sq[2];
 138        const char *path0[2];
 139        const char *path1[2];
 140        const char *name_sq[2];
 141        char *cmd;
 142
 143        name_sq[0] = sq_quote(name_a);
 144        name_sq[1] = sq_quote(name_b);
 145
 146        /* diff_cmd and diff_arg have 6 %s in total which makes
 147         * the sum of these strings 12 bytes larger than required.
 148         * we use 2 spaces around diff-opts, and we need to count
 149         * terminating NUL, so we subtract 9 here.
 150         */
 151        cmd_size = (strlen(diff_cmd) + strlen(diff_opts) +
 152                        strlen(diff_arg) - 9);
 153        for (i = 0; i < 2; i++) {
 154                input_name_sq[i] = sq_quote(temp[i].name);
 155                if (!strcmp(temp[i].name, "/dev/null")) {
 156                        path0[i] = "/dev/null";
 157                        path1[i] = "";
 158                } else {
 159                        path0[i] = i ? "b/" : "a/";
 160                        path1[i] = name_sq[i];
 161                }
 162                cmd_size += (strlen(path0[i]) + strlen(path1[i]) +
 163                             strlen(input_name_sq[i]));
 164        }
 165
 166        cmd = xmalloc(cmd_size);
 167
 168        next_at = 0;
 169        next_at += snprintf(cmd+next_at, cmd_size-next_at,
 170                            diff_cmd,
 171                            path0[0], path1[0], path0[1], path1[1]);
 172        next_at += snprintf(cmd+next_at, cmd_size-next_at,
 173                            " %s ", diff_opts);
 174        next_at += snprintf(cmd+next_at, cmd_size-next_at,
 175                            diff_arg, input_name_sq[0], input_name_sq[1]);
 176
 177        printf("diff --git a/%s b/%s\n", name_a, name_b);
 178        if (!path1[0][0]) {
 179                printf("new file mode %s\n", temp[1].mode);
 180                if (xfrm_msg && xfrm_msg[0])
 181                        puts(xfrm_msg);
 182        }
 183        else if (!path1[1][0]) {
 184                printf("deleted file mode %s\n", temp[0].mode);
 185                if (xfrm_msg && xfrm_msg[0])
 186                        puts(xfrm_msg);
 187        }
 188        else {
 189                if (strcmp(temp[0].mode, temp[1].mode)) {
 190                        printf("old mode %s\n", temp[0].mode);
 191                        printf("new mode %s\n", temp[1].mode);
 192                }
 193                if (xfrm_msg && xfrm_msg[0])
 194                        puts(xfrm_msg);
 195                if (strncmp(temp[0].mode, temp[1].mode, 3))
 196                        /* we do not run diff between different kind
 197                         * of objects.
 198                         */
 199                        exit(0);
 200                if (complete_rewrite) {
 201                        fflush(NULL);
 202                        emit_rewrite_diff(name_a, name_b, temp);
 203                        exit(0);
 204                }
 205        }
 206        fflush(NULL);
 207        execlp("/bin/sh","sh", "-c", cmd, NULL);
 208}
 209
 210struct diff_filespec *alloc_filespec(const char *path)
 211{
 212        int namelen = strlen(path);
 213        struct diff_filespec *spec = xmalloc(sizeof(*spec) + namelen + 1);
 214        spec->path = (char *)(spec + 1);
 215        strcpy(spec->path, path);
 216        spec->should_free = spec->should_munmap = 0;
 217        spec->xfrm_flags = 0;
 218        spec->size = 0;
 219        spec->data = NULL;
 220        spec->mode = 0;
 221        memset(spec->sha1, 0, 20);
 222        return spec;
 223}
 224
 225void fill_filespec(struct diff_filespec *spec, const unsigned char *sha1,
 226                   unsigned short mode)
 227{
 228        if (mode) {
 229                spec->mode = DIFF_FILE_CANON_MODE(mode);
 230                memcpy(spec->sha1, sha1, 20);
 231                spec->sha1_valid = !!memcmp(sha1, null_sha1, 20);
 232        }
 233}
 234
 235/*
 236 * Given a name and sha1 pair, if the dircache tells us the file in
 237 * the work tree has that object contents, return true, so that
 238 * prepare_temp_file() does not have to inflate and extract.
 239 */
 240static int work_tree_matches(const char *name, const unsigned char *sha1)
 241{
 242        struct cache_entry *ce;
 243        struct stat st;
 244        int pos, len;
 245
 246        /* We do not read the cache ourselves here, because the
 247         * benchmark with my previous version that always reads cache
 248         * shows that it makes things worse for diff-tree comparing
 249         * two linux-2.6 kernel trees in an already checked out work
 250         * tree.  This is because most diff-tree comparisons deal with
 251         * only a small number of files, while reading the cache is
 252         * expensive for a large project, and its cost outweighs the
 253         * savings we get by not inflating the object to a temporary
 254         * file.  Practically, this code only helps when we are used
 255         * by diff-cache --cached, which does read the cache before
 256         * calling us.
 257         */
 258        if (!active_cache)
 259                return 0;
 260
 261        len = strlen(name);
 262        pos = cache_name_pos(name, len);
 263        if (pos < 0)
 264                return 0;
 265        ce = active_cache[pos];
 266        if ((lstat(name, &st) < 0) ||
 267            !S_ISREG(st.st_mode) || /* careful! */
 268            ce_match_stat(ce, &st) ||
 269            memcmp(sha1, ce->sha1, 20))
 270                return 0;
 271        /* we return 1 only when we can stat, it is a regular file,
 272         * stat information matches, and sha1 recorded in the cache
 273         * matches.  I.e. we know the file in the work tree really is
 274         * the same as the <name, sha1> pair.
 275         */
 276        return 1;
 277}
 278
 279static struct sha1_size_cache {
 280        unsigned char sha1[20];
 281        unsigned long size;
 282} **sha1_size_cache;
 283static int sha1_size_cache_nr, sha1_size_cache_alloc;
 284
 285static struct sha1_size_cache *locate_size_cache(unsigned char *sha1,
 286                                                 int find_only,
 287                                                 unsigned long size)
 288{
 289        int first, last;
 290        struct sha1_size_cache *e;
 291
 292        first = 0;
 293        last = sha1_size_cache_nr;
 294        while (last > first) {
 295                int cmp, next = (last + first) >> 1;
 296                e = sha1_size_cache[next];
 297                cmp = memcmp(e->sha1, sha1, 20);
 298                if (!cmp)
 299                        return e;
 300                if (cmp < 0) {
 301                        last = next;
 302                        continue;
 303                }
 304                first = next+1;
 305        }
 306        /* not found */
 307        if (find_only)
 308                return NULL;
 309        /* insert to make it at "first" */
 310        if (sha1_size_cache_alloc <= sha1_size_cache_nr) {
 311                sha1_size_cache_alloc = alloc_nr(sha1_size_cache_alloc);
 312                sha1_size_cache = xrealloc(sha1_size_cache,
 313                                           sha1_size_cache_alloc *
 314                                           sizeof(*sha1_size_cache));
 315        }
 316        sha1_size_cache_nr++;
 317        if (first < sha1_size_cache_nr)
 318                memmove(sha1_size_cache + first + 1, sha1_size_cache + first,
 319                        (sha1_size_cache_nr - first - 1) *
 320                        sizeof(*sha1_size_cache));
 321        e = xmalloc(sizeof(struct sha1_size_cache));
 322        sha1_size_cache[first] = e;
 323        memcpy(e->sha1, sha1, 20);
 324        e->size = size;
 325        return e;
 326}
 327
 328/*
 329 * While doing rename detection and pickaxe operation, we may need to
 330 * grab the data for the blob (or file) for our own in-core comparison.
 331 * diff_filespec has data and size fields for this purpose.
 332 */
 333int diff_populate_filespec(struct diff_filespec *s, int size_only)
 334{
 335        int err = 0;
 336        if (!DIFF_FILE_VALID(s))
 337                die("internal error: asking to populate invalid file.");
 338        if (S_ISDIR(s->mode))
 339                return -1;
 340
 341        if (!use_size_cache)
 342                size_only = 0;
 343
 344        if (s->data)
 345                return err;
 346        if (!s->sha1_valid ||
 347            work_tree_matches(s->path, s->sha1)) {
 348                struct stat st;
 349                int fd;
 350                if (lstat(s->path, &st) < 0) {
 351                        if (errno == ENOENT) {
 352                        err_empty:
 353                                err = -1;
 354                        empty:
 355                                s->data = "";
 356                                s->size = 0;
 357                                return err;
 358                        }
 359                }
 360                s->size = st.st_size;
 361                if (!s->size)
 362                        goto empty;
 363                if (size_only)
 364                        return 0;
 365                if (S_ISLNK(st.st_mode)) {
 366                        int ret;
 367                        s->data = xmalloc(s->size);
 368                        s->should_free = 1;
 369                        ret = readlink(s->path, s->data, s->size);
 370                        if (ret < 0) {
 371                                free(s->data);
 372                                goto err_empty;
 373                        }
 374                        return 0;
 375                }
 376                fd = open(s->path, O_RDONLY);
 377                if (fd < 0)
 378                        goto err_empty;
 379                s->data = mmap(NULL, s->size, PROT_READ, MAP_PRIVATE, fd, 0);
 380                s->should_munmap = 1;
 381                close(fd);
 382        }
 383        else {
 384                char type[20];
 385                struct sha1_size_cache *e;
 386
 387                if (size_only) {
 388                        e = locate_size_cache(s->sha1, 1, 0);
 389                        if (e) {
 390                                s->size = e->size;
 391                                return 0;
 392                        }
 393                        if (!sha1_object_info(s->sha1, type, &s->size))
 394                                locate_size_cache(s->sha1, 0, s->size);
 395                }
 396                else {
 397                        s->data = read_sha1_file(s->sha1, type, &s->size);
 398                        s->should_free = 1;
 399                }
 400        }
 401        return 0;
 402}
 403
 404void diff_free_filespec_data(struct diff_filespec *s)
 405{
 406        if (s->should_free)
 407                free(s->data);
 408        else if (s->should_munmap)
 409                munmap(s->data, s->size);
 410        s->should_free = s->should_munmap = 0;
 411        s->data = NULL;
 412}
 413
 414static void prep_temp_blob(struct diff_tempfile *temp,
 415                           void *blob,
 416                           unsigned long size,
 417                           unsigned char *sha1,
 418                           int mode)
 419{
 420        int fd;
 421
 422        strcpy(temp->tmp_path, ".diff_XXXXXX");
 423        fd = mkstemp(temp->tmp_path);
 424        if (fd < 0)
 425                die("unable to create temp-file");
 426        if (write(fd, blob, size) != size)
 427                die("unable to write temp-file");
 428        close(fd);
 429        temp->name = temp->tmp_path;
 430        strcpy(temp->hex, sha1_to_hex(sha1));
 431        temp->hex[40] = 0;
 432        sprintf(temp->mode, "%06o", mode);
 433}
 434
 435static void prepare_temp_file(const char *name,
 436                              struct diff_tempfile *temp,
 437                              struct diff_filespec *one)
 438{
 439        if (!DIFF_FILE_VALID(one)) {
 440        not_a_valid_file:
 441                /* A '-' entry produces this for file-2, and
 442                 * a '+' entry produces this for file-1.
 443                 */
 444                temp->name = "/dev/null";
 445                strcpy(temp->hex, ".");
 446                strcpy(temp->mode, ".");
 447                return;
 448        }
 449
 450        if (!one->sha1_valid ||
 451            work_tree_matches(name, one->sha1)) {
 452                struct stat st;
 453                if (lstat(name, &st) < 0) {
 454                        if (errno == ENOENT)
 455                                goto not_a_valid_file;
 456                        die("stat(%s): %s", name, strerror(errno));
 457                }
 458                if (S_ISLNK(st.st_mode)) {
 459                        int ret;
 460                        char *buf, buf_[1024];
 461                        buf = ((sizeof(buf_) < st.st_size) ?
 462                               xmalloc(st.st_size) : buf_);
 463                        ret = readlink(name, buf, st.st_size);
 464                        if (ret < 0)
 465                                die("readlink(%s)", name);
 466                        prep_temp_blob(temp, buf, st.st_size,
 467                                       (one->sha1_valid ?
 468                                        one->sha1 : null_sha1),
 469                                       (one->sha1_valid ?
 470                                        one->mode : S_IFLNK));
 471                }
 472                else {
 473                        /* we can borrow from the file in the work tree */
 474                        temp->name = name;
 475                        if (!one->sha1_valid)
 476                                strcpy(temp->hex, sha1_to_hex(null_sha1));
 477                        else
 478                                strcpy(temp->hex, sha1_to_hex(one->sha1));
 479                        /* Even though we may sometimes borrow the
 480                         * contents from the work tree, we always want
 481                         * one->mode.  mode is trustworthy even when
 482                         * !(one->sha1_valid), as long as
 483                         * DIFF_FILE_VALID(one).
 484                         */
 485                        sprintf(temp->mode, "%06o", one->mode);
 486                }
 487                return;
 488        }
 489        else {
 490                if (diff_populate_filespec(one, 0))
 491                        die("cannot read data blob for %s", one->path);
 492                prep_temp_blob(temp, one->data, one->size,
 493                               one->sha1, one->mode);
 494        }
 495}
 496
 497static void remove_tempfile(void)
 498{
 499        int i;
 500
 501        for (i = 0; i < 2; i++)
 502                if (diff_temp[i].name == diff_temp[i].tmp_path) {
 503                        unlink(diff_temp[i].name);
 504                        diff_temp[i].name = NULL;
 505                }
 506}
 507
 508static void remove_tempfile_on_signal(int signo)
 509{
 510        remove_tempfile();
 511}
 512
 513/* An external diff command takes:
 514 *
 515 * diff-cmd name infile1 infile1-sha1 infile1-mode \
 516 *               infile2 infile2-sha1 infile2-mode [ rename-to ]
 517 *
 518 */
 519static void run_external_diff(const char *pgm,
 520                              const char *name,
 521                              const char *other,
 522                              struct diff_filespec *one,
 523                              struct diff_filespec *two,
 524                              const char *xfrm_msg,
 525                              int complete_rewrite)
 526{
 527        struct diff_tempfile *temp = diff_temp;
 528        pid_t pid;
 529        int status;
 530        static int atexit_asked = 0;
 531
 532        if (one && two) {
 533                prepare_temp_file(name, &temp[0], one);
 534                prepare_temp_file(other ? : name, &temp[1], two);
 535                if (! atexit_asked &&
 536                    (temp[0].name == temp[0].tmp_path ||
 537                     temp[1].name == temp[1].tmp_path)) {
 538                        atexit_asked = 1;
 539                        atexit(remove_tempfile);
 540                }
 541                signal(SIGINT, remove_tempfile_on_signal);
 542        }
 543
 544        fflush(NULL);
 545        pid = fork();
 546        if (pid < 0)
 547                die("unable to fork");
 548        if (!pid) {
 549                if (pgm) {
 550                        if (one && two) {
 551                                const char *exec_arg[10];
 552                                const char **arg = &exec_arg[0];
 553                                *arg++ = pgm;
 554                                *arg++ = name;
 555                                *arg++ = temp[0].name;
 556                                *arg++ = temp[0].hex;
 557                                *arg++ = temp[0].mode;
 558                                *arg++ = temp[1].name;
 559                                *arg++ = temp[1].hex;
 560                                *arg++ = temp[1].mode;
 561                                if (other) {
 562                                        *arg++ = other;
 563                                        *arg++ = xfrm_msg;
 564                                }
 565                                *arg = NULL;
 566                                execvp(pgm, (char *const*) exec_arg);
 567                        }
 568                        else
 569                                execlp(pgm, pgm, name, NULL);
 570                }
 571                /*
 572                 * otherwise we use the built-in one.
 573                 */
 574                if (one && two)
 575                        builtin_diff(name, other ? : name, temp, xfrm_msg,
 576                                     complete_rewrite);
 577                else
 578                        printf("* Unmerged path %s\n", name);
 579                exit(0);
 580        }
 581        if (waitpid(pid, &status, 0) < 0 ||
 582            !WIFEXITED(status) || WEXITSTATUS(status)) {
 583                /* Earlier we did not check the exit status because
 584                 * diff exits non-zero if files are different, and
 585                 * we are not interested in knowing that.  It was a
 586                 * mistake which made it harder to quit a diff-*
 587                 * session that uses the git-apply-patch-script as
 588                 * the GIT_EXTERNAL_DIFF.  A custom GIT_EXTERNAL_DIFF
 589                 * should also exit non-zero only when it wants to
 590                 * abort the entire diff-* session.
 591                 */
 592                remove_tempfile();
 593                fprintf(stderr, "external diff died, stopping at %s.\n", name);
 594                exit(1);
 595        }
 596        remove_tempfile();
 597}
 598
 599static void run_diff(struct diff_filepair *p)
 600{
 601        const char *pgm = external_diff();
 602        char msg_[PATH_MAX*2+200], *xfrm_msg;
 603        struct diff_filespec *one;
 604        struct diff_filespec *two;
 605        const char *name;
 606        const char *other;
 607        int complete_rewrite = 0;
 608
 609        if (DIFF_PAIR_UNMERGED(p)) {
 610                /* unmerged */
 611                run_external_diff(pgm, p->one->path, NULL, NULL, NULL, NULL,
 612                                  0);
 613                return;
 614        }
 615
 616        name = p->one->path;
 617        other = (strcmp(name, p->two->path) ? p->two->path : NULL);
 618        one = p->one; two = p->two;
 619        switch (p->status) {
 620        case 'C':
 621                sprintf(msg_,
 622                        "similarity index %d%%\n"
 623                        "copy from %s\n"
 624                        "copy to %s",
 625                        (int)(0.5 + p->score * 100.0/MAX_SCORE),
 626                        name, other);
 627                xfrm_msg = msg_;
 628                break;
 629        case 'R':
 630                sprintf(msg_,
 631                        "similarity index %d%%\n"
 632                        "rename from %s\n"
 633                        "rename to %s",
 634                        (int)(0.5 + p->score * 100.0/MAX_SCORE),
 635                        name, other);
 636                xfrm_msg = msg_;
 637                break;
 638        case 'M':
 639                if (p->score) {
 640                        sprintf(msg_,
 641                                "dissimilarity index %d%%",
 642                                (int)(0.5 + p->score * 100.0/MAX_SCORE));
 643                        xfrm_msg = msg_;
 644                        complete_rewrite = 1;
 645                        break;
 646                }
 647                /* fallthru */
 648        default:
 649                xfrm_msg = NULL;
 650        }
 651
 652        if (!pgm &&
 653            DIFF_FILE_VALID(one) && DIFF_FILE_VALID(two) &&
 654            (S_IFMT & one->mode) != (S_IFMT & two->mode)) {
 655                /* a filepair that changes between file and symlink
 656                 * needs to be split into deletion and creation.
 657                 */
 658                struct diff_filespec *null = alloc_filespec(two->path);
 659                run_external_diff(NULL, name, other, one, null, xfrm_msg, 0);
 660                free(null);
 661                null = alloc_filespec(one->path);
 662                run_external_diff(NULL, name, other, null, two, xfrm_msg, 0);
 663                free(null);
 664        }
 665        else
 666                run_external_diff(pgm, name, other, one, two, xfrm_msg,
 667                                  complete_rewrite);
 668}
 669
 670void diff_setup(int flags)
 671{
 672        if (flags & DIFF_SETUP_REVERSE)
 673                reverse_diff = 1;
 674        if (flags & DIFF_SETUP_USE_CACHE) {
 675                if (!active_cache)
 676                        /* read-cache does not die even when it fails
 677                         * so it is safe for us to do this here.  Also
 678                         * it does not smudge active_cache or active_nr
 679                         * when it fails, so we do not have to worry about
 680                         * cleaning it up oufselves either.
 681                         */
 682                        read_cache();
 683        }
 684        if (flags & DIFF_SETUP_USE_SIZE_CACHE)
 685                use_size_cache = 1;
 686        
 687}
 688
 689static int parse_num(const char **cp_p)
 690{
 691        int num, scale, ch, cnt;
 692        const char *cp = *cp_p;
 693
 694        cnt = num = 0;
 695        scale = 1;
 696        while ('0' <= (ch = *cp) && ch <= '9') {
 697                if (cnt++ < 5) {
 698                        /* We simply ignore more than 5 digits precision. */
 699                        scale *= 10;
 700                        num = num * 10 + ch - '0';
 701                }
 702                *cp++;
 703        }
 704        *cp_p = cp;
 705
 706        /* user says num divided by scale and we say internally that
 707         * is MAX_SCORE * num / scale.
 708         */
 709        return (MAX_SCORE * num / scale);
 710}
 711
 712int diff_scoreopt_parse(const char *opt)
 713{
 714        int opt1, opt2, cmd;
 715
 716        if (*opt++ != '-')
 717                return -1;
 718        cmd = *opt++;
 719        if (cmd != 'M' && cmd != 'C' && cmd != 'B')
 720                return -1; /* that is not a -M, -C nor -B option */
 721
 722        opt1 = parse_num(&opt);
 723        if (cmd != 'B')
 724                opt2 = 0;
 725        else {
 726                if (*opt == 0)
 727                        opt2 = 0;
 728                else if (*opt != '/')
 729                        return -1; /* we expect -B80/99 or -B80 */
 730                else {
 731                        opt++;
 732                        opt2 = parse_num(&opt);
 733                }
 734        }
 735        if (*opt != 0)
 736                return -1;
 737        return opt1 | (opt2 << 16);
 738}
 739
 740struct diff_queue_struct diff_queued_diff;
 741
 742void diff_q(struct diff_queue_struct *queue, struct diff_filepair *dp)
 743{
 744        if (queue->alloc <= queue->nr) {
 745                queue->alloc = alloc_nr(queue->alloc);
 746                queue->queue = xrealloc(queue->queue,
 747                                        sizeof(dp) * queue->alloc);
 748        }
 749        queue->queue[queue->nr++] = dp;
 750}
 751
 752struct diff_filepair *diff_queue(struct diff_queue_struct *queue,
 753                                 struct diff_filespec *one,
 754                                 struct diff_filespec *two)
 755{
 756        struct diff_filepair *dp = xmalloc(sizeof(*dp));
 757        dp->one = one;
 758        dp->two = two;
 759        dp->score = 0;
 760        dp->status = 0;
 761        dp->source_stays = 0;
 762        dp->broken_pair = 0;
 763        diff_q(queue, dp);
 764        return dp;
 765}
 766
 767void diff_free_filepair(struct diff_filepair *p)
 768{
 769        diff_free_filespec_data(p->one);
 770        diff_free_filespec_data(p->two);
 771        free(p);
 772}
 773
 774static void diff_flush_raw(struct diff_filepair *p,
 775                           int line_termination,
 776                           int inter_name_termination)
 777{
 778        int two_paths;
 779        char status[10];
 780
 781        if (line_termination) {
 782                const char *err = "path %s cannot be expressed without -z";
 783                if (strchr(p->one->path, line_termination) ||
 784                    strchr(p->one->path, inter_name_termination))
 785                        die(err, p->one->path);
 786                if (strchr(p->two->path, line_termination) ||
 787                    strchr(p->two->path, inter_name_termination))
 788                        die(err, p->two->path);
 789        }
 790
 791        if (p->score)
 792                sprintf(status, "%c%03d", p->status,
 793                        (int)(0.5 + p->score * 100.0/MAX_SCORE));
 794        else {
 795                status[0] = p->status;
 796                status[1] = 0;
 797        }
 798        switch (p->status) {
 799        case 'C': case 'R':
 800                two_paths = 1;
 801                break;
 802        case 'N': case 'D':
 803                two_paths = 0;
 804                break;
 805        default:
 806                two_paths = 0;
 807                break;
 808        }
 809        printf(":%06o %06o %s ",
 810               p->one->mode, p->two->mode, sha1_to_hex(p->one->sha1));
 811        printf("%s %s%c%s",
 812               sha1_to_hex(p->two->sha1),
 813               status,
 814               inter_name_termination,
 815               p->one->path);
 816        if (two_paths)
 817                printf("%c%s", inter_name_termination, p->two->path);
 818        putchar(line_termination);
 819}
 820
 821static void diff_flush_name(struct diff_filepair *p,
 822                            int line_termination)
 823{
 824        printf("%s%c", p->two->path, line_termination);
 825}
 826
 827int diff_unmodified_pair(struct diff_filepair *p)
 828{
 829        /* This function is written stricter than necessary to support
 830         * the currently implemented transformers, but the idea is to
 831         * let transformers to produce diff_filepairs any way they want,
 832         * and filter and clean them up here before producing the output.
 833         */
 834        struct diff_filespec *one, *two;
 835
 836        if (DIFF_PAIR_UNMERGED(p))
 837                return 0; /* unmerged is interesting */
 838
 839        one = p->one;
 840        two = p->two;
 841
 842        /* deletion, addition, mode or type change
 843         * and rename are all interesting.
 844         */
 845        if (DIFF_FILE_VALID(one) != DIFF_FILE_VALID(two) ||
 846            DIFF_PAIR_MODE_CHANGED(p) ||
 847            strcmp(one->path, two->path))
 848                return 0;
 849
 850        /* both are valid and point at the same path.  that is, we are
 851         * dealing with a change.
 852         */
 853        if (one->sha1_valid && two->sha1_valid &&
 854            !memcmp(one->sha1, two->sha1, sizeof(one->sha1)))
 855                return 1; /* no change */
 856        if (!one->sha1_valid && !two->sha1_valid)
 857                return 1; /* both look at the same file on the filesystem. */
 858        return 0;
 859}
 860
 861static void diff_flush_patch(struct diff_filepair *p)
 862{
 863        if (diff_unmodified_pair(p))
 864                return;
 865
 866        if ((DIFF_FILE_VALID(p->one) && S_ISDIR(p->one->mode)) ||
 867            (DIFF_FILE_VALID(p->two) && S_ISDIR(p->two->mode)))
 868                return; /* no tree diffs in patch format */ 
 869
 870        run_diff(p);
 871}
 872
 873int diff_queue_is_empty(void)
 874{
 875        struct diff_queue_struct *q = &diff_queued_diff;
 876        int i;
 877        for (i = 0; i < q->nr; i++)
 878                if (!diff_unmodified_pair(q->queue[i]))
 879                        return 0;
 880        return 1;
 881}
 882
 883#if DIFF_DEBUG
 884void diff_debug_filespec(struct diff_filespec *s, int x, const char *one)
 885{
 886        fprintf(stderr, "queue[%d] %s (%s) %s %06o %s\n",
 887                x, one ? : "",
 888                s->path,
 889                DIFF_FILE_VALID(s) ? "valid" : "invalid",
 890                s->mode,
 891                s->sha1_valid ? sha1_to_hex(s->sha1) : "");
 892        fprintf(stderr, "queue[%d] %s size %lu flags %d\n",
 893                x, one ? : "",
 894                s->size, s->xfrm_flags);
 895}
 896
 897void diff_debug_filepair(const struct diff_filepair *p, int i)
 898{
 899        diff_debug_filespec(p->one, i, "one");
 900        diff_debug_filespec(p->two, i, "two");
 901        fprintf(stderr, "score %d, status %c stays %d broken %d\n",
 902                p->score, p->status ? : '?',
 903                p->source_stays, p->broken_pair);
 904}
 905
 906void diff_debug_queue(const char *msg, struct diff_queue_struct *q)
 907{
 908        int i;
 909        if (msg)
 910                fprintf(stderr, "%s\n", msg);
 911        fprintf(stderr, "q->nr = %d\n", q->nr);
 912        for (i = 0; i < q->nr; i++) {
 913                struct diff_filepair *p = q->queue[i];
 914                diff_debug_filepair(p, i);
 915        }
 916}
 917#endif
 918
 919static void diff_resolve_rename_copy(void)
 920{
 921        int i, j;
 922        struct diff_filepair *p, *pp;
 923        struct diff_queue_struct *q = &diff_queued_diff;
 924
 925        diff_debug_queue("resolve-rename-copy", q);
 926
 927        for (i = 0; i < q->nr; i++) {
 928                p = q->queue[i];
 929                p->status = 0; /* undecided */
 930                if (DIFF_PAIR_UNMERGED(p))
 931                        p->status = 'U';
 932                else if (!DIFF_FILE_VALID(p->one))
 933                        p->status = 'N';
 934                else if (!DIFF_FILE_VALID(p->two))
 935                        p->status = 'D';
 936                else if (DIFF_PAIR_TYPE_CHANGED(p))
 937                        p->status = 'T';
 938
 939                /* from this point on, we are dealing with a pair
 940                 * whose both sides are valid and of the same type, i.e.
 941                 * either in-place edit or rename/copy edit.
 942                 */
 943                else if (DIFF_PAIR_RENAME(p)) {
 944                        if (p->source_stays) {
 945                                p->status = 'C';
 946                                continue;
 947                        }
 948                        /* See if there is some other filepair that
 949                         * copies from the same source as us.  If so
 950                         * we are a copy.  Otherwise we are a rename.
 951                         */
 952                        for (j = i + 1; j < q->nr; j++) {
 953                                pp = q->queue[j];
 954                                if (strcmp(pp->one->path, p->one->path))
 955                                        continue; /* not us */
 956                                if (!DIFF_PAIR_RENAME(pp))
 957                                        continue; /* not a rename/copy */
 958                                /* pp is a rename/copy from the same source */
 959                                p->status = 'C';
 960                                break;
 961                        }
 962                        if (!p->status)
 963                                p->status = 'R';
 964                }
 965                else if (memcmp(p->one->sha1, p->two->sha1, 20) ||
 966                         p->one->mode != p->two->mode)
 967                        p->status = 'M';
 968                else {
 969                        /* This is a "no-change" entry and should not
 970                         * happen anymore, but prepare for broken callers.
 971                         */
 972                        error("feeding unmodified %s to diffcore",
 973                              p->one->path);
 974                        p->status = 'X';
 975                }
 976        }
 977        diff_debug_queue("resolve-rename-copy done", q);
 978}
 979
 980void diff_flush(int diff_output_style, int line_termination)
 981{
 982        struct diff_queue_struct *q = &diff_queued_diff;
 983        int i;
 984        int inter_name_termination = '\t';
 985
 986        if (!line_termination)
 987                inter_name_termination = 0;
 988
 989        for (i = 0; i < q->nr; i++) {
 990                struct diff_filepair *p = q->queue[i];
 991                if ((diff_output_style == DIFF_FORMAT_NO_OUTPUT) ||
 992                    (p->status == 'X'))
 993                        continue;
 994                if (p->status == 0)
 995                        die("internal error in diff-resolve-rename-copy");
 996                switch (diff_output_style) {
 997                case DIFF_FORMAT_PATCH:
 998                        diff_flush_patch(p);
 999                        break;
1000                case DIFF_FORMAT_RAW:
1001                        diff_flush_raw(p, line_termination,
1002                                       inter_name_termination);
1003                        break;
1004                case DIFF_FORMAT_NAME:
1005                        diff_flush_name(p, line_termination);
1006                        break;
1007                }
1008        }
1009        for (i = 0; i < q->nr; i++)
1010                diff_free_filepair(q->queue[i]);
1011        free(q->queue);
1012        q->queue = NULL;
1013        q->nr = q->alloc = 0;
1014}
1015
1016static void diffcore_apply_filter(const char *filter)
1017{
1018        int i;
1019        struct diff_queue_struct *q = &diff_queued_diff;
1020        struct diff_queue_struct outq;
1021        outq.queue = NULL;
1022        outq.nr = outq.alloc = 0;
1023
1024        if (!filter)
1025                return;
1026
1027        if (strchr(filter, 'A')) {
1028                /* All-or-none */
1029                int found;
1030                for (i = found = 0; !found && i < q->nr; i++) {
1031                        struct diff_filepair *p = q->queue[i];
1032                        if (((p->status == 'M') &&
1033                             ((p->score && strchr(filter, 'B')) ||
1034                              (!p->score && strchr(filter, 'M')))) ||
1035                            ((p->status != 'M') && strchr(filter, p->status)))
1036                                found++;
1037                }
1038                if (found)
1039                        return;
1040
1041                /* otherwise we will clear the whole queue
1042                 * by copying the empty outq at the end of this
1043                 * function, but first clear the current entries
1044                 * in the queue.
1045                 */
1046                for (i = 0; i < q->nr; i++)
1047                        diff_free_filepair(q->queue[i]);
1048        }
1049        else {
1050                /* Only the matching ones */
1051                for (i = 0; i < q->nr; i++) {
1052                        struct diff_filepair *p = q->queue[i];
1053                        if (((p->status == 'M') &&
1054                             ((p->score && strchr(filter, 'B')) ||
1055                              (!p->score && strchr(filter, 'M')))) ||
1056                            ((p->status != 'M') && strchr(filter, p->status)))
1057                                diff_q(&outq, p);
1058                        else
1059                                diff_free_filepair(p);
1060                }
1061        }
1062        free(q->queue);
1063        *q = outq;
1064}
1065
1066void diffcore_std(const char **paths,
1067                  int detect_rename, int rename_score,
1068                  const char *pickaxe, int pickaxe_opts,
1069                  int break_opt,
1070                  const char *orderfile,
1071                  const char *filter)
1072{
1073        if (paths && paths[0])
1074                diffcore_pathspec(paths);
1075        if (break_opt != -1)
1076                diffcore_break(break_opt);
1077        if (detect_rename)
1078                diffcore_rename(detect_rename, rename_score);
1079        if (break_opt != -1)
1080                diffcore_merge_broken();
1081        if (pickaxe)
1082                diffcore_pickaxe(pickaxe, pickaxe_opts);
1083        if (orderfile)
1084                diffcore_order(orderfile);
1085        diff_resolve_rename_copy();
1086        diffcore_apply_filter(filter);
1087}
1088
1089
1090void diffcore_std_no_resolve(const char **paths,
1091                             const char *pickaxe, int pickaxe_opts,
1092                             const char *orderfile,
1093                             const char *filter)
1094{
1095        if (paths && paths[0])
1096                diffcore_pathspec(paths);
1097        if (pickaxe)
1098                diffcore_pickaxe(pickaxe, pickaxe_opts);
1099        if (orderfile)
1100                diffcore_order(orderfile);
1101        diffcore_apply_filter(filter);
1102}
1103
1104void diff_addremove(int addremove, unsigned mode,
1105                    const unsigned char *sha1,
1106                    const char *base, const char *path)
1107{
1108        char concatpath[PATH_MAX];
1109        struct diff_filespec *one, *two;
1110
1111        /* This may look odd, but it is a preparation for
1112         * feeding "there are unchanged files which should
1113         * not produce diffs, but when you are doing copy
1114         * detection you would need them, so here they are"
1115         * entries to the diff-core.  They will be prefixed
1116         * with something like '=' or '*' (I haven't decided
1117         * which but should not make any difference).
1118         * Feeding the same new and old to diff_change() 
1119         * also has the same effect.
1120         * Before the final output happens, they are pruned after
1121         * merged into rename/copy pairs as appropriate.
1122         */
1123        if (reverse_diff)
1124                addremove = (addremove == '+' ? '-' :
1125                             addremove == '-' ? '+' : addremove);
1126
1127        if (!path) path = "";
1128        sprintf(concatpath, "%s%s", base, path);
1129        one = alloc_filespec(concatpath);
1130        two = alloc_filespec(concatpath);
1131
1132        if (addremove != '+')
1133                fill_filespec(one, sha1, mode);
1134        if (addremove != '-')
1135                fill_filespec(two, sha1, mode);
1136
1137        diff_queue(&diff_queued_diff, one, two);
1138}
1139
1140void diff_helper_input(unsigned old_mode,
1141                       unsigned new_mode,
1142                       const unsigned char *old_sha1,
1143                       const unsigned char *new_sha1,
1144                       const char *old_path,
1145                       int status,
1146                       int score,
1147                       const char *new_path)
1148{
1149        struct diff_filespec *one, *two;
1150        struct diff_filepair *dp;
1151
1152        one = alloc_filespec(old_path);
1153        two = alloc_filespec(new_path);
1154        if (old_mode)
1155                fill_filespec(one, old_sha1, old_mode);
1156        if (new_mode)
1157                fill_filespec(two, new_sha1, new_mode);
1158        dp = diff_queue(&diff_queued_diff, one, two);
1159        dp->score = score * MAX_SCORE / 100;
1160        dp->status = status;
1161}
1162
1163void diff_change(unsigned old_mode, unsigned new_mode,
1164                 const unsigned char *old_sha1,
1165                 const unsigned char *new_sha1,
1166                 const char *base, const char *path) 
1167{
1168        char concatpath[PATH_MAX];
1169        struct diff_filespec *one, *two;
1170
1171        if (reverse_diff) {
1172                unsigned tmp;
1173                const unsigned char *tmp_c;
1174                tmp = old_mode; old_mode = new_mode; new_mode = tmp;
1175                tmp_c = old_sha1; old_sha1 = new_sha1; new_sha1 = tmp_c;
1176        }
1177        if (!path) path = "";
1178        sprintf(concatpath, "%s%s", base, path);
1179        one = alloc_filespec(concatpath);
1180        two = alloc_filespec(concatpath);
1181        fill_filespec(one, old_sha1, old_mode);
1182        fill_filespec(two, new_sha1, new_mode);
1183
1184        diff_queue(&diff_queued_diff, one, two);
1185}
1186
1187void diff_unmerge(const char *path)
1188{
1189        struct diff_filespec *one, *two;
1190        one = alloc_filespec(path);
1191        two = alloc_filespec(path);
1192        diff_queue(&diff_queued_diff, one, two);
1193}