1 /*-
2 * Copyright (c) 2008 Peter Holm <pho@FreeBSD.org>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 *
26 */
27
28 /* Write and check read a file */
29
30 #include <stdio.h>
31 #include <stdlib.h>
32 #include <unistd.h>
33 #include <fcntl.h>
34 #include <sys/stat.h>
35 #include <sys/param.h>
36 #include <err.h>
37
38 #include "stress.h"
39
40 static char path[128];
41 static int starting_dir;
42 static unsigned long size;
43
44 #define MAXSIZE 256 * 1024
45
46 int
setup(int nb)47 setup(int nb)
48 {
49 int64_t bl;
50 int64_t in;
51 int64_t reserve_in;
52 int64_t reserve_bl;
53 int pct;
54
55 if (nb == 0) {
56 getdf(&bl, &in);
57 size = bl / op->incarnations / 1024;
58
59 pct = 90;
60 if (op->hog == 0)
61 pct = random_int(1, 90);
62 size = size / 100 * pct + 1;
63
64 if (size > MAXSIZE)
65 size = MAXSIZE; /* arbitrary limit size pr. incarnation */
66
67 /* Resource requirements: */
68 while (size > 0) {
69 reserve_in = 2 * op->incarnations + 1;
70 reserve_bl = size * 1024 * op->incarnations +
71 (512 * 1024 * op->incarnations) +
72 64 * 1024;
73 // printf("-- size = %lu, reserve(%jd, %jd)\n", size, reserve_bl/1024, reserve_in);
74 if (reserve_bl <= bl && reserve_in <= in)
75 break;
76 size = size - 1024;
77 }
78 if (size == 0)
79 reserve_bl = reserve_in = 0;
80
81 if (op->verbose > 1)
82 printf("rw(size=%lu, incarnations=%d). Free(%jdk, %jd), reserve(%jdk, %jd)\n",
83 size, op->incarnations, bl/1024, in, reserve_bl/1024, reserve_in);
84 reservedf(reserve_bl, reserve_in);
85 putval(size);
86 size = size * 1024;
87 } else {
88 size = getval();
89 size = size * 1024;
90 }
91
92 umask(0);
93 sprintf(path,"%s.%05d", getprogname(), getpid());
94 (void)mkdir(path, 0770);
95 if (chdir(path) == -1)
96 err(1, "chdir(%s), %s:%d", path, __FILE__, __LINE__);
97 if ((starting_dir = open(".", 0)) < 0)
98 err(1, ".");
99
100
101 return (0);
102 }
103
104 void
cleanup(void)105 cleanup(void)
106 {
107 if (size == 0)
108 return;
109 if (fchdir(starting_dir) == -1)
110 err(1, "fchdir()");
111 if (close(starting_dir) < 0)
112 err(1, "close(starting_dir:%d)", starting_dir);
113
114 (void)system("find . -delete");
115
116 if (chdir("..") == -1)
117 err(1, "chdir(..)");
118 if (rmdir(path) == -1)
119 err(1, "rmdir(%s), %s:%d", path, __FILE__, __LINE__);
120 size = 0;
121 }
122
123 int
test(void)124 test(void)
125 {
126 int buf[1024], index, to;
127 #ifdef TEST
128 int i;
129 #endif
130 int fd;
131 char file[128];
132
133
134 sprintf(file,"p%05d", getpid());
135 if ((fd = creat(file, 0660)) == -1)
136 err(1, "creat(%s)", file);
137
138 to = sizeof(buf);
139 index = 0;
140 while (index < size) {
141 if (index + to > size)
142 to = size - index;
143 #ifdef TEST
144 for (i = 0; i < to; i++)
145 buf[i] = index + i;
146 #endif
147 index += to;
148 if (write(fd, buf, to) != to)
149 err(1, "write(%s), %s:%d", file, __FILE__, __LINE__);
150 }
151 if (close(fd) == -1)
152 err(1, "close(%s), %s:%d", file, __FILE__, __LINE__);
153
154 if ((fd = open(file, O_RDONLY)) == -1)
155 err(1, "open(%s), %s:%d", file, __FILE__, __LINE__);
156
157 index = 0;
158 while (index < size && done_testing == 0) {
159 if (index + to > size)
160 to = size - index;
161 if (read(fd, buf, to) != to)
162 err(1, "rw read. %s.%d", __FILE__, __LINE__);
163 #ifdef TEST
164 for (i = 0; i < to; i++) {
165 if (buf[i] != index + i) {
166 fprintf(stderr,
167 "%s, pid %d: expected %d @ %d, got %d\n",
168 getprogname(), getpid(), index+i, index+i,
169 buf[i]);
170 exit(EXIT_FAILURE);
171 }
172 }
173 #endif
174 index += to;
175 }
176 if (close(fd) == -1)
177 err(1, "close(%s), %s:%d", file, __FILE__, __LINE__);
178 if (unlink(file) == -1)
179 err(1, "unlink(%s), %s:%d", file, __FILE__, __LINE__);
180 return (0);
181 }
182