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