glob.c 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219
  1. #include <u.h>
  2. #include <libc.h>
  3. #include <regexp.h>
  4. #include <String.h>
  5. #include "glob.h"
  6. /*
  7. * I wrote this glob so that there would be no limit
  8. * on element or path size. The one in rc is probably
  9. * better, certainly faster. - presotto
  10. */
  11. static Glob*
  12. globnew(void)
  13. {
  14. Glob *g;
  15. g = mallocz(sizeof(*g), 1);
  16. if(g == nil)
  17. sysfatal("globnew: %r");
  18. return g;
  19. }
  20. static void
  21. globfree1(Glob *g)
  22. {
  23. s_free(g->glob);
  24. free(g);
  25. }
  26. static void
  27. globfree(Glob *g)
  28. {
  29. Glob *next;
  30. for(; g != nil; g = next){
  31. next = g->next;
  32. globfree1(g);
  33. }
  34. }
  35. static Globlist*
  36. globlistnew(char *x)
  37. {
  38. Globlist *gl;
  39. gl = mallocz(sizeof *gl, 1);
  40. if(gl == nil)
  41. sysfatal("globlistnew: %r");
  42. gl->first = globnew();
  43. gl->first->glob = s_copy(x);
  44. gl->l = &gl->first->next;
  45. return gl;
  46. }
  47. void
  48. globlistfree(Globlist *gl)
  49. {
  50. if(gl == nil)
  51. return;
  52. globfree(gl->first);
  53. free(gl);
  54. }
  55. void
  56. globadd(Globlist *gl, char *dir, char *file)
  57. {
  58. Glob *g;
  59. g = globnew();
  60. g->glob = s_copy(dir);
  61. if(strcmp(dir, "/") != 0 && *dir != 0)
  62. s_append(g->glob, "/");
  63. s_append(g->glob, file);
  64. *(gl->l) = g;
  65. gl->l = &(g->next);
  66. }
  67. static void
  68. globdir(Globlist *gl, char *dir, Reprog *re)
  69. {
  70. Dir *d;
  71. int i, n, fd;
  72. if(*dir == 0)
  73. fd = open(".", OREAD);
  74. else
  75. fd = open(dir, OREAD);
  76. if(fd < 0)
  77. return;
  78. n = dirreadall(fd, &d);
  79. if(n == 0)
  80. return;
  81. close(fd);
  82. for(i = 0; i < n; i++)
  83. if(regexec(re, d[i].name, nil, 0))
  84. globadd(gl, dir, d[i].name);
  85. free(d);
  86. }
  87. static void
  88. globdot(Globlist *gl, char *dir)
  89. {
  90. Dir *d;
  91. if(*dir == 0){
  92. globadd(gl, "", ".");
  93. return;
  94. }
  95. d = dirstat(dir);
  96. if(d == nil)
  97. return;
  98. if(d->qid.type & QTDIR)
  99. globadd(gl, dir, ".");
  100. free(d);
  101. }
  102. static void
  103. globnext(Globlist *gl, char *pattern)
  104. {
  105. String *np;
  106. Glob *g, *inlist;
  107. Reprog *re;
  108. int c;
  109. /* nothing left */
  110. if(*pattern == 0)
  111. return;
  112. inlist = gl->first;
  113. gl->first = nil;
  114. gl->l = &gl->first;
  115. /* pick off next pattern and turn into a reg exp */
  116. np = s_new();
  117. s_putc(np, '^');
  118. for(; c = *pattern; pattern++){
  119. if(c == '/'){
  120. pattern++;
  121. break;
  122. }
  123. switch(c){
  124. case '|':
  125. case '+':
  126. case '.':
  127. case '^':
  128. case '$':
  129. case '(':
  130. case ')':
  131. s_putc(np, '\\');
  132. s_putc(np, c);
  133. break;
  134. case '?':
  135. s_putc(np, '.');
  136. break;
  137. case '*':
  138. s_putc(np, '.');
  139. s_putc(np, '*');
  140. break;
  141. default:
  142. s_putc(np, c);
  143. break;
  144. }
  145. }
  146. s_putc(np, '$');
  147. s_terminate(np);
  148. if(strcmp(s_to_c(np), "^\\.$") == 0){
  149. /* anything that's a directory works */
  150. for(g = inlist; g != nil; g = g->next)
  151. globdot(gl, s_to_c(g->glob));
  152. } else {
  153. re = regcomp(s_to_c(np));
  154. /* run input list as directories */
  155. for(g = inlist; g != nil; g = g->next)
  156. globdir(gl, s_to_c(g->glob), re);
  157. free(re);
  158. }
  159. s_free(np);
  160. globfree(inlist);
  161. if(gl->first != nil)
  162. globnext(gl, pattern);
  163. }
  164. char *
  165. globiter(Globlist *gl)
  166. {
  167. Glob *g;
  168. char *s;
  169. if(gl->first == nil)
  170. return nil;
  171. g = gl->first;
  172. gl->first = g->next;
  173. if(gl->first == nil)
  174. gl->l = &gl->first;
  175. s = strdup(s_to_c(g->glob));
  176. if(s == nil)
  177. sysfatal("globiter: %r");
  178. globfree1(g);
  179. return s;
  180. }
  181. Globlist*
  182. glob(char *pattern)
  183. {
  184. Globlist *gl;
  185. if(pattern == nil || *pattern == 0)
  186. return nil;
  187. if(*pattern == '/'){
  188. pattern++;
  189. gl = globlistnew("/");
  190. } else
  191. gl = globlistnew("");
  192. globnext(gl, pattern);
  193. return gl;
  194. }