xref: /onnv-gate/usr/src/cmd/ztest/ztest.c (revision 9480)
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>
799412SAleksandr.Guzovskiy@Sun.COM #include <sys/dbuf.h>
80789Sahrens #include <sys/zap.h>
81789Sahrens #include <sys/dmu_objset.h>
82789Sahrens #include <sys/poll.h>
83789Sahrens #include <sys/stat.h>
84789Sahrens #include <sys/time.h>
85789Sahrens #include <sys/wait.h>
86789Sahrens #include <sys/mman.h>
87789Sahrens #include <sys/resource.h>
88789Sahrens #include <sys/zio.h>
89789Sahrens #include <sys/zio_checksum.h>
90789Sahrens #include <sys/zio_compress.h>
91789Sahrens #include <sys/zil.h>
92789Sahrens #include <sys/vdev_impl.h>
937754SJeff.Bonwick@Sun.COM #include <sys/vdev_file.h>
94789Sahrens #include <sys/spa_impl.h>
95789Sahrens #include <sys/dsl_prop.h>
968779SMark.Musante@Sun.COM #include <sys/dsl_dataset.h>
97789Sahrens #include <sys/refcount.h>
98789Sahrens #include <stdio.h>
991914Scasper #include <stdio_ext.h>
100789Sahrens #include <stdlib.h>
101789Sahrens #include <unistd.h>
102789Sahrens #include <signal.h>
103789Sahrens #include <umem.h>
104789Sahrens #include <dlfcn.h>
105789Sahrens #include <ctype.h>
106789Sahrens #include <math.h>
107789Sahrens #include <sys/fs/zfs.h>
108789Sahrens 
109789Sahrens static char cmdname[] = "ztest";
110789Sahrens static char *zopt_pool = cmdname;
111789Sahrens 
112789Sahrens static uint64_t zopt_vdevs = 5;
113789Sahrens static uint64_t zopt_vdevtime;
1141732Sbonwick static int zopt_ashift = SPA_MINBLOCKSHIFT;
115789Sahrens static int zopt_mirrors = 2;
116789Sahrens static int zopt_raidz = 4;
1172082Seschrock static int zopt_raidz_parity = 1;
118789Sahrens static size_t zopt_vdev_size = SPA_MINDEVSIZE;
1191732Sbonwick static int zopt_datasets = 7;
120789Sahrens static int zopt_threads = 23;
121789Sahrens static uint64_t zopt_passtime = 60;	/* 60 seconds */
122789Sahrens static uint64_t zopt_killrate = 70;	/* 70% kill rate */
123789Sahrens static int zopt_verbose = 0;
124789Sahrens static int zopt_init = 1;
125789Sahrens static char *zopt_dir = "/tmp";
126789Sahrens static uint64_t zopt_time = 300;	/* 5 minutes */
127789Sahrens static int zopt_maxfaults;
128789Sahrens 
1295530Sbonwick typedef struct ztest_block_tag {
1305530Sbonwick 	uint64_t	bt_objset;
1315530Sbonwick 	uint64_t	bt_object;
1325530Sbonwick 	uint64_t	bt_offset;
1335530Sbonwick 	uint64_t	bt_txg;
1345530Sbonwick 	uint64_t	bt_thread;
1355530Sbonwick 	uint64_t	bt_seq;
1365530Sbonwick } ztest_block_tag_t;
1375530Sbonwick 
138789Sahrens typedef struct ztest_args {
1395530Sbonwick 	char		za_pool[MAXNAMELEN];
1405530Sbonwick 	spa_t		*za_spa;
141789Sahrens 	objset_t	*za_os;
142789Sahrens 	zilog_t		*za_zilog;
143789Sahrens 	thread_t	za_thread;
144789Sahrens 	uint64_t	za_instance;
145789Sahrens 	uint64_t	za_random;
146789Sahrens 	uint64_t	za_diroff;
147789Sahrens 	uint64_t	za_diroff_shared;
1481544Seschrock 	uint64_t	za_zil_seq;
149789Sahrens 	hrtime_t	za_start;
150789Sahrens 	hrtime_t	za_stop;
151789Sahrens 	hrtime_t	za_kill;
1525530Sbonwick 	/*
1535530Sbonwick 	 * Thread-local variables can go here to aid debugging.
1545530Sbonwick 	 */
1555530Sbonwick 	ztest_block_tag_t za_rbt;
1565530Sbonwick 	ztest_block_tag_t za_wbt;
1575530Sbonwick 	dmu_object_info_t za_doi;
1585530Sbonwick 	dmu_buf_t	*za_dbuf;
159789Sahrens } ztest_args_t;
160789Sahrens 
161789Sahrens typedef void ztest_func_t(ztest_args_t *);
162789Sahrens 
163789Sahrens /*
164789Sahrens  * Note: these aren't static because we want dladdr() to work.
165789Sahrens  */
166789Sahrens ztest_func_t ztest_dmu_read_write;
1679412SAleksandr.Guzovskiy@Sun.COM ztest_func_t ztest_dmu_read_write_zcopy;
168789Sahrens ztest_func_t ztest_dmu_write_parallel;
169789Sahrens ztest_func_t ztest_dmu_object_alloc_free;
170789Sahrens ztest_func_t ztest_zap;
171789Sahrens ztest_func_t ztest_zap_parallel;
172789Sahrens ztest_func_t ztest_traverse;
173789Sahrens ztest_func_t ztest_dsl_prop_get_set;
174789Sahrens ztest_func_t ztest_dmu_objset_create_destroy;
175789Sahrens ztest_func_t ztest_dmu_snapshot_create_destroy;
1768779SMark.Musante@Sun.COM ztest_func_t ztest_dsl_dataset_promote_busy;
177789Sahrens ztest_func_t ztest_spa_create_destroy;
178789Sahrens ztest_func_t ztest_fault_inject;
1797754SJeff.Bonwick@Sun.COM ztest_func_t ztest_spa_rename;
180789Sahrens ztest_func_t ztest_vdev_attach_detach;
181789Sahrens ztest_func_t ztest_vdev_LUN_growth;
182789Sahrens ztest_func_t ztest_vdev_add_remove;
1837754SJeff.Bonwick@Sun.COM ztest_func_t ztest_vdev_aux_add_remove;
184789Sahrens ztest_func_t ztest_scrub;
185789Sahrens 
186789Sahrens typedef struct ztest_info {
187789Sahrens 	ztest_func_t	*zi_func;	/* test function */
1885530Sbonwick 	uint64_t	zi_iters;	/* iterations per execution */
189789Sahrens 	uint64_t	*zi_interval;	/* execute every <interval> seconds */
190789Sahrens 	uint64_t	zi_calls;	/* per-pass count */
191789Sahrens 	uint64_t	zi_call_time;	/* per-pass time */
192789Sahrens 	uint64_t	zi_call_total;	/* cumulative total */
193789Sahrens 	uint64_t	zi_call_target;	/* target cumulative total */
194789Sahrens } ztest_info_t;
195789Sahrens 
196789Sahrens uint64_t zopt_always = 0;		/* all the time */
197789Sahrens uint64_t zopt_often = 1;		/* every second */
198789Sahrens uint64_t zopt_sometimes = 10;		/* every 10 seconds */
199789Sahrens uint64_t zopt_rarely = 60;		/* every 60 seconds */
200789Sahrens 
201789Sahrens ztest_info_t ztest_info[] = {
2025530Sbonwick 	{ ztest_dmu_read_write,			1,	&zopt_always	},
2039412SAleksandr.Guzovskiy@Sun.COM 	{ ztest_dmu_read_write_zcopy,		1,	&zopt_always	},
2045530Sbonwick 	{ ztest_dmu_write_parallel,		30,	&zopt_always	},
2055530Sbonwick 	{ ztest_dmu_object_alloc_free,		1,	&zopt_always	},
2065530Sbonwick 	{ ztest_zap,				30,	&zopt_always	},
2075530Sbonwick 	{ ztest_zap_parallel,			100,	&zopt_always	},
2085530Sbonwick 	{ ztest_dsl_prop_get_set,		1,	&zopt_sometimes	},
2097046Sahrens 	{ ztest_dmu_objset_create_destroy,	1,	&zopt_sometimes },
2107754SJeff.Bonwick@Sun.COM 	{ ztest_dmu_snapshot_create_destroy,	1,	&zopt_sometimes },
2117046Sahrens 	{ ztest_spa_create_destroy,		1,	&zopt_sometimes },
2125530Sbonwick 	{ ztest_fault_inject,			1,	&zopt_sometimes	},
2135530Sbonwick 	{ ztest_spa_rename,			1,	&zopt_rarely	},
2147754SJeff.Bonwick@Sun.COM 	{ ztest_vdev_attach_detach,		1,	&zopt_rarely	},
2157754SJeff.Bonwick@Sun.COM 	{ ztest_vdev_LUN_growth,		1,	&zopt_rarely	},
2169229SGeorge.Wilson@Sun.COM 	{ ztest_dsl_dataset_promote_busy,	1,	&zopt_rarely	},
2177754SJeff.Bonwick@Sun.COM 	{ ztest_vdev_add_remove,		1,	&zopt_vdevtime	},
2187754SJeff.Bonwick@Sun.COM 	{ ztest_vdev_aux_add_remove,		1,	&zopt_vdevtime	},
2195530Sbonwick 	{ ztest_scrub,				1,	&zopt_vdevtime	},
220789Sahrens };
221789Sahrens 
222789Sahrens #define	ZTEST_FUNCS	(sizeof (ztest_info) / sizeof (ztest_info_t))
223789Sahrens 
224789Sahrens #define	ZTEST_SYNC_LOCKS	16
225789Sahrens 
226789Sahrens /*
227789Sahrens  * Stuff we need to share writably between parent and child.
228789Sahrens  */
229789Sahrens typedef struct ztest_shared {
230789Sahrens 	mutex_t		zs_vdev_lock;
231789Sahrens 	rwlock_t	zs_name_lock;
232789Sahrens 	uint64_t	zs_vdev_primaries;
2337754SJeff.Bonwick@Sun.COM 	uint64_t	zs_vdev_aux;
234789Sahrens 	uint64_t	zs_enospc_count;
235789Sahrens 	hrtime_t	zs_start_time;
236789Sahrens 	hrtime_t	zs_stop_time;
237789Sahrens 	uint64_t	zs_alloc;
238789Sahrens 	uint64_t	zs_space;
239789Sahrens 	ztest_info_t	zs_info[ZTEST_FUNCS];
240789Sahrens 	mutex_t		zs_sync_lock[ZTEST_SYNC_LOCKS];
241789Sahrens 	uint64_t	zs_seq[ZTEST_SYNC_LOCKS];
242789Sahrens } ztest_shared_t;
243789Sahrens 
244789Sahrens static char ztest_dev_template[] = "%s/%s.%llua";
2457754SJeff.Bonwick@Sun.COM static char ztest_aux_template[] = "%s/%s.%s.%llu";
246789Sahrens static ztest_shared_t *ztest_shared;
247789Sahrens 
248789Sahrens static int ztest_random_fd;
249789Sahrens static int ztest_dump_core = 1;
250789Sahrens 
251*9480SGeorge.Wilson@Sun.COM static uint64_t metaslab_sz;
2527754SJeff.Bonwick@Sun.COM static boolean_t ztest_exiting;
2535329Sgw25295 
2545530Sbonwick extern uint64_t metaslab_gang_bang;
255*9480SGeorge.Wilson@Sun.COM extern uint64_t metaslab_df_alloc_threshold;
256789Sahrens 
257789Sahrens #define	ZTEST_DIROBJ		1
258789Sahrens #define	ZTEST_MICROZAP_OBJ	2
259789Sahrens #define	ZTEST_FATZAP_OBJ	3
260789Sahrens 
261789Sahrens #define	ZTEST_DIROBJ_BLOCKSIZE	(1 << 10)
262789Sahrens #define	ZTEST_DIRSIZE		256
263789Sahrens 
2644008Sraf static void usage(boolean_t) __NORETURN;
2653972Svb160487 
266789Sahrens /*
267789Sahrens  * These libumem hooks provide a reasonable set of defaults for the allocator's
268789Sahrens  * debugging facilities.
269789Sahrens  */
270789Sahrens const char *
271789Sahrens _umem_debug_init()
272789Sahrens {
273789Sahrens 	return ("default,verbose"); /* $UMEM_DEBUG setting */
274789Sahrens }
275789Sahrens 
276789Sahrens const char *
277789Sahrens _umem_logging_init(void)
278789Sahrens {
279789Sahrens 	return ("fail,contents"); /* $UMEM_LOGGING setting */
280789Sahrens }
281789Sahrens 
282789Sahrens #define	FATAL_MSG_SZ	1024
283789Sahrens 
284789Sahrens char *fatal_msg;
285789Sahrens 
286789Sahrens static void
287789Sahrens fatal(int do_perror, char *message, ...)
288789Sahrens {
289789Sahrens 	va_list args;
290789Sahrens 	int save_errno = errno;
291789Sahrens 	char buf[FATAL_MSG_SZ];
292789Sahrens 
293789Sahrens 	(void) fflush(stdout);
294789Sahrens 
295789Sahrens 	va_start(args, message);
296789Sahrens 	(void) sprintf(buf, "ztest: ");
297789Sahrens 	/* LINTED */
298789Sahrens 	(void) vsprintf(buf + strlen(buf), message, args);
299789Sahrens 	va_end(args);
300789Sahrens 	if (do_perror) {
301789Sahrens 		(void) snprintf(buf + strlen(buf), FATAL_MSG_SZ - strlen(buf),
302789Sahrens 		    ": %s", strerror(save_errno));
303789Sahrens 	}
304789Sahrens 	(void) fprintf(stderr, "%s\n", buf);
305789Sahrens 	fatal_msg = buf;			/* to ease debugging */
306789Sahrens 	if (ztest_dump_core)
307789Sahrens 		abort();
308789Sahrens 	exit(3);
309789Sahrens }
310789Sahrens 
311789Sahrens static int
312789Sahrens str2shift(const char *buf)
313789Sahrens {
314789Sahrens 	const char *ends = "BKMGTPEZ";
315789Sahrens 	int i;
316789Sahrens 
317789Sahrens 	if (buf[0] == '\0')
318789Sahrens 		return (0);
319789Sahrens 	for (i = 0; i < strlen(ends); i++) {
320789Sahrens 		if (toupper(buf[0]) == ends[i])
321789Sahrens 			break;
322789Sahrens 	}
3233972Svb160487 	if (i == strlen(ends)) {
3243972Svb160487 		(void) fprintf(stderr, "ztest: invalid bytes suffix: %s\n",
3253972Svb160487 		    buf);
3263972Svb160487 		usage(B_FALSE);
3273972Svb160487 	}
328789Sahrens 	if (buf[1] == '\0' || (toupper(buf[1]) == 'B' && buf[2] == '\0')) {
329789Sahrens 		return (10*i);
330789Sahrens 	}
3313972Svb160487 	(void) fprintf(stderr, "ztest: invalid bytes suffix: %s\n", buf);
3323972Svb160487 	usage(B_FALSE);
3333972Svb160487 	/* NOTREACHED */
334789Sahrens }
335789Sahrens 
336789Sahrens static uint64_t
337789Sahrens nicenumtoull(const char *buf)
338789Sahrens {
339789Sahrens 	char *end;
340789Sahrens 	uint64_t val;
341789Sahrens 
342789Sahrens 	val = strtoull(buf, &end, 0);
343789Sahrens 	if (end == buf) {
3443972Svb160487 		(void) fprintf(stderr, "ztest: bad numeric value: %s\n", buf);
3453972Svb160487 		usage(B_FALSE);
346789Sahrens 	} else if (end[0] == '.') {
347789Sahrens 		double fval = strtod(buf, &end);
348789Sahrens 		fval *= pow(2, str2shift(end));
3493972Svb160487 		if (fval > UINT64_MAX) {
3503972Svb160487 			(void) fprintf(stderr, "ztest: value too large: %s\n",
3513972Svb160487 			    buf);
3523972Svb160487 			usage(B_FALSE);
3533972Svb160487 		}
354789Sahrens 		val = (uint64_t)fval;
355789Sahrens 	} else {
356789Sahrens 		int shift = str2shift(end);
3573972Svb160487 		if (shift >= 64 || (val << shift) >> shift != val) {
3583972Svb160487 			(void) fprintf(stderr, "ztest: value too large: %s\n",
3593972Svb160487 			    buf);
3603972Svb160487 			usage(B_FALSE);
3613972Svb160487 		}
362789Sahrens 		val <<= shift;
363789Sahrens 	}
364789Sahrens 	return (val);
365789Sahrens }
366789Sahrens 
367789Sahrens static void
3683972Svb160487 usage(boolean_t requested)
369789Sahrens {
370789Sahrens 	char nice_vdev_size[10];
371789Sahrens 	char nice_gang_bang[10];
3723972Svb160487 	FILE *fp = requested ? stdout : stderr;
373789Sahrens 
374789Sahrens 	nicenum(zopt_vdev_size, nice_vdev_size);
3755530Sbonwick 	nicenum(metaslab_gang_bang, nice_gang_bang);
376789Sahrens 
3773972Svb160487 	(void) fprintf(fp, "Usage: %s\n"
378789Sahrens 	    "\t[-v vdevs (default: %llu)]\n"
379789Sahrens 	    "\t[-s size_of_each_vdev (default: %s)]\n"
3801732Sbonwick 	    "\t[-a alignment_shift (default: %d) (use 0 for random)]\n"
381789Sahrens 	    "\t[-m mirror_copies (default: %d)]\n"
382789Sahrens 	    "\t[-r raidz_disks (default: %d)]\n"
3832082Seschrock 	    "\t[-R raidz_parity (default: %d)]\n"
384789Sahrens 	    "\t[-d datasets (default: %d)]\n"
385789Sahrens 	    "\t[-t threads (default: %d)]\n"
386789Sahrens 	    "\t[-g gang_block_threshold (default: %s)]\n"
387789Sahrens 	    "\t[-i initialize pool i times (default: %d)]\n"
388789Sahrens 	    "\t[-k kill percentage (default: %llu%%)]\n"
389789Sahrens 	    "\t[-p pool_name (default: %s)]\n"
390789Sahrens 	    "\t[-f file directory for vdev files (default: %s)]\n"
391789Sahrens 	    "\t[-V(erbose)] (use multiple times for ever more blather)\n"
3921732Sbonwick 	    "\t[-E(xisting)] (use existing pool instead of creating new one)\n"
393789Sahrens 	    "\t[-T time] total run time (default: %llu sec)\n"
394789Sahrens 	    "\t[-P passtime] time per pass (default: %llu sec)\n"
3953972Svb160487 	    "\t[-h] (print help)\n"
396789Sahrens 	    "",
397789Sahrens 	    cmdname,
3985329Sgw25295 	    (u_longlong_t)zopt_vdevs,			/* -v */
3995329Sgw25295 	    nice_vdev_size,				/* -s */
4005329Sgw25295 	    zopt_ashift,				/* -a */
4015329Sgw25295 	    zopt_mirrors,				/* -m */
4025329Sgw25295 	    zopt_raidz,					/* -r */
4035329Sgw25295 	    zopt_raidz_parity,				/* -R */
4045329Sgw25295 	    zopt_datasets,				/* -d */
4055329Sgw25295 	    zopt_threads,				/* -t */
4065329Sgw25295 	    nice_gang_bang,				/* -g */
4075329Sgw25295 	    zopt_init,					/* -i */
4085329Sgw25295 	    (u_longlong_t)zopt_killrate,		/* -k */
4095329Sgw25295 	    zopt_pool,					/* -p */
4105329Sgw25295 	    zopt_dir,					/* -f */
4115329Sgw25295 	    (u_longlong_t)zopt_time,			/* -T */
4127754SJeff.Bonwick@Sun.COM 	    (u_longlong_t)zopt_passtime);		/* -P */
4133972Svb160487 	exit(requested ? 0 : 1);
414789Sahrens }
415789Sahrens 
416789Sahrens static uint64_t
417789Sahrens ztest_random(uint64_t range)
418789Sahrens {
419789Sahrens 	uint64_t r;
420789Sahrens 
421789Sahrens 	if (range == 0)
422789Sahrens 		return (0);
423789Sahrens 
424789Sahrens 	if (read(ztest_random_fd, &r, sizeof (r)) != sizeof (r))
425789Sahrens 		fatal(1, "short read from /dev/urandom");
426789Sahrens 
427789Sahrens 	return (r % range);
428789Sahrens }
429789Sahrens 
4308241SJeff.Bonwick@Sun.COM /* ARGSUSED */
431789Sahrens static void
432789Sahrens ztest_record_enospc(char *s)
433789Sahrens {
434789Sahrens 	ztest_shared->zs_enospc_count++;
435789Sahrens }
436789Sahrens 
437789Sahrens static void
438789Sahrens process_options(int argc, char **argv)
439789Sahrens {
440789Sahrens 	int opt;
441789Sahrens 	uint64_t value;
442789Sahrens 
443789Sahrens 	/* By default, test gang blocks for blocks 32K and greater */
4445530Sbonwick 	metaslab_gang_bang = 32 << 10;
445789Sahrens 
446789Sahrens 	while ((opt = getopt(argc, argv,
4477754SJeff.Bonwick@Sun.COM 	    "v:s:a:m:r:R:d:t:g:i:k:p:f:VET:P:h")) != EOF) {
448789Sahrens 		value = 0;
449789Sahrens 		switch (opt) {
4504451Seschrock 		case 'v':
4514451Seschrock 		case 's':
4524451Seschrock 		case 'a':
4534451Seschrock 		case 'm':
4544451Seschrock 		case 'r':
4554451Seschrock 		case 'R':
4564451Seschrock 		case 'd':
4574451Seschrock 		case 't':
4584451Seschrock 		case 'g':
4594451Seschrock 		case 'i':
4604451Seschrock 		case 'k':
4614451Seschrock 		case 'T':
4624451Seschrock 		case 'P':
463789Sahrens 			value = nicenumtoull(optarg);
464789Sahrens 		}
465789Sahrens 		switch (opt) {
4664451Seschrock 		case 'v':
467789Sahrens 			zopt_vdevs = value;
468789Sahrens 			break;
4694451Seschrock 		case 's':
470789Sahrens 			zopt_vdev_size = MAX(SPA_MINDEVSIZE, value);
471789Sahrens 			break;
4724451Seschrock 		case 'a':
4731732Sbonwick 			zopt_ashift = value;
4741732Sbonwick 			break;
4754451Seschrock 		case 'm':
476789Sahrens 			zopt_mirrors = value;
477789Sahrens 			break;
4784451Seschrock 		case 'r':
479789Sahrens 			zopt_raidz = MAX(1, value);
480789Sahrens 			break;
4814451Seschrock 		case 'R':
4822082Seschrock 			zopt_raidz_parity = MIN(MAX(value, 1), 2);
4832082Seschrock 			break;
4844451Seschrock 		case 'd':
4851732Sbonwick 			zopt_datasets = MAX(1, value);
486789Sahrens 			break;
4874451Seschrock 		case 't':
488789Sahrens 			zopt_threads = MAX(1, value);
489789Sahrens 			break;
4904451Seschrock 		case 'g':
4915530Sbonwick 			metaslab_gang_bang = MAX(SPA_MINBLOCKSIZE << 1, value);
492789Sahrens 			break;
4934451Seschrock 		case 'i':
494789Sahrens 			zopt_init = value;
495789Sahrens 			break;
4964451Seschrock 		case 'k':
497789Sahrens 			zopt_killrate = value;
498789Sahrens 			break;
4994451Seschrock 		case 'p':
500789Sahrens 			zopt_pool = strdup(optarg);
501789Sahrens 			break;
5024451Seschrock 		case 'f':
503789Sahrens 			zopt_dir = strdup(optarg);
504789Sahrens 			break;
5054451Seschrock 		case 'V':
506789Sahrens 			zopt_verbose++;
507789Sahrens 			break;
5084451Seschrock 		case 'E':
509789Sahrens 			zopt_init = 0;
510789Sahrens 			break;
5114451Seschrock 		case 'T':
512789Sahrens 			zopt_time = value;
513789Sahrens 			break;
5144451Seschrock 		case 'P':
515789Sahrens 			zopt_passtime = MAX(1, value);
516789Sahrens 			break;
5174451Seschrock 		case 'h':
5183972Svb160487 			usage(B_TRUE);
5193972Svb160487 			break;
5204451Seschrock 		case '?':
5214451Seschrock 		default:
5223972Svb160487 			usage(B_FALSE);
523789Sahrens 			break;
524789Sahrens 		}
525789Sahrens 	}
526789Sahrens 
5272082Seschrock 	zopt_raidz_parity = MIN(zopt_raidz_parity, zopt_raidz - 1);
5282082Seschrock 
529789Sahrens 	zopt_vdevtime = (zopt_vdevs > 0 ? zopt_time / zopt_vdevs : UINT64_MAX);
5302082Seschrock 	zopt_maxfaults = MAX(zopt_mirrors, 1) * (zopt_raidz_parity + 1) - 1;
531789Sahrens }
532789Sahrens 
5331732Sbonwick static uint64_t
5341732Sbonwick ztest_get_ashift(void)
5351732Sbonwick {
5361732Sbonwick 	if (zopt_ashift == 0)
5371732Sbonwick 		return (SPA_MINBLOCKSHIFT + ztest_random(3));
5381732Sbonwick 	return (zopt_ashift);
5391732Sbonwick }
5401732Sbonwick 
541789Sahrens static nvlist_t *
5427754SJeff.Bonwick@Sun.COM make_vdev_file(char *path, char *aux, size_t size, uint64_t ashift)
543789Sahrens {
5447754SJeff.Bonwick@Sun.COM 	char pathbuf[MAXPATHLEN];
545789Sahrens 	uint64_t vdev;
546789Sahrens 	nvlist_t *file;
547789Sahrens 
5487754SJeff.Bonwick@Sun.COM 	if (ashift == 0)
5497754SJeff.Bonwick@Sun.COM 		ashift = ztest_get_ashift();
5507754SJeff.Bonwick@Sun.COM 
5517754SJeff.Bonwick@Sun.COM 	if (path == NULL) {
5527754SJeff.Bonwick@Sun.COM 		path = pathbuf;
5537754SJeff.Bonwick@Sun.COM 
5547754SJeff.Bonwick@Sun.COM 		if (aux != NULL) {
5557754SJeff.Bonwick@Sun.COM 			vdev = ztest_shared->zs_vdev_aux;
5567754SJeff.Bonwick@Sun.COM 			(void) sprintf(path, ztest_aux_template,
5577754SJeff.Bonwick@Sun.COM 			    zopt_dir, zopt_pool, aux, vdev);
5587754SJeff.Bonwick@Sun.COM 		} else {
5597754SJeff.Bonwick@Sun.COM 			vdev = ztest_shared->zs_vdev_primaries++;
5607754SJeff.Bonwick@Sun.COM 			(void) sprintf(path, ztest_dev_template,
5617754SJeff.Bonwick@Sun.COM 			    zopt_dir, zopt_pool, vdev);
5627754SJeff.Bonwick@Sun.COM 		}
5637754SJeff.Bonwick@Sun.COM 	}
5647754SJeff.Bonwick@Sun.COM 
5657754SJeff.Bonwick@Sun.COM 	if (size != 0) {
5667754SJeff.Bonwick@Sun.COM 		int fd = open(path, O_RDWR | O_CREAT | O_TRUNC, 0666);
567789Sahrens 		if (fd == -1)
5687754SJeff.Bonwick@Sun.COM 			fatal(1, "can't open %s", path);
569789Sahrens 		if (ftruncate(fd, size) != 0)
5707754SJeff.Bonwick@Sun.COM 			fatal(1, "can't ftruncate %s", path);
571789Sahrens 		(void) close(fd);
572789Sahrens 	}
573789Sahrens 
574789Sahrens 	VERIFY(nvlist_alloc(&file, NV_UNIQUE_NAME, 0) == 0);
575789Sahrens 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_TYPE, VDEV_TYPE_FILE) == 0);
5767754SJeff.Bonwick@Sun.COM 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_PATH, path) == 0);
5771732Sbonwick 	VERIFY(nvlist_add_uint64(file, ZPOOL_CONFIG_ASHIFT, ashift) == 0);
578789Sahrens 
579789Sahrens 	return (file);
580789Sahrens }
581789Sahrens 
582789Sahrens static nvlist_t *
5837754SJeff.Bonwick@Sun.COM make_vdev_raidz(char *path, char *aux, size_t size, uint64_t ashift, int r)
584789Sahrens {
585789Sahrens 	nvlist_t *raidz, **child;
586789Sahrens 	int c;
587789Sahrens 
588789Sahrens 	if (r < 2)
5897754SJeff.Bonwick@Sun.COM 		return (make_vdev_file(path, aux, size, ashift));
590789Sahrens 	child = umem_alloc(r * sizeof (nvlist_t *), UMEM_NOFAIL);
591789Sahrens 
592789Sahrens 	for (c = 0; c < r; c++)
5937754SJeff.Bonwick@Sun.COM 		child[c] = make_vdev_file(path, aux, size, ashift);
594789Sahrens 
595789Sahrens 	VERIFY(nvlist_alloc(&raidz, NV_UNIQUE_NAME, 0) == 0);
596789Sahrens 	VERIFY(nvlist_add_string(raidz, ZPOOL_CONFIG_TYPE,
597789Sahrens 	    VDEV_TYPE_RAIDZ) == 0);
5982082Seschrock 	VERIFY(nvlist_add_uint64(raidz, ZPOOL_CONFIG_NPARITY,
5992082Seschrock 	    zopt_raidz_parity) == 0);
600789Sahrens 	VERIFY(nvlist_add_nvlist_array(raidz, ZPOOL_CONFIG_CHILDREN,
601789Sahrens 	    child, r) == 0);
602789Sahrens 
603789Sahrens 	for (c = 0; c < r; c++)
604789Sahrens 		nvlist_free(child[c]);
605789Sahrens 
606789Sahrens 	umem_free(child, r * sizeof (nvlist_t *));
607789Sahrens 
608789Sahrens 	return (raidz);
609789Sahrens }
610789Sahrens 
611789Sahrens static nvlist_t *
6127754SJeff.Bonwick@Sun.COM make_vdev_mirror(char *path, char *aux, size_t size, uint64_t ashift,
6137768SJeff.Bonwick@Sun.COM 	int r, int m)
614789Sahrens {
615789Sahrens 	nvlist_t *mirror, **child;
616789Sahrens 	int c;
617789Sahrens 
618789Sahrens 	if (m < 1)
6197754SJeff.Bonwick@Sun.COM 		return (make_vdev_raidz(path, aux, size, ashift, r));
620789Sahrens 
621789Sahrens 	child = umem_alloc(m * sizeof (nvlist_t *), UMEM_NOFAIL);
622789Sahrens 
623789Sahrens 	for (c = 0; c < m; c++)
6247754SJeff.Bonwick@Sun.COM 		child[c] = make_vdev_raidz(path, aux, size, ashift, r);
625789Sahrens 
626789Sahrens 	VERIFY(nvlist_alloc(&mirror, NV_UNIQUE_NAME, 0) == 0);
627789Sahrens 	VERIFY(nvlist_add_string(mirror, ZPOOL_CONFIG_TYPE,
628789Sahrens 	    VDEV_TYPE_MIRROR) == 0);
629789Sahrens 	VERIFY(nvlist_add_nvlist_array(mirror, ZPOOL_CONFIG_CHILDREN,
630789Sahrens 	    child, m) == 0);
631789Sahrens 
632789Sahrens 	for (c = 0; c < m; c++)
633789Sahrens 		nvlist_free(child[c]);
634789Sahrens 
635789Sahrens 	umem_free(child, m * sizeof (nvlist_t *));
636789Sahrens 
637789Sahrens 	return (mirror);
638789Sahrens }
639789Sahrens 
640789Sahrens static nvlist_t *
6417754SJeff.Bonwick@Sun.COM make_vdev_root(char *path, char *aux, size_t size, uint64_t ashift,
6427754SJeff.Bonwick@Sun.COM 	int log, int r, int m, int t)
643789Sahrens {
644789Sahrens 	nvlist_t *root, **child;
645789Sahrens 	int c;
646789Sahrens 
647789Sahrens 	ASSERT(t > 0);
648789Sahrens 
649789Sahrens 	child = umem_alloc(t * sizeof (nvlist_t *), UMEM_NOFAIL);
650789Sahrens 
6517768SJeff.Bonwick@Sun.COM 	for (c = 0; c < t; c++) {
6527768SJeff.Bonwick@Sun.COM 		child[c] = make_vdev_mirror(path, aux, size, ashift, r, m);
6537768SJeff.Bonwick@Sun.COM 		VERIFY(nvlist_add_uint64(child[c], ZPOOL_CONFIG_IS_LOG,
6547768SJeff.Bonwick@Sun.COM 		    log) == 0);
6557768SJeff.Bonwick@Sun.COM 	}
656789Sahrens 
657789Sahrens 	VERIFY(nvlist_alloc(&root, NV_UNIQUE_NAME, 0) == 0);
658789Sahrens 	VERIFY(nvlist_add_string(root, ZPOOL_CONFIG_TYPE, VDEV_TYPE_ROOT) == 0);
6597754SJeff.Bonwick@Sun.COM 	VERIFY(nvlist_add_nvlist_array(root, aux ? aux : ZPOOL_CONFIG_CHILDREN,
660789Sahrens 	    child, t) == 0);
661789Sahrens 
662789Sahrens 	for (c = 0; c < t; c++)
663789Sahrens 		nvlist_free(child[c]);
664789Sahrens 
665789Sahrens 	umem_free(child, t * sizeof (nvlist_t *));
666789Sahrens 
667789Sahrens 	return (root);
668789Sahrens }
669789Sahrens 
670789Sahrens static void
671789Sahrens ztest_set_random_blocksize(objset_t *os, uint64_t object, dmu_tx_t *tx)
672789Sahrens {
673789Sahrens 	int bs = SPA_MINBLOCKSHIFT +
674789Sahrens 	    ztest_random(SPA_MAXBLOCKSHIFT - SPA_MINBLOCKSHIFT + 1);
675789Sahrens 	int ibs = DN_MIN_INDBLKSHIFT +
676789Sahrens 	    ztest_random(DN_MAX_INDBLKSHIFT - DN_MIN_INDBLKSHIFT + 1);
677789Sahrens 	int error;
678789Sahrens 
679789Sahrens 	error = dmu_object_set_blocksize(os, object, 1ULL << bs, ibs, tx);
680789Sahrens 	if (error) {
681789Sahrens 		char osname[300];
682789Sahrens 		dmu_objset_name(os, osname);
683789Sahrens 		fatal(0, "dmu_object_set_blocksize('%s', %llu, %d, %d) = %d",
684789Sahrens 		    osname, object, 1 << bs, ibs, error);
685789Sahrens 	}
686789Sahrens }
687789Sahrens 
688789Sahrens static uint8_t
689789Sahrens ztest_random_checksum(void)
690789Sahrens {
691789Sahrens 	uint8_t checksum;
692789Sahrens 
693789Sahrens 	do {
694789Sahrens 		checksum = ztest_random(ZIO_CHECKSUM_FUNCTIONS);
695789Sahrens 	} while (zio_checksum_table[checksum].ci_zbt);
696789Sahrens 
697789Sahrens 	if (checksum == ZIO_CHECKSUM_OFF)
698789Sahrens 		checksum = ZIO_CHECKSUM_ON;
699789Sahrens 
700789Sahrens 	return (checksum);
701789Sahrens }
702789Sahrens 
703789Sahrens static uint8_t
704789Sahrens ztest_random_compress(void)
705789Sahrens {
706789Sahrens 	return ((uint8_t)ztest_random(ZIO_COMPRESS_FUNCTIONS));
707789Sahrens }
708789Sahrens 
709789Sahrens static int
7108227SNeil.Perrin@Sun.COM ztest_replay_create(objset_t *os, lr_create_t *lr, boolean_t byteswap)
711789Sahrens {
712789Sahrens 	dmu_tx_t *tx;
713789Sahrens 	int error;
714789Sahrens 
715789Sahrens 	if (byteswap)
716789Sahrens 		byteswap_uint64_array(lr, sizeof (*lr));
717789Sahrens 
718789Sahrens 	tx = dmu_tx_create(os);
719789Sahrens 	dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
7208227SNeil.Perrin@Sun.COM 	error = dmu_tx_assign(tx, TXG_WAIT);
721789Sahrens 	if (error) {
722789Sahrens 		dmu_tx_abort(tx);
723789Sahrens 		return (error);
724789Sahrens 	}
725789Sahrens 
726789Sahrens 	error = dmu_object_claim(os, lr->lr_doid, lr->lr_mode, 0,
727789Sahrens 	    DMU_OT_NONE, 0, tx);
7282082Seschrock 	ASSERT3U(error, ==, 0);
729789Sahrens 	dmu_tx_commit(tx);
730789Sahrens 
731789Sahrens 	if (zopt_verbose >= 5) {
732789Sahrens 		char osname[MAXNAMELEN];
733789Sahrens 		dmu_objset_name(os, osname);
734789Sahrens 		(void) printf("replay create of %s object %llu"
735789Sahrens 		    " in txg %llu = %d\n",
736789Sahrens 		    osname, (u_longlong_t)lr->lr_doid,
7378227SNeil.Perrin@Sun.COM 		    (u_longlong_t)dmu_tx_get_txg(tx), error);
738789Sahrens 	}
739789Sahrens 
740789Sahrens 	return (error);
741789Sahrens }
742789Sahrens 
743789Sahrens static int
7448227SNeil.Perrin@Sun.COM ztest_replay_remove(objset_t *os, lr_remove_t *lr, boolean_t byteswap)
745789Sahrens {
746789Sahrens 	dmu_tx_t *tx;
747789Sahrens 	int error;
748789Sahrens 
749789Sahrens 	if (byteswap)
750789Sahrens 		byteswap_uint64_array(lr, sizeof (*lr));
751789Sahrens 
752789Sahrens 	tx = dmu_tx_create(os);
753789Sahrens 	dmu_tx_hold_free(tx, lr->lr_doid, 0, DMU_OBJECT_END);
7548227SNeil.Perrin@Sun.COM 	error = dmu_tx_assign(tx, TXG_WAIT);
755789Sahrens 	if (error) {
756789Sahrens 		dmu_tx_abort(tx);
757789Sahrens 		return (error);
758789Sahrens 	}
759789Sahrens 
760789Sahrens 	error = dmu_object_free(os, lr->lr_doid, tx);
761789Sahrens 	dmu_tx_commit(tx);
762789Sahrens 
763789Sahrens 	return (error);
764789Sahrens }
765789Sahrens 
766789Sahrens zil_replay_func_t *ztest_replay_vector[TX_MAX_TYPE] = {
767789Sahrens 	NULL,			/* 0 no such transaction type */
768789Sahrens 	ztest_replay_create,	/* TX_CREATE */
769789Sahrens 	NULL,			/* TX_MKDIR */
770789Sahrens 	NULL,			/* TX_MKXATTR */
771789Sahrens 	NULL,			/* TX_SYMLINK */
772789Sahrens 	ztest_replay_remove,	/* TX_REMOVE */
773789Sahrens 	NULL,			/* TX_RMDIR */
774789Sahrens 	NULL,			/* TX_LINK */
775789Sahrens 	NULL,			/* TX_RENAME */
776789Sahrens 	NULL,			/* TX_WRITE */
777789Sahrens 	NULL,			/* TX_TRUNCATE */
778789Sahrens 	NULL,			/* TX_SETATTR */
779789Sahrens 	NULL,			/* TX_ACL */
780789Sahrens };
781789Sahrens 
782789Sahrens /*
783789Sahrens  * Verify that we can't destroy an active pool, create an existing pool,
784789Sahrens  * or create a pool with a bad vdev spec.
785789Sahrens  */
786789Sahrens void
787789Sahrens ztest_spa_create_destroy(ztest_args_t *za)
788789Sahrens {
789789Sahrens 	int error;
790789Sahrens 	spa_t *spa;
791789Sahrens 	nvlist_t *nvroot;
792789Sahrens 
793789Sahrens 	/*
794789Sahrens 	 * Attempt to create using a bad file.
795789Sahrens 	 */
7967754SJeff.Bonwick@Sun.COM 	nvroot = make_vdev_root("/dev/bogus", NULL, 0, 0, 0, 0, 0, 1);
7977184Stimh 	error = spa_create("ztest_bad_file", nvroot, NULL, NULL, NULL);
798789Sahrens 	nvlist_free(nvroot);
799789Sahrens 	if (error != ENOENT)
800789Sahrens 		fatal(0, "spa_create(bad_file) = %d", error);
801789Sahrens 
802789Sahrens 	/*
803789Sahrens 	 * Attempt to create using a bad mirror.
804789Sahrens 	 */
8057754SJeff.Bonwick@Sun.COM 	nvroot = make_vdev_root("/dev/bogus", NULL, 0, 0, 0, 0, 2, 1);
8067184Stimh 	error = spa_create("ztest_bad_mirror", nvroot, NULL, NULL, NULL);
807789Sahrens 	nvlist_free(nvroot);
808789Sahrens 	if (error != ENOENT)
809789Sahrens 		fatal(0, "spa_create(bad_mirror) = %d", error);
810789Sahrens 
811789Sahrens 	/*
812789Sahrens 	 * Attempt to create an existing pool.  It shouldn't matter
813789Sahrens 	 * what's in the nvroot; we should fail with EEXIST.
814789Sahrens 	 */
815789Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
8167754SJeff.Bonwick@Sun.COM 	nvroot = make_vdev_root("/dev/bogus", NULL, 0, 0, 0, 0, 0, 1);
8177184Stimh 	error = spa_create(za->za_pool, nvroot, NULL, NULL, NULL);
818789Sahrens 	nvlist_free(nvroot);
819789Sahrens 	if (error != EEXIST)
820789Sahrens 		fatal(0, "spa_create(whatever) = %d", error);
821789Sahrens 
822789Sahrens 	error = spa_open(za->za_pool, &spa, FTAG);
823789Sahrens 	if (error)
824789Sahrens 		fatal(0, "spa_open() = %d", error);
825789Sahrens 
826789Sahrens 	error = spa_destroy(za->za_pool);
827789Sahrens 	if (error != EBUSY)
828789Sahrens 		fatal(0, "spa_destroy() = %d", error);
829789Sahrens 
830789Sahrens 	spa_close(spa, FTAG);
831789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
832789Sahrens }
833789Sahrens 
8347768SJeff.Bonwick@Sun.COM static vdev_t *
8357768SJeff.Bonwick@Sun.COM vdev_lookup_by_path(vdev_t *vd, const char *path)
8367768SJeff.Bonwick@Sun.COM {
8377768SJeff.Bonwick@Sun.COM 	vdev_t *mvd;
8387768SJeff.Bonwick@Sun.COM 
8397768SJeff.Bonwick@Sun.COM 	if (vd->vdev_path != NULL && strcmp(path, vd->vdev_path) == 0)
8407768SJeff.Bonwick@Sun.COM 		return (vd);
8417768SJeff.Bonwick@Sun.COM 
8427768SJeff.Bonwick@Sun.COM 	for (int c = 0; c < vd->vdev_children; c++)
8437768SJeff.Bonwick@Sun.COM 		if ((mvd = vdev_lookup_by_path(vd->vdev_child[c], path)) !=
8447768SJeff.Bonwick@Sun.COM 		    NULL)
8457768SJeff.Bonwick@Sun.COM 			return (mvd);
8467768SJeff.Bonwick@Sun.COM 
8477768SJeff.Bonwick@Sun.COM 	return (NULL);
8487768SJeff.Bonwick@Sun.COM }
8497768SJeff.Bonwick@Sun.COM 
850789Sahrens /*
851789Sahrens  * Verify that vdev_add() works as expected.
852789Sahrens  */
853789Sahrens void
854789Sahrens ztest_vdev_add_remove(ztest_args_t *za)
855789Sahrens {
8565530Sbonwick 	spa_t *spa = za->za_spa;
857789Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
858789Sahrens 	nvlist_t *nvroot;
859789Sahrens 	int error;
860789Sahrens 
861789Sahrens 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
862789Sahrens 
8637754SJeff.Bonwick@Sun.COM 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
864789Sahrens 
865789Sahrens 	ztest_shared->zs_vdev_primaries =
866789Sahrens 	    spa->spa_root_vdev->vdev_children * leaves;
867789Sahrens 
8687754SJeff.Bonwick@Sun.COM 	spa_config_exit(spa, SCL_VDEV, FTAG);
869789Sahrens 
8704527Sperrin 	/*
8714527Sperrin 	 * Make 1/4 of the devices be log devices.
8724527Sperrin 	 */
8737754SJeff.Bonwick@Sun.COM 	nvroot = make_vdev_root(NULL, NULL, zopt_vdev_size, 0,
8744527Sperrin 	    ztest_random(4) == 0, zopt_raidz, zopt_mirrors, 1);
8754527Sperrin 
876789Sahrens 	error = spa_vdev_add(spa, nvroot);
877789Sahrens 	nvlist_free(nvroot);
878789Sahrens 
879789Sahrens 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
880789Sahrens 
881789Sahrens 	if (error == ENOSPC)
882789Sahrens 		ztest_record_enospc("spa_vdev_add");
883789Sahrens 	else if (error != 0)
884789Sahrens 		fatal(0, "spa_vdev_add() = %d", error);
8857754SJeff.Bonwick@Sun.COM }
8867754SJeff.Bonwick@Sun.COM 
8877754SJeff.Bonwick@Sun.COM /*
8887754SJeff.Bonwick@Sun.COM  * Verify that adding/removing aux devices (l2arc, hot spare) works as expected.
8897754SJeff.Bonwick@Sun.COM  */
8907754SJeff.Bonwick@Sun.COM void
8917754SJeff.Bonwick@Sun.COM ztest_vdev_aux_add_remove(ztest_args_t *za)
8927754SJeff.Bonwick@Sun.COM {
8937754SJeff.Bonwick@Sun.COM 	spa_t *spa = za->za_spa;
8947768SJeff.Bonwick@Sun.COM 	vdev_t *rvd = spa->spa_root_vdev;
8957754SJeff.Bonwick@Sun.COM 	spa_aux_vdev_t *sav;
8967754SJeff.Bonwick@Sun.COM 	char *aux;
8977754SJeff.Bonwick@Sun.COM 	uint64_t guid = 0;
8987754SJeff.Bonwick@Sun.COM 	int error;
8997754SJeff.Bonwick@Sun.COM 
9007768SJeff.Bonwick@Sun.COM 	if (ztest_random(2) == 0) {
9017754SJeff.Bonwick@Sun.COM 		sav = &spa->spa_spares;
9027754SJeff.Bonwick@Sun.COM 		aux = ZPOOL_CONFIG_SPARES;
9037754SJeff.Bonwick@Sun.COM 	} else {
9047754SJeff.Bonwick@Sun.COM 		sav = &spa->spa_l2cache;
9057754SJeff.Bonwick@Sun.COM 		aux = ZPOOL_CONFIG_L2CACHE;
9067754SJeff.Bonwick@Sun.COM 	}
9077754SJeff.Bonwick@Sun.COM 
9087754SJeff.Bonwick@Sun.COM 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
9097754SJeff.Bonwick@Sun.COM 
9107754SJeff.Bonwick@Sun.COM 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
9117754SJeff.Bonwick@Sun.COM 
9127754SJeff.Bonwick@Sun.COM 	if (sav->sav_count != 0 && ztest_random(4) == 0) {
9137754SJeff.Bonwick@Sun.COM 		/*
9147754SJeff.Bonwick@Sun.COM 		 * Pick a random device to remove.
9157754SJeff.Bonwick@Sun.COM 		 */
9167754SJeff.Bonwick@Sun.COM 		guid = sav->sav_vdevs[ztest_random(sav->sav_count)]->vdev_guid;
9177754SJeff.Bonwick@Sun.COM 	} else {
9187754SJeff.Bonwick@Sun.COM 		/*
9197754SJeff.Bonwick@Sun.COM 		 * Find an unused device we can add.
9207754SJeff.Bonwick@Sun.COM 		 */
9217754SJeff.Bonwick@Sun.COM 		ztest_shared->zs_vdev_aux = 0;
9227754SJeff.Bonwick@Sun.COM 		for (;;) {
9237754SJeff.Bonwick@Sun.COM 			char path[MAXPATHLEN];
9247754SJeff.Bonwick@Sun.COM 			int c;
9257754SJeff.Bonwick@Sun.COM 			(void) sprintf(path, ztest_aux_template, zopt_dir,
9267754SJeff.Bonwick@Sun.COM 			    zopt_pool, aux, ztest_shared->zs_vdev_aux);
9277754SJeff.Bonwick@Sun.COM 			for (c = 0; c < sav->sav_count; c++)
9287754SJeff.Bonwick@Sun.COM 				if (strcmp(sav->sav_vdevs[c]->vdev_path,
9297754SJeff.Bonwick@Sun.COM 				    path) == 0)
9307754SJeff.Bonwick@Sun.COM 					break;
9317768SJeff.Bonwick@Sun.COM 			if (c == sav->sav_count &&
9327768SJeff.Bonwick@Sun.COM 			    vdev_lookup_by_path(rvd, path) == NULL)
9337754SJeff.Bonwick@Sun.COM 				break;
9347754SJeff.Bonwick@Sun.COM 			ztest_shared->zs_vdev_aux++;
9357754SJeff.Bonwick@Sun.COM 		}
9367754SJeff.Bonwick@Sun.COM 	}
9377754SJeff.Bonwick@Sun.COM 
9387754SJeff.Bonwick@Sun.COM 	spa_config_exit(spa, SCL_VDEV, FTAG);
9397754SJeff.Bonwick@Sun.COM 
9407754SJeff.Bonwick@Sun.COM 	if (guid == 0) {
9417754SJeff.Bonwick@Sun.COM 		/*
9427754SJeff.Bonwick@Sun.COM 		 * Add a new device.
9437754SJeff.Bonwick@Sun.COM 		 */
9447768SJeff.Bonwick@Sun.COM 		nvlist_t *nvroot = make_vdev_root(NULL, aux,
9457768SJeff.Bonwick@Sun.COM 		    (zopt_vdev_size * 5) / 4, 0, 0, 0, 0, 1);
9467754SJeff.Bonwick@Sun.COM 		error = spa_vdev_add(spa, nvroot);
9477754SJeff.Bonwick@Sun.COM 		if (error != 0)
9487754SJeff.Bonwick@Sun.COM 			fatal(0, "spa_vdev_add(%p) = %d", nvroot, error);
9497754SJeff.Bonwick@Sun.COM 		nvlist_free(nvroot);
9507754SJeff.Bonwick@Sun.COM 	} else {
9517754SJeff.Bonwick@Sun.COM 		/*
9527754SJeff.Bonwick@Sun.COM 		 * Remove an existing device.  Sometimes, dirty its
9537754SJeff.Bonwick@Sun.COM 		 * vdev state first to make sure we handle removal
9547754SJeff.Bonwick@Sun.COM 		 * of devices that have pending state changes.
9557754SJeff.Bonwick@Sun.COM 		 */
9567754SJeff.Bonwick@Sun.COM 		if (ztest_random(2) == 0)
9577754SJeff.Bonwick@Sun.COM 			(void) vdev_online(spa, guid, B_FALSE, NULL);
9587754SJeff.Bonwick@Sun.COM 
9597754SJeff.Bonwick@Sun.COM 		error = spa_vdev_remove(spa, guid, B_FALSE);
9607754SJeff.Bonwick@Sun.COM 		if (error != 0 && error != EBUSY)
9617754SJeff.Bonwick@Sun.COM 			fatal(0, "spa_vdev_remove(%llu) = %d", guid, error);
9627754SJeff.Bonwick@Sun.COM 	}
9637754SJeff.Bonwick@Sun.COM 
9647754SJeff.Bonwick@Sun.COM 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
965789Sahrens }
966789Sahrens 
967789Sahrens /*
968789Sahrens  * Verify that we can attach and detach devices.
969789Sahrens  */
970789Sahrens void
971789Sahrens ztest_vdev_attach_detach(ztest_args_t *za)
972789Sahrens {
9735530Sbonwick 	spa_t *spa = za->za_spa;
9747768SJeff.Bonwick@Sun.COM 	spa_aux_vdev_t *sav = &spa->spa_spares;
975789Sahrens 	vdev_t *rvd = spa->spa_root_vdev;
9761544Seschrock 	vdev_t *oldvd, *newvd, *pvd;
9777754SJeff.Bonwick@Sun.COM 	nvlist_t *root;
978789Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
979789Sahrens 	uint64_t leaf, top;
9801732Sbonwick 	uint64_t ashift = ztest_get_ashift();
9818241SJeff.Bonwick@Sun.COM 	uint64_t oldguid, pguid;
9821544Seschrock 	size_t oldsize, newsize;
9831544Seschrock 	char oldpath[MAXPATHLEN], newpath[MAXPATHLEN];
984789Sahrens 	int replacing;
9857793SJeff.Bonwick@Sun.COM 	int oldvd_has_siblings = B_FALSE;
9867768SJeff.Bonwick@Sun.COM 	int newvd_is_spare = B_FALSE;
9877768SJeff.Bonwick@Sun.COM 	int oldvd_is_log;
988789Sahrens 	int error, expected_error;
989789Sahrens 
990789Sahrens 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
991789Sahrens 
9927754SJeff.Bonwick@Sun.COM 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
993789Sahrens 
994789Sahrens 	/*
995789Sahrens 	 * Decide whether to do an attach or a replace.
996789Sahrens 	 */
997789Sahrens 	replacing = ztest_random(2);
998789Sahrens 
999789Sahrens 	/*
1000789Sahrens 	 * Pick a random top-level vdev.
1001789Sahrens 	 */
1002789Sahrens 	top = ztest_random(rvd->vdev_children);
1003789Sahrens 
1004789Sahrens 	/*
1005789Sahrens 	 * Pick a random leaf within it.
1006789Sahrens 	 */
1007789Sahrens 	leaf = ztest_random(leaves);
1008789Sahrens 
1009789Sahrens 	/*
10107768SJeff.Bonwick@Sun.COM 	 * Locate this vdev.
1011789Sahrens 	 */
10127768SJeff.Bonwick@Sun.COM 	oldvd = rvd->vdev_child[top];
10138241SJeff.Bonwick@Sun.COM 	if (zopt_mirrors >= 1) {
10148241SJeff.Bonwick@Sun.COM 		ASSERT(oldvd->vdev_ops == &vdev_mirror_ops);
10158241SJeff.Bonwick@Sun.COM 		ASSERT(oldvd->vdev_children >= zopt_mirrors);
10167768SJeff.Bonwick@Sun.COM 		oldvd = oldvd->vdev_child[leaf / zopt_raidz];
10178241SJeff.Bonwick@Sun.COM 	}
10188241SJeff.Bonwick@Sun.COM 	if (zopt_raidz > 1) {
10198241SJeff.Bonwick@Sun.COM 		ASSERT(oldvd->vdev_ops == &vdev_raidz_ops);
10208241SJeff.Bonwick@Sun.COM 		ASSERT(oldvd->vdev_children == zopt_raidz);
10217768SJeff.Bonwick@Sun.COM 		oldvd = oldvd->vdev_child[leaf % zopt_raidz];
10228241SJeff.Bonwick@Sun.COM 	}
1023789Sahrens 
1024789Sahrens 	/*
10257768SJeff.Bonwick@Sun.COM 	 * If we're already doing an attach or replace, oldvd may be a
10267768SJeff.Bonwick@Sun.COM 	 * mirror vdev -- in which case, pick a random child.
1027789Sahrens 	 */
10287768SJeff.Bonwick@Sun.COM 	while (oldvd->vdev_children != 0) {
10297793SJeff.Bonwick@Sun.COM 		oldvd_has_siblings = B_TRUE;
10308241SJeff.Bonwick@Sun.COM 		ASSERT(oldvd->vdev_children >= 2);
10318241SJeff.Bonwick@Sun.COM 		oldvd = oldvd->vdev_child[ztest_random(oldvd->vdev_children)];
10327768SJeff.Bonwick@Sun.COM 	}
10337768SJeff.Bonwick@Sun.COM 
10347768SJeff.Bonwick@Sun.COM 	oldguid = oldvd->vdev_guid;
10357768SJeff.Bonwick@Sun.COM 	oldsize = vdev_get_rsize(oldvd);
10367768SJeff.Bonwick@Sun.COM 	oldvd_is_log = oldvd->vdev_top->vdev_islog;
10377768SJeff.Bonwick@Sun.COM 	(void) strcpy(oldpath, oldvd->vdev_path);
10381544Seschrock 	pvd = oldvd->vdev_parent;
10398241SJeff.Bonwick@Sun.COM 	pguid = pvd->vdev_guid;
1040789Sahrens 
1041789Sahrens 	/*
10427793SJeff.Bonwick@Sun.COM 	 * If oldvd has siblings, then half of the time, detach it.
10437793SJeff.Bonwick@Sun.COM 	 */
10447793SJeff.Bonwick@Sun.COM 	if (oldvd_has_siblings && ztest_random(2) == 0) {
10457793SJeff.Bonwick@Sun.COM 		spa_config_exit(spa, SCL_VDEV, FTAG);
10468241SJeff.Bonwick@Sun.COM 		error = spa_vdev_detach(spa, oldguid, pguid, B_FALSE);
10478241SJeff.Bonwick@Sun.COM 		if (error != 0 && error != ENODEV && error != EBUSY &&
10488241SJeff.Bonwick@Sun.COM 		    error != ENOTSUP)
10498241SJeff.Bonwick@Sun.COM 			fatal(0, "detach (%s) returned %d", oldpath, error);
10507793SJeff.Bonwick@Sun.COM 		(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
10517793SJeff.Bonwick@Sun.COM 		return;
10527793SJeff.Bonwick@Sun.COM 	}
10537793SJeff.Bonwick@Sun.COM 
10547793SJeff.Bonwick@Sun.COM 	/*
10557768SJeff.Bonwick@Sun.COM 	 * For the new vdev, choose with equal probability between the two
10567768SJeff.Bonwick@Sun.COM 	 * standard paths (ending in either 'a' or 'b') or a random hot spare.
1057789Sahrens 	 */
10587768SJeff.Bonwick@Sun.COM 	if (sav->sav_count != 0 && ztest_random(3) == 0) {
10597768SJeff.Bonwick@Sun.COM 		newvd = sav->sav_vdevs[ztest_random(sav->sav_count)];
10607768SJeff.Bonwick@Sun.COM 		newvd_is_spare = B_TRUE;
10617768SJeff.Bonwick@Sun.COM 		(void) strcpy(newpath, newvd->vdev_path);
10627768SJeff.Bonwick@Sun.COM 	} else {
10637768SJeff.Bonwick@Sun.COM 		(void) snprintf(newpath, sizeof (newpath), ztest_dev_template,
10647768SJeff.Bonwick@Sun.COM 		    zopt_dir, zopt_pool, top * leaves + leaf);
10657768SJeff.Bonwick@Sun.COM 		if (ztest_random(2) == 0)
10667768SJeff.Bonwick@Sun.COM 			newpath[strlen(newpath) - 1] = 'b';
10677768SJeff.Bonwick@Sun.COM 		newvd = vdev_lookup_by_path(rvd, newpath);
10687768SJeff.Bonwick@Sun.COM 	}
10697768SJeff.Bonwick@Sun.COM 
10707768SJeff.Bonwick@Sun.COM 	if (newvd) {
10717768SJeff.Bonwick@Sun.COM 		newsize = vdev_get_rsize(newvd);
10727768SJeff.Bonwick@Sun.COM 	} else {
10737768SJeff.Bonwick@Sun.COM 		/*
10747768SJeff.Bonwick@Sun.COM 		 * Make newsize a little bigger or smaller than oldsize.
10757768SJeff.Bonwick@Sun.COM 		 * If it's smaller, the attach should fail.
10767768SJeff.Bonwick@Sun.COM 		 * If it's larger, and we're doing a replace,
10777768SJeff.Bonwick@Sun.COM 		 * we should get dynamic LUN growth when we're done.
10787768SJeff.Bonwick@Sun.COM 		 */
10797768SJeff.Bonwick@Sun.COM 		newsize = 10 * oldsize / (9 + ztest_random(3));
10807768SJeff.Bonwick@Sun.COM 	}
1081789Sahrens 
1082789Sahrens 	/*
1083789Sahrens 	 * If pvd is not a mirror or root, the attach should fail with ENOTSUP,
1084789Sahrens 	 * unless it's a replace; in that case any non-replacing parent is OK.
1085789Sahrens 	 *
10861544Seschrock 	 * If newvd is already part of the pool, it should fail with EBUSY.
1087789Sahrens 	 *
10881544Seschrock 	 * If newvd is too small, it should fail with EOVERFLOW.
1089789Sahrens 	 */
10907768SJeff.Bonwick@Sun.COM 	if (pvd->vdev_ops != &vdev_mirror_ops &&
10917768SJeff.Bonwick@Sun.COM 	    pvd->vdev_ops != &vdev_root_ops && (!replacing ||
10927768SJeff.Bonwick@Sun.COM 	    pvd->vdev_ops == &vdev_replacing_ops ||
10937768SJeff.Bonwick@Sun.COM 	    pvd->vdev_ops == &vdev_spare_ops))
10947768SJeff.Bonwick@Sun.COM 		expected_error = ENOTSUP;
10957768SJeff.Bonwick@Sun.COM 	else if (newvd_is_spare && (!replacing || oldvd_is_log))
10967768SJeff.Bonwick@Sun.COM 		expected_error = ENOTSUP;
10977768SJeff.Bonwick@Sun.COM 	else if (newvd == oldvd)
10987768SJeff.Bonwick@Sun.COM 		expected_error = replacing ? 0 : EBUSY;
10997768SJeff.Bonwick@Sun.COM 	else if (vdev_lookup_by_path(rvd, newpath) != NULL)
11002174Seschrock 		expected_error = EBUSY;
11011544Seschrock 	else if (newsize < oldsize)
1102789Sahrens 		expected_error = EOVERFLOW;
11031732Sbonwick 	else if (ashift > oldvd->vdev_top->vdev_ashift)
11041732Sbonwick 		expected_error = EDOM;
1105789Sahrens 	else
1106789Sahrens 		expected_error = 0;
1107789Sahrens 
11087754SJeff.Bonwick@Sun.COM 	spa_config_exit(spa, SCL_VDEV, FTAG);
1109789Sahrens 
1110789Sahrens 	/*
11111544Seschrock 	 * Build the nvlist describing newpath.
1112789Sahrens 	 */
11137754SJeff.Bonwick@Sun.COM 	root = make_vdev_root(newpath, NULL, newvd == NULL ? newsize : 0,
11147754SJeff.Bonwick@Sun.COM 	    ashift, 0, 0, 0, 1);
1115789Sahrens 
11167768SJeff.Bonwick@Sun.COM 	error = spa_vdev_attach(spa, oldguid, root, replacing);
1117789Sahrens 
1118789Sahrens 	nvlist_free(root);
1119789Sahrens 
1120789Sahrens 	/*
1121789Sahrens 	 * If our parent was the replacing vdev, but the replace completed,
1122789Sahrens 	 * then instead of failing with ENOTSUP we may either succeed,
1123789Sahrens 	 * fail with ENODEV, or fail with EOVERFLOW.
1124789Sahrens 	 */
1125789Sahrens 	if (expected_error == ENOTSUP &&
1126789Sahrens 	    (error == 0 || error == ENODEV || error == EOVERFLOW))
1127789Sahrens 		expected_error = error;
1128789Sahrens 
1129797Sbonwick 	/*
1130797Sbonwick 	 * If someone grew the LUN, the replacement may be too small.
1131797Sbonwick 	 */
11327046Sahrens 	if (error == EOVERFLOW || error == EBUSY)
1133797Sbonwick 		expected_error = error;
1134797Sbonwick 
11357046Sahrens 	/* XXX workaround 6690467 */
11367046Sahrens 	if (error != expected_error && expected_error != EBUSY) {
11377046Sahrens 		fatal(0, "attach (%s %llu, %s %llu, %d) "
11387046Sahrens 		    "returned %d, expected %d",
11397046Sahrens 		    oldpath, (longlong_t)oldsize, newpath,
11407046Sahrens 		    (longlong_t)newsize, replacing, error, expected_error);
1141789Sahrens 	}
1142789Sahrens 
1143789Sahrens 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
1144789Sahrens }
1145789Sahrens 
1146789Sahrens /*
1147789Sahrens  * Verify that dynamic LUN growth works as expected.
1148789Sahrens  */
1149789Sahrens void
1150789Sahrens ztest_vdev_LUN_growth(ztest_args_t *za)
1151789Sahrens {
11525530Sbonwick 	spa_t *spa = za->za_spa;
1153789Sahrens 	char dev_name[MAXPATHLEN];
1154789Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
1155789Sahrens 	uint64_t vdev;
1156789Sahrens 	size_t fsize;
1157789Sahrens 	int fd;
1158789Sahrens 
1159789Sahrens 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
1160789Sahrens 
1161789Sahrens 	/*
1162789Sahrens 	 * Pick a random leaf vdev.
1163789Sahrens 	 */
11647754SJeff.Bonwick@Sun.COM 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
1165789Sahrens 	vdev = ztest_random(spa->spa_root_vdev->vdev_children * leaves);
11667754SJeff.Bonwick@Sun.COM 	spa_config_exit(spa, SCL_VDEV, FTAG);
1167789Sahrens 
1168789Sahrens 	(void) sprintf(dev_name, ztest_dev_template, zopt_dir, zopt_pool, vdev);
1169789Sahrens 
1170789Sahrens 	if ((fd = open(dev_name, O_RDWR)) != -1) {
1171789Sahrens 		/*
1172789Sahrens 		 * Determine the size.
1173789Sahrens 		 */
1174789Sahrens 		fsize = lseek(fd, 0, SEEK_END);
1175789Sahrens 
1176789Sahrens 		/*
1177789Sahrens 		 * If it's less than 2x the original size, grow by around 3%.
1178789Sahrens 		 */
1179789Sahrens 		if (fsize < 2 * zopt_vdev_size) {
1180789Sahrens 			size_t newsize = fsize + ztest_random(fsize / 32);
1181789Sahrens 			(void) ftruncate(fd, newsize);
1182789Sahrens 			if (zopt_verbose >= 6) {
1183789Sahrens 				(void) printf("%s grew from %lu to %lu bytes\n",
1184789Sahrens 				    dev_name, (ulong_t)fsize, (ulong_t)newsize);
1185789Sahrens 			}
1186789Sahrens 		}
1187789Sahrens 		(void) close(fd);
1188789Sahrens 	}
1189789Sahrens 
1190789Sahrens 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
1191789Sahrens }
1192789Sahrens 
1193789Sahrens /* ARGSUSED */
1194789Sahrens static void
11954543Smarks ztest_create_cb(objset_t *os, void *arg, cred_t *cr, dmu_tx_t *tx)
1196789Sahrens {
1197789Sahrens 	/*
1198789Sahrens 	 * Create the directory object.
1199789Sahrens 	 */
1200789Sahrens 	VERIFY(dmu_object_claim(os, ZTEST_DIROBJ,
1201789Sahrens 	    DMU_OT_UINT64_OTHER, ZTEST_DIROBJ_BLOCKSIZE,
12025530Sbonwick 	    DMU_OT_UINT64_OTHER, 5 * sizeof (ztest_block_tag_t), tx) == 0);
1203789Sahrens 
1204789Sahrens 	VERIFY(zap_create_claim(os, ZTEST_MICROZAP_OBJ,
1205789Sahrens 	    DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx) == 0);
1206789Sahrens 
1207789Sahrens 	VERIFY(zap_create_claim(os, ZTEST_FATZAP_OBJ,
1208789Sahrens 	    DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx) == 0);
1209789Sahrens }
1210789Sahrens 
12112199Sahrens static int
1212789Sahrens ztest_destroy_cb(char *name, void *arg)
1213789Sahrens {
12145530Sbonwick 	ztest_args_t *za = arg;
1215789Sahrens 	objset_t *os;
12165530Sbonwick 	dmu_object_info_t *doi = &za->za_doi;
1217789Sahrens 	int error;
1218789Sahrens 
1219789Sahrens 	/*
1220789Sahrens 	 * Verify that the dataset contains a directory object.
1221789Sahrens 	 */
1222789Sahrens 	error = dmu_objset_open(name, DMU_OST_OTHER,
12236689Smaybee 	    DS_MODE_USER | DS_MODE_READONLY, &os);
1224789Sahrens 	ASSERT3U(error, ==, 0);
12255530Sbonwick 	error = dmu_object_info(os, ZTEST_DIROBJ, doi);
12261731Sbonwick 	if (error != ENOENT) {
12271731Sbonwick 		/* We could have crashed in the middle of destroying it */
12281731Sbonwick 		ASSERT3U(error, ==, 0);
12295530Sbonwick 		ASSERT3U(doi->doi_type, ==, DMU_OT_UINT64_OTHER);
12305530Sbonwick 		ASSERT3S(doi->doi_physical_blks, >=, 0);
12311731Sbonwick 	}
1232789Sahrens 	dmu_objset_close(os);
1233789Sahrens 
1234789Sahrens 	/*
1235789Sahrens 	 * Destroy the dataset.
1236789Sahrens 	 */
1237789Sahrens 	error = dmu_objset_destroy(name);
12386689Smaybee 	if (error) {
12396689Smaybee 		(void) dmu_objset_open(name, DMU_OST_OTHER,
12406689Smaybee 		    DS_MODE_USER | DS_MODE_READONLY, &os);
12416689Smaybee 		fatal(0, "dmu_objset_destroy(os=%p) = %d\n", &os, error);
12426689Smaybee 	}
12432199Sahrens 	return (0);
1244789Sahrens }
1245789Sahrens 
1246789Sahrens /*
1247789Sahrens  * Verify that dmu_objset_{create,destroy,open,close} work as expected.
1248789Sahrens  */
1249789Sahrens static uint64_t
1250789Sahrens ztest_log_create(zilog_t *zilog, dmu_tx_t *tx, uint64_t object, int mode)
1251789Sahrens {
1252789Sahrens 	itx_t *itx;
1253789Sahrens 	lr_create_t *lr;
1254789Sahrens 	size_t namesize;
1255789Sahrens 	char name[24];
1256789Sahrens 
1257789Sahrens 	(void) sprintf(name, "ZOBJ_%llu", (u_longlong_t)object);
1258789Sahrens 	namesize = strlen(name) + 1;
1259789Sahrens 
1260789Sahrens 	itx = zil_itx_create(TX_CREATE, sizeof (*lr) + namesize +
1261789Sahrens 	    ztest_random(ZIL_MAX_BLKSZ));
1262789Sahrens 	lr = (lr_create_t *)&itx->itx_lr;
1263789Sahrens 	bzero(lr + 1, lr->lr_common.lrc_reclen - sizeof (*lr));
1264789Sahrens 	lr->lr_doid = object;
1265789Sahrens 	lr->lr_foid = 0;
1266789Sahrens 	lr->lr_mode = mode;
1267789Sahrens 	lr->lr_uid = 0;
1268789Sahrens 	lr->lr_gid = 0;
1269789Sahrens 	lr->lr_gen = dmu_tx_get_txg(tx);
1270789Sahrens 	lr->lr_crtime[0] = time(NULL);
1271789Sahrens 	lr->lr_crtime[1] = 0;
1272789Sahrens 	lr->lr_rdev = 0;
1273789Sahrens 	bcopy(name, (char *)(lr + 1), namesize);
1274789Sahrens 
1275789Sahrens 	return (zil_itx_assign(zilog, itx, tx));
1276789Sahrens }
1277789Sahrens 
1278789Sahrens void
1279789Sahrens ztest_dmu_objset_create_destroy(ztest_args_t *za)
1280789Sahrens {
1281789Sahrens 	int error;
12826689Smaybee 	objset_t *os, *os2;
1283789Sahrens 	char name[100];
12846689Smaybee 	int basemode, expected_error;
1285789Sahrens 	zilog_t *zilog;
1286789Sahrens 	uint64_t seq;
1287789Sahrens 	uint64_t objects;
1288789Sahrens 
1289789Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
1290789Sahrens 	(void) snprintf(name, 100, "%s/%s_temp_%llu", za->za_pool, za->za_pool,
1291789Sahrens 	    (u_longlong_t)za->za_instance);
1292789Sahrens 
12936689Smaybee 	basemode = DS_MODE_TYPE(za->za_instance);
12946689Smaybee 	if (basemode != DS_MODE_USER && basemode != DS_MODE_OWNER)
12956689Smaybee 		basemode = DS_MODE_USER;
1296789Sahrens 
1297789Sahrens 	/*
1298789Sahrens 	 * If this dataset exists from a previous run, process its replay log
1299789Sahrens 	 * half of the time.  If we don't replay it, then dmu_objset_destroy()
1300789Sahrens 	 * (invoked from ztest_destroy_cb() below) should just throw it away.
1301789Sahrens 	 */
1302789Sahrens 	if (ztest_random(2) == 0 &&
13036689Smaybee 	    dmu_objset_open(name, DMU_OST_OTHER, DS_MODE_OWNER, &os) == 0) {
13048227SNeil.Perrin@Sun.COM 		zil_replay(os, os, ztest_replay_vector);
1305789Sahrens 		dmu_objset_close(os);
1306789Sahrens 	}
1307789Sahrens 
1308789Sahrens 	/*
1309789Sahrens 	 * There may be an old instance of the dataset we're about to
1310789Sahrens 	 * create lying around from a previous run.  If so, destroy it
1311789Sahrens 	 * and all of its snapshots.
1312789Sahrens 	 */
13135530Sbonwick 	(void) dmu_objset_find(name, ztest_destroy_cb, za,
13142417Sahrens 	    DS_FIND_CHILDREN | DS_FIND_SNAPSHOTS);
1315789Sahrens 
1316789Sahrens 	/*
1317789Sahrens 	 * Verify that the destroyed dataset is no longer in the namespace.
1318789Sahrens 	 */
1319789Sahrens 	error = dmu_objset_open(name, DMU_OST_OTHER, basemode, &os);
1320789Sahrens 	if (error != ENOENT)
1321789Sahrens 		fatal(1, "dmu_objset_open(%s) found destroyed dataset %p",
1322789Sahrens 		    name, os);
1323789Sahrens 
1324789Sahrens 	/*
1325789Sahrens 	 * Verify that we can create a new dataset.
1326789Sahrens 	 */
13276492Stimh 	error = dmu_objset_create(name, DMU_OST_OTHER, NULL, 0,
13286492Stimh 	    ztest_create_cb, NULL);
1329789Sahrens 	if (error) {
1330789Sahrens 		if (error == ENOSPC) {
1331789Sahrens 			ztest_record_enospc("dmu_objset_create");
1332789Sahrens 			(void) rw_unlock(&ztest_shared->zs_name_lock);
1333789Sahrens 			return;
1334789Sahrens 		}
1335789Sahrens 		fatal(0, "dmu_objset_create(%s) = %d", name, error);
1336789Sahrens 	}
1337789Sahrens 
1338789Sahrens 	error = dmu_objset_open(name, DMU_OST_OTHER, basemode, &os);
1339789Sahrens 	if (error) {
1340789Sahrens 		fatal(0, "dmu_objset_open(%s) = %d", name, error);
1341789Sahrens 	}
1342789Sahrens 
1343789Sahrens 	/*
1344789Sahrens 	 * Open the intent log for it.
1345789Sahrens 	 */
1346789Sahrens 	zilog = zil_open(os, NULL);
1347789Sahrens 
1348789Sahrens 	/*
1349789Sahrens 	 * Put a random number of objects in there.
1350789Sahrens 	 */
13511807Sbonwick 	objects = ztest_random(20);
1352789Sahrens 	seq = 0;
1353789Sahrens 	while (objects-- != 0) {
1354789Sahrens 		uint64_t object;
1355789Sahrens 		dmu_tx_t *tx = dmu_tx_create(os);
1356789Sahrens 		dmu_tx_hold_write(tx, DMU_NEW_OBJECT, 0, sizeof (name));
1357789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1358789Sahrens 		if (error) {
1359789Sahrens 			dmu_tx_abort(tx);
1360789Sahrens 		} else {
1361789Sahrens 			object = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1362789Sahrens 			    DMU_OT_NONE, 0, tx);
1363789Sahrens 			ztest_set_random_blocksize(os, object, tx);
1364789Sahrens 			seq = ztest_log_create(zilog, tx, object,
1365789Sahrens 			    DMU_OT_UINT64_OTHER);
1366789Sahrens 			dmu_write(os, object, 0, sizeof (name), name, tx);
1367789Sahrens 			dmu_tx_commit(tx);
1368789Sahrens 		}
1369789Sahrens 		if (ztest_random(5) == 0) {
13702638Sperrin 			zil_commit(zilog, seq, object);
1371789Sahrens 		}
13721807Sbonwick 		if (ztest_random(100) == 0) {
1373789Sahrens 			error = zil_suspend(zilog);
1374789Sahrens 			if (error == 0) {
1375789Sahrens 				zil_resume(zilog);
1376789Sahrens 			}
1377789Sahrens 		}
1378789Sahrens 	}
1379789Sahrens 
1380789Sahrens 	/*
1381789Sahrens 	 * Verify that we cannot create an existing dataset.
1382789Sahrens 	 */
13836492Stimh 	error = dmu_objset_create(name, DMU_OST_OTHER, NULL, 0, NULL, NULL);
1384789Sahrens 	if (error != EEXIST)
1385789Sahrens 		fatal(0, "created existing dataset, error = %d", error);
1386789Sahrens 
1387789Sahrens 	/*
13886689Smaybee 	 * Verify that multiple dataset holds are allowed, but only when
1389789Sahrens 	 * the new access mode is compatible with the base mode.
1390789Sahrens 	 */
13916689Smaybee 	if (basemode == DS_MODE_OWNER) {
13926689Smaybee 		error = dmu_objset_open(name, DMU_OST_OTHER, DS_MODE_USER,
13936689Smaybee 		    &os2);
13946689Smaybee 		if (error)
13956689Smaybee 			fatal(0, "dmu_objset_open('%s') = %d", name, error);
13966689Smaybee 		else
1397789Sahrens 			dmu_objset_close(os2);
1398789Sahrens 	}
13996689Smaybee 	error = dmu_objset_open(name, DMU_OST_OTHER, DS_MODE_OWNER, &os2);
14006689Smaybee 	expected_error = (basemode == DS_MODE_OWNER) ? EBUSY : 0;
14016689Smaybee 	if (error != expected_error)
14026689Smaybee 		fatal(0, "dmu_objset_open('%s') = %d, expected %d",
14036689Smaybee 		    name, error, expected_error);
14046689Smaybee 	if (error == 0)
14056689Smaybee 		dmu_objset_close(os2);
1406789Sahrens 
1407789Sahrens 	zil_close(zilog);
1408789Sahrens 	dmu_objset_close(os);
1409789Sahrens 
1410789Sahrens 	error = dmu_objset_destroy(name);
1411789Sahrens 	if (error)
1412789Sahrens 		fatal(0, "dmu_objset_destroy(%s) = %d", name, error);
1413789Sahrens 
1414789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
1415789Sahrens }
1416789Sahrens 
1417789Sahrens /*
1418789Sahrens  * Verify that dmu_snapshot_{create,destroy,open,close} work as expected.
1419789Sahrens  */
1420789Sahrens void
1421789Sahrens ztest_dmu_snapshot_create_destroy(ztest_args_t *za)
1422789Sahrens {
1423789Sahrens 	int error;
1424789Sahrens 	objset_t *os = za->za_os;
1425789Sahrens 	char snapname[100];
1426789Sahrens 	char osname[MAXNAMELEN];
1427789Sahrens 
1428789Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
1429789Sahrens 	dmu_objset_name(os, osname);
1430789Sahrens 	(void) snprintf(snapname, 100, "%s@%llu", osname,
1431789Sahrens 	    (u_longlong_t)za->za_instance);
1432789Sahrens 
1433789Sahrens 	error = dmu_objset_destroy(snapname);
1434789Sahrens 	if (error != 0 && error != ENOENT)
1435789Sahrens 		fatal(0, "dmu_objset_destroy() = %d", error);
14369355SMatthew.Ahrens@Sun.COM 	error = dmu_objset_snapshot(osname, strchr(snapname, '@')+1,
14379355SMatthew.Ahrens@Sun.COM 	    NULL, FALSE);
1438789Sahrens 	if (error == ENOSPC)
1439789Sahrens 		ztest_record_enospc("dmu_take_snapshot");
1440789Sahrens 	else if (error != 0 && error != EEXIST)
1441789Sahrens 		fatal(0, "dmu_take_snapshot() = %d", error);
1442789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
1443789Sahrens }
1444789Sahrens 
1445789Sahrens /*
14468779SMark.Musante@Sun.COM  * Verify dsl_dataset_promote handles EBUSY
14478779SMark.Musante@Sun.COM  */
14488779SMark.Musante@Sun.COM void
14498779SMark.Musante@Sun.COM ztest_dsl_dataset_promote_busy(ztest_args_t *za)
14508779SMark.Musante@Sun.COM {
14518779SMark.Musante@Sun.COM 	int error;
14528779SMark.Musante@Sun.COM 	objset_t *os = za->za_os;
14538779SMark.Musante@Sun.COM 	objset_t *clone;
14548779SMark.Musante@Sun.COM 	dsl_dataset_t *ds;
14558779SMark.Musante@Sun.COM 	char snap1name[100];
14568779SMark.Musante@Sun.COM 	char clone1name[100];
14578779SMark.Musante@Sun.COM 	char snap2name[100];
14588779SMark.Musante@Sun.COM 	char clone2name[100];
14598779SMark.Musante@Sun.COM 	char snap3name[100];
14608779SMark.Musante@Sun.COM 	char osname[MAXNAMELEN];
14618779SMark.Musante@Sun.COM 	uint64_t curval;
14628779SMark.Musante@Sun.COM 
14638837SMark.Musante@Sun.COM 	curval = za->za_instance;
14648779SMark.Musante@Sun.COM 
14658779SMark.Musante@Sun.COM 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
14668779SMark.Musante@Sun.COM 
14678779SMark.Musante@Sun.COM 	dmu_objset_name(os, osname);
14688837SMark.Musante@Sun.COM 	(void) snprintf(snap1name, 100, "%s@s1_%llu", osname, curval);
14698837SMark.Musante@Sun.COM 	(void) snprintf(clone1name, 100, "%s/c1_%llu", osname, curval);
14708837SMark.Musante@Sun.COM 	(void) snprintf(snap2name, 100, "%s@s2_%llu", clone1name, curval);
14718837SMark.Musante@Sun.COM 	(void) snprintf(clone2name, 100, "%s/c2_%llu", osname, curval);
14728837SMark.Musante@Sun.COM 	(void) snprintf(snap3name, 100, "%s@s3_%llu", clone1name, curval);
14738837SMark.Musante@Sun.COM 
14748837SMark.Musante@Sun.COM 	error = dmu_objset_destroy(clone2name);
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(snap3name);
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(snap2name);
14818837SMark.Musante@Sun.COM 	if (error != 0 && error != ENOENT)
14828837SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy() = %d", error);
14838837SMark.Musante@Sun.COM 	error = dmu_objset_destroy(clone1name);
14848837SMark.Musante@Sun.COM 	if (error != 0 && error != ENOENT)
14858837SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy() = %d", error);
14868837SMark.Musante@Sun.COM 	error = dmu_objset_destroy(snap1name);
14878837SMark.Musante@Sun.COM 	if (error != 0 && error != ENOENT)
14888837SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy() = %d", error);
14898779SMark.Musante@Sun.COM 
14909355SMatthew.Ahrens@Sun.COM 	error = dmu_objset_snapshot(osname, strchr(snap1name, '@')+1,
14919355SMatthew.Ahrens@Sun.COM 	    NULL, FALSE);
14929229SGeorge.Wilson@Sun.COM 	if (error && error != EEXIST) {
14939229SGeorge.Wilson@Sun.COM 		if (error == ENOSPC) {
14949229SGeorge.Wilson@Sun.COM 			ztest_record_enospc("dmu_take_snapshot");
14959229SGeorge.Wilson@Sun.COM 			goto out;
14969229SGeorge.Wilson@Sun.COM 		}
14979229SGeorge.Wilson@Sun.COM 		fatal(0, "dmu_take_snapshot(%s) = %d", snap1name, error);
14989229SGeorge.Wilson@Sun.COM 	}
14998779SMark.Musante@Sun.COM 
15008779SMark.Musante@Sun.COM 	error = dmu_objset_open(snap1name, DMU_OST_OTHER,
15018779SMark.Musante@Sun.COM 	    DS_MODE_USER | DS_MODE_READONLY, &clone);
15028779SMark.Musante@Sun.COM 	if (error)
15038779SMark.Musante@Sun.COM 		fatal(0, "dmu_open_snapshot(%s) = %d", snap1name, error);
15048779SMark.Musante@Sun.COM 
15058779SMark.Musante@Sun.COM 	error = dmu_objset_create(clone1name, DMU_OST_OTHER, clone, 0,
15068779SMark.Musante@Sun.COM 	    NULL, NULL);
15079229SGeorge.Wilson@Sun.COM 	dmu_objset_close(clone);
15089229SGeorge.Wilson@Sun.COM 	if (error) {
15099229SGeorge.Wilson@Sun.COM 		if (error == ENOSPC) {
15109229SGeorge.Wilson@Sun.COM 			ztest_record_enospc("dmu_objset_create");
15119229SGeorge.Wilson@Sun.COM 			goto out;
15129229SGeorge.Wilson@Sun.COM 		}
15138779SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_create(%s) = %d", clone1name, error);
15149229SGeorge.Wilson@Sun.COM 	}
15158779SMark.Musante@Sun.COM 
15168779SMark.Musante@Sun.COM 	error = dmu_objset_snapshot(clone1name, strchr(snap2name, '@')+1,
15179355SMatthew.Ahrens@Sun.COM 	    NULL, FALSE);
15189229SGeorge.Wilson@Sun.COM 	if (error && error != EEXIST) {
15199229SGeorge.Wilson@Sun.COM 		if (error == ENOSPC) {
15209229SGeorge.Wilson@Sun.COM 			ztest_record_enospc("dmu_take_snapshot");
15219229SGeorge.Wilson@Sun.COM 			goto out;
15229229SGeorge.Wilson@Sun.COM 		}
15239229SGeorge.Wilson@Sun.COM 		fatal(0, "dmu_open_snapshot(%s) = %d", snap2name, error);
15249229SGeorge.Wilson@Sun.COM 	}
15258779SMark.Musante@Sun.COM 
15268779SMark.Musante@Sun.COM 	error = dmu_objset_snapshot(clone1name, strchr(snap3name, '@')+1,
15279355SMatthew.Ahrens@Sun.COM 	    NULL, FALSE);
15289229SGeorge.Wilson@Sun.COM 	if (error && error != EEXIST) {
15299229SGeorge.Wilson@Sun.COM 		if (error == ENOSPC) {
15309229SGeorge.Wilson@Sun.COM 			ztest_record_enospc("dmu_take_snapshot");
15319229SGeorge.Wilson@Sun.COM 			goto out;
15329229SGeorge.Wilson@Sun.COM 		}
15339229SGeorge.Wilson@Sun.COM 		fatal(0, "dmu_open_snapshot(%s) = %d", snap3name, error);
15349229SGeorge.Wilson@Sun.COM 	}
15358779SMark.Musante@Sun.COM 
15368779SMark.Musante@Sun.COM 	error = dmu_objset_open(snap3name, DMU_OST_OTHER,
15378779SMark.Musante@Sun.COM 	    DS_MODE_USER | DS_MODE_READONLY, &clone);
15388779SMark.Musante@Sun.COM 	if (error)
15398779SMark.Musante@Sun.COM 		fatal(0, "dmu_open_snapshot(%s) = %d", snap3name, error);
15408779SMark.Musante@Sun.COM 
15418779SMark.Musante@Sun.COM 	error = dmu_objset_create(clone2name, DMU_OST_OTHER, clone, 0,
15428779SMark.Musante@Sun.COM 	    NULL, NULL);
15439229SGeorge.Wilson@Sun.COM 	dmu_objset_close(clone);
15449229SGeorge.Wilson@Sun.COM 	if (error) {
15459229SGeorge.Wilson@Sun.COM 		if (error == ENOSPC) {
15469229SGeorge.Wilson@Sun.COM 			ztest_record_enospc("dmu_objset_create");
15479229SGeorge.Wilson@Sun.COM 			goto out;
15489229SGeorge.Wilson@Sun.COM 		}
15498779SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_create(%s) = %d", clone2name, error);
15509229SGeorge.Wilson@Sun.COM 	}
15518779SMark.Musante@Sun.COM 
15528837SMark.Musante@Sun.COM 	error = dsl_dataset_own(snap1name, DS_MODE_READONLY, FTAG, &ds);
15538779SMark.Musante@Sun.COM 	if (error)
15548779SMark.Musante@Sun.COM 		fatal(0, "dsl_dataset_own(%s) = %d", snap1name, error);
15558779SMark.Musante@Sun.COM 	error = dsl_dataset_promote(clone2name);
15568779SMark.Musante@Sun.COM 	if (error != EBUSY)
15578779SMark.Musante@Sun.COM 		fatal(0, "dsl_dataset_promote(%s), %d, not EBUSY", clone2name,
15588779SMark.Musante@Sun.COM 		    error);
15598779SMark.Musante@Sun.COM 	dsl_dataset_disown(ds, FTAG);
15608779SMark.Musante@Sun.COM 
15619229SGeorge.Wilson@Sun.COM out:
15628779SMark.Musante@Sun.COM 	error = dmu_objset_destroy(clone2name);
15639229SGeorge.Wilson@Sun.COM 	if (error && error != ENOENT)
15648779SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy(%s) = %d", clone2name, error);
15658779SMark.Musante@Sun.COM 	error = dmu_objset_destroy(snap3name);
15669229SGeorge.Wilson@Sun.COM 	if (error && error != ENOENT)
15678837SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy(%s) = %d", snap3name, error);
15688779SMark.Musante@Sun.COM 	error = dmu_objset_destroy(snap2name);
15699229SGeorge.Wilson@Sun.COM 	if (error && error != ENOENT)
15708779SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy(%s) = %d", snap2name, error);
15718779SMark.Musante@Sun.COM 	error = dmu_objset_destroy(clone1name);
15729229SGeorge.Wilson@Sun.COM 	if (error && error != ENOENT)
15738779SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy(%s) = %d", clone1name, error);
15748779SMark.Musante@Sun.COM 	error = dmu_objset_destroy(snap1name);
15759229SGeorge.Wilson@Sun.COM 	if (error && error != ENOENT)
15768779SMark.Musante@Sun.COM 		fatal(0, "dmu_objset_destroy(%s) = %d", snap1name, error);
15778779SMark.Musante@Sun.COM 
15788779SMark.Musante@Sun.COM 	(void) rw_unlock(&ztest_shared->zs_name_lock);
15798779SMark.Musante@Sun.COM }
15808779SMark.Musante@Sun.COM 
15818779SMark.Musante@Sun.COM /*
1582789Sahrens  * Verify that dmu_object_{alloc,free} work as expected.
1583789Sahrens  */
1584789Sahrens void
1585789Sahrens ztest_dmu_object_alloc_free(ztest_args_t *za)
1586789Sahrens {
1587789Sahrens 	objset_t *os = za->za_os;
1588789Sahrens 	dmu_buf_t *db;
1589789Sahrens 	dmu_tx_t *tx;
1590789Sahrens 	uint64_t batchobj, object, batchsize, endoff, temp;
1591789Sahrens 	int b, c, error, bonuslen;
15925530Sbonwick 	dmu_object_info_t *doi = &za->za_doi;
1593789Sahrens 	char osname[MAXNAMELEN];
1594789Sahrens 
1595789Sahrens 	dmu_objset_name(os, osname);
1596789Sahrens 
1597789Sahrens 	endoff = -8ULL;
1598789Sahrens 	batchsize = 2;
1599789Sahrens 
1600789Sahrens 	/*
1601789Sahrens 	 * Create a batch object if necessary, and record it in the directory.
1602789Sahrens 	 */
16037754SJeff.Bonwick@Sun.COM 	VERIFY3U(0, ==, dmu_read(os, ZTEST_DIROBJ, za->za_diroff,
16041544Seschrock 	    sizeof (uint64_t), &batchobj));
1605789Sahrens 	if (batchobj == 0) {
1606789Sahrens 		tx = dmu_tx_create(os);
1607789Sahrens 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff,
1608789Sahrens 		    sizeof (uint64_t));
1609789Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1610789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1611789Sahrens 		if (error) {
1612789Sahrens 			ztest_record_enospc("create a batch object");
1613789Sahrens 			dmu_tx_abort(tx);
1614789Sahrens 			return;
1615789Sahrens 		}
1616789Sahrens 		batchobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1617789Sahrens 		    DMU_OT_NONE, 0, tx);
1618789Sahrens 		ztest_set_random_blocksize(os, batchobj, tx);
1619789Sahrens 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff,
1620789Sahrens 		    sizeof (uint64_t), &batchobj, tx);
1621789Sahrens 		dmu_tx_commit(tx);
1622789Sahrens 	}
1623789Sahrens 
1624789Sahrens 	/*
1625789Sahrens 	 * Destroy the previous batch of objects.
1626789Sahrens 	 */
1627789Sahrens 	for (b = 0; b < batchsize; b++) {
16287046Sahrens 		VERIFY3U(0, ==, dmu_read(os, batchobj, b * sizeof (uint64_t),
16291544Seschrock 		    sizeof (uint64_t), &object));
1630789Sahrens 		if (object == 0)
1631789Sahrens 			continue;
1632789Sahrens 		/*
1633789Sahrens 		 * Read and validate contents.
1634789Sahrens 		 * We expect the nth byte of the bonus buffer to be n.
1635789Sahrens 		 */
16361544Seschrock 		VERIFY(0 == dmu_bonus_hold(os, object, FTAG, &db));
16375530Sbonwick 		za->za_dbuf = db;
16385530Sbonwick 
16395530Sbonwick 		dmu_object_info_from_db(db, doi);
16405530Sbonwick 		ASSERT(doi->doi_type == DMU_OT_UINT64_OTHER);
16415530Sbonwick 		ASSERT(doi->doi_bonus_type == DMU_OT_PLAIN_OTHER);
16425530Sbonwick 		ASSERT3S(doi->doi_physical_blks, >=, 0);
16435530Sbonwick 
16445530Sbonwick 		bonuslen = doi->doi_bonus_size;
1645789Sahrens 
1646789Sahrens 		for (c = 0; c < bonuslen; c++) {
1647789Sahrens 			if (((uint8_t *)db->db_data)[c] !=
1648789Sahrens 			    (uint8_t)(c + bonuslen)) {
1649789Sahrens 				fatal(0,
1650789Sahrens 				    "bad bonus: %s, obj %llu, off %d: %u != %u",
1651789Sahrens 				    osname, object, c,
1652789Sahrens 				    ((uint8_t *)db->db_data)[c],
1653789Sahrens 				    (uint8_t)(c + bonuslen));
1654789Sahrens 			}
1655789Sahrens 		}
1656789Sahrens 
16571544Seschrock 		dmu_buf_rele(db, FTAG);
16585530Sbonwick 		za->za_dbuf = NULL;
1659789Sahrens 
1660789Sahrens 		/*
1661789Sahrens 		 * We expect the word at endoff to be our object number.
1662789Sahrens 		 */
16631544Seschrock 		VERIFY(0 == dmu_read(os, object, endoff,
16641544Seschrock 		    sizeof (uint64_t), &temp));
1665789Sahrens 
1666789Sahrens 		if (temp != object) {
1667789Sahrens 			fatal(0, "bad data in %s, got %llu, expected %llu",
1668789Sahrens 			    osname, temp, object);
1669789Sahrens 		}
1670789Sahrens 
1671789Sahrens 		/*
1672789Sahrens 		 * Destroy old object and clear batch entry.
1673789Sahrens 		 */
1674789Sahrens 		tx = dmu_tx_create(os);
1675789Sahrens 		dmu_tx_hold_write(tx, batchobj,
1676789Sahrens 		    b * sizeof (uint64_t), sizeof (uint64_t));
1677789Sahrens 		dmu_tx_hold_free(tx, object, 0, DMU_OBJECT_END);
1678789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1679789Sahrens 		if (error) {
1680789Sahrens 			ztest_record_enospc("free object");
1681789Sahrens 			dmu_tx_abort(tx);
1682789Sahrens 			return;
1683789Sahrens 		}
1684789Sahrens 		error = dmu_object_free(os, object, tx);
1685789Sahrens 		if (error) {
1686789Sahrens 			fatal(0, "dmu_object_free('%s', %llu) = %d",
1687789Sahrens 			    osname, object, error);
1688789Sahrens 		}
1689789Sahrens 		object = 0;
1690789Sahrens 
1691789Sahrens 		dmu_object_set_checksum(os, batchobj,
1692789Sahrens 		    ztest_random_checksum(), tx);
1693789Sahrens 		dmu_object_set_compress(os, batchobj,
1694789Sahrens 		    ztest_random_compress(), tx);
1695789Sahrens 
1696789Sahrens 		dmu_write(os, batchobj, b * sizeof (uint64_t),
1697789Sahrens 		    sizeof (uint64_t), &object, tx);
1698789Sahrens 
1699789Sahrens 		dmu_tx_commit(tx);
1700789Sahrens 	}
1701789Sahrens 
1702789Sahrens 	/*
1703789Sahrens 	 * Before creating the new batch of objects, generate a bunch of churn.
1704789Sahrens 	 */
1705789Sahrens 	for (b = ztest_random(100); b > 0; b--) {
1706789Sahrens 		tx = dmu_tx_create(os);
1707789Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1708789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1709789Sahrens 		if (error) {
1710789Sahrens 			ztest_record_enospc("churn objects");
1711789Sahrens 			dmu_tx_abort(tx);
1712789Sahrens 			return;
1713789Sahrens 		}
1714789Sahrens 		object = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1715789Sahrens 		    DMU_OT_NONE, 0, tx);
1716789Sahrens 		ztest_set_random_blocksize(os, object, tx);
1717789Sahrens 		error = dmu_object_free(os, object, tx);
1718789Sahrens 		if (error) {
1719789Sahrens 			fatal(0, "dmu_object_free('%s', %llu) = %d",
1720789Sahrens 			    osname, object, error);
1721789Sahrens 		}
1722789Sahrens 		dmu_tx_commit(tx);
1723789Sahrens 	}
1724789Sahrens 
1725789Sahrens 	/*
1726789Sahrens 	 * Create a new batch of objects with randomly chosen
1727789Sahrens 	 * blocksizes and record them in the batch directory.
1728789Sahrens 	 */
1729789Sahrens 	for (b = 0; b < batchsize; b++) {
1730789Sahrens 		uint32_t va_blksize;
1731789Sahrens 		u_longlong_t va_nblocks;
1732789Sahrens 
1733789Sahrens 		tx = dmu_tx_create(os);
1734789Sahrens 		dmu_tx_hold_write(tx, batchobj, b * sizeof (uint64_t),
1735789Sahrens 		    sizeof (uint64_t));
1736789Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1737789Sahrens 		dmu_tx_hold_write(tx, DMU_NEW_OBJECT, endoff,
1738789Sahrens 		    sizeof (uint64_t));
1739789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1740789Sahrens 		if (error) {
1741789Sahrens 			ztest_record_enospc("create batchobj");
1742789Sahrens 			dmu_tx_abort(tx);
1743789Sahrens 			return;
1744789Sahrens 		}
1745789Sahrens 		bonuslen = (int)ztest_random(dmu_bonus_max()) + 1;
1746789Sahrens 
1747789Sahrens 		object = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1748789Sahrens 		    DMU_OT_PLAIN_OTHER, bonuslen, tx);
1749789Sahrens 
1750789Sahrens 		ztest_set_random_blocksize(os, object, tx);
1751789Sahrens 
1752789Sahrens 		dmu_object_set_checksum(os, object,
1753789Sahrens 		    ztest_random_checksum(), tx);
1754789Sahrens 		dmu_object_set_compress(os, object,
1755789Sahrens 		    ztest_random_compress(), tx);
1756789Sahrens 
1757789Sahrens 		dmu_write(os, batchobj, b * sizeof (uint64_t),
1758789Sahrens 		    sizeof (uint64_t), &object, tx);
1759789Sahrens 
1760789Sahrens 		/*
1761789Sahrens 		 * Write to both the bonus buffer and the regular data.
1762789Sahrens 		 */
17635530Sbonwick 		VERIFY(dmu_bonus_hold(os, object, FTAG, &db) == 0);
17645530Sbonwick 		za->za_dbuf = db;
17654944Smaybee 		ASSERT3U(bonuslen, <=, db->db_size);
1766789Sahrens 
1767789Sahrens 		dmu_object_size_from_db(db, &va_blksize, &va_nblocks);
1768789Sahrens 		ASSERT3S(va_nblocks, >=, 0);
1769789Sahrens 
1770789Sahrens 		dmu_buf_will_dirty(db, tx);
1771789Sahrens 
1772789Sahrens 		/*
1773789Sahrens 		 * See comments above regarding the contents of
1774789Sahrens 		 * the bonus buffer and the word at endoff.
1775789Sahrens 		 */
17764944Smaybee 		for (c = 0; c < bonuslen; c++)
1777789Sahrens 			((uint8_t *)db->db_data)[c] = (uint8_t)(c + bonuslen);
1778789Sahrens 
17791544Seschrock 		dmu_buf_rele(db, FTAG);
17805530Sbonwick 		za->za_dbuf = NULL;
1781789Sahrens 
1782789Sahrens 		/*
1783789Sahrens 		 * Write to a large offset to increase indirection.
1784789Sahrens 		 */
1785789Sahrens 		dmu_write(os, object, endoff, sizeof (uint64_t), &object, tx);
1786789Sahrens 
1787789Sahrens 		dmu_tx_commit(tx);
1788789Sahrens 	}
1789789Sahrens }
1790789Sahrens 
1791789Sahrens /*
1792789Sahrens  * Verify that dmu_{read,write} work as expected.
1793789Sahrens  */
1794789Sahrens typedef struct bufwad {
1795789Sahrens 	uint64_t	bw_index;
1796789Sahrens 	uint64_t	bw_txg;
1797789Sahrens 	uint64_t	bw_data;
1798789Sahrens } bufwad_t;
1799789Sahrens 
1800789Sahrens typedef struct dmu_read_write_dir {
1801789Sahrens 	uint64_t	dd_packobj;
1802789Sahrens 	uint64_t	dd_bigobj;
1803789Sahrens 	uint64_t	dd_chunk;
1804789Sahrens } dmu_read_write_dir_t;
1805789Sahrens 
1806789Sahrens void
1807789Sahrens ztest_dmu_read_write(ztest_args_t *za)
1808789Sahrens {
1809789Sahrens 	objset_t *os = za->za_os;
1810789Sahrens 	dmu_read_write_dir_t dd;
1811789Sahrens 	dmu_tx_t *tx;
1812789Sahrens 	int i, freeit, error;
1813789Sahrens 	uint64_t n, s, txg;
1814789Sahrens 	bufwad_t *packbuf, *bigbuf, *pack, *bigH, *bigT;
1815789Sahrens 	uint64_t packoff, packsize, bigoff, bigsize;
1816789Sahrens 	uint64_t regions = 997;
1817789Sahrens 	uint64_t stride = 123456789ULL;
1818789Sahrens 	uint64_t width = 40;
1819789Sahrens 	int free_percent = 5;
1820789Sahrens 
1821789Sahrens 	/*
1822789Sahrens 	 * This test uses two objects, packobj and bigobj, that are always
1823789Sahrens 	 * updated together (i.e. in the same tx) so that their contents are
1824789Sahrens 	 * in sync and can be compared.  Their contents relate to each other
1825789Sahrens 	 * in a simple way: packobj is a dense array of 'bufwad' structures,
1826789Sahrens 	 * while bigobj is a sparse array of the same bufwads.  Specifically,
1827789Sahrens 	 * for any index n, there are three bufwads that should be identical:
1828789Sahrens 	 *
1829789Sahrens 	 *	packobj, at offset n * sizeof (bufwad_t)
1830789Sahrens 	 *	bigobj, at the head of the nth chunk
1831789Sahrens 	 *	bigobj, at the tail of the nth chunk
1832789Sahrens 	 *
1833789Sahrens 	 * The chunk size is arbitrary. It doesn't have to be a power of two,
1834789Sahrens 	 * and it doesn't have any relation to the object blocksize.
1835789Sahrens 	 * The only requirement is that it can hold at least two bufwads.
1836789Sahrens 	 *
1837789Sahrens 	 * Normally, we write the bufwad to each of these locations.
1838789Sahrens 	 * However, free_percent of the time we instead write zeroes to
1839789Sahrens 	 * packobj and perform a dmu_free_range() on bigobj.  By comparing
1840789Sahrens 	 * bigobj to packobj, we can verify that the DMU is correctly
1841789Sahrens 	 * tracking which parts of an object are allocated and free,
1842789Sahrens 	 * and that the contents of the allocated blocks are correct.
1843789Sahrens 	 */
1844789Sahrens 
1845789Sahrens 	/*
1846789Sahrens 	 * Read the directory info.  If it's the first time, set things up.
1847789Sahrens 	 */
18481544Seschrock 	VERIFY(0 == dmu_read(os, ZTEST_DIROBJ, za->za_diroff,
18491544Seschrock 	    sizeof (dd), &dd));
1850789Sahrens 	if (dd.dd_chunk == 0) {
1851789Sahrens 		ASSERT(dd.dd_packobj == 0);
1852789Sahrens 		ASSERT(dd.dd_bigobj == 0);
1853789Sahrens 		tx = dmu_tx_create(os);
1854789Sahrens 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff, sizeof (dd));
1855789Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1856789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1857789Sahrens 		if (error) {
1858789Sahrens 			ztest_record_enospc("create r/w directory");
1859789Sahrens 			dmu_tx_abort(tx);
1860789Sahrens 			return;
1861789Sahrens 		}
1862789Sahrens 
1863789Sahrens 		dd.dd_packobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1864789Sahrens 		    DMU_OT_NONE, 0, tx);
1865789Sahrens 		dd.dd_bigobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1866789Sahrens 		    DMU_OT_NONE, 0, tx);
1867789Sahrens 		dd.dd_chunk = (1000 + ztest_random(1000)) * sizeof (uint64_t);
1868789Sahrens 
1869789Sahrens 		ztest_set_random_blocksize(os, dd.dd_packobj, tx);
1870789Sahrens 		ztest_set_random_blocksize(os, dd.dd_bigobj, tx);
1871789Sahrens 
1872789Sahrens 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff, sizeof (dd), &dd,
1873789Sahrens 		    tx);
1874789Sahrens 		dmu_tx_commit(tx);
1875789Sahrens 	}
1876789Sahrens 
1877789Sahrens 	/*
1878789Sahrens 	 * Prefetch a random chunk of the big object.
1879789Sahrens 	 * Our aim here is to get some async reads in flight
1880789Sahrens 	 * for blocks that we may free below; the DMU should
1881789Sahrens 	 * handle this race correctly.
1882789Sahrens 	 */
1883789Sahrens 	n = ztest_random(regions) * stride + ztest_random(width);
1884789Sahrens 	s = 1 + ztest_random(2 * width - 1);
1885789Sahrens 	dmu_prefetch(os, dd.dd_bigobj, n * dd.dd_chunk, s * dd.dd_chunk);
1886789Sahrens 
1887789Sahrens 	/*
1888789Sahrens 	 * Pick a random index and compute the offsets into packobj and bigobj.
1889789Sahrens 	 */
1890789Sahrens 	n = ztest_random(regions) * stride + ztest_random(width);
1891789Sahrens 	s = 1 + ztest_random(width - 1);
1892789Sahrens 
1893789Sahrens 	packoff = n * sizeof (bufwad_t);
1894789Sahrens 	packsize = s * sizeof (bufwad_t);
1895789Sahrens 
1896789Sahrens 	bigoff = n * dd.dd_chunk;
1897789Sahrens 	bigsize = s * dd.dd_chunk;
1898789Sahrens 
1899789Sahrens 	packbuf = umem_alloc(packsize, UMEM_NOFAIL);
1900789Sahrens 	bigbuf = umem_alloc(bigsize, UMEM_NOFAIL);
1901789Sahrens 
1902789Sahrens 	/*
1903789Sahrens 	 * free_percent of the time, free a range of bigobj rather than
1904789Sahrens 	 * overwriting it.
1905789Sahrens 	 */
1906789Sahrens 	freeit = (ztest_random(100) < free_percent);
1907789Sahrens 
1908789Sahrens 	/*
1909789Sahrens 	 * Read the current contents of our objects.
1910789Sahrens 	 */
19111544Seschrock 	error = dmu_read(os, dd.dd_packobj, packoff, packsize, packbuf);
19121544Seschrock 	ASSERT3U(error, ==, 0);
19131544Seschrock 	error = dmu_read(os, dd.dd_bigobj, bigoff, bigsize, bigbuf);
19141544Seschrock 	ASSERT3U(error, ==, 0);
1915789Sahrens 
1916789Sahrens 	/*
1917789Sahrens 	 * Get a tx for the mods to both packobj and bigobj.
1918789Sahrens 	 */
1919789Sahrens 	tx = dmu_tx_create(os);
1920789Sahrens 
1921789Sahrens 	dmu_tx_hold_write(tx, dd.dd_packobj, packoff, packsize);
1922789Sahrens 
1923789Sahrens 	if (freeit)
1924789Sahrens 		dmu_tx_hold_free(tx, dd.dd_bigobj, bigoff, bigsize);
1925789Sahrens 	else
1926789Sahrens 		dmu_tx_hold_write(tx, dd.dd_bigobj, bigoff, bigsize);
1927789Sahrens 
1928789Sahrens 	error = dmu_tx_assign(tx, TXG_WAIT);
1929789Sahrens 
1930789Sahrens 	if (error) {
1931789Sahrens 		ztest_record_enospc("dmu r/w range");
1932789Sahrens 		dmu_tx_abort(tx);
1933789Sahrens 		umem_free(packbuf, packsize);
1934789Sahrens 		umem_free(bigbuf, bigsize);
1935789Sahrens 		return;
1936789Sahrens 	}
1937789Sahrens 
1938789Sahrens 	txg = dmu_tx_get_txg(tx);
1939789Sahrens 
1940789Sahrens 	/*
1941789Sahrens 	 * For each index from n to n + s, verify that the existing bufwad
1942789Sahrens 	 * in packobj matches the bufwads at the head and tail of the
1943789Sahrens 	 * corresponding chunk in bigobj.  Then update all three bufwads
1944789Sahrens 	 * with the new values we want to write out.
1945789Sahrens 	 */
1946789Sahrens 	for (i = 0; i < s; i++) {
1947789Sahrens 		/* LINTED */
1948789Sahrens 		pack = (bufwad_t *)((char *)packbuf + i * sizeof (bufwad_t));
1949789Sahrens 		/* LINTED */
1950789Sahrens 		bigH = (bufwad_t *)((char *)bigbuf + i * dd.dd_chunk);
1951789Sahrens 		/* LINTED */
1952789Sahrens 		bigT = (bufwad_t *)((char *)bigH + dd.dd_chunk) - 1;
1953789Sahrens 
1954789Sahrens 		ASSERT((uintptr_t)bigH - (uintptr_t)bigbuf < bigsize);
1955789Sahrens 		ASSERT((uintptr_t)bigT - (uintptr_t)bigbuf < bigsize);
1956789Sahrens 
1957789Sahrens 		if (pack->bw_txg > txg)
1958789Sahrens 			fatal(0, "future leak: got %llx, open txg is %llx",
1959789Sahrens 			    pack->bw_txg, txg);
1960789Sahrens 
1961789Sahrens 		if (pack->bw_data != 0 && pack->bw_index != n + i)
1962789Sahrens 			fatal(0, "wrong index: got %llx, wanted %llx+%llx",
1963789Sahrens 			    pack->bw_index, n, i);
1964789Sahrens 
1965789Sahrens 		if (bcmp(pack, bigH, sizeof (bufwad_t)) != 0)
1966789Sahrens 			fatal(0, "pack/bigH mismatch in %p/%p", pack, bigH);
1967789Sahrens 
1968789Sahrens 		if (bcmp(pack, bigT, sizeof (bufwad_t)) != 0)
1969789Sahrens 			fatal(0, "pack/bigT mismatch in %p/%p", pack, bigT);
1970789Sahrens 
1971789Sahrens 		if (freeit) {
1972789Sahrens 			bzero(pack, sizeof (bufwad_t));
1973789Sahrens 		} else {
1974789Sahrens 			pack->bw_index = n + i;
1975789Sahrens 			pack->bw_txg = txg;
1976789Sahrens 			pack->bw_data = 1 + ztest_random(-2ULL);
1977789Sahrens 		}
1978789Sahrens 		*bigH = *pack;
1979789Sahrens 		*bigT = *pack;
1980789Sahrens 	}
1981789Sahrens 
1982789Sahrens 	/*
1983789Sahrens 	 * We've verified all the old bufwads, and made new ones.
1984789Sahrens 	 * Now write them out.
1985789Sahrens 	 */
1986789Sahrens 	dmu_write(os, dd.dd_packobj, packoff, packsize, packbuf, tx);
1987789Sahrens 
1988789Sahrens 	if (freeit) {
1989789Sahrens 		if (zopt_verbose >= 6) {
1990789Sahrens 			(void) printf("freeing offset %llx size %llx"
1991789Sahrens 			    " txg %llx\n",
1992789Sahrens 			    (u_longlong_t)bigoff,
1993789Sahrens 			    (u_longlong_t)bigsize,
1994789Sahrens 			    (u_longlong_t)txg);
1995789Sahrens 		}
19961544Seschrock 		VERIFY(0 == dmu_free_range(os, dd.dd_bigobj, bigoff,
19971544Seschrock 		    bigsize, tx));
1998789Sahrens 	} else {
1999789Sahrens 		if (zopt_verbose >= 6) {
2000789Sahrens 			(void) printf("writing offset %llx size %llx"
2001789Sahrens 			    " txg %llx\n",
2002789Sahrens 			    (u_longlong_t)bigoff,
2003789Sahrens 			    (u_longlong_t)bigsize,
2004789Sahrens 			    (u_longlong_t)txg);
2005789Sahrens 		}
2006789Sahrens 		dmu_write(os, dd.dd_bigobj, bigoff, bigsize, bigbuf, tx);
2007789Sahrens 	}
2008789Sahrens 
2009789Sahrens 	dmu_tx_commit(tx);
2010789Sahrens 
2011789Sahrens 	/*
2012789Sahrens 	 * Sanity check the stuff we just wrote.
2013789Sahrens 	 */
2014789Sahrens 	{
2015789Sahrens 		void *packcheck = umem_alloc(packsize, UMEM_NOFAIL);
2016789Sahrens 		void *bigcheck = umem_alloc(bigsize, UMEM_NOFAIL);
2017789Sahrens 
20181544Seschrock 		VERIFY(0 == dmu_read(os, dd.dd_packobj, packoff,
20191544Seschrock 		    packsize, packcheck));
20201544Seschrock 		VERIFY(0 == dmu_read(os, dd.dd_bigobj, bigoff,
20211544Seschrock 		    bigsize, bigcheck));
2022789Sahrens 
2023789Sahrens 		ASSERT(bcmp(packbuf, packcheck, packsize) == 0);
2024789Sahrens 		ASSERT(bcmp(bigbuf, bigcheck, bigsize) == 0);
2025789Sahrens 
2026789Sahrens 		umem_free(packcheck, packsize);
2027789Sahrens 		umem_free(bigcheck, bigsize);
2028789Sahrens 	}
2029789Sahrens 
2030789Sahrens 	umem_free(packbuf, packsize);
2031789Sahrens 	umem_free(bigbuf, bigsize);
2032789Sahrens }
2033789Sahrens 
2034789Sahrens void
20359412SAleksandr.Guzovskiy@Sun.COM compare_and_update_pbbufs(uint64_t s, bufwad_t *packbuf, bufwad_t *bigbuf,
20369412SAleksandr.Guzovskiy@Sun.COM     uint64_t bigsize, uint64_t n, dmu_read_write_dir_t dd, uint64_t txg)
20379412SAleksandr.Guzovskiy@Sun.COM {
20389412SAleksandr.Guzovskiy@Sun.COM 	uint64_t i;
20399412SAleksandr.Guzovskiy@Sun.COM 	bufwad_t *pack;
20409412SAleksandr.Guzovskiy@Sun.COM 	bufwad_t *bigH;
20419412SAleksandr.Guzovskiy@Sun.COM 	bufwad_t *bigT;
20429412SAleksandr.Guzovskiy@Sun.COM 
20439412SAleksandr.Guzovskiy@Sun.COM 	/*
20449412SAleksandr.Guzovskiy@Sun.COM 	 * For each index from n to n + s, verify that the existing bufwad
20459412SAleksandr.Guzovskiy@Sun.COM 	 * in packobj matches the bufwads at the head and tail of the
20469412SAleksandr.Guzovskiy@Sun.COM 	 * corresponding chunk in bigobj.  Then update all three bufwads
20479412SAleksandr.Guzovskiy@Sun.COM 	 * with the new values we want to write out.
20489412SAleksandr.Guzovskiy@Sun.COM 	 */
20499412SAleksandr.Guzovskiy@Sun.COM 	for (i = 0; i < s; i++) {
20509412SAleksandr.Guzovskiy@Sun.COM 		/* LINTED */
20519412SAleksandr.Guzovskiy@Sun.COM 		pack = (bufwad_t *)((char *)packbuf + i * sizeof (bufwad_t));
20529412SAleksandr.Guzovskiy@Sun.COM 		/* LINTED */
20539412SAleksandr.Guzovskiy@Sun.COM 		bigH = (bufwad_t *)((char *)bigbuf + i * dd.dd_chunk);
20549412SAleksandr.Guzovskiy@Sun.COM 		/* LINTED */
20559412SAleksandr.Guzovskiy@Sun.COM 		bigT = (bufwad_t *)((char *)bigH + dd.dd_chunk) - 1;
20569412SAleksandr.Guzovskiy@Sun.COM 
20579412SAleksandr.Guzovskiy@Sun.COM 		ASSERT((uintptr_t)bigH - (uintptr_t)bigbuf < bigsize);
20589412SAleksandr.Guzovskiy@Sun.COM 		ASSERT((uintptr_t)bigT - (uintptr_t)bigbuf < bigsize);
20599412SAleksandr.Guzovskiy@Sun.COM 
20609412SAleksandr.Guzovskiy@Sun.COM 		if (pack->bw_txg > txg)
20619412SAleksandr.Guzovskiy@Sun.COM 			fatal(0, "future leak: got %llx, open txg is %llx",
20629412SAleksandr.Guzovskiy@Sun.COM 			    pack->bw_txg, txg);
20639412SAleksandr.Guzovskiy@Sun.COM 
20649412SAleksandr.Guzovskiy@Sun.COM 		if (pack->bw_data != 0 && pack->bw_index != n + i)
20659412SAleksandr.Guzovskiy@Sun.COM 			fatal(0, "wrong index: got %llx, wanted %llx+%llx",
20669412SAleksandr.Guzovskiy@Sun.COM 			    pack->bw_index, n, i);
20679412SAleksandr.Guzovskiy@Sun.COM 
20689412SAleksandr.Guzovskiy@Sun.COM 		if (bcmp(pack, bigH, sizeof (bufwad_t)) != 0)
20699412SAleksandr.Guzovskiy@Sun.COM 			fatal(0, "pack/bigH mismatch in %p/%p", pack, bigH);
20709412SAleksandr.Guzovskiy@Sun.COM 
20719412SAleksandr.Guzovskiy@Sun.COM 		if (bcmp(pack, bigT, sizeof (bufwad_t)) != 0)
20729412SAleksandr.Guzovskiy@Sun.COM 			fatal(0, "pack/bigT mismatch in %p/%p", pack, bigT);
20739412SAleksandr.Guzovskiy@Sun.COM 
20749412SAleksandr.Guzovskiy@Sun.COM 		pack->bw_index = n + i;
20759412SAleksandr.Guzovskiy@Sun.COM 		pack->bw_txg = txg;
20769412SAleksandr.Guzovskiy@Sun.COM 		pack->bw_data = 1 + ztest_random(-2ULL);
20779412SAleksandr.Guzovskiy@Sun.COM 
20789412SAleksandr.Guzovskiy@Sun.COM 		*bigH = *pack;
20799412SAleksandr.Guzovskiy@Sun.COM 		*bigT = *pack;
20809412SAleksandr.Guzovskiy@Sun.COM 	}
20819412SAleksandr.Guzovskiy@Sun.COM }
20829412SAleksandr.Guzovskiy@Sun.COM 
20839412SAleksandr.Guzovskiy@Sun.COM void
20849412SAleksandr.Guzovskiy@Sun.COM ztest_dmu_read_write_zcopy(ztest_args_t *za)
20859412SAleksandr.Guzovskiy@Sun.COM {
20869412SAleksandr.Guzovskiy@Sun.COM 	objset_t *os = za->za_os;
20879412SAleksandr.Guzovskiy@Sun.COM 	dmu_read_write_dir_t dd;
20889412SAleksandr.Guzovskiy@Sun.COM 	dmu_tx_t *tx;
20899412SAleksandr.Guzovskiy@Sun.COM 	uint64_t i;
20909412SAleksandr.Guzovskiy@Sun.COM 	int error;
20919412SAleksandr.Guzovskiy@Sun.COM 	uint64_t n, s, txg;
20929412SAleksandr.Guzovskiy@Sun.COM 	bufwad_t *packbuf, *bigbuf;
20939412SAleksandr.Guzovskiy@Sun.COM 	uint64_t packoff, packsize, bigoff, bigsize;
20949412SAleksandr.Guzovskiy@Sun.COM 	uint64_t regions = 997;
20959412SAleksandr.Guzovskiy@Sun.COM 	uint64_t stride = 123456789ULL;
20969412SAleksandr.Guzovskiy@Sun.COM 	uint64_t width = 9;
20979412SAleksandr.Guzovskiy@Sun.COM 	dmu_buf_t *bonus_db;
20989412SAleksandr.Guzovskiy@Sun.COM 	arc_buf_t **bigbuf_arcbufs;
20999412SAleksandr.Guzovskiy@Sun.COM 	dmu_object_info_t *doi = &za->za_doi;
21009412SAleksandr.Guzovskiy@Sun.COM 
21019412SAleksandr.Guzovskiy@Sun.COM 	/*
21029412SAleksandr.Guzovskiy@Sun.COM 	 * This test uses two objects, packobj and bigobj, that are always
21039412SAleksandr.Guzovskiy@Sun.COM 	 * updated together (i.e. in the same tx) so that their contents are
21049412SAleksandr.Guzovskiy@Sun.COM 	 * in sync and can be compared.  Their contents relate to each other
21059412SAleksandr.Guzovskiy@Sun.COM 	 * in a simple way: packobj is a dense array of 'bufwad' structures,
21069412SAleksandr.Guzovskiy@Sun.COM 	 * while bigobj is a sparse array of the same bufwads.  Specifically,
21079412SAleksandr.Guzovskiy@Sun.COM 	 * for any index n, there are three bufwads that should be identical:
21089412SAleksandr.Guzovskiy@Sun.COM 	 *
21099412SAleksandr.Guzovskiy@Sun.COM 	 *	packobj, at offset n * sizeof (bufwad_t)
21109412SAleksandr.Guzovskiy@Sun.COM 	 *	bigobj, at the head of the nth chunk
21119412SAleksandr.Guzovskiy@Sun.COM 	 *	bigobj, at the tail of the nth chunk
21129412SAleksandr.Guzovskiy@Sun.COM 	 *
21139412SAleksandr.Guzovskiy@Sun.COM 	 * The chunk size is set equal to bigobj block size so that
21149412SAleksandr.Guzovskiy@Sun.COM 	 * dmu_assign_arcbuf() can be tested for object updates.
21159412SAleksandr.Guzovskiy@Sun.COM 	 */
21169412SAleksandr.Guzovskiy@Sun.COM 
21179412SAleksandr.Guzovskiy@Sun.COM 	/*
21189412SAleksandr.Guzovskiy@Sun.COM 	 * Read the directory info.  If it's the first time, set things up.
21199412SAleksandr.Guzovskiy@Sun.COM 	 */
21209412SAleksandr.Guzovskiy@Sun.COM 	VERIFY(0 == dmu_read(os, ZTEST_DIROBJ, za->za_diroff,
21219412SAleksandr.Guzovskiy@Sun.COM 	    sizeof (dd), &dd));
21229412SAleksandr.Guzovskiy@Sun.COM 	if (dd.dd_chunk == 0) {
21239412SAleksandr.Guzovskiy@Sun.COM 		ASSERT(dd.dd_packobj == 0);
21249412SAleksandr.Guzovskiy@Sun.COM 		ASSERT(dd.dd_bigobj == 0);
21259412SAleksandr.Guzovskiy@Sun.COM 		tx = dmu_tx_create(os);
21269412SAleksandr.Guzovskiy@Sun.COM 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff, sizeof (dd));
21279412SAleksandr.Guzovskiy@Sun.COM 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
21289412SAleksandr.Guzovskiy@Sun.COM 		error = dmu_tx_assign(tx, TXG_WAIT);
21299412SAleksandr.Guzovskiy@Sun.COM 		if (error) {
21309412SAleksandr.Guzovskiy@Sun.COM 			ztest_record_enospc("create r/w directory");
21319412SAleksandr.Guzovskiy@Sun.COM 			dmu_tx_abort(tx);
21329412SAleksandr.Guzovskiy@Sun.COM 			return;
21339412SAleksandr.Guzovskiy@Sun.COM 		}
21349412SAleksandr.Guzovskiy@Sun.COM 
21359412SAleksandr.Guzovskiy@Sun.COM 		dd.dd_packobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
21369412SAleksandr.Guzovskiy@Sun.COM 		    DMU_OT_NONE, 0, tx);
21379412SAleksandr.Guzovskiy@Sun.COM 		dd.dd_bigobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
21389412SAleksandr.Guzovskiy@Sun.COM 		    DMU_OT_NONE, 0, tx);
21399412SAleksandr.Guzovskiy@Sun.COM 		ztest_set_random_blocksize(os, dd.dd_packobj, tx);
21409412SAleksandr.Guzovskiy@Sun.COM 		ztest_set_random_blocksize(os, dd.dd_bigobj, tx);
21419412SAleksandr.Guzovskiy@Sun.COM 
21429412SAleksandr.Guzovskiy@Sun.COM 		VERIFY(dmu_object_info(os, dd.dd_bigobj, doi) == 0);
21439412SAleksandr.Guzovskiy@Sun.COM 		ASSERT(doi->doi_data_block_size >= 2 * sizeof (bufwad_t));
21449412SAleksandr.Guzovskiy@Sun.COM 		ASSERT(ISP2(doi->doi_data_block_size));
21459412SAleksandr.Guzovskiy@Sun.COM 		dd.dd_chunk = doi->doi_data_block_size;
21469412SAleksandr.Guzovskiy@Sun.COM 
21479412SAleksandr.Guzovskiy@Sun.COM 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff, sizeof (dd), &dd,
21489412SAleksandr.Guzovskiy@Sun.COM 		    tx);
21499412SAleksandr.Guzovskiy@Sun.COM 		dmu_tx_commit(tx);
21509412SAleksandr.Guzovskiy@Sun.COM 	} else {
21519412SAleksandr.Guzovskiy@Sun.COM 		VERIFY(dmu_object_info(os, dd.dd_bigobj, doi) == 0);
21529412SAleksandr.Guzovskiy@Sun.COM 		VERIFY(ISP2(doi->doi_data_block_size));
21539412SAleksandr.Guzovskiy@Sun.COM 		VERIFY(dd.dd_chunk == doi->doi_data_block_size);
21549412SAleksandr.Guzovskiy@Sun.COM 		VERIFY(dd.dd_chunk >= 2 * sizeof (bufwad_t));
21559412SAleksandr.Guzovskiy@Sun.COM 	}
21569412SAleksandr.Guzovskiy@Sun.COM 
21579412SAleksandr.Guzovskiy@Sun.COM 	/*
21589412SAleksandr.Guzovskiy@Sun.COM 	 * Pick a random index and compute the offsets into packobj and bigobj.
21599412SAleksandr.Guzovskiy@Sun.COM 	 */
21609412SAleksandr.Guzovskiy@Sun.COM 	n = ztest_random(regions) * stride + ztest_random(width);
21619412SAleksandr.Guzovskiy@Sun.COM 	s = 1 + ztest_random(width - 1);
21629412SAleksandr.Guzovskiy@Sun.COM 
21639412SAleksandr.Guzovskiy@Sun.COM 	packoff = n * sizeof (bufwad_t);
21649412SAleksandr.Guzovskiy@Sun.COM 	packsize = s * sizeof (bufwad_t);
21659412SAleksandr.Guzovskiy@Sun.COM 
21669412SAleksandr.Guzovskiy@Sun.COM 	bigoff = n * dd.dd_chunk;
21679412SAleksandr.Guzovskiy@Sun.COM 	bigsize = s * dd.dd_chunk;
21689412SAleksandr.Guzovskiy@Sun.COM 
21699412SAleksandr.Guzovskiy@Sun.COM 	packbuf = umem_zalloc(packsize, UMEM_NOFAIL);
21709412SAleksandr.Guzovskiy@Sun.COM 	bigbuf = umem_zalloc(bigsize, UMEM_NOFAIL);
21719412SAleksandr.Guzovskiy@Sun.COM 
21729412SAleksandr.Guzovskiy@Sun.COM 	VERIFY(dmu_bonus_hold(os, dd.dd_bigobj, FTAG, &bonus_db) == 0);
21739412SAleksandr.Guzovskiy@Sun.COM 
21749412SAleksandr.Guzovskiy@Sun.COM 	bigbuf_arcbufs = umem_zalloc(2 * s * sizeof (arc_buf_t *), UMEM_NOFAIL);
21759412SAleksandr.Guzovskiy@Sun.COM 
21769412SAleksandr.Guzovskiy@Sun.COM 	/*
21779412SAleksandr.Guzovskiy@Sun.COM 	 * Iteration 0 test zcopy for DB_UNCACHED dbufs.
21789412SAleksandr.Guzovskiy@Sun.COM 	 * Iteration 1 test zcopy to already referenced dbufs.
21799412SAleksandr.Guzovskiy@Sun.COM 	 * Iteration 2 test zcopy to dirty dbuf in the same txg.
21809412SAleksandr.Guzovskiy@Sun.COM 	 * Iteration 3 test zcopy to dbuf dirty in previous txg.
21819412SAleksandr.Guzovskiy@Sun.COM 	 * Iteration 4 test zcopy when dbuf is no longer dirty.
21829412SAleksandr.Guzovskiy@Sun.COM 	 * Iteration 5 test zcopy when it can't be done.
21839412SAleksandr.Guzovskiy@Sun.COM 	 * Iteration 6 one more zcopy write.
21849412SAleksandr.Guzovskiy@Sun.COM 	 */
21859412SAleksandr.Guzovskiy@Sun.COM 	for (i = 0; i < 7; i++) {
21869412SAleksandr.Guzovskiy@Sun.COM 		uint64_t j;
21879412SAleksandr.Guzovskiy@Sun.COM 		uint64_t off;
21889412SAleksandr.Guzovskiy@Sun.COM 
21899412SAleksandr.Guzovskiy@Sun.COM 		/*
21909412SAleksandr.Guzovskiy@Sun.COM 		 * In iteration 5 (i == 5) use arcbufs
21919412SAleksandr.Guzovskiy@Sun.COM 		 * that don't match bigobj blksz to test
21929412SAleksandr.Guzovskiy@Sun.COM 		 * dmu_assign_arcbuf() when it can't directly
21939412SAleksandr.Guzovskiy@Sun.COM 		 * assign an arcbuf to a dbuf.
21949412SAleksandr.Guzovskiy@Sun.COM 		 */
21959412SAleksandr.Guzovskiy@Sun.COM 		for (j = 0; j < s; j++) {
21969412SAleksandr.Guzovskiy@Sun.COM 			if (i != 5) {
21979412SAleksandr.Guzovskiy@Sun.COM 				bigbuf_arcbufs[j] =
21989412SAleksandr.Guzovskiy@Sun.COM 				    dmu_request_arcbuf(bonus_db,
21999412SAleksandr.Guzovskiy@Sun.COM 				    dd.dd_chunk);
22009412SAleksandr.Guzovskiy@Sun.COM 			} else {
22019412SAleksandr.Guzovskiy@Sun.COM 				bigbuf_arcbufs[2 * j] =
22029412SAleksandr.Guzovskiy@Sun.COM 				    dmu_request_arcbuf(bonus_db,
22039412SAleksandr.Guzovskiy@Sun.COM 				    dd.dd_chunk / 2);
22049412SAleksandr.Guzovskiy@Sun.COM 				bigbuf_arcbufs[2 * j + 1] =
22059412SAleksandr.Guzovskiy@Sun.COM 				    dmu_request_arcbuf(bonus_db,
22069412SAleksandr.Guzovskiy@Sun.COM 				    dd.dd_chunk / 2);
22079412SAleksandr.Guzovskiy@Sun.COM 			}
22089412SAleksandr.Guzovskiy@Sun.COM 		}
22099412SAleksandr.Guzovskiy@Sun.COM 
22109412SAleksandr.Guzovskiy@Sun.COM 		/*
22119412SAleksandr.Guzovskiy@Sun.COM 		 * Get a tx for the mods to both packobj and bigobj.
22129412SAleksandr.Guzovskiy@Sun.COM 		 */
22139412SAleksandr.Guzovskiy@Sun.COM 		tx = dmu_tx_create(os);
22149412SAleksandr.Guzovskiy@Sun.COM 
22159412SAleksandr.Guzovskiy@Sun.COM 		dmu_tx_hold_write(tx, dd.dd_packobj, packoff, packsize);
22169412SAleksandr.Guzovskiy@Sun.COM 		dmu_tx_hold_write(tx, dd.dd_bigobj, bigoff, bigsize);
22179412SAleksandr.Guzovskiy@Sun.COM 
22189412SAleksandr.Guzovskiy@Sun.COM 		if (ztest_random(100) == 0) {
22199412SAleksandr.Guzovskiy@Sun.COM 			error = -1;
22209412SAleksandr.Guzovskiy@Sun.COM 		} else {
22219412SAleksandr.Guzovskiy@Sun.COM 			error = dmu_tx_assign(tx, TXG_WAIT);
22229412SAleksandr.Guzovskiy@Sun.COM 		}
22239412SAleksandr.Guzovskiy@Sun.COM 
22249412SAleksandr.Guzovskiy@Sun.COM 		if (error) {
22259412SAleksandr.Guzovskiy@Sun.COM 			if (error != -1) {
22269412SAleksandr.Guzovskiy@Sun.COM 				ztest_record_enospc("dmu r/w range");
22279412SAleksandr.Guzovskiy@Sun.COM 			}
22289412SAleksandr.Guzovskiy@Sun.COM 			dmu_tx_abort(tx);
22299412SAleksandr.Guzovskiy@Sun.COM 			umem_free(packbuf, packsize);
22309412SAleksandr.Guzovskiy@Sun.COM 			umem_free(bigbuf, bigsize);
22319412SAleksandr.Guzovskiy@Sun.COM 			for (j = 0; j < s; j++) {
22329412SAleksandr.Guzovskiy@Sun.COM 				if (i != 5) {
22339412SAleksandr.Guzovskiy@Sun.COM 					dmu_return_arcbuf(bigbuf_arcbufs[j]);
22349412SAleksandr.Guzovskiy@Sun.COM 				} else {
22359412SAleksandr.Guzovskiy@Sun.COM 					dmu_return_arcbuf(
22369412SAleksandr.Guzovskiy@Sun.COM 					    bigbuf_arcbufs[2 * j]);
22379412SAleksandr.Guzovskiy@Sun.COM 					dmu_return_arcbuf(
22389412SAleksandr.Guzovskiy@Sun.COM 					    bigbuf_arcbufs[2 * j + 1]);
22399412SAleksandr.Guzovskiy@Sun.COM 				}
22409412SAleksandr.Guzovskiy@Sun.COM 			}
22419412SAleksandr.Guzovskiy@Sun.COM 			umem_free(bigbuf_arcbufs, 2 * s * sizeof (arc_buf_t *));
22429412SAleksandr.Guzovskiy@Sun.COM 			dmu_buf_rele(bonus_db, FTAG);
22439412SAleksandr.Guzovskiy@Sun.COM 			return;
22449412SAleksandr.Guzovskiy@Sun.COM 		}
22459412SAleksandr.Guzovskiy@Sun.COM 
22469412SAleksandr.Guzovskiy@Sun.COM 		txg = dmu_tx_get_txg(tx);
22479412SAleksandr.Guzovskiy@Sun.COM 
22489412SAleksandr.Guzovskiy@Sun.COM 		/*
22499412SAleksandr.Guzovskiy@Sun.COM 		 * 50% of the time don't read objects in the 1st iteration to
22509412SAleksandr.Guzovskiy@Sun.COM 		 * test dmu_assign_arcbuf() for the case when there're no
22519412SAleksandr.Guzovskiy@Sun.COM 		 * existing dbufs for the specified offsets.
22529412SAleksandr.Guzovskiy@Sun.COM 		 */
22539412SAleksandr.Guzovskiy@Sun.COM 		if (i != 0 || ztest_random(2) != 0) {
22549412SAleksandr.Guzovskiy@Sun.COM 			error = dmu_read(os, dd.dd_packobj, packoff,
22559412SAleksandr.Guzovskiy@Sun.COM 			    packsize, packbuf);
22569412SAleksandr.Guzovskiy@Sun.COM 			ASSERT3U(error, ==, 0);
22579412SAleksandr.Guzovskiy@Sun.COM 			error = dmu_read(os, dd.dd_bigobj, bigoff, bigsize,
22589412SAleksandr.Guzovskiy@Sun.COM 			    bigbuf);
22599412SAleksandr.Guzovskiy@Sun.COM 			ASSERT3U(error, ==, 0);
22609412SAleksandr.Guzovskiy@Sun.COM 		}
22619412SAleksandr.Guzovskiy@Sun.COM 		compare_and_update_pbbufs(s, packbuf, bigbuf, bigsize,
22629412SAleksandr.Guzovskiy@Sun.COM 		    n, dd, txg);
22639412SAleksandr.Guzovskiy@Sun.COM 
22649412SAleksandr.Guzovskiy@Sun.COM 		/*
22659412SAleksandr.Guzovskiy@Sun.COM 		 * We've verified all the old bufwads, and made new ones.
22669412SAleksandr.Guzovskiy@Sun.COM 		 * Now write them out.
22679412SAleksandr.Guzovskiy@Sun.COM 		 */
22689412SAleksandr.Guzovskiy@Sun.COM 		dmu_write(os, dd.dd_packobj, packoff, packsize, packbuf, tx);
22699412SAleksandr.Guzovskiy@Sun.COM 		if (zopt_verbose >= 6) {
22709412SAleksandr.Guzovskiy@Sun.COM 			(void) printf("writing offset %llx size %llx"
22719412SAleksandr.Guzovskiy@Sun.COM 			    " txg %llx\n",
22729412SAleksandr.Guzovskiy@Sun.COM 			    (u_longlong_t)bigoff,
22739412SAleksandr.Guzovskiy@Sun.COM 			    (u_longlong_t)bigsize,
22749412SAleksandr.Guzovskiy@Sun.COM 			    (u_longlong_t)txg);
22759412SAleksandr.Guzovskiy@Sun.COM 		}
22769412SAleksandr.Guzovskiy@Sun.COM 		for (off = bigoff, j = 0; j < s; j++, off += dd.dd_chunk) {
22779412SAleksandr.Guzovskiy@Sun.COM 			dmu_buf_t *dbt;
22789412SAleksandr.Guzovskiy@Sun.COM 			if (i != 5) {
22799412SAleksandr.Guzovskiy@Sun.COM 				bcopy((caddr_t)bigbuf + (off - bigoff),
22809412SAleksandr.Guzovskiy@Sun.COM 				    bigbuf_arcbufs[j]->b_data, dd.dd_chunk);
22819412SAleksandr.Guzovskiy@Sun.COM 			} else {
22829412SAleksandr.Guzovskiy@Sun.COM 				bcopy((caddr_t)bigbuf + (off - bigoff),
22839412SAleksandr.Guzovskiy@Sun.COM 				    bigbuf_arcbufs[2 * j]->b_data,
22849412SAleksandr.Guzovskiy@Sun.COM 				    dd.dd_chunk / 2);
22859412SAleksandr.Guzovskiy@Sun.COM 				bcopy((caddr_t)bigbuf + (off - bigoff) +
22869412SAleksandr.Guzovskiy@Sun.COM 				    dd.dd_chunk / 2,
22879412SAleksandr.Guzovskiy@Sun.COM 				    bigbuf_arcbufs[2 * j + 1]->b_data,
22889412SAleksandr.Guzovskiy@Sun.COM 				    dd.dd_chunk / 2);
22899412SAleksandr.Guzovskiy@Sun.COM 			}
22909412SAleksandr.Guzovskiy@Sun.COM 
22919412SAleksandr.Guzovskiy@Sun.COM 			if (i == 1) {
22929412SAleksandr.Guzovskiy@Sun.COM 				VERIFY(dmu_buf_hold(os, dd.dd_bigobj, off,
22939412SAleksandr.Guzovskiy@Sun.COM 				    FTAG, &dbt) == 0);
22949412SAleksandr.Guzovskiy@Sun.COM 			}
22959412SAleksandr.Guzovskiy@Sun.COM 			if (i != 5) {
22969412SAleksandr.Guzovskiy@Sun.COM 				dmu_assign_arcbuf(bonus_db, off,
22979412SAleksandr.Guzovskiy@Sun.COM 				    bigbuf_arcbufs[j], tx);
22989412SAleksandr.Guzovskiy@Sun.COM 			} else {
22999412SAleksandr.Guzovskiy@Sun.COM 				dmu_assign_arcbuf(bonus_db, off,
23009412SAleksandr.Guzovskiy@Sun.COM 				    bigbuf_arcbufs[2 * j], tx);
23019412SAleksandr.Guzovskiy@Sun.COM 				dmu_assign_arcbuf(bonus_db,
23029412SAleksandr.Guzovskiy@Sun.COM 				    off + dd.dd_chunk / 2,
23039412SAleksandr.Guzovskiy@Sun.COM 				    bigbuf_arcbufs[2 * j + 1], tx);
23049412SAleksandr.Guzovskiy@Sun.COM 			}
23059412SAleksandr.Guzovskiy@Sun.COM 			if (i == 1) {
23069412SAleksandr.Guzovskiy@Sun.COM 				dmu_buf_rele(dbt, FTAG);
23079412SAleksandr.Guzovskiy@Sun.COM 			}
23089412SAleksandr.Guzovskiy@Sun.COM 		}
23099412SAleksandr.Guzovskiy@Sun.COM 		dmu_tx_commit(tx);
23109412SAleksandr.Guzovskiy@Sun.COM 
23119412SAleksandr.Guzovskiy@Sun.COM 		/*
23129412SAleksandr.Guzovskiy@Sun.COM 		 * Sanity check the stuff we just wrote.
23139412SAleksandr.Guzovskiy@Sun.COM 		 */
23149412SAleksandr.Guzovskiy@Sun.COM 		{
23159412SAleksandr.Guzovskiy@Sun.COM 			void *packcheck = umem_alloc(packsize, UMEM_NOFAIL);
23169412SAleksandr.Guzovskiy@Sun.COM 			void *bigcheck = umem_alloc(bigsize, UMEM_NOFAIL);
23179412SAleksandr.Guzovskiy@Sun.COM 
23189412SAleksandr.Guzovskiy@Sun.COM 			VERIFY(0 == dmu_read(os, dd.dd_packobj, packoff,
23199412SAleksandr.Guzovskiy@Sun.COM 			    packsize, packcheck));
23209412SAleksandr.Guzovskiy@Sun.COM 			VERIFY(0 == dmu_read(os, dd.dd_bigobj, bigoff,
23219412SAleksandr.Guzovskiy@Sun.COM 			    bigsize, bigcheck));
23229412SAleksandr.Guzovskiy@Sun.COM 
23239412SAleksandr.Guzovskiy@Sun.COM 			ASSERT(bcmp(packbuf, packcheck, packsize) == 0);
23249412SAleksandr.Guzovskiy@Sun.COM 			ASSERT(bcmp(bigbuf, bigcheck, bigsize) == 0);
23259412SAleksandr.Guzovskiy@Sun.COM 
23269412SAleksandr.Guzovskiy@Sun.COM 			umem_free(packcheck, packsize);
23279412SAleksandr.Guzovskiy@Sun.COM 			umem_free(bigcheck, bigsize);
23289412SAleksandr.Guzovskiy@Sun.COM 		}
23299412SAleksandr.Guzovskiy@Sun.COM 		if (i == 2) {
23309412SAleksandr.Guzovskiy@Sun.COM 			txg_wait_open(dmu_objset_pool(os), 0);
23319412SAleksandr.Guzovskiy@Sun.COM 		} else if (i == 3) {
23329412SAleksandr.Guzovskiy@Sun.COM 			txg_wait_synced(dmu_objset_pool(os), 0);
23339412SAleksandr.Guzovskiy@Sun.COM 		}
23349412SAleksandr.Guzovskiy@Sun.COM 	}
23359412SAleksandr.Guzovskiy@Sun.COM 
23369412SAleksandr.Guzovskiy@Sun.COM 	dmu_buf_rele(bonus_db, FTAG);
23379412SAleksandr.Guzovskiy@Sun.COM 	umem_free(packbuf, packsize);
23389412SAleksandr.Guzovskiy@Sun.COM 	umem_free(bigbuf, bigsize);
23399412SAleksandr.Guzovskiy@Sun.COM 	umem_free(bigbuf_arcbufs, 2 * s * sizeof (arc_buf_t *));
23409412SAleksandr.Guzovskiy@Sun.COM }
23419412SAleksandr.Guzovskiy@Sun.COM 
23429412SAleksandr.Guzovskiy@Sun.COM void
23435530Sbonwick ztest_dmu_check_future_leak(ztest_args_t *za)
23443711Smaybee {
23455530Sbonwick 	objset_t *os = za->za_os;
23463711Smaybee 	dmu_buf_t *db;
23475530Sbonwick 	ztest_block_tag_t *bt;
23485530Sbonwick 	dmu_object_info_t *doi = &za->za_doi;
23493711Smaybee 
23503711Smaybee 	/*
23513711Smaybee 	 * Make sure that, if there is a write record in the bonus buffer
23523711Smaybee 	 * of the ZTEST_DIROBJ, that the txg for this record is <= the
23533711Smaybee 	 * last synced txg of the pool.
23543711Smaybee 	 */
23555530Sbonwick 	VERIFY(dmu_bonus_hold(os, ZTEST_DIROBJ, FTAG, &db) == 0);
23565530Sbonwick 	za->za_dbuf = db;
23575530Sbonwick 	VERIFY(dmu_object_info(os, ZTEST_DIROBJ, doi) == 0);
23585530Sbonwick 	ASSERT3U(doi->doi_bonus_size, >=, sizeof (*bt));
23595530Sbonwick 	ASSERT3U(doi->doi_bonus_size, <=, db->db_size);
23605530Sbonwick 	ASSERT3U(doi->doi_bonus_size % sizeof (*bt), ==, 0);
23615530Sbonwick 	bt = (void *)((char *)db->db_data + doi->doi_bonus_size - sizeof (*bt));
23625530Sbonwick 	if (bt->bt_objset != 0) {
23635530Sbonwick 		ASSERT3U(bt->bt_objset, ==, dmu_objset_id(os));
23645530Sbonwick 		ASSERT3U(bt->bt_object, ==, ZTEST_DIROBJ);
23655530Sbonwick 		ASSERT3U(bt->bt_offset, ==, -1ULL);
23665530Sbonwick 		ASSERT3U(bt->bt_txg, <, spa_first_txg(za->za_spa));
23673711Smaybee 	}
23683711Smaybee 	dmu_buf_rele(db, FTAG);
23695530Sbonwick 	za->za_dbuf = NULL;
23703711Smaybee }
23713711Smaybee 
23723711Smaybee void
2373789Sahrens ztest_dmu_write_parallel(ztest_args_t *za)
2374789Sahrens {
2375789Sahrens 	objset_t *os = za->za_os;
23765530Sbonwick 	ztest_block_tag_t *rbt = &za->za_rbt;
23775530Sbonwick 	ztest_block_tag_t *wbt = &za->za_wbt;
23785530Sbonwick 	const size_t btsize = sizeof (ztest_block_tag_t);
2379789Sahrens 	dmu_buf_t *db;
23805530Sbonwick 	int b, error;
23815530Sbonwick 	int bs = ZTEST_DIROBJ_BLOCKSIZE;
23825530Sbonwick 	int do_free = 0;
23837762SJeff.Bonwick@Sun.COM 	uint64_t off, txg, txg_how;
2384789Sahrens 	mutex_t *lp;
2385789Sahrens 	char osname[MAXNAMELEN];
2386789Sahrens 	char iobuf[SPA_MAXBLOCKSIZE];
23875530Sbonwick 	blkptr_t blk = { 0 };
23885530Sbonwick 	uint64_t blkoff;
23895530Sbonwick 	zbookmark_t zb;
23905530Sbonwick 	dmu_tx_t *tx = dmu_tx_create(os);
23919412SAleksandr.Guzovskiy@Sun.COM 	dmu_buf_t *bonus_db;
23929412SAleksandr.Guzovskiy@Sun.COM 	arc_buf_t *abuf = NULL;
2393789Sahrens 
2394789Sahrens 	dmu_objset_name(os, osname);
2395789Sahrens 
2396789Sahrens 	/*
2397789Sahrens 	 * Have multiple threads write to large offsets in ZTEST_DIROBJ
2398789Sahrens 	 * to verify that having multiple threads writing to the same object
2399789Sahrens 	 * in parallel doesn't cause any trouble.
2400789Sahrens 	 */
24015530Sbonwick 	if (ztest_random(4) == 0) {
24025530Sbonwick 		/*
24035530Sbonwick 		 * Do the bonus buffer instead of a regular block.
24045530Sbonwick 		 * We need a lock to serialize resize vs. others,
24055530Sbonwick 		 * so we hash on the objset ID.
24065530Sbonwick 		 */
24075530Sbonwick 		b = dmu_objset_id(os) % ZTEST_SYNC_LOCKS;
24085530Sbonwick 		off = -1ULL;
24095530Sbonwick 		dmu_tx_hold_bonus(tx, ZTEST_DIROBJ);
24105530Sbonwick 	} else {
2411789Sahrens 		b = ztest_random(ZTEST_SYNC_LOCKS);
24125530Sbonwick 		off = za->za_diroff_shared + (b << SPA_MAXBLOCKSHIFT);
2413789Sahrens 		if (ztest_random(4) == 0) {
24145530Sbonwick 			do_free = 1;
2415789Sahrens 			dmu_tx_hold_free(tx, ZTEST_DIROBJ, off, bs);
24165530Sbonwick 		} else {
2417789Sahrens 			dmu_tx_hold_write(tx, ZTEST_DIROBJ, off, bs);
2418789Sahrens 		}
2419789Sahrens 	}
24205530Sbonwick 
24219412SAleksandr.Guzovskiy@Sun.COM 	if (off != -1ULL && P2PHASE(off, bs) == 0 && !do_free &&
24229412SAleksandr.Guzovskiy@Sun.COM 	    ztest_random(8) == 0) {
24239412SAleksandr.Guzovskiy@Sun.COM 		VERIFY(dmu_bonus_hold(os, ZTEST_DIROBJ, FTAG, &bonus_db) == 0);
24249412SAleksandr.Guzovskiy@Sun.COM 		abuf = dmu_request_arcbuf(bonus_db, bs);
24259412SAleksandr.Guzovskiy@Sun.COM 	}
24269412SAleksandr.Guzovskiy@Sun.COM 
24275530Sbonwick 	txg_how = ztest_random(2) == 0 ? TXG_WAIT : TXG_NOWAIT;
24285530Sbonwick 	error = dmu_tx_assign(tx, txg_how);
24295530Sbonwick 	if (error) {
24305530Sbonwick 		if (error == ERESTART) {
24315530Sbonwick 			ASSERT(txg_how == TXG_NOWAIT);
24325530Sbonwick 			dmu_tx_wait(tx);
24335530Sbonwick 		} else {
24345530Sbonwick 			ztest_record_enospc("dmu write parallel");
24355530Sbonwick 		}
24365530Sbonwick 		dmu_tx_abort(tx);
24379412SAleksandr.Guzovskiy@Sun.COM 		if (abuf != NULL) {
24389412SAleksandr.Guzovskiy@Sun.COM 			dmu_return_arcbuf(abuf);
24399412SAleksandr.Guzovskiy@Sun.COM 			dmu_buf_rele(bonus_db, FTAG);
24409412SAleksandr.Guzovskiy@Sun.COM 		}
24415530Sbonwick 		return;
24425530Sbonwick 	}
24437762SJeff.Bonwick@Sun.COM 	txg = dmu_tx_get_txg(tx);
24445530Sbonwick 
24455530Sbonwick 	lp = &ztest_shared->zs_sync_lock[b];
24465530Sbonwick 	(void) mutex_lock(lp);
24475530Sbonwick 
24485530Sbonwick 	wbt->bt_objset = dmu_objset_id(os);
24495530Sbonwick 	wbt->bt_object = ZTEST_DIROBJ;
24505530Sbonwick 	wbt->bt_offset = off;
24517762SJeff.Bonwick@Sun.COM 	wbt->bt_txg = txg;
24525530Sbonwick 	wbt->bt_thread = za->za_instance;
24535530Sbonwick 	wbt->bt_seq = ztest_shared->zs_seq[b]++;	/* protected by lp */
24545530Sbonwick 
24557762SJeff.Bonwick@Sun.COM 	/*
24567762SJeff.Bonwick@Sun.COM 	 * Occasionally, write an all-zero block to test the behavior
24577762SJeff.Bonwick@Sun.COM 	 * of blocks that compress into holes.
24587762SJeff.Bonwick@Sun.COM 	 */
24597762SJeff.Bonwick@Sun.COM 	if (off != -1ULL && ztest_random(8) == 0)
24607762SJeff.Bonwick@Sun.COM 		bzero(wbt, btsize);
24617762SJeff.Bonwick@Sun.COM 
24625530Sbonwick 	if (off == -1ULL) {
24635530Sbonwick 		dmu_object_info_t *doi = &za->za_doi;
24645530Sbonwick 		char *dboff;
24655530Sbonwick 
24665530Sbonwick 		VERIFY(dmu_bonus_hold(os, ZTEST_DIROBJ, FTAG, &db) == 0);
24675530Sbonwick 		za->za_dbuf = db;
24685530Sbonwick 		dmu_object_info_from_db(db, doi);
24695530Sbonwick 		ASSERT3U(doi->doi_bonus_size, <=, db->db_size);
24705530Sbonwick 		ASSERT3U(doi->doi_bonus_size, >=, btsize);
24715530Sbonwick 		ASSERT3U(doi->doi_bonus_size % btsize, ==, 0);
24725530Sbonwick 		dboff = (char *)db->db_data + doi->doi_bonus_size - btsize;
24735530Sbonwick 		bcopy(dboff, rbt, btsize);
24745530Sbonwick 		if (rbt->bt_objset != 0) {
24755530Sbonwick 			ASSERT3U(rbt->bt_objset, ==, wbt->bt_objset);
24765530Sbonwick 			ASSERT3U(rbt->bt_object, ==, wbt->bt_object);
24775530Sbonwick 			ASSERT3U(rbt->bt_offset, ==, wbt->bt_offset);
24785530Sbonwick 			ASSERT3U(rbt->bt_txg, <=, wbt->bt_txg);
24795530Sbonwick 		}
24805530Sbonwick 		if (ztest_random(10) == 0) {
24815530Sbonwick 			int newsize = (ztest_random(db->db_size /
24825530Sbonwick 			    btsize) + 1) * btsize;
24835530Sbonwick 
24845530Sbonwick 			ASSERT3U(newsize, >=, btsize);
24855530Sbonwick 			ASSERT3U(newsize, <=, db->db_size);
24865530Sbonwick 			VERIFY3U(dmu_set_bonus(db, newsize, tx), ==, 0);
24875530Sbonwick 			dboff = (char *)db->db_data + newsize - btsize;
24885530Sbonwick 		}
24895530Sbonwick 		dmu_buf_will_dirty(db, tx);
24905530Sbonwick 		bcopy(wbt, dboff, btsize);
24915530Sbonwick 		dmu_buf_rele(db, FTAG);
24925530Sbonwick 		za->za_dbuf = NULL;
24935530Sbonwick 	} else if (do_free) {
24945530Sbonwick 		VERIFY(dmu_free_range(os, ZTEST_DIROBJ, off, bs, tx) == 0);
24959412SAleksandr.Guzovskiy@Sun.COM 	} else if (abuf == NULL) {
24969412SAleksandr.Guzovskiy@Sun.COM 		dmu_write(os, ZTEST_DIROBJ, off, btsize, wbt, tx);
24975530Sbonwick 	} else {
24989412SAleksandr.Guzovskiy@Sun.COM 		bcopy(wbt, abuf->b_data, btsize);
24999412SAleksandr.Guzovskiy@Sun.COM 		dmu_assign_arcbuf(bonus_db, off, abuf, tx);
25009412SAleksandr.Guzovskiy@Sun.COM 		dmu_buf_rele(bonus_db, FTAG);
25015530Sbonwick 	}
25025530Sbonwick 
25035530Sbonwick 	(void) mutex_unlock(lp);
25045530Sbonwick 
25055530Sbonwick 	if (ztest_random(1000) == 0)
25065530Sbonwick 		(void) poll(NULL, 0, 1); /* open dn_notxholds window */
25075530Sbonwick 
25085530Sbonwick 	dmu_tx_commit(tx);
25095530Sbonwick 
25105530Sbonwick 	if (ztest_random(10000) == 0)
25117762SJeff.Bonwick@Sun.COM 		txg_wait_synced(dmu_objset_pool(os), txg);
25125530Sbonwick 
25137754SJeff.Bonwick@Sun.COM 	if (off == -1ULL || do_free)
25145530Sbonwick 		return;
25155530Sbonwick 
25165530Sbonwick 	if (ztest_random(2) != 0)
25175530Sbonwick 		return;
25185530Sbonwick 
25195530Sbonwick 	/*
25205530Sbonwick 	 * dmu_sync() the block we just wrote.
25215530Sbonwick 	 */
25225530Sbonwick 	(void) mutex_lock(lp);
25235530Sbonwick 
25245530Sbonwick 	blkoff = P2ALIGN_TYPED(off, bs, uint64_t);
25255530Sbonwick 	error = dmu_buf_hold(os, ZTEST_DIROBJ, blkoff, FTAG, &db);
25265530Sbonwick 	za->za_dbuf = db;
25275530Sbonwick 	if (error) {
25285530Sbonwick 		(void) mutex_unlock(lp);
25295530Sbonwick 		return;
25305530Sbonwick 	}
25315530Sbonwick 	blkoff = off - blkoff;
25327762SJeff.Bonwick@Sun.COM 	error = dmu_sync(NULL, db, &blk, txg, NULL, NULL);
25335530Sbonwick 	dmu_buf_rele(db, FTAG);
25345530Sbonwick 	za->za_dbuf = NULL;
25355530Sbonwick 
25369412SAleksandr.Guzovskiy@Sun.COM 	if (error) {
25379412SAleksandr.Guzovskiy@Sun.COM 		(void) mutex_unlock(lp);
25385530Sbonwick 		return;
25399412SAleksandr.Guzovskiy@Sun.COM 	}
25409412SAleksandr.Guzovskiy@Sun.COM 
25419412SAleksandr.Guzovskiy@Sun.COM 	if (blk.blk_birth == 0)	{	/* concurrent free */
25429412SAleksandr.Guzovskiy@Sun.COM 		(void) mutex_unlock(lp);
25435530Sbonwick 		return;
25449412SAleksandr.Guzovskiy@Sun.COM 	}
25455530Sbonwick 
25465530Sbonwick 	txg_suspend(dmu_objset_pool(os));
25475530Sbonwick 
25489412SAleksandr.Guzovskiy@Sun.COM 	(void) mutex_unlock(lp);
25499412SAleksandr.Guzovskiy@Sun.COM 
25505530Sbonwick 	ASSERT(blk.blk_fill == 1);
25515530Sbonwick 	ASSERT3U(BP_GET_TYPE(&blk), ==, DMU_OT_UINT64_OTHER);
25525530Sbonwick 	ASSERT3U(BP_GET_LEVEL(&blk), ==, 0);
25535530Sbonwick 	ASSERT3U(BP_GET_LSIZE(&blk), ==, bs);
25545530Sbonwick 
25555530Sbonwick 	/*
25565530Sbonwick 	 * Read the block that dmu_sync() returned to make sure its contents
25575530Sbonwick 	 * match what we wrote.  We do this while still txg_suspend()ed
25585530Sbonwick 	 * to ensure that the block can't be reused before we read it.
25595530Sbonwick 	 */
25605530Sbonwick 	zb.zb_objset = dmu_objset_id(os);
25615530Sbonwick 	zb.zb_object = ZTEST_DIROBJ;
25625530Sbonwick 	zb.zb_level = 0;
25635530Sbonwick 	zb.zb_blkid = off / bs;
25645530Sbonwick 	error = zio_wait(zio_read(NULL, za->za_spa, &blk, iobuf, bs,
25655530Sbonwick 	    NULL, NULL, ZIO_PRIORITY_SYNC_READ, ZIO_FLAG_MUSTSUCCEED, &zb));
25665530Sbonwick 	ASSERT3U(error, ==, 0);
25675530Sbonwick 
25685530Sbonwick 	txg_resume(dmu_objset_pool(os));
25695530Sbonwick 
25705530Sbonwick 	bcopy(&iobuf[blkoff], rbt, btsize);
25715530Sbonwick 
25725530Sbonwick 	if (rbt->bt_objset == 0)		/* concurrent free */
25735530Sbonwick 		return;
25745530Sbonwick 
25757762SJeff.Bonwick@Sun.COM 	if (wbt->bt_objset == 0)		/* all-zero overwrite */
25767762SJeff.Bonwick@Sun.COM 		return;
25777762SJeff.Bonwick@Sun.COM 
25785530Sbonwick 	ASSERT3U(rbt->bt_objset, ==, wbt->bt_objset);
25795530Sbonwick 	ASSERT3U(rbt->bt_object, ==, wbt->bt_object);
25805530Sbonwick 	ASSERT3U(rbt->bt_offset, ==, wbt->bt_offset);
25815530Sbonwick 
25825530Sbonwick 	/*
25835530Sbonwick 	 * The semantic of dmu_sync() is that we always push the most recent
25845530Sbonwick 	 * version of the data, so in the face of concurrent updates we may
25855530Sbonwick 	 * see a newer version of the block.  That's OK.
25865530Sbonwick 	 */
25875530Sbonwick 	ASSERT3U(rbt->bt_txg, >=, wbt->bt_txg);
25885530Sbonwick 	if (rbt->bt_thread == wbt->bt_thread)
25895530Sbonwick 		ASSERT3U(rbt->bt_seq, ==, wbt->bt_seq);
25905530Sbonwick 	else
25915530Sbonwick 		ASSERT3U(rbt->bt_seq, >, wbt->bt_seq);
2592789Sahrens }
2593789Sahrens 
2594789Sahrens /*
2595789Sahrens  * Verify that zap_{create,destroy,add,remove,update} work as expected.
2596789Sahrens  */
2597789Sahrens #define	ZTEST_ZAP_MIN_INTS	1
2598789Sahrens #define	ZTEST_ZAP_MAX_INTS	4
2599789Sahrens #define	ZTEST_ZAP_MAX_PROPS	1000
2600789Sahrens 
2601789Sahrens void
2602789Sahrens ztest_zap(ztest_args_t *za)
2603789Sahrens {
2604789Sahrens 	objset_t *os = za->za_os;
2605789Sahrens 	uint64_t object;
2606789Sahrens 	uint64_t txg, last_txg;
2607789Sahrens 	uint64_t value[ZTEST_ZAP_MAX_INTS];
2608789Sahrens 	uint64_t zl_ints, zl_intsize, prop;
2609789Sahrens 	int i, ints;
2610789Sahrens 	dmu_tx_t *tx;
2611789Sahrens 	char propname[100], txgname[100];
2612789Sahrens 	int error;
2613789Sahrens 	char osname[MAXNAMELEN];
2614789Sahrens 	char *hc[2] = { "s.acl.h", ".s.open.h.hyLZlg" };
2615789Sahrens 
2616789Sahrens 	dmu_objset_name(os, osname);
2617789Sahrens 
2618789Sahrens 	/*
2619789Sahrens 	 * Create a new object if necessary, and record it in the directory.
2620789Sahrens 	 */
26211544Seschrock 	VERIFY(0 == dmu_read(os, ZTEST_DIROBJ, za->za_diroff,
26221544Seschrock 	    sizeof (uint64_t), &object));
2623789Sahrens 
2624789Sahrens 	if (object == 0) {
2625789Sahrens 		tx = dmu_tx_create(os);
2626789Sahrens 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff,
2627789Sahrens 		    sizeof (uint64_t));
26281544Seschrock 		dmu_tx_hold_zap(tx, DMU_NEW_OBJECT, TRUE, NULL);
2629789Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
2630789Sahrens 		if (error) {
2631789Sahrens 			ztest_record_enospc("create zap test obj");
2632789Sahrens 			dmu_tx_abort(tx);
2633789Sahrens 			return;
2634789Sahrens 		}
2635789Sahrens 		object = zap_create(os, DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx);
2636789Sahrens 		if (error) {
2637789Sahrens 			fatal(0, "zap_create('%s', %llu) = %d",
2638789Sahrens 			    osname, object, error);
2639789Sahrens 		}
2640789Sahrens 		ASSERT(object != 0);
2641789Sahrens 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff,
2642789Sahrens 		    sizeof (uint64_t), &object, tx);
2643789Sahrens 		/*
2644789Sahrens 		 * Generate a known hash collision, and verify that
2645789Sahrens 		 * we can lookup and remove both entries.
2646789Sahrens 		 */
2647789Sahrens 		for (i = 0; i < 2; i++) {
2648789Sahrens 			value[i] = i;
2649789Sahrens 			error = zap_add(os, object, hc[i], sizeof (uint64_t),
2650789Sahrens 			    1, &value[i], tx);
2651789Sahrens 			ASSERT3U(error, ==, 0);
2652789Sahrens 		}
2653789Sahrens 		for (i = 0; i < 2; i++) {
2654789Sahrens 			error = zap_add(os, object, hc[i], sizeof (uint64_t),
2655789Sahrens 			    1, &value[i], tx);
2656789Sahrens 			ASSERT3U(error, ==, EEXIST);
2657789Sahrens 			error = zap_length(os, object, hc[i],
2658789Sahrens 			    &zl_intsize, &zl_ints);
2659789Sahrens 			ASSERT3U(error, ==, 0);
2660789Sahrens 			ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
2661789Sahrens 			ASSERT3U(zl_ints, ==, 1);
2662789Sahrens 		}
2663789Sahrens 		for (i = 0; i < 2; i++) {
2664789Sahrens 			error = zap_remove(os, object, hc[i], tx);
2665789Sahrens 			ASSERT3U(error, ==, 0);
2666789Sahrens 		}
2667789Sahrens 
2668789Sahrens 		dmu_tx_commit(tx);
2669789Sahrens 	}
2670789Sahrens 
2671789Sahrens 	ints = MAX(ZTEST_ZAP_MIN_INTS, object % ZTEST_ZAP_MAX_INTS);
2672789Sahrens 
26735530Sbonwick 	prop = ztest_random(ZTEST_ZAP_MAX_PROPS);
26745530Sbonwick 	(void) sprintf(propname, "prop_%llu", (u_longlong_t)prop);
26755530Sbonwick 	(void) sprintf(txgname, "txg_%llu", (u_longlong_t)prop);
26765530Sbonwick 	bzero(value, sizeof (value));
26775530Sbonwick 	last_txg = 0;
26785530Sbonwick 
26795530Sbonwick 	/*
26805530Sbonwick 	 * If these zap entries already exist, validate their contents.
26815530Sbonwick 	 */
26825530Sbonwick 	error = zap_length(os, object, txgname, &zl_intsize, &zl_ints);
26835530Sbonwick 	if (error == 0) {
26845530Sbonwick 		ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
26855530Sbonwick 		ASSERT3U(zl_ints, ==, 1);
26865530Sbonwick 
26875530Sbonwick 		VERIFY(zap_lookup(os, object, txgname, zl_intsize,
26885530Sbonwick 		    zl_ints, &last_txg) == 0);
26895530Sbonwick 
26905530Sbonwick 		VERIFY(zap_length(os, object, propname, &zl_intsize,
26915530Sbonwick 		    &zl_ints) == 0);
26925530Sbonwick 
26935530Sbonwick 		ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
26945530Sbonwick 		ASSERT3U(zl_ints, ==, ints);
26955530Sbonwick 
26965530Sbonwick 		VERIFY(zap_lookup(os, object, propname, zl_intsize,
26975530Sbonwick 		    zl_ints, value) == 0);
26985530Sbonwick 
26995530Sbonwick 		for (i = 0; i < ints; i++) {
27005530Sbonwick 			ASSERT3U(value[i], ==, last_txg + object + i);
2701789Sahrens 		}
27025530Sbonwick 	} else {
27035530Sbonwick 		ASSERT3U(error, ==, ENOENT);
27045530Sbonwick 	}
27055530Sbonwick 
27065530Sbonwick 	/*
27075530Sbonwick 	 * Atomically update two entries in our zap object.
27085530Sbonwick 	 * The first is named txg_%llu, and contains the txg
27095530Sbonwick 	 * in which the property was last updated.  The second
27105530Sbonwick 	 * is named prop_%llu, and the nth element of its value
27115530Sbonwick 	 * should be txg + object + n.
27125530Sbonwick 	 */
27135530Sbonwick 	tx = dmu_tx_create(os);
27145530Sbonwick 	dmu_tx_hold_zap(tx, object, TRUE, NULL);
27155530Sbonwick 	error = dmu_tx_assign(tx, TXG_WAIT);
27165530Sbonwick 	if (error) {
27175530Sbonwick 		ztest_record_enospc("create zap entry");
27185530Sbonwick 		dmu_tx_abort(tx);
27195530Sbonwick 		return;
2720789Sahrens 	}
27215530Sbonwick 	txg = dmu_tx_get_txg(tx);
27225530Sbonwick 
27235530Sbonwick 	if (last_txg > txg)
27245530Sbonwick 		fatal(0, "zap future leak: old %llu new %llu", last_txg, txg);
27255530Sbonwick 
27265530Sbonwick 	for (i = 0; i < ints; i++)
27275530Sbonwick 		value[i] = txg + object + i;
27285530Sbonwick 
27295530Sbonwick 	error = zap_update(os, object, txgname, sizeof (uint64_t), 1, &txg, tx);
27305530Sbonwick 	if (error)
27315530Sbonwick 		fatal(0, "zap_update('%s', %llu, '%s') = %d",
27325530Sbonwick 		    osname, object, txgname, error);
27335530Sbonwick 
27345530Sbonwick 	error = zap_update(os, object, propname, sizeof (uint64_t),
27355530Sbonwick 	    ints, value, tx);
27365530Sbonwick 	if (error)
27375530Sbonwick 		fatal(0, "zap_update('%s', %llu, '%s') = %d",
27385530Sbonwick 		    osname, object, propname, error);
27395530Sbonwick 
27405530Sbonwick 	dmu_tx_commit(tx);
27415530Sbonwick 
27425530Sbonwick 	/*
27435530Sbonwick 	 * Remove a random pair of entries.
27445530Sbonwick 	 */
27455530Sbonwick 	prop = ztest_random(ZTEST_ZAP_MAX_PROPS);
27465530Sbonwick 	(void) sprintf(propname, "prop_%llu", (u_longlong_t)prop);
27475530Sbonwick 	(void) sprintf(txgname, "txg_%llu", (u_longlong_t)prop);
27485530Sbonwick 
27495530Sbonwick 	error = zap_length(os, object, txgname, &zl_intsize, &zl_ints);
27505530Sbonwick 
27515530Sbonwick 	if (error == ENOENT)
27525530Sbonwick 		return;
27535530Sbonwick 
27545530Sbonwick 	ASSERT3U(error, ==, 0);
27555530Sbonwick 
27565530Sbonwick 	tx = dmu_tx_create(os);
27575530Sbonwick 	dmu_tx_hold_zap(tx, object, TRUE, NULL);
27585530Sbonwick 	error = dmu_tx_assign(tx, TXG_WAIT);
27595530Sbonwick 	if (error) {
27605530Sbonwick 		ztest_record_enospc("remove zap entry");
27615530Sbonwick 		dmu_tx_abort(tx);
27625530Sbonwick 		return;
27635530Sbonwick 	}
27645530Sbonwick 	error = zap_remove(os, object, txgname, tx);
27655530Sbonwick 	if (error)
27665530Sbonwick 		fatal(0, "zap_remove('%s', %llu, '%s') = %d",
27675530Sbonwick 		    osname, object, txgname, error);
27685530Sbonwick 
27695530Sbonwick 	error = zap_remove(os, object, propname, tx);
27705530Sbonwick 	if (error)
27715530Sbonwick 		fatal(0, "zap_remove('%s', %llu, '%s') = %d",
27725530Sbonwick 		    osname, object, propname, error);
27735530Sbonwick 
27745530Sbonwick 	dmu_tx_commit(tx);
2775789Sahrens 
2776789Sahrens 	/*
2777789Sahrens 	 * Once in a while, destroy the object.
2778789Sahrens 	 */
27795530Sbonwick 	if (ztest_random(1000) != 0)
2780789Sahrens 		return;
2781789Sahrens 
2782789Sahrens 	tx = dmu_tx_create(os);
2783789Sahrens 	dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff, sizeof (uint64_t));
2784789Sahrens 	dmu_tx_hold_free(tx, object, 0, DMU_OBJECT_END);
2785789Sahrens 	error = dmu_tx_assign(tx, TXG_WAIT);
2786789Sahrens 	if (error) {
2787789Sahrens 		ztest_record_enospc("destroy zap object");
2788789Sahrens 		dmu_tx_abort(tx);
2789789Sahrens 		return;
2790789Sahrens 	}
2791789Sahrens 	error = zap_destroy(os, object, tx);
2792789Sahrens 	if (error)
2793789Sahrens 		fatal(0, "zap_destroy('%s', %llu) = %d",
2794789Sahrens 		    osname, object, error);
2795789Sahrens 	object = 0;
2796789Sahrens 	dmu_write(os, ZTEST_DIROBJ, za->za_diroff, sizeof (uint64_t),
2797789Sahrens 	    &object, tx);
2798789Sahrens 	dmu_tx_commit(tx);
2799789Sahrens }
2800789Sahrens 
2801789Sahrens void
2802789Sahrens ztest_zap_parallel(ztest_args_t *za)
2803789Sahrens {
2804789Sahrens 	objset_t *os = za->za_os;
2805789Sahrens 	uint64_t txg, object, count, wsize, wc, zl_wsize, zl_wc;
2806789Sahrens 	dmu_tx_t *tx;
2807789Sahrens 	int i, namelen, error;
2808789Sahrens 	char name[20], string_value[20];
2809789Sahrens 	void *data;
2810789Sahrens 
28115530Sbonwick 	/*
28125530Sbonwick 	 * Generate a random name of the form 'xxx.....' where each
28135530Sbonwick 	 * x is a random printable character and the dots are dots.
28145530Sbonwick 	 * There are 94 such characters, and the name length goes from
28155530Sbonwick 	 * 6 to 20, so there are 94^3 * 15 = 12,458,760 possible names.
28165530Sbonwick 	 */
28175530Sbonwick 	namelen = ztest_random(sizeof (name) - 5) + 5 + 1;
28185530Sbonwick 
28195530Sbonwick 	for (i = 0; i < 3; i++)
28205530Sbonwick 		name[i] = '!' + ztest_random('~' - '!' + 1);
28215530Sbonwick 	for (; i < namelen - 1; i++)
28225530Sbonwick 		name[i] = '.';
28235530Sbonwick 	name[i] = '\0';
28245530Sbonwick 
28255530Sbonwick 	if (ztest_random(2) == 0)
28265530Sbonwick 		object = ZTEST_MICROZAP_OBJ;
28275530Sbonwick 	else
28285530Sbonwick 		object = ZTEST_FATZAP_OBJ;
28295530Sbonwick 
28305530Sbonwick 	if ((namelen & 1) || object == ZTEST_MICROZAP_OBJ) {
28315530Sbonwick 		wsize = sizeof (txg);
28325530Sbonwick 		wc = 1;
28335530Sbonwick 		data = &txg;
28345530Sbonwick 	} else {
28355530Sbonwick 		wsize = 1;
28365530Sbonwick 		wc = namelen;
28375530Sbonwick 		data = string_value;
28385530Sbonwick 	}
28395530Sbonwick 
28405530Sbonwick 	count = -1ULL;
28415530Sbonwick 	VERIFY(zap_count(os, object, &count) == 0);
28425530Sbonwick 	ASSERT(count != -1ULL);
28435530Sbonwick 
28445530Sbonwick 	/*
28455530Sbonwick 	 * Select an operation: length, lookup, add, update, remove.
28465530Sbonwick 	 */
28475530Sbonwick 	i = ztest_random(5);
28485530Sbonwick 
28495530Sbonwick 	if (i >= 2) {
28505530Sbonwick 		tx = dmu_tx_create(os);
28515530Sbonwick 		dmu_tx_hold_zap(tx, object, TRUE, NULL);
28525530Sbonwick 		error = dmu_tx_assign(tx, TXG_WAIT);
28535530Sbonwick 		if (error) {
28545530Sbonwick 			ztest_record_enospc("zap parallel");
28555530Sbonwick 			dmu_tx_abort(tx);
28565530Sbonwick 			return;
2857789Sahrens 		}
28585530Sbonwick 		txg = dmu_tx_get_txg(tx);
28595530Sbonwick 		bcopy(name, string_value, namelen);
28605530Sbonwick 	} else {
28615530Sbonwick 		tx = NULL;
28625530Sbonwick 		txg = 0;
28635530Sbonwick 		bzero(string_value, namelen);
28645530Sbonwick 	}
28655530Sbonwick 
28665530Sbonwick 	switch (i) {
28675530Sbonwick 
28685530Sbonwick 	case 0:
28695530Sbonwick 		error = zap_length(os, object, name, &zl_wsize, &zl_wc);
28705530Sbonwick 		if (error == 0) {
28715530Sbonwick 			ASSERT3U(wsize, ==, zl_wsize);
28725530Sbonwick 			ASSERT3U(wc, ==, zl_wc);
2873789Sahrens 		} else {
28745530Sbonwick 			ASSERT3U(error, ==, ENOENT);
2875789Sahrens 		}
28765530Sbonwick 		break;
28775530Sbonwick 
28785530Sbonwick 	case 1:
28795530Sbonwick 		error = zap_lookup(os, object, name, wsize, wc, data);
28805530Sbonwick 		if (error == 0) {
28815530Sbonwick 			if (data == string_value &&
28825530Sbonwick 			    bcmp(name, data, namelen) != 0)
28835530Sbonwick 				fatal(0, "name '%s' != val '%s' len %d",
28845530Sbonwick 				    name, data, namelen);
28855530Sbonwick 		} else {
28865530Sbonwick 			ASSERT3U(error, ==, ENOENT);
2887789Sahrens 		}
28885530Sbonwick 		break;
28895530Sbonwick 
28905530Sbonwick 	case 2:
28915530Sbonwick 		error = zap_add(os, object, name, wsize, wc, data, tx);
28925530Sbonwick 		ASSERT(error == 0 || error == EEXIST);
28935530Sbonwick 		break;
28945530Sbonwick 
28955530Sbonwick 	case 3:
28965530Sbonwick 		VERIFY(zap_update(os, object, name, wsize, wc, data, tx) == 0);
28975530Sbonwick 		break;
28985530Sbonwick 
28995530Sbonwick 	case 4:
29005530Sbonwick 		error = zap_remove(os, object, name, tx);
29015530Sbonwick 		ASSERT(error == 0 || error == ENOENT);
29025530Sbonwick 		break;
2903789Sahrens 	}
29045530Sbonwick 
29055530Sbonwick 	if (tx != NULL)
29065530Sbonwick 		dmu_tx_commit(tx);
2907789Sahrens }
2908789Sahrens 
2909789Sahrens void
2910789Sahrens ztest_dsl_prop_get_set(ztest_args_t *za)
2911789Sahrens {
2912789Sahrens 	objset_t *os = za->za_os;
2913789Sahrens 	int i, inherit;
2914789Sahrens 	uint64_t value;
2915789Sahrens 	const char *prop, *valname;
2916789Sahrens 	char setpoint[MAXPATHLEN];
2917789Sahrens 	char osname[MAXNAMELEN];
29181544Seschrock 	int error;
2919789Sahrens 
2920789Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
2921789Sahrens 
2922789Sahrens 	dmu_objset_name(os, osname);
2923789Sahrens 
2924789Sahrens 	for (i = 0; i < 2; i++) {
2925789Sahrens 		if (i == 0) {
2926789Sahrens 			prop = "checksum";
2927789Sahrens 			value = ztest_random_checksum();
2928789Sahrens 			inherit = (value == ZIO_CHECKSUM_INHERIT);
2929789Sahrens 		} else {
2930789Sahrens 			prop = "compression";
2931789Sahrens 			value = ztest_random_compress();
2932789Sahrens 			inherit = (value == ZIO_COMPRESS_INHERIT);
2933789Sahrens 		}
2934789Sahrens 
29351544Seschrock 		error = dsl_prop_set(osname, prop, sizeof (value),
29361544Seschrock 		    !inherit, &value);
29371544Seschrock 
29381544Seschrock 		if (error == ENOSPC) {
29391544Seschrock 			ztest_record_enospc("dsl_prop_set");
29401544Seschrock 			break;
29411544Seschrock 		}
29421544Seschrock 
29431544Seschrock 		ASSERT3U(error, ==, 0);
2944789Sahrens 
2945789Sahrens 		VERIFY3U(dsl_prop_get(osname, prop, sizeof (value),
2946789Sahrens 		    1, &value, setpoint), ==, 0);
2947789Sahrens 
2948789Sahrens 		if (i == 0)
2949789Sahrens 			valname = zio_checksum_table[value].ci_name;
2950789Sahrens 		else
2951789Sahrens 			valname = zio_compress_table[value].ci_name;
2952789Sahrens 
2953789Sahrens 		if (zopt_verbose >= 6) {
2954789Sahrens 			(void) printf("%s %s = %s for '%s'\n",
2955789Sahrens 			    osname, prop, valname, setpoint);
2956789Sahrens 		}
2957789Sahrens 	}
2958789Sahrens 
2959789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
2960789Sahrens }
2961789Sahrens 
2962789Sahrens /*
2963789Sahrens  * Inject random faults into the on-disk data.
2964789Sahrens  */
2965789Sahrens void
2966789Sahrens ztest_fault_inject(ztest_args_t *za)
2967789Sahrens {
2968789Sahrens 	int fd;
2969789Sahrens 	uint64_t offset;
2970789Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
2971789Sahrens 	uint64_t bad = 0x1990c0ffeedecade;
2972789Sahrens 	uint64_t top, leaf;
2973789Sahrens 	char path0[MAXPATHLEN];
2974789Sahrens 	char pathrand[MAXPATHLEN];
2975789Sahrens 	size_t fsize;
29765530Sbonwick 	spa_t *spa = za->za_spa;
2977789Sahrens 	int bshift = SPA_MAXBLOCKSHIFT + 2;	/* don't scrog all labels */
2978789Sahrens 	int iters = 1000;
29797754SJeff.Bonwick@Sun.COM 	int maxfaults = zopt_maxfaults;
29807754SJeff.Bonwick@Sun.COM 	vdev_t *vd0 = NULL;
29811544Seschrock 	uint64_t guid0 = 0;
29821544Seschrock 
29837754SJeff.Bonwick@Sun.COM 	ASSERT(leaves >= 1);
2984789Sahrens 
2985789Sahrens 	/*
29867754SJeff.Bonwick@Sun.COM 	 * We need SCL_STATE here because we're going to look at vd0->vdev_tsd.
2987789Sahrens 	 */
29887754SJeff.Bonwick@Sun.COM 	spa_config_enter(spa, SCL_STATE, FTAG, RW_READER);
29897754SJeff.Bonwick@Sun.COM 
29907754SJeff.Bonwick@Sun.COM 	if (ztest_random(2) == 0) {
29917754SJeff.Bonwick@Sun.COM 		/*
29927754SJeff.Bonwick@Sun.COM 		 * Inject errors on a normal data device.
29937754SJeff.Bonwick@Sun.COM 		 */
29947754SJeff.Bonwick@Sun.COM 		top = ztest_random(spa->spa_root_vdev->vdev_children);
29957754SJeff.Bonwick@Sun.COM 		leaf = ztest_random(leaves);
29967754SJeff.Bonwick@Sun.COM 
29977754SJeff.Bonwick@Sun.COM 		/*
29987754SJeff.Bonwick@Sun.COM 		 * Generate paths to the first leaf in this top-level vdev,
29997754SJeff.Bonwick@Sun.COM 		 * and to the random leaf we selected.  We'll induce transient
30007754SJeff.Bonwick@Sun.COM 		 * write failures and random online/offline activity on leaf 0,
30017754SJeff.Bonwick@Sun.COM 		 * and we'll write random garbage to the randomly chosen leaf.
30027754SJeff.Bonwick@Sun.COM 		 */
30037754SJeff.Bonwick@Sun.COM 		(void) snprintf(path0, sizeof (path0), ztest_dev_template,
30047754SJeff.Bonwick@Sun.COM 		    zopt_dir, zopt_pool, top * leaves + 0);
30057754SJeff.Bonwick@Sun.COM 		(void) snprintf(pathrand, sizeof (pathrand), ztest_dev_template,
30067754SJeff.Bonwick@Sun.COM 		    zopt_dir, zopt_pool, top * leaves + leaf);
30077754SJeff.Bonwick@Sun.COM 
30087754SJeff.Bonwick@Sun.COM 		vd0 = vdev_lookup_by_path(spa->spa_root_vdev, path0);
30097754SJeff.Bonwick@Sun.COM 		if (vd0 != NULL && maxfaults != 1) {
30107754SJeff.Bonwick@Sun.COM 			/*
30117754SJeff.Bonwick@Sun.COM 			 * Make vd0 explicitly claim to be unreadable,
30127754SJeff.Bonwick@Sun.COM 			 * or unwriteable, or reach behind its back
30137754SJeff.Bonwick@Sun.COM 			 * and close the underlying fd.  We can do this if
30147754SJeff.Bonwick@Sun.COM 			 * maxfaults == 0 because we'll fail and reexecute,
30157754SJeff.Bonwick@Sun.COM 			 * and we can do it if maxfaults >= 2 because we'll
30167754SJeff.Bonwick@Sun.COM 			 * have enough redundancy.  If maxfaults == 1, the
30177754SJeff.Bonwick@Sun.COM 			 * combination of this with injection of random data
30187754SJeff.Bonwick@Sun.COM 			 * corruption below exceeds the pool's fault tolerance.
30197754SJeff.Bonwick@Sun.COM 			 */
30207754SJeff.Bonwick@Sun.COM 			vdev_file_t *vf = vd0->vdev_tsd;
30217754SJeff.Bonwick@Sun.COM 
30227754SJeff.Bonwick@Sun.COM 			if (vf != NULL && ztest_random(3) == 0) {
30237754SJeff.Bonwick@Sun.COM 				(void) close(vf->vf_vnode->v_fd);
30247754SJeff.Bonwick@Sun.COM 				vf->vf_vnode->v_fd = -1;
30257754SJeff.Bonwick@Sun.COM 			} else if (ztest_random(2) == 0) {
30267754SJeff.Bonwick@Sun.COM 				vd0->vdev_cant_read = B_TRUE;
30277754SJeff.Bonwick@Sun.COM 			} else {
30287754SJeff.Bonwick@Sun.COM 				vd0->vdev_cant_write = B_TRUE;
30297754SJeff.Bonwick@Sun.COM 			}
30307754SJeff.Bonwick@Sun.COM 			guid0 = vd0->vdev_guid;
30317754SJeff.Bonwick@Sun.COM 		}
30327754SJeff.Bonwick@Sun.COM 	} else {
30337754SJeff.Bonwick@Sun.COM 		/*
30347754SJeff.Bonwick@Sun.COM 		 * Inject errors on an l2cache device.
30357754SJeff.Bonwick@Sun.COM 		 */
30367754SJeff.Bonwick@Sun.COM 		spa_aux_vdev_t *sav = &spa->spa_l2cache;
30377754SJeff.Bonwick@Sun.COM 
30387754SJeff.Bonwick@Sun.COM 		if (sav->sav_count == 0) {
30397754SJeff.Bonwick@Sun.COM 			spa_config_exit(spa, SCL_STATE, FTAG);
30407754SJeff.Bonwick@Sun.COM 			return;
30417754SJeff.Bonwick@Sun.COM 		}
30427754SJeff.Bonwick@Sun.COM 		vd0 = sav->sav_vdevs[ztest_random(sav->sav_count)];
30437754SJeff.Bonwick@Sun.COM 		guid0 = vd0->vdev_guid;
30447754SJeff.Bonwick@Sun.COM 		(void) strcpy(path0, vd0->vdev_path);
30457754SJeff.Bonwick@Sun.COM 		(void) strcpy(pathrand, vd0->vdev_path);
30467754SJeff.Bonwick@Sun.COM 
30477754SJeff.Bonwick@Sun.COM 		leaf = 0;
30487754SJeff.Bonwick@Sun.COM 		leaves = 1;
30497754SJeff.Bonwick@Sun.COM 		maxfaults = INT_MAX;	/* no limit on cache devices */
30507754SJeff.Bonwick@Sun.COM 	}
3051789Sahrens 
30527754SJeff.Bonwick@Sun.COM 	spa_config_exit(spa, SCL_STATE, FTAG);
30537754SJeff.Bonwick@Sun.COM 
30547754SJeff.Bonwick@Sun.COM 	if (maxfaults == 0)
30557754SJeff.Bonwick@Sun.COM 		return;
30561544Seschrock 
30571544Seschrock 	/*
30581544Seschrock 	 * If we can tolerate two or more faults, randomly online/offline vd0.
30591544Seschrock 	 */
30607754SJeff.Bonwick@Sun.COM 	if (maxfaults >= 2 && guid0 != 0) {
30618241SJeff.Bonwick@Sun.COM 		if (ztest_random(10) < 6) {
30628241SJeff.Bonwick@Sun.COM 			int flags = (ztest_random(2) == 0 ?
30638241SJeff.Bonwick@Sun.COM 			    ZFS_OFFLINE_TEMPORARY : 0);
30648241SJeff.Bonwick@Sun.COM 			VERIFY(vdev_offline(spa, guid0, flags) != EBUSY);
30658241SJeff.Bonwick@Sun.COM 		} else {
30668241SJeff.Bonwick@Sun.COM 			(void) vdev_online(spa, guid0, 0, NULL);
30678241SJeff.Bonwick@Sun.COM 		}
3068789Sahrens 	}
3069789Sahrens 
3070789Sahrens 	/*
30711544Seschrock 	 * We have at least single-fault tolerance, so inject data corruption.
3072789Sahrens 	 */
3073789Sahrens 	fd = open(pathrand, O_RDWR);
3074789Sahrens 
3075789Sahrens 	if (fd == -1)	/* we hit a gap in the device namespace */
3076789Sahrens 		return;
3077789Sahrens 
3078789Sahrens 	fsize = lseek(fd, 0, SEEK_END);
3079789Sahrens 
3080789Sahrens 	while (--iters != 0) {
3081789Sahrens 		offset = ztest_random(fsize / (leaves << bshift)) *
3082789Sahrens 		    (leaves << bshift) + (leaf << bshift) +
3083789Sahrens 		    (ztest_random(1ULL << (bshift - 1)) & -8ULL);
3084789Sahrens 
3085789Sahrens 		if (offset >= fsize)
3086789Sahrens 			continue;
3087789Sahrens 
3088789Sahrens 		if (zopt_verbose >= 6)
3089789Sahrens 			(void) printf("injecting bad word into %s,"
3090789Sahrens 			    " offset 0x%llx\n", pathrand, (u_longlong_t)offset);
3091789Sahrens 
3092789Sahrens 		if (pwrite(fd, &bad, sizeof (bad), offset) != sizeof (bad))
3093789Sahrens 			fatal(1, "can't inject bad word at 0x%llx in %s",
3094789Sahrens 			    offset, pathrand);
3095789Sahrens 	}
3096789Sahrens 
3097789Sahrens 	(void) close(fd);
3098789Sahrens }
3099789Sahrens 
3100789Sahrens /*
3101789Sahrens  * Scrub the pool.
3102789Sahrens  */
3103789Sahrens void
3104789Sahrens ztest_scrub(ztest_args_t *za)
3105789Sahrens {
31065530Sbonwick 	spa_t *spa = za->za_spa;
3107789Sahrens 
31087046Sahrens 	(void) spa_scrub(spa, POOL_SCRUB_EVERYTHING);
3109789Sahrens 	(void) poll(NULL, 0, 1000); /* wait a second, then force a restart */
31107046Sahrens 	(void) spa_scrub(spa, POOL_SCRUB_EVERYTHING);
3111789Sahrens }
3112789Sahrens 
3113789Sahrens /*
3114789Sahrens  * Rename the pool to a different name and then rename it back.
3115789Sahrens  */
3116789Sahrens void
3117789Sahrens ztest_spa_rename(ztest_args_t *za)
3118789Sahrens {
3119789Sahrens 	char *oldname, *newname;
3120789Sahrens 	int error;
3121789Sahrens 	spa_t *spa;
3122789Sahrens 
3123789Sahrens 	(void) rw_wrlock(&ztest_shared->zs_name_lock);
3124789Sahrens 
3125789Sahrens 	oldname = za->za_pool;
3126789Sahrens 	newname = umem_alloc(strlen(oldname) + 5, UMEM_NOFAIL);
3127789Sahrens 	(void) strcpy(newname, oldname);
3128789Sahrens 	(void) strcat(newname, "_tmp");
3129789Sahrens 
3130789Sahrens 	/*
3131789Sahrens 	 * Do the rename
3132789Sahrens 	 */
3133789Sahrens 	error = spa_rename(oldname, newname);
3134789Sahrens 	if (error)
3135789Sahrens 		fatal(0, "spa_rename('%s', '%s') = %d", oldname,
3136789Sahrens 		    newname, error);
3137789Sahrens 
3138789Sahrens 	/*
3139789Sahrens 	 * Try to open it under the old name, which shouldn't exist
3140789Sahrens 	 */
3141789Sahrens 	error = spa_open(oldname, &spa, FTAG);
3142789Sahrens 	if (error != ENOENT)
3143789Sahrens 		fatal(0, "spa_open('%s') = %d", oldname, error);
3144789Sahrens 
3145789Sahrens 	/*
3146789Sahrens 	 * Open it under the new name and make sure it's still the same spa_t.
3147789Sahrens 	 */
3148789Sahrens 	error = spa_open(newname, &spa, FTAG);
3149789Sahrens 	if (error != 0)
3150789Sahrens 		fatal(0, "spa_open('%s') = %d", newname, error);
3151789Sahrens 
31525530Sbonwick 	ASSERT(spa == za->za_spa);
3153789Sahrens 	spa_close(spa, FTAG);
3154789Sahrens 
3155789Sahrens 	/*
3156789Sahrens 	 * Rename it back to the original
3157789Sahrens 	 */
3158789Sahrens 	error = spa_rename(newname, oldname);
3159789Sahrens 	if (error)
3160789Sahrens 		fatal(0, "spa_rename('%s', '%s') = %d", newname,
3161789Sahrens 		    oldname, error);
3162789Sahrens 
3163789Sahrens 	/*
3164789Sahrens 	 * Make sure it can still be opened
3165789Sahrens 	 */
3166789Sahrens 	error = spa_open(oldname, &spa, FTAG);
3167789Sahrens 	if (error != 0)
3168789Sahrens 		fatal(0, "spa_open('%s') = %d", oldname, error);
3169789Sahrens 
31705530Sbonwick 	ASSERT(spa == za->za_spa);
3171789Sahrens 	spa_close(spa, FTAG);
3172789Sahrens 
3173789Sahrens 	umem_free(newname, strlen(newname) + 1);
3174789Sahrens 
3175789Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
3176789Sahrens }
3177789Sahrens 
3178789Sahrens 
3179789Sahrens /*
3180789Sahrens  * Completely obliterate one disk.
3181789Sahrens  */
3182789Sahrens static void
3183789Sahrens ztest_obliterate_one_disk(uint64_t vdev)
3184789Sahrens {
3185789Sahrens 	int fd;
31861807Sbonwick 	char dev_name[MAXPATHLEN], copy_name[MAXPATHLEN];
3187789Sahrens 	size_t fsize;
3188789Sahrens 
3189789Sahrens 	if (zopt_maxfaults < 2)
3190789Sahrens 		return;
3191789Sahrens 
3192789Sahrens 	(void) sprintf(dev_name, ztest_dev_template, zopt_dir, zopt_pool, vdev);
31931807Sbonwick 	(void) snprintf(copy_name, MAXPATHLEN, "%s.old", dev_name);
3194789Sahrens 
3195789Sahrens 	fd = open(dev_name, O_RDWR);
3196789Sahrens 
3197789Sahrens 	if (fd == -1)
3198789Sahrens 		fatal(1, "can't open %s", dev_name);
3199789Sahrens 
3200789Sahrens 	/*
3201789Sahrens 	 * Determine the size.
3202789Sahrens 	 */
3203789Sahrens 	fsize = lseek(fd, 0, SEEK_END);
32041807Sbonwick 
3205789Sahrens 	(void) close(fd);
3206789Sahrens 
3207789Sahrens 	/*
32081807Sbonwick 	 * Rename the old device to dev_name.old (useful for debugging).
3209789Sahrens 	 */
32101807Sbonwick 	VERIFY(rename(dev_name, copy_name) == 0);
3211789Sahrens 
3212789Sahrens 	/*
3213789Sahrens 	 * Create a new one.
3214789Sahrens 	 */
3215789Sahrens 	VERIFY((fd = open(dev_name, O_RDWR | O_CREAT | O_TRUNC, 0666)) >= 0);
3216789Sahrens 	VERIFY(ftruncate(fd, fsize) == 0);
3217789Sahrens 	(void) close(fd);
3218789Sahrens }
3219789Sahrens 
3220789Sahrens static void
3221789Sahrens ztest_replace_one_disk(spa_t *spa, uint64_t vdev)
3222789Sahrens {
3223789Sahrens 	char dev_name[MAXPATHLEN];
32247754SJeff.Bonwick@Sun.COM 	nvlist_t *root;
3225789Sahrens 	int error;
32261544Seschrock 	uint64_t guid;
32271544Seschrock 	vdev_t *vd;
3228789Sahrens 
3229789Sahrens 	(void) sprintf(dev_name, ztest_dev_template, zopt_dir, zopt_pool, vdev);
3230789Sahrens 
3231789Sahrens 	/*
3232789Sahrens 	 * Build the nvlist describing dev_name.
3233789Sahrens 	 */
32347754SJeff.Bonwick@Sun.COM 	root = make_vdev_root(dev_name, NULL, 0, 0, 0, 0, 0, 1);
32357754SJeff.Bonwick@Sun.COM 
32367754SJeff.Bonwick@Sun.COM 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
32371544Seschrock 	if ((vd = vdev_lookup_by_path(spa->spa_root_vdev, dev_name)) == NULL)
32381544Seschrock 		guid = 0;
32391544Seschrock 	else
32401544Seschrock 		guid = vd->vdev_guid;
32417754SJeff.Bonwick@Sun.COM 	spa_config_exit(spa, SCL_VDEV, FTAG);
32421544Seschrock 	error = spa_vdev_attach(spa, guid, root, B_TRUE);
32431732Sbonwick 	if (error != 0 &&
32441732Sbonwick 	    error != EBUSY &&
32451732Sbonwick 	    error != ENOTSUP &&
32461732Sbonwick 	    error != ENODEV &&
32471732Sbonwick 	    error != EDOM)
3248789Sahrens 		fatal(0, "spa_vdev_attach(in-place) = %d", error);
3249789Sahrens 
3250789Sahrens 	nvlist_free(root);
3251789Sahrens }
3252789Sahrens 
3253789Sahrens static void
3254789Sahrens ztest_verify_blocks(char *pool)
3255789Sahrens {
3256789Sahrens 	int status;
3257789Sahrens 	char zdb[MAXPATHLEN + MAXNAMELEN + 20];
3258789Sahrens 	char zbuf[1024];
3259789Sahrens 	char *bin;
32604527Sperrin 	char *ztest;
32614527Sperrin 	char *isa;
32624527Sperrin 	int isalen;
3263789Sahrens 	FILE *fp;
3264789Sahrens 
3265789Sahrens 	(void) realpath(getexecname(), zdb);
3266789Sahrens 
3267789Sahrens 	/* zdb lives in /usr/sbin, while ztest lives in /usr/bin */
3268789Sahrens 	bin = strstr(zdb, "/usr/bin/");
32694527Sperrin 	ztest = strstr(bin, "/ztest");
32704527Sperrin 	isa = bin + 8;
32714527Sperrin 	isalen = ztest - isa;
32724527Sperrin 	isa = strdup(isa);
3273789Sahrens 	/* LINTED */
32745994Sck153898 	(void) sprintf(bin,
32759463SVictor.Latushkin@Sun.COM 	    "/usr/sbin%.*s/zdb -bcc%s%s -U /tmp/zpool.cache %s",
32764527Sperrin 	    isalen,
32774527Sperrin 	    isa,
3278789Sahrens 	    zopt_verbose >= 3 ? "s" : "",
3279789Sahrens 	    zopt_verbose >= 4 ? "v" : "",
32807837SMatthew.Ahrens@Sun.COM 	    pool);
32814527Sperrin 	free(isa);
3282789Sahrens 
3283789Sahrens 	if (zopt_verbose >= 5)
3284789Sahrens 		(void) printf("Executing %s\n", strstr(zdb, "zdb "));
3285789Sahrens 
3286789Sahrens 	fp = popen(zdb, "r");
3287789Sahrens 
3288789Sahrens 	while (fgets(zbuf, sizeof (zbuf), fp) != NULL)
3289789Sahrens 		if (zopt_verbose >= 3)
3290789Sahrens 			(void) printf("%s", zbuf);
3291789Sahrens 
3292789Sahrens 	status = pclose(fp);
3293789Sahrens 
3294789Sahrens 	if (status == 0)
3295789Sahrens 		return;
3296789Sahrens 
3297789Sahrens 	ztest_dump_core = 0;
3298789Sahrens 	if (WIFEXITED(status))
3299789Sahrens 		fatal(0, "'%s' exit code %d", zdb, WEXITSTATUS(status));
3300789Sahrens 	else
3301789Sahrens 		fatal(0, "'%s' died with signal %d", zdb, WTERMSIG(status));
3302789Sahrens }
3303789Sahrens 
3304789Sahrens static void
3305789Sahrens ztest_walk_pool_directory(char *header)
3306789Sahrens {
3307789Sahrens 	spa_t *spa = NULL;
3308789Sahrens 
3309789Sahrens 	if (zopt_verbose >= 6)
3310789Sahrens 		(void) printf("%s\n", header);
3311789Sahrens 
3312789Sahrens 	mutex_enter(&spa_namespace_lock);
3313789Sahrens 	while ((spa = spa_next(spa)) != NULL)
3314789Sahrens 		if (zopt_verbose >= 6)
3315789Sahrens 			(void) printf("\t%s\n", spa_name(spa));
3316789Sahrens 	mutex_exit(&spa_namespace_lock);
3317789Sahrens }
3318789Sahrens 
3319789Sahrens static void
3320789Sahrens ztest_spa_import_export(char *oldname, char *newname)
3321789Sahrens {
33228241SJeff.Bonwick@Sun.COM 	nvlist_t *config, *newconfig;
3323789Sahrens 	uint64_t pool_guid;
3324789Sahrens 	spa_t *spa;
3325789Sahrens 	int error;
3326789Sahrens 
3327789Sahrens 	if (zopt_verbose >= 4) {
3328789Sahrens 		(void) printf("import/export: old = %s, new = %s\n",
3329789Sahrens 		    oldname, newname);
3330789Sahrens 	}
3331789Sahrens 
3332789Sahrens 	/*
3333789Sahrens 	 * Clean up from previous runs.
3334789Sahrens 	 */
3335789Sahrens 	(void) spa_destroy(newname);
3336789Sahrens 
3337789Sahrens 	/*
3338789Sahrens 	 * Get the pool's configuration and guid.
3339789Sahrens 	 */
3340789Sahrens 	error = spa_open(oldname, &spa, FTAG);
3341789Sahrens 	if (error)
3342789Sahrens 		fatal(0, "spa_open('%s') = %d", oldname, error);
3343789Sahrens 
33448241SJeff.Bonwick@Sun.COM 	/*
33458241SJeff.Bonwick@Sun.COM 	 * Kick off a scrub to tickle scrub/export races.
33468241SJeff.Bonwick@Sun.COM 	 */
33478241SJeff.Bonwick@Sun.COM 	if (ztest_random(2) == 0)
33488241SJeff.Bonwick@Sun.COM 		(void) spa_scrub(spa, POOL_SCRUB_EVERYTHING);
33498241SJeff.Bonwick@Sun.COM 
3350789Sahrens 	pool_guid = spa_guid(spa);
3351789Sahrens 	spa_close(spa, FTAG);
3352789Sahrens 
3353789Sahrens 	ztest_walk_pool_directory("pools before export");
3354789Sahrens 
3355789Sahrens 	/*
3356789Sahrens 	 * Export it.
3357789Sahrens 	 */
33588211SGeorge.Wilson@Sun.COM 	error = spa_export(oldname, &config, B_FALSE, B_FALSE);
3359789Sahrens 	if (error)
3360789Sahrens 		fatal(0, "spa_export('%s') = %d", oldname, error);
3361789Sahrens 
3362789Sahrens 	ztest_walk_pool_directory("pools after export");
3363789Sahrens 
3364789Sahrens 	/*
33658241SJeff.Bonwick@Sun.COM 	 * Try to import it.
33668241SJeff.Bonwick@Sun.COM 	 */
33678241SJeff.Bonwick@Sun.COM 	newconfig = spa_tryimport(config);
33688241SJeff.Bonwick@Sun.COM 	ASSERT(newconfig != NULL);
33698241SJeff.Bonwick@Sun.COM 	nvlist_free(newconfig);
33708241SJeff.Bonwick@Sun.COM 
33718241SJeff.Bonwick@Sun.COM 	/*
3372789Sahrens 	 * Import it under the new name.
3373789Sahrens 	 */
3374789Sahrens 	error = spa_import(newname, config, NULL);
3375789Sahrens 	if (error)
3376789Sahrens 		fatal(0, "spa_import('%s') = %d", newname, error);
3377789Sahrens 
3378789Sahrens 	ztest_walk_pool_directory("pools after import");
3379789Sahrens 
3380789Sahrens 	/*
3381789Sahrens 	 * Try to import it again -- should fail with EEXIST.
3382789Sahrens 	 */
3383789Sahrens 	error = spa_import(newname, config, NULL);
3384789Sahrens 	if (error != EEXIST)
3385789Sahrens 		fatal(0, "spa_import('%s') twice", newname);
3386789Sahrens 
3387789Sahrens 	/*
3388789Sahrens 	 * Try to import it under a different name -- should fail with EEXIST.
3389789Sahrens 	 */
3390789Sahrens 	error = spa_import(oldname, config, NULL);
3391789Sahrens 	if (error != EEXIST)
3392789Sahrens 		fatal(0, "spa_import('%s') under multiple names", newname);
3393789Sahrens 
3394789Sahrens 	/*
3395789Sahrens 	 * Verify that the pool is no longer visible under the old name.
3396789Sahrens 	 */
3397789Sahrens 	error = spa_open(oldname, &spa, FTAG);
3398789Sahrens 	if (error != ENOENT)
3399789Sahrens 		fatal(0, "spa_open('%s') = %d", newname, error);
3400789Sahrens 
3401789Sahrens 	/*
3402789Sahrens 	 * Verify that we can open and close the pool using the new name.
3403789Sahrens 	 */
3404789Sahrens 	error = spa_open(newname, &spa, FTAG);
3405789Sahrens 	if (error)
3406789Sahrens 		fatal(0, "spa_open('%s') = %d", newname, error);
3407789Sahrens 	ASSERT(pool_guid == spa_guid(spa));
3408789Sahrens 	spa_close(spa, FTAG);
3409789Sahrens 
3410789Sahrens 	nvlist_free(config);
3411789Sahrens }
3412789Sahrens 
34138241SJeff.Bonwick@Sun.COM static void
34148241SJeff.Bonwick@Sun.COM ztest_resume(spa_t *spa)
34158241SJeff.Bonwick@Sun.COM {
34168241SJeff.Bonwick@Sun.COM 	if (spa_suspended(spa)) {
34178241SJeff.Bonwick@Sun.COM 		spa_vdev_state_enter(spa);
34188241SJeff.Bonwick@Sun.COM 		vdev_clear(spa, NULL);
34198241SJeff.Bonwick@Sun.COM 		(void) spa_vdev_state_exit(spa, NULL, 0);
34209234SGeorge.Wilson@Sun.COM 		(void) zio_resume(spa);
34218241SJeff.Bonwick@Sun.COM 	}
34228241SJeff.Bonwick@Sun.COM }
34238241SJeff.Bonwick@Sun.COM 
34245329Sgw25295 static void *
34258241SJeff.Bonwick@Sun.COM ztest_resume_thread(void *arg)
34265329Sgw25295 {
34277754SJeff.Bonwick@Sun.COM 	spa_t *spa = arg;
34285329Sgw25295 
34295329Sgw25295 	while (!ztest_exiting) {
34307754SJeff.Bonwick@Sun.COM 		(void) poll(NULL, 0, 1000);
34318241SJeff.Bonwick@Sun.COM 		ztest_resume(spa);
34325329Sgw25295 	}
34335329Sgw25295 	return (NULL);
34345329Sgw25295 }
34355329Sgw25295 
3436789Sahrens static void *
3437789Sahrens ztest_thread(void *arg)
3438789Sahrens {
3439789Sahrens 	ztest_args_t *za = arg;
3440789Sahrens 	ztest_shared_t *zs = ztest_shared;
3441789Sahrens 	hrtime_t now, functime;
3442789Sahrens 	ztest_info_t *zi;
34435530Sbonwick 	int f, i;
3444789Sahrens 
3445789Sahrens 	while ((now = gethrtime()) < za->za_stop) {
3446789Sahrens 		/*
3447789Sahrens 		 * See if it's time to force a crash.
3448789Sahrens 		 */
3449789Sahrens 		if (now > za->za_kill) {
34505530Sbonwick 			zs->zs_alloc = spa_get_alloc(za->za_spa);
34515530Sbonwick 			zs->zs_space = spa_get_space(za->za_spa);
3452789Sahrens 			(void) kill(getpid(), SIGKILL);
3453789Sahrens 		}
3454789Sahrens 
3455789Sahrens 		/*
3456789Sahrens 		 * Pick a random function.
3457789Sahrens 		 */
3458789Sahrens 		f = ztest_random(ZTEST_FUNCS);
3459789Sahrens 		zi = &zs->zs_info[f];
3460789Sahrens 
3461789Sahrens 		/*
3462789Sahrens 		 * Decide whether to call it, based on the requested frequency.
3463789Sahrens 		 */
3464789Sahrens 		if (zi->zi_call_target == 0 ||
3465789Sahrens 		    (double)zi->zi_call_total / zi->zi_call_target >
3466789Sahrens 		    (double)(now - zs->zs_start_time) / (zopt_time * NANOSEC))
3467789Sahrens 			continue;
3468789Sahrens 
3469789Sahrens 		atomic_add_64(&zi->zi_calls, 1);
3470789Sahrens 		atomic_add_64(&zi->zi_call_total, 1);
3471789Sahrens 
3472789Sahrens 		za->za_diroff = (za->za_instance * ZTEST_FUNCS + f) *
3473789Sahrens 		    ZTEST_DIRSIZE;
3474789Sahrens 		za->za_diroff_shared = (1ULL << 63);
3475789Sahrens 
34765530Sbonwick 		for (i = 0; i < zi->zi_iters; i++)
34775530Sbonwick 			zi->zi_func(za);
3478789Sahrens 
3479789Sahrens 		functime = gethrtime() - now;
3480789Sahrens 
3481789Sahrens 		atomic_add_64(&zi->zi_call_time, functime);
3482789Sahrens 
3483789Sahrens 		if (zopt_verbose >= 4) {
3484789Sahrens 			Dl_info dli;
3485789Sahrens 			(void) dladdr((void *)zi->zi_func, &dli);
3486789Sahrens 			(void) printf("%6.2f sec in %s\n",
3487789Sahrens 			    (double)functime / NANOSEC, dli.dli_sname);
3488789Sahrens 		}
3489789Sahrens 
3490789Sahrens 		/*
3491789Sahrens 		 * If we're getting ENOSPC with some regularity, stop.
3492789Sahrens 		 */
3493789Sahrens 		if (zs->zs_enospc_count > 10)
3494789Sahrens 			break;
3495789Sahrens 	}
3496789Sahrens 
3497789Sahrens 	return (NULL);
3498789Sahrens }
3499789Sahrens 
3500789Sahrens /*
3501789Sahrens  * Kick off threads to run tests on all datasets in parallel.
3502789Sahrens  */
3503789Sahrens static void
3504789Sahrens ztest_run(char *pool)
3505789Sahrens {
3506789Sahrens 	int t, d, error;
3507789Sahrens 	ztest_shared_t *zs = ztest_shared;
3508789Sahrens 	ztest_args_t *za;
3509789Sahrens 	spa_t *spa;
3510789Sahrens 	char name[100];
35117754SJeff.Bonwick@Sun.COM 	thread_t resume_tid;
35127754SJeff.Bonwick@Sun.COM 
35137754SJeff.Bonwick@Sun.COM 	ztest_exiting = B_FALSE;
3514789Sahrens 
3515789Sahrens 	(void) _mutex_init(&zs->zs_vdev_lock, USYNC_THREAD, NULL);
3516789Sahrens 	(void) rwlock_init(&zs->zs_name_lock, USYNC_THREAD, NULL);
3517789Sahrens 
3518789Sahrens 	for (t = 0; t < ZTEST_SYNC_LOCKS; t++)
3519789Sahrens 		(void) _mutex_init(&zs->zs_sync_lock[t], USYNC_THREAD, NULL);
3520789Sahrens 
3521789Sahrens 	/*
3522789Sahrens 	 * Destroy one disk before we even start.
3523789Sahrens 	 * It's mirrored, so everything should work just fine.
3524789Sahrens 	 * This makes us exercise fault handling very early in spa_load().
3525789Sahrens 	 */
3526789Sahrens 	ztest_obliterate_one_disk(0);
3527789Sahrens 
3528789Sahrens 	/*
3529789Sahrens 	 * Verify that the sum of the sizes of all blocks in the pool
3530789Sahrens 	 * equals the SPA's allocated space total.
3531789Sahrens 	 */
3532789Sahrens 	ztest_verify_blocks(pool);
3533789Sahrens 
3534789Sahrens 	/*
3535789Sahrens 	 * Kick off a replacement of the disk we just obliterated.
3536789Sahrens 	 */
3537789Sahrens 	kernel_init(FREAD | FWRITE);
35387754SJeff.Bonwick@Sun.COM 	VERIFY(spa_open(pool, &spa, FTAG) == 0);
3539789Sahrens 	ztest_replace_one_disk(spa, 0);
3540789Sahrens 	if (zopt_verbose >= 5)
3541789Sahrens 		show_pool_stats(spa);
3542789Sahrens 	spa_close(spa, FTAG);
3543789Sahrens 	kernel_fini();
3544789Sahrens 
3545789Sahrens 	kernel_init(FREAD | FWRITE);
3546789Sahrens 
3547789Sahrens 	/*
3548789Sahrens 	 * Verify that we can export the pool and reimport it under a
3549789Sahrens 	 * different name.
3550789Sahrens 	 */
35511585Sbonwick 	if (ztest_random(2) == 0) {
35521585Sbonwick 		(void) snprintf(name, 100, "%s_import", pool);
35531585Sbonwick 		ztest_spa_import_export(pool, name);
35541585Sbonwick 		ztest_spa_import_export(name, pool);
35551585Sbonwick 	}
3556789Sahrens 
3557789Sahrens 	/*
3558789Sahrens 	 * Verify that we can loop over all pools.
3559789Sahrens 	 */
3560789Sahrens 	mutex_enter(&spa_namespace_lock);
3561789Sahrens 	for (spa = spa_next(NULL); spa != NULL; spa = spa_next(spa)) {
3562789Sahrens 		if (zopt_verbose > 3) {
3563789Sahrens 			(void) printf("spa_next: found %s\n", spa_name(spa));
3564789Sahrens 		}
3565789Sahrens 	}
3566789Sahrens 	mutex_exit(&spa_namespace_lock);
3567789Sahrens 
3568789Sahrens 	/*
35697754SJeff.Bonwick@Sun.COM 	 * Open our pool.
35705329Sgw25295 	 */
35717754SJeff.Bonwick@Sun.COM 	VERIFY(spa_open(pool, &spa, FTAG) == 0);
35725329Sgw25295 
35735329Sgw25295 	/*
35748241SJeff.Bonwick@Sun.COM 	 * We don't expect the pool to suspend unless maxfaults == 0,
35758241SJeff.Bonwick@Sun.COM 	 * in which case ztest_fault_inject() temporarily takes away
35768241SJeff.Bonwick@Sun.COM 	 * the only valid replica.
35778241SJeff.Bonwick@Sun.COM 	 */
35788241SJeff.Bonwick@Sun.COM 	if (zopt_maxfaults == 0)
35798241SJeff.Bonwick@Sun.COM 		spa->spa_failmode = ZIO_FAILURE_MODE_WAIT;
35808241SJeff.Bonwick@Sun.COM 	else
35818241SJeff.Bonwick@Sun.COM 		spa->spa_failmode = ZIO_FAILURE_MODE_PANIC;
35828241SJeff.Bonwick@Sun.COM 
35838241SJeff.Bonwick@Sun.COM 	/*
35847754SJeff.Bonwick@Sun.COM 	 * Create a thread to periodically resume suspended I/O.
3585789Sahrens 	 */
35868241SJeff.Bonwick@Sun.COM 	VERIFY(thr_create(0, 0, ztest_resume_thread, spa, THR_BOUND,
35877754SJeff.Bonwick@Sun.COM 	    &resume_tid) == 0);
3588789Sahrens 
3589789Sahrens 	/*
3590789Sahrens 	 * Verify that we can safely inquire about about any object,
3591789Sahrens 	 * whether it's allocated or not.  To make it interesting,
3592789Sahrens 	 * we probe a 5-wide window around each power of two.
3593789Sahrens 	 * This hits all edge cases, including zero and the max.
3594789Sahrens 	 */
3595789Sahrens 	for (t = 0; t < 64; t++) {
3596789Sahrens 		for (d = -5; d <= 5; d++) {
3597789Sahrens 			error = dmu_object_info(spa->spa_meta_objset,
3598789Sahrens 			    (1ULL << t) + d, NULL);
35991544Seschrock 			ASSERT(error == 0 || error == ENOENT ||
36001544Seschrock 			    error == EINVAL);
3601789Sahrens 		}
3602789Sahrens 	}
3603789Sahrens 
3604789Sahrens 	/*
3605789Sahrens 	 * Now kick off all the tests that run in parallel.
3606789Sahrens 	 */
3607789Sahrens 	zs->zs_enospc_count = 0;
3608789Sahrens 
3609789Sahrens 	za = umem_zalloc(zopt_threads * sizeof (ztest_args_t), UMEM_NOFAIL);
3610789Sahrens 
3611789Sahrens 	if (zopt_verbose >= 4)
3612789Sahrens 		(void) printf("starting main threads...\n");
3613789Sahrens 
3614789Sahrens 	za[0].za_start = gethrtime();
3615789Sahrens 	za[0].za_stop = za[0].za_start + zopt_passtime * NANOSEC;
3616789Sahrens 	za[0].za_stop = MIN(za[0].za_stop, zs->zs_stop_time);
3617789Sahrens 	za[0].za_kill = za[0].za_stop;
3618789Sahrens 	if (ztest_random(100) < zopt_killrate)
3619789Sahrens 		za[0].za_kill -= ztest_random(zopt_passtime * NANOSEC);
3620789Sahrens 
3621789Sahrens 	for (t = 0; t < zopt_threads; t++) {
36221732Sbonwick 		d = t % zopt_datasets;
36235530Sbonwick 
36245530Sbonwick 		(void) strcpy(za[t].za_pool, pool);
36255530Sbonwick 		za[t].za_os = za[d].za_os;
36265530Sbonwick 		za[t].za_spa = spa;
36275530Sbonwick 		za[t].za_zilog = za[d].za_zilog;
36285530Sbonwick 		za[t].za_instance = t;
36295530Sbonwick 		za[t].za_random = ztest_random(-1ULL);
36305530Sbonwick 		za[t].za_start = za[0].za_start;
36315530Sbonwick 		za[t].za_stop = za[0].za_stop;
36325530Sbonwick 		za[t].za_kill = za[0].za_kill;
36335530Sbonwick 
36341732Sbonwick 		if (t < zopt_datasets) {
36353711Smaybee 			int test_future = FALSE;
3636789Sahrens 			(void) rw_rdlock(&ztest_shared->zs_name_lock);
3637789Sahrens 			(void) snprintf(name, 100, "%s/%s_%d", pool, pool, d);
36386492Stimh 			error = dmu_objset_create(name, DMU_OST_OTHER, NULL, 0,
3639789Sahrens 			    ztest_create_cb, NULL);
36403711Smaybee 			if (error == EEXIST) {
36413711Smaybee 				test_future = TRUE;
36425530Sbonwick 			} else if (error == ENOSPC) {
36435530Sbonwick 				zs->zs_enospc_count++;
36445530Sbonwick 				(void) rw_unlock(&ztest_shared->zs_name_lock);
36455530Sbonwick 				break;
36463711Smaybee 			} else if (error != 0) {
3647789Sahrens 				fatal(0, "dmu_objset_create(%s) = %d",
3648789Sahrens 				    name, error);
3649789Sahrens 			}
3650789Sahrens 			error = dmu_objset_open(name, DMU_OST_OTHER,
36516689Smaybee 			    DS_MODE_USER, &za[d].za_os);
3652789Sahrens 			if (error)
3653789Sahrens 				fatal(0, "dmu_objset_open('%s') = %d",
3654789Sahrens 				    name, error);
3655789Sahrens 			(void) rw_unlock(&ztest_shared->zs_name_lock);
36565530Sbonwick 			if (test_future)
36575530Sbonwick 				ztest_dmu_check_future_leak(&za[t]);
36588227SNeil.Perrin@Sun.COM 			zil_replay(za[d].za_os, za[d].za_os,
36598227SNeil.Perrin@Sun.COM 			    ztest_replay_vector);
3660789Sahrens 			za[d].za_zilog = zil_open(za[d].za_os, NULL);
3661789Sahrens 		}
3662789Sahrens 
36637754SJeff.Bonwick@Sun.COM 		VERIFY(thr_create(0, 0, ztest_thread, &za[t], THR_BOUND,
36647754SJeff.Bonwick@Sun.COM 		    &za[t].za_thread) == 0);
3665789Sahrens 	}
3666789Sahrens 
3667789Sahrens 	while (--t >= 0) {
36687754SJeff.Bonwick@Sun.COM 		VERIFY(thr_join(za[t].za_thread, NULL, NULL) == 0);
36691732Sbonwick 		if (t < zopt_datasets) {
3670789Sahrens 			zil_close(za[t].za_zilog);
3671789Sahrens 			dmu_objset_close(za[t].za_os);
3672789Sahrens 		}
3673789Sahrens 	}
3674789Sahrens 
3675789Sahrens 	if (zopt_verbose >= 3)
3676789Sahrens 		show_pool_stats(spa);
3677789Sahrens 
3678789Sahrens 	txg_wait_synced(spa_get_dsl(spa), 0);
3679789Sahrens 
3680789Sahrens 	zs->zs_alloc = spa_get_alloc(spa);
3681789Sahrens 	zs->zs_space = spa_get_space(spa);
3682789Sahrens 
3683789Sahrens 	/*
36845530Sbonwick 	 * If we had out-of-space errors, destroy a random objset.
3685789Sahrens 	 */
3686789Sahrens 	if (zs->zs_enospc_count != 0) {
3687789Sahrens 		(void) rw_rdlock(&ztest_shared->zs_name_lock);
36885530Sbonwick 		d = (int)ztest_random(zopt_datasets);
36895530Sbonwick 		(void) snprintf(name, 100, "%s/%s_%d", pool, pool, d);
3690789Sahrens 		if (zopt_verbose >= 3)
3691789Sahrens 			(void) printf("Destroying %s to free up space\n", name);
36925530Sbonwick 		(void) dmu_objset_find(name, ztest_destroy_cb, &za[d],
36932417Sahrens 		    DS_FIND_SNAPSHOTS | DS_FIND_CHILDREN);
3694789Sahrens 		(void) rw_unlock(&ztest_shared->zs_name_lock);
3695789Sahrens 	}
3696789Sahrens 
36971544Seschrock 	txg_wait_synced(spa_get_dsl(spa), 0);
3698789Sahrens 
36997754SJeff.Bonwick@Sun.COM 	umem_free(za, zopt_threads * sizeof (ztest_args_t));
37007754SJeff.Bonwick@Sun.COM 
37017754SJeff.Bonwick@Sun.COM 	/* Kill the resume thread */
37027754SJeff.Bonwick@Sun.COM 	ztest_exiting = B_TRUE;
37037754SJeff.Bonwick@Sun.COM 	VERIFY(thr_join(resume_tid, NULL, NULL) == 0);
37048241SJeff.Bonwick@Sun.COM 	ztest_resume(spa);
37057754SJeff.Bonwick@Sun.COM 
3706789Sahrens 	/*
3707789Sahrens 	 * Right before closing the pool, kick off a bunch of async I/O;
3708789Sahrens 	 * spa_close() should wait for it to complete.
3709789Sahrens 	 */
3710789Sahrens 	for (t = 1; t < 50; t++)
3711789Sahrens 		dmu_prefetch(spa->spa_meta_objset, t, 0, 1 << 15);
3712789Sahrens 
37135530Sbonwick 	spa_close(spa, FTAG);
37145530Sbonwick 
3715789Sahrens 	kernel_fini();
3716789Sahrens }
3717789Sahrens 
3718789Sahrens void
3719789Sahrens print_time(hrtime_t t, char *timebuf)
3720789Sahrens {
3721789Sahrens 	hrtime_t s = t / NANOSEC;
3722789Sahrens 	hrtime_t m = s / 60;
3723789Sahrens 	hrtime_t h = m / 60;
3724789Sahrens 	hrtime_t d = h / 24;
3725789Sahrens 
3726789Sahrens 	s -= m * 60;
3727789Sahrens 	m -= h * 60;
3728789Sahrens 	h -= d * 24;
3729789Sahrens 
3730789Sahrens 	timebuf[0] = '\0';
3731789Sahrens 
3732789Sahrens 	if (d)
3733789Sahrens 		(void) sprintf(timebuf,
3734789Sahrens 		    "%llud%02lluh%02llum%02llus", d, h, m, s);
3735789Sahrens 	else if (h)
3736789Sahrens 		(void) sprintf(timebuf, "%lluh%02llum%02llus", h, m, s);
3737789Sahrens 	else if (m)
3738789Sahrens 		(void) sprintf(timebuf, "%llum%02llus", m, s);
3739789Sahrens 	else
3740789Sahrens 		(void) sprintf(timebuf, "%llus", s);
3741789Sahrens }
3742789Sahrens 
3743789Sahrens /*
3744789Sahrens  * Create a storage pool with the given name and initial vdev size.
3745789Sahrens  * Then create the specified number of datasets in the pool.
3746789Sahrens  */
3747789Sahrens static void
3748789Sahrens ztest_init(char *pool)
3749789Sahrens {
3750789Sahrens 	spa_t *spa;
3751789Sahrens 	int error;
3752789Sahrens 	nvlist_t *nvroot;
3753789Sahrens 
3754789Sahrens 	kernel_init(FREAD | FWRITE);
3755789Sahrens 
3756789Sahrens 	/*
3757789Sahrens 	 * Create the storage pool.
3758789Sahrens 	 */
3759789Sahrens 	(void) spa_destroy(pool);
3760789Sahrens 	ztest_shared->zs_vdev_primaries = 0;
37617754SJeff.Bonwick@Sun.COM 	nvroot = make_vdev_root(NULL, NULL, zopt_vdev_size, 0,
37627754SJeff.Bonwick@Sun.COM 	    0, zopt_raidz, zopt_mirrors, 1);
37637184Stimh 	error = spa_create(pool, nvroot, NULL, NULL, NULL);
3764789Sahrens 	nvlist_free(nvroot);
3765789Sahrens 
3766789Sahrens 	if (error)
3767789Sahrens 		fatal(0, "spa_create() = %d", error);
3768789Sahrens 	error = spa_open(pool, &spa, FTAG);
3769789Sahrens 	if (error)
3770789Sahrens 		fatal(0, "spa_open() = %d", error);
3771789Sahrens 
3772*9480SGeorge.Wilson@Sun.COM 	metaslab_sz = 1ULL << spa->spa_root_vdev->vdev_child[0]->vdev_ms_shift;
3773*9480SGeorge.Wilson@Sun.COM 
3774789Sahrens 	if (zopt_verbose >= 3)
3775789Sahrens 		show_pool_stats(spa);
3776789Sahrens 
3777789Sahrens 	spa_close(spa, FTAG);
3778789Sahrens 
3779789Sahrens 	kernel_fini();
3780789Sahrens }
3781789Sahrens 
3782789Sahrens int
3783789Sahrens main(int argc, char **argv)
3784789Sahrens {
3785789Sahrens 	int kills = 0;
3786789Sahrens 	int iters = 0;
3787789Sahrens 	int i, f;
3788789Sahrens 	ztest_shared_t *zs;
3789789Sahrens 	ztest_info_t *zi;
3790789Sahrens 	char timebuf[100];
3791789Sahrens 	char numbuf[6];
3792789Sahrens 
3793789Sahrens 	(void) setvbuf(stdout, NULL, _IOLBF, 0);
3794789Sahrens 
3795789Sahrens 	/* Override location of zpool.cache */
37966643Seschrock 	spa_config_path = "/tmp/zpool.cache";
3797789Sahrens 
3798789Sahrens 	ztest_random_fd = open("/dev/urandom", O_RDONLY);
3799789Sahrens 
3800789Sahrens 	process_options(argc, argv);
3801789Sahrens 
38021544Seschrock 	/*
38031544Seschrock 	 * Blow away any existing copy of zpool.cache
38041544Seschrock 	 */
38051544Seschrock 	if (zopt_init != 0)
38061544Seschrock 		(void) remove("/tmp/zpool.cache");
38071544Seschrock 
3808789Sahrens 	zs = ztest_shared = (void *)mmap(0,
3809789Sahrens 	    P2ROUNDUP(sizeof (ztest_shared_t), getpagesize()),
3810789Sahrens 	    PROT_READ | PROT_WRITE, MAP_SHARED | MAP_ANON, -1, 0);
3811789Sahrens 
3812789Sahrens 	if (zopt_verbose >= 1) {
3813789Sahrens 		(void) printf("%llu vdevs, %d datasets, %d threads,"
3814789Sahrens 		    " %llu seconds...\n",
38151732Sbonwick 		    (u_longlong_t)zopt_vdevs, zopt_datasets, zopt_threads,
3816789Sahrens 		    (u_longlong_t)zopt_time);
3817789Sahrens 	}
3818789Sahrens 
3819789Sahrens 	/*
3820789Sahrens 	 * Create and initialize our storage pool.
3821789Sahrens 	 */
3822789Sahrens 	for (i = 1; i <= zopt_init; i++) {
3823789Sahrens 		bzero(zs, sizeof (ztest_shared_t));
3824789Sahrens 		if (zopt_verbose >= 3 && zopt_init != 1)
3825789Sahrens 			(void) printf("ztest_init(), pass %d\n", i);
3826789Sahrens 		ztest_init(zopt_pool);
3827789Sahrens 	}
3828789Sahrens 
3829789Sahrens 	/*
3830789Sahrens 	 * Initialize the call targets for each function.
3831789Sahrens 	 */
3832789Sahrens 	for (f = 0; f < ZTEST_FUNCS; f++) {
3833789Sahrens 		zi = &zs->zs_info[f];
3834789Sahrens 
3835789Sahrens 		*zi = ztest_info[f];
3836789Sahrens 
3837789Sahrens 		if (*zi->zi_interval == 0)
3838789Sahrens 			zi->zi_call_target = UINT64_MAX;
3839789Sahrens 		else
3840789Sahrens 			zi->zi_call_target = zopt_time / *zi->zi_interval;
3841789Sahrens 	}
3842789Sahrens 
3843789Sahrens 	zs->zs_start_time = gethrtime();
3844789Sahrens 	zs->zs_stop_time = zs->zs_start_time + zopt_time * NANOSEC;
3845789Sahrens 
3846789Sahrens 	/*
3847789Sahrens 	 * Run the tests in a loop.  These tests include fault injection
3848789Sahrens 	 * to verify that self-healing data works, and forced crashes
3849789Sahrens 	 * to verify that we never lose on-disk consistency.
3850789Sahrens 	 */
3851789Sahrens 	while (gethrtime() < zs->zs_stop_time) {
3852789Sahrens 		int status;
3853789Sahrens 		pid_t pid;
3854789Sahrens 		char *tmp;
3855789Sahrens 
3856789Sahrens 		/*
3857789Sahrens 		 * Initialize the workload counters for each function.
3858789Sahrens 		 */
3859789Sahrens 		for (f = 0; f < ZTEST_FUNCS; f++) {
3860789Sahrens 			zi = &zs->zs_info[f];
3861789Sahrens 			zi->zi_calls = 0;
3862789Sahrens 			zi->zi_call_time = 0;
3863789Sahrens 		}
3864789Sahrens 
3865*9480SGeorge.Wilson@Sun.COM 		/* Set the allocation switch size */
3866*9480SGeorge.Wilson@Sun.COM 		metaslab_df_alloc_threshold = ztest_random(metaslab_sz / 4) + 1;
3867*9480SGeorge.Wilson@Sun.COM 
3868789Sahrens 		pid = fork();
3869789Sahrens 
3870789Sahrens 		if (pid == -1)
3871789Sahrens 			fatal(1, "fork failed");
3872789Sahrens 
3873789Sahrens 		if (pid == 0) {	/* child */
3874789Sahrens 			struct rlimit rl = { 1024, 1024 };
3875789Sahrens 			(void) setrlimit(RLIMIT_NOFILE, &rl);
38761914Scasper 			(void) enable_extended_FILE_stdio(-1, -1);
3877789Sahrens 			ztest_run(zopt_pool);
3878789Sahrens 			exit(0);
3879789Sahrens 		}
3880789Sahrens 
38812856Snd150628 		while (waitpid(pid, &status, 0) != pid)
3882789Sahrens 			continue;
3883789Sahrens 
3884789Sahrens 		if (WIFEXITED(status)) {
3885789Sahrens 			if (WEXITSTATUS(status) != 0) {
3886789Sahrens 				(void) fprintf(stderr,
3887789Sahrens 				    "child exited with code %d\n",
3888789Sahrens 				    WEXITSTATUS(status));
3889789Sahrens 				exit(2);
3890789Sahrens 			}
38912856Snd150628 		} else if (WIFSIGNALED(status)) {
3892789Sahrens 			if (WTERMSIG(status) != SIGKILL) {
3893789Sahrens 				(void) fprintf(stderr,
3894789Sahrens 				    "child died with signal %d\n",
3895789Sahrens 				    WTERMSIG(status));
3896789Sahrens 				exit(3);
3897789Sahrens 			}
3898789Sahrens 			kills++;
38992856Snd150628 		} else {
39002856Snd150628 			(void) fprintf(stderr, "something strange happened "
39012856Snd150628 			    "to child\n");
39022856Snd150628 			exit(4);
3903789Sahrens 		}
3904789Sahrens 
3905789Sahrens 		iters++;
3906789Sahrens 
3907789Sahrens 		if (zopt_verbose >= 1) {
3908789Sahrens 			hrtime_t now = gethrtime();
3909789Sahrens 
3910789Sahrens 			now = MIN(now, zs->zs_stop_time);
3911789Sahrens 			print_time(zs->zs_stop_time - now, timebuf);
3912789Sahrens 			nicenum(zs->zs_space, numbuf);
3913789Sahrens 
3914789Sahrens 			(void) printf("Pass %3d, %8s, %3llu ENOSPC, "
3915789Sahrens 			    "%4.1f%% of %5s used, %3.0f%% done, %8s to go\n",
3916789Sahrens 			    iters,
3917789Sahrens 			    WIFEXITED(status) ? "Complete" : "SIGKILL",
3918789Sahrens 			    (u_longlong_t)zs->zs_enospc_count,
3919789Sahrens 			    100.0 * zs->zs_alloc / zs->zs_space,
3920789Sahrens 			    numbuf,
3921789Sahrens 			    100.0 * (now - zs->zs_start_time) /
3922789Sahrens 			    (zopt_time * NANOSEC), timebuf);
3923789Sahrens 		}
3924789Sahrens 
3925789Sahrens 		if (zopt_verbose >= 2) {
3926789Sahrens 			(void) printf("\nWorkload summary:\n\n");
3927789Sahrens 			(void) printf("%7s %9s   %s\n",
3928789Sahrens 			    "Calls", "Time", "Function");
3929789Sahrens 			(void) printf("%7s %9s   %s\n",
3930789Sahrens 			    "-----", "----", "--------");
3931789Sahrens 			for (f = 0; f < ZTEST_FUNCS; f++) {
3932789Sahrens 				Dl_info dli;
3933789Sahrens 
3934789Sahrens 				zi = &zs->zs_info[f];
3935789Sahrens 				print_time(zi->zi_call_time, timebuf);
3936789Sahrens 				(void) dladdr((void *)zi->zi_func, &dli);
3937789Sahrens 				(void) printf("%7llu %9s   %s\n",
3938789Sahrens 				    (u_longlong_t)zi->zi_calls, timebuf,
3939789Sahrens 				    dli.dli_sname);
3940789Sahrens 			}
3941789Sahrens 			(void) printf("\n");
3942789Sahrens 		}
3943789Sahrens 
3944789Sahrens 		/*
3945789Sahrens 		 * It's possible that we killed a child during a rename test, in
3946789Sahrens 		 * which case we'll have a 'ztest_tmp' pool lying around instead
3947789Sahrens 		 * of 'ztest'.  Do a blind rename in case this happened.
3948789Sahrens 		 */
3949789Sahrens 		tmp = umem_alloc(strlen(zopt_pool) + 5, UMEM_NOFAIL);
3950789Sahrens 		(void) strcpy(tmp, zopt_pool);
3951789Sahrens 		(void) strcat(tmp, "_tmp");
3952789Sahrens 		kernel_init(FREAD | FWRITE);
3953789Sahrens 		(void) spa_rename(tmp, zopt_pool);
3954789Sahrens 		kernel_fini();
3955789Sahrens 		umem_free(tmp, strlen(tmp) + 1);
3956789Sahrens 	}
3957789Sahrens 
3958789Sahrens 	ztest_verify_blocks(zopt_pool);
3959789Sahrens 
3960789Sahrens 	if (zopt_verbose >= 1) {
3961789Sahrens 		(void) printf("%d killed, %d completed, %.0f%% kill rate\n",
3962789Sahrens 		    kills, iters - kills, (100.0 * kills) / MAX(1, iters));
3963789Sahrens 	}
3964789Sahrens 
3965789Sahrens 	return (0);
3966789Sahrens }
3967