dir.con commit dir.c::match_pathname(): pay attention to the length of string parameters (ab3aebc)
   1/*
   2 * This handles recursive filename detection with exclude
   3 * files, index knowledge etc..
   4 *
   5 * Copyright (C) Linus Torvalds, 2005-2006
   6 *               Junio Hamano, 2005-2006
   7 */
   8#include "cache.h"
   9#include "dir.h"
  10#include "refs.h"
  11
  12struct path_simplify {
  13        int len;
  14        const char *path;
  15};
  16
  17static int read_directory_recursive(struct dir_struct *dir, const char *path, int len,
  18        int check_only, const struct path_simplify *simplify);
  19static int get_dtype(struct dirent *de, const char *path, int len);
  20
  21/* helper string functions with support for the ignore_case flag */
  22int strcmp_icase(const char *a, const char *b)
  23{
  24        return ignore_case ? strcasecmp(a, b) : strcmp(a, b);
  25}
  26
  27int strncmp_icase(const char *a, const char *b, size_t count)
  28{
  29        return ignore_case ? strncasecmp(a, b, count) : strncmp(a, b, count);
  30}
  31
  32int fnmatch_icase(const char *pattern, const char *string, int flags)
  33{
  34        return fnmatch(pattern, string, flags | (ignore_case ? FNM_CASEFOLD : 0));
  35}
  36
  37static int fnmatch_icase_mem(const char *pattern, int patternlen,
  38                             const char *string, int stringlen,
  39                             int flags)
  40{
  41        int match_status;
  42        struct strbuf pat_buf = STRBUF_INIT;
  43        struct strbuf str_buf = STRBUF_INIT;
  44        const char *use_pat = pattern;
  45        const char *use_str = string;
  46
  47        if (pattern[patternlen]) {
  48                strbuf_add(&pat_buf, pattern, patternlen);
  49                use_pat = pat_buf.buf;
  50        }
  51        if (string[stringlen]) {
  52                strbuf_add(&str_buf, string, stringlen);
  53                use_str = str_buf.buf;
  54        }
  55
  56        match_status = fnmatch_icase(use_pat, use_str, flags);
  57
  58        strbuf_release(&pat_buf);
  59        strbuf_release(&str_buf);
  60
  61        return match_status;
  62}
  63
  64static size_t common_prefix_len(const char **pathspec)
  65{
  66        const char *n, *first;
  67        size_t max = 0;
  68
  69        if (!pathspec)
  70                return max;
  71
  72        first = *pathspec;
  73        while ((n = *pathspec++)) {
  74                size_t i, len = 0;
  75                for (i = 0; first == n || i < max; i++) {
  76                        char c = n[i];
  77                        if (!c || c != first[i] || is_glob_special(c))
  78                                break;
  79                        if (c == '/')
  80                                len = i + 1;
  81                }
  82                if (first == n || len < max) {
  83                        max = len;
  84                        if (!max)
  85                                break;
  86                }
  87        }
  88        return max;
  89}
  90
  91/*
  92 * Returns a copy of the longest leading path common among all
  93 * pathspecs.
  94 */
  95char *common_prefix(const char **pathspec)
  96{
  97        unsigned long len = common_prefix_len(pathspec);
  98
  99        return len ? xmemdupz(*pathspec, len) : NULL;
 100}
 101
 102int fill_directory(struct dir_struct *dir, const char **pathspec)
 103{
 104        size_t len;
 105
 106        /*
 107         * Calculate common prefix for the pathspec, and
 108         * use that to optimize the directory walk
 109         */
 110        len = common_prefix_len(pathspec);
 111
 112        /* Read the directory and prune it */
 113        read_directory(dir, pathspec ? *pathspec : "", len, pathspec);
 114        return len;
 115}
 116
 117int within_depth(const char *name, int namelen,
 118                        int depth, int max_depth)
 119{
 120        const char *cp = name, *cpe = name + namelen;
 121
 122        while (cp < cpe) {
 123                if (*cp++ != '/')
 124                        continue;
 125                depth++;
 126                if (depth > max_depth)
 127                        return 0;
 128        }
 129        return 1;
 130}
 131
 132/*
 133 * Does 'match' match the given name?
 134 * A match is found if
 135 *
 136 * (1) the 'match' string is leading directory of 'name', or
 137 * (2) the 'match' string is a wildcard and matches 'name', or
 138 * (3) the 'match' string is exactly the same as 'name'.
 139 *
 140 * and the return value tells which case it was.
 141 *
 142 * It returns 0 when there is no match.
 143 */
 144static int match_one(const char *match, const char *name, int namelen)
 145{
 146        int matchlen;
 147
 148        /* If the match was just the prefix, we matched */
 149        if (!*match)
 150                return MATCHED_RECURSIVELY;
 151
 152        if (ignore_case) {
 153                for (;;) {
 154                        unsigned char c1 = tolower(*match);
 155                        unsigned char c2 = tolower(*name);
 156                        if (c1 == '\0' || is_glob_special(c1))
 157                                break;
 158                        if (c1 != c2)
 159                                return 0;
 160                        match++;
 161                        name++;
 162                        namelen--;
 163                }
 164        } else {
 165                for (;;) {
 166                        unsigned char c1 = *match;
 167                        unsigned char c2 = *name;
 168                        if (c1 == '\0' || is_glob_special(c1))
 169                                break;
 170                        if (c1 != c2)
 171                                return 0;
 172                        match++;
 173                        name++;
 174                        namelen--;
 175                }
 176        }
 177
 178
 179        /*
 180         * If we don't match the matchstring exactly,
 181         * we need to match by fnmatch
 182         */
 183        matchlen = strlen(match);
 184        if (strncmp_icase(match, name, matchlen))
 185                return !fnmatch_icase(match, name, 0) ? MATCHED_FNMATCH : 0;
 186
 187        if (namelen == matchlen)
 188                return MATCHED_EXACTLY;
 189        if (match[matchlen-1] == '/' || name[matchlen] == '/')
 190                return MATCHED_RECURSIVELY;
 191        return 0;
 192}
 193
 194/*
 195 * Given a name and a list of pathspecs, see if the name matches
 196 * any of the pathspecs.  The caller is also interested in seeing
 197 * all pathspec matches some names it calls this function with
 198 * (otherwise the user could have mistyped the unmatched pathspec),
 199 * and a mark is left in seen[] array for pathspec element that
 200 * actually matched anything.
 201 */
 202int match_pathspec(const char **pathspec, const char *name, int namelen,
 203                int prefix, char *seen)
 204{
 205        int i, retval = 0;
 206
 207        if (!pathspec)
 208                return 1;
 209
 210        name += prefix;
 211        namelen -= prefix;
 212
 213        for (i = 0; pathspec[i] != NULL; i++) {
 214                int how;
 215                const char *match = pathspec[i] + prefix;
 216                if (seen && seen[i] == MATCHED_EXACTLY)
 217                        continue;
 218                how = match_one(match, name, namelen);
 219                if (how) {
 220                        if (retval < how)
 221                                retval = how;
 222                        if (seen && seen[i] < how)
 223                                seen[i] = how;
 224                }
 225        }
 226        return retval;
 227}
 228
 229/*
 230 * Does 'match' match the given name?
 231 * A match is found if
 232 *
 233 * (1) the 'match' string is leading directory of 'name', or
 234 * (2) the 'match' string is a wildcard and matches 'name', or
 235 * (3) the 'match' string is exactly the same as 'name'.
 236 *
 237 * and the return value tells which case it was.
 238 *
 239 * It returns 0 when there is no match.
 240 */
 241static int match_pathspec_item(const struct pathspec_item *item, int prefix,
 242                               const char *name, int namelen)
 243{
 244        /* name/namelen has prefix cut off by caller */
 245        const char *match = item->match + prefix;
 246        int matchlen = item->len - prefix;
 247
 248        /* If the match was just the prefix, we matched */
 249        if (!*match)
 250                return MATCHED_RECURSIVELY;
 251
 252        if (matchlen <= namelen && !strncmp(match, name, matchlen)) {
 253                if (matchlen == namelen)
 254                        return MATCHED_EXACTLY;
 255
 256                if (match[matchlen-1] == '/' || name[matchlen] == '/')
 257                        return MATCHED_RECURSIVELY;
 258        }
 259
 260        if (item->use_wildcard && !fnmatch(match, name, 0))
 261                return MATCHED_FNMATCH;
 262
 263        return 0;
 264}
 265
 266/*
 267 * Given a name and a list of pathspecs, see if the name matches
 268 * any of the pathspecs.  The caller is also interested in seeing
 269 * all pathspec matches some names it calls this function with
 270 * (otherwise the user could have mistyped the unmatched pathspec),
 271 * and a mark is left in seen[] array for pathspec element that
 272 * actually matched anything.
 273 */
 274int match_pathspec_depth(const struct pathspec *ps,
 275                         const char *name, int namelen,
 276                         int prefix, char *seen)
 277{
 278        int i, retval = 0;
 279
 280        if (!ps->nr) {
 281                if (!ps->recursive || ps->max_depth == -1)
 282                        return MATCHED_RECURSIVELY;
 283
 284                if (within_depth(name, namelen, 0, ps->max_depth))
 285                        return MATCHED_EXACTLY;
 286                else
 287                        return 0;
 288        }
 289
 290        name += prefix;
 291        namelen -= prefix;
 292
 293        for (i = ps->nr - 1; i >= 0; i--) {
 294                int how;
 295                if (seen && seen[i] == MATCHED_EXACTLY)
 296                        continue;
 297                how = match_pathspec_item(ps->items+i, prefix, name, namelen);
 298                if (ps->recursive && ps->max_depth != -1 &&
 299                    how && how != MATCHED_FNMATCH) {
 300                        int len = ps->items[i].len;
 301                        if (name[len] == '/')
 302                                len++;
 303                        if (within_depth(name+len, namelen-len, 0, ps->max_depth))
 304                                how = MATCHED_EXACTLY;
 305                        else
 306                                how = 0;
 307                }
 308                if (how) {
 309                        if (retval < how)
 310                                retval = how;
 311                        if (seen && seen[i] < how)
 312                                seen[i] = how;
 313                }
 314        }
 315        return retval;
 316}
 317
 318/*
 319 * Return the length of the "simple" part of a path match limiter.
 320 */
 321static int simple_length(const char *match)
 322{
 323        int len = -1;
 324
 325        for (;;) {
 326                unsigned char c = *match++;
 327                len++;
 328                if (c == '\0' || is_glob_special(c))
 329                        return len;
 330        }
 331}
 332
 333static int no_wildcard(const char *string)
 334{
 335        return string[simple_length(string)] == '\0';
 336}
 337
 338void parse_exclude_pattern(const char **pattern,
 339                           int *patternlen,
 340                           int *flags,
 341                           int *nowildcardlen)
 342{
 343        const char *p = *pattern;
 344        size_t i, len;
 345
 346        *flags = 0;
 347        if (*p == '!') {
 348                *flags |= EXC_FLAG_NEGATIVE;
 349                p++;
 350        }
 351        len = strlen(p);
 352        if (len && p[len - 1] == '/') {
 353                len--;
 354                *flags |= EXC_FLAG_MUSTBEDIR;
 355        }
 356        for (i = 0; i < len; i++) {
 357                if (p[i] == '/')
 358                        break;
 359        }
 360        if (i == len)
 361                *flags |= EXC_FLAG_NODIR;
 362        *nowildcardlen = simple_length(p);
 363        /*
 364         * we should have excluded the trailing slash from 'p' too,
 365         * but that's one more allocation. Instead just make sure
 366         * nowildcardlen does not exceed real patternlen
 367         */
 368        if (*nowildcardlen > len)
 369                *nowildcardlen = len;
 370        if (*p == '*' && no_wildcard(p + 1))
 371                *flags |= EXC_FLAG_ENDSWITH;
 372        *pattern = p;
 373        *patternlen = len;
 374}
 375
 376void add_exclude(const char *string, const char *base,
 377                 int baselen, struct exclude_list *which)
 378{
 379        struct exclude *x;
 380        int patternlen;
 381        int flags;
 382        int nowildcardlen;
 383
 384        parse_exclude_pattern(&string, &patternlen, &flags, &nowildcardlen);
 385        if (flags & EXC_FLAG_MUSTBEDIR) {
 386                char *s;
 387                x = xmalloc(sizeof(*x) + patternlen + 1);
 388                s = (char *)(x+1);
 389                memcpy(s, string, patternlen);
 390                s[patternlen] = '\0';
 391                x->pattern = s;
 392        } else {
 393                x = xmalloc(sizeof(*x));
 394                x->pattern = string;
 395        }
 396        x->patternlen = patternlen;
 397        x->nowildcardlen = nowildcardlen;
 398        x->base = base;
 399        x->baselen = baselen;
 400        x->flags = flags;
 401        ALLOC_GROW(which->excludes, which->nr + 1, which->alloc);
 402        which->excludes[which->nr++] = x;
 403}
 404
 405static void *read_skip_worktree_file_from_index(const char *path, size_t *size)
 406{
 407        int pos, len;
 408        unsigned long sz;
 409        enum object_type type;
 410        void *data;
 411        struct index_state *istate = &the_index;
 412
 413        len = strlen(path);
 414        pos = index_name_pos(istate, path, len);
 415        if (pos < 0)
 416                return NULL;
 417        if (!ce_skip_worktree(istate->cache[pos]))
 418                return NULL;
 419        data = read_sha1_file(istate->cache[pos]->sha1, &type, &sz);
 420        if (!data || type != OBJ_BLOB) {
 421                free(data);
 422                return NULL;
 423        }
 424        *size = xsize_t(sz);
 425        return data;
 426}
 427
 428void free_excludes(struct exclude_list *el)
 429{
 430        int i;
 431
 432        for (i = 0; i < el->nr; i++)
 433                free(el->excludes[i]);
 434        free(el->excludes);
 435
 436        el->nr = 0;
 437        el->excludes = NULL;
 438}
 439
 440int add_excludes_from_file_to_list(const char *fname,
 441                                   const char *base,
 442                                   int baselen,
 443                                   char **buf_p,
 444                                   struct exclude_list *which,
 445                                   int check_index)
 446{
 447        struct stat st;
 448        int fd, i;
 449        size_t size = 0;
 450        char *buf, *entry;
 451
 452        fd = open(fname, O_RDONLY);
 453        if (fd < 0 || fstat(fd, &st) < 0) {
 454                if (errno != ENOENT)
 455                        warn_on_inaccessible(fname);
 456                if (0 <= fd)
 457                        close(fd);
 458                if (!check_index ||
 459                    (buf = read_skip_worktree_file_from_index(fname, &size)) == NULL)
 460                        return -1;
 461                if (size == 0) {
 462                        free(buf);
 463                        return 0;
 464                }
 465                if (buf[size-1] != '\n') {
 466                        buf = xrealloc(buf, size+1);
 467                        buf[size++] = '\n';
 468                }
 469        }
 470        else {
 471                size = xsize_t(st.st_size);
 472                if (size == 0) {
 473                        close(fd);
 474                        return 0;
 475                }
 476                buf = xmalloc(size+1);
 477                if (read_in_full(fd, buf, size) != size) {
 478                        free(buf);
 479                        close(fd);
 480                        return -1;
 481                }
 482                buf[size++] = '\n';
 483                close(fd);
 484        }
 485
 486        if (buf_p)
 487                *buf_p = buf;
 488        entry = buf;
 489        for (i = 0; i < size; i++) {
 490                if (buf[i] == '\n') {
 491                        if (entry != buf + i && entry[0] != '#') {
 492                                buf[i - (i && buf[i-1] == '\r')] = 0;
 493                                add_exclude(entry, base, baselen, which);
 494                        }
 495                        entry = buf + i + 1;
 496                }
 497        }
 498        return 0;
 499}
 500
 501void add_excludes_from_file(struct dir_struct *dir, const char *fname)
 502{
 503        if (add_excludes_from_file_to_list(fname, "", 0, NULL,
 504                                           &dir->exclude_list[EXC_FILE], 0) < 0)
 505                die("cannot use %s as an exclude file", fname);
 506}
 507
 508static void prep_exclude(struct dir_struct *dir, const char *base, int baselen)
 509{
 510        struct exclude_list *el;
 511        struct exclude_stack *stk = NULL;
 512        int current;
 513
 514        if ((!dir->exclude_per_dir) ||
 515            (baselen + strlen(dir->exclude_per_dir) >= PATH_MAX))
 516                return; /* too long a path -- ignore */
 517
 518        /* Pop the ones that are not the prefix of the path being checked. */
 519        el = &dir->exclude_list[EXC_DIRS];
 520        while ((stk = dir->exclude_stack) != NULL) {
 521                if (stk->baselen <= baselen &&
 522                    !strncmp(dir->basebuf, base, stk->baselen))
 523                        break;
 524                dir->exclude_stack = stk->prev;
 525                while (stk->exclude_ix < el->nr)
 526                        free(el->excludes[--el->nr]);
 527                free(stk->filebuf);
 528                free(stk);
 529        }
 530
 531        /* Read from the parent directories and push them down. */
 532        current = stk ? stk->baselen : -1;
 533        while (current < baselen) {
 534                struct exclude_stack *stk = xcalloc(1, sizeof(*stk));
 535                const char *cp;
 536
 537                if (current < 0) {
 538                        cp = base;
 539                        current = 0;
 540                }
 541                else {
 542                        cp = strchr(base + current + 1, '/');
 543                        if (!cp)
 544                                die("oops in prep_exclude");
 545                        cp++;
 546                }
 547                stk->prev = dir->exclude_stack;
 548                stk->baselen = cp - base;
 549                stk->exclude_ix = el->nr;
 550                memcpy(dir->basebuf + current, base + current,
 551                       stk->baselen - current);
 552                strcpy(dir->basebuf + stk->baselen, dir->exclude_per_dir);
 553                add_excludes_from_file_to_list(dir->basebuf,
 554                                               dir->basebuf, stk->baselen,
 555                                               &stk->filebuf, el, 1);
 556                dir->exclude_stack = stk;
 557                current = stk->baselen;
 558        }
 559        dir->basebuf[baselen] = '\0';
 560}
 561
 562int match_basename(const char *basename, int basenamelen,
 563                   const char *pattern, int prefix, int patternlen,
 564                   int flags)
 565{
 566        if (prefix == patternlen) {
 567                if (patternlen == basenamelen &&
 568                    !strncmp_icase(pattern, basename, basenamelen))
 569                        return 1;
 570        } else if (flags & EXC_FLAG_ENDSWITH) {
 571                /* "*literal" matching against "fooliteral" */
 572                if (patternlen - 1 <= basenamelen &&
 573                    !strncmp_icase(pattern + 1,
 574                                   basename + basenamelen - (patternlen - 1),
 575                                   patternlen - 1))
 576                        return 1;
 577        } else {
 578                if (fnmatch_icase_mem(pattern, patternlen,
 579                                      basename, basenamelen,
 580                                      0) == 0)
 581                        return 1;
 582        }
 583        return 0;
 584}
 585
 586int match_pathname(const char *pathname, int pathlen,
 587                   const char *base, int baselen,
 588                   const char *pattern, int prefix, int patternlen,
 589                   int flags)
 590{
 591        const char *name;
 592        int namelen;
 593
 594        /*
 595         * match with FNM_PATHNAME; the pattern has base implicitly
 596         * in front of it.
 597         */
 598        if (*pattern == '/') {
 599                pattern++;
 600                patternlen--;
 601                prefix--;
 602        }
 603
 604        /*
 605         * baselen does not count the trailing slash. base[] may or
 606         * may not end with a trailing slash though.
 607         */
 608        if (pathlen < baselen + 1 ||
 609            (baselen && pathname[baselen] != '/') ||
 610            strncmp_icase(pathname, base, baselen))
 611                return 0;
 612
 613        namelen = baselen ? pathlen - baselen - 1 : pathlen;
 614        name = pathname + pathlen - namelen;
 615
 616        if (prefix) {
 617                /*
 618                 * if the non-wildcard part is longer than the
 619                 * remaining pathname, surely it cannot match.
 620                 */
 621                if (prefix > namelen)
 622                        return 0;
 623
 624                if (strncmp_icase(pattern, name, prefix))
 625                        return 0;
 626                pattern += prefix;
 627                patternlen -= prefix;
 628                name    += prefix;
 629                namelen -= prefix;
 630
 631                /*
 632                 * If the whole pattern did not have a wildcard,
 633                 * then our prefix match is all we need; we
 634                 * do not need to call fnmatch at all.
 635                 */
 636                if (!patternlen && !namelen)
 637                        return 1;
 638        }
 639
 640        return fnmatch_icase_mem(pattern, patternlen,
 641                                 name, namelen,
 642                                 FNM_PATHNAME) == 0;
 643}
 644
 645/* Scan the list and let the last match determine the fate.
 646 * Return 1 for exclude, 0 for include and -1 for undecided.
 647 */
 648int excluded_from_list(const char *pathname,
 649                       int pathlen, const char *basename, int *dtype,
 650                       struct exclude_list *el)
 651{
 652        int i;
 653
 654        if (!el->nr)
 655                return -1;      /* undefined */
 656
 657        for (i = el->nr - 1; 0 <= i; i--) {
 658                struct exclude *x = el->excludes[i];
 659                const char *exclude = x->pattern;
 660                int to_exclude = x->flags & EXC_FLAG_NEGATIVE ? 0 : 1;
 661                int prefix = x->nowildcardlen;
 662
 663                if (x->flags & EXC_FLAG_MUSTBEDIR) {
 664                        if (*dtype == DT_UNKNOWN)
 665                                *dtype = get_dtype(NULL, pathname, pathlen);
 666                        if (*dtype != DT_DIR)
 667                                continue;
 668                }
 669
 670                if (x->flags & EXC_FLAG_NODIR) {
 671                        if (match_basename(basename,
 672                                           pathlen - (basename - pathname),
 673                                           exclude, prefix, x->patternlen,
 674                                           x->flags))
 675                                return to_exclude;
 676                        continue;
 677                }
 678
 679                assert(x->baselen == 0 || x->base[x->baselen - 1] == '/');
 680                if (match_pathname(pathname, pathlen,
 681                                   x->base, x->baselen ? x->baselen - 1 : 0,
 682                                   exclude, prefix, x->patternlen, x->flags))
 683                        return to_exclude;
 684        }
 685        return -1; /* undecided */
 686}
 687
 688static int excluded(struct dir_struct *dir, const char *pathname, int *dtype_p)
 689{
 690        int pathlen = strlen(pathname);
 691        int st;
 692        const char *basename = strrchr(pathname, '/');
 693        basename = (basename) ? basename+1 : pathname;
 694
 695        prep_exclude(dir, pathname, basename-pathname);
 696        for (st = EXC_CMDL; st <= EXC_FILE; st++) {
 697                switch (excluded_from_list(pathname, pathlen, basename,
 698                                           dtype_p, &dir->exclude_list[st])) {
 699                case 0:
 700                        return 0;
 701                case 1:
 702                        return 1;
 703                }
 704        }
 705        return 0;
 706}
 707
 708void path_exclude_check_init(struct path_exclude_check *check,
 709                             struct dir_struct *dir)
 710{
 711        check->dir = dir;
 712        strbuf_init(&check->path, 256);
 713}
 714
 715void path_exclude_check_clear(struct path_exclude_check *check)
 716{
 717        strbuf_release(&check->path);
 718}
 719
 720/*
 721 * Is this name excluded?  This is for a caller like show_files() that
 722 * do not honor directory hierarchy and iterate through paths that are
 723 * possibly in an ignored directory.
 724 *
 725 * A path to a directory known to be excluded is left in check->path to
 726 * optimize for repeated checks for files in the same excluded directory.
 727 */
 728int path_excluded(struct path_exclude_check *check,
 729                  const char *name, int namelen, int *dtype)
 730{
 731        int i;
 732        struct strbuf *path = &check->path;
 733
 734        /*
 735         * we allow the caller to pass namelen as an optimization; it
 736         * must match the length of the name, as we eventually call
 737         * excluded() on the whole name string.
 738         */
 739        if (namelen < 0)
 740                namelen = strlen(name);
 741
 742        if (path->len &&
 743            path->len <= namelen &&
 744            !memcmp(name, path->buf, path->len) &&
 745            (!name[path->len] || name[path->len] == '/'))
 746                return 1;
 747
 748        strbuf_setlen(path, 0);
 749        for (i = 0; name[i]; i++) {
 750                int ch = name[i];
 751
 752                if (ch == '/') {
 753                        int dt = DT_DIR;
 754                        if (excluded(check->dir, path->buf, &dt))
 755                                return 1;
 756                }
 757                strbuf_addch(path, ch);
 758        }
 759
 760        /* An entry in the index; cannot be a directory with subentries */
 761        strbuf_setlen(path, 0);
 762
 763        return excluded(check->dir, name, dtype);
 764}
 765
 766static struct dir_entry *dir_entry_new(const char *pathname, int len)
 767{
 768        struct dir_entry *ent;
 769
 770        ent = xmalloc(sizeof(*ent) + len + 1);
 771        ent->len = len;
 772        memcpy(ent->name, pathname, len);
 773        ent->name[len] = 0;
 774        return ent;
 775}
 776
 777static struct dir_entry *dir_add_name(struct dir_struct *dir, const char *pathname, int len)
 778{
 779        if (cache_name_exists(pathname, len, ignore_case))
 780                return NULL;
 781
 782        ALLOC_GROW(dir->entries, dir->nr+1, dir->alloc);
 783        return dir->entries[dir->nr++] = dir_entry_new(pathname, len);
 784}
 785
 786struct dir_entry *dir_add_ignored(struct dir_struct *dir, const char *pathname, int len)
 787{
 788        if (!cache_name_is_other(pathname, len))
 789                return NULL;
 790
 791        ALLOC_GROW(dir->ignored, dir->ignored_nr+1, dir->ignored_alloc);
 792        return dir->ignored[dir->ignored_nr++] = dir_entry_new(pathname, len);
 793}
 794
 795enum exist_status {
 796        index_nonexistent = 0,
 797        index_directory,
 798        index_gitdir
 799};
 800
 801/*
 802 * Do not use the alphabetically stored index to look up
 803 * the directory name; instead, use the case insensitive
 804 * name hash.
 805 */
 806static enum exist_status directory_exists_in_index_icase(const char *dirname, int len)
 807{
 808        struct cache_entry *ce = index_name_exists(&the_index, dirname, len + 1, ignore_case);
 809        unsigned char endchar;
 810
 811        if (!ce)
 812                return index_nonexistent;
 813        endchar = ce->name[len];
 814
 815        /*
 816         * The cache_entry structure returned will contain this dirname
 817         * and possibly additional path components.
 818         */
 819        if (endchar == '/')
 820                return index_directory;
 821
 822        /*
 823         * If there are no additional path components, then this cache_entry
 824         * represents a submodule.  Submodules, despite being directories,
 825         * are stored in the cache without a closing slash.
 826         */
 827        if (!endchar && S_ISGITLINK(ce->ce_mode))
 828                return index_gitdir;
 829
 830        /* This should never be hit, but it exists just in case. */
 831        return index_nonexistent;
 832}
 833
 834/*
 835 * The index sorts alphabetically by entry name, which
 836 * means that a gitlink sorts as '\0' at the end, while
 837 * a directory (which is defined not as an entry, but as
 838 * the files it contains) will sort with the '/' at the
 839 * end.
 840 */
 841static enum exist_status directory_exists_in_index(const char *dirname, int len)
 842{
 843        int pos;
 844
 845        if (ignore_case)
 846                return directory_exists_in_index_icase(dirname, len);
 847
 848        pos = cache_name_pos(dirname, len);
 849        if (pos < 0)
 850                pos = -pos-1;
 851        while (pos < active_nr) {
 852                struct cache_entry *ce = active_cache[pos++];
 853                unsigned char endchar;
 854
 855                if (strncmp(ce->name, dirname, len))
 856                        break;
 857                endchar = ce->name[len];
 858                if (endchar > '/')
 859                        break;
 860                if (endchar == '/')
 861                        return index_directory;
 862                if (!endchar && S_ISGITLINK(ce->ce_mode))
 863                        return index_gitdir;
 864        }
 865        return index_nonexistent;
 866}
 867
 868/*
 869 * When we find a directory when traversing the filesystem, we
 870 * have three distinct cases:
 871 *
 872 *  - ignore it
 873 *  - see it as a directory
 874 *  - recurse into it
 875 *
 876 * and which one we choose depends on a combination of existing
 877 * git index contents and the flags passed into the directory
 878 * traversal routine.
 879 *
 880 * Case 1: If we *already* have entries in the index under that
 881 * directory name, we always recurse into the directory to see
 882 * all the files.
 883 *
 884 * Case 2: If we *already* have that directory name as a gitlink,
 885 * we always continue to see it as a gitlink, regardless of whether
 886 * there is an actual git directory there or not (it might not
 887 * be checked out as a subproject!)
 888 *
 889 * Case 3: if we didn't have it in the index previously, we
 890 * have a few sub-cases:
 891 *
 892 *  (a) if "show_other_directories" is true, we show it as
 893 *      just a directory, unless "hide_empty_directories" is
 894 *      also true and the directory is empty, in which case
 895 *      we just ignore it entirely.
 896 *  (b) if it looks like a git directory, and we don't have
 897 *      'no_gitlinks' set we treat it as a gitlink, and show it
 898 *      as a directory.
 899 *  (c) otherwise, we recurse into it.
 900 */
 901enum directory_treatment {
 902        show_directory,
 903        ignore_directory,
 904        recurse_into_directory
 905};
 906
 907static enum directory_treatment treat_directory(struct dir_struct *dir,
 908        const char *dirname, int len,
 909        const struct path_simplify *simplify)
 910{
 911        /* The "len-1" is to strip the final '/' */
 912        switch (directory_exists_in_index(dirname, len-1)) {
 913        case index_directory:
 914                return recurse_into_directory;
 915
 916        case index_gitdir:
 917                if (dir->flags & DIR_SHOW_OTHER_DIRECTORIES)
 918                        return ignore_directory;
 919                return show_directory;
 920
 921        case index_nonexistent:
 922                if (dir->flags & DIR_SHOW_OTHER_DIRECTORIES)
 923                        break;
 924                if (!(dir->flags & DIR_NO_GITLINKS)) {
 925                        unsigned char sha1[20];
 926                        if (resolve_gitlink_ref(dirname, "HEAD", sha1) == 0)
 927                                return show_directory;
 928                }
 929                return recurse_into_directory;
 930        }
 931
 932        /* This is the "show_other_directories" case */
 933        if (!(dir->flags & DIR_HIDE_EMPTY_DIRECTORIES))
 934                return show_directory;
 935        if (!read_directory_recursive(dir, dirname, len, 1, simplify))
 936                return ignore_directory;
 937        return show_directory;
 938}
 939
 940/*
 941 * This is an inexact early pruning of any recursive directory
 942 * reading - if the path cannot possibly be in the pathspec,
 943 * return true, and we'll skip it early.
 944 */
 945static int simplify_away(const char *path, int pathlen, const struct path_simplify *simplify)
 946{
 947        if (simplify) {
 948                for (;;) {
 949                        const char *match = simplify->path;
 950                        int len = simplify->len;
 951
 952                        if (!match)
 953                                break;
 954                        if (len > pathlen)
 955                                len = pathlen;
 956                        if (!memcmp(path, match, len))
 957                                return 0;
 958                        simplify++;
 959                }
 960                return 1;
 961        }
 962        return 0;
 963}
 964
 965/*
 966 * This function tells us whether an excluded path matches a
 967 * list of "interesting" pathspecs. That is, whether a path matched
 968 * by any of the pathspecs could possibly be ignored by excluding
 969 * the specified path. This can happen if:
 970 *
 971 *   1. the path is mentioned explicitly in the pathspec
 972 *
 973 *   2. the path is a directory prefix of some element in the
 974 *      pathspec
 975 */
 976static int exclude_matches_pathspec(const char *path, int len,
 977                const struct path_simplify *simplify)
 978{
 979        if (simplify) {
 980                for (; simplify->path; simplify++) {
 981                        if (len == simplify->len
 982                            && !memcmp(path, simplify->path, len))
 983                                return 1;
 984                        if (len < simplify->len
 985                            && simplify->path[len] == '/'
 986                            && !memcmp(path, simplify->path, len))
 987                                return 1;
 988                }
 989        }
 990        return 0;
 991}
 992
 993static int get_index_dtype(const char *path, int len)
 994{
 995        int pos;
 996        struct cache_entry *ce;
 997
 998        ce = cache_name_exists(path, len, 0);
 999        if (ce) {
1000                if (!ce_uptodate(ce))
1001                        return DT_UNKNOWN;
1002                if (S_ISGITLINK(ce->ce_mode))
1003                        return DT_DIR;
1004                /*
1005                 * Nobody actually cares about the
1006                 * difference between DT_LNK and DT_REG
1007                 */
1008                return DT_REG;
1009        }
1010
1011        /* Try to look it up as a directory */
1012        pos = cache_name_pos(path, len);
1013        if (pos >= 0)
1014                return DT_UNKNOWN;
1015        pos = -pos-1;
1016        while (pos < active_nr) {
1017                ce = active_cache[pos++];
1018                if (strncmp(ce->name, path, len))
1019                        break;
1020                if (ce->name[len] > '/')
1021                        break;
1022                if (ce->name[len] < '/')
1023                        continue;
1024                if (!ce_uptodate(ce))
1025                        break;  /* continue? */
1026                return DT_DIR;
1027        }
1028        return DT_UNKNOWN;
1029}
1030
1031static int get_dtype(struct dirent *de, const char *path, int len)
1032{
1033        int dtype = de ? DTYPE(de) : DT_UNKNOWN;
1034        struct stat st;
1035
1036        if (dtype != DT_UNKNOWN)
1037                return dtype;
1038        dtype = get_index_dtype(path, len);
1039        if (dtype != DT_UNKNOWN)
1040                return dtype;
1041        if (lstat(path, &st))
1042                return dtype;
1043        if (S_ISREG(st.st_mode))
1044                return DT_REG;
1045        if (S_ISDIR(st.st_mode))
1046                return DT_DIR;
1047        if (S_ISLNK(st.st_mode))
1048                return DT_LNK;
1049        return dtype;
1050}
1051
1052enum path_treatment {
1053        path_ignored,
1054        path_handled,
1055        path_recurse
1056};
1057
1058static enum path_treatment treat_one_path(struct dir_struct *dir,
1059                                          struct strbuf *path,
1060                                          const struct path_simplify *simplify,
1061                                          int dtype, struct dirent *de)
1062{
1063        int exclude = excluded(dir, path->buf, &dtype);
1064        if (exclude && (dir->flags & DIR_COLLECT_IGNORED)
1065            && exclude_matches_pathspec(path->buf, path->len, simplify))
1066                dir_add_ignored(dir, path->buf, path->len);
1067
1068        /*
1069         * Excluded? If we don't explicitly want to show
1070         * ignored files, ignore it
1071         */
1072        if (exclude && !(dir->flags & DIR_SHOW_IGNORED))
1073                return path_ignored;
1074
1075        if (dtype == DT_UNKNOWN)
1076                dtype = get_dtype(de, path->buf, path->len);
1077
1078        /*
1079         * Do we want to see just the ignored files?
1080         * We still need to recurse into directories,
1081         * even if we don't ignore them, since the
1082         * directory may contain files that we do..
1083         */
1084        if (!exclude && (dir->flags & DIR_SHOW_IGNORED)) {
1085                if (dtype != DT_DIR)
1086                        return path_ignored;
1087        }
1088
1089        switch (dtype) {
1090        default:
1091                return path_ignored;
1092        case DT_DIR:
1093                strbuf_addch(path, '/');
1094                switch (treat_directory(dir, path->buf, path->len, simplify)) {
1095                case show_directory:
1096                        if (exclude != !!(dir->flags
1097                                          & DIR_SHOW_IGNORED))
1098                                return path_ignored;
1099                        break;
1100                case recurse_into_directory:
1101                        return path_recurse;
1102                case ignore_directory:
1103                        return path_ignored;
1104                }
1105                break;
1106        case DT_REG:
1107        case DT_LNK:
1108                break;
1109        }
1110        return path_handled;
1111}
1112
1113static enum path_treatment treat_path(struct dir_struct *dir,
1114                                      struct dirent *de,
1115                                      struct strbuf *path,
1116                                      int baselen,
1117                                      const struct path_simplify *simplify)
1118{
1119        int dtype;
1120
1121        if (is_dot_or_dotdot(de->d_name) || !strcmp(de->d_name, ".git"))
1122                return path_ignored;
1123        strbuf_setlen(path, baselen);
1124        strbuf_addstr(path, de->d_name);
1125        if (simplify_away(path->buf, path->len, simplify))
1126                return path_ignored;
1127
1128        dtype = DTYPE(de);
1129        return treat_one_path(dir, path, simplify, dtype, de);
1130}
1131
1132/*
1133 * Read a directory tree. We currently ignore anything but
1134 * directories, regular files and symlinks. That's because git
1135 * doesn't handle them at all yet. Maybe that will change some
1136 * day.
1137 *
1138 * Also, we ignore the name ".git" (even if it is not a directory).
1139 * That likely will not change.
1140 */
1141static int read_directory_recursive(struct dir_struct *dir,
1142                                    const char *base, int baselen,
1143                                    int check_only,
1144                                    const struct path_simplify *simplify)
1145{
1146        DIR *fdir;
1147        int contents = 0;
1148        struct dirent *de;
1149        struct strbuf path = STRBUF_INIT;
1150
1151        strbuf_add(&path, base, baselen);
1152
1153        fdir = opendir(path.len ? path.buf : ".");
1154        if (!fdir)
1155                goto out;
1156
1157        while ((de = readdir(fdir)) != NULL) {
1158                switch (treat_path(dir, de, &path, baselen, simplify)) {
1159                case path_recurse:
1160                        contents += read_directory_recursive(dir, path.buf,
1161                                                             path.len, 0,
1162                                                             simplify);
1163                        continue;
1164                case path_ignored:
1165                        continue;
1166                case path_handled:
1167                        break;
1168                }
1169                contents++;
1170                if (check_only)
1171                        break;
1172                dir_add_name(dir, path.buf, path.len);
1173        }
1174        closedir(fdir);
1175 out:
1176        strbuf_release(&path);
1177
1178        return contents;
1179}
1180
1181static int cmp_name(const void *p1, const void *p2)
1182{
1183        const struct dir_entry *e1 = *(const struct dir_entry **)p1;
1184        const struct dir_entry *e2 = *(const struct dir_entry **)p2;
1185
1186        return cache_name_compare(e1->name, e1->len,
1187                                  e2->name, e2->len);
1188}
1189
1190static struct path_simplify *create_simplify(const char **pathspec)
1191{
1192        int nr, alloc = 0;
1193        struct path_simplify *simplify = NULL;
1194
1195        if (!pathspec)
1196                return NULL;
1197
1198        for (nr = 0 ; ; nr++) {
1199                const char *match;
1200                if (nr >= alloc) {
1201                        alloc = alloc_nr(alloc);
1202                        simplify = xrealloc(simplify, alloc * sizeof(*simplify));
1203                }
1204                match = *pathspec++;
1205                if (!match)
1206                        break;
1207                simplify[nr].path = match;
1208                simplify[nr].len = simple_length(match);
1209        }
1210        simplify[nr].path = NULL;
1211        simplify[nr].len = 0;
1212        return simplify;
1213}
1214
1215static void free_simplify(struct path_simplify *simplify)
1216{
1217        free(simplify);
1218}
1219
1220static int treat_leading_path(struct dir_struct *dir,
1221                              const char *path, int len,
1222                              const struct path_simplify *simplify)
1223{
1224        struct strbuf sb = STRBUF_INIT;
1225        int baselen, rc = 0;
1226        const char *cp;
1227
1228        while (len && path[len - 1] == '/')
1229                len--;
1230        if (!len)
1231                return 1;
1232        baselen = 0;
1233        while (1) {
1234                cp = path + baselen + !!baselen;
1235                cp = memchr(cp, '/', path + len - cp);
1236                if (!cp)
1237                        baselen = len;
1238                else
1239                        baselen = cp - path;
1240                strbuf_setlen(&sb, 0);
1241                strbuf_add(&sb, path, baselen);
1242                if (!is_directory(sb.buf))
1243                        break;
1244                if (simplify_away(sb.buf, sb.len, simplify))
1245                        break;
1246                if (treat_one_path(dir, &sb, simplify,
1247                                   DT_DIR, NULL) == path_ignored)
1248                        break; /* do not recurse into it */
1249                if (len <= baselen) {
1250                        rc = 1;
1251                        break; /* finished checking */
1252                }
1253        }
1254        strbuf_release(&sb);
1255        return rc;
1256}
1257
1258int read_directory(struct dir_struct *dir, const char *path, int len, const char **pathspec)
1259{
1260        struct path_simplify *simplify;
1261
1262        if (has_symlink_leading_path(path, len))
1263                return dir->nr;
1264
1265        simplify = create_simplify(pathspec);
1266        if (!len || treat_leading_path(dir, path, len, simplify))
1267                read_directory_recursive(dir, path, len, 0, simplify);
1268        free_simplify(simplify);
1269        qsort(dir->entries, dir->nr, sizeof(struct dir_entry *), cmp_name);
1270        qsort(dir->ignored, dir->ignored_nr, sizeof(struct dir_entry *), cmp_name);
1271        return dir->nr;
1272}
1273
1274int file_exists(const char *f)
1275{
1276        struct stat sb;
1277        return lstat(f, &sb) == 0;
1278}
1279
1280/*
1281 * Given two normalized paths (a trailing slash is ok), if subdir is
1282 * outside dir, return -1.  Otherwise return the offset in subdir that
1283 * can be used as relative path to dir.
1284 */
1285int dir_inside_of(const char *subdir, const char *dir)
1286{
1287        int offset = 0;
1288
1289        assert(dir && subdir && *dir && *subdir);
1290
1291        while (*dir && *subdir && *dir == *subdir) {
1292                dir++;
1293                subdir++;
1294                offset++;
1295        }
1296
1297        /* hel[p]/me vs hel[l]/yeah */
1298        if (*dir && *subdir)
1299                return -1;
1300
1301        if (!*subdir)
1302                return !*dir ? offset : -1; /* same dir */
1303
1304        /* foo/[b]ar vs foo/[] */
1305        if (is_dir_sep(dir[-1]))
1306                return is_dir_sep(subdir[-1]) ? offset : -1;
1307
1308        /* foo[/]bar vs foo[] */
1309        return is_dir_sep(*subdir) ? offset + 1 : -1;
1310}
1311
1312int is_inside_dir(const char *dir)
1313{
1314        char cwd[PATH_MAX];
1315        if (!dir)
1316                return 0;
1317        if (!getcwd(cwd, sizeof(cwd)))
1318                die_errno("can't find the current directory");
1319        return dir_inside_of(cwd, dir) >= 0;
1320}
1321
1322int is_empty_dir(const char *path)
1323{
1324        DIR *dir = opendir(path);
1325        struct dirent *e;
1326        int ret = 1;
1327
1328        if (!dir)
1329                return 0;
1330
1331        while ((e = readdir(dir)) != NULL)
1332                if (!is_dot_or_dotdot(e->d_name)) {
1333                        ret = 0;
1334                        break;
1335                }
1336
1337        closedir(dir);
1338        return ret;
1339}
1340
1341static int remove_dir_recurse(struct strbuf *path, int flag, int *kept_up)
1342{
1343        DIR *dir;
1344        struct dirent *e;
1345        int ret = 0, original_len = path->len, len, kept_down = 0;
1346        int only_empty = (flag & REMOVE_DIR_EMPTY_ONLY);
1347        int keep_toplevel = (flag & REMOVE_DIR_KEEP_TOPLEVEL);
1348        unsigned char submodule_head[20];
1349
1350        if ((flag & REMOVE_DIR_KEEP_NESTED_GIT) &&
1351            !resolve_gitlink_ref(path->buf, "HEAD", submodule_head)) {
1352                /* Do not descend and nuke a nested git work tree. */
1353                if (kept_up)
1354                        *kept_up = 1;
1355                return 0;
1356        }
1357
1358        flag &= ~REMOVE_DIR_KEEP_TOPLEVEL;
1359        dir = opendir(path->buf);
1360        if (!dir) {
1361                /* an empty dir could be removed even if it is unreadble */
1362                if (!keep_toplevel)
1363                        return rmdir(path->buf);
1364                else
1365                        return -1;
1366        }
1367        if (path->buf[original_len - 1] != '/')
1368                strbuf_addch(path, '/');
1369
1370        len = path->len;
1371        while ((e = readdir(dir)) != NULL) {
1372                struct stat st;
1373                if (is_dot_or_dotdot(e->d_name))
1374                        continue;
1375
1376                strbuf_setlen(path, len);
1377                strbuf_addstr(path, e->d_name);
1378                if (lstat(path->buf, &st))
1379                        ; /* fall thru */
1380                else if (S_ISDIR(st.st_mode)) {
1381                        if (!remove_dir_recurse(path, flag, &kept_down))
1382                                continue; /* happy */
1383                } else if (!only_empty && !unlink(path->buf))
1384                        continue; /* happy, too */
1385
1386                /* path too long, stat fails, or non-directory still exists */
1387                ret = -1;
1388                break;
1389        }
1390        closedir(dir);
1391
1392        strbuf_setlen(path, original_len);
1393        if (!ret && !keep_toplevel && !kept_down)
1394                ret = rmdir(path->buf);
1395        else if (kept_up)
1396                /*
1397                 * report the uplevel that it is not an error that we
1398                 * did not rmdir() our directory.
1399                 */
1400                *kept_up = !ret;
1401        return ret;
1402}
1403
1404int remove_dir_recursively(struct strbuf *path, int flag)
1405{
1406        return remove_dir_recurse(path, flag, NULL);
1407}
1408
1409void setup_standard_excludes(struct dir_struct *dir)
1410{
1411        const char *path;
1412        char *xdg_path;
1413
1414        dir->exclude_per_dir = ".gitignore";
1415        path = git_path("info/exclude");
1416        if (!excludes_file) {
1417                home_config_paths(NULL, &xdg_path, "ignore");
1418                excludes_file = xdg_path;
1419        }
1420        if (!access_or_warn(path, R_OK))
1421                add_excludes_from_file(dir, path);
1422        if (excludes_file && !access_or_warn(excludes_file, R_OK))
1423                add_excludes_from_file(dir, excludes_file);
1424}
1425
1426int remove_path(const char *name)
1427{
1428        char *slash;
1429
1430        if (unlink(name) && errno != ENOENT)
1431                return -1;
1432
1433        slash = strrchr(name, '/');
1434        if (slash) {
1435                char *dirs = xstrdup(name);
1436                slash = dirs + (slash - name);
1437                do {
1438                        *slash = '\0';
1439                } while (rmdir(dirs) == 0 && (slash = strrchr(dirs, '/')));
1440                free(dirs);
1441        }
1442        return 0;
1443}
1444
1445static int pathspec_item_cmp(const void *a_, const void *b_)
1446{
1447        struct pathspec_item *a, *b;
1448
1449        a = (struct pathspec_item *)a_;
1450        b = (struct pathspec_item *)b_;
1451        return strcmp(a->match, b->match);
1452}
1453
1454int init_pathspec(struct pathspec *pathspec, const char **paths)
1455{
1456        const char **p = paths;
1457        int i;
1458
1459        memset(pathspec, 0, sizeof(*pathspec));
1460        if (!p)
1461                return 0;
1462        while (*p)
1463                p++;
1464        pathspec->raw = paths;
1465        pathspec->nr = p - paths;
1466        if (!pathspec->nr)
1467                return 0;
1468
1469        pathspec->items = xmalloc(sizeof(struct pathspec_item)*pathspec->nr);
1470        for (i = 0; i < pathspec->nr; i++) {
1471                struct pathspec_item *item = pathspec->items+i;
1472                const char *path = paths[i];
1473
1474                item->match = path;
1475                item->len = strlen(path);
1476                item->use_wildcard = !no_wildcard(path);
1477                if (item->use_wildcard)
1478                        pathspec->has_wildcard = 1;
1479        }
1480
1481        qsort(pathspec->items, pathspec->nr,
1482              sizeof(struct pathspec_item), pathspec_item_cmp);
1483
1484        return 0;
1485}
1486
1487void free_pathspec(struct pathspec *pathspec)
1488{
1489        free(pathspec->items);
1490        pathspec->items = NULL;
1491}