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 /* Get various resource limits for the tests */
29
30 #include <sys/types.h>
31 #include <sys/sysctl.h>
32 #include <unistd.h>
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <fcntl.h>
36 #include <string.h>
37 #include <sys/stat.h>
38 #include <sys/statvfs.h>
39 #include <sys/param.h>
40 #include <sys/mount.h>
41 #include <kvm.h>
42 #include <vm/vm_param.h>
43 #include <errno.h>
44 #include <err.h>
45 #include <stdarg.h>
46 #include <libutil.h>
47
48 #include "stress.h"
49
50 static int lockfd;
51 static int dffd;
52 static int flags;
53 static char lockpath[128];
54 static char dfpath[128];
55
56 static int64_t
inodes(void)57 inodes(void)
58 {
59 char path[MAXPATHLEN+1];
60 struct statfs sbuf;
61 struct statvfs buf;
62
63 if (op->inodes != 0)
64 return (op->inodes);
65 if (getcwd(path, sizeof(path)) == NULL)
66 err(1, "getcwd()");
67
68 if (statfs(path, &sbuf) < 0)
69 err(1, "statfs(%s)", path);
70 if (statvfs(path, &buf) < 0)
71 err(1, "statvfs(%s)", path);
72 if (!strcmp(sbuf.f_fstypename, "msdosfs"))
73 buf.f_ffree = 9999;
74 flags = sbuf.f_flags & MNT_VISFLAGMASK;
75 if (op->verbose > 2)
76 printf("Free inodes on %s (%s): %jd\n", path, sbuf.f_mntonname,
77 (intmax_t)buf.f_ffree);
78 return (buf.f_ffree);
79 }
80
81 static int64_t
df(void)82 df(void)
83 {
84 char path[MAXPATHLEN+1];
85 struct statvfs buf;
86
87 if (op->kblocks != 0)
88 return (op->kblocks * (uint64_t)1024);
89
90 if (getcwd(path, sizeof(path)) == NULL)
91 err(1, "getcwd()");
92
93 if (statvfs(path, &buf) < 0)
94 err(1, "statfs(%s)", path);
95
96 if (buf.f_bavail > buf.f_blocks || buf.f_bavail < 0) {
97 warnx("Corrupt statfs(%s). f_bavail = %jd!", path,
98 (intmax_t)buf.f_bavail);
99 buf.f_bavail = 100;
100 }
101 if (op->verbose > 2)
102 printf("Free space on %s: %jd Mb\n", path,
103 (intmax_t)(buf.f_bavail * buf.f_bsize / 1024 / 1024));
104 return (buf.f_bavail * buf.f_bsize);
105 }
106
107
108 int64_t
swap(void)109 swap(void)
110 {
111 #error "please put the amount of free swap (in bytes) in sz" /* REMOVE HERE! */
112 /*
113 * Currently, DragonFly has no way of determining free swap as non-root
114 *
115 * Please remove the #error line above this comment and modify the
116 * line below it with the amount of free swap you have (in bytes).
117 */
118 int64_t sz = 4294967296LL; /* EDIT HERE! */
119
120 if (op->verbose > 2)
121 printf("Total free swap space %jd Mb\n",
122 (intmax_t)(sz / 1024 / 1024));
123
124 return (sz);
125 }
126
127 unsigned long
usermem(void)128 usermem(void)
129 {
130 unsigned long mem;
131 size_t nlen = sizeof(mem);
132
133 if (sysctlbyname("hw.usermem", &mem, &nlen, NULL, 0) == -1)
134 err(1, "sysctlbyname() %s:%d", __FILE__, __LINE__);
135
136 if (op->verbose > 2)
137 printf("Total free user memory %lu Mb\n",
138 mem / 1024 / 1024);
139
140 return (mem);
141 }
142
cleanupdf()143 void cleanupdf()
144 {
145 unlink(dfpath);
146 }
147
148 void
getdf(int64_t * block,int64_t * inode)149 getdf(int64_t *block, int64_t *inode)
150 {
151 int i, j;
152 char buf[128];
153
154 snprintf(lockpath, sizeof(lockpath), "%s/lock", op->cd);
155 for (j = 0; j < 10; j++) {
156 for (i = 0; i < 10000; i++) {
157 if ((lockfd = open(lockpath,
158 O_CREAT | O_TRUNC | O_WRONLY | O_EXCL, 0644)) != -1)
159 break;
160 usleep(10000); /* sleep 1/100 sec */
161 }
162 if (lockfd != -1)
163 break;
164 fprintf(stderr, "%s. Removing stale %s\n", getprogname(), lockpath);
165 unlink(lockpath);
166 }
167 if (lockfd == -1)
168 errx(1, "%s. Can not create %s\n", getprogname(), lockpath);
169 snprintf(dfpath, sizeof(dfpath), "%s/df", op->cd);
170 if ((dffd = open(dfpath, O_RDWR, 0644)) == -1) {
171 if ((dffd = open(dfpath,
172 O_CREAT | O_TRUNC | O_WRONLY, 0644)) == -1) {
173 unlink(lockpath);
174 err(1, "creat(%s) %s:%d", dfpath, __FILE__, __LINE__);
175 }
176 atexit(cleanupdf);
177 *block = df();
178 *inode = inodes();
179 snprintf(buf, sizeof(buf), "%jd %jd", *block, *inode);
180
181 if (write(dffd, buf, strlen(buf) + 1) != strlen(buf) +1)
182 err(1, "write df. %s:%d", __FILE__, __LINE__);
183 } else {
184 if (read(dffd, buf, sizeof(buf)) < 1) {
185 system("ls -l /tmp/stressX.control");
186 unlink(lockpath);
187 err(1, "read df. %s:%d", __FILE__, __LINE__);
188 }
189 sscanf(buf, "%jd %jd", block, inode);
190 }
191 close(dffd);
192 }
193
194
195 void
reservedf(int64_t blks,int64_t inos)196 reservedf(int64_t blks, int64_t inos)
197 {
198 char buf[128];
199 int64_t blocks, inodes;
200
201 if ((dffd = open(dfpath, O_RDWR, 0644)) == -1) {
202 warn("open(%s) %s:%d. %s", dfpath, __FILE__, __LINE__, getprogname());
203 goto err;
204 }
205 if (read(dffd, buf, sizeof(buf)) < 1) {
206 warn("read df. %s:%d", __FILE__, __LINE__);
207 goto err;
208 }
209 sscanf(buf, "%jd %jd", &blocks, &inodes);
210
211 if (op->verbose > 2)
212 printf("%-8s: reservefd(%9jdK, %6jd) out of (%9jdK, %6jd)\n",
213 getprogname(), blks/1024, inos, blocks/1024, inodes);
214 blocks -= blks;
215 inodes -= inos;
216
217 snprintf(buf, sizeof(buf), "%jd %jd", blocks, inodes);
218 if (blocks < 0 || inodes < 0)
219 printf("******************************** %s: %s\n", getprogname(), buf);
220 if (lseek(dffd, 0, 0) == -1)
221 err(1, "lseek. %s:%d", __FILE__, __LINE__);
222 if (write(dffd, buf, strlen(buf) + 1) != strlen(buf) +1)
223 warn("write df. %s:%d", __FILE__, __LINE__);
224 err:
225 close(dffd);
226 close(lockfd);
227 if (unlink(lockpath) == -1)
228 err(1, "unlink(%s)", lockpath);
229 }
230
231 /* The UFS2 soft update lag problem causes a lot of confusion, so for now add the err() function here */
232
233 static void
vpr(int code,const char * fmt,va_list ap)234 vpr(int code, const char *fmt, va_list ap)
235 {
236 char path[MAXPATHLEN+1];
237 char siz[5], ino[5];
238 int64_t s, i;
239
240 s = df();
241 i = inodes();
242
243 if (errno == ENOSPC && (flags & MNT_SOFTDEP) && (flags & MNT_QUOTA) == 0 &&
244 s > 100 && i > 100) {
245 if (getcwd(path, sizeof(path)) == NULL)
246 err(1, "getcwd()");
247
248 humanize_number(siz, sizeof(siz), s, "",
249 HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
250 humanize_number(ino, sizeof(ino), i, "",
251 HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
252
253 printf("A syscall has failed with ENOSPC even though free disk "
254 "space for %s is reported as %s and %s inodes.\n",
255 path, siz, ino);
256 }
257
258
259 fprintf(stderr, "%s: ", getprogname());
260 if (fmt != NULL) {
261 vfprintf(stderr, fmt, ap);
262 fprintf(stderr, ": ");
263 }
264 fprintf(stderr, "%s\n", strerror(code));
265 }
266
267 void
err(int eval,const char * fmt,...)268 err(int eval, const char *fmt, ...)
269 {
270 va_list ap;
271 int code = errno;
272
273 va_start(ap, fmt);
274 vpr(code, fmt, ap);
275 va_end(ap);
276 exit(eval);
277 }
278
279 void
warn(const char * fmt,...)280 warn(const char *fmt, ...)
281 {
282 va_list ap;
283 int code = errno;
284
285 va_start(ap, fmt);
286 vpr(code, fmt, ap);
287 va_end(ap);
288 }
289