test/clib/flock.c

   1 #include <sys/file.h>
   2 #include <stdio.h>
   3 #include <fcntl.h>
   4 #include <stdlib.h>
   5 #include <unistd.h>
   6 #include <exec/types.h>
   7 #include <proto/dos.h>
   8 #include <dos/dos.h>
   9 #include <proto/exec.h>
  10 #include "test.h"
  11
  12 char *tmpname;
  13
  14 #define ITERATIONS 50
  15 #define NPROCS 10
  16
  17 LONG entry()
  18 {
  19     struct Task *this = FindTask(NULL);
  20     Wait(SIGBREAKF_CTRL_C);
  21     int *counter = this->tc_UserData;
  22     int i;
  23     struct Library *aroscbase;
  24
  25     aroscbase = OpenLibrary("arosc.library", 0);
  26     if(!aroscbase)
  27         return -1;
  28
  29     for(i = 0; i < ITERATIONS; i++)
  30     {
  31         int fd = open(tmpname, 0);
  32         if(!flock(fd, LOCK_EX))
  33         {
  34             int tmp = *counter;
  35             Delay(1);
  36             *counter = tmp + 1;
  37             printf("\rprogress: %.2f%%", *counter * 100.0 / (NPROCS * ITERATIONS));
  38             flock(fd, LOCK_UN);
  39         }
  40         else
  41         {
  42             close(fd);
  43             return -1;
  44         }
  45         close(fd);
  46     }
  47
  48     CloseLibrary(aroscbase);
  49     return 0;
  50 }
  51
  52 int fd;
  53
  54 int main()
  55 {
  56     tmpname = mktemp("T:flockXXXXXX");
  57     int fd = open(tmpname, O_CREAT);
  58     TEST((fd != -1));
  59
  60     TEST((flock(fd, LOCK_SH|LOCK_NB) == 0));
  61     TEST((flock(fd, LOCK_UN) == 0));
  62
  63     TEST((flock(fd, LOCK_EX|LOCK_NB) == 0));
  64     TEST((flock(fd, LOCK_UN) == 0));
  65
  66     TEST((flock(fd, LOCK_SH) == 0));
  67     TEST((flock(fd, LOCK_UN) == 0));
  68
  69     TEST((flock(fd, LOCK_EX) == 0));
  70     TEST((flock(fd, LOCK_UN) == 0));
  71
  72     /* Create NPROCS processes increasing counter ITERATIONS times in an ugly
  73        way */
  74     int counter = 0;
  75     struct Process *procs[NPROCS];
  76     APTR ids[NPROCS];
  77     struct TagItem tags[] =
  78     {
  79         { NP_Entry,         (IPTR) entry     },
  80         { NP_Name,          (IPTR) "flocker" },
  81         { NP_Output,        (IPTR) Output()  },
  82         { NP_CloseOutput,   (IPTR) FALSE     },
  83         { NP_UserData,      (IPTR) &counter  },
  84         { NP_NotifyOnDeath, (IPTR) TRUE      },
  85         { TAG_DONE,         0                }
  86     };
  87
  88     int i;
  89     for(i = 0; i < NPROCS; i++)
  90     {
  91         procs[i] = CreateNewProc(tags);
  92         TEST((procs[i]));
  93         ids[i] = GetETask(procs[i])->et_UniqueID;
  94         Signal(procs[i], SIGBREAKF_CTRL_C);
  95     }
  96
  97     for(i = 0; i < NPROCS; i++)
  98     {
  99         ChildWait(ids[i]);
 100         ChildFree(ids[i]);
 101     }
 102     putchar('\n');
 103
 104     TEST((counter == NPROCS * ITERATIONS));
 105
 106     cleanup();
 107     return OK;
 108 }
 109
 110 void cleanup()
 111 {
 112     close(fd);
 113     remove(tmpname);
 114 }