1 /*
2 * test_icount.c
3 *
4 * Copyright (C) 1997 Theodore Ts'o.
5 *
6 * %Begin-Header%
7 * This file may be redistributed under the terms of the GNU Public
8 * License.
9 * %End-Header%
10 */
11
12 #include "config.h"
13 #include <stdio.h>
14 #include <stdlib.h>
15 #include <string.h>
16 #include <unistd.h>
17 #ifdef HAVE_GETOPT_H
18 #include <getopt.h>
19 #endif
20 #include <fcntl.h>
21
22 #include <ext2fs/ext2_fs.h>
23
24 #include <et/com_err.h>
25 #include <ss/ss.h>
26 #include <ext2fs/ext2fs.h>
27 #include <ext2fs/irel.h>
28 #include <ext2fs/brel.h>
29
30 extern ss_request_table test_cmds;
31
32 #include "test_icount.h"
33
34 ext2_filsys test_fs;
35 ext2_icount_t test_icount;
36
37 /*
38 * Helper function which assures that the icount structure is valid
39 */
check_icount(char * request)40 static int check_icount(char *request)
41 {
42 if (test_icount)
43 return 0;
44 com_err(request, 0, "The icount structure must be allocated.");
45 return 1;
46 }
47
48 /*
49 * Helper function which parses an inode number.
50 */
parse_inode(const char * request,const char * desc,const char * str,ext2_ino_t * ino)51 static int parse_inode(const char *request, const char *desc,
52 const char *str, ext2_ino_t *ino)
53 {
54 char *tmp;
55
56 *ino = strtoul(str, &tmp, 0);
57 if (*tmp) {
58 com_err(request, 0, "Bad %s - %s", desc, str);
59 return 1;
60 }
61 return 0;
62 }
63
do_create_icount(int argc,char ** argv)64 void do_create_icount(int argc, char **argv)
65 {
66 errcode_t retval;
67 char *progname;
68 int flags = 0;
69 ext2_ino_t size = 5;
70
71 progname = *argv;
72 argv++; argc --;
73
74 if (argc && !strcmp("-i", *argv)) {
75 flags |= EXT2_ICOUNT_OPT_INCREMENT;
76 argv++; argc--;
77 }
78 if (argc) {
79 if (parse_inode(progname, "icount size", argv[0], &size))
80 return;
81 argv++; argc--;
82 }
83 #if 0
84 printf("Creating icount... flags=%d, size=%d\n", flags, (int) size);
85 #endif
86 retval = ext2fs_create_icount(test_fs, flags, (int) size,
87 &test_icount);
88 if (retval) {
89 com_err(progname, retval, "while creating icount");
90 return;
91 }
92 }
93
do_free_icount(int argc,char ** argv)94 void do_free_icount(int argc, char **argv)
95 {
96 if (check_icount(argv[0]))
97 return;
98
99 ext2fs_free_icount(test_icount);
100 test_icount = 0;
101 }
102
do_fetch(int argc,char ** argv)103 void do_fetch(int argc, char **argv)
104 {
105 const char *usage = "usage: %s inode\n";
106 errcode_t retval;
107 ext2_ino_t ino;
108 __u16 count;
109
110 if (argc < 2) {
111 printf(usage, argv[0]);
112 return;
113 }
114 if (check_icount(argv[0]))
115 return;
116 if (parse_inode(argv[0], "inode", argv[1], &ino))
117 return;
118 retval = ext2fs_icount_fetch(test_icount, ino, &count);
119 if (retval) {
120 com_err(argv[0], retval, "while calling ext2fs_icount_fetch");
121 return;
122 }
123 printf("Count is %u\n", count);
124 }
125
do_increment(int argc,char ** argv)126 void do_increment(int argc, char **argv)
127 {
128 const char *usage = "usage: %s inode\n";
129 errcode_t retval;
130 ext2_ino_t ino;
131 __u16 count;
132
133 if (argc < 2) {
134 printf(usage, argv[0]);
135 return;
136 }
137 if (check_icount(argv[0]))
138 return;
139 if (parse_inode(argv[0], "inode", argv[1], &ino))
140 return;
141 retval = ext2fs_icount_increment(test_icount, ino, &count);
142 if (retval) {
143 com_err(argv[0], retval,
144 "while calling ext2fs_icount_increment");
145 return;
146 }
147 printf("Count is now %u\n", count);
148 }
149
do_decrement(int argc,char ** argv)150 void do_decrement(int argc, char **argv)
151 {
152 const char *usage = "usage: %s inode\n";
153 errcode_t retval;
154 ext2_ino_t ino;
155 __u16 count;
156
157 if (argc < 2) {
158 printf(usage, argv[0]);
159 return;
160 }
161 if (check_icount(argv[0]))
162 return;
163 if (parse_inode(argv[0], "inode", argv[1], &ino))
164 return;
165 retval = ext2fs_icount_decrement(test_icount, ino, &count);
166 if (retval) {
167 com_err(argv[0], retval,
168 "while calling ext2fs_icount_decrement");
169 return;
170 }
171 printf("Count is now %u\n", count);
172 }
173
do_store(int argc,char ** argv)174 void do_store(int argc, char **argv)
175 {
176 const char *usage = "usage: %s inode count\n";
177 errcode_t retval;
178 ext2_ino_t ino;
179 ext2_ino_t count;
180
181 if (argc < 3) {
182 printf(usage, argv[0]);
183 return;
184 }
185 if (check_icount(argv[0]))
186 return;
187 if (parse_inode(argv[0], "inode", argv[1], &ino))
188 return;
189 if (parse_inode(argv[0], "count", argv[2], &count))
190 return;
191 if (count > 65535) {
192 printf("Count too large.\n");
193 return;
194 }
195 retval = ext2fs_icount_store(test_icount, ino, (__u16) count);
196 if (retval) {
197 com_err(argv[0], retval,
198 "while calling ext2fs_icount_store");
199 return;
200 }
201 }
202
do_dump(int argc,char ** argv)203 void do_dump(int argc, char **argv)
204 {
205 errcode_t retval;
206 ext2_ino_t i;
207 __u16 count;
208
209 if (check_icount(argv[0]))
210 return;
211 for (i=1; i <= test_fs->super->s_inodes_count; i++) {
212 retval = ext2fs_icount_fetch(test_icount, i, &count);
213 if (retval) {
214 com_err(argv[0], retval,
215 "while fetching icount for %lu", (unsigned long)i);
216 return;
217 }
218 if (count)
219 printf("%lu: %u\n", (unsigned long)i, count);
220 }
221 }
222
do_validate(int argc,char ** argv)223 void do_validate(int argc, char **argv)
224 {
225 errcode_t retval;
226
227 if (check_icount(argv[0]))
228 return;
229 retval = ext2fs_icount_validate(test_icount, stdout);
230 if (retval) {
231 com_err(argv[0], retval, "while validating icount structure");
232 return;
233 }
234 printf("Icount structure successfully validated\n");
235 }
236
do_get_size(int argc,char ** argv)237 void do_get_size(int argc, char **argv)
238 {
239 ext2_ino_t size;
240
241 if (check_icount(argv[0]))
242 return;
243 size = ext2fs_get_icount_size(test_icount);
244 printf("Size of icount is: %lu\n", (unsigned long)size);
245 }
246
source_file(const char * cmd_file,int sci_idx)247 static int source_file(const char *cmd_file, int sci_idx)
248 {
249 FILE *f;
250 char buf[256];
251 char *cp;
252 int exit_status = 0;
253 int retval;
254 int noecho;
255
256 if (strcmp(cmd_file, "-") == 0)
257 f = stdin;
258 else {
259 f = fopen(cmd_file, "r");
260 if (!f) {
261 perror(cmd_file);
262 exit(1);
263 }
264 }
265 fflush(stdout);
266 fflush(stderr);
267 setbuf(stdout, NULL);
268 setbuf(stderr, NULL);
269 while (!feof(f)) {
270 if (fgets(buf, sizeof(buf), f) == NULL)
271 break;
272 if (buf[0] == '#')
273 continue;
274 noecho = 0;
275 if (buf[0] == '-') {
276 noecho = 1;
277 buf[0] = ' ';
278 }
279 cp = strchr(buf, '\n');
280 if (cp)
281 *cp = 0;
282 cp = strchr(buf, '\r');
283 if (cp)
284 *cp = 0;
285 if (!noecho)
286 printf("test_icount: %s\n", buf);
287 retval = ss_execute_line(sci_idx, buf);
288 if (retval) {
289 ss_perror(sci_idx, retval, buf);
290 exit_status++;
291 }
292 }
293 if (f != stdin)
294 fclose(f);
295 return exit_status;
296 }
297
main(int argc,char ** argv)298 int main(int argc, char **argv)
299 {
300 int retval;
301 int sci_idx;
302 int c;
303 char *request = 0;
304 int exit_status = 0;
305 char *cmd_file = 0;
306 struct ext2_super_block param;
307
308 initialize_ext2_error_table();
309
310 /*
311 * Create a sample filesystem structure
312 */
313 memset(¶m, 0, sizeof(struct ext2_super_block));
314 ext2fs_blocks_count_set(¶m, 80000);
315 param.s_inodes_count = 20000;
316 retval = ext2fs_initialize("/dev/null", 0, ¶m,
317 unix_io_manager, &test_fs);
318 if (retval) {
319 com_err("/dev/null", retval, "while setting up test fs");
320 exit(1);
321 }
322
323 while ((c = getopt (argc, argv, "wR:f:")) != EOF) {
324 switch (c) {
325 case 'R':
326 request = optarg;
327 break;
328 case 'f':
329 cmd_file = optarg;
330 break;
331 default:
332 com_err(argv[0], 0, "Usage: test_icount "
333 "[-R request] [-f cmd_file]");
334 exit(1);
335 }
336 }
337 sci_idx = ss_create_invocation("test_icount", "0.0", (char *) NULL,
338 &test_cmds, &retval);
339 if (retval) {
340 ss_perror(sci_idx, retval, "creating invocation");
341 exit(1);
342 }
343
344 (void) ss_add_request_table (sci_idx, &ss_std_requests, 1, &retval);
345 if (retval) {
346 ss_perror(sci_idx, retval, "adding standard requests");
347 exit (1);
348 }
349 if (request) {
350 retval = 0;
351 retval = ss_execute_line(sci_idx, request);
352 if (retval) {
353 ss_perror(sci_idx, retval, request);
354 exit_status++;
355 }
356 } else if (cmd_file) {
357 exit_status = source_file(cmd_file, sci_idx);
358 } else {
359 ss_listen(sci_idx);
360 }
361
362 return(exit_status);
363 }
364