lib/libc/gen/glob.c

   1 /*
   2  * Copyright (c) 1989, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * This code is derived from software contributed to Berkeley by
   6  * Guido van Rossum.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  * 3. Neither the name of the University nor the names of its contributors
  17  *    may be used to endorse or promote products derived from this software
  18  *    without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  30  * SUCH DAMAGE.
  31  *
  32  * @(#)glob.c   8.3 (Berkeley) 10/13/93
  33  * $FreeBSD: src/lib/libc/gen/glob.c,v 1.27 2008/06/26 07:12:35 mtm Exp $
  34  * $DragonFly: src/lib/libc/gen/glob.c,v 1.6 2005/12/07 02:28:15 corecode Exp $
  35  */
  36
  37 /*
  38  * glob(3) -- a superset of the one defined in POSIX 1003.2.
  39  *
  40  * The [!...] convention to negate a range is supported (SysV, Posix, ksh).
  41  *
  42  * Optional extra services, controlled by flags not defined by POSIX:
  43  *
  44  * GLOB_QUOTE:
  45  *      Escaping convention: \ inhibits any special meaning the following
  46  *      character might have (except \ at end of string is retained).
  47  * GLOB_MAGCHAR:
  48  *      Set in gl_flags if pattern contained a globbing character.
  49  * GLOB_NOMAGIC:
  50  *      Same as GLOB_NOCHECK, but it will only append pattern if it did
  51  *      not contain any magic characters.  [Used in csh style globbing]
  52  * GLOB_ALTDIRFUNC:
  53  *      Use alternately specified directory access functions.
  54  * GLOB_TILDE:
  55  *      expand ~user/foo to the /home/dir/of/user/foo
  56  * GLOB_BRACE:
  57  *      expand {1,2}{a,b} to 1a 1b 2a 2b
  58  * gl_matchc:
  59  *      Number of matches in the current invocation of glob.
  60  */
  61
  62 /*
  63  * Some notes on multibyte character support:
  64  * 1. Patterns with illegal byte sequences match nothing - even if
  65  *    GLOB_NOCHECK is specified.
  66  * 2. Illegal byte sequences in filenames are handled by treating them as
  67  *    single-byte characters with a value of the first byte of the sequence
  68  *    cast to wchar_t.
  69  * 3. State-dependent encodings are not currently supported.
  70  */
  71
  72 #include <sys/param.h>
  73 #include <sys/stat.h>
  74
  75 #include <ctype.h>
  76 #include <dirent.h>
  77 #include <errno.h>
  78 #include <glob.h>
  79 #include <limits.h>
  80 #include <pwd.h>
  81 #include <stdint.h>
  82 #include <stdio.h>
  83 #include <stdlib.h>
  84 #include <string.h>
  85 #include <unistd.h>
  86 #include <wchar.h>
  87
  88 #include "collate.h"
  89
  90 #define DOLLAR          '$'
  91 #define DOT             '.'
  92 #define EOS             '\0'
  93 #define LBRACKET        '['
  94 #define NOT             '!'
  95 #define QUESTION        '?'
  96 #define QUOTE           '\\'
  97 #define RANGE           '-'
  98 #define RBRACKET        ']'
  99 #define SEP             '/'
 100 #define STAR            '*'
 101 #define TILDE           '~'
 102 #define UNDERSCORE      '_'
 103 #define LBRACE          '{'
 104 #define RBRACE          '}'
 105 #define SLASH           '/'
 106 #define COMMA           ','
 107
 108 #ifndef DEBUG
 109
 110 #define M_QUOTE         0x8000000000ULL
 111 #define M_PROTECT       0x4000000000ULL
 112 #define M_MASK          0xffffffffffULL
 113 #define M_CHAR          0x00ffffffffULL
 114
 115 typedef uint_fast64_t Char;
 116
 117 #else
 118
 119 #define M_QUOTE         0x80
 120 #define M_PROTECT       0x40
 121 #define M_MASK          0xff
 122 #define M_CHAR          0x7f
 123
 124 typedef char Char;
 125
 126 #endif
 127
 128
 129 #define CHAR(c)         ((Char)((c)&M_CHAR))
 130 #define META(c)         ((Char)((c)|M_QUOTE))
 131 #define M_ALL           META('*')
 132 #define M_END           META(']')
 133 #define M_NOT           META('!')
 134 #define M_ONE           META('?')
 135 #define M_RNG           META('-')
 136 #define M_SET           META('[')
 137 #define ismeta(c)       (((c)&M_QUOTE) != 0)
 138
 139
 140 #define GLOB_LIMIT_MALLOC       65536
 141 #define GLOB_LIMIT_STAT         128
 142 #define GLOB_LIMIT_READDIR      16384
 143
 144 #define GLOB_INDEX_MALLOC       0
 145 #define GLOB_INDEX_STAT         1
 146 #define GLOB_INDEX_READDIR      2
 147
 148 static int       compare(const void *, const void *);
 149 static int       g_Ctoc(const Char *, char *, size_t);
 150 static int       g_lstat(const Char *, struct stat *, glob_t *);
 151 static DIR      *g_opendir(const Char *, glob_t *);
 152 static const Char *g_strchr(const Char *, wchar_t);
 153 #ifdef notdef
 154 static Char     *g_strcat(Char *, const Char *);
 155 #endif
 156 static int       g_stat(const Char *, struct stat *, glob_t *);
 157 static int       glob0(const Char *, glob_t *, size_t *);
 158 static int       glob1(const Char *, glob_t *, size_t *);
 159 static int       glob2(Char *, Char *, Char *, const Char *, glob_t *,
 160                        size_t *);
 161 static int       glob3(Char *, Char *, Char *, const Char *, const Char *,
 162                        glob_t *, size_t *);
 163 static int       globextend(const Char *, glob_t *, size_t *);
 164 static const Char *
 165                  globtilde(const Char *, Char *, size_t, glob_t *);
 166 static int       globexp1(const Char *, glob_t *, size_t *);
 167 static int       globexp2(const Char *, const Char *, glob_t *, int *, size_t *);
 168 static int       match(const Char *, const Char *, const Char *);
 169 #ifdef DEBUG
 170 static void      qprintf(const char *, const Char *);
 171 #endif
 172
 173 /* 0 = malloc(), 1 = stat(), 2 = readdir() */
 174 static size_t limits[] = { 0, 0, 0 };
 175
 176 int
 177 glob(const char *pattern, int flags, int (*errfunc)(const char *, int),
 178      glob_t *pglob)
 179 {
 180         const char *patnext;
 181         size_t limit;
 182         Char *bufnext, *bufend, patbuf[MAXPATHLEN], prot;
 183         mbstate_t mbs;
 184         wchar_t wc;
 185         size_t clen;
 186
 187         patnext = pattern;
 188         if (!(flags & GLOB_APPEND)) {
 189                 pglob->gl_pathc = 0;
 190                 pglob->gl_pathv = NULL;
 191                 if (!(flags & GLOB_DOOFFS))
 192                         pglob->gl_offs = 0;
 193         }
 194         if (flags & GLOB_LIMIT) {
 195                 limit = pglob->gl_matchc;
 196                 if (limit == 0)
 197                         limit = ARG_MAX;
 198         } else
 199                 limit = 0;
 200         pglob->gl_flags = flags & ~GLOB_MAGCHAR;
 201         pglob->gl_errfunc = errfunc;
 202         pglob->gl_matchc = 0;
 203
 204         bufnext = patbuf;
 205         bufend = bufnext + MAXPATHLEN - 1;
 206         if (flags & GLOB_NOESCAPE) {
 207                 memset(&mbs, 0, sizeof(mbs));
 208                 while (bufend - bufnext >= MB_CUR_MAX) {
 209                         clen = mbrtowc(&wc, patnext, MB_LEN_MAX, &mbs);
 210                         if (clen == (size_t)-1 || clen == (size_t)-2)
 211                                 return (GLOB_NOMATCH);
 212                         else if (clen == 0)
 213                                 break;
 214                         *bufnext++ = wc;
 215                         patnext += clen;
 216                 }
 217         } else {
 218                 /* Protect the quoted characters. */
 219                 memset(&mbs, 0, sizeof(mbs));
 220                 while (bufend - bufnext >= MB_CUR_MAX) {
 221                         if (*patnext == QUOTE) {
 222                                 if (*++patnext == EOS) {
 223                                         *bufnext++ = QUOTE | M_PROTECT;
 224                                         continue;
 225                                 }
 226                                 prot = M_PROTECT;
 227                         } else
 228                                 prot = 0;
 229                         clen = mbrtowc(&wc, patnext, MB_LEN_MAX, &mbs);
 230                         if (clen == (size_t)-1 || clen == (size_t)-2)
 231                                 return (GLOB_NOMATCH);
 232                         else if (clen == 0)
 233                                 break;
 234                         *bufnext++ = wc | prot;
 235                         patnext += clen;
 236                 }
 237         }
 238         *bufnext = EOS;
 239
 240         if (flags & GLOB_BRACE)
 241             return globexp1(patbuf, pglob, &limit);
 242         else
 243             return glob0(patbuf, pglob, &limit);
 244 }
 245
 246 /*
 247  * Expand recursively a glob {} pattern. When there is no more expansion
 248  * invoke the standard globbing routine to glob the rest of the magic
 249  * characters
 250  */
 251 static int
 252 globexp1(const Char *pattern, glob_t *pglob, size_t *limit)
 253 {
 254         const Char *ptr = pattern;
 255         int rv;
 256
 257         /* Protect a single {}, for find(1), like csh */
 258         if (pattern[0] == LBRACE && pattern[1] == RBRACE && pattern[2] == EOS)
 259                 return glob0(pattern, pglob, limit);
 260
 261         while ((ptr = g_strchr(ptr, LBRACE)) != NULL)
 262                 if (!globexp2(ptr, pattern, pglob, &rv, limit))
 263                         return rv;
 264
 265         return glob0(pattern, pglob, limit);
 266 }
 267
 268
 269 /*
 270  * Recursive brace globbing helper. Tries to expand a single brace.
 271  * If it succeeds then it invokes globexp1 with the new pattern.
 272  * If it fails then it tries to glob the rest of the pattern and returns.
 273  */
 274 static int
 275 globexp2(const Char *ptr, const Char *pattern, glob_t *pglob, int *rv,
 276          size_t *limit)
 277 {
 278         int     i;
 279         Char   *lm, *ls;
 280         const Char *pe, *pm, *pm1, *pl;
 281         Char    patbuf[MAXPATHLEN];
 282
 283         /* copy part up to the brace */
 284         for (lm = patbuf, pm = pattern; pm != ptr; *lm++ = *pm++)
 285                 continue;
 286         *lm = EOS;
 287         ls = lm;
 288
 289         /* Find the balanced brace */
 290         for (i = 0, pe = ++ptr; *pe; pe++)
 291                 if (*pe == LBRACKET) {
 292                         /* Ignore everything between [] */
 293                         for (pm = pe++; *pe != RBRACKET && *pe != EOS; pe++)
 294                                 continue;
 295                         if (*pe == EOS) {
 296                                 /*
 297                                  * We could not find a matching RBRACKET.
 298                                  * Ignore and just look for RBRACE
 299                                  */
 300                                 pe = pm;
 301                         }
 302                 }
 303                 else if (*pe == LBRACE)
 304                         i++;
 305                 else if (*pe == RBRACE) {
 306                         if (i == 0)
 307                                 break;
 308                         i--;
 309                 }
 310
 311         /* Non matching braces; just glob the pattern */
 312         if (i != 0 || *pe == EOS) {
 313                 *rv = glob0(patbuf, pglob, limit);
 314                 return 0;
 315         }
 316
 317         for (i = 0, pl = pm = ptr; pm <= pe; pm++)
 318                 switch (*pm) {
 319                 case LBRACKET:
 320                         /* Ignore everything between [] */
 321                         for (pm1 = pm++; *pm != RBRACKET && *pm != EOS; pm++)
 322                                 continue;
 323                         if (*pm == EOS) {
 324                                 /*
 325                                  * We could not find a matching RBRACKET.
 326                                  * Ignore and just look for RBRACE
 327                                  */
 328                                 pm = pm1;
 329                         }
 330                         break;
 331
 332                 case LBRACE:
 333                         i++;
 334                         break;
 335
 336                 case RBRACE:
 337                         if (i) {
 338                             i--;
 339                             break;
 340                         }
 341                         /* FALLTHROUGH */
 342                 case COMMA:
 343                         if (i && *pm == COMMA)
 344                                 break;
 345                         else {
 346                                 /* Append the current string */
 347                                 for (lm = ls; (pl < pm); *lm++ = *pl++)
 348                                         continue;
 349                                 /*
 350                                  * Append the rest of the pattern after the
 351                                  * closing brace
 352                                  */
 353                                 for (pl = pe + 1; (*lm++ = *pl++) != EOS;)
 354                                         continue;
 355
 356                                 /* Expand the current pattern */
 357 #ifdef DEBUG
 358                                 qprintf("globexp2:", patbuf);
 359 #endif
 360                                 *rv = globexp1(patbuf, pglob, limit);
 361
 362                                 /* move after the comma, to the next string */
 363                                 pl = pm + 1;
 364                         }
 365                         break;
 366
 367                 default:
 368                         break;
 369                 }
 370         *rv = 0;
 371         return 0;
 372 }
 373
 374
 375
 376 /*
 377  * expand tilde from the passwd file.
 378  */
 379 static const Char *
 380 globtilde(const Char *pattern, Char *patbuf, size_t patbuf_len, glob_t *pglob)
 381 {
 382         struct passwd *pwd;
 383         char *h;
 384         const Char *p;
 385         Char *b, *eb;
 386
 387         if (*pattern != TILDE || !(pglob->gl_flags & GLOB_TILDE))
 388                 return pattern;
 389
 390         /*
 391          * Copy up to the end of the string or /
 392          */
 393         eb = &patbuf[patbuf_len - 1];
 394         for (p = pattern + 1, h = (char *) patbuf;
 395             h < (char *)eb && *p && *p != SLASH; *h++ = *p++)
 396                 continue;
 397
 398         *h = EOS;
 399
 400         if (((char *) patbuf)[0] == EOS) {
 401                 /*
 402                  * handle a plain ~ or ~/ by expanding $HOME first (iff
 403                  * we're not running setuid or setgid) and then trying
 404                  * the password file
 405                  */
 406                 if (issetugid() != 0 ||
 407                     (h = getenv("HOME")) == NULL) {
 408                         if (((h = getlogin()) != NULL &&
 409                              (pwd = getpwnam(h)) != NULL) ||
 410                             (pwd = getpwuid(getuid())) != NULL)
 411                                 h = pwd->pw_dir;
 412                         else
 413                                 return pattern;
 414                 }
 415         }
 416         else {
 417                 /*
 418                  * Expand a ~user
 419                  */
 420                 if ((pwd = getpwnam((char*) patbuf)) == NULL)
 421                         return pattern;
 422                 else
 423                         h = pwd->pw_dir;
 424         }
 425
 426         /* Copy the home directory */
 427         for (b = patbuf; b < eb && *h; *b++ = *h++)
 428                 continue;
 429
 430         /* Append the rest of the pattern */
 431         while (b < eb && (*b++ = *p++) != EOS)
 432                 continue;
 433         *b = EOS;
 434
 435         return patbuf;
 436 }
 437
 438
 439 /*
 440  * The main glob() routine: compiles the pattern (optionally processing
 441  * quotes), calls glob1() to do the real pattern matching, and finally
 442  * sorts the list (unless unsorted operation is requested).  Returns 0
 443  * if things went well, nonzero if errors occurred.
 444  */
 445 static int
 446 glob0(const Char *pattern, glob_t *pglob, size_t *limit)
 447 {
 448         const Char *qpatnext;
 449         int c, err;
 450         size_t oldpathc;
 451         Char *bufnext, patbuf[MAXPATHLEN];
 452
 453         qpatnext = globtilde(pattern, patbuf, MAXPATHLEN, pglob);
 454         oldpathc = pglob->gl_pathc;
 455         bufnext = patbuf;
 456
 457         /* We don't need to check for buffer overflow any more. */
 458         while ((c = *qpatnext++) != EOS) {
 459                 switch (c) {
 460                 case LBRACKET:
 461                         c = *qpatnext;
 462                         if (c == NOT)
 463                                 ++qpatnext;
 464                         if (*qpatnext == EOS ||
 465                             g_strchr(qpatnext+1, RBRACKET) == NULL) {
 466                                 *bufnext++ = LBRACKET;
 467                                 if (c == NOT)
 468                                         --qpatnext;
 469                                 break;
 470                         }
 471                         *bufnext++ = M_SET;
 472                         if (c == NOT)
 473                                 *bufnext++ = M_NOT;
 474                         c = *qpatnext++;
 475                         do {
 476                                 *bufnext++ = CHAR(c);
 477                                 if (*qpatnext == RANGE &&
 478                                     (c = qpatnext[1]) != RBRACKET) {
 479                                         *bufnext++ = M_RNG;
 480                                         *bufnext++ = CHAR(c);
 481                                         qpatnext += 2;
 482                                 }
 483                         } while ((c = *qpatnext++) != RBRACKET);
 484                         pglob->gl_flags |= GLOB_MAGCHAR;
 485                         *bufnext++ = M_END;
 486                         break;
 487                 case QUESTION:
 488                         pglob->gl_flags |= GLOB_MAGCHAR;
 489                         *bufnext++ = M_ONE;
 490                         break;
 491                 case STAR:
 492                         pglob->gl_flags |= GLOB_MAGCHAR;
 493                         /* collapse adjacent stars to one,
 494                          * to avoid exponential behavior
 495                          */
 496                         if (bufnext == patbuf || bufnext[-1] != M_ALL)
 497                             *bufnext++ = M_ALL;
 498                         break;
 499                 default:
 500                         *bufnext++ = CHAR(c);
 501                         break;
 502                 }
 503         }
 504         *bufnext = EOS;
 505 #ifdef DEBUG
 506         qprintf("glob0:", patbuf);
 507 #endif
 508
 509         if ((err = glob1(patbuf, pglob, limit)) != 0)
 510                 return(err);
 511
 512         /*
 513          * If there was no match we are going to append the pattern
 514          * if GLOB_NOCHECK was specified or if GLOB_NOMAGIC was specified
 515          * and the pattern did not contain any magic characters
 516          * GLOB_NOMAGIC is there just for compatibility with csh.
 517          */
 518         if (pglob->gl_pathc == oldpathc) {
 519                 if (((pglob->gl_flags & GLOB_NOCHECK) ||
 520                     ((pglob->gl_flags & GLOB_NOMAGIC) &&
 521                         !(pglob->gl_flags & GLOB_MAGCHAR))))
 522                         return(globextend(pattern, pglob, limit));
 523                 else
 524                         return(GLOB_NOMATCH);
 525         }
 526         if (!(pglob->gl_flags & GLOB_NOSORT))
 527                 qsort(pglob->gl_pathv + pglob->gl_offs + oldpathc,
 528                     pglob->gl_pathc - oldpathc, sizeof(char *), compare);
 529         return(0);
 530 }
 531
 532 static int
 533 compare(const void *p, const void *q)
 534 {
 535         return(strcmp(*(const char * const *)p, *(const char * const *)q));
 536 }
 537
 538 static int
 539 glob1(const Char *pattern, glob_t *pglob, size_t *limit)
 540 {
 541         Char pathbuf[MAXPATHLEN];
 542
 543         /* A null pathname is invalid -- POSIX 1003.1 sect. 2.4. */
 544         if (*pattern == EOS)
 545                 return(0);
 546         return(glob2(pathbuf, pathbuf, pathbuf + MAXPATHLEN - 1,
 547             pattern, pglob, limit));
 548 }
 549
 550 /*
 551  * The functions glob2 and glob3 are mutually recursive; there is one level
 552  * of recursion for each segment in the pattern that contains one or more
 553  * meta characters.
 554  */
 555 static int
 556 glob2(Char *pathbuf, Char *pathend, Char *pathend_last, const Char *pattern,
 557       glob_t *pglob, size_t *limit)
 558 {
 559         struct stat sb;
 560         const Char *p;
 561         Char *q;
 562         int anymeta;
 563
 564         /*
 565          * Loop over pattern segments until end of pattern or until
 566          * segment with meta character found.
 567          */
 568         for (anymeta = 0;;) {
 569                 if (*pattern == EOS) {          /* End of pattern? */
 570                         *pathend = EOS;
 571                         if (g_lstat(pathbuf, &sb, pglob))
 572                                 return(0);
 573
 574                         if ((pglob->gl_flags & GLOB_LIMIT) &&
 575                             limits[GLOB_INDEX_STAT]++ >= GLOB_LIMIT_STAT) {
 576                                 errno = 0;
 577                                 *pathend++ = SEP;
 578                                 *pathend = EOS;
 579                                 return GLOB_NOSPACE;
 580                         }
 581
 582                         if (((pglob->gl_flags & GLOB_MARK) &&
 583                             pathend[-1] != SEP) && (S_ISDIR(sb.st_mode)
 584                             || (S_ISLNK(sb.st_mode) &&
 585                             (g_stat(pathbuf, &sb, pglob) == 0) &&
 586                             S_ISDIR(sb.st_mode)))) {
 587                                 if (pathend + 1 > pathend_last)
 588                                         return (GLOB_ABORTED);
 589                                 *pathend++ = SEP;
 590                                 *pathend = EOS;
 591                         }
 592                         ++pglob->gl_matchc;
 593                         return(globextend(pathbuf, pglob, limit));
 594                 }
 595
 596                 /* Find end of next segment, copy tentatively to pathend. */
 597                 q = pathend;
 598                 p = pattern;
 599                 while (*p != EOS && *p != SEP) {
 600                         if (ismeta(*p))
 601                                 anymeta = 1;
 602                         if (q + 1 > pathend_last)
 603                                 return (GLOB_ABORTED);
 604                         *q++ = *p++;
 605                 }
 606
 607                 if (!anymeta) {         /* No expansion, do next segment. */
 608                         pathend = q;
 609                         pattern = p;
 610                         while (*pattern == SEP) {
 611                                 if (pathend + 1 > pathend_last)
 612                                         return (GLOB_ABORTED);
 613                                 *pathend++ = *pattern++;
 614                         }
 615                 } else                  /* Need expansion, recurse. */
 616                         return(glob3(pathbuf, pathend, pathend_last, pattern, p,
 617                             pglob, limit));
 618         }
 619         /* NOTREACHED */
 620 }
 621
 622 static int
 623 glob3(Char *pathbuf, Char *pathend, Char *pathend_last, const Char *pattern,
 624       const Char *restpattern, glob_t *pglob, size_t *limit)
 625 {
 626         struct dirent *dp;
 627         DIR *dirp;
 628         int err;
 629         char buf[MAXPATHLEN];
 630         struct dirent *(*readdirfunc)(DIR *);
 631
 632         if (pathend > pathend_last)
 633                 return (GLOB_ABORTED);
 634         *pathend = EOS;
 635         errno = 0;
 636
 637         if ((dirp = g_opendir(pathbuf, pglob)) == NULL) {
 638                 /* TODO: don't call for ENOENT or ENOTDIR? */
 639                 if (pglob->gl_errfunc) {
 640                         if (g_Ctoc(pathbuf, buf, sizeof(buf)))
 641                                 return (GLOB_ABORTED);
 642                         if (pglob->gl_errfunc(buf, errno) ||
 643                             pglob->gl_flags & GLOB_ERR)
 644                                 return (GLOB_ABORTED);
 645                 }
 646                 return(0);
 647         }
 648
 649         err = 0;
 650
 651         /* pglob->gl_readdir takes a void *, fix this manually. */
 652         if (pglob->gl_flags & GLOB_ALTDIRFUNC)
 653                 readdirfunc = (struct dirent *(*)(DIR *))pglob->gl_readdir;
 654         else
 655                 readdirfunc = readdir;
 656
 657         /* Search directory for matching names. */
 658         while ((dp = (*readdirfunc)(dirp)) != NULL) {
 659                 char *sc;
 660                 Char *dc;
 661                 wchar_t wc;
 662                 size_t clen;
 663                 mbstate_t mbs;
 664
 665                 if ((pglob->gl_flags & GLOB_LIMIT) &&
 666                     limits[GLOB_INDEX_READDIR]++ >= GLOB_LIMIT_READDIR) {
 667                         errno = 0;
 668                         *pathend++ = SEP;
 669                         *pathend = EOS;
 670                         return GLOB_NOSPACE;
 671                 }
 672
 673                 /* Initial DOT must be matched literally. */
 674                 if (dp->d_name[0] == DOT && *pattern != DOT &&
 675                     !(pglob->gl_flags & GLOB_PERIOD))
 676                         continue;
 677                 memset(&mbs, 0, sizeof(mbs));
 678                 dc = pathend;
 679                 sc = dp->d_name;
 680                 while (dc < pathend_last) {
 681                         clen = mbrtowc(&wc, sc, MB_LEN_MAX, &mbs);
 682                         if (clen == (size_t)-1 || clen == (size_t)-2) {
 683                                 wc = *sc;
 684                                 clen = 1;
 685                                 memset(&mbs, 0, sizeof(mbs));
 686                         }
 687                         if ((*dc++ = wc) == EOS)
 688                                 break;
 689                         sc += clen;
 690                 }
 691                 if (!match(pathend, pattern, restpattern)) {
 692                         *pathend = EOS;
 693                         continue;
 694                 }
 695                 err = glob2(pathbuf, --dc, pathend_last, restpattern,
 696                     pglob, limit);
 697                 if (err)
 698                         break;
 699         }
 700
 701         if (pglob->gl_flags & GLOB_ALTDIRFUNC)
 702                 (*pglob->gl_closedir)(dirp);
 703         else
 704                 closedir(dirp);
 705         return(err);
 706 }
 707
 708
 709 /*
 710  * Extend the gl_pathv member of a glob_t structure to accomodate a new item,
 711  * add the new item, and update gl_pathc.
 712  *
 713  * This assumes the BSD realloc, which only copies the block when its size
 714  * crosses a power-of-two boundary; for v7 realloc, this would cause quadratic
 715  * behavior.
 716  *
 717  * Return 0 if new item added, error code if memory couldn't be allocated.
 718  *
 719  * Invariant of the glob_t structure:
 720  *      Either gl_pathc is zero and gl_pathv is NULL; or gl_pathc > 0 and
 721  *      gl_pathv points to (gl_offs + gl_pathc + 1) items.
 722  */
 723 static int
 724 globextend(const Char *path, glob_t *pglob, size_t *limit)
 725 {
 726         char **pathv;
 727         size_t i, newsize, len;
 728         char *copy;
 729         const Char *p;
 730
 731         if (*limit && pglob->gl_pathc > *limit) {
 732                 errno = 0;
 733                 return (GLOB_NOSPACE);
 734         }
 735
 736         newsize = sizeof(*pathv) * (2 + pglob->gl_pathc + pglob->gl_offs);
 737         pathv = pglob->gl_pathv ?
 738                     realloc((char *)pglob->gl_pathv, newsize) :
 739                     malloc(newsize);
 740         if (pathv == NULL) {
 741                 if (pglob->gl_pathv) {
 742                         free(pglob->gl_pathv);
 743                         pglob->gl_pathv = NULL;
 744                 }
 745                 return(GLOB_NOSPACE);
 746         }
 747
 748         if (pglob->gl_pathv == NULL && pglob->gl_offs > 0) {
 749                 /* first time around -- clear initial gl_offs items */
 750                 pathv += pglob->gl_offs;
 751                 for (i = pglob->gl_offs + 1; --i > 0; )
 752                         *--pathv = NULL;
 753         }
 754         pglob->gl_pathv = pathv;
 755
 756         for (p = path; *p++;)
 757                 continue;
 758         len = MB_CUR_MAX * (size_t)(p - path);  /* XXX overallocation */
 759         limits[GLOB_INDEX_MALLOC] += len;
 760         if ((copy = malloc(len)) != NULL) {
 761                 if (g_Ctoc(path, copy, len)) {
 762                         free(copy);
 763                         return (GLOB_NOSPACE);
 764                 }
 765                 pathv[pglob->gl_offs + pglob->gl_pathc++] = copy;
 766         }
 767         pathv[pglob->gl_offs + pglob->gl_pathc] = NULL;
 768         if ((pglob->gl_flags & GLOB_LIMIT) &&
 769                 (newsize + limits[GLOB_INDEX_MALLOC]) >= GLOB_LIMIT_MALLOC) {
 770                 errno = 0;
 771                 return GLOB_NOSPACE;
 772         }
 773
 774         return(copy == NULL ? GLOB_NOSPACE : 0);
 775 }
 776
 777 /*
 778  * pattern matching function for filenames.  Each occurrence of the *
 779  * pattern causes a recursion level.
 780  */
 781 static int
 782 match(const Char * name, const Char *pat, const Char *patend)
 783 {
 784         int ok, negate_range;
 785         Char c, k;
 786
 787         while (pat < patend) {
 788                 c = *pat++;
 789                 switch (c & M_MASK) {
 790                 case M_ALL:
 791                         if (pat == patend)
 792                                 return(1);
 793                         do
 794                             if (match(name, pat, patend))
 795                                     return(1);
 796                         while (*name++ != EOS);
 797                         return(0);
 798                 case M_ONE:
 799                         if (*name++ == EOS)
 800                                 return(0);
 801                         break;
 802                 case M_SET:
 803                         ok = 0;
 804                         if ((k = *name++) == EOS)
 805                                 return(0);
 806                         if ((negate_range = ((*pat & M_MASK) == M_NOT)) != EOS)
 807                                 ++pat;
 808                         while (((c = *pat++) & M_MASK) != M_END)
 809                                 if ((*pat & M_MASK) == M_RNG) {
 810                                         if (__collate_load_error ?
 811                                             CHAR(c) <= CHAR(k) && CHAR(k) <= CHAR(pat[1]) :
 812                                                __collate_range_cmp(CHAR(c), CHAR(k)) <= 0
 813                                             && __collate_range_cmp(CHAR(k), CHAR(pat[1])) <= 0
 814                                            )
 815                                                 ok = 1;
 816                                         pat += 2;
 817                                 } else if (c == k)
 818                                         ok = 1;
 819                         if (ok == negate_range)
 820                                 return(0);
 821                         break;
 822                 default:
 823                         if (*name++ != c)
 824                                 return(0);
 825                         break;
 826                 }
 827         }
 828         return(*name == EOS);
 829 }
 830
 831 /* Free allocated data belonging to a glob_t structure. */
 832 void
 833 globfree(glob_t *pglob)
 834 {
 835         size_t i;
 836         char **pp;
 837
 838         if (pglob->gl_pathv != NULL) {
 839                 pp = pglob->gl_pathv + pglob->gl_offs;
 840                 for (i = pglob->gl_pathc; i--; ++pp)
 841                         if (*pp)
 842                                 free(*pp);
 843                 free(pglob->gl_pathv);
 844                 pglob->gl_pathv = NULL;
 845         }
 846 }
 847
 848 static DIR *
 849 g_opendir(const Char *str, glob_t *pglob)
 850 {
 851         char buf[MAXPATHLEN];
 852
 853         if (*str == '\0')
 854                 strcpy(buf, ".");
 855         else if (g_Ctoc(str, buf, sizeof(buf)))
 856                 return (NULL);
 857
 858         if (pglob->gl_flags & GLOB_ALTDIRFUNC)
 859                 return((*pglob->gl_opendir)(buf));
 860
 861         return(opendir(buf));
 862 }
 863
 864 static int
 865 g_lstat(const Char *fn, struct stat *sb, glob_t *pglob)
 866 {
 867         char buf[MAXPATHLEN];
 868
 869         if (g_Ctoc(fn, buf, sizeof(buf))) {
 870                 errno = ENAMETOOLONG;
 871                 return (-1);
 872         }
 873         if (pglob->gl_flags & GLOB_ALTDIRFUNC)
 874                 return((*pglob->gl_lstat)(buf, sb));
 875         return(lstat(buf, sb));
 876 }
 877
 878 static int
 879 g_stat(const Char *fn, struct stat *sb, glob_t *pglob)
 880 {
 881         char buf[MAXPATHLEN];
 882
 883         if (g_Ctoc(fn, buf, sizeof(buf))) {
 884                 errno = ENAMETOOLONG;
 885                 return (-1);
 886         }
 887         if (pglob->gl_flags & GLOB_ALTDIRFUNC)
 888                 return((*pglob->gl_stat)(buf, sb));
 889         return(stat(buf, sb));
 890 }
 891
 892 static const Char *
 893 g_strchr(const Char *str, wchar_t ch)
 894 {
 895         for (;; ++str) {
 896                 if (*str == ch)
 897                         return(str);
 898                 if (*str == '\0')
 899                         return(NULL);
 900         }
 901 }
 902
 903 static int
 904 g_Ctoc(const Char *str, char *buf, size_t len)
 905 {
 906         mbstate_t mbs;
 907         size_t clen;
 908
 909         memset(&mbs, 0, sizeof(mbs));
 910         while (len >= MB_CUR_MAX) {
 911                 clen = wcrtomb(buf, *str, &mbs);
 912                 if (clen == (size_t)-1)
 913                         return (1);
 914                 if (*str == L'\0')
 915                         return (0);
 916                 str++;
 917                 buf += clen;
 918                 len -= clen;
 919         }
 920         return (1);
 921 }
 922
 923 #ifdef DEBUG
 924 static void
 925 qprintf(const char *str, const Char *s)
 926 {
 927         const Char *p;
 928
 929         printf("%s:\n", str);
 930         for (p = s; *p; p++)
 931                 printf("%c", CHAR(*p));
 932         printf("\n");
 933         for (p = s; *p; p++)
 934                 printf("%c", *p & M_PROTECT ? '"' : ' ');
 935         printf("\n");
 936         for (p = s; *p; p++)
 937                 printf("%c", ismeta(*p) ? '_' : ' ');
 938         printf("\n");
 939 }
 940 #endif