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 }