home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Professional / OS2PRO194.ISO / os2 / editor / sed / glob.c < prev    next >
C/C++ Source or Header  |  1994-01-31  |  14KB  |  591 lines
  1. /* File-name wildcard pattern matching for GNU.
  2.    Copyright (C) 1985, 1988, 1989 Free Software Foundation, Inc.
  3.  
  4.    This program is free software; you can redistribute it and/or modify
  5.    it under the terms of the GNU General Public License as published by
  6.    the Free Software Foundation; either version 1, or (at your option)
  7.    any later version.
  8.  
  9.    This program is distributed in the hope that it will be useful,
  10.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.    GNU General Public License for more details.
  13.  
  14.    You should have received a copy of the GNU General Public License
  15.    along with this program; if not, write to the Free Software
  16.    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  17.  
  18. /* To whomever it may concern: I have never seen the code which most
  19.  Unix programs use to perform this function.  I wrote this from scratch
  20.  based on specifications for the pattern matching.  */
  21.  
  22. #include <sys/types.h>
  23.  
  24. #ifdef    USGr3
  25. #include <dirent.h>
  26. #define direct dirent
  27. #define    D_NAMLEN(d) strlen((d)->d_name)
  28. #else    /* not USGr3    */
  29. #define D_NAMLEN(d) ((d)->d_namlen)
  30. #ifdef MSDOS
  31. #include "dir.h"
  32. #else
  33. #ifdef    USG
  34. #include "ndir.h"   /* Get ndir.h from the Emacs distribution.  */
  35. #else    /* not USG    */
  36. #include <sys/dir.h>
  37. #endif    /* USG        */
  38. #endif
  39. #endif    /* USGr3    */
  40.  
  41. #ifdef USG
  42. #include <memory.h>
  43. #include <string.h>
  44. #define bcopy(s, d, n) ((void) memcpy ((d), (s), (n)))
  45. #define rindex strrchr
  46. struct passwd *getpwent(), *getpwuid(), *getpwnam();
  47. #ifndef MSDOS
  48. extern char *memcpy ();
  49. #endif
  50. #else /* not USG */
  51. #include <strings.h>
  52.  
  53. extern void bcopy ();
  54. #endif /* not USG */
  55.  
  56. #ifdef    __GNUC__
  57. #define    alloca(n)    __builtin_alloca (n)
  58. #else    /* Not GCC.  */
  59. #ifdef    sparc
  60. #include <alloca.h>
  61. #else    /* Not sparc.  */
  62. extern char *alloca ();
  63. #endif    /* sparc.  */
  64. #endif    /* GCC.  */
  65.  
  66. #ifndef MSDOS
  67. #include <pwd.h>
  68. #endif
  69.  
  70. extern char *malloc (), *realloc ();
  71. extern void free ();
  72.  
  73. #ifndef NULL
  74. #define NULL 0
  75. #endif
  76.  
  77. /* Zero if * matches .*.  */
  78. int noglob_dot_filenames = 1;
  79.  
  80. /* Nonzero if ~ and ~USER are expanded by glob_filename.  */
  81. int glob_tilde = 0;
  82.  
  83.  
  84. static int glob_match_after_star ();
  85.  
  86. /* Return nonzero if PATTERN has any special globbing chars in it.  */
  87. int
  88. glob_pattern_p (pattern)
  89.      char *pattern;
  90. {
  91.   register char *p = pattern;
  92.   register char c;
  93.  
  94.   while ((c = *p++))
  95.     {
  96.       switch (c)
  97.     {
  98.     case '?':
  99.     case '[':
  100.     case '*':
  101.       return 1;
  102.  
  103.     case '\\':
  104.       if (*p++ == 0) return 0;
  105.     default:
  106.       ;
  107.     }
  108.     }
  109.  
  110.   return 0;
  111. }
  112.  
  113.  
  114. /* Match the pattern PATTERN against the string TEXT;
  115.    return 1 if it matches, 0 otherwise.
  116.  
  117.    A match means the entire string TEXT is used up in matching.
  118.  
  119.    In the pattern string, `*' matches any sequence of characters,
  120.    `?' matches any character, [SET] matches any character in the specified set,
  121.    [^SET] matches any character not in the specified set.
  122.  
  123.    A set is composed of characters or ranges; a range looks like
  124.    character hyphen character (as in 0-9 or A-Z).
  125.    [0-9a-zA-Z_] is the set of characters allowed in C identifiers.
  126.    Any other character in the pattern must be matched exactly.
  127.  
  128.    To suppress the special syntactic significance of any of `[]*?^-\',
  129.    and match the character exactly, precede it with a `\'.
  130.  
  131.    If DOT_SPECIAL is nonzero,
  132.    `*' and `?' do not match `.' at the beginning of TEXT.  */
  133.  
  134. int
  135. glob_match (pattern, text, dot_special)
  136.      char *pattern, *text;
  137.      int dot_special;
  138. {
  139.   register char *p = pattern, *t = text;
  140.   register char c;
  141.  
  142.   while ((c = *p++))
  143.     {
  144.       switch (c)
  145.     {
  146.     case '?':
  147.       if (*t == 0 || (dot_special && t == text && *t == '.')) return 0;
  148.       else ++t;
  149.       break;
  150.  
  151.     case '\\':
  152.       if (*p++ != *t++) return 0;
  153.       break;
  154.  
  155.     case '*':
  156.       if (dot_special && t == text && *t == '.')
  157.         return 0;
  158.       return glob_match_after_star (p, t);
  159.  
  160.     case '[':
  161.       {
  162.         register char c1 = *t++;
  163.         register int invert = (*p == '^');
  164.  
  165.         if (invert) p++;
  166.  
  167.         c = *p++;
  168.         while (1)
  169.           {
  170.         register char cstart = c, cend = c;
  171.  
  172.         if (c == '\\')
  173.           {
  174.             cstart = *p++; cend = cstart;
  175.           }
  176.  
  177.         if (!c) return (0);
  178.  
  179.         c = *p++;
  180.  
  181.         if (c == '-')
  182.           {
  183.             cend = *p++;
  184.             if (cend == '\\')
  185.               cend = *p++;
  186.             if (!cend) return (0);
  187.             c = *p++;
  188.           }
  189.         if (c1 >= cstart && c1 <= cend) goto match;
  190.         if (c == ']')
  191.           break;
  192.           }
  193.         if (!invert) return 0;
  194.         break;
  195.  
  196.       match:
  197.         /* Skip the rest of the [...] construct that already matched.  */
  198.         while (c != ']')
  199.           {
  200.             if (!c || !(c = *p++)) return (0);
  201.         if (c == '\\') p++;
  202.           }
  203.         if (invert) return 0;
  204.         break;
  205.       }
  206.  
  207.     default:
  208.       if (c != *t++) return 0;
  209.     }
  210.     }
  211.  
  212.   if (*t) return 0;
  213.   return 1;
  214. }
  215.  
  216. /* Like glob_match, but match PATTERN against any final segment of TEXT.  */
  217.  
  218. static int
  219. glob_match_after_star (pattern, text)
  220.      char *pattern, *text;
  221. {
  222.   register char *p = pattern, *t = text;
  223.   register char c, c1;
  224.  
  225.   while ((c = *p++) == '?' || c == '*')
  226.     {
  227.       if (c == '?' && *t++ == 0)
  228.     return 0;
  229.     }
  230.  
  231.   if (c == 0)
  232.     return 1;
  233.  
  234.   if (c == '\\') c1 = *p;
  235.   else c1 = c;
  236.  
  237.   for (;;)
  238.     {
  239.       if ((c == '[' || *t == c1)
  240.           && glob_match (p - 1, t, 0))
  241.     return 1;
  242.       if (*t++ == 0) return 0;
  243.     }
  244. }
  245.  
  246. /* Return a vector of names of files in directory DIR
  247.    whose names match glob pattern PAT.
  248.    The names are not in any particular order.
  249.    Wildcards at the beginning of PAT do not match an initial period.
  250.  
  251.    The vector is terminated by an element that is a null pointer.
  252.  
  253.    To free the space allocated, first free the vector's elements,
  254.    then free the vector.
  255.  
  256.    Return 0 if cannot get enough memory to hold the pointer
  257.    and the names.
  258.  
  259.    Return -1 if cannot access directory DIR.
  260.    Look in errno for more information.  */
  261.  
  262. char **
  263. glob_vector (pat, dir)
  264.      char *pat;
  265.      char *dir;
  266. {
  267.   struct globval
  268.     {
  269.       struct globval *next;
  270.       char *name;
  271.     };
  272.  
  273.   DIR *d;
  274.   register struct direct *dp;
  275.   struct globval *lastlink;
  276.   register struct globval *nextlink;
  277.   register char *nextname;
  278.   int count;
  279.   int lose;
  280.   register char **name_vector;
  281.   register int i;
  282.  
  283.   if (!(d = opendir (dir)))
  284.     return (char **) -1;
  285.  
  286.   lastlink = 0;
  287.   count = 0;
  288.   lose = 0;
  289.  
  290.   /* Scan the directory, finding all names that match.
  291.      For each name that matches, allocate a struct globval
  292.      on the stack and store the name in it.
  293.      Chain those structs together; lastlink is the front of the chain.  */
  294.   /* Loop reading blocks */
  295.   while (1)
  296.     {
  297.       dp = readdir (d);
  298.       if (!dp) break;
  299.       if (dp->d_ino && glob_match (pat, dp->d_name, noglob_dot_filenames))
  300.     {
  301.       nextlink = (struct globval *) alloca (sizeof (struct globval));
  302.       nextlink->next = lastlink;
  303.       nextname = (char *) malloc (D_NAMLEN(dp) + 1);
  304.       if (!nextname)
  305.         {
  306.           lose = 1;
  307.           break;
  308.         }
  309.       lastlink = nextlink;
  310.       nextlink->name = nextname;
  311.       bcopy (dp->d_name, nextname, D_NAMLEN(dp) + 1);
  312.       count++;
  313.     }
  314.     }
  315.   closedir (d);
  316.  
  317.   name_vector = (char **) malloc ((count + 1) * sizeof (char *));
  318.  
  319.   /* Have we run out of memory?  */
  320.   if (!name_vector || lose)
  321.     {
  322.       /* Here free the strings we have got */
  323.       while (lastlink)
  324.     {
  325.       free (lastlink->name);
  326.       lastlink = lastlink->next;
  327.     }
  328.       return 0;
  329.     }
  330.  
  331.   /* Copy the name pointers from the linked list into the vector */
  332.   for (i = 0; i < count; i++)
  333.     {
  334.       name_vector[i] = lastlink->name;
  335.       lastlink = lastlink->next;
  336.     }
  337.  
  338.   name_vector[count] = 0;
  339.   return name_vector;
  340. }
  341.  
  342. /* Return a new array which is the concatenation of each string in
  343.    ARRAY to DIR. */
  344.  
  345. static char **
  346. glob_dir_to_array (dir, array)
  347.      char *dir, **array;
  348. {
  349.   register int i, l;
  350.   int add_slash = 0;
  351.   char **result;
  352.  
  353.   l = strlen (dir);
  354.   if (!l) return (array);
  355.  
  356.   if (dir[l - 1] != '/') add_slash++;
  357.  
  358.   for (i = 0; array[i]; i++);
  359.  
  360.   result = (char **)malloc ((1 + i) * sizeof (char *));
  361.   if (!result) return (result);
  362.  
  363.   for (i = 0; array[i]; i++) {
  364.     result[i] = (char *)malloc (1 + l + add_slash + strlen (array[i]));
  365.     if (!result[i]) return (char **)NULL;
  366.     strcpy (result[i], dir);
  367.     if (add_slash) strcat (result[i], "/");
  368.     strcat (result[i], array[i]);
  369.   }
  370.   result[i] = (char *)NULL;
  371.  
  372.   /* Free the input array. */
  373.   for (i = 0; array[i]; i++) free (array[i]);
  374.   free (array);
  375.   return (result);
  376. }
  377.  
  378. /* Do globbing on PATHNAME.  Return an array of pathnames that match,
  379.    marking the end of the array with a null-pointer as an element.
  380.    If no pathnames match, then the array is empty (first element is null).
  381.    If there isn't enough memory, then return NULL.
  382.    If a file system error occurs, return -1; `errno' has the error code.
  383.  
  384.    Wildcards at the beginning of PAT, or following a slash,
  385.    do not match an initial period.  */
  386.  
  387. char **
  388. glob_filename (pathname)
  389.      char *pathname;
  390. {
  391.   char **result;
  392.   unsigned int result_size;
  393.   char *directory_name, *filename;
  394.   unsigned int directory_len;
  395.  
  396.   result = (char **) malloc (sizeof (char *));
  397.   result_size = 1;
  398.   if (result == NULL)
  399.     return NULL;
  400.  
  401.   result[0] = NULL;
  402.  
  403.   /* Find the filename.  */
  404.   filename = rindex (pathname, '/');
  405.   if (filename == 0)
  406.     {
  407.       filename = pathname;
  408.       directory_name = "";
  409.       directory_len = 0;
  410.     }
  411.   else
  412.     {
  413.       directory_len = filename - pathname;
  414.       directory_name = (char *) alloca (directory_len + 1);
  415.       bcopy (pathname, directory_name, directory_len);
  416.       directory_name[directory_len] = '\0';
  417.       ++filename;
  418.     }
  419.  
  420. #ifndef MSDOS
  421.   if (glob_tilde && *pathname == '~')
  422.     {
  423.       if (directory_len == 0)
  424.     {
  425.       filename = directory_name;
  426.       directory_name = pathname;
  427.       directory_len = strlen (directory_name);
  428.     }
  429.  
  430.       if (directory_len == 1)
  431.     {
  432.       extern char *getenv ();
  433.       static char *home_directory = 0;
  434.       static unsigned int home_len;
  435.  
  436.       if (home_directory == 0)
  437.         {
  438.           home_directory = getenv ("HOME");
  439.           if (home_directory == NULL)
  440.         {
  441.           home_directory == "";
  442.           home_len = 0;
  443.         }
  444.           else
  445.         home_len = strlen (home_directory);
  446.         }
  447.  
  448.       directory_name = home_directory;
  449.       directory_len = home_len;
  450.     }
  451.       else
  452.     {
  453.       struct passwd *pwent = getpwnam (directory_name + 1);
  454.       if (pwent == 0)
  455.         {
  456.           directory_name = "";
  457.           directory_len = 0;
  458.         }
  459.       else
  460.         {
  461.           directory_name = pwent->pw_dir;
  462.           directory_len = strlen (directory_name);
  463.         }
  464.     }
  465.     }
  466.   else
  467. #endif
  468.      if (glob_pattern_p (directory_name))
  469.     {
  470.       /* If directory_name contains globbing characters, then we
  471.      have to expand the previous levels.  Just recurse. */
  472.       char **directories;
  473.       register unsigned int i;
  474.  
  475.       if (directory_name[directory_len - 1] == '/')
  476.     directory_name[directory_len - 1] = '\0';
  477.  
  478.       directories = glob_filename (directory_name);
  479.       if (directories == NULL)
  480.     goto memory_error;
  481.       else if ((int) directories == -1)
  482.     return (char **) -1;
  483.       else if (*directories == NULL)
  484.     {
  485.       free ((char *) directories);
  486.       return (char **) -1;
  487.     }
  488.  
  489.       /* We have successfully globbed the preceding directory name.
  490.      For each name in DIRECTORIES, call glob_vector on it and
  491.      FILENAME.  Concatenate the results together.  */
  492.       for (i = 0; directories[i] != NULL; ++i)
  493.     {
  494.       char **temp_results = glob_vector (filename, directories[i]);
  495.       if (temp_results == NULL)
  496.         goto memory_error;
  497.       else if ((int) temp_results == -1)
  498.         /* This filename is probably not a directory.  Ignore it.  */
  499.         ;
  500.       else
  501.         {
  502.           char **array = glob_dir_to_array (directories[i], temp_results);
  503.  
  504.           register unsigned int l = 0;
  505.           while (array[l] != NULL)
  506.         ++l;
  507.  
  508.           result = (char **) realloc (result,
  509.                       (result_size + 1) * sizeof (char *));
  510.           if (result == NULL)
  511.         goto memory_error;
  512.  
  513.           for (l = 0; array[l] != NULL; ++l)
  514.         result[result_size++ - 1] = array[l];
  515.           result[result_size - 1] = NULL;
  516.           free ((char *) array);
  517.         }
  518.     }
  519.       /* Free the directories.  */
  520.       for (i = 0; directories[i]; i++)
  521.     free (directories[i]);
  522.       free ((char *) directories);
  523.  
  524.       return result;
  525.     }
  526.  
  527.   if (*filename == '\0')
  528.     {
  529.       /* If there is only a directory name, return it.  */
  530.       result = (char **) realloc ((char *) result, 2 * sizeof (char *));
  531.       if (result == NULL)
  532.     return NULL;
  533.       result[0] = (char *) malloc (directory_len + 1);
  534.       if (result[0] == NULL)
  535.     goto memory_error;
  536.       bcopy (directory_name, result[0], directory_len + 1);
  537.       result[1] = NULL;
  538.       return result;
  539.     }
  540.   else
  541.     {
  542.       /* Otherwise, just return what glob_vector
  543.      returns appended to the directory name. */
  544.       char **temp_results = glob_vector (filename,
  545.                      (directory_len == 0
  546.                       ? "." : directory_name));
  547.  
  548.       if (temp_results == NULL || (int) temp_results == -1)
  549.     return temp_results;
  550.  
  551.       return glob_dir_to_array (directory_name, temp_results);
  552.     }
  553.  
  554.   memory_error:;
  555.   if (result != NULL)
  556.     {
  557.       register unsigned int i;
  558.       for (i = 0; result[i] != NULL; ++i)
  559.     free (result[i]);
  560.       free ((char *) result);
  561.     }
  562.   return NULL;
  563. }
  564.  
  565.  
  566.  
  567. #ifdef TEST
  568.  
  569. main (argc, argv)
  570.      int argc;
  571.      char **argv;
  572. {
  573.   char **value;
  574.   int i, index = 1;
  575.  
  576.   while (index < argc) {
  577.     value = glob_filename (argv[index]);
  578.     if ((int) value == 0)
  579.       printf ("Memory exhausted.\n");
  580.     else if ((int) value == -1)
  581.       perror (argv[index]);
  582.     else
  583.       for (i = 0; value[i]; i++)
  584.     printf ("%s\n", value[i]);
  585.     index++;
  586.   }
  587.   return 0;
  588. }
  589.  
  590. #endif /* TEST */
  591.