xref: /onnv-gate/usr/src/cmd/ztest/ztest.c (revision 9229)
1789Sahrens /*
2789Sahrens  * CDDL HEADER START
3789Sahrens  *
4789Sahrens  * The contents of this file are subject to the terms of the
51485Slling  * Common Development and Distribution License (the "License").
61485Slling  * 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 /*
228779SMark.Musante@Sun.COM  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
23789Sahrens  * Use is subject to license terms.
24789Sahrens  */
25789Sahrens 
26789Sahrens /*
27789Sahrens  * The objective of this program is to provide a DMU/ZAP/SPA stress test
28789Sahrens  * that runs entirely in userland, is easy to use, and easy to extend.
29789Sahrens  *
30789Sahrens  * The overall design of the ztest program is as follows:
31789Sahrens  *
32789Sahrens  * (1) For each major functional area (e.g. adding vdevs to a pool,
33789Sahrens  *     creating and destroying datasets, reading and writing objects, etc)
34789Sahrens  *     we have a simple routine to test that functionality.  These
35789Sahrens  *     individual routines do not have to do anything "stressful".
36789Sahrens  *
37789Sahrens  * (2) We turn these simple functionality tests into a stress test by
38789Sahrens  *     running them all in parallel, with as many threads as desired,
39789Sahrens  *     and spread across as many datasets, objects, and vdevs as desired.
40789Sahrens  *
41789Sahrens  * (3) While all this is happening, we inject faults into the pool to
42789Sahrens  *     verify that self-healing data really works.
43789Sahrens  *
44789Sahrens  * (4) Every time we open a dataset, we change its checksum and compression
45789Sahrens  *     functions.  Thus even individual objects vary from block to block
46789Sahrens  *     in which checksum they use and whether they're compressed.
47789Sahrens  *
48789Sahrens  * (5) To verify that we never lose on-disk consistency after a crash,
49789Sahrens  *     we run the entire test in a child of the main process.
50789Sahrens  *     At random times, the child self-immolates with a SIGKILL.
51789Sahrens  *     This is the software equivalent of pulling the power cord.
52789Sahrens  *     The parent then runs the test again, using the existing
53789Sahrens  *     storage pool, as many times as desired.
54789Sahrens  *
55789Sahrens  * (6) To verify that we don't have future leaks or temporal incursions,
56789Sahrens  *     many of the functional tests record the transaction group number
57789Sahrens  *     as part of their data.  When reading old data, they verify that
58789Sahrens  *     the transaction group number is less than the current, open txg.
59789Sahrens  *     If you add a new test, please do this if applicable.
60789Sahrens  *
61789Sahrens  * When run with no arguments, ztest runs for about five minutes and
62789Sahrens  * produces no output if successful.  To get a little bit of information,
63789Sahrens  * specify -V.  To get more information, specify -VV, and so on.
64789Sahrens  *
65789Sahrens  * To turn this into an overnight stress test, use -T to specify run time.
66789Sahrens  *
67789Sahrens  * You can ask more more vdevs [-v], datasets [-d], or threads [-t]
68789Sahrens  * to increase the pool capacity, fanout, and overall stress level.
69789Sahrens  *
70789Sahrens  * The -N(okill) option will suppress kills, so each child runs to completion.
71789Sahrens  * This can be useful when you're trying to distinguish temporal incursions
72789Sahrens  * from plain old race conditions.
73789Sahrens  */
74789Sahrens 
75789Sahrens #include <sys/zfs_context.h>
76789Sahrens #include <sys/spa.h>
77789Sahrens #include <sys/dmu.h>
78789Sahrens #include <sys/txg.h>
79789Sahrens #include <sys/zap.h>
80789Sahrens #include <sys/dmu_objset.h>
81789Sahrens #include <sys/poll.h>
82789Sahrens #include <sys/stat.h>
83789Sahrens #include <sys/time.h>
84789Sahrens #include <sys/wait.h>
85789Sahrens #include <sys/mman.h>
86789Sahrens #include <sys/resource.h>
87789Sahrens #include <sys/zio.h>
88789Sahrens #include <sys/zio_checksum.h>
89789Sahrens #include <sys/zio_compress.h>
90789Sahrens #include <sys/zil.h>
91789Sahrens #include <sys/vdev_impl.h>
927754SJeff.Bonwick@Sun.COM #include <sys/vdev_file.h>
93789Sahrens #include <sys/spa_impl.h>
94789Sahrens #include <sys/dsl_prop.h>
958779SMark.Musante@Sun.COM #include <sys/dsl_dataset.h>
96789Sahrens #include <sys/refcount.h>
97789Sahrens #include <stdio.h>
981914Scasper #include <stdio_ext.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;
1131732Sbonwick static int zopt_ashift = SPA_MINBLOCKSHIFT;
114789Sahrens static int zopt_mirrors = 2;
115789Sahrens static int zopt_raidz = 4;
1162082Seschrock static int zopt_raidz_parity = 1;
117789Sahrens static size_t zopt_vdev_size = SPA_MINDEVSIZE;
1181732Sbonwick static int zopt_datasets = 7;
119789Sahrens static int zopt_threads = 23;
120789Sahrens static uint64_t zopt_passtime = 60;	/* 60 seconds */
121789Sahrens static uint64_t zopt_killrate = 70;	/* 70% kill rate */
122789Sahrens static int zopt_verbose = 0;
123789Sahrens static int zopt_init = 1;
124789Sahrens static char *zopt_dir = "/tmp";
125789Sahrens static uint64_t zopt_time = 300;	/* 5 minutes */
126789Sahrens static int zopt_maxfaults;
127789Sahrens 
1285530Sbonwick typedef struct ztest_block_tag {
1295530Sbonwick 	uint64_t	bt_objset;
1305530Sbonwick 	uint64_t	bt_object;
1315530Sbonwick 	uint64_t	bt_offset;
1325530Sbonwick 	uint64_t	bt_txg;
1335530Sbonwick 	uint64_t	bt_thread;
1345530Sbonwick 	uint64_t	bt_seq;
1355530Sbonwick } ztest_block_tag_t;
1365530Sbonwick 
137789Sahrens typedef struct ztest_args {
1385530Sbonwick 	char		za_pool[MAXNAMELEN];
1395530Sbonwick 	spa_t		*za_spa;
140789Sahrens 	objset_t	*za_os;
141789Sahrens 	zilog_t		*za_zilog;
142789Sahrens 	thread_t	za_thread;
143789Sahrens 	uint64_t	za_instance;
144789Sahrens 	uint64_t	za_random;
145789Sahrens 	uint64_t	za_diroff;
146789Sahrens 	uint64_t	za_diroff_shared;
1471544Seschrock 	uint64_t	za_zil_seq;
148789Sahrens 	hrtime_t	za_start;
149789Sahrens 	hrtime_t	za_stop;
150789Sahrens 	hrtime_t	za_kill;
1515530Sbonwick 	/*
1525530Sbonwick 	 * Thread-local variables can go here to aid debugging.
1535530Sbonwick 	 */
1545530Sbonwick 	ztest_block_tag_t za_rbt;
1555530Sbonwick 	ztest_block_tag_t za_wbt;
1565530Sbonwick 	dmu_object_info_t za_doi;
1575530Sbonwick 	dmu_buf_t	*za_dbuf;
158789Sahrens } ztest_args_t;
159789Sahrens 
160789Sahrens typedef void ztest_func_t(ztest_args_t *);
161789Sahrens 
162789Sahrens /*
163789Sahrens  * Note: these aren't static because we want dladdr() to work.
164789Sahrens  */
165789Sahrens ztest_func_t ztest_dmu_read_write;
166789Sahrens ztest_func_t ztest_dmu_write_parallel;
167789Sahrens ztest_func_t ztest_dmu_object_alloc_free;
168789Sahrens ztest_func_t ztest_zap;
169789Sahrens ztest_func_t ztest_zap_parallel;
170789Sahrens ztest_func_t ztest_traverse;
171789Sahrens ztest_func_t ztest_dsl_prop_get_set;
172789Sahrens ztest_func_t ztest_dmu_objset_create_destroy;
173789Sahrens ztest_func_t ztest_dmu_snapshot_create_destroy;
1748779SMark.Musante@Sun.COM ztest_func_t ztest_dsl_dataset_promote_busy;
175789Sahrens ztest_func_t ztest_spa_create_destroy;
176789Sahrens ztest_func_t ztest_fault_inject;
1777754SJeff.Bonwick@Sun.COM ztest_func_t ztest_spa_rename;
178789Sahrens ztest_func_t ztest_vdev_attach_detach;
179789Sahrens ztest_func_t ztest_vdev_LUN_growth;
180789Sahrens ztest_func_t ztest_vdev_add_remove;
1817754SJeff.Bonwick@Sun.COM ztest_func_t ztest_vdev_aux_add_remove;
182789Sahrens ztest_func_t ztest_scrub;
183789Sahrens 
184789Sahrens typedef struct ztest_info {
185789Sahrens 	ztest_func_t	*zi_func;	/* test function */
1865530Sbonwick 	uint64_t	zi_iters;	/* iterations per execution */
187789Sahrens 	uint64_t	*zi_interval;	/* execute every <interval> seconds */
188789Sahrens 	uint64_t	zi_calls;	/* per-pass count */
189789Sahrens 	uint64_t	zi_call_time;	/* per-pass time */
190789Sahrens 	uint64_t	zi_call_total;	/* cumulative total */
191789Sahrens 	uint64_t	zi_call_target;	/* target cumulative total */
192789Sahrens } ztest_info_t;
193789Sahrens 
194789Sahrens uint64_t zopt_always = 0;		/* all the time */
195789Sahrens uint64_t zopt_often = 1;		/* every second */
196789Sahrens uint64_t zopt_sometimes = 10;		/* every 10 seconds */
197789Sahrens uint64_t zopt_rarely = 60;		/* every 60 seconds */
198789Sahrens 
199789Sahrens ztest_info_t ztest_info[] = {
2005530Sbonwick 	{ ztest_dmu_read_write,			1,	&zopt_always	},
2015530Sbonwick 	{ ztest_dmu_write_parallel,		30,	&zopt_always	},
2025530Sbonwick 	{ ztest_dmu_object_alloc_free,		1,	&zopt_always	},
2035530Sbonwick 	{ ztest_zap,				30,	&zopt_always	},
2045530Sbonwick 	{ ztest_zap_parallel,			100,	&zopt_always	},
2055530Sbonwick 	{ ztest_dsl_prop_get_set,		1,	&zopt_sometimes	},
2067046Sahrens 	{ ztest_dmu_objset_create_destroy,	1,	&zopt_sometimes },
2077754SJeff.Bonwick@Sun.COM 	{ ztest_dmu_snapshot_create_destroy,	1,	&zopt_sometimes },
2087046Sahrens 	{ ztest_spa_create_destroy,		1,	&zopt_sometimes },
2095530Sbonwick 	{ ztest_fault_inject,			1,	&zopt_sometimes	},
2105530Sbonwick 	{ ztest_spa_rename,			1,	&zopt_rarely	},
2117754SJeff.Bonwick@Sun.COM 	{ ztest_vdev_attach_detach,		1,	&zopt_rarely	},
2127754SJeff.Bonwick@Sun.COM 	{ ztest_vdev_LUN_growth,		1,	&zopt_rarely	},
213*9229SGeorge.Wilson@Sun.COM 	{ ztest_dsl_dataset_promote_busy,	1,	&zopt_rarely	},
2147754SJeff.Bonwick@Sun.COM 	{ ztest_vdev_add_remove,		1,	&zopt_vdevtime	},
2157754SJeff.Bonwick@Sun.COM 	{ ztest_vdev_aux_add_remove,		1,	&zopt_vdevtime	},
2165530Sbonwick 	{ ztest_scrub,				1,	&zopt_vdevtime	},
217789Sahrens };
218789Sahrens 
219789Sahrens #define	ZTEST_FUNCS	(sizeof (ztest_info) / sizeof (ztest_info_t))
220789Sahrens 
221789Sahrens #define	ZTEST_SYNC_LOCKS	16
222789Sahrens 
223789Sahrens /*
224789Sahrens  * Stuff we need to share writably between parent and child.
225789Sahrens  */
226789Sahrens typedef struct ztest_shared {
227789Sahrens 	mutex_t		zs_vdev_lock;
228789Sahrens 	rwlock_t	zs_name_lock;
229789Sahrens 	uint64_t	zs_vdev_primaries;
2307754SJeff.Bonwick@Sun.COM 	uint64_t	zs_vdev_aux;
231789Sahrens 	uint64_t	zs_enospc_count;
232789Sahrens 	hrtime_t	zs_start_time;
233789Sahrens 	hrtime_t	zs_stop_time;
234789Sahrens 	uint64_t	zs_alloc;
235789Sahrens 	uint64_t	zs_space;
236789Sahrens 	ztest_info_t	zs_info[ZTEST_FUNCS];
237789Sahrens 	mutex_t		zs_sync_lock[ZTEST_SYNC_LOCKS];
238789Sahrens 	uint64_t	zs_seq[ZTEST_SYNC_LOCKS];
239789Sahrens } ztest_shared_t;
240789Sahrens 
241789Sahrens static char ztest_dev_template[] = "%s/%s.%llua";
2427754SJeff.Bonwick@Sun.COM static char ztest_aux_template[] = "%s/%s.%s.%llu";
243789Sahrens static ztest_shared_t *ztest_shared;
244789Sahrens 
245789Sahrens static int ztest_random_fd;
246789Sahrens static int ztest_dump_core = 1;
247789Sahrens 
2487754SJeff.Bonwick@Sun.COM static boolean_t ztest_exiting;
2495329Sgw25295 
2505530Sbonwick extern uint64_t metaslab_gang_bang;
251789Sahrens 
252789Sahrens #define	ZTEST_DIROBJ		1
253789Sahrens #define	ZTEST_MICROZAP_OBJ	2
254789Sahrens #define	ZTEST_FATZAP_OBJ	3
255789Sahrens 
256789Sahrens #define	ZTEST_DIROBJ_BLOCKSIZE	(1 << 10)
257789Sahrens #define	ZTEST_DIRSIZE		256
258789Sahrens 
2594008Sraf static void usage(boolean_t) __NORETURN;
2603972Svb160487 
261789Sahrens /*
262789Sahrens  * These libumem hooks provide a reasonable set of defaults for the allocator's
263789Sahrens  * debugging facilities.
264789Sahrens  */
265789Sahrens const char *
266789Sahrens _umem_debug_init()
267789Sahrens {
268789Sahrens 	return ("default,verbose"); /* $UMEM_DEBUG setting */
269789Sahrens }
270789Sahrens 
271789Sahrens const char *
272789Sahrens _umem_logging_init(void)
273789Sahrens {
274789Sahrens 	return ("fail,contents"); /* $UMEM_LOGGING setting */
275789Sahrens }
276789Sahrens 
277789Sahrens #define	FATAL_MSG_SZ	1024
278789Sahrens 
279789Sahrens char *fatal_msg;
280789Sahrens 
281789Sahrens static void
282789Sahrens fatal(int do_perror, char *message, ...)
283789Sahrens {
284789Sahrens 	va_list args;
285789Sahrens 	int save_errno = errno;
286789Sahrens 	char buf[FATAL_MSG_SZ];
287789Sahrens 
288789Sahrens 	(void) fflush(stdout);
289789Sahrens 
290789Sahrens 	va_start(args, message);
291789Sahrens 	(void) sprintf(buf, "ztest: ");
292789Sahrens 	/* LINTED */
293789Sahrens 	(void) vsprintf(buf + strlen(buf), message, args);
294789Sahrens 	va_end(args);
295789Sahrens 	if (do_perror) {
296789Sahrens 		(void) snprintf(buf + strlen(buf), FATAL_MSG_SZ - strlen(buf),
297789Sahrens 		    ": %s", strerror(save_errno));
298789Sahrens 	}
299789Sahrens 	(void) fprintf(stderr, "%s\n", buf);
300789Sahrens 	fatal_msg = buf;			/* to ease debugging */
301789Sahrens 	if (ztest_dump_core)
302789Sahrens 		abort();
303789Sahrens 	exit(3);
304789Sahrens }
305789Sahrens 
306789Sahrens static int
307789Sahrens str2shift(const char *buf)
308789Sahrens {
309789Sahrens 	const char *ends = "BKMGTPEZ";
310789Sahrens 	int i;
311789Sahrens 
312789Sahrens 	if (buf[0] == '\0')
313789Sahrens 		return (0);
314789Sahrens 	for (i = 0; i < strlen(ends); i++) {
315789Sahrens 		if (toupper(buf[0]) == ends[i])
316789Sahrens 			break;
317789Sahrens 	}
3183972Svb160487 	if (i == strlen(ends)) {
3193972Svb160487 		(void) fprintf(stderr, "ztest: invalid bytes suffix: %s\n",
3203972Svb160487 		    buf);
3213972Svb160487 		usage(B_FALSE);
3223972Svb160487 	}
323789Sahrens 	if (buf[1] == '\0' || (toupper(buf[1]) == 'B' && buf[2] == '\0')) {
324789Sahrens 		return (10*i);
325789Sahrens 	}
3263972Svb160487 	(void) fprintf(stderr, "ztest: invalid bytes suffix: %s\n", buf);
3273972Svb160487 	usage(B_FALSE);
3283972Svb160487 	/* NOTREACHED */
329789Sahrens }
330789Sahrens 
331789Sahrens static uint64_t
332789Sahrens nicenumtoull(const char *buf)
333789Sahrens {
334789Sahrens 	char *end;
335789Sahrens 	uint64_t val;
336789Sahrens 
337789Sahrens 	val = strtoull(buf, &end, 0);
338789Sahrens 	if (end == buf) {
3393972Svb160487 		(void) fprintf(stderr, "ztest: bad numeric value: %s\n", buf);
3403972Svb160487 		usage(B_FALSE);
341789Sahrens 	} else if (end[0] == '.') {
342789Sahrens 		double fval = strtod(buf, &end);
343789Sahrens 		fval *= pow(2, str2shift(end));
3443972Svb160487 		if (fval > UINT64_MAX) {
3453972Svb160487 			(void) fprintf(stderr, "ztest: value too large: %s\n",
3463972Svb160487 			    buf);
3473972Svb160487 			usage(B_FALSE);
3483972Svb160487 		}
349789Sahrens 		val = (uint64_t)fval;
350789Sahrens 	} else {
351789Sahrens 		int shift = str2shift(end);
3523972Svb160487 		if (shift >= 64 || (val << shift) >> shift != val) {
3533972Svb160487 			(void) fprintf(stderr, "ztest: value too large: %s\n",
3543972Svb160487 			    buf);
3553972Svb160487 			usage(B_FALSE);
3563972Svb160487 		}
357789Sahrens 		val <<= shift;
358789Sahrens 	}
359789Sahrens 	return (val);
360789Sahrens }
361789Sahrens 
362789Sahrens static void
3633972Svb160487 usage(boolean_t requested)
364789Sahrens {
365789Sahrens 	char nice_vdev_size[10];
366789Sahrens 	char nice_gang_bang[10];
3673972Svb160487 	FILE *fp = requested ? stdout : stderr;
368789Sahrens 
369789Sahrens 	nicenum(zopt_vdev_size, nice_vdev_size);
3705530Sbonwick 	nicenum(metaslab_gang_bang, nice_gang_bang);
371789Sahrens 
3723972Svb160487 	(void) fprintf(fp, "Usage: %s\n"
373789Sahrens 	    "\t[-v vdevs (default: %llu)]\n"
374789Sahrens 	    "\t[-s size_of_each_vdev (default: %s)]\n"
3751732Sbonwick 	    "\t[-a alignment_shift (default: %d) (use 0 for random)]\n"
376789Sahrens 	    "\t[-m mirror_copies (default: %d)]\n"
377789Sahrens 	    "\t[-r raidz_disks (default: %d)]\n"
3782082Seschrock 	    "\t[-R raidz_parity (default: %d)]\n"
379789Sahrens 	    "\t[-d datasets (default: %d)]\n"
380789Sahrens 	    "\t[-t threads (default: %d)]\n"
381789Sahrens 	    "\t[-g gang_block_threshold (default: %s)]\n"
382789Sahrens 	    "\t[-i initialize pool i times (default: %d)]\n"
383789Sahrens 	    "\t[-k kill percentage (default: %llu%%)]\n"
384789Sahrens 	    "\t[-p pool_name (default: %s)]\n"
385789Sahrens 	    "\t[-f file directory for vdev files (default: %s)]\n"
386789Sahrens 	    "\t[-V(erbose)] (use multiple times for ever more blather)\n"
3871732Sbonwick 	    "\t[-E(xisting)] (use existing pool instead of creating new one)\n"
388789Sahrens 	    "\t[-T time] total run time (default: %llu sec)\n"
389789Sahrens 	    "\t[-P passtime] time per pass (default: %llu sec)\n"
3903972Svb160487 	    "\t[-h] (print help)\n"
391789Sahrens 	    "",
392789Sahrens 	    cmdname,
3935329Sgw25295 	    (u_longlong_t)zopt_vdevs,			/* -v */
3945329Sgw25295 	    nice_vdev_size,				/* -s */
3955329Sgw25295 	    zopt_ashift,				/* -a */
3965329Sgw25295 	    zopt_mirrors,				/* -m */
3975329Sgw25295 	    zopt_raidz,					/* -r */
3985329Sgw25295 	    zopt_raidz_parity,				/* -R */
3995329Sgw25295 	    zopt_datasets,				/* -d */
4005329Sgw25295 	    zopt_threads,				/* -t */
4015329Sgw25295 	    nice_gang_bang,				/* -g */
4025329Sgw25295 	    zopt_init,					/* -i */
4035329Sgw25295 	    (u_longlong_t)zopt_killrate,		/* -k */
4045329Sgw25295 	    zopt_pool,					/* -p */
4055329Sgw25295 	    zopt_dir,					/* -f */
4065329Sgw25295 	    (u_longlong_t)zopt_time,			/* -T */
4077754SJeff.Bonwick@Sun.COM 	    (u_longlong_t)zopt_passtime);		/* -P */
4083972Svb160487 	exit(requested ? 0 : 1);
409789Sahrens }
410789Sahrens 
411789Sahrens static uint64_t
412789Sahrens ztest_random(uint64_t range)
413789Sahrens {
414789Sahrens 	uint64_t r;
415789Sahrens 
416789Sahrens 	if (range == 0)
417789Sahrens 		return (0);
418789Sahrens 
419789Sahrens 	if (read(ztest_random_fd, &r, sizeof (r)) != sizeof (r))
420789Sahrens 		fatal(1, "short read from /dev/urandom");
421789Sahrens 
422789Sahrens 	return (r % range);
423789Sahrens }
424789Sahrens 
4258241SJeff.Bonwick@Sun.COM /* ARGSUSED */
426789Sahrens static void
427789Sahrens ztest_record_enospc(char *s)
428789Sahrens {
429789Sahrens 	ztest_shared->zs_enospc_count++;
430789Sahrens }
431789Sahrens 
432789Sahrens static void
433789Sahrens process_options(int argc, char **argv)
434789Sahrens {
435789Sahrens 	int opt;
436789Sahrens 	uint64_t value;
437789Sahrens 
438789Sahrens 	/* By default, test gang blocks for blocks 32K and greater */
4395530Sbonwick 	metaslab_gang_bang = 32 << 10;
440789Sahrens 
441789Sahrens 	while ((opt = getopt(argc, argv,
4427754SJeff.Bonwick@Sun.COM 	    "v:s:a:m:r:R:d:t:g:i:k:p:f:VET:P:h")) != EOF) {
443789Sahrens 		value = 0;
444789Sahrens 		switch (opt) {
4454451Seschrock 		case 'v':
4464451Seschrock 		case 's':
4474451Seschrock 		case 'a':
4484451Seschrock 		case 'm':
4494451Seschrock 		case 'r':
4504451Seschrock 		case 'R':
4514451Seschrock 		case 'd':
4524451Seschrock 		case 't':
4534451Seschrock 		case 'g':
4544451Seschrock 		case 'i':
4554451Seschrock 		case 'k':
4564451Seschrock 		case 'T':
4574451Seschrock 		case 'P':
458789Sahrens 			value = nicenumtoull(optarg);
459789Sahrens 		}
460789Sahrens 		switch (opt) {
4614451Seschrock 		case 'v':
462789Sahrens 			zopt_vdevs = value;
463789Sahrens 			break;
4644451Seschrock 		case 's':
465789Sahrens 			zopt_vdev_size = MAX(SPA_MINDEVSIZE, value);
466789Sahrens 			break;
4674451Seschrock 		case 'a':
4681732Sbonwick 			zopt_ashift = value;
4691732Sbonwick 			break;
4704451Seschrock 		case 'm':
471789Sahrens 			zopt_mirrors = value;
472789Sahrens 			break;
4734451Seschrock 		case 'r':
474789Sahrens 			zopt_raidz = MAX(1, value);
475789Sahrens 			break;
4764451Seschrock 		case 'R':
4772082Seschrock 			zopt_raidz_parity = MIN(MAX(value, 1), 2);
4782082Seschrock 			break;
4794451Seschrock 		case 'd':
4801732Sbonwick 			zopt_datasets = MAX(1, value);
481789Sahrens 			break;
4824451Seschrock 		case 't':
483789Sahrens 			zopt_threads = MAX(1, value);
484789Sahrens 			break;
4854451Seschrock 		case 'g':
4865530Sbonwick 			metaslab_gang_bang = MAX(SPA_MINBLOCKSIZE << 1, value);
487789Sahrens 			break;
4884451Seschrock 		case 'i':
489789Sahrens 			zopt_init = value;
490789Sahrens 			break;
4914451Seschrock 		case 'k':
492789Sahrens 			zopt_killrate = value;
493789Sahrens 			break;
4944451Seschrock 		case 'p':
495789Sahrens 			zopt_pool = strdup(optarg);
496789Sahrens 			break;
4974451Seschrock 		case 'f':
498789Sahrens 			zopt_dir = strdup(optarg);
499789Sahrens 			break;
5004451Seschrock 		case 'V':
501789Sahrens 			zopt_verbose++;
502789Sahrens 			break;
5034451Seschrock 		case 'E':
504789Sahrens 			zopt_init = 0;
505789Sahrens 			break;
5064451Seschrock 		case 'T':
507789Sahrens 			zopt_time = value;
508789Sahrens 			break;
5094451Seschrock 		case 'P':
510789Sahrens 			zopt_passtime = MAX(1, value);
511789Sahrens 			break;
5124451Seschrock 		case 'h':
5133972Svb160487 			usage(B_TRUE);
5143972Svb160487 			break;
5154451Seschrock 		case '?':
5164451Seschrock 		default:
5173972Svb160487 			usage(B_FALSE);
518789Sahrens 			break;
519789Sahrens 		}
520789Sahrens 	}
521789Sahrens 
5222082Seschrock 	zopt_raidz_parity = MIN(zopt_raidz_parity, zopt_raidz - 1);
5232082Seschrock 
524789Sahrens 	zopt_vdevtime = (zopt_vdevs > 0 ? zopt_time / zopt_vdevs : UINT64_MAX);
5252082Seschrock 	zopt_maxfaults = MAX(zopt_mirrors, 1) * (zopt_raidz_parity + 1) - 1;
526789Sahrens }
527789Sahrens 
5281732Sbonwick static uint64_t
5291732Sbonwick ztest_get_ashift(void)
5301732Sbonwick {
5311732Sbonwick 	if (zopt_ashift == 0)
5321732Sbonwick 		return (SPA_MINBLOCKSHIFT + ztest_random(3));
5331732Sbonwick 	return (zopt_ashift);
5341732Sbonwick }
5351732Sbonwick 
536789Sahrens static nvlist_t *
5377754SJeff.Bonwick@Sun.COM make_vdev_file(char *path, char *aux, size_t size, uint64_t ashift)
538789Sahrens {
5397754SJeff.Bonwick@Sun.COM 	char pathbuf[MAXPATHLEN];
540789Sahrens 	uint64_t vdev;
541789Sahrens 	nvlist_t *file;
542789Sahrens 
5437754SJeff.Bonwick@Sun.COM 	if (ashift == 0)
5447754SJeff.Bonwick@Sun.COM 		ashift = ztest_get_ashift();
5457754SJeff.Bonwick@Sun.COM 
5467754SJeff.Bonwick@Sun.COM 	if (path == NULL) {
5477754SJeff.Bonwick@Sun.COM 		path = pathbuf;
5487754SJeff.Bonwick@Sun.COM 
5497754SJeff.Bonwick@Sun.COM 		if (aux != NULL) {
5507754SJeff.Bonwick@Sun.COM 			vdev = ztest_shared->zs_vdev_aux;
5517754SJeff.Bonwick@Sun.COM 			(void) sprintf(path, ztest_aux_template,
5527754SJeff.Bonwick@Sun.COM 			    zopt_dir, zopt_pool, aux, vdev);
5537754SJeff.Bonwick@Sun.COM 		} else {
5547754SJeff.Bonwick@Sun.COM 			vdev = ztest_shared->zs_vdev_primaries++;
5557754SJeff.Bonwick@Sun.COM 			(void) sprintf(path, ztest_dev_template,
5567754SJeff.Bonwick@Sun.COM 			    zopt_dir, zopt_pool, vdev);
5577754SJeff.Bonwick@Sun.COM 		}
5587754SJeff.Bonwick@Sun.COM 	}
5597754SJeff.Bonwick@Sun.COM 
5607754SJeff.Bonwick@Sun.COM 	if (size != 0) {
5617754SJeff.Bonwick@Sun.COM 		int fd = open(path, O_RDWR | O_CREAT | O_TRUNC, 0666);
562789Sahrens 		if (fd == -1)
5637754SJeff.Bonwick@Sun.COM 			fatal(1, "can't open %s", path);
564789Sahrens 		if (ftruncate(fd, size) != 0)
5657754SJeff.Bonwick@Sun.COM 			fatal(1, "can't ftruncate %s", path);
566789Sahrens 		(void) close(fd);
567789Sahrens 	}
568789Sahrens 
569789Sahrens 	VERIFY(nvlist_alloc(&file, NV_UNIQUE_NAME, 0) == 0);
570789Sahrens 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_TYPE, VDEV_TYPE_FILE) == 0);
5717754SJeff.Bonwick@Sun.COM 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_PATH, path) == 0);
5721732Sbonwick 	VERIFY(nvlist_add_uint64(file, ZPOOL_CONFIG_ASHIFT, ashift) == 0);
573789Sahrens 
574789Sahrens 	return (file);
575789Sahrens }
576789Sahrens 
577789Sahrens static nvlist_t *
5787754SJeff.Bonwick@Sun.COM make_vdev_raidz(char *path, char *aux, size_t size, uint64_t ashift, int r)
579789Sahrens {
580789Sahrens 	nvlist_t *raidz, **child;
581789Sahrens 	int c;
582789Sahrens 
583789Sahrens 	if (r < 2)
5847754SJeff.Bonwick@Sun.COM 		return (make_vdev_file(path, aux, size, ashift));
585789Sahrens 	child = umem_alloc(r * sizeof (nvlist_t *), UMEM_NOFAIL);
586789Sahrens 
587789Sahrens 	for (c = 0; c < r; c++)
5887754SJeff.Bonwick@Sun.COM 		child[c] = make_vdev_file(path, aux, size, ashift);
589789Sahrens 
590789Sahrens 	VERIFY(nvlist_alloc(&raidz, NV_UNIQUE_NAME, 0) == 0);
591789Sahrens 	VERIFY(nvlist_add_string(raidz, ZPOOL_CONFIG_TYPE,
592789Sahrens 	    VDEV_TYPE_RAIDZ) == 0);
5932082Seschrock 	VERIFY(nvlist_add_uint64(raidz, ZPOOL_CONFIG_NPARITY,
5942082Seschrock 	    zopt_raidz_parity) == 0);
595789Sahrens 	VERIFY(nvlist_add_nvlist_array(raidz, ZPOOL_CONFIG_CHILDREN,
596789Sahrens 	    child, r) == 0);
597789Sahrens 
598789Sahrens 	for (c = 0; c < r; c++)
599789Sahrens 		nvlist_free(child[c]);
600789Sahrens 
601789Sahrens 	umem_free(child, r * sizeof (nvlist_t *));
602789Sahrens 
603789Sahrens 	return (raidz);
604789Sahrens }
605789Sahrens 
606789Sahrens static nvlist_t *
6077754SJeff.Bonwick@Sun.COM make_vdev_mirror(char *path, char *aux, size_t size, uint64_t ashift,
6087768SJeff.Bonwick@Sun.COM 	int r, int m)
609789Sahrens {
610789Sahrens 	nvlist_t *mirror, **child;
611789Sahrens 	int c;
612789Sahrens 
613789Sahrens 	if (m < 1)
6147754SJeff.Bonwick@Sun.COM 		return (make_vdev_raidz(path, aux, size, ashift, r));
615789Sahrens 
616789Sahrens 	child = umem_alloc(m * sizeof (nvlist_t *), UMEM_NOFAIL);
617789Sahrens 
618789Sahrens 	for (c = 0; c < m; c++)
6197754SJeff.Bonwick@Sun.COM 		child[c] = make_vdev_raidz(path, aux, size, ashift, r);
620789Sahrens 
621789Sahrens 	VERIFY(nvlist_alloc(&mirror, NV_UNIQUE_NAME, 0) == 0);
622789Sahrens 	VERIFY(nvlist_add_string(mirror, ZPOOL_CONFIG_TYPE,
623789Sahrens 	    VDEV_TYPE_MIRROR) == 0);
624789Sahrens 	VERIFY(nvlist_add_nvlist_array(mirror, ZPOOL_CONFIG_CHILDREN,
625789Sahrens 	    child, m) == 0);
626789Sahrens 
627789Sahrens 	for (c = 0; c < m; c++)
628789Sahrens 		nvlist_free(child[c]);
629789Sahrens 
630789Sahrens 	umem_free(child, m * sizeof (nvlist_t *));
631789Sahrens 
632789Sahrens 	return (mirror);
633789Sahrens }
634789Sahrens 
635789Sahrens static nvlist_t *
6367754SJeff.Bonwick@Sun.COM make_vdev_root(char *path, char *aux, size_t size, uint64_t ashift,
6377754SJeff.Bonwick@Sun.COM 	int log, int r, int m, int t)
638789Sahrens {
639789Sahrens 	nvlist_t *root, **child;
640789Sahrens 	int c;
641789Sahrens 
642789Sahrens 	ASSERT(t > 0);
643789Sahrens 
644789Sahrens 	child = umem_alloc(t * sizeof (nvlist_t *), UMEM_NOFAIL);
645789Sahrens 
6467768SJeff.Bonwick@Sun.COM 	for (c = 0; c < t; c++) {
6477768SJeff.Bonwick@Sun.COM 		child[c] = make_vdev_mirror(path, aux, size, ashift, r, m);
6487768SJeff.Bonwick@Sun.COM 		VERIFY(nvlist_add_uint64(child[c], ZPOOL_CONFIG_IS_LOG,
6497768SJeff.Bonwick@Sun.COM 		    log) == 0);
6507768SJeff.Bonwick@Sun.COM 	}
651789Sahrens 
652789Sahrens 	VERIFY(nvlist_alloc(&root, NV_UNIQUE_NAME, 0) == 0);
653789Sahrens 	VERIFY(nvlist_add_string(root, ZPOOL_CONFIG_TYPE, VDEV_TYPE_ROOT) == 0);
6547754SJeff.Bonwick@Sun.COM 	VERIFY(nvlist_add_nvlist_array(root, aux ? aux : ZPOOL_CONFIG_CHILDREN,
655789Sahrens 	    child, t) == 0);
656789Sahrens 
657789Sahrens 	for (c = 0; c < t; c++)
658789Sahrens 		nvlist_free(child[c]);
659789Sahrens 
660789Sahrens 	umem_free(child, t * sizeof (nvlist_t *));
661789Sahrens 
662789Sahrens 	return (root);
663789Sahrens }
664789Sahrens 
665789Sahrens static void
666789Sahrens ztest_set_random_blocksize(objset_t *os, uint64_t object, dmu_tx_t *tx)
667789Sahrens {
668789Sahrens 	int bs = SPA_MINBLOCKSHIFT +
669789Sahrens 	    ztest_random(SPA_MAXBLOCKSHIFT - SPA_MINBLOCKSHIFT + 1);
670789Sahrens 	int ibs = DN_MIN_INDBLKSHIFT +
671789Sahrens 	    ztest_random(DN_MAX_INDBLKSHIFT - DN_MIN_INDBLKSHIFT + 1);
672789Sahrens 	int error;
673789Sahrens 
674789Sahrens 	error = dmu_object_set_blocksize(os, object, 1ULL << bs, ibs, tx);
675789Sahrens 	if (error) {
676789Sahrens 		char osname[300];
677789Sahrens 		dmu_objset_name(os, osname);
678789Sahrens 		fatal(0, "dmu_object_set_blocksize('%s', %llu, %d, %d) = %d",
679789Sahrens 		    osname, object, 1 << bs, ibs, error);
680789Sahrens 	}
681789Sahrens }
682789Sahrens 
683789Sahrens static uint8_t
684789Sahrens ztest_random_checksum(void)
685789Sahrens {
686789Sahrens 	uint8_t checksum;
687789Sahrens 
688789Sahrens 	do {
689789Sahrens 		checksum = ztest_random(ZIO_CHECKSUM_FUNCTIONS);
690789Sahrens 	} while (zio_checksum_table[checksum].ci_zbt);
691789Sahrens 
692789Sahrens 	if (checksum == ZIO_CHECKSUM_OFF)
693789Sahrens 		checksum = ZIO_CHECKSUM_ON;
694789Sahrens 
695789Sahrens 	return (checksum);
696789Sahrens }
697789Sahrens 
698789Sahrens static uint8_t
699789Sahrens ztest_random_compress(void)
700789Sahrens {
701789Sahrens 	return ((uint8_t)ztest_random(ZIO_COMPRESS_FUNCTIONS));
702789Sahrens }
703789Sahrens 
704789Sahrens static int
7058227SNeil.Perrin@Sun.COM ztest_replay_create(objset_t *os, lr_create_t *lr, boolean_t byteswap)
706789Sahrens {
707789Sahrens 	dmu_tx_t *tx;
708789Sahrens 	int error;
709789Sahrens 
710789Sahrens 	if (byteswap)
711789Sahrens 		byteswap_uint64_array(lr, sizeof (*lr));
712789Sahrens 
713789Sahrens 	tx = dmu_tx_create(os);
714789Sahrens 	dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
7158227SNeil.Perrin@Sun.COM 	error = dmu_tx_assign(tx, TXG_WAIT);
716789Sahrens 	if (error) {
717789Sahrens 		dmu_tx_abort(tx);
718789Sahrens 		return (error);
719789Sahrens 	}
720789Sahrens 
721789Sahrens 	error = dmu_object_claim(os, lr->lr_doid, lr->lr_mode, 0,
722789Sahrens 	    DMU_OT_NONE, 0, tx);
7232082Seschrock 	ASSERT3U(error, ==, 0);
724789Sahrens 	dmu_tx_commit(tx);
725789Sahrens 
726789Sahrens 	if (zopt_verbose >= 5) {
727789Sahrens 		char osname[MAXNAMELEN];
728789Sahrens 		dmu_objset_name(os, osname);
729789Sahrens 		(void) printf("replay create of %s object %llu"
730789Sahrens 		    " in txg %llu = %d\n",
731789Sahrens 		    osname, (u_longlong_t)lr->lr_doid,
7328227SNeil.Perrin@Sun.COM 		    (u_longlong_t)dmu_tx_get_txg(tx), error);
733789Sahrens 	}
734789Sahrens 
735789Sahrens 	return (error);
736789Sahrens }
737789Sahrens 
738789Sahrens static int
7398227SNeil.Perrin@Sun.COM ztest_replay_remove(objset_t *os, lr_remove_t *lr, boolean_t byteswap)
740789Sahrens {
741789Sahrens 	dmu_tx_t *tx;
742789Sahrens 	int error;
743789Sahrens 
744789Sahrens 	if (byteswap)
745789Sahrens 		byteswap_uint64_array(lr, sizeof (*lr));
746789Sahrens 
747789Sahrens 	tx = dmu_tx_create(os);
748789Sahrens 	dmu_tx_hold_free(tx, lr->lr_doid, 0, DMU_OBJECT_END);
7498227SNeil.Perrin@Sun.COM 	error = dmu_tx_assign(tx, TXG_WAIT);
750789Sahrens 	if (error) {
751789Sahrens 		dmu_tx_abort(tx);
752789Sahrens 		return (error);
753789Sahrens 	}
754789Sahrens 
755789Sahrens 	error = dmu_object_free(os, lr->lr_doid, tx);
756789Sahrens 	dmu_tx_commit(tx);
757789Sahrens 
758789Sahrens 	return (error);
759789Sahrens }
760789Sahrens 
761789Sahrens zil_replay_func_t *ztest_replay_vector[TX_MAX_TYPE] = {
762789Sahrens 	NULL,			/* 0 no such transaction type */
763789Sahrens 	ztest_replay_create,	/* TX_CREATE */
764789Sahrens 	NULL,			/* TX_MKDIR */
765789Sahrens 	NULL,			/* TX_MKXATTR */
766789Sahrens 	NULL,			/* TX_SYMLINK */
767789Sahrens 	ztest_replay_remove,	/* TX_REMOVE */
768789Sahrens 	NULL,			/* TX_RMDIR */
769789Sahrens 	NULL,			/* TX_LINK */
770789Sahrens 	NULL,			/* TX_RENAME */
771789Sahrens 	NULL,			/* TX_WRITE */
772789Sahrens 	NULL,			/* TX_TRUNCATE */
773789Sahrens 	NULL,			/* TX_SETATTR */
774789Sahrens 	NULL,			/* TX_ACL */
775789Sahrens };
776789Sahrens 
777789Sahrens /*
778789Sahrens  * Verify that we can't destroy an active pool, create an existing pool,
779789Sahrens  * or create a pool with a bad vdev spec.
780789Sahrens  */
781789Sahrens void
782789Sahrens ztest_spa_create_destroy(ztest_args_t *za)
783789Sahrens {
784789Sahrens 	int error;
785789Sahrens 	spa_t *spa;
786789Sahrens 	nvlist_t *nvroot;
787789Sahrens 
788789Sahrens 	/*
789789Sahrens 	 * Attempt to create using a bad file.
790789Sahrens 	 */
7917754SJeff.Bonwick@Sun.COM 	nvroot = make_vdev_root("/dev/bogus", NULL, 0, 0, 0, 0, 0, 1);
7927184Stimh 	error = spa_create("ztest_bad_file", nvroot, NULL, NULL, NULL);
793789Sahrens 	nvlist_free(nvroot);
794789Sahrens 	if (error != ENOENT)
795789Sahrens 		fatal(0, "spa_create(bad_file) = %d", error);
796789Sahrens 
797789Sahrens 	/*
798789Sahrens 	 * Attempt to create using a bad mirror.
799789Sahrens 	 */
8007754SJeff.Bonwick@Sun.COM 	nvroot = make_vdev_root("/dev/bogus", NULL, 0, 0, 0, 0, 2, 1);
8017184Stimh 	error = spa_create("ztest_bad_mirror", nvroot, NULL, NULL, NULL);
802789Sahrens 	nvlist_free(nvroot);
803789Sahrens 	if (error != ENOENT)
804789Sahrens 		fatal(0, "spa_create(bad_mirror) = %d", error);
805789Sahrens 
806789Sahrens 	/*
807789Sahrens 	 * Attempt to create an existing pool.  It shouldn't matter
808789Sahrens 	 * what's in the nvroot; we should fail with EEXIST.
809789Sahrens 	 */
810789Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
8117754SJeff.Bonwick@Sun.COM 	nvroot = make_vdev_root("/dev/bogus", NULL, 0, 0, 0, 0, 0, 1);
8127184Stimh 	error = spa_create(za->za_pool, nvroot, NULL, NULL, NULL);
813789Sahrens 	nvlist_free(nvroot);
814789Sahrens 	if (error != EEXIST)
815789Sahrens 		fatal(0, "spa_create(whatever) = %d", error);
816789Sahrens 
817789Sahrens 	error = spa_open(za->za_pool, &spa, FTAG);
818789Sahrens 	if (error)
819789Sahrens 		fatal(0, "spa_open() = %d", error);
820789Sahrens 
821789Sahrens 	error = spa_destroy(za->za_pool);
822789Sahrens 	if (error != EBUSY)
823789Sahrens 		fatal(0, "spa_destroy() = %d", error);
824789Sahrens 
825789Sahrens 	spa_close(spa, FTAG);
826789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
827789Sahrens }
828789Sahrens 
8297768SJeff.Bonwick@Sun.COM static vdev_t *
8307768SJeff.Bonwick@Sun.COM vdev_lookup_by_path(vdev_t *vd, const char *path)
8317768SJeff.Bonwick@Sun.COM {
8327768SJeff.Bonwick@Sun.COM 	vdev_t *mvd;
8337768SJeff.Bonwick@Sun.COM 
8347768SJeff.Bonwick@Sun.COM 	if (vd->vdev_path != NULL && strcmp(path, vd->vdev_path) == 0)
8357768SJeff.Bonwick@Sun.COM 		return (vd);
8367768SJeff.Bonwick@Sun.COM 
8377768SJeff.Bonwick@Sun.COM 	for (int c = 0; c < vd->vdev_children; c++)
8387768SJeff.Bonwick@Sun.COM 		if ((mvd = vdev_lookup_by_path(vd->vdev_child[c], path)) !=
8397768SJeff.Bonwick@Sun.COM 		    NULL)
8407768SJeff.Bonwick@Sun.COM 			return (mvd);
8417768SJeff.Bonwick@Sun.COM 
8427768SJeff.Bonwick@Sun.COM 	return (NULL);
8437768SJeff.Bonwick@Sun.COM }
8447768SJeff.Bonwick@Sun.COM 
845789Sahrens /*
846789Sahrens  * Verify that vdev_add() works as expected.
847789Sahrens  */
848789Sahrens void
849789Sahrens ztest_vdev_add_remove(ztest_args_t *za)
850789Sahrens {
8515530Sbonwick 	spa_t *spa = za->za_spa;
852789Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
853789Sahrens 	nvlist_t *nvroot;
854789Sahrens 	int error;
855789Sahrens 
856789Sahrens 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
857789Sahrens 
8587754SJeff.Bonwick@Sun.COM 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
859789Sahrens 
860789Sahrens 	ztest_shared->zs_vdev_primaries =
861789Sahrens 	    spa->spa_root_vdev->vdev_children * leaves;
862789Sahrens 
8637754SJeff.Bonwick@Sun.COM 	spa_config_exit(spa, SCL_VDEV, FTAG);
864789Sahrens 
8654527Sperrin 	/*
8664527Sperrin 	 * Make 1/4 of the devices be log devices.
8674527Sperrin 	 */
8687754SJeff.Bonwick@Sun.COM 	nvroot = make_vdev_root(NULL, NULL, zopt_vdev_size, 0,
8694527Sperrin 	    ztest_random(4) == 0, zopt_raidz, zopt_mirrors, 1);
8704527Sperrin 
871789Sahrens 	error = spa_vdev_add(spa, nvroot);
872789Sahrens 	nvlist_free(nvroot);
873789Sahrens 
874789Sahrens 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
875789Sahrens 
876789Sahrens 	if (error == ENOSPC)
877789Sahrens 		ztest_record_enospc("spa_vdev_add");
878789Sahrens 	else if (error != 0)
879789Sahrens 		fatal(0, "spa_vdev_add() = %d", error);
8807754SJeff.Bonwick@Sun.COM }
8817754SJeff.Bonwick@Sun.COM 
8827754SJeff.Bonwick@Sun.COM /*
8837754SJeff.Bonwick@Sun.COM  * Verify that adding/removing aux devices (l2arc, hot spare) works as expected.
8847754SJeff.Bonwick@Sun.COM  */
8857754SJeff.Bonwick@Sun.COM void
8867754SJeff.Bonwick@Sun.COM ztest_vdev_aux_add_remove(ztest_args_t *za)
8877754SJeff.Bonwick@Sun.COM {
8887754SJeff.Bonwick@Sun.COM 	spa_t *spa = za->za_spa;
8897768SJeff.Bonwick@Sun.COM 	vdev_t *rvd = spa->spa_root_vdev;
8907754SJeff.Bonwick@Sun.COM 	spa_aux_vdev_t *sav;
8917754SJeff.Bonwick@Sun.COM 	char *aux;
8927754SJeff.Bonwick@Sun.COM 	uint64_t guid = 0;
8937754SJeff.Bonwick@Sun.COM 	int error;
8947754SJeff.Bonwick@Sun.COM 
8957768SJeff.Bonwick@Sun.COM 	if (ztest_random(2) == 0) {
8967754SJeff.Bonwick@Sun.COM 		sav = &spa->spa_spares;
8977754SJeff.Bonwick@Sun.COM 		aux = ZPOOL_CONFIG_SPARES;
8987754SJeff.Bonwick@Sun.COM 	} else {
8997754SJeff.Bonwick@Sun.COM 		sav = &spa->spa_l2cache;
9007754SJeff.Bonwick@Sun.COM 		aux = ZPOOL_CONFIG_L2CACHE;
9017754SJeff.Bonwick@Sun.COM 	}
9027754SJeff.Bonwick@Sun.COM 
9037754SJeff.Bonwick@Sun.COM 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
9047754SJeff.Bonwick@Sun.COM 
9057754SJeff.Bonwick@Sun.COM 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
9067754SJeff.Bonwick@Sun.COM 
9077754SJeff.Bonwick@Sun.COM 	if (sav->sav_count != 0 && ztest_random(4) == 0) {
9087754SJeff.Bonwick@Sun.COM 		/*
9097754SJeff.Bonwick@Sun.COM 		 * Pick a random device to remove.
9107754SJeff.Bonwick@Sun.COM 		 */
9117754SJeff.Bonwick@Sun.COM 		guid = sav->sav_vdevs[ztest_random(sav->sav_count)]->vdev_guid;
9127754SJeff.Bonwick@Sun.COM 	} else {
9137754SJeff.Bonwick@Sun.COM 		/*
9147754SJeff.Bonwick@Sun.COM 		 * Find an unused device we can add.
9157754SJeff.Bonwick@Sun.COM 		 */
9167754SJeff.Bonwick@Sun.COM 		ztest_shared->zs_vdev_aux = 0;
9177754SJeff.Bonwick@Sun.COM 		for (;;) {
9187754SJeff.Bonwick@Sun.COM 			char path[MAXPATHLEN];
9197754SJeff.Bonwick@Sun.COM 			int c;
9207754SJeff.Bonwick@Sun.COM 			(void) sprintf(path, ztest_aux_template, zopt_dir,
9217754SJeff.Bonwick@Sun.COM 			    zopt_pool, aux, ztest_shared->zs_vdev_aux);
9227754SJeff.Bonwick@Sun.COM 			for (c = 0; c < sav->sav_count; c++)
9237754SJeff.Bonwick@Sun.COM 				if (strcmp(sav->sav_vdevs[c]->vdev_path,
9247754SJeff.Bonwick@Sun.COM 				    path) == 0)
9257754SJeff.Bonwick@Sun.COM 					break;
9267768SJeff.Bonwick@Sun.COM 			if (c == sav->sav_count &&
9277768SJeff.Bonwick@Sun.COM 			    vdev_lookup_by_path(rvd, path) == NULL)
9287754SJeff.Bonwick@Sun.COM 				break;
9297754SJeff.Bonwick@Sun.COM 			ztest_shared->zs_vdev_aux++;
9307754SJeff.Bonwick@Sun.COM 		}
9317754SJeff.Bonwick@Sun.COM 	}
9327754SJeff.Bonwick@Sun.COM 
9337754SJeff.Bonwick@Sun.COM 	spa_config_exit(spa, SCL_VDEV, FTAG);
9347754SJeff.Bonwick@Sun.COM 
9357754SJeff.Bonwick@Sun.COM 	if (guid == 0) {
9367754SJeff.Bonwick@Sun.COM 		/*
9377754SJeff.Bonwick@Sun.COM 		 * Add a new device.
9387754SJeff.Bonwick@Sun.COM 		 */
9397768SJeff.Bonwick@Sun.COM 		nvlist_t *nvroot = make_vdev_root(NULL, aux,
9407768SJeff.Bonwick@Sun.COM 		    (zopt_vdev_size * 5) / 4, 0, 0, 0, 0, 1);
9417754SJeff.Bonwick@Sun.COM 		error = spa_vdev_add(spa, nvroot);
9427754SJeff.Bonwick@Sun.COM 		if (error != 0)
9437754SJeff.Bonwick@Sun.COM 			fatal(0, "spa_vdev_add(%p) = %d", nvroot, error);
9447754SJeff.Bonwick@Sun.COM 		nvlist_free(nvroot);
9457754SJeff.Bonwick@Sun.COM 	} else {
9467754SJeff.Bonwick@Sun.COM 		/*
9477754SJeff.Bonwick@Sun.COM 		 * Remove an existing device.  Sometimes, dirty its
9487754SJeff.Bonwick@Sun.COM 		 * vdev state first to make sure we handle removal
9497754SJeff.Bonwick@Sun.COM 		 * of devices that have pending state changes.
9507754SJeff.Bonwick@Sun.COM 		 */
9517754SJeff.Bonwick@Sun.COM 		if (ztest_random(2) == 0)
9527754SJeff.Bonwick@Sun.COM 			(void) vdev_online(spa, guid, B_FALSE, NULL);
9537754SJeff.Bonwick@Sun.COM 
9547754SJeff.Bonwick@Sun.COM 		error = spa_vdev_remove(spa, guid, B_FALSE);
9557754SJeff.Bonwick@Sun.COM 		if (error != 0 && error != EBUSY)
9567754SJeff.Bonwick@Sun.COM 			fatal(0, "spa_vdev_remove(%llu) = %d", guid, error);
9577754SJeff.Bonwick@Sun.COM 	}
9587754SJeff.Bonwick@Sun.COM 
9597754SJeff.Bonwick@Sun.COM 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
960789Sahrens }
961789Sahrens 
962789Sahrens /*
963789Sahrens  * Verify that we can attach and detach devices.
964789Sahrens  */
965789Sahrens void
966789Sahrens ztest_vdev_attach_detach(ztest_args_t *za)
967789Sahrens {
9685530Sbonwick 	spa_t *spa = za->za_spa;
9697768SJeff.Bonwick@Sun.COM 	spa_aux_vdev_t *sav = &spa->spa_spares;
970789Sahrens 	vdev_t *rvd = spa->spa_root_vdev;
9711544Seschrock 	vdev_t *oldvd, *newvd, *pvd;
9727754SJeff.Bonwick@Sun.COM 	nvlist_t *root;
973789Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
974789Sahrens 	uint64_t leaf, top;
9751732Sbonwick 	uint64_t ashift = ztest_get_ashift();
9768241SJeff.Bonwick@Sun.COM 	uint64_t oldguid, pguid;
9771544Seschrock 	size_t oldsize, newsize;
9781544Seschrock 	char oldpath[MAXPATHLEN], newpath[MAXPATHLEN];
979789Sahrens 	int replacing;
9807793SJeff.Bonwick@Sun.COM 	int oldvd_has_siblings = B_FALSE;
9817768SJeff.Bonwick@Sun.COM 	int newvd_is_spare = B_FALSE;
9827768SJeff.Bonwick@Sun.COM 	int oldvd_is_log;
983789Sahrens 	int error, expected_error;
984789Sahrens 
985789Sahrens 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
986789Sahrens 
9877754SJeff.Bonwick@Sun.COM 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
988789Sahrens 
989789Sahrens 	/*
990789Sahrens 	 * Decide whether to do an attach or a replace.
991789Sahrens 	 */
992789Sahrens 	replacing = ztest_random(2);
993789Sahrens 
994789Sahrens 	/*
995789Sahrens 	 * Pick a random top-level vdev.
996789Sahrens 	 */
997789Sahrens 	top = ztest_random(rvd->vdev_children);
998789Sahrens 
999789Sahrens 	/*
1000789Sahrens 	 * Pick a random leaf within it.
1001789Sahrens 	 */
1002789Sahrens 	leaf = ztest_random(leaves);
1003789Sahrens 
1004789Sahrens 	/*
10057768SJeff.Bonwick@Sun.COM 	 * Locate this vdev.
1006789Sahrens 	 */
10077768SJeff.Bonwick@Sun.COM 	oldvd = rvd->vdev_child[top];
10088241SJeff.Bonwick@Sun.COM 	if (zopt_mirrors >= 1) {
10098241SJeff.Bonwick@Sun.COM 		ASSERT(oldvd->vdev_ops == &vdev_mirror_ops);
10108241SJeff.Bonwick@Sun.COM 		ASSERT(oldvd->vdev_children >= zopt_mirrors);
10117768SJeff.Bonwick@Sun.COM 		oldvd = oldvd->vdev_child[leaf / zopt_raidz];
10128241SJeff.Bonwick@Sun.COM 	}
10138241SJeff.Bonwick@Sun.COM 	if (zopt_raidz > 1) {
10148241SJeff.Bonwick@Sun.COM 		ASSERT(oldvd->vdev_ops == &vdev_raidz_ops);
10158241SJeff.Bonwick@Sun.COM 		ASSERT(oldvd->vdev_children == zopt_raidz);
10167768SJeff.Bonwick@Sun.COM 		oldvd = oldvd->vdev_child[leaf % zopt_raidz];
10178241SJeff.Bonwick@Sun.COM 	}
1018789Sahrens 
1019789Sahrens 	/*
10207768SJeff.Bonwick@Sun.COM 	 * If we're already doing an attach or replace, oldvd may be a
10217768SJeff.Bonwick@Sun.COM 	 * mirror vdev -- in which case, pick a random child.
1022789Sahrens 	 */
10237768SJeff.Bonwick@Sun.COM 	while (oldvd->vdev_children != 0) {
10247793SJeff.Bonwick@Sun.COM 		oldvd_has_siblings = B_TRUE;
10258241SJeff.Bonwick@Sun.COM 		ASSERT(oldvd->vdev_children >= 2);
10268241SJeff.Bonwick@Sun.COM 		oldvd = oldvd->vdev_child[ztest_random(oldvd->vdev_children)];
10277768SJeff.Bonwick@Sun.COM 	}
10287768SJeff.Bonwick@Sun.COM 
10297768SJeff.Bonwick@Sun.COM 	oldguid = oldvd->vdev_guid;
10307768SJeff.Bonwick@Sun.COM 	oldsize = vdev_get_rsize(oldvd);
10317768SJeff.Bonwick@Sun.COM 	oldvd_is_log = oldvd->vdev_top->vdev_islog;
10327768SJeff.Bonwick@Sun.COM 	(void) strcpy(oldpath, oldvd->vdev_path);
10331544Seschrock 	pvd = oldvd->vdev_parent;
10348241SJeff.Bonwick@Sun.COM 	pguid = pvd->vdev_guid;
1035789Sahrens 
1036789Sahrens 	/*
10377793SJeff.Bonwick@Sun.COM 	 * If oldvd has siblings, then half of the time, detach it.
10387793SJeff.Bonwick@Sun.COM 	 */
10397793SJeff.Bonwick@Sun.COM 	if (oldvd_has_siblings && ztest_random(2) == 0) {
10407793SJeff.Bonwick@Sun.COM 		spa_config_exit(spa, SCL_VDEV, FTAG);
10418241SJeff.Bonwick@Sun.COM 		error = spa_vdev_detach(spa, oldguid, pguid, B_FALSE);
10428241SJeff.Bonwick@Sun.COM 		if (error != 0 && error != ENODEV && error != EBUSY &&
10438241SJeff.Bonwick@Sun.COM 		    error != ENOTSUP)
10448241SJeff.Bonwick@Sun.COM 			fatal(0, "detach (%s) returned %d", oldpath, error);
10457793SJeff.Bonwick@Sun.COM 		(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
10467793SJeff.Bonwick@Sun.COM 		return;
10477793SJeff.Bonwick@Sun.COM 	}
10487793SJeff.Bonwick@Sun.COM 
10497793SJeff.Bonwick@Sun.COM 	/*
10507768SJeff.Bonwick@Sun.COM 	 * For the new vdev, choose with equal probability between the two
10517768SJeff.Bonwick@Sun.COM 	 * standard paths (ending in either 'a' or 'b') or a random hot spare.
1052789Sahrens 	 */
10537768SJeff.Bonwick@Sun.COM 	if (sav->sav_count != 0 && ztest_random(3) == 0) {
10547768SJeff.Bonwick@Sun.COM 		newvd = sav->sav_vdevs[ztest_random(sav->sav_count)];
10557768SJeff.Bonwick@Sun.COM 		newvd_is_spare = B_TRUE;
10567768SJeff.Bonwick@Sun.COM 		(void) strcpy(newpath, newvd->vdev_path);
10577768SJeff.Bonwick@Sun.COM 	} else {
10587768SJeff.Bonwick@Sun.COM 		(void) snprintf(newpath, sizeof (newpath), ztest_dev_template,
10597768SJeff.Bonwick@Sun.COM 		    zopt_dir, zopt_pool, top * leaves + leaf);
10607768SJeff.Bonwick@Sun.COM 		if (ztest_random(2) == 0)
10617768SJeff.Bonwick@Sun.COM 			newpath[strlen(newpath) - 1] = 'b';
10627768SJeff.Bonwick@Sun.COM 		newvd = vdev_lookup_by_path(rvd, newpath);
10637768SJeff.Bonwick@Sun.COM 	}
10647768SJeff.Bonwick@Sun.COM 
10657768SJeff.Bonwick@Sun.COM 	if (newvd) {
10667768SJeff.Bonwick@Sun.COM 		newsize = vdev_get_rsize(newvd);
10677768SJeff.Bonwick@Sun.COM 	} else {
10687768SJeff.Bonwick@Sun.COM 		/*
10697768SJeff.Bonwick@Sun.COM 		 * Make newsize a little bigger or smaller than oldsize.
10707768SJeff.Bonwick@Sun.COM 		 * If it's smaller, the attach should fail.
10717768SJeff.Bonwick@Sun.COM 		 * If it's larger, and we're doing a replace,
10727768SJeff.Bonwick@Sun.COM 		 * we should get dynamic LUN growth when we're done.
10737768SJeff.Bonwick@Sun.COM 		 */
10747768SJeff.Bonwick@Sun.COM 		newsize = 10 * oldsize / (9 + ztest_random(3));
10757768SJeff.Bonwick@Sun.COM 	}
1076789Sahrens 
1077789Sahrens 	/*
1078789Sahrens 	 * If pvd is not a mirror or root, the attach should fail with ENOTSUP,
1079789Sahrens 	 * unless it's a replace; in that case any non-replacing parent is OK.
1080789Sahrens 	 *
10811544Seschrock 	 * If newvd is already part of the pool, it should fail with EBUSY.
1082789Sahrens 	 *
10831544Seschrock 	 * If newvd is too small, it should fail with EOVERFLOW.
1084789Sahrens 	 */
10857768SJeff.Bonwick@Sun.COM 	if (pvd->vdev_ops != &vdev_mirror_ops &&
10867768SJeff.Bonwick@Sun.COM 	    pvd->vdev_ops != &vdev_root_ops && (!replacing ||
10877768SJeff.Bonwick@Sun.COM 	    pvd->vdev_ops == &vdev_replacing_ops ||
10887768SJeff.Bonwick@Sun.COM 	    pvd->vdev_ops == &vdev_spare_ops))
10897768SJeff.Bonwick@Sun.COM 		expected_error = ENOTSUP;
10907768SJeff.Bonwick@Sun.COM 	else if (newvd_is_spare && (!replacing || oldvd_is_log))
10917768SJeff.Bonwick@Sun.COM 		expected_error = ENOTSUP;
10927768SJeff.Bonwick@Sun.COM 	else if (newvd == oldvd)
10937768SJeff.Bonwick@Sun.COM 		expected_error = replacing ? 0 : EBUSY;
10947768SJeff.Bonwick@Sun.COM 	else if (vdev_lookup_by_path(rvd, newpath) != NULL)
10952174Seschrock 		expected_error = EBUSY;
10961544Seschrock 	else if (newsize < oldsize)
1097789Sahrens 		expected_error = EOVERFLOW;
10981732Sbonwick 	else if (ashift > oldvd->vdev_top->vdev_ashift)
10991732Sbonwick 		expected_error = EDOM;
1100789Sahrens 	else
1101789Sahrens 		expected_error = 0;
1102789Sahrens 
11037754SJeff.Bonwick@Sun.COM 	spa_config_exit(spa, SCL_VDEV, FTAG);
1104789Sahrens 
1105789Sahrens 	/*
11061544Seschrock 	 * Build the nvlist describing newpath.
1107789Sahrens 	 */
11087754SJeff.Bonwick@Sun.COM 	root = make_vdev_root(newpath, NULL, newvd == NULL ? newsize : 0,
11097754SJeff.Bonwick@Sun.COM 	    ashift, 0, 0, 0, 1);
1110789Sahrens 
11117768SJeff.Bonwick@Sun.COM 	error = spa_vdev_attach(spa, oldguid, root, replacing);
1112789Sahrens 
1113789Sahrens 	nvlist_free(root);
1114789Sahrens 
1115789Sahrens 	/*
1116789Sahrens 	 * If our parent was the replacing vdev, but the replace completed,
1117789Sahrens 	 * then instead of failing with ENOTSUP we may either succeed,
1118789Sahrens 	 * fail with ENODEV, or fail with EOVERFLOW.
1119789Sahrens 	 */
1120789Sahrens 	if (expected_error == ENOTSUP &&
1121789Sahrens 	    (error == 0 || error == ENODEV || error == EOVERFLOW))
1122789Sahrens 		expected_error = error;
1123789Sahrens 
1124797Sbonwick 	/*
1125797Sbonwick 	 * If someone grew the LUN, the replacement may be too small.
1126797Sbonwick 	 */
11277046Sahrens 	if (error == EOVERFLOW || error == EBUSY)
1128797Sbonwick 		expected_error = error;
1129797Sbonwick 
11307046Sahrens 	/* XXX workaround 6690467 */
11317046Sahrens 	if (error != expected_error && expected_error != EBUSY) {
11327046Sahrens 		fatal(0, "attach (%s %llu, %s %llu, %d) "
11337046Sahrens 		    "returned %d, expected %d",
11347046Sahrens 		    oldpath, (longlong_t)oldsize, newpath,
11357046Sahrens 		    (longlong_t)newsize, replacing, error, expected_error);
1136789Sahrens 	}
1137789Sahrens 
1138789Sahrens 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
1139789Sahrens }
1140789Sahrens 
1141789Sahrens /*
1142789Sahrens  * Verify that dynamic LUN growth works as expected.
1143789Sahrens  */
1144789Sahrens void
1145789Sahrens ztest_vdev_LUN_growth(ztest_args_t *za)
1146789Sahrens {
11475530Sbonwick 	spa_t *spa = za->za_spa;
1148789Sahrens 	char dev_name[MAXPATHLEN];
1149789Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
1150789Sahrens 	uint64_t vdev;
1151789Sahrens 	size_t fsize;
1152789Sahrens 	int fd;
1153789Sahrens 
1154789Sahrens 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
1155789Sahrens 
1156789Sahrens 	/*
1157789Sahrens 	 * Pick a random leaf vdev.
1158789Sahrens 	 */
11597754SJeff.Bonwick@Sun.COM 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
1160789Sahrens 	vdev = ztest_random(spa->spa_root_vdev->vdev_children * leaves);
11617754SJeff.Bonwick@Sun.COM 	spa_config_exit(spa, SCL_VDEV, FTAG);
1162789Sahrens 
1163789Sahrens 	(void) sprintf(dev_name, ztest_dev_template, zopt_dir, zopt_pool, vdev);
1164789Sahrens 
1165789Sahrens 	if ((fd = open(dev_name, O_RDWR)) != -1) {
1166789Sahrens 		/*
1167789Sahrens 		 * Determine the size.
1168789Sahrens 		 */
1169789Sahrens 		fsize = lseek(fd, 0, SEEK_END);
1170789Sahrens 
1171789Sahrens 		/*
1172789Sahrens 		 * If it's less than 2x the original size, grow by around 3%.
1173789Sahrens 		 */
1174789Sahrens 		if (fsize < 2 * zopt_vdev_size) {
1175789Sahrens 			size_t newsize = fsize + ztest_random(fsize / 32);
1176789Sahrens 			(void) ftruncate(fd, newsize);
1177789Sahrens 			if (zopt_verbose >= 6) {
1178789Sahrens 				(void) printf("%s grew from %lu to %lu bytes\n",
1179789Sahrens 				    dev_name, (ulong_t)fsize, (ulong_t)newsize);
1180789Sahrens 			}
1181789Sahrens 		}
1182789Sahrens 		(void) close(fd);
1183789Sahrens 	}
1184789Sahrens 
1185789Sahrens 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
1186789Sahrens }
1187789Sahrens 
1188789Sahrens /* ARGSUSED */
1189789Sahrens static void
11904543Smarks ztest_create_cb(objset_t *os, void *arg, cred_t *cr, dmu_tx_t *tx)
1191789Sahrens {
1192789Sahrens 	/*
1193789Sahrens 	 * Create the directory object.
1194789Sahrens 	 */
1195789Sahrens 	VERIFY(dmu_object_claim(os, ZTEST_DIROBJ,
1196789Sahrens 	    DMU_OT_UINT64_OTHER, ZTEST_DIROBJ_BLOCKSIZE,
11975530Sbonwick 	    DMU_OT_UINT64_OTHER, 5 * sizeof (ztest_block_tag_t), tx) == 0);
1198789Sahrens 
1199789Sahrens 	VERIFY(zap_create_claim(os, ZTEST_MICROZAP_OBJ,
1200789Sahrens 	    DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx) == 0);
1201789Sahrens 
1202789Sahrens 	VERIFY(zap_create_claim(os, ZTEST_FATZAP_OBJ,
1203789Sahrens 	    DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx) == 0);
1204789Sahrens }
1205789Sahrens 
12062199Sahrens static int
1207789Sahrens ztest_destroy_cb(char *name, void *arg)
1208789Sahrens {
12095530Sbonwick 	ztest_args_t *za = arg;
1210789Sahrens 	objset_t *os;
12115530Sbonwick 	dmu_object_info_t *doi = &za->za_doi;
1212789Sahrens 	int error;
1213789Sahrens 
1214789Sahrens 	/*
1215789Sahrens 	 * Verify that the dataset contains a directory object.
1216789Sahrens 	 */
1217789Sahrens 	error = dmu_objset_open(name, DMU_OST_OTHER,
12186689Smaybee 	    DS_MODE_USER | DS_MODE_READONLY, &os);
1219789Sahrens 	ASSERT3U(error, ==, 0);
12205530Sbonwick 	error = dmu_object_info(os, ZTEST_DIROBJ, doi);
12211731Sbonwick 	if (error != ENOENT) {
12221731Sbonwick 		/* We could have crashed in the middle of destroying it */
12231731Sbonwick 		ASSERT3U(error, ==, 0);
12245530Sbonwick 		ASSERT3U(doi->doi_type, ==, DMU_OT_UINT64_OTHER);
12255530Sbonwick 		ASSERT3S(doi->doi_physical_blks, >=, 0);
12261731Sbonwick 	}
1227789Sahrens 	dmu_objset_close(os);
1228789Sahrens 
1229789Sahrens 	/*
1230789Sahrens 	 * Destroy the dataset.
1231789Sahrens 	 */
1232789Sahrens 	error = dmu_objset_destroy(name);
12336689Smaybee 	if (error) {
12346689Smaybee 		(void) dmu_objset_open(name, DMU_OST_OTHER,
12356689Smaybee 		    DS_MODE_USER | DS_MODE_READONLY, &os);
12366689Smaybee 		fatal(0, "dmu_objset_destroy(os=%p) = %d\n", &os, error);
12376689Smaybee 	}
12382199Sahrens 	return (0);
1239789Sahrens }
1240789Sahrens 
1241789Sahrens /*
1242789Sahrens  * Verify that dmu_objset_{create,destroy,open,close} work as expected.
1243789Sahrens  */
1244789Sahrens static uint64_t
1245789Sahrens ztest_log_create(zilog_t *zilog, dmu_tx_t *tx, uint64_t object, int mode)
1246789Sahrens {
1247789Sahrens 	itx_t *itx;
1248789Sahrens 	lr_create_t *lr;
1249789Sahrens 	size_t namesize;
1250789Sahrens 	char name[24];
1251789Sahrens 
1252789Sahrens 	(void) sprintf(name, "ZOBJ_%llu", (u_longlong_t)object);
1253789Sahrens 	namesize = strlen(name) + 1;
1254789Sahrens 
1255789Sahrens 	itx = zil_itx_create(TX_CREATE, sizeof (*lr) + namesize +
1256789Sahrens 	    ztest_random(ZIL_MAX_BLKSZ));
1257789Sahrens 	lr = (lr_create_t *)&itx->itx_lr;
1258789Sahrens 	bzero(lr + 1, lr->lr_common.lrc_reclen - sizeof (*lr));
1259789Sahrens 	lr->lr_doid = object;
1260789Sahrens 	lr->lr_foid = 0;
1261789Sahrens 	lr->lr_mode = mode;
1262789Sahrens 	lr->lr_uid = 0;
1263789Sahrens 	lr->lr_gid = 0;
1264789Sahrens 	lr->lr_gen = dmu_tx_get_txg(tx);
1265789Sahrens 	lr->lr_crtime[0] = time(NULL);
1266789Sahrens 	lr->lr_crtime[1] = 0;
1267789Sahrens 	lr->lr_rdev = 0;
1268789Sahrens 	bcopy(name, (char *)(lr + 1), namesize);
1269789Sahrens 
1270789Sahrens 	return (zil_itx_assign(zilog, itx, tx));
1271789Sahrens }
1272789Sahrens 
1273789Sahrens void
1274789Sahrens ztest_dmu_objset_create_destroy(ztest_args_t *za)
1275789Sahrens {
1276789Sahrens 	int error;
12776689Smaybee 	objset_t *os, *os2;
1278789Sahrens 	char name[100];
12796689Smaybee 	int basemode, expected_error;
1280789Sahrens 	zilog_t *zilog;
1281789Sahrens 	uint64_t seq;
1282789Sahrens 	uint64_t objects;
1283789Sahrens 
1284789Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
1285789Sahrens 	(void) snprintf(name, 100, "%s/%s_temp_%llu", za->za_pool, za->za_pool,
1286789Sahrens 	    (u_longlong_t)za->za_instance);
1287789Sahrens 
12886689Smaybee 	basemode = DS_MODE_TYPE(za->za_instance);
12896689Smaybee 	if (basemode != DS_MODE_USER && basemode != DS_MODE_OWNER)
12906689Smaybee 		basemode = DS_MODE_USER;
1291789Sahrens 
1292789Sahrens 	/*
1293789Sahrens 	 * If this dataset exists from a previous run, process its replay log
1294789Sahrens 	 * half of the time.  If we don't replay it, then dmu_objset_destroy()
1295789Sahrens 	 * (invoked from ztest_destroy_cb() below) should just throw it away.
1296789Sahrens 	 */
1297789Sahrens 	if (ztest_random(2) == 0 &&
12986689Smaybee 	    dmu_objset_open(name, DMU_OST_OTHER, DS_MODE_OWNER, &os) == 0) {
12998227SNeil.Perrin@Sun.COM 		zil_replay(os, os, ztest_replay_vector);
1300789Sahrens 		dmu_objset_close(os);
1301789Sahrens 	}
1302789Sahrens 
1303789Sahrens 	/*
1304789Sahrens 	 * There may be an old instance of the dataset we're about to
1305789Sahrens 	 * create lying around from a previous run.  If so, destroy it
1306789Sahrens 	 * and all of its snapshots.
1307789Sahrens 	 */
13085530Sbonwick 	(void) dmu_objset_find(name, ztest_destroy_cb, za,
13092417Sahrens 	    DS_FIND_CHILDREN | DS_FIND_SNAPSHOTS);
1310789Sahrens 
1311789Sahrens 	/*
1312789Sahrens 	 * Verify that the destroyed dataset is no longer in the namespace.
1313789Sahrens 	 */
1314789Sahrens 	error = dmu_objset_open(name, DMU_OST_OTHER, basemode, &os);
1315789Sahrens 	if (error != ENOENT)
1316789Sahrens 		fatal(1, "dmu_objset_open(%s) found destroyed dataset %p",
1317789Sahrens 		    name, os);
1318789Sahrens 
1319789Sahrens 	/*
1320789Sahrens 	 * Verify that we can create a new dataset.
1321789Sahrens 	 */
13226492Stimh 	error = dmu_objset_create(name, DMU_OST_OTHER, NULL, 0,
13236492Stimh 	    ztest_create_cb, NULL);
1324789Sahrens 	if (error) {
1325789Sahrens 		if (error == ENOSPC) {
1326789Sahrens 			ztest_record_enospc("dmu_objset_create");
1327789Sahrens 			(void) rw_unlock(&ztest_shared->zs_name_lock);
1328789Sahrens 			return;
1329789Sahrens 		}
1330789Sahrens 		fatal(0, "dmu_objset_create(%s) = %d", name, error);
1331789Sahrens 	}
1332789Sahrens 
1333789Sahrens 	error = dmu_objset_open(name, DMU_OST_OTHER, basemode, &os);
1334789Sahrens 	if (error) {
1335789Sahrens 		fatal(0, "dmu_objset_open(%s) = %d", name, error);
1336789Sahrens 	}
1337789Sahrens 
1338789Sahrens 	/*
1339789Sahrens 	 * Open the intent log for it.
1340789Sahrens 	 */
1341789Sahrens 	zilog = zil_open(os, NULL);
1342789Sahrens 
1343789Sahrens 	/*
1344789Sahrens 	 * Put a random number of objects in there.
1345789Sahrens 	 */
13461807Sbonwick 	objects = ztest_random(20);
1347789Sahrens 	seq = 0;
1348789Sahrens 	while (objects-- != 0) {
1349789Sahrens 		uint64_t object;
1350789Sahrens 		dmu_tx_t *tx = dmu_tx_create(os);
1351789Sahrens 		dmu_tx_hold_write(tx, DMU_NEW_OBJECT, 0, sizeof (name));
1352789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1353789Sahrens 		if (error) {
1354789Sahrens 			dmu_tx_abort(tx);
1355789Sahrens 		} else {
1356789Sahrens 			object = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1357789Sahrens 			    DMU_OT_NONE, 0, tx);
1358789Sahrens 			ztest_set_random_blocksize(os, object, tx);
1359789Sahrens 			seq = ztest_log_create(zilog, tx, object,
1360789Sahrens 			    DMU_OT_UINT64_OTHER);
1361789Sahrens 			dmu_write(os, object, 0, sizeof (name), name, tx);
1362789Sahrens 			dmu_tx_commit(tx);
1363789Sahrens 		}
1364789Sahrens 		if (ztest_random(5) == 0) {
13652638Sperrin 			zil_commit(zilog, seq, object);
1366789Sahrens 		}
13671807Sbonwick 		if (ztest_random(100) == 0) {
1368789Sahrens 			error = zil_suspend(zilog);
1369789Sahrens 			if (error == 0) {
1370789Sahrens 				zil_resume(zilog);
1371789Sahrens 			}
1372789Sahrens 		}
1373789Sahrens 	}
1374789Sahrens 
1375789Sahrens 	/*
1376789Sahrens 	 * Verify that we cannot create an existing dataset.
1377789Sahrens 	 */
13786492Stimh 	error = dmu_objset_create(name, DMU_OST_OTHER, NULL, 0, NULL, NULL);
1379789Sahrens 	if (error != EEXIST)
1380789Sahrens 		fatal(0, "created existing dataset, error = %d", error);
1381789Sahrens 
1382789Sahrens 	/*
13836689Smaybee 	 * Verify that multiple dataset holds are allowed, but only when
1384789Sahrens 	 * the new access mode is compatible with the base mode.
1385789Sahrens 	 */
13866689Smaybee 	if (basemode == DS_MODE_OWNER) {
13876689Smaybee 		error = dmu_objset_open(name, DMU_OST_OTHER, DS_MODE_USER,
13886689Smaybee 		    &os2);
13896689Smaybee 		if (error)
13906689Smaybee 			fatal(0, "dmu_objset_open('%s') = %d", name, error);
13916689Smaybee 		else
1392789Sahrens 			dmu_objset_close(os2);
1393789Sahrens 	}
13946689Smaybee 	error = dmu_objset_open(name, DMU_OST_OTHER, DS_MODE_OWNER, &os2);
13956689Smaybee 	expected_error = (basemode == DS_MODE_OWNER) ? EBUSY : 0;
13966689Smaybee 	if (error != expected_error)
13976689Smaybee 		fatal(0, "dmu_objset_open('%s') = %d, expected %d",
13986689Smaybee 		    name, error, expected_error);
13996689Smaybee 	if (error == 0)
14006689Smaybee 		dmu_objset_close(os2);
1401789Sahrens 
1402789Sahrens 	zil_close(zilog);
1403789Sahrens 	dmu_objset_close(os);
1404789Sahrens 
1405789Sahrens 	error = dmu_objset_destroy(name);
1406789Sahrens 	if (error)
1407789Sahrens 		fatal(0, "dmu_objset_destroy(%s) = %d", name, error);
1408789Sahrens 
1409789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
1410789Sahrens }
1411789Sahrens 
1412789Sahrens /*
1413789Sahrens  * Verify that dmu_snapshot_{create,destroy,open,close} work as expected.
1414789Sahrens  */
1415789Sahrens void
1416789Sahrens ztest_dmu_snapshot_create_destroy(ztest_args_t *za)
1417789Sahrens {
1418789Sahrens 	int error;
1419789Sahrens 	objset_t *os = za->za_os;
1420789Sahrens 	char snapname[100];
1421789Sahrens 	char osname[MAXNAMELEN];
1422789Sahrens 
1423789Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
1424789Sahrens 	dmu_objset_name(os, osname);
1425789Sahrens 	(void) snprintf(snapname, 100, "%s@%llu", osname,
1426789Sahrens 	    (u_longlong_t)za->za_instance);
1427789Sahrens 
1428789Sahrens 	error = dmu_objset_destroy(snapname);
1429789Sahrens 	if (error != 0 && error != ENOENT)
1430789Sahrens 		fatal(0, "dmu_objset_destroy() = %d", error);
14312199Sahrens 	error = dmu_objset_snapshot(osname, strchr(snapname, '@')+1, FALSE);
1432789Sahrens 	if (error == ENOSPC)
1433789Sahrens 		ztest_record_enospc("dmu_take_snapshot");
1434789Sahrens 	else if (error != 0 && error != EEXIST)
1435789Sahrens 		fatal(0, "dmu_take_snapshot() = %d", error);
1436789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
1437789Sahrens }
1438789Sahrens 
1439789Sahrens /*
14408779SMark.Musante@Sun.COM  * Verify dsl_dataset_promote handles EBUSY
14418779SMark.Musante@Sun.COM  */
14428779SMark.Musante@Sun.COM void
14438779SMark.Musante@Sun.COM ztest_dsl_dataset_promote_busy(ztest_args_t *za)
14448779SMark.Musante@Sun.COM {
14458779SMark.Musante@Sun.COM 	int error;
14468779SMark.Musante@Sun.COM 	objset_t *os = za->za_os;
14478779SMark.Musante@Sun.COM 	objset_t *clone;
14488779SMark.Musante@Sun.COM 	dsl_dataset_t *ds;
14498779SMark.Musante@Sun.COM 	char snap1name[100];
14508779SMark.Musante@Sun.COM 	char clone1name[100];
14518779SMark.Musante@Sun.COM 	char snap2name[100];
14528779SMark.Musante@Sun.COM 	char clone2name[100];
14538779SMark.Musante@Sun.COM 	char snap3name[100];
14548779SMark.Musante@Sun.COM 	char osname[MAXNAMELEN];
14558779SMark.Musante@Sun.COM 	uint64_t curval;
14568779SMark.Musante@Sun.COM 
14578837SMark.Musante@Sun.COM 	curval = za->za_instance;
14588779SMark.Musante@Sun.COM 
14598779SMark.Musante@Sun.COM 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
14608779SMark.Musante@Sun.COM 
14618779SMark.Musante@Sun.COM 	dmu_objset_name(os, osname);
14628837SMark.Musante@Sun.COM 	(void) snprintf(snap1name, 100, "%s@s1_%llu", osname, curval);
14638837SMark.Musante@Sun.COM 	(void) snprintf(clone1name, 100, "%s/c1_%llu", osname, curval);
14648837SMark.Musante@Sun.COM 	(void) snprintf(snap2name, 100, "%s@s2_%llu", clone1name, curval);
14658837SMark.Musante@Sun.COM 	(void) snprintf(clone2name, 100, "%s/c2_%llu", osname, curval);
14668837SMark.Musante@Sun.COM 	(void) snprintf(snap3name, 100, "%s@s3_%llu", clone1name, curval);
14678837SMark.Musante@Sun.COM 
14688837SMark.Musante@Sun.COM 	error = dmu_objset_destroy(clone2name);
14698837SMark.Musante@Sun.COM 	if (error != 0 && error != ENOENT)
14708837SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy() = %d", error);
14718837SMark.Musante@Sun.COM 	error = dmu_objset_destroy(snap3name);
14728837SMark.Musante@Sun.COM 	if (error != 0 && error != ENOENT)
14738837SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy() = %d", error);
14748837SMark.Musante@Sun.COM 	error = dmu_objset_destroy(snap2name);
14758837SMark.Musante@Sun.COM 	if (error != 0 && error != ENOENT)
14768837SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy() = %d", error);
14778837SMark.Musante@Sun.COM 	error = dmu_objset_destroy(clone1name);
14788837SMark.Musante@Sun.COM 	if (error != 0 && error != ENOENT)
14798837SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy() = %d", error);
14808837SMark.Musante@Sun.COM 	error = dmu_objset_destroy(snap1name);
14818837SMark.Musante@Sun.COM 	if (error != 0 && error != ENOENT)
14828837SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy() = %d", error);
14838779SMark.Musante@Sun.COM 
14848779SMark.Musante@Sun.COM 	error = dmu_objset_snapshot(osname, strchr(snap1name, '@')+1, FALSE);
1485*9229SGeorge.Wilson@Sun.COM 	if (error && error != EEXIST) {
1486*9229SGeorge.Wilson@Sun.COM 		if (error == ENOSPC) {
1487*9229SGeorge.Wilson@Sun.COM 			ztest_record_enospc("dmu_take_snapshot");
1488*9229SGeorge.Wilson@Sun.COM 			goto out;
1489*9229SGeorge.Wilson@Sun.COM 		}
1490*9229SGeorge.Wilson@Sun.COM 		fatal(0, "dmu_take_snapshot(%s) = %d", snap1name, error);
1491*9229SGeorge.Wilson@Sun.COM 	}
14928779SMark.Musante@Sun.COM 
14938779SMark.Musante@Sun.COM 	error = dmu_objset_open(snap1name, DMU_OST_OTHER,
14948779SMark.Musante@Sun.COM 	    DS_MODE_USER | DS_MODE_READONLY, &clone);
14958779SMark.Musante@Sun.COM 	if (error)
14968779SMark.Musante@Sun.COM 		fatal(0, "dmu_open_snapshot(%s) = %d", snap1name, error);
14978779SMark.Musante@Sun.COM 
14988779SMark.Musante@Sun.COM 	error = dmu_objset_create(clone1name, DMU_OST_OTHER, clone, 0,
14998779SMark.Musante@Sun.COM 	    NULL, NULL);
1500*9229SGeorge.Wilson@Sun.COM 	dmu_objset_close(clone);
1501*9229SGeorge.Wilson@Sun.COM 	if (error) {
1502*9229SGeorge.Wilson@Sun.COM 		if (error == ENOSPC) {
1503*9229SGeorge.Wilson@Sun.COM 			ztest_record_enospc("dmu_objset_create");
1504*9229SGeorge.Wilson@Sun.COM 			goto out;
1505*9229SGeorge.Wilson@Sun.COM 		}
15068779SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_create(%s) = %d", clone1name, error);
1507*9229SGeorge.Wilson@Sun.COM 	}
15088779SMark.Musante@Sun.COM 
15098779SMark.Musante@Sun.COM 	error = dmu_objset_snapshot(clone1name, strchr(snap2name, '@')+1,
15108779SMark.Musante@Sun.COM 	    FALSE);
1511*9229SGeorge.Wilson@Sun.COM 	if (error && error != EEXIST) {
1512*9229SGeorge.Wilson@Sun.COM 		if (error == ENOSPC) {
1513*9229SGeorge.Wilson@Sun.COM 			ztest_record_enospc("dmu_take_snapshot");
1514*9229SGeorge.Wilson@Sun.COM 			goto out;
1515*9229SGeorge.Wilson@Sun.COM 		}
1516*9229SGeorge.Wilson@Sun.COM 		fatal(0, "dmu_open_snapshot(%s) = %d", snap2name, error);
1517*9229SGeorge.Wilson@Sun.COM 	}
15188779SMark.Musante@Sun.COM 
15198779SMark.Musante@Sun.COM 	error = dmu_objset_snapshot(clone1name, strchr(snap3name, '@')+1,
15208779SMark.Musante@Sun.COM 	    FALSE);
1521*9229SGeorge.Wilson@Sun.COM 	if (error && error != EEXIST) {
1522*9229SGeorge.Wilson@Sun.COM 		if (error == ENOSPC) {
1523*9229SGeorge.Wilson@Sun.COM 			ztest_record_enospc("dmu_take_snapshot");
1524*9229SGeorge.Wilson@Sun.COM 			goto out;
1525*9229SGeorge.Wilson@Sun.COM 		}
1526*9229SGeorge.Wilson@Sun.COM 		fatal(0, "dmu_open_snapshot(%s) = %d", snap3name, error);
1527*9229SGeorge.Wilson@Sun.COM 	}
15288779SMark.Musante@Sun.COM 
15298779SMark.Musante@Sun.COM 	error = dmu_objset_open(snap3name, DMU_OST_OTHER,
15308779SMark.Musante@Sun.COM 	    DS_MODE_USER | DS_MODE_READONLY, &clone);
15318779SMark.Musante@Sun.COM 	if (error)
15328779SMark.Musante@Sun.COM 		fatal(0, "dmu_open_snapshot(%s) = %d", snap3name, error);
15338779SMark.Musante@Sun.COM 
15348779SMark.Musante@Sun.COM 	error = dmu_objset_create(clone2name, DMU_OST_OTHER, clone, 0,
15358779SMark.Musante@Sun.COM 	    NULL, NULL);
1536*9229SGeorge.Wilson@Sun.COM 	dmu_objset_close(clone);
1537*9229SGeorge.Wilson@Sun.COM 	if (error) {
1538*9229SGeorge.Wilson@Sun.COM 		if (error == ENOSPC) {
1539*9229SGeorge.Wilson@Sun.COM 			ztest_record_enospc("dmu_objset_create");
1540*9229SGeorge.Wilson@Sun.COM 			goto out;
1541*9229SGeorge.Wilson@Sun.COM 		}
15428779SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_create(%s) = %d", clone2name, error);
1543*9229SGeorge.Wilson@Sun.COM 	}
15448779SMark.Musante@Sun.COM 
15458837SMark.Musante@Sun.COM 	error = dsl_dataset_own(snap1name, DS_MODE_READONLY, FTAG, &ds);
15468779SMark.Musante@Sun.COM 	if (error)
15478779SMark.Musante@Sun.COM 		fatal(0, "dsl_dataset_own(%s) = %d", snap1name, error);
15488779SMark.Musante@Sun.COM 	error = dsl_dataset_promote(clone2name);
15498779SMark.Musante@Sun.COM 	if (error != EBUSY)
15508779SMark.Musante@Sun.COM 		fatal(0, "dsl_dataset_promote(%s), %d, not EBUSY", clone2name,
15518779SMark.Musante@Sun.COM 		    error);
15528779SMark.Musante@Sun.COM 	dsl_dataset_disown(ds, FTAG);
15538779SMark.Musante@Sun.COM 
1554*9229SGeorge.Wilson@Sun.COM out:
15558779SMark.Musante@Sun.COM 	error = dmu_objset_destroy(clone2name);
1556*9229SGeorge.Wilson@Sun.COM 	if (error && error != ENOENT)
15578779SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy(%s) = %d", clone2name, error);
15588779SMark.Musante@Sun.COM 	error = dmu_objset_destroy(snap3name);
1559*9229SGeorge.Wilson@Sun.COM 	if (error && error != ENOENT)
15608837SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy(%s) = %d", snap3name, error);
15618779SMark.Musante@Sun.COM 	error = dmu_objset_destroy(snap2name);
1562*9229SGeorge.Wilson@Sun.COM 	if (error && error != ENOENT)
15638779SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy(%s) = %d", snap2name, error);
15648779SMark.Musante@Sun.COM 	error = dmu_objset_destroy(clone1name);
1565*9229SGeorge.Wilson@Sun.COM 	if (error && error != ENOENT)
15668779SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy(%s) = %d", clone1name, error);
15678779SMark.Musante@Sun.COM 	error = dmu_objset_destroy(snap1name);
1568*9229SGeorge.Wilson@Sun.COM 	if (error && error != ENOENT)
15698779SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy(%s) = %d", snap1name, error);
15708779SMark.Musante@Sun.COM 
15718779SMark.Musante@Sun.COM 	(void) rw_unlock(&ztest_shared->zs_name_lock);
15728779SMark.Musante@Sun.COM }
15738779SMark.Musante@Sun.COM 
15748779SMark.Musante@Sun.COM /*
1575789Sahrens  * Verify that dmu_object_{alloc,free} work as expected.
1576789Sahrens  */
1577789Sahrens void
1578789Sahrens ztest_dmu_object_alloc_free(ztest_args_t *za)
1579789Sahrens {
1580789Sahrens 	objset_t *os = za->za_os;
1581789Sahrens 	dmu_buf_t *db;
1582789Sahrens 	dmu_tx_t *tx;
1583789Sahrens 	uint64_t batchobj, object, batchsize, endoff, temp;
1584789Sahrens 	int b, c, error, bonuslen;
15855530Sbonwick 	dmu_object_info_t *doi = &za->za_doi;
1586789Sahrens 	char osname[MAXNAMELEN];
1587789Sahrens 
1588789Sahrens 	dmu_objset_name(os, osname);
1589789Sahrens 
1590789Sahrens 	endoff = -8ULL;
1591789Sahrens 	batchsize = 2;
1592789Sahrens 
1593789Sahrens 	/*
1594789Sahrens 	 * Create a batch object if necessary, and record it in the directory.
1595789Sahrens 	 */
15967754SJeff.Bonwick@Sun.COM 	VERIFY3U(0, ==, dmu_read(os, ZTEST_DIROBJ, za->za_diroff,
15971544Seschrock 	    sizeof (uint64_t), &batchobj));
1598789Sahrens 	if (batchobj == 0) {
1599789Sahrens 		tx = dmu_tx_create(os);
1600789Sahrens 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff,
1601789Sahrens 		    sizeof (uint64_t));
1602789Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1603789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1604789Sahrens 		if (error) {
1605789Sahrens 			ztest_record_enospc("create a batch object");
1606789Sahrens 			dmu_tx_abort(tx);
1607789Sahrens 			return;
1608789Sahrens 		}
1609789Sahrens 		batchobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1610789Sahrens 		    DMU_OT_NONE, 0, tx);
1611789Sahrens 		ztest_set_random_blocksize(os, batchobj, tx);
1612789Sahrens 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff,
1613789Sahrens 		    sizeof (uint64_t), &batchobj, tx);
1614789Sahrens 		dmu_tx_commit(tx);
1615789Sahrens 	}
1616789Sahrens 
1617789Sahrens 	/*
1618789Sahrens 	 * Destroy the previous batch of objects.
1619789Sahrens 	 */
1620789Sahrens 	for (b = 0; b < batchsize; b++) {
16217046Sahrens 		VERIFY3U(0, ==, dmu_read(os, batchobj, b * sizeof (uint64_t),
16221544Seschrock 		    sizeof (uint64_t), &object));
1623789Sahrens 		if (object == 0)
1624789Sahrens 			continue;
1625789Sahrens 		/*
1626789Sahrens 		 * Read and validate contents.
1627789Sahrens 		 * We expect the nth byte of the bonus buffer to be n.
1628789Sahrens 		 */
16291544Seschrock 		VERIFY(0 == dmu_bonus_hold(os, object, FTAG, &db));
16305530Sbonwick 		za->za_dbuf = db;
16315530Sbonwick 
16325530Sbonwick 		dmu_object_info_from_db(db, doi);
16335530Sbonwick 		ASSERT(doi->doi_type == DMU_OT_UINT64_OTHER);
16345530Sbonwick 		ASSERT(doi->doi_bonus_type == DMU_OT_PLAIN_OTHER);
16355530Sbonwick 		ASSERT3S(doi->doi_physical_blks, >=, 0);
16365530Sbonwick 
16375530Sbonwick 		bonuslen = doi->doi_bonus_size;
1638789Sahrens 
1639789Sahrens 		for (c = 0; c < bonuslen; c++) {
1640789Sahrens 			if (((uint8_t *)db->db_data)[c] !=
1641789Sahrens 			    (uint8_t)(c + bonuslen)) {
1642789Sahrens 				fatal(0,
1643789Sahrens 				    "bad bonus: %s, obj %llu, off %d: %u != %u",
1644789Sahrens 				    osname, object, c,
1645789Sahrens 				    ((uint8_t *)db->db_data)[c],
1646789Sahrens 				    (uint8_t)(c + bonuslen));
1647789Sahrens 			}
1648789Sahrens 		}
1649789Sahrens 
16501544Seschrock 		dmu_buf_rele(db, FTAG);
16515530Sbonwick 		za->za_dbuf = NULL;
1652789Sahrens 
1653789Sahrens 		/*
1654789Sahrens 		 * We expect the word at endoff to be our object number.
1655789Sahrens 		 */
16561544Seschrock 		VERIFY(0 == dmu_read(os, object, endoff,
16571544Seschrock 		    sizeof (uint64_t), &temp));
1658789Sahrens 
1659789Sahrens 		if (temp != object) {
1660789Sahrens 			fatal(0, "bad data in %s, got %llu, expected %llu",
1661789Sahrens 			    osname, temp, object);
1662789Sahrens 		}
1663789Sahrens 
1664789Sahrens 		/*
1665789Sahrens 		 * Destroy old object and clear batch entry.
1666789Sahrens 		 */
1667789Sahrens 		tx = dmu_tx_create(os);
1668789Sahrens 		dmu_tx_hold_write(tx, batchobj,
1669789Sahrens 		    b * sizeof (uint64_t), sizeof (uint64_t));
1670789Sahrens 		dmu_tx_hold_free(tx, object, 0, DMU_OBJECT_END);
1671789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1672789Sahrens 		if (error) {
1673789Sahrens 			ztest_record_enospc("free object");
1674789Sahrens 			dmu_tx_abort(tx);
1675789Sahrens 			return;
1676789Sahrens 		}
1677789Sahrens 		error = dmu_object_free(os, object, tx);
1678789Sahrens 		if (error) {
1679789Sahrens 			fatal(0, "dmu_object_free('%s', %llu) = %d",
1680789Sahrens 			    osname, object, error);
1681789Sahrens 		}
1682789Sahrens 		object = 0;
1683789Sahrens 
1684789Sahrens 		dmu_object_set_checksum(os, batchobj,
1685789Sahrens 		    ztest_random_checksum(), tx);
1686789Sahrens 		dmu_object_set_compress(os, batchobj,
1687789Sahrens 		    ztest_random_compress(), tx);
1688789Sahrens 
1689789Sahrens 		dmu_write(os, batchobj, b * sizeof (uint64_t),
1690789Sahrens 		    sizeof (uint64_t), &object, tx);
1691789Sahrens 
1692789Sahrens 		dmu_tx_commit(tx);
1693789Sahrens 	}
1694789Sahrens 
1695789Sahrens 	/*
1696789Sahrens 	 * Before creating the new batch of objects, generate a bunch of churn.
1697789Sahrens 	 */
1698789Sahrens 	for (b = ztest_random(100); b > 0; b--) {
1699789Sahrens 		tx = dmu_tx_create(os);
1700789Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1701789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1702789Sahrens 		if (error) {
1703789Sahrens 			ztest_record_enospc("churn objects");
1704789Sahrens 			dmu_tx_abort(tx);
1705789Sahrens 			return;
1706789Sahrens 		}
1707789Sahrens 		object = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1708789Sahrens 		    DMU_OT_NONE, 0, tx);
1709789Sahrens 		ztest_set_random_blocksize(os, object, tx);
1710789Sahrens 		error = dmu_object_free(os, object, tx);
1711789Sahrens 		if (error) {
1712789Sahrens 			fatal(0, "dmu_object_free('%s', %llu) = %d",
1713789Sahrens 			    osname, object, error);
1714789Sahrens 		}
1715789Sahrens 		dmu_tx_commit(tx);
1716789Sahrens 	}
1717789Sahrens 
1718789Sahrens 	/*
1719789Sahrens 	 * Create a new batch of objects with randomly chosen
1720789Sahrens 	 * blocksizes and record them in the batch directory.
1721789Sahrens 	 */
1722789Sahrens 	for (b = 0; b < batchsize; b++) {
1723789Sahrens 		uint32_t va_blksize;
1724789Sahrens 		u_longlong_t va_nblocks;
1725789Sahrens 
1726789Sahrens 		tx = dmu_tx_create(os);
1727789Sahrens 		dmu_tx_hold_write(tx, batchobj, b * sizeof (uint64_t),
1728789Sahrens 		    sizeof (uint64_t));
1729789Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1730789Sahrens 		dmu_tx_hold_write(tx, DMU_NEW_OBJECT, endoff,
1731789Sahrens 		    sizeof (uint64_t));
1732789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1733789Sahrens 		if (error) {
1734789Sahrens 			ztest_record_enospc("create batchobj");
1735789Sahrens 			dmu_tx_abort(tx);
1736789Sahrens 			return;
1737789Sahrens 		}
1738789Sahrens 		bonuslen = (int)ztest_random(dmu_bonus_max()) + 1;
1739789Sahrens 
1740789Sahrens 		object = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1741789Sahrens 		    DMU_OT_PLAIN_OTHER, bonuslen, tx);
1742789Sahrens 
1743789Sahrens 		ztest_set_random_blocksize(os, object, tx);
1744789Sahrens 
1745789Sahrens 		dmu_object_set_checksum(os, object,
1746789Sahrens 		    ztest_random_checksum(), tx);
1747789Sahrens 		dmu_object_set_compress(os, object,
1748789Sahrens 		    ztest_random_compress(), tx);
1749789Sahrens 
1750789Sahrens 		dmu_write(os, batchobj, b * sizeof (uint64_t),
1751789Sahrens 		    sizeof (uint64_t), &object, tx);
1752789Sahrens 
1753789Sahrens 		/*
1754789Sahrens 		 * Write to both the bonus buffer and the regular data.
1755789Sahrens 		 */
17565530Sbonwick 		VERIFY(dmu_bonus_hold(os, object, FTAG, &db) == 0);
17575530Sbonwick 		za->za_dbuf = db;
17584944Smaybee 		ASSERT3U(bonuslen, <=, db->db_size);
1759789Sahrens 
1760789Sahrens 		dmu_object_size_from_db(db, &va_blksize, &va_nblocks);
1761789Sahrens 		ASSERT3S(va_nblocks, >=, 0);
1762789Sahrens 
1763789Sahrens 		dmu_buf_will_dirty(db, tx);
1764789Sahrens 
1765789Sahrens 		/*
1766789Sahrens 		 * See comments above regarding the contents of
1767789Sahrens 		 * the bonus buffer and the word at endoff.
1768789Sahrens 		 */
17694944Smaybee 		for (c = 0; c < bonuslen; c++)
1770789Sahrens 			((uint8_t *)db->db_data)[c] = (uint8_t)(c + bonuslen);
1771789Sahrens 
17721544Seschrock 		dmu_buf_rele(db, FTAG);
17735530Sbonwick 		za->za_dbuf = NULL;
1774789Sahrens 
1775789Sahrens 		/*
1776789Sahrens 		 * Write to a large offset to increase indirection.
1777789Sahrens 		 */
1778789Sahrens 		dmu_write(os, object, endoff, sizeof (uint64_t), &object, tx);
1779789Sahrens 
1780789Sahrens 		dmu_tx_commit(tx);
1781789Sahrens 	}
1782789Sahrens }
1783789Sahrens 
1784789Sahrens /*
1785789Sahrens  * Verify that dmu_{read,write} work as expected.
1786789Sahrens  */
1787789Sahrens typedef struct bufwad {
1788789Sahrens 	uint64_t	bw_index;
1789789Sahrens 	uint64_t	bw_txg;
1790789Sahrens 	uint64_t	bw_data;
1791789Sahrens } bufwad_t;
1792789Sahrens 
1793789Sahrens typedef struct dmu_read_write_dir {
1794789Sahrens 	uint64_t	dd_packobj;
1795789Sahrens 	uint64_t	dd_bigobj;
1796789Sahrens 	uint64_t	dd_chunk;
1797789Sahrens } dmu_read_write_dir_t;
1798789Sahrens 
1799789Sahrens void
1800789Sahrens ztest_dmu_read_write(ztest_args_t *za)
1801789Sahrens {
1802789Sahrens 	objset_t *os = za->za_os;
1803789Sahrens 	dmu_read_write_dir_t dd;
1804789Sahrens 	dmu_tx_t *tx;
1805789Sahrens 	int i, freeit, error;
1806789Sahrens 	uint64_t n, s, txg;
1807789Sahrens 	bufwad_t *packbuf, *bigbuf, *pack, *bigH, *bigT;
1808789Sahrens 	uint64_t packoff, packsize, bigoff, bigsize;
1809789Sahrens 	uint64_t regions = 997;
1810789Sahrens 	uint64_t stride = 123456789ULL;
1811789Sahrens 	uint64_t width = 40;
1812789Sahrens 	int free_percent = 5;
1813789Sahrens 
1814789Sahrens 	/*
1815789Sahrens 	 * This test uses two objects, packobj and bigobj, that are always
1816789Sahrens 	 * updated together (i.e. in the same tx) so that their contents are
1817789Sahrens 	 * in sync and can be compared.  Their contents relate to each other
1818789Sahrens 	 * in a simple way: packobj is a dense array of 'bufwad' structures,
1819789Sahrens 	 * while bigobj is a sparse array of the same bufwads.  Specifically,
1820789Sahrens 	 * for any index n, there are three bufwads that should be identical:
1821789Sahrens 	 *
1822789Sahrens 	 *	packobj, at offset n * sizeof (bufwad_t)
1823789Sahrens 	 *	bigobj, at the head of the nth chunk
1824789Sahrens 	 *	bigobj, at the tail of the nth chunk
1825789Sahrens 	 *
1826789Sahrens 	 * The chunk size is arbitrary. It doesn't have to be a power of two,
1827789Sahrens 	 * and it doesn't have any relation to the object blocksize.
1828789Sahrens 	 * The only requirement is that it can hold at least two bufwads.
1829789Sahrens 	 *
1830789Sahrens 	 * Normally, we write the bufwad to each of these locations.
1831789Sahrens 	 * However, free_percent of the time we instead write zeroes to
1832789Sahrens 	 * packobj and perform a dmu_free_range() on bigobj.  By comparing
1833789Sahrens 	 * bigobj to packobj, we can verify that the DMU is correctly
1834789Sahrens 	 * tracking which parts of an object are allocated and free,
1835789Sahrens 	 * and that the contents of the allocated blocks are correct.
1836789Sahrens 	 */
1837789Sahrens 
1838789Sahrens 	/*
1839789Sahrens 	 * Read the directory info.  If it's the first time, set things up.
1840789Sahrens 	 */
18411544Seschrock 	VERIFY(0 == dmu_read(os, ZTEST_DIROBJ, za->za_diroff,
18421544Seschrock 	    sizeof (dd), &dd));
1843789Sahrens 	if (dd.dd_chunk == 0) {
1844789Sahrens 		ASSERT(dd.dd_packobj == 0);
1845789Sahrens 		ASSERT(dd.dd_bigobj == 0);
1846789Sahrens 		tx = dmu_tx_create(os);
1847789Sahrens 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff, sizeof (dd));
1848789Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1849789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1850789Sahrens 		if (error) {
1851789Sahrens 			ztest_record_enospc("create r/w directory");
1852789Sahrens 			dmu_tx_abort(tx);
1853789Sahrens 			return;
1854789Sahrens 		}
1855789Sahrens 
1856789Sahrens 		dd.dd_packobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1857789Sahrens 		    DMU_OT_NONE, 0, tx);
1858789Sahrens 		dd.dd_bigobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1859789Sahrens 		    DMU_OT_NONE, 0, tx);
1860789Sahrens 		dd.dd_chunk = (1000 + ztest_random(1000)) * sizeof (uint64_t);
1861789Sahrens 
1862789Sahrens 		ztest_set_random_blocksize(os, dd.dd_packobj, tx);
1863789Sahrens 		ztest_set_random_blocksize(os, dd.dd_bigobj, tx);
1864789Sahrens 
1865789Sahrens 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff, sizeof (dd), &dd,
1866789Sahrens 		    tx);
1867789Sahrens 		dmu_tx_commit(tx);
1868789Sahrens 	}
1869789Sahrens 
1870789Sahrens 	/*
1871789Sahrens 	 * Prefetch a random chunk of the big object.
1872789Sahrens 	 * Our aim here is to get some async reads in flight
1873789Sahrens 	 * for blocks that we may free below; the DMU should
1874789Sahrens 	 * handle this race correctly.
1875789Sahrens 	 */
1876789Sahrens 	n = ztest_random(regions) * stride + ztest_random(width);
1877789Sahrens 	s = 1 + ztest_random(2 * width - 1);
1878789Sahrens 	dmu_prefetch(os, dd.dd_bigobj, n * dd.dd_chunk, s * dd.dd_chunk);
1879789Sahrens 
1880789Sahrens 	/*
1881789Sahrens 	 * Pick a random index and compute the offsets into packobj and bigobj.
1882789Sahrens 	 */
1883789Sahrens 	n = ztest_random(regions) * stride + ztest_random(width);
1884789Sahrens 	s = 1 + ztest_random(width - 1);
1885789Sahrens 
1886789Sahrens 	packoff = n * sizeof (bufwad_t);
1887789Sahrens 	packsize = s * sizeof (bufwad_t);
1888789Sahrens 
1889789Sahrens 	bigoff = n * dd.dd_chunk;
1890789Sahrens 	bigsize = s * dd.dd_chunk;
1891789Sahrens 
1892789Sahrens 	packbuf = umem_alloc(packsize, UMEM_NOFAIL);
1893789Sahrens 	bigbuf = umem_alloc(bigsize, UMEM_NOFAIL);
1894789Sahrens 
1895789Sahrens 	/*
1896789Sahrens 	 * free_percent of the time, free a range of bigobj rather than
1897789Sahrens 	 * overwriting it.
1898789Sahrens 	 */
1899789Sahrens 	freeit = (ztest_random(100) < free_percent);
1900789Sahrens 
1901789Sahrens 	/*
1902789Sahrens 	 * Read the current contents of our objects.
1903789Sahrens 	 */
19041544Seschrock 	error = dmu_read(os, dd.dd_packobj, packoff, packsize, packbuf);
19051544Seschrock 	ASSERT3U(error, ==, 0);
19061544Seschrock 	error = dmu_read(os, dd.dd_bigobj, bigoff, bigsize, bigbuf);
19071544Seschrock 	ASSERT3U(error, ==, 0);
1908789Sahrens 
1909789Sahrens 	/*
1910789Sahrens 	 * Get a tx for the mods to both packobj and bigobj.
1911789Sahrens 	 */
1912789Sahrens 	tx = dmu_tx_create(os);
1913789Sahrens 
1914789Sahrens 	dmu_tx_hold_write(tx, dd.dd_packobj, packoff, packsize);
1915789Sahrens 
1916789Sahrens 	if (freeit)
1917789Sahrens 		dmu_tx_hold_free(tx, dd.dd_bigobj, bigoff, bigsize);
1918789Sahrens 	else
1919789Sahrens 		dmu_tx_hold_write(tx, dd.dd_bigobj, bigoff, bigsize);
1920789Sahrens 
1921789Sahrens 	error = dmu_tx_assign(tx, TXG_WAIT);
1922789Sahrens 
1923789Sahrens 	if (error) {
1924789Sahrens 		ztest_record_enospc("dmu r/w range");
1925789Sahrens 		dmu_tx_abort(tx);
1926789Sahrens 		umem_free(packbuf, packsize);
1927789Sahrens 		umem_free(bigbuf, bigsize);
1928789Sahrens 		return;
1929789Sahrens 	}
1930789Sahrens 
1931789Sahrens 	txg = dmu_tx_get_txg(tx);
1932789Sahrens 
1933789Sahrens 	/*
1934789Sahrens 	 * For each index from n to n + s, verify that the existing bufwad
1935789Sahrens 	 * in packobj matches the bufwads at the head and tail of the
1936789Sahrens 	 * corresponding chunk in bigobj.  Then update all three bufwads
1937789Sahrens 	 * with the new values we want to write out.
1938789Sahrens 	 */
1939789Sahrens 	for (i = 0; i < s; i++) {
1940789Sahrens 		/* LINTED */
1941789Sahrens 		pack = (bufwad_t *)((char *)packbuf + i * sizeof (bufwad_t));
1942789Sahrens 		/* LINTED */
1943789Sahrens 		bigH = (bufwad_t *)((char *)bigbuf + i * dd.dd_chunk);
1944789Sahrens 		/* LINTED */
1945789Sahrens 		bigT = (bufwad_t *)((char *)bigH + dd.dd_chunk) - 1;
1946789Sahrens 
1947789Sahrens 		ASSERT((uintptr_t)bigH - (uintptr_t)bigbuf < bigsize);
1948789Sahrens 		ASSERT((uintptr_t)bigT - (uintptr_t)bigbuf < bigsize);
1949789Sahrens 
1950789Sahrens 		if (pack->bw_txg > txg)
1951789Sahrens 			fatal(0, "future leak: got %llx, open txg is %llx",
1952789Sahrens 			    pack->bw_txg, txg);
1953789Sahrens 
1954789Sahrens 		if (pack->bw_data != 0 && pack->bw_index != n + i)
1955789Sahrens 			fatal(0, "wrong index: got %llx, wanted %llx+%llx",
1956789Sahrens 			    pack->bw_index, n, i);
1957789Sahrens 
1958789Sahrens 		if (bcmp(pack, bigH, sizeof (bufwad_t)) != 0)
1959789Sahrens 			fatal(0, "pack/bigH mismatch in %p/%p", pack, bigH);
1960789Sahrens 
1961789Sahrens 		if (bcmp(pack, bigT, sizeof (bufwad_t)) != 0)
1962789Sahrens 			fatal(0, "pack/bigT mismatch in %p/%p", pack, bigT);
1963789Sahrens 
1964789Sahrens 		if (freeit) {
1965789Sahrens 			bzero(pack, sizeof (bufwad_t));
1966789Sahrens 		} else {
1967789Sahrens 			pack->bw_index = n + i;
1968789Sahrens 			pack->bw_txg = txg;
1969789Sahrens 			pack->bw_data = 1 + ztest_random(-2ULL);
1970789Sahrens 		}
1971789Sahrens 		*bigH = *pack;
1972789Sahrens 		*bigT = *pack;
1973789Sahrens 	}
1974789Sahrens 
1975789Sahrens 	/*
1976789Sahrens 	 * We've verified all the old bufwads, and made new ones.
1977789Sahrens 	 * Now write them out.
1978789Sahrens 	 */
1979789Sahrens 	dmu_write(os, dd.dd_packobj, packoff, packsize, packbuf, tx);
1980789Sahrens 
1981789Sahrens 	if (freeit) {
1982789Sahrens 		if (zopt_verbose >= 6) {
1983789Sahrens 			(void) printf("freeing offset %llx size %llx"
1984789Sahrens 			    " txg %llx\n",
1985789Sahrens 			    (u_longlong_t)bigoff,
1986789Sahrens 			    (u_longlong_t)bigsize,
1987789Sahrens 			    (u_longlong_t)txg);
1988789Sahrens 		}
19891544Seschrock 		VERIFY(0 == dmu_free_range(os, dd.dd_bigobj, bigoff,
19901544Seschrock 		    bigsize, tx));
1991789Sahrens 	} else {
1992789Sahrens 		if (zopt_verbose >= 6) {
1993789Sahrens 			(void) printf("writing offset %llx size %llx"
1994789Sahrens 			    " txg %llx\n",
1995789Sahrens 			    (u_longlong_t)bigoff,
1996789Sahrens 			    (u_longlong_t)bigsize,
1997789Sahrens 			    (u_longlong_t)txg);
1998789Sahrens 		}
1999789Sahrens 		dmu_write(os, dd.dd_bigobj, bigoff, bigsize, bigbuf, tx);
2000789Sahrens 	}
2001789Sahrens 
2002789Sahrens 	dmu_tx_commit(tx);
2003789Sahrens 
2004789Sahrens 	/*
2005789Sahrens 	 * Sanity check the stuff we just wrote.
2006789Sahrens 	 */
2007789Sahrens 	{
2008789Sahrens 		void *packcheck = umem_alloc(packsize, UMEM_NOFAIL);
2009789Sahrens 		void *bigcheck = umem_alloc(bigsize, UMEM_NOFAIL);
2010789Sahrens 
20111544Seschrock 		VERIFY(0 == dmu_read(os, dd.dd_packobj, packoff,
20121544Seschrock 		    packsize, packcheck));
20131544Seschrock 		VERIFY(0 == dmu_read(os, dd.dd_bigobj, bigoff,
20141544Seschrock 		    bigsize, bigcheck));
2015789Sahrens 
2016789Sahrens 		ASSERT(bcmp(packbuf, packcheck, packsize) == 0);
2017789Sahrens 		ASSERT(bcmp(bigbuf, bigcheck, bigsize) == 0);
2018789Sahrens 
2019789Sahrens 		umem_free(packcheck, packsize);
2020789Sahrens 		umem_free(bigcheck, bigsize);
2021789Sahrens 	}
2022789Sahrens 
2023789Sahrens 	umem_free(packbuf, packsize);
2024789Sahrens 	umem_free(bigbuf, bigsize);
2025789Sahrens }
2026789Sahrens 
2027789Sahrens void
20285530Sbonwick ztest_dmu_check_future_leak(ztest_args_t *za)
20293711Smaybee {
20305530Sbonwick 	objset_t *os = za->za_os;
20313711Smaybee 	dmu_buf_t *db;
20325530Sbonwick 	ztest_block_tag_t *bt;
20335530Sbonwick 	dmu_object_info_t *doi = &za->za_doi;
20343711Smaybee 
20353711Smaybee 	/*
20363711Smaybee 	 * Make sure that, if there is a write record in the bonus buffer
20373711Smaybee 	 * of the ZTEST_DIROBJ, that the txg for this record is <= the
20383711Smaybee 	 * last synced txg of the pool.
20393711Smaybee 	 */
20405530Sbonwick 	VERIFY(dmu_bonus_hold(os, ZTEST_DIROBJ, FTAG, &db) == 0);
20415530Sbonwick 	za->za_dbuf = db;
20425530Sbonwick 	VERIFY(dmu_object_info(os, ZTEST_DIROBJ, doi) == 0);
20435530Sbonwick 	ASSERT3U(doi->doi_bonus_size, >=, sizeof (*bt));
20445530Sbonwick 	ASSERT3U(doi->doi_bonus_size, <=, db->db_size);
20455530Sbonwick 	ASSERT3U(doi->doi_bonus_size % sizeof (*bt), ==, 0);
20465530Sbonwick 	bt = (void *)((char *)db->db_data + doi->doi_bonus_size - sizeof (*bt));
20475530Sbonwick 	if (bt->bt_objset != 0) {
20485530Sbonwick 		ASSERT3U(bt->bt_objset, ==, dmu_objset_id(os));
20495530Sbonwick 		ASSERT3U(bt->bt_object, ==, ZTEST_DIROBJ);
20505530Sbonwick 		ASSERT3U(bt->bt_offset, ==, -1ULL);
20515530Sbonwick 		ASSERT3U(bt->bt_txg, <, spa_first_txg(za->za_spa));
20523711Smaybee 	}
20533711Smaybee 	dmu_buf_rele(db, FTAG);
20545530Sbonwick 	za->za_dbuf = NULL;
20553711Smaybee }
20563711Smaybee 
20573711Smaybee void
2058789Sahrens ztest_dmu_write_parallel(ztest_args_t *za)
2059789Sahrens {
2060789Sahrens 	objset_t *os = za->za_os;
20615530Sbonwick 	ztest_block_tag_t *rbt = &za->za_rbt;
20625530Sbonwick 	ztest_block_tag_t *wbt = &za->za_wbt;
20635530Sbonwick 	const size_t btsize = sizeof (ztest_block_tag_t);
2064789Sahrens 	dmu_buf_t *db;
20655530Sbonwick 	int b, error;
20665530Sbonwick 	int bs = ZTEST_DIROBJ_BLOCKSIZE;
20675530Sbonwick 	int do_free = 0;
20687762SJeff.Bonwick@Sun.COM 	uint64_t off, txg, txg_how;
2069789Sahrens 	mutex_t *lp;
2070789Sahrens 	char osname[MAXNAMELEN];
2071789Sahrens 	char iobuf[SPA_MAXBLOCKSIZE];
20725530Sbonwick 	blkptr_t blk = { 0 };
20735530Sbonwick 	uint64_t blkoff;
20745530Sbonwick 	zbookmark_t zb;
20755530Sbonwick 	dmu_tx_t *tx = dmu_tx_create(os);
2076789Sahrens 
2077789Sahrens 	dmu_objset_name(os, osname);
2078789Sahrens 
2079789Sahrens 	/*
2080789Sahrens 	 * Have multiple threads write to large offsets in ZTEST_DIROBJ
2081789Sahrens 	 * to verify that having multiple threads writing to the same object
2082789Sahrens 	 * in parallel doesn't cause any trouble.
2083789Sahrens 	 */
20845530Sbonwick 	if (ztest_random(4) == 0) {
20855530Sbonwick 		/*
20865530Sbonwick 		 * Do the bonus buffer instead of a regular block.
20875530Sbonwick 		 * We need a lock to serialize resize vs. others,
20885530Sbonwick 		 * so we hash on the objset ID.
20895530Sbonwick 		 */
20905530Sbonwick 		b = dmu_objset_id(os) % ZTEST_SYNC_LOCKS;
20915530Sbonwick 		off = -1ULL;
20925530Sbonwick 		dmu_tx_hold_bonus(tx, ZTEST_DIROBJ);
20935530Sbonwick 	} else {
2094789Sahrens 		b = ztest_random(ZTEST_SYNC_LOCKS);
20955530Sbonwick 		off = za->za_diroff_shared + (b << SPA_MAXBLOCKSHIFT);
2096789Sahrens 		if (ztest_random(4) == 0) {
20975530Sbonwick 			do_free = 1;
2098789Sahrens 			dmu_tx_hold_free(tx, ZTEST_DIROBJ, off, bs);
20995530Sbonwick 		} else {
2100789Sahrens 			dmu_tx_hold_write(tx, ZTEST_DIROBJ, off, bs);
2101789Sahrens 		}
2102789Sahrens 	}
21035530Sbonwick 
21045530Sbonwick 	txg_how = ztest_random(2) == 0 ? TXG_WAIT : TXG_NOWAIT;
21055530Sbonwick 	error = dmu_tx_assign(tx, txg_how);
21065530Sbonwick 	if (error) {
21075530Sbonwick 		if (error == ERESTART) {
21085530Sbonwick 			ASSERT(txg_how == TXG_NOWAIT);
21095530Sbonwick 			dmu_tx_wait(tx);
21105530Sbonwick 		} else {
21115530Sbonwick 			ztest_record_enospc("dmu write parallel");
21125530Sbonwick 		}
21135530Sbonwick 		dmu_tx_abort(tx);
21145530Sbonwick 		return;
21155530Sbonwick 	}
21167762SJeff.Bonwick@Sun.COM 	txg = dmu_tx_get_txg(tx);
21175530Sbonwick 
21185530Sbonwick 	lp = &ztest_shared->zs_sync_lock[b];
21195530Sbonwick 	(void) mutex_lock(lp);
21205530Sbonwick 
21215530Sbonwick 	wbt->bt_objset = dmu_objset_id(os);
21225530Sbonwick 	wbt->bt_object = ZTEST_DIROBJ;
21235530Sbonwick 	wbt->bt_offset = off;
21247762SJeff.Bonwick@Sun.COM 	wbt->bt_txg = txg;
21255530Sbonwick 	wbt->bt_thread = za->za_instance;
21265530Sbonwick 	wbt->bt_seq = ztest_shared->zs_seq[b]++;	/* protected by lp */
21275530Sbonwick 
21287762SJeff.Bonwick@Sun.COM 	/*
21297762SJeff.Bonwick@Sun.COM 	 * Occasionally, write an all-zero block to test the behavior
21307762SJeff.Bonwick@Sun.COM 	 * of blocks that compress into holes.
21317762SJeff.Bonwick@Sun.COM 	 */
21327762SJeff.Bonwick@Sun.COM 	if (off != -1ULL && ztest_random(8) == 0)
21337762SJeff.Bonwick@Sun.COM 		bzero(wbt, btsize);
21347762SJeff.Bonwick@Sun.COM 
21355530Sbonwick 	if (off == -1ULL) {
21365530Sbonwick 		dmu_object_info_t *doi = &za->za_doi;
21375530Sbonwick 		char *dboff;
21385530Sbonwick 
21395530Sbonwick 		VERIFY(dmu_bonus_hold(os, ZTEST_DIROBJ, FTAG, &db) == 0);
21405530Sbonwick 		za->za_dbuf = db;
21415530Sbonwick 		dmu_object_info_from_db(db, doi);
21425530Sbonwick 		ASSERT3U(doi->doi_bonus_size, <=, db->db_size);
21435530Sbonwick 		ASSERT3U(doi->doi_bonus_size, >=, btsize);
21445530Sbonwick 		ASSERT3U(doi->doi_bonus_size % btsize, ==, 0);
21455530Sbonwick 		dboff = (char *)db->db_data + doi->doi_bonus_size - btsize;
21465530Sbonwick 		bcopy(dboff, rbt, btsize);
21475530Sbonwick 		if (rbt->bt_objset != 0) {
21485530Sbonwick 			ASSERT3U(rbt->bt_objset, ==, wbt->bt_objset);
21495530Sbonwick 			ASSERT3U(rbt->bt_object, ==, wbt->bt_object);
21505530Sbonwick 			ASSERT3U(rbt->bt_offset, ==, wbt->bt_offset);
21515530Sbonwick 			ASSERT3U(rbt->bt_txg, <=, wbt->bt_txg);
21525530Sbonwick 		}
21535530Sbonwick 		if (ztest_random(10) == 0) {
21545530Sbonwick 			int newsize = (ztest_random(db->db_size /
21555530Sbonwick 			    btsize) + 1) * btsize;
21565530Sbonwick 
21575530Sbonwick 			ASSERT3U(newsize, >=, btsize);
21585530Sbonwick 			ASSERT3U(newsize, <=, db->db_size);
21595530Sbonwick 			VERIFY3U(dmu_set_bonus(db, newsize, tx), ==, 0);
21605530Sbonwick 			dboff = (char *)db->db_data + newsize - btsize;
21615530Sbonwick 		}
21625530Sbonwick 		dmu_buf_will_dirty(db, tx);
21635530Sbonwick 		bcopy(wbt, dboff, btsize);
21645530Sbonwick 		dmu_buf_rele(db, FTAG);
21655530Sbonwick 		za->za_dbuf = NULL;
21665530Sbonwick 	} else if (do_free) {
21675530Sbonwick 		VERIFY(dmu_free_range(os, ZTEST_DIROBJ, off, bs, tx) == 0);
21685530Sbonwick 	} else {
21695530Sbonwick 		dmu_write(os, ZTEST_DIROBJ, off, btsize, wbt, tx);
21705530Sbonwick 	}
21715530Sbonwick 
21725530Sbonwick 	(void) mutex_unlock(lp);
21735530Sbonwick 
21745530Sbonwick 	if (ztest_random(1000) == 0)
21755530Sbonwick 		(void) poll(NULL, 0, 1); /* open dn_notxholds window */
21765530Sbonwick 
21775530Sbonwick 	dmu_tx_commit(tx);
21785530Sbonwick 
21795530Sbonwick 	if (ztest_random(10000) == 0)
21807762SJeff.Bonwick@Sun.COM 		txg_wait_synced(dmu_objset_pool(os), txg);
21815530Sbonwick 
21827754SJeff.Bonwick@Sun.COM 	if (off == -1ULL || do_free)
21835530Sbonwick 		return;
21845530Sbonwick 
21855530Sbonwick 	if (ztest_random(2) != 0)
21865530Sbonwick 		return;
21875530Sbonwick 
21885530Sbonwick 	/*
21895530Sbonwick 	 * dmu_sync() the block we just wrote.
21905530Sbonwick 	 */
21915530Sbonwick 	(void) mutex_lock(lp);
21925530Sbonwick 
21935530Sbonwick 	blkoff = P2ALIGN_TYPED(off, bs, uint64_t);
21945530Sbonwick 	error = dmu_buf_hold(os, ZTEST_DIROBJ, blkoff, FTAG, &db);
21955530Sbonwick 	za->za_dbuf = db;
21965530Sbonwick 	if (error) {
21975530Sbonwick 		(void) mutex_unlock(lp);
21985530Sbonwick 		return;
21995530Sbonwick 	}
22005530Sbonwick 	blkoff = off - blkoff;
22017762SJeff.Bonwick@Sun.COM 	error = dmu_sync(NULL, db, &blk, txg, NULL, NULL);
22025530Sbonwick 	dmu_buf_rele(db, FTAG);
22035530Sbonwick 	za->za_dbuf = NULL;
22045530Sbonwick 
22055530Sbonwick 	(void) mutex_unlock(lp);
22065530Sbonwick 
22078241SJeff.Bonwick@Sun.COM 	if (error)
22085530Sbonwick 		return;
22095530Sbonwick 
22105530Sbonwick 	if (blk.blk_birth == 0)		/* concurrent free */
22115530Sbonwick 		return;
22125530Sbonwick 
22135530Sbonwick 	txg_suspend(dmu_objset_pool(os));
22145530Sbonwick 
22155530Sbonwick 	ASSERT(blk.blk_fill == 1);
22165530Sbonwick 	ASSERT3U(BP_GET_TYPE(&blk), ==, DMU_OT_UINT64_OTHER);
22175530Sbonwick 	ASSERT3U(BP_GET_LEVEL(&blk), ==, 0);
22185530Sbonwick 	ASSERT3U(BP_GET_LSIZE(&blk), ==, bs);
22195530Sbonwick 
22205530Sbonwick 	/*
22215530Sbonwick 	 * Read the block that dmu_sync() returned to make sure its contents
22225530Sbonwick 	 * match what we wrote.  We do this while still txg_suspend()ed
22235530Sbonwick 	 * to ensure that the block can't be reused before we read it.
22245530Sbonwick 	 */
22255530Sbonwick 	zb.zb_objset = dmu_objset_id(os);
22265530Sbonwick 	zb.zb_object = ZTEST_DIROBJ;
22275530Sbonwick 	zb.zb_level = 0;
22285530Sbonwick 	zb.zb_blkid = off / bs;
22295530Sbonwick 	error = zio_wait(zio_read(NULL, za->za_spa, &blk, iobuf, bs,
22305530Sbonwick 	    NULL, NULL, ZIO_PRIORITY_SYNC_READ, ZIO_FLAG_MUSTSUCCEED, &zb));
22315530Sbonwick 	ASSERT3U(error, ==, 0);
22325530Sbonwick 
22335530Sbonwick 	txg_resume(dmu_objset_pool(os));
22345530Sbonwick 
22355530Sbonwick 	bcopy(&iobuf[blkoff], rbt, btsize);
22365530Sbonwick 
22375530Sbonwick 	if (rbt->bt_objset == 0)		/* concurrent free */
22385530Sbonwick 		return;
22395530Sbonwick 
22407762SJeff.Bonwick@Sun.COM 	if (wbt->bt_objset == 0)		/* all-zero overwrite */
22417762SJeff.Bonwick@Sun.COM 		return;
22427762SJeff.Bonwick@Sun.COM 
22435530Sbonwick 	ASSERT3U(rbt->bt_objset, ==, wbt->bt_objset);
22445530Sbonwick 	ASSERT3U(rbt->bt_object, ==, wbt->bt_object);
22455530Sbonwick 	ASSERT3U(rbt->bt_offset, ==, wbt->bt_offset);
22465530Sbonwick 
22475530Sbonwick 	/*
22485530Sbonwick 	 * The semantic of dmu_sync() is that we always push the most recent
22495530Sbonwick 	 * version of the data, so in the face of concurrent updates we may
22505530Sbonwick 	 * see a newer version of the block.  That's OK.
22515530Sbonwick 	 */
22525530Sbonwick 	ASSERT3U(rbt->bt_txg, >=, wbt->bt_txg);
22535530Sbonwick 	if (rbt->bt_thread == wbt->bt_thread)
22545530Sbonwick 		ASSERT3U(rbt->bt_seq, ==, wbt->bt_seq);
22555530Sbonwick 	else
22565530Sbonwick 		ASSERT3U(rbt->bt_seq, >, wbt->bt_seq);
2257789Sahrens }
2258789Sahrens 
2259789Sahrens /*
2260789Sahrens  * Verify that zap_{create,destroy,add,remove,update} work as expected.
2261789Sahrens  */
2262789Sahrens #define	ZTEST_ZAP_MIN_INTS	1
2263789Sahrens #define	ZTEST_ZAP_MAX_INTS	4
2264789Sahrens #define	ZTEST_ZAP_MAX_PROPS	1000
2265789Sahrens 
2266789Sahrens void
2267789Sahrens ztest_zap(ztest_args_t *za)
2268789Sahrens {
2269789Sahrens 	objset_t *os = za->za_os;
2270789Sahrens 	uint64_t object;
2271789Sahrens 	uint64_t txg, last_txg;
2272789Sahrens 	uint64_t value[ZTEST_ZAP_MAX_INTS];
2273789Sahrens 	uint64_t zl_ints, zl_intsize, prop;
2274789Sahrens 	int i, ints;
2275789Sahrens 	dmu_tx_t *tx;
2276789Sahrens 	char propname[100], txgname[100];
2277789Sahrens 	int error;
2278789Sahrens 	char osname[MAXNAMELEN];
2279789Sahrens 	char *hc[2] = { "s.acl.h", ".s.open.h.hyLZlg" };
2280789Sahrens 
2281789Sahrens 	dmu_objset_name(os, osname);
2282789Sahrens 
2283789Sahrens 	/*
2284789Sahrens 	 * Create a new object if necessary, and record it in the directory.
2285789Sahrens 	 */
22861544Seschrock 	VERIFY(0 == dmu_read(os, ZTEST_DIROBJ, za->za_diroff,
22871544Seschrock 	    sizeof (uint64_t), &object));
2288789Sahrens 
2289789Sahrens 	if (object == 0) {
2290789Sahrens 		tx = dmu_tx_create(os);
2291789Sahrens 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff,
2292789Sahrens 		    sizeof (uint64_t));
22931544Seschrock 		dmu_tx_hold_zap(tx, DMU_NEW_OBJECT, TRUE, NULL);
2294789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
2295789Sahrens 		if (error) {
2296789Sahrens 			ztest_record_enospc("create zap test obj");
2297789Sahrens 			dmu_tx_abort(tx);
2298789Sahrens 			return;
2299789Sahrens 		}
2300789Sahrens 		object = zap_create(os, DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx);
2301789Sahrens 		if (error) {
2302789Sahrens 			fatal(0, "zap_create('%s', %llu) = %d",
2303789Sahrens 			    osname, object, error);
2304789Sahrens 		}
2305789Sahrens 		ASSERT(object != 0);
2306789Sahrens 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff,
2307789Sahrens 		    sizeof (uint64_t), &object, tx);
2308789Sahrens 		/*
2309789Sahrens 		 * Generate a known hash collision, and verify that
2310789Sahrens 		 * we can lookup and remove both entries.
2311789Sahrens 		 */
2312789Sahrens 		for (i = 0; i < 2; i++) {
2313789Sahrens 			value[i] = i;
2314789Sahrens 			error = zap_add(os, object, hc[i], sizeof (uint64_t),
2315789Sahrens 			    1, &value[i], tx);
2316789Sahrens 			ASSERT3U(error, ==, 0);
2317789Sahrens 		}
2318789Sahrens 		for (i = 0; i < 2; i++) {
2319789Sahrens 			error = zap_add(os, object, hc[i], sizeof (uint64_t),
2320789Sahrens 			    1, &value[i], tx);
2321789Sahrens 			ASSERT3U(error, ==, EEXIST);
2322789Sahrens 			error = zap_length(os, object, hc[i],
2323789Sahrens 			    &zl_intsize, &zl_ints);
2324789Sahrens 			ASSERT3U(error, ==, 0);
2325789Sahrens 			ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
2326789Sahrens 			ASSERT3U(zl_ints, ==, 1);
2327789Sahrens 		}
2328789Sahrens 		for (i = 0; i < 2; i++) {
2329789Sahrens 			error = zap_remove(os, object, hc[i], tx);
2330789Sahrens 			ASSERT3U(error, ==, 0);
2331789Sahrens 		}
2332789Sahrens 
2333789Sahrens 		dmu_tx_commit(tx);
2334789Sahrens 	}
2335789Sahrens 
2336789Sahrens 	ints = MAX(ZTEST_ZAP_MIN_INTS, object % ZTEST_ZAP_MAX_INTS);
2337789Sahrens 
23385530Sbonwick 	prop = ztest_random(ZTEST_ZAP_MAX_PROPS);
23395530Sbonwick 	(void) sprintf(propname, "prop_%llu", (u_longlong_t)prop);
23405530Sbonwick 	(void) sprintf(txgname, "txg_%llu", (u_longlong_t)prop);
23415530Sbonwick 	bzero(value, sizeof (value));
23425530Sbonwick 	last_txg = 0;
23435530Sbonwick 
23445530Sbonwick 	/*
23455530Sbonwick 	 * If these zap entries already exist, validate their contents.
23465530Sbonwick 	 */
23475530Sbonwick 	error = zap_length(os, object, txgname, &zl_intsize, &zl_ints);
23485530Sbonwick 	if (error == 0) {
23495530Sbonwick 		ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
23505530Sbonwick 		ASSERT3U(zl_ints, ==, 1);
23515530Sbonwick 
23525530Sbonwick 		VERIFY(zap_lookup(os, object, txgname, zl_intsize,
23535530Sbonwick 		    zl_ints, &last_txg) == 0);
23545530Sbonwick 
23555530Sbonwick 		VERIFY(zap_length(os, object, propname, &zl_intsize,
23565530Sbonwick 		    &zl_ints) == 0);
23575530Sbonwick 
23585530Sbonwick 		ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
23595530Sbonwick 		ASSERT3U(zl_ints, ==, ints);
23605530Sbonwick 
23615530Sbonwick 		VERIFY(zap_lookup(os, object, propname, zl_intsize,
23625530Sbonwick 		    zl_ints, value) == 0);
23635530Sbonwick 
23645530Sbonwick 		for (i = 0; i < ints; i++) {
23655530Sbonwick 			ASSERT3U(value[i], ==, last_txg + object + i);
2366789Sahrens 		}
23675530Sbonwick 	} else {
23685530Sbonwick 		ASSERT3U(error, ==, ENOENT);
23695530Sbonwick 	}
23705530Sbonwick 
23715530Sbonwick 	/*
23725530Sbonwick 	 * Atomically update two entries in our zap object.
23735530Sbonwick 	 * The first is named txg_%llu, and contains the txg
23745530Sbonwick 	 * in which the property was last updated.  The second
23755530Sbonwick 	 * is named prop_%llu, and the nth element of its value
23765530Sbonwick 	 * should be txg + object + n.
23775530Sbonwick 	 */
23785530Sbonwick 	tx = dmu_tx_create(os);
23795530Sbonwick 	dmu_tx_hold_zap(tx, object, TRUE, NULL);
23805530Sbonwick 	error = dmu_tx_assign(tx, TXG_WAIT);
23815530Sbonwick 	if (error) {
23825530Sbonwick 		ztest_record_enospc("create zap entry");
23835530Sbonwick 		dmu_tx_abort(tx);
23845530Sbonwick 		return;
2385789Sahrens 	}
23865530Sbonwick 	txg = dmu_tx_get_txg(tx);
23875530Sbonwick 
23885530Sbonwick 	if (last_txg > txg)
23895530Sbonwick 		fatal(0, "zap future leak: old %llu new %llu", last_txg, txg);
23905530Sbonwick 
23915530Sbonwick 	for (i = 0; i < ints; i++)
23925530Sbonwick 		value[i] = txg + object + i;
23935530Sbonwick 
23945530Sbonwick 	error = zap_update(os, object, txgname, sizeof (uint64_t), 1, &txg, tx);
23955530Sbonwick 	if (error)
23965530Sbonwick 		fatal(0, "zap_update('%s', %llu, '%s') = %d",
23975530Sbonwick 		    osname, object, txgname, error);
23985530Sbonwick 
23995530Sbonwick 	error = zap_update(os, object, propname, sizeof (uint64_t),
24005530Sbonwick 	    ints, value, tx);
24015530Sbonwick 	if (error)
24025530Sbonwick 		fatal(0, "zap_update('%s', %llu, '%s') = %d",
24035530Sbonwick 		    osname, object, propname, error);
24045530Sbonwick 
24055530Sbonwick 	dmu_tx_commit(tx);
24065530Sbonwick 
24075530Sbonwick 	/*
24085530Sbonwick 	 * Remove a random pair of entries.
24095530Sbonwick 	 */
24105530Sbonwick 	prop = ztest_random(ZTEST_ZAP_MAX_PROPS);
24115530Sbonwick 	(void) sprintf(propname, "prop_%llu", (u_longlong_t)prop);
24125530Sbonwick 	(void) sprintf(txgname, "txg_%llu", (u_longlong_t)prop);
24135530Sbonwick 
24145530Sbonwick 	error = zap_length(os, object, txgname, &zl_intsize, &zl_ints);
24155530Sbonwick 
24165530Sbonwick 	if (error == ENOENT)
24175530Sbonwick 		return;
24185530Sbonwick 
24195530Sbonwick 	ASSERT3U(error, ==, 0);
24205530Sbonwick 
24215530Sbonwick 	tx = dmu_tx_create(os);
24225530Sbonwick 	dmu_tx_hold_zap(tx, object, TRUE, NULL);
24235530Sbonwick 	error = dmu_tx_assign(tx, TXG_WAIT);
24245530Sbonwick 	if (error) {
24255530Sbonwick 		ztest_record_enospc("remove zap entry");
24265530Sbonwick 		dmu_tx_abort(tx);
24275530Sbonwick 		return;
24285530Sbonwick 	}
24295530Sbonwick 	error = zap_remove(os, object, txgname, tx);
24305530Sbonwick 	if (error)
24315530Sbonwick 		fatal(0, "zap_remove('%s', %llu, '%s') = %d",
24325530Sbonwick 		    osname, object, txgname, error);
24335530Sbonwick 
24345530Sbonwick 	error = zap_remove(os, object, propname, tx);
24355530Sbonwick 	if (error)
24365530Sbonwick 		fatal(0, "zap_remove('%s', %llu, '%s') = %d",
24375530Sbonwick 		    osname, object, propname, error);
24385530Sbonwick 
24395530Sbonwick 	dmu_tx_commit(tx);
2440789Sahrens 
2441789Sahrens 	/*
2442789Sahrens 	 * Once in a while, destroy the object.
2443789Sahrens 	 */
24445530Sbonwick 	if (ztest_random(1000) != 0)
2445789Sahrens 		return;
2446789Sahrens 
2447789Sahrens 	tx = dmu_tx_create(os);
2448789Sahrens 	dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff, sizeof (uint64_t));
2449789Sahrens 	dmu_tx_hold_free(tx, object, 0, DMU_OBJECT_END);
2450789Sahrens 	error = dmu_tx_assign(tx, TXG_WAIT);
2451789Sahrens 	if (error) {
2452789Sahrens 		ztest_record_enospc("destroy zap object");
2453789Sahrens 		dmu_tx_abort(tx);
2454789Sahrens 		return;
2455789Sahrens 	}
2456789Sahrens 	error = zap_destroy(os, object, tx);
2457789Sahrens 	if (error)
2458789Sahrens 		fatal(0, "zap_destroy('%s', %llu) = %d",
2459789Sahrens 		    osname, object, error);
2460789Sahrens 	object = 0;
2461789Sahrens 	dmu_write(os, ZTEST_DIROBJ, za->za_diroff, sizeof (uint64_t),
2462789Sahrens 	    &object, tx);
2463789Sahrens 	dmu_tx_commit(tx);
2464789Sahrens }
2465789Sahrens 
2466789Sahrens void
2467789Sahrens ztest_zap_parallel(ztest_args_t *za)
2468789Sahrens {
2469789Sahrens 	objset_t *os = za->za_os;
2470789Sahrens 	uint64_t txg, object, count, wsize, wc, zl_wsize, zl_wc;
2471789Sahrens 	dmu_tx_t *tx;
2472789Sahrens 	int i, namelen, error;
2473789Sahrens 	char name[20], string_value[20];
2474789Sahrens 	void *data;
2475789Sahrens 
24765530Sbonwick 	/*
24775530Sbonwick 	 * Generate a random name of the form 'xxx.....' where each
24785530Sbonwick 	 * x is a random printable character and the dots are dots.
24795530Sbonwick 	 * There are 94 such characters, and the name length goes from
24805530Sbonwick 	 * 6 to 20, so there are 94^3 * 15 = 12,458,760 possible names.
24815530Sbonwick 	 */
24825530Sbonwick 	namelen = ztest_random(sizeof (name) - 5) + 5 + 1;
24835530Sbonwick 
24845530Sbonwick 	for (i = 0; i < 3; i++)
24855530Sbonwick 		name[i] = '!' + ztest_random('~' - '!' + 1);
24865530Sbonwick 	for (; i < namelen - 1; i++)
24875530Sbonwick 		name[i] = '.';
24885530Sbonwick 	name[i] = '\0';
24895530Sbonwick 
24905530Sbonwick 	if (ztest_random(2) == 0)
24915530Sbonwick 		object = ZTEST_MICROZAP_OBJ;
24925530Sbonwick 	else
24935530Sbonwick 		object = ZTEST_FATZAP_OBJ;
24945530Sbonwick 
24955530Sbonwick 	if ((namelen & 1) || object == ZTEST_MICROZAP_OBJ) {
24965530Sbonwick 		wsize = sizeof (txg);
24975530Sbonwick 		wc = 1;
24985530Sbonwick 		data = &txg;
24995530Sbonwick 	} else {
25005530Sbonwick 		wsize = 1;
25015530Sbonwick 		wc = namelen;
25025530Sbonwick 		data = string_value;
25035530Sbonwick 	}
25045530Sbonwick 
25055530Sbonwick 	count = -1ULL;
25065530Sbonwick 	VERIFY(zap_count(os, object, &count) == 0);
25075530Sbonwick 	ASSERT(count != -1ULL);
25085530Sbonwick 
25095530Sbonwick 	/*
25105530Sbonwick 	 * Select an operation: length, lookup, add, update, remove.
25115530Sbonwick 	 */
25125530Sbonwick 	i = ztest_random(5);
25135530Sbonwick 
25145530Sbonwick 	if (i >= 2) {
25155530Sbonwick 		tx = dmu_tx_create(os);
25165530Sbonwick 		dmu_tx_hold_zap(tx, object, TRUE, NULL);
25175530Sbonwick 		error = dmu_tx_assign(tx, TXG_WAIT);
25185530Sbonwick 		if (error) {
25195530Sbonwick 			ztest_record_enospc("zap parallel");
25205530Sbonwick 			dmu_tx_abort(tx);
25215530Sbonwick 			return;
2522789Sahrens 		}
25235530Sbonwick 		txg = dmu_tx_get_txg(tx);
25245530Sbonwick 		bcopy(name, string_value, namelen);
25255530Sbonwick 	} else {
25265530Sbonwick 		tx = NULL;
25275530Sbonwick 		txg = 0;
25285530Sbonwick 		bzero(string_value, namelen);
25295530Sbonwick 	}
25305530Sbonwick 
25315530Sbonwick 	switch (i) {
25325530Sbonwick 
25335530Sbonwick 	case 0:
25345530Sbonwick 		error = zap_length(os, object, name, &zl_wsize, &zl_wc);
25355530Sbonwick 		if (error == 0) {
25365530Sbonwick 			ASSERT3U(wsize, ==, zl_wsize);
25375530Sbonwick 			ASSERT3U(wc, ==, zl_wc);
2538789Sahrens 		} else {
25395530Sbonwick 			ASSERT3U(error, ==, ENOENT);
2540789Sahrens 		}
25415530Sbonwick 		break;
25425530Sbonwick 
25435530Sbonwick 	case 1:
25445530Sbonwick 		error = zap_lookup(os, object, name, wsize, wc, data);
25455530Sbonwick 		if (error == 0) {
25465530Sbonwick 			if (data == string_value &&
25475530Sbonwick 			    bcmp(name, data, namelen) != 0)
25485530Sbonwick 				fatal(0, "name '%s' != val '%s' len %d",
25495530Sbonwick 				    name, data, namelen);
25505530Sbonwick 		} else {
25515530Sbonwick 			ASSERT3U(error, ==, ENOENT);
2552789Sahrens 		}
25535530Sbonwick 		break;
25545530Sbonwick 
25555530Sbonwick 	case 2:
25565530Sbonwick 		error = zap_add(os, object, name, wsize, wc, data, tx);
25575530Sbonwick 		ASSERT(error == 0 || error == EEXIST);
25585530Sbonwick 		break;
25595530Sbonwick 
25605530Sbonwick 	case 3:
25615530Sbonwick 		VERIFY(zap_update(os, object, name, wsize, wc, data, tx) == 0);
25625530Sbonwick 		break;
25635530Sbonwick 
25645530Sbonwick 	case 4:
25655530Sbonwick 		error = zap_remove(os, object, name, tx);
25665530Sbonwick 		ASSERT(error == 0 || error == ENOENT);
25675530Sbonwick 		break;
2568789Sahrens 	}
25695530Sbonwick 
25705530Sbonwick 	if (tx != NULL)
25715530Sbonwick 		dmu_tx_commit(tx);
2572789Sahrens }
2573789Sahrens 
2574789Sahrens void
2575789Sahrens ztest_dsl_prop_get_set(ztest_args_t *za)
2576789Sahrens {
2577789Sahrens 	objset_t *os = za->za_os;
2578789Sahrens 	int i, inherit;
2579789Sahrens 	uint64_t value;
2580789Sahrens 	const char *prop, *valname;
2581789Sahrens 	char setpoint[MAXPATHLEN];
2582789Sahrens 	char osname[MAXNAMELEN];
25831544Seschrock 	int error;
2584789Sahrens 
2585789Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
2586789Sahrens 
2587789Sahrens 	dmu_objset_name(os, osname);
2588789Sahrens 
2589789Sahrens 	for (i = 0; i < 2; i++) {
2590789Sahrens 		if (i == 0) {
2591789Sahrens 			prop = "checksum";
2592789Sahrens 			value = ztest_random_checksum();
2593789Sahrens 			inherit = (value == ZIO_CHECKSUM_INHERIT);
2594789Sahrens 		} else {
2595789Sahrens 			prop = "compression";
2596789Sahrens 			value = ztest_random_compress();
2597789Sahrens 			inherit = (value == ZIO_COMPRESS_INHERIT);
2598789Sahrens 		}
2599789Sahrens 
26001544Seschrock 		error = dsl_prop_set(osname, prop, sizeof (value),
26011544Seschrock 		    !inherit, &value);
26021544Seschrock 
26031544Seschrock 		if (error == ENOSPC) {
26041544Seschrock 			ztest_record_enospc("dsl_prop_set");
26051544Seschrock 			break;
26061544Seschrock 		}
26071544Seschrock 
26081544Seschrock 		ASSERT3U(error, ==, 0);
2609789Sahrens 
2610789Sahrens 		VERIFY3U(dsl_prop_get(osname, prop, sizeof (value),
2611789Sahrens 		    1, &value, setpoint), ==, 0);
2612789Sahrens 
2613789Sahrens 		if (i == 0)
2614789Sahrens 			valname = zio_checksum_table[value].ci_name;
2615789Sahrens 		else
2616789Sahrens 			valname = zio_compress_table[value].ci_name;
2617789Sahrens 
2618789Sahrens 		if (zopt_verbose >= 6) {
2619789Sahrens 			(void) printf("%s %s = %s for '%s'\n",
2620789Sahrens 			    osname, prop, valname, setpoint);
2621789Sahrens 		}
2622789Sahrens 	}
2623789Sahrens 
2624789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
2625789Sahrens }
2626789Sahrens 
2627789Sahrens /*
2628789Sahrens  * Inject random faults into the on-disk data.
2629789Sahrens  */
2630789Sahrens void
2631789Sahrens ztest_fault_inject(ztest_args_t *za)
2632789Sahrens {
2633789Sahrens 	int fd;
2634789Sahrens 	uint64_t offset;
2635789Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
2636789Sahrens 	uint64_t bad = 0x1990c0ffeedecade;
2637789Sahrens 	uint64_t top, leaf;
2638789Sahrens 	char path0[MAXPATHLEN];
2639789Sahrens 	char pathrand[MAXPATHLEN];
2640789Sahrens 	size_t fsize;
26415530Sbonwick 	spa_t *spa = za->za_spa;
2642789Sahrens 	int bshift = SPA_MAXBLOCKSHIFT + 2;	/* don't scrog all labels */
2643789Sahrens 	int iters = 1000;
26447754SJeff.Bonwick@Sun.COM 	int maxfaults = zopt_maxfaults;
26457754SJeff.Bonwick@Sun.COM 	vdev_t *vd0 = NULL;
26461544Seschrock 	uint64_t guid0 = 0;
26471544Seschrock 
26487754SJeff.Bonwick@Sun.COM 	ASSERT(leaves >= 1);
2649789Sahrens 
2650789Sahrens 	/*
26517754SJeff.Bonwick@Sun.COM 	 * We need SCL_STATE here because we're going to look at vd0->vdev_tsd.
2652789Sahrens 	 */
26537754SJeff.Bonwick@Sun.COM 	spa_config_enter(spa, SCL_STATE, FTAG, RW_READER);
26547754SJeff.Bonwick@Sun.COM 
26557754SJeff.Bonwick@Sun.COM 	if (ztest_random(2) == 0) {
26567754SJeff.Bonwick@Sun.COM 		/*
26577754SJeff.Bonwick@Sun.COM 		 * Inject errors on a normal data device.
26587754SJeff.Bonwick@Sun.COM 		 */
26597754SJeff.Bonwick@Sun.COM 		top = ztest_random(spa->spa_root_vdev->vdev_children);
26607754SJeff.Bonwick@Sun.COM 		leaf = ztest_random(leaves);
26617754SJeff.Bonwick@Sun.COM 
26627754SJeff.Bonwick@Sun.COM 		/*
26637754SJeff.Bonwick@Sun.COM 		 * Generate paths to the first leaf in this top-level vdev,
26647754SJeff.Bonwick@Sun.COM 		 * and to the random leaf we selected.  We'll induce transient
26657754SJeff.Bonwick@Sun.COM 		 * write failures and random online/offline activity on leaf 0,
26667754SJeff.Bonwick@Sun.COM 		 * and we'll write random garbage to the randomly chosen leaf.
26677754SJeff.Bonwick@Sun.COM 		 */
26687754SJeff.Bonwick@Sun.COM 		(void) snprintf(path0, sizeof (path0), ztest_dev_template,
26697754SJeff.Bonwick@Sun.COM 		    zopt_dir, zopt_pool, top * leaves + 0);
26707754SJeff.Bonwick@Sun.COM 		(void) snprintf(pathrand, sizeof (pathrand), ztest_dev_template,
26717754SJeff.Bonwick@Sun.COM 		    zopt_dir, zopt_pool, top * leaves + leaf);
26727754SJeff.Bonwick@Sun.COM 
26737754SJeff.Bonwick@Sun.COM 		vd0 = vdev_lookup_by_path(spa->spa_root_vdev, path0);
26747754SJeff.Bonwick@Sun.COM 		if (vd0 != NULL && maxfaults != 1) {
26757754SJeff.Bonwick@Sun.COM 			/*
26767754SJeff.Bonwick@Sun.COM 			 * Make vd0 explicitly claim to be unreadable,
26777754SJeff.Bonwick@Sun.COM 			 * or unwriteable, or reach behind its back
26787754SJeff.Bonwick@Sun.COM 			 * and close the underlying fd.  We can do this if
26797754SJeff.Bonwick@Sun.COM 			 * maxfaults == 0 because we'll fail and reexecute,
26807754SJeff.Bonwick@Sun.COM 			 * and we can do it if maxfaults >= 2 because we'll
26817754SJeff.Bonwick@Sun.COM 			 * have enough redundancy.  If maxfaults == 1, the
26827754SJeff.Bonwick@Sun.COM 			 * combination of this with injection of random data
26837754SJeff.Bonwick@Sun.COM 			 * corruption below exceeds the pool's fault tolerance.
26847754SJeff.Bonwick@Sun.COM 			 */
26857754SJeff.Bonwick@Sun.COM 			vdev_file_t *vf = vd0->vdev_tsd;
26867754SJeff.Bonwick@Sun.COM 
26877754SJeff.Bonwick@Sun.COM 			if (vf != NULL && ztest_random(3) == 0) {
26887754SJeff.Bonwick@Sun.COM 				(void) close(vf->vf_vnode->v_fd);
26897754SJeff.Bonwick@Sun.COM 				vf->vf_vnode->v_fd = -1;
26907754SJeff.Bonwick@Sun.COM 			} else if (ztest_random(2) == 0) {
26917754SJeff.Bonwick@Sun.COM 				vd0->vdev_cant_read = B_TRUE;
26927754SJeff.Bonwick@Sun.COM 			} else {
26937754SJeff.Bonwick@Sun.COM 				vd0->vdev_cant_write = B_TRUE;
26947754SJeff.Bonwick@Sun.COM 			}
26957754SJeff.Bonwick@Sun.COM 			guid0 = vd0->vdev_guid;
26967754SJeff.Bonwick@Sun.COM 		}
26977754SJeff.Bonwick@Sun.COM 	} else {
26987754SJeff.Bonwick@Sun.COM 		/*
26997754SJeff.Bonwick@Sun.COM 		 * Inject errors on an l2cache device.
27007754SJeff.Bonwick@Sun.COM 		 */
27017754SJeff.Bonwick@Sun.COM 		spa_aux_vdev_t *sav = &spa->spa_l2cache;
27027754SJeff.Bonwick@Sun.COM 
27037754SJeff.Bonwick@Sun.COM 		if (sav->sav_count == 0) {
27047754SJeff.Bonwick@Sun.COM 			spa_config_exit(spa, SCL_STATE, FTAG);
27057754SJeff.Bonwick@Sun.COM 			return;
27067754SJeff.Bonwick@Sun.COM 		}
27077754SJeff.Bonwick@Sun.COM 		vd0 = sav->sav_vdevs[ztest_random(sav->sav_count)];
27087754SJeff.Bonwick@Sun.COM 		guid0 = vd0->vdev_guid;
27097754SJeff.Bonwick@Sun.COM 		(void) strcpy(path0, vd0->vdev_path);
27107754SJeff.Bonwick@Sun.COM 		(void) strcpy(pathrand, vd0->vdev_path);
27117754SJeff.Bonwick@Sun.COM 
27127754SJeff.Bonwick@Sun.COM 		leaf = 0;
27137754SJeff.Bonwick@Sun.COM 		leaves = 1;
27147754SJeff.Bonwick@Sun.COM 		maxfaults = INT_MAX;	/* no limit on cache devices */
27157754SJeff.Bonwick@Sun.COM 	}
2716789Sahrens 
27177754SJeff.Bonwick@Sun.COM 	spa_config_exit(spa, SCL_STATE, FTAG);
27187754SJeff.Bonwick@Sun.COM 
27197754SJeff.Bonwick@Sun.COM 	if (maxfaults == 0)
27207754SJeff.Bonwick@Sun.COM 		return;
27211544Seschrock 
27221544Seschrock 	/*
27231544Seschrock 	 * If we can tolerate two or more faults, randomly online/offline vd0.
27241544Seschrock 	 */
27257754SJeff.Bonwick@Sun.COM 	if (maxfaults >= 2 && guid0 != 0) {
27268241SJeff.Bonwick@Sun.COM 		if (ztest_random(10) < 6) {
27278241SJeff.Bonwick@Sun.COM 			int flags = (ztest_random(2) == 0 ?
27288241SJeff.Bonwick@Sun.COM 			    ZFS_OFFLINE_TEMPORARY : 0);
27298241SJeff.Bonwick@Sun.COM 			VERIFY(vdev_offline(spa, guid0, flags) != EBUSY);
27308241SJeff.Bonwick@Sun.COM 		} else {
27318241SJeff.Bonwick@Sun.COM 			(void) vdev_online(spa, guid0, 0, NULL);
27328241SJeff.Bonwick@Sun.COM 		}
2733789Sahrens 	}
2734789Sahrens 
2735789Sahrens 	/*
27361544Seschrock 	 * We have at least single-fault tolerance, so inject data corruption.
2737789Sahrens 	 */
2738789Sahrens 	fd = open(pathrand, O_RDWR);
2739789Sahrens 
2740789Sahrens 	if (fd == -1)	/* we hit a gap in the device namespace */
2741789Sahrens 		return;
2742789Sahrens 
2743789Sahrens 	fsize = lseek(fd, 0, SEEK_END);
2744789Sahrens 
2745789Sahrens 	while (--iters != 0) {
2746789Sahrens 		offset = ztest_random(fsize / (leaves << bshift)) *
2747789Sahrens 		    (leaves << bshift) + (leaf << bshift) +
2748789Sahrens 		    (ztest_random(1ULL << (bshift - 1)) & -8ULL);
2749789Sahrens 
2750789Sahrens 		if (offset >= fsize)
2751789Sahrens 			continue;
2752789Sahrens 
2753789Sahrens 		if (zopt_verbose >= 6)
2754789Sahrens 			(void) printf("injecting bad word into %s,"
2755789Sahrens 			    " offset 0x%llx\n", pathrand, (u_longlong_t)offset);
2756789Sahrens 
2757789Sahrens 		if (pwrite(fd, &bad, sizeof (bad), offset) != sizeof (bad))
2758789Sahrens 			fatal(1, "can't inject bad word at 0x%llx in %s",
2759789Sahrens 			    offset, pathrand);
2760789Sahrens 	}
2761789Sahrens 
2762789Sahrens 	(void) close(fd);
2763789Sahrens }
2764789Sahrens 
2765789Sahrens /*
2766789Sahrens  * Scrub the pool.
2767789Sahrens  */
2768789Sahrens void
2769789Sahrens ztest_scrub(ztest_args_t *za)
2770789Sahrens {
27715530Sbonwick 	spa_t *spa = za->za_spa;
2772789Sahrens 
27737046Sahrens 	(void) spa_scrub(spa, POOL_SCRUB_EVERYTHING);
2774789Sahrens 	(void) poll(NULL, 0, 1000); /* wait a second, then force a restart */
27757046Sahrens 	(void) spa_scrub(spa, POOL_SCRUB_EVERYTHING);
2776789Sahrens }
2777789Sahrens 
2778789Sahrens /*
2779789Sahrens  * Rename the pool to a different name and then rename it back.
2780789Sahrens  */
2781789Sahrens void
2782789Sahrens ztest_spa_rename(ztest_args_t *za)
2783789Sahrens {
2784789Sahrens 	char *oldname, *newname;
2785789Sahrens 	int error;
2786789Sahrens 	spa_t *spa;
2787789Sahrens 
2788789Sahrens 	(void) rw_wrlock(&ztest_shared->zs_name_lock);
2789789Sahrens 
2790789Sahrens 	oldname = za->za_pool;
2791789Sahrens 	newname = umem_alloc(strlen(oldname) + 5, UMEM_NOFAIL);
2792789Sahrens 	(void) strcpy(newname, oldname);
2793789Sahrens 	(void) strcat(newname, "_tmp");
2794789Sahrens 
2795789Sahrens 	/*
2796789Sahrens 	 * Do the rename
2797789Sahrens 	 */
2798789Sahrens 	error = spa_rename(oldname, newname);
2799789Sahrens 	if (error)
2800789Sahrens 		fatal(0, "spa_rename('%s', '%s') = %d", oldname,
2801789Sahrens 		    newname, error);
2802789Sahrens 
2803789Sahrens 	/*
2804789Sahrens 	 * Try to open it under the old name, which shouldn't exist
2805789Sahrens 	 */
2806789Sahrens 	error = spa_open(oldname, &spa, FTAG);
2807789Sahrens 	if (error != ENOENT)
2808789Sahrens 		fatal(0, "spa_open('%s') = %d", oldname, error);
2809789Sahrens 
2810789Sahrens 	/*
2811789Sahrens 	 * Open it under the new name and make sure it's still the same spa_t.
2812789Sahrens 	 */
2813789Sahrens 	error = spa_open(newname, &spa, FTAG);
2814789Sahrens 	if (error != 0)
2815789Sahrens 		fatal(0, "spa_open('%s') = %d", newname, error);
2816789Sahrens 
28175530Sbonwick 	ASSERT(spa == za->za_spa);
2818789Sahrens 	spa_close(spa, FTAG);
2819789Sahrens 
2820789Sahrens 	/*
2821789Sahrens 	 * Rename it back to the original
2822789Sahrens 	 */
2823789Sahrens 	error = spa_rename(newname, oldname);
2824789Sahrens 	if (error)
2825789Sahrens 		fatal(0, "spa_rename('%s', '%s') = %d", newname,
2826789Sahrens 		    oldname, error);
2827789Sahrens 
2828789Sahrens 	/*
2829789Sahrens 	 * Make sure it can still be opened
2830789Sahrens 	 */
2831789Sahrens 	error = spa_open(oldname, &spa, FTAG);
2832789Sahrens 	if (error != 0)
2833789Sahrens 		fatal(0, "spa_open('%s') = %d", oldname, error);
2834789Sahrens 
28355530Sbonwick 	ASSERT(spa == za->za_spa);
2836789Sahrens 	spa_close(spa, FTAG);
2837789Sahrens 
2838789Sahrens 	umem_free(newname, strlen(newname) + 1);
2839789Sahrens 
2840789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
2841789Sahrens }
2842789Sahrens 
2843789Sahrens 
2844789Sahrens /*
2845789Sahrens  * Completely obliterate one disk.
2846789Sahrens  */
2847789Sahrens static void
2848789Sahrens ztest_obliterate_one_disk(uint64_t vdev)
2849789Sahrens {
2850789Sahrens 	int fd;
28511807Sbonwick 	char dev_name[MAXPATHLEN], copy_name[MAXPATHLEN];
2852789Sahrens 	size_t fsize;
2853789Sahrens 
2854789Sahrens 	if (zopt_maxfaults < 2)
2855789Sahrens 		return;
2856789Sahrens 
2857789Sahrens 	(void) sprintf(dev_name, ztest_dev_template, zopt_dir, zopt_pool, vdev);
28581807Sbonwick 	(void) snprintf(copy_name, MAXPATHLEN, "%s.old", dev_name);
2859789Sahrens 
2860789Sahrens 	fd = open(dev_name, O_RDWR);
2861789Sahrens 
2862789Sahrens 	if (fd == -1)
2863789Sahrens 		fatal(1, "can't open %s", dev_name);
2864789Sahrens 
2865789Sahrens 	/*
2866789Sahrens 	 * Determine the size.
2867789Sahrens 	 */
2868789Sahrens 	fsize = lseek(fd, 0, SEEK_END);
28691807Sbonwick 
2870789Sahrens 	(void) close(fd);
2871789Sahrens 
2872789Sahrens 	/*
28731807Sbonwick 	 * Rename the old device to dev_name.old (useful for debugging).
2874789Sahrens 	 */
28751807Sbonwick 	VERIFY(rename(dev_name, copy_name) == 0);
2876789Sahrens 
2877789Sahrens 	/*
2878789Sahrens 	 * Create a new one.
2879789Sahrens 	 */
2880789Sahrens 	VERIFY((fd = open(dev_name, O_RDWR | O_CREAT | O_TRUNC, 0666)) >= 0);
2881789Sahrens 	VERIFY(ftruncate(fd, fsize) == 0);
2882789Sahrens 	(void) close(fd);
2883789Sahrens }
2884789Sahrens 
2885789Sahrens static void
2886789Sahrens ztest_replace_one_disk(spa_t *spa, uint64_t vdev)
2887789Sahrens {
2888789Sahrens 	char dev_name[MAXPATHLEN];
28897754SJeff.Bonwick@Sun.COM 	nvlist_t *root;
2890789Sahrens 	int error;
28911544Seschrock 	uint64_t guid;
28921544Seschrock 	vdev_t *vd;
2893789Sahrens 
2894789Sahrens 	(void) sprintf(dev_name, ztest_dev_template, zopt_dir, zopt_pool, vdev);
2895789Sahrens 
2896789Sahrens 	/*
2897789Sahrens 	 * Build the nvlist describing dev_name.
2898789Sahrens 	 */
28997754SJeff.Bonwick@Sun.COM 	root = make_vdev_root(dev_name, NULL, 0, 0, 0, 0, 0, 1);
29007754SJeff.Bonwick@Sun.COM 
29017754SJeff.Bonwick@Sun.COM 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
29021544Seschrock 	if ((vd = vdev_lookup_by_path(spa->spa_root_vdev, dev_name)) == NULL)
29031544Seschrock 		guid = 0;
29041544Seschrock 	else
29051544Seschrock 		guid = vd->vdev_guid;
29067754SJeff.Bonwick@Sun.COM 	spa_config_exit(spa, SCL_VDEV, FTAG);
29071544Seschrock 	error = spa_vdev_attach(spa, guid, root, B_TRUE);
29081732Sbonwick 	if (error != 0 &&
29091732Sbonwick 	    error != EBUSY &&
29101732Sbonwick 	    error != ENOTSUP &&
29111732Sbonwick 	    error != ENODEV &&
29121732Sbonwick 	    error != EDOM)
2913789Sahrens 		fatal(0, "spa_vdev_attach(in-place) = %d", error);
2914789Sahrens 
2915789Sahrens 	nvlist_free(root);
2916789Sahrens }
2917789Sahrens 
2918789Sahrens static void
2919789Sahrens ztest_verify_blocks(char *pool)
2920789Sahrens {
2921789Sahrens 	int status;
2922789Sahrens 	char zdb[MAXPATHLEN + MAXNAMELEN + 20];
2923789Sahrens 	char zbuf[1024];
2924789Sahrens 	char *bin;
29254527Sperrin 	char *ztest;
29264527Sperrin 	char *isa;
29274527Sperrin 	int isalen;
2928789Sahrens 	FILE *fp;
2929789Sahrens 
2930789Sahrens 	(void) realpath(getexecname(), zdb);
2931789Sahrens 
2932789Sahrens 	/* zdb lives in /usr/sbin, while ztest lives in /usr/bin */
2933789Sahrens 	bin = strstr(zdb, "/usr/bin/");
29344527Sperrin 	ztest = strstr(bin, "/ztest");
29354527Sperrin 	isa = bin + 8;
29364527Sperrin 	isalen = ztest - isa;
29374527Sperrin 	isa = strdup(isa);
2938789Sahrens 	/* LINTED */
29395994Sck153898 	(void) sprintf(bin,
29407837SMatthew.Ahrens@Sun.COM 	    "/usr/sbin%.*s/zdb -bc%s%s -U /tmp/zpool.cache %s",
29414527Sperrin 	    isalen,
29424527Sperrin 	    isa,
2943789Sahrens 	    zopt_verbose >= 3 ? "s" : "",
2944789Sahrens 	    zopt_verbose >= 4 ? "v" : "",
29457837SMatthew.Ahrens@Sun.COM 	    pool);
29464527Sperrin 	free(isa);
2947789Sahrens 
2948789Sahrens 	if (zopt_verbose >= 5)
2949789Sahrens 		(void) printf("Executing %s\n", strstr(zdb, "zdb "));
2950789Sahrens 
2951789Sahrens 	fp = popen(zdb, "r");
2952789Sahrens 
2953789Sahrens 	while (fgets(zbuf, sizeof (zbuf), fp) != NULL)
2954789Sahrens 		if (zopt_verbose >= 3)
2955789Sahrens 			(void) printf("%s", zbuf);
2956789Sahrens 
2957789Sahrens 	status = pclose(fp);
2958789Sahrens 
2959789Sahrens 	if (status == 0)
2960789Sahrens 		return;
2961789Sahrens 
2962789Sahrens 	ztest_dump_core = 0;
2963789Sahrens 	if (WIFEXITED(status))
2964789Sahrens 		fatal(0, "'%s' exit code %d", zdb, WEXITSTATUS(status));
2965789Sahrens 	else
2966789Sahrens 		fatal(0, "'%s' died with signal %d", zdb, WTERMSIG(status));
2967789Sahrens }
2968789Sahrens 
2969789Sahrens static void
2970789Sahrens ztest_walk_pool_directory(char *header)
2971789Sahrens {
2972789Sahrens 	spa_t *spa = NULL;
2973789Sahrens 
2974789Sahrens 	if (zopt_verbose >= 6)
2975789Sahrens 		(void) printf("%s\n", header);
2976789Sahrens 
2977789Sahrens 	mutex_enter(&spa_namespace_lock);
2978789Sahrens 	while ((spa = spa_next(spa)) != NULL)
2979789Sahrens 		if (zopt_verbose >= 6)
2980789Sahrens 			(void) printf("\t%s\n", spa_name(spa));
2981789Sahrens 	mutex_exit(&spa_namespace_lock);
2982789Sahrens }
2983789Sahrens 
2984789Sahrens static void
2985789Sahrens ztest_spa_import_export(char *oldname, char *newname)
2986789Sahrens {
29878241SJeff.Bonwick@Sun.COM 	nvlist_t *config, *newconfig;
2988789Sahrens 	uint64_t pool_guid;
2989789Sahrens 	spa_t *spa;
2990789Sahrens 	int error;
2991789Sahrens 
2992789Sahrens 	if (zopt_verbose >= 4) {
2993789Sahrens 		(void) printf("import/export: old = %s, new = %s\n",
2994789Sahrens 		    oldname, newname);
2995789Sahrens 	}
2996789Sahrens 
2997789Sahrens 	/*
2998789Sahrens 	 * Clean up from previous runs.
2999789Sahrens 	 */
3000789Sahrens 	(void) spa_destroy(newname);
3001789Sahrens 
3002789Sahrens 	/*
3003789Sahrens 	 * Get the pool's configuration and guid.
3004789Sahrens 	 */
3005789Sahrens 	error = spa_open(oldname, &spa, FTAG);
3006789Sahrens 	if (error)
3007789Sahrens 		fatal(0, "spa_open('%s') = %d", oldname, error);
3008789Sahrens 
30098241SJeff.Bonwick@Sun.COM 	/*
30108241SJeff.Bonwick@Sun.COM 	 * Kick off a scrub to tickle scrub/export races.
30118241SJeff.Bonwick@Sun.COM 	 */
30128241SJeff.Bonwick@Sun.COM 	if (ztest_random(2) == 0)
30138241SJeff.Bonwick@Sun.COM 		(void) spa_scrub(spa, POOL_SCRUB_EVERYTHING);
30148241SJeff.Bonwick@Sun.COM 
3015789Sahrens 	pool_guid = spa_guid(spa);
3016789Sahrens 	spa_close(spa, FTAG);
3017789Sahrens 
3018789Sahrens 	ztest_walk_pool_directory("pools before export");
3019789Sahrens 
3020789Sahrens 	/*
3021789Sahrens 	 * Export it.
3022789Sahrens 	 */
30238211SGeorge.Wilson@Sun.COM 	error = spa_export(oldname, &config, B_FALSE, B_FALSE);
3024789Sahrens 	if (error)
3025789Sahrens 		fatal(0, "spa_export('%s') = %d", oldname, error);
3026789Sahrens 
3027789Sahrens 	ztest_walk_pool_directory("pools after export");
3028789Sahrens 
3029789Sahrens 	/*
30308241SJeff.Bonwick@Sun.COM 	 * Try to import it.
30318241SJeff.Bonwick@Sun.COM 	 */
30328241SJeff.Bonwick@Sun.COM 	newconfig = spa_tryimport(config);
30338241SJeff.Bonwick@Sun.COM 	ASSERT(newconfig != NULL);
30348241SJeff.Bonwick@Sun.COM 	nvlist_free(newconfig);
30358241SJeff.Bonwick@Sun.COM 
30368241SJeff.Bonwick@Sun.COM 	/*
3037789Sahrens 	 * Import it under the new name.
3038789Sahrens 	 */
3039789Sahrens 	error = spa_import(newname, config, NULL);
3040789Sahrens 	if (error)
3041789Sahrens 		fatal(0, "spa_import('%s') = %d", newname, error);
3042789Sahrens 
3043789Sahrens 	ztest_walk_pool_directory("pools after import");
3044789Sahrens 
3045789Sahrens 	/*
3046789Sahrens 	 * Try to import it again -- should fail with EEXIST.
3047789Sahrens 	 */
3048789Sahrens 	error = spa_import(newname, config, NULL);
3049789Sahrens 	if (error != EEXIST)
3050789Sahrens 		fatal(0, "spa_import('%s') twice", newname);
3051789Sahrens 
3052789Sahrens 	/*
3053789Sahrens 	 * Try to import it under a different name -- should fail with EEXIST.
3054789Sahrens 	 */
3055789Sahrens 	error = spa_import(oldname, config, NULL);
3056789Sahrens 	if (error != EEXIST)
3057789Sahrens 		fatal(0, "spa_import('%s') under multiple names", newname);
3058789Sahrens 
3059789Sahrens 	/*
3060789Sahrens 	 * Verify that the pool is no longer visible under the old name.
3061789Sahrens 	 */
3062789Sahrens 	error = spa_open(oldname, &spa, FTAG);
3063789Sahrens 	if (error != ENOENT)
3064789Sahrens 		fatal(0, "spa_open('%s') = %d", newname, error);
3065789Sahrens 
3066789Sahrens 	/*
3067789Sahrens 	 * Verify that we can open and close the pool using the new name.
3068789Sahrens 	 */
3069789Sahrens 	error = spa_open(newname, &spa, FTAG);
3070789Sahrens 	if (error)
3071789Sahrens 		fatal(0, "spa_open('%s') = %d", newname, error);
3072789Sahrens 	ASSERT(pool_guid == spa_guid(spa));
3073789Sahrens 	spa_close(spa, FTAG);
3074789Sahrens 
3075789Sahrens 	nvlist_free(config);
3076789Sahrens }
3077789Sahrens 
30788241SJeff.Bonwick@Sun.COM static void
30798241SJeff.Bonwick@Sun.COM ztest_resume(spa_t *spa)
30808241SJeff.Bonwick@Sun.COM {
30818241SJeff.Bonwick@Sun.COM 	if (spa_suspended(spa)) {
30828241SJeff.Bonwick@Sun.COM 		spa_vdev_state_enter(spa);
30838241SJeff.Bonwick@Sun.COM 		vdev_clear(spa, NULL);
30848241SJeff.Bonwick@Sun.COM 		(void) spa_vdev_state_exit(spa, NULL, 0);
30858241SJeff.Bonwick@Sun.COM 		zio_resume(spa);
30868241SJeff.Bonwick@Sun.COM 	}
30878241SJeff.Bonwick@Sun.COM }
30888241SJeff.Bonwick@Sun.COM 
30895329Sgw25295 static void *
30908241SJeff.Bonwick@Sun.COM ztest_resume_thread(void *arg)
30915329Sgw25295 {
30927754SJeff.Bonwick@Sun.COM 	spa_t *spa = arg;
30935329Sgw25295 
30945329Sgw25295 	while (!ztest_exiting) {
30957754SJeff.Bonwick@Sun.COM 		(void) poll(NULL, 0, 1000);
30968241SJeff.Bonwick@Sun.COM 		ztest_resume(spa);
30975329Sgw25295 	}
30985329Sgw25295 	return (NULL);
30995329Sgw25295 }
31005329Sgw25295 
3101789Sahrens static void *
3102789Sahrens ztest_thread(void *arg)
3103789Sahrens {
3104789Sahrens 	ztest_args_t *za = arg;
3105789Sahrens 	ztest_shared_t *zs = ztest_shared;
3106789Sahrens 	hrtime_t now, functime;
3107789Sahrens 	ztest_info_t *zi;
31085530Sbonwick 	int f, i;
3109789Sahrens 
3110789Sahrens 	while ((now = gethrtime()) < za->za_stop) {
3111789Sahrens 		/*
3112789Sahrens 		 * See if it's time to force a crash.
3113789Sahrens 		 */
3114789Sahrens 		if (now > za->za_kill) {
31155530Sbonwick 			zs->zs_alloc = spa_get_alloc(za->za_spa);
31165530Sbonwick 			zs->zs_space = spa_get_space(za->za_spa);
3117789Sahrens 			(void) kill(getpid(), SIGKILL);
3118789Sahrens 		}
3119789Sahrens 
3120789Sahrens 		/*
3121789Sahrens 		 * Pick a random function.
3122789Sahrens 		 */
3123789Sahrens 		f = ztest_random(ZTEST_FUNCS);
3124789Sahrens 		zi = &zs->zs_info[f];
3125789Sahrens 
3126789Sahrens 		/*
3127789Sahrens 		 * Decide whether to call it, based on the requested frequency.
3128789Sahrens 		 */
3129789Sahrens 		if (zi->zi_call_target == 0 ||
3130789Sahrens 		    (double)zi->zi_call_total / zi->zi_call_target >
3131789Sahrens 		    (double)(now - zs->zs_start_time) / (zopt_time * NANOSEC))
3132789Sahrens 			continue;
3133789Sahrens 
3134789Sahrens 		atomic_add_64(&zi->zi_calls, 1);
3135789Sahrens 		atomic_add_64(&zi->zi_call_total, 1);
3136789Sahrens 
3137789Sahrens 		za->za_diroff = (za->za_instance * ZTEST_FUNCS + f) *
3138789Sahrens 		    ZTEST_DIRSIZE;
3139789Sahrens 		za->za_diroff_shared = (1ULL << 63);
3140789Sahrens 
31415530Sbonwick 		for (i = 0; i < zi->zi_iters; i++)
31425530Sbonwick 			zi->zi_func(za);
3143789Sahrens 
3144789Sahrens 		functime = gethrtime() - now;
3145789Sahrens 
3146789Sahrens 		atomic_add_64(&zi->zi_call_time, functime);
3147789Sahrens 
3148789Sahrens 		if (zopt_verbose >= 4) {
3149789Sahrens 			Dl_info dli;
3150789Sahrens 			(void) dladdr((void *)zi->zi_func, &dli);
3151789Sahrens 			(void) printf("%6.2f sec in %s\n",
3152789Sahrens 			    (double)functime / NANOSEC, dli.dli_sname);
3153789Sahrens 		}
3154789Sahrens 
3155789Sahrens 		/*
3156789Sahrens 		 * If we're getting ENOSPC with some regularity, stop.
3157789Sahrens 		 */
3158789Sahrens 		if (zs->zs_enospc_count > 10)
3159789Sahrens 			break;
3160789Sahrens 	}
3161789Sahrens 
3162789Sahrens 	return (NULL);
3163789Sahrens }
3164789Sahrens 
3165789Sahrens /*
3166789Sahrens  * Kick off threads to run tests on all datasets in parallel.
3167789Sahrens  */
3168789Sahrens static void
3169789Sahrens ztest_run(char *pool)
3170789Sahrens {
3171789Sahrens 	int t, d, error;
3172789Sahrens 	ztest_shared_t *zs = ztest_shared;
3173789Sahrens 	ztest_args_t *za;
3174789Sahrens 	spa_t *spa;
3175789Sahrens 	char name[100];
31767754SJeff.Bonwick@Sun.COM 	thread_t resume_tid;
31777754SJeff.Bonwick@Sun.COM 
31787754SJeff.Bonwick@Sun.COM 	ztest_exiting = B_FALSE;
3179789Sahrens 
3180789Sahrens 	(void) _mutex_init(&zs->zs_vdev_lock, USYNC_THREAD, NULL);
3181789Sahrens 	(void) rwlock_init(&zs->zs_name_lock, USYNC_THREAD, NULL);
3182789Sahrens 
3183789Sahrens 	for (t = 0; t < ZTEST_SYNC_LOCKS; t++)
3184789Sahrens 		(void) _mutex_init(&zs->zs_sync_lock[t], USYNC_THREAD, NULL);
3185789Sahrens 
3186789Sahrens 	/*
3187789Sahrens 	 * Destroy one disk before we even start.
3188789Sahrens 	 * It's mirrored, so everything should work just fine.
3189789Sahrens 	 * This makes us exercise fault handling very early in spa_load().
3190789Sahrens 	 */
3191789Sahrens 	ztest_obliterate_one_disk(0);
3192789Sahrens 
3193789Sahrens 	/*
3194789Sahrens 	 * Verify that the sum of the sizes of all blocks in the pool
3195789Sahrens 	 * equals the SPA's allocated space total.
3196789Sahrens 	 */
3197789Sahrens 	ztest_verify_blocks(pool);
3198789Sahrens 
3199789Sahrens 	/*
3200789Sahrens 	 * Kick off a replacement of the disk we just obliterated.
3201789Sahrens 	 */
3202789Sahrens 	kernel_init(FREAD | FWRITE);
32037754SJeff.Bonwick@Sun.COM 	VERIFY(spa_open(pool, &spa, FTAG) == 0);
3204789Sahrens 	ztest_replace_one_disk(spa, 0);
3205789Sahrens 	if (zopt_verbose >= 5)
3206789Sahrens 		show_pool_stats(spa);
3207789Sahrens 	spa_close(spa, FTAG);
3208789Sahrens 	kernel_fini();
3209789Sahrens 
3210789Sahrens 	kernel_init(FREAD | FWRITE);
3211789Sahrens 
3212789Sahrens 	/*
3213789Sahrens 	 * Verify that we can export the pool and reimport it under a
3214789Sahrens 	 * different name.
3215789Sahrens 	 */
32161585Sbonwick 	if (ztest_random(2) == 0) {
32171585Sbonwick 		(void) snprintf(name, 100, "%s_import", pool);
32181585Sbonwick 		ztest_spa_import_export(pool, name);
32191585Sbonwick 		ztest_spa_import_export(name, pool);
32201585Sbonwick 	}
3221789Sahrens 
3222789Sahrens 	/*
3223789Sahrens 	 * Verify that we can loop over all pools.
3224789Sahrens 	 */
3225789Sahrens 	mutex_enter(&spa_namespace_lock);
3226789Sahrens 	for (spa = spa_next(NULL); spa != NULL; spa = spa_next(spa)) {
3227789Sahrens 		if (zopt_verbose > 3) {
3228789Sahrens 			(void) printf("spa_next: found %s\n", spa_name(spa));
3229789Sahrens 		}
3230789Sahrens 	}
3231789Sahrens 	mutex_exit(&spa_namespace_lock);
3232789Sahrens 
3233789Sahrens 	/*
32347754SJeff.Bonwick@Sun.COM 	 * Open our pool.
32355329Sgw25295 	 */
32367754SJeff.Bonwick@Sun.COM 	VERIFY(spa_open(pool, &spa, FTAG) == 0);
32375329Sgw25295 
32385329Sgw25295 	/*
32398241SJeff.Bonwick@Sun.COM 	 * We don't expect the pool to suspend unless maxfaults == 0,
32408241SJeff.Bonwick@Sun.COM 	 * in which case ztest_fault_inject() temporarily takes away
32418241SJeff.Bonwick@Sun.COM 	 * the only valid replica.
32428241SJeff.Bonwick@Sun.COM 	 */
32438241SJeff.Bonwick@Sun.COM 	if (zopt_maxfaults == 0)
32448241SJeff.Bonwick@Sun.COM 		spa->spa_failmode = ZIO_FAILURE_MODE_WAIT;
32458241SJeff.Bonwick@Sun.COM 	else
32468241SJeff.Bonwick@Sun.COM 		spa->spa_failmode = ZIO_FAILURE_MODE_PANIC;
32478241SJeff.Bonwick@Sun.COM 
32488241SJeff.Bonwick@Sun.COM 	/*
32497754SJeff.Bonwick@Sun.COM 	 * Create a thread to periodically resume suspended I/O.
3250789Sahrens 	 */
32518241SJeff.Bonwick@Sun.COM 	VERIFY(thr_create(0, 0, ztest_resume_thread, spa, THR_BOUND,
32527754SJeff.Bonwick@Sun.COM 	    &resume_tid) == 0);
3253789Sahrens 
3254789Sahrens 	/*
3255789Sahrens 	 * Verify that we can safely inquire about about any object,
3256789Sahrens 	 * whether it's allocated or not.  To make it interesting,
3257789Sahrens 	 * we probe a 5-wide window around each power of two.
3258789Sahrens 	 * This hits all edge cases, including zero and the max.
3259789Sahrens 	 */
3260789Sahrens 	for (t = 0; t < 64; t++) {
3261789Sahrens 		for (d = -5; d <= 5; d++) {
3262789Sahrens 			error = dmu_object_info(spa->spa_meta_objset,
3263789Sahrens 			    (1ULL << t) + d, NULL);
32641544Seschrock 			ASSERT(error == 0 || error == ENOENT ||
32651544Seschrock 			    error == EINVAL);
3266789Sahrens 		}
3267789Sahrens 	}
3268789Sahrens 
3269789Sahrens 	/*
3270789Sahrens 	 * Now kick off all the tests that run in parallel.
3271789Sahrens 	 */
3272789Sahrens 	zs->zs_enospc_count = 0;
3273789Sahrens 
3274789Sahrens 	za = umem_zalloc(zopt_threads * sizeof (ztest_args_t), UMEM_NOFAIL);
3275789Sahrens 
3276789Sahrens 	if (zopt_verbose >= 4)
3277789Sahrens 		(void) printf("starting main threads...\n");
3278789Sahrens 
3279789Sahrens 	za[0].za_start = gethrtime();
3280789Sahrens 	za[0].za_stop = za[0].za_start + zopt_passtime * NANOSEC;
3281789Sahrens 	za[0].za_stop = MIN(za[0].za_stop, zs->zs_stop_time);
3282789Sahrens 	za[0].za_kill = za[0].za_stop;
3283789Sahrens 	if (ztest_random(100) < zopt_killrate)
3284789Sahrens 		za[0].za_kill -= ztest_random(zopt_passtime * NANOSEC);
3285789Sahrens 
3286789Sahrens 	for (t = 0; t < zopt_threads; t++) {
32871732Sbonwick 		d = t % zopt_datasets;
32885530Sbonwick 
32895530Sbonwick 		(void) strcpy(za[t].za_pool, pool);
32905530Sbonwick 		za[t].za_os = za[d].za_os;
32915530Sbonwick 		za[t].za_spa = spa;
32925530Sbonwick 		za[t].za_zilog = za[d].za_zilog;
32935530Sbonwick 		za[t].za_instance = t;
32945530Sbonwick 		za[t].za_random = ztest_random(-1ULL);
32955530Sbonwick 		za[t].za_start = za[0].za_start;
32965530Sbonwick 		za[t].za_stop = za[0].za_stop;
32975530Sbonwick 		za[t].za_kill = za[0].za_kill;
32985530Sbonwick 
32991732Sbonwick 		if (t < zopt_datasets) {
33003711Smaybee 			int test_future = FALSE;
3301789Sahrens 			(void) rw_rdlock(&ztest_shared->zs_name_lock);
3302789Sahrens 			(void) snprintf(name, 100, "%s/%s_%d", pool, pool, d);
33036492Stimh 			error = dmu_objset_create(name, DMU_OST_OTHER, NULL, 0,
3304789Sahrens 			    ztest_create_cb, NULL);
33053711Smaybee 			if (error == EEXIST) {
33063711Smaybee 				test_future = TRUE;
33075530Sbonwick 			} else if (error == ENOSPC) {
33085530Sbonwick 				zs->zs_enospc_count++;
33095530Sbonwick 				(void) rw_unlock(&ztest_shared->zs_name_lock);
33105530Sbonwick 				break;
33113711Smaybee 			} else if (error != 0) {
3312789Sahrens 				fatal(0, "dmu_objset_create(%s) = %d",
3313789Sahrens 				    name, error);
3314789Sahrens 			}
3315789Sahrens 			error = dmu_objset_open(name, DMU_OST_OTHER,
33166689Smaybee 			    DS_MODE_USER, &za[d].za_os);
3317789Sahrens 			if (error)
3318789Sahrens 				fatal(0, "dmu_objset_open('%s') = %d",
3319789Sahrens 				    name, error);
3320789Sahrens 			(void) rw_unlock(&ztest_shared->zs_name_lock);
33215530Sbonwick 			if (test_future)
33225530Sbonwick 				ztest_dmu_check_future_leak(&za[t]);
33238227SNeil.Perrin@Sun.COM 			zil_replay(za[d].za_os, za[d].za_os,
33248227SNeil.Perrin@Sun.COM 			    ztest_replay_vector);
3325789Sahrens 			za[d].za_zilog = zil_open(za[d].za_os, NULL);
3326789Sahrens 		}
3327789Sahrens 
33287754SJeff.Bonwick@Sun.COM 		VERIFY(thr_create(0, 0, ztest_thread, &za[t], THR_BOUND,
33297754SJeff.Bonwick@Sun.COM 		    &za[t].za_thread) == 0);
3330789Sahrens 	}
3331789Sahrens 
3332789Sahrens 	while (--t >= 0) {
33337754SJeff.Bonwick@Sun.COM 		VERIFY(thr_join(za[t].za_thread, NULL, NULL) == 0);
33341732Sbonwick 		if (t < zopt_datasets) {
3335789Sahrens 			zil_close(za[t].za_zilog);
3336789Sahrens 			dmu_objset_close(za[t].za_os);
3337789Sahrens 		}
3338789Sahrens 	}
3339789Sahrens 
3340789Sahrens 	if (zopt_verbose >= 3)
3341789Sahrens 		show_pool_stats(spa);
3342789Sahrens 
3343789Sahrens 	txg_wait_synced(spa_get_dsl(spa), 0);
3344789Sahrens 
3345789Sahrens 	zs->zs_alloc = spa_get_alloc(spa);
3346789Sahrens 	zs->zs_space = spa_get_space(spa);
3347789Sahrens 
3348789Sahrens 	/*
33495530Sbonwick 	 * If we had out-of-space errors, destroy a random objset.
3350789Sahrens 	 */
3351789Sahrens 	if (zs->zs_enospc_count != 0) {
3352789Sahrens 		(void) rw_rdlock(&ztest_shared->zs_name_lock);
33535530Sbonwick 		d = (int)ztest_random(zopt_datasets);
33545530Sbonwick 		(void) snprintf(name, 100, "%s/%s_%d", pool, pool, d);
3355789Sahrens 		if (zopt_verbose >= 3)
3356789Sahrens 			(void) printf("Destroying %s to free up space\n", name);
33575530Sbonwick 		(void) dmu_objset_find(name, ztest_destroy_cb, &za[d],
33582417Sahrens 		    DS_FIND_SNAPSHOTS | DS_FIND_CHILDREN);
3359789Sahrens 		(void) rw_unlock(&ztest_shared->zs_name_lock);
3360789Sahrens 	}
3361789Sahrens 
33621544Seschrock 	txg_wait_synced(spa_get_dsl(spa), 0);
3363789Sahrens 
33647754SJeff.Bonwick@Sun.COM 	umem_free(za, zopt_threads * sizeof (ztest_args_t));
33657754SJeff.Bonwick@Sun.COM 
33667754SJeff.Bonwick@Sun.COM 	/* Kill the resume thread */
33677754SJeff.Bonwick@Sun.COM 	ztest_exiting = B_TRUE;
33687754SJeff.Bonwick@Sun.COM 	VERIFY(thr_join(resume_tid, NULL, NULL) == 0);
33698241SJeff.Bonwick@Sun.COM 	ztest_resume(spa);
33707754SJeff.Bonwick@Sun.COM 
3371789Sahrens 	/*
3372789Sahrens 	 * Right before closing the pool, kick off a bunch of async I/O;
3373789Sahrens 	 * spa_close() should wait for it to complete.
3374789Sahrens 	 */
3375789Sahrens 	for (t = 1; t < 50; t++)
3376789Sahrens 		dmu_prefetch(spa->spa_meta_objset, t, 0, 1 << 15);
3377789Sahrens 
33785530Sbonwick 	spa_close(spa, FTAG);
33795530Sbonwick 
3380789Sahrens 	kernel_fini();
3381789Sahrens }
3382789Sahrens 
3383789Sahrens void
3384789Sahrens print_time(hrtime_t t, char *timebuf)
3385789Sahrens {
3386789Sahrens 	hrtime_t s = t / NANOSEC;
3387789Sahrens 	hrtime_t m = s / 60;
3388789Sahrens 	hrtime_t h = m / 60;
3389789Sahrens 	hrtime_t d = h / 24;
3390789Sahrens 
3391789Sahrens 	s -= m * 60;
3392789Sahrens 	m -= h * 60;
3393789Sahrens 	h -= d * 24;
3394789Sahrens 
3395789Sahrens 	timebuf[0] = '\0';
3396789Sahrens 
3397789Sahrens 	if (d)
3398789Sahrens 		(void) sprintf(timebuf,
3399789Sahrens 		    "%llud%02lluh%02llum%02llus", d, h, m, s);
3400789Sahrens 	else if (h)
3401789Sahrens 		(void) sprintf(timebuf, "%lluh%02llum%02llus", h, m, s);
3402789Sahrens 	else if (m)
3403789Sahrens 		(void) sprintf(timebuf, "%llum%02llus", m, s);
3404789Sahrens 	else
3405789Sahrens 		(void) sprintf(timebuf, "%llus", s);
3406789Sahrens }
3407789Sahrens 
3408789Sahrens /*
3409789Sahrens  * Create a storage pool with the given name and initial vdev size.
3410789Sahrens  * Then create the specified number of datasets in the pool.
3411789Sahrens  */
3412789Sahrens static void
3413789Sahrens ztest_init(char *pool)
3414789Sahrens {
3415789Sahrens 	spa_t *spa;
3416789Sahrens 	int error;
3417789Sahrens 	nvlist_t *nvroot;
3418789Sahrens 
3419789Sahrens 	kernel_init(FREAD | FWRITE);
3420789Sahrens 
3421789Sahrens 	/*
3422789Sahrens 	 * Create the storage pool.
3423789Sahrens 	 */
3424789Sahrens 	(void) spa_destroy(pool);
3425789Sahrens 	ztest_shared->zs_vdev_primaries = 0;
34267754SJeff.Bonwick@Sun.COM 	nvroot = make_vdev_root(NULL, NULL, zopt_vdev_size, 0,
34277754SJeff.Bonwick@Sun.COM 	    0, zopt_raidz, zopt_mirrors, 1);
34287184Stimh 	error = spa_create(pool, nvroot, NULL, NULL, NULL);
3429789Sahrens 	nvlist_free(nvroot);
3430789Sahrens 
3431789Sahrens 	if (error)
3432789Sahrens 		fatal(0, "spa_create() = %d", error);
3433789Sahrens 	error = spa_open(pool, &spa, FTAG);
3434789Sahrens 	if (error)
3435789Sahrens 		fatal(0, "spa_open() = %d", error);
3436789Sahrens 
3437789Sahrens 	if (zopt_verbose >= 3)
3438789Sahrens 		show_pool_stats(spa);
3439789Sahrens 
3440789Sahrens 	spa_close(spa, FTAG);
3441789Sahrens 
3442789Sahrens 	kernel_fini();
3443789Sahrens }
3444789Sahrens 
3445789Sahrens int
3446789Sahrens main(int argc, char **argv)
3447789Sahrens {
3448789Sahrens 	int kills = 0;
3449789Sahrens 	int iters = 0;
3450789Sahrens 	int i, f;
3451789Sahrens 	ztest_shared_t *zs;
3452789Sahrens 	ztest_info_t *zi;
3453789Sahrens 	char timebuf[100];
3454789Sahrens 	char numbuf[6];
3455789Sahrens 
3456789Sahrens 	(void) setvbuf(stdout, NULL, _IOLBF, 0);
3457789Sahrens 
3458789Sahrens 	/* Override location of zpool.cache */
34596643Seschrock 	spa_config_path = "/tmp/zpool.cache";
3460789Sahrens 
3461789Sahrens 	ztest_random_fd = open("/dev/urandom", O_RDONLY);
3462789Sahrens 
3463789Sahrens 	process_options(argc, argv);
3464789Sahrens 
34651544Seschrock 	/*
34661544Seschrock 	 * Blow away any existing copy of zpool.cache
34671544Seschrock 	 */
34681544Seschrock 	if (zopt_init != 0)
34691544Seschrock 		(void) remove("/tmp/zpool.cache");
34701544Seschrock 
3471789Sahrens 	zs = ztest_shared = (void *)mmap(0,
3472789Sahrens 	    P2ROUNDUP(sizeof (ztest_shared_t), getpagesize()),
3473789Sahrens 	    PROT_READ | PROT_WRITE, MAP_SHARED | MAP_ANON, -1, 0);
3474789Sahrens 
3475789Sahrens 	if (zopt_verbose >= 1) {
3476789Sahrens 		(void) printf("%llu vdevs, %d datasets, %d threads,"
3477789Sahrens 		    " %llu seconds...\n",
34781732Sbonwick 		    (u_longlong_t)zopt_vdevs, zopt_datasets, zopt_threads,
3479789Sahrens 		    (u_longlong_t)zopt_time);
3480789Sahrens 	}
3481789Sahrens 
3482789Sahrens 	/*
3483789Sahrens 	 * Create and initialize our storage pool.
3484789Sahrens 	 */
3485789Sahrens 	for (i = 1; i <= zopt_init; i++) {
3486789Sahrens 		bzero(zs, sizeof (ztest_shared_t));
3487789Sahrens 		if (zopt_verbose >= 3 && zopt_init != 1)
3488789Sahrens 			(void) printf("ztest_init(), pass %d\n", i);
3489789Sahrens 		ztest_init(zopt_pool);
3490789Sahrens 	}
3491789Sahrens 
3492789Sahrens 	/*
3493789Sahrens 	 * Initialize the call targets for each function.
3494789Sahrens 	 */
3495789Sahrens 	for (f = 0; f < ZTEST_FUNCS; f++) {
3496789Sahrens 		zi = &zs->zs_info[f];
3497789Sahrens 
3498789Sahrens 		*zi = ztest_info[f];
3499789Sahrens 
3500789Sahrens 		if (*zi->zi_interval == 0)
3501789Sahrens 			zi->zi_call_target = UINT64_MAX;
3502789Sahrens 		else
3503789Sahrens 			zi->zi_call_target = zopt_time / *zi->zi_interval;
3504789Sahrens 	}
3505789Sahrens 
3506789Sahrens 	zs->zs_start_time = gethrtime();
3507789Sahrens 	zs->zs_stop_time = zs->zs_start_time + zopt_time * NANOSEC;
3508789Sahrens 
3509789Sahrens 	/*
3510789Sahrens 	 * Run the tests in a loop.  These tests include fault injection
3511789Sahrens 	 * to verify that self-healing data works, and forced crashes
3512789Sahrens 	 * to verify that we never lose on-disk consistency.
3513789Sahrens 	 */
3514789Sahrens 	while (gethrtime() < zs->zs_stop_time) {
3515789Sahrens 		int status;
3516789Sahrens 		pid_t pid;
3517789Sahrens 		char *tmp;
3518789Sahrens 
3519789Sahrens 		/*
3520789Sahrens 		 * Initialize the workload counters for each function.
3521789Sahrens 		 */
3522789Sahrens 		for (f = 0; f < ZTEST_FUNCS; f++) {
3523789Sahrens 			zi = &zs->zs_info[f];
3524789Sahrens 			zi->zi_calls = 0;
3525789Sahrens 			zi->zi_call_time = 0;
3526789Sahrens 		}
3527789Sahrens 
3528789Sahrens 		pid = fork();
3529789Sahrens 
3530789Sahrens 		if (pid == -1)
3531789Sahrens 			fatal(1, "fork failed");
3532789Sahrens 
3533789Sahrens 		if (pid == 0) {	/* child */
3534789Sahrens 			struct rlimit rl = { 1024, 1024 };
3535789Sahrens 			(void) setrlimit(RLIMIT_NOFILE, &rl);
35361914Scasper 			(void) enable_extended_FILE_stdio(-1, -1);
3537789Sahrens 			ztest_run(zopt_pool);
3538789Sahrens 			exit(0);
3539789Sahrens 		}
3540789Sahrens 
35412856Snd150628 		while (waitpid(pid, &status, 0) != pid)
3542789Sahrens 			continue;
3543789Sahrens 
3544789Sahrens 		if (WIFEXITED(status)) {
3545789Sahrens 			if (WEXITSTATUS(status) != 0) {
3546789Sahrens 				(void) fprintf(stderr,
3547789Sahrens 				    "child exited with code %d\n",
3548789Sahrens 				    WEXITSTATUS(status));
3549789Sahrens 				exit(2);
3550789Sahrens 			}
35512856Snd150628 		} else if (WIFSIGNALED(status)) {
3552789Sahrens 			if (WTERMSIG(status) != SIGKILL) {
3553789Sahrens 				(void) fprintf(stderr,
3554789Sahrens 				    "child died with signal %d\n",
3555789Sahrens 				    WTERMSIG(status));
3556789Sahrens 				exit(3);
3557789Sahrens 			}
3558789Sahrens 			kills++;
35592856Snd150628 		} else {
35602856Snd150628 			(void) fprintf(stderr, "something strange happened "
35612856Snd150628 			    "to child\n");
35622856Snd150628 			exit(4);
3563789Sahrens 		}
3564789Sahrens 
3565789Sahrens 		iters++;
3566789Sahrens 
3567789Sahrens 		if (zopt_verbose >= 1) {
3568789Sahrens 			hrtime_t now = gethrtime();
3569789Sahrens 
3570789Sahrens 			now = MIN(now, zs->zs_stop_time);
3571789Sahrens 			print_time(zs->zs_stop_time - now, timebuf);
3572789Sahrens 			nicenum(zs->zs_space, numbuf);
3573789Sahrens 
3574789Sahrens 			(void) printf("Pass %3d, %8s, %3llu ENOSPC, "
3575789Sahrens 			    "%4.1f%% of %5s used, %3.0f%% done, %8s to go\n",
3576789Sahrens 			    iters,
3577789Sahrens 			    WIFEXITED(status) ? "Complete" : "SIGKILL",
3578789Sahrens 			    (u_longlong_t)zs->zs_enospc_count,
3579789Sahrens 			    100.0 * zs->zs_alloc / zs->zs_space,
3580789Sahrens 			    numbuf,
3581789Sahrens 			    100.0 * (now - zs->zs_start_time) /
3582789Sahrens 			    (zopt_time * NANOSEC), timebuf);
3583789Sahrens 		}
3584789Sahrens 
3585789Sahrens 		if (zopt_verbose >= 2) {
3586789Sahrens 			(void) printf("\nWorkload summary:\n\n");
3587789Sahrens 			(void) printf("%7s %9s   %s\n",
3588789Sahrens 			    "Calls", "Time", "Function");
3589789Sahrens 			(void) printf("%7s %9s   %s\n",
3590789Sahrens 			    "-----", "----", "--------");
3591789Sahrens 			for (f = 0; f < ZTEST_FUNCS; f++) {
3592789Sahrens 				Dl_info dli;
3593789Sahrens 
3594789Sahrens 				zi = &zs->zs_info[f];
3595789Sahrens 				print_time(zi->zi_call_time, timebuf);
3596789Sahrens 				(void) dladdr((void *)zi->zi_func, &dli);
3597789Sahrens 				(void) printf("%7llu %9s   %s\n",
3598789Sahrens 				    (u_longlong_t)zi->zi_calls, timebuf,
3599789Sahrens 				    dli.dli_sname);
3600789Sahrens 			}
3601789Sahrens 			(void) printf("\n");
3602789Sahrens 		}
3603789Sahrens 
3604789Sahrens 		/*
3605789Sahrens 		 * It's possible that we killed a child during a rename test, in
3606789Sahrens 		 * which case we'll have a 'ztest_tmp' pool lying around instead
3607789Sahrens 		 * of 'ztest'.  Do a blind rename in case this happened.
3608789Sahrens 		 */
3609789Sahrens 		tmp = umem_alloc(strlen(zopt_pool) + 5, UMEM_NOFAIL);
3610789Sahrens 		(void) strcpy(tmp, zopt_pool);
3611789Sahrens 		(void) strcat(tmp, "_tmp");
3612789Sahrens 		kernel_init(FREAD | FWRITE);
3613789Sahrens 		(void) spa_rename(tmp, zopt_pool);
3614789Sahrens 		kernel_fini();
3615789Sahrens 		umem_free(tmp, strlen(tmp) + 1);
3616789Sahrens 	}
3617789Sahrens 
3618789Sahrens 	ztest_verify_blocks(zopt_pool);
3619789Sahrens 
3620789Sahrens 	if (zopt_verbose >= 1) {
3621789Sahrens 		(void) printf("%d killed, %d completed, %.0f%% kill rate\n",
3622789Sahrens 		    kills, iters - kills, (100.0 * kills) / MAX(1, iters));
3623789Sahrens 	}
3624789Sahrens 
3625789Sahrens 	return (0);
3626789Sahrens }
3627