1 /* This file is part of the Zebra server.
2 Copyright (C) 2004-2013 Index Data
3
4 Zebra is free software; you can redistribute it and/or modify it under
5 the terms of the GNU General Public License as published by the Free
6 Software Foundation; either version 2, or (at your option) any later
7 version.
8
9 Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
10 WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
17
18 */
19
20
21 #if HAVE_CONFIG_H
22 #include <config.h>
23 #endif
24 #include <stdio.h>
25 #include <assert.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <stdarg.h>
29
30 #include <sys/types.h>
31 #include <sys/stat.h>
32 #include <fcntl.h>
33
34 #ifdef WIN32
35 #include <io.h>
36 #endif
37 #if HAVE_UNISTD_H
38 #include <unistd.h>
39 #endif
40
41 #include <idzebra/util.h>
42 #include <dfa.h>
43 #include "imalloc.h"
44
45 #ifndef O_BINARY
46 #define O_BINARY 0
47 #endif
48
49 static char *prog;
50
error(const char * format,...)51 void error (const char *format, ...)
52 {
53 va_list argptr;
54 va_start (argptr, format);
55 fprintf (stderr, "%s error: ", prog);
56 (void) vfprintf (stderr, format, argptr);
57 putc ('\n', stderr);
58 exit (1);
59 }
60
61 static int show_lines = 0;
62
agrep_options(int argc,char ** argv)63 int agrep_options (int argc, char **argv)
64 {
65 while (--argc > 0)
66 if (**++argv == '-')
67 while (*++*argv)
68 {
69 switch (**argv)
70 {
71 case 'V':
72 fprintf (stderr, "%s: %s %s\n", prog, __DATE__, __TIME__);
73 continue;
74 case 'v':
75 dfa_verbose = 1;
76 continue;
77 case 'n':
78 show_lines = 1;
79 continue;
80 case 'd':
81 switch (*++*argv)
82 {
83 case 's':
84 debug_dfa_tran = 1;
85 break;
86 case 't':
87 debug_dfa_trav = 1;
88 break;
89 case 'f':
90 debug_dfa_followpos = 1;
91 break;
92 default:
93 --*argv;
94 debug_dfa_tran = 1;
95 debug_dfa_followpos = 1;
96 debug_dfa_trav = 1;
97 }
98 continue;
99 default:
100 fprintf (stderr, "%s: unknown option `-%s'\n", prog, *argv);
101 return 1;
102 }
103 break;
104 }
105 return 0;
106 }
107
108 #define INF_BUF_SIZE 32768U
109 static char *inf_buf;
110 static char *inf_ptr, *inf_flsh;
111 static int inf_eof, line_no;
112
inf_flush(int fd)113 static int inf_flush (int fd)
114 {
115 char *p;
116 unsigned b, r;
117
118 r = (unsigned) (inf_buf+INF_BUF_SIZE - inf_ptr); /* no of `wrap' bytes */
119 if (r)
120 memcpy (inf_buf, inf_ptr, r);
121 inf_ptr = p = inf_buf + r;
122 b = INF_BUF_SIZE - r;
123 do
124 if ((r = read (fd, p, b)) == (unsigned) -1)
125 return -1;
126 else if (r)
127 p += r;
128 else
129 {
130 *p++ = '\n';
131 inf_eof = 1;
132 break;
133 }
134 while ((b -= r) > 0);
135 while (p != inf_buf && *--p != '\n')
136 ;
137 while (p != inf_buf && *--p != '\n')
138 ;
139 inf_flsh = p+1;
140 return 0;
141 }
142
prline(char * p)143 static char *prline (char *p)
144 {
145 char *p0;
146
147 --p;
148 while (p != inf_buf && p[-1] != '\n')
149 --p;
150 p0 = p;
151 while (*p++ != '\n')
152 ;
153 p[-1] = '\0';
154 if (show_lines)
155 printf ("%5d:\t%s\n", line_no, p0);
156 else
157 puts (p0);
158 p[-1] = '\n';
159 return p;
160 }
161
go(int fd,struct DFA_state ** dfaar)162 static int go (int fd, struct DFA_state **dfaar)
163 {
164 struct DFA_state *s = dfaar[0];
165 struct DFA_tran *t;
166 char *p;
167 int i;
168 unsigned char c;
169 int start_line = 1;
170
171 while (1)
172 {
173 for (c = *inf_ptr++, t=s->trans, i=s->tran_no; --i >= 0; t++)
174 if (c >= t->ch[0] && c <= t->ch[1])
175 {
176 p = inf_ptr;
177 do
178 {
179 if ((s = dfaar[t->to])->rule_no &&
180 (start_line || s->rule_nno))
181 {
182 inf_ptr = prline (inf_ptr);
183 c = '\n';
184 break;
185 }
186 for (t=s->trans, i=s->tran_no; --i >= 0; t++)
187 if ((unsigned) *p >= t->ch[0]
188 && (unsigned) *p <= t->ch[1])
189 break;
190 p++;
191 } while (i >= 0);
192 s = dfaar[0];
193 break;
194 }
195 if (c == '\n')
196 {
197 start_line = 1;
198 ++line_no;
199 if (inf_ptr == inf_flsh)
200 {
201 if (inf_eof)
202 break;
203 ++line_no;
204 if (inf_flush (fd))
205 {
206 fprintf (stderr, "%s: read error\n", prog);
207 return -1;
208 }
209 }
210 }
211 else
212 start_line = 0;
213 }
214 return 0;
215 }
216
agrep(struct DFA_state ** dfas,int fd)217 int agrep (struct DFA_state **dfas, int fd)
218 {
219 inf_buf = imalloc (sizeof(char)*INF_BUF_SIZE);
220 inf_eof = 0;
221 inf_ptr = inf_buf+INF_BUF_SIZE;
222 inf_flush (fd);
223 line_no = 1;
224
225 go (fd, dfas);
226
227 ifree (inf_buf);
228 return 0;
229 }
230
231
main(int argc,char ** argv)232 int main (int argc, char **argv)
233 {
234 const char *pattern = NULL;
235 char outbuf[BUFSIZ];
236 int fd, i, no = 0;
237 struct DFA *dfa = dfa_init();
238
239 prog = *argv;
240 if (argc < 2)
241 {
242 fprintf (stderr, "usage: agrep [options] pattern file..\n");
243 fprintf (stderr, " -v dfa verbose\n");
244 fprintf (stderr, " -n show lines\n");
245 fprintf (stderr, " -d debug\n");
246 fprintf (stderr, " -V show version\n");
247 exit (1);
248 }
249 setbuf (stdout, outbuf);
250 i = agrep_options (argc, argv);
251 if (i)
252 return i;
253 while (--argc > 0)
254 if (**++argv != '-' && **argv)
255 {
256 if (!pattern)
257 {
258 pattern = *argv;
259 i = dfa_parse (dfa, &pattern);
260 if (i || *pattern)
261 {
262 fprintf (stderr, "%s: illegal pattern\n", prog);
263 return 1;
264 }
265 dfa_mkstate (dfa);
266 }
267 else
268 {
269 ++no;
270 fd = open (*argv, O_RDONLY | O_BINARY);
271 if (fd == -1)
272 {
273 fprintf (stderr, "%s: couldn't open `%s'\n", prog, *argv);
274 return 1;
275 }
276 i = agrep (dfa->states, fd);
277 close (fd);
278 if (i)
279 return i;
280 }
281 }
282 if (!no)
283 {
284 fprintf (stderr, "usage:\n "
285 " %s [-d] [-v] [-n] [-f] pattern file ..\n", prog);
286 return 2;
287 }
288 fflush(stdout);
289 dfa_delete (&dfa);
290 return 0;
291 }
292 /*
293 * Local variables:
294 * c-basic-offset: 4
295 * c-file-style: "Stroustrup"
296 * indent-tabs-mode: nil
297 * End:
298 * vim: shiftwidth=4 tabstop=8 expandtab
299 */
300
301