18a272653SPeter Holm /*-
28a272653SPeter Holm  * Copyright (c) 2008 Peter Holm <pho@FreeBSD.org>
38a272653SPeter Holm  * All rights reserved.
48a272653SPeter Holm  *
58a272653SPeter Holm  * Redistribution and use in source and binary forms, with or without
68a272653SPeter Holm  * modification, are permitted provided that the following conditions
78a272653SPeter Holm  * are met:
88a272653SPeter Holm  * 1. Redistributions of source code must retain the above copyright
98a272653SPeter Holm  *    notice, this list of conditions and the following disclaimer.
108a272653SPeter Holm  * 2. Redistributions in binary form must reproduce the above copyright
118a272653SPeter Holm  *    notice, this list of conditions and the following disclaimer in the
128a272653SPeter Holm  *    documentation and/or other materials provided with the distribution.
138a272653SPeter Holm  *
148a272653SPeter Holm  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
158a272653SPeter Holm  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
168a272653SPeter Holm  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
178a272653SPeter Holm  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
188a272653SPeter Holm  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
198a272653SPeter Holm  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
208a272653SPeter Holm  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
218a272653SPeter Holm  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
228a272653SPeter Holm  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
238a272653SPeter Holm  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
248a272653SPeter Holm  * SUCH DAMAGE.
258a272653SPeter Holm  *
268a272653SPeter Holm  */
278a272653SPeter Holm 
288a272653SPeter Holm #include <sys/param.h>
298a272653SPeter Holm #include <sys/mount.h>
308a272653SPeter Holm #include <sys/stat.h>
318a272653SPeter Holm #include <err.h>
328a272653SPeter Holm #include <errno.h>
338a272653SPeter Holm #include <fcntl.h>
348a272653SPeter Holm #include <stdio.h>
358a272653SPeter Holm #include <stdlib.h>
368a272653SPeter Holm #include <unistd.h>
378a272653SPeter Holm 
388a272653SPeter Holm #include "stress.h"
398a272653SPeter Holm 
408a272653SPeter Holm static char path[128];
418a272653SPeter Holm static unsigned long size;
428a272653SPeter Holm 
438a272653SPeter Holm int
setup(int nb)448a272653SPeter Holm setup(int nb)
458a272653SPeter Holm {
468a272653SPeter Holm 	int64_t in;
478a272653SPeter Holm 	int64_t bl;
488a272653SPeter Holm 	int64_t reserve_in;
498a272653SPeter Holm 	int64_t reserve_bl;
508a272653SPeter Holm 	int pct;
518a272653SPeter Holm 
528a272653SPeter Holm 	umask(0);
538a272653SPeter Holm 	if (nb == 0) {
548a272653SPeter Holm 		getdf(&bl, &in);
558a272653SPeter Holm 		size = in / op->incarnations;
568a272653SPeter Holm 
578a272653SPeter Holm 		pct = 90;
588a272653SPeter Holm 		if (op->hog == 0)
598a272653SPeter Holm 			pct = random_int(1, 90);
608a272653SPeter Holm 		size = size / 100 * pct + 1;
618a272653SPeter Holm 
628a272653SPeter Holm 		if (size > 16000)
638a272653SPeter Holm 			size = 16000;	/* arbitrary limit number of files pr. dir */
648a272653SPeter Holm 
658a272653SPeter Holm 		/* Resource requirements: */
668a272653SPeter Holm 		while (size > 0) {
678a272653SPeter Holm 			reserve_in =  1 * size * op->incarnations + op->incarnations;
688a272653SPeter Holm 			reserve_bl = 26 * size * op->incarnations;
698a272653SPeter Holm 			if (reserve_bl <= bl && reserve_in <= in)
708a272653SPeter Holm 				break;
718a272653SPeter Holm 			size = size / 2;
728a272653SPeter Holm 		}
738a272653SPeter Holm 		if (size == 0)
748a272653SPeter Holm 			reserve_bl = reserve_in = 0;
758a272653SPeter Holm 
768a272653SPeter Holm 		if (op->verbose > 1)
778a272653SPeter Holm 			printf("symlink(size=%lu, incarnations=%d). Free(%jdk, %jd), reserve(%jdk, %jd)\n",
788a272653SPeter Holm 				size, op->incarnations, bl/1024, in, reserve_bl/1024, reserve_in);
798a272653SPeter Holm 		reservedf(reserve_bl, reserve_in);
808a272653SPeter Holm 		putval(size);
818a272653SPeter Holm 	} else {
828a272653SPeter Holm 		size = getval();
838a272653SPeter Holm 	}
848a272653SPeter Holm 	if (size == 0)
858a272653SPeter Holm 		exit(0);
868a272653SPeter Holm 
878a272653SPeter Holm 	sprintf(path,"%s.%05d", getprogname(), getpid());
888a272653SPeter Holm 	if (mkdir(path, 0770) < 0)
898a272653SPeter Holm 		err(1, "mkdir(%s), %s:%d", path, __FILE__, __LINE__);
908a272653SPeter Holm 
918a272653SPeter Holm 	if (chdir(path) == -1)
928a272653SPeter Holm 		err(1, "chdir(%s), %s:%d", path, __FILE__, __LINE__);
938a272653SPeter Holm 
948a272653SPeter Holm 	return (0);
958a272653SPeter Holm }
968a272653SPeter Holm 
978a272653SPeter Holm void
cleanup(void)988a272653SPeter Holm cleanup(void)
998a272653SPeter Holm {
1008a272653SPeter Holm 	if (path[0] != 0) {
1018a272653SPeter Holm 		(void)chdir("..");
1028a272653SPeter Holm 		if (rmdir(path) == -1) {
1038a272653SPeter Holm 			warn("rmdir(%s), %s:%d", path, __FILE__, __LINE__);
1048a272653SPeter Holm 		}
1058a272653SPeter Holm 	}
1068a272653SPeter Holm }
1078a272653SPeter Holm 
1088a272653SPeter Holm int
test(void)1098a272653SPeter Holm test(void)
1108a272653SPeter Holm {
1118a272653SPeter Holm 	pid_t pid;
1128a272653SPeter Holm 	int i, j, error = 0;
1138a272653SPeter Holm 	char file[128];
1148a272653SPeter Holm 
1158a272653SPeter Holm 	pid = getpid();
1168a272653SPeter Holm 	for (j = 0; j < (int)size && done_testing == 0; j++) {
1178a272653SPeter Holm 		sprintf(file,"p%05d.%05d", pid, j);
1188a272653SPeter Holm 		if (symlink("/tmp/not/there", file) == -1) {
1198a272653SPeter Holm 			if (errno != EINTR) {
1208a272653SPeter Holm 				warn("symlink(%s). %s.%d", file, __FILE__, __LINE__);
1218a272653SPeter Holm 				error = 1;
1228a272653SPeter Holm 				exit(1);
1238a272653SPeter Holm 				break;
1248a272653SPeter Holm 			}
1258a272653SPeter Holm 		}
1268a272653SPeter Holm 	}
1278a272653SPeter Holm 
1288a272653SPeter Holm 	for (i = --j; i >= 0; i--) {
1298a272653SPeter Holm 		sprintf(file,"p%05d.%05d", pid, i);
1308a272653SPeter Holm 		if (unlink(file) == -1)
1318a272653SPeter Holm 			err(3, "unlink(%s)", file);
1328a272653SPeter Holm 	}
1338a272653SPeter Holm 
1348a272653SPeter Holm 	if (error != 0)
1358a272653SPeter Holm 		exit(1);
1368a272653SPeter Holm 
1378a272653SPeter Holm 	return (0);
1388a272653SPeter Holm }
139