Whamcloud - gitweb
b=22040 use df POSIX output format
[fs/lustre-release.git] / lustre / tests / flock_test.c
1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
2  * vim:expandtab:shiftwidth=8:tabstop=8:
3  *
4  * GPL HEADER START
5  *
6  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License version 2 only,
10  * as published by the Free Software Foundation.
11  *
12  * This program is distributed in the hope that it will be useful, but
13  * WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * General Public License version 2 for more details (a copy is included
16  * in the LICENSE file that accompanied this code).
17  *
18  * You should have received a copy of the GNU General Public License
19  * version 2 along with this program; If not, see
20  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
21  *
22  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
23  * CA 95054 USA or visit www.sun.com if you need additional information or
24  * have any questions.
25  *
26  * GPL HEADER END
27  */
28 /*
29  * Copyright  2008 Sun Microsystems, Inc. All rights reserved
30  * Use is subject to license terms.
31  */
32 /*
33  * This file is part of Lustre, http://www.lustre.org/
34  * Lustre is a trademark of Sun Microsystems, Inc.
35  */
36
37 #include <stdio.h>
38 #include <stdlib.h>
39 #include <errno.h>
40 #include <fcntl.h>
41 #include <unistd.h>
42 #include <string.h>
43 #include <sys/types.h>
44 #include <sys/stat.h>
45 #include <sys/file.h>
46 #include <sys/wait.h>
47
48 void chd_lock_unlock(int);
49 char fname[1024];
50
51 int main(int argc, char **argv)
52 {
53     pid_t pid;
54     int cfd, fd, rc;
55
56     if (argc != 2) {
57         fprintf(stderr, "\nUSAGE: flock_test filepath\n");
58         exit(2);
59     }
60     strncpy(fname, argv[1], 1023);
61     fname[1023] ='\0';
62     fd = open(fname, O_RDWR|O_CREAT, (mode_t)0666);
63     if (fd == -1) {
64         fprintf(stderr, "flock_test: failed to open %s : ", fname);
65         perror("");
66         exit(1);
67     }
68     if (flock(fd, LOCK_EX | LOCK_NB) == -1) {
69         fprintf(stderr, "flock_test: parent attempt to lock %s failed : ", \
70             fname);
71         perror("");
72         exit(1);
73     }
74
75     pid = fork();
76     if (pid == -1) {
77         fprintf(stderr, "flock_test: fork failed : ");
78         perror("");
79         exit(1);
80     }
81
82     if (pid == 0) {
83         pid = getpid();
84         sleep(2);
85         if ((cfd = open(fname, O_RDWR)) == -1) {
86             fprintf(stderr, "flock_test child (%d) cannot open %s: ", \
87                 pid, fname);
88             perror("");
89             exit(1);
90         }
91         if(flock(cfd, LOCK_EX | LOCK_NB) != -1) {
92             fprintf(stderr, "flock_test child (%d): %s not yet locked  : ", \
93                 pid, fname);
94             exit(1);
95         }
96         if(flock(fd, LOCK_UN) == -1) {
97             fprintf(stderr, "flock_test child (%d): cannot unlock %s: ", \
98                 pid, fname);
99             perror("");
100             exit(1);
101         }
102         if(flock(cfd, LOCK_EX | LOCK_NB) == -1 ) {
103             fprintf(stderr, \
104                 "flock_test: child (%d) cannot re-lock %s after unlocking : ", \
105                 pid, fname);
106             perror("");
107             exit(1);
108         }
109         close(cfd);
110         exit(0);
111     }
112
113     waitpid(pid, &rc, 0);
114     close(fd);
115     unlink(fname);
116     if (WIFEXITED(rc) && WEXITSTATUS(rc) != 0) {
117         fprintf(stderr, "flock_test: child (%d) exit code = %d\n", \
118             pid, WEXITSTATUS(rc));
119         exit(1);
120     }
121     exit(0);
122 }