ffd1ee69fab1acc4828117af6e206752314777f0
[samba.git] / source4 / build / tests / fcntl_lock.c
1 /* test whether fcntl locking works on this system */
2
3 #if defined(HAVE_UNISTD_H)
4 #include <unistd.h>
5 #endif
6
7 #include <stdio.h>
8 #include <stdlib.h>
9 #include <sys/types.h>
10
11 #ifdef HAVE_FCNTL_H
12 #include <fcntl.h>
13 #endif
14
15 #ifdef HAVE_SYS_FCNTL_H
16 #include <sys/fcntl.h>
17 #endif
18
19 #ifdef HAVE_SYS_WAIT_H
20 #include <sys/wait.h>
21 #endif
22
23 #include <errno.h>
24
25 #define DATA "conftest.fcntl"
26
27 #ifndef SEEK_SET
28 #define SEEK_SET 0
29 #endif
30
31 /* lock a byte range in a open file */
32 int main(int argc, char *argv[])
33 {
34         struct flock lock;
35         int fd, ret, status=1;
36         pid_t pid;
37         char *testdir = NULL;
38
39         testdir = getenv("TESTDIR");
40         if (testdir) chdir(testdir);
41
42         alarm(10);
43
44         if (!(pid=fork())) {
45                 sleep(2);
46                 fd = open(DATA, O_RDONLY);
47
48                 if (fd == -1) {
49                         fprintf(stderr,"ERROR: failed to open %s (errno=%d)\n", 
50                                 DATA, (int)errno);
51                         exit(1);
52                 }
53
54                 lock.l_type = F_WRLCK;
55                 lock.l_whence = SEEK_SET;
56                 lock.l_start = 0;
57                 lock.l_len = 4;
58                 lock.l_pid = getpid();
59                 
60                 lock.l_type = F_WRLCK;
61                 
62                 /* check if a lock applies */
63                 ret = fcntl(fd,F_GETLK,&lock);
64
65                 if ((ret == -1) ||
66                     (lock.l_type == F_UNLCK)) {
67                         fprintf(stderr,"ERROR: lock test failed (ret=%d errno=%d)\n", ret, (int)errno);
68                         exit(1);
69                 } else {
70                         exit(0);
71                 }
72         }
73
74         unlink(DATA);
75         fd = open(DATA, O_RDWR|O_CREAT|O_EXCL, 0600);
76
77         if (fd == -1) {
78                 fprintf(stderr,"ERROR: failed to open %s (errno=%d)\n", 
79                         DATA, (int)errno);
80                 exit(1);
81         }
82
83         lock.l_type = F_WRLCK;
84         lock.l_whence = SEEK_SET;
85         lock.l_start = 0;
86         lock.l_len = 4;
87         lock.l_pid = getpid();
88
89         /* set a 4 byte write lock */
90         fcntl(fd,F_SETLK,&lock);
91
92         waitpid(pid, &status, 0);
93
94         unlink(DATA);
95
96 #if defined(WIFEXITED) && defined(WEXITSTATUS)
97     if(WIFEXITED(status)) {
98         status = WEXITSTATUS(status);
99     } else {
100         status = 1;
101     }
102 #else /* defined(WIFEXITED) && defined(WEXITSTATUS) */
103         status = (status == 0) ? 0 : 1;
104 #endif /* defined(WIFEXITED) && defined(WEXITSTATUS) */
105
106         if (status) {
107                 fprintf(stderr,"ERROR: lock test failed with status=%d\n", 
108                         status);
109         }
110
111         exit(status);
112 }