glob.c
上传用户:sddyfurun
上传日期:2007-01-04
资源大小:525k
文件大小:12k
源码类别:

代理服务器

开发平台:

Unix_Linux

  1. /*
  2.  * Copyright (c) 1980 Regents of the University of California.
  3.  * All rights reserved.
  4.  *
  5.  * Redistribution and use in source and binary forms, with or without
  6.  * modification, are permitted provided that the following conditions
  7.  * are met:
  8.  * 1. Redistributions of source code must retain the above copyright
  9.  *    notice, this list of conditions and the following disclaimer.
  10.  * 2. Redistributions in binary form must reproduce the above copyright
  11.  *    notice, this list of conditions and the following disclaimer in the
  12.  *    documentation and/or other materials provided with the distribution.
  13.  * 3. All advertising materials mentioning features or use of this software
  14.  *    must display the following acknowledgement:
  15.  * This product includes software developed by the University of
  16.  * California, Berkeley and its contributors.
  17.  * 4. Neither the name of the University nor the names of its contributors
  18.  *    may be used to endorse or promote products derived from this software
  19.  *    without specific prior written permission.
  20.  *
  21.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31.  * SUCH DAMAGE.
  32.  */
  33. /*
  34.  * C-shell glob for random programs.
  35.  */
  36. #include "ftp_var.h"
  37. #ifdef HAVE_SYS_STAT_H
  38. #include <sys/stat.h>
  39. #endif
  40. #ifdef HAVE_DIRENT_H
  41. # include <dirent.h>
  42. # define NLENGTH(direct) (strlen((direct)->d_name))
  43. #else /* not HAVE_DIRENT_H */
  44. # define dirent direct
  45. # define NLENGTH(direct) ((direct)->d_namlen)
  46. # ifdef HAVE_SYS_NDIR_H
  47. #  include <sys/ndir.h>
  48. # endif /* HAVE_SYS_NDIR_H */
  49. # ifdef HAVE_SYS_DIR_H
  50. #  include <sys/dir.h>
  51. # endif /* HAVE_SYS_DIR_H */
  52. # ifdef HAVE_NDIR_H
  53. #  include <ndir.h>
  54. # endif /* HAVE_NDIR_H */
  55. #endif /* HAVE_DIRENT_H */
  56. #include <pwd.h>
  57. #include <errno.h>
  58. #include <stdio.h>
  59. #include <stdlib.h>
  60. #include <string.h>
  61. #define QUOTE 0200
  62. #define TRIM 0177
  63. #define eq(a,b) (strcmp(a, b)==0)
  64. #ifndef NCARGS
  65. #define NCARGS 256
  66. #endif
  67. #define GAVSIZ (NCARGS/6)
  68. #define isdir(d) ((d.st_mode & S_IFMT) == S_IFDIR)
  69. static char **gargv; /* Pointer to the (stack) arglist */
  70. static int gargc; /* Number args in gargv */
  71. static int gnleft;
  72. static short gflag;
  73. static char *gpath, *gpathp, *lastgpathp;
  74. static int globbed, globcnt;
  75. static char **sortbas;
  76. static char *entp;
  77. char *globchars = "`{[*?";
  78. char *globerr;
  79. char *home;
  80. extern int errno;
  81. static int match P((char *, char *));
  82. static int execbrc P((char *, char *));
  83. int any(register int c, register char *s) {
  84.     while (*s) if (*s++ == c) return(1);
  85.     return(0);
  86. }
  87. static int tglob(register char c) {
  88.     if (any(c, globchars)) gflag |= c == '{' ? 2 : 1;
  89.     return (c);
  90. }
  91. int letter(register char c) {
  92.     return ((c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') || c == '_');
  93. }
  94. int digit(register char c) {
  95.     return (c >= '0' && c <= '9');
  96. }
  97. int blklen(register char **av) {
  98.     register int i = 0;
  99.     
  100.     while (*av++) i++;
  101.     return (i);
  102. }
  103. char ** blkcpy(char **oav, register char **bv) {
  104.     register char **av = oav;
  105.     
  106.     while ((*av++ = *bv++));
  107.     return (oav);
  108. }
  109. void blkfree(char **av0) {
  110.     register char **av = av0;
  111.     
  112.     /* if gflag is 0, we did not malloc() the space! */
  113.     if (!av || !gflag) return;
  114.     while (*av) free(*av++);
  115. }
  116. static void ginit(char **agargv) {
  117.     agargv[0] = 0; gargv = agargv; sortbas = agargv; gargc = 0;
  118.     gnleft = NCARGS - 4;
  119. }
  120. static void sort() {
  121.     register char **p1, **p2, *c;
  122.     char **Gvp = &gargv[gargc];
  123.     
  124.     p1 = sortbas;
  125.     while (p1 < Gvp-1) {
  126. p2 = p1;
  127. while (++p2 < Gvp)
  128.     if (strcmp(*p1, *p2) > 0)
  129. c = *p1, *p1 = *p2, *p2 = c;
  130. p1++;
  131.     }
  132.     sortbas = Gvp;
  133. }
  134. static void addpath(char c) {
  135.     if (gpathp >= lastgpathp)
  136. globerr = "Pathname too long";
  137.     else {
  138. *gpathp++ = c;
  139. *gpathp = 0;
  140.     }
  141. }
  142. static char *strspl(register char *cp, register char *dp) {
  143.     register char *ep = (char *)malloc((unsigned)(strlen(cp) + strlen(dp) + 1));
  144.     if (ep == NULL) fatal("Out of memory");
  145.     strcpy(ep, cp);
  146.     strcat(ep, dp);
  147.     return (ep);
  148. }
  149. static char *strend(register char *cp) {
  150.     while (*cp) cp++;
  151.     return (cp);
  152. }
  153. static void Gcat(register char *s1, register char *s2) {
  154.     register int len = strlen(s1) + strlen(s2) + 1;
  155.     
  156.     if (len >= gnleft || gargc >= GAVSIZ - 1)
  157. globerr = "Arguments too long";
  158.     else {
  159. gargc++;
  160. gnleft -= len;
  161. gargv[gargc] = 0;
  162. gargv[gargc - 1] = strspl(s1, s2);
  163.     }
  164. }
  165. /*
  166.  * Extract a home directory from the password file
  167.  * The argument points to a buffer where the name of the
  168.  * user whose home directory is sought is currently.
  169.  * We write the home directory of the user back there.
  170.  */
  171. int gethdir(char *home) {
  172.     register struct passwd *pp = getpwnam(home);
  173.     
  174.     if (!pp || home + strlen(pp->pw_dir) >= lastgpathp) return (1);
  175.     strcpy(home, pp->pw_dir);
  176.     return (0);
  177. }
  178. static void matchdir(char *pattern) {
  179.     register struct dirent *dp;
  180.     DIR *dirp;
  181.     
  182.     /* 
  183.      * opendir() does not accept a null directory string
  184.      * to mean the current directory.  Must use ".". But leave
  185.      *  the gpath prefix alone
  186.      */  
  187.     if (*gpath == 0)
  188. dirp = opendir(".");
  189.     else
  190. dirp = opendir(gpath);
  191.     if (dirp == NULL) {
  192. if (globbed) return;
  193. goto patherr2;
  194.     }
  195. #ifndef linux
  196.     {
  197. struct stat stb;
  198. if (fstat(dirp->dd_fd, &stb) < 0)
  199.     goto patherr1;
  200. if (!isdir(stb)) {
  201.     errno = ENOTDIR;
  202.     goto patherr1;
  203. }
  204.     }
  205. #endif /* linux */
  206.     while ((dp = readdir(dirp)) != NULL) {
  207. if (dp->d_ino == 0)
  208.     continue;
  209. if (match(dp->d_name, pattern)) {
  210.     Gcat(gpath, dp->d_name);
  211.     globcnt++;
  212. }
  213.     }
  214.     closedir(dirp);
  215.     return;
  216. #ifndef linux
  217.   patherr1:
  218. #endif
  219.     closedir(dirp);
  220.   patherr2:
  221.     globerr = "Bad directory components";
  222. }
  223. static void expand(char *as) {
  224.     register char *cs;
  225.     register char *sgpathp, *oldcs;
  226.     struct stat stb;
  227.     
  228.     sgpathp = gpathp;
  229.     cs = as;
  230.     if (*cs == '~' && gpathp == gpath) {
  231. addpath('~');
  232. for (cs++; letter(*cs) || digit(*cs) || *cs == '-';)
  233.     addpath(*cs++);
  234. if (!*cs || *cs == '/') {
  235.     if (gpathp != gpath + 1) {
  236. *gpathp = 0;
  237. if (gethdir(gpath + 1))
  238.     globerr = "Unknown user name after ~";
  239. strcpy(gpath, gpath + 1);
  240.     } else
  241. strcpy(gpath, home);
  242.     gpathp = strend(gpath);
  243. }
  244.     }
  245.     while (!any(*cs, globchars)) {
  246. if (*cs == 0) {
  247.     if (!globbed)
  248. Gcat(gpath, "");
  249.     else if (stat(gpath, &stb) >= 0) {
  250. Gcat(gpath, "");
  251. globcnt++;
  252.     }
  253.     goto endit;
  254. }
  255. addpath(*cs++);
  256.     }
  257.     oldcs = cs;
  258.     while (cs > as && *cs != '/')
  259. cs--, gpathp--;
  260.     if (*cs == '/')
  261. cs++, gpathp++;
  262.     *gpathp = 0;
  263.     if (*oldcs == '{') {
  264. execbrc(cs, ((char *)0));
  265. return;
  266.     }
  267.     matchdir(cs);
  268.   endit:
  269.     gpathp = sgpathp;
  270.     *gpathp = 0;
  271. }
  272. static int amatch(register char *s, register char *p) {
  273. register int scc;
  274. int ok, lc;
  275. char *sgpathp;
  276. struct stat stb;
  277. int c, cc;
  278. globbed = 1;
  279. for (;;) {
  280. scc = *s++ & TRIM;
  281. switch (c = *p++) {
  282.     case '{':
  283. return (execbrc(p - 1, s - 1));
  284.     case '[':
  285. ok = 0;
  286. lc = 077777;
  287. while ((cc = *p++)) {
  288.     if (cc == ']') {
  289. if (ok) break;
  290. return (0);
  291.     }
  292.     if (cc == '-') {
  293. if (lc <= scc && scc <= *p++) ok++;
  294.     } else 
  295.     if (scc == (lc = cc)) ok++;
  296. }
  297. if (cc == 0)
  298.     if (!ok) return 0;
  299.     else p--;
  300. continue;
  301.     case '*':
  302. if (!*p) return (1);
  303. if (*p == '/') {
  304.     p++;
  305.     goto slash;
  306. }
  307. s--;
  308. do {
  309.     if (amatch(s, p)) return (1);
  310. } while (*s++);
  311. return (0);
  312.     case 0:
  313. return (scc == 0);
  314.     default:
  315. if (c != scc) return (0);
  316. continue;
  317.     case '?':
  318. if (scc == 0) return (0);
  319. continue;
  320.     case '/':
  321. if (scc) return (0);
  322.   slash:
  323. s = entp;
  324. sgpathp = gpathp;
  325. while (*s) addpath(*s++);
  326. addpath('/');
  327. if (stat(gpath, &stb) == 0 && isdir(stb))
  328.     if (*p == 0) {
  329. Gcat(gpath, "");
  330. globcnt++;
  331.     } else
  332. expand(p);
  333. gpathp = sgpathp;
  334. *gpathp = 0;
  335. return (0);
  336. }
  337. }
  338. }
  339. static int match(char *s, char *p) {
  340. register int c;
  341. register char *sentp;
  342. char sglobbed = globbed;
  343. if (*s == '.' && *p != '.')
  344. return (0);
  345. sentp = entp;
  346. entp = s;
  347. c = amatch(s, p);
  348. entp = sentp;
  349. globbed = sglobbed;
  350. return (c);
  351. }
  352. static int execbrc(char *p, char *s) {
  353.     char restbuf[BUFSIZ + 2];
  354.     register char *pe, *pm, *pl;
  355.     int brclev = 0;
  356.     char *lm, savec, *sgpathp;
  357.     
  358.     for (lm = restbuf; *p != '{'; *lm++ = *p++); 
  359.     for (pe = ++p; *pe; pe++)
  360. switch (*pe) {
  361.     case '{':
  362. brclev++;
  363. continue;
  364.     case '}':
  365. if (brclev == 0) goto pend;
  366. brclev--;
  367. continue;
  368.     case '[':
  369. for (pe++; *pe && *pe != ']'; pe++);
  370. continue;
  371. }
  372. pend:
  373.     brclev = 0;
  374.     for (pl = pm = p; pm <= pe; pm++)
  375. switch (*pm & (QUOTE|TRIM)) {
  376.     case '{':
  377. brclev++;
  378. continue;
  379.     case '}':
  380. if (brclev) {
  381.     brclev--;
  382.     continue;
  383. }
  384. goto doit;
  385.     case ','|QUOTE:
  386.     case ',':
  387. if (brclev) continue;
  388. doit:
  389. savec = *pm;
  390. *pm = 0;
  391. strcpy(lm, pl);
  392. strcat(restbuf, pe + 1);
  393. *pm = savec;
  394. if (s == 0) {
  395.     sgpathp = gpathp;
  396.     expand(restbuf);
  397.     gpathp = sgpathp;
  398.     *gpathp = 0;
  399. } else if (amatch(s, restbuf)) {
  400.     return (1);
  401. }
  402. sort();
  403. pl = pm + 1;
  404. if (brclev) return (0);
  405. continue;
  406.     case '[':
  407. for (pm++; *pm && *pm != ']'; pm++);
  408. if (!*pm) pm--;
  409. continue;
  410. }
  411.     if (brclev) goto doit;
  412.     return (0);
  413. }
  414. static void acollect(register char *as) {
  415.     register int ogargc = gargc;
  416.     
  417.     gpathp = gpath; *gpathp = 0; globbed = 0;
  418.     expand(as);
  419.     if (gargc != ogargc) sort();
  420. }
  421. static void collect(register char *as) {
  422.     if (eq(as, "{") || eq(as, "{}")) {
  423. Gcat(as, "");
  424. sort();
  425.     } else
  426. acollect(as);
  427. }
  428. static int Gmatch(register char *s, register char *p) {
  429.     register int scc;
  430.     int ok, lc;
  431.     int c, cc;
  432.     
  433.     for (;;) {
  434. scc = *s++ & TRIM;
  435. switch (c = *p++) {
  436.     case '[':
  437. ok = 0;
  438. lc = 077777;
  439. while ((cc = *p++)) {
  440.     if (cc == ']') {
  441. if (ok) break;
  442. return (0);
  443.     }
  444.     if (cc == '-') {
  445. if (lc <= scc && scc <= *p++) ok++;
  446.     } else
  447. if (scc == (lc = cc)) ok++;
  448. }
  449. if (cc == 0)
  450.     if (!ok) return 0;
  451.     else p--;
  452. continue;
  453.     case '*':
  454. if (!*p) return (1);
  455. for (s--; *s; s++)
  456.     if (Gmatch(s, p))
  457. return (1);
  458. return (0);
  459.     case 0:
  460. return (scc == 0);
  461.     default:
  462. if ((c & TRIM) != scc) return (0);
  463. continue;
  464.     case '?':
  465. if (scc == 0) return (0);
  466. continue;
  467. }
  468.     }
  469. }
  470. static void rscan(register char **t, int (*f)P((char))) {
  471.     register char *p, c;
  472.     
  473.     while ((p = *t++)) {
  474. if (f == tglob)
  475.     if (*p == '~')
  476. gflag |= 2;
  477.     else if (eq(p, "{") || eq(p, "{}"))
  478. continue;
  479. while ((c = *p++)) (*f)(c);
  480.     }
  481. }
  482. char **copyblk(register char **v) {
  483.     register char **nv = (char **)malloc((unsigned)((blklen(v) + 1) * sizeof(char **)));
  484.     if (nv == (char **)0) fatal("Out of memory");
  485.     return (blkcpy(nv, v));
  486. }
  487. char **ftpglob(register char *v) {
  488. char agpath[BUFSIZ];
  489. char *agargv[GAVSIZ];
  490. char *vv[2];
  491. vv[0] = v;
  492. vv[1] = 0;
  493. gflag = 0;
  494. rscan(vv, tglob);
  495. if (gflag == 0)
  496. return (copyblk(vv));
  497. globerr = 0;
  498. gpath = agpath; gpathp = gpath; *gpathp = 0;
  499. lastgpathp = &gpath[sizeof agpath - 2];
  500. ginit(agargv); globcnt = 0;
  501. collect(v);
  502. if (globcnt == 0 && (gflag&1)) {
  503.     blkfree(gargv), gargv = 0;
  504.     return (0);
  505. } else {
  506.     return (gargv = copyblk(gargv));
  507. }
  508. }