added demo of signal/uid handling feature
[tridge/junkcode.git] / os2_delete.c
1 /*
2   test readdir/unlink pattern that OS/2 uses
3   tridge@samba.org July 2005
4 */
5
6 #define _GNU_SOURCE
7 #include <stdio.h>
8 #include <stdlib.h>
9 #include <sys/stat.h>
10 #include <unistd.h>
11 #include <sys/types.h>
12 #include <dirent.h>
13 #include <errno.h>
14 #include <string.h>
15 #include <fcntl.h>
16
17 #define NUM_FILES 700
18 #define READDIR_SIZE 100
19 #define DELETE_SIZE 4
20
21 #define TESTDIR "test.dir"
22
23 #define FAILED() (fprintf(stderr, "Failed at %s:%d - %s\n", __FUNCTION__, __LINE__, strerror(errno)), exit(1), 1)
24
25 #ifndef MIN
26 #define MIN(a,b) ((a)<(b)?(a):(b))
27 #endif
28
29 static void cleanup(void)
30 {
31         /* I'm a lazy bastard */
32         system("rm -rf " TESTDIR);
33         mkdir(TESTDIR, 0700) == 0 || FAILED();
34 }
35
36 static void create_files()
37 {
38         int i;
39         for (i=0;i<NUM_FILES;i++) {
40                 char fname[40];
41                 snprintf(fname, sizeof(fname), TESTDIR "/test%u.txt", i);
42                 close(open(fname, O_CREAT|O_RDWR, 0600)) == 0 || FAILED();
43         }
44 }
45
46 static int os2_delete(DIR *d)
47 {
48         off_t offsets[READDIR_SIZE];
49         int i, j;
50         struct dirent *de;
51         char names[READDIR_SIZE][30];
52
53         /* scan, remembering offsets */
54         for (i=0, de=readdir(d); 
55              de && i < READDIR_SIZE; 
56              de=readdir(d), i++) {
57                 offsets[i] = telldir(d);
58                 strcpy(names[i], de->d_name);
59         }
60
61         if (i == 0) {
62                 return 0;
63         }
64
65         /* delete the first few */
66         for (j=0; j<MIN(i, DELETE_SIZE); j++) {
67                 char fname[40];
68                 snprintf(fname, sizeof(fname), TESTDIR "/%s", names[j]);
69                 unlink(fname) == 0 || FAILED();
70         }
71
72         /* seek to just after the deletion */
73         seekdir(d, offsets[j-1]);
74
75         /* return number deleted */
76         return j;
77 }
78
79 int main(void)
80 {
81         int total_deleted = 0;
82         DIR *d;
83         struct dirent *de;
84
85         cleanup();
86         create_files();
87         
88         d = opendir(TESTDIR);
89
90         /* skip past . and .. */
91         de = readdir(d);
92         strcmp(de->d_name, ".") == 0 || FAILED();
93         de = readdir(d);
94         strcmp(de->d_name, "..") == 0 || FAILED();
95
96         while (1) {
97                 int n = os2_delete(d);
98                 if (n == 0) break;
99                 total_deleted += n;
100         }
101         closedir(d);
102
103         printf("Deleted %d files of %d\n", total_deleted, NUM_FILES);
104
105         rmdir(TESTDIR) == 0 || FAILED();
106
107         return 0;
108 }