import HEAD into svn+ssh://svn.samba.org/home/svn/samba/trunk
[metze/old/v3-2-winbind-ndr.git] / source / 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 static int sys_waitpid(pid_t pid,int *status,int options)
26 {
27 #ifdef HAVE_WAITPID
28   return waitpid(pid,status,options);
29 #else /* USE_WAITPID */
30   return wait4(pid, status, options, NULL);
31 #endif /* USE_WAITPID */
32 }
33
34 #define DATA "conftest.fcntl"
35
36 #ifndef SEEK_SET
37 #define SEEK_SET 0
38 #endif
39
40 /* lock a byte range in a open file */
41 int main(int argc, char *argv[])
42 {
43         struct flock lock;
44         int fd, ret, status=1;
45         pid_t pid;
46         char *testdir = NULL;
47
48         testdir = getenv("TESTDIR");
49         if (testdir) chdir(testdir);
50
51         alarm(10);
52
53         if (!(pid=fork())) {
54                 sleep(2);
55                 fd = open(DATA, O_RDONLY);
56
57                 if (fd == -1) {
58                         fprintf(stderr,"ERROR: failed to open %s (errno=%d)\n", 
59                                 DATA, (int)errno);
60                         exit(1);
61                 }
62
63                 lock.l_type = F_WRLCK;
64                 lock.l_whence = SEEK_SET;
65                 lock.l_start = 0;
66                 lock.l_len = 4;
67                 lock.l_pid = getpid();
68                 
69                 lock.l_type = F_WRLCK;
70                 
71                 /* check if a lock applies */
72                 ret = fcntl(fd,F_GETLK,&lock);
73
74                 if ((ret == -1) ||
75                     (lock.l_type == F_UNLCK)) {
76                         fprintf(stderr,"ERROR: lock test failed (ret=%d errno=%d)\n", ret, (int)errno);
77                         exit(1);
78                 } else {
79                         exit(0);
80                 }
81         }
82
83         unlink(DATA);
84         fd = open(DATA, O_RDWR|O_CREAT|O_EXCL, 0600);
85
86         if (fd == -1) {
87                 fprintf(stderr,"ERROR: failed to open %s (errno=%d)\n", 
88                         DATA, (int)errno);
89                 exit(1);
90         }
91
92         lock.l_type = F_WRLCK;
93         lock.l_whence = SEEK_SET;
94         lock.l_start = 0;
95         lock.l_len = 4;
96         lock.l_pid = getpid();
97
98         /* set a 4 byte write lock */
99         fcntl(fd,F_SETLK,&lock);
100
101         sys_waitpid(pid, &status, 0);
102
103         unlink(DATA);
104
105 #if defined(WIFEXITED) && defined(WEXITSTATUS)
106     if(WIFEXITED(status)) {
107         status = WEXITSTATUS(status);
108     } else {
109         status = 1;
110     }
111 #else /* defined(WIFEXITED) && defined(WEXITSTATUS) */
112         status = (status == 0) ? 0 : 1;
113 #endif /* defined(WIFEXITED) && defined(WEXITSTATUS) */
114
115         if (status) {
116                 fprintf(stderr,"ERROR: lock test failed with status=%d\n", 
117                         status);
118         }
119
120         exit(status);
121 }