dir: Make test utility build again
[openafs.git] / src / dir / test / dtest.c
1 /*
2  * Copyright 2000, International Business Machines Corporation and others.
3  * All Rights Reserved.
4  * 
5  * This software has been released under the terms of the IBM Public
6  * License.  For details, see the LICENSE file in the top-level source
7  * directory or online at http://www.openafs.org/dl/license10.html
8  */
9
10 #define PAGESIZE 2048
11 #include <afsconfig.h>
12 #include <afs/param.h>
13
14 #include <roken.h>
15
16 #include <stdarg.h>
17
18 #include <afs/dir.h>
19
20 long fidCounter = 0;
21
22 typedef struct dirhandle {
23     int fd;
24     int uniq;
25 } dirhandle;
26 int Uniq;
27
28 static void OpenDir(char *name, dirhandle *dir);
29 static void CreateDir(char *name, dirhandle *dir);
30
31 static void
32 Usage(void)
33 {
34     printf("Usage: dtest <command [args]>, where command is one of:\n");
35     printf("-l file - lists directory in file\n");
36     printf("-c file - checks directory in file\n");
37     printf("-s ifile ofile - salvages directory in ifile, places in ofile\n");
38     printf
39         ("-f file name count - creates count names based on \"name\" in directory in file\n");
40     printf("-d file name - delete name from directory in file\n");
41     printf("-r file name - lookup name in directory\n");
42     printf("-a file name - add name to directory in file\n");
43     exit(1);
44 }
45
46 static void
47 LookupDir(char *dname, char *ename)
48 {
49     dirhandle dir;
50     long fid[3];
51     int code;
52
53     OpenDir(dname, &dir);
54     code = Lookup(&dir, ename, fid);
55     if (code)
56         printf("lookup code %d\n", code);
57     else {
58         printf("Found fid %ld.%ld for file '%s'\n", fid[1], fid[2], ename);
59     }
60     DFlush();
61 }
62
63 static void
64 AddEntry(char *dname, char *ename)
65 {
66     dirhandle dir;
67     long fid[3];
68     int code;
69
70     fid[1] = fidCounter++;
71     fid[2] = 3;
72     OpenDir(dname, &dir);
73     code = Create(&dir, ename, fid);
74     if (code)
75         printf("create code %d\n", code);
76     DFlush();
77 }
78
79 static int
80 ListEntry(void * handle, char *name, afs_int32 vnode, afs_int32 unique)
81 {
82     printf("%s\t%ld\t%ld\n", name, afs_printable_int32_ld(vnode),
83            afs_printable_int32_ld(unique));
84
85     return 0;
86 }
87
88 static void
89 ListDir(char *name)
90 {
91     dirhandle dir;
92     OpenDir(name, &dir);
93     EnumerateDir(&dir, ListEntry, 0);
94 }
95
96 static void
97 CheckDir(char *name)
98 {
99     dirhandle dir;
100     OpenDir(name, &dir);
101     if (DirOK(&dir))
102         printf("Directory ok.\n");
103     else
104         printf("Directory bad\n");
105 }
106
107 static void
108 SalvageDir(char *iname, char *oname)
109 {
110     dirhandle in, out;
111     afs_int32 myFid[3], parentFid[3];
112
113     OpenDir(iname, &in);
114     if (Lookup(&in, ".", myFid) || Lookup(&in, "..", parentFid)) {
115         printf("Lookup of \".\" and/or \"..\" failed: ");
116         printf("%d %d %d %d\n", myFid[1], myFid[2], parentFid[1],
117                parentFid[2]);
118         printf("Directory cannnot be salvaged\n");
119     }
120     CreateDir(oname, &out);
121     DirSalvage(&in, &out, myFid[1], myFid[2], parentFid[1], parentFid[2]);
122     DFlush();
123 }
124
125 static void
126 DelTest(char *dname, char *ename)
127 {
128     dirhandle dir;
129     int code;
130
131     OpenDir(dname, &dir);
132     code = Delete(&dir, ename);
133     if (code)
134         printf("delete code is %d\n", code);
135     DFlush();
136 }
137
138 static void
139 CRTest(char *dname, char *ename, int count)
140 {
141     char tbuffer[200];
142     int i;
143     afs_int32 fid[3];
144     dirhandle dir;
145     int code;
146
147     CreateDir(dname, &dir);
148     memset(fid, 0, sizeof(fid));
149     MakeDir(&dir, fid, fid);
150     for (i = 0; i < count; i++) {
151         sprintf(tbuffer, "%s%d", ename, i);
152         fid[1] = fidCounter++;
153         fid[2] = count;
154         code = Create(&dir, tbuffer, &fid);
155         if (i % 100 == 0) {
156             printf("#");
157             fflush(stdout);
158         }
159         if (code) {
160             printf("code for '%s' is %d\n", tbuffer, code);
161             return;
162         }
163     }
164     DFlush();
165 }
166
167 static void
168 OpenDir(char *name, dirhandle *dir)
169 {
170     dir->fd = open(name, O_RDWR, 0666);
171     if (dir->fd == -1) {
172         printf("Couldn't open %s\n", name);
173         exit(1);
174     }
175     dir->uniq = ++Uniq;
176 }
177
178 static void
179 CreateDir(char *name, dirhandle *dir)
180 {
181     dir->fd = open(name, O_CREAT | O_RDWR | O_TRUNC, 0666);
182     dir->uniq = ++Uniq;
183     if (dir->fd == -1) {
184         printf("Couldn't create %s\n", name);
185         exit(1);
186     }
187 }
188
189 int
190 ReallyRead(dirhandle *dir, int block, char *data)
191 {
192     int code;
193     if (lseek(dir->fd, block * PAGESIZE, 0) == -1)
194         return errno;
195     code = read(dir->fd, data, PAGESIZE);
196     if (code < 0)
197         return errno;
198     if (code != PAGESIZE)
199         return EIO;
200     return 0;
201 }
202
203 int
204 ReallyWrite(dirhandle *dir, int block, char *data)
205 {
206     int code;
207     if (lseek(dir->fd, block * PAGESIZE, 0) == -1)
208         return errno;
209     code = write(dir->fd, data, PAGESIZE);
210     if (code < 0)
211         return errno;
212     if (code != PAGESIZE)
213         return EIO;
214     return 0;
215 }
216
217 void
218 FidZap(dirhandle *dir)
219 {
220     dir->fd = -1;
221 }
222
223 void
224 FidZero(long *afid)
225 {
226     *afid = 0;
227 }
228
229 int
230 FidEq(dirhandle *dir1, dirhandle *dir2)
231 {
232     return (dir1->uniq == dir2->uniq);
233 }
234
235 int
236 FidVolEq(long *afid, long *bfid)
237 {
238     return 1;
239 }
240
241 void
242 FidCpy(dirhandle *todir, dirhandle *fromdir)
243 {
244     *todir = *fromdir;
245 }
246
247 void
248 Die(char *msg)
249 {
250     printf("Something died with this message:  %s\n", msg);
251 }
252
253 void
254 Log(const char *fmt, ...)
255 {
256     va_list ap;
257
258     va_start(ap, fmt);
259     vprintf(fmt, ap);
260     va_end(ap);
261 }
262
263 int
264 main(int argc, char **argv)
265 {
266     DInit(600);
267     argc--;
268     argv++;
269
270     if (argc == 0)
271         Usage();
272
273     switch ((*argv++)[1]) {
274     case 'l':
275         ListDir(*argv);
276         break;
277     case 'c':
278         CheckDir(*argv);
279         break;
280     case 's':
281         SalvageDir(*argv, argv[1]);
282         break;
283     case 'f':
284         CRTest(*argv, argv[1], atoi(argv[2]));
285         break;
286     case 'd':
287         DelTest(*argv, argv[1]);
288         break;
289     case 'r':
290         LookupDir(*argv, argv[1]);
291         break;
292     case 'a':
293         AddEntry(*argv, argv[1]);
294         break;
295     default:
296         Usage();
297     }
298     exit(0);
299 }