1*86d7f5d3SJohn Marino /*-
2*86d7f5d3SJohn Marino * Copyright (c) 2008 Peter Holm <pho@FreeBSD.org>
3*86d7f5d3SJohn Marino * All rights reserved.
4*86d7f5d3SJohn Marino *
5*86d7f5d3SJohn Marino * Redistribution and use in source and binary forms, with or without
6*86d7f5d3SJohn Marino * modification, are permitted provided that the following conditions
7*86d7f5d3SJohn Marino * are met:
8*86d7f5d3SJohn Marino * 1. Redistributions of source code must retain the above copyright
9*86d7f5d3SJohn Marino * notice, this list of conditions and the following disclaimer.
10*86d7f5d3SJohn Marino * 2. Redistributions in binary form must reproduce the above copyright
11*86d7f5d3SJohn Marino * notice, this list of conditions and the following disclaimer in the
12*86d7f5d3SJohn Marino * documentation and/or other materials provided with the distribution.
13*86d7f5d3SJohn Marino *
14*86d7f5d3SJohn Marino * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15*86d7f5d3SJohn Marino * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16*86d7f5d3SJohn Marino * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17*86d7f5d3SJohn Marino * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18*86d7f5d3SJohn Marino * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19*86d7f5d3SJohn Marino * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20*86d7f5d3SJohn Marino * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21*86d7f5d3SJohn Marino * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22*86d7f5d3SJohn Marino * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23*86d7f5d3SJohn Marino * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24*86d7f5d3SJohn Marino * SUCH DAMAGE.
25*86d7f5d3SJohn Marino *
26*86d7f5d3SJohn Marino */
27*86d7f5d3SJohn Marino
28*86d7f5d3SJohn Marino /* Test lockf(3) */
29*86d7f5d3SJohn Marino
30*86d7f5d3SJohn Marino #include <stdio.h>
31*86d7f5d3SJohn Marino #include <stdlib.h>
32*86d7f5d3SJohn Marino #include <unistd.h>
33*86d7f5d3SJohn Marino #include <fcntl.h>
34*86d7f5d3SJohn Marino #include <sys/types.h>
35*86d7f5d3SJohn Marino #include <sys/param.h>
36*86d7f5d3SJohn Marino #include <sys/wait.h>
37*86d7f5d3SJohn Marino #include <signal.h>
38*86d7f5d3SJohn Marino #include <errno.h>
39*86d7f5d3SJohn Marino #include <err.h>
40*86d7f5d3SJohn Marino
41*86d7f5d3SJohn Marino #include <stress.h>
42*86d7f5d3SJohn Marino
43*86d7f5d3SJohn Marino char file[128];
44*86d7f5d3SJohn Marino int fd;
45*86d7f5d3SJohn Marino pid_t pid;
46*86d7f5d3SJohn Marino
47*86d7f5d3SJohn Marino int
get(void)48*86d7f5d3SJohn Marino get(void) {
49*86d7f5d3SJohn Marino int sem;
50*86d7f5d3SJohn Marino if (lockf(fd, F_LOCK, 0) == -1)
51*86d7f5d3SJohn Marino err(1, "lockf(%s, F_LOCK)", file);
52*86d7f5d3SJohn Marino if (read(fd, &sem, sizeof(sem)) != sizeof(sem))
53*86d7f5d3SJohn Marino err(1, "get: read(%d)", fd);
54*86d7f5d3SJohn Marino if (lseek(fd, 0, SEEK_SET) == -1)
55*86d7f5d3SJohn Marino err(1, "lseek");
56*86d7f5d3SJohn Marino if (lockf(fd, F_ULOCK, 0) == -1)
57*86d7f5d3SJohn Marino err(1, "lockf(%s, F_ULOCK)", file);
58*86d7f5d3SJohn Marino return (sem);
59*86d7f5d3SJohn Marino }
60*86d7f5d3SJohn Marino
61*86d7f5d3SJohn Marino void
incr(void)62*86d7f5d3SJohn Marino incr(void) {
63*86d7f5d3SJohn Marino int sem;
64*86d7f5d3SJohn Marino if (lockf(fd, F_LOCK, 0) == -1)
65*86d7f5d3SJohn Marino err(1, "lockf(%s, F_LOCK)", file);
66*86d7f5d3SJohn Marino if (read(fd, &sem, sizeof(sem)) != sizeof(sem))
67*86d7f5d3SJohn Marino err(1, "incr: read(%d)", fd);
68*86d7f5d3SJohn Marino if (lseek(fd, 0, SEEK_SET) == -1)
69*86d7f5d3SJohn Marino err(1, "lseek");
70*86d7f5d3SJohn Marino sem++;
71*86d7f5d3SJohn Marino if (write(fd, &sem, sizeof(sem)) != sizeof(sem))
72*86d7f5d3SJohn Marino err(1, "incr: read");
73*86d7f5d3SJohn Marino if (lseek(fd, 0, SEEK_SET) == -1)
74*86d7f5d3SJohn Marino err(1, "lseek");
75*86d7f5d3SJohn Marino if (lockf(fd, F_ULOCK, 0) == -1)
76*86d7f5d3SJohn Marino err(1, "lockf(%s, F_ULOCK)", file);
77*86d7f5d3SJohn Marino }
78*86d7f5d3SJohn Marino int
setup(int nb)79*86d7f5d3SJohn Marino setup(int nb)
80*86d7f5d3SJohn Marino {
81*86d7f5d3SJohn Marino return (0);
82*86d7f5d3SJohn Marino }
83*86d7f5d3SJohn Marino
84*86d7f5d3SJohn Marino void
cleanup(void)85*86d7f5d3SJohn Marino cleanup(void)
86*86d7f5d3SJohn Marino {
87*86d7f5d3SJohn Marino }
88*86d7f5d3SJohn Marino
89*86d7f5d3SJohn Marino int
test(void)90*86d7f5d3SJohn Marino test(void)
91*86d7f5d3SJohn Marino {
92*86d7f5d3SJohn Marino int i;
93*86d7f5d3SJohn Marino int sem = 0;
94*86d7f5d3SJohn Marino
95*86d7f5d3SJohn Marino sprintf(file, "lockf.0.%d", getpid());
96*86d7f5d3SJohn Marino if ((fd = open(file,O_CREAT | O_TRUNC | O_RDWR, 0600)) == -1)
97*86d7f5d3SJohn Marino err(1, "creat(%s)", file);
98*86d7f5d3SJohn Marino if (write(fd, &sem, sizeof(sem)) != sizeof(sem))
99*86d7f5d3SJohn Marino err(1, "write");
100*86d7f5d3SJohn Marino if (lseek(fd, 0, SEEK_SET) == -1)
101*86d7f5d3SJohn Marino err(1, "lseek");
102*86d7f5d3SJohn Marino
103*86d7f5d3SJohn Marino pid = fork();
104*86d7f5d3SJohn Marino if (pid == -1) {
105*86d7f5d3SJohn Marino perror("fork");
106*86d7f5d3SJohn Marino exit(2);
107*86d7f5d3SJohn Marino }
108*86d7f5d3SJohn Marino
109*86d7f5d3SJohn Marino if (pid == 0) { /* child */
110*86d7f5d3SJohn Marino for (i = 0; i < 100; i++) {
111*86d7f5d3SJohn Marino while ((get() & 1) == 0)
112*86d7f5d3SJohn Marino ;
113*86d7f5d3SJohn Marino if (op->verbose > 2)
114*86d7f5d3SJohn Marino printf("Child %d, sem = %d\n", i, get()),
115*86d7f5d3SJohn Marino fflush(stdout);
116*86d7f5d3SJohn Marino incr();
117*86d7f5d3SJohn Marino }
118*86d7f5d3SJohn Marino exit(0);
119*86d7f5d3SJohn Marino } else { /* parent */
120*86d7f5d3SJohn Marino for (i = 0; i < 100; i++) {
121*86d7f5d3SJohn Marino while ((get() & 1) == 1)
122*86d7f5d3SJohn Marino ;
123*86d7f5d3SJohn Marino if (op->verbose > 2)
124*86d7f5d3SJohn Marino printf("Parent %d, sem = %d\n", i, get()),
125*86d7f5d3SJohn Marino fflush(stdout);
126*86d7f5d3SJohn Marino incr();
127*86d7f5d3SJohn Marino }
128*86d7f5d3SJohn Marino }
129*86d7f5d3SJohn Marino close(fd);
130*86d7f5d3SJohn Marino waitpid(pid, &i, 0);
131*86d7f5d3SJohn Marino unlink(file);
132*86d7f5d3SJohn Marino
133*86d7f5d3SJohn Marino return (0);
134*86d7f5d3SJohn Marino }
135