xref: /freebsd/tools/test/stress2/misc/lockf4.sh (revision 608c97bff921c8936eb71f4b1a177625cfa60250)
18a272653SPeter Holm#!/bin/sh
28a272653SPeter Holm
38a272653SPeter Holm#
48a272653SPeter Holm# Copyright (c) 2015 EMC Corp.
58a272653SPeter Holm# All rights reserved.
68a272653SPeter Holm#
78a272653SPeter Holm# Redistribution and use in source and binary forms, with or without
88a272653SPeter Holm# modification, are permitted provided that the following conditions
98a272653SPeter Holm# are met:
108a272653SPeter Holm# 1. Redistributions of source code must retain the above copyright
118a272653SPeter Holm#    notice, this list of conditions and the following disclaimer.
128a272653SPeter Holm# 2. Redistributions in binary form must reproduce the above copyright
138a272653SPeter Holm#    notice, this list of conditions and the following disclaimer in the
148a272653SPeter Holm#    documentation and/or other materials provided with the distribution.
158a272653SPeter Holm#
168a272653SPeter Holm# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
178a272653SPeter Holm# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
188a272653SPeter Holm# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
198a272653SPeter Holm# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
208a272653SPeter Holm# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
218a272653SPeter Holm# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
228a272653SPeter Holm# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
238a272653SPeter Holm# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
248a272653SPeter Holm# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
258a272653SPeter Holm# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
268a272653SPeter Holm# SUCH DAMAGE.
278a272653SPeter Holm#
288a272653SPeter Holm
298a272653SPeter Holm# lockf(3) test scenario.
308a272653SPeter Holm
318a272653SPeter Holm[ `id -u ` -ne 0 ] && echo "Must be root!" && exit 1
328a272653SPeter Holm
338a272653SPeter Holm. ../default.cfg
348a272653SPeter Holm
358a272653SPeter Holmodir=`pwd`
368a272653SPeter Holmcd /tmp
378a272653SPeter Holmsed '1,/^EOF/d' < $odir/$0 > lockf4.c
388a272653SPeter Holmrm -f /tmp/lockf4
398a272653SPeter Holmmycc -o lockf4 -Wall -Wextra -g -O2 lockf4.c || exit 1
408a272653SPeter Holmrm -f lockf4.c
418a272653SPeter Holmcd $odir
428a272653SPeter Holm
438a272653SPeter Holm[ -c /dev/md$mdstart ] && mdconfig -d -u $mdstart
448a272653SPeter Holmmdconfig -a -t swap -s 1g -u $mdstart || exit 1
45*608c97bfSPeter Holmnewfs $newfs_flags md$mdstart > /dev/null
46*608c97bfSPeter Holmmount /dev/md$mdstart $mntpoint
478a272653SPeter Holmchmod 777 $mntpoint
488a272653SPeter Holm
498a272653SPeter Holmsu $testuser -c "(cd $mntpoint; /tmp/lockf4)" &
508a272653SPeter Holmsu $testuser -c "(cd $mntpoint; /tmp/lockf4)" &
518a272653SPeter Holmsu $testuser -c "(cd $mntpoint; /tmp/lockf4)" &
528a272653SPeter Holmwait
538a272653SPeter Holm
548a272653SPeter Holmwhile mount | grep "on $mntpoint " | grep -q /dev/md; do
558a272653SPeter Holm	umount $mntpoint || sleep 1
568a272653SPeter Holmdone
578a272653SPeter Holmmdconfig -d -u $mdstart
588a272653SPeter Holmrm -f /tmp/lockf4
598a272653SPeter Holmexit 0
608a272653SPeter HolmEOF
618a272653SPeter Holm#include <fcntl.h>
628a272653SPeter Holm#include <err.h>
638a272653SPeter Holm#include <errno.h>
648a272653SPeter Holm#include <stdio.h>
658a272653SPeter Holm#include <stdlib.h>
668a272653SPeter Holm#include <time.h>
678a272653SPeter Holm#include <unistd.h>
688a272653SPeter Holm
698a272653SPeter Holm#define N (5 * 1024 * 1024) /* 40 MB */
708a272653SPeter Holmint fd;
718a272653SPeter Holmint idx[N];
728a272653SPeter Holmchar file[] = "lockf4.file";
738a272653SPeter Holm
748a272653SPeter Holm#define TIMEOUT 600
758a272653SPeter Holm
768a272653SPeter Holmint64_t
778a272653SPeter Holmadd(int ix, int val)
788a272653SPeter Holm{
798a272653SPeter Holm	int64_t v;
808a272653SPeter Holm	off_t offset;
818a272653SPeter Holm	time_t start;
828a272653SPeter Holm	int r;
838a272653SPeter Holm
848a272653SPeter Holm	offset = ix * sizeof(v);
858a272653SPeter Holm	if (lseek(fd, offset, SEEK_SET) == -1)
868a272653SPeter Holm		err(1, "lseek");
878a272653SPeter Holm	start = time(NULL);
888a272653SPeter Holm	while (lockf(fd, F_LOCK, sizeof(v)) == -1) {
898a272653SPeter Holm		if (errno != EDEADLK)
908a272653SPeter Holm			err(1, "lockf(%s, F_LOCK)", file);
918a272653SPeter Holm		if (time(NULL) - start > TIMEOUT)
928a272653SPeter Holm			errx(1, "lockf timedout");
938a272653SPeter Holm	}
948a272653SPeter Holm	v = 0;
958a272653SPeter Holm	r = read(fd, &v, sizeof(v));
968a272653SPeter Holm	if (r == 0)
978a272653SPeter Holm		v = 0;
988a272653SPeter Holm	else
998a272653SPeter Holm		if (r == -1)
1008a272653SPeter Holm			err(1, "read");
1018a272653SPeter Holm	v += val;
1028a272653SPeter Holm
1038a272653SPeter Holm	if (lseek(fd, offset, SEEK_SET) == -1)
1048a272653SPeter Holm		err(1, "lseek");
1058a272653SPeter Holm	if (write(fd, &v, sizeof(v)) < 0)
1068a272653SPeter Holm		err(1, "write");
1078a272653SPeter Holm	if (lseek(fd, offset, SEEK_SET) == -1)
1088a272653SPeter Holm		err(1, "lseek");
1098a272653SPeter Holm	if (lockf(fd, F_ULOCK, sizeof(v)) == -1)
1108a272653SPeter Holm		err(1, "lockf(%s, F_ULOCK)", file);
1118a272653SPeter Holm
1128a272653SPeter Holm	return (v);
1138a272653SPeter Holm}
1148a272653SPeter Holm
1158a272653SPeter Holmint
1168a272653SPeter Holmcheck(void)
1178a272653SPeter Holm{
1188a272653SPeter Holm	int64_t v;
1198a272653SPeter Holm	int i;
1208a272653SPeter Holm
1218a272653SPeter Holm	setproctitle("check");
1228a272653SPeter Holm	if (lseek(fd, 0, SEEK_SET) == -1)
1238a272653SPeter Holm		err(1, "lseek");
1248a272653SPeter Holm	for (i = 1; i < N; i++) {
1258a272653SPeter Holm		if (read(fd, &v, sizeof(v)) < 0)
1268a272653SPeter Holm			err(1, "read");
1278a272653SPeter Holm		if (v != 0)
1288a272653SPeter Holm			return (1);
1298a272653SPeter Holm	}
1308a272653SPeter Holm
1318a272653SPeter Holm	return (0);
1328a272653SPeter Holm}
1338a272653SPeter Holm
1348a272653SPeter Holmint
1358a272653SPeter Holmmain(void)
1368a272653SPeter Holm{
1378a272653SPeter Holm	int64_t r;
1388a272653SPeter Holm	int e, i, j, t;
1398a272653SPeter Holm	char help[80];
1408a272653SPeter Holm
1418a272653SPeter Holm	for (i = 1; i < N; i++)
1428a272653SPeter Holm		idx[i] = i;
1438a272653SPeter Holm
1448a272653SPeter Holm	for (i = 1; i < N; i++) {
1458a272653SPeter Holm		j = arc4random() % (N - 1) + 1;
1468a272653SPeter Holm		t = idx[i];
1478a272653SPeter Holm		idx[i] = idx[j];
1488a272653SPeter Holm		idx[j] = t;
1498a272653SPeter Holm	}
1508a272653SPeter Holm
1518a272653SPeter Holm	if ((fd = open(file, O_RDWR | O_CREAT, 0644)) == -1)
1528a272653SPeter Holm		err(1, "open(%s)", file);
1538a272653SPeter Holm
1548a272653SPeter Holm	add(0, 1);
1558a272653SPeter Holm	for (i = 1; i < N; i++) {
1568a272653SPeter Holm		if (i % 500 == 0) {
1578a272653SPeter Holm			snprintf(help, sizeof(help), "add %d%%", i * 100 / N);
1588a272653SPeter Holm			setproctitle("%s", help);
1598a272653SPeter Holm		}
1608a272653SPeter Holm		add(idx[i], 1);
1618a272653SPeter Holm	}
1628a272653SPeter Holm	for (i = 1; i < N; i++) {
1638a272653SPeter Holm		if (i % 500 == 0) {
1648a272653SPeter Holm			snprintf(help, sizeof(help), "sub %d%%", i * 100 / N);
1658a272653SPeter Holm			setproctitle("%s", help);
1668a272653SPeter Holm		}
1678a272653SPeter Holm		add(idx[i], -1);
1688a272653SPeter Holm	}
1698a272653SPeter Holm
1708a272653SPeter Holm	e = 0;
1718a272653SPeter Holm	if ((r = add(0, -1)) == 0) {
1728a272653SPeter Holm		e = check();
1738a272653SPeter Holm		if (e == 0)
1748a272653SPeter Holm			unlink(file);
1758a272653SPeter Holm		else
1768a272653SPeter Holm			fprintf(stderr, "FAIL\n");
1778a272653SPeter Holm	}
1788a272653SPeter Holm
1798a272653SPeter Holm	close(fd);
1808a272653SPeter Holm
1818a272653SPeter Holm	return (e);
1828a272653SPeter Holm}
183