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