xref: /illumos-gate/usr/src/cmd/ztest/ztest.c (revision 2fdbea25c2ba89186b8a6b7c6840ebc9f4dff245)
1fa9e4066Sahrens /*
2fa9e4066Sahrens  * CDDL HEADER START
3fa9e4066Sahrens  *
4fa9e4066Sahrens  * The contents of this file are subject to the terms of the
5441d80aaSlling  * Common Development and Distribution License (the "License").
6441d80aaSlling  * You may not use this file except in compliance with the License.
7fa9e4066Sahrens  *
8fa9e4066Sahrens  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9fa9e4066Sahrens  * or http://www.opensolaris.org/os/licensing.
10fa9e4066Sahrens  * See the License for the specific language governing permissions
11fa9e4066Sahrens  * and limitations under the License.
12fa9e4066Sahrens  *
13fa9e4066Sahrens  * When distributing Covered Code, include this CDDL HEADER in each
14fa9e4066Sahrens  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15fa9e4066Sahrens  * If applicable, add the following below this CDDL HEADER, with the
16fa9e4066Sahrens  * fields enclosed by brackets "[]" replaced with your own identifying
17fa9e4066Sahrens  * information: Portions Copyright [yyyy] [name of copyright owner]
18fa9e4066Sahrens  *
19fa9e4066Sahrens  * CDDL HEADER END
20fa9e4066Sahrens  */
21fa9e4066Sahrens /*
224f5064b7SMark J Musante  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
23fa9e4066Sahrens  * Use is subject to license terms.
24fa9e4066Sahrens  */
25fa9e4066Sahrens 
26fa9e4066Sahrens /*
27fa9e4066Sahrens  * The objective of this program is to provide a DMU/ZAP/SPA stress test
28fa9e4066Sahrens  * that runs entirely in userland, is easy to use, and easy to extend.
29fa9e4066Sahrens  *
30fa9e4066Sahrens  * The overall design of the ztest program is as follows:
31fa9e4066Sahrens  *
32fa9e4066Sahrens  * (1) For each major functional area (e.g. adding vdevs to a pool,
33fa9e4066Sahrens  *     creating and destroying datasets, reading and writing objects, etc)
34fa9e4066Sahrens  *     we have a simple routine to test that functionality.  These
35fa9e4066Sahrens  *     individual routines do not have to do anything "stressful".
36fa9e4066Sahrens  *
37fa9e4066Sahrens  * (2) We turn these simple functionality tests into a stress test by
38fa9e4066Sahrens  *     running them all in parallel, with as many threads as desired,
39fa9e4066Sahrens  *     and spread across as many datasets, objects, and vdevs as desired.
40fa9e4066Sahrens  *
41fa9e4066Sahrens  * (3) While all this is happening, we inject faults into the pool to
42fa9e4066Sahrens  *     verify that self-healing data really works.
43fa9e4066Sahrens  *
44fa9e4066Sahrens  * (4) Every time we open a dataset, we change its checksum and compression
45fa9e4066Sahrens  *     functions.  Thus even individual objects vary from block to block
46fa9e4066Sahrens  *     in which checksum they use and whether they're compressed.
47fa9e4066Sahrens  *
48fa9e4066Sahrens  * (5) To verify that we never lose on-disk consistency after a crash,
49fa9e4066Sahrens  *     we run the entire test in a child of the main process.
50fa9e4066Sahrens  *     At random times, the child self-immolates with a SIGKILL.
51fa9e4066Sahrens  *     This is the software equivalent of pulling the power cord.
52fa9e4066Sahrens  *     The parent then runs the test again, using the existing
53fa9e4066Sahrens  *     storage pool, as many times as desired.
54fa9e4066Sahrens  *
55fa9e4066Sahrens  * (6) To verify that we don't have future leaks or temporal incursions,
56fa9e4066Sahrens  *     many of the functional tests record the transaction group number
57fa9e4066Sahrens  *     as part of their data.  When reading old data, they verify that
58fa9e4066Sahrens  *     the transaction group number is less than the current, open txg.
59fa9e4066Sahrens  *     If you add a new test, please do this if applicable.
60fa9e4066Sahrens  *
61fa9e4066Sahrens  * When run with no arguments, ztest runs for about five minutes and
62fa9e4066Sahrens  * produces no output if successful.  To get a little bit of information,
63fa9e4066Sahrens  * specify -V.  To get more information, specify -VV, and so on.
64fa9e4066Sahrens  *
65fa9e4066Sahrens  * To turn this into an overnight stress test, use -T to specify run time.
66fa9e4066Sahrens  *
67fa9e4066Sahrens  * You can ask more more vdevs [-v], datasets [-d], or threads [-t]
68fa9e4066Sahrens  * to increase the pool capacity, fanout, and overall stress level.
69fa9e4066Sahrens  *
70fa9e4066Sahrens  * The -N(okill) option will suppress kills, so each child runs to completion.
71fa9e4066Sahrens  * This can be useful when you're trying to distinguish temporal incursions
72fa9e4066Sahrens  * from plain old race conditions.
73fa9e4066Sahrens  */
74fa9e4066Sahrens 
75fa9e4066Sahrens #include <sys/zfs_context.h>
76fa9e4066Sahrens #include <sys/spa.h>
77fa9e4066Sahrens #include <sys/dmu.h>
78fa9e4066Sahrens #include <sys/txg.h>
79*2fdbea25SAleksandr Guzovskiy #include <sys/dbuf.h>
80fa9e4066Sahrens #include <sys/zap.h>
81fa9e4066Sahrens #include <sys/dmu_objset.h>
82fa9e4066Sahrens #include <sys/poll.h>
83fa9e4066Sahrens #include <sys/stat.h>
84fa9e4066Sahrens #include <sys/time.h>
85fa9e4066Sahrens #include <sys/wait.h>
86fa9e4066Sahrens #include <sys/mman.h>
87fa9e4066Sahrens #include <sys/resource.h>
88fa9e4066Sahrens #include <sys/zio.h>
89fa9e4066Sahrens #include <sys/zio_checksum.h>
90fa9e4066Sahrens #include <sys/zio_compress.h>
91fa9e4066Sahrens #include <sys/zil.h>
92fa9e4066Sahrens #include <sys/vdev_impl.h>
93e14bb325SJeff Bonwick #include <sys/vdev_file.h>
94fa9e4066Sahrens #include <sys/spa_impl.h>
95fa9e4066Sahrens #include <sys/dsl_prop.h>
964f5064b7SMark J Musante #include <sys/dsl_dataset.h>
97fa9e4066Sahrens #include <sys/refcount.h>
98fa9e4066Sahrens #include <stdio.h>
99004388ebScasper #include <stdio_ext.h>
100fa9e4066Sahrens #include <stdlib.h>
101fa9e4066Sahrens #include <unistd.h>
102fa9e4066Sahrens #include <signal.h>
103fa9e4066Sahrens #include <umem.h>
104fa9e4066Sahrens #include <dlfcn.h>
105fa9e4066Sahrens #include <ctype.h>
106fa9e4066Sahrens #include <math.h>
107fa9e4066Sahrens #include <sys/fs/zfs.h>
108fa9e4066Sahrens 
109fa9e4066Sahrens static char cmdname[] = "ztest";
110fa9e4066Sahrens static char *zopt_pool = cmdname;
111fa9e4066Sahrens 
112fa9e4066Sahrens static uint64_t zopt_vdevs = 5;
113fa9e4066Sahrens static uint64_t zopt_vdevtime;
114ecc2d604Sbonwick static int zopt_ashift = SPA_MINBLOCKSHIFT;
115fa9e4066Sahrens static int zopt_mirrors = 2;
116fa9e4066Sahrens static int zopt_raidz = 4;
11799653d4eSeschrock static int zopt_raidz_parity = 1;
118fa9e4066Sahrens static size_t zopt_vdev_size = SPA_MINDEVSIZE;
119ecc2d604Sbonwick static int zopt_datasets = 7;
120fa9e4066Sahrens static int zopt_threads = 23;
121fa9e4066Sahrens static uint64_t zopt_passtime = 60;	/* 60 seconds */
122fa9e4066Sahrens static uint64_t zopt_killrate = 70;	/* 70% kill rate */
123fa9e4066Sahrens static int zopt_verbose = 0;
124fa9e4066Sahrens static int zopt_init = 1;
125fa9e4066Sahrens static char *zopt_dir = "/tmp";
126fa9e4066Sahrens static uint64_t zopt_time = 300;	/* 5 minutes */
127fa9e4066Sahrens static int zopt_maxfaults;
128fa9e4066Sahrens 
129e05725b1Sbonwick typedef struct ztest_block_tag {
130e05725b1Sbonwick 	uint64_t	bt_objset;
131e05725b1Sbonwick 	uint64_t	bt_object;
132e05725b1Sbonwick 	uint64_t	bt_offset;
133e05725b1Sbonwick 	uint64_t	bt_txg;
134e05725b1Sbonwick 	uint64_t	bt_thread;
135e05725b1Sbonwick 	uint64_t	bt_seq;
136e05725b1Sbonwick } ztest_block_tag_t;
137e05725b1Sbonwick 
138fa9e4066Sahrens typedef struct ztest_args {
139e05725b1Sbonwick 	char		za_pool[MAXNAMELEN];
140e05725b1Sbonwick 	spa_t		*za_spa;
141fa9e4066Sahrens 	objset_t	*za_os;
142fa9e4066Sahrens 	zilog_t		*za_zilog;
143fa9e4066Sahrens 	thread_t	za_thread;
144fa9e4066Sahrens 	uint64_t	za_instance;
145fa9e4066Sahrens 	uint64_t	za_random;
146fa9e4066Sahrens 	uint64_t	za_diroff;
147fa9e4066Sahrens 	uint64_t	za_diroff_shared;
148ea8dc4b6Seschrock 	uint64_t	za_zil_seq;
149fa9e4066Sahrens 	hrtime_t	za_start;
150fa9e4066Sahrens 	hrtime_t	za_stop;
151fa9e4066Sahrens 	hrtime_t	za_kill;
152e05725b1Sbonwick 	/*
153e05725b1Sbonwick 	 * Thread-local variables can go here to aid debugging.
154e05725b1Sbonwick 	 */
155e05725b1Sbonwick 	ztest_block_tag_t za_rbt;
156e05725b1Sbonwick 	ztest_block_tag_t za_wbt;
157e05725b1Sbonwick 	dmu_object_info_t za_doi;
158e05725b1Sbonwick 	dmu_buf_t	*za_dbuf;
159fa9e4066Sahrens } ztest_args_t;
160fa9e4066Sahrens 
161fa9e4066Sahrens typedef void ztest_func_t(ztest_args_t *);
162fa9e4066Sahrens 
163fa9e4066Sahrens /*
164fa9e4066Sahrens  * Note: these aren't static because we want dladdr() to work.
165fa9e4066Sahrens  */
166fa9e4066Sahrens ztest_func_t ztest_dmu_read_write;
167*2fdbea25SAleksandr Guzovskiy ztest_func_t ztest_dmu_read_write_zcopy;
168fa9e4066Sahrens ztest_func_t ztest_dmu_write_parallel;
169fa9e4066Sahrens ztest_func_t ztest_dmu_object_alloc_free;
170fa9e4066Sahrens ztest_func_t ztest_zap;
171fa9e4066Sahrens ztest_func_t ztest_zap_parallel;
172fa9e4066Sahrens ztest_func_t ztest_traverse;
173fa9e4066Sahrens ztest_func_t ztest_dsl_prop_get_set;
174fa9e4066Sahrens ztest_func_t ztest_dmu_objset_create_destroy;
175fa9e4066Sahrens ztest_func_t ztest_dmu_snapshot_create_destroy;
1764f5064b7SMark J Musante ztest_func_t ztest_dsl_dataset_promote_busy;
177fa9e4066Sahrens ztest_func_t ztest_spa_create_destroy;
178fa9e4066Sahrens ztest_func_t ztest_fault_inject;
179e14bb325SJeff Bonwick ztest_func_t ztest_spa_rename;
180fa9e4066Sahrens ztest_func_t ztest_vdev_attach_detach;
181fa9e4066Sahrens ztest_func_t ztest_vdev_LUN_growth;
182fa9e4066Sahrens ztest_func_t ztest_vdev_add_remove;
183e14bb325SJeff Bonwick ztest_func_t ztest_vdev_aux_add_remove;
184fa9e4066Sahrens ztest_func_t ztest_scrub;
185fa9e4066Sahrens 
186fa9e4066Sahrens typedef struct ztest_info {
187fa9e4066Sahrens 	ztest_func_t	*zi_func;	/* test function */
188e05725b1Sbonwick 	uint64_t	zi_iters;	/* iterations per execution */
189fa9e4066Sahrens 	uint64_t	*zi_interval;	/* execute every <interval> seconds */
190fa9e4066Sahrens 	uint64_t	zi_calls;	/* per-pass count */
191fa9e4066Sahrens 	uint64_t	zi_call_time;	/* per-pass time */
192fa9e4066Sahrens 	uint64_t	zi_call_total;	/* cumulative total */
193fa9e4066Sahrens 	uint64_t	zi_call_target;	/* target cumulative total */
194fa9e4066Sahrens } ztest_info_t;
195fa9e4066Sahrens 
196fa9e4066Sahrens uint64_t zopt_always = 0;		/* all the time */
197fa9e4066Sahrens uint64_t zopt_often = 1;		/* every second */
198fa9e4066Sahrens uint64_t zopt_sometimes = 10;		/* every 10 seconds */
199fa9e4066Sahrens uint64_t zopt_rarely = 60;		/* every 60 seconds */
200fa9e4066Sahrens 
201fa9e4066Sahrens ztest_info_t ztest_info[] = {
202e05725b1Sbonwick 	{ ztest_dmu_read_write,			1,	&zopt_always	},
203*2fdbea25SAleksandr Guzovskiy 	{ ztest_dmu_read_write_zcopy,		1,	&zopt_always	},
204e05725b1Sbonwick 	{ ztest_dmu_write_parallel,		30,	&zopt_always	},
205e05725b1Sbonwick 	{ ztest_dmu_object_alloc_free,		1,	&zopt_always	},
206e05725b1Sbonwick 	{ ztest_zap,				30,	&zopt_always	},
207e05725b1Sbonwick 	{ ztest_zap_parallel,			100,	&zopt_always	},
208e05725b1Sbonwick 	{ ztest_dsl_prop_get_set,		1,	&zopt_sometimes	},
209088f3894Sahrens 	{ ztest_dmu_objset_create_destroy,	1,	&zopt_sometimes },
210e14bb325SJeff Bonwick 	{ ztest_dmu_snapshot_create_destroy,	1,	&zopt_sometimes },
211088f3894Sahrens 	{ ztest_spa_create_destroy,		1,	&zopt_sometimes },
212e05725b1Sbonwick 	{ ztest_fault_inject,			1,	&zopt_sometimes	},
213e05725b1Sbonwick 	{ ztest_spa_rename,			1,	&zopt_rarely	},
214e14bb325SJeff Bonwick 	{ ztest_vdev_attach_detach,		1,	&zopt_rarely	},
215e14bb325SJeff Bonwick 	{ ztest_vdev_LUN_growth,		1,	&zopt_rarely	},
216b2d22b66SGeorge Wilson 	{ ztest_dsl_dataset_promote_busy,	1,	&zopt_rarely	},
217e14bb325SJeff Bonwick 	{ ztest_vdev_add_remove,		1,	&zopt_vdevtime	},
218e14bb325SJeff Bonwick 	{ ztest_vdev_aux_add_remove,		1,	&zopt_vdevtime	},
219e05725b1Sbonwick 	{ ztest_scrub,				1,	&zopt_vdevtime	},
220fa9e4066Sahrens };
221fa9e4066Sahrens 
222fa9e4066Sahrens #define	ZTEST_FUNCS	(sizeof (ztest_info) / sizeof (ztest_info_t))
223fa9e4066Sahrens 
224fa9e4066Sahrens #define	ZTEST_SYNC_LOCKS	16
225fa9e4066Sahrens 
226fa9e4066Sahrens /*
227fa9e4066Sahrens  * Stuff we need to share writably between parent and child.
228fa9e4066Sahrens  */
229fa9e4066Sahrens typedef struct ztest_shared {
230fa9e4066Sahrens 	mutex_t		zs_vdev_lock;
231fa9e4066Sahrens 	rwlock_t	zs_name_lock;
232fa9e4066Sahrens 	uint64_t	zs_vdev_primaries;
233e14bb325SJeff Bonwick 	uint64_t	zs_vdev_aux;
234fa9e4066Sahrens 	uint64_t	zs_enospc_count;
235fa9e4066Sahrens 	hrtime_t	zs_start_time;
236fa9e4066Sahrens 	hrtime_t	zs_stop_time;
237fa9e4066Sahrens 	uint64_t	zs_alloc;
238fa9e4066Sahrens 	uint64_t	zs_space;
239fa9e4066Sahrens 	ztest_info_t	zs_info[ZTEST_FUNCS];
240fa9e4066Sahrens 	mutex_t		zs_sync_lock[ZTEST_SYNC_LOCKS];
241fa9e4066Sahrens 	uint64_t	zs_seq[ZTEST_SYNC_LOCKS];
242fa9e4066Sahrens } ztest_shared_t;
243fa9e4066Sahrens 
244fa9e4066Sahrens static char ztest_dev_template[] = "%s/%s.%llua";
245e14bb325SJeff Bonwick static char ztest_aux_template[] = "%s/%s.%s.%llu";
246fa9e4066Sahrens static ztest_shared_t *ztest_shared;
247fa9e4066Sahrens 
248fa9e4066Sahrens static int ztest_random_fd;
249fa9e4066Sahrens static int ztest_dump_core = 1;
250fa9e4066Sahrens 
251e14bb325SJeff Bonwick static boolean_t ztest_exiting;
2520a4e9518Sgw 
253e05725b1Sbonwick extern uint64_t metaslab_gang_bang;
254fa9e4066Sahrens 
255fa9e4066Sahrens #define	ZTEST_DIROBJ		1
256fa9e4066Sahrens #define	ZTEST_MICROZAP_OBJ	2
257fa9e4066Sahrens #define	ZTEST_FATZAP_OBJ	3
258fa9e4066Sahrens 
259fa9e4066Sahrens #define	ZTEST_DIROBJ_BLOCKSIZE	(1 << 10)
260fa9e4066Sahrens #define	ZTEST_DIRSIZE		256
261fa9e4066Sahrens 
2621ce825d8Sraf static void usage(boolean_t) __NORETURN;
263f1b4288bSvb 
264fa9e4066Sahrens /*
265fa9e4066Sahrens  * These libumem hooks provide a reasonable set of defaults for the allocator's
266fa9e4066Sahrens  * debugging facilities.
267fa9e4066Sahrens  */
268fa9e4066Sahrens const char *
269fa9e4066Sahrens _umem_debug_init()
270fa9e4066Sahrens {
271fa9e4066Sahrens 	return ("default,verbose"); /* $UMEM_DEBUG setting */
272fa9e4066Sahrens }
273fa9e4066Sahrens 
274fa9e4066Sahrens const char *
275fa9e4066Sahrens _umem_logging_init(void)
276fa9e4066Sahrens {
277fa9e4066Sahrens 	return ("fail,contents"); /* $UMEM_LOGGING setting */
278fa9e4066Sahrens }
279fa9e4066Sahrens 
280fa9e4066Sahrens #define	FATAL_MSG_SZ	1024
281fa9e4066Sahrens 
282fa9e4066Sahrens char *fatal_msg;
283fa9e4066Sahrens 
284fa9e4066Sahrens static void
285fa9e4066Sahrens fatal(int do_perror, char *message, ...)
286fa9e4066Sahrens {
287fa9e4066Sahrens 	va_list args;
288fa9e4066Sahrens 	int save_errno = errno;
289fa9e4066Sahrens 	char buf[FATAL_MSG_SZ];
290fa9e4066Sahrens 
291fa9e4066Sahrens 	(void) fflush(stdout);
292fa9e4066Sahrens 
293fa9e4066Sahrens 	va_start(args, message);
294fa9e4066Sahrens 	(void) sprintf(buf, "ztest: ");
295fa9e4066Sahrens 	/* LINTED */
296fa9e4066Sahrens 	(void) vsprintf(buf + strlen(buf), message, args);
297fa9e4066Sahrens 	va_end(args);
298fa9e4066Sahrens 	if (do_perror) {
299fa9e4066Sahrens 		(void) snprintf(buf + strlen(buf), FATAL_MSG_SZ - strlen(buf),
300fa9e4066Sahrens 		    ": %s", strerror(save_errno));
301fa9e4066Sahrens 	}
302fa9e4066Sahrens 	(void) fprintf(stderr, "%s\n", buf);
303fa9e4066Sahrens 	fatal_msg = buf;			/* to ease debugging */
304fa9e4066Sahrens 	if (ztest_dump_core)
305fa9e4066Sahrens 		abort();
306fa9e4066Sahrens 	exit(3);
307fa9e4066Sahrens }
308fa9e4066Sahrens 
309fa9e4066Sahrens static int
310fa9e4066Sahrens str2shift(const char *buf)
311fa9e4066Sahrens {
312fa9e4066Sahrens 	const char *ends = "BKMGTPEZ";
313fa9e4066Sahrens 	int i;
314fa9e4066Sahrens 
315fa9e4066Sahrens 	if (buf[0] == '\0')
316fa9e4066Sahrens 		return (0);
317fa9e4066Sahrens 	for (i = 0; i < strlen(ends); i++) {
318fa9e4066Sahrens 		if (toupper(buf[0]) == ends[i])
319fa9e4066Sahrens 			break;
320fa9e4066Sahrens 	}
321f1b4288bSvb 	if (i == strlen(ends)) {
322f1b4288bSvb 		(void) fprintf(stderr, "ztest: invalid bytes suffix: %s\n",
323f1b4288bSvb 		    buf);
324f1b4288bSvb 		usage(B_FALSE);
325f1b4288bSvb 	}
326fa9e4066Sahrens 	if (buf[1] == '\0' || (toupper(buf[1]) == 'B' && buf[2] == '\0')) {
327fa9e4066Sahrens 		return (10*i);
328fa9e4066Sahrens 	}
329f1b4288bSvb 	(void) fprintf(stderr, "ztest: invalid bytes suffix: %s\n", buf);
330f1b4288bSvb 	usage(B_FALSE);
331f1b4288bSvb 	/* NOTREACHED */
332fa9e4066Sahrens }
333fa9e4066Sahrens 
334fa9e4066Sahrens static uint64_t
335fa9e4066Sahrens nicenumtoull(const char *buf)
336fa9e4066Sahrens {
337fa9e4066Sahrens 	char *end;
338fa9e4066Sahrens 	uint64_t val;
339fa9e4066Sahrens 
340fa9e4066Sahrens 	val = strtoull(buf, &end, 0);
341fa9e4066Sahrens 	if (end == buf) {
342f1b4288bSvb 		(void) fprintf(stderr, "ztest: bad numeric value: %s\n", buf);
343f1b4288bSvb 		usage(B_FALSE);
344fa9e4066Sahrens 	} else if (end[0] == '.') {
345fa9e4066Sahrens 		double fval = strtod(buf, &end);
346fa9e4066Sahrens 		fval *= pow(2, str2shift(end));
347f1b4288bSvb 		if (fval > UINT64_MAX) {
348f1b4288bSvb 			(void) fprintf(stderr, "ztest: value too large: %s\n",
349f1b4288bSvb 			    buf);
350f1b4288bSvb 			usage(B_FALSE);
351f1b4288bSvb 		}
352fa9e4066Sahrens 		val = (uint64_t)fval;
353fa9e4066Sahrens 	} else {
354fa9e4066Sahrens 		int shift = str2shift(end);
355f1b4288bSvb 		if (shift >= 64 || (val << shift) >> shift != val) {
356f1b4288bSvb 			(void) fprintf(stderr, "ztest: value too large: %s\n",
357f1b4288bSvb 			    buf);
358f1b4288bSvb 			usage(B_FALSE);
359f1b4288bSvb 		}
360fa9e4066Sahrens 		val <<= shift;
361fa9e4066Sahrens 	}
362fa9e4066Sahrens 	return (val);
363fa9e4066Sahrens }
364fa9e4066Sahrens 
365fa9e4066Sahrens static void
366f1b4288bSvb usage(boolean_t requested)
367fa9e4066Sahrens {
368fa9e4066Sahrens 	char nice_vdev_size[10];
369fa9e4066Sahrens 	char nice_gang_bang[10];
370f1b4288bSvb 	FILE *fp = requested ? stdout : stderr;
371fa9e4066Sahrens 
372fa9e4066Sahrens 	nicenum(zopt_vdev_size, nice_vdev_size);
373e05725b1Sbonwick 	nicenum(metaslab_gang_bang, nice_gang_bang);
374fa9e4066Sahrens 
375f1b4288bSvb 	(void) fprintf(fp, "Usage: %s\n"
376fa9e4066Sahrens 	    "\t[-v vdevs (default: %llu)]\n"
377fa9e4066Sahrens 	    "\t[-s size_of_each_vdev (default: %s)]\n"
378ecc2d604Sbonwick 	    "\t[-a alignment_shift (default: %d) (use 0 for random)]\n"
379fa9e4066Sahrens 	    "\t[-m mirror_copies (default: %d)]\n"
380fa9e4066Sahrens 	    "\t[-r raidz_disks (default: %d)]\n"
38199653d4eSeschrock 	    "\t[-R raidz_parity (default: %d)]\n"
382fa9e4066Sahrens 	    "\t[-d datasets (default: %d)]\n"
383fa9e4066Sahrens 	    "\t[-t threads (default: %d)]\n"
384fa9e4066Sahrens 	    "\t[-g gang_block_threshold (default: %s)]\n"
385fa9e4066Sahrens 	    "\t[-i initialize pool i times (default: %d)]\n"
386fa9e4066Sahrens 	    "\t[-k kill percentage (default: %llu%%)]\n"
387fa9e4066Sahrens 	    "\t[-p pool_name (default: %s)]\n"
388fa9e4066Sahrens 	    "\t[-f file directory for vdev files (default: %s)]\n"
389fa9e4066Sahrens 	    "\t[-V(erbose)] (use multiple times for ever more blather)\n"
390ecc2d604Sbonwick 	    "\t[-E(xisting)] (use existing pool instead of creating new one)\n"
391fa9e4066Sahrens 	    "\t[-T time] total run time (default: %llu sec)\n"
392fa9e4066Sahrens 	    "\t[-P passtime] time per pass (default: %llu sec)\n"
393f1b4288bSvb 	    "\t[-h] (print help)\n"
394fa9e4066Sahrens 	    "",
395fa9e4066Sahrens 	    cmdname,
3960a4e9518Sgw 	    (u_longlong_t)zopt_vdevs,			/* -v */
3970a4e9518Sgw 	    nice_vdev_size,				/* -s */
3980a4e9518Sgw 	    zopt_ashift,				/* -a */
3990a4e9518Sgw 	    zopt_mirrors,				/* -m */
4000a4e9518Sgw 	    zopt_raidz,					/* -r */
4010a4e9518Sgw 	    zopt_raidz_parity,				/* -R */
4020a4e9518Sgw 	    zopt_datasets,				/* -d */
4030a4e9518Sgw 	    zopt_threads,				/* -t */
4040a4e9518Sgw 	    nice_gang_bang,				/* -g */
4050a4e9518Sgw 	    zopt_init,					/* -i */
4060a4e9518Sgw 	    (u_longlong_t)zopt_killrate,		/* -k */
4070a4e9518Sgw 	    zopt_pool,					/* -p */
4080a4e9518Sgw 	    zopt_dir,					/* -f */
4090a4e9518Sgw 	    (u_longlong_t)zopt_time,			/* -T */
410e14bb325SJeff Bonwick 	    (u_longlong_t)zopt_passtime);		/* -P */
411f1b4288bSvb 	exit(requested ? 0 : 1);
412fa9e4066Sahrens }
413fa9e4066Sahrens 
414fa9e4066Sahrens static uint64_t
415fa9e4066Sahrens ztest_random(uint64_t range)
416fa9e4066Sahrens {
417fa9e4066Sahrens 	uint64_t r;
418fa9e4066Sahrens 
419fa9e4066Sahrens 	if (range == 0)
420fa9e4066Sahrens 		return (0);
421fa9e4066Sahrens 
422fa9e4066Sahrens 	if (read(ztest_random_fd, &r, sizeof (r)) != sizeof (r))
423fa9e4066Sahrens 		fatal(1, "short read from /dev/urandom");
424fa9e4066Sahrens 
425fa9e4066Sahrens 	return (r % range);
426fa9e4066Sahrens }
427fa9e4066Sahrens 
4288ad4d6ddSJeff Bonwick /* ARGSUSED */
429fa9e4066Sahrens static void
430fa9e4066Sahrens ztest_record_enospc(char *s)
431fa9e4066Sahrens {
432fa9e4066Sahrens 	ztest_shared->zs_enospc_count++;
433fa9e4066Sahrens }
434fa9e4066Sahrens 
435fa9e4066Sahrens static void
436fa9e4066Sahrens process_options(int argc, char **argv)
437fa9e4066Sahrens {
438fa9e4066Sahrens 	int opt;
439fa9e4066Sahrens 	uint64_t value;
440fa9e4066Sahrens 
441fa9e4066Sahrens 	/* By default, test gang blocks for blocks 32K and greater */
442e05725b1Sbonwick 	metaslab_gang_bang = 32 << 10;
443fa9e4066Sahrens 
444fa9e4066Sahrens 	while ((opt = getopt(argc, argv,
445e14bb325SJeff Bonwick 	    "v:s:a:m:r:R:d:t:g:i:k:p:f:VET:P:h")) != EOF) {
446fa9e4066Sahrens 		value = 0;
447fa9e4066Sahrens 		switch (opt) {
4483d7072f8Seschrock 		case 'v':
4493d7072f8Seschrock 		case 's':
4503d7072f8Seschrock 		case 'a':
4513d7072f8Seschrock 		case 'm':
4523d7072f8Seschrock 		case 'r':
4533d7072f8Seschrock 		case 'R':
4543d7072f8Seschrock 		case 'd':
4553d7072f8Seschrock 		case 't':
4563d7072f8Seschrock 		case 'g':
4573d7072f8Seschrock 		case 'i':
4583d7072f8Seschrock 		case 'k':
4593d7072f8Seschrock 		case 'T':
4603d7072f8Seschrock 		case 'P':
461fa9e4066Sahrens 			value = nicenumtoull(optarg);
462fa9e4066Sahrens 		}
463fa9e4066Sahrens 		switch (opt) {
4643d7072f8Seschrock 		case 'v':
465fa9e4066Sahrens 			zopt_vdevs = value;
466fa9e4066Sahrens 			break;
4673d7072f8Seschrock 		case 's':
468fa9e4066Sahrens 			zopt_vdev_size = MAX(SPA_MINDEVSIZE, value);
469fa9e4066Sahrens 			break;
4703d7072f8Seschrock 		case 'a':
471ecc2d604Sbonwick 			zopt_ashift = value;
472ecc2d604Sbonwick 			break;
4733d7072f8Seschrock 		case 'm':
474fa9e4066Sahrens 			zopt_mirrors = value;
475fa9e4066Sahrens 			break;
4763d7072f8Seschrock 		case 'r':
477fa9e4066Sahrens 			zopt_raidz = MAX(1, value);
478fa9e4066Sahrens 			break;
4793d7072f8Seschrock 		case 'R':
48099653d4eSeschrock 			zopt_raidz_parity = MIN(MAX(value, 1), 2);
48199653d4eSeschrock 			break;
4823d7072f8Seschrock 		case 'd':
483ecc2d604Sbonwick 			zopt_datasets = MAX(1, value);
484fa9e4066Sahrens 			break;
4853d7072f8Seschrock 		case 't':
486fa9e4066Sahrens 			zopt_threads = MAX(1, value);
487fa9e4066Sahrens 			break;
4883d7072f8Seschrock 		case 'g':
489e05725b1Sbonwick 			metaslab_gang_bang = MAX(SPA_MINBLOCKSIZE << 1, value);
490fa9e4066Sahrens 			break;
4913d7072f8Seschrock 		case 'i':
492fa9e4066Sahrens 			zopt_init = value;
493fa9e4066Sahrens 			break;
4943d7072f8Seschrock 		case 'k':
495fa9e4066Sahrens 			zopt_killrate = value;
496fa9e4066Sahrens 			break;
4973d7072f8Seschrock 		case 'p':
498fa9e4066Sahrens 			zopt_pool = strdup(optarg);
499fa9e4066Sahrens 			break;
5003d7072f8Seschrock 		case 'f':
501fa9e4066Sahrens 			zopt_dir = strdup(optarg);
502fa9e4066Sahrens 			break;
5033d7072f8Seschrock 		case 'V':
504fa9e4066Sahrens 			zopt_verbose++;
505fa9e4066Sahrens 			break;
5063d7072f8Seschrock 		case 'E':
507fa9e4066Sahrens 			zopt_init = 0;
508fa9e4066Sahrens 			break;
5093d7072f8Seschrock 		case 'T':
510fa9e4066Sahrens 			zopt_time = value;
511fa9e4066Sahrens 			break;
5123d7072f8Seschrock 		case 'P':
513fa9e4066Sahrens 			zopt_passtime = MAX(1, value);
514fa9e4066Sahrens 			break;
5153d7072f8Seschrock 		case 'h':
516f1b4288bSvb 			usage(B_TRUE);
517f1b4288bSvb 			break;
5183d7072f8Seschrock 		case '?':
5193d7072f8Seschrock 		default:
520f1b4288bSvb 			usage(B_FALSE);
521fa9e4066Sahrens 			break;
522fa9e4066Sahrens 		}
523fa9e4066Sahrens 	}
524fa9e4066Sahrens 
52599653d4eSeschrock 	zopt_raidz_parity = MIN(zopt_raidz_parity, zopt_raidz - 1);
52699653d4eSeschrock 
527fa9e4066Sahrens 	zopt_vdevtime = (zopt_vdevs > 0 ? zopt_time / zopt_vdevs : UINT64_MAX);
52899653d4eSeschrock 	zopt_maxfaults = MAX(zopt_mirrors, 1) * (zopt_raidz_parity + 1) - 1;
529fa9e4066Sahrens }
530fa9e4066Sahrens 
531ecc2d604Sbonwick static uint64_t
532ecc2d604Sbonwick ztest_get_ashift(void)
533ecc2d604Sbonwick {
534ecc2d604Sbonwick 	if (zopt_ashift == 0)
535ecc2d604Sbonwick 		return (SPA_MINBLOCKSHIFT + ztest_random(3));
536ecc2d604Sbonwick 	return (zopt_ashift);
537ecc2d604Sbonwick }
538ecc2d604Sbonwick 
539fa9e4066Sahrens static nvlist_t *
540e14bb325SJeff Bonwick make_vdev_file(char *path, char *aux, size_t size, uint64_t ashift)
541fa9e4066Sahrens {
542e14bb325SJeff Bonwick 	char pathbuf[MAXPATHLEN];
543fa9e4066Sahrens 	uint64_t vdev;
544fa9e4066Sahrens 	nvlist_t *file;
545fa9e4066Sahrens 
546e14bb325SJeff Bonwick 	if (ashift == 0)
547e14bb325SJeff Bonwick 		ashift = ztest_get_ashift();
548e14bb325SJeff Bonwick 
549e14bb325SJeff Bonwick 	if (path == NULL) {
550e14bb325SJeff Bonwick 		path = pathbuf;
551fa9e4066Sahrens 
552e14bb325SJeff Bonwick 		if (aux != NULL) {
553e14bb325SJeff Bonwick 			vdev = ztest_shared->zs_vdev_aux;
554e14bb325SJeff Bonwick 			(void) sprintf(path, ztest_aux_template,
555e14bb325SJeff Bonwick 			    zopt_dir, zopt_pool, aux, vdev);
556e14bb325SJeff Bonwick 		} else {
557e14bb325SJeff Bonwick 			vdev = ztest_shared->zs_vdev_primaries++;
558e14bb325SJeff Bonwick 			(void) sprintf(path, ztest_dev_template,
559e14bb325SJeff Bonwick 			    zopt_dir, zopt_pool, vdev);
560e14bb325SJeff Bonwick 		}
561e14bb325SJeff Bonwick 	}
562e14bb325SJeff Bonwick 
563e14bb325SJeff Bonwick 	if (size != 0) {
564e14bb325SJeff Bonwick 		int fd = open(path, O_RDWR | O_CREAT | O_TRUNC, 0666);
565fa9e4066Sahrens 		if (fd == -1)
566e14bb325SJeff Bonwick 			fatal(1, "can't open %s", path);
567fa9e4066Sahrens 		if (ftruncate(fd, size) != 0)
568e14bb325SJeff Bonwick 			fatal(1, "can't ftruncate %s", path);
569fa9e4066Sahrens 		(void) close(fd);
570fa9e4066Sahrens 	}
571fa9e4066Sahrens 
572fa9e4066Sahrens 	VERIFY(nvlist_alloc(&file, NV_UNIQUE_NAME, 0) == 0);
573fa9e4066Sahrens 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_TYPE, VDEV_TYPE_FILE) == 0);
574e14bb325SJeff Bonwick 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_PATH, path) == 0);
575ecc2d604Sbonwick 	VERIFY(nvlist_add_uint64(file, ZPOOL_CONFIG_ASHIFT, ashift) == 0);
576fa9e4066Sahrens 
577fa9e4066Sahrens 	return (file);
578fa9e4066Sahrens }
579fa9e4066Sahrens 
580fa9e4066Sahrens static nvlist_t *
581e14bb325SJeff Bonwick make_vdev_raidz(char *path, char *aux, size_t size, uint64_t ashift, int r)
582fa9e4066Sahrens {
583fa9e4066Sahrens 	nvlist_t *raidz, **child;
584fa9e4066Sahrens 	int c;
585fa9e4066Sahrens 
586fa9e4066Sahrens 	if (r < 2)
587e14bb325SJeff Bonwick 		return (make_vdev_file(path, aux, size, ashift));
588fa9e4066Sahrens 	child = umem_alloc(r * sizeof (nvlist_t *), UMEM_NOFAIL);
589fa9e4066Sahrens 
590fa9e4066Sahrens 	for (c = 0; c < r; c++)
591e14bb325SJeff Bonwick 		child[c] = make_vdev_file(path, aux, size, ashift);
592fa9e4066Sahrens 
593fa9e4066Sahrens 	VERIFY(nvlist_alloc(&raidz, NV_UNIQUE_NAME, 0) == 0);
594fa9e4066Sahrens 	VERIFY(nvlist_add_string(raidz, ZPOOL_CONFIG_TYPE,
595fa9e4066Sahrens 	    VDEV_TYPE_RAIDZ) == 0);
59699653d4eSeschrock 	VERIFY(nvlist_add_uint64(raidz, ZPOOL_CONFIG_NPARITY,
59799653d4eSeschrock 	    zopt_raidz_parity) == 0);
598fa9e4066Sahrens 	VERIFY(nvlist_add_nvlist_array(raidz, ZPOOL_CONFIG_CHILDREN,
599fa9e4066Sahrens 	    child, r) == 0);
600fa9e4066Sahrens 
601fa9e4066Sahrens 	for (c = 0; c < r; c++)
602fa9e4066Sahrens 		nvlist_free(child[c]);
603fa9e4066Sahrens 
604fa9e4066Sahrens 	umem_free(child, r * sizeof (nvlist_t *));
605fa9e4066Sahrens 
606fa9e4066Sahrens 	return (raidz);
607fa9e4066Sahrens }
608fa9e4066Sahrens 
609fa9e4066Sahrens static nvlist_t *
610e14bb325SJeff Bonwick make_vdev_mirror(char *path, char *aux, size_t size, uint64_t ashift,
61131157203SJeff Bonwick 	int r, int m)
612fa9e4066Sahrens {
613fa9e4066Sahrens 	nvlist_t *mirror, **child;
614fa9e4066Sahrens 	int c;
615fa9e4066Sahrens 
616fa9e4066Sahrens 	if (m < 1)
617e14bb325SJeff Bonwick 		return (make_vdev_raidz(path, aux, size, ashift, r));
618fa9e4066Sahrens 
619fa9e4066Sahrens 	child = umem_alloc(m * sizeof (nvlist_t *), UMEM_NOFAIL);
620fa9e4066Sahrens 
621fa9e4066Sahrens 	for (c = 0; c < m; c++)
622e14bb325SJeff Bonwick 		child[c] = make_vdev_raidz(path, aux, size, ashift, r);
623fa9e4066Sahrens 
624fa9e4066Sahrens 	VERIFY(nvlist_alloc(&mirror, NV_UNIQUE_NAME, 0) == 0);
625fa9e4066Sahrens 	VERIFY(nvlist_add_string(mirror, ZPOOL_CONFIG_TYPE,
626fa9e4066Sahrens 	    VDEV_TYPE_MIRROR) == 0);
627fa9e4066Sahrens 	VERIFY(nvlist_add_nvlist_array(mirror, ZPOOL_CONFIG_CHILDREN,
628fa9e4066Sahrens 	    child, m) == 0);
629fa9e4066Sahrens 
630fa9e4066Sahrens 	for (c = 0; c < m; c++)
631fa9e4066Sahrens 		nvlist_free(child[c]);
632fa9e4066Sahrens 
633fa9e4066Sahrens 	umem_free(child, m * sizeof (nvlist_t *));
634fa9e4066Sahrens 
635fa9e4066Sahrens 	return (mirror);
636fa9e4066Sahrens }
637fa9e4066Sahrens 
638fa9e4066Sahrens static nvlist_t *
639e14bb325SJeff Bonwick make_vdev_root(char *path, char *aux, size_t size, uint64_t ashift,
640e14bb325SJeff Bonwick 	int log, int r, int m, int t)
641fa9e4066Sahrens {
642fa9e4066Sahrens 	nvlist_t *root, **child;
643fa9e4066Sahrens 	int c;
644fa9e4066Sahrens 
645fa9e4066Sahrens 	ASSERT(t > 0);
646fa9e4066Sahrens 
647fa9e4066Sahrens 	child = umem_alloc(t * sizeof (nvlist_t *), UMEM_NOFAIL);
648fa9e4066Sahrens 
64931157203SJeff Bonwick 	for (c = 0; c < t; c++) {
65031157203SJeff Bonwick 		child[c] = make_vdev_mirror(path, aux, size, ashift, r, m);
65131157203SJeff Bonwick 		VERIFY(nvlist_add_uint64(child[c], ZPOOL_CONFIG_IS_LOG,
65231157203SJeff Bonwick 		    log) == 0);
65331157203SJeff Bonwick 	}
654fa9e4066Sahrens 
655fa9e4066Sahrens 	VERIFY(nvlist_alloc(&root, NV_UNIQUE_NAME, 0) == 0);
656fa9e4066Sahrens 	VERIFY(nvlist_add_string(root, ZPOOL_CONFIG_TYPE, VDEV_TYPE_ROOT) == 0);
657e14bb325SJeff Bonwick 	VERIFY(nvlist_add_nvlist_array(root, aux ? aux : ZPOOL_CONFIG_CHILDREN,
658fa9e4066Sahrens 	    child, t) == 0);
659fa9e4066Sahrens 
660fa9e4066Sahrens 	for (c = 0; c < t; c++)
661fa9e4066Sahrens 		nvlist_free(child[c]);
662fa9e4066Sahrens 
663fa9e4066Sahrens 	umem_free(child, t * sizeof (nvlist_t *));
664fa9e4066Sahrens 
665fa9e4066Sahrens 	return (root);
666fa9e4066Sahrens }
667fa9e4066Sahrens 
668fa9e4066Sahrens static void
669fa9e4066Sahrens ztest_set_random_blocksize(objset_t *os, uint64_t object, dmu_tx_t *tx)
670fa9e4066Sahrens {
671fa9e4066Sahrens 	int bs = SPA_MINBLOCKSHIFT +
672fa9e4066Sahrens 	    ztest_random(SPA_MAXBLOCKSHIFT - SPA_MINBLOCKSHIFT + 1);
673fa9e4066Sahrens 	int ibs = DN_MIN_INDBLKSHIFT +
674fa9e4066Sahrens 	    ztest_random(DN_MAX_INDBLKSHIFT - DN_MIN_INDBLKSHIFT + 1);
675fa9e4066Sahrens 	int error;
676fa9e4066Sahrens 
677fa9e4066Sahrens 	error = dmu_object_set_blocksize(os, object, 1ULL << bs, ibs, tx);
678fa9e4066Sahrens 	if (error) {
679fa9e4066Sahrens 		char osname[300];
680fa9e4066Sahrens 		dmu_objset_name(os, osname);
681fa9e4066Sahrens 		fatal(0, "dmu_object_set_blocksize('%s', %llu, %d, %d) = %d",
682fa9e4066Sahrens 		    osname, object, 1 << bs, ibs, error);
683fa9e4066Sahrens 	}
684fa9e4066Sahrens }
685fa9e4066Sahrens 
686fa9e4066Sahrens static uint8_t
687fa9e4066Sahrens ztest_random_checksum(void)
688fa9e4066Sahrens {
689fa9e4066Sahrens 	uint8_t checksum;
690fa9e4066Sahrens 
691fa9e4066Sahrens 	do {
692fa9e4066Sahrens 		checksum = ztest_random(ZIO_CHECKSUM_FUNCTIONS);
693fa9e4066Sahrens 	} while (zio_checksum_table[checksum].ci_zbt);
694fa9e4066Sahrens 
695fa9e4066Sahrens 	if (checksum == ZIO_CHECKSUM_OFF)
696fa9e4066Sahrens 		checksum = ZIO_CHECKSUM_ON;
697fa9e4066Sahrens 
698fa9e4066Sahrens 	return (checksum);
699fa9e4066Sahrens }
700fa9e4066Sahrens 
701fa9e4066Sahrens static uint8_t
702fa9e4066Sahrens ztest_random_compress(void)
703fa9e4066Sahrens {
704fa9e4066Sahrens 	return ((uint8_t)ztest_random(ZIO_COMPRESS_FUNCTIONS));
705fa9e4066Sahrens }
706fa9e4066Sahrens 
707fa9e4066Sahrens static int
7081209a471SNeil Perrin ztest_replay_create(objset_t *os, lr_create_t *lr, boolean_t byteswap)
709fa9e4066Sahrens {
710fa9e4066Sahrens 	dmu_tx_t *tx;
711fa9e4066Sahrens 	int error;
712fa9e4066Sahrens 
713fa9e4066Sahrens 	if (byteswap)
714fa9e4066Sahrens 		byteswap_uint64_array(lr, sizeof (*lr));
715fa9e4066Sahrens 
716fa9e4066Sahrens 	tx = dmu_tx_create(os);
717fa9e4066Sahrens 	dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
7181209a471SNeil Perrin 	error = dmu_tx_assign(tx, TXG_WAIT);
719fa9e4066Sahrens 	if (error) {
720fa9e4066Sahrens 		dmu_tx_abort(tx);
721fa9e4066Sahrens 		return (error);
722fa9e4066Sahrens 	}
723fa9e4066Sahrens 
724fa9e4066Sahrens 	error = dmu_object_claim(os, lr->lr_doid, lr->lr_mode, 0,
725fa9e4066Sahrens 	    DMU_OT_NONE, 0, tx);
72699653d4eSeschrock 	ASSERT3U(error, ==, 0);
727fa9e4066Sahrens 	dmu_tx_commit(tx);
728fa9e4066Sahrens 
729fa9e4066Sahrens 	if (zopt_verbose >= 5) {
730fa9e4066Sahrens 		char osname[MAXNAMELEN];
731fa9e4066Sahrens 		dmu_objset_name(os, osname);
732fa9e4066Sahrens 		(void) printf("replay create of %s object %llu"
733fa9e4066Sahrens 		    " in txg %llu = %d\n",
734fa9e4066Sahrens 		    osname, (u_longlong_t)lr->lr_doid,
7351209a471SNeil Perrin 		    (u_longlong_t)dmu_tx_get_txg(tx), error);
736fa9e4066Sahrens 	}
737fa9e4066Sahrens 
738fa9e4066Sahrens 	return (error);
739fa9e4066Sahrens }
740fa9e4066Sahrens 
741fa9e4066Sahrens static int
7421209a471SNeil Perrin ztest_replay_remove(objset_t *os, lr_remove_t *lr, boolean_t byteswap)
743fa9e4066Sahrens {
744fa9e4066Sahrens 	dmu_tx_t *tx;
745fa9e4066Sahrens 	int error;
746fa9e4066Sahrens 
747fa9e4066Sahrens 	if (byteswap)
748fa9e4066Sahrens 		byteswap_uint64_array(lr, sizeof (*lr));
749fa9e4066Sahrens 
750fa9e4066Sahrens 	tx = dmu_tx_create(os);
751fa9e4066Sahrens 	dmu_tx_hold_free(tx, lr->lr_doid, 0, DMU_OBJECT_END);
7521209a471SNeil Perrin 	error = dmu_tx_assign(tx, TXG_WAIT);
753fa9e4066Sahrens 	if (error) {
754fa9e4066Sahrens 		dmu_tx_abort(tx);
755fa9e4066Sahrens 		return (error);
756fa9e4066Sahrens 	}
757fa9e4066Sahrens 
758fa9e4066Sahrens 	error = dmu_object_free(os, lr->lr_doid, tx);
759fa9e4066Sahrens 	dmu_tx_commit(tx);
760fa9e4066Sahrens 
761fa9e4066Sahrens 	return (error);
762fa9e4066Sahrens }
763fa9e4066Sahrens 
764fa9e4066Sahrens zil_replay_func_t *ztest_replay_vector[TX_MAX_TYPE] = {
765fa9e4066Sahrens 	NULL,			/* 0 no such transaction type */
766fa9e4066Sahrens 	ztest_replay_create,	/* TX_CREATE */
767fa9e4066Sahrens 	NULL,			/* TX_MKDIR */
768fa9e4066Sahrens 	NULL,			/* TX_MKXATTR */
769fa9e4066Sahrens 	NULL,			/* TX_SYMLINK */
770fa9e4066Sahrens 	ztest_replay_remove,	/* TX_REMOVE */
771fa9e4066Sahrens 	NULL,			/* TX_RMDIR */
772fa9e4066Sahrens 	NULL,			/* TX_LINK */
773fa9e4066Sahrens 	NULL,			/* TX_RENAME */
774fa9e4066Sahrens 	NULL,			/* TX_WRITE */
775fa9e4066Sahrens 	NULL,			/* TX_TRUNCATE */
776fa9e4066Sahrens 	NULL,			/* TX_SETATTR */
777fa9e4066Sahrens 	NULL,			/* TX_ACL */
778fa9e4066Sahrens };
779fa9e4066Sahrens 
780fa9e4066Sahrens /*
781fa9e4066Sahrens  * Verify that we can't destroy an active pool, create an existing pool,
782fa9e4066Sahrens  * or create a pool with a bad vdev spec.
783fa9e4066Sahrens  */
784fa9e4066Sahrens void
785fa9e4066Sahrens ztest_spa_create_destroy(ztest_args_t *za)
786fa9e4066Sahrens {
787fa9e4066Sahrens 	int error;
788fa9e4066Sahrens 	spa_t *spa;
789fa9e4066Sahrens 	nvlist_t *nvroot;
790fa9e4066Sahrens 
791fa9e4066Sahrens 	/*
792fa9e4066Sahrens 	 * Attempt to create using a bad file.
793fa9e4066Sahrens 	 */
794e14bb325SJeff Bonwick 	nvroot = make_vdev_root("/dev/bogus", NULL, 0, 0, 0, 0, 0, 1);
7950a48a24eStimh 	error = spa_create("ztest_bad_file", nvroot, NULL, NULL, NULL);
796fa9e4066Sahrens 	nvlist_free(nvroot);
797fa9e4066Sahrens 	if (error != ENOENT)
798fa9e4066Sahrens 		fatal(0, "spa_create(bad_file) = %d", error);
799fa9e4066Sahrens 
800fa9e4066Sahrens 	/*
801fa9e4066Sahrens 	 * Attempt to create using a bad mirror.
802fa9e4066Sahrens 	 */
803e14bb325SJeff Bonwick 	nvroot = make_vdev_root("/dev/bogus", NULL, 0, 0, 0, 0, 2, 1);
8040a48a24eStimh 	error = spa_create("ztest_bad_mirror", nvroot, NULL, NULL, NULL);
805fa9e4066Sahrens 	nvlist_free(nvroot);
806fa9e4066Sahrens 	if (error != ENOENT)
807fa9e4066Sahrens 		fatal(0, "spa_create(bad_mirror) = %d", error);
808fa9e4066Sahrens 
809fa9e4066Sahrens 	/*
810fa9e4066Sahrens 	 * Attempt to create an existing pool.  It shouldn't matter
811fa9e4066Sahrens 	 * what's in the nvroot; we should fail with EEXIST.
812fa9e4066Sahrens 	 */
813fa9e4066Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
814e14bb325SJeff Bonwick 	nvroot = make_vdev_root("/dev/bogus", NULL, 0, 0, 0, 0, 0, 1);
8150a48a24eStimh 	error = spa_create(za->za_pool, nvroot, NULL, NULL, NULL);
816fa9e4066Sahrens 	nvlist_free(nvroot);
817fa9e4066Sahrens 	if (error != EEXIST)
818fa9e4066Sahrens 		fatal(0, "spa_create(whatever) = %d", error);
819fa9e4066Sahrens 
820fa9e4066Sahrens 	error = spa_open(za->za_pool, &spa, FTAG);
821fa9e4066Sahrens 	if (error)
822fa9e4066Sahrens 		fatal(0, "spa_open() = %d", error);
823fa9e4066Sahrens 
824fa9e4066Sahrens 	error = spa_destroy(za->za_pool);
825fa9e4066Sahrens 	if (error != EBUSY)
826fa9e4066Sahrens 		fatal(0, "spa_destroy() = %d", error);
827fa9e4066Sahrens 
828fa9e4066Sahrens 	spa_close(spa, FTAG);
829fa9e4066Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
830fa9e4066Sahrens }
831fa9e4066Sahrens 
83231157203SJeff Bonwick static vdev_t *
83331157203SJeff Bonwick vdev_lookup_by_path(vdev_t *vd, const char *path)
83431157203SJeff Bonwick {
83531157203SJeff Bonwick 	vdev_t *mvd;
83631157203SJeff Bonwick 
83731157203SJeff Bonwick 	if (vd->vdev_path != NULL && strcmp(path, vd->vdev_path) == 0)
83831157203SJeff Bonwick 		return (vd);
83931157203SJeff Bonwick 
84031157203SJeff Bonwick 	for (int c = 0; c < vd->vdev_children; c++)
84131157203SJeff Bonwick 		if ((mvd = vdev_lookup_by_path(vd->vdev_child[c], path)) !=
84231157203SJeff Bonwick 		    NULL)
84331157203SJeff Bonwick 			return (mvd);
84431157203SJeff Bonwick 
84531157203SJeff Bonwick 	return (NULL);
84631157203SJeff Bonwick }
84731157203SJeff Bonwick 
848fa9e4066Sahrens /*
849fa9e4066Sahrens  * Verify that vdev_add() works as expected.
850fa9e4066Sahrens  */
851fa9e4066Sahrens void
852fa9e4066Sahrens ztest_vdev_add_remove(ztest_args_t *za)
853fa9e4066Sahrens {
854e05725b1Sbonwick 	spa_t *spa = za->za_spa;
855fa9e4066Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
856fa9e4066Sahrens 	nvlist_t *nvroot;
857fa9e4066Sahrens 	int error;
858fa9e4066Sahrens 
859fa9e4066Sahrens 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
860fa9e4066Sahrens 
861e14bb325SJeff Bonwick 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
862fa9e4066Sahrens 
863fa9e4066Sahrens 	ztest_shared->zs_vdev_primaries =
864fa9e4066Sahrens 	    spa->spa_root_vdev->vdev_children * leaves;
865fa9e4066Sahrens 
866e14bb325SJeff Bonwick 	spa_config_exit(spa, SCL_VDEV, FTAG);
867fa9e4066Sahrens 
8688654d025Sperrin 	/*
8698654d025Sperrin 	 * Make 1/4 of the devices be log devices.
8708654d025Sperrin 	 */
871e14bb325SJeff Bonwick 	nvroot = make_vdev_root(NULL, NULL, zopt_vdev_size, 0,
8728654d025Sperrin 	    ztest_random(4) == 0, zopt_raidz, zopt_mirrors, 1);
8738654d025Sperrin 
874fa9e4066Sahrens 	error = spa_vdev_add(spa, nvroot);
875fa9e4066Sahrens 	nvlist_free(nvroot);
876fa9e4066Sahrens 
877fa9e4066Sahrens 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
878fa9e4066Sahrens 
879fa9e4066Sahrens 	if (error == ENOSPC)
880fa9e4066Sahrens 		ztest_record_enospc("spa_vdev_add");
881fa9e4066Sahrens 	else if (error != 0)
882fa9e4066Sahrens 		fatal(0, "spa_vdev_add() = %d", error);
883e14bb325SJeff Bonwick }
884fa9e4066Sahrens 
885e14bb325SJeff Bonwick /*
886e14bb325SJeff Bonwick  * Verify that adding/removing aux devices (l2arc, hot spare) works as expected.
887e14bb325SJeff Bonwick  */
888e14bb325SJeff Bonwick void
889e14bb325SJeff Bonwick ztest_vdev_aux_add_remove(ztest_args_t *za)
890e14bb325SJeff Bonwick {
891e14bb325SJeff Bonwick 	spa_t *spa = za->za_spa;
89231157203SJeff Bonwick 	vdev_t *rvd = spa->spa_root_vdev;
893e14bb325SJeff Bonwick 	spa_aux_vdev_t *sav;
894e14bb325SJeff Bonwick 	char *aux;
895e14bb325SJeff Bonwick 	uint64_t guid = 0;
896e14bb325SJeff Bonwick 	int error;
897e14bb325SJeff Bonwick 
89831157203SJeff Bonwick 	if (ztest_random(2) == 0) {
899e14bb325SJeff Bonwick 		sav = &spa->spa_spares;
900e14bb325SJeff Bonwick 		aux = ZPOOL_CONFIG_SPARES;
901e14bb325SJeff Bonwick 	} else {
902e14bb325SJeff Bonwick 		sav = &spa->spa_l2cache;
903e14bb325SJeff Bonwick 		aux = ZPOOL_CONFIG_L2CACHE;
904e14bb325SJeff Bonwick 	}
905e14bb325SJeff Bonwick 
906e14bb325SJeff Bonwick 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
907e14bb325SJeff Bonwick 
908e14bb325SJeff Bonwick 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
909e14bb325SJeff Bonwick 
910e14bb325SJeff Bonwick 	if (sav->sav_count != 0 && ztest_random(4) == 0) {
911e14bb325SJeff Bonwick 		/*
912e14bb325SJeff Bonwick 		 * Pick a random device to remove.
913e14bb325SJeff Bonwick 		 */
914e14bb325SJeff Bonwick 		guid = sav->sav_vdevs[ztest_random(sav->sav_count)]->vdev_guid;
915e14bb325SJeff Bonwick 	} else {
916e14bb325SJeff Bonwick 		/*
917e14bb325SJeff Bonwick 		 * Find an unused device we can add.
918e14bb325SJeff Bonwick 		 */
919e14bb325SJeff Bonwick 		ztest_shared->zs_vdev_aux = 0;
920e14bb325SJeff Bonwick 		for (;;) {
921e14bb325SJeff Bonwick 			char path[MAXPATHLEN];
922e14bb325SJeff Bonwick 			int c;
923e14bb325SJeff Bonwick 			(void) sprintf(path, ztest_aux_template, zopt_dir,
924e14bb325SJeff Bonwick 			    zopt_pool, aux, ztest_shared->zs_vdev_aux);
925e14bb325SJeff Bonwick 			for (c = 0; c < sav->sav_count; c++)
926e14bb325SJeff Bonwick 				if (strcmp(sav->sav_vdevs[c]->vdev_path,
927e14bb325SJeff Bonwick 				    path) == 0)
928e14bb325SJeff Bonwick 					break;
92931157203SJeff Bonwick 			if (c == sav->sav_count &&
93031157203SJeff Bonwick 			    vdev_lookup_by_path(rvd, path) == NULL)
931e14bb325SJeff Bonwick 				break;
932e14bb325SJeff Bonwick 			ztest_shared->zs_vdev_aux++;
933e14bb325SJeff Bonwick 		}
934e14bb325SJeff Bonwick 	}
935e14bb325SJeff Bonwick 
936e14bb325SJeff Bonwick 	spa_config_exit(spa, SCL_VDEV, FTAG);
937e14bb325SJeff Bonwick 
938e14bb325SJeff Bonwick 	if (guid == 0) {
939e14bb325SJeff Bonwick 		/*
940e14bb325SJeff Bonwick 		 * Add a new device.
941e14bb325SJeff Bonwick 		 */
94231157203SJeff Bonwick 		nvlist_t *nvroot = make_vdev_root(NULL, aux,
94331157203SJeff Bonwick 		    (zopt_vdev_size * 5) / 4, 0, 0, 0, 0, 1);
944e14bb325SJeff Bonwick 		error = spa_vdev_add(spa, nvroot);
945e14bb325SJeff Bonwick 		if (error != 0)
946e14bb325SJeff Bonwick 			fatal(0, "spa_vdev_add(%p) = %d", nvroot, error);
947e14bb325SJeff Bonwick 		nvlist_free(nvroot);
948e14bb325SJeff Bonwick 	} else {
949e14bb325SJeff Bonwick 		/*
950e14bb325SJeff Bonwick 		 * Remove an existing device.  Sometimes, dirty its
951e14bb325SJeff Bonwick 		 * vdev state first to make sure we handle removal
952e14bb325SJeff Bonwick 		 * of devices that have pending state changes.
953e14bb325SJeff Bonwick 		 */
954e14bb325SJeff Bonwick 		if (ztest_random(2) == 0)
955e14bb325SJeff Bonwick 			(void) vdev_online(spa, guid, B_FALSE, NULL);
956e14bb325SJeff Bonwick 
957e14bb325SJeff Bonwick 		error = spa_vdev_remove(spa, guid, B_FALSE);
958e14bb325SJeff Bonwick 		if (error != 0 && error != EBUSY)
959e14bb325SJeff Bonwick 			fatal(0, "spa_vdev_remove(%llu) = %d", guid, error);
960e14bb325SJeff Bonwick 	}
961e14bb325SJeff Bonwick 
962e14bb325SJeff Bonwick 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
963fa9e4066Sahrens }
964fa9e4066Sahrens 
965fa9e4066Sahrens /*
966fa9e4066Sahrens  * Verify that we can attach and detach devices.
967fa9e4066Sahrens  */
968fa9e4066Sahrens void
969fa9e4066Sahrens ztest_vdev_attach_detach(ztest_args_t *za)
970fa9e4066Sahrens {
971e05725b1Sbonwick 	spa_t *spa = za->za_spa;
97231157203SJeff Bonwick 	spa_aux_vdev_t *sav = &spa->spa_spares;
973fa9e4066Sahrens 	vdev_t *rvd = spa->spa_root_vdev;
974ea8dc4b6Seschrock 	vdev_t *oldvd, *newvd, *pvd;
975e14bb325SJeff Bonwick 	nvlist_t *root;
976fa9e4066Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
977fa9e4066Sahrens 	uint64_t leaf, top;
978ecc2d604Sbonwick 	uint64_t ashift = ztest_get_ashift();
9798ad4d6ddSJeff Bonwick 	uint64_t oldguid, pguid;
980ea8dc4b6Seschrock 	size_t oldsize, newsize;
981ea8dc4b6Seschrock 	char oldpath[MAXPATHLEN], newpath[MAXPATHLEN];
982fa9e4066Sahrens 	int replacing;
9839af0a4dfSJeff Bonwick 	int oldvd_has_siblings = B_FALSE;
98431157203SJeff Bonwick 	int newvd_is_spare = B_FALSE;
98531157203SJeff Bonwick 	int oldvd_is_log;
986fa9e4066Sahrens 	int error, expected_error;
987fa9e4066Sahrens 
988fa9e4066Sahrens 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
989fa9e4066Sahrens 
990e14bb325SJeff Bonwick 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
991fa9e4066Sahrens 
992fa9e4066Sahrens 	/*
993fa9e4066Sahrens 	 * Decide whether to do an attach or a replace.
994fa9e4066Sahrens 	 */
995fa9e4066Sahrens 	replacing = ztest_random(2);
996fa9e4066Sahrens 
997fa9e4066Sahrens 	/*
998fa9e4066Sahrens 	 * Pick a random top-level vdev.
999fa9e4066Sahrens 	 */
1000fa9e4066Sahrens 	top = ztest_random(rvd->vdev_children);
1001fa9e4066Sahrens 
1002fa9e4066Sahrens 	/*
1003fa9e4066Sahrens 	 * Pick a random leaf within it.
1004fa9e4066Sahrens 	 */
1005fa9e4066Sahrens 	leaf = ztest_random(leaves);
1006fa9e4066Sahrens 
1007fa9e4066Sahrens 	/*
100831157203SJeff Bonwick 	 * Locate this vdev.
1009fa9e4066Sahrens 	 */
101031157203SJeff Bonwick 	oldvd = rvd->vdev_child[top];
10118ad4d6ddSJeff Bonwick 	if (zopt_mirrors >= 1) {
10128ad4d6ddSJeff Bonwick 		ASSERT(oldvd->vdev_ops == &vdev_mirror_ops);
10138ad4d6ddSJeff Bonwick 		ASSERT(oldvd->vdev_children >= zopt_mirrors);
101431157203SJeff Bonwick 		oldvd = oldvd->vdev_child[leaf / zopt_raidz];
10158ad4d6ddSJeff Bonwick 	}
10168ad4d6ddSJeff Bonwick 	if (zopt_raidz > 1) {
10178ad4d6ddSJeff Bonwick 		ASSERT(oldvd->vdev_ops == &vdev_raidz_ops);
10188ad4d6ddSJeff Bonwick 		ASSERT(oldvd->vdev_children == zopt_raidz);
101931157203SJeff Bonwick 		oldvd = oldvd->vdev_child[leaf % zopt_raidz];
10208ad4d6ddSJeff Bonwick 	}
1021fa9e4066Sahrens 
1022fa9e4066Sahrens 	/*
102331157203SJeff Bonwick 	 * If we're already doing an attach or replace, oldvd may be a
102431157203SJeff Bonwick 	 * mirror vdev -- in which case, pick a random child.
1025fa9e4066Sahrens 	 */
102631157203SJeff Bonwick 	while (oldvd->vdev_children != 0) {
10279af0a4dfSJeff Bonwick 		oldvd_has_siblings = B_TRUE;
10288ad4d6ddSJeff Bonwick 		ASSERT(oldvd->vdev_children >= 2);
10298ad4d6ddSJeff Bonwick 		oldvd = oldvd->vdev_child[ztest_random(oldvd->vdev_children)];
103031157203SJeff Bonwick 	}
1031fa9e4066Sahrens 
103231157203SJeff Bonwick 	oldguid = oldvd->vdev_guid;
103331157203SJeff Bonwick 	oldsize = vdev_get_rsize(oldvd);
103431157203SJeff Bonwick 	oldvd_is_log = oldvd->vdev_top->vdev_islog;
103531157203SJeff Bonwick 	(void) strcpy(oldpath, oldvd->vdev_path);
1036ea8dc4b6Seschrock 	pvd = oldvd->vdev_parent;
10378ad4d6ddSJeff Bonwick 	pguid = pvd->vdev_guid;
1038fa9e4066Sahrens 
10399af0a4dfSJeff Bonwick 	/*
10409af0a4dfSJeff Bonwick 	 * If oldvd has siblings, then half of the time, detach it.
10419af0a4dfSJeff Bonwick 	 */
10429af0a4dfSJeff Bonwick 	if (oldvd_has_siblings && ztest_random(2) == 0) {
10439af0a4dfSJeff Bonwick 		spa_config_exit(spa, SCL_VDEV, FTAG);
10448ad4d6ddSJeff Bonwick 		error = spa_vdev_detach(spa, oldguid, pguid, B_FALSE);
10458ad4d6ddSJeff Bonwick 		if (error != 0 && error != ENODEV && error != EBUSY &&
10468ad4d6ddSJeff Bonwick 		    error != ENOTSUP)
10478ad4d6ddSJeff Bonwick 			fatal(0, "detach (%s) returned %d", oldpath, error);
10489af0a4dfSJeff Bonwick 		(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
10499af0a4dfSJeff Bonwick 		return;
10509af0a4dfSJeff Bonwick 	}
10519af0a4dfSJeff Bonwick 
1052fa9e4066Sahrens 	/*
105331157203SJeff Bonwick 	 * For the new vdev, choose with equal probability between the two
105431157203SJeff Bonwick 	 * standard paths (ending in either 'a' or 'b') or a random hot spare.
1055fa9e4066Sahrens 	 */
105631157203SJeff Bonwick 	if (sav->sav_count != 0 && ztest_random(3) == 0) {
105731157203SJeff Bonwick 		newvd = sav->sav_vdevs[ztest_random(sav->sav_count)];
105831157203SJeff Bonwick 		newvd_is_spare = B_TRUE;
105931157203SJeff Bonwick 		(void) strcpy(newpath, newvd->vdev_path);
106031157203SJeff Bonwick 	} else {
106131157203SJeff Bonwick 		(void) snprintf(newpath, sizeof (newpath), ztest_dev_template,
106231157203SJeff Bonwick 		    zopt_dir, zopt_pool, top * leaves + leaf);
106331157203SJeff Bonwick 		if (ztest_random(2) == 0)
106431157203SJeff Bonwick 			newpath[strlen(newpath) - 1] = 'b';
106531157203SJeff Bonwick 		newvd = vdev_lookup_by_path(rvd, newpath);
106631157203SJeff Bonwick 	}
106731157203SJeff Bonwick 
106831157203SJeff Bonwick 	if (newvd) {
106931157203SJeff Bonwick 		newsize = vdev_get_rsize(newvd);
107031157203SJeff Bonwick 	} else {
107131157203SJeff Bonwick 		/*
107231157203SJeff Bonwick 		 * Make newsize a little bigger or smaller than oldsize.
107331157203SJeff Bonwick 		 * If it's smaller, the attach should fail.
107431157203SJeff Bonwick 		 * If it's larger, and we're doing a replace,
107531157203SJeff Bonwick 		 * we should get dynamic LUN growth when we're done.
107631157203SJeff Bonwick 		 */
107731157203SJeff Bonwick 		newsize = 10 * oldsize / (9 + ztest_random(3));
107831157203SJeff Bonwick 	}
1079fa9e4066Sahrens 
1080fa9e4066Sahrens 	/*
1081fa9e4066Sahrens 	 * If pvd is not a mirror or root, the attach should fail with ENOTSUP,
1082fa9e4066Sahrens 	 * unless it's a replace; in that case any non-replacing parent is OK.
1083fa9e4066Sahrens 	 *
1084ea8dc4b6Seschrock 	 * If newvd is already part of the pool, it should fail with EBUSY.
1085fa9e4066Sahrens 	 *
1086ea8dc4b6Seschrock 	 * If newvd is too small, it should fail with EOVERFLOW.
1087fa9e4066Sahrens 	 */
108831157203SJeff Bonwick 	if (pvd->vdev_ops != &vdev_mirror_ops &&
108931157203SJeff Bonwick 	    pvd->vdev_ops != &vdev_root_ops && (!replacing ||
109031157203SJeff Bonwick 	    pvd->vdev_ops == &vdev_replacing_ops ||
109131157203SJeff Bonwick 	    pvd->vdev_ops == &vdev_spare_ops))
109231157203SJeff Bonwick 		expected_error = ENOTSUP;
109331157203SJeff Bonwick 	else if (newvd_is_spare && (!replacing || oldvd_is_log))
1094fa9e4066Sahrens 		expected_error = ENOTSUP;
109531157203SJeff Bonwick 	else if (newvd == oldvd)
109631157203SJeff Bonwick 		expected_error = replacing ? 0 : EBUSY;
109731157203SJeff Bonwick 	else if (vdev_lookup_by_path(rvd, newpath) != NULL)
109831157203SJeff Bonwick 		expected_error = EBUSY;
1099ea8dc4b6Seschrock 	else if (newsize < oldsize)
1100fa9e4066Sahrens 		expected_error = EOVERFLOW;
1101ecc2d604Sbonwick 	else if (ashift > oldvd->vdev_top->vdev_ashift)
1102ecc2d604Sbonwick 		expected_error = EDOM;
1103fa9e4066Sahrens 	else
1104fa9e4066Sahrens 		expected_error = 0;
1105fa9e4066Sahrens 
1106e14bb325SJeff Bonwick 	spa_config_exit(spa, SCL_VDEV, FTAG);
1107fa9e4066Sahrens 
1108fa9e4066Sahrens 	/*
1109ea8dc4b6Seschrock 	 * Build the nvlist describing newpath.
1110fa9e4066Sahrens 	 */
1111e14bb325SJeff Bonwick 	root = make_vdev_root(newpath, NULL, newvd == NULL ? newsize : 0,
1112e14bb325SJeff Bonwick 	    ashift, 0, 0, 0, 1);
1113fa9e4066Sahrens 
111431157203SJeff Bonwick 	error = spa_vdev_attach(spa, oldguid, root, replacing);
1115fa9e4066Sahrens 
1116fa9e4066Sahrens 	nvlist_free(root);
1117fa9e4066Sahrens 
1118fa9e4066Sahrens 	/*
1119fa9e4066Sahrens 	 * If our parent was the replacing vdev, but the replace completed,
1120fa9e4066Sahrens 	 * then instead of failing with ENOTSUP we may either succeed,
1121fa9e4066Sahrens 	 * fail with ENODEV, or fail with EOVERFLOW.
1122fa9e4066Sahrens 	 */
1123fa9e4066Sahrens 	if (expected_error == ENOTSUP &&
1124fa9e4066Sahrens 	    (error == 0 || error == ENODEV || error == EOVERFLOW))
1125fa9e4066Sahrens 		expected_error = error;
1126fa9e4066Sahrens 
1127f0aa80d4Sbonwick 	/*
1128f0aa80d4Sbonwick 	 * If someone grew the LUN, the replacement may be too small.
1129f0aa80d4Sbonwick 	 */
1130088f3894Sahrens 	if (error == EOVERFLOW || error == EBUSY)
1131f0aa80d4Sbonwick 		expected_error = error;
1132f0aa80d4Sbonwick 
1133088f3894Sahrens 	/* XXX workaround 6690467 */
1134088f3894Sahrens 	if (error != expected_error && expected_error != EBUSY) {
1135088f3894Sahrens 		fatal(0, "attach (%s %llu, %s %llu, %d) "
1136088f3894Sahrens 		    "returned %d, expected %d",
1137088f3894Sahrens 		    oldpath, (longlong_t)oldsize, newpath,
1138088f3894Sahrens 		    (longlong_t)newsize, replacing, error, expected_error);
1139fa9e4066Sahrens 	}
1140fa9e4066Sahrens 
1141fa9e4066Sahrens 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
1142fa9e4066Sahrens }
1143fa9e4066Sahrens 
1144fa9e4066Sahrens /*
1145fa9e4066Sahrens  * Verify that dynamic LUN growth works as expected.
1146fa9e4066Sahrens  */
1147fa9e4066Sahrens void
1148fa9e4066Sahrens ztest_vdev_LUN_growth(ztest_args_t *za)
1149fa9e4066Sahrens {
1150e05725b1Sbonwick 	spa_t *spa = za->za_spa;
1151fa9e4066Sahrens 	char dev_name[MAXPATHLEN];
1152fa9e4066Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
1153fa9e4066Sahrens 	uint64_t vdev;
1154fa9e4066Sahrens 	size_t fsize;
1155fa9e4066Sahrens 	int fd;
1156fa9e4066Sahrens 
1157fa9e4066Sahrens 	(void) mutex_lock(&ztest_shared->zs_vdev_lock);
1158fa9e4066Sahrens 
1159fa9e4066Sahrens 	/*
1160fa9e4066Sahrens 	 * Pick a random leaf vdev.
1161fa9e4066Sahrens 	 */
1162e14bb325SJeff Bonwick 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
1163fa9e4066Sahrens 	vdev = ztest_random(spa->spa_root_vdev->vdev_children * leaves);
1164e14bb325SJeff Bonwick 	spa_config_exit(spa, SCL_VDEV, FTAG);
1165fa9e4066Sahrens 
1166fa9e4066Sahrens 	(void) sprintf(dev_name, ztest_dev_template, zopt_dir, zopt_pool, vdev);
1167fa9e4066Sahrens 
1168fa9e4066Sahrens 	if ((fd = open(dev_name, O_RDWR)) != -1) {
1169fa9e4066Sahrens 		/*
1170fa9e4066Sahrens 		 * Determine the size.
1171fa9e4066Sahrens 		 */
1172fa9e4066Sahrens 		fsize = lseek(fd, 0, SEEK_END);
1173fa9e4066Sahrens 
1174fa9e4066Sahrens 		/*
1175fa9e4066Sahrens 		 * If it's less than 2x the original size, grow by around 3%.
1176fa9e4066Sahrens 		 */
1177fa9e4066Sahrens 		if (fsize < 2 * zopt_vdev_size) {
1178fa9e4066Sahrens 			size_t newsize = fsize + ztest_random(fsize / 32);
1179fa9e4066Sahrens 			(void) ftruncate(fd, newsize);
1180fa9e4066Sahrens 			if (zopt_verbose >= 6) {
1181fa9e4066Sahrens 				(void) printf("%s grew from %lu to %lu bytes\n",
1182fa9e4066Sahrens 				    dev_name, (ulong_t)fsize, (ulong_t)newsize);
1183fa9e4066Sahrens 			}
1184fa9e4066Sahrens 		}
1185fa9e4066Sahrens 		(void) close(fd);
1186fa9e4066Sahrens 	}
1187fa9e4066Sahrens 
1188fa9e4066Sahrens 	(void) mutex_unlock(&ztest_shared->zs_vdev_lock);
1189fa9e4066Sahrens }
1190fa9e4066Sahrens 
1191fa9e4066Sahrens /* ARGSUSED */
1192fa9e4066Sahrens static void
1193ecd6cf80Smarks ztest_create_cb(objset_t *os, void *arg, cred_t *cr, dmu_tx_t *tx)
1194fa9e4066Sahrens {
1195fa9e4066Sahrens 	/*
1196fa9e4066Sahrens 	 * Create the directory object.
1197fa9e4066Sahrens 	 */
1198fa9e4066Sahrens 	VERIFY(dmu_object_claim(os, ZTEST_DIROBJ,
1199fa9e4066Sahrens 	    DMU_OT_UINT64_OTHER, ZTEST_DIROBJ_BLOCKSIZE,
1200e05725b1Sbonwick 	    DMU_OT_UINT64_OTHER, 5 * sizeof (ztest_block_tag_t), tx) == 0);
1201fa9e4066Sahrens 
1202fa9e4066Sahrens 	VERIFY(zap_create_claim(os, ZTEST_MICROZAP_OBJ,
1203fa9e4066Sahrens 	    DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx) == 0);
1204fa9e4066Sahrens 
1205fa9e4066Sahrens 	VERIFY(zap_create_claim(os, ZTEST_FATZAP_OBJ,
1206fa9e4066Sahrens 	    DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx) == 0);
1207fa9e4066Sahrens }
1208fa9e4066Sahrens 
12091d452cf5Sahrens static int
1210fa9e4066Sahrens ztest_destroy_cb(char *name, void *arg)
1211fa9e4066Sahrens {
1212e05725b1Sbonwick 	ztest_args_t *za = arg;
1213fa9e4066Sahrens 	objset_t *os;
1214e05725b1Sbonwick 	dmu_object_info_t *doi = &za->za_doi;
1215fa9e4066Sahrens 	int error;
1216fa9e4066Sahrens 
1217fa9e4066Sahrens 	/*
1218fa9e4066Sahrens 	 * Verify that the dataset contains a directory object.
1219fa9e4066Sahrens 	 */
1220fa9e4066Sahrens 	error = dmu_objset_open(name, DMU_OST_OTHER,
1221745cd3c5Smaybee 	    DS_MODE_USER | DS_MODE_READONLY, &os);
1222fa9e4066Sahrens 	ASSERT3U(error, ==, 0);
1223e05725b1Sbonwick 	error = dmu_object_info(os, ZTEST_DIROBJ, doi);
1224e1930233Sbonwick 	if (error != ENOENT) {
1225e1930233Sbonwick 		/* We could have crashed in the middle of destroying it */
1226e1930233Sbonwick 		ASSERT3U(error, ==, 0);
1227e05725b1Sbonwick 		ASSERT3U(doi->doi_type, ==, DMU_OT_UINT64_OTHER);
1228e05725b1Sbonwick 		ASSERT3S(doi->doi_physical_blks, >=, 0);
1229e1930233Sbonwick 	}
1230fa9e4066Sahrens 	dmu_objset_close(os);
1231fa9e4066Sahrens 
1232fa9e4066Sahrens 	/*
1233fa9e4066Sahrens 	 * Destroy the dataset.
1234fa9e4066Sahrens 	 */
1235fa9e4066Sahrens 	error = dmu_objset_destroy(name);
1236745cd3c5Smaybee 	if (error) {
1237745cd3c5Smaybee 		(void) dmu_objset_open(name, DMU_OST_OTHER,
1238745cd3c5Smaybee 		    DS_MODE_USER | DS_MODE_READONLY, &os);
1239745cd3c5Smaybee 		fatal(0, "dmu_objset_destroy(os=%p) = %d\n", &os, error);
1240745cd3c5Smaybee 	}
12411d452cf5Sahrens 	return (0);
1242fa9e4066Sahrens }
1243fa9e4066Sahrens 
1244fa9e4066Sahrens /*
1245fa9e4066Sahrens  * Verify that dmu_objset_{create,destroy,open,close} work as expected.
1246fa9e4066Sahrens  */
1247fa9e4066Sahrens static uint64_t
1248fa9e4066Sahrens ztest_log_create(zilog_t *zilog, dmu_tx_t *tx, uint64_t object, int mode)
1249fa9e4066Sahrens {
1250fa9e4066Sahrens 	itx_t *itx;
1251fa9e4066Sahrens 	lr_create_t *lr;
1252fa9e4066Sahrens 	size_t namesize;
1253fa9e4066Sahrens 	char name[24];
1254fa9e4066Sahrens 
1255fa9e4066Sahrens 	(void) sprintf(name, "ZOBJ_%llu", (u_longlong_t)object);
1256fa9e4066Sahrens 	namesize = strlen(name) + 1;
1257fa9e4066Sahrens 
1258fa9e4066Sahrens 	itx = zil_itx_create(TX_CREATE, sizeof (*lr) + namesize +
1259fa9e4066Sahrens 	    ztest_random(ZIL_MAX_BLKSZ));
1260fa9e4066Sahrens 	lr = (lr_create_t *)&itx->itx_lr;
1261fa9e4066Sahrens 	bzero(lr + 1, lr->lr_common.lrc_reclen - sizeof (*lr));
1262fa9e4066Sahrens 	lr->lr_doid = object;
1263fa9e4066Sahrens 	lr->lr_foid = 0;
1264fa9e4066Sahrens 	lr->lr_mode = mode;
1265fa9e4066Sahrens 	lr->lr_uid = 0;
1266fa9e4066Sahrens 	lr->lr_gid = 0;
1267fa9e4066Sahrens 	lr->lr_gen = dmu_tx_get_txg(tx);
1268fa9e4066Sahrens 	lr->lr_crtime[0] = time(NULL);
1269fa9e4066Sahrens 	lr->lr_crtime[1] = 0;
1270fa9e4066Sahrens 	lr->lr_rdev = 0;
1271fa9e4066Sahrens 	bcopy(name, (char *)(lr + 1), namesize);
1272fa9e4066Sahrens 
1273fa9e4066Sahrens 	return (zil_itx_assign(zilog, itx, tx));
1274fa9e4066Sahrens }
1275fa9e4066Sahrens 
1276fa9e4066Sahrens void
1277fa9e4066Sahrens ztest_dmu_objset_create_destroy(ztest_args_t *za)
1278fa9e4066Sahrens {
1279fa9e4066Sahrens 	int error;
1280745cd3c5Smaybee 	objset_t *os, *os2;
1281fa9e4066Sahrens 	char name[100];
1282745cd3c5Smaybee 	int basemode, expected_error;
1283fa9e4066Sahrens 	zilog_t *zilog;
1284fa9e4066Sahrens 	uint64_t seq;
1285fa9e4066Sahrens 	uint64_t objects;
1286fa9e4066Sahrens 
1287fa9e4066Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
1288fa9e4066Sahrens 	(void) snprintf(name, 100, "%s/%s_temp_%llu", za->za_pool, za->za_pool,
1289fa9e4066Sahrens 	    (u_longlong_t)za->za_instance);
1290fa9e4066Sahrens 
1291745cd3c5Smaybee 	basemode = DS_MODE_TYPE(za->za_instance);
1292745cd3c5Smaybee 	if (basemode != DS_MODE_USER && basemode != DS_MODE_OWNER)
1293745cd3c5Smaybee 		basemode = DS_MODE_USER;
1294fa9e4066Sahrens 
1295fa9e4066Sahrens 	/*
1296fa9e4066Sahrens 	 * If this dataset exists from a previous run, process its replay log
1297fa9e4066Sahrens 	 * half of the time.  If we don't replay it, then dmu_objset_destroy()
1298fa9e4066Sahrens 	 * (invoked from ztest_destroy_cb() below) should just throw it away.
1299fa9e4066Sahrens 	 */
1300fa9e4066Sahrens 	if (ztest_random(2) == 0 &&
1301745cd3c5Smaybee 	    dmu_objset_open(name, DMU_OST_OTHER, DS_MODE_OWNER, &os) == 0) {
13021209a471SNeil Perrin 		zil_replay(os, os, ztest_replay_vector);
1303fa9e4066Sahrens 		dmu_objset_close(os);
1304fa9e4066Sahrens 	}
1305fa9e4066Sahrens 
1306fa9e4066Sahrens 	/*
1307fa9e4066Sahrens 	 * There may be an old instance of the dataset we're about to
1308fa9e4066Sahrens 	 * create lying around from a previous run.  If so, destroy it
1309fa9e4066Sahrens 	 * and all of its snapshots.
1310fa9e4066Sahrens 	 */
1311e05725b1Sbonwick 	(void) dmu_objset_find(name, ztest_destroy_cb, za,
13120b69c2f0Sahrens 	    DS_FIND_CHILDREN | DS_FIND_SNAPSHOTS);
1313fa9e4066Sahrens 
1314fa9e4066Sahrens 	/*
1315fa9e4066Sahrens 	 * Verify that the destroyed dataset is no longer in the namespace.
1316fa9e4066Sahrens 	 */
1317fa9e4066Sahrens 	error = dmu_objset_open(name, DMU_OST_OTHER, basemode, &os);
1318fa9e4066Sahrens 	if (error != ENOENT)
1319fa9e4066Sahrens 		fatal(1, "dmu_objset_open(%s) found destroyed dataset %p",
1320fa9e4066Sahrens 		    name, os);
1321fa9e4066Sahrens 
1322fa9e4066Sahrens 	/*
1323fa9e4066Sahrens 	 * Verify that we can create a new dataset.
1324fa9e4066Sahrens 	 */
1325ab04eb8eStimh 	error = dmu_objset_create(name, DMU_OST_OTHER, NULL, 0,
1326ab04eb8eStimh 	    ztest_create_cb, NULL);
1327fa9e4066Sahrens 	if (error) {
1328fa9e4066Sahrens 		if (error == ENOSPC) {
1329fa9e4066Sahrens 			ztest_record_enospc("dmu_objset_create");
1330fa9e4066Sahrens 			(void) rw_unlock(&ztest_shared->zs_name_lock);
1331fa9e4066Sahrens 			return;
1332fa9e4066Sahrens 		}
1333fa9e4066Sahrens 		fatal(0, "dmu_objset_create(%s) = %d", name, error);
1334fa9e4066Sahrens 	}
1335fa9e4066Sahrens 
1336fa9e4066Sahrens 	error = dmu_objset_open(name, DMU_OST_OTHER, basemode, &os);
1337fa9e4066Sahrens 	if (error) {
1338fa9e4066Sahrens 		fatal(0, "dmu_objset_open(%s) = %d", name, error);
1339fa9e4066Sahrens 	}
1340fa9e4066Sahrens 
1341fa9e4066Sahrens 	/*
1342fa9e4066Sahrens 	 * Open the intent log for it.
1343fa9e4066Sahrens 	 */
1344fa9e4066Sahrens 	zilog = zil_open(os, NULL);
1345fa9e4066Sahrens 
1346fa9e4066Sahrens 	/*
1347fa9e4066Sahrens 	 * Put a random number of objects in there.
1348fa9e4066Sahrens 	 */
1349d80c45e0Sbonwick 	objects = ztest_random(20);
1350fa9e4066Sahrens 	seq = 0;
1351fa9e4066Sahrens 	while (objects-- != 0) {
1352fa9e4066Sahrens 		uint64_t object;
1353fa9e4066Sahrens 		dmu_tx_t *tx = dmu_tx_create(os);
1354fa9e4066Sahrens 		dmu_tx_hold_write(tx, DMU_NEW_OBJECT, 0, sizeof (name));
1355fa9e4066Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1356fa9e4066Sahrens 		if (error) {
1357fa9e4066Sahrens 			dmu_tx_abort(tx);
1358fa9e4066Sahrens 		} else {
1359fa9e4066Sahrens 			object = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1360fa9e4066Sahrens 			    DMU_OT_NONE, 0, tx);
1361fa9e4066Sahrens 			ztest_set_random_blocksize(os, object, tx);
1362fa9e4066Sahrens 			seq = ztest_log_create(zilog, tx, object,
1363fa9e4066Sahrens 			    DMU_OT_UINT64_OTHER);
1364fa9e4066Sahrens 			dmu_write(os, object, 0, sizeof (name), name, tx);
1365fa9e4066Sahrens 			dmu_tx_commit(tx);
1366fa9e4066Sahrens 		}
1367fa9e4066Sahrens 		if (ztest_random(5) == 0) {
1368b19a79ecSperrin 			zil_commit(zilog, seq, object);
1369fa9e4066Sahrens 		}
1370d80c45e0Sbonwick 		if (ztest_random(100) == 0) {
1371fa9e4066Sahrens 			error = zil_suspend(zilog);
1372fa9e4066Sahrens 			if (error == 0) {
1373fa9e4066Sahrens 				zil_resume(zilog);
1374fa9e4066Sahrens 			}
1375fa9e4066Sahrens 		}
1376fa9e4066Sahrens 	}
1377fa9e4066Sahrens 
1378fa9e4066Sahrens 	/*
1379fa9e4066Sahrens 	 * Verify that we cannot create an existing dataset.
1380fa9e4066Sahrens 	 */
1381ab04eb8eStimh 	error = dmu_objset_create(name, DMU_OST_OTHER, NULL, 0, NULL, NULL);
1382fa9e4066Sahrens 	if (error != EEXIST)
1383fa9e4066Sahrens 		fatal(0, "created existing dataset, error = %d", error);
1384fa9e4066Sahrens 
1385fa9e4066Sahrens 	/*
1386745cd3c5Smaybee 	 * Verify that multiple dataset holds are allowed, but only when
1387fa9e4066Sahrens 	 * the new access mode is compatible with the base mode.
1388745cd3c5Smaybee 	 */
1389745cd3c5Smaybee 	if (basemode == DS_MODE_OWNER) {
1390745cd3c5Smaybee 		error = dmu_objset_open(name, DMU_OST_OTHER, DS_MODE_USER,
1391745cd3c5Smaybee 		    &os2);
1392745cd3c5Smaybee 		if (error)
1393745cd3c5Smaybee 			fatal(0, "dmu_objset_open('%s') = %d", name, error);
1394745cd3c5Smaybee 		else
1395fa9e4066Sahrens 			dmu_objset_close(os2);
1396fa9e4066Sahrens 	}
1397745cd3c5Smaybee 	error = dmu_objset_open(name, DMU_OST_OTHER, DS_MODE_OWNER, &os2);
1398745cd3c5Smaybee 	expected_error = (basemode == DS_MODE_OWNER) ? EBUSY : 0;
1399745cd3c5Smaybee 	if (error != expected_error)
1400745cd3c5Smaybee 		fatal(0, "dmu_objset_open('%s') = %d, expected %d",
1401745cd3c5Smaybee 		    name, error, expected_error);
1402745cd3c5Smaybee 	if (error == 0)
1403745cd3c5Smaybee 		dmu_objset_close(os2);
1404fa9e4066Sahrens 
1405fa9e4066Sahrens 	zil_close(zilog);
1406fa9e4066Sahrens 	dmu_objset_close(os);
1407fa9e4066Sahrens 
1408fa9e4066Sahrens 	error = dmu_objset_destroy(name);
1409fa9e4066Sahrens 	if (error)
1410fa9e4066Sahrens 		fatal(0, "dmu_objset_destroy(%s) = %d", name, error);
1411fa9e4066Sahrens 
1412fa9e4066Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
1413fa9e4066Sahrens }
1414fa9e4066Sahrens 
1415fa9e4066Sahrens /*
1416fa9e4066Sahrens  * Verify that dmu_snapshot_{create,destroy,open,close} work as expected.
1417fa9e4066Sahrens  */
1418fa9e4066Sahrens void
1419fa9e4066Sahrens ztest_dmu_snapshot_create_destroy(ztest_args_t *za)
1420fa9e4066Sahrens {
1421fa9e4066Sahrens 	int error;
1422fa9e4066Sahrens 	objset_t *os = za->za_os;
1423fa9e4066Sahrens 	char snapname[100];
1424fa9e4066Sahrens 	char osname[MAXNAMELEN];
1425fa9e4066Sahrens 
1426fa9e4066Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
1427fa9e4066Sahrens 	dmu_objset_name(os, osname);
1428fa9e4066Sahrens 	(void) snprintf(snapname, 100, "%s@%llu", osname,
1429fa9e4066Sahrens 	    (u_longlong_t)za->za_instance);
1430fa9e4066Sahrens 
1431fa9e4066Sahrens 	error = dmu_objset_destroy(snapname);
1432fa9e4066Sahrens 	if (error != 0 && error != ENOENT)
1433fa9e4066Sahrens 		fatal(0, "dmu_objset_destroy() = %d", error);
1434ea2f5b9eSMatthew Ahrens 	error = dmu_objset_snapshot(osname, strchr(snapname, '@')+1,
1435ea2f5b9eSMatthew Ahrens 	    NULL, FALSE);
1436fa9e4066Sahrens 	if (error == ENOSPC)
1437fa9e4066Sahrens 		ztest_record_enospc("dmu_take_snapshot");
1438fa9e4066Sahrens 	else if (error != 0 && error != EEXIST)
1439fa9e4066Sahrens 		fatal(0, "dmu_take_snapshot() = %d", error);
1440fa9e4066Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
14414f5064b7SMark J Musante }
14424f5064b7SMark J Musante 
14434f5064b7SMark J Musante /*
14444f5064b7SMark J Musante  * Verify dsl_dataset_promote handles EBUSY
14454f5064b7SMark J Musante  */
14464f5064b7SMark J Musante void
14474f5064b7SMark J Musante ztest_dsl_dataset_promote_busy(ztest_args_t *za)
14484f5064b7SMark J Musante {
14494f5064b7SMark J Musante 	int error;
14504f5064b7SMark J Musante 	objset_t *os = za->za_os;
14514f5064b7SMark J Musante 	objset_t *clone;
14524f5064b7SMark J Musante 	dsl_dataset_t *ds;
14534f5064b7SMark J Musante 	char snap1name[100];
14544f5064b7SMark J Musante 	char clone1name[100];
14554f5064b7SMark J Musante 	char snap2name[100];
14564f5064b7SMark J Musante 	char clone2name[100];
14574f5064b7SMark J Musante 	char snap3name[100];
14584f5064b7SMark J Musante 	char osname[MAXNAMELEN];
14594f5064b7SMark J Musante 	uint64_t curval;
14604f5064b7SMark J Musante 
146105312e2cSMark J Musante 	curval = za->za_instance;
14624f5064b7SMark J Musante 
14634f5064b7SMark J Musante 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
14644f5064b7SMark J Musante 
14654f5064b7SMark J Musante 	dmu_objset_name(os, osname);
146605312e2cSMark J Musante 	(void) snprintf(snap1name, 100, "%s@s1_%llu", osname, curval);
146705312e2cSMark J Musante 	(void) snprintf(clone1name, 100, "%s/c1_%llu", osname, curval);
146805312e2cSMark J Musante 	(void) snprintf(snap2name, 100, "%s@s2_%llu", clone1name, curval);
146905312e2cSMark J Musante 	(void) snprintf(clone2name, 100, "%s/c2_%llu", osname, curval);
147005312e2cSMark J Musante 	(void) snprintf(snap3name, 100, "%s@s3_%llu", clone1name, curval);
147105312e2cSMark J Musante 
147205312e2cSMark J Musante 	error = dmu_objset_destroy(clone2name);
147305312e2cSMark J Musante 	if (error != 0 && error != ENOENT)
147405312e2cSMark J Musante 		fatal(0, "dmu_objset_destroy() = %d", error);
147505312e2cSMark J Musante 	error = dmu_objset_destroy(snap3name);
147605312e2cSMark J Musante 	if (error != 0 && error != ENOENT)
147705312e2cSMark J Musante 		fatal(0, "dmu_objset_destroy() = %d", error);
147805312e2cSMark J Musante 	error = dmu_objset_destroy(snap2name);
147905312e2cSMark J Musante 	if (error != 0 && error != ENOENT)
148005312e2cSMark J Musante 		fatal(0, "dmu_objset_destroy() = %d", error);
148105312e2cSMark J Musante 	error = dmu_objset_destroy(clone1name);
148205312e2cSMark J Musante 	if (error != 0 && error != ENOENT)
148305312e2cSMark J Musante 		fatal(0, "dmu_objset_destroy() = %d", error);
148405312e2cSMark J Musante 	error = dmu_objset_destroy(snap1name);
148505312e2cSMark J Musante 	if (error != 0 && error != ENOENT)
148605312e2cSMark J Musante 		fatal(0, "dmu_objset_destroy() = %d", error);
14874f5064b7SMark J Musante 
1488ea2f5b9eSMatthew Ahrens 	error = dmu_objset_snapshot(osname, strchr(snap1name, '@')+1,
1489ea2f5b9eSMatthew Ahrens 	    NULL, FALSE);
1490b2d22b66SGeorge Wilson 	if (error && error != EEXIST) {
1491b2d22b66SGeorge Wilson 		if (error == ENOSPC) {
1492b2d22b66SGeorge Wilson 			ztest_record_enospc("dmu_take_snapshot");
1493b2d22b66SGeorge Wilson 			goto out;
1494b2d22b66SGeorge Wilson 		}
1495b2d22b66SGeorge Wilson 		fatal(0, "dmu_take_snapshot(%s) = %d", snap1name, error);
1496b2d22b66SGeorge Wilson 	}
14974f5064b7SMark J Musante 
14984f5064b7SMark J Musante 	error = dmu_objset_open(snap1name, DMU_OST_OTHER,
14994f5064b7SMark J Musante 	    DS_MODE_USER | DS_MODE_READONLY, &clone);
15004f5064b7SMark J Musante 	if (error)
15014f5064b7SMark J Musante 		fatal(0, "dmu_open_snapshot(%s) = %d", snap1name, error);
15024f5064b7SMark J Musante 
15034f5064b7SMark J Musante 	error = dmu_objset_create(clone1name, DMU_OST_OTHER, clone, 0,
15044f5064b7SMark J Musante 	    NULL, NULL);
15054f5064b7SMark J Musante 	dmu_objset_close(clone);
1506b2d22b66SGeorge Wilson 	if (error) {
1507b2d22b66SGeorge Wilson 		if (error == ENOSPC) {
1508b2d22b66SGeorge Wilson 			ztest_record_enospc("dmu_objset_create");
1509b2d22b66SGeorge Wilson 			goto out;
1510b2d22b66SGeorge Wilson 		}
1511b2d22b66SGeorge Wilson 		fatal(0, "dmu_objset_create(%s) = %d", clone1name, error);
1512b2d22b66SGeorge Wilson 	}
15134f5064b7SMark J Musante 
15144f5064b7SMark J Musante 	error = dmu_objset_snapshot(clone1name, strchr(snap2name, '@')+1,
1515ea2f5b9eSMatthew Ahrens 	    NULL, FALSE);
1516b2d22b66SGeorge Wilson 	if (error && error != EEXIST) {
1517b2d22b66SGeorge Wilson 		if (error == ENOSPC) {
1518b2d22b66SGeorge Wilson 			ztest_record_enospc("dmu_take_snapshot");
1519b2d22b66SGeorge Wilson 			goto out;
1520b2d22b66SGeorge Wilson 		}
1521b2d22b66SGeorge Wilson 		fatal(0, "dmu_open_snapshot(%s) = %d", snap2name, error);
1522b2d22b66SGeorge Wilson 	}
15234f5064b7SMark J Musante 
15244f5064b7SMark J Musante 	error = dmu_objset_snapshot(clone1name, strchr(snap3name, '@')+1,
1525ea2f5b9eSMatthew Ahrens 	    NULL, FALSE);
1526b2d22b66SGeorge Wilson 	if (error && error != EEXIST) {
1527b2d22b66SGeorge Wilson 		if (error == ENOSPC) {
1528b2d22b66SGeorge Wilson 			ztest_record_enospc("dmu_take_snapshot");
1529b2d22b66SGeorge Wilson 			goto out;
1530b2d22b66SGeorge Wilson 		}
1531b2d22b66SGeorge Wilson 		fatal(0, "dmu_open_snapshot(%s) = %d", snap3name, error);
1532b2d22b66SGeorge Wilson 	}
15334f5064b7SMark J Musante 
15344f5064b7SMark J Musante 	error = dmu_objset_open(snap3name, DMU_OST_OTHER,
15354f5064b7SMark J Musante 	    DS_MODE_USER | DS_MODE_READONLY, &clone);
15364f5064b7SMark J Musante 	if (error)
15374f5064b7SMark J Musante 		fatal(0, "dmu_open_snapshot(%s) = %d", snap3name, error);
15384f5064b7SMark J Musante 
15394f5064b7SMark J Musante 	error = dmu_objset_create(clone2name, DMU_OST_OTHER, clone, 0,
15404f5064b7SMark J Musante 	    NULL, NULL);
15414f5064b7SMark J Musante 	dmu_objset_close(clone);
1542b2d22b66SGeorge Wilson 	if (error) {
1543b2d22b66SGeorge Wilson 		if (error == ENOSPC) {
1544b2d22b66SGeorge Wilson 			ztest_record_enospc("dmu_objset_create");
1545b2d22b66SGeorge Wilson 			goto out;
1546b2d22b66SGeorge Wilson 		}
1547b2d22b66SGeorge Wilson 		fatal(0, "dmu_objset_create(%s) = %d", clone2name, error);
1548b2d22b66SGeorge Wilson 	}
15494f5064b7SMark J Musante 
155005312e2cSMark J Musante 	error = dsl_dataset_own(snap1name, DS_MODE_READONLY, FTAG, &ds);
15514f5064b7SMark J Musante 	if (error)
15524f5064b7SMark J Musante 		fatal(0, "dsl_dataset_own(%s) = %d", snap1name, error);
15534f5064b7SMark J Musante 	error = dsl_dataset_promote(clone2name);
15544f5064b7SMark J Musante 	if (error != EBUSY)
15554f5064b7SMark J Musante 		fatal(0, "dsl_dataset_promote(%s), %d, not EBUSY", clone2name,
15564f5064b7SMark J Musante 		    error);
15574f5064b7SMark J Musante 	dsl_dataset_disown(ds, FTAG);
15584f5064b7SMark J Musante 
1559b2d22b66SGeorge Wilson out:
15604f5064b7SMark J Musante 	error = dmu_objset_destroy(clone2name);
1561b2d22b66SGeorge Wilson 	if (error && error != ENOENT)
15624f5064b7SMark J Musante 		fatal(0, "dmu_objset_destroy(%s) = %d", clone2name, error);
15634f5064b7SMark J Musante 	error = dmu_objset_destroy(snap3name);
1564b2d22b66SGeorge Wilson 	if (error && error != ENOENT)
156505312e2cSMark J Musante 		fatal(0, "dmu_objset_destroy(%s) = %d", snap3name, error);
15664f5064b7SMark J Musante 	error = dmu_objset_destroy(snap2name);
1567b2d22b66SGeorge Wilson 	if (error && error != ENOENT)
15684f5064b7SMark J Musante 		fatal(0, "dmu_objset_destroy(%s) = %d", snap2name, error);
15694f5064b7SMark J Musante 	error = dmu_objset_destroy(clone1name);
1570b2d22b66SGeorge Wilson 	if (error && error != ENOENT)
15714f5064b7SMark J Musante 		fatal(0, "dmu_objset_destroy(%s) = %d", clone1name, error);
15724f5064b7SMark J Musante 	error = dmu_objset_destroy(snap1name);
1573b2d22b66SGeorge Wilson 	if (error && error != ENOENT)
15744f5064b7SMark J Musante 		fatal(0, "dmu_objset_destroy(%s) = %d", snap1name, error);
15754f5064b7SMark J Musante 
15764f5064b7SMark J Musante 	(void) rw_unlock(&ztest_shared->zs_name_lock);
1577fa9e4066Sahrens }
1578fa9e4066Sahrens 
1579fa9e4066Sahrens /*
1580fa9e4066Sahrens  * Verify that dmu_object_{alloc,free} work as expected.
1581fa9e4066Sahrens  */
1582fa9e4066Sahrens void
1583fa9e4066Sahrens ztest_dmu_object_alloc_free(ztest_args_t *za)
1584fa9e4066Sahrens {
1585fa9e4066Sahrens 	objset_t *os = za->za_os;
1586fa9e4066Sahrens 	dmu_buf_t *db;
1587fa9e4066Sahrens 	dmu_tx_t *tx;
1588fa9e4066Sahrens 	uint64_t batchobj, object, batchsize, endoff, temp;
1589fa9e4066Sahrens 	int b, c, error, bonuslen;
1590e05725b1Sbonwick 	dmu_object_info_t *doi = &za->za_doi;
1591fa9e4066Sahrens 	char osname[MAXNAMELEN];
1592fa9e4066Sahrens 
1593fa9e4066Sahrens 	dmu_objset_name(os, osname);
1594fa9e4066Sahrens 
1595fa9e4066Sahrens 	endoff = -8ULL;
1596fa9e4066Sahrens 	batchsize = 2;
1597fa9e4066Sahrens 
1598fa9e4066Sahrens 	/*
1599fa9e4066Sahrens 	 * Create a batch object if necessary, and record it in the directory.
1600fa9e4066Sahrens 	 */
1601e14bb325SJeff Bonwick 	VERIFY3U(0, ==, dmu_read(os, ZTEST_DIROBJ, za->za_diroff,
1602ea8dc4b6Seschrock 	    sizeof (uint64_t), &batchobj));
1603fa9e4066Sahrens 	if (batchobj == 0) {
1604fa9e4066Sahrens 		tx = dmu_tx_create(os);
1605fa9e4066Sahrens 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff,
1606fa9e4066Sahrens 		    sizeof (uint64_t));
1607fa9e4066Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1608fa9e4066Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1609fa9e4066Sahrens 		if (error) {
1610fa9e4066Sahrens 			ztest_record_enospc("create a batch object");
1611fa9e4066Sahrens 			dmu_tx_abort(tx);
1612fa9e4066Sahrens 			return;
1613fa9e4066Sahrens 		}
1614fa9e4066Sahrens 		batchobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1615fa9e4066Sahrens 		    DMU_OT_NONE, 0, tx);
1616fa9e4066Sahrens 		ztest_set_random_blocksize(os, batchobj, tx);
1617fa9e4066Sahrens 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff,
1618fa9e4066Sahrens 		    sizeof (uint64_t), &batchobj, tx);
1619fa9e4066Sahrens 		dmu_tx_commit(tx);
1620fa9e4066Sahrens 	}
1621fa9e4066Sahrens 
1622fa9e4066Sahrens 	/*
1623fa9e4066Sahrens 	 * Destroy the previous batch of objects.
1624fa9e4066Sahrens 	 */
1625fa9e4066Sahrens 	for (b = 0; b < batchsize; b++) {
1626088f3894Sahrens 		VERIFY3U(0, ==, dmu_read(os, batchobj, b * sizeof (uint64_t),
1627ea8dc4b6Seschrock 		    sizeof (uint64_t), &object));
1628fa9e4066Sahrens 		if (object == 0)
1629fa9e4066Sahrens 			continue;
1630fa9e4066Sahrens 		/*
1631fa9e4066Sahrens 		 * Read and validate contents.
1632fa9e4066Sahrens 		 * We expect the nth byte of the bonus buffer to be n.
1633fa9e4066Sahrens 		 */
1634ea8dc4b6Seschrock 		VERIFY(0 == dmu_bonus_hold(os, object, FTAG, &db));
1635e05725b1Sbonwick 		za->za_dbuf = db;
1636fa9e4066Sahrens 
1637e05725b1Sbonwick 		dmu_object_info_from_db(db, doi);
1638e05725b1Sbonwick 		ASSERT(doi->doi_type == DMU_OT_UINT64_OTHER);
1639e05725b1Sbonwick 		ASSERT(doi->doi_bonus_type == DMU_OT_PLAIN_OTHER);
1640e05725b1Sbonwick 		ASSERT3S(doi->doi_physical_blks, >=, 0);
1641fa9e4066Sahrens 
1642e05725b1Sbonwick 		bonuslen = doi->doi_bonus_size;
1643fa9e4066Sahrens 
1644fa9e4066Sahrens 		for (c = 0; c < bonuslen; c++) {
1645fa9e4066Sahrens 			if (((uint8_t *)db->db_data)[c] !=
1646fa9e4066Sahrens 			    (uint8_t)(c + bonuslen)) {
1647fa9e4066Sahrens 				fatal(0,
1648fa9e4066Sahrens 				    "bad bonus: %s, obj %llu, off %d: %u != %u",
1649fa9e4066Sahrens 				    osname, object, c,
1650fa9e4066Sahrens 				    ((uint8_t *)db->db_data)[c],
1651fa9e4066Sahrens 				    (uint8_t)(c + bonuslen));
1652fa9e4066Sahrens 			}
1653fa9e4066Sahrens 		}
1654fa9e4066Sahrens 
1655ea8dc4b6Seschrock 		dmu_buf_rele(db, FTAG);
1656e05725b1Sbonwick 		za->za_dbuf = NULL;
1657fa9e4066Sahrens 
1658fa9e4066Sahrens 		/*
1659fa9e4066Sahrens 		 * We expect the word at endoff to be our object number.
1660fa9e4066Sahrens 		 */
1661ea8dc4b6Seschrock 		VERIFY(0 == dmu_read(os, object, endoff,
1662ea8dc4b6Seschrock 		    sizeof (uint64_t), &temp));
1663fa9e4066Sahrens 
1664fa9e4066Sahrens 		if (temp != object) {
1665fa9e4066Sahrens 			fatal(0, "bad data in %s, got %llu, expected %llu",
1666fa9e4066Sahrens 			    osname, temp, object);
1667fa9e4066Sahrens 		}
1668fa9e4066Sahrens 
1669fa9e4066Sahrens 		/*
1670fa9e4066Sahrens 		 * Destroy old object and clear batch entry.
1671fa9e4066Sahrens 		 */
1672fa9e4066Sahrens 		tx = dmu_tx_create(os);
1673fa9e4066Sahrens 		dmu_tx_hold_write(tx, batchobj,
1674fa9e4066Sahrens 		    b * sizeof (uint64_t), sizeof (uint64_t));
1675fa9e4066Sahrens 		dmu_tx_hold_free(tx, object, 0, DMU_OBJECT_END);
1676fa9e4066Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1677fa9e4066Sahrens 		if (error) {
1678fa9e4066Sahrens 			ztest_record_enospc("free object");
1679fa9e4066Sahrens 			dmu_tx_abort(tx);
1680fa9e4066Sahrens 			return;
1681fa9e4066Sahrens 		}
1682fa9e4066Sahrens 		error = dmu_object_free(os, object, tx);
1683fa9e4066Sahrens 		if (error) {
1684fa9e4066Sahrens 			fatal(0, "dmu_object_free('%s', %llu) = %d",
1685fa9e4066Sahrens 			    osname, object, error);
1686fa9e4066Sahrens 		}
1687fa9e4066Sahrens 		object = 0;
1688fa9e4066Sahrens 
1689fa9e4066Sahrens 		dmu_object_set_checksum(os, batchobj,
1690fa9e4066Sahrens 		    ztest_random_checksum(), tx);
1691fa9e4066Sahrens 		dmu_object_set_compress(os, batchobj,
1692fa9e4066Sahrens 		    ztest_random_compress(), tx);
1693fa9e4066Sahrens 
1694fa9e4066Sahrens 		dmu_write(os, batchobj, b * sizeof (uint64_t),
1695fa9e4066Sahrens 		    sizeof (uint64_t), &object, tx);
1696fa9e4066Sahrens 
1697fa9e4066Sahrens 		dmu_tx_commit(tx);
1698fa9e4066Sahrens 	}
1699fa9e4066Sahrens 
1700fa9e4066Sahrens 	/*
1701fa9e4066Sahrens 	 * Before creating the new batch of objects, generate a bunch of churn.
1702fa9e4066Sahrens 	 */
1703fa9e4066Sahrens 	for (b = ztest_random(100); b > 0; b--) {
1704fa9e4066Sahrens 		tx = dmu_tx_create(os);
1705fa9e4066Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1706fa9e4066Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1707fa9e4066Sahrens 		if (error) {
1708fa9e4066Sahrens 			ztest_record_enospc("churn objects");
1709fa9e4066Sahrens 			dmu_tx_abort(tx);
1710fa9e4066Sahrens 			return;
1711fa9e4066Sahrens 		}
1712fa9e4066Sahrens 		object = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1713fa9e4066Sahrens 		    DMU_OT_NONE, 0, tx);
1714fa9e4066Sahrens 		ztest_set_random_blocksize(os, object, tx);
1715fa9e4066Sahrens 		error = dmu_object_free(os, object, tx);
1716fa9e4066Sahrens 		if (error) {
1717fa9e4066Sahrens 			fatal(0, "dmu_object_free('%s', %llu) = %d",
1718fa9e4066Sahrens 			    osname, object, error);
1719fa9e4066Sahrens 		}
1720fa9e4066Sahrens 		dmu_tx_commit(tx);
1721fa9e4066Sahrens 	}
1722fa9e4066Sahrens 
1723fa9e4066Sahrens 	/*
1724fa9e4066Sahrens 	 * Create a new batch of objects with randomly chosen
1725fa9e4066Sahrens 	 * blocksizes and record them in the batch directory.
1726fa9e4066Sahrens 	 */
1727fa9e4066Sahrens 	for (b = 0; b < batchsize; b++) {
1728fa9e4066Sahrens 		uint32_t va_blksize;
1729fa9e4066Sahrens 		u_longlong_t va_nblocks;
1730fa9e4066Sahrens 
1731fa9e4066Sahrens 		tx = dmu_tx_create(os);
1732fa9e4066Sahrens 		dmu_tx_hold_write(tx, batchobj, b * sizeof (uint64_t),
1733fa9e4066Sahrens 		    sizeof (uint64_t));
1734fa9e4066Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1735fa9e4066Sahrens 		dmu_tx_hold_write(tx, DMU_NEW_OBJECT, endoff,
1736fa9e4066Sahrens 		    sizeof (uint64_t));
1737fa9e4066Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1738fa9e4066Sahrens 		if (error) {
1739fa9e4066Sahrens 			ztest_record_enospc("create batchobj");
1740fa9e4066Sahrens 			dmu_tx_abort(tx);
1741fa9e4066Sahrens 			return;
1742fa9e4066Sahrens 		}
1743fa9e4066Sahrens 		bonuslen = (int)ztest_random(dmu_bonus_max()) + 1;
1744fa9e4066Sahrens 
1745fa9e4066Sahrens 		object = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1746fa9e4066Sahrens 		    DMU_OT_PLAIN_OTHER, bonuslen, tx);
1747fa9e4066Sahrens 
1748fa9e4066Sahrens 		ztest_set_random_blocksize(os, object, tx);
1749fa9e4066Sahrens 
1750fa9e4066Sahrens 		dmu_object_set_checksum(os, object,
1751fa9e4066Sahrens 		    ztest_random_checksum(), tx);
1752fa9e4066Sahrens 		dmu_object_set_compress(os, object,
1753fa9e4066Sahrens 		    ztest_random_compress(), tx);
1754fa9e4066Sahrens 
1755fa9e4066Sahrens 		dmu_write(os, batchobj, b * sizeof (uint64_t),
1756fa9e4066Sahrens 		    sizeof (uint64_t), &object, tx);
1757fa9e4066Sahrens 
1758fa9e4066Sahrens 		/*
1759fa9e4066Sahrens 		 * Write to both the bonus buffer and the regular data.
1760fa9e4066Sahrens 		 */
1761e05725b1Sbonwick 		VERIFY(dmu_bonus_hold(os, object, FTAG, &db) == 0);
1762e05725b1Sbonwick 		za->za_dbuf = db;
17631934e92fSmaybee 		ASSERT3U(bonuslen, <=, db->db_size);
1764fa9e4066Sahrens 
1765fa9e4066Sahrens 		dmu_object_size_from_db(db, &va_blksize, &va_nblocks);
1766fa9e4066Sahrens 		ASSERT3S(va_nblocks, >=, 0);
1767fa9e4066Sahrens 
1768fa9e4066Sahrens 		dmu_buf_will_dirty(db, tx);
1769fa9e4066Sahrens 
1770fa9e4066Sahrens 		/*
1771fa9e4066Sahrens 		 * See comments above regarding the contents of
1772fa9e4066Sahrens 		 * the bonus buffer and the word at endoff.
1773fa9e4066Sahrens 		 */
17741934e92fSmaybee 		for (c = 0; c < bonuslen; c++)
1775fa9e4066Sahrens 			((uint8_t *)db->db_data)[c] = (uint8_t)(c + bonuslen);
1776fa9e4066Sahrens 
1777ea8dc4b6Seschrock 		dmu_buf_rele(db, FTAG);
1778e05725b1Sbonwick 		za->za_dbuf = NULL;
1779fa9e4066Sahrens 
1780fa9e4066Sahrens 		/*
1781fa9e4066Sahrens 		 * Write to a large offset to increase indirection.
1782fa9e4066Sahrens 		 */
1783fa9e4066Sahrens 		dmu_write(os, object, endoff, sizeof (uint64_t), &object, tx);
1784fa9e4066Sahrens 
1785fa9e4066Sahrens 		dmu_tx_commit(tx);
1786fa9e4066Sahrens 	}
1787fa9e4066Sahrens }
1788fa9e4066Sahrens 
1789fa9e4066Sahrens /*
1790fa9e4066Sahrens  * Verify that dmu_{read,write} work as expected.
1791fa9e4066Sahrens  */
1792fa9e4066Sahrens typedef struct bufwad {
1793fa9e4066Sahrens 	uint64_t	bw_index;
1794fa9e4066Sahrens 	uint64_t	bw_txg;
1795fa9e4066Sahrens 	uint64_t	bw_data;
1796fa9e4066Sahrens } bufwad_t;
1797fa9e4066Sahrens 
1798fa9e4066Sahrens typedef struct dmu_read_write_dir {
1799fa9e4066Sahrens 	uint64_t	dd_packobj;
1800fa9e4066Sahrens 	uint64_t	dd_bigobj;
1801fa9e4066Sahrens 	uint64_t	dd_chunk;
1802fa9e4066Sahrens } dmu_read_write_dir_t;
1803fa9e4066Sahrens 
1804fa9e4066Sahrens void
1805fa9e4066Sahrens ztest_dmu_read_write(ztest_args_t *za)
1806fa9e4066Sahrens {
1807fa9e4066Sahrens 	objset_t *os = za->za_os;
1808fa9e4066Sahrens 	dmu_read_write_dir_t dd;
1809fa9e4066Sahrens 	dmu_tx_t *tx;
1810fa9e4066Sahrens 	int i, freeit, error;
1811fa9e4066Sahrens 	uint64_t n, s, txg;
1812fa9e4066Sahrens 	bufwad_t *packbuf, *bigbuf, *pack, *bigH, *bigT;
1813fa9e4066Sahrens 	uint64_t packoff, packsize, bigoff, bigsize;
1814fa9e4066Sahrens 	uint64_t regions = 997;
1815fa9e4066Sahrens 	uint64_t stride = 123456789ULL;
1816fa9e4066Sahrens 	uint64_t width = 40;
1817fa9e4066Sahrens 	int free_percent = 5;
1818fa9e4066Sahrens 
1819fa9e4066Sahrens 	/*
1820fa9e4066Sahrens 	 * This test uses two objects, packobj and bigobj, that are always
1821fa9e4066Sahrens 	 * updated together (i.e. in the same tx) so that their contents are
1822fa9e4066Sahrens 	 * in sync and can be compared.  Their contents relate to each other
1823fa9e4066Sahrens 	 * in a simple way: packobj is a dense array of 'bufwad' structures,
1824fa9e4066Sahrens 	 * while bigobj is a sparse array of the same bufwads.  Specifically,
1825fa9e4066Sahrens 	 * for any index n, there are three bufwads that should be identical:
1826fa9e4066Sahrens 	 *
1827fa9e4066Sahrens 	 *	packobj, at offset n * sizeof (bufwad_t)
1828fa9e4066Sahrens 	 *	bigobj, at the head of the nth chunk
1829fa9e4066Sahrens 	 *	bigobj, at the tail of the nth chunk
1830fa9e4066Sahrens 	 *
1831fa9e4066Sahrens 	 * The chunk size is arbitrary. It doesn't have to be a power of two,
1832fa9e4066Sahrens 	 * and it doesn't have any relation to the object blocksize.
1833fa9e4066Sahrens 	 * The only requirement is that it can hold at least two bufwads.
1834fa9e4066Sahrens 	 *
1835fa9e4066Sahrens 	 * Normally, we write the bufwad to each of these locations.
1836fa9e4066Sahrens 	 * However, free_percent of the time we instead write zeroes to
1837fa9e4066Sahrens 	 * packobj and perform a dmu_free_range() on bigobj.  By comparing
1838fa9e4066Sahrens 	 * bigobj to packobj, we can verify that the DMU is correctly
1839fa9e4066Sahrens 	 * tracking which parts of an object are allocated and free,
1840fa9e4066Sahrens 	 * and that the contents of the allocated blocks are correct.
1841fa9e4066Sahrens 	 */
1842fa9e4066Sahrens 
1843fa9e4066Sahrens 	/*
1844fa9e4066Sahrens 	 * Read the directory info.  If it's the first time, set things up.
1845fa9e4066Sahrens 	 */
1846ea8dc4b6Seschrock 	VERIFY(0 == dmu_read(os, ZTEST_DIROBJ, za->za_diroff,
1847ea8dc4b6Seschrock 	    sizeof (dd), &dd));
1848fa9e4066Sahrens 	if (dd.dd_chunk == 0) {
1849fa9e4066Sahrens 		ASSERT(dd.dd_packobj == 0);
1850fa9e4066Sahrens 		ASSERT(dd.dd_bigobj == 0);
1851fa9e4066Sahrens 		tx = dmu_tx_create(os);
1852fa9e4066Sahrens 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff, sizeof (dd));
1853fa9e4066Sahrens 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1854fa9e4066Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
1855fa9e4066Sahrens 		if (error) {
1856fa9e4066Sahrens 			ztest_record_enospc("create r/w directory");
1857fa9e4066Sahrens 			dmu_tx_abort(tx);
1858fa9e4066Sahrens 			return;
1859fa9e4066Sahrens 		}
1860fa9e4066Sahrens 
1861fa9e4066Sahrens 		dd.dd_packobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1862fa9e4066Sahrens 		    DMU_OT_NONE, 0, tx);
1863fa9e4066Sahrens 		dd.dd_bigobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
1864fa9e4066Sahrens 		    DMU_OT_NONE, 0, tx);
1865fa9e4066Sahrens 		dd.dd_chunk = (1000 + ztest_random(1000)) * sizeof (uint64_t);
1866fa9e4066Sahrens 
1867fa9e4066Sahrens 		ztest_set_random_blocksize(os, dd.dd_packobj, tx);
1868fa9e4066Sahrens 		ztest_set_random_blocksize(os, dd.dd_bigobj, tx);
1869fa9e4066Sahrens 
1870fa9e4066Sahrens 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff, sizeof (dd), &dd,
1871fa9e4066Sahrens 		    tx);
1872fa9e4066Sahrens 		dmu_tx_commit(tx);
1873fa9e4066Sahrens 	}
1874fa9e4066Sahrens 
1875fa9e4066Sahrens 	/*
1876fa9e4066Sahrens 	 * Prefetch a random chunk of the big object.
1877fa9e4066Sahrens 	 * Our aim here is to get some async reads in flight
1878fa9e4066Sahrens 	 * for blocks that we may free below; the DMU should
1879fa9e4066Sahrens 	 * handle this race correctly.
1880fa9e4066Sahrens 	 */
1881fa9e4066Sahrens 	n = ztest_random(regions) * stride + ztest_random(width);
1882fa9e4066Sahrens 	s = 1 + ztest_random(2 * width - 1);
1883fa9e4066Sahrens 	dmu_prefetch(os, dd.dd_bigobj, n * dd.dd_chunk, s * dd.dd_chunk);
1884fa9e4066Sahrens 
1885fa9e4066Sahrens 	/*
1886fa9e4066Sahrens 	 * Pick a random index and compute the offsets into packobj and bigobj.
1887fa9e4066Sahrens 	 */
1888fa9e4066Sahrens 	n = ztest_random(regions) * stride + ztest_random(width);
1889fa9e4066Sahrens 	s = 1 + ztest_random(width - 1);
1890fa9e4066Sahrens 
1891fa9e4066Sahrens 	packoff = n * sizeof (bufwad_t);
1892fa9e4066Sahrens 	packsize = s * sizeof (bufwad_t);
1893fa9e4066Sahrens 
1894fa9e4066Sahrens 	bigoff = n * dd.dd_chunk;
1895fa9e4066Sahrens 	bigsize = s * dd.dd_chunk;
1896fa9e4066Sahrens 
1897fa9e4066Sahrens 	packbuf = umem_alloc(packsize, UMEM_NOFAIL);
1898fa9e4066Sahrens 	bigbuf = umem_alloc(bigsize, UMEM_NOFAIL);
1899fa9e4066Sahrens 
1900fa9e4066Sahrens 	/*
1901fa9e4066Sahrens 	 * free_percent of the time, free a range of bigobj rather than
1902fa9e4066Sahrens 	 * overwriting it.
1903fa9e4066Sahrens 	 */
1904fa9e4066Sahrens 	freeit = (ztest_random(100) < free_percent);
1905fa9e4066Sahrens 
1906fa9e4066Sahrens 	/*
1907fa9e4066Sahrens 	 * Read the current contents of our objects.
1908fa9e4066Sahrens 	 */
1909ea8dc4b6Seschrock 	error = dmu_read(os, dd.dd_packobj, packoff, packsize, packbuf);
1910ea8dc4b6Seschrock 	ASSERT3U(error, ==, 0);
1911ea8dc4b6Seschrock 	error = dmu_read(os, dd.dd_bigobj, bigoff, bigsize, bigbuf);
1912ea8dc4b6Seschrock 	ASSERT3U(error, ==, 0);
1913fa9e4066Sahrens 
1914fa9e4066Sahrens 	/*
1915fa9e4066Sahrens 	 * Get a tx for the mods to both packobj and bigobj.
1916fa9e4066Sahrens 	 */
1917fa9e4066Sahrens 	tx = dmu_tx_create(os);
1918fa9e4066Sahrens 
1919fa9e4066Sahrens 	dmu_tx_hold_write(tx, dd.dd_packobj, packoff, packsize);
1920fa9e4066Sahrens 
1921fa9e4066Sahrens 	if (freeit)
1922fa9e4066Sahrens 		dmu_tx_hold_free(tx, dd.dd_bigobj, bigoff, bigsize);
1923fa9e4066Sahrens 	else
1924fa9e4066Sahrens 		dmu_tx_hold_write(tx, dd.dd_bigobj, bigoff, bigsize);
1925fa9e4066Sahrens 
1926fa9e4066Sahrens 	error = dmu_tx_assign(tx, TXG_WAIT);
1927fa9e4066Sahrens 
1928fa9e4066Sahrens 	if (error) {
1929fa9e4066Sahrens 		ztest_record_enospc("dmu r/w range");
1930fa9e4066Sahrens 		dmu_tx_abort(tx);
1931fa9e4066Sahrens 		umem_free(packbuf, packsize);
1932fa9e4066Sahrens 		umem_free(bigbuf, bigsize);
1933fa9e4066Sahrens 		return;
1934fa9e4066Sahrens 	}
1935fa9e4066Sahrens 
1936fa9e4066Sahrens 	txg = dmu_tx_get_txg(tx);
1937fa9e4066Sahrens 
1938fa9e4066Sahrens 	/*
1939fa9e4066Sahrens 	 * For each index from n to n + s, verify that the existing bufwad
1940fa9e4066Sahrens 	 * in packobj matches the bufwads at the head and tail of the
1941fa9e4066Sahrens 	 * corresponding chunk in bigobj.  Then update all three bufwads
1942fa9e4066Sahrens 	 * with the new values we want to write out.
1943fa9e4066Sahrens 	 */
1944fa9e4066Sahrens 	for (i = 0; i < s; i++) {
1945fa9e4066Sahrens 		/* LINTED */
1946fa9e4066Sahrens 		pack = (bufwad_t *)((char *)packbuf + i * sizeof (bufwad_t));
1947fa9e4066Sahrens 		/* LINTED */
1948fa9e4066Sahrens 		bigH = (bufwad_t *)((char *)bigbuf + i * dd.dd_chunk);
1949fa9e4066Sahrens 		/* LINTED */
1950fa9e4066Sahrens 		bigT = (bufwad_t *)((char *)bigH + dd.dd_chunk) - 1;
1951fa9e4066Sahrens 
1952fa9e4066Sahrens 		ASSERT((uintptr_t)bigH - (uintptr_t)bigbuf < bigsize);
1953fa9e4066Sahrens 		ASSERT((uintptr_t)bigT - (uintptr_t)bigbuf < bigsize);
1954fa9e4066Sahrens 
1955fa9e4066Sahrens 		if (pack->bw_txg > txg)
1956fa9e4066Sahrens 			fatal(0, "future leak: got %llx, open txg is %llx",
1957fa9e4066Sahrens 			    pack->bw_txg, txg);
1958fa9e4066Sahrens 
1959fa9e4066Sahrens 		if (pack->bw_data != 0 && pack->bw_index != n + i)
1960fa9e4066Sahrens 			fatal(0, "wrong index: got %llx, wanted %llx+%llx",
1961fa9e4066Sahrens 			    pack->bw_index, n, i);
1962fa9e4066Sahrens 
1963fa9e4066Sahrens 		if (bcmp(pack, bigH, sizeof (bufwad_t)) != 0)
1964fa9e4066Sahrens 			fatal(0, "pack/bigH mismatch in %p/%p", pack, bigH);
1965fa9e4066Sahrens 
1966fa9e4066Sahrens 		if (bcmp(pack, bigT, sizeof (bufwad_t)) != 0)
1967fa9e4066Sahrens 			fatal(0, "pack/bigT mismatch in %p/%p", pack, bigT);
1968fa9e4066Sahrens 
1969fa9e4066Sahrens 		if (freeit) {
1970fa9e4066Sahrens 			bzero(pack, sizeof (bufwad_t));
1971fa9e4066Sahrens 		} else {
1972fa9e4066Sahrens 			pack->bw_index = n + i;
1973fa9e4066Sahrens 			pack->bw_txg = txg;
1974fa9e4066Sahrens 			pack->bw_data = 1 + ztest_random(-2ULL);
1975fa9e4066Sahrens 		}
1976fa9e4066Sahrens 		*bigH = *pack;
1977fa9e4066Sahrens 		*bigT = *pack;
1978fa9e4066Sahrens 	}
1979fa9e4066Sahrens 
1980fa9e4066Sahrens 	/*
1981fa9e4066Sahrens 	 * We've verified all the old bufwads, and made new ones.
1982fa9e4066Sahrens 	 * Now write them out.
1983fa9e4066Sahrens 	 */
1984fa9e4066Sahrens 	dmu_write(os, dd.dd_packobj, packoff, packsize, packbuf, tx);
1985fa9e4066Sahrens 
1986fa9e4066Sahrens 	if (freeit) {
1987fa9e4066Sahrens 		if (zopt_verbose >= 6) {
1988fa9e4066Sahrens 			(void) printf("freeing offset %llx size %llx"
1989fa9e4066Sahrens 			    " txg %llx\n",
1990fa9e4066Sahrens 			    (u_longlong_t)bigoff,
1991fa9e4066Sahrens 			    (u_longlong_t)bigsize,
1992fa9e4066Sahrens 			    (u_longlong_t)txg);
1993fa9e4066Sahrens 		}
1994ea8dc4b6Seschrock 		VERIFY(0 == dmu_free_range(os, dd.dd_bigobj, bigoff,
1995ea8dc4b6Seschrock 		    bigsize, tx));
1996fa9e4066Sahrens 	} else {
1997fa9e4066Sahrens 		if (zopt_verbose >= 6) {
1998fa9e4066Sahrens 			(void) printf("writing offset %llx size %llx"
1999fa9e4066Sahrens 			    " txg %llx\n",
2000fa9e4066Sahrens 			    (u_longlong_t)bigoff,
2001fa9e4066Sahrens 			    (u_longlong_t)bigsize,
2002fa9e4066Sahrens 			    (u_longlong_t)txg);
2003fa9e4066Sahrens 		}
2004fa9e4066Sahrens 		dmu_write(os, dd.dd_bigobj, bigoff, bigsize, bigbuf, tx);
2005fa9e4066Sahrens 	}
2006fa9e4066Sahrens 
2007fa9e4066Sahrens 	dmu_tx_commit(tx);
2008fa9e4066Sahrens 
2009fa9e4066Sahrens 	/*
2010fa9e4066Sahrens 	 * Sanity check the stuff we just wrote.
2011fa9e4066Sahrens 	 */
2012fa9e4066Sahrens 	{
2013fa9e4066Sahrens 		void *packcheck = umem_alloc(packsize, UMEM_NOFAIL);
2014fa9e4066Sahrens 		void *bigcheck = umem_alloc(bigsize, UMEM_NOFAIL);
2015fa9e4066Sahrens 
2016ea8dc4b6Seschrock 		VERIFY(0 == dmu_read(os, dd.dd_packobj, packoff,
2017ea8dc4b6Seschrock 		    packsize, packcheck));
2018ea8dc4b6Seschrock 		VERIFY(0 == dmu_read(os, dd.dd_bigobj, bigoff,
2019ea8dc4b6Seschrock 		    bigsize, bigcheck));
2020fa9e4066Sahrens 
2021fa9e4066Sahrens 		ASSERT(bcmp(packbuf, packcheck, packsize) == 0);
2022fa9e4066Sahrens 		ASSERT(bcmp(bigbuf, bigcheck, bigsize) == 0);
2023fa9e4066Sahrens 
2024fa9e4066Sahrens 		umem_free(packcheck, packsize);
2025fa9e4066Sahrens 		umem_free(bigcheck, bigsize);
2026fa9e4066Sahrens 	}
2027fa9e4066Sahrens 
2028fa9e4066Sahrens 	umem_free(packbuf, packsize);
2029fa9e4066Sahrens 	umem_free(bigbuf, bigsize);
2030fa9e4066Sahrens }
2031fa9e4066Sahrens 
2032*2fdbea25SAleksandr Guzovskiy void
2033*2fdbea25SAleksandr Guzovskiy compare_and_update_pbbufs(uint64_t s, bufwad_t *packbuf, bufwad_t *bigbuf,
2034*2fdbea25SAleksandr Guzovskiy     uint64_t bigsize, uint64_t n, dmu_read_write_dir_t dd, uint64_t txg)
2035*2fdbea25SAleksandr Guzovskiy {
2036*2fdbea25SAleksandr Guzovskiy 	uint64_t i;
2037*2fdbea25SAleksandr Guzovskiy 	bufwad_t *pack;
2038*2fdbea25SAleksandr Guzovskiy 	bufwad_t *bigH;
2039*2fdbea25SAleksandr Guzovskiy 	bufwad_t *bigT;
2040*2fdbea25SAleksandr Guzovskiy 
2041*2fdbea25SAleksandr Guzovskiy 	/*
2042*2fdbea25SAleksandr Guzovskiy 	 * For each index from n to n + s, verify that the existing bufwad
2043*2fdbea25SAleksandr Guzovskiy 	 * in packobj matches the bufwads at the head and tail of the
2044*2fdbea25SAleksandr Guzovskiy 	 * corresponding chunk in bigobj.  Then update all three bufwads
2045*2fdbea25SAleksandr Guzovskiy 	 * with the new values we want to write out.
2046*2fdbea25SAleksandr Guzovskiy 	 */
2047*2fdbea25SAleksandr Guzovskiy 	for (i = 0; i < s; i++) {
2048*2fdbea25SAleksandr Guzovskiy 		/* LINTED */
2049*2fdbea25SAleksandr Guzovskiy 		pack = (bufwad_t *)((char *)packbuf + i * sizeof (bufwad_t));
2050*2fdbea25SAleksandr Guzovskiy 		/* LINTED */
2051*2fdbea25SAleksandr Guzovskiy 		bigH = (bufwad_t *)((char *)bigbuf + i * dd.dd_chunk);
2052*2fdbea25SAleksandr Guzovskiy 		/* LINTED */
2053*2fdbea25SAleksandr Guzovskiy 		bigT = (bufwad_t *)((char *)bigH + dd.dd_chunk) - 1;
2054*2fdbea25SAleksandr Guzovskiy 
2055*2fdbea25SAleksandr Guzovskiy 		ASSERT((uintptr_t)bigH - (uintptr_t)bigbuf < bigsize);
2056*2fdbea25SAleksandr Guzovskiy 		ASSERT((uintptr_t)bigT - (uintptr_t)bigbuf < bigsize);
2057*2fdbea25SAleksandr Guzovskiy 
2058*2fdbea25SAleksandr Guzovskiy 		if (pack->bw_txg > txg)
2059*2fdbea25SAleksandr Guzovskiy 			fatal(0, "future leak: got %llx, open txg is %llx",
2060*2fdbea25SAleksandr Guzovskiy 			    pack->bw_txg, txg);
2061*2fdbea25SAleksandr Guzovskiy 
2062*2fdbea25SAleksandr Guzovskiy 		if (pack->bw_data != 0 && pack->bw_index != n + i)
2063*2fdbea25SAleksandr Guzovskiy 			fatal(0, "wrong index: got %llx, wanted %llx+%llx",
2064*2fdbea25SAleksandr Guzovskiy 			    pack->bw_index, n, i);
2065*2fdbea25SAleksandr Guzovskiy 
2066*2fdbea25SAleksandr Guzovskiy 		if (bcmp(pack, bigH, sizeof (bufwad_t)) != 0)
2067*2fdbea25SAleksandr Guzovskiy 			fatal(0, "pack/bigH mismatch in %p/%p", pack, bigH);
2068*2fdbea25SAleksandr Guzovskiy 
2069*2fdbea25SAleksandr Guzovskiy 		if (bcmp(pack, bigT, sizeof (bufwad_t)) != 0)
2070*2fdbea25SAleksandr Guzovskiy 			fatal(0, "pack/bigT mismatch in %p/%p", pack, bigT);
2071*2fdbea25SAleksandr Guzovskiy 
2072*2fdbea25SAleksandr Guzovskiy 		pack->bw_index = n + i;
2073*2fdbea25SAleksandr Guzovskiy 		pack->bw_txg = txg;
2074*2fdbea25SAleksandr Guzovskiy 		pack->bw_data = 1 + ztest_random(-2ULL);
2075*2fdbea25SAleksandr Guzovskiy 
2076*2fdbea25SAleksandr Guzovskiy 		*bigH = *pack;
2077*2fdbea25SAleksandr Guzovskiy 		*bigT = *pack;
2078*2fdbea25SAleksandr Guzovskiy 	}
2079*2fdbea25SAleksandr Guzovskiy }
2080*2fdbea25SAleksandr Guzovskiy 
2081*2fdbea25SAleksandr Guzovskiy void
2082*2fdbea25SAleksandr Guzovskiy ztest_dmu_read_write_zcopy(ztest_args_t *za)
2083*2fdbea25SAleksandr Guzovskiy {
2084*2fdbea25SAleksandr Guzovskiy 	objset_t *os = za->za_os;
2085*2fdbea25SAleksandr Guzovskiy 	dmu_read_write_dir_t dd;
2086*2fdbea25SAleksandr Guzovskiy 	dmu_tx_t *tx;
2087*2fdbea25SAleksandr Guzovskiy 	uint64_t i;
2088*2fdbea25SAleksandr Guzovskiy 	int error;
2089*2fdbea25SAleksandr Guzovskiy 	uint64_t n, s, txg;
2090*2fdbea25SAleksandr Guzovskiy 	bufwad_t *packbuf, *bigbuf;
2091*2fdbea25SAleksandr Guzovskiy 	uint64_t packoff, packsize, bigoff, bigsize;
2092*2fdbea25SAleksandr Guzovskiy 	uint64_t regions = 997;
2093*2fdbea25SAleksandr Guzovskiy 	uint64_t stride = 123456789ULL;
2094*2fdbea25SAleksandr Guzovskiy 	uint64_t width = 9;
2095*2fdbea25SAleksandr Guzovskiy 	dmu_buf_t *bonus_db;
2096*2fdbea25SAleksandr Guzovskiy 	arc_buf_t **bigbuf_arcbufs;
2097*2fdbea25SAleksandr Guzovskiy 	dmu_object_info_t *doi = &za->za_doi;
2098*2fdbea25SAleksandr Guzovskiy 
2099*2fdbea25SAleksandr Guzovskiy 	/*
2100*2fdbea25SAleksandr Guzovskiy 	 * This test uses two objects, packobj and bigobj, that are always
2101*2fdbea25SAleksandr Guzovskiy 	 * updated together (i.e. in the same tx) so that their contents are
2102*2fdbea25SAleksandr Guzovskiy 	 * in sync and can be compared.  Their contents relate to each other
2103*2fdbea25SAleksandr Guzovskiy 	 * in a simple way: packobj is a dense array of 'bufwad' structures,
2104*2fdbea25SAleksandr Guzovskiy 	 * while bigobj is a sparse array of the same bufwads.  Specifically,
2105*2fdbea25SAleksandr Guzovskiy 	 * for any index n, there are three bufwads that should be identical:
2106*2fdbea25SAleksandr Guzovskiy 	 *
2107*2fdbea25SAleksandr Guzovskiy 	 *	packobj, at offset n * sizeof (bufwad_t)
2108*2fdbea25SAleksandr Guzovskiy 	 *	bigobj, at the head of the nth chunk
2109*2fdbea25SAleksandr Guzovskiy 	 *	bigobj, at the tail of the nth chunk
2110*2fdbea25SAleksandr Guzovskiy 	 *
2111*2fdbea25SAleksandr Guzovskiy 	 * The chunk size is set equal to bigobj block size so that
2112*2fdbea25SAleksandr Guzovskiy 	 * dmu_assign_arcbuf() can be tested for object updates.
2113*2fdbea25SAleksandr Guzovskiy 	 */
2114*2fdbea25SAleksandr Guzovskiy 
2115*2fdbea25SAleksandr Guzovskiy 	/*
2116*2fdbea25SAleksandr Guzovskiy 	 * Read the directory info.  If it's the first time, set things up.
2117*2fdbea25SAleksandr Guzovskiy 	 */
2118*2fdbea25SAleksandr Guzovskiy 	VERIFY(0 == dmu_read(os, ZTEST_DIROBJ, za->za_diroff,
2119*2fdbea25SAleksandr Guzovskiy 	    sizeof (dd), &dd));
2120*2fdbea25SAleksandr Guzovskiy 	if (dd.dd_chunk == 0) {
2121*2fdbea25SAleksandr Guzovskiy 		ASSERT(dd.dd_packobj == 0);
2122*2fdbea25SAleksandr Guzovskiy 		ASSERT(dd.dd_bigobj == 0);
2123*2fdbea25SAleksandr Guzovskiy 		tx = dmu_tx_create(os);
2124*2fdbea25SAleksandr Guzovskiy 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff, sizeof (dd));
2125*2fdbea25SAleksandr Guzovskiy 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
2126*2fdbea25SAleksandr Guzovskiy 		error = dmu_tx_assign(tx, TXG_WAIT);
2127*2fdbea25SAleksandr Guzovskiy 		if (error) {
2128*2fdbea25SAleksandr Guzovskiy 			ztest_record_enospc("create r/w directory");
2129*2fdbea25SAleksandr Guzovskiy 			dmu_tx_abort(tx);
2130*2fdbea25SAleksandr Guzovskiy 			return;
2131*2fdbea25SAleksandr Guzovskiy 		}
2132*2fdbea25SAleksandr Guzovskiy 
2133*2fdbea25SAleksandr Guzovskiy 		dd.dd_packobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
2134*2fdbea25SAleksandr Guzovskiy 		    DMU_OT_NONE, 0, tx);
2135*2fdbea25SAleksandr Guzovskiy 		dd.dd_bigobj = dmu_object_alloc(os, DMU_OT_UINT64_OTHER, 0,
2136*2fdbea25SAleksandr Guzovskiy 		    DMU_OT_NONE, 0, tx);
2137*2fdbea25SAleksandr Guzovskiy 		ztest_set_random_blocksize(os, dd.dd_packobj, tx);
2138*2fdbea25SAleksandr Guzovskiy 		ztest_set_random_blocksize(os, dd.dd_bigobj, tx);
2139*2fdbea25SAleksandr Guzovskiy 
2140*2fdbea25SAleksandr Guzovskiy 		VERIFY(dmu_object_info(os, dd.dd_bigobj, doi) == 0);
2141*2fdbea25SAleksandr Guzovskiy 		ASSERT(doi->doi_data_block_size >= 2 * sizeof (bufwad_t));
2142*2fdbea25SAleksandr Guzovskiy 		ASSERT(ISP2(doi->doi_data_block_size));
2143*2fdbea25SAleksandr Guzovskiy 		dd.dd_chunk = doi->doi_data_block_size;
2144*2fdbea25SAleksandr Guzovskiy 
2145*2fdbea25SAleksandr Guzovskiy 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff, sizeof (dd), &dd,
2146*2fdbea25SAleksandr Guzovskiy 		    tx);
2147*2fdbea25SAleksandr Guzovskiy 		dmu_tx_commit(tx);
2148*2fdbea25SAleksandr Guzovskiy 	} else {
2149*2fdbea25SAleksandr Guzovskiy 		VERIFY(dmu_object_info(os, dd.dd_bigobj, doi) == 0);
2150*2fdbea25SAleksandr Guzovskiy 		VERIFY(ISP2(doi->doi_data_block_size));
2151*2fdbea25SAleksandr Guzovskiy 		VERIFY(dd.dd_chunk == doi->doi_data_block_size);
2152*2fdbea25SAleksandr Guzovskiy 		VERIFY(dd.dd_chunk >= 2 * sizeof (bufwad_t));
2153*2fdbea25SAleksandr Guzovskiy 	}
2154*2fdbea25SAleksandr Guzovskiy 
2155*2fdbea25SAleksandr Guzovskiy 	/*
2156*2fdbea25SAleksandr Guzovskiy 	 * Pick a random index and compute the offsets into packobj and bigobj.
2157*2fdbea25SAleksandr Guzovskiy 	 */
2158*2fdbea25SAleksandr Guzovskiy 	n = ztest_random(regions) * stride + ztest_random(width);
2159*2fdbea25SAleksandr Guzovskiy 	s = 1 + ztest_random(width - 1);
2160*2fdbea25SAleksandr Guzovskiy 
2161*2fdbea25SAleksandr Guzovskiy 	packoff = n * sizeof (bufwad_t);
2162*2fdbea25SAleksandr Guzovskiy 	packsize = s * sizeof (bufwad_t);
2163*2fdbea25SAleksandr Guzovskiy 
2164*2fdbea25SAleksandr Guzovskiy 	bigoff = n * dd.dd_chunk;
2165*2fdbea25SAleksandr Guzovskiy 	bigsize = s * dd.dd_chunk;
2166*2fdbea25SAleksandr Guzovskiy 
2167*2fdbea25SAleksandr Guzovskiy 	packbuf = umem_zalloc(packsize, UMEM_NOFAIL);
2168*2fdbea25SAleksandr Guzovskiy 	bigbuf = umem_zalloc(bigsize, UMEM_NOFAIL);
2169*2fdbea25SAleksandr Guzovskiy 
2170*2fdbea25SAleksandr Guzovskiy 	VERIFY(dmu_bonus_hold(os, dd.dd_bigobj, FTAG, &bonus_db) == 0);
2171*2fdbea25SAleksandr Guzovskiy 
2172*2fdbea25SAleksandr Guzovskiy 	bigbuf_arcbufs = umem_zalloc(2 * s * sizeof (arc_buf_t *), UMEM_NOFAIL);
2173*2fdbea25SAleksandr Guzovskiy 
2174*2fdbea25SAleksandr Guzovskiy 	/*
2175*2fdbea25SAleksandr Guzovskiy 	 * Iteration 0 test zcopy for DB_UNCACHED dbufs.
2176*2fdbea25SAleksandr Guzovskiy 	 * Iteration 1 test zcopy to already referenced dbufs.
2177*2fdbea25SAleksandr Guzovskiy 	 * Iteration 2 test zcopy to dirty dbuf in the same txg.
2178*2fdbea25SAleksandr Guzovskiy 	 * Iteration 3 test zcopy to dbuf dirty in previous txg.
2179*2fdbea25SAleksandr Guzovskiy 	 * Iteration 4 test zcopy when dbuf is no longer dirty.
2180*2fdbea25SAleksandr Guzovskiy 	 * Iteration 5 test zcopy when it can't be done.
2181*2fdbea25SAleksandr Guzovskiy 	 * Iteration 6 one more zcopy write.
2182*2fdbea25SAleksandr Guzovskiy 	 */
2183*2fdbea25SAleksandr Guzovskiy 	for (i = 0; i < 7; i++) {
2184*2fdbea25SAleksandr Guzovskiy 		uint64_t j;
2185*2fdbea25SAleksandr Guzovskiy 		uint64_t off;
2186*2fdbea25SAleksandr Guzovskiy 
2187*2fdbea25SAleksandr Guzovskiy 		/*
2188*2fdbea25SAleksandr Guzovskiy 		 * In iteration 5 (i == 5) use arcbufs
2189*2fdbea25SAleksandr Guzovskiy 		 * that don't match bigobj blksz to test
2190*2fdbea25SAleksandr Guzovskiy 		 * dmu_assign_arcbuf() when it can't directly
2191*2fdbea25SAleksandr Guzovskiy 		 * assign an arcbuf to a dbuf.
2192*2fdbea25SAleksandr Guzovskiy 		 */
2193*2fdbea25SAleksandr Guzovskiy 		for (j = 0; j < s; j++) {
2194*2fdbea25SAleksandr Guzovskiy 			if (i != 5) {
2195*2fdbea25SAleksandr Guzovskiy 				bigbuf_arcbufs[j] =
2196*2fdbea25SAleksandr Guzovskiy 				    dmu_request_arcbuf(bonus_db,
2197*2fdbea25SAleksandr Guzovskiy 				    dd.dd_chunk);
2198*2fdbea25SAleksandr Guzovskiy 			} else {
2199*2fdbea25SAleksandr Guzovskiy 				bigbuf_arcbufs[2 * j] =
2200*2fdbea25SAleksandr Guzovskiy 				    dmu_request_arcbuf(bonus_db,
2201*2fdbea25SAleksandr Guzovskiy 				    dd.dd_chunk / 2);
2202*2fdbea25SAleksandr Guzovskiy 				bigbuf_arcbufs[2 * j + 1] =
2203*2fdbea25SAleksandr Guzovskiy 				    dmu_request_arcbuf(bonus_db,
2204*2fdbea25SAleksandr Guzovskiy 				    dd.dd_chunk / 2);
2205*2fdbea25SAleksandr Guzovskiy 			}
2206*2fdbea25SAleksandr Guzovskiy 		}
2207*2fdbea25SAleksandr Guzovskiy 
2208*2fdbea25SAleksandr Guzovskiy 		/*
2209*2fdbea25SAleksandr Guzovskiy 		 * Get a tx for the mods to both packobj and bigobj.
2210*2fdbea25SAleksandr Guzovskiy 		 */
2211*2fdbea25SAleksandr Guzovskiy 		tx = dmu_tx_create(os);
2212*2fdbea25SAleksandr Guzovskiy 
2213*2fdbea25SAleksandr Guzovskiy 		dmu_tx_hold_write(tx, dd.dd_packobj, packoff, packsize);
2214*2fdbea25SAleksandr Guzovskiy 		dmu_tx_hold_write(tx, dd.dd_bigobj, bigoff, bigsize);
2215*2fdbea25SAleksandr Guzovskiy 
2216*2fdbea25SAleksandr Guzovskiy 		if (ztest_random(100) == 0) {
2217*2fdbea25SAleksandr Guzovskiy 			error = -1;
2218*2fdbea25SAleksandr Guzovskiy 		} else {
2219*2fdbea25SAleksandr Guzovskiy 			error = dmu_tx_assign(tx, TXG_WAIT);
2220*2fdbea25SAleksandr Guzovskiy 		}
2221*2fdbea25SAleksandr Guzovskiy 
2222*2fdbea25SAleksandr Guzovskiy 		if (error) {
2223*2fdbea25SAleksandr Guzovskiy 			if (error != -1) {
2224*2fdbea25SAleksandr Guzovskiy 				ztest_record_enospc("dmu r/w range");
2225*2fdbea25SAleksandr Guzovskiy 			}
2226*2fdbea25SAleksandr Guzovskiy 			dmu_tx_abort(tx);
2227*2fdbea25SAleksandr Guzovskiy 			umem_free(packbuf, packsize);
2228*2fdbea25SAleksandr Guzovskiy 			umem_free(bigbuf, bigsize);
2229*2fdbea25SAleksandr Guzovskiy 			for (j = 0; j < s; j++) {
2230*2fdbea25SAleksandr Guzovskiy 				if (i != 5) {
2231*2fdbea25SAleksandr Guzovskiy 					dmu_return_arcbuf(bigbuf_arcbufs[j]);
2232*2fdbea25SAleksandr Guzovskiy 				} else {
2233*2fdbea25SAleksandr Guzovskiy 					dmu_return_arcbuf(
2234*2fdbea25SAleksandr Guzovskiy 					    bigbuf_arcbufs[2 * j]);
2235*2fdbea25SAleksandr Guzovskiy 					dmu_return_arcbuf(
2236*2fdbea25SAleksandr Guzovskiy 					    bigbuf_arcbufs[2 * j + 1]);
2237*2fdbea25SAleksandr Guzovskiy 				}
2238*2fdbea25SAleksandr Guzovskiy 			}
2239*2fdbea25SAleksandr Guzovskiy 			umem_free(bigbuf_arcbufs, 2 * s * sizeof (arc_buf_t *));
2240*2fdbea25SAleksandr Guzovskiy 			dmu_buf_rele(bonus_db, FTAG);
2241*2fdbea25SAleksandr Guzovskiy 			return;
2242*2fdbea25SAleksandr Guzovskiy 		}
2243*2fdbea25SAleksandr Guzovskiy 
2244*2fdbea25SAleksandr Guzovskiy 		txg = dmu_tx_get_txg(tx);
2245*2fdbea25SAleksandr Guzovskiy 
2246*2fdbea25SAleksandr Guzovskiy 		/*
2247*2fdbea25SAleksandr Guzovskiy 		 * 50% of the time don't read objects in the 1st iteration to
2248*2fdbea25SAleksandr Guzovskiy 		 * test dmu_assign_arcbuf() for the case when there're no
2249*2fdbea25SAleksandr Guzovskiy 		 * existing dbufs for the specified offsets.
2250*2fdbea25SAleksandr Guzovskiy 		 */
2251*2fdbea25SAleksandr Guzovskiy 		if (i != 0 || ztest_random(2) != 0) {
2252*2fdbea25SAleksandr Guzovskiy 			error = dmu_read(os, dd.dd_packobj, packoff,
2253*2fdbea25SAleksandr Guzovskiy 			    packsize, packbuf);
2254*2fdbea25SAleksandr Guzovskiy 			ASSERT3U(error, ==, 0);
2255*2fdbea25SAleksandr Guzovskiy 			error = dmu_read(os, dd.dd_bigobj, bigoff, bigsize,
2256*2fdbea25SAleksandr Guzovskiy 			    bigbuf);
2257*2fdbea25SAleksandr Guzovskiy 			ASSERT3U(error, ==, 0);
2258*2fdbea25SAleksandr Guzovskiy 		}
2259*2fdbea25SAleksandr Guzovskiy 		compare_and_update_pbbufs(s, packbuf, bigbuf, bigsize,
2260*2fdbea25SAleksandr Guzovskiy 		    n, dd, txg);
2261*2fdbea25SAleksandr Guzovskiy 
2262*2fdbea25SAleksandr Guzovskiy 		/*
2263*2fdbea25SAleksandr Guzovskiy 		 * We've verified all the old bufwads, and made new ones.
2264*2fdbea25SAleksandr Guzovskiy 		 * Now write them out.
2265*2fdbea25SAleksandr Guzovskiy 		 */
2266*2fdbea25SAleksandr Guzovskiy 		dmu_write(os, dd.dd_packobj, packoff, packsize, packbuf, tx);
2267*2fdbea25SAleksandr Guzovskiy 		if (zopt_verbose >= 6) {
2268*2fdbea25SAleksandr Guzovskiy 			(void) printf("writing offset %llx size %llx"
2269*2fdbea25SAleksandr Guzovskiy 			    " txg %llx\n",
2270*2fdbea25SAleksandr Guzovskiy 			    (u_longlong_t)bigoff,
2271*2fdbea25SAleksandr Guzovskiy 			    (u_longlong_t)bigsize,
2272*2fdbea25SAleksandr Guzovskiy 			    (u_longlong_t)txg);
2273*2fdbea25SAleksandr Guzovskiy 		}
2274*2fdbea25SAleksandr Guzovskiy 		for (off = bigoff, j = 0; j < s; j++, off += dd.dd_chunk) {
2275*2fdbea25SAleksandr Guzovskiy 			dmu_buf_t *dbt;
2276*2fdbea25SAleksandr Guzovskiy 			if (i != 5) {
2277*2fdbea25SAleksandr Guzovskiy 				bcopy((caddr_t)bigbuf + (off - bigoff),
2278*2fdbea25SAleksandr Guzovskiy 				    bigbuf_arcbufs[j]->b_data, dd.dd_chunk);
2279*2fdbea25SAleksandr Guzovskiy 			} else {
2280*2fdbea25SAleksandr Guzovskiy 				bcopy((caddr_t)bigbuf + (off - bigoff),
2281*2fdbea25SAleksandr Guzovskiy 				    bigbuf_arcbufs[2 * j]->b_data,
2282*2fdbea25SAleksandr Guzovskiy 				    dd.dd_chunk / 2);
2283*2fdbea25SAleksandr Guzovskiy 				bcopy((caddr_t)bigbuf + (off - bigoff) +
2284*2fdbea25SAleksandr Guzovskiy 				    dd.dd_chunk / 2,
2285*2fdbea25SAleksandr Guzovskiy 				    bigbuf_arcbufs[2 * j + 1]->b_data,
2286*2fdbea25SAleksandr Guzovskiy 				    dd.dd_chunk / 2);
2287*2fdbea25SAleksandr Guzovskiy 			}
2288*2fdbea25SAleksandr Guzovskiy 
2289*2fdbea25SAleksandr Guzovskiy 			if (i == 1) {
2290*2fdbea25SAleksandr Guzovskiy 				VERIFY(dmu_buf_hold(os, dd.dd_bigobj, off,
2291*2fdbea25SAleksandr Guzovskiy 				    FTAG, &dbt) == 0);
2292*2fdbea25SAleksandr Guzovskiy 			}
2293*2fdbea25SAleksandr Guzovskiy 			if (i != 5) {
2294*2fdbea25SAleksandr Guzovskiy 				dmu_assign_arcbuf(bonus_db, off,
2295*2fdbea25SAleksandr Guzovskiy 				    bigbuf_arcbufs[j], tx);
2296*2fdbea25SAleksandr Guzovskiy 			} else {
2297*2fdbea25SAleksandr Guzovskiy 				dmu_assign_arcbuf(bonus_db, off,
2298*2fdbea25SAleksandr Guzovskiy 				    bigbuf_arcbufs[2 * j], tx);
2299*2fdbea25SAleksandr Guzovskiy 				dmu_assign_arcbuf(bonus_db,
2300*2fdbea25SAleksandr Guzovskiy 				    off + dd.dd_chunk / 2,
2301*2fdbea25SAleksandr Guzovskiy 				    bigbuf_arcbufs[2 * j + 1], tx);
2302*2fdbea25SAleksandr Guzovskiy 			}
2303*2fdbea25SAleksandr Guzovskiy 			if (i == 1) {
2304*2fdbea25SAleksandr Guzovskiy 				dmu_buf_rele(dbt, FTAG);
2305*2fdbea25SAleksandr Guzovskiy 			}
2306*2fdbea25SAleksandr Guzovskiy 		}
2307*2fdbea25SAleksandr Guzovskiy 		dmu_tx_commit(tx);
2308*2fdbea25SAleksandr Guzovskiy 
2309*2fdbea25SAleksandr Guzovskiy 		/*
2310*2fdbea25SAleksandr Guzovskiy 		 * Sanity check the stuff we just wrote.
2311*2fdbea25SAleksandr Guzovskiy 		 */
2312*2fdbea25SAleksandr Guzovskiy 		{
2313*2fdbea25SAleksandr Guzovskiy 			void *packcheck = umem_alloc(packsize, UMEM_NOFAIL);
2314*2fdbea25SAleksandr Guzovskiy 			void *bigcheck = umem_alloc(bigsize, UMEM_NOFAIL);
2315*2fdbea25SAleksandr Guzovskiy 
2316*2fdbea25SAleksandr Guzovskiy 			VERIFY(0 == dmu_read(os, dd.dd_packobj, packoff,
2317*2fdbea25SAleksandr Guzovskiy 			    packsize, packcheck));
2318*2fdbea25SAleksandr Guzovskiy 			VERIFY(0 == dmu_read(os, dd.dd_bigobj, bigoff,
2319*2fdbea25SAleksandr Guzovskiy 			    bigsize, bigcheck));
2320*2fdbea25SAleksandr Guzovskiy 
2321*2fdbea25SAleksandr Guzovskiy 			ASSERT(bcmp(packbuf, packcheck, packsize) == 0);
2322*2fdbea25SAleksandr Guzovskiy 			ASSERT(bcmp(bigbuf, bigcheck, bigsize) == 0);
2323*2fdbea25SAleksandr Guzovskiy 
2324*2fdbea25SAleksandr Guzovskiy 			umem_free(packcheck, packsize);
2325*2fdbea25SAleksandr Guzovskiy 			umem_free(bigcheck, bigsize);
2326*2fdbea25SAleksandr Guzovskiy 		}
2327*2fdbea25SAleksandr Guzovskiy 		if (i == 2) {
2328*2fdbea25SAleksandr Guzovskiy 			txg_wait_open(dmu_objset_pool(os), 0);
2329*2fdbea25SAleksandr Guzovskiy 		} else if (i == 3) {
2330*2fdbea25SAleksandr Guzovskiy 			txg_wait_synced(dmu_objset_pool(os), 0);
2331*2fdbea25SAleksandr Guzovskiy 		}
2332*2fdbea25SAleksandr Guzovskiy 	}
2333*2fdbea25SAleksandr Guzovskiy 
2334*2fdbea25SAleksandr Guzovskiy 	dmu_buf_rele(bonus_db, FTAG);
2335*2fdbea25SAleksandr Guzovskiy 	umem_free(packbuf, packsize);
2336*2fdbea25SAleksandr Guzovskiy 	umem_free(bigbuf, bigsize);
2337*2fdbea25SAleksandr Guzovskiy 	umem_free(bigbuf_arcbufs, 2 * s * sizeof (arc_buf_t *));
2338*2fdbea25SAleksandr Guzovskiy }
2339*2fdbea25SAleksandr Guzovskiy 
23404d31c452Smaybee void
2341e05725b1Sbonwick ztest_dmu_check_future_leak(ztest_args_t *za)
23424d31c452Smaybee {
2343e05725b1Sbonwick 	objset_t *os = za->za_os;
23444d31c452Smaybee 	dmu_buf_t *db;
2345e05725b1Sbonwick 	ztest_block_tag_t *bt;
2346e05725b1Sbonwick 	dmu_object_info_t *doi = &za->za_doi;
23474d31c452Smaybee 
23484d31c452Smaybee 	/*
23494d31c452Smaybee 	 * Make sure that, if there is a write record in the bonus buffer
23504d31c452Smaybee 	 * of the ZTEST_DIROBJ, that the txg for this record is <= the
23514d31c452Smaybee 	 * last synced txg of the pool.
23524d31c452Smaybee 	 */
2353e05725b1Sbonwick 	VERIFY(dmu_bonus_hold(os, ZTEST_DIROBJ, FTAG, &db) == 0);
2354e05725b1Sbonwick 	za->za_dbuf = db;
2355e05725b1Sbonwick 	VERIFY(dmu_object_info(os, ZTEST_DIROBJ, doi) == 0);
2356e05725b1Sbonwick 	ASSERT3U(doi->doi_bonus_size, >=, sizeof (*bt));
2357e05725b1Sbonwick 	ASSERT3U(doi->doi_bonus_size, <=, db->db_size);
2358e05725b1Sbonwick 	ASSERT3U(doi->doi_bonus_size % sizeof (*bt), ==, 0);
2359e05725b1Sbonwick 	bt = (void *)((char *)db->db_data + doi->doi_bonus_size - sizeof (*bt));
2360e05725b1Sbonwick 	if (bt->bt_objset != 0) {
2361e05725b1Sbonwick 		ASSERT3U(bt->bt_objset, ==, dmu_objset_id(os));
2362e05725b1Sbonwick 		ASSERT3U(bt->bt_object, ==, ZTEST_DIROBJ);
2363e05725b1Sbonwick 		ASSERT3U(bt->bt_offset, ==, -1ULL);
2364e05725b1Sbonwick 		ASSERT3U(bt->bt_txg, <, spa_first_txg(za->za_spa));
23654d31c452Smaybee 	}
23664d31c452Smaybee 	dmu_buf_rele(db, FTAG);
2367e05725b1Sbonwick 	za->za_dbuf = NULL;
23684d31c452Smaybee }
23694d31c452Smaybee 
2370fa9e4066Sahrens void
2371fa9e4066Sahrens ztest_dmu_write_parallel(ztest_args_t *za)
2372fa9e4066Sahrens {
2373fa9e4066Sahrens 	objset_t *os = za->za_os;
2374e05725b1Sbonwick 	ztest_block_tag_t *rbt = &za->za_rbt;
2375e05725b1Sbonwick 	ztest_block_tag_t *wbt = &za->za_wbt;
2376e05725b1Sbonwick 	const size_t btsize = sizeof (ztest_block_tag_t);
2377fa9e4066Sahrens 	dmu_buf_t *db;
2378e05725b1Sbonwick 	int b, error;
2379e05725b1Sbonwick 	int bs = ZTEST_DIROBJ_BLOCKSIZE;
2380e05725b1Sbonwick 	int do_free = 0;
2381f4a72450SJeff Bonwick 	uint64_t off, txg, txg_how;
2382fa9e4066Sahrens 	mutex_t *lp;
2383fa9e4066Sahrens 	char osname[MAXNAMELEN];
2384fa9e4066Sahrens 	char iobuf[SPA_MAXBLOCKSIZE];
2385e05725b1Sbonwick 	blkptr_t blk = { 0 };
2386e05725b1Sbonwick 	uint64_t blkoff;
2387e05725b1Sbonwick 	zbookmark_t zb;
2388e05725b1Sbonwick 	dmu_tx_t *tx = dmu_tx_create(os);
2389*2fdbea25SAleksandr Guzovskiy 	dmu_buf_t *bonus_db;
2390*2fdbea25SAleksandr Guzovskiy 	arc_buf_t *abuf = NULL;
2391fa9e4066Sahrens 
2392fa9e4066Sahrens 	dmu_objset_name(os, osname);
2393fa9e4066Sahrens 
2394fa9e4066Sahrens 	/*
2395fa9e4066Sahrens 	 * Have multiple threads write to large offsets in ZTEST_DIROBJ
2396fa9e4066Sahrens 	 * to verify that having multiple threads writing to the same object
2397fa9e4066Sahrens 	 * in parallel doesn't cause any trouble.
2398fa9e4066Sahrens 	 */
2399e05725b1Sbonwick 	if (ztest_random(4) == 0) {
2400e05725b1Sbonwick 		/*
2401e05725b1Sbonwick 		 * Do the bonus buffer instead of a regular block.
2402e05725b1Sbonwick 		 * We need a lock to serialize resize vs. others,
2403e05725b1Sbonwick 		 * so we hash on the objset ID.
2404e05725b1Sbonwick 		 */
2405e05725b1Sbonwick 		b = dmu_objset_id(os) % ZTEST_SYNC_LOCKS;
2406e05725b1Sbonwick 		off = -1ULL;
2407e05725b1Sbonwick 		dmu_tx_hold_bonus(tx, ZTEST_DIROBJ);
2408e05725b1Sbonwick 	} else {
2409fa9e4066Sahrens 		b = ztest_random(ZTEST_SYNC_LOCKS);
2410e05725b1Sbonwick 		off = za->za_diroff_shared + (b << SPA_MAXBLOCKSHIFT);
2411fa9e4066Sahrens 		if (ztest_random(4) == 0) {
2412e05725b1Sbonwick 			do_free = 1;
2413fa9e4066Sahrens 			dmu_tx_hold_free(tx, ZTEST_DIROBJ, off, bs);
2414e05725b1Sbonwick 		} else {
2415fa9e4066Sahrens 			dmu_tx_hold_write(tx, ZTEST_DIROBJ, off, bs);
2416e05725b1Sbonwick 		}
2417e05725b1Sbonwick 	}
2418fa9e4066Sahrens 
2419*2fdbea25SAleksandr Guzovskiy 	if (off != -1ULL && P2PHASE(off, bs) == 0 && !do_free &&
2420*2fdbea25SAleksandr Guzovskiy 	    ztest_random(8) == 0) {
2421*2fdbea25SAleksandr Guzovskiy 		VERIFY(dmu_bonus_hold(os, ZTEST_DIROBJ, FTAG, &bonus_db) == 0);
2422*2fdbea25SAleksandr Guzovskiy 		abuf = dmu_request_arcbuf(bonus_db, bs);
2423*2fdbea25SAleksandr Guzovskiy 	}
2424*2fdbea25SAleksandr Guzovskiy 
2425e05725b1Sbonwick 	txg_how = ztest_random(2) == 0 ? TXG_WAIT : TXG_NOWAIT;
2426e05725b1Sbonwick 	error = dmu_tx_assign(tx, txg_how);
2427e05725b1Sbonwick 	if (error) {
2428e05725b1Sbonwick 		if (error == ERESTART) {
2429e05725b1Sbonwick 			ASSERT(txg_how == TXG_NOWAIT);
2430e05725b1Sbonwick 			dmu_tx_wait(tx);
2431e05725b1Sbonwick 		} else {
2432fa9e4066Sahrens 			ztest_record_enospc("dmu write parallel");
2433fa9e4066Sahrens 		}
2434e05725b1Sbonwick 		dmu_tx_abort(tx);
2435*2fdbea25SAleksandr Guzovskiy 		if (abuf != NULL) {
2436*2fdbea25SAleksandr Guzovskiy 			dmu_return_arcbuf(abuf);
2437*2fdbea25SAleksandr Guzovskiy 			dmu_buf_rele(bonus_db, FTAG);
2438*2fdbea25SAleksandr Guzovskiy 		}
2439e05725b1Sbonwick 		return;
2440e05725b1Sbonwick 	}
2441f4a72450SJeff Bonwick 	txg = dmu_tx_get_txg(tx);
2442fa9e4066Sahrens 
2443e05725b1Sbonwick 	lp = &ztest_shared->zs_sync_lock[b];
2444e05725b1Sbonwick 	(void) mutex_lock(lp);
2445e05725b1Sbonwick 
2446e05725b1Sbonwick 	wbt->bt_objset = dmu_objset_id(os);
2447e05725b1Sbonwick 	wbt->bt_object = ZTEST_DIROBJ;
2448e05725b1Sbonwick 	wbt->bt_offset = off;
2449f4a72450SJeff Bonwick 	wbt->bt_txg = txg;
2450e05725b1Sbonwick 	wbt->bt_thread = za->za_instance;
2451e05725b1Sbonwick 	wbt->bt_seq = ztest_shared->zs_seq[b]++;	/* protected by lp */
2452e05725b1Sbonwick 
2453f4a72450SJeff Bonwick 	/*
2454f4a72450SJeff Bonwick 	 * Occasionally, write an all-zero block to test the behavior
2455f4a72450SJeff Bonwick 	 * of blocks that compress into holes.
2456f4a72450SJeff Bonwick 	 */
2457f4a72450SJeff Bonwick 	if (off != -1ULL && ztest_random(8) == 0)
2458f4a72450SJeff Bonwick 		bzero(wbt, btsize);
2459f4a72450SJeff Bonwick 
2460e05725b1Sbonwick 	if (off == -1ULL) {
2461e05725b1Sbonwick 		dmu_object_info_t *doi = &za->za_doi;
2462e05725b1Sbonwick 		char *dboff;
2463e05725b1Sbonwick 
2464e05725b1Sbonwick 		VERIFY(dmu_bonus_hold(os, ZTEST_DIROBJ, FTAG, &db) == 0);
2465e05725b1Sbonwick 		za->za_dbuf = db;
2466e05725b1Sbonwick 		dmu_object_info_from_db(db, doi);
2467e05725b1Sbonwick 		ASSERT3U(doi->doi_bonus_size, <=, db->db_size);
2468e05725b1Sbonwick 		ASSERT3U(doi->doi_bonus_size, >=, btsize);
2469e05725b1Sbonwick 		ASSERT3U(doi->doi_bonus_size % btsize, ==, 0);
2470e05725b1Sbonwick 		dboff = (char *)db->db_data + doi->doi_bonus_size - btsize;
2471e05725b1Sbonwick 		bcopy(dboff, rbt, btsize);
2472e05725b1Sbonwick 		if (rbt->bt_objset != 0) {
2473e05725b1Sbonwick 			ASSERT3U(rbt->bt_objset, ==, wbt->bt_objset);
2474e05725b1Sbonwick 			ASSERT3U(rbt->bt_object, ==, wbt->bt_object);
2475e05725b1Sbonwick 			ASSERT3U(rbt->bt_offset, ==, wbt->bt_offset);
2476e05725b1Sbonwick 			ASSERT3U(rbt->bt_txg, <=, wbt->bt_txg);
2477fa9e4066Sahrens 		}
2478e05725b1Sbonwick 		if (ztest_random(10) == 0) {
2479e05725b1Sbonwick 			int newsize = (ztest_random(db->db_size /
2480e05725b1Sbonwick 			    btsize) + 1) * btsize;
2481e05725b1Sbonwick 
2482e05725b1Sbonwick 			ASSERT3U(newsize, >=, btsize);
2483e05725b1Sbonwick 			ASSERT3U(newsize, <=, db->db_size);
2484e05725b1Sbonwick 			VERIFY3U(dmu_set_bonus(db, newsize, tx), ==, 0);
2485e05725b1Sbonwick 			dboff = (char *)db->db_data + newsize - btsize;
2486fa9e4066Sahrens 		}
2487e05725b1Sbonwick 		dmu_buf_will_dirty(db, tx);
2488e05725b1Sbonwick 		bcopy(wbt, dboff, btsize);
2489e05725b1Sbonwick 		dmu_buf_rele(db, FTAG);
2490e05725b1Sbonwick 		za->za_dbuf = NULL;
2491e05725b1Sbonwick 	} else if (do_free) {
2492e05725b1Sbonwick 		VERIFY(dmu_free_range(os, ZTEST_DIROBJ, off, bs, tx) == 0);
2493*2fdbea25SAleksandr Guzovskiy 	} else if (abuf == NULL) {
2494e05725b1Sbonwick 		dmu_write(os, ZTEST_DIROBJ, off, btsize, wbt, tx);
2495*2fdbea25SAleksandr Guzovskiy 	} else {
2496*2fdbea25SAleksandr Guzovskiy 		bcopy(wbt, abuf->b_data, btsize);
2497*2fdbea25SAleksandr Guzovskiy 		dmu_assign_arcbuf(bonus_db, off, abuf, tx);
2498*2fdbea25SAleksandr Guzovskiy 		dmu_buf_rele(bonus_db, FTAG);
2499e05725b1Sbonwick 	}
2500e05725b1Sbonwick 
2501e05725b1Sbonwick 	(void) mutex_unlock(lp);
2502e05725b1Sbonwick 
2503e05725b1Sbonwick 	if (ztest_random(1000) == 0)
2504e05725b1Sbonwick 		(void) poll(NULL, 0, 1); /* open dn_notxholds window */
2505e05725b1Sbonwick 
2506e05725b1Sbonwick 	dmu_tx_commit(tx);
2507e05725b1Sbonwick 
2508e05725b1Sbonwick 	if (ztest_random(10000) == 0)
2509f4a72450SJeff Bonwick 		txg_wait_synced(dmu_objset_pool(os), txg);
2510fa9e4066Sahrens 
2511e14bb325SJeff Bonwick 	if (off == -1ULL || do_free)
2512e05725b1Sbonwick 		return;
2513fa9e4066Sahrens 
2514e05725b1Sbonwick 	if (ztest_random(2) != 0)
2515e05725b1Sbonwick 		return;
2516fa9e4066Sahrens 
2517e05725b1Sbonwick 	/*
2518e05725b1Sbonwick 	 * dmu_sync() the block we just wrote.
2519e05725b1Sbonwick 	 */
2520e05725b1Sbonwick 	(void) mutex_lock(lp);
2521fa9e4066Sahrens 
2522e05725b1Sbonwick 	blkoff = P2ALIGN_TYPED(off, bs, uint64_t);
2523e05725b1Sbonwick 	error = dmu_buf_hold(os, ZTEST_DIROBJ, blkoff, FTAG, &db);
2524e05725b1Sbonwick 	za->za_dbuf = db;
2525e05725b1Sbonwick 	if (error) {
2526fa9e4066Sahrens 		(void) mutex_unlock(lp);
2527e05725b1Sbonwick 		return;
2528e05725b1Sbonwick 	}
2529e05725b1Sbonwick 	blkoff = off - blkoff;
2530f4a72450SJeff Bonwick 	error = dmu_sync(NULL, db, &blk, txg, NULL, NULL);
2531e05725b1Sbonwick 	dmu_buf_rele(db, FTAG);
2532e05725b1Sbonwick 	za->za_dbuf = NULL;
2533fa9e4066Sahrens 
2534*2fdbea25SAleksandr Guzovskiy 	if (error) {
2535*2fdbea25SAleksandr Guzovskiy 		(void) mutex_unlock(lp);
2536e05725b1Sbonwick 		return;
2537*2fdbea25SAleksandr Guzovskiy 	}
2538fa9e4066Sahrens 
2539*2fdbea25SAleksandr Guzovskiy 	if (blk.blk_birth == 0)	{	/* concurrent free */
2540*2fdbea25SAleksandr Guzovskiy 		(void) mutex_unlock(lp);
2541e05725b1Sbonwick 		return;
2542*2fdbea25SAleksandr Guzovskiy 	}
2543fa9e4066Sahrens 
2544e05725b1Sbonwick 	txg_suspend(dmu_objset_pool(os));
2545fa9e4066Sahrens 
2546*2fdbea25SAleksandr Guzovskiy 	(void) mutex_unlock(lp);
2547*2fdbea25SAleksandr Guzovskiy 
2548e05725b1Sbonwick 	ASSERT(blk.blk_fill == 1);
2549e05725b1Sbonwick 	ASSERT3U(BP_GET_TYPE(&blk), ==, DMU_OT_UINT64_OTHER);
2550e05725b1Sbonwick 	ASSERT3U(BP_GET_LEVEL(&blk), ==, 0);
2551e05725b1Sbonwick 	ASSERT3U(BP_GET_LSIZE(&blk), ==, bs);
2552fa9e4066Sahrens 
2553e05725b1Sbonwick 	/*
2554e05725b1Sbonwick 	 * Read the block that dmu_sync() returned to make sure its contents
2555e05725b1Sbonwick 	 * match what we wrote.  We do this while still txg_suspend()ed
2556e05725b1Sbonwick 	 * to ensure that the block can't be reused before we read it.
2557e05725b1Sbonwick 	 */
2558e05725b1Sbonwick 	zb.zb_objset = dmu_objset_id(os);
2559e05725b1Sbonwick 	zb.zb_object = ZTEST_DIROBJ;
2560e05725b1Sbonwick 	zb.zb_level = 0;
2561e05725b1Sbonwick 	zb.zb_blkid = off / bs;
2562e05725b1Sbonwick 	error = zio_wait(zio_read(NULL, za->za_spa, &blk, iobuf, bs,
2563e05725b1Sbonwick 	    NULL, NULL, ZIO_PRIORITY_SYNC_READ, ZIO_FLAG_MUSTSUCCEED, &zb));
2564e05725b1Sbonwick 	ASSERT3U(error, ==, 0);
2565fa9e4066Sahrens 
2566e05725b1Sbonwick 	txg_resume(dmu_objset_pool(os));
2567fa9e4066Sahrens 
2568e05725b1Sbonwick 	bcopy(&iobuf[blkoff], rbt, btsize);
2569fa9e4066Sahrens 
2570e05725b1Sbonwick 	if (rbt->bt_objset == 0)		/* concurrent free */
2571e05725b1Sbonwick 		return;
25720e34b6a7Sbonwick 
2573f4a72450SJeff Bonwick 	if (wbt->bt_objset == 0)		/* all-zero overwrite */
2574f4a72450SJeff Bonwick 		return;
2575f4a72450SJeff Bonwick 
2576e05725b1Sbonwick 	ASSERT3U(rbt->bt_objset, ==, wbt->bt_objset);
2577e05725b1Sbonwick 	ASSERT3U(rbt->bt_object, ==, wbt->bt_object);
2578e05725b1Sbonwick 	ASSERT3U(rbt->bt_offset, ==, wbt->bt_offset);
2579fa9e4066Sahrens 
2580e05725b1Sbonwick 	/*
2581e05725b1Sbonwick 	 * The semantic of dmu_sync() is that we always push the most recent
2582e05725b1Sbonwick 	 * version of the data, so in the face of concurrent updates we may
2583e05725b1Sbonwick 	 * see a newer version of the block.  That's OK.
2584e05725b1Sbonwick 	 */
2585e05725b1Sbonwick 	ASSERT3U(rbt->bt_txg, >=, wbt->bt_txg);
2586e05725b1Sbonwick 	if (rbt->bt_thread == wbt->bt_thread)
2587e05725b1Sbonwick 		ASSERT3U(rbt->bt_seq, ==, wbt->bt_seq);
2588e05725b1Sbonwick 	else
2589e05725b1Sbonwick 		ASSERT3U(rbt->bt_seq, >, wbt->bt_seq);
2590fa9e4066Sahrens }
2591fa9e4066Sahrens 
2592fa9e4066Sahrens /*
2593fa9e4066Sahrens  * Verify that zap_{create,destroy,add,remove,update} work as expected.
2594fa9e4066Sahrens  */
2595fa9e4066Sahrens #define	ZTEST_ZAP_MIN_INTS	1
2596fa9e4066Sahrens #define	ZTEST_ZAP_MAX_INTS	4
2597fa9e4066Sahrens #define	ZTEST_ZAP_MAX_PROPS	1000
2598fa9e4066Sahrens 
2599fa9e4066Sahrens void
2600fa9e4066Sahrens ztest_zap(ztest_args_t *za)
2601fa9e4066Sahrens {
2602fa9e4066Sahrens 	objset_t *os = za->za_os;
2603fa9e4066Sahrens 	uint64_t object;
2604fa9e4066Sahrens 	uint64_t txg, last_txg;
2605fa9e4066Sahrens 	uint64_t value[ZTEST_ZAP_MAX_INTS];
2606fa9e4066Sahrens 	uint64_t zl_ints, zl_intsize, prop;
2607fa9e4066Sahrens 	int i, ints;
2608fa9e4066Sahrens 	dmu_tx_t *tx;
2609fa9e4066Sahrens 	char propname[100], txgname[100];
2610fa9e4066Sahrens 	int error;
2611fa9e4066Sahrens 	char osname[MAXNAMELEN];
2612fa9e4066Sahrens 	char *hc[2] = { "s.acl.h", ".s.open.h.hyLZlg" };
2613fa9e4066Sahrens 
2614fa9e4066Sahrens 	dmu_objset_name(os, osname);
2615fa9e4066Sahrens 
2616fa9e4066Sahrens 	/*
2617fa9e4066Sahrens 	 * Create a new object if necessary, and record it in the directory.
2618fa9e4066Sahrens 	 */
2619ea8dc4b6Seschrock 	VERIFY(0 == dmu_read(os, ZTEST_DIROBJ, za->za_diroff,
2620ea8dc4b6Seschrock 	    sizeof (uint64_t), &object));
2621fa9e4066Sahrens 
2622fa9e4066Sahrens 	if (object == 0) {
2623fa9e4066Sahrens 		tx = dmu_tx_create(os);
2624fa9e4066Sahrens 		dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff,
2625fa9e4066Sahrens 		    sizeof (uint64_t));
2626ea8dc4b6Seschrock 		dmu_tx_hold_zap(tx, DMU_NEW_OBJECT, TRUE, NULL);
2627fa9e4066Sahrens 		error = dmu_tx_assign(tx, TXG_WAIT);
2628fa9e4066Sahrens 		if (error) {
2629fa9e4066Sahrens 			ztest_record_enospc("create zap test obj");
2630fa9e4066Sahrens 			dmu_tx_abort(tx);
2631fa9e4066Sahrens 			return;
2632fa9e4066Sahrens 		}
2633fa9e4066Sahrens 		object = zap_create(os, DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx);
2634fa9e4066Sahrens 		if (error) {
2635fa9e4066Sahrens 			fatal(0, "zap_create('%s', %llu) = %d",
2636fa9e4066Sahrens 			    osname, object, error);
2637fa9e4066Sahrens 		}
2638fa9e4066Sahrens 		ASSERT(object != 0);
2639fa9e4066Sahrens 		dmu_write(os, ZTEST_DIROBJ, za->za_diroff,
2640fa9e4066Sahrens 		    sizeof (uint64_t), &object, tx);
2641fa9e4066Sahrens 		/*
2642fa9e4066Sahrens 		 * Generate a known hash collision, and verify that
2643fa9e4066Sahrens 		 * we can lookup and remove both entries.
2644fa9e4066Sahrens 		 */
2645fa9e4066Sahrens 		for (i = 0; i < 2; i++) {
2646fa9e4066Sahrens 			value[i] = i;
2647fa9e4066Sahrens 			error = zap_add(os, object, hc[i], sizeof (uint64_t),
2648fa9e4066Sahrens 			    1, &value[i], tx);
2649fa9e4066Sahrens 			ASSERT3U(error, ==, 0);
2650fa9e4066Sahrens 		}
2651fa9e4066Sahrens 		for (i = 0; i < 2; i++) {
2652fa9e4066Sahrens 			error = zap_add(os, object, hc[i], sizeof (uint64_t),
2653fa9e4066Sahrens 			    1, &value[i], tx);
2654fa9e4066Sahrens 			ASSERT3U(error, ==, EEXIST);
2655fa9e4066Sahrens 			error = zap_length(os, object, hc[i],
2656fa9e4066Sahrens 			    &zl_intsize, &zl_ints);
2657fa9e4066Sahrens 			ASSERT3U(error, ==, 0);
2658fa9e4066Sahrens 			ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
2659fa9e4066Sahrens 			ASSERT3U(zl_ints, ==, 1);
2660fa9e4066Sahrens 		}
2661fa9e4066Sahrens 		for (i = 0; i < 2; i++) {
2662fa9e4066Sahrens 			error = zap_remove(os, object, hc[i], tx);
2663fa9e4066Sahrens 			ASSERT3U(error, ==, 0);
2664fa9e4066Sahrens 		}
2665fa9e4066Sahrens 
2666fa9e4066Sahrens 		dmu_tx_commit(tx);
2667fa9e4066Sahrens 	}
2668fa9e4066Sahrens 
2669fa9e4066Sahrens 	ints = MAX(ZTEST_ZAP_MIN_INTS, object % ZTEST_ZAP_MAX_INTS);
2670fa9e4066Sahrens 
2671e05725b1Sbonwick 	prop = ztest_random(ZTEST_ZAP_MAX_PROPS);
2672e05725b1Sbonwick 	(void) sprintf(propname, "prop_%llu", (u_longlong_t)prop);
2673e05725b1Sbonwick 	(void) sprintf(txgname, "txg_%llu", (u_longlong_t)prop);
2674e05725b1Sbonwick 	bzero(value, sizeof (value));
2675e05725b1Sbonwick 	last_txg = 0;
2676fa9e4066Sahrens 
2677e05725b1Sbonwick 	/*
2678e05725b1Sbonwick 	 * If these zap entries already exist, validate their contents.
2679e05725b1Sbonwick 	 */
2680e05725b1Sbonwick 	error = zap_length(os, object, txgname, &zl_intsize, &zl_ints);
2681e05725b1Sbonwick 	if (error == 0) {
2682e05725b1Sbonwick 		ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
2683e05725b1Sbonwick 		ASSERT3U(zl_ints, ==, 1);
2684fa9e4066Sahrens 
2685e05725b1Sbonwick 		VERIFY(zap_lookup(os, object, txgname, zl_intsize,
2686e05725b1Sbonwick 		    zl_ints, &last_txg) == 0);
2687fa9e4066Sahrens 
2688e05725b1Sbonwick 		VERIFY(zap_length(os, object, propname, &zl_intsize,
2689e05725b1Sbonwick 		    &zl_ints) == 0);
2690fa9e4066Sahrens 
2691e05725b1Sbonwick 		ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
2692e05725b1Sbonwick 		ASSERT3U(zl_ints, ==, ints);
2693fa9e4066Sahrens 
2694e05725b1Sbonwick 		VERIFY(zap_lookup(os, object, propname, zl_intsize,
2695e05725b1Sbonwick 		    zl_ints, value) == 0);
2696fa9e4066Sahrens 
2697e05725b1Sbonwick 		for (i = 0; i < ints; i++) {
2698e05725b1Sbonwick 			ASSERT3U(value[i], ==, last_txg + object + i);
2699fa9e4066Sahrens 		}
2700e05725b1Sbonwick 	} else {
2701e05725b1Sbonwick 		ASSERT3U(error, ==, ENOENT);
2702e05725b1Sbonwick 	}
2703fa9e4066Sahrens 
2704e05725b1Sbonwick 	/*
2705e05725b1Sbonwick 	 * Atomically update two entries in our zap object.
2706e05725b1Sbonwick 	 * The first is named txg_%llu, and contains the txg
2707e05725b1Sbonwick 	 * in which the property was last updated.  The second
2708e05725b1Sbonwick 	 * is named prop_%llu, and the nth element of its value
2709e05725b1Sbonwick 	 * should be txg + object + n.
2710e05725b1Sbonwick 	 */
2711e05725b1Sbonwick 	tx = dmu_tx_create(os);
2712e05725b1Sbonwick 	dmu_tx_hold_zap(tx, object, TRUE, NULL);
2713e05725b1Sbonwick 	error = dmu_tx_assign(tx, TXG_WAIT);
2714e05725b1Sbonwick 	if (error) {
2715e05725b1Sbonwick 		ztest_record_enospc("create zap entry");
2716e05725b1Sbonwick 		dmu_tx_abort(tx);
2717e05725b1Sbonwick 		return;
2718e05725b1Sbonwick 	}
2719e05725b1Sbonwick 	txg = dmu_tx_get_txg(tx);
2720fa9e4066Sahrens 
2721e05725b1Sbonwick 	if (last_txg > txg)
2722e05725b1Sbonwick 		fatal(0, "zap future leak: old %llu new %llu", last_txg, txg);
2723fa9e4066Sahrens 
2724e05725b1Sbonwick 	for (i = 0; i < ints; i++)
2725e05725b1Sbonwick 		value[i] = txg + object + i;
2726fa9e4066Sahrens 
2727e05725b1Sbonwick 	error = zap_update(os, object, txgname, sizeof (uint64_t), 1, &txg, tx);
2728e05725b1Sbonwick 	if (error)
2729e05725b1Sbonwick 		fatal(0, "zap_update('%s', %llu, '%s') = %d",
2730e05725b1Sbonwick 		    osname, object, txgname, error);
2731fa9e4066Sahrens 
2732e05725b1Sbonwick 	error = zap_update(os, object, propname, sizeof (uint64_t),
2733e05725b1Sbonwick 	    ints, value, tx);
2734e05725b1Sbonwick 	if (error)
2735e05725b1Sbonwick 		fatal(0, "zap_update('%s', %llu, '%s') = %d",
2736e05725b1Sbonwick 		    osname, object, propname, error);
2737fa9e4066Sahrens 
2738e05725b1Sbonwick 	dmu_tx_commit(tx);
2739fa9e4066Sahrens 
2740e05725b1Sbonwick 	/*
2741e05725b1Sbonwick 	 * Remove a random pair of entries.
2742e05725b1Sbonwick 	 */
2743e05725b1Sbonwick 	prop = ztest_random(ZTEST_ZAP_MAX_PROPS);
2744e05725b1Sbonwick 	(void) sprintf(propname, "prop_%llu", (u_longlong_t)prop);
2745e05725b1Sbonwick 	(void) sprintf(txgname, "txg_%llu", (u_longlong_t)prop);
2746fa9e4066Sahrens 
2747e05725b1Sbonwick 	error = zap_length(os, object, txgname, &zl_intsize, &zl_ints);
2748fa9e4066Sahrens 
2749e05725b1Sbonwick 	if (error == ENOENT)
2750e05725b1Sbonwick 		return;
2751fa9e4066Sahrens 
2752e05725b1Sbonwick 	ASSERT3U(error, ==, 0);
2753fa9e4066Sahrens 
2754e05725b1Sbonwick 	tx = dmu_tx_create(os);
2755e05725b1Sbonwick 	dmu_tx_hold_zap(tx, object, TRUE, NULL);
2756e05725b1Sbonwick 	error = dmu_tx_assign(tx, TXG_WAIT);
2757e05725b1Sbonwick 	if (error) {
2758e05725b1Sbonwick 		ztest_record_enospc("remove zap entry");
2759e05725b1Sbonwick 		dmu_tx_abort(tx);
2760e05725b1Sbonwick 		return;
2761e05725b1Sbonwick 	}
2762e05725b1Sbonwick 	error = zap_remove(os, object, txgname, tx);
2763e05725b1Sbonwick 	if (error)
2764e05725b1Sbonwick 		fatal(0, "zap_remove('%s', %llu, '%s') = %d",
2765e05725b1Sbonwick 		    osname, object, txgname, error);
2766fa9e4066Sahrens 
2767e05725b1Sbonwick 	error = zap_remove(os, object, propname, tx);
2768e05725b1Sbonwick 	if (error)
2769e05725b1Sbonwick 		fatal(0, "zap_remove('%s', %llu, '%s') = %d",
2770e05725b1Sbonwick 		    osname, object, propname, error);
2771fa9e4066Sahrens 
2772e05725b1Sbonwick 	dmu_tx_commit(tx);
2773fa9e4066Sahrens 
2774fa9e4066Sahrens 	/*
2775fa9e4066Sahrens 	 * Once in a while, destroy the object.
2776fa9e4066Sahrens 	 */
2777e05725b1Sbonwick 	if (ztest_random(1000) != 0)
2778fa9e4066Sahrens 		return;
2779fa9e4066Sahrens 
2780fa9e4066Sahrens 	tx = dmu_tx_create(os);
2781fa9e4066Sahrens 	dmu_tx_hold_write(tx, ZTEST_DIROBJ, za->za_diroff, sizeof (uint64_t));
2782fa9e4066Sahrens 	dmu_tx_hold_free(tx, object, 0, DMU_OBJECT_END);
2783fa9e4066Sahrens 	error = dmu_tx_assign(tx, TXG_WAIT);
2784fa9e4066Sahrens 	if (error) {
2785fa9e4066Sahrens 		ztest_record_enospc("destroy zap object");
2786fa9e4066Sahrens 		dmu_tx_abort(tx);
2787fa9e4066Sahrens 		return;
2788fa9e4066Sahrens 	}
2789fa9e4066Sahrens 	error = zap_destroy(os, object, tx);
2790fa9e4066Sahrens 	if (error)
2791fa9e4066Sahrens 		fatal(0, "zap_destroy('%s', %llu) = %d",
2792fa9e4066Sahrens 		    osname, object, error);
2793fa9e4066Sahrens 	object = 0;
2794fa9e4066Sahrens 	dmu_write(os, ZTEST_DIROBJ, za->za_diroff, sizeof (uint64_t),
2795fa9e4066Sahrens 	    &object, tx);
2796fa9e4066Sahrens 	dmu_tx_commit(tx);
2797fa9e4066Sahrens }
2798fa9e4066Sahrens 
2799fa9e4066Sahrens void
2800fa9e4066Sahrens ztest_zap_parallel(ztest_args_t *za)
2801fa9e4066Sahrens {
2802fa9e4066Sahrens 	objset_t *os = za->za_os;
2803fa9e4066Sahrens 	uint64_t txg, object, count, wsize, wc, zl_wsize, zl_wc;
2804fa9e4066Sahrens 	dmu_tx_t *tx;
2805fa9e4066Sahrens 	int i, namelen, error;
2806fa9e4066Sahrens 	char name[20], string_value[20];
2807fa9e4066Sahrens 	void *data;
2808fa9e4066Sahrens 
2809e05725b1Sbonwick 	/*
2810e05725b1Sbonwick 	 * Generate a random name of the form 'xxx.....' where each
2811e05725b1Sbonwick 	 * x is a random printable character and the dots are dots.
2812e05725b1Sbonwick 	 * There are 94 such characters, and the name length goes from
2813e05725b1Sbonwick 	 * 6 to 20, so there are 94^3 * 15 = 12,458,760 possible names.
2814e05725b1Sbonwick 	 */
2815e05725b1Sbonwick 	namelen = ztest_random(sizeof (name) - 5) + 5 + 1;
2816fa9e4066Sahrens 
2817e05725b1Sbonwick 	for (i = 0; i < 3; i++)
2818e05725b1Sbonwick 		name[i] = '!' + ztest_random('~' - '!' + 1);
2819e05725b1Sbonwick 	for (; i < namelen - 1; i++)
2820e05725b1Sbonwick 		name[i] = '.';
2821e05725b1Sbonwick 	name[i] = '\0';
2822fa9e4066Sahrens 
2823e05725b1Sbonwick 	if (ztest_random(2) == 0)
2824e05725b1Sbonwick 		object = ZTEST_MICROZAP_OBJ;
2825e05725b1Sbonwick 	else
2826e05725b1Sbonwick 		object = ZTEST_FATZAP_OBJ;
2827fa9e4066Sahrens 
2828e05725b1Sbonwick 	if ((namelen & 1) || object == ZTEST_MICROZAP_OBJ) {
2829e05725b1Sbonwick 		wsize = sizeof (txg);
2830e05725b1Sbonwick 		wc = 1;
2831e05725b1Sbonwick 		data = &txg;
2832e05725b1Sbonwick 	} else {
2833e05725b1Sbonwick 		wsize = 1;
2834e05725b1Sbonwick 		wc = namelen;
2835e05725b1Sbonwick 		data = string_value;
2836e05725b1Sbonwick 	}
2837fa9e4066Sahrens 
2838e05725b1Sbonwick 	count = -1ULL;
2839e05725b1Sbonwick 	VERIFY(zap_count(os, object, &count) == 0);
2840e05725b1Sbonwick 	ASSERT(count != -1ULL);
2841fa9e4066Sahrens 
2842e05725b1Sbonwick 	/*
2843e05725b1Sbonwick 	 * Select an operation: length, lookup, add, update, remove.
2844e05725b1Sbonwick 	 */
2845e05725b1Sbonwick 	i = ztest_random(5);
2846e05725b1Sbonwick 
2847e05725b1Sbonwick 	if (i >= 2) {
2848e05725b1Sbonwick 		tx = dmu_tx_create(os);
2849e05725b1Sbonwick 		dmu_tx_hold_zap(tx, object, TRUE, NULL);
2850e05725b1Sbonwick 		error = dmu_tx_assign(tx, TXG_WAIT);
2851e05725b1Sbonwick 		if (error) {
2852e05725b1Sbonwick 			ztest_record_enospc("zap parallel");
2853e05725b1Sbonwick 			dmu_tx_abort(tx);
2854e05725b1Sbonwick 			return;
2855fa9e4066Sahrens 		}
2856e05725b1Sbonwick 		txg = dmu_tx_get_txg(tx);
2857e05725b1Sbonwick 		bcopy(name, string_value, namelen);
2858e05725b1Sbonwick 	} else {
2859e05725b1Sbonwick 		tx = NULL;
2860e05725b1Sbonwick 		txg = 0;
2861e05725b1Sbonwick 		bzero(string_value, namelen);
2862e05725b1Sbonwick 	}
2863fa9e4066Sahrens 
2864e05725b1Sbonwick 	switch (i) {
2865fa9e4066Sahrens 
2866e05725b1Sbonwick 	case 0:
2867e05725b1Sbonwick 		error = zap_length(os, object, name, &zl_wsize, &zl_wc);
2868e05725b1Sbonwick 		if (error == 0) {
2869e05725b1Sbonwick 			ASSERT3U(wsize, ==, zl_wsize);
2870e05725b1Sbonwick 			ASSERT3U(wc, ==, zl_wc);
2871e05725b1Sbonwick 		} else {
2872e05725b1Sbonwick 			ASSERT3U(error, ==, ENOENT);
2873e05725b1Sbonwick 		}
2874e05725b1Sbonwick 		break;
2875fa9e4066Sahrens 
2876e05725b1Sbonwick 	case 1:
2877e05725b1Sbonwick 		error = zap_lookup(os, object, name, wsize, wc, data);
2878e05725b1Sbonwick 		if (error == 0) {
2879e05725b1Sbonwick 			if (data == string_value &&
2880e05725b1Sbonwick 			    bcmp(name, data, namelen) != 0)
2881e05725b1Sbonwick 				fatal(0, "name '%s' != val '%s' len %d",
2882e05725b1Sbonwick 				    name, data, namelen);
2883e05725b1Sbonwick 		} else {
2884e05725b1Sbonwick 			ASSERT3U(error, ==, ENOENT);
2885e05725b1Sbonwick 		}
2886e05725b1Sbonwick 		break;
2887fa9e4066Sahrens 
2888e05725b1Sbonwick 	case 2:
2889e05725b1Sbonwick 		error = zap_add(os, object, name, wsize, wc, data, tx);
2890e05725b1Sbonwick 		ASSERT(error == 0 || error == EEXIST);
2891e05725b1Sbonwick 		break;
2892fa9e4066Sahrens 
2893e05725b1Sbonwick 	case 3:
2894e05725b1Sbonwick 		VERIFY(zap_update(os, object, name, wsize, wc, data, tx) == 0);
2895e05725b1Sbonwick 		break;
2896fa9e4066Sahrens 
2897e05725b1Sbonwick 	case 4:
2898e05725b1Sbonwick 		error = zap_remove(os, object, name, tx);
2899e05725b1Sbonwick 		ASSERT(error == 0 || error == ENOENT);
2900e05725b1Sbonwick 		break;
2901fa9e4066Sahrens 	}
2902e05725b1Sbonwick 
2903e05725b1Sbonwick 	if (tx != NULL)
2904e05725b1Sbonwick 		dmu_tx_commit(tx);
2905fa9e4066Sahrens }
2906fa9e4066Sahrens 
2907fa9e4066Sahrens void
2908fa9e4066Sahrens ztest_dsl_prop_get_set(ztest_args_t *za)
2909fa9e4066Sahrens {
2910fa9e4066Sahrens 	objset_t *os = za->za_os;
2911fa9e4066Sahrens 	int i, inherit;
2912fa9e4066Sahrens 	uint64_t value;
2913fa9e4066Sahrens 	const char *prop, *valname;
2914fa9e4066Sahrens 	char setpoint[MAXPATHLEN];
2915fa9e4066Sahrens 	char osname[MAXNAMELEN];
2916ea8dc4b6Seschrock 	int error;
2917fa9e4066Sahrens 
2918fa9e4066Sahrens 	(void) rw_rdlock(&ztest_shared->zs_name_lock);
2919fa9e4066Sahrens 
2920fa9e4066Sahrens 	dmu_objset_name(os, osname);
2921fa9e4066Sahrens 
2922fa9e4066Sahrens 	for (i = 0; i < 2; i++) {
2923fa9e4066Sahrens 		if (i == 0) {
2924fa9e4066Sahrens 			prop = "checksum";
2925fa9e4066Sahrens 			value = ztest_random_checksum();
2926fa9e4066Sahrens 			inherit = (value == ZIO_CHECKSUM_INHERIT);
2927fa9e4066Sahrens 		} else {
2928fa9e4066Sahrens 			prop = "compression";
2929fa9e4066Sahrens 			value = ztest_random_compress();
2930fa9e4066Sahrens 			inherit = (value == ZIO_COMPRESS_INHERIT);
2931fa9e4066Sahrens 		}
2932fa9e4066Sahrens 
2933ea8dc4b6Seschrock 		error = dsl_prop_set(osname, prop, sizeof (value),
2934ea8dc4b6Seschrock 		    !inherit, &value);
2935ea8dc4b6Seschrock 
2936ea8dc4b6Seschrock 		if (error == ENOSPC) {
2937ea8dc4b6Seschrock 			ztest_record_enospc("dsl_prop_set");
2938ea8dc4b6Seschrock 			break;
2939ea8dc4b6Seschrock 		}
2940ea8dc4b6Seschrock 
2941ea8dc4b6Seschrock 		ASSERT3U(error, ==, 0);
2942fa9e4066Sahrens 
2943fa9e4066Sahrens 		VERIFY3U(dsl_prop_get(osname, prop, sizeof (value),
2944fa9e4066Sahrens 		    1, &value, setpoint), ==, 0);
2945fa9e4066Sahrens 
2946fa9e4066Sahrens 		if (i == 0)
2947fa9e4066Sahrens 			valname = zio_checksum_table[value].ci_name;
2948fa9e4066Sahrens 		else
2949fa9e4066Sahrens 			valname = zio_compress_table[value].ci_name;
2950fa9e4066Sahrens 
2951fa9e4066Sahrens 		if (zopt_verbose >= 6) {
2952fa9e4066Sahrens 			(void) printf("%s %s = %s for '%s'\n",
2953fa9e4066Sahrens 			    osname, prop, valname, setpoint);
2954fa9e4066Sahrens 		}
2955fa9e4066Sahrens 	}
2956fa9e4066Sahrens 
2957fa9e4066Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
2958fa9e4066Sahrens }
2959fa9e4066Sahrens 
2960fa9e4066Sahrens /*
2961fa9e4066Sahrens  * Inject random faults into the on-disk data.
2962fa9e4066Sahrens  */
2963fa9e4066Sahrens void
2964fa9e4066Sahrens ztest_fault_inject(ztest_args_t *za)
2965fa9e4066Sahrens {
2966fa9e4066Sahrens 	int fd;
2967fa9e4066Sahrens 	uint64_t offset;
2968fa9e4066Sahrens 	uint64_t leaves = MAX(zopt_mirrors, 1) * zopt_raidz;
2969fa9e4066Sahrens 	uint64_t bad = 0x1990c0ffeedecade;
2970fa9e4066Sahrens 	uint64_t top, leaf;
2971fa9e4066Sahrens 	char path0[MAXPATHLEN];
2972fa9e4066Sahrens 	char pathrand[MAXPATHLEN];
2973fa9e4066Sahrens 	size_t fsize;
2974e05725b1Sbonwick 	spa_t *spa = za->za_spa;
2975fa9e4066Sahrens 	int bshift = SPA_MAXBLOCKSHIFT + 2;	/* don't scrog all labels */
2976fa9e4066Sahrens 	int iters = 1000;
2977e14bb325SJeff Bonwick 	int maxfaults = zopt_maxfaults;
2978e14bb325SJeff Bonwick 	vdev_t *vd0 = NULL;
2979ea8dc4b6Seschrock 	uint64_t guid0 = 0;
2980ea8dc4b6Seschrock 
2981e14bb325SJeff Bonwick 	ASSERT(leaves >= 1);
2982fa9e4066Sahrens 
2983fa9e4066Sahrens 	/*
2984e14bb325SJeff Bonwick 	 * We need SCL_STATE here because we're going to look at vd0->vdev_tsd.
2985fa9e4066Sahrens 	 */
2986e14bb325SJeff Bonwick 	spa_config_enter(spa, SCL_STATE, FTAG, RW_READER);
2987fa9e4066Sahrens 
2988e14bb325SJeff Bonwick 	if (ztest_random(2) == 0) {
2989e14bb325SJeff Bonwick 		/*
2990e14bb325SJeff Bonwick 		 * Inject errors on a normal data device.
2991e14bb325SJeff Bonwick 		 */
2992e14bb325SJeff Bonwick 		top = ztest_random(spa->spa_root_vdev->vdev_children);
2993e14bb325SJeff Bonwick 		leaf = ztest_random(leaves);
2994fa9e4066Sahrens 
2995e14bb325SJeff Bonwick 		/*
2996e14bb325SJeff Bonwick 		 * Generate paths to the first leaf in this top-level vdev,
2997e14bb325SJeff Bonwick 		 * and to the random leaf we selected.  We'll induce transient
2998e14bb325SJeff Bonwick 		 * write failures and random online/offline activity on leaf 0,
2999e14bb325SJeff Bonwick 		 * and we'll write random garbage to the randomly chosen leaf.
3000e14bb325SJeff Bonwick 		 */
3001e14bb325SJeff Bonwick 		(void) snprintf(path0, sizeof (path0), ztest_dev_template,
3002e14bb325SJeff Bonwick 		    zopt_dir, zopt_pool, top * leaves + 0);
3003e14bb325SJeff Bonwick 		(void) snprintf(pathrand, sizeof (pathrand), ztest_dev_template,
3004e14bb325SJeff Bonwick 		    zopt_dir, zopt_pool, top * leaves + leaf);
3005fa9e4066Sahrens 
3006e14bb325SJeff Bonwick 		vd0 = vdev_lookup_by_path(spa->spa_root_vdev, path0);
3007e14bb325SJeff Bonwick 		if (vd0 != NULL && maxfaults != 1) {
3008e14bb325SJeff Bonwick 			/*
3009e14bb325SJeff Bonwick 			 * Make vd0 explicitly claim to be unreadable,
3010e14bb325SJeff Bonwick 			 * or unwriteable, or reach behind its back
3011e14bb325SJeff Bonwick 			 * and close the underlying fd.  We can do this if
3012e14bb325SJeff Bonwick 			 * maxfaults == 0 because we'll fail and reexecute,
3013e14bb325SJeff Bonwick 			 * and we can do it if maxfaults >= 2 because we'll
3014e14bb325SJeff Bonwick 			 * have enough redundancy.  If maxfaults == 1, the
3015e14bb325SJeff Bonwick 			 * combination of this with injection of random data
3016e14bb325SJeff Bonwick 			 * corruption below exceeds the pool's fault tolerance.
3017e14bb325SJeff Bonwick 			 */
3018e14bb325SJeff Bonwick 			vdev_file_t *vf = vd0->vdev_tsd;
3019e14bb325SJeff Bonwick 
3020e14bb325SJeff Bonwick 			if (vf != NULL && ztest_random(3) == 0) {
3021e14bb325SJeff Bonwick 				(void) close(vf->vf_vnode->v_fd);
3022e14bb325SJeff Bonwick 				vf->vf_vnode->v_fd = -1;
3023e14bb325SJeff Bonwick 			} else if (ztest_random(2) == 0) {
3024e14bb325SJeff Bonwick 				vd0->vdev_cant_read = B_TRUE;
3025e14bb325SJeff Bonwick 			} else {
3026e14bb325SJeff Bonwick 				vd0->vdev_cant_write = B_TRUE;
3027e14bb325SJeff Bonwick 			}
3028e14bb325SJeff Bonwick 			guid0 = vd0->vdev_guid;
3029e14bb325SJeff Bonwick 		}
3030e14bb325SJeff Bonwick 	} else {
3031e14bb325SJeff Bonwick 		/*
3032e14bb325SJeff Bonwick 		 * Inject errors on an l2cache device.
3033e14bb325SJeff Bonwick 		 */
3034e14bb325SJeff Bonwick 		spa_aux_vdev_t *sav = &spa->spa_l2cache;
3035fa9e4066Sahrens 
3036e14bb325SJeff Bonwick 		if (sav->sav_count == 0) {
3037e14bb325SJeff Bonwick 			spa_config_exit(spa, SCL_STATE, FTAG);
3038e14bb325SJeff Bonwick 			return;
3039e14bb325SJeff Bonwick 		}
3040e14bb325SJeff Bonwick 		vd0 = sav->sav_vdevs[ztest_random(sav->sav_count)];
3041ea8dc4b6Seschrock 		guid0 = vd0->vdev_guid;
3042e14bb325SJeff Bonwick 		(void) strcpy(path0, vd0->vdev_path);
3043e14bb325SJeff Bonwick 		(void) strcpy(pathrand, vd0->vdev_path);
3044e14bb325SJeff Bonwick 
3045e14bb325SJeff Bonwick 		leaf = 0;
3046e14bb325SJeff Bonwick 		leaves = 1;
3047e14bb325SJeff Bonwick 		maxfaults = INT_MAX;	/* no limit on cache devices */
3048fa9e4066Sahrens 	}
3049fa9e4066Sahrens 
3050e14bb325SJeff Bonwick 	spa_config_exit(spa, SCL_STATE, FTAG);
3051e14bb325SJeff Bonwick 
3052e14bb325SJeff Bonwick 	if (maxfaults == 0)
3053e14bb325SJeff Bonwick 		return;
3054fa9e4066Sahrens 
3055fa9e4066Sahrens 	/*
3056ea8dc4b6Seschrock 	 * If we can tolerate two or more faults, randomly online/offline vd0.
3057fa9e4066Sahrens 	 */
3058e14bb325SJeff Bonwick 	if (maxfaults >= 2 && guid0 != 0) {
30598ad4d6ddSJeff Bonwick 		if (ztest_random(10) < 6) {
30608ad4d6ddSJeff Bonwick 			int flags = (ztest_random(2) == 0 ?
30618ad4d6ddSJeff Bonwick 			    ZFS_OFFLINE_TEMPORARY : 0);
30628ad4d6ddSJeff Bonwick 			VERIFY(vdev_offline(spa, guid0, flags) != EBUSY);
30638ad4d6ddSJeff Bonwick 		} else {
30648ad4d6ddSJeff Bonwick 			(void) vdev_online(spa, guid0, 0, NULL);
30658ad4d6ddSJeff Bonwick 		}
3066fa9e4066Sahrens 	}
3067fa9e4066Sahrens 
3068fa9e4066Sahrens 	/*
3069ea8dc4b6Seschrock 	 * We have at least single-fault tolerance, so inject data corruption.
3070fa9e4066Sahrens 	 */
3071fa9e4066Sahrens 	fd = open(pathrand, O_RDWR);
3072fa9e4066Sahrens 
3073fa9e4066Sahrens 	if (fd == -1)	/* we hit a gap in the device namespace */
3074fa9e4066Sahrens 		return;
3075fa9e4066Sahrens 
3076fa9e4066Sahrens 	fsize = lseek(fd, 0, SEEK_END);
3077fa9e4066Sahrens 
3078fa9e4066Sahrens 	while (--iters != 0) {
3079fa9e4066Sahrens 		offset = ztest_random(fsize / (leaves << bshift)) *
3080fa9e4066Sahrens 		    (leaves << bshift) + (leaf << bshift) +
3081fa9e4066Sahrens 		    (ztest_random(1ULL << (bshift - 1)) & -8ULL);
3082fa9e4066Sahrens 
3083fa9e4066Sahrens 		if (offset >= fsize)
3084fa9e4066Sahrens 			continue;
3085fa9e4066Sahrens 
3086fa9e4066Sahrens 		if (zopt_verbose >= 6)
3087fa9e4066Sahrens 			(void) printf("injecting bad word into %s,"
3088fa9e4066Sahrens 			    " offset 0x%llx\n", pathrand, (u_longlong_t)offset);
3089fa9e4066Sahrens 
3090fa9e4066Sahrens 		if (pwrite(fd, &bad, sizeof (bad), offset) != sizeof (bad))
3091fa9e4066Sahrens 			fatal(1, "can't inject bad word at 0x%llx in %s",
3092fa9e4066Sahrens 			    offset, pathrand);
3093fa9e4066Sahrens 	}
3094fa9e4066Sahrens 
3095fa9e4066Sahrens 	(void) close(fd);
3096fa9e4066Sahrens }
3097fa9e4066Sahrens 
3098fa9e4066Sahrens /*
3099fa9e4066Sahrens  * Scrub the pool.
3100fa9e4066Sahrens  */
3101fa9e4066Sahrens void
3102fa9e4066Sahrens ztest_scrub(ztest_args_t *za)
3103fa9e4066Sahrens {
3104e05725b1Sbonwick 	spa_t *spa = za->za_spa;
3105fa9e4066Sahrens 
3106088f3894Sahrens 	(void) spa_scrub(spa, POOL_SCRUB_EVERYTHING);
3107fa9e4066Sahrens 	(void) poll(NULL, 0, 1000); /* wait a second, then force a restart */
3108088f3894Sahrens 	(void) spa_scrub(spa, POOL_SCRUB_EVERYTHING);
3109fa9e4066Sahrens }
3110fa9e4066Sahrens 
3111fa9e4066Sahrens /*
3112fa9e4066Sahrens  * Rename the pool to a different name and then rename it back.
3113fa9e4066Sahrens  */
3114fa9e4066Sahrens void
3115fa9e4066Sahrens ztest_spa_rename(ztest_args_t *za)
3116fa9e4066Sahrens {
3117fa9e4066Sahrens 	char *oldname, *newname;
3118fa9e4066Sahrens 	int error;
3119fa9e4066Sahrens 	spa_t *spa;
3120fa9e4066Sahrens 
3121fa9e4066Sahrens 	(void) rw_wrlock(&ztest_shared->zs_name_lock);
3122fa9e4066Sahrens 
3123fa9e4066Sahrens 	oldname = za->za_pool;
3124fa9e4066Sahrens 	newname = umem_alloc(strlen(oldname) + 5, UMEM_NOFAIL);
3125fa9e4066Sahrens 	(void) strcpy(newname, oldname);
3126fa9e4066Sahrens 	(void) strcat(newname, "_tmp");
3127fa9e4066Sahrens 
3128fa9e4066Sahrens 	/*
3129fa9e4066Sahrens 	 * Do the rename
3130fa9e4066Sahrens 	 */
3131fa9e4066Sahrens 	error = spa_rename(oldname, newname);
3132fa9e4066Sahrens 	if (error)
3133fa9e4066Sahrens 		fatal(0, "spa_rename('%s', '%s') = %d", oldname,
3134fa9e4066Sahrens 		    newname, error);
3135fa9e4066Sahrens 
3136fa9e4066Sahrens 	/*
3137fa9e4066Sahrens 	 * Try to open it under the old name, which shouldn't exist
3138fa9e4066Sahrens 	 */
3139fa9e4066Sahrens 	error = spa_open(oldname, &spa, FTAG);
3140fa9e4066Sahrens 	if (error != ENOENT)
3141fa9e4066Sahrens 		fatal(0, "spa_open('%s') = %d", oldname, error);
3142fa9e4066Sahrens 
3143fa9e4066Sahrens 	/*
3144fa9e4066Sahrens 	 * Open it under the new name and make sure it's still the same spa_t.
3145fa9e4066Sahrens 	 */
3146fa9e4066Sahrens 	error = spa_open(newname, &spa, FTAG);
3147fa9e4066Sahrens 	if (error != 0)
3148fa9e4066Sahrens 		fatal(0, "spa_open('%s') = %d", newname, error);
3149fa9e4066Sahrens 
3150e05725b1Sbonwick 	ASSERT(spa == za->za_spa);
3151fa9e4066Sahrens 	spa_close(spa, FTAG);
3152fa9e4066Sahrens 
3153fa9e4066Sahrens 	/*
3154fa9e4066Sahrens 	 * Rename it back to the original
3155fa9e4066Sahrens 	 */
3156fa9e4066Sahrens 	error = spa_rename(newname, oldname);
3157fa9e4066Sahrens 	if (error)
3158fa9e4066Sahrens 		fatal(0, "spa_rename('%s', '%s') = %d", newname,
3159fa9e4066Sahrens 		    oldname, error);
3160fa9e4066Sahrens 
3161fa9e4066Sahrens 	/*
3162fa9e4066Sahrens 	 * Make sure it can still be opened
3163fa9e4066Sahrens 	 */
3164fa9e4066Sahrens 	error = spa_open(oldname, &spa, FTAG);
3165fa9e4066Sahrens 	if (error != 0)
3166fa9e4066Sahrens 		fatal(0, "spa_open('%s') = %d", oldname, error);
3167fa9e4066Sahrens 
3168e05725b1Sbonwick 	ASSERT(spa == za->za_spa);
3169fa9e4066Sahrens 	spa_close(spa, FTAG);
3170fa9e4066Sahrens 
3171fa9e4066Sahrens 	umem_free(newname, strlen(newname) + 1);
3172fa9e4066Sahrens 
3173fa9e4066Sahrens 	(void) rw_unlock(&ztest_shared->zs_name_lock);
3174fa9e4066Sahrens }
3175fa9e4066Sahrens 
3176fa9e4066Sahrens 
3177fa9e4066Sahrens /*
3178fa9e4066Sahrens  * Completely obliterate one disk.
3179fa9e4066Sahrens  */
3180fa9e4066Sahrens static void
3181fa9e4066Sahrens ztest_obliterate_one_disk(uint64_t vdev)
3182fa9e4066Sahrens {
3183fa9e4066Sahrens 	int fd;
3184d80c45e0Sbonwick 	char dev_name[MAXPATHLEN], copy_name[MAXPATHLEN];
3185fa9e4066Sahrens 	size_t fsize;
3186fa9e4066Sahrens 
3187fa9e4066Sahrens 	if (zopt_maxfaults < 2)
3188fa9e4066Sahrens 		return;
3189fa9e4066Sahrens 
3190fa9e4066Sahrens 	(void) sprintf(dev_name, ztest_dev_template, zopt_dir, zopt_pool, vdev);
3191d80c45e0Sbonwick 	(void) snprintf(copy_name, MAXPATHLEN, "%s.old", dev_name);
3192fa9e4066Sahrens 
3193fa9e4066Sahrens 	fd = open(dev_name, O_RDWR);
3194fa9e4066Sahrens 
3195fa9e4066Sahrens 	if (fd == -1)
3196fa9e4066Sahrens 		fatal(1, "can't open %s", dev_name);
3197fa9e4066Sahrens 
3198fa9e4066Sahrens 	/*
3199fa9e4066Sahrens 	 * Determine the size.
3200fa9e4066Sahrens 	 */
3201fa9e4066Sahrens 	fsize = lseek(fd, 0, SEEK_END);
3202d80c45e0Sbonwick 
3203fa9e4066Sahrens 	(void) close(fd);
3204fa9e4066Sahrens 
3205fa9e4066Sahrens 	/*
3206d80c45e0Sbonwick 	 * Rename the old device to dev_name.old (useful for debugging).
3207fa9e4066Sahrens 	 */
3208d80c45e0Sbonwick 	VERIFY(rename(dev_name, copy_name) == 0);
3209fa9e4066Sahrens 
3210fa9e4066Sahrens 	/*
3211fa9e4066Sahrens 	 * Create a new one.
3212fa9e4066Sahrens 	 */
3213fa9e4066Sahrens 	VERIFY((fd = open(dev_name, O_RDWR | O_CREAT | O_TRUNC, 0666)) >= 0);
3214fa9e4066Sahrens 	VERIFY(ftruncate(fd, fsize) == 0);
3215fa9e4066Sahrens 	(void) close(fd);
3216fa9e4066Sahrens }
3217fa9e4066Sahrens 
3218fa9e4066Sahrens static void
3219fa9e4066Sahrens ztest_replace_one_disk(spa_t *spa, uint64_t vdev)
3220fa9e4066Sahrens {
3221fa9e4066Sahrens 	char dev_name[MAXPATHLEN];
3222e14bb325SJeff Bonwick 	nvlist_t *root;
3223fa9e4066Sahrens 	int error;
3224ea8dc4b6Seschrock 	uint64_t guid;
3225ea8dc4b6Seschrock 	vdev_t *vd;
3226fa9e4066Sahrens 
3227fa9e4066Sahrens 	(void) sprintf(dev_name, ztest_dev_template, zopt_dir, zopt_pool, vdev);
3228fa9e4066Sahrens 
3229fa9e4066Sahrens 	/*
3230fa9e4066Sahrens 	 * Build the nvlist describing dev_name.
3231fa9e4066Sahrens 	 */
3232e14bb325SJeff Bonwick 	root = make_vdev_root(dev_name, NULL, 0, 0, 0, 0, 0, 1);
3233fa9e4066Sahrens 
3234e14bb325SJeff Bonwick 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
3235ea8dc4b6Seschrock 	if ((vd = vdev_lookup_by_path(spa->spa_root_vdev, dev_name)) == NULL)
3236ea8dc4b6Seschrock 		guid = 0;
3237ea8dc4b6Seschrock 	else
3238ea8dc4b6Seschrock 		guid = vd->vdev_guid;
3239e14bb325SJeff Bonwick 	spa_config_exit(spa, SCL_VDEV, FTAG);
3240ea8dc4b6Seschrock 	error = spa_vdev_attach(spa, guid, root, B_TRUE);
3241ecc2d604Sbonwick 	if (error != 0 &&
3242ecc2d604Sbonwick 	    error != EBUSY &&
3243ecc2d604Sbonwick 	    error != ENOTSUP &&
3244ecc2d604Sbonwick 	    error != ENODEV &&
3245ecc2d604Sbonwick 	    error != EDOM)
3246fa9e4066Sahrens 		fatal(0, "spa_vdev_attach(in-place) = %d", error);
3247fa9e4066Sahrens 
3248fa9e4066Sahrens 	nvlist_free(root);
3249fa9e4066Sahrens }
3250fa9e4066Sahrens 
3251fa9e4066Sahrens static void
3252fa9e4066Sahrens ztest_verify_blocks(char *pool)
3253fa9e4066Sahrens {
3254fa9e4066Sahrens 	int status;
3255fa9e4066Sahrens 	char zdb[MAXPATHLEN + MAXNAMELEN + 20];
3256fa9e4066Sahrens 	char zbuf[1024];
3257fa9e4066Sahrens 	char *bin;
32588654d025Sperrin 	char *ztest;
32598654d025Sperrin 	char *isa;
32608654d025Sperrin 	int isalen;
3261fa9e4066Sahrens 	FILE *fp;
3262fa9e4066Sahrens 
3263fa9e4066Sahrens 	(void) realpath(getexecname(), zdb);
3264fa9e4066Sahrens 
3265fa9e4066Sahrens 	/* zdb lives in /usr/sbin, while ztest lives in /usr/bin */
3266fa9e4066Sahrens 	bin = strstr(zdb, "/usr/bin/");
32678654d025Sperrin 	ztest = strstr(bin, "/ztest");
32688654d025Sperrin 	isa = bin + 8;
32698654d025Sperrin 	isalen = ztest - isa;
32708654d025Sperrin 	isa = strdup(isa);
3271fa9e4066Sahrens 	/* LINTED */
32723a57275aSck 	(void) sprintf(bin,
327388b7b0f2SMatthew Ahrens 	    "/usr/sbin%.*s/zdb -bc%s%s -U /tmp/zpool.cache %s",
32748654d025Sperrin 	    isalen,
32758654d025Sperrin 	    isa,
3276fa9e4066Sahrens 	    zopt_verbose >= 3 ? "s" : "",
3277fa9e4066Sahrens 	    zopt_verbose >= 4 ? "v" : "",
327888b7b0f2SMatthew Ahrens 	    pool);
32798654d025Sperrin 	free(isa);
3280fa9e4066Sahrens 
3281fa9e4066Sahrens 	if (zopt_verbose >= 5)
3282fa9e4066Sahrens 		(void) printf("Executing %s\n", strstr(zdb, "zdb "));
3283fa9e4066Sahrens 
3284fa9e4066Sahrens 	fp = popen(zdb, "r");
3285fa9e4066Sahrens 
3286fa9e4066Sahrens 	while (fgets(zbuf, sizeof (zbuf), fp) != NULL)
3287fa9e4066Sahrens 		if (zopt_verbose >= 3)
3288fa9e4066Sahrens 			(void) printf("%s", zbuf);
3289fa9e4066Sahrens 
3290fa9e4066Sahrens 	status = pclose(fp);
3291fa9e4066Sahrens 
3292fa9e4066Sahrens 	if (status == 0)
3293fa9e4066Sahrens 		return;
3294fa9e4066Sahrens 
3295fa9e4066Sahrens 	ztest_dump_core = 0;
3296fa9e4066Sahrens 	if (WIFEXITED(status))
3297fa9e4066Sahrens 		fatal(0, "'%s' exit code %d", zdb, WEXITSTATUS(status));
3298fa9e4066Sahrens 	else
3299fa9e4066Sahrens 		fatal(0, "'%s' died with signal %d", zdb, WTERMSIG(status));
3300fa9e4066Sahrens }
3301fa9e4066Sahrens 
3302fa9e4066Sahrens static void
3303fa9e4066Sahrens ztest_walk_pool_directory(char *header)
3304fa9e4066Sahrens {
3305fa9e4066Sahrens 	spa_t *spa = NULL;
3306fa9e4066Sahrens 
3307fa9e4066Sahrens 	if (zopt_verbose >= 6)
3308fa9e4066Sahrens 		(void) printf("%s\n", header);
3309fa9e4066Sahrens 
3310fa9e4066Sahrens 	mutex_enter(&spa_namespace_lock);
3311fa9e4066Sahrens 	while ((spa = spa_next(spa)) != NULL)
3312fa9e4066Sahrens 		if (zopt_verbose >= 6)
3313fa9e4066Sahrens 			(void) printf("\t%s\n", spa_name(spa));
3314fa9e4066Sahrens 	mutex_exit(&spa_namespace_lock);
3315fa9e4066Sahrens }
3316fa9e4066Sahrens 
3317fa9e4066Sahrens static void
3318fa9e4066Sahrens ztest_spa_import_export(char *oldname, char *newname)
3319fa9e4066Sahrens {
33208ad4d6ddSJeff Bonwick 	nvlist_t *config, *newconfig;
3321fa9e4066Sahrens 	uint64_t pool_guid;
3322fa9e4066Sahrens 	spa_t *spa;
3323fa9e4066Sahrens 	int error;
3324fa9e4066Sahrens 
3325fa9e4066Sahrens 	if (zopt_verbose >= 4) {
3326fa9e4066Sahrens 		(void) printf("import/export: old = %s, new = %s\n",
3327fa9e4066Sahrens 		    oldname, newname);
3328fa9e4066Sahrens 	}
3329fa9e4066Sahrens 
3330fa9e4066Sahrens 	/*
3331fa9e4066Sahrens 	 * Clean up from previous runs.
3332fa9e4066Sahrens 	 */
3333fa9e4066Sahrens 	(void) spa_destroy(newname);
3334fa9e4066Sahrens 
3335fa9e4066Sahrens 	/*
3336fa9e4066Sahrens 	 * Get the pool's configuration and guid.
3337fa9e4066Sahrens 	 */
3338fa9e4066Sahrens 	error = spa_open(oldname, &spa, FTAG);
3339fa9e4066Sahrens 	if (error)
3340fa9e4066Sahrens 		fatal(0, "spa_open('%s') = %d", oldname, error);
3341fa9e4066Sahrens 
33428ad4d6ddSJeff Bonwick 	/*
33438ad4d6ddSJeff Bonwick 	 * Kick off a scrub to tickle scrub/export races.
33448ad4d6ddSJeff Bonwick 	 */
33458ad4d6ddSJeff Bonwick 	if (ztest_random(2) == 0)
33468ad4d6ddSJeff Bonwick 		(void) spa_scrub(spa, POOL_SCRUB_EVERYTHING);
33478ad4d6ddSJeff Bonwick 
3348fa9e4066Sahrens 	pool_guid = spa_guid(spa);
3349fa9e4066Sahrens 	spa_close(spa, FTAG);
3350fa9e4066Sahrens 
3351fa9e4066Sahrens 	ztest_walk_pool_directory("pools before export");
3352fa9e4066Sahrens 
3353fa9e4066Sahrens 	/*
3354fa9e4066Sahrens 	 * Export it.
3355fa9e4066Sahrens 	 */
3356394ab0cbSGeorge Wilson 	error = spa_export(oldname, &config, B_FALSE, B_FALSE);
3357fa9e4066Sahrens 	if (error)
3358fa9e4066Sahrens 		fatal(0, "spa_export('%s') = %d", oldname, error);
3359fa9e4066Sahrens 
3360fa9e4066Sahrens 	ztest_walk_pool_directory("pools after export");
3361fa9e4066Sahrens 
33628ad4d6ddSJeff Bonwick 	/*
33638ad4d6ddSJeff Bonwick 	 * Try to import it.
33648ad4d6ddSJeff Bonwick 	 */
33658ad4d6ddSJeff Bonwick 	newconfig = spa_tryimport(config);
33668ad4d6ddSJeff Bonwick 	ASSERT(newconfig != NULL);
33678ad4d6ddSJeff Bonwick 	nvlist_free(newconfig);
33688ad4d6ddSJeff Bonwick 
3369fa9e4066Sahrens 	/*
3370fa9e4066Sahrens 	 * Import it under the new name.
3371fa9e4066Sahrens 	 */
3372fa9e4066Sahrens 	error = spa_import(newname, config, NULL);
3373fa9e4066Sahrens 	if (error)
3374fa9e4066Sahrens 		fatal(0, "spa_import('%s') = %d", newname, error);
3375fa9e4066Sahrens 
3376fa9e4066Sahrens 	ztest_walk_pool_directory("pools after import");
3377fa9e4066Sahrens 
3378fa9e4066Sahrens 	/*
3379fa9e4066Sahrens 	 * Try to import it again -- should fail with EEXIST.
3380fa9e4066Sahrens 	 */
3381fa9e4066Sahrens 	error = spa_import(newname, config, NULL);
3382fa9e4066Sahrens 	if (error != EEXIST)
3383fa9e4066Sahrens 		fatal(0, "spa_import('%s') twice", newname);
3384fa9e4066Sahrens 
3385fa9e4066Sahrens 	/*
3386fa9e4066Sahrens 	 * Try to import it under a different name -- should fail with EEXIST.
3387fa9e4066Sahrens 	 */
3388fa9e4066Sahrens 	error = spa_import(oldname, config, NULL);
3389fa9e4066Sahrens 	if (error != EEXIST)
3390fa9e4066Sahrens 		fatal(0, "spa_import('%s') under multiple names", newname);
3391fa9e4066Sahrens 
3392fa9e4066Sahrens 	/*
3393fa9e4066Sahrens 	 * Verify that the pool is no longer visible under the old name.
3394fa9e4066Sahrens 	 */
3395fa9e4066Sahrens 	error = spa_open(oldname, &spa, FTAG);
3396fa9e4066Sahrens 	if (error != ENOENT)
3397fa9e4066Sahrens 		fatal(0, "spa_open('%s') = %d", newname, error);
3398fa9e4066Sahrens 
3399fa9e4066Sahrens 	/*
3400fa9e4066Sahrens 	 * Verify that we can open and close the pool using the new name.
3401fa9e4066Sahrens 	 */
3402fa9e4066Sahrens 	error = spa_open(newname, &spa, FTAG);
3403fa9e4066Sahrens 	if (error)
3404fa9e4066Sahrens 		fatal(0, "spa_open('%s') = %d", newname, error);
3405fa9e4066Sahrens 	ASSERT(pool_guid == spa_guid(spa));
3406fa9e4066Sahrens 	spa_close(spa, FTAG);
3407fa9e4066Sahrens 
3408fa9e4066Sahrens 	nvlist_free(config);
3409fa9e4066Sahrens }
3410fa9e4066Sahrens 
34118ad4d6ddSJeff Bonwick static void
34128ad4d6ddSJeff Bonwick ztest_resume(spa_t *spa)
34138ad4d6ddSJeff Bonwick {
34148ad4d6ddSJeff Bonwick 	if (spa_suspended(spa)) {
34158ad4d6ddSJeff Bonwick 		spa_vdev_state_enter(spa);
34168ad4d6ddSJeff Bonwick 		vdev_clear(spa, NULL);
34178ad4d6ddSJeff Bonwick 		(void) spa_vdev_state_exit(spa, NULL, 0);
341854d692b7SGeorge Wilson 		(void) zio_resume(spa);
34198ad4d6ddSJeff Bonwick 	}
34208ad4d6ddSJeff Bonwick }
34218ad4d6ddSJeff Bonwick 
34220a4e9518Sgw static void *
34238ad4d6ddSJeff Bonwick ztest_resume_thread(void *arg)
34240a4e9518Sgw {
3425e14bb325SJeff Bonwick 	spa_t *spa = arg;
34260a4e9518Sgw 
34270a4e9518Sgw 	while (!ztest_exiting) {
3428e14bb325SJeff Bonwick 		(void) poll(NULL, 0, 1000);
34298ad4d6ddSJeff Bonwick 		ztest_resume(spa);
34300a4e9518Sgw 	}
34310a4e9518Sgw 	return (NULL);
34320a4e9518Sgw }
34330a4e9518Sgw 
3434fa9e4066Sahrens static void *
3435fa9e4066Sahrens ztest_thread(void *arg)
3436fa9e4066Sahrens {
3437fa9e4066Sahrens 	ztest_args_t *za = arg;
3438fa9e4066Sahrens 	ztest_shared_t *zs = ztest_shared;
3439fa9e4066Sahrens 	hrtime_t now, functime;
3440fa9e4066Sahrens 	ztest_info_t *zi;
3441e05725b1Sbonwick 	int f, i;
3442fa9e4066Sahrens 
3443fa9e4066Sahrens 	while ((now = gethrtime()) < za->za_stop) {
3444fa9e4066Sahrens 		/*
3445fa9e4066Sahrens 		 * See if it's time to force a crash.
3446fa9e4066Sahrens 		 */
3447fa9e4066Sahrens 		if (now > za->za_kill) {
3448e05725b1Sbonwick 			zs->zs_alloc = spa_get_alloc(za->za_spa);
3449e05725b1Sbonwick 			zs->zs_space = spa_get_space(za->za_spa);
3450fa9e4066Sahrens 			(void) kill(getpid(), SIGKILL);
3451fa9e4066Sahrens 		}
3452fa9e4066Sahrens 
3453fa9e4066Sahrens 		/*
3454fa9e4066Sahrens 		 * Pick a random function.
3455fa9e4066Sahrens 		 */
3456fa9e4066Sahrens 		f = ztest_random(ZTEST_FUNCS);
3457fa9e4066Sahrens 		zi = &zs->zs_info[f];
3458fa9e4066Sahrens 
3459fa9e4066Sahrens 		/*
3460fa9e4066Sahrens 		 * Decide whether to call it, based on the requested frequency.
3461fa9e4066Sahrens 		 */
3462fa9e4066Sahrens 		if (zi->zi_call_target == 0 ||
3463fa9e4066Sahrens 		    (double)zi->zi_call_total / zi->zi_call_target >
3464fa9e4066Sahrens 		    (double)(now - zs->zs_start_time) / (zopt_time * NANOSEC))
3465fa9e4066Sahrens 			continue;
3466fa9e4066Sahrens 
3467fa9e4066Sahrens 		atomic_add_64(&zi->zi_calls, 1);
3468fa9e4066Sahrens 		atomic_add_64(&zi->zi_call_total, 1);
3469fa9e4066Sahrens 
3470fa9e4066Sahrens 		za->za_diroff = (za->za_instance * ZTEST_FUNCS + f) *
3471fa9e4066Sahrens 		    ZTEST_DIRSIZE;
3472fa9e4066Sahrens 		za->za_diroff_shared = (1ULL << 63);
3473fa9e4066Sahrens 
3474e05725b1Sbonwick 		for (i = 0; i < zi->zi_iters; i++)
3475e05725b1Sbonwick 			zi->zi_func(za);
3476fa9e4066Sahrens 
3477fa9e4066Sahrens 		functime = gethrtime() - now;
3478fa9e4066Sahrens 
3479fa9e4066Sahrens 		atomic_add_64(&zi->zi_call_time, functime);
3480fa9e4066Sahrens 
3481fa9e4066Sahrens 		if (zopt_verbose >= 4) {
3482fa9e4066Sahrens 			Dl_info dli;
3483fa9e4066Sahrens 			(void) dladdr((void *)zi->zi_func, &dli);
3484fa9e4066Sahrens 			(void) printf("%6.2f sec in %s\n",
3485fa9e4066Sahrens 			    (double)functime / NANOSEC, dli.dli_sname);
3486fa9e4066Sahrens 		}
3487fa9e4066Sahrens 
3488fa9e4066Sahrens 		/*
3489fa9e4066Sahrens 		 * If we're getting ENOSPC with some regularity, stop.
3490fa9e4066Sahrens 		 */
3491fa9e4066Sahrens 		if (zs->zs_enospc_count > 10)
3492fa9e4066Sahrens 			break;
3493fa9e4066Sahrens 	}
3494fa9e4066Sahrens 
3495fa9e4066Sahrens 	return (NULL);
3496fa9e4066Sahrens }
3497fa9e4066Sahrens 
3498fa9e4066Sahrens /*
3499fa9e4066Sahrens  * Kick off threads to run tests on all datasets in parallel.
3500fa9e4066Sahrens  */
3501fa9e4066Sahrens static void
3502fa9e4066Sahrens ztest_run(char *pool)
3503fa9e4066Sahrens {
3504fa9e4066Sahrens 	int t, d, error;
3505fa9e4066Sahrens 	ztest_shared_t *zs = ztest_shared;
3506fa9e4066Sahrens 	ztest_args_t *za;
3507fa9e4066Sahrens 	spa_t *spa;
3508fa9e4066Sahrens 	char name[100];
3509e14bb325SJeff Bonwick 	thread_t resume_tid;
3510e14bb325SJeff Bonwick 
3511e14bb325SJeff Bonwick 	ztest_exiting = B_FALSE;
3512fa9e4066Sahrens 
3513fa9e4066Sahrens 	(void) _mutex_init(&zs->zs_vdev_lock, USYNC_THREAD, NULL);
3514fa9e4066Sahrens 	(void) rwlock_init(&zs->zs_name_lock, USYNC_THREAD, NULL);
3515fa9e4066Sahrens 
3516fa9e4066Sahrens 	for (t = 0; t < ZTEST_SYNC_LOCKS; t++)
3517fa9e4066Sahrens 		(void) _mutex_init(&zs->zs_sync_lock[t], USYNC_THREAD, NULL);
3518fa9e4066Sahrens 
3519fa9e4066Sahrens 	/*
3520fa9e4066Sahrens 	 * Destroy one disk before we even start.
3521fa9e4066Sahrens 	 * It's mirrored, so everything should work just fine.
3522fa9e4066Sahrens 	 * This makes us exercise fault handling very early in spa_load().
3523fa9e4066Sahrens 	 */
3524fa9e4066Sahrens 	ztest_obliterate_one_disk(0);
3525fa9e4066Sahrens 
3526fa9e4066Sahrens 	/*
3527fa9e4066Sahrens 	 * Verify that the sum of the sizes of all blocks in the pool
3528fa9e4066Sahrens 	 * equals the SPA's allocated space total.
3529fa9e4066Sahrens 	 */
3530fa9e4066Sahrens 	ztest_verify_blocks(pool);
3531fa9e4066Sahrens 
3532fa9e4066Sahrens 	/*
3533fa9e4066Sahrens 	 * Kick off a replacement of the disk we just obliterated.
3534fa9e4066Sahrens 	 */
3535fa9e4066Sahrens 	kernel_init(FREAD | FWRITE);
3536e14bb325SJeff Bonwick 	VERIFY(spa_open(pool, &spa, FTAG) == 0);
3537fa9e4066Sahrens 	ztest_replace_one_disk(spa, 0);
3538fa9e4066Sahrens 	if (zopt_verbose >= 5)
3539fa9e4066Sahrens 		show_pool_stats(spa);
3540fa9e4066Sahrens 	spa_close(spa, FTAG);
3541fa9e4066Sahrens 	kernel_fini();
3542fa9e4066Sahrens 
3543fa9e4066Sahrens 	kernel_init(FREAD | FWRITE);
3544fa9e4066Sahrens 
3545fa9e4066Sahrens 	/*
3546fa9e4066Sahrens 	 * Verify that we can export the pool and reimport it under a
3547fa9e4066Sahrens 	 * different name.
3548fa9e4066Sahrens 	 */
35490e34b6a7Sbonwick 	if (ztest_random(2) == 0) {
35500e34b6a7Sbonwick 		(void) snprintf(name, 100, "%s_import", pool);
35510e34b6a7Sbonwick 		ztest_spa_import_export(pool, name);
35520e34b6a7Sbonwick 		ztest_spa_import_export(name, pool);
35530e34b6a7Sbonwick 	}
3554fa9e4066Sahrens 
3555fa9e4066Sahrens 	/*
3556fa9e4066Sahrens 	 * Verify that we can loop over all pools.
3557fa9e4066Sahrens 	 */
3558fa9e4066Sahrens 	mutex_enter(&spa_namespace_lock);
3559fa9e4066Sahrens 	for (spa = spa_next(NULL); spa != NULL; spa = spa_next(spa)) {
3560fa9e4066Sahrens 		if (zopt_verbose > 3) {
3561fa9e4066Sahrens 			(void) printf("spa_next: found %s\n", spa_name(spa));
3562fa9e4066Sahrens 		}
3563fa9e4066Sahrens 	}
3564fa9e4066Sahrens 	mutex_exit(&spa_namespace_lock);
3565fa9e4066Sahrens 
35660a4e9518Sgw 	/*
3567e14bb325SJeff Bonwick 	 * Open our pool.
35680a4e9518Sgw 	 */
3569e14bb325SJeff Bonwick 	VERIFY(spa_open(pool, &spa, FTAG) == 0);
35700a4e9518Sgw 
35718ad4d6ddSJeff Bonwick 	/*
35728ad4d6ddSJeff Bonwick 	 * We don't expect the pool to suspend unless maxfaults == 0,
35738ad4d6ddSJeff Bonwick 	 * in which case ztest_fault_inject() temporarily takes away
35748ad4d6ddSJeff Bonwick 	 * the only valid replica.
35758ad4d6ddSJeff Bonwick 	 */
35768ad4d6ddSJeff Bonwick 	if (zopt_maxfaults == 0)
35778ad4d6ddSJeff Bonwick 		spa->spa_failmode = ZIO_FAILURE_MODE_WAIT;
35788ad4d6ddSJeff Bonwick 	else
35798ad4d6ddSJeff Bonwick 		spa->spa_failmode = ZIO_FAILURE_MODE_PANIC;
35808ad4d6ddSJeff Bonwick 
3581fa9e4066Sahrens 	/*
3582e14bb325SJeff Bonwick 	 * Create a thread to periodically resume suspended I/O.
3583fa9e4066Sahrens 	 */
35848ad4d6ddSJeff Bonwick 	VERIFY(thr_create(0, 0, ztest_resume_thread, spa, THR_BOUND,
3585e14bb325SJeff Bonwick 	    &resume_tid) == 0);
3586fa9e4066Sahrens 
3587fa9e4066Sahrens 	/*
3588fa9e4066Sahrens 	 * Verify that we can safely inquire about about any object,
3589fa9e4066Sahrens 	 * whether it's allocated or not.  To make it interesting,
3590fa9e4066Sahrens 	 * we probe a 5-wide window around each power of two.
3591fa9e4066Sahrens 	 * This hits all edge cases, including zero and the max.
3592fa9e4066Sahrens 	 */
3593fa9e4066Sahrens 	for (t = 0; t < 64; t++) {
3594fa9e4066Sahrens 		for (d = -5; d <= 5; d++) {
3595fa9e4066Sahrens 			error = dmu_object_info(spa->spa_meta_objset,
3596fa9e4066Sahrens 			    (1ULL << t) + d, NULL);
3597ea8dc4b6Seschrock 			ASSERT(error == 0 || error == ENOENT ||
3598ea8dc4b6Seschrock 			    error == EINVAL);
3599fa9e4066Sahrens 		}
3600fa9e4066Sahrens 	}
3601fa9e4066Sahrens 
3602fa9e4066Sahrens 	/*
3603fa9e4066Sahrens 	 * Now kick off all the tests that run in parallel.
3604fa9e4066Sahrens 	 */
3605fa9e4066Sahrens 	zs->zs_enospc_count = 0;
3606fa9e4066Sahrens 
3607fa9e4066Sahrens 	za = umem_zalloc(zopt_threads * sizeof (ztest_args_t), UMEM_NOFAIL);
3608fa9e4066Sahrens 
3609fa9e4066Sahrens 	if (zopt_verbose >= 4)
3610fa9e4066Sahrens 		(void) printf("starting main threads...\n");
3611fa9e4066Sahrens 
3612fa9e4066Sahrens 	za[0].za_start = gethrtime();
3613fa9e4066Sahrens 	za[0].za_stop = za[0].za_start + zopt_passtime * NANOSEC;
3614fa9e4066Sahrens 	za[0].za_stop = MIN(za[0].za_stop, zs->zs_stop_time);
3615fa9e4066Sahrens 	za[0].za_kill = za[0].za_stop;
3616fa9e4066Sahrens 	if (ztest_random(100) < zopt_killrate)
3617fa9e4066Sahrens 		za[0].za_kill -= ztest_random(zopt_passtime * NANOSEC);
3618fa9e4066Sahrens 
3619fa9e4066Sahrens 	for (t = 0; t < zopt_threads; t++) {
3620ecc2d604Sbonwick 		d = t % zopt_datasets;
3621e05725b1Sbonwick 
3622e05725b1Sbonwick 		(void) strcpy(za[t].za_pool, pool);
3623e05725b1Sbonwick 		za[t].za_os = za[d].za_os;
3624e05725b1Sbonwick 		za[t].za_spa = spa;
3625e05725b1Sbonwick 		za[t].za_zilog = za[d].za_zilog;
3626e05725b1Sbonwick 		za[t].za_instance = t;
3627e05725b1Sbonwick 		za[t].za_random = ztest_random(-1ULL);
3628e05725b1Sbonwick 		za[t].za_start = za[0].za_start;
3629e05725b1Sbonwick 		za[t].za_stop = za[0].za_stop;
3630e05725b1Sbonwick 		za[t].za_kill = za[0].za_kill;
3631e05725b1Sbonwick 
3632ecc2d604Sbonwick 		if (t < zopt_datasets) {
36334d31c452Smaybee 			int test_future = FALSE;
3634fa9e4066Sahrens 			(void) rw_rdlock(&ztest_shared->zs_name_lock);
3635fa9e4066Sahrens 			(void) snprintf(name, 100, "%s/%s_%d", pool, pool, d);
3636ab04eb8eStimh 			error = dmu_objset_create(name, DMU_OST_OTHER, NULL, 0,
3637fa9e4066Sahrens 			    ztest_create_cb, NULL);
36384d31c452Smaybee 			if (error == EEXIST) {
36394d31c452Smaybee 				test_future = TRUE;
3640e05725b1Sbonwick 			} else if (error == ENOSPC) {
3641e05725b1Sbonwick 				zs->zs_enospc_count++;
3642e05725b1Sbonwick 				(void) rw_unlock(&ztest_shared->zs_name_lock);
3643e05725b1Sbonwick 				break;
36444d31c452Smaybee 			} else if (error != 0) {
3645fa9e4066Sahrens 				fatal(0, "dmu_objset_create(%s) = %d",
3646fa9e4066Sahrens 				    name, error);
3647fa9e4066Sahrens 			}
3648fa9e4066Sahrens 			error = dmu_objset_open(name, DMU_OST_OTHER,
3649745cd3c5Smaybee 			    DS_MODE_USER, &za[d].za_os);
3650fa9e4066Sahrens 			if (error)
3651fa9e4066Sahrens 				fatal(0, "dmu_objset_open('%s') = %d",
3652fa9e4066Sahrens 				    name, error);
3653fa9e4066Sahrens 			(void) rw_unlock(&ztest_shared->zs_name_lock);
3654e05725b1Sbonwick 			if (test_future)
3655e05725b1Sbonwick 				ztest_dmu_check_future_leak(&za[t]);
36561209a471SNeil Perrin 			zil_replay(za[d].za_os, za[d].za_os,
36571209a471SNeil Perrin 			    ztest_replay_vector);
3658fa9e4066Sahrens 			za[d].za_zilog = zil_open(za[d].za_os, NULL);
3659fa9e4066Sahrens 		}
3660fa9e4066Sahrens 
3661e14bb325SJeff Bonwick 		VERIFY(thr_create(0, 0, ztest_thread, &za[t], THR_BOUND,
3662e14bb325SJeff Bonwick 		    &za[t].za_thread) == 0);
3663fa9e4066Sahrens 	}
3664fa9e4066Sahrens 
3665fa9e4066Sahrens 	while (--t >= 0) {
3666e14bb325SJeff Bonwick 		VERIFY(thr_join(za[t].za_thread, NULL, NULL) == 0);
3667ecc2d604Sbonwick 		if (t < zopt_datasets) {
3668fa9e4066Sahrens 			zil_close(za[t].za_zilog);
3669fa9e4066Sahrens 			dmu_objset_close(za[t].za_os);
3670fa9e4066Sahrens 		}
3671fa9e4066Sahrens 	}
3672fa9e4066Sahrens 
3673fa9e4066Sahrens 	if (zopt_verbose >= 3)
3674fa9e4066Sahrens 		show_pool_stats(spa);
3675fa9e4066Sahrens 
3676fa9e4066Sahrens 	txg_wait_synced(spa_get_dsl(spa), 0);
3677fa9e4066Sahrens 
3678fa9e4066Sahrens 	zs->zs_alloc = spa_get_alloc(spa);
3679fa9e4066Sahrens 	zs->zs_space = spa_get_space(spa);
3680fa9e4066Sahrens 
3681fa9e4066Sahrens 	/*
3682e05725b1Sbonwick 	 * If we had out-of-space errors, destroy a random objset.
3683fa9e4066Sahrens 	 */
3684fa9e4066Sahrens 	if (zs->zs_enospc_count != 0) {
3685fa9e4066Sahrens 		(void) rw_rdlock(&ztest_shared->zs_name_lock);
3686e05725b1Sbonwick 		d = (int)ztest_random(zopt_datasets);
3687e05725b1Sbonwick 		(void) snprintf(name, 100, "%s/%s_%d", pool, pool, d);
3688fa9e4066Sahrens 		if (zopt_verbose >= 3)
3689fa9e4066Sahrens 			(void) printf("Destroying %s to free up space\n", name);
3690e05725b1Sbonwick 		(void) dmu_objset_find(name, ztest_destroy_cb, &za[d],
36910b69c2f0Sahrens 		    DS_FIND_SNAPSHOTS | DS_FIND_CHILDREN);
3692fa9e4066Sahrens 		(void) rw_unlock(&ztest_shared->zs_name_lock);
3693fa9e4066Sahrens 	}
3694fa9e4066Sahrens 
3695ea8dc4b6Seschrock 	txg_wait_synced(spa_get_dsl(spa), 0);
3696fa9e4066Sahrens 
3697e14bb325SJeff Bonwick 	umem_free(za, zopt_threads * sizeof (ztest_args_t));
3698e14bb325SJeff Bonwick 
3699e14bb325SJeff Bonwick 	/* Kill the resume thread */
3700e14bb325SJeff Bonwick 	ztest_exiting = B_TRUE;
3701e14bb325SJeff Bonwick 	VERIFY(thr_join(resume_tid, NULL, NULL) == 0);
37028ad4d6ddSJeff Bonwick 	ztest_resume(spa);
3703e14bb325SJeff Bonwick 
3704fa9e4066Sahrens 	/*
3705fa9e4066Sahrens 	 * Right before closing the pool, kick off a bunch of async I/O;
3706fa9e4066Sahrens 	 * spa_close() should wait for it to complete.
3707fa9e4066Sahrens 	 */
3708fa9e4066Sahrens 	for (t = 1; t < 50; t++)
3709fa9e4066Sahrens 		dmu_prefetch(spa->spa_meta_objset, t, 0, 1 << 15);
3710fa9e4066Sahrens 
3711e05725b1Sbonwick 	spa_close(spa, FTAG);
3712e05725b1Sbonwick 
3713fa9e4066Sahrens 	kernel_fini();
3714fa9e4066Sahrens }
3715fa9e4066Sahrens 
3716fa9e4066Sahrens void
3717fa9e4066Sahrens print_time(hrtime_t t, char *timebuf)
3718fa9e4066Sahrens {
3719fa9e4066Sahrens 	hrtime_t s = t / NANOSEC;
3720fa9e4066Sahrens 	hrtime_t m = s / 60;
3721fa9e4066Sahrens 	hrtime_t h = m / 60;
3722fa9e4066Sahrens 	hrtime_t d = h / 24;
3723fa9e4066Sahrens 
3724fa9e4066Sahrens 	s -= m * 60;
3725fa9e4066Sahrens 	m -= h * 60;
3726fa9e4066Sahrens 	h -= d * 24;
3727fa9e4066Sahrens 
3728fa9e4066Sahrens 	timebuf[0] = '\0';
3729fa9e4066Sahrens 
3730fa9e4066Sahrens 	if (d)
3731fa9e4066Sahrens 		(void) sprintf(timebuf,
3732fa9e4066Sahrens 		    "%llud%02lluh%02llum%02llus", d, h, m, s);
3733fa9e4066Sahrens 	else if (h)
3734fa9e4066Sahrens 		(void) sprintf(timebuf, "%lluh%02llum%02llus", h, m, s);
3735fa9e4066Sahrens 	else if (m)
3736fa9e4066Sahrens 		(void) sprintf(timebuf, "%llum%02llus", m, s);
3737fa9e4066Sahrens 	else
3738fa9e4066Sahrens 		(void) sprintf(timebuf, "%llus", s);
3739fa9e4066Sahrens }
3740fa9e4066Sahrens 
3741fa9e4066Sahrens /*
3742fa9e4066Sahrens  * Create a storage pool with the given name and initial vdev size.
3743fa9e4066Sahrens  * Then create the specified number of datasets in the pool.
3744fa9e4066Sahrens  */
3745fa9e4066Sahrens static void
3746fa9e4066Sahrens ztest_init(char *pool)
3747fa9e4066Sahrens {
3748fa9e4066Sahrens 	spa_t *spa;
3749fa9e4066Sahrens 	int error;
3750fa9e4066Sahrens 	nvlist_t *nvroot;
3751fa9e4066Sahrens 
3752fa9e4066Sahrens 	kernel_init(FREAD | FWRITE);
3753fa9e4066Sahrens 
3754fa9e4066Sahrens 	/*
3755fa9e4066Sahrens 	 * Create the storage pool.
3756fa9e4066Sahrens 	 */
3757fa9e4066Sahrens 	(void) spa_destroy(pool);
3758fa9e4066Sahrens 	ztest_shared->zs_vdev_primaries = 0;
3759e14bb325SJeff Bonwick 	nvroot = make_vdev_root(NULL, NULL, zopt_vdev_size, 0,
3760e14bb325SJeff Bonwick 	    0, zopt_raidz, zopt_mirrors, 1);
37610a48a24eStimh 	error = spa_create(pool, nvroot, NULL, NULL, NULL);
3762fa9e4066Sahrens 	nvlist_free(nvroot);
3763fa9e4066Sahrens 
3764fa9e4066Sahrens 	if (error)
3765fa9e4066Sahrens 		fatal(0, "spa_create() = %d", error);
3766fa9e4066Sahrens 	error = spa_open(pool, &spa, FTAG);
3767fa9e4066Sahrens 	if (error)
3768fa9e4066Sahrens 		fatal(0, "spa_open() = %d", error);
3769fa9e4066Sahrens 
3770fa9e4066Sahrens 	if (zopt_verbose >= 3)
3771fa9e4066Sahrens 		show_pool_stats(spa);
3772fa9e4066Sahrens 
3773fa9e4066Sahrens 	spa_close(spa, FTAG);
3774fa9e4066Sahrens 
3775fa9e4066Sahrens 	kernel_fini();
3776fa9e4066Sahrens }
3777fa9e4066Sahrens 
3778fa9e4066Sahrens int
3779fa9e4066Sahrens main(int argc, char **argv)
3780fa9e4066Sahrens {
3781fa9e4066Sahrens 	int kills = 0;
3782fa9e4066Sahrens 	int iters = 0;
3783fa9e4066Sahrens 	int i, f;
3784fa9e4066Sahrens 	ztest_shared_t *zs;
3785fa9e4066Sahrens 	ztest_info_t *zi;
3786fa9e4066Sahrens 	char timebuf[100];
3787fa9e4066Sahrens 	char numbuf[6];
3788fa9e4066Sahrens 
3789fa9e4066Sahrens 	(void) setvbuf(stdout, NULL, _IOLBF, 0);
3790fa9e4066Sahrens 
3791fa9e4066Sahrens 	/* Override location of zpool.cache */
3792c5904d13Seschrock 	spa_config_path = "/tmp/zpool.cache";
3793fa9e4066Sahrens 
3794fa9e4066Sahrens 	ztest_random_fd = open("/dev/urandom", O_RDONLY);
3795fa9e4066Sahrens 
3796fa9e4066Sahrens 	process_options(argc, argv);
3797fa9e4066Sahrens 
3798ea8dc4b6Seschrock 	/*
3799ea8dc4b6Seschrock 	 * Blow away any existing copy of zpool.cache
3800ea8dc4b6Seschrock 	 */
3801ea8dc4b6Seschrock 	if (zopt_init != 0)
3802ea8dc4b6Seschrock 		(void) remove("/tmp/zpool.cache");
3803ea8dc4b6Seschrock 
3804fa9e4066Sahrens 	zs = ztest_shared = (void *)mmap(0,
3805fa9e4066Sahrens 	    P2ROUNDUP(sizeof (ztest_shared_t), getpagesize()),
3806fa9e4066Sahrens 	    PROT_READ | PROT_WRITE, MAP_SHARED | MAP_ANON, -1, 0);
3807fa9e4066Sahrens 
3808fa9e4066Sahrens 	if (zopt_verbose >= 1) {
3809fa9e4066Sahrens 		(void) printf("%llu vdevs, %d datasets, %d threads,"
3810fa9e4066Sahrens 		    " %llu seconds...\n",
3811ecc2d604Sbonwick 		    (u_longlong_t)zopt_vdevs, zopt_datasets, zopt_threads,
3812fa9e4066Sahrens 		    (u_longlong_t)zopt_time);
3813fa9e4066Sahrens 	}
3814fa9e4066Sahrens 
3815fa9e4066Sahrens 	/*
3816fa9e4066Sahrens 	 * Create and initialize our storage pool.
3817fa9e4066Sahrens 	 */
3818fa9e4066Sahrens 	for (i = 1; i <= zopt_init; i++) {
3819fa9e4066Sahrens 		bzero(zs, sizeof (ztest_shared_t));
3820fa9e4066Sahrens 		if (zopt_verbose >= 3 && zopt_init != 1)
3821fa9e4066Sahrens 			(void) printf("ztest_init(), pass %d\n", i);
3822fa9e4066Sahrens 		ztest_init(zopt_pool);
3823fa9e4066Sahrens 	}
3824fa9e4066Sahrens 
3825fa9e4066Sahrens 	/*
3826fa9e4066Sahrens 	 * Initialize the call targets for each function.
3827fa9e4066Sahrens 	 */
3828fa9e4066Sahrens 	for (f = 0; f < ZTEST_FUNCS; f++) {
3829fa9e4066Sahrens 		zi = &zs->zs_info[f];
3830fa9e4066Sahrens 
3831fa9e4066Sahrens 		*zi = ztest_info[f];
3832fa9e4066Sahrens 
3833fa9e4066Sahrens 		if (*zi->zi_interval == 0)
3834fa9e4066Sahrens 			zi->zi_call_target = UINT64_MAX;
3835fa9e4066Sahrens 		else
3836fa9e4066Sahrens 			zi->zi_call_target = zopt_time / *zi->zi_interval;
3837fa9e4066Sahrens 	}
3838fa9e4066Sahrens 
3839fa9e4066Sahrens 	zs->zs_start_time = gethrtime();
3840fa9e4066Sahrens 	zs->zs_stop_time = zs->zs_start_time + zopt_time * NANOSEC;
3841fa9e4066Sahrens 
3842fa9e4066Sahrens 	/*
3843fa9e4066Sahrens 	 * Run the tests in a loop.  These tests include fault injection
3844fa9e4066Sahrens 	 * to verify that self-healing data works, and forced crashes
3845fa9e4066Sahrens 	 * to verify that we never lose on-disk consistency.
3846fa9e4066Sahrens 	 */
3847fa9e4066Sahrens 	while (gethrtime() < zs->zs_stop_time) {
3848fa9e4066Sahrens 		int status;
3849fa9e4066Sahrens 		pid_t pid;
3850fa9e4066Sahrens 		char *tmp;
3851fa9e4066Sahrens 
3852fa9e4066Sahrens 		/*
3853fa9e4066Sahrens 		 * Initialize the workload counters for each function.
3854fa9e4066Sahrens 		 */
3855fa9e4066Sahrens 		for (f = 0; f < ZTEST_FUNCS; f++) {
3856fa9e4066Sahrens 			zi = &zs->zs_info[f];
3857fa9e4066Sahrens 			zi->zi_calls = 0;
3858fa9e4066Sahrens 			zi->zi_call_time = 0;
3859fa9e4066Sahrens 		}
3860fa9e4066Sahrens 
3861fa9e4066Sahrens 		pid = fork();
3862fa9e4066Sahrens 
3863fa9e4066Sahrens 		if (pid == -1)
3864fa9e4066Sahrens 			fatal(1, "fork failed");
3865fa9e4066Sahrens 
3866fa9e4066Sahrens 		if (pid == 0) {	/* child */
3867fa9e4066Sahrens 			struct rlimit rl = { 1024, 1024 };
3868fa9e4066Sahrens 			(void) setrlimit(RLIMIT_NOFILE, &rl);
3869004388ebScasper 			(void) enable_extended_FILE_stdio(-1, -1);
3870fa9e4066Sahrens 			ztest_run(zopt_pool);
3871fa9e4066Sahrens 			exit(0);
3872fa9e4066Sahrens 		}
3873fa9e4066Sahrens 
38745ad82045Snd 		while (waitpid(pid, &status, 0) != pid)
3875fa9e4066Sahrens 			continue;
3876fa9e4066Sahrens 
3877fa9e4066Sahrens 		if (WIFEXITED(status)) {
3878fa9e4066Sahrens 			if (WEXITSTATUS(status) != 0) {
3879fa9e4066Sahrens 				(void) fprintf(stderr,
3880fa9e4066Sahrens 				    "child exited with code %d\n",
3881fa9e4066Sahrens 				    WEXITSTATUS(status));
3882fa9e4066Sahrens 				exit(2);
3883fa9e4066Sahrens 			}
38845ad82045Snd 		} else if (WIFSIGNALED(status)) {
3885fa9e4066Sahrens 			if (WTERMSIG(status) != SIGKILL) {
3886fa9e4066Sahrens 				(void) fprintf(stderr,
3887fa9e4066Sahrens 				    "child died with signal %d\n",
3888fa9e4066Sahrens 				    WTERMSIG(status));
3889fa9e4066Sahrens 				exit(3);
3890fa9e4066Sahrens 			}
3891fa9e4066Sahrens 			kills++;
38925ad82045Snd 		} else {
38935ad82045Snd 			(void) fprintf(stderr, "something strange happened "
38945ad82045Snd 			    "to child\n");
38955ad82045Snd 			exit(4);
3896fa9e4066Sahrens 		}
3897fa9e4066Sahrens 
3898fa9e4066Sahrens 		iters++;
3899fa9e4066Sahrens 
3900fa9e4066Sahrens 		if (zopt_verbose >= 1) {
3901fa9e4066Sahrens 			hrtime_t now = gethrtime();
3902fa9e4066Sahrens 
3903fa9e4066Sahrens 			now = MIN(now, zs->zs_stop_time);
3904fa9e4066Sahrens 			print_time(zs->zs_stop_time - now, timebuf);
3905fa9e4066Sahrens 			nicenum(zs->zs_space, numbuf);
3906fa9e4066Sahrens 
3907fa9e4066Sahrens 			(void) printf("Pass %3d, %8s, %3llu ENOSPC, "
3908fa9e4066Sahrens 			    "%4.1f%% of %5s used, %3.0f%% done, %8s to go\n",
3909fa9e4066Sahrens 			    iters,
3910fa9e4066Sahrens 			    WIFEXITED(status) ? "Complete" : "SIGKILL",
3911fa9e4066Sahrens 			    (u_longlong_t)zs->zs_enospc_count,
3912fa9e4066Sahrens 			    100.0 * zs->zs_alloc / zs->zs_space,
3913fa9e4066Sahrens 			    numbuf,
3914fa9e4066Sahrens 			    100.0 * (now - zs->zs_start_time) /
3915fa9e4066Sahrens 			    (zopt_time * NANOSEC), timebuf);
3916fa9e4066Sahrens 		}
3917fa9e4066Sahrens 
3918fa9e4066Sahrens 		if (zopt_verbose >= 2) {
3919fa9e4066Sahrens 			(void) printf("\nWorkload summary:\n\n");
3920fa9e4066Sahrens 			(void) printf("%7s %9s   %s\n",
3921fa9e4066Sahrens 			    "Calls", "Time", "Function");
3922fa9e4066Sahrens 			(void) printf("%7s %9s   %s\n",
3923fa9e4066Sahrens 			    "-----", "----", "--------");
3924fa9e4066Sahrens 			for (f = 0; f < ZTEST_FUNCS; f++) {
3925fa9e4066Sahrens 				Dl_info dli;
3926fa9e4066Sahrens 
3927fa9e4066Sahrens 				zi = &zs->zs_info[f];
3928fa9e4066Sahrens 				print_time(zi->zi_call_time, timebuf);
3929fa9e4066Sahrens 				(void) dladdr((void *)zi->zi_func, &dli);
3930fa9e4066Sahrens 				(void) printf("%7llu %9s   %s\n",
3931fa9e4066Sahrens 				    (u_longlong_t)zi->zi_calls, timebuf,
3932fa9e4066Sahrens 				    dli.dli_sname);
3933fa9e4066Sahrens 			}
3934fa9e4066Sahrens 			(void) printf("\n");
3935fa9e4066Sahrens 		}
3936fa9e4066Sahrens 
3937fa9e4066Sahrens 		/*
3938fa9e4066Sahrens 		 * It's possible that we killed a child during a rename test, in
3939fa9e4066Sahrens 		 * which case we'll have a 'ztest_tmp' pool lying around instead
3940fa9e4066Sahrens 		 * of 'ztest'.  Do a blind rename in case this happened.
3941fa9e4066Sahrens 		 */
3942fa9e4066Sahrens 		tmp = umem_alloc(strlen(zopt_pool) + 5, UMEM_NOFAIL);
3943fa9e4066Sahrens 		(void) strcpy(tmp, zopt_pool);
3944fa9e4066Sahrens 		(void) strcat(tmp, "_tmp");
3945fa9e4066Sahrens 		kernel_init(FREAD | FWRITE);
3946fa9e4066Sahrens 		(void) spa_rename(tmp, zopt_pool);
3947fa9e4066Sahrens 		kernel_fini();
3948fa9e4066Sahrens 		umem_free(tmp, strlen(tmp) + 1);
3949fa9e4066Sahrens 	}
3950fa9e4066Sahrens 
3951fa9e4066Sahrens 	ztest_verify_blocks(zopt_pool);
3952fa9e4066Sahrens 
3953fa9e4066Sahrens 	if (zopt_verbose >= 1) {
3954fa9e4066Sahrens 		(void) printf("%d killed, %d completed, %.0f%% kill rate\n",
3955fa9e4066Sahrens 		    kills, iters - kills, (100.0 * kills) / MAX(1, iters));
3956fa9e4066Sahrens 	}
3957fa9e4066Sahrens 
3958fa9e4066Sahrens 	return (0);
3959fa9e4066Sahrens }
3960