506e12a031c2301e0fdc40aab3e4b2296cca1b6b
   1#include "cache.h"
   2#include "tree-walk.h"
   3#include "unpack-trees.h"
   4#include "dir.h"
   5#include "object-store.h"
   6#include "tree.h"
   7#include "pathspec.h"
   8
   9static const char *get_mode(const char *str, unsigned int *modep)
  10{
  11        unsigned char c;
  12        unsigned int mode = 0;
  13
  14        if (*str == ' ')
  15                return NULL;
  16
  17        while ((c = *str++) != ' ') {
  18                if (c < '0' || c > '7')
  19                        return NULL;
  20                mode = (mode << 3) + (c - '0');
  21        }
  22        *modep = mode;
  23        return str;
  24}
  25
  26static int decode_tree_entry(struct tree_desc *desc, const char *buf, unsigned long size, struct strbuf *err)
  27{
  28        const char *path;
  29        unsigned int mode, len;
  30        const unsigned hashsz = the_hash_algo->rawsz;
  31
  32        if (size < hashsz + 3 || buf[size - (hashsz + 1)]) {
  33                strbuf_addstr(err, _("too-short tree object"));
  34                return -1;
  35        }
  36
  37        path = get_mode(buf, &mode);
  38        if (!path) {
  39                strbuf_addstr(err, _("malformed mode in tree entry"));
  40                return -1;
  41        }
  42        if (!*path) {
  43                strbuf_addstr(err, _("empty filename in tree entry"));
  44                return -1;
  45        }
  46        len = strlen(path) + 1;
  47
  48        /* Initialize the descriptor entry */
  49        desc->entry.path = path;
  50        desc->entry.mode = canon_mode(mode);
  51        desc->entry.pathlen = len - 1;
  52        hashcpy(desc->entry.oid.hash, (const unsigned char *)path + len);
  53
  54        return 0;
  55}
  56
  57static int init_tree_desc_internal(struct tree_desc *desc, const void *buffer, unsigned long size, struct strbuf *err)
  58{
  59        desc->buffer = buffer;
  60        desc->size = size;
  61        if (size)
  62                return decode_tree_entry(desc, buffer, size, err);
  63        return 0;
  64}
  65
  66void init_tree_desc(struct tree_desc *desc, const void *buffer, unsigned long size)
  67{
  68        struct strbuf err = STRBUF_INIT;
  69        if (init_tree_desc_internal(desc, buffer, size, &err))
  70                die("%s", err.buf);
  71        strbuf_release(&err);
  72}
  73
  74int init_tree_desc_gently(struct tree_desc *desc, const void *buffer, unsigned long size)
  75{
  76        struct strbuf err = STRBUF_INIT;
  77        int result = init_tree_desc_internal(desc, buffer, size, &err);
  78        if (result)
  79                error("%s", err.buf);
  80        strbuf_release(&err);
  81        return result;
  82}
  83
  84void *fill_tree_descriptor(struct repository *r,
  85                           struct tree_desc *desc,
  86                           const struct object_id *oid)
  87{
  88        unsigned long size = 0;
  89        void *buf = NULL;
  90
  91        if (oid) {
  92                buf = read_object_with_reference(r, oid, tree_type, &size, NULL);
  93                if (!buf)
  94                        die("unable to read tree %s", oid_to_hex(oid));
  95        }
  96        init_tree_desc(desc, buf, size);
  97        return buf;
  98}
  99
 100static void entry_clear(struct name_entry *a)
 101{
 102        memset(a, 0, sizeof(*a));
 103}
 104
 105static void entry_extract(struct tree_desc *t, struct name_entry *a)
 106{
 107        *a = t->entry;
 108}
 109
 110static int update_tree_entry_internal(struct tree_desc *desc, struct strbuf *err)
 111{
 112        const void *buf = desc->buffer;
 113        const unsigned char *end = (const unsigned char *)desc->entry.path + desc->entry.pathlen + 1 + the_hash_algo->rawsz;
 114        unsigned long size = desc->size;
 115        unsigned long len = end - (const unsigned char *)buf;
 116
 117        if (size < len)
 118                die(_("too-short tree file"));
 119        buf = end;
 120        size -= len;
 121        desc->buffer = buf;
 122        desc->size = size;
 123        if (size)
 124                return decode_tree_entry(desc, buf, size, err);
 125        return 0;
 126}
 127
 128void update_tree_entry(struct tree_desc *desc)
 129{
 130        struct strbuf err = STRBUF_INIT;
 131        if (update_tree_entry_internal(desc, &err))
 132                die("%s", err.buf);
 133        strbuf_release(&err);
 134}
 135
 136int update_tree_entry_gently(struct tree_desc *desc)
 137{
 138        struct strbuf err = STRBUF_INIT;
 139        if (update_tree_entry_internal(desc, &err)) {
 140                error("%s", err.buf);
 141                strbuf_release(&err);
 142                /* Stop processing this tree after error */
 143                desc->size = 0;
 144                return -1;
 145        }
 146        strbuf_release(&err);
 147        return 0;
 148}
 149
 150int tree_entry(struct tree_desc *desc, struct name_entry *entry)
 151{
 152        if (!desc->size)
 153                return 0;
 154
 155        *entry = desc->entry;
 156        update_tree_entry(desc);
 157        return 1;
 158}
 159
 160int tree_entry_gently(struct tree_desc *desc, struct name_entry *entry)
 161{
 162        if (!desc->size)
 163                return 0;
 164
 165        *entry = desc->entry;
 166        if (update_tree_entry_gently(desc))
 167                return 0;
 168        return 1;
 169}
 170
 171void setup_traverse_info(struct traverse_info *info, const char *base)
 172{
 173        int pathlen = strlen(base);
 174        static struct traverse_info dummy;
 175
 176        memset(info, 0, sizeof(*info));
 177        if (pathlen && base[pathlen-1] == '/')
 178                pathlen--;
 179        info->pathlen = pathlen ? pathlen + 1 : 0;
 180        info->name.path = base;
 181        info->name.pathlen = pathlen;
 182        if (pathlen) {
 183                hashcpy(info->name.oid.hash, (const unsigned char *)base + pathlen + 1);
 184                info->prev = &dummy;
 185        }
 186}
 187
 188char *make_traverse_path(char *path, const struct traverse_info *info, const struct name_entry *n)
 189{
 190        int len = tree_entry_len(n);
 191        int pathlen = info->pathlen;
 192
 193        path[pathlen + len] = 0;
 194        for (;;) {
 195                memcpy(path + pathlen, n->path, len);
 196                if (!pathlen)
 197                        break;
 198                path[--pathlen] = '/';
 199                n = &info->name;
 200                len = tree_entry_len(n);
 201                info = info->prev;
 202                pathlen -= len;
 203        }
 204        return path;
 205}
 206
 207struct tree_desc_skip {
 208        struct tree_desc_skip *prev;
 209        const void *ptr;
 210};
 211
 212struct tree_desc_x {
 213        struct tree_desc d;
 214        struct tree_desc_skip *skip;
 215};
 216
 217static int check_entry_match(const char *a, int a_len, const char *b, int b_len)
 218{
 219        /*
 220         * The caller wants to pick *a* from a tree or nothing.
 221         * We are looking at *b* in a tree.
 222         *
 223         * (0) If a and b are the same name, we are trivially happy.
 224         *
 225         * There are three possibilities where *a* could be hiding
 226         * behind *b*.
 227         *
 228         * (1) *a* == "t",   *b* == "ab"  i.e. *b* sorts earlier than *a* no
 229         *                                matter what.
 230         * (2) *a* == "t",   *b* == "t-2" and "t" is a subtree in the tree;
 231         * (3) *a* == "t-2", *b* == "t"   and "t-2" is a blob in the tree.
 232         *
 233         * Otherwise we know *a* won't appear in the tree without
 234         * scanning further.
 235         */
 236
 237        int cmp = name_compare(a, a_len, b, b_len);
 238
 239        /* Most common case first -- reading sync'd trees */
 240        if (!cmp)
 241                return cmp;
 242
 243        if (0 < cmp) {
 244                /* a comes after b; it does not matter if it is case (3)
 245                if (b_len < a_len && !memcmp(a, b, b_len) && a[b_len] < '/')
 246                        return 1;
 247                */
 248                return 1; /* keep looking */
 249        }
 250
 251        /* b comes after a; are we looking at case (2)? */
 252        if (a_len < b_len && !memcmp(a, b, a_len) && b[a_len] < '/')
 253                return 1; /* keep looking */
 254
 255        return -1; /* a cannot appear in the tree */
 256}
 257
 258/*
 259 * From the extended tree_desc, extract the first name entry, while
 260 * paying attention to the candidate "first" name.  Most importantly,
 261 * when looking for an entry, if there are entries that sorts earlier
 262 * in the tree object representation than that name, skip them and
 263 * process the named entry first.  We will remember that we haven't
 264 * processed the first entry yet, and in the later call skip the
 265 * entry we processed early when update_extended_entry() is called.
 266 *
 267 * E.g. if the underlying tree object has these entries:
 268 *
 269 *    blob    "t-1"
 270 *    blob    "t-2"
 271 *    tree    "t"
 272 *    blob    "t=1"
 273 *
 274 * and the "first" asks for "t", remember that we still need to
 275 * process "t-1" and "t-2" but extract "t".  After processing the
 276 * entry "t" from this call, the caller will let us know by calling
 277 * update_extended_entry() that we can remember "t" has been processed
 278 * already.
 279 */
 280
 281static void extended_entry_extract(struct tree_desc_x *t,
 282                                   struct name_entry *a,
 283                                   const char *first,
 284                                   int first_len)
 285{
 286        const char *path;
 287        int len;
 288        struct tree_desc probe;
 289        struct tree_desc_skip *skip;
 290
 291        /*
 292         * Extract the first entry from the tree_desc, but skip the
 293         * ones that we already returned in earlier rounds.
 294         */
 295        while (1) {
 296                if (!t->d.size) {
 297                        entry_clear(a);
 298                        break; /* not found */
 299                }
 300                entry_extract(&t->d, a);
 301                for (skip = t->skip; skip; skip = skip->prev)
 302                        if (a->path == skip->ptr)
 303                                break; /* found */
 304                if (!skip)
 305                        break;
 306                /* We have processed this entry already. */
 307                update_tree_entry(&t->d);
 308        }
 309
 310        if (!first || !a->path)
 311                return;
 312
 313        /*
 314         * The caller wants "first" from this tree, or nothing.
 315         */
 316        path = a->path;
 317        len = tree_entry_len(a);
 318        switch (check_entry_match(first, first_len, path, len)) {
 319        case -1:
 320                entry_clear(a);
 321        case 0:
 322                return;
 323        default:
 324                break;
 325        }
 326
 327        /*
 328         * We need to look-ahead -- we suspect that a subtree whose
 329         * name is "first" may be hiding behind the current entry "path".
 330         */
 331        probe = t->d;
 332        while (probe.size) {
 333                entry_extract(&probe, a);
 334                path = a->path;
 335                len = tree_entry_len(a);
 336                switch (check_entry_match(first, first_len, path, len)) {
 337                case -1:
 338                        entry_clear(a);
 339                case 0:
 340                        return;
 341                default:
 342                        update_tree_entry(&probe);
 343                        break;
 344                }
 345                /* keep looking */
 346        }
 347        entry_clear(a);
 348}
 349
 350static void update_extended_entry(struct tree_desc_x *t, struct name_entry *a)
 351{
 352        if (t->d.entry.path == a->path) {
 353                update_tree_entry(&t->d);
 354        } else {
 355                /* we have returned this entry early */
 356                struct tree_desc_skip *skip = xmalloc(sizeof(*skip));
 357                skip->ptr = a->path;
 358                skip->prev = t->skip;
 359                t->skip = skip;
 360        }
 361}
 362
 363static void free_extended_entry(struct tree_desc_x *t)
 364{
 365        struct tree_desc_skip *p, *s;
 366
 367        for (s = t->skip; s; s = p) {
 368                p = s->prev;
 369                free(s);
 370        }
 371}
 372
 373static inline int prune_traversal(struct index_state *istate,
 374                                  struct name_entry *e,
 375                                  struct traverse_info *info,
 376                                  struct strbuf *base,
 377                                  int still_interesting)
 378{
 379        if (!info->pathspec || still_interesting == 2)
 380                return 2;
 381        if (still_interesting < 0)
 382                return still_interesting;
 383        return tree_entry_interesting(istate, e, base,
 384                                      0, info->pathspec);
 385}
 386
 387int traverse_trees(struct index_state *istate,
 388                   int n, struct tree_desc *t,
 389                   struct traverse_info *info)
 390{
 391        int error = 0;
 392        struct name_entry *entry = xmalloc(n*sizeof(*entry));
 393        int i;
 394        struct tree_desc_x *tx = xcalloc(n, sizeof(*tx));
 395        struct strbuf base = STRBUF_INIT;
 396        int interesting = 1;
 397        char *traverse_path;
 398
 399        for (i = 0; i < n; i++)
 400                tx[i].d = t[i];
 401
 402        if (info->prev) {
 403                strbuf_grow(&base, info->pathlen);
 404                make_traverse_path(base.buf, info->prev, &info->name);
 405                base.buf[info->pathlen-1] = '/';
 406                strbuf_setlen(&base, info->pathlen);
 407                traverse_path = xstrndup(base.buf, info->pathlen);
 408        } else {
 409                traverse_path = xstrndup(info->name.path, info->pathlen);
 410        }
 411        info->traverse_path = traverse_path;
 412        for (;;) {
 413                int trees_used;
 414                unsigned long mask, dirmask;
 415                const char *first = NULL;
 416                int first_len = 0;
 417                struct name_entry *e = NULL;
 418                int len;
 419
 420                for (i = 0; i < n; i++) {
 421                        e = entry + i;
 422                        extended_entry_extract(tx + i, e, NULL, 0);
 423                }
 424
 425                /*
 426                 * A tree may have "t-2" at the current location even
 427                 * though it may have "t" that is a subtree behind it,
 428                 * and another tree may return "t".  We want to grab
 429                 * all "t" from all trees to match in such a case.
 430                 */
 431                for (i = 0; i < n; i++) {
 432                        e = entry + i;
 433                        if (!e->path)
 434                                continue;
 435                        len = tree_entry_len(e);
 436                        if (!first) {
 437                                first = e->path;
 438                                first_len = len;
 439                                continue;
 440                        }
 441                        if (name_compare(e->path, len, first, first_len) < 0) {
 442                                first = e->path;
 443                                first_len = len;
 444                        }
 445                }
 446
 447                if (first) {
 448                        for (i = 0; i < n; i++) {
 449                                e = entry + i;
 450                                extended_entry_extract(tx + i, e, first, first_len);
 451                                /* Cull the ones that are not the earliest */
 452                                if (!e->path)
 453                                        continue;
 454                                len = tree_entry_len(e);
 455                                if (name_compare(e->path, len, first, first_len))
 456                                        entry_clear(e);
 457                        }
 458                }
 459
 460                /* Now we have in entry[i] the earliest name from the trees */
 461                mask = 0;
 462                dirmask = 0;
 463                for (i = 0; i < n; i++) {
 464                        if (!entry[i].path)
 465                                continue;
 466                        mask |= 1ul << i;
 467                        if (S_ISDIR(entry[i].mode))
 468                                dirmask |= 1ul << i;
 469                        e = &entry[i];
 470                }
 471                if (!mask)
 472                        break;
 473                interesting = prune_traversal(istate, e, info, &base, interesting);
 474                if (interesting < 0)
 475                        break;
 476                if (interesting) {
 477                        trees_used = info->fn(n, mask, dirmask, entry, info);
 478                        if (trees_used < 0) {
 479                                error = trees_used;
 480                                if (!info->show_all_errors)
 481                                        break;
 482                        }
 483                        mask &= trees_used;
 484                }
 485                for (i = 0; i < n; i++)
 486                        if (mask & (1ul << i))
 487                                update_extended_entry(tx + i, entry + i);
 488        }
 489        free(entry);
 490        for (i = 0; i < n; i++)
 491                free_extended_entry(tx + i);
 492        free(tx);
 493        free(traverse_path);
 494        info->traverse_path = NULL;
 495        strbuf_release(&base);
 496        return error;
 497}
 498
 499struct dir_state {
 500        void *tree;
 501        unsigned long size;
 502        struct object_id oid;
 503};
 504
 505static int find_tree_entry(struct repository *r, struct tree_desc *t,
 506                           const char *name, struct object_id *result,
 507                           unsigned short *mode)
 508{
 509        int namelen = strlen(name);
 510        while (t->size) {
 511                const char *entry;
 512                struct object_id oid;
 513                int entrylen, cmp;
 514
 515                oidcpy(&oid, tree_entry_extract(t, &entry, mode));
 516                entrylen = tree_entry_len(&t->entry);
 517                update_tree_entry(t);
 518                if (entrylen > namelen)
 519                        continue;
 520                cmp = memcmp(name, entry, entrylen);
 521                if (cmp > 0)
 522                        continue;
 523                if (cmp < 0)
 524                        break;
 525                if (entrylen == namelen) {
 526                        oidcpy(result, &oid);
 527                        return 0;
 528                }
 529                if (name[entrylen] != '/')
 530                        continue;
 531                if (!S_ISDIR(*mode))
 532                        break;
 533                if (++entrylen == namelen) {
 534                        oidcpy(result, &oid);
 535                        return 0;
 536                }
 537                return get_tree_entry(r, &oid, name + entrylen, result, mode);
 538        }
 539        return -1;
 540}
 541
 542int get_tree_entry(struct repository *r,
 543                   const struct object_id *tree_oid,
 544                   const char *name,
 545                   struct object_id *oid,
 546                   unsigned short *mode)
 547{
 548        int retval;
 549        void *tree;
 550        unsigned long size;
 551        struct object_id root;
 552
 553        tree = read_object_with_reference(r, tree_oid, tree_type, &size, &root);
 554        if (!tree)
 555                return -1;
 556
 557        if (name[0] == '\0') {
 558                oidcpy(oid, &root);
 559                free(tree);
 560                return 0;
 561        }
 562
 563        if (!size) {
 564                retval = -1;
 565        } else {
 566                struct tree_desc t;
 567                init_tree_desc(&t, tree, size);
 568                retval = find_tree_entry(r, &t, name, oid, mode);
 569        }
 570        free(tree);
 571        return retval;
 572}
 573
 574/*
 575 * This is Linux's built-in max for the number of symlinks to follow.
 576 * That limit, of course, does not affect git, but it's a reasonable
 577 * choice.
 578 */
 579#define GET_TREE_ENTRY_FOLLOW_SYMLINKS_MAX_LINKS 40
 580
 581/**
 582 * Find a tree entry by following symlinks in tree_sha (which is
 583 * assumed to be the root of the repository).  In the event that a
 584 * symlink points outside the repository (e.g. a link to /foo or a
 585 * root-level link to ../foo), the portion of the link which is
 586 * outside the repository will be returned in result_path, and *mode
 587 * will be set to 0.  It is assumed that result_path is uninitialized.
 588 * If there are no symlinks, or the end result of the symlink chain
 589 * points to an object inside the repository, result will be filled in
 590 * with the sha1 of the found object, and *mode will hold the mode of
 591 * the object.
 592 *
 593 * See the code for enum get_oid_result for a description of
 594 * the return values.
 595 */
 596enum get_oid_result get_tree_entry_follow_symlinks(struct object_id *tree_oid, const char *name, struct object_id *result, struct strbuf *result_path, unsigned short *mode)
 597{
 598        int retval = MISSING_OBJECT;
 599        struct dir_state *parents = NULL;
 600        size_t parents_alloc = 0;
 601        size_t i, parents_nr = 0;
 602        struct object_id current_tree_oid;
 603        struct strbuf namebuf = STRBUF_INIT;
 604        struct tree_desc t;
 605        int follows_remaining = GET_TREE_ENTRY_FOLLOW_SYMLINKS_MAX_LINKS;
 606
 607        init_tree_desc(&t, NULL, 0UL);
 608        strbuf_addstr(&namebuf, name);
 609        oidcpy(&current_tree_oid, tree_oid);
 610
 611        while (1) {
 612                int find_result;
 613                char *first_slash;
 614                char *remainder = NULL;
 615
 616                if (!t.buffer) {
 617                        void *tree;
 618                        struct object_id root;
 619                        unsigned long size;
 620                        tree = read_object_with_reference(the_repository,
 621                                                          &current_tree_oid,
 622                                                          tree_type, &size,
 623                                                          &root);
 624                        if (!tree)
 625                                goto done;
 626
 627                        ALLOC_GROW(parents, parents_nr + 1, parents_alloc);
 628                        parents[parents_nr].tree = tree;
 629                        parents[parents_nr].size = size;
 630                        oidcpy(&parents[parents_nr].oid, &root);
 631                        parents_nr++;
 632
 633                        if (namebuf.buf[0] == '\0') {
 634                                oidcpy(result, &root);
 635                                retval = FOUND;
 636                                goto done;
 637                        }
 638
 639                        if (!size)
 640                                goto done;
 641
 642                        /* descend */
 643                        init_tree_desc(&t, tree, size);
 644                }
 645
 646                /* Handle symlinks to e.g. a//b by removing leading slashes */
 647                while (namebuf.buf[0] == '/') {
 648                        strbuf_remove(&namebuf, 0, 1);
 649                }
 650
 651                /* Split namebuf into a first component and a remainder */
 652                if ((first_slash = strchr(namebuf.buf, '/'))) {
 653                        *first_slash = 0;
 654                        remainder = first_slash + 1;
 655                }
 656
 657                if (!strcmp(namebuf.buf, "..")) {
 658                        struct dir_state *parent;
 659                        /*
 660                         * We could end up with .. in the namebuf if it
 661                         * appears in a symlink.
 662                         */
 663
 664                        if (parents_nr == 1) {
 665                                if (remainder)
 666                                        *first_slash = '/';
 667                                strbuf_add(result_path, namebuf.buf,
 668                                           namebuf.len);
 669                                *mode = 0;
 670                                retval = FOUND;
 671                                goto done;
 672                        }
 673                        parent = &parents[parents_nr - 1];
 674                        free(parent->tree);
 675                        parents_nr--;
 676                        parent = &parents[parents_nr - 1];
 677                        init_tree_desc(&t, parent->tree, parent->size);
 678                        strbuf_remove(&namebuf, 0, remainder ? 3 : 2);
 679                        continue;
 680                }
 681
 682                /* We could end up here via a symlink to dir/.. */
 683                if (namebuf.buf[0] == '\0') {
 684                        oidcpy(result, &parents[parents_nr - 1].oid);
 685                        retval = FOUND;
 686                        goto done;
 687                }
 688
 689                /* Look up the first (or only) path component in the tree. */
 690                find_result = find_tree_entry(the_repository, &t, namebuf.buf,
 691                                              &current_tree_oid, mode);
 692                if (find_result) {
 693                        goto done;
 694                }
 695
 696                if (S_ISDIR(*mode)) {
 697                        if (!remainder) {
 698                                oidcpy(result, &current_tree_oid);
 699                                retval = FOUND;
 700                                goto done;
 701                        }
 702                        /* Descend the tree */
 703                        t.buffer = NULL;
 704                        strbuf_remove(&namebuf, 0,
 705                                      1 + first_slash - namebuf.buf);
 706                } else if (S_ISREG(*mode)) {
 707                        if (!remainder) {
 708                                oidcpy(result, &current_tree_oid);
 709                                retval = FOUND;
 710                        } else {
 711                                retval = NOT_DIR;
 712                        }
 713                        goto done;
 714                } else if (S_ISLNK(*mode)) {
 715                        /* Follow a symlink */
 716                        unsigned long link_len;
 717                        size_t len;
 718                        char *contents, *contents_start;
 719                        struct dir_state *parent;
 720                        enum object_type type;
 721
 722                        if (follows_remaining-- == 0) {
 723                                /* Too many symlinks followed */
 724                                retval = SYMLINK_LOOP;
 725                                goto done;
 726                        }
 727
 728                        /*
 729                         * At this point, we have followed at a least
 730                         * one symlink, so on error we need to report this.
 731                         */
 732                        retval = DANGLING_SYMLINK;
 733
 734                        contents = read_object_file(&current_tree_oid, &type,
 735                                                    &link_len);
 736
 737                        if (!contents)
 738                                goto done;
 739
 740                        if (contents[0] == '/') {
 741                                strbuf_addstr(result_path, contents);
 742                                free(contents);
 743                                *mode = 0;
 744                                retval = FOUND;
 745                                goto done;
 746                        }
 747
 748                        if (remainder)
 749                                len = first_slash - namebuf.buf;
 750                        else
 751                                len = namebuf.len;
 752
 753                        contents_start = contents;
 754
 755                        parent = &parents[parents_nr - 1];
 756                        init_tree_desc(&t, parent->tree, parent->size);
 757                        strbuf_splice(&namebuf, 0, len,
 758                                      contents_start, link_len);
 759                        if (remainder)
 760                                namebuf.buf[link_len] = '/';
 761                        free(contents);
 762                }
 763        }
 764done:
 765        for (i = 0; i < parents_nr; i++)
 766                free(parents[i].tree);
 767        free(parents);
 768
 769        strbuf_release(&namebuf);
 770        return retval;
 771}
 772
 773static int match_entry(const struct pathspec_item *item,
 774                       const struct name_entry *entry, int pathlen,
 775                       const char *match, int matchlen,
 776                       enum interesting *never_interesting)
 777{
 778        int m = -1; /* signals that we haven't called strncmp() */
 779
 780        if (item->magic & PATHSPEC_ICASE)
 781                /*
 782                 * "Never interesting" trick requires exact
 783                 * matching. We could do something clever with inexact
 784                 * matching, but it's trickier (and not to forget that
 785                 * strcasecmp is locale-dependent, at least in
 786                 * glibc). Just disable it for now. It can't be worse
 787                 * than the wildcard's codepath of '[Tt][Hi][Is][Ss]'
 788                 * pattern.
 789                 */
 790                *never_interesting = entry_not_interesting;
 791        else if (*never_interesting != entry_not_interesting) {
 792                /*
 793                 * We have not seen any match that sorts later
 794                 * than the current path.
 795                 */
 796
 797                /*
 798                 * Does match sort strictly earlier than path
 799                 * with their common parts?
 800                 */
 801                m = strncmp(match, entry->path,
 802                            (matchlen < pathlen) ? matchlen : pathlen);
 803                if (m < 0)
 804                        return 0;
 805
 806                /*
 807                 * If we come here even once, that means there is at
 808                 * least one pathspec that would sort equal to or
 809                 * later than the path we are currently looking at.
 810                 * In other words, if we have never reached this point
 811                 * after iterating all pathspecs, it means all
 812                 * pathspecs are either outside of base, or inside the
 813                 * base but sorts strictly earlier than the current
 814                 * one.  In either case, they will never match the
 815                 * subsequent entries.  In such a case, we initialized
 816                 * the variable to -1 and that is what will be
 817                 * returned, allowing the caller to terminate early.
 818                 */
 819                *never_interesting = entry_not_interesting;
 820        }
 821
 822        if (pathlen > matchlen)
 823                return 0;
 824
 825        if (matchlen > pathlen) {
 826                if (match[pathlen] != '/')
 827                        return 0;
 828                if (!S_ISDIR(entry->mode) && !S_ISGITLINK(entry->mode))
 829                        return 0;
 830        }
 831
 832        if (m == -1)
 833                /*
 834                 * we cheated and did not do strncmp(), so we do
 835                 * that here.
 836                 */
 837                m = ps_strncmp(item, match, entry->path, pathlen);
 838
 839        /*
 840         * If common part matched earlier then it is a hit,
 841         * because we rejected the case where path is not a
 842         * leading directory and is shorter than match.
 843         */
 844        if (!m)
 845                /*
 846                 * match_entry does not check if the prefix part is
 847                 * matched case-sensitively. If the entry is a
 848                 * directory and part of prefix, it'll be rematched
 849                 * eventually by basecmp with special treatment for
 850                 * the prefix.
 851                 */
 852                return 1;
 853
 854        return 0;
 855}
 856
 857/* :(icase)-aware string compare */
 858static int basecmp(const struct pathspec_item *item,
 859                   const char *base, const char *match, int len)
 860{
 861        if (item->magic & PATHSPEC_ICASE) {
 862                int ret, n = len > item->prefix ? item->prefix : len;
 863                ret = strncmp(base, match, n);
 864                if (ret)
 865                        return ret;
 866                base += n;
 867                match += n;
 868                len -= n;
 869        }
 870        return ps_strncmp(item, base, match, len);
 871}
 872
 873static int match_dir_prefix(const struct pathspec_item *item,
 874                            const char *base,
 875                            const char *match, int matchlen)
 876{
 877        if (basecmp(item, base, match, matchlen))
 878                return 0;
 879
 880        /*
 881         * If the base is a subdirectory of a path which
 882         * was specified, all of them are interesting.
 883         */
 884        if (!matchlen ||
 885            base[matchlen] == '/' ||
 886            match[matchlen - 1] == '/')
 887                return 1;
 888
 889        /* Just a random prefix match */
 890        return 0;
 891}
 892
 893/*
 894 * Perform matching on the leading non-wildcard part of
 895 * pathspec. item->nowildcard_len must be greater than zero. Return
 896 * non-zero if base is matched.
 897 */
 898static int match_wildcard_base(const struct pathspec_item *item,
 899                               const char *base, int baselen,
 900                               int *matched)
 901{
 902        const char *match = item->match;
 903        /* the wildcard part is not considered in this function */
 904        int matchlen = item->nowildcard_len;
 905
 906        if (baselen) {
 907                int dirlen;
 908                /*
 909                 * Return early if base is longer than the
 910                 * non-wildcard part but it does not match.
 911                 */
 912                if (baselen >= matchlen) {
 913                        *matched = matchlen;
 914                        return !basecmp(item, base, match, matchlen);
 915                }
 916
 917                dirlen = matchlen;
 918                while (dirlen && match[dirlen - 1] != '/')
 919                        dirlen--;
 920
 921                /*
 922                 * Return early if base is shorter than the
 923                 * non-wildcard part but it does not match. Note that
 924                 * base ends with '/' so we are sure it really matches
 925                 * directory
 926                 */
 927                if (basecmp(item, base, match, baselen))
 928                        return 0;
 929                *matched = baselen;
 930        } else
 931                *matched = 0;
 932        /*
 933         * we could have checked entry against the non-wildcard part
 934         * that is not in base and does similar never_interesting
 935         * optimization as in match_entry. For now just be happy with
 936         * base comparison.
 937         */
 938        return entry_interesting;
 939}
 940
 941/*
 942 * Is a tree entry interesting given the pathspec we have?
 943 *
 944 * Pre-condition: either baselen == base_offset (i.e. empty path)
 945 * or base[baselen-1] == '/' (i.e. with trailing slash).
 946 */
 947static enum interesting do_match(struct index_state *istate,
 948                                 const struct name_entry *entry,
 949                                 struct strbuf *base, int base_offset,
 950                                 const struct pathspec *ps,
 951                                 int exclude)
 952{
 953        int i;
 954        int pathlen, baselen = base->len - base_offset;
 955        enum interesting never_interesting = ps->has_wildcard ?
 956                entry_not_interesting : all_entries_not_interesting;
 957
 958        GUARD_PATHSPEC(ps,
 959                       PATHSPEC_FROMTOP |
 960                       PATHSPEC_MAXDEPTH |
 961                       PATHSPEC_LITERAL |
 962                       PATHSPEC_GLOB |
 963                       PATHSPEC_ICASE |
 964                       PATHSPEC_EXCLUDE |
 965                       PATHSPEC_ATTR);
 966
 967        if (!ps->nr) {
 968                if (!ps->recursive ||
 969                    !(ps->magic & PATHSPEC_MAXDEPTH) ||
 970                    ps->max_depth == -1)
 971                        return all_entries_interesting;
 972                return within_depth(base->buf + base_offset, baselen,
 973                                    !!S_ISDIR(entry->mode),
 974                                    ps->max_depth) ?
 975                        entry_interesting : entry_not_interesting;
 976        }
 977
 978        pathlen = tree_entry_len(entry);
 979
 980        for (i = ps->nr - 1; i >= 0; i--) {
 981                const struct pathspec_item *item = ps->items+i;
 982                const char *match = item->match;
 983                const char *base_str = base->buf + base_offset;
 984                int matchlen = item->len, matched = 0;
 985
 986                if ((!exclude &&   item->magic & PATHSPEC_EXCLUDE) ||
 987                    ( exclude && !(item->magic & PATHSPEC_EXCLUDE)))
 988                        continue;
 989
 990                if (baselen >= matchlen) {
 991                        /* If it doesn't match, move along... */
 992                        if (!match_dir_prefix(item, base_str, match, matchlen))
 993                                goto match_wildcards;
 994
 995                        if (!ps->recursive ||
 996                            !(ps->magic & PATHSPEC_MAXDEPTH) ||
 997                            ps->max_depth == -1) {
 998                                if (!item->attr_match_nr)
 999                                        return all_entries_interesting;
1000                                else
1001                                        goto interesting;
1002                        }
1003
1004                        if (within_depth(base_str + matchlen + 1,
1005                                         baselen - matchlen - 1,
1006                                         !!S_ISDIR(entry->mode),
1007                                         ps->max_depth))
1008                                goto interesting;
1009                        else
1010                                return entry_not_interesting;
1011                }
1012
1013                /* Either there must be no base, or the base must match. */
1014                if (baselen == 0 || !basecmp(item, base_str, match, baselen)) {
1015                        if (match_entry(item, entry, pathlen,
1016                                        match + baselen, matchlen - baselen,
1017                                        &never_interesting))
1018                                goto interesting;
1019
1020                        if (item->nowildcard_len < item->len) {
1021                                if (!git_fnmatch(item, match + baselen, entry->path,
1022                                                 item->nowildcard_len - baselen))
1023                                        goto interesting;
1024
1025                                /*
1026                                 * Match all directories. We'll try to
1027                                 * match files later on.
1028                                 */
1029                                if (ps->recursive && S_ISDIR(entry->mode))
1030                                        return entry_interesting;
1031
1032                                /*
1033                                 * When matching against submodules with
1034                                 * wildcard characters, ensure that the entry
1035                                 * at least matches up to the first wild
1036                                 * character.  More accurate matching can then
1037                                 * be performed in the submodule itself.
1038                                 */
1039                                if (ps->recurse_submodules &&
1040                                    S_ISGITLINK(entry->mode) &&
1041                                    !ps_strncmp(item, match + baselen,
1042                                                entry->path,
1043                                                item->nowildcard_len - baselen))
1044                                        goto interesting;
1045                        }
1046
1047                        continue;
1048                }
1049
1050match_wildcards:
1051                if (item->nowildcard_len == item->len)
1052                        continue;
1053
1054                if (item->nowildcard_len &&
1055                    !match_wildcard_base(item, base_str, baselen, &matched))
1056                        continue;
1057
1058                /*
1059                 * Concatenate base and entry->path into one and do
1060                 * fnmatch() on it.
1061                 *
1062                 * While we could avoid concatenation in certain cases
1063                 * [1], which saves a memcpy and potentially a
1064                 * realloc, it turns out not worth it. Measurement on
1065                 * linux-2.6 does not show any clear improvements,
1066                 * partly because of the nowildcard_len optimization
1067                 * in git_fnmatch(). Avoid micro-optimizations here.
1068                 *
1069                 * [1] if match_wildcard_base() says the base
1070                 * directory is already matched, we only need to match
1071                 * the rest, which is shorter so _in theory_ faster.
1072                 */
1073
1074                strbuf_add(base, entry->path, pathlen);
1075
1076                if (!git_fnmatch(item, match, base->buf + base_offset,
1077                                 item->nowildcard_len)) {
1078                        strbuf_setlen(base, base_offset + baselen);
1079                        goto interesting;
1080                }
1081
1082                /*
1083                 * When matching against submodules with
1084                 * wildcard characters, ensure that the entry
1085                 * at least matches up to the first wild
1086                 * character.  More accurate matching can then
1087                 * be performed in the submodule itself.
1088                 */
1089                if (ps->recurse_submodules && S_ISGITLINK(entry->mode) &&
1090                    !ps_strncmp(item, match, base->buf + base_offset,
1091                                item->nowildcard_len)) {
1092                        strbuf_setlen(base, base_offset + baselen);
1093                        goto interesting;
1094                }
1095
1096                strbuf_setlen(base, base_offset + baselen);
1097
1098                /*
1099                 * Match all directories. We'll try to match files
1100                 * later on.
1101                 * max_depth is ignored but we may consider support it
1102                 * in future, see
1103                 * https://public-inbox.org/git/7vmxo5l2g4.fsf@alter.siamese.dyndns.org/
1104                 */
1105                if (ps->recursive && S_ISDIR(entry->mode))
1106                        return entry_interesting;
1107                continue;
1108interesting:
1109                if (item->attr_match_nr) {
1110                        int ret;
1111
1112                        /*
1113                         * Must not return all_entries_not_interesting
1114                         * prematurely. We do not know if all entries do not
1115                         * match some attributes with current attr API.
1116                         */
1117                        never_interesting = entry_not_interesting;
1118
1119                        /*
1120                         * Consider all directories interesting (because some
1121                         * of those files inside may match some attributes
1122                         * even though the parent dir does not)
1123                         *
1124                         * FIXME: attributes _can_ match directories and we
1125                         * can probably return all_entries_interesting or
1126                         * all_entries_not_interesting here if matched.
1127                         */
1128                        if (S_ISDIR(entry->mode))
1129                                return entry_interesting;
1130
1131                        strbuf_add(base, entry->path, pathlen);
1132                        ret = match_pathspec_attrs(istate, base->buf + base_offset,
1133                                                   base->len - base_offset, item);
1134                        strbuf_setlen(base, base_offset + baselen);
1135                        if (!ret)
1136                                continue;
1137                }
1138                return entry_interesting;
1139        }
1140        return never_interesting; /* No matches */
1141}
1142
1143/*
1144 * Is a tree entry interesting given the pathspec we have?
1145 *
1146 * Pre-condition: either baselen == base_offset (i.e. empty path)
1147 * or base[baselen-1] == '/' (i.e. with trailing slash).
1148 */
1149enum interesting tree_entry_interesting(struct index_state *istate,
1150                                        const struct name_entry *entry,
1151                                        struct strbuf *base, int base_offset,
1152                                        const struct pathspec *ps)
1153{
1154        enum interesting positive, negative;
1155        positive = do_match(istate, entry, base, base_offset, ps, 0);
1156
1157        /*
1158         * case | entry | positive | negative | result
1159         * -----+-------+----------+----------+-------
1160         *   1  |  file |   -1     |  -1..2   |  -1
1161         *   2  |  file |    0     |  -1..2   |   0
1162         *   3  |  file |    1     |   -1     |   1
1163         *   4  |  file |    1     |    0     |   1
1164         *   5  |  file |    1     |    1     |   0
1165         *   6  |  file |    1     |    2     |   0
1166         *   7  |  file |    2     |   -1     |   2
1167         *   8  |  file |    2     |    0     |   1
1168         *   9  |  file |    2     |    1     |   0
1169         *  10  |  file |    2     |    2     |  -1
1170         * -----+-------+----------+----------+-------
1171         *  11  |  dir  |   -1     |  -1..2   |  -1
1172         *  12  |  dir  |    0     |  -1..2   |   0
1173         *  13  |  dir  |    1     |   -1     |   1
1174         *  14  |  dir  |    1     |    0     |   1
1175         *  15  |  dir  |    1     |    1     |   1 (*)
1176         *  16  |  dir  |    1     |    2     |   0
1177         *  17  |  dir  |    2     |   -1     |   2
1178         *  18  |  dir  |    2     |    0     |   1
1179         *  19  |  dir  |    2     |    1     |   1 (*)
1180         *  20  |  dir  |    2     |    2     |  -1
1181         *
1182         * (*) An exclude pattern interested in a directory does not
1183         * necessarily mean it will exclude all of the directory. In
1184         * wildcard case, it can't decide until looking at individual
1185         * files inside. So don't write such directories off yet.
1186         */
1187
1188        if (!(ps->magic & PATHSPEC_EXCLUDE) ||
1189            positive <= entry_not_interesting) /* #1, #2, #11, #12 */
1190                return positive;
1191
1192        negative = do_match(istate, entry, base, base_offset, ps, 1);
1193
1194        /* #8, #18 */
1195        if (positive == all_entries_interesting &&
1196            negative == entry_not_interesting)
1197                return entry_interesting;
1198
1199        /* #3, #4, #7, #13, #14, #17 */
1200        if (negative <= entry_not_interesting)
1201                return positive;
1202
1203        /* #15, #19 */
1204        if (S_ISDIR(entry->mode) &&
1205            positive >= entry_interesting &&
1206            negative == entry_interesting)
1207                return entry_interesting;
1208
1209        if ((positive == entry_interesting &&
1210             negative >= entry_interesting) || /* #5, #6, #16 */
1211            (positive == all_entries_interesting &&
1212             negative == entry_interesting)) /* #9 */
1213                return entry_not_interesting;
1214
1215        return all_entries_not_interesting; /* #10, #20 */
1216}