18a272653SPeter Holm#!/bin/sh 28a272653SPeter Holm 38a272653SPeter Holm# 48a272653SPeter Holm# Copyright (c) 2014 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# Threaded producer-consumer test. 308a272653SPeter Holm 318a272653SPeter Holm. ../default.cfg 328a272653SPeter Holm 338a272653SPeter Holmexport LANG=C 348a272653SPeter Holmhere=`pwd` 358a272653SPeter Holmcd /tmp 368a272653SPeter Holmsed '1,/^EOF/d' < $here/$0 > pthread2.c 378a272653SPeter Holmmycc -o pthread2 -Wall -Wextra -O2 -g pthread2.c -lpthread || exit 1 388a272653SPeter Holmrm -f pthread2.c /tmp/pthread2.core 398a272653SPeter Holm 408a272653SPeter Holmlog=/tmp/pthread2.`date '+%Y%m%d-%H%M'` 418a272653SPeter Holmfor i in `jot 5`; do 428a272653SPeter Holm [ $i -eq 1 ] && echo "# `uname -v`" 438a272653SPeter Holm time sh -c ' 448a272653SPeter Holm for i in `jot 8`; do 458a272653SPeter Holm /tmp/pthread2 & 468a272653SPeter Holm done 478a272653SPeter Holm wait 488a272653SPeter Holm ' 498a272653SPeter Holmdone > $log 2>&1 508a272653SPeter Holmrm -f /tmp/pthread2 518a272653SPeter Holm 528a272653SPeter Holmif [ -n "$bench" ]; then 538a272653SPeter Holm pair=`ls /tmp/pthread2* | egrep "pthread2\.[0-9]{8}-" | sort | 548a272653SPeter Holm tail -2 | tr '\n' ' '` 558a272653SPeter Holm ministat -w 72 $pair 568a272653SPeter Holmelse 578a272653SPeter Holm rm -f $log 588a272653SPeter Holmfi 598a272653SPeter Holm 608a272653SPeter Holm# __thr_umutex_lock() may call abort(3) under VM pressure. 618a272653SPeter Holm[ -r /tmp/pthread2.core ] && echo FAIL 628a272653SPeter Holmexit 0 638a272653SPeter HolmEOF 648a272653SPeter Holm/* 658a272653SPeter Holm * Threaded producer-consumer test. 668a272653SPeter Holm * Loosly based on work by 678a272653SPeter Holm * Andrey Zonov (c) 2012 688a272653SPeter Holm */ 698a272653SPeter Holm 708a272653SPeter Holm#include <sys/types.h> 718a272653SPeter Holm#include <sys/stat.h> 728a272653SPeter Holm#include <sys/time.h> 738a272653SPeter Holm#include <sys/queue.h> 748a272653SPeter Holm#include <err.h> 758a272653SPeter Holm#include <errno.h> 768a272653SPeter Holm#include <fcntl.h> 778a272653SPeter Holm#include <pthread.h> 788a272653SPeter Holm#ifdef __FreeBSD__ 798a272653SPeter Holm#include <pthread_np.h> 808a272653SPeter Holm#define __NP__ 818a272653SPeter Holm#endif 828a272653SPeter Holm#include <sched.h> 838a272653SPeter Holm#include <signal.h> 848a272653SPeter Holm#include <stdio.h> 858a272653SPeter Holm#include <stdlib.h> 868a272653SPeter Holm#include <string.h> 878a272653SPeter Holm#include <time.h> 888a272653SPeter Holm#include <unistd.h> 898a272653SPeter Holm 908a272653SPeter Holm#define LOCK(x) plock(&x.mtx) 918a272653SPeter Holm#define UNLOCK(x) punlock(&x.mtx) 928a272653SPeter Holm#define SIGNAL(x) psig(&x.wait) 938a272653SPeter Holm#define WAIT(x) pwait(&x.wait, &x.mtx) 948a272653SPeter Holm 958a272653SPeter Holmlong ncreate, nrename, nunlink; 968a272653SPeter Holmint bench, max; 978a272653SPeter Holmchar *dirname1; 988a272653SPeter Holmchar *dirname2; 998a272653SPeter Holm 1008a272653SPeter Holmstruct file { 1018a272653SPeter Holm char *name; 1028a272653SPeter Holm STAILQ_ENTRY(file) next; 1038a272653SPeter Holm}; 1048a272653SPeter Holm 1058a272653SPeter Holmstruct files { 1068a272653SPeter Holm pthread_mutex_t mtx; 1078a272653SPeter Holm pthread_cond_t wait; 1088a272653SPeter Holm STAILQ_HEAD(, file) list; 1098a272653SPeter Holm}; 1108a272653SPeter Holm 1118a272653SPeter Holmstatic struct files newfiles; 1128a272653SPeter Holmstatic struct files renamedfiles; 1138a272653SPeter Holm 1148a272653SPeter Holm#define MAXQ 100000 /* Max create queue length */ 1158a272653SPeter Holm#define MESSAGES 10000000; 1168a272653SPeter Holm 1178a272653SPeter Holmstatic void 1188a272653SPeter Holmhand(int i __unused) { /* handler */ 1198a272653SPeter Holm fprintf(stderr, "max = %d, ncreate = %ld, nrename = %ld, nunlink = %ld\n", 1208a272653SPeter Holm max, ncreate, nrename, nunlink); 1218a272653SPeter Holm} 1228a272653SPeter Holm 1238a272653SPeter Holmstatic void 1248a272653SPeter Holmahand(int i __unused) { /* handler */ 1258a272653SPeter Holm fprintf(stderr, "FAIL\n"); 1268a272653SPeter Holm hand(0); 1278a272653SPeter Holm _exit(0); 1288a272653SPeter Holm} 1298a272653SPeter Holm 1308a272653SPeter Holmvoid 1318a272653SPeter Holmplock(pthread_mutex_t *l) 1328a272653SPeter Holm{ 1338a272653SPeter Holm int rc; 1348a272653SPeter Holm 1358a272653SPeter Holm if ((rc = pthread_mutex_lock(l)) != 0) 1368a272653SPeter Holm errc(1, rc, "pthread_mutex_lock"); 1378a272653SPeter Holm} 1388a272653SPeter Holm 1398a272653SPeter Holmvoid 1408a272653SPeter Holmpunlock(pthread_mutex_t *l) 1418a272653SPeter Holm{ 1428a272653SPeter Holm int rc; 1438a272653SPeter Holm 1448a272653SPeter Holm if ((rc = pthread_mutex_unlock(l)) != 0) 1458a272653SPeter Holm errc(1, rc, "pthread_mutex_unlock"); 1468a272653SPeter Holm} 1478a272653SPeter Holm 1488a272653SPeter Holmvoid 1498a272653SPeter Holmpsig(pthread_cond_t *c) 1508a272653SPeter Holm{ 1518a272653SPeter Holm int rc; 1528a272653SPeter Holm 1538a272653SPeter Holm if ((rc = pthread_cond_signal(c)) != 0) 1548a272653SPeter Holm errc(1, rc, "pthread_cond_signal"); 1558a272653SPeter Holm} 1568a272653SPeter Holm 1578a272653SPeter Holmvoid 1588a272653SPeter Holmpwait(pthread_cond_t *c, pthread_mutex_t *l) 1598a272653SPeter Holm{ 1608a272653SPeter Holm int rc; 1618a272653SPeter Holm 1628a272653SPeter Holm if ((rc = pthread_cond_wait(c, l)) != 0) 1638a272653SPeter Holm errc(1, rc, "pthread_cond_wait"); 1648a272653SPeter Holm} 1658a272653SPeter Holm 1668a272653SPeter Holmvoid * 1678a272653SPeter Holmloop_create(void *arg __unused) 1688a272653SPeter Holm{ 1698a272653SPeter Holm int i; 1708a272653SPeter Holm struct file *file; 1718a272653SPeter Holm 1728a272653SPeter Holm#ifdef __NP__ 1738a272653SPeter Holm pthread_set_name_np(pthread_self(), __func__); 1748a272653SPeter Holm#endif 1758a272653SPeter Holm 1768a272653SPeter Holm for (i = 0; i < max; i++) { 1778a272653SPeter Holm file = malloc(sizeof(*file)); 1788a272653SPeter Holm asprintf(&file->name, "%s/filename_too-long:%d", dirname1, i); 1798a272653SPeter Holm LOCK(newfiles); 1808a272653SPeter Holm STAILQ_INSERT_TAIL(&newfiles.list, file, next); 1818a272653SPeter Holm ncreate++; 1828a272653SPeter Holm UNLOCK(newfiles); 1838a272653SPeter Holm SIGNAL(newfiles); 1848a272653SPeter Holm if (ncreate - nrename > MAXQ) 1858a272653SPeter Holm usleep(400); 1868a272653SPeter Holm } 1878a272653SPeter Holm return (NULL); 1888a272653SPeter Holm} 1898a272653SPeter Holm 1908a272653SPeter Holmvoid * 1918a272653SPeter Holmloop_rename(void *arg __unused) 1928a272653SPeter Holm{ 1938a272653SPeter Holm char *filename, *newname; 1948a272653SPeter Holm struct file *file; 1958a272653SPeter Holm 1968a272653SPeter Holm#ifdef __NP__ 1978a272653SPeter Holm pthread_set_name_np(pthread_self(), __func__); 1988a272653SPeter Holm#endif 1998a272653SPeter Holm 2008a272653SPeter Holm while (nrename < max) { 2018a272653SPeter Holm LOCK(newfiles); 2028a272653SPeter Holm while (STAILQ_EMPTY(&newfiles.list)) { 2038a272653SPeter Holm WAIT(newfiles); 2048a272653SPeter Holm } 2058a272653SPeter Holm file = STAILQ_FIRST(&newfiles.list); 2068a272653SPeter Holm STAILQ_REMOVE_HEAD(&newfiles.list, next); 2078a272653SPeter Holm UNLOCK(newfiles); 2088a272653SPeter Holm filename = strrchr(file->name, '/'); 2098a272653SPeter Holm asprintf(&newname, "%s/%s", dirname2, filename); 2108a272653SPeter Holm nrename++; 2118a272653SPeter Holm free(file->name); 2128a272653SPeter Holm file->name = newname; 2138a272653SPeter Holm LOCK(renamedfiles); 2148a272653SPeter Holm STAILQ_INSERT_TAIL(&renamedfiles.list, file, next); 2158a272653SPeter Holm UNLOCK(renamedfiles); 2168a272653SPeter Holm SIGNAL(renamedfiles); 2178a272653SPeter Holm } 2188a272653SPeter Holm return (NULL); 2198a272653SPeter Holm} 2208a272653SPeter Holm 2218a272653SPeter Holmvoid * 2228a272653SPeter Holmloop_unlink(void *arg __unused) 2238a272653SPeter Holm{ 2248a272653SPeter Holm struct file *file; 2258a272653SPeter Holm 2268a272653SPeter Holm#ifdef __NP__ 2278a272653SPeter Holm pthread_set_name_np(pthread_self(), __func__); 2288a272653SPeter Holm#endif 2298a272653SPeter Holm 2308a272653SPeter Holm while (nunlink < max) { 2318a272653SPeter Holm LOCK(renamedfiles); 2328a272653SPeter Holm while (STAILQ_EMPTY(&renamedfiles.list)) { 2338a272653SPeter Holm WAIT(renamedfiles); 2348a272653SPeter Holm } 2358a272653SPeter Holm file = STAILQ_FIRST(&renamedfiles.list); 2368a272653SPeter Holm STAILQ_REMOVE_HEAD(&renamedfiles.list, next); 2378a272653SPeter Holm nunlink++; 2388a272653SPeter Holm UNLOCK(renamedfiles); 2398a272653SPeter Holm free(file->name); 2408a272653SPeter Holm free(file); 2418a272653SPeter Holm } 2428a272653SPeter Holm return (NULL); 2438a272653SPeter Holm} 2448a272653SPeter Holm 2458a272653SPeter Holmint 2468a272653SPeter Holmmain(void) 2478a272653SPeter Holm{ 2488a272653SPeter Holm int i; 2498a272653SPeter Holm int rc; 2508a272653SPeter Holm pthread_t tid[3]; 2518a272653SPeter Holm 2528a272653SPeter Holm bench = getenv("bench") != NULL; 2538a272653SPeter Holm asprintf(&dirname1, "%s.1", "f1"); 2548a272653SPeter Holm asprintf(&dirname2, "%s.2", "f2"); 2558a272653SPeter Holm max = MESSAGES; 2568a272653SPeter Holm 2578a272653SPeter Holm STAILQ_INIT(&newfiles.list); 2588a272653SPeter Holm STAILQ_INIT(&renamedfiles.list); 2598a272653SPeter Holm 2608a272653SPeter Holm if ((rc = pthread_mutex_init(&newfiles.mtx, NULL)) != 0) 2618a272653SPeter Holm errc(1, rc, "pthread_mutex_init()"); 2628a272653SPeter Holm if ((rc = pthread_cond_init(&newfiles.wait, NULL)) != 0) 2638a272653SPeter Holm errc(1, rc, "pthread_cond_init()"); 2648a272653SPeter Holm if ((rc = pthread_mutex_init(&renamedfiles.mtx, NULL)) != 0) 2658a272653SPeter Holm errc(1, rc, "pthread_mutex_init()"); 2668a272653SPeter Holm if ((rc = pthread_cond_init(&renamedfiles.wait, NULL)) != 0) 2678a272653SPeter Holm errc(1, rc, "pthread_cond_init()"); 2688a272653SPeter Holm 2698a272653SPeter Holm signal(SIGINFO, hand); 2708a272653SPeter Holm signal(SIGALRM, ahand); 2718a272653SPeter Holm alarm(300); 2728a272653SPeter Holm if ((rc = pthread_create(&tid[0], NULL, loop_create, NULL)) != 0) 2738a272653SPeter Holm errc(1, rc, "pthread_create()"); 2748a272653SPeter Holm if ((rc = pthread_create(&tid[1], NULL, loop_rename, NULL)) != 0) 2758a272653SPeter Holm errc(1, rc, "pthread_create()"); 2768a272653SPeter Holm if ((rc = pthread_create(&tid[2], NULL, loop_unlink, NULL)) != 0) 2778a272653SPeter Holm errc(1, rc, "pthread_create()"); 2788a272653SPeter Holm 2798a272653SPeter Holm for (i = 0; i < 3; i++) { 2808a272653SPeter Holm if ((rc = pthread_join(tid[i], NULL)) != 0) 2818a272653SPeter Holm errc(1, rc, "pthread_join(%d)", i); 2828a272653SPeter Holm } 2838a272653SPeter Holm 2848a272653SPeter Holm if ((rc = pthread_mutex_destroy(&newfiles.mtx)) != 0) 2858a272653SPeter Holm errc(1, rc, "pthread_mutex_destroy(newfiles)"); 2868a272653SPeter Holm if ((rc = pthread_cond_destroy(&newfiles.wait)) != 0) 2878a272653SPeter Holm errc(1, rc, "pthread_cond_destroy(newfiles)"); 2888a272653SPeter Holm if ((rc = pthread_mutex_destroy(&renamedfiles.mtx)) != 0) 2898a272653SPeter Holm errc(1, rc, "pthread_mutex_destroy(renamedfiles)"); 2908a272653SPeter Holm if ((rc = pthread_cond_destroy(&renamedfiles.wait)) != 0) 2918a272653SPeter Holm errc(1, rc, "pthread_cond_destroy(renamedfiles)"); 2928a272653SPeter Holm free(dirname1); 2938a272653SPeter Holm free(dirname2); 2948a272653SPeter Holm 2958a272653SPeter Holm return (0); 2968a272653SPeter Holm} 297