xref: /onnv-gate/usr/src/cmd/ztest/ztest.c (revision 1485)
1789Sahrens /*
2789Sahrens  * CDDL HEADER START
3789Sahrens  *
4789Sahrens  * The contents of this file are subject to the terms of the
5*1485Slling  * Common Development and Distribution License (the "License").
6*1485Slling  * You may not use this file except in compliance with the License.
7789Sahrens  *
8789Sahrens  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9789Sahrens  * or http://www.opensolaris.org/os/licensing.
10789Sahrens  * See the License for the specific language governing permissions
11789Sahrens  * and limitations under the License.
12789Sahrens  *
13789Sahrens  * When distributing Covered Code, include this CDDL HEADER in each
14789Sahrens  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15789Sahrens  * If applicable, add the following below this CDDL HEADER, with the
16789Sahrens  * fields enclosed by brackets "[]" replaced with your own identifying
17789Sahrens  * information: Portions Copyright [yyyy] [name of copyright owner]
18789Sahrens  *
19789Sahrens  * CDDL HEADER END
20789Sahrens  */
21789Sahrens /*
22*1485Slling  * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
23789Sahrens  * Use is subject to license terms.
24789Sahrens  */
25789Sahrens 
26789Sahrens #pragma ident	"%Z%%M%	%I%	%E% SMI"
27789Sahrens 
28789Sahrens /*
29789Sahrens  * The objective of this program is to provide a DMU/ZAP/SPA stress test
30789Sahrens  * that runs entirely in userland, is easy to use, and easy to extend.
31789Sahrens  *
32789Sahrens  * The overall design of the ztest program is as follows:
33789Sahrens  *
34789Sahrens  * (1) For each major functional area (e.g. adding vdevs to a pool,
35789Sahrens  *     creating and destroying datasets, reading and writing objects, etc)
36789Sahrens  *     we have a simple routine to test that functionality.  These
37789Sahrens  *     individual routines do not have to do anything "stressful".
38789Sahrens  *
39789Sahrens  * (2) We turn these simple functionality tests into a stress test by
40789Sahrens  *     running them all in parallel, with as many threads as desired,
41789Sahrens  *     and spread across as many datasets, objects, and vdevs as desired.
42789Sahrens  *
43789Sahrens  * (3) While all this is happening, we inject faults into the pool to
44789Sahrens  *     verify that self-healing data really works.
45789Sahrens  *
46789Sahrens  * (4) Every time we open a dataset, we change its checksum and compression
47789Sahrens  *     functions.  Thus even individual objects vary from block to block
48789Sahrens  *     in which checksum they use and whether they're compressed.
49789Sahrens  *
50789Sahrens  * (5) To verify that we never lose on-disk consistency after a crash,
51789Sahrens  *     we run the entire test in a child of the main process.
52789Sahrens  *     At random times, the child self-immolates with a SIGKILL.
53789Sahrens  *     This is the software equivalent of pulling the power cord.
54789Sahrens  *     The parent then runs the test again, using the existing
55789Sahrens  *     storage pool, as many times as desired.
56789Sahrens  *
57789Sahrens  * (6) To verify that we don't have future leaks or temporal incursions,
58789Sahrens  *     many of the functional tests record the transaction group number
59789Sahrens  *     as part of their data.  When reading old data, they verify that
60789Sahrens  *     the transaction group number is less than the current, open txg.
61789Sahrens  *     If you add a new test, please do this if applicable.
62789Sahrens  *
63789Sahrens  * When run with no arguments, ztest runs for about five minutes and
64789Sahrens  * produces no output if successful.  To get a little bit of information,
65789Sahrens  * specify -V.  To get more information, specify -VV, and so on.
66789Sahrens  *
67789Sahrens  * To turn this into an overnight stress test, use -T to specify run time.
68789Sahrens  *
69789Sahrens  * You can ask more more vdevs [-v], datasets [-d], or threads [-t]
70789Sahrens  * to increase the pool capacity, fanout, and overall stress level.
71789Sahrens  *
72789Sahrens  * The -N(okill) option will suppress kills, so each child runs to completion.
73789Sahrens  * This can be useful when you're trying to distinguish temporal incursions
74789Sahrens  * from plain old race conditions.
75789Sahrens  */
76789Sahrens 
77789Sahrens #include <sys/zfs_context.h>
78789Sahrens #include <sys/spa.h>
79789Sahrens #include <sys/dmu.h>
80789Sahrens #include <sys/txg.h>
81789Sahrens #include <sys/zap.h>
82789Sahrens #include <sys/dmu_traverse.h>
83789Sahrens #include <sys/dmu_objset.h>
84789Sahrens #include <sys/poll.h>
85789Sahrens #include <sys/stat.h>
86789Sahrens #include <sys/time.h>
87789Sahrens #include <sys/wait.h>
88789Sahrens #include <sys/mman.h>
89789Sahrens #include <sys/resource.h>
90789Sahrens #include <sys/zio.h>
91789Sahrens #include <sys/zio_checksum.h>
92789Sahrens #include <sys/zio_compress.h>
93789Sahrens #include <sys/zil.h>
94789Sahrens #include <sys/vdev_impl.h>
95789Sahrens #include <sys/spa_impl.h>
96789Sahrens #include <sys/dsl_prop.h>
97789Sahrens #include <sys/refcount.h>
98789Sahrens #include <stdio.h>
99789Sahrens #include <stdlib.h>
100789Sahrens #include <unistd.h>
101789Sahrens #include <signal.h>
102789Sahrens #include <umem.h>
103789Sahrens #include <dlfcn.h>
104789Sahrens #include <ctype.h>
105789Sahrens #include <math.h>
106789Sahrens #include <sys/fs/zfs.h>
107789Sahrens 
108789Sahrens static char cmdname[] = "ztest";
109789Sahrens static char *zopt_pool = cmdname;
110789Sahrens 
111789Sahrens static uint64_t zopt_vdevs = 5;
112789Sahrens static uint64_t zopt_vdevtime;
113789Sahrens static int zopt_mirrors = 2;
114789Sahrens static int zopt_raidz = 4;
115789Sahrens static size_t zopt_vdev_size = SPA_MINDEVSIZE;
116789Sahrens static int zopt_dirs = 7;
117789Sahrens static int zopt_threads = 23;
118789Sahrens static uint64_t zopt_passtime = 60;	/* 60 seconds */
119789Sahrens static uint64_t zopt_killrate = 70;	/* 70% kill rate */
120789Sahrens static int zopt_verbose = 0;
121789Sahrens static int zopt_init = 1;
122789Sahrens static char *zopt_dir = "/tmp";
123789Sahrens static uint64_t zopt_time = 300;	/* 5 minutes */
124789Sahrens static int zopt_maxfaults;
125789Sahrens 
126789Sahrens typedef struct ztest_args {
127789Sahrens 	char		*za_pool;
128789Sahrens 	objset_t	*za_os;
129789Sahrens 	zilog_t		*za_zilog;
130789Sahrens 	thread_t	za_thread;
131789Sahrens 	uint64_t	za_instance;
132789Sahrens 	uint64_t	za_random;
133789Sahrens 	uint64_t	za_diroff;
134789Sahrens 	uint64_t	za_diroff_shared;
135789Sahrens 	hrtime_t	za_start;
136789Sahrens 	hrtime_t	za_stop;
137789Sahrens 	hrtime_t	za_kill;
138789Sahrens 	traverse_handle_t *za_th;
139789Sahrens } ztest_args_t;
140789Sahrens 
141789Sahrens typedef void ztest_func_t(ztest_args_t *);
142789Sahrens 
143789Sahrens /*
144789Sahrens  * Note: these aren't static because we want dladdr() to work.
145789Sahrens  */
146789Sahrens ztest_func_t ztest_dmu_read_write;
147789Sahrens ztest_func_t ztest_dmu_write_parallel;
148789Sahrens ztest_func_t ztest_dmu_object_alloc_free;
149789Sahrens ztest_func_t ztest_zap;
150789Sahrens ztest_func_t ztest_zap_parallel;
151789Sahrens ztest_func_t ztest_traverse;
152789Sahrens ztest_func_t ztest_dsl_prop_get_set;
153789Sahrens ztest_func_t ztest_dmu_objset_create_destroy;
154789Sahrens ztest_func_t ztest_dmu_snapshot_create_destroy;
155789Sahrens ztest_func_t ztest_spa_create_destroy;
156789Sahrens ztest_func_t ztest_fault_inject;
157789Sahrens ztest_func_t ztest_vdev_attach_detach;
158789Sahrens ztest_func_t ztest_vdev_LUN_growth;
159789Sahrens ztest_func_t ztest_vdev_add_remove;
160789Sahrens ztest_func_t ztest_scrub;
161789Sahrens ztest_func_t ztest_spa_rename;
162789Sahrens 
163789Sahrens typedef struct ztest_info {
164789Sahrens 	ztest_func_t	*zi_func;	/* test function */
165789Sahrens 	uint64_t	*zi_interval;	/* execute every <interval> seconds */
166789Sahrens 	uint64_t	zi_calls;	/* per-pass count */
167789Sahrens 	uint64_t	zi_call_time;	/* per-pass time */
168789Sahrens 	uint64_t	zi_call_total;	/* cumulative total */
169789Sahrens 	uint64_t	zi_call_target;	/* target cumulative total */
170789Sahrens } ztest_info_t;
171789Sahrens 
172789Sahrens uint64_t zopt_always = 0;		/* all the time */
173789Sahrens uint64_t zopt_often = 1;		/* every second */
174789Sahrens uint64_t zopt_sometimes = 10;		/* every 10 seconds */
175789Sahrens uint64_t zopt_rarely = 60;		/* every 60 seconds */
176789Sahrens 
177789Sahrens ztest_info_t ztest_info[] = {
178789Sahrens 	{ ztest_dmu_read_write,			&zopt_always	},
179789Sahrens 	{ ztest_dmu_write_parallel,		&zopt_always	},
180789Sahrens 	{ ztest_dmu_object_alloc_free,		&zopt_always	},
181789Sahrens 	{ ztest_zap,				&zopt_always	},
182789Sahrens 	{ ztest_zap_parallel,			&zopt_always	},
183789Sahrens 	{ ztest_traverse,			&zopt_often	},
184789Sahrens 	{ ztest_dsl_prop_get_set,		&zopt_sometimes	},
185789Sahrens 	{ ztest_dmu_objset_create_destroy,	&zopt_sometimes	},
186789Sahrens 	{ ztest_dmu_snapshot_create_destroy,	&zopt_sometimes	},
187789Sahrens 	{ ztest_spa_create_destroy,		&zopt_sometimes	},
188789Sahrens 	{ ztest_fault_inject,			&zopt_sometimes	},
189789Sahrens 	{ ztest_spa_rename,			&zopt_rarely	},
190789Sahrens 	{ ztest_vdev_attach_detach,		&zopt_rarely	},
191789Sahrens 	{ ztest_vdev_LUN_growth,		&zopt_rarely	},
192789Sahrens 	{ ztest_vdev_add_remove,		&zopt_vdevtime	},
193789Sahrens 	{ ztest_scrub,				&zopt_vdevtime	},
194789Sahrens };
195789Sahrens 
196789Sahrens #define	ZTEST_FUNCS	(sizeof (ztest_info) / sizeof (ztest_info_t))
197789Sahrens 
198789Sahrens #define	ZTEST_SYNC_LOCKS	16
199789Sahrens 
200789Sahrens /*
201789Sahrens  * Stuff we need to share writably between parent and child.
202789Sahrens  */
203789Sahrens typedef struct ztest_shared {
204789Sahrens 	mutex_t		zs_vdev_lock;
205789Sahrens 	rwlock_t	zs_name_lock;
206789Sahrens 	uint64_t	zs_vdev_primaries;
207789Sahrens 	uint64_t	zs_enospc_count;
208789Sahrens 	hrtime_t	zs_start_time;
209789Sahrens 	hrtime_t	zs_stop_time;
210789Sahrens 	uint64_t	zs_alloc;
211789Sahrens 	uint64_t	zs_space;
212789Sahrens 	ztest_info_t	zs_info[ZTEST_FUNCS];
213789Sahrens 	mutex_t		zs_sync_lock[ZTEST_SYNC_LOCKS];
214789Sahrens 	uint64_t	zs_seq[ZTEST_SYNC_LOCKS];
215789Sahrens } ztest_shared_t;
216789Sahrens 
217789Sahrens typedef struct ztest_block_tag {
218789Sahrens 	uint64_t	bt_objset;
219789Sahrens 	uint64_t	bt_object;
220789Sahrens 	uint64_t	bt_offset;
221789Sahrens 	uint64_t	bt_txg;
222789Sahrens 	uint64_t	bt_thread;
223789Sahrens 	uint64_t	bt_seq;
224789Sahrens } ztest_block_tag_t;
225789Sahrens 
226789Sahrens static char ztest_dev_template[] = "%s/%s.%llua";
227789Sahrens static ztest_shared_t *ztest_shared;
228789Sahrens 
229789Sahrens static int ztest_random_fd;
230789Sahrens static int ztest_dump_core = 1;
231789Sahrens 
232789Sahrens extern uint64_t zio_gang_bang;
233789Sahrens 
234789Sahrens #define	ZTEST_DIROBJ		1
235789Sahrens #define	ZTEST_MICROZAP_OBJ	2
236789Sahrens #define	ZTEST_FATZAP_OBJ	3
237789Sahrens 
238789Sahrens #define	ZTEST_DIROBJ_BLOCKSIZE	(1 << 10)
239789Sahrens #define	ZTEST_DIRSIZE		256
240789Sahrens 
241789Sahrens /*
242789Sahrens  * These libumem hooks provide a reasonable set of defaults for the allocator's
243789Sahrens  * debugging facilities.
244789Sahrens  */
245789Sahrens const char *
246789Sahrens _umem_debug_init()
247789Sahrens {
248789Sahrens 	return ("default,verbose"); /* $UMEM_DEBUG setting */
249789Sahrens }
250789Sahrens 
251789Sahrens const char *
252789Sahrens _umem_logging_init(void)
253789Sahrens {
254789Sahrens 	return ("fail,contents"); /* $UMEM_LOGGING setting */
255789Sahrens }
256789Sahrens 
257789Sahrens #define	FATAL_MSG_SZ	1024
258789Sahrens 
259789Sahrens char *fatal_msg;
260789Sahrens 
261789Sahrens static void
262789Sahrens fatal(int do_perror, char *message, ...)
263789Sahrens {
264789Sahrens 	va_list args;
265789Sahrens 	int save_errno = errno;
266789Sahrens 	char buf[FATAL_MSG_SZ];
267789Sahrens 
268789Sahrens 	(void) fflush(stdout);
269789Sahrens 
270789Sahrens 	va_start(args, message);
271789Sahrens 	(void) sprintf(buf, "ztest: ");
272789Sahrens 	/* LINTED */
273789Sahrens 	(void) vsprintf(buf + strlen(buf), message, args);
274789Sahrens 	va_end(args);
275789Sahrens 	if (do_perror) {
276789Sahrens 		(void) snprintf(buf + strlen(buf), FATAL_MSG_SZ - strlen(buf),
277789Sahrens 		    ": %s", strerror(save_errno));
278789Sahrens 	}
279789Sahrens 	(void) fprintf(stderr, "%s\n", buf);
280789Sahrens 	fatal_msg = buf;			/* to ease debugging */
281789Sahrens 	if (ztest_dump_core)
282789Sahrens 		abort();
283789Sahrens 	exit(3);
284789Sahrens }
285789Sahrens 
286789Sahrens static int
287789Sahrens str2shift(const char *buf)
288789Sahrens {
289789Sahrens 	const char *ends = "BKMGTPEZ";
290789Sahrens 	int i;
291789Sahrens 
292789Sahrens 	if (buf[0] == '\0')
293789Sahrens 		return (0);
294789Sahrens 	for (i = 0; i < strlen(ends); i++) {
295789Sahrens 		if (toupper(buf[0]) == ends[i])
296789Sahrens 			break;
297789Sahrens 	}
298789Sahrens 	if (i == strlen(ends))
299789Sahrens 		fatal(0, "invalid bytes suffix: %s", buf);
300789Sahrens 	if (buf[1] == '\0' || (toupper(buf[1]) == 'B' && buf[2] == '\0')) {
301789Sahrens 		return (10*i);
302789Sahrens 	}
303789Sahrens 	fatal(0, "invalid bytes suffix: %s", buf);
304789Sahrens 	return (-1);
305789Sahrens }
306789Sahrens 
307789Sahrens static uint64_t
308789Sahrens nicenumtoull(const char *buf)
309789Sahrens {
310789Sahrens 	char *end;
311789Sahrens 	uint64_t val;
312789Sahrens 
313789Sahrens 	val = strtoull(buf, &end, 0);
314789Sahrens 	if (end == buf) {
315789Sahrens 		fatal(0, "bad numeric value: %s", buf);
316789Sahrens 	} else if (end[0] == '.') {
317789Sahrens 		double fval = strtod(buf, &end);
318789Sahrens 		fval *= pow(2, str2shift(end));
319789Sahrens 		if (fval > UINT64_MAX)
320789Sahrens 			fatal(0, "value too large: %s", buf);
321789Sahrens 		val = (uint64_t)fval;
322789Sahrens 	} else {
323789Sahrens 		int shift = str2shift(end);
324789Sahrens 		if (shift >= 64 || (val << shift) >> shift != val)
325789Sahrens 			fatal(0, "value too large: %s", buf);
326789Sahrens 		val <<= shift;
327789Sahrens 	}
328789Sahrens 	return (val);
329789Sahrens }
330789Sahrens 
331789Sahrens static void
332789Sahrens usage(void)
333789Sahrens {
334789Sahrens 	char nice_vdev_size[10];
335789Sahrens 	char nice_gang_bang[10];
336789Sahrens 
337789Sahrens 	nicenum(zopt_vdev_size, nice_vdev_size);
338789Sahrens 	nicenum(zio_gang_bang, nice_gang_bang);
339789Sahrens 
340789Sahrens 	(void) printf("Usage: %s\n"
341789Sahrens 	    "\t[-v vdevs (default: %llu)]\n"
342789Sahrens 	    "\t[-s size_of_each_vdev (default: %s)]\n"
343789Sahrens 	    "\t[-m mirror_copies (default: %d)]\n"
344789Sahrens 	    "\t[-r raidz_disks (default: %d)]\n"
345789Sahrens 	    "\t[-d datasets (default: %d)]\n"
346789Sahrens 	    "\t[-t threads (default: %d)]\n"
347789Sahrens 	    "\t[-g gang_block_threshold (default: %s)]\n"
348789Sahrens 	    "\t[-i initialize pool i times (default: %d)]\n"
349789Sahrens 	    "\t[-k kill percentage (default: %llu%%)]\n"
350789Sahrens 	    "\t[-p pool_name (default: %s)]\n"
351789Sahrens 	    "\t[-f file directory for vdev files (default: %s)]\n"
352789Sahrens 	    "\t[-V(erbose)] (use multiple times for ever more blather)\n"
353789Sahrens 	    "\t[-E(xisting)] (use existing pool instead of creating new one\n"
354789Sahrens 	    "\t[-I(mport)] (discover and import existing pools)\n"
355789Sahrens 	    "\t[-T time] total run time (default: %llu sec)\n"
356789Sahrens 	    "\t[-P passtime] time per pass (default: %llu sec)\n"
357789Sahrens 	    "",
358789Sahrens 	    cmdname,
359789Sahrens 	    (u_longlong_t)zopt_vdevs,		/* -v */
360789Sahrens 	    nice_vdev_size,			/* -s */
361789Sahrens 	    zopt_mirrors,			/* -m */
362789Sahrens 	    zopt_raidz,				/* -r */
363789Sahrens 	    zopt_dirs,			/* -d */
364789Sahrens 	    zopt_threads,			/* -t */
365789Sahrens 	    nice_gang_bang,			/* -g */
366789Sahrens 	    zopt_init,				/* -i */
367789Sahrens 	    (u_longlong_t)zopt_killrate,	/* -k */
368789Sahrens 	    zopt_pool,				/* -p */
369789Sahrens 	    zopt_dir,				/* -f */
370789Sahrens 	    (u_longlong_t)zopt_time,		/* -T */
371789Sahrens 	    (u_longlong_t)zopt_passtime);	/* -P */
372789Sahrens 	exit(1);
373789Sahrens }
374789Sahrens 
375789Sahrens static uint64_t
376789Sahrens ztest_random(uint64_t range)
377789Sahrens {
378789Sahrens 	uint64_t r;
379789Sahrens 
380789Sahrens 	if (range == 0)
381789Sahrens 		return (0);
382789Sahrens 
383789Sahrens 	if (read(ztest_random_fd, &r, sizeof (r)) != sizeof (r))
384789Sahrens 		fatal(1, "short read from /dev/urandom");
385789Sahrens 
386789Sahrens 	return (r % range);
387789Sahrens }
388789Sahrens 
389789Sahrens static void
390789Sahrens ztest_record_enospc(char *s)
391789Sahrens {
392789Sahrens 	dprintf("ENOSPC doing: %s\n", s ? s : "<unknown>");
393789Sahrens 	ztest_shared->zs_enospc_count++;
394789Sahrens }
395789Sahrens 
396789Sahrens static void
397789Sahrens process_options(int argc, char **argv)
398789Sahrens {
399789Sahrens 	int opt;
400789Sahrens 	uint64_t value;
401789Sahrens 
402789Sahrens 	/* By default, test gang blocks for blocks 32K and greater */
403789Sahrens 	zio_gang_bang = 32 << 10;
404789Sahrens 
405789Sahrens 	while ((opt = getopt(argc, argv,
406789Sahrens 	    "v:s:m:r:c:d:t:g:i:k:p:f:VEIT:P:S")) != EOF) {
407789Sahrens 		value = 0;
408789Sahrens 		switch (opt) {
409789Sahrens 		    case 'v':
410789Sahrens 		    case 's':
411789Sahrens 		    case 'm':
412789Sahrens 		    case 'r':
413789Sahrens 		    case 'c':
414789Sahrens 		    case 'd':
415789Sahrens 		    case 't':
416789Sahrens 		    case 'g':
417789Sahrens 		    case 'i':
418789Sahrens 		    case 'k':
419789Sahrens 		    case 'T':
420789Sahrens 		    case 'P':
421789Sahrens 			value = nicenumtoull(optarg);
422789Sahrens 		}
423789Sahrens 		switch (opt) {
424789Sahrens 		    case 'v':
425789Sahrens 			zopt_vdevs = value;
426789Sahrens 			break;
427789Sahrens 		    case 's':
428789Sahrens 			zopt_vdev_size = MAX(SPA_MINDEVSIZE, value);
429789Sahrens 			break;
430789Sahrens 		    case 'm':
431789Sahrens 			zopt_mirrors = value;
432789Sahrens 			break;
433789Sahrens 		    case 'r':
434789Sahrens 			zopt_raidz = MAX(1, value);
435789Sahrens 			break;
436789Sahrens 		    case 'd':
437789Sahrens 			zopt_dirs = MAX(1, value);
438789Sahrens 			break;
439789Sahrens 		    case 't':
440789Sahrens 			zopt_threads = MAX(1, value);
441789Sahrens 			break;
442789Sahrens 		    case 'g':
443789Sahrens 			zio_gang_bang = MAX(SPA_MINBLOCKSIZE << 1, value);
444789Sahrens 			break;
445789Sahrens 		    case 'i':
446789Sahrens 			zopt_init = value;
447789Sahrens 			break;
448789Sahrens 		    case 'k':
449789Sahrens 			zopt_killrate = value;
450789Sahrens 			break;
451789Sahrens 		    case 'p':
452789Sahrens 			zopt_pool = strdup(optarg);
453789Sahrens 			break;
454789Sahrens 		    case 'f':
455789Sahrens 			zopt_dir = strdup(optarg);
456789Sahrens 			break;
457789Sahrens 		    case 'V':
458789Sahrens 			zopt_verbose++;
459789Sahrens 			break;
460789Sahrens 		    case 'E':
461789Sahrens 			zopt_init = 0;
462789Sahrens 			break;
463789Sahrens 		    case 'T':
464789Sahrens 			zopt_time = value;
465789Sahrens 			break;
466789Sahrens 		    case 'P':
467789Sahrens 			zopt_passtime = MAX(1, value);
468789Sahrens 			break;
469789Sahrens 		    case '?':
470789Sahrens 		    default:
471789Sahrens 			usage();
472789Sahrens 			break;
473789Sahrens 		}
474789Sahrens 	}
475789Sahrens 
476789Sahrens 	zopt_vdevtime = (zopt_vdevs > 0 ? zopt_time / zopt_vdevs : UINT64_MAX);
477789Sahrens 	zopt_maxfaults = MAX(zopt_mirrors, 1) * (zopt_raidz >= 2 ? 2 : 1) - 1;
478789Sahrens }
479789Sahrens 
480789Sahrens static nvlist_t *
481789Sahrens make_vdev_file(size_t size)
482789Sahrens {
483789Sahrens 	char dev_name[MAXPATHLEN];
484789Sahrens 	uint64_t vdev;
485789Sahrens 	int fd;
486789Sahrens 	nvlist_t *file;
487789Sahrens 
488789Sahrens 	if (size == 0) {
489789Sahrens 		(void) snprintf(dev_name, sizeof (dev_name), "%s",
490789Sahrens 		    "/dev/bogus");
491789Sahrens 	} else {
492789Sahrens 		vdev = ztest_shared->zs_vdev_primaries++;
493789Sahrens 		(void) sprintf(dev_name, ztest_dev_template,
494789Sahrens 		    zopt_dir, zopt_pool, vdev);
495789Sahrens 
496789Sahrens 		fd = open(dev_name, O_RDWR | O_CREAT | O_TRUNC, 0666);
497789Sahrens 		if (fd == -1)
498789Sahrens 			fatal(1, "can't open %s", dev_name);
499789Sahrens 		if (ftruncate(fd, size) != 0)
500789Sahrens 			fatal(1, "can't ftruncate %s", dev_name);
501789Sahrens 		(void) close(fd);
502789Sahrens 	}
503789Sahrens 
504789Sahrens 	VERIFY(nvlist_alloc(&file, NV_UNIQUE_NAME, 0) == 0);
505789Sahrens 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_TYPE, VDEV_TYPE_FILE) == 0);
506789Sahrens 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_PATH, dev_name) == 0);
507789Sahrens 
508789Sahrens 	return (file);
509789Sahrens }
510789Sahrens 
511789Sahrens static nvlist_t *
512789Sahrens make_vdev_raidz(size_t size, int r)
513789Sahrens {
514789Sahrens 	nvlist_t *raidz, **child;
515789Sahrens 	int c;
516789Sahrens 
517789Sahrens 	if (r < 2)
518789Sahrens 		return (make_vdev_file(size));
519789Sahrens 
520789Sahrens 	child = umem_alloc(r * sizeof (nvlist_t *), UMEM_NOFAIL);
521789Sahrens 
522789Sahrens 	for (c = 0; c < r; c++)
523789Sahrens 		child[c] = make_vdev_file(size);
524789Sahrens 
525789Sahrens 	VERIFY(nvlist_alloc(&raidz, NV_UNIQUE_NAME, 0) == 0);
526789Sahrens 	VERIFY(nvlist_add_string(raidz, ZPOOL_CONFIG_TYPE,
527789Sahrens 	    VDEV_TYPE_RAIDZ) == 0);
528789Sahrens 	VERIFY(nvlist_add_nvlist_array(raidz, ZPOOL_CONFIG_CHILDREN,
529789Sahrens 	    child, r) == 0);
530789Sahrens 
531789Sahrens 	for (c = 0; c < r; c++)
532789Sahrens 		nvlist_free(child[c]);
533789Sahrens 
534789Sahrens 	umem_free(child, r * sizeof (nvlist_t *));
535789Sahrens 
536789Sahrens 	return (raidz);
537789Sahrens }
538789Sahrens 
539789Sahrens static nvlist_t *
540789Sahrens make_vdev_mirror(size_t size, int r, int m)
541789Sahrens {
542789Sahrens 	nvlist_t *mirror, **child;
543789Sahrens 	int c;
544789Sahrens 
545789Sahrens 	if (m < 1)
546789Sahrens 		return (make_vdev_raidz(size, r));
547789Sahrens 
548789Sahrens 	child = umem_alloc(m * sizeof (nvlist_t *), UMEM_NOFAIL);
549789Sahrens 
550789Sahrens 	for (c = 0; c < m; c++)
551789Sahrens 		child[c] = make_vdev_raidz(size, r);
552789Sahrens 
553789Sahrens 	VERIFY(nvlist_alloc(&mirror, NV_UNIQUE_NAME, 0) == 0);
554789Sahrens 	VERIFY(nvlist_add_string(mirror, ZPOOL_CONFIG_TYPE,
555789Sahrens 	    VDEV_TYPE_MIRROR) == 0);
556789Sahrens 	VERIFY(nvlist_add_nvlist_array(mirror, ZPOOL_CONFIG_CHILDREN,
557789Sahrens 	    child, m) == 0);
558789Sahrens 
559789Sahrens 	for (c = 0; c < m; c++)
560789Sahrens 		nvlist_free(child[c]);
561789Sahrens 
562789Sahrens 	umem_free(child, m * sizeof (nvlist_t *));
563789Sahrens 
564789Sahrens 	return (mirror);
565789Sahrens }
566789Sahrens 
567789Sahrens static nvlist_t *
568789Sahrens make_vdev_root(size_t size, int r, int m, int t)
569789Sahrens {
570789Sahrens 	nvlist_t *root, **child;
571789Sahrens 	int c;
572789Sahrens 
573789Sahrens 	ASSERT(t > 0);
574789Sahrens 
575789Sahrens 	child = umem_alloc(t * sizeof (nvlist_t *), UMEM_NOFAIL);
576789Sahrens 
577789Sahrens 	for (c = 0; c < t; c++)
578789Sahrens 		child[c] = make_vdev_mirror(size, r, m);
579789Sahrens 
580789Sahrens 	VERIFY(nvlist_alloc(&root, NV_UNIQUE_NAME, 0) == 0);
581789Sahrens 	VERIFY(nvlist_add_string(root, ZPOOL_CONFIG_TYPE, VDEV_TYPE_ROOT) == 0);
582789Sahrens 	VERIFY(nvlist_add_nvlist_array(root, ZPOOL_CONFIG_CHILDREN,
583789Sahrens 	    child, t) == 0);
584789Sahrens 
585789Sahrens 	for (c = 0; c < t; c++)
586789Sahrens 		nvlist_free(child[c]);
587789Sahrens 
588789Sahrens 	umem_free(child, t * sizeof (nvlist_t *));
589789Sahrens 
590789Sahrens 	return (root);
591789Sahrens }
592789Sahrens 
593789Sahrens static void
594789Sahrens ztest_set_random_blocksize(objset_t *os, uint64_t object, dmu_tx_t *tx)
595789Sahrens {
596789Sahrens 	int bs = SPA_MINBLOCKSHIFT +
597789Sahrens 	    ztest_random(SPA_MAXBLOCKSHIFT - SPA_MINBLOCKSHIFT + 1);
598789Sahrens 	int ibs = DN_MIN_INDBLKSHIFT +
599789Sahrens 	    ztest_random(DN_MAX_INDBLKSHIFT - DN_MIN_INDBLKSHIFT + 1);
600789Sahrens 	int error;
601789Sahrens 
602789Sahrens 	error = dmu_object_set_blocksize(os, object, 1ULL << bs, ibs, tx);
603789Sahrens 	if (error) {
604789Sahrens 		char osname[300];
605789Sahrens 		dmu_objset_name(os, osname);
606789Sahrens 		fatal(0, "dmu_object_set_blocksize('%s', %llu, %d, %d) = %d",
607789Sahrens 		    osname, object, 1 << bs, ibs, error);
608789Sahrens 	}
609789Sahrens }
610789Sahrens 
611789Sahrens static uint8_t
612789Sahrens ztest_random_checksum(void)
613789Sahrens {
614789Sahrens 	uint8_t checksum;
615789Sahrens 
616789Sahrens 	do {
617789Sahrens 		checksum = ztest_random(ZIO_CHECKSUM_FUNCTIONS);
618789Sahrens 	} while (zio_checksum_table[checksum].ci_zbt);
619789Sahrens 
620789Sahrens 	if (checksum == ZIO_CHECKSUM_OFF)
621789Sahrens 		checksum = ZIO_CHECKSUM_ON;
622789Sahrens 
623789Sahrens 	return (checksum);
624789Sahrens }
625789Sahrens 
626789Sahrens static uint8_t
627789Sahrens ztest_random_compress(void)
628789Sahrens {
629789Sahrens 	return ((uint8_t)ztest_random(ZIO_COMPRESS_FUNCTIONS));
630789Sahrens }
631789Sahrens 
632789Sahrens typedef struct ztest_replay {
633789Sahrens 	objset_t	*zr_os;
634789Sahrens 	uint64_t	zr_assign;
635789Sahrens } ztest_replay_t;
636789Sahrens 
637789Sahrens static int
638789Sahrens ztest_replay_create(ztest_replay_t *zr, lr_create_t *lr, boolean_t byteswap)
639789Sahrens {
640789Sahrens 	objset_t *os = zr->zr_os;
641789Sahrens 	dmu_tx_t *tx;
642789Sahrens 	int error;
643789Sahrens 
644789Sahrens 	if (byteswap)
645789Sahrens 		byteswap_uint64_array(lr, sizeof (*lr));
646789Sahrens 
647789Sahrens 	tx = dmu_tx_create(os);
648789Sahrens 	dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
649789Sahrens 	error = dmu_tx_assign(tx, zr->zr_assign);
650789Sahrens 	if (error) {
651789Sahrens 		dmu_tx_abort(tx);
652789Sahrens 		return (error);
653789Sahrens 	}
654789Sahrens 
655789Sahrens 	error = dmu_object_claim(os, lr->lr_doid, lr->lr_mode, 0,
656789Sahrens 	    DMU_OT_NONE, 0, tx);
657789Sahrens 	ASSERT(error == 0);
658789Sahrens 	dmu_tx_commit(tx);
659789Sahrens 
660789Sahrens 	if (zopt_verbose >= 5) {
661789Sahrens 		char osname[MAXNAMELEN];
662789Sahrens 		dmu_objset_name(os, osname);
663789Sahrens 		(void) printf("replay create of %s object %llu"
664789Sahrens 		    " in txg %llu = %d\n",
665789Sahrens 		    osname, (u_longlong_t)lr->lr_doid,
666789Sahrens 		    (u_longlong_t)zr->zr_assign, error);
667789Sahrens 	}
668789Sahrens 
669789Sahrens 	return (error);
670789Sahrens }
671789Sahrens 
672789Sahrens static int
673789Sahrens ztest_replay_remove(ztest_replay_t *zr, lr_remove_t *lr, boolean_t byteswap)
674789Sahrens {
675789Sahrens 	objset_t *os = zr->zr_os;
676789Sahrens 	dmu_tx_t *tx;
677789Sahrens 	int error;
678789Sahrens 
679789Sahrens 	if (byteswap)
680789Sahrens 		byteswap_uint64_array(lr, sizeof (*lr));
681789Sahrens 
682789Sahrens 	tx = dmu_tx_create(os);
683789Sahrens 	dmu_tx_hold_free(tx, lr->lr_doid, 0, DMU_OBJECT_END);
684789Sahrens 	error = dmu_tx_assign(tx, zr->zr_assign);
685789Sahrens 	if (error) {
686789Sahrens 		dmu_tx_abort(tx);
687789Sahrens 		return (error);
688789Sahrens 	}
689789Sahrens 
690789Sahrens 	error = dmu_object_free(os, lr->lr_doid, tx);
691789Sahrens 	dmu_tx_commit(tx);
692789Sahrens 
693789Sahrens 	return (error);
694789Sahrens }
695789Sahrens 
696789Sahrens zil_replay_func_t *ztest_replay_vector[TX_MAX_TYPE] = {
697789Sahrens 	NULL,			/* 0 no such transaction type */
698789Sahrens 	ztest_replay_create,	/* TX_CREATE */
699789Sahrens 	NULL,			/* TX_MKDIR */
700789Sahrens 	NULL,			/* TX_MKXATTR */
701789Sahrens 	NULL,			/* TX_SYMLINK */
702789Sahrens 	ztest_replay_remove,	/* TX_REMOVE */
703789Sahrens 	NULL,			/* TX_RMDIR */
704789Sahrens 	NULL,			/* TX_LINK */
705789Sahrens 	NULL,			/* TX_RENAME */
706789Sahrens 	NULL,			/* TX_WRITE */
707789Sahrens 	NULL,			/* TX_TRUNCATE */
708789Sahrens 	NULL,			/* TX_SETATTR */
709789Sahrens 	NULL,			/* TX_ACL */
710789Sahrens };
711789Sahrens 
712789Sahrens /*
713789Sahrens  * Verify that we can't destroy an active pool, create an existing pool,
714789Sahrens  * or create a pool with a bad vdev spec.
715789Sahrens  */
716789Sahrens void
717789Sahrens ztest_spa_create_destroy(ztest_args_t *za)
718789Sahrens {
719789Sahrens 	int error;
720789Sahrens 	spa_t *spa;
721789Sahrens 	nvlist_t *nvroot;
722789Sahrens 
723789Sahrens 	/*
724789Sahrens 	 * Attempt to create using a bad file.
725789Sahrens 	 */
726789Sahrens 	nvroot = make_vdev_root(0, 0, 0, 1);
727789Sahrens 	error = spa_create("ztest_bad_file", nvroot, NULL);
728789Sahrens 	nvlist_free(nvroot);
729789Sahrens 	if (error != ENOENT)
730789Sahrens 		fatal(0, "spa_create(bad_file) = %d", error);
731789Sahrens 
732789Sahrens 	/*
733789Sahrens 	 * Attempt to create using a bad mirror.
734789Sahrens 	 */
735789Sahrens 	nvroot = make_vdev_root(0, 0, 2, 1);
736789Sahrens 	error = spa_create("ztest_bad_mirror", nvroot, NULL);
737789Sahrens 	nvlist_free(nvroot);
738789Sahrens 	if (error != ENOENT)
739789Sahrens 		fatal(0, "spa_create(bad_mirror) = %d", error);
740789Sahrens 
741789Sahrens 	/*
742789Sahrens 	 * Attempt to create an existing pool.  It shouldn't matter
743789Sahrens 	 * what's in the nvroot; we should fail with EEXIST.
744789Sahrens 	 */
745789Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
746789Sahrens 	nvroot = make_vdev_root(0, 0, 0, 1);
747789Sahrens 	error = spa_create(za->za_pool, nvroot, NULL);
748789Sahrens 	nvlist_free(nvroot);
749789Sahrens 	if (error != EEXIST)
750789Sahrens 		fatal(0, "spa_create(whatever) = %d", error);
751789Sahrens 
752789Sahrens 	error = spa_open(za->za_pool, &spa, FTAG);
753789Sahrens 	if (error)
754789Sahrens 		fatal(0, "spa_open() = %d", error);
755789Sahrens 
756789Sahrens 	error = spa_destroy(za->za_pool);
757789Sahrens 	if (error != EBUSY)
758789Sahrens 		fatal(0, "spa_destroy() = %d", error);
759789Sahrens 
760789Sahrens 	spa_close(spa, FTAG);
761789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
762789Sahrens }
763789Sahrens 
764789Sahrens /*
765789Sahrens  * Verify that vdev_add() works as expected.
766789Sahrens  */
767789Sahrens void
768789Sahrens ztest_vdev_add_remove(ztest_args_t *za)
769789Sahrens {
770789Sahrens 	spa_t *spa = dmu_objset_spa(za->za_os);
771789Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
772789Sahrens 	nvlist_t *nvroot;
773789Sahrens 	int error;
774789Sahrens 
775789Sahrens 	if (zopt_verbose >= 6)
776789Sahrens 		(void) printf("adding vdev\n");
777789Sahrens 
778789Sahrens 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
779789Sahrens 
780789Sahrens 	spa_config_enter(spa, RW_READER);
781789Sahrens 
782789Sahrens 	ztest_shared->zs_vdev_primaries =
783789Sahrens 	    spa->spa_root_vdev->vdev_children * leaves;
784789Sahrens 
785789Sahrens 	spa_config_exit(spa);
786789Sahrens 
787789Sahrens 	nvroot = make_vdev_root(zopt_vdev_size, zopt_raidz, zopt_mirrors, 1);
788789Sahrens 	error = spa_vdev_add(spa, nvroot);
789789Sahrens 	nvlist_free(nvroot);
790789Sahrens 
791789Sahrens 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
792789Sahrens 
793789Sahrens 	if (error == ENOSPC)
794789Sahrens 		ztest_record_enospc("spa_vdev_add");
795789Sahrens 	else if (error != 0)
796789Sahrens 		fatal(0, "spa_vdev_add() = %d", error);
797789Sahrens 
798789Sahrens 	if (zopt_verbose >= 6)
799789Sahrens 		(void) printf("spa_vdev_add = %d, as expected\n", error);
800789Sahrens }
801789Sahrens 
802789Sahrens /*
803789Sahrens  * Verify that we can attach and detach devices.
804789Sahrens  */
805789Sahrens void
806789Sahrens ztest_vdev_attach_detach(ztest_args_t *za)
807789Sahrens {
808789Sahrens 	spa_t *spa = dmu_objset_spa(za->za_os);
809789Sahrens 	vdev_t *rvd = spa->spa_root_vdev;
810789Sahrens 	vdev_t *vd0, *vd1, *pvd;
811789Sahrens 	nvlist_t *root, *file;
812789Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
813789Sahrens 	uint64_t leaf, top;
814789Sahrens 	size_t size0, size1;
815789Sahrens 	char path0[MAXPATHLEN], path1[MAXPATHLEN];
816789Sahrens 	int replacing;
817789Sahrens 	int error, expected_error;
818789Sahrens 	int fd;
819789Sahrens 
820789Sahrens 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
821789Sahrens 
822789Sahrens 	spa_config_enter(spa, RW_READER);
823789Sahrens 
824789Sahrens 	/*
825789Sahrens 	 * Decide whether to do an attach or a replace.
826789Sahrens 	 */
827789Sahrens 	replacing = ztest_random(2);
828789Sahrens 
829789Sahrens 	/*
830789Sahrens 	 * Pick a random top-level vdev.
831789Sahrens 	 */
832789Sahrens 	top = ztest_random(rvd->vdev_children);
833789Sahrens 
834789Sahrens 	/*
835789Sahrens 	 * Pick a random leaf within it.
836789Sahrens 	 */
837789Sahrens 	leaf = ztest_random(leaves);
838789Sahrens 
839789Sahrens 	/*
840789Sahrens 	 * Generate the path to this leaf.  The filename will end with 'a'.
841789Sahrens 	 * We'll alternate replacements with a filename that ends with 'b'.
842789Sahrens 	 */
843789Sahrens 	(void) snprintf(path0, sizeof (path0),
844789Sahrens 	    ztest_dev_template, zopt_dir, zopt_pool, top * leaves + leaf);
845789Sahrens 
846789Sahrens 	bcopy(path0, path1, MAXPATHLEN);
847789Sahrens 
848789Sahrens 	/*
849789Sahrens 	 * If the 'a' file isn't part of the pool, the 'b' file must be.
850789Sahrens 	 */
851789Sahrens 	if (vdev_lookup_by_path(rvd, path0) == NULL)
852789Sahrens 		path0[strlen(path0) - 1] = 'b';
853789Sahrens 	else
854789Sahrens 		path1[strlen(path1) - 1] = 'b';
855789Sahrens 
856789Sahrens 	/*
857789Sahrens 	 * Now path0 represents something that's already in the pool,
858789Sahrens 	 * and path1 is the thing we'll try to attach.
859789Sahrens 	 */
860789Sahrens 	vd0 = vdev_lookup_by_path(rvd, path0);
861789Sahrens 	vd1 = vdev_lookup_by_path(rvd, path1);
862789Sahrens 	ASSERT(vd0 != NULL);
863789Sahrens 	pvd = vd0->vdev_parent;
864789Sahrens 
865789Sahrens 
866789Sahrens 	/*
867789Sahrens 	 * Make size1 a little bigger or smaller than size0.
868789Sahrens 	 * If it's smaller, the attach should fail.
869789Sahrens 	 * If it's larger, and we're doing a replace,
870789Sahrens 	 * we should get dynamic LUN growth when we're done.
871789Sahrens 	 */
872789Sahrens 	size0 = vd0->vdev_psize;
873789Sahrens 	size1 = 10 * size0 / (9 + ztest_random(3));
874789Sahrens 
875789Sahrens 	/*
876789Sahrens 	 * If pvd is not a mirror or root, the attach should fail with ENOTSUP,
877789Sahrens 	 * unless it's a replace; in that case any non-replacing parent is OK.
878789Sahrens 	 *
879789Sahrens 	 * If vd1 is already part of the pool, it should fail with EBUSY.
880789Sahrens 	 *
881789Sahrens 	 * If vd1 is too small, it should fail with EOVERFLOW.
882789Sahrens 	 */
883789Sahrens 	if (pvd->vdev_ops != &vdev_mirror_ops &&
884789Sahrens 	    pvd->vdev_ops != &vdev_root_ops &&
885789Sahrens 	    (!replacing || pvd->vdev_ops == &vdev_replacing_ops))
886789Sahrens 		expected_error = ENOTSUP;
887789Sahrens 	else if (vd1 != NULL)
888789Sahrens 		expected_error = EBUSY;
889789Sahrens 	else if (size1 < size0)
890789Sahrens 		expected_error = EOVERFLOW;
891789Sahrens 	else
892789Sahrens 		expected_error = 0;
893789Sahrens 
894789Sahrens 	/*
895789Sahrens 	 * If vd1 isn't already part of the pool, create it.
896789Sahrens 	 */
897789Sahrens 	if (vd1 == NULL) {
898789Sahrens 		fd = open(path1, O_RDWR | O_CREAT | O_TRUNC, 0666);
899789Sahrens 		if (fd == -1)
900789Sahrens 			fatal(1, "can't open %s", path1);
901789Sahrens 		if (ftruncate(fd, size1) != 0)
902789Sahrens 			fatal(1, "can't ftruncate %s", path1);
903789Sahrens 		(void) close(fd);
904789Sahrens 	}
905789Sahrens 
906789Sahrens 	spa_config_exit(spa);
907789Sahrens 
908789Sahrens 	/*
909789Sahrens 	 * Build the nvlist describing path1.
910789Sahrens 	 */
911789Sahrens 	VERIFY(nvlist_alloc(&file, NV_UNIQUE_NAME, 0) == 0);
912789Sahrens 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_TYPE, VDEV_TYPE_FILE) == 0);
913789Sahrens 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_PATH, path1) == 0);
914789Sahrens 
915789Sahrens 	VERIFY(nvlist_alloc(&root, NV_UNIQUE_NAME, 0) == 0);
916789Sahrens 	VERIFY(nvlist_add_string(root, ZPOOL_CONFIG_TYPE, VDEV_TYPE_ROOT) == 0);
917789Sahrens 	VERIFY(nvlist_add_nvlist_array(root, ZPOOL_CONFIG_CHILDREN,
918789Sahrens 	    &file, 1) == 0);
919789Sahrens 
920789Sahrens 	error = spa_vdev_attach(spa, path0, root, replacing);
921789Sahrens 
922789Sahrens 	nvlist_free(file);
923789Sahrens 	nvlist_free(root);
924789Sahrens 
925789Sahrens 	/*
926789Sahrens 	 * If our parent was the replacing vdev, but the replace completed,
927789Sahrens 	 * then instead of failing with ENOTSUP we may either succeed,
928789Sahrens 	 * fail with ENODEV, or fail with EOVERFLOW.
929789Sahrens 	 */
930789Sahrens 	if (expected_error == ENOTSUP &&
931789Sahrens 	    (error == 0 || error == ENODEV || error == EOVERFLOW))
932789Sahrens 		expected_error = error;
933789Sahrens 
934797Sbonwick 	/*
935797Sbonwick 	 * If someone grew the LUN, the replacement may be too small.
936797Sbonwick 	 */
937797Sbonwick 	if (error == EOVERFLOW)
938797Sbonwick 		expected_error = error;
939797Sbonwick 
940789Sahrens 	if (error != expected_error) {
941789Sahrens 		fatal(0, "attach (%s, %s, %d) returned %d, expected %d",
942789Sahrens 		    path0, path1, replacing, error, expected_error);
943789Sahrens 	}
944789Sahrens 
945789Sahrens 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
946789Sahrens }
947789Sahrens 
948789Sahrens /*
949789Sahrens  * Verify that dynamic LUN growth works as expected.
950789Sahrens  */
951789Sahrens /* ARGSUSED */
952789Sahrens void
953789Sahrens ztest_vdev_LUN_growth(ztest_args_t *za)
954789Sahrens {
955789Sahrens 	spa_t *spa = dmu_objset_spa(za->za_os);
956789Sahrens 	char dev_name[MAXPATHLEN];
957789Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
958789Sahrens 	uint64_t vdev;
959789Sahrens 	size_t fsize;
960789Sahrens 	int fd;
961789Sahrens 
962789Sahrens 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
963789Sahrens 
964789Sahrens 	/*
965789Sahrens 	 * Pick a random leaf vdev.
966789Sahrens 	 */
967789Sahrens 	spa_config_enter(spa, RW_READER);
968789Sahrens 	vdev = ztest_random(spa->spa_root_vdev->vdev_children * leaves);
969789Sahrens 	spa_config_exit(spa);
970789Sahrens 
971789Sahrens 	(void) sprintf(dev_name, ztest_dev_template, zopt_dir, zopt_pool, vdev);
972789Sahrens 
973789Sahrens 	if ((fd = open(dev_name, O_RDWR)) != -1) {
974789Sahrens 		/*
975789Sahrens 		 * Determine the size.
976789Sahrens 		 */
977789Sahrens 		fsize = lseek(fd, 0, SEEK_END);
978789Sahrens 
979789Sahrens 		/*
980789Sahrens 		 * If it's less than 2x the original size, grow by around 3%.
981789Sahrens 		 */
982789Sahrens 		if (fsize < 2 * zopt_vdev_size) {
983789Sahrens 			size_t newsize = fsize + ztest_random(fsize / 32);
984789Sahrens 			(void) ftruncate(fd, newsize);
985789Sahrens 			if (zopt_verbose >= 6) {
986789Sahrens 				(void) printf("%s grew from %lu to %lu bytes\n",
987789Sahrens 				    dev_name, (ulong_t)fsize, (ulong_t)newsize);
988789Sahrens 			}
989789Sahrens 		}
990789Sahrens 		(void) close(fd);
991789Sahrens 	}
992789Sahrens 
993789Sahrens 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
994789Sahrens }
995789Sahrens 
996789Sahrens /* ARGSUSED */
997789Sahrens static void
998789Sahrens ztest_create_cb(objset_t *os, void *arg, dmu_tx_t *tx)
999789Sahrens {
1000789Sahrens 	/*
1001789Sahrens 	 * Create the directory object.
1002789Sahrens 	 */
1003789Sahrens 	VERIFY(dmu_object_claim(os, ZTEST_DIROBJ,
1004789Sahrens 	    DMU_OT_UINT64_OTHER, ZTEST_DIROBJ_BLOCKSIZE,
1005789Sahrens 	    DMU_OT_UINT64_OTHER, sizeof (ztest_block_tag_t), tx) == 0);
1006789Sahrens 
1007789Sahrens 	VERIFY(zap_create_claim(os, ZTEST_MICROZAP_OBJ,
1008789Sahrens 	    DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx) == 0);
1009789Sahrens 
1010789Sahrens 	VERIFY(zap_create_claim(os, ZTEST_FATZAP_OBJ,
1011789Sahrens 	    DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx) == 0);
1012789Sahrens }
1013789Sahrens 
1014789Sahrens /* ARGSUSED */
1015789Sahrens static void
1016789Sahrens ztest_destroy_cb(char *name, void *arg)
1017789Sahrens {
1018789Sahrens 	objset_t *os;
1019789Sahrens 	dmu_object_info_t doi;
1020789Sahrens 	int error;
1021789Sahrens 
1022789Sahrens 	/*
1023789Sahrens 	 * Verify that the dataset contains a directory object.
1024789Sahrens 	 */
1025789Sahrens 	error = dmu_objset_open(name, DMU_OST_OTHER,
1026789Sahrens 	    DS_MODE_STANDARD | DS_MODE_READONLY, &os);
1027789Sahrens 	ASSERT3U(error, ==, 0);
1028789Sahrens 	error = dmu_object_info(os, ZTEST_DIROBJ, &doi);
1029789Sahrens 	ASSERT3U(error, ==, 0);
1030789Sahrens 	ASSERT3U(doi.doi_type, ==, DMU_OT_UINT64_OTHER);
1031789Sahrens 	ASSERT3S(doi.doi_physical_blks, >=, 0);
1032789Sahrens 	dmu_objset_close(os);
1033789Sahrens 
1034789Sahrens 	/*
1035789Sahrens 	 * Destroy the dataset.
1036789Sahrens 	 */
1037789Sahrens 	error = dmu_objset_destroy(name);
1038789Sahrens 	ASSERT3U(error, ==, 0);
1039789Sahrens }
1040789Sahrens 
1041789Sahrens /*
1042789Sahrens  * Verify that dmu_objset_{create,destroy,open,close} work as expected.
1043789Sahrens  */
1044789Sahrens static uint64_t
1045789Sahrens ztest_log_create(zilog_t *zilog, dmu_tx_t *tx, uint64_t object, int mode)
1046789Sahrens {
1047789Sahrens 	itx_t *itx;
1048789Sahrens 	lr_create_t *lr;
1049789Sahrens 	size_t namesize;
1050789Sahrens 	char name[24];
1051789Sahrens 
1052789Sahrens 	(void) sprintf(name, "ZOBJ_%llu", (u_longlong_t)object);
1053789Sahrens 	namesize = strlen(name) + 1;
1054789Sahrens 
1055789Sahrens 	itx = zil_itx_create(TX_CREATE, sizeof (*lr) + namesize +
1056789Sahrens 	    ztest_random(ZIL_MAX_BLKSZ));
1057789Sahrens 	lr = (lr_create_t *)&itx->itx_lr;
1058789Sahrens 	bzero(lr + 1, lr->lr_common.lrc_reclen - sizeof (*lr));
1059789Sahrens 	lr->lr_doid = object;
1060789Sahrens 	lr->lr_foid = 0;
1061789Sahrens 	lr->lr_mode = mode;
1062789Sahrens 	lr->lr_uid = 0;
1063789Sahrens 	lr->lr_gid = 0;
1064789Sahrens 	lr->lr_gen = dmu_tx_get_txg(tx);
1065789Sahrens 	lr->lr_crtime[0] = time(NULL);
1066789Sahrens 	lr->lr_crtime[1] = 0;
1067789Sahrens 	lr->lr_rdev = 0;
1068789Sahrens 	bcopy(name, (char *)(lr + 1), namesize);
1069789Sahrens 
1070789Sahrens 	return (zil_itx_assign(zilog, itx, tx));
1071789Sahrens }
1072789Sahrens 
1073789Sahrens #ifndef lint
1074789Sahrens static uint64_t
1075789Sahrens ztest_log_remove(zilog_t *zilog, dmu_tx_t *tx, uint64_t object)
1076789Sahrens {
1077789Sahrens 	itx_t *itx;
1078789Sahrens 	lr_remove_t *lr;
1079789Sahrens 	size_t namesize;
1080789Sahrens 	char name[24];
1081789Sahrens 
1082789Sahrens 	(void) sprintf(name, "ZOBJ_%llu", (u_longlong_t)object);
1083789Sahrens 	namesize = strlen(name) + 1;
1084789Sahrens 
1085789Sahrens 	itx = zil_itx_create(TX_REMOVE, sizeof (*lr) + namesize +
1086789Sahrens 	    ztest_random(8000));
1087789Sahrens 	lr = (lr_remove_t *)&itx->itx_lr;
1088789Sahrens 	lr->lr_doid = object;
1089789Sahrens 	bcopy(name, (char *)(lr + 1), namesize);
1090789Sahrens 
1091789Sahrens 	return (zil_itx_assign(zilog, itx, tx));
1092789Sahrens }
1093789Sahrens #endif /* lint */
1094789Sahrens 
1095789Sahrens void
1096789Sahrens ztest_dmu_objset_create_destroy(ztest_args_t *za)
1097789Sahrens {
1098789Sahrens 	int error;
1099789Sahrens 	objset_t *os;
1100789Sahrens 	char name[100];
1101789Sahrens 	int mode, basemode, expected_error;
1102789Sahrens 	zilog_t *zilog;
1103789Sahrens 	uint64_t seq;
1104789Sahrens 	uint64_t objects;
1105789Sahrens 	ztest_replay_t zr;
1106789Sahrens 
1107789Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
1108789Sahrens 	(void) snprintf(name, 100, "%s/%s_temp_%llu", za->za_pool, za->za_pool,
1109789Sahrens 	    (u_longlong_t)za->za_instance);
1110789Sahrens 
1111789Sahrens 	basemode = DS_MODE_LEVEL(za->za_instance);
1112789Sahrens 	if (basemode == DS_MODE_NONE)
1113789Sahrens 		basemode++;
1114789Sahrens 
1115789Sahrens 	/*
1116789Sahrens 	 * If this dataset exists from a previous run, process its replay log
1117789Sahrens 	 * half of the time.  If we don't replay it, then dmu_objset_destroy()
1118789Sahrens 	 * (invoked from ztest_destroy_cb() below) should just throw it away.
1119789Sahrens 	 */
1120789Sahrens 	if (ztest_random(2) == 0 &&
1121789Sahrens 	    dmu_objset_open(name, DMU_OST_OTHER, DS_MODE_PRIMARY, &os) == 0) {
1122789Sahrens 		zr.zr_os = os;
1123789Sahrens 		zil_replay(os, &zr, &zr.zr_assign, ztest_replay_vector, NULL);
1124789Sahrens 		dmu_objset_close(os);
1125789Sahrens 	}
1126789Sahrens 
1127789Sahrens 	/*
1128789Sahrens 	 * There may be an old instance of the dataset we're about to
1129789Sahrens 	 * create lying around from a previous run.  If so, destroy it
1130789Sahrens 	 * and all of its snapshots.
1131789Sahrens 	 */
1132789Sahrens 	dmu_objset_find(name, ztest_destroy_cb, NULL, DS_FIND_SNAPSHOTS);
1133789Sahrens 
1134789Sahrens 	/*
1135789Sahrens 	 * Verify that the destroyed dataset is no longer in the namespace.
1136789Sahrens 	 */
1137789Sahrens 	error = dmu_objset_open(name, DMU_OST_OTHER, basemode, &os);
1138789Sahrens 	if (error != ENOENT)
1139789Sahrens 		fatal(1, "dmu_objset_open(%s) found destroyed dataset %p",
1140789Sahrens 		    name, os);
1141789Sahrens 
1142789Sahrens 	/*
1143789Sahrens 	 * Verify that we can create a new dataset.
1144789Sahrens 	 */
1145789Sahrens 	error = dmu_objset_create(name, DMU_OST_OTHER, NULL, ztest_create_cb,
1146789Sahrens 	    NULL);
1147789Sahrens 	if (error) {
1148789Sahrens 		if (error == ENOSPC) {
1149789Sahrens 			ztest_record_enospc("dmu_objset_create");
1150789Sahrens 			(void) rw_unlock(&ztest_shared->zs_name_lock);
1151789Sahrens 			return;
1152789Sahrens 		}
1153789Sahrens 		fatal(0, "dmu_objset_create(%s) = %d", name, error);
1154789Sahrens 	}
1155789Sahrens 
1156789Sahrens 	error = dmu_objset_open(name, DMU_OST_OTHER, basemode, &os);
1157789Sahrens 	if (error) {
1158789Sahrens 		fatal(0, "dmu_objset_open(%s) = %d", name, error);
1159789Sahrens 	}
1160789Sahrens 
1161789Sahrens 	/*
1162789Sahrens 	 * Open the intent log for it.
1163789Sahrens 	 */
1164789Sahrens 	zilog = zil_open(os, NULL);
1165789Sahrens 
1166789Sahrens 	/*
1167789Sahrens 	 * Put a random number of objects in there.
1168789Sahrens 	 */
1169789Sahrens 	objects = ztest_random(50);
1170789Sahrens 	seq = 0;
1171789Sahrens 	while (objects-- != 0) {
1172789Sahrens 		uint64_t object;
1173789Sahrens 		dmu_tx_t *tx = dmu_tx_create(os);
1174789Sahrens 		dmu_tx_hold_write(tx, DMU_NEW_OBJECT, 0, sizeof (name));
1175789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1176789Sahrens 		if (error) {
1177789Sahrens 			dmu_tx_abort(tx);
1178789Sahrens 		} else {
1179789Sahrens 			object = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1180789Sahrens 			    DMU_OT_NONE, 0, tx);
1181789Sahrens 			ztest_set_random_blocksize(os, object, tx);
1182789Sahrens 			seq = ztest_log_create(zilog, tx, object,
1183789Sahrens 			    DMU_OT_UINT64_OTHER);
1184789Sahrens 			dmu_write(os, object, 0, sizeof (name), name, tx);
1185789Sahrens 			dmu_tx_commit(tx);
1186789Sahrens 		}
1187789Sahrens 		if (ztest_random(5) == 0) {
1188789Sahrens 			zil_commit(zilog, seq, FSYNC);
1189789Sahrens 		}
1190789Sahrens 		if (ztest_random(5) == 0) {
1191789Sahrens 			error = zil_suspend(zilog);
1192789Sahrens 			if (error == 0) {
1193789Sahrens 				zil_resume(zilog);
1194789Sahrens 			}
1195789Sahrens 		}
1196789Sahrens 	}
1197789Sahrens 
1198789Sahrens 	/*
1199789Sahrens 	 * Verify that we cannot create an existing dataset.
1200789Sahrens 	 */
1201789Sahrens 	error = dmu_objset_create(name, DMU_OST_OTHER, NULL, NULL, NULL);
1202789Sahrens 	if (error != EEXIST)
1203789Sahrens 		fatal(0, "created existing dataset, error = %d", error);
1204789Sahrens 
1205789Sahrens 	/*
1206789Sahrens 	 * Verify that multiple dataset opens are allowed, but only when
1207789Sahrens 	 * the new access mode is compatible with the base mode.
1208789Sahrens 	 * We use a mixture of typed and typeless opens, and when the
1209789Sahrens 	 * open succeeds, verify that the discovered type is correct.
1210789Sahrens 	 */
1211789Sahrens 	for (mode = DS_MODE_STANDARD; mode < DS_MODE_LEVELS; mode++) {
1212789Sahrens 		objset_t *os2;
1213789Sahrens 		error = dmu_objset_open(name, DMU_OST_OTHER, mode, &os2);
1214789Sahrens 		expected_error = (basemode + mode < DS_MODE_LEVELS) ? 0 : EBUSY;
1215789Sahrens 		if (error != expected_error)
1216789Sahrens 			fatal(0, "dmu_objset_open('%s') = %d, expected %d",
1217789Sahrens 			    name, error, expected_error);
1218789Sahrens 		if (error == 0)
1219789Sahrens 			dmu_objset_close(os2);
1220789Sahrens 	}
1221789Sahrens 
1222789Sahrens 	zil_close(zilog);
1223789Sahrens 	dmu_objset_close(os);
1224789Sahrens 
1225789Sahrens 	error = dmu_objset_destroy(name);
1226789Sahrens 	if (error)
1227789Sahrens 		fatal(0, "dmu_objset_destroy(%s) = %d", name, error);
1228789Sahrens 
1229789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
1230789Sahrens }
1231789Sahrens 
1232789Sahrens /*
1233789Sahrens  * Verify that dmu_snapshot_{create,destroy,open,close} work as expected.
1234789Sahrens  */
1235789Sahrens void
1236789Sahrens ztest_dmu_snapshot_create_destroy(ztest_args_t *za)
1237789Sahrens {
1238789Sahrens 	int error;
1239789Sahrens 	objset_t *os = za->za_os;
1240789Sahrens 	char snapname[100];
1241789Sahrens 	char osname[MAXNAMELEN];
1242789Sahrens 
1243789Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
1244789Sahrens 	dmu_objset_name(os, osname);
1245789Sahrens 	(void) snprintf(snapname, 100, "%s@%llu", osname,
1246789Sahrens 	    (u_longlong_t)za->za_instance);
1247789Sahrens 
1248789Sahrens 	error = dmu_objset_destroy(snapname);
1249789Sahrens 	if (error != 0 && error != ENOENT)
1250789Sahrens 		fatal(0, "dmu_objset_destroy() = %d", error);
1251789Sahrens 	error = dmu_objset_create(snapname, DMU_OST_OTHER, NULL, NULL, NULL);
1252789Sahrens 	if (error == ENOSPC)
1253789Sahrens 		ztest_record_enospc("dmu_take_snapshot");
1254789Sahrens 	else if (error != 0 && error != EEXIST)
1255789Sahrens 		fatal(0, "dmu_take_snapshot() = %d", error);
1256789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
1257789Sahrens }
1258789Sahrens 
1259789Sahrens #define	ZTEST_TRAVERSE_BLOCKS	1000
1260789Sahrens 
1261789Sahrens static int
1262789Sahrens ztest_blk_cb(traverse_blk_cache_t *bc, spa_t *spa, void *arg)
1263789Sahrens {
1264789Sahrens 	ztest_args_t *za = arg;
1265789Sahrens 	zbookmark_t *zb = &bc->bc_bookmark;
1266789Sahrens 	blkptr_t *bp = &bc->bc_blkptr;
1267789Sahrens 	dnode_phys_t *dnp = bc->bc_dnode;
1268789Sahrens 	traverse_handle_t *th = za->za_th;
1269789Sahrens 	uint64_t size = BP_GET_LSIZE(bp);
1270789Sahrens 
1271789Sahrens 	ASSERT(dnp != NULL);
1272789Sahrens 
1273789Sahrens 	if (bc->bc_errno)
1274789Sahrens 		return (ERESTART);
1275789Sahrens 
1276789Sahrens 	/*
1277789Sahrens 	 * Once in a while, abort the traverse.   We only do this to odd
1278789Sahrens 	 * instance numbers to ensure that even ones can run to completion.
1279789Sahrens 	 */
1280789Sahrens 	if ((za->za_instance & 1) && ztest_random(10000) == 0)
1281789Sahrens 		return (EINTR);
1282789Sahrens 
1283789Sahrens 	if (bp->blk_birth == 0) {
1284789Sahrens 		ASSERT(th->th_advance & ADVANCE_HOLES);
1285789Sahrens 		return (0);
1286789Sahrens 	}
1287789Sahrens 
1288789Sahrens 	if (zb->zb_level == 0 && !(th->th_advance & ADVANCE_DATA) &&
1289789Sahrens 	    bc == &th->th_cache[ZB_DN_CACHE][0]) {
1290789Sahrens 		ASSERT(bc->bc_data == NULL);
1291789Sahrens 		return (0);
1292789Sahrens 	}
1293789Sahrens 
1294789Sahrens 	ASSERT(bc->bc_data != NULL);
1295789Sahrens 
1296789Sahrens 	/*
1297789Sahrens 	 * This is an expensive question, so don't ask it too often.
1298789Sahrens 	 */
1299789Sahrens 	if (((za->za_random ^ th->th_callbacks) & 0xff) == 0) {
1300789Sahrens 		void *xbuf = umem_alloc(size, UMEM_NOFAIL);
1301789Sahrens 		if (arc_tryread(spa, bp, xbuf) == 0) {
1302789Sahrens 			ASSERT(bcmp(bc->bc_data, xbuf, size) == 0);
1303789Sahrens 		}
1304789Sahrens 		umem_free(xbuf, size);
1305789Sahrens 	}
1306789Sahrens 
1307789Sahrens 	if (zb->zb_level > 0) {
1308789Sahrens 		ASSERT3U(size, ==, 1ULL << dnp->dn_indblkshift);
1309789Sahrens 		return (0);
1310789Sahrens 	}
1311789Sahrens 
1312789Sahrens 	if (zb->zb_level == -1) {
1313789Sahrens 		ASSERT3U(size, ==, sizeof (objset_phys_t));
1314789Sahrens 		return (0);
1315789Sahrens 	}
1316789Sahrens 
1317789Sahrens 	ASSERT(zb->zb_level == 0);
1318789Sahrens 	ASSERT3U(size, ==, dnp->dn_datablkszsec << DEV_BSHIFT);
1319789Sahrens 
1320789Sahrens 	return (0);
1321789Sahrens }
1322789Sahrens 
1323789Sahrens /*
1324789Sahrens  * Verify that live pool traversal works.
1325789Sahrens  */
1326789Sahrens void
1327789Sahrens ztest_traverse(ztest_args_t *za)
1328789Sahrens {
1329789Sahrens 	spa_t *spa = dmu_objset_spa(za->za_os);
1330789Sahrens 	traverse_handle_t *th = za->za_th;
1331789Sahrens 	int rc, advance;
1332789Sahrens 	uint64_t cbstart, cblimit;
1333789Sahrens 
1334789Sahrens 	if (th == NULL) {
1335789Sahrens 		advance = 0;
1336789Sahrens 
1337789Sahrens 		if (ztest_random(2) == 0)
1338789Sahrens 			advance |= ADVANCE_PRE;
1339789Sahrens 
1340789Sahrens 		if (ztest_random(2) == 0)
1341789Sahrens 			advance |= ADVANCE_PRUNE;
1342789Sahrens 
1343789Sahrens 		if (ztest_random(2) == 0)
1344789Sahrens 			advance |= ADVANCE_DATA;
1345789Sahrens 
1346789Sahrens 		if (ztest_random(2) == 0)
1347789Sahrens 			advance |= ADVANCE_HOLES;
1348789Sahrens 
1349789Sahrens 		th = za->za_th = traverse_init(spa, ztest_blk_cb, za, advance,
1350789Sahrens 		    ZIO_FLAG_CANFAIL);
1351789Sahrens 
1352789Sahrens 		traverse_add_pool(th, 0, -1ULL);
1353789Sahrens 	}
1354789Sahrens 
1355789Sahrens 	advance = th->th_advance;
1356789Sahrens 	cbstart = th->th_callbacks;
1357789Sahrens 	cblimit = cbstart + ((advance & ADVANCE_DATA) ? 100 : 1000);
1358789Sahrens 
1359789Sahrens 	while ((rc = traverse_more(th)) == EAGAIN && th->th_callbacks < cblimit)
1360789Sahrens 		continue;
1361789Sahrens 
1362789Sahrens 	if (zopt_verbose >= 5)
1363789Sahrens 		(void) printf("traverse %s%s%s%s %llu blocks to "
1364789Sahrens 		    "<%llu, %llu, %d, %llx>%s\n",
1365789Sahrens 		    (advance & ADVANCE_PRE) ? "pre" : "post",
1366789Sahrens 		    (advance & ADVANCE_PRUNE) ? "|prune" : "",
1367789Sahrens 		    (advance & ADVANCE_DATA) ? "|data" : "",
1368789Sahrens 		    (advance & ADVANCE_HOLES) ? "|holes" : "",
1369789Sahrens 		    (u_longlong_t)(th->th_callbacks - cbstart),
1370789Sahrens 		    (u_longlong_t)th->th_lastcb.zb_objset,
1371789Sahrens 		    (u_longlong_t)th->th_lastcb.zb_object,
1372789Sahrens 		    th->th_lastcb.zb_level,
1373789Sahrens 		    (u_longlong_t)th->th_lastcb.zb_blkid,
1374789Sahrens 		    rc == 0 ? " [done]" :
1375789Sahrens 		    rc == EINTR ? " [aborted]" :
1376789Sahrens 		    rc == EAGAIN ? "" :
1377789Sahrens 		    strerror(rc));
1378789Sahrens 
1379789Sahrens 	if (rc != EAGAIN) {
1380789Sahrens 		if (rc != 0 && rc != EINTR)
1381789Sahrens 			fatal(0, "traverse_more(%p) = %d", th, rc);
1382789Sahrens 		traverse_fini(th);
1383789Sahrens 		za->za_th = NULL;
1384789Sahrens 	}
1385789Sahrens }
1386789Sahrens 
1387789Sahrens /*
1388789Sahrens  * Verify that dmu_object_{alloc,free} work as expected.
1389789Sahrens  */
1390789Sahrens void
1391789Sahrens ztest_dmu_object_alloc_free(ztest_args_t *za)
1392789Sahrens {
1393789Sahrens 	objset_t *os = za->za_os;
1394789Sahrens 	dmu_buf_t *db;
1395789Sahrens 	dmu_tx_t *tx;
1396789Sahrens 	uint64_t batchobj, object, batchsize, endoff, temp;
1397789Sahrens 	int b, c, error, bonuslen;
1398789Sahrens 	dmu_object_info_t doi;
1399789Sahrens 	char osname[MAXNAMELEN];
1400789Sahrens 
1401789Sahrens 	dmu_objset_name(os, osname);
1402789Sahrens 
1403789Sahrens 	endoff = -8ULL;
1404789Sahrens 	batchsize = 2;
1405789Sahrens 
1406789Sahrens 	/*
1407789Sahrens 	 * Create a batch object if necessary, and record it in the directory.
1408789Sahrens 	 */
1409789Sahrens 	dmu_read(os, ZTEST_DIROBJ, za->za_diroff, sizeof (uint64_t), &batchobj);
1410789Sahrens 	if (batchobj == 0) {
1411789Sahrens 		tx = dmu_tx_create(os);
1412789Sahrens 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff,
1413789Sahrens 		    sizeof (uint64_t));
1414789Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1415789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1416789Sahrens 		if (error) {
1417789Sahrens 			ztest_record_enospc("create a batch object");
1418789Sahrens 			dmu_tx_abort(tx);
1419789Sahrens 			return;
1420789Sahrens 		}
1421789Sahrens 		batchobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1422789Sahrens 		    DMU_OT_NONE, 0, tx);
1423789Sahrens 		ztest_set_random_blocksize(os, batchobj, tx);
1424789Sahrens 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff,
1425789Sahrens 		    sizeof (uint64_t), &batchobj, tx);
1426789Sahrens 		dmu_tx_commit(tx);
1427789Sahrens 	}
1428789Sahrens 
1429789Sahrens 	/*
1430789Sahrens 	 * Destroy the previous batch of objects.
1431789Sahrens 	 */
1432789Sahrens 	for (b = 0; b < batchsize; b++) {
1433789Sahrens 		dmu_read(os, batchobj, b * sizeof (uint64_t),
1434789Sahrens 		    sizeof (uint64_t), &object);
1435789Sahrens 		if (object == 0)
1436789Sahrens 			continue;
1437789Sahrens 		/*
1438789Sahrens 		 * Read and validate contents.
1439789Sahrens 		 * We expect the nth byte of the bonus buffer to be n.
1440789Sahrens 		 */
1441789Sahrens 		db = dmu_bonus_hold(os, object);
1442789Sahrens 
1443789Sahrens 		dmu_object_info_from_db(db, &doi);
1444789Sahrens 		ASSERT(doi.doi_type == DMU_OT_UINT64_OTHER);
1445789Sahrens 		ASSERT(doi.doi_bonus_type == DMU_OT_PLAIN_OTHER);
1446789Sahrens 		ASSERT3S(doi.doi_physical_blks, >=, 0);
1447789Sahrens 
1448789Sahrens 		dmu_buf_read(db);
1449789Sahrens 
1450789Sahrens 		bonuslen = db->db_size;
1451789Sahrens 
1452789Sahrens 		for (c = 0; c < bonuslen; c++) {
1453789Sahrens 			if (((uint8_t *)db->db_data)[c] !=
1454789Sahrens 			    (uint8_t)(c + bonuslen)) {
1455789Sahrens 				fatal(0,
1456789Sahrens 				    "bad bonus: %s, obj %llu, off %d: %u != %u",
1457789Sahrens 				    osname, object, c,
1458789Sahrens 				    ((uint8_t *)db->db_data)[c],
1459789Sahrens 				    (uint8_t)(c + bonuslen));
1460789Sahrens 			}
1461789Sahrens 		}
1462789Sahrens 
1463789Sahrens 		dmu_buf_rele(db);
1464789Sahrens 
1465789Sahrens 		/*
1466789Sahrens 		 * We expect the word at endoff to be our object number.
1467789Sahrens 		 */
1468789Sahrens 		dmu_read(os, object, endoff, sizeof (uint64_t), &temp);
1469789Sahrens 
1470789Sahrens 		if (temp != object) {
1471789Sahrens 			fatal(0, "bad data in %s, got %llu, expected %llu",
1472789Sahrens 			    osname, temp, object);
1473789Sahrens 		}
1474789Sahrens 
1475789Sahrens 		/*
1476789Sahrens 		 * Destroy old object and clear batch entry.
1477789Sahrens 		 */
1478789Sahrens 		tx = dmu_tx_create(os);
1479789Sahrens 		dmu_tx_hold_write(tx, batchobj,
1480789Sahrens 		    b * sizeof (uint64_t), sizeof (uint64_t));
1481789Sahrens 		dmu_tx_hold_free(tx, object, 0, DMU_OBJECT_END);
1482789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1483789Sahrens 		if (error) {
1484789Sahrens 			ztest_record_enospc("free object");
1485789Sahrens 			dmu_tx_abort(tx);
1486789Sahrens 			return;
1487789Sahrens 		}
1488789Sahrens 		error = dmu_object_free(os, object, tx);
1489789Sahrens 		if (error) {
1490789Sahrens 			fatal(0, "dmu_object_free('%s', %llu) = %d",
1491789Sahrens 			    osname, object, error);
1492789Sahrens 		}
1493789Sahrens 		object = 0;
1494789Sahrens 
1495789Sahrens 		dmu_object_set_checksum(os, batchobj,
1496789Sahrens 		    ztest_random_checksum(), tx);
1497789Sahrens 		dmu_object_set_compress(os, batchobj,
1498789Sahrens 		    ztest_random_compress(), tx);
1499789Sahrens 
1500789Sahrens 		dmu_write(os, batchobj, b * sizeof (uint64_t),
1501789Sahrens 		    sizeof (uint64_t), &object, tx);
1502789Sahrens 
1503789Sahrens 		dmu_tx_commit(tx);
1504789Sahrens 	}
1505789Sahrens 
1506789Sahrens 	/*
1507789Sahrens 	 * Before creating the new batch of objects, generate a bunch of churn.
1508789Sahrens 	 */
1509789Sahrens 	for (b = ztest_random(100); b > 0; b--) {
1510789Sahrens 		tx = dmu_tx_create(os);
1511789Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1512789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1513789Sahrens 		if (error) {
1514789Sahrens 			ztest_record_enospc("churn objects");
1515789Sahrens 			dmu_tx_abort(tx);
1516789Sahrens 			return;
1517789Sahrens 		}
1518789Sahrens 		object = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1519789Sahrens 		    DMU_OT_NONE, 0, tx);
1520789Sahrens 		ztest_set_random_blocksize(os, object, tx);
1521789Sahrens 		error = dmu_object_free(os, object, tx);
1522789Sahrens 		if (error) {
1523789Sahrens 			fatal(0, "dmu_object_free('%s', %llu) = %d",
1524789Sahrens 			    osname, object, error);
1525789Sahrens 		}
1526789Sahrens 		dmu_tx_commit(tx);
1527789Sahrens 	}
1528789Sahrens 
1529789Sahrens 	/*
1530789Sahrens 	 * Create a new batch of objects with randomly chosen
1531789Sahrens 	 * blocksizes and record them in the batch directory.
1532789Sahrens 	 */
1533789Sahrens 	for (b = 0; b < batchsize; b++) {
1534789Sahrens 		uint32_t va_blksize;
1535789Sahrens 		u_longlong_t va_nblocks;
1536789Sahrens 
1537789Sahrens 		tx = dmu_tx_create(os);
1538789Sahrens 		dmu_tx_hold_write(tx, batchobj, b * sizeof (uint64_t),
1539789Sahrens 		    sizeof (uint64_t));
1540789Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1541789Sahrens 		dmu_tx_hold_write(tx, DMU_NEW_OBJECT, endoff,
1542789Sahrens 		    sizeof (uint64_t));
1543789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1544789Sahrens 		if (error) {
1545789Sahrens 			ztest_record_enospc("create batchobj");
1546789Sahrens 			dmu_tx_abort(tx);
1547789Sahrens 			return;
1548789Sahrens 		}
1549789Sahrens 		bonuslen = (int)ztest_random(dmu_bonus_max()) + 1;
1550789Sahrens 
1551789Sahrens 		object = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1552789Sahrens 		    DMU_OT_PLAIN_OTHER, bonuslen, tx);
1553789Sahrens 
1554789Sahrens 		ztest_set_random_blocksize(os, object, tx);
1555789Sahrens 
1556789Sahrens 		dmu_object_set_checksum(os, object,
1557789Sahrens 		    ztest_random_checksum(), tx);
1558789Sahrens 		dmu_object_set_compress(os, object,
1559789Sahrens 		    ztest_random_compress(), tx);
1560789Sahrens 
1561789Sahrens 		dmu_write(os, batchobj, b * sizeof (uint64_t),
1562789Sahrens 		    sizeof (uint64_t), &object, tx);
1563789Sahrens 
1564789Sahrens 		/*
1565789Sahrens 		 * Write to both the bonus buffer and the regular data.
1566789Sahrens 		 */
1567789Sahrens 		db = dmu_bonus_hold(os, object);
1568789Sahrens 		ASSERT3U(bonuslen, ==, db->db_size);
1569789Sahrens 
1570789Sahrens 		dmu_object_size_from_db(db, &va_blksize, &va_nblocks);
1571789Sahrens 		ASSERT3S(va_nblocks, >=, 0);
1572789Sahrens 
1573789Sahrens 		dmu_buf_will_dirty(db, tx);
1574789Sahrens 
1575789Sahrens 		/*
1576789Sahrens 		 * See comments above regarding the contents of
1577789Sahrens 		 * the bonus buffer and the word at endoff.
1578789Sahrens 		 */
1579789Sahrens 		for (c = 0; c < db->db_size; c++)
1580789Sahrens 			((uint8_t *)db->db_data)[c] = (uint8_t)(c + bonuslen);
1581789Sahrens 
1582789Sahrens 		dmu_buf_rele(db);
1583789Sahrens 
1584789Sahrens 		/*
1585789Sahrens 		 * Write to a large offset to increase indirection.
1586789Sahrens 		 */
1587789Sahrens 		dmu_write(os, object, endoff, sizeof (uint64_t), &object, tx);
1588789Sahrens 
1589789Sahrens 		dmu_tx_commit(tx);
1590789Sahrens 	}
1591789Sahrens }
1592789Sahrens 
1593789Sahrens /*
1594789Sahrens  * Verify that dmu_{read,write} work as expected.
1595789Sahrens  */
1596789Sahrens typedef struct bufwad {
1597789Sahrens 	uint64_t	bw_index;
1598789Sahrens 	uint64_t	bw_txg;
1599789Sahrens 	uint64_t	bw_data;
1600789Sahrens } bufwad_t;
1601789Sahrens 
1602789Sahrens typedef struct dmu_read_write_dir {
1603789Sahrens 	uint64_t	dd_packobj;
1604789Sahrens 	uint64_t	dd_bigobj;
1605789Sahrens 	uint64_t	dd_chunk;
1606789Sahrens } dmu_read_write_dir_t;
1607789Sahrens 
1608789Sahrens void
1609789Sahrens ztest_dmu_read_write(ztest_args_t *za)
1610789Sahrens {
1611789Sahrens 	objset_t *os = za->za_os;
1612789Sahrens 	dmu_read_write_dir_t dd;
1613789Sahrens 	dmu_tx_t *tx;
1614789Sahrens 	int i, freeit, error;
1615789Sahrens 	uint64_t n, s, txg;
1616789Sahrens 	bufwad_t *packbuf, *bigbuf, *pack, *bigH, *bigT;
1617789Sahrens 	uint64_t packoff, packsize, bigoff, bigsize;
1618789Sahrens 	uint64_t regions = 997;
1619789Sahrens 	uint64_t stride = 123456789ULL;
1620789Sahrens 	uint64_t width = 40;
1621789Sahrens 	int free_percent = 5;
1622789Sahrens 
1623789Sahrens 	/*
1624789Sahrens 	 * This test uses two objects, packobj and bigobj, that are always
1625789Sahrens 	 * updated together (i.e. in the same tx) so that their contents are
1626789Sahrens 	 * in sync and can be compared.  Their contents relate to each other
1627789Sahrens 	 * in a simple way: packobj is a dense array of 'bufwad' structures,
1628789Sahrens 	 * while bigobj is a sparse array of the same bufwads.  Specifically,
1629789Sahrens 	 * for any index n, there are three bufwads that should be identical:
1630789Sahrens 	 *
1631789Sahrens 	 *	packobj, at offset n * sizeof (bufwad_t)
1632789Sahrens 	 *	bigobj, at the head of the nth chunk
1633789Sahrens 	 *	bigobj, at the tail of the nth chunk
1634789Sahrens 	 *
1635789Sahrens 	 * The chunk size is arbitrary. It doesn't have to be a power of two,
1636789Sahrens 	 * and it doesn't have any relation to the object blocksize.
1637789Sahrens 	 * The only requirement is that it can hold at least two bufwads.
1638789Sahrens 	 *
1639789Sahrens 	 * Normally, we write the bufwad to each of these locations.
1640789Sahrens 	 * However, free_percent of the time we instead write zeroes to
1641789Sahrens 	 * packobj and perform a dmu_free_range() on bigobj.  By comparing
1642789Sahrens 	 * bigobj to packobj, we can verify that the DMU is correctly
1643789Sahrens 	 * tracking which parts of an object are allocated and free,
1644789Sahrens 	 * and that the contents of the allocated blocks are correct.
1645789Sahrens 	 */
1646789Sahrens 
1647789Sahrens 	/*
1648789Sahrens 	 * Read the directory info.  If it's the first time, set things up.
1649789Sahrens 	 */
1650789Sahrens 	dmu_read(os, ZTEST_DIROBJ, za->za_diroff, sizeof (dd), &dd);
1651789Sahrens 	if (dd.dd_chunk == 0) {
1652789Sahrens 		ASSERT(dd.dd_packobj == 0);
1653789Sahrens 		ASSERT(dd.dd_bigobj == 0);
1654789Sahrens 		tx = dmu_tx_create(os);
1655789Sahrens 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff, sizeof (dd));
1656789Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1657789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1658789Sahrens 		if (error) {
1659789Sahrens 			ztest_record_enospc("create r/w directory");
1660789Sahrens 			dmu_tx_abort(tx);
1661789Sahrens 			return;
1662789Sahrens 		}
1663789Sahrens 
1664789Sahrens 		dd.dd_packobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1665789Sahrens 		    DMU_OT_NONE, 0, tx);
1666789Sahrens 		dd.dd_bigobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1667789Sahrens 		    DMU_OT_NONE, 0, tx);
1668789Sahrens 		dd.dd_chunk = (1000 + ztest_random(1000)) * sizeof (uint64_t);
1669789Sahrens 
1670789Sahrens 		ztest_set_random_blocksize(os, dd.dd_packobj, tx);
1671789Sahrens 		ztest_set_random_blocksize(os, dd.dd_bigobj, tx);
1672789Sahrens 
1673789Sahrens 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff, sizeof (dd), &dd,
1674789Sahrens 		    tx);
1675789Sahrens 		dmu_tx_commit(tx);
1676789Sahrens 	}
1677789Sahrens 
1678789Sahrens 	/*
1679789Sahrens 	 * Prefetch a random chunk of the big object.
1680789Sahrens 	 * Our aim here is to get some async reads in flight
1681789Sahrens 	 * for blocks that we may free below; the DMU should
1682789Sahrens 	 * handle this race correctly.
1683789Sahrens 	 */
1684789Sahrens 	n = ztest_random(regions) * stride + ztest_random(width);
1685789Sahrens 	s = 1 + ztest_random(2 * width - 1);
1686789Sahrens 	dmu_prefetch(os, dd.dd_bigobj, n * dd.dd_chunk, s * dd.dd_chunk);
1687789Sahrens 
1688789Sahrens 	/*
1689789Sahrens 	 * Pick a random index and compute the offsets into packobj and bigobj.
1690789Sahrens 	 */
1691789Sahrens 	n = ztest_random(regions) * stride + ztest_random(width);
1692789Sahrens 	s = 1 + ztest_random(width - 1);
1693789Sahrens 
1694789Sahrens 	packoff = n * sizeof (bufwad_t);
1695789Sahrens 	packsize = s * sizeof (bufwad_t);
1696789Sahrens 
1697789Sahrens 	bigoff = n * dd.dd_chunk;
1698789Sahrens 	bigsize = s * dd.dd_chunk;
1699789Sahrens 
1700789Sahrens 	packbuf = umem_alloc(packsize, UMEM_NOFAIL);
1701789Sahrens 	bigbuf = umem_alloc(bigsize, UMEM_NOFAIL);
1702789Sahrens 
1703789Sahrens 	/*
1704789Sahrens 	 * free_percent of the time, free a range of bigobj rather than
1705789Sahrens 	 * overwriting it.
1706789Sahrens 	 */
1707789Sahrens 	freeit = (ztest_random(100) < free_percent);
1708789Sahrens 
1709789Sahrens 	/*
1710789Sahrens 	 * Read the current contents of our objects.
1711789Sahrens 	 */
1712789Sahrens 	dmu_read(os, dd.dd_packobj, packoff, packsize, packbuf);
1713789Sahrens 	dmu_read(os, dd.dd_bigobj, bigoff, bigsize, bigbuf);
1714789Sahrens 
1715789Sahrens 	/*
1716789Sahrens 	 * Get a tx for the mods to both packobj and bigobj.
1717789Sahrens 	 */
1718789Sahrens 	tx = dmu_tx_create(os);
1719789Sahrens 
1720789Sahrens 	dmu_tx_hold_write(tx, dd.dd_packobj, packoff, packsize);
1721789Sahrens 
1722789Sahrens 	if (freeit)
1723789Sahrens 		dmu_tx_hold_free(tx, dd.dd_bigobj, bigoff, bigsize);
1724789Sahrens 	else
1725789Sahrens 		dmu_tx_hold_write(tx, dd.dd_bigobj, bigoff, bigsize);
1726789Sahrens 
1727789Sahrens 	error = dmu_tx_assign(tx, TXG_WAIT);
1728789Sahrens 
1729789Sahrens 	if (error) {
1730789Sahrens 		ztest_record_enospc("dmu r/w range");
1731789Sahrens 		dmu_tx_abort(tx);
1732789Sahrens 		umem_free(packbuf, packsize);
1733789Sahrens 		umem_free(bigbuf, bigsize);
1734789Sahrens 		return;
1735789Sahrens 	}
1736789Sahrens 
1737789Sahrens 	txg = dmu_tx_get_txg(tx);
1738789Sahrens 
1739789Sahrens 	/*
1740789Sahrens 	 * For each index from n to n + s, verify that the existing bufwad
1741789Sahrens 	 * in packobj matches the bufwads at the head and tail of the
1742789Sahrens 	 * corresponding chunk in bigobj.  Then update all three bufwads
1743789Sahrens 	 * with the new values we want to write out.
1744789Sahrens 	 */
1745789Sahrens 	for (i = 0; i < s; i++) {
1746789Sahrens 		/* LINTED */
1747789Sahrens 		pack = (bufwad_t *)((char *)packbuf + i * sizeof (bufwad_t));
1748789Sahrens 		/* LINTED */
1749789Sahrens 		bigH = (bufwad_t *)((char *)bigbuf + i * dd.dd_chunk);
1750789Sahrens 		/* LINTED */
1751789Sahrens 		bigT = (bufwad_t *)((char *)bigH + dd.dd_chunk) - 1;
1752789Sahrens 
1753789Sahrens 		ASSERT((uintptr_t)bigH - (uintptr_t)bigbuf < bigsize);
1754789Sahrens 		ASSERT((uintptr_t)bigT - (uintptr_t)bigbuf < bigsize);
1755789Sahrens 
1756789Sahrens 		if (pack->bw_txg > txg)
1757789Sahrens 			fatal(0, "future leak: got %llx, open txg is %llx",
1758789Sahrens 			    pack->bw_txg, txg);
1759789Sahrens 
1760789Sahrens 		if (pack->bw_data != 0 && pack->bw_index != n + i)
1761789Sahrens 			fatal(0, "wrong index: got %llx, wanted %llx+%llx",
1762789Sahrens 			    pack->bw_index, n, i);
1763789Sahrens 
1764789Sahrens 		if (bcmp(pack, bigH, sizeof (bufwad_t)) != 0)
1765789Sahrens 			fatal(0, "pack/bigH mismatch in %p/%p", pack, bigH);
1766789Sahrens 
1767789Sahrens 		if (bcmp(pack, bigT, sizeof (bufwad_t)) != 0)
1768789Sahrens 			fatal(0, "pack/bigT mismatch in %p/%p", pack, bigT);
1769789Sahrens 
1770789Sahrens 		if (freeit) {
1771789Sahrens 			bzero(pack, sizeof (bufwad_t));
1772789Sahrens 		} else {
1773789Sahrens 			pack->bw_index = n + i;
1774789Sahrens 			pack->bw_txg = txg;
1775789Sahrens 			pack->bw_data = 1 + ztest_random(-2ULL);
1776789Sahrens 		}
1777789Sahrens 		*bigH = *pack;
1778789Sahrens 		*bigT = *pack;
1779789Sahrens 	}
1780789Sahrens 
1781789Sahrens 	/*
1782789Sahrens 	 * We've verified all the old bufwads, and made new ones.
1783789Sahrens 	 * Now write them out.
1784789Sahrens 	 */
1785789Sahrens 	dmu_write(os, dd.dd_packobj, packoff, packsize, packbuf, tx);
1786789Sahrens 
1787789Sahrens 	if (freeit) {
1788789Sahrens 		if (zopt_verbose >= 6) {
1789789Sahrens 			(void) printf("freeing offset %llx size %llx"
1790789Sahrens 			    " txg %llx\n",
1791789Sahrens 			    (u_longlong_t)bigoff,
1792789Sahrens 			    (u_longlong_t)bigsize,
1793789Sahrens 			    (u_longlong_t)txg);
1794789Sahrens 		}
1795789Sahrens 		dmu_free_range(os, dd.dd_bigobj, bigoff, bigsize, tx);
1796789Sahrens 	} else {
1797789Sahrens 		if (zopt_verbose >= 6) {
1798789Sahrens 			(void) printf("writing offset %llx size %llx"
1799789Sahrens 			    " txg %llx\n",
1800789Sahrens 			    (u_longlong_t)bigoff,
1801789Sahrens 			    (u_longlong_t)bigsize,
1802789Sahrens 			    (u_longlong_t)txg);
1803789Sahrens 		}
1804789Sahrens 		dmu_write(os, dd.dd_bigobj, bigoff, bigsize, bigbuf, tx);
1805789Sahrens 	}
1806789Sahrens 
1807789Sahrens 	dmu_tx_commit(tx);
1808789Sahrens 
1809789Sahrens 	/*
1810789Sahrens 	 * Sanity check the stuff we just wrote.
1811789Sahrens 	 */
1812789Sahrens 	{
1813789Sahrens 		void *packcheck = umem_alloc(packsize, UMEM_NOFAIL);
1814789Sahrens 		void *bigcheck = umem_alloc(bigsize, UMEM_NOFAIL);
1815789Sahrens 
1816789Sahrens 		dmu_read(os, dd.dd_packobj, packoff, packsize, packcheck);
1817789Sahrens 		dmu_read(os, dd.dd_bigobj, bigoff, bigsize, bigcheck);
1818789Sahrens 
1819789Sahrens 		ASSERT(bcmp(packbuf, packcheck, packsize) == 0);
1820789Sahrens 		ASSERT(bcmp(bigbuf, bigcheck, bigsize) == 0);
1821789Sahrens 
1822789Sahrens 		umem_free(packcheck, packsize);
1823789Sahrens 		umem_free(bigcheck, bigsize);
1824789Sahrens 	}
1825789Sahrens 
1826789Sahrens 	umem_free(packbuf, packsize);
1827789Sahrens 	umem_free(bigbuf, bigsize);
1828789Sahrens }
1829789Sahrens 
1830789Sahrens void
1831789Sahrens ztest_dmu_write_parallel(ztest_args_t *za)
1832789Sahrens {
1833789Sahrens 	objset_t *os = za->za_os;
1834789Sahrens 	dmu_tx_t *tx;
1835789Sahrens 	dmu_buf_t *db;
1836789Sahrens 	int i, b, error, do_free, bs;
1837789Sahrens 	uint64_t off, txg_how, txg;
1838789Sahrens 	mutex_t *lp;
1839789Sahrens 	char osname[MAXNAMELEN];
1840789Sahrens 	char iobuf[SPA_MAXBLOCKSIZE];
1841789Sahrens 	ztest_block_tag_t rbt, wbt;
1842789Sahrens 
1843789Sahrens 	dmu_objset_name(os, osname);
1844789Sahrens 	bs = ZTEST_DIROBJ_BLOCKSIZE;
1845789Sahrens 
1846789Sahrens 	/*
1847789Sahrens 	 * Have multiple threads write to large offsets in ZTEST_DIROBJ
1848789Sahrens 	 * to verify that having multiple threads writing to the same object
1849789Sahrens 	 * in parallel doesn't cause any trouble.
1850789Sahrens 	 * Also do parallel writes to the bonus buffer on occasion.
1851789Sahrens 	 */
1852789Sahrens 	for (i = 0; i < 50; i++) {
1853789Sahrens 		b = ztest_random(ZTEST_SYNC_LOCKS);
1854789Sahrens 		lp = &ztest_shared->zs_sync_lock[b];
1855789Sahrens 
1856789Sahrens 		do_free = (ztest_random(4) == 0);
1857789Sahrens 
1858789Sahrens 		off = za->za_diroff_shared + ((uint64_t)b << SPA_MAXBLOCKSHIFT);
1859789Sahrens 
1860789Sahrens 		if (ztest_random(4) == 0) {
1861789Sahrens 			/*
1862789Sahrens 			 * Do the bonus buffer instead of a regular block.
1863789Sahrens 			 */
1864789Sahrens 			do_free = 0;
1865789Sahrens 			off = -1ULL;
1866789Sahrens 		}
1867789Sahrens 
1868789Sahrens 		tx = dmu_tx_create(os);
1869789Sahrens 
1870789Sahrens 		if (off == -1ULL)
1871789Sahrens 			dmu_tx_hold_bonus(tx, ZTEST_DIROBJ);
1872789Sahrens 		else if (do_free)
1873789Sahrens 			dmu_tx_hold_free(tx, ZTEST_DIROBJ, off, bs);
1874789Sahrens 		else
1875789Sahrens 			dmu_tx_hold_write(tx, ZTEST_DIROBJ, off, bs);
1876789Sahrens 
1877789Sahrens 		txg_how = ztest_random(2) == 0 ? TXG_WAIT : TXG_NOWAIT;
1878789Sahrens 		error = dmu_tx_assign(tx, txg_how);
1879789Sahrens 		if (error) {
1880789Sahrens 			dmu_tx_abort(tx);
1881789Sahrens 			if (error == ERESTART) {
1882789Sahrens 				ASSERT(txg_how == TXG_NOWAIT);
1883789Sahrens 				txg_wait_open(dmu_objset_pool(os), 0);
1884789Sahrens 				continue;
1885789Sahrens 			}
1886789Sahrens 			ztest_record_enospc("dmu write parallel");
1887789Sahrens 			return;
1888789Sahrens 		}
1889789Sahrens 		txg = dmu_tx_get_txg(tx);
1890789Sahrens 
1891789Sahrens 		if (do_free) {
1892789Sahrens 			(void) mutex_lock(lp);
1893789Sahrens 			dmu_free_range(os, ZTEST_DIROBJ, off, bs, tx);
1894789Sahrens 			(void) mutex_unlock(lp);
1895789Sahrens 			dmu_tx_commit(tx);
1896789Sahrens 			continue;
1897789Sahrens 		}
1898789Sahrens 
1899789Sahrens 		wbt.bt_objset = dmu_objset_id(os);
1900789Sahrens 		wbt.bt_object = ZTEST_DIROBJ;
1901789Sahrens 		wbt.bt_offset = off;
1902789Sahrens 		wbt.bt_txg = txg;
1903789Sahrens 		wbt.bt_thread = za->za_instance;
1904789Sahrens 
1905789Sahrens 		if (off == -1ULL) {
1906789Sahrens 			wbt.bt_seq = 0;
1907789Sahrens 			db = dmu_bonus_hold(os, ZTEST_DIROBJ);
1908789Sahrens 			ASSERT3U(db->db_size, ==, sizeof (wbt));
1909789Sahrens 			dmu_buf_read(db);
1910789Sahrens 			bcopy(db->db_data, &rbt, db->db_size);
1911789Sahrens 			if (rbt.bt_objset != 0) {
1912789Sahrens 				ASSERT3U(rbt.bt_objset, ==, wbt.bt_objset);
1913789Sahrens 				ASSERT3U(rbt.bt_object, ==, wbt.bt_object);
1914789Sahrens 				ASSERT3U(rbt.bt_offset, ==, wbt.bt_offset);
1915789Sahrens 				ASSERT3U(rbt.bt_txg, <=, wbt.bt_txg);
1916789Sahrens 			}
1917789Sahrens 			dmu_buf_will_dirty(db, tx);
1918789Sahrens 			bcopy(&wbt, db->db_data, db->db_size);
1919789Sahrens 			dmu_buf_rele(db);
1920789Sahrens 			dmu_tx_commit(tx);
1921789Sahrens 			continue;
1922789Sahrens 		}
1923789Sahrens 
1924789Sahrens 		(void) mutex_lock(lp);
1925789Sahrens 
1926789Sahrens 		wbt.bt_seq = ztest_shared->zs_seq[b]++;
1927789Sahrens 
1928789Sahrens 		dmu_write(os, ZTEST_DIROBJ, off, sizeof (wbt), &wbt, tx);
1929789Sahrens 
1930789Sahrens 		(void) mutex_unlock(lp);
1931789Sahrens 
1932789Sahrens 		if (ztest_random(100) == 0)
1933789Sahrens 			(void) poll(NULL, 0, 1); /* open dn_notxholds window */
1934789Sahrens 
1935789Sahrens 		dmu_tx_commit(tx);
1936789Sahrens 
1937789Sahrens 		if (ztest_random(1000) == 0)
1938789Sahrens 			txg_wait_synced(dmu_objset_pool(os), txg);
1939789Sahrens 
1940789Sahrens 		if (ztest_random(2) == 0) {
1941789Sahrens 			blkptr_t blk = { 0 };
1942789Sahrens 			uint64_t blkoff;
1943789Sahrens 
1944789Sahrens 			txg_suspend(dmu_objset_pool(os));
1945789Sahrens 			(void) mutex_lock(lp);
1946789Sahrens 			error = dmu_sync(os, ZTEST_DIROBJ, off, &blkoff, &blk,
1947789Sahrens 			    txg);
1948789Sahrens 			(void) mutex_unlock(lp);
1949789Sahrens 			if (error) {
1950789Sahrens 				txg_resume(dmu_objset_pool(os));
1951789Sahrens 				dprintf("dmu_sync(%s, %d, %llx) = %d\n",
1952789Sahrens 				    osname, ZTEST_DIROBJ, off, error);
1953789Sahrens 				continue;
1954789Sahrens 			}
1955789Sahrens 
1956789Sahrens 			if (blk.blk_birth == 0)	{	/* concurrent free */
1957789Sahrens 				txg_resume(dmu_objset_pool(os));
1958789Sahrens 				continue;
1959789Sahrens 			}
1960789Sahrens 
1961789Sahrens 			ASSERT(blk.blk_fill == 1);
1962789Sahrens 			ASSERT3U(BP_GET_TYPE(&blk), ==, DMU_OT_UINT64_OTHER);
1963789Sahrens 			ASSERT3U(BP_GET_LEVEL(&blk), ==, 0);
1964789Sahrens 			ASSERT3U(BP_GET_LSIZE(&blk), ==, bs);
1965789Sahrens 
1966789Sahrens 			/*
1967789Sahrens 			 * Read the block that dmu_sync() returned to
1968789Sahrens 			 * make sure its contents match what we wrote.
1969789Sahrens 			 * We do this while still txg_suspend()ed to ensure
1970789Sahrens 			 * that the block can't be reused before we read it.
1971789Sahrens 			 */
1972789Sahrens 			error = zio_wait(zio_read(NULL, dmu_objset_spa(os),
1973789Sahrens 			    &blk, iobuf, bs, NULL, NULL,
1974789Sahrens 			    ZIO_PRIORITY_SYNC_READ, ZIO_FLAG_MUSTSUCCEED));
1975789Sahrens 			ASSERT(error == 0);
1976789Sahrens 
1977789Sahrens 			txg_resume(dmu_objset_pool(os));
1978789Sahrens 
1979789Sahrens 			bcopy(&iobuf[blkoff], &rbt, sizeof (rbt));
1980789Sahrens 
1981789Sahrens 			ASSERT3U(rbt.bt_objset, ==, wbt.bt_objset);
1982789Sahrens 			ASSERT3U(rbt.bt_object, ==, wbt.bt_object);
1983789Sahrens 			ASSERT3U(rbt.bt_offset, ==, wbt.bt_offset);
1984789Sahrens 
1985789Sahrens 			/*
1986789Sahrens 			 * The semantic of dmu_sync() is that we always
1987789Sahrens 			 * push the most recent version of the data,
1988789Sahrens 			 * so in the face of concurrent updates we may
1989789Sahrens 			 * see a newer version of the block.  That's OK.
1990789Sahrens 			 */
1991789Sahrens 			ASSERT3U(rbt.bt_txg, >=, wbt.bt_txg);
1992789Sahrens 			if (rbt.bt_thread == wbt.bt_thread)
1993789Sahrens 				ASSERT3U(rbt.bt_seq, ==, wbt.bt_seq);
1994789Sahrens 			else
1995789Sahrens 				ASSERT3U(rbt.bt_seq, >, wbt.bt_seq);
1996789Sahrens 		}
1997789Sahrens 	}
1998789Sahrens }
1999789Sahrens 
2000789Sahrens /*
2001789Sahrens  * Verify that zap_{create,destroy,add,remove,update} work as expected.
2002789Sahrens  */
2003789Sahrens #define	ZTEST_ZAP_MIN_INTS	1
2004789Sahrens #define	ZTEST_ZAP_MAX_INTS	4
2005789Sahrens #define	ZTEST_ZAP_MAX_PROPS	1000
2006789Sahrens 
2007789Sahrens void
2008789Sahrens ztest_zap(ztest_args_t *za)
2009789Sahrens {
2010789Sahrens 	objset_t *os = za->za_os;
2011789Sahrens 	uint64_t object;
2012789Sahrens 	uint64_t txg, last_txg;
2013789Sahrens 	uint64_t value[ZTEST_ZAP_MAX_INTS];
2014789Sahrens 	uint64_t zl_ints, zl_intsize, prop;
2015789Sahrens 	int i, ints;
2016789Sahrens 	int iters = 100;
2017789Sahrens 	dmu_tx_t *tx;
2018789Sahrens 	char propname[100], txgname[100];
2019789Sahrens 	int error;
2020789Sahrens 	char osname[MAXNAMELEN];
2021789Sahrens 	char *hc[2] = { "s.acl.h", ".s.open.h.hyLZlg" };
2022789Sahrens 
2023789Sahrens 	dmu_objset_name(os, osname);
2024789Sahrens 
2025789Sahrens 	/*
2026789Sahrens 	 * Create a new object if necessary, and record it in the directory.
2027789Sahrens 	 */
2028789Sahrens 	dmu_read(os, ZTEST_DIROBJ, za->za_diroff, sizeof (uint64_t), &object);
2029789Sahrens 
2030789Sahrens 	if (object == 0) {
2031789Sahrens 		tx = dmu_tx_create(os);
2032789Sahrens 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff,
2033789Sahrens 		    sizeof (uint64_t));
2034789Sahrens 		dmu_tx_hold_zap(tx, DMU_NEW_OBJECT, 2);
2035789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
2036789Sahrens 		if (error) {
2037789Sahrens 			ztest_record_enospc("create zap test obj");
2038789Sahrens 			dmu_tx_abort(tx);
2039789Sahrens 			return;
2040789Sahrens 		}
2041789Sahrens 		object = zap_create(os, DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx);
2042789Sahrens 		if (error) {
2043789Sahrens 			fatal(0, "zap_create('%s', %llu) = %d",
2044789Sahrens 			    osname, object, error);
2045789Sahrens 		}
2046789Sahrens 		ASSERT(object != 0);
2047789Sahrens 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff,
2048789Sahrens 		    sizeof (uint64_t), &object, tx);
2049789Sahrens 		/*
2050789Sahrens 		 * Generate a known hash collision, and verify that
2051789Sahrens 		 * we can lookup and remove both entries.
2052789Sahrens 		 */
2053789Sahrens 		for (i = 0; i < 2; i++) {
2054789Sahrens 			value[i] = i;
2055789Sahrens 			error = zap_add(os, object, hc[i], sizeof (uint64_t),
2056789Sahrens 			    1, &value[i], tx);
2057789Sahrens 			ASSERT3U(error, ==, 0);
2058789Sahrens 		}
2059789Sahrens 		for (i = 0; i < 2; i++) {
2060789Sahrens 			error = zap_add(os, object, hc[i], sizeof (uint64_t),
2061789Sahrens 			    1, &value[i], tx);
2062789Sahrens 			ASSERT3U(error, ==, EEXIST);
2063789Sahrens 			error = zap_length(os, object, hc[i],
2064789Sahrens 			    &zl_intsize, &zl_ints);
2065789Sahrens 			ASSERT3U(error, ==, 0);
2066789Sahrens 			ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
2067789Sahrens 			ASSERT3U(zl_ints, ==, 1);
2068789Sahrens 		}
2069789Sahrens 		for (i = 0; i < 2; i++) {
2070789Sahrens 			error = zap_remove(os, object, hc[i], tx);
2071789Sahrens 			ASSERT3U(error, ==, 0);
2072789Sahrens 		}
2073789Sahrens 
2074789Sahrens 		dmu_tx_commit(tx);
2075789Sahrens 	}
2076789Sahrens 
2077789Sahrens 	ints = MAX(ZTEST_ZAP_MIN_INTS, object % ZTEST_ZAP_MAX_INTS);
2078789Sahrens 
2079789Sahrens 	while (--iters >= 0) {
2080789Sahrens 		prop = ztest_random(ZTEST_ZAP_MAX_PROPS);
2081789Sahrens 		(void) sprintf(propname, "prop_%llu", (u_longlong_t)prop);
2082789Sahrens 		(void) sprintf(txgname, "txg_%llu", (u_longlong_t)prop);
2083789Sahrens 		bzero(value, sizeof (value));
2084789Sahrens 		last_txg = 0;
2085789Sahrens 
2086789Sahrens 		/*
2087789Sahrens 		 * If these zap entries already exist, validate their contents.
2088789Sahrens 		 */
2089789Sahrens 		error = zap_length(os, object, txgname, &zl_intsize, &zl_ints);
2090789Sahrens 		if (error == 0) {
2091789Sahrens 			ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
2092789Sahrens 			ASSERT3U(zl_ints, ==, 1);
2093789Sahrens 
2094789Sahrens 			error = zap_lookup(os, object, txgname, zl_intsize,
2095789Sahrens 			    zl_ints, &last_txg);
2096789Sahrens 
2097789Sahrens 			ASSERT3U(error, ==, 0);
2098789Sahrens 
2099789Sahrens 			error = zap_length(os, object, propname, &zl_intsize,
2100789Sahrens 			    &zl_ints);
2101789Sahrens 
2102789Sahrens 			ASSERT3U(error, ==, 0);
2103789Sahrens 			ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
2104789Sahrens 			ASSERT3U(zl_ints, ==, ints);
2105789Sahrens 
2106789Sahrens 			error = zap_lookup(os, object, propname, zl_intsize,
2107789Sahrens 			    zl_ints, value);
2108789Sahrens 
2109789Sahrens 			ASSERT3U(error, ==, 0);
2110789Sahrens 
2111789Sahrens 			for (i = 0; i < ints; i++) {
2112789Sahrens 				ASSERT3U(value[i], ==, last_txg + object + i);
2113789Sahrens 			}
2114789Sahrens 		} else {
2115789Sahrens 			ASSERT3U(error, ==, ENOENT);
2116789Sahrens 		}
2117789Sahrens 
2118789Sahrens 		/*
2119789Sahrens 		 * Atomically update two entries in our zap object.
2120789Sahrens 		 * The first is named txg_%llu, and contains the txg
2121789Sahrens 		 * in which the property was last updated.  The second
2122789Sahrens 		 * is named prop_%llu, and the nth element of its value
2123789Sahrens 		 * should be txg + object + n.
2124789Sahrens 		 */
2125789Sahrens 		tx = dmu_tx_create(os);
2126789Sahrens 		dmu_tx_hold_zap(tx, object, 2);
2127789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
2128789Sahrens 		if (error) {
2129789Sahrens 			ztest_record_enospc("create zap entry");
2130789Sahrens 			dmu_tx_abort(tx);
2131789Sahrens 			return;
2132789Sahrens 		}
2133789Sahrens 		txg = dmu_tx_get_txg(tx);
2134789Sahrens 
2135789Sahrens 		if (last_txg > txg)
2136789Sahrens 			fatal(0, "zap future leak: old %llu new %llu",
2137789Sahrens 			    last_txg, txg);
2138789Sahrens 
2139789Sahrens 		for (i = 0; i < ints; i++)
2140789Sahrens 			value[i] = txg + object + i;
2141789Sahrens 
2142789Sahrens 		error = zap_update(os, object, txgname, sizeof (uint64_t),
2143789Sahrens 		    1, &txg, tx);
2144789Sahrens 		if (error)
2145789Sahrens 			fatal(0, "zap_update('%s', %llu, '%s') = %d",
2146789Sahrens 			    osname, object, txgname, error);
2147789Sahrens 
2148789Sahrens 		error = zap_update(os, object, propname, sizeof (uint64_t),
2149789Sahrens 		    ints, value, tx);
2150789Sahrens 		if (error)
2151789Sahrens 			fatal(0, "zap_update('%s', %llu, '%s') = %d",
2152789Sahrens 			    osname, object, propname, error);
2153789Sahrens 
2154789Sahrens 		dmu_tx_commit(tx);
2155789Sahrens 
2156789Sahrens 		/*
2157789Sahrens 		 * Remove a random pair of entries.
2158789Sahrens 		 */
2159789Sahrens 		prop = ztest_random(ZTEST_ZAP_MAX_PROPS);
2160789Sahrens 		(void) sprintf(propname, "prop_%llu", (u_longlong_t)prop);
2161789Sahrens 		(void) sprintf(txgname, "txg_%llu", (u_longlong_t)prop);
2162789Sahrens 
2163789Sahrens 		error = zap_length(os, object, txgname, &zl_intsize, &zl_ints);
2164789Sahrens 
2165789Sahrens 		if (error == ENOENT)
2166789Sahrens 			continue;
2167789Sahrens 
2168789Sahrens 		ASSERT3U(error, ==, 0);
2169789Sahrens 
2170789Sahrens 		tx = dmu_tx_create(os);
2171789Sahrens 		dmu_tx_hold_zap(tx, object, 2);
2172789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
2173789Sahrens 		if (error) {
2174789Sahrens 			ztest_record_enospc("remove zap entry");
2175789Sahrens 			dmu_tx_abort(tx);
2176789Sahrens 			return;
2177789Sahrens 		}
2178789Sahrens 		error = zap_remove(os, object, txgname, tx);
2179789Sahrens 		if (error)
2180789Sahrens 			fatal(0, "zap_remove('%s', %llu, '%s') = %d",
2181789Sahrens 			    osname, object, txgname, error);
2182789Sahrens 
2183789Sahrens 		error = zap_remove(os, object, propname, tx);
2184789Sahrens 		if (error)
2185789Sahrens 			fatal(0, "zap_remove('%s', %llu, '%s') = %d",
2186789Sahrens 			    osname, object, propname, error);
2187789Sahrens 
2188789Sahrens 		dmu_tx_commit(tx);
2189789Sahrens 	}
2190789Sahrens 
2191789Sahrens 	/*
2192789Sahrens 	 * Once in a while, destroy the object.
2193789Sahrens 	 */
2194789Sahrens 	if (ztest_random(100) != 0)
2195789Sahrens 		return;
2196789Sahrens 
2197789Sahrens 	tx = dmu_tx_create(os);
2198789Sahrens 	dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff, sizeof (uint64_t));
2199789Sahrens 	dmu_tx_hold_free(tx, object, 0, DMU_OBJECT_END);
2200789Sahrens 	error = dmu_tx_assign(tx, TXG_WAIT);
2201789Sahrens 	if (error) {
2202789Sahrens 		ztest_record_enospc("destroy zap object");
2203789Sahrens 		dmu_tx_abort(tx);
2204789Sahrens 		return;
2205789Sahrens 	}
2206789Sahrens 	error = zap_destroy(os, object, tx);
2207789Sahrens 	if (error)
2208789Sahrens 		fatal(0, "zap_destroy('%s', %llu) = %d",
2209789Sahrens 		    osname, object, error);
2210789Sahrens 	object = 0;
2211789Sahrens 	dmu_write(os, ZTEST_DIROBJ, za->za_diroff, sizeof (uint64_t),
2212789Sahrens 	    &object, tx);
2213789Sahrens 	dmu_tx_commit(tx);
2214789Sahrens }
2215789Sahrens 
2216789Sahrens void
2217789Sahrens ztest_zap_parallel(ztest_args_t *za)
2218789Sahrens {
2219789Sahrens 	objset_t *os = za->za_os;
2220789Sahrens 	uint64_t txg, object, count, wsize, wc, zl_wsize, zl_wc;
2221789Sahrens 	int iters = 100;
2222789Sahrens 	dmu_tx_t *tx;
2223789Sahrens 	int i, namelen, error;
2224789Sahrens 	char name[20], string_value[20];
2225789Sahrens 	void *data;
2226789Sahrens 
2227789Sahrens 	while (--iters >= 0) {
2228789Sahrens 		/*
2229789Sahrens 		 * Generate a random name of the form 'xxx.....' where each
2230789Sahrens 		 * x is a random printable character and the dots are dots.
2231789Sahrens 		 * There are 94 such characters, and the name length goes from
2232789Sahrens 		 * 6 to 20, so there are 94^3 * 15 = 12,458,760 possible names.
2233789Sahrens 		 */
2234789Sahrens 		namelen = ztest_random(sizeof (name) - 5) + 5 + 1;
2235789Sahrens 
2236789Sahrens 		for (i = 0; i < 3; i++)
2237789Sahrens 			name[i] = '!' + ztest_random('~' - '!' + 1);
2238789Sahrens 		for (; i < namelen - 1; i++)
2239789Sahrens 			name[i] = '.';
2240789Sahrens 		name[i] = '\0';
2241789Sahrens 
2242789Sahrens 		if (ztest_random(2) == 0)
2243789Sahrens 			object = ZTEST_MICROZAP_OBJ;
2244789Sahrens 		else
2245789Sahrens 			object = ZTEST_FATZAP_OBJ;
2246789Sahrens 
2247789Sahrens 		if ((namelen & 1) || object == ZTEST_MICROZAP_OBJ) {
2248789Sahrens 			wsize = sizeof (txg);
2249789Sahrens 			wc = 1;
2250789Sahrens 			data = &txg;
2251789Sahrens 		} else {
2252789Sahrens 			wsize = 1;
2253789Sahrens 			wc = namelen;
2254789Sahrens 			data = string_value;
2255789Sahrens 		}
2256789Sahrens 
2257789Sahrens 		count = -1ULL;
2258789Sahrens 		VERIFY(zap_count(os, object, &count) == 0);
2259789Sahrens 		ASSERT(count != -1ULL);
2260789Sahrens 
2261789Sahrens 		/*
2262789Sahrens 		 * Select an operation: length, lookup, add, update, remove.
2263789Sahrens 		 */
2264789Sahrens 		i = ztest_random(5);
2265789Sahrens 
2266789Sahrens 		if (i >= 2) {
2267789Sahrens 			tx = dmu_tx_create(os);
2268789Sahrens 			dmu_tx_hold_zap(tx, object, 1);
2269789Sahrens 			error = dmu_tx_assign(tx, TXG_WAIT);
2270789Sahrens 			if (error) {
2271789Sahrens 				ztest_record_enospc("zap parallel");
2272789Sahrens 				dmu_tx_abort(tx);
2273789Sahrens 				return;
2274789Sahrens 			}
2275789Sahrens 			txg = dmu_tx_get_txg(tx);
2276789Sahrens 			bcopy(name, string_value, namelen);
2277789Sahrens 		} else {
2278789Sahrens 			tx = NULL;
2279789Sahrens 			txg = 0;
2280789Sahrens 			bzero(string_value, namelen);
2281789Sahrens 		}
2282789Sahrens 
2283789Sahrens 		switch (i) {
2284789Sahrens 
2285789Sahrens 		case 0:
2286789Sahrens 			error = zap_length(os, object, name, &zl_wsize, &zl_wc);
2287789Sahrens 			if (error == 0) {
2288789Sahrens 				ASSERT3U(wsize, ==, zl_wsize);
2289789Sahrens 				ASSERT3U(wc, ==, zl_wc);
2290789Sahrens 			} else {
2291789Sahrens 				ASSERT3U(error, ==, ENOENT);
2292789Sahrens 			}
2293789Sahrens 			break;
2294789Sahrens 
2295789Sahrens 		case 1:
2296789Sahrens 			error = zap_lookup(os, object, name, wsize, wc, data);
2297789Sahrens 			if (error == 0) {
2298789Sahrens 				if (data == string_value &&
2299789Sahrens 				    bcmp(name, data, namelen) != 0)
2300789Sahrens 					fatal(0, "name '%s' != val '%s' len %d",
2301789Sahrens 					    name, data, namelen);
2302789Sahrens 			} else {
2303789Sahrens 				ASSERT3U(error, ==, ENOENT);
2304789Sahrens 			}
2305789Sahrens 			break;
2306789Sahrens 
2307789Sahrens 		case 2:
2308789Sahrens 			error = zap_add(os, object, name, wsize, wc, data, tx);
2309789Sahrens 			ASSERT(error == 0 || error == EEXIST);
2310789Sahrens 			break;
2311789Sahrens 
2312789Sahrens 		case 3:
2313789Sahrens 			VERIFY(zap_update(os, object, name, wsize, wc,
2314789Sahrens 			    data, tx) == 0);
2315789Sahrens 			break;
2316789Sahrens 
2317789Sahrens 		case 4:
2318789Sahrens 			error = zap_remove(os, object, name, tx);
2319789Sahrens 			ASSERT(error == 0 || error == ENOENT);
2320789Sahrens 			break;
2321789Sahrens 		}
2322789Sahrens 
2323789Sahrens 		if (tx != NULL)
2324789Sahrens 			dmu_tx_commit(tx);
2325789Sahrens 	}
2326789Sahrens }
2327789Sahrens 
2328789Sahrens void
2329789Sahrens ztest_dsl_prop_get_set(ztest_args_t *za)
2330789Sahrens {
2331789Sahrens 	objset_t *os = za->za_os;
2332789Sahrens 	int i, inherit;
2333789Sahrens 	uint64_t value;
2334789Sahrens 	const char *prop, *valname;
2335789Sahrens 	char setpoint[MAXPATHLEN];
2336789Sahrens 	char osname[MAXNAMELEN];
2337789Sahrens 
2338789Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
2339789Sahrens 
2340789Sahrens 	dmu_objset_name(os, osname);
2341789Sahrens 
2342789Sahrens 	for (i = 0; i < 2; i++) {
2343789Sahrens 		if (i == 0) {
2344789Sahrens 			prop = "checksum";
2345789Sahrens 			value = ztest_random_checksum();
2346789Sahrens 			inherit = (value == ZIO_CHECKSUM_INHERIT);
2347789Sahrens 		} else {
2348789Sahrens 			prop = "compression";
2349789Sahrens 			value = ztest_random_compress();
2350789Sahrens 			inherit = (value == ZIO_COMPRESS_INHERIT);
2351789Sahrens 		}
2352789Sahrens 
2353789Sahrens 		VERIFY3U(dsl_prop_set(osname, prop, sizeof (value),
2354789Sahrens 		    !inherit, &value), ==, 0);
2355789Sahrens 
2356789Sahrens 		VERIFY3U(dsl_prop_get(osname, prop, sizeof (value),
2357789Sahrens 		    1, &value, setpoint), ==, 0);
2358789Sahrens 
2359789Sahrens 		if (i == 0)
2360789Sahrens 			valname = zio_checksum_table[value].ci_name;
2361789Sahrens 		else
2362789Sahrens 			valname = zio_compress_table[value].ci_name;
2363789Sahrens 
2364789Sahrens 		if (zopt_verbose >= 6) {
2365789Sahrens 			(void) printf("%s %s = %s for '%s'\n",
2366789Sahrens 			    osname, prop, valname, setpoint);
2367789Sahrens 		}
2368789Sahrens 	}
2369789Sahrens 
2370789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
2371789Sahrens }
2372789Sahrens 
2373789Sahrens /*
2374789Sahrens  * Inject random faults into the on-disk data.
2375789Sahrens  */
2376789Sahrens void
2377789Sahrens ztest_fault_inject(ztest_args_t *za)
2378789Sahrens {
2379789Sahrens 	int fd;
2380789Sahrens 	uint64_t offset;
2381789Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
2382789Sahrens 	uint64_t bad = 0x1990c0ffeedecade;
2383789Sahrens 	uint64_t top, leaf;
2384789Sahrens 	char path0[MAXPATHLEN];
2385789Sahrens 	char path1[MAXPATHLEN];
2386789Sahrens 	char pathrand[MAXPATHLEN];
2387789Sahrens 	size_t fsize;
2388789Sahrens 	spa_t *spa = dmu_objset_spa(za->za_os);
2389789Sahrens 	int bshift = SPA_MAXBLOCKSHIFT + 2;	/* don't scrog all labels */
2390789Sahrens 	int iters = 1000;
2391789Sahrens 	int ftype;
2392789Sahrens 
2393789Sahrens 	/*
2394789Sahrens 	 * Pick a random top-level vdev.
2395789Sahrens 	 */
2396789Sahrens 	spa_config_enter(spa, RW_READER);
2397789Sahrens 	top = ztest_random(spa->spa_root_vdev->vdev_children);
2398789Sahrens 	spa_config_exit(spa);
2399789Sahrens 
2400789Sahrens 	/*
2401789Sahrens 	 * Pick a random leaf.
2402789Sahrens 	 */
2403789Sahrens 	leaf = ztest_random(leaves);
2404789Sahrens 
2405789Sahrens 	/*
2406789Sahrens 	 * Generate paths to the first to leaves in this top-level vdev,
2407789Sahrens 	 * and to the random leaf we selected.  We'll induce transient
2408789Sahrens 	 * faults on leaves 0 and 1, we'll online/offline leaf 1,
2409789Sahrens 	 * and we'll write random garbage to the randomly chosen leaf.
2410789Sahrens 	 */
2411789Sahrens 	(void) snprintf(path0, sizeof (path0),
2412789Sahrens 	    ztest_dev_template, zopt_dir, zopt_pool, top * leaves + 0);
2413789Sahrens 	(void) snprintf(path1, sizeof (path1),
2414789Sahrens 	    ztest_dev_template, zopt_dir, zopt_pool, top * leaves + 1);
2415789Sahrens 	(void) snprintf(pathrand, sizeof (pathrand),
2416789Sahrens 	    ztest_dev_template, zopt_dir, zopt_pool, top * leaves + leaf);
2417789Sahrens 
2418789Sahrens 	if (leaves < 2)			/* there is no second leaf */
2419789Sahrens 		path1[0] = '\0';
2420789Sahrens 
2421789Sahrens 	dprintf("damaging %s, %s, and %s\n", path0, path1, pathrand);
2422789Sahrens 
2423789Sahrens 	/*
2424789Sahrens 	 * If we have exactly one-fault tolerance, just randomly offline
2425789Sahrens 	 * and online one device.
2426789Sahrens 	 */
2427789Sahrens 	if (zopt_maxfaults == 1 && path1[0] != '\0') {
2428789Sahrens 		if (ztest_random(10) < 6)
2429*1485Slling 			(void) vdev_offline(spa, path1, B_TRUE);
2430789Sahrens 		else
2431789Sahrens 			(void) vdev_online(spa, path1);
2432789Sahrens 		return;
2433789Sahrens 	}
2434789Sahrens 
2435789Sahrens 	/*
2436789Sahrens 	 * Always inject a little random device failure, regardless of
2437789Sahrens 	 * the replication level.  The I/Os should be retried successfully.
2438789Sahrens 	 * If we only have single-fault tolerance, don't inject write
2439789Sahrens 	 * faults, because then we'll be doing partial writes and won't
2440789Sahrens 	 * be able to recover when we inject data corruption.
2441789Sahrens 	 */
2442789Sahrens 	if (zopt_maxfaults <= 1)
2443789Sahrens 		ftype = (1U << ZIO_TYPE_READ);
2444789Sahrens 	else
2445789Sahrens 		ftype = (1U << ZIO_TYPE_READ) | (1U << ZIO_TYPE_WRITE);
2446789Sahrens 
2447789Sahrens 	(void) vdev_error_setup(spa, path0, VDEV_FAULT_COUNT, ftype, 10);
2448789Sahrens 
2449789Sahrens 	/*
2450789Sahrens 	 * If we can tolerate three or more faults, make one of the
2451789Sahrens 	 * devices fail quite a lot.
2452789Sahrens 	 */
2453789Sahrens 	if (zopt_maxfaults >= 3 && path1[0] != '\0')
2454789Sahrens 		(void) vdev_error_setup(spa, path1, VDEV_FAULT_COUNT,
2455789Sahrens 		    ftype, 100);
2456789Sahrens 
2457789Sahrens 	/*
2458789Sahrens 	 * If we can tolerate four or more faults, offline one of the devices.
2459789Sahrens 	 */
2460789Sahrens 	if (zopt_maxfaults >= 4 && path1[0] != '\0') {
2461789Sahrens 		if (ztest_random(10) < 6)
2462*1485Slling 			(void) vdev_offline(spa, path1, B_TRUE);
2463789Sahrens 		else
2464789Sahrens 			(void) vdev_online(spa, path1);
2465789Sahrens 	}
2466789Sahrens 
2467789Sahrens 	/*
2468789Sahrens 	 * If we have at least single-fault tolerance, inject data corruption.
2469789Sahrens 	 */
2470789Sahrens 	if (zopt_maxfaults < 1)
2471789Sahrens 		return;
2472789Sahrens 
2473789Sahrens 	fd = open(pathrand, O_RDWR);
2474789Sahrens 
2475789Sahrens 	if (fd == -1)	/* we hit a gap in the device namespace */
2476789Sahrens 		return;
2477789Sahrens 
2478789Sahrens 	fsize = lseek(fd, 0, SEEK_END);
2479789Sahrens 
2480789Sahrens 	while (--iters != 0) {
2481789Sahrens 		offset = ztest_random(fsize / (leaves << bshift)) *
2482789Sahrens 		    (leaves << bshift) + (leaf << bshift) +
2483789Sahrens 		    (ztest_random(1ULL << (bshift - 1)) & -8ULL);
2484789Sahrens 
2485789Sahrens 		if (offset >= fsize)
2486789Sahrens 			continue;
2487789Sahrens 
2488789Sahrens 		if (zopt_verbose >= 6)
2489789Sahrens 			(void) printf("injecting bad word into %s,"
2490789Sahrens 			    " offset 0x%llx\n", pathrand, (u_longlong_t)offset);
2491789Sahrens 
2492789Sahrens 		if (pwrite(fd, &bad, sizeof (bad), offset) != sizeof (bad))
2493789Sahrens 			fatal(1, "can't inject bad word at 0x%llx in %s",
2494789Sahrens 			    offset, pathrand);
2495789Sahrens 	}
2496789Sahrens 
2497789Sahrens 	(void) close(fd);
2498789Sahrens }
2499789Sahrens 
2500789Sahrens static void
2501789Sahrens ztest_error_setup(vdev_t *vd, int mode, int mask, uint64_t arg)
2502789Sahrens {
2503789Sahrens 	int c;
2504789Sahrens 
2505789Sahrens 	for (c = 0; c < vd->vdev_children; c++)
2506789Sahrens 		ztest_error_setup(vd->vdev_child[c], mode, mask, arg);
2507789Sahrens 
2508789Sahrens 	if (vd->vdev_path != NULL)
2509789Sahrens 		(void) vdev_error_setup(vd->vdev_spa, vd->vdev_path,
2510789Sahrens 		    mode, mask, arg);
2511789Sahrens }
2512789Sahrens 
2513789Sahrens /*
2514789Sahrens  * Scrub the pool.
2515789Sahrens  */
2516789Sahrens void
2517789Sahrens ztest_scrub(ztest_args_t *za)
2518789Sahrens {
2519789Sahrens 	spa_t *spa = dmu_objset_spa(za->za_os);
2520789Sahrens 
2521789Sahrens 	(void) spa_scrub(spa, POOL_SCRUB_EVERYTHING, B_FALSE);
2522789Sahrens 	(void) poll(NULL, 0, 1000); /* wait a second, then force a restart */
2523789Sahrens 	(void) spa_scrub(spa, POOL_SCRUB_EVERYTHING, B_FALSE);
2524789Sahrens }
2525789Sahrens 
2526789Sahrens /*
2527789Sahrens  * Rename the pool to a different name and then rename it back.
2528789Sahrens  */
2529789Sahrens void
2530789Sahrens ztest_spa_rename(ztest_args_t *za)
2531789Sahrens {
2532789Sahrens 	char *oldname, *newname;
2533789Sahrens 	int error;
2534789Sahrens 	spa_t *spa;
2535789Sahrens 
2536789Sahrens 	(void) rw_wrlock(&ztest_shared->zs_name_lock);
2537789Sahrens 
2538789Sahrens 	oldname = za->za_pool;
2539789Sahrens 	newname = umem_alloc(strlen(oldname) + 5, UMEM_NOFAIL);
2540789Sahrens 	(void) strcpy(newname, oldname);
2541789Sahrens 	(void) strcat(newname, "_tmp");
2542789Sahrens 
2543789Sahrens 	/*
2544789Sahrens 	 * Do the rename
2545789Sahrens 	 */
2546789Sahrens 	error = spa_rename(oldname, newname);
2547789Sahrens 	if (error)
2548789Sahrens 		fatal(0, "spa_rename('%s', '%s') = %d", oldname,
2549789Sahrens 		    newname, error);
2550789Sahrens 
2551789Sahrens 	/*
2552789Sahrens 	 * Try to open it under the old name, which shouldn't exist
2553789Sahrens 	 */
2554789Sahrens 	error = spa_open(oldname, &spa, FTAG);
2555789Sahrens 	if (error != ENOENT)
2556789Sahrens 		fatal(0, "spa_open('%s') = %d", oldname, error);
2557789Sahrens 
2558789Sahrens 	/*
2559789Sahrens 	 * Open it under the new name and make sure it's still the same spa_t.
2560789Sahrens 	 */
2561789Sahrens 	error = spa_open(newname, &spa, FTAG);
2562789Sahrens 	if (error != 0)
2563789Sahrens 		fatal(0, "spa_open('%s') = %d", newname, error);
2564789Sahrens 
2565789Sahrens 	ASSERT(spa == dmu_objset_spa(za->za_os));
2566789Sahrens 	spa_close(spa, FTAG);
2567789Sahrens 
2568789Sahrens 	/*
2569789Sahrens 	 * Rename it back to the original
2570789Sahrens 	 */
2571789Sahrens 	error = spa_rename(newname, oldname);
2572789Sahrens 	if (error)
2573789Sahrens 		fatal(0, "spa_rename('%s', '%s') = %d", newname,
2574789Sahrens 		    oldname, error);
2575789Sahrens 
2576789Sahrens 	/*
2577789Sahrens 	 * Make sure it can still be opened
2578789Sahrens 	 */
2579789Sahrens 	error = spa_open(oldname, &spa, FTAG);
2580789Sahrens 	if (error != 0)
2581789Sahrens 		fatal(0, "spa_open('%s') = %d", oldname, error);
2582789Sahrens 
2583789Sahrens 	ASSERT(spa == dmu_objset_spa(za->za_os));
2584789Sahrens 	spa_close(spa, FTAG);
2585789Sahrens 
2586789Sahrens 	umem_free(newname, strlen(newname) + 1);
2587789Sahrens 
2588789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
2589789Sahrens }
2590789Sahrens 
2591789Sahrens 
2592789Sahrens /*
2593789Sahrens  * Completely obliterate one disk.
2594789Sahrens  */
2595789Sahrens static void
2596789Sahrens ztest_obliterate_one_disk(uint64_t vdev)
2597789Sahrens {
2598789Sahrens 	int fd;
2599789Sahrens 	char dev_name[MAXPATHLEN];
2600789Sahrens 	size_t fsize;
2601789Sahrens 
2602789Sahrens 	if (zopt_maxfaults < 2)
2603789Sahrens 		return;
2604789Sahrens 
2605789Sahrens 	(void) sprintf(dev_name, ztest_dev_template, zopt_dir, zopt_pool, vdev);
2606789Sahrens 
2607789Sahrens 	fd = open(dev_name, O_RDWR);
2608789Sahrens 
2609789Sahrens 	if (fd == -1)
2610789Sahrens 		fatal(1, "can't open %s", dev_name);
2611789Sahrens 
2612789Sahrens 	/*
2613789Sahrens 	 * Determine the size.
2614789Sahrens 	 */
2615789Sahrens 	fsize = lseek(fd, 0, SEEK_END);
2616789Sahrens 	(void) close(fd);
2617789Sahrens 
2618789Sahrens 	/*
2619789Sahrens 	 * Remove it.
2620789Sahrens 	 */
2621789Sahrens 	VERIFY(remove(dev_name) == 0);
2622789Sahrens 
2623789Sahrens 	/*
2624789Sahrens 	 * Create a new one.
2625789Sahrens 	 */
2626789Sahrens 	VERIFY((fd = open(dev_name, O_RDWR | O_CREAT | O_TRUNC, 0666)) >= 0);
2627789Sahrens 	VERIFY(ftruncate(fd, fsize) == 0);
2628789Sahrens 	(void) close(fd);
2629789Sahrens }
2630789Sahrens 
2631789Sahrens static void
2632789Sahrens ztest_replace_one_disk(spa_t *spa, uint64_t vdev)
2633789Sahrens {
2634789Sahrens 	char dev_name[MAXPATHLEN];
2635789Sahrens 	nvlist_t *file, *root;
2636789Sahrens 	int error;
2637789Sahrens 
2638789Sahrens 	(void) sprintf(dev_name, ztest_dev_template, zopt_dir, zopt_pool, vdev);
2639789Sahrens 
2640789Sahrens 	/*
2641789Sahrens 	 * Build the nvlist describing dev_name.
2642789Sahrens 	 */
2643789Sahrens 	VERIFY(nvlist_alloc(&file, NV_UNIQUE_NAME, 0) == 0);
2644789Sahrens 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_TYPE, VDEV_TYPE_FILE) == 0);
2645789Sahrens 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_PATH, dev_name) == 0);
2646789Sahrens 
2647789Sahrens 	VERIFY(nvlist_alloc(&root, NV_UNIQUE_NAME, 0) == 0);
2648789Sahrens 	VERIFY(nvlist_add_string(root, ZPOOL_CONFIG_TYPE, VDEV_TYPE_ROOT) == 0);
2649789Sahrens 	VERIFY(nvlist_add_nvlist_array(root, ZPOOL_CONFIG_CHILDREN,
2650789Sahrens 	    &file, 1) == 0);
2651789Sahrens 
2652789Sahrens 	error = spa_vdev_attach(spa, dev_name, root, B_TRUE);
2653789Sahrens 	if (error != 0 && error != EBUSY && error != ENOTSUP && error != ENODEV)
2654789Sahrens 		fatal(0, "spa_vdev_attach(in-place) = %d", error);
2655789Sahrens 
2656789Sahrens 	nvlist_free(file);
2657789Sahrens 	nvlist_free(root);
2658789Sahrens }
2659789Sahrens 
2660789Sahrens static void
2661789Sahrens ztest_verify_blocks(char *pool)
2662789Sahrens {
2663789Sahrens 	int status;
2664789Sahrens 	char zdb[MAXPATHLEN + MAXNAMELEN + 20];
2665789Sahrens 	char zbuf[1024];
2666789Sahrens 	char *bin;
2667789Sahrens 	FILE *fp;
2668789Sahrens 
2669789Sahrens 	(void) realpath(getexecname(), zdb);
2670789Sahrens 
2671789Sahrens 	/* zdb lives in /usr/sbin, while ztest lives in /usr/bin */
2672789Sahrens 	bin = strstr(zdb, "/usr/bin/");
2673789Sahrens 	/* LINTED */
2674789Sahrens 	(void) sprintf(bin, "/usr/sbin/zdb -bc%s%s -U -O %s %s",
2675789Sahrens 	    zopt_verbose >= 3 ? "s" : "",
2676789Sahrens 	    zopt_verbose >= 4 ? "v" : "",
2677789Sahrens 	    ztest_random(2) == 0 ? "pre" : "post", pool);
2678789Sahrens 
2679789Sahrens 	if (zopt_verbose >= 5)
2680789Sahrens 		(void) printf("Executing %s\n", strstr(zdb, "zdb "));
2681789Sahrens 
2682789Sahrens 	fp = popen(zdb, "r");
2683789Sahrens 
2684789Sahrens 	while (fgets(zbuf, sizeof (zbuf), fp) != NULL)
2685789Sahrens 		if (zopt_verbose >= 3)
2686789Sahrens 			(void) printf("%s", zbuf);
2687789Sahrens 
2688789Sahrens 	status = pclose(fp);
2689789Sahrens 
2690789Sahrens 	if (status == 0)
2691789Sahrens 		return;
2692789Sahrens 
2693789Sahrens 	ztest_dump_core = 0;
2694789Sahrens 	if (WIFEXITED(status))
2695789Sahrens 		fatal(0, "'%s' exit code %d", zdb, WEXITSTATUS(status));
2696789Sahrens 	else
2697789Sahrens 		fatal(0, "'%s' died with signal %d", zdb, WTERMSIG(status));
2698789Sahrens }
2699789Sahrens 
2700789Sahrens static void
2701789Sahrens ztest_walk_pool_directory(char *header)
2702789Sahrens {
2703789Sahrens 	spa_t *spa = NULL;
2704789Sahrens 
2705789Sahrens 	if (zopt_verbose >= 6)
2706789Sahrens 		(void) printf("%s\n", header);
2707789Sahrens 
2708789Sahrens 	mutex_enter(&spa_namespace_lock);
2709789Sahrens 	while ((spa = spa_next(spa)) != NULL)
2710789Sahrens 		if (zopt_verbose >= 6)
2711789Sahrens 			(void) printf("\t%s\n", spa_name(spa));
2712789Sahrens 	mutex_exit(&spa_namespace_lock);
2713789Sahrens }
2714789Sahrens 
2715789Sahrens static void
2716789Sahrens ztest_spa_import_export(char *oldname, char *newname)
2717789Sahrens {
2718789Sahrens 	nvlist_t *config;
2719789Sahrens 	uint64_t pool_guid;
2720789Sahrens 	spa_t *spa;
2721789Sahrens 	int error;
2722789Sahrens 
2723789Sahrens 	if (zopt_verbose >= 4) {
2724789Sahrens 		(void) printf("import/export: old = %s, new = %s\n",
2725789Sahrens 		    oldname, newname);
2726789Sahrens 	}
2727789Sahrens 
2728789Sahrens 	/*
2729789Sahrens 	 * Clean up from previous runs.
2730789Sahrens 	 */
2731789Sahrens 	(void) spa_destroy(newname);
2732789Sahrens 
2733789Sahrens 	/*
2734789Sahrens 	 * Get the pool's configuration and guid.
2735789Sahrens 	 */
2736789Sahrens 	error = spa_open(oldname, &spa, FTAG);
2737789Sahrens 	if (error)
2738789Sahrens 		fatal(0, "spa_open('%s') = %d", oldname, error);
2739789Sahrens 
2740789Sahrens 	ASSERT(spa->spa_config != NULL);
2741789Sahrens 
2742789Sahrens 	VERIFY(nvlist_dup(spa->spa_config, &config, 0) == 0);
2743789Sahrens 	pool_guid = spa_guid(spa);
2744789Sahrens 	spa_close(spa, FTAG);
2745789Sahrens 
2746789Sahrens 	ztest_walk_pool_directory("pools before export");
2747789Sahrens 
2748789Sahrens 	/*
2749789Sahrens 	 * Export it.
2750789Sahrens 	 */
2751789Sahrens 	error = spa_export(oldname);
2752789Sahrens 	if (error)
2753789Sahrens 		fatal(0, "spa_export('%s') = %d", oldname, error);
2754789Sahrens 
2755789Sahrens 	ztest_walk_pool_directory("pools after export");
2756789Sahrens 
2757789Sahrens 	/*
2758789Sahrens 	 * Import it under the new name.
2759789Sahrens 	 */
2760789Sahrens 	error = spa_import(newname, config, NULL);
2761789Sahrens 	if (error)
2762789Sahrens 		fatal(0, "spa_import('%s') = %d", newname, error);
2763789Sahrens 
2764789Sahrens 	ztest_walk_pool_directory("pools after import");
2765789Sahrens 
2766789Sahrens 	/*
2767789Sahrens 	 * Try to import it again -- should fail with EEXIST.
2768789Sahrens 	 */
2769789Sahrens 	error = spa_import(newname, config, NULL);
2770789Sahrens 	if (error != EEXIST)
2771789Sahrens 		fatal(0, "spa_import('%s') twice", newname);
2772789Sahrens 
2773789Sahrens 	/*
2774789Sahrens 	 * Try to import it under a different name -- should fail with EEXIST.
2775789Sahrens 	 */
2776789Sahrens 	error = spa_import(oldname, config, NULL);
2777789Sahrens 	if (error != EEXIST)
2778789Sahrens 		fatal(0, "spa_import('%s') under multiple names", newname);
2779789Sahrens 
2780789Sahrens 	/*
2781789Sahrens 	 * Verify that the pool is no longer visible under the old name.
2782789Sahrens 	 */
2783789Sahrens 	error = spa_open(oldname, &spa, FTAG);
2784789Sahrens 	if (error != ENOENT)
2785789Sahrens 		fatal(0, "spa_open('%s') = %d", newname, error);
2786789Sahrens 
2787789Sahrens 	/*
2788789Sahrens 	 * Verify that we can open and close the pool using the new name.
2789789Sahrens 	 */
2790789Sahrens 	error = spa_open(newname, &spa, FTAG);
2791789Sahrens 	if (error)
2792789Sahrens 		fatal(0, "spa_open('%s') = %d", newname, error);
2793789Sahrens 	ASSERT(pool_guid == spa_guid(spa));
2794789Sahrens 	spa_close(spa, FTAG);
2795789Sahrens 
2796789Sahrens 	nvlist_free(config);
2797789Sahrens }
2798789Sahrens 
2799789Sahrens static void *
2800789Sahrens ztest_thread(void *arg)
2801789Sahrens {
2802789Sahrens 	ztest_args_t *za = arg;
2803789Sahrens 	ztest_shared_t *zs = ztest_shared;
2804789Sahrens 	hrtime_t now, functime;
2805789Sahrens 	ztest_info_t *zi;
2806789Sahrens 	int f;
2807789Sahrens 
2808789Sahrens 	while ((now = gethrtime()) < za->za_stop) {
2809789Sahrens 		/*
2810789Sahrens 		 * See if it's time to force a crash.
2811789Sahrens 		 */
2812789Sahrens 		if (now > za->za_kill) {
2813789Sahrens 			zs->zs_alloc = spa_get_alloc(dmu_objset_spa(za->za_os));
2814789Sahrens 			zs->zs_space = spa_get_space(dmu_objset_spa(za->za_os));
2815789Sahrens 			(void) kill(getpid(), SIGKILL);
2816789Sahrens 		}
2817789Sahrens 
2818789Sahrens 		/*
2819789Sahrens 		 * Pick a random function.
2820789Sahrens 		 */
2821789Sahrens 		f = ztest_random(ZTEST_FUNCS);
2822789Sahrens 		zi = &zs->zs_info[f];
2823789Sahrens 
2824789Sahrens 		/*
2825789Sahrens 		 * Decide whether to call it, based on the requested frequency.
2826789Sahrens 		 */
2827789Sahrens 		if (zi->zi_call_target == 0 ||
2828789Sahrens 		    (double)zi->zi_call_total / zi->zi_call_target >
2829789Sahrens 		    (double)(now - zs->zs_start_time) / (zopt_time * NANOSEC))
2830789Sahrens 			continue;
2831789Sahrens 
2832789Sahrens 		atomic_add_64(&zi->zi_calls, 1);
2833789Sahrens 		atomic_add_64(&zi->zi_call_total, 1);
2834789Sahrens 
2835789Sahrens 		za->za_diroff = (za->za_instance * ZTEST_FUNCS + f) *
2836789Sahrens 		    ZTEST_DIRSIZE;
2837789Sahrens 		za->za_diroff_shared = (1ULL << 63);
2838789Sahrens 
2839789Sahrens 		ztest_dmu_write_parallel(za);
2840789Sahrens 
2841789Sahrens 		zi->zi_func(za);
2842789Sahrens 
2843789Sahrens 		functime = gethrtime() - now;
2844789Sahrens 
2845789Sahrens 		atomic_add_64(&zi->zi_call_time, functime);
2846789Sahrens 
2847789Sahrens 		if (zopt_verbose >= 4) {
2848789Sahrens 			Dl_info dli;
2849789Sahrens 			(void) dladdr((void *)zi->zi_func, &dli);
2850789Sahrens 			(void) printf("%6.2f sec in %s\n",
2851789Sahrens 			    (double)functime / NANOSEC, dli.dli_sname);
2852789Sahrens 		}
2853789Sahrens 
2854789Sahrens 		/*
2855789Sahrens 		 * If we're getting ENOSPC with some regularity, stop.
2856789Sahrens 		 */
2857789Sahrens 		if (zs->zs_enospc_count > 10)
2858789Sahrens 			break;
2859789Sahrens 	}
2860789Sahrens 
2861789Sahrens 	return (NULL);
2862789Sahrens }
2863789Sahrens 
2864789Sahrens /*
2865789Sahrens  * Kick off threads to run tests on all datasets in parallel.
2866789Sahrens  */
2867789Sahrens static void
2868789Sahrens ztest_run(char *pool)
2869789Sahrens {
2870789Sahrens 	int t, d, error;
2871789Sahrens 	ztest_shared_t *zs = ztest_shared;
2872789Sahrens 	ztest_args_t *za;
2873789Sahrens 	spa_t *spa;
2874789Sahrens 	char name[100];
2875789Sahrens 
2876789Sahrens 	(void) _mutex_init(&zs->zs_vdev_lock, USYNC_THREAD, NULL);
2877789Sahrens 	(void) rwlock_init(&zs->zs_name_lock, USYNC_THREAD, NULL);
2878789Sahrens 
2879789Sahrens 	for (t = 0; t < ZTEST_SYNC_LOCKS; t++)
2880789Sahrens 		(void) _mutex_init(&zs->zs_sync_lock[t], USYNC_THREAD, NULL);
2881789Sahrens 
2882789Sahrens 	/*
2883789Sahrens 	 * Destroy one disk before we even start.
2884789Sahrens 	 * It's mirrored, so everything should work just fine.
2885789Sahrens 	 * This makes us exercise fault handling very early in spa_load().
2886789Sahrens 	 */
2887789Sahrens 	ztest_obliterate_one_disk(0);
2888789Sahrens 
2889789Sahrens 	/*
2890789Sahrens 	 * Verify that the sum of the sizes of all blocks in the pool
2891789Sahrens 	 * equals the SPA's allocated space total.
2892789Sahrens 	 */
2893789Sahrens 	ztest_verify_blocks(pool);
2894789Sahrens 
2895789Sahrens 	/*
2896789Sahrens 	 * Kick off a replacement of the disk we just obliterated.
2897789Sahrens 	 */
2898789Sahrens 	kernel_init(FREAD | FWRITE);
2899789Sahrens 	error = spa_open(pool, &spa, FTAG);
2900789Sahrens 	if (error)
2901789Sahrens 		fatal(0, "spa_open(%s) = %d", pool, error);
2902789Sahrens 	ztest_replace_one_disk(spa, 0);
2903789Sahrens 	if (zopt_verbose >= 5)
2904789Sahrens 		show_pool_stats(spa);
2905789Sahrens 	spa_close(spa, FTAG);
2906789Sahrens 	kernel_fini();
2907789Sahrens 
2908789Sahrens 	kernel_init(FREAD | FWRITE);
2909789Sahrens 
2910789Sahrens 	/*
2911789Sahrens 	 * Verify that we can export the pool and reimport it under a
2912789Sahrens 	 * different name.
2913789Sahrens 	 */
2914789Sahrens 	(void) snprintf(name, 100, "%s_import", pool);
2915789Sahrens 	ztest_spa_import_export(pool, name);
2916789Sahrens 	ztest_spa_import_export(name, pool);
2917789Sahrens 
2918789Sahrens 	/*
2919789Sahrens 	 * Verify that we can loop over all pools.
2920789Sahrens 	 */
2921789Sahrens 	mutex_enter(&spa_namespace_lock);
2922789Sahrens 	for (spa = spa_next(NULL); spa != NULL; spa = spa_next(spa)) {
2923789Sahrens 		if (zopt_verbose > 3) {
2924789Sahrens 			(void) printf("spa_next: found %s\n", spa_name(spa));
2925789Sahrens 		}
2926789Sahrens 	}
2927789Sahrens 	mutex_exit(&spa_namespace_lock);
2928789Sahrens 
2929789Sahrens 	/*
2930789Sahrens 	 * Open our pool.
2931789Sahrens 	 */
2932789Sahrens 	error = spa_open(pool, &spa, FTAG);
2933789Sahrens 	if (error)
2934789Sahrens 		fatal(0, "spa_open() = %d", error);
2935789Sahrens 
2936789Sahrens 	/*
2937789Sahrens 	 * Verify that we can safely inquire about about any object,
2938789Sahrens 	 * whether it's allocated or not.  To make it interesting,
2939789Sahrens 	 * we probe a 5-wide window around each power of two.
2940789Sahrens 	 * This hits all edge cases, including zero and the max.
2941789Sahrens 	 */
2942789Sahrens 	for (t = 0; t < 64; t++) {
2943789Sahrens 		for (d = -5; d <= 5; d++) {
2944789Sahrens 			error = dmu_object_info(spa->spa_meta_objset,
2945789Sahrens 			    (1ULL << t) + d, NULL);
2946789Sahrens 			ASSERT(error == 0 || error == ENOENT);
2947789Sahrens 		}
2948789Sahrens 	}
2949789Sahrens 
2950789Sahrens 	/*
2951789Sahrens 	 * Now kick off all the tests that run in parallel.
2952789Sahrens 	 */
2953789Sahrens 	zs->zs_enospc_count = 0;
2954789Sahrens 
2955789Sahrens 	za = umem_zalloc(zopt_threads * sizeof (ztest_args_t), UMEM_NOFAIL);
2956789Sahrens 
2957789Sahrens 	if (zopt_verbose >= 4)
2958789Sahrens 		(void) printf("starting main threads...\n");
2959789Sahrens 
2960789Sahrens 	za[0].za_start = gethrtime();
2961789Sahrens 	za[0].za_stop = za[0].za_start + zopt_passtime * NANOSEC;
2962789Sahrens 	za[0].za_stop = MIN(za[0].za_stop, zs->zs_stop_time);
2963789Sahrens 	za[0].za_kill = za[0].za_stop;
2964789Sahrens 	if (ztest_random(100) < zopt_killrate)
2965789Sahrens 		za[0].za_kill -= ztest_random(zopt_passtime * NANOSEC);
2966789Sahrens 
2967789Sahrens 	for (t = 0; t < zopt_threads; t++) {
2968789Sahrens 		d = t % zopt_dirs;
2969789Sahrens 		if (t < zopt_dirs) {
2970789Sahrens 			ztest_replay_t zr;
2971789Sahrens 			(void) rw_rdlock(&ztest_shared->zs_name_lock);
2972789Sahrens 			(void) snprintf(name, 100, "%s/%s_%d", pool, pool, d);
2973789Sahrens 			error = dmu_objset_create(name, DMU_OST_OTHER, NULL,
2974789Sahrens 			    ztest_create_cb, NULL);
2975789Sahrens 			if (error != 0 && error != EEXIST) {
2976789Sahrens 				if (error == ENOSPC) {
2977789Sahrens 					zs->zs_enospc_count++;
2978789Sahrens 					(void) rw_unlock(
2979789Sahrens 					    &ztest_shared->zs_name_lock);
2980789Sahrens 					break;
2981789Sahrens 				}
2982789Sahrens 				fatal(0, "dmu_objset_create(%s) = %d",
2983789Sahrens 				    name, error);
2984789Sahrens 			}
2985789Sahrens 			error = dmu_objset_open(name, DMU_OST_OTHER,
2986789Sahrens 			    DS_MODE_STANDARD, &za[d].za_os);
2987789Sahrens 			if (error)
2988789Sahrens 				fatal(0, "dmu_objset_open('%s') = %d",
2989789Sahrens 				    name, error);
2990789Sahrens 			(void) rw_unlock(&ztest_shared->zs_name_lock);
2991789Sahrens 			zr.zr_os = za[d].za_os;
2992789Sahrens 			zil_replay(zr.zr_os, &zr, &zr.zr_assign,
2993789Sahrens 			    ztest_replay_vector, NULL);
2994789Sahrens 			za[d].za_zilog = zil_open(za[d].za_os, NULL);
2995789Sahrens 		}
2996789Sahrens 		za[t].za_pool = spa_strdup(pool);
2997789Sahrens 		za[t].za_os = za[d].za_os;
2998789Sahrens 		za[t].za_zilog = za[d].za_zilog;
2999789Sahrens 		za[t].za_instance = t;
3000789Sahrens 		za[t].za_random = ztest_random(-1ULL);
3001789Sahrens 		za[t].za_start = za[0].za_start;
3002789Sahrens 		za[t].za_stop = za[0].za_stop;
3003789Sahrens 		za[t].za_kill = za[0].za_kill;
3004789Sahrens 
3005789Sahrens 		error = thr_create(0, 0, ztest_thread, &za[t], THR_BOUND,
3006789Sahrens 		    &za[t].za_thread);
3007789Sahrens 		if (error)
3008789Sahrens 			fatal(0, "can't create thread %d: error %d",
3009789Sahrens 			    t, error);
3010789Sahrens 	}
3011789Sahrens 
3012789Sahrens 	while (--t >= 0) {
3013789Sahrens 		error = thr_join(za[t].za_thread, NULL, NULL);
3014789Sahrens 		if (error)
3015789Sahrens 			fatal(0, "thr_join(%d) = %d", t, error);
3016789Sahrens 		if (za[t].za_th)
3017789Sahrens 			traverse_fini(za[t].za_th);
3018789Sahrens 		if (t < zopt_dirs) {
3019789Sahrens 			zil_close(za[t].za_zilog);
3020789Sahrens 			dmu_objset_close(za[t].za_os);
3021789Sahrens 		}
3022789Sahrens 		spa_strfree(za[t].za_pool);
3023789Sahrens 	}
3024789Sahrens 
3025789Sahrens 	umem_free(za, zopt_threads * sizeof (ztest_args_t));
3026789Sahrens 
3027789Sahrens 	if (zopt_verbose >= 3)
3028789Sahrens 		show_pool_stats(spa);
3029789Sahrens 
3030789Sahrens 	txg_wait_synced(spa_get_dsl(spa), 0);
3031789Sahrens 
3032789Sahrens 	zs->zs_alloc = spa_get_alloc(spa);
3033789Sahrens 	zs->zs_space = spa_get_space(spa);
3034789Sahrens 
3035789Sahrens 	/*
3036789Sahrens 	 * Did we have out-of-space errors?  If so, destroy a random objset.
3037789Sahrens 	 */
3038789Sahrens 	if (zs->zs_enospc_count != 0) {
3039789Sahrens 		(void) rw_rdlock(&ztest_shared->zs_name_lock);
3040789Sahrens 		(void) snprintf(name, 100, "%s/%s_%d", pool, pool,
3041789Sahrens 		    (int)ztest_random(zopt_dirs));
3042789Sahrens 		if (zopt_verbose >= 3)
3043789Sahrens 			(void) printf("Destroying %s to free up space\n", name);
3044789Sahrens 		dmu_objset_find(name, ztest_destroy_cb, NULL,
3045789Sahrens 		    DS_FIND_SNAPSHOTS);
3046789Sahrens 		(void) rw_unlock(&ztest_shared->zs_name_lock);
3047789Sahrens 	}
3048789Sahrens 
3049789Sahrens 	/*
3050789Sahrens 	 * Prepare every leaf device to inject a few random read faults.
3051789Sahrens 	 */
3052789Sahrens 	ztest_error_setup(spa->spa_root_vdev, VDEV_FAULT_COUNT,
3053789Sahrens 	    (1U << ZIO_TYPE_READ), 10);
3054789Sahrens 
3055789Sahrens 	/*
3056789Sahrens 	 * Right before closing the pool, kick off a bunch of async I/O;
3057789Sahrens 	 * spa_close() should wait for it to complete.
3058789Sahrens 	 */
3059789Sahrens 	for (t = 1; t < 50; t++)
3060789Sahrens 		dmu_prefetch(spa->spa_meta_objset, t, 0, 1 << 15);
3061789Sahrens 
3062789Sahrens 	spa_close(spa, FTAG);
3063789Sahrens 
3064789Sahrens 	kernel_fini();
3065789Sahrens }
3066789Sahrens 
3067789Sahrens void
3068789Sahrens print_time(hrtime_t t, char *timebuf)
3069789Sahrens {
3070789Sahrens 	hrtime_t s = t / NANOSEC;
3071789Sahrens 	hrtime_t m = s / 60;
3072789Sahrens 	hrtime_t h = m / 60;
3073789Sahrens 	hrtime_t d = h / 24;
3074789Sahrens 
3075789Sahrens 	s -= m * 60;
3076789Sahrens 	m -= h * 60;
3077789Sahrens 	h -= d * 24;
3078789Sahrens 
3079789Sahrens 	timebuf[0] = '\0';
3080789Sahrens 
3081789Sahrens 	if (d)
3082789Sahrens 		(void) sprintf(timebuf,
3083789Sahrens 		    "%llud%02lluh%02llum%02llus", d, h, m, s);
3084789Sahrens 	else if (h)
3085789Sahrens 		(void) sprintf(timebuf, "%lluh%02llum%02llus", h, m, s);
3086789Sahrens 	else if (m)
3087789Sahrens 		(void) sprintf(timebuf, "%llum%02llus", m, s);
3088789Sahrens 	else
3089789Sahrens 		(void) sprintf(timebuf, "%llus", s);
3090789Sahrens }
3091789Sahrens 
3092789Sahrens /*
3093789Sahrens  * Create a storage pool with the given name and initial vdev size.
3094789Sahrens  * Then create the specified number of datasets in the pool.
3095789Sahrens  */
3096789Sahrens static void
3097789Sahrens ztest_init(char *pool)
3098789Sahrens {
3099789Sahrens 	spa_t *spa;
3100789Sahrens 	int error;
3101789Sahrens 	nvlist_t *nvroot;
3102789Sahrens 
3103789Sahrens 	kernel_init(FREAD | FWRITE);
3104789Sahrens 
3105789Sahrens 	/*
3106789Sahrens 	 * Create the storage pool.
3107789Sahrens 	 */
3108789Sahrens 	(void) spa_destroy(pool);
3109789Sahrens 	ztest_shared->zs_vdev_primaries = 0;
3110789Sahrens 	nvroot = make_vdev_root(zopt_vdev_size, zopt_raidz, zopt_mirrors, 1);
3111789Sahrens 	error = spa_create(pool, nvroot, NULL);
3112789Sahrens 	nvlist_free(nvroot);
3113789Sahrens 
3114789Sahrens 	if (error)
3115789Sahrens 		fatal(0, "spa_create() = %d", error);
3116789Sahrens 	error = spa_open(pool, &spa, FTAG);
3117789Sahrens 	if (error)
3118789Sahrens 		fatal(0, "spa_open() = %d", error);
3119789Sahrens 
3120789Sahrens 	if (zopt_verbose >= 3)
3121789Sahrens 		show_pool_stats(spa);
3122789Sahrens 
3123789Sahrens 	spa_close(spa, FTAG);
3124789Sahrens 
3125789Sahrens 	kernel_fini();
3126789Sahrens }
3127789Sahrens 
3128789Sahrens int
3129789Sahrens main(int argc, char **argv)
3130789Sahrens {
3131789Sahrens 	int kills = 0;
3132789Sahrens 	int iters = 0;
3133789Sahrens 	int i, f;
3134789Sahrens 	ztest_shared_t *zs;
3135789Sahrens 	ztest_info_t *zi;
3136789Sahrens 	char timebuf[100];
3137789Sahrens 	char numbuf[6];
3138789Sahrens 
3139789Sahrens 	(void) setvbuf(stdout, NULL, _IOLBF, 0);
3140789Sahrens 
3141789Sahrens 	/* Override location of zpool.cache */
3142789Sahrens 	spa_config_dir = "/tmp";
3143789Sahrens 
3144789Sahrens 	/*
3145789Sahrens 	 * Blow away any existing copy of zpool.cache
3146789Sahrens 	 */
3147789Sahrens 	(void) remove("/tmp/zpool.cache");
3148789Sahrens 
3149789Sahrens 	ztest_random_fd = open("/dev/urandom", O_RDONLY);
3150789Sahrens 
3151789Sahrens 	process_options(argc, argv);
3152789Sahrens 
3153789Sahrens 	argc -= optind;
3154789Sahrens 	argv += optind;
3155789Sahrens 
3156789Sahrens 	dprintf_setup(&argc, argv);
3157789Sahrens 
3158789Sahrens 	zs = ztest_shared = (void *)mmap(0,
3159789Sahrens 	    P2ROUNDUP(sizeof (ztest_shared_t), getpagesize()),
3160789Sahrens 	    PROT_READ | PROT_WRITE, MAP_SHARED | MAP_ANON, -1, 0);
3161789Sahrens 
3162789Sahrens 	if (zopt_verbose >= 1) {
3163789Sahrens 		(void) printf("%llu vdevs, %d datasets, %d threads,"
3164789Sahrens 		    " %llu seconds...\n",
3165789Sahrens 		    (u_longlong_t)zopt_vdevs, zopt_dirs, zopt_threads,
3166789Sahrens 		    (u_longlong_t)zopt_time);
3167789Sahrens 	}
3168789Sahrens 
3169789Sahrens 	/*
3170789Sahrens 	 * Create and initialize our storage pool.
3171789Sahrens 	 */
3172789Sahrens 	for (i = 1; i <= zopt_init; i++) {
3173789Sahrens 		bzero(zs, sizeof (ztest_shared_t));
3174789Sahrens 		if (zopt_verbose >= 3 && zopt_init != 1)
3175789Sahrens 			(void) printf("ztest_init(), pass %d\n", i);
3176789Sahrens 		ztest_init(zopt_pool);
3177789Sahrens 	}
3178789Sahrens 
3179789Sahrens 	/*
3180789Sahrens 	 * Initialize the call targets for each function.
3181789Sahrens 	 */
3182789Sahrens 	for (f = 0; f < ZTEST_FUNCS; f++) {
3183789Sahrens 		zi = &zs->zs_info[f];
3184789Sahrens 
3185789Sahrens 		*zi = ztest_info[f];
3186789Sahrens 
3187789Sahrens 		if (*zi->zi_interval == 0)
3188789Sahrens 			zi->zi_call_target = UINT64_MAX;
3189789Sahrens 		else
3190789Sahrens 			zi->zi_call_target = zopt_time / *zi->zi_interval;
3191789Sahrens 	}
3192789Sahrens 
3193789Sahrens 	zs->zs_start_time = gethrtime();
3194789Sahrens 	zs->zs_stop_time = zs->zs_start_time + zopt_time * NANOSEC;
3195789Sahrens 
3196789Sahrens 	/*
3197789Sahrens 	 * Run the tests in a loop.  These tests include fault injection
3198789Sahrens 	 * to verify that self-healing data works, and forced crashes
3199789Sahrens 	 * to verify that we never lose on-disk consistency.
3200789Sahrens 	 */
3201789Sahrens 	while (gethrtime() < zs->zs_stop_time) {
3202789Sahrens 		int status;
3203789Sahrens 		pid_t pid;
3204789Sahrens 		char *tmp;
3205789Sahrens 
3206789Sahrens 		/*
3207789Sahrens 		 * Initialize the workload counters for each function.
3208789Sahrens 		 */
3209789Sahrens 		for (f = 0; f < ZTEST_FUNCS; f++) {
3210789Sahrens 			zi = &zs->zs_info[f];
3211789Sahrens 			zi->zi_calls = 0;
3212789Sahrens 			zi->zi_call_time = 0;
3213789Sahrens 		}
3214789Sahrens 
3215789Sahrens 		pid = fork();
3216789Sahrens 
3217789Sahrens 		if (pid == -1)
3218789Sahrens 			fatal(1, "fork failed");
3219789Sahrens 
3220789Sahrens 		if (pid == 0) {	/* child */
3221789Sahrens 			struct rlimit rl = { 1024, 1024 };
3222789Sahrens 			(void) setrlimit(RLIMIT_NOFILE, &rl);
3223789Sahrens 			ztest_run(zopt_pool);
3224789Sahrens 			exit(0);
3225789Sahrens 		}
3226789Sahrens 
3227789Sahrens 		while (waitpid(pid, &status, WEXITED) != pid)
3228789Sahrens 			continue;
3229789Sahrens 
3230789Sahrens 		if (WIFEXITED(status)) {
3231789Sahrens 			if (WEXITSTATUS(status) != 0) {
3232789Sahrens 				(void) fprintf(stderr,
3233789Sahrens 				    "child exited with code %d\n",
3234789Sahrens 				    WEXITSTATUS(status));
3235789Sahrens 				exit(2);
3236789Sahrens 			}
3237789Sahrens 		} else {
3238789Sahrens 			if (WTERMSIG(status) != SIGKILL) {
3239789Sahrens 				(void) fprintf(stderr,
3240789Sahrens 				    "child died with signal %d\n",
3241789Sahrens 				    WTERMSIG(status));
3242789Sahrens 				exit(3);
3243789Sahrens 			}
3244789Sahrens 			kills++;
3245789Sahrens 		}
3246789Sahrens 
3247789Sahrens 		iters++;
3248789Sahrens 
3249789Sahrens 		if (zopt_verbose >= 1) {
3250789Sahrens 			hrtime_t now = gethrtime();
3251789Sahrens 
3252789Sahrens 			now = MIN(now, zs->zs_stop_time);
3253789Sahrens 			print_time(zs->zs_stop_time - now, timebuf);
3254789Sahrens 			nicenum(zs->zs_space, numbuf);
3255789Sahrens 
3256789Sahrens 			(void) printf("Pass %3d, %8s, %3llu ENOSPC, "
3257789Sahrens 			    "%4.1f%% of %5s used, %3.0f%% done, %8s to go\n",
3258789Sahrens 			    iters,
3259789Sahrens 			    WIFEXITED(status) ? "Complete" : "SIGKILL",
3260789Sahrens 			    (u_longlong_t)zs->zs_enospc_count,
3261789Sahrens 			    100.0 * zs->zs_alloc / zs->zs_space,
3262789Sahrens 			    numbuf,
3263789Sahrens 			    100.0 * (now - zs->zs_start_time) /
3264789Sahrens 			    (zopt_time * NANOSEC), timebuf);
3265789Sahrens 		}
3266789Sahrens 
3267789Sahrens 		if (zopt_verbose >= 2) {
3268789Sahrens 			(void) printf("\nWorkload summary:\n\n");
3269789Sahrens 			(void) printf("%7s %9s   %s\n",
3270789Sahrens 			    "Calls", "Time", "Function");
3271789Sahrens 			(void) printf("%7s %9s   %s\n",
3272789Sahrens 			    "-----", "----", "--------");
3273789Sahrens 			for (f = 0; f < ZTEST_FUNCS; f++) {
3274789Sahrens 				Dl_info dli;
3275789Sahrens 
3276789Sahrens 				zi = &zs->zs_info[f];
3277789Sahrens 				print_time(zi->zi_call_time, timebuf);
3278789Sahrens 				(void) dladdr((void *)zi->zi_func, &dli);
3279789Sahrens 				(void) printf("%7llu %9s   %s\n",
3280789Sahrens 				    (u_longlong_t)zi->zi_calls, timebuf,
3281789Sahrens 				    dli.dli_sname);
3282789Sahrens 			}
3283789Sahrens 			(void) printf("\n");
3284789Sahrens 		}
3285789Sahrens 
3286789Sahrens 		/*
3287789Sahrens 		 * It's possible that we killed a child during a rename test, in
3288789Sahrens 		 * which case we'll have a 'ztest_tmp' pool lying around instead
3289789Sahrens 		 * of 'ztest'.  Do a blind rename in case this happened.
3290789Sahrens 		 */
3291789Sahrens 		tmp = umem_alloc(strlen(zopt_pool) + 5, UMEM_NOFAIL);
3292789Sahrens 		(void) strcpy(tmp, zopt_pool);
3293789Sahrens 		(void) strcat(tmp, "_tmp");
3294789Sahrens 		kernel_init(FREAD | FWRITE);
3295789Sahrens 		(void) spa_rename(tmp, zopt_pool);
3296789Sahrens 		kernel_fini();
3297789Sahrens 		umem_free(tmp, strlen(tmp) + 1);
3298789Sahrens 	}
3299789Sahrens 
3300789Sahrens 	ztest_verify_blocks(zopt_pool);
3301789Sahrens 
3302789Sahrens 	if (zopt_verbose >= 1) {
3303789Sahrens 		(void) printf("%d killed, %d completed, %.0f%% kill rate\n",
3304789Sahrens 		    kills, iters - kills, (100.0 * kills) / MAX(1, iters));
3305789Sahrens 	}
3306789Sahrens 
3307789Sahrens 	return (0);
3308789Sahrens }
3309