Commit | Line | Data |
463ee0b2 |
1 | #include <stdio.h> |
2 | #include <sys/file.h> |
3 | #ifdef SDBM |
17f28c40 |
4 | #include "EXTERN.h" |
463ee0b2 |
5 | #include "sdbm.h" |
6 | #else |
7 | #include <ndbm.h> |
8 | #endif |
9 | #include <string.h> |
10 | |
11 | #ifdef BSD42 |
12 | #define strchr index |
13 | #endif |
14 | |
15 | extern int getopt(); |
16 | extern char *strchr(); |
17 | extern void oops(); |
18 | |
19 | char *progname; |
20 | |
21 | static int rflag; |
22 | static char *usage = "%s [-R] cat | look |... dbmname"; |
23 | |
24 | #define DERROR 0 |
25 | #define DLOOK 1 |
26 | #define DINSERT 2 |
27 | #define DDELETE 3 |
28 | #define DCAT 4 |
29 | #define DBUILD 5 |
30 | #define DPRESS 6 |
31 | #define DCREAT 7 |
32 | |
33 | #define LINEMAX 8192 |
34 | |
35 | typedef struct { |
36 | char *sname; |
37 | int scode; |
38 | int flags; |
39 | } cmd; |
40 | |
41 | static cmd cmds[] = { |
42 | |
43 | "fetch", DLOOK, O_RDONLY, |
44 | "get", DLOOK, O_RDONLY, |
45 | "look", DLOOK, O_RDONLY, |
46 | "add", DINSERT, O_RDWR, |
47 | "insert", DINSERT, O_RDWR, |
48 | "store", DINSERT, O_RDWR, |
49 | "delete", DDELETE, O_RDWR, |
50 | "remove", DDELETE, O_RDWR, |
51 | "dump", DCAT, O_RDONLY, |
52 | "list", DCAT, O_RDONLY, |
53 | "cat", DCAT, O_RDONLY, |
54 | "creat", DCREAT, O_RDWR | O_CREAT | O_TRUNC, |
55 | "new", DCREAT, O_RDWR | O_CREAT | O_TRUNC, |
56 | "build", DBUILD, O_RDWR | O_CREAT, |
57 | "squash", DPRESS, O_RDWR, |
58 | "compact", DPRESS, O_RDWR, |
59 | "compress", DPRESS, O_RDWR |
60 | }; |
61 | |
62 | #define CTABSIZ (sizeof (cmds)/sizeof (cmd)) |
63 | |
64 | static cmd *parse(); |
65 | static void badk(), doit(), prdatum(); |
66 | |
67 | int |
ba106d47 |
68 | main(int argc, char **argv) |
463ee0b2 |
69 | { |
70 | int c; |
71 | register cmd *act; |
72 | extern int optind; |
73 | extern char *optarg; |
74 | |
75 | progname = argv[0]; |
76 | |
77 | while ((c = getopt(argc, argv, "R")) != EOF) |
78 | switch (c) { |
79 | case 'R': /* raw processing */ |
80 | rflag++; |
81 | break; |
82 | |
83 | default: |
84 | oops("usage: %s", usage); |
85 | break; |
86 | } |
87 | |
88 | if ((argc -= optind) < 2) |
89 | oops("usage: %s", usage); |
90 | |
91 | if ((act = parse(argv[optind])) == NULL) |
92 | badk(argv[optind]); |
93 | optind++; |
94 | doit(act, argv[optind]); |
95 | return 0; |
96 | } |
97 | |
98 | static void |
ba106d47 |
99 | doit(register cmd *act, char *file) |
463ee0b2 |
100 | { |
101 | datum key; |
102 | datum val; |
103 | register DBM *db; |
104 | register char *op; |
105 | register int n; |
106 | char *line; |
107 | #ifdef TIME |
108 | long start; |
109 | extern long time(); |
110 | #endif |
111 | |
112 | if ((db = dbm_open(file, act->flags, 0644)) == NULL) |
113 | oops("cannot open: %s", file); |
114 | |
115 | if ((line = (char *) malloc(LINEMAX)) == NULL) |
116 | oops("%s: cannot get memory", "line alloc"); |
117 | |
118 | switch (act->scode) { |
119 | |
120 | case DLOOK: |
121 | while (fgets(line, LINEMAX, stdin) != NULL) { |
122 | n = strlen(line) - 1; |
123 | line[n] = 0; |
124 | key.dptr = line; |
125 | key.dsize = n; |
126 | val = dbm_fetch(db, key); |
127 | if (val.dptr != NULL) { |
128 | prdatum(stdout, val); |
129 | putchar('\n'); |
130 | continue; |
131 | } |
132 | prdatum(stderr, key); |
133 | fprintf(stderr, ": not found.\n"); |
134 | } |
135 | break; |
136 | case DINSERT: |
137 | break; |
138 | case DDELETE: |
139 | while (fgets(line, LINEMAX, stdin) != NULL) { |
140 | n = strlen(line) - 1; |
141 | line[n] = 0; |
142 | key.dptr = line; |
143 | key.dsize = n; |
144 | if (dbm_delete(db, key) == -1) { |
145 | prdatum(stderr, key); |
146 | fprintf(stderr, ": not found.\n"); |
147 | } |
148 | } |
149 | break; |
150 | case DCAT: |
151 | for (key = dbm_firstkey(db); key.dptr != 0; |
152 | key = dbm_nextkey(db)) { |
153 | prdatum(stdout, key); |
154 | putchar('\t'); |
155 | prdatum(stdout, dbm_fetch(db, key)); |
156 | putchar('\n'); |
157 | } |
158 | break; |
159 | case DBUILD: |
160 | #ifdef TIME |
161 | start = time(0); |
162 | #endif |
163 | while (fgets(line, LINEMAX, stdin) != NULL) { |
164 | n = strlen(line) - 1; |
165 | line[n] = 0; |
166 | key.dptr = line; |
167 | if ((op = strchr(line, '\t')) != 0) { |
168 | key.dsize = op - line; |
169 | *op++ = 0; |
170 | val.dptr = op; |
171 | val.dsize = line + n - op; |
172 | } |
173 | else |
174 | oops("bad input; %s", line); |
175 | |
176 | if (dbm_store(db, key, val, DBM_REPLACE) < 0) { |
177 | prdatum(stderr, key); |
178 | fprintf(stderr, ": "); |
179 | oops("store: %s", "failed"); |
180 | } |
181 | } |
182 | #ifdef TIME |
183 | printf("done: %d seconds.\n", time(0) - start); |
184 | #endif |
185 | break; |
186 | case DPRESS: |
187 | break; |
188 | case DCREAT: |
189 | break; |
190 | } |
191 | |
192 | dbm_close(db); |
193 | } |
194 | |
195 | static void |
ba106d47 |
196 | badk(char *word) |
463ee0b2 |
197 | { |
198 | register int i; |
199 | |
200 | if (progname) |
201 | fprintf(stderr, "%s: ", progname); |
202 | fprintf(stderr, "bad keywd %s. use one of\n", word); |
203 | for (i = 0; i < (int)CTABSIZ; i++) |
204 | fprintf(stderr, "%-8s%c", cmds[i].sname, |
205 | ((i + 1) % 6 == 0) ? '\n' : ' '); |
206 | fprintf(stderr, "\n"); |
207 | exit(1); |
208 | /*NOTREACHED*/ |
209 | } |
210 | |
211 | static cmd * |
ba106d47 |
212 | parse(register char *str) |
463ee0b2 |
213 | { |
214 | register int i = CTABSIZ; |
215 | register cmd *p; |
216 | |
217 | for (p = cmds; i--; p++) |
218 | if (strcmp(p->sname, str) == 0) |
219 | return p; |
220 | return NULL; |
221 | } |
222 | |
223 | static void |
ba106d47 |
224 | prdatum(FILE *stream, datum d) |
463ee0b2 |
225 | { |
226 | register int c; |
227 | register char *p = d.dptr; |
228 | register int n = d.dsize; |
229 | |
230 | while (n--) { |
231 | c = *p++ & 0377; |
232 | if (c & 0200) { |
233 | fprintf(stream, "M-"); |
234 | c &= 0177; |
235 | } |
236 | if (c == 0177 || c < ' ') |
237 | fprintf(stream, "^%c", (c == 0177) ? '?' : c + '@'); |
238 | else |
239 | putc(c, stream); |
240 | } |
241 | } |
242 | |
243 | |