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