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