xref: /illumos-gate/usr/src/cmd/ztest/ztest.c (revision fb09f5aa)
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 /*
2247cb52daSJeff Bonwick  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
23420dfc95SChris Siden  * Copyright (c) 2012 by Delphix. All rights reserved.
24e9103aaeSGarrett D'Amore  * Copyright 2011 Nexenta Systems, Inc.  All rights reserved.
25fa9e4066Sahrens  */
26fa9e4066Sahrens 
27fa9e4066Sahrens /*
28fa9e4066Sahrens  * The objective of this program is to provide a DMU/ZAP/SPA stress test
29fa9e4066Sahrens  * that runs entirely in userland, is easy to use, and easy to extend.
30fa9e4066Sahrens  *
31fa9e4066Sahrens  * The overall design of the ztest program is as follows:
32fa9e4066Sahrens  *
33fa9e4066Sahrens  * (1) For each major functional area (e.g. adding vdevs to a pool,
34fa9e4066Sahrens  *     creating and destroying datasets, reading and writing objects, etc)
35fa9e4066Sahrens  *     we have a simple routine to test that functionality.  These
36fa9e4066Sahrens  *     individual routines do not have to do anything "stressful".
37fa9e4066Sahrens  *
38fa9e4066Sahrens  * (2) We turn these simple functionality tests into a stress test by
39fa9e4066Sahrens  *     running them all in parallel, with as many threads as desired,
40fa9e4066Sahrens  *     and spread across as many datasets, objects, and vdevs as desired.
41fa9e4066Sahrens  *
42fa9e4066Sahrens  * (3) While all this is happening, we inject faults into the pool to
43fa9e4066Sahrens  *     verify that self-healing data really works.
44fa9e4066Sahrens  *
45fa9e4066Sahrens  * (4) Every time we open a dataset, we change its checksum and compression
46fa9e4066Sahrens  *     functions.  Thus even individual objects vary from block to block
47fa9e4066Sahrens  *     in which checksum they use and whether they're compressed.
48fa9e4066Sahrens  *
49fa9e4066Sahrens  * (5) To verify that we never lose on-disk consistency after a crash,
50fa9e4066Sahrens  *     we run the entire test in a child of the main process.
51fa9e4066Sahrens  *     At random times, the child self-immolates with a SIGKILL.
52fa9e4066Sahrens  *     This is the software equivalent of pulling the power cord.
53fa9e4066Sahrens  *     The parent then runs the test again, using the existing
54420dfc95SChris Siden  *     storage pool, as many times as desired. If backwards compatability
55420dfc95SChris Siden  *     testing is enabled ztest will sometimes run the "older" version
56420dfc95SChris Siden  *     of ztest after a SIGKILL.
57fa9e4066Sahrens  *
58fa9e4066Sahrens  * (6) To verify that we don't have future leaks or temporal incursions,
59fa9e4066Sahrens  *     many of the functional tests record the transaction group number
60fa9e4066Sahrens  *     as part of their data.  When reading old data, they verify that
61fa9e4066Sahrens  *     the transaction group number is less than the current, open txg.
62fa9e4066Sahrens  *     If you add a new test, please do this if applicable.
63fa9e4066Sahrens  *
64fa9e4066Sahrens  * When run with no arguments, ztest runs for about five minutes and
65fa9e4066Sahrens  * produces no output if successful.  To get a little bit of information,
66fa9e4066Sahrens  * specify -V.  To get more information, specify -VV, and so on.
67fa9e4066Sahrens  *
68fa9e4066Sahrens  * To turn this into an overnight stress test, use -T to specify run time.
69fa9e4066Sahrens  *
70fa9e4066Sahrens  * You can ask more more vdevs [-v], datasets [-d], or threads [-t]
71fa9e4066Sahrens  * to increase the pool capacity, fanout, and overall stress level.
72fa9e4066Sahrens  *
73420dfc95SChris Siden  * Use the -k option to set the desired frequency of kills.
74420dfc95SChris Siden  *
75420dfc95SChris Siden  * When ztest invokes itself it passes all relevant information through a
76420dfc95SChris Siden  * temporary file which is mmap-ed in the child process. This allows shared
77420dfc95SChris Siden  * memory to survive the exec syscall. The ztest_shared_hdr_t struct is always
78420dfc95SChris Siden  * stored at offset 0 of this file and contains information on the size and
79420dfc95SChris Siden  * number of shared structures in the file. The information stored in this file
80420dfc95SChris Siden  * must remain backwards compatible with older versions of ztest so that
81420dfc95SChris Siden  * ztest can invoke them during backwards compatibility testing (-B).
82fa9e4066Sahrens  */
83fa9e4066Sahrens 
84fa9e4066Sahrens #include <sys/zfs_context.h>
85fa9e4066Sahrens #include <sys/spa.h>
86fa9e4066Sahrens #include <sys/dmu.h>
87fa9e4066Sahrens #include <sys/txg.h>
882fdbea25SAleksandr Guzovskiy #include <sys/dbuf.h>
89fa9e4066Sahrens #include <sys/zap.h>
90fa9e4066Sahrens #include <sys/dmu_objset.h>
91fa9e4066Sahrens #include <sys/poll.h>
92fa9e4066Sahrens #include <sys/stat.h>
93fa9e4066Sahrens #include <sys/time.h>
94fa9e4066Sahrens #include <sys/wait.h>
95fa9e4066Sahrens #include <sys/mman.h>
96fa9e4066Sahrens #include <sys/resource.h>
97fa9e4066Sahrens #include <sys/zio.h>
98fa9e4066Sahrens #include <sys/zil.h>
99b24ab676SJeff Bonwick #include <sys/zil_impl.h>
100fa9e4066Sahrens #include <sys/vdev_impl.h>
101e14bb325SJeff Bonwick #include <sys/vdev_file.h>
102fa9e4066Sahrens #include <sys/spa_impl.h>
10388ecc943SGeorge Wilson #include <sys/metaslab_impl.h>
104fa9e4066Sahrens #include <sys/dsl_prop.h>
1054f5064b7SMark J Musante #include <sys/dsl_dataset.h>
1063f9d6ad7SLin Ling #include <sys/dsl_scan.h>
107cde58dbcSMatthew Ahrens #include <sys/zio_checksum.h>
108fa9e4066Sahrens #include <sys/refcount.h>
109ad135b5dSChristopher Siden #include <sys/zfeature.h>
110fa9e4066Sahrens #include <stdio.h>
111004388ebScasper #include <stdio_ext.h>
112fa9e4066Sahrens #include <stdlib.h>
113fa9e4066Sahrens #include <unistd.h>
114fa9e4066Sahrens #include <signal.h>
115fa9e4066Sahrens #include <umem.h>
116fa9e4066Sahrens #include <dlfcn.h>
117fa9e4066Sahrens #include <ctype.h>
118fa9e4066Sahrens #include <math.h>
119fa9e4066Sahrens #include <sys/fs/zfs.h>
120b24ab676SJeff Bonwick #include <libnvpair.h>
121fa9e4066Sahrens 
122420dfc95SChris Siden #define	ZTEST_FD_DATA 3
123420dfc95SChris Siden #define	ZTEST_FD_RAND 4
124420dfc95SChris Siden 
125420dfc95SChris Siden typedef struct ztest_shared_hdr {
126420dfc95SChris Siden 	uint64_t	zh_hdr_size;
127420dfc95SChris Siden 	uint64_t	zh_opts_size;
128420dfc95SChris Siden 	uint64_t	zh_size;
129420dfc95SChris Siden 	uint64_t	zh_stats_size;
130420dfc95SChris Siden 	uint64_t	zh_stats_count;
131420dfc95SChris Siden 	uint64_t	zh_ds_size;
132420dfc95SChris Siden 	uint64_t	zh_ds_count;
133420dfc95SChris Siden } ztest_shared_hdr_t;
134420dfc95SChris Siden 
135420dfc95SChris Siden static ztest_shared_hdr_t *ztest_shared_hdr;
136420dfc95SChris Siden 
137420dfc95SChris Siden typedef struct ztest_shared_opts {
138420dfc95SChris Siden 	char zo_pool[MAXNAMELEN];
139420dfc95SChris Siden 	char zo_dir[MAXNAMELEN];
140420dfc95SChris Siden 	char zo_alt_ztest[MAXNAMELEN];
141420dfc95SChris Siden 	char zo_alt_libpath[MAXNAMELEN];
142420dfc95SChris Siden 	uint64_t zo_vdevs;
143420dfc95SChris Siden 	uint64_t zo_vdevtime;
144420dfc95SChris Siden 	size_t zo_vdev_size;
145420dfc95SChris Siden 	int zo_ashift;
146420dfc95SChris Siden 	int zo_mirrors;
147420dfc95SChris Siden 	int zo_raidz;
148420dfc95SChris Siden 	int zo_raidz_parity;
149420dfc95SChris Siden 	int zo_datasets;
150420dfc95SChris Siden 	int zo_threads;
151420dfc95SChris Siden 	uint64_t zo_passtime;
152420dfc95SChris Siden 	uint64_t zo_killrate;
153420dfc95SChris Siden 	int zo_verbose;
154420dfc95SChris Siden 	int zo_init;
155420dfc95SChris Siden 	uint64_t zo_time;
156420dfc95SChris Siden 	uint64_t zo_maxloops;
157420dfc95SChris Siden 	uint64_t zo_metaslab_gang_bang;
158420dfc95SChris Siden } ztest_shared_opts_t;
159420dfc95SChris Siden 
160420dfc95SChris Siden static const ztest_shared_opts_t ztest_opts_defaults = {
161420dfc95SChris Siden 	.zo_pool = { 'z', 't', 'e', 's', 't', '\0' },
162420dfc95SChris Siden 	.zo_dir = { '/', 't', 'm', 'p', '\0' },
163420dfc95SChris Siden 	.zo_alt_ztest = { '\0' },
164420dfc95SChris Siden 	.zo_alt_libpath = { '\0' },
165420dfc95SChris Siden 	.zo_vdevs = 5,
166420dfc95SChris Siden 	.zo_ashift = SPA_MINBLOCKSHIFT,
167420dfc95SChris Siden 	.zo_mirrors = 2,
168420dfc95SChris Siden 	.zo_raidz = 4,
169420dfc95SChris Siden 	.zo_raidz_parity = 1,
170420dfc95SChris Siden 	.zo_vdev_size = SPA_MINDEVSIZE,
171420dfc95SChris Siden 	.zo_datasets = 7,
172420dfc95SChris Siden 	.zo_threads = 23,
173420dfc95SChris Siden 	.zo_passtime = 60,		/* 60 seconds */
174420dfc95SChris Siden 	.zo_killrate = 70,		/* 70% kill rate */
175420dfc95SChris Siden 	.zo_verbose = 0,
176420dfc95SChris Siden 	.zo_init = 1,
177420dfc95SChris Siden 	.zo_time = 300,			/* 5 minutes */
178420dfc95SChris Siden 	.zo_maxloops = 50,		/* max loops during spa_freeze() */
179420dfc95SChris Siden 	.zo_metaslab_gang_bang = 32 << 10
180420dfc95SChris Siden };
181420dfc95SChris Siden 
182420dfc95SChris Siden extern uint64_t metaslab_gang_bang;
183420dfc95SChris Siden extern uint64_t metaslab_df_alloc_threshold;
184420dfc95SChris Siden 
185420dfc95SChris Siden static ztest_shared_opts_t *ztest_shared_opts;
186420dfc95SChris Siden static ztest_shared_opts_t ztest_opts;
187420dfc95SChris Siden 
188420dfc95SChris Siden typedef struct ztest_shared_ds {
189420dfc95SChris Siden 	uint64_t	zd_seq;
190420dfc95SChris Siden } ztest_shared_ds_t;
191420dfc95SChris Siden 
192420dfc95SChris Siden static ztest_shared_ds_t *ztest_shared_ds;
193420dfc95SChris Siden #define	ZTEST_GET_SHARED_DS(d) (&ztest_shared_ds[d])
194fa9e4066Sahrens 
195b24ab676SJeff Bonwick #define	BT_MAGIC	0x123456789abcdefULL
196420dfc95SChris Siden #define	MAXFAULTS() \
197420dfc95SChris Siden 	(MAX(zs->zs_mirrors, 1) * (ztest_opts.zo_raidz_parity + 1) - 1)
198b24ab676SJeff Bonwick 
199b24ab676SJeff Bonwick enum ztest_io_type {
200b24ab676SJeff Bonwick 	ZTEST_IO_WRITE_TAG,
201b24ab676SJeff Bonwick 	ZTEST_IO_WRITE_PATTERN,
202b24ab676SJeff Bonwick 	ZTEST_IO_WRITE_ZEROES,
203b24ab676SJeff Bonwick 	ZTEST_IO_TRUNCATE,
204b24ab676SJeff Bonwick 	ZTEST_IO_SETATTR,
205b24ab676SJeff Bonwick 	ZTEST_IO_TYPES
206b24ab676SJeff Bonwick };
207b24ab676SJeff Bonwick 
208e05725b1Sbonwick typedef struct ztest_block_tag {
209b24ab676SJeff Bonwick 	uint64_t	bt_magic;
210e05725b1Sbonwick 	uint64_t	bt_objset;
211e05725b1Sbonwick 	uint64_t	bt_object;
212e05725b1Sbonwick 	uint64_t	bt_offset;
213b24ab676SJeff Bonwick 	uint64_t	bt_gen;
214e05725b1Sbonwick 	uint64_t	bt_txg;
215b24ab676SJeff Bonwick 	uint64_t	bt_crtxg;
216e05725b1Sbonwick } ztest_block_tag_t;
217e05725b1Sbonwick 
218b24ab676SJeff Bonwick typedef struct bufwad {
219b24ab676SJeff Bonwick 	uint64_t	bw_index;
220b24ab676SJeff Bonwick 	uint64_t	bw_txg;
221b24ab676SJeff Bonwick 	uint64_t	bw_data;
222b24ab676SJeff Bonwick } bufwad_t;
223b24ab676SJeff Bonwick 
224b24ab676SJeff Bonwick /*
225b24ab676SJeff Bonwick  * XXX -- fix zfs range locks to be generic so we can use them here.
226b24ab676SJeff Bonwick  */
227b24ab676SJeff Bonwick typedef enum {
228b24ab676SJeff Bonwick 	RL_READER,
229b24ab676SJeff Bonwick 	RL_WRITER,
230b24ab676SJeff Bonwick 	RL_APPEND
231b24ab676SJeff Bonwick } rl_type_t;
232b24ab676SJeff Bonwick 
233b24ab676SJeff Bonwick typedef struct rll {
234b24ab676SJeff Bonwick 	void		*rll_writer;
235b24ab676SJeff Bonwick 	int		rll_readers;
236b24ab676SJeff Bonwick 	mutex_t		rll_lock;
237b24ab676SJeff Bonwick 	cond_t		rll_cv;
238b24ab676SJeff Bonwick } rll_t;
239b24ab676SJeff Bonwick 
240b24ab676SJeff Bonwick typedef struct rl {
241b24ab676SJeff Bonwick 	uint64_t	rl_object;
242b24ab676SJeff Bonwick 	uint64_t	rl_offset;
243b24ab676SJeff Bonwick 	uint64_t	rl_size;
244b24ab676SJeff Bonwick 	rll_t		*rl_lock;
245b24ab676SJeff Bonwick } rl_t;
246b24ab676SJeff Bonwick 
247b24ab676SJeff Bonwick #define	ZTEST_RANGE_LOCKS	64
248b24ab676SJeff Bonwick #define	ZTEST_OBJECT_LOCKS	64
249b24ab676SJeff Bonwick 
250b24ab676SJeff Bonwick /*
251b24ab676SJeff Bonwick  * Object descriptor.  Used as a template for object lookup/create/remove.
252b24ab676SJeff Bonwick  */
253b24ab676SJeff Bonwick typedef struct ztest_od {
254b24ab676SJeff Bonwick 	uint64_t	od_dir;
255b24ab676SJeff Bonwick 	uint64_t	od_object;
256b24ab676SJeff Bonwick 	dmu_object_type_t od_type;
257b24ab676SJeff Bonwick 	dmu_object_type_t od_crtype;
258b24ab676SJeff Bonwick 	uint64_t	od_blocksize;
259b24ab676SJeff Bonwick 	uint64_t	od_crblocksize;
260b24ab676SJeff Bonwick 	uint64_t	od_gen;
261b24ab676SJeff Bonwick 	uint64_t	od_crgen;
262b24ab676SJeff Bonwick 	char		od_name[MAXNAMELEN];
263b24ab676SJeff Bonwick } ztest_od_t;
264fa9e4066Sahrens 
265b24ab676SJeff Bonwick /*
266b24ab676SJeff Bonwick  * Per-dataset state.
267b24ab676SJeff Bonwick  */
268b24ab676SJeff Bonwick typedef struct ztest_ds {
269420dfc95SChris Siden 	ztest_shared_ds_t *zd_shared;
270b24ab676SJeff Bonwick 	objset_t	*zd_os;
271c9ba2a43SEric Schrock 	rwlock_t	zd_zilog_lock;
272b24ab676SJeff Bonwick 	zilog_t		*zd_zilog;
273b24ab676SJeff Bonwick 	ztest_od_t	*zd_od;		/* debugging aid */
274b24ab676SJeff Bonwick 	char		zd_name[MAXNAMELEN];
275b24ab676SJeff Bonwick 	mutex_t		zd_dirobj_lock;
276b24ab676SJeff Bonwick 	rll_t		zd_object_lock[ZTEST_OBJECT_LOCKS];
277b24ab676SJeff Bonwick 	rll_t		zd_range_lock[ZTEST_RANGE_LOCKS];
278b24ab676SJeff Bonwick } ztest_ds_t;
279b24ab676SJeff Bonwick 
280b24ab676SJeff Bonwick /*
281b24ab676SJeff Bonwick  * Per-iteration state.
282b24ab676SJeff Bonwick  */
283b24ab676SJeff Bonwick typedef void ztest_func_t(ztest_ds_t *zd, uint64_t id);
284b24ab676SJeff Bonwick 
285b24ab676SJeff Bonwick typedef struct ztest_info {
286b24ab676SJeff Bonwick 	ztest_func_t	*zi_func;	/* test function */
287b24ab676SJeff Bonwick 	uint64_t	zi_iters;	/* iterations per execution */
288b24ab676SJeff Bonwick 	uint64_t	*zi_interval;	/* execute every <interval> seconds */
289b24ab676SJeff Bonwick } ztest_info_t;
290fa9e4066Sahrens 
291420dfc95SChris Siden typedef struct ztest_shared_callstate {
292420dfc95SChris Siden 	uint64_t	zc_count;	/* per-pass count */
293420dfc95SChris Siden 	uint64_t	zc_time;	/* per-pass time */
294420dfc95SChris Siden 	uint64_t	zc_next;	/* next time to call this function */
295420dfc95SChris Siden } ztest_shared_callstate_t;
296420dfc95SChris Siden 
297420dfc95SChris Siden static ztest_shared_callstate_t *ztest_shared_callstate;
298420dfc95SChris Siden #define	ZTEST_GET_SHARED_CALLSTATE(c) (&ztest_shared_callstate[c])
299420dfc95SChris Siden 
300fa9e4066Sahrens /*
301fa9e4066Sahrens  * Note: these aren't static because we want dladdr() to work.
302fa9e4066Sahrens  */
303fa9e4066Sahrens ztest_func_t ztest_dmu_read_write;
304fa9e4066Sahrens ztest_func_t ztest_dmu_write_parallel;
305fa9e4066Sahrens ztest_func_t ztest_dmu_object_alloc_free;
306d20e665cSRicardo M. Correia ztest_func_t ztest_dmu_commit_callbacks;
307fa9e4066Sahrens ztest_func_t ztest_zap;
308fa9e4066Sahrens ztest_func_t ztest_zap_parallel;
309b24ab676SJeff Bonwick ztest_func_t ztest_zil_commit;
310c9ba2a43SEric Schrock ztest_func_t ztest_zil_remount;
311b24ab676SJeff Bonwick ztest_func_t ztest_dmu_read_write_zcopy;
312fa9e4066Sahrens ztest_func_t ztest_dmu_objset_create_destroy;
313b24ab676SJeff Bonwick ztest_func_t ztest_dmu_prealloc;
314b24ab676SJeff Bonwick ztest_func_t ztest_fzap;
315fa9e4066Sahrens ztest_func_t ztest_dmu_snapshot_create_destroy;
316b24ab676SJeff Bonwick ztest_func_t ztest_dsl_prop_get_set;
317b24ab676SJeff Bonwick ztest_func_t ztest_spa_prop_get_set;
318fa9e4066Sahrens ztest_func_t ztest_spa_create_destroy;
319fa9e4066Sahrens ztest_func_t ztest_fault_inject;
320b24ab676SJeff Bonwick ztest_func_t ztest_ddt_repair;
321b24ab676SJeff Bonwick ztest_func_t ztest_dmu_snapshot_hold;
322e14bb325SJeff Bonwick ztest_func_t ztest_spa_rename;
323b24ab676SJeff Bonwick ztest_func_t ztest_scrub;
324b24ab676SJeff Bonwick ztest_func_t ztest_dsl_dataset_promote_busy;
325fa9e4066Sahrens ztest_func_t ztest_vdev_attach_detach;
326fa9e4066Sahrens ztest_func_t ztest_vdev_LUN_growth;
327fa9e4066Sahrens ztest_func_t ztest_vdev_add_remove;
328e14bb325SJeff Bonwick ztest_func_t ztest_vdev_aux_add_remove;
3291195e687SMark J Musante ztest_func_t ztest_split_pool;
330e9103aaeSGarrett D'Amore ztest_func_t ztest_reguid;
331fa9e4066Sahrens 
332b24ab676SJeff Bonwick uint64_t zopt_always = 0ULL * NANOSEC;		/* all the time */
333b24ab676SJeff Bonwick uint64_t zopt_incessant = 1ULL * NANOSEC / 10;	/* every 1/10 second */
334b24ab676SJeff Bonwick uint64_t zopt_often = 1ULL * NANOSEC;		/* every second */
335b24ab676SJeff Bonwick uint64_t zopt_sometimes = 10ULL * NANOSEC;	/* every 10 seconds */
336b24ab676SJeff Bonwick uint64_t zopt_rarely = 60ULL * NANOSEC;		/* every 60 seconds */
337fa9e4066Sahrens 
338fa9e4066Sahrens ztest_info_t ztest_info[] = {
339e05725b1Sbonwick 	{ ztest_dmu_read_write,			1,	&zopt_always	},
340b24ab676SJeff Bonwick 	{ ztest_dmu_write_parallel,		10,	&zopt_always	},
341e05725b1Sbonwick 	{ ztest_dmu_object_alloc_free,		1,	&zopt_always	},
342b24ab676SJeff Bonwick 	{ ztest_dmu_commit_callbacks,		1,	&zopt_always	},
343e05725b1Sbonwick 	{ ztest_zap,				30,	&zopt_always	},
344e05725b1Sbonwick 	{ ztest_zap_parallel,			100,	&zopt_always	},
3451195e687SMark J Musante 	{ ztest_split_pool,			1,	&zopt_always	},
346b24ab676SJeff Bonwick 	{ ztest_zil_commit,			1,	&zopt_incessant	},
347c9ba2a43SEric Schrock 	{ ztest_zil_remount,			1,	&zopt_sometimes	},
348b24ab676SJeff Bonwick 	{ ztest_dmu_read_write_zcopy,		1,	&zopt_often	},
349b24ab676SJeff Bonwick 	{ ztest_dmu_objset_create_destroy,	1,	&zopt_often	},
350b24ab676SJeff Bonwick 	{ ztest_dsl_prop_get_set,		1,	&zopt_often	},
351b24ab676SJeff Bonwick 	{ ztest_spa_prop_get_set,		1,	&zopt_sometimes	},
352b24ab676SJeff Bonwick #if 0
353b24ab676SJeff Bonwick 	{ ztest_dmu_prealloc,			1,	&zopt_sometimes	},
354b24ab676SJeff Bonwick #endif
355b24ab676SJeff Bonwick 	{ ztest_fzap,				1,	&zopt_sometimes	},
356b24ab676SJeff Bonwick 	{ ztest_dmu_snapshot_create_destroy,	1,	&zopt_sometimes	},
357b24ab676SJeff Bonwick 	{ ztest_spa_create_destroy,		1,	&zopt_sometimes	},
358e05725b1Sbonwick 	{ ztest_fault_inject,			1,	&zopt_sometimes	},
359b24ab676SJeff Bonwick 	{ ztest_ddt_repair,			1,	&zopt_sometimes	},
3605c987a37SChris Kirby 	{ ztest_dmu_snapshot_hold,		1,	&zopt_sometimes	},
361e9103aaeSGarrett D'Amore 	{ ztest_reguid,				1,	&zopt_sometimes },
362e05725b1Sbonwick 	{ ztest_spa_rename,			1,	&zopt_rarely	},
363b24ab676SJeff Bonwick 	{ ztest_scrub,				1,	&zopt_rarely	},
364b24ab676SJeff Bonwick 	{ ztest_dsl_dataset_promote_busy,	1,	&zopt_rarely	},
365dfbb9432SGeorge Wilson 	{ ztest_vdev_attach_detach,		1,	&zopt_rarely	},
366e14bb325SJeff Bonwick 	{ ztest_vdev_LUN_growth,		1,	&zopt_rarely	},
367420dfc95SChris Siden 	{ ztest_vdev_add_remove,		1,
368420dfc95SChris Siden 	    &ztest_opts.zo_vdevtime				},
369420dfc95SChris Siden 	{ ztest_vdev_aux_add_remove,		1,
370420dfc95SChris Siden 	    &ztest_opts.zo_vdevtime				},
371fa9e4066Sahrens };
372fa9e4066Sahrens 
373fa9e4066Sahrens #define	ZTEST_FUNCS	(sizeof (ztest_info) / sizeof (ztest_info_t))
374fa9e4066Sahrens 
375d20e665cSRicardo M. Correia /*
376d20e665cSRicardo M. Correia  * The following struct is used to hold a list of uncalled commit callbacks.
377d20e665cSRicardo M. Correia  * The callbacks are ordered by txg number.
378d20e665cSRicardo M. Correia  */
379d20e665cSRicardo M. Correia typedef struct ztest_cb_list {
380d20e665cSRicardo M. Correia 	mutex_t	zcl_callbacks_lock;
381d20e665cSRicardo M. Correia 	list_t	zcl_callbacks;
382d20e665cSRicardo M. Correia } ztest_cb_list_t;
383d20e665cSRicardo M. Correia 
384fa9e4066Sahrens /*
385fa9e4066Sahrens  * Stuff we need to share writably between parent and child.
386fa9e4066Sahrens  */
387fa9e4066Sahrens typedef struct ztest_shared {
388420dfc95SChris Siden 	boolean_t	zs_do_init;
389b24ab676SJeff Bonwick 	hrtime_t	zs_proc_start;
390b24ab676SJeff Bonwick 	hrtime_t	zs_proc_stop;
391b24ab676SJeff Bonwick 	hrtime_t	zs_thread_start;
392b24ab676SJeff Bonwick 	hrtime_t	zs_thread_stop;
393b24ab676SJeff Bonwick 	hrtime_t	zs_thread_kill;
394b24ab676SJeff Bonwick 	uint64_t	zs_enospc_count;
39588ecc943SGeorge Wilson 	uint64_t	zs_vdev_next_leaf;
396e14bb325SJeff Bonwick 	uint64_t	zs_vdev_aux;
397fa9e4066Sahrens 	uint64_t	zs_alloc;
398fa9e4066Sahrens 	uint64_t	zs_space;
3991195e687SMark J Musante 	uint64_t	zs_splits;
4001195e687SMark J Musante 	uint64_t	zs_mirrors;
401420dfc95SChris Siden 	uint64_t	zs_metaslab_sz;
402420dfc95SChris Siden 	uint64_t	zs_metaslab_df_alloc_threshold;
403420dfc95SChris Siden 	uint64_t	zs_guid;
404fa9e4066Sahrens } ztest_shared_t;
405fa9e4066Sahrens 
406b24ab676SJeff Bonwick #define	ID_PARALLEL	-1ULL
407b24ab676SJeff Bonwick 
408fa9e4066Sahrens static char ztest_dev_template[] = "%s/%s.%llua";
409e14bb325SJeff Bonwick static char ztest_aux_template[] = "%s/%s.%s.%llu";
410b24ab676SJeff Bonwick ztest_shared_t *ztest_shared;
411fa9e4066Sahrens 
412420dfc95SChris Siden static spa_t *ztest_spa = NULL;
413420dfc95SChris Siden static ztest_ds_t *ztest_ds;
414420dfc95SChris Siden 
415420dfc95SChris Siden static mutex_t ztest_vdev_lock;
416dfbb9432SGeorge Wilson 
417dfbb9432SGeorge Wilson /*
418dfbb9432SGeorge Wilson  * The ztest_name_lock protects the pool and dataset namespace used by
419dfbb9432SGeorge Wilson  * the individual tests. To modify the namespace, consumers must grab
420dfbb9432SGeorge Wilson  * this lock as writer. Grabbing the lock as reader will ensure that the
421dfbb9432SGeorge Wilson  * namespace does not change while the lock is held.
422dfbb9432SGeorge Wilson  */
423420dfc95SChris Siden static rwlock_t ztest_name_lock;
424fa9e4066Sahrens 
425420dfc95SChris Siden static boolean_t ztest_dump_core = B_TRUE;
426e14bb325SJeff Bonwick static boolean_t ztest_exiting;
4270a4e9518Sgw 
428d20e665cSRicardo M. Correia /* Global commit callback list */
429d20e665cSRicardo M. Correia static ztest_cb_list_t zcl;
430d20e665cSRicardo M. Correia 
431b24ab676SJeff Bonwick enum ztest_object {
432b24ab676SJeff Bonwick 	ZTEST_META_DNODE = 0,
433b24ab676SJeff Bonwick 	ZTEST_DIROBJ,
434b24ab676SJeff Bonwick 	ZTEST_OBJECTS
435b24ab676SJeff Bonwick };
436fa9e4066Sahrens 
4371ce825d8Sraf static void usage(boolean_t) __NORETURN;
438f1b4288bSvb 
439fa9e4066Sahrens /*
440fa9e4066Sahrens  * These libumem hooks provide a reasonable set of defaults for the allocator's
441fa9e4066Sahrens  * debugging facilities.
442fa9e4066Sahrens  */
443fa9e4066Sahrens const char *
444fa9e4066Sahrens _umem_debug_init()
445fa9e4066Sahrens {
446fa9e4066Sahrens 	return ("default,verbose"); /* $UMEM_DEBUG setting */
447fa9e4066Sahrens }
448fa9e4066Sahrens 
449fa9e4066Sahrens const char *
450fa9e4066Sahrens _umem_logging_init(void)
451fa9e4066Sahrens {
452fa9e4066Sahrens 	return ("fail,contents"); /* $UMEM_LOGGING setting */
453fa9e4066Sahrens }
454fa9e4066Sahrens 
455fa9e4066Sahrens #define	FATAL_MSG_SZ	1024
456fa9e4066Sahrens 
457fa9e4066Sahrens char *fatal_msg;
458fa9e4066Sahrens 
459fa9e4066Sahrens static void
460fa9e4066Sahrens fatal(int do_perror, char *message, ...)
461fa9e4066Sahrens {
462fa9e4066Sahrens 	va_list args;
463fa9e4066Sahrens 	int save_errno = errno;
464fa9e4066Sahrens 	char buf[FATAL_MSG_SZ];
465fa9e4066Sahrens 
466fa9e4066Sahrens 	(void) fflush(stdout);
467fa9e4066Sahrens 
468fa9e4066Sahrens 	va_start(args, message);
469fa9e4066Sahrens 	(void) sprintf(buf, "ztest: ");
470fa9e4066Sahrens 	/* LINTED */
471fa9e4066Sahrens 	(void) vsprintf(buf + strlen(buf), message, args);
472fa9e4066Sahrens 	va_end(args);
473fa9e4066Sahrens 	if (do_perror) {
474fa9e4066Sahrens 		(void) snprintf(buf + strlen(buf), FATAL_MSG_SZ - strlen(buf),
475fa9e4066Sahrens 		    ": %s", strerror(save_errno));
476fa9e4066Sahrens 	}
477fa9e4066Sahrens 	(void) fprintf(stderr, "%s\n", buf);
478fa9e4066Sahrens 	fatal_msg = buf;			/* to ease debugging */
479fa9e4066Sahrens 	if (ztest_dump_core)
480fa9e4066Sahrens 		abort();
481fa9e4066Sahrens 	exit(3);
482fa9e4066Sahrens }
483fa9e4066Sahrens 
484fa9e4066Sahrens static int
485fa9e4066Sahrens str2shift(const char *buf)
486fa9e4066Sahrens {
487fa9e4066Sahrens 	const char *ends = "BKMGTPEZ";
488fa9e4066Sahrens 	int i;
489fa9e4066Sahrens 
490fa9e4066Sahrens 	if (buf[0] == '\0')
491fa9e4066Sahrens 		return (0);
492fa9e4066Sahrens 	for (i = 0; i < strlen(ends); i++) {
493fa9e4066Sahrens 		if (toupper(buf[0]) == ends[i])
494fa9e4066Sahrens 			break;
495fa9e4066Sahrens 	}
496f1b4288bSvb 	if (i == strlen(ends)) {
497f1b4288bSvb 		(void) fprintf(stderr, "ztest: invalid bytes suffix: %s\n",
498f1b4288bSvb 		    buf);
499f1b4288bSvb 		usage(B_FALSE);
500f1b4288bSvb 	}
501fa9e4066Sahrens 	if (buf[1] == '\0' || (toupper(buf[1]) == 'B' && buf[2] == '\0')) {
502fa9e4066Sahrens 		return (10*i);
503fa9e4066Sahrens 	}
504f1b4288bSvb 	(void) fprintf(stderr, "ztest: invalid bytes suffix: %s\n", buf);
505f1b4288bSvb 	usage(B_FALSE);
506f1b4288bSvb 	/* NOTREACHED */
507fa9e4066Sahrens }
508fa9e4066Sahrens 
509fa9e4066Sahrens static uint64_t
510fa9e4066Sahrens nicenumtoull(const char *buf)
511fa9e4066Sahrens {
512fa9e4066Sahrens 	char *end;
513fa9e4066Sahrens 	uint64_t val;
514fa9e4066Sahrens 
515fa9e4066Sahrens 	val = strtoull(buf, &end, 0);
516fa9e4066Sahrens 	if (end == buf) {
517f1b4288bSvb 		(void) fprintf(stderr, "ztest: bad numeric value: %s\n", buf);
518f1b4288bSvb 		usage(B_FALSE);
519fa9e4066Sahrens 	} else if (end[0] == '.') {
520fa9e4066Sahrens 		double fval = strtod(buf, &end);
521fa9e4066Sahrens 		fval *= pow(2, str2shift(end));
522f1b4288bSvb 		if (fval > UINT64_MAX) {
523f1b4288bSvb 			(void) fprintf(stderr, "ztest: value too large: %s\n",
524f1b4288bSvb 			    buf);
525f1b4288bSvb 			usage(B_FALSE);
526f1b4288bSvb 		}
527fa9e4066Sahrens 		val = (uint64_t)fval;
528fa9e4066Sahrens 	} else {
529fa9e4066Sahrens 		int shift = str2shift(end);
530f1b4288bSvb 		if (shift >= 64 || (val << shift) >> shift != val) {
531f1b4288bSvb 			(void) fprintf(stderr, "ztest: value too large: %s\n",
532f1b4288bSvb 			    buf);
533f1b4288bSvb 			usage(B_FALSE);
534f1b4288bSvb 		}
535fa9e4066Sahrens 		val <<= shift;
536fa9e4066Sahrens 	}
537fa9e4066Sahrens 	return (val);
538fa9e4066Sahrens }
539fa9e4066Sahrens 
540fa9e4066Sahrens static void
541f1b4288bSvb usage(boolean_t requested)
542fa9e4066Sahrens {
543420dfc95SChris Siden 	const ztest_shared_opts_t *zo = &ztest_opts_defaults;
544420dfc95SChris Siden 
545fa9e4066Sahrens 	char nice_vdev_size[10];
546fa9e4066Sahrens 	char nice_gang_bang[10];
547f1b4288bSvb 	FILE *fp = requested ? stdout : stderr;
548fa9e4066Sahrens 
549420dfc95SChris Siden 	nicenum(zo->zo_vdev_size, nice_vdev_size);
550420dfc95SChris Siden 	nicenum(zo->zo_metaslab_gang_bang, nice_gang_bang);
551fa9e4066Sahrens 
552f1b4288bSvb 	(void) fprintf(fp, "Usage: %s\n"
553fa9e4066Sahrens 	    "\t[-v vdevs (default: %llu)]\n"
554fa9e4066Sahrens 	    "\t[-s size_of_each_vdev (default: %s)]\n"
5552215e990SMark J Musante 	    "\t[-a alignment_shift (default: %d)] use 0 for random\n"
556fa9e4066Sahrens 	    "\t[-m mirror_copies (default: %d)]\n"
557fa9e4066Sahrens 	    "\t[-r raidz_disks (default: %d)]\n"
55899653d4eSeschrock 	    "\t[-R raidz_parity (default: %d)]\n"
559fa9e4066Sahrens 	    "\t[-d datasets (default: %d)]\n"
560fa9e4066Sahrens 	    "\t[-t threads (default: %d)]\n"
561fa9e4066Sahrens 	    "\t[-g gang_block_threshold (default: %s)]\n"
5622215e990SMark J Musante 	    "\t[-i init_count (default: %d)] initialize pool i times\n"
5632215e990SMark J Musante 	    "\t[-k kill_percentage (default: %llu%%)]\n"
564fa9e4066Sahrens 	    "\t[-p pool_name (default: %s)]\n"
5652215e990SMark J Musante 	    "\t[-f dir (default: %s)] file directory for vdev files\n"
5662215e990SMark J Musante 	    "\t[-V] verbose (use multiple times for ever more blather)\n"
5672215e990SMark J Musante 	    "\t[-E] use existing pool instead of creating new one\n"
5682215e990SMark J Musante 	    "\t[-T time (default: %llu sec)] total run time\n"
5692215e990SMark J Musante 	    "\t[-F freezeloops (default: %llu)] max loops in spa_freeze()\n"
5702215e990SMark J Musante 	    "\t[-P passtime (default: %llu sec)] time per pass\n"
571420dfc95SChris Siden 	    "\t[-B alt_ztest (default: <none>)] alternate ztest path\n"
572f1b4288bSvb 	    "\t[-h] (print help)\n"
573fa9e4066Sahrens 	    "",
574420dfc95SChris Siden 	    zo->zo_pool,
575420dfc95SChris Siden 	    (u_longlong_t)zo->zo_vdevs,			/* -v */
5760a4e9518Sgw 	    nice_vdev_size,				/* -s */
577420dfc95SChris Siden 	    zo->zo_ashift,				/* -a */
578420dfc95SChris Siden 	    zo->zo_mirrors,				/* -m */
579420dfc95SChris Siden 	    zo->zo_raidz,				/* -r */
580420dfc95SChris Siden 	    zo->zo_raidz_parity,			/* -R */
581420dfc95SChris Siden 	    zo->zo_datasets,				/* -d */
582420dfc95SChris Siden 	    zo->zo_threads,				/* -t */
5830a4e9518Sgw 	    nice_gang_bang,				/* -g */
584420dfc95SChris Siden 	    zo->zo_init,				/* -i */
585420dfc95SChris Siden 	    (u_longlong_t)zo->zo_killrate,		/* -k */
586420dfc95SChris Siden 	    zo->zo_pool,				/* -p */
587420dfc95SChris Siden 	    zo->zo_dir,					/* -f */
588420dfc95SChris Siden 	    (u_longlong_t)zo->zo_time,			/* -T */
589420dfc95SChris Siden 	    (u_longlong_t)zo->zo_maxloops,		/* -F */
590420dfc95SChris Siden 	    (u_longlong_t)zo->zo_passtime);
591f1b4288bSvb 	exit(requested ? 0 : 1);
592fa9e4066Sahrens }
593fa9e4066Sahrens 
594fa9e4066Sahrens static void
595fa9e4066Sahrens process_options(int argc, char **argv)
596fa9e4066Sahrens {
597420dfc95SChris Siden 	char *path;
598420dfc95SChris Siden 	ztest_shared_opts_t *zo = &ztest_opts;
599420dfc95SChris Siden 
600fa9e4066Sahrens 	int opt;
601fa9e4066Sahrens 	uint64_t value;
602420dfc95SChris Siden 	char altdir[MAXNAMELEN] = { 0 };
603fa9e4066Sahrens 
604420dfc95SChris Siden 	bcopy(&ztest_opts_defaults, zo, sizeof (*zo));
605fa9e4066Sahrens 
606fa9e4066Sahrens 	while ((opt = getopt(argc, argv,
607420dfc95SChris Siden 	    "v:s:a:m:r:R:d:t:g:i:k:p:f:VET:P:hF:B:")) != EOF) {
608fa9e4066Sahrens 		value = 0;
609fa9e4066Sahrens 		switch (opt) {
6103d7072f8Seschrock 		case 'v':
6113d7072f8Seschrock 		case 's':
6123d7072f8Seschrock 		case 'a':
6133d7072f8Seschrock 		case 'm':
6143d7072f8Seschrock 		case 'r':
6153d7072f8Seschrock 		case 'R':
6163d7072f8Seschrock 		case 'd':
6173d7072f8Seschrock 		case 't':
6183d7072f8Seschrock 		case 'g':
6193d7072f8Seschrock 		case 'i':
6203d7072f8Seschrock 		case 'k':
6213d7072f8Seschrock 		case 'T':
6223d7072f8Seschrock 		case 'P':
6232215e990SMark J Musante 		case 'F':
624fa9e4066Sahrens 			value = nicenumtoull(optarg);
625fa9e4066Sahrens 		}
626fa9e4066Sahrens 		switch (opt) {
6273d7072f8Seschrock 		case 'v':
628420dfc95SChris Siden 			zo->zo_vdevs = value;
629fa9e4066Sahrens 			break;
6303d7072f8Seschrock 		case 's':
631420dfc95SChris Siden 			zo->zo_vdev_size = MAX(SPA_MINDEVSIZE, value);
632fa9e4066Sahrens 			break;
6333d7072f8Seschrock 		case 'a':
634420dfc95SChris Siden 			zo->zo_ashift = value;
635ecc2d604Sbonwick 			break;
6363d7072f8Seschrock 		case 'm':
637420dfc95SChris Siden 			zo->zo_mirrors = value;
638fa9e4066Sahrens 			break;
6393d7072f8Seschrock 		case 'r':
640420dfc95SChris Siden 			zo->zo_raidz = MAX(1, value);
641fa9e4066Sahrens 			break;
6423d7072f8Seschrock 		case 'R':
643420dfc95SChris Siden 			zo->zo_raidz_parity = MIN(MAX(value, 1), 3);
64499653d4eSeschrock 			break;
6453d7072f8Seschrock 		case 'd':
646420dfc95SChris Siden 			zo->zo_datasets = MAX(1, value);
647fa9e4066Sahrens 			break;
6483d7072f8Seschrock 		case 't':
649420dfc95SChris Siden 			zo->zo_threads = MAX(1, value);
650fa9e4066Sahrens 			break;
6513d7072f8Seschrock 		case 'g':
652420dfc95SChris Siden 			zo->zo_metaslab_gang_bang = MAX(SPA_MINBLOCKSIZE << 1,
653420dfc95SChris Siden 			    value);
654fa9e4066Sahrens 			break;
6553d7072f8Seschrock 		case 'i':
656420dfc95SChris Siden 			zo->zo_init = value;
657fa9e4066Sahrens 			break;
6583d7072f8Seschrock 		case 'k':
659420dfc95SChris Siden 			zo->zo_killrate = value;
660fa9e4066Sahrens 			break;
6613d7072f8Seschrock 		case 'p':
662420dfc95SChris Siden 			(void) strlcpy(zo->zo_pool, optarg,
663420dfc95SChris Siden 			    sizeof (zo->zo_pool));
664fa9e4066Sahrens 			break;
6653d7072f8Seschrock 		case 'f':
666420dfc95SChris Siden 			path = realpath(optarg, NULL);
667420dfc95SChris Siden 			if (path == NULL) {
668420dfc95SChris Siden 				(void) fprintf(stderr, "error: %s: %s\n",
669420dfc95SChris Siden 				    optarg, strerror(errno));
670420dfc95SChris Siden 				usage(B_FALSE);
671420dfc95SChris Siden 			} else {
672420dfc95SChris Siden 				(void) strlcpy(zo->zo_dir, path,
673420dfc95SChris Siden 				    sizeof (zo->zo_dir));
674420dfc95SChris Siden 			}
675fa9e4066Sahrens 			break;
6763d7072f8Seschrock 		case 'V':
677420dfc95SChris Siden 			zo->zo_verbose++;
678fa9e4066Sahrens 			break;
6793d7072f8Seschrock 		case 'E':
680420dfc95SChris Siden 			zo->zo_init = 0;
681fa9e4066Sahrens 			break;
6823d7072f8Seschrock 		case 'T':
683420dfc95SChris Siden 			zo->zo_time = value;
684fa9e4066Sahrens 			break;
6853d7072f8Seschrock 		case 'P':
686420dfc95SChris Siden 			zo->zo_passtime = MAX(1, value);
687fa9e4066Sahrens 			break;
6882215e990SMark J Musante 		case 'F':
689420dfc95SChris Siden 			zo->zo_maxloops = MAX(1, value);
690420dfc95SChris Siden 			break;
691420dfc95SChris Siden 		case 'B':
692420dfc95SChris Siden 			(void) strlcpy(altdir, optarg, sizeof (altdir));
6932215e990SMark J Musante 			break;
6943d7072f8Seschrock 		case 'h':
695f1b4288bSvb 			usage(B_TRUE);
696f1b4288bSvb 			break;
6973d7072f8Seschrock 		case '?':
6983d7072f8Seschrock 		default:
699f1b4288bSvb 			usage(B_FALSE);
700fa9e4066Sahrens 			break;
701fa9e4066Sahrens 		}
702fa9e4066Sahrens 	}
703fa9e4066Sahrens 
704420dfc95SChris Siden 	zo->zo_raidz_parity = MIN(zo->zo_raidz_parity, zo->zo_raidz - 1);
70599653d4eSeschrock 
706420dfc95SChris Siden 	zo->zo_vdevtime =
707420dfc95SChris Siden 	    (zo->zo_vdevs > 0 ? zo->zo_time * NANOSEC / zo->zo_vdevs :
708b24ab676SJeff Bonwick 	    UINT64_MAX >> 2);
709420dfc95SChris Siden 
710420dfc95SChris Siden 	if (strlen(altdir) > 0) {
711420dfc95SChris Siden 		char cmd[MAXNAMELEN];
712420dfc95SChris Siden 		char realaltdir[MAXNAMELEN];
713420dfc95SChris Siden 		char *bin;
714420dfc95SChris Siden 		char *ztest;
715420dfc95SChris Siden 		char *isa;
716420dfc95SChris Siden 		int isalen;
717420dfc95SChris Siden 
718420dfc95SChris Siden 		(void) realpath(getexecname(), cmd);
719420dfc95SChris Siden 		if (0 != access(altdir, F_OK)) {
720420dfc95SChris Siden 			ztest_dump_core = B_FALSE;
721420dfc95SChris Siden 			fatal(B_TRUE, "invalid alternate ztest path: %s",
722420dfc95SChris Siden 			    altdir);
723420dfc95SChris Siden 		}
724420dfc95SChris Siden 		VERIFY(NULL != realpath(altdir, realaltdir));
725420dfc95SChris Siden 
726420dfc95SChris Siden 		/*
727420dfc95SChris Siden 		 * 'cmd' should be of the form "<anything>/usr/bin/<isa>/ztest".
728420dfc95SChris Siden 		 * We want to extract <isa> to determine if we should use
729420dfc95SChris Siden 		 * 32 or 64 bit binaries.
730420dfc95SChris Siden 		 */
731420dfc95SChris Siden 		bin = strstr(cmd, "/usr/bin/");
732420dfc95SChris Siden 		ztest = strstr(bin, "/ztest");
733420dfc95SChris Siden 		isa = bin + 9;
734420dfc95SChris Siden 		isalen = ztest - isa;
735420dfc95SChris Siden 		(void) snprintf(zo->zo_alt_ztest, sizeof (zo->zo_alt_ztest),
736420dfc95SChris Siden 		    "%s/usr/bin/%.*s/ztest", realaltdir, isalen, isa);
737420dfc95SChris Siden 		(void) snprintf(zo->zo_alt_libpath, sizeof (zo->zo_alt_libpath),
738420dfc95SChris Siden 		    "%s/usr/lib/%.*s", realaltdir, isalen, isa);
739420dfc95SChris Siden 
740420dfc95SChris Siden 		if (0 != access(zo->zo_alt_ztest, X_OK)) {
741420dfc95SChris Siden 			ztest_dump_core = B_FALSE;
742420dfc95SChris Siden 			fatal(B_TRUE, "invalid alternate ztest: %s",
743420dfc95SChris Siden 			    zo->zo_alt_ztest);
744420dfc95SChris Siden 		} else if (0 != access(zo->zo_alt_libpath, X_OK)) {
745420dfc95SChris Siden 			ztest_dump_core = B_FALSE;
746420dfc95SChris Siden 			fatal(B_TRUE, "invalid alternate lib directory %s",
747420dfc95SChris Siden 			    zo->zo_alt_libpath);
748420dfc95SChris Siden 		}
749420dfc95SChris Siden 	}
750fa9e4066Sahrens }
751fa9e4066Sahrens 
752b24ab676SJeff Bonwick static void
753b24ab676SJeff Bonwick ztest_kill(ztest_shared_t *zs)
754b24ab676SJeff Bonwick {
755420dfc95SChris Siden 	zs->zs_alloc = metaslab_class_get_alloc(spa_normal_class(ztest_spa));
756420dfc95SChris Siden 	zs->zs_space = metaslab_class_get_space(spa_normal_class(ztest_spa));
757b24ab676SJeff Bonwick 	(void) kill(getpid(), SIGKILL);
758b24ab676SJeff Bonwick }
759b24ab676SJeff Bonwick 
760b24ab676SJeff Bonwick static uint64_t
761b24ab676SJeff Bonwick ztest_random(uint64_t range)
762b24ab676SJeff Bonwick {
763b24ab676SJeff Bonwick 	uint64_t r;
764b24ab676SJeff Bonwick 
765b24ab676SJeff Bonwick 	if (range == 0)
766b24ab676SJeff Bonwick 		return (0);
767b24ab676SJeff Bonwick 
768420dfc95SChris Siden 	if (read(ZTEST_FD_RAND, &r, sizeof (r)) != sizeof (r))
769b24ab676SJeff Bonwick 		fatal(1, "short read from /dev/urandom");
770b24ab676SJeff Bonwick 
771b24ab676SJeff Bonwick 	return (r % range);
772b24ab676SJeff Bonwick }
773b24ab676SJeff Bonwick 
774b24ab676SJeff Bonwick /* ARGSUSED */
775b24ab676SJeff Bonwick static void
776b24ab676SJeff Bonwick ztest_record_enospc(const char *s)
777b24ab676SJeff Bonwick {
778b24ab676SJeff Bonwick 	ztest_shared->zs_enospc_count++;
779b24ab676SJeff Bonwick }
780b24ab676SJeff Bonwick 
781ecc2d604Sbonwick static uint64_t
782ecc2d604Sbonwick ztest_get_ashift(void)
783ecc2d604Sbonwick {
784420dfc95SChris Siden 	if (ztest_opts.zo_ashift == 0)
785ecc2d604Sbonwick 		return (SPA_MINBLOCKSHIFT + ztest_random(3));
786420dfc95SChris Siden 	return (ztest_opts.zo_ashift);
787ecc2d604Sbonwick }
788ecc2d604Sbonwick 
789fa9e4066Sahrens static nvlist_t *
790e14bb325SJeff Bonwick make_vdev_file(char *path, char *aux, size_t size, uint64_t ashift)
791fa9e4066Sahrens {
792e14bb325SJeff Bonwick 	char pathbuf[MAXPATHLEN];
793fa9e4066Sahrens 	uint64_t vdev;
794fa9e4066Sahrens 	nvlist_t *file;
795fa9e4066Sahrens 
796e14bb325SJeff Bonwick 	if (ashift == 0)
797e14bb325SJeff Bonwick 		ashift = ztest_get_ashift();
798e14bb325SJeff Bonwick 
799e14bb325SJeff Bonwick 	if (path == NULL) {
800e14bb325SJeff Bonwick 		path = pathbuf;
801fa9e4066Sahrens 
802e14bb325SJeff Bonwick 		if (aux != NULL) {
803e14bb325SJeff Bonwick 			vdev = ztest_shared->zs_vdev_aux;
804420dfc95SChris Siden 			(void) snprintf(path, sizeof (pathbuf),
805420dfc95SChris Siden 			    ztest_aux_template, ztest_opts.zo_dir,
806420dfc95SChris Siden 			    ztest_opts.zo_pool, aux, vdev);
807e14bb325SJeff Bonwick 		} else {
80888ecc943SGeorge Wilson 			vdev = ztest_shared->zs_vdev_next_leaf++;
809420dfc95SChris Siden 			(void) snprintf(path, sizeof (pathbuf),
810420dfc95SChris Siden 			    ztest_dev_template, ztest_opts.zo_dir,
811420dfc95SChris Siden 			    ztest_opts.zo_pool, vdev);
812e14bb325SJeff Bonwick 		}
813e14bb325SJeff Bonwick 	}
814e14bb325SJeff Bonwick 
815e14bb325SJeff Bonwick 	if (size != 0) {
816e14bb325SJeff Bonwick 		int fd = open(path, O_RDWR | O_CREAT | O_TRUNC, 0666);
817fa9e4066Sahrens 		if (fd == -1)
818e14bb325SJeff Bonwick 			fatal(1, "can't open %s", path);
819fa9e4066Sahrens 		if (ftruncate(fd, size) != 0)
820e14bb325SJeff Bonwick 			fatal(1, "can't ftruncate %s", path);
821fa9e4066Sahrens 		(void) close(fd);
822fa9e4066Sahrens 	}
823fa9e4066Sahrens 
824fa9e4066Sahrens 	VERIFY(nvlist_alloc(&file, NV_UNIQUE_NAME, 0) == 0);
825fa9e4066Sahrens 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_TYPE, VDEV_TYPE_FILE) == 0);
826e14bb325SJeff Bonwick 	VERIFY(nvlist_add_string(file, ZPOOL_CONFIG_PATH, path) == 0);
827ecc2d604Sbonwick 	VERIFY(nvlist_add_uint64(file, ZPOOL_CONFIG_ASHIFT, ashift) == 0);
828fa9e4066Sahrens 
829fa9e4066Sahrens 	return (file);
830fa9e4066Sahrens }
831fa9e4066Sahrens 
832fa9e4066Sahrens static nvlist_t *
833e14bb325SJeff Bonwick make_vdev_raidz(char *path, char *aux, size_t size, uint64_t ashift, int r)
834fa9e4066Sahrens {
835fa9e4066Sahrens 	nvlist_t *raidz, **child;
836fa9e4066Sahrens 	int c;
837fa9e4066Sahrens 
838fa9e4066Sahrens 	if (r < 2)
839e14bb325SJeff Bonwick 		return (make_vdev_file(path, aux, size, ashift));
840fa9e4066Sahrens 	child = umem_alloc(r * sizeof (nvlist_t *), UMEM_NOFAIL);
841fa9e4066Sahrens 
842fa9e4066Sahrens 	for (c = 0; c < r; c++)
843e14bb325SJeff Bonwick 		child[c] = make_vdev_file(path, aux, size, ashift);
844fa9e4066Sahrens 
845fa9e4066Sahrens 	VERIFY(nvlist_alloc(&raidz, NV_UNIQUE_NAME, 0) == 0);
846fa9e4066Sahrens 	VERIFY(nvlist_add_string(raidz, ZPOOL_CONFIG_TYPE,
847fa9e4066Sahrens 	    VDEV_TYPE_RAIDZ) == 0);
84899653d4eSeschrock 	VERIFY(nvlist_add_uint64(raidz, ZPOOL_CONFIG_NPARITY,
849420dfc95SChris Siden 	    ztest_opts.zo_raidz_parity) == 0);
850fa9e4066Sahrens 	VERIFY(nvlist_add_nvlist_array(raidz, ZPOOL_CONFIG_CHILDREN,
851fa9e4066Sahrens 	    child, r) == 0);
852fa9e4066Sahrens 
853fa9e4066Sahrens 	for (c = 0; c < r; c++)
854fa9e4066Sahrens 		nvlist_free(child[c]);
855fa9e4066Sahrens 
856fa9e4066Sahrens 	umem_free(child, r * sizeof (nvlist_t *));
857fa9e4066Sahrens 
858fa9e4066Sahrens 	return (raidz);
859fa9e4066Sahrens }
860fa9e4066Sahrens 
861fa9e4066Sahrens static nvlist_t *
862e14bb325SJeff Bonwick make_vdev_mirror(char *path, char *aux, size_t size, uint64_t ashift,
86331157203SJeff Bonwick 	int r, int m)
864fa9e4066Sahrens {
865fa9e4066Sahrens 	nvlist_t *mirror, **child;
866fa9e4066Sahrens 	int c;
867fa9e4066Sahrens 
868fa9e4066Sahrens 	if (m < 1)
869e14bb325SJeff Bonwick 		return (make_vdev_raidz(path, aux, size, ashift, r));
870fa9e4066Sahrens 
871fa9e4066Sahrens 	child = umem_alloc(m * sizeof (nvlist_t *), UMEM_NOFAIL);
872fa9e4066Sahrens 
873fa9e4066Sahrens 	for (c = 0; c < m; c++)
874e14bb325SJeff Bonwick 		child[c] = make_vdev_raidz(path, aux, size, ashift, r);
875fa9e4066Sahrens 
876fa9e4066Sahrens 	VERIFY(nvlist_alloc(&mirror, NV_UNIQUE_NAME, 0) == 0);
877fa9e4066Sahrens 	VERIFY(nvlist_add_string(mirror, ZPOOL_CONFIG_TYPE,
878fa9e4066Sahrens 	    VDEV_TYPE_MIRROR) == 0);
879fa9e4066Sahrens 	VERIFY(nvlist_add_nvlist_array(mirror, ZPOOL_CONFIG_CHILDREN,
880fa9e4066Sahrens 	    child, m) == 0);
881fa9e4066Sahrens 
882fa9e4066Sahrens 	for (c = 0; c < m; c++)
883fa9e4066Sahrens 		nvlist_free(child[c]);
884fa9e4066Sahrens 
885fa9e4066Sahrens 	umem_free(child, m * sizeof (nvlist_t *));
886fa9e4066Sahrens 
887fa9e4066Sahrens 	return (mirror);
888fa9e4066Sahrens }
889fa9e4066Sahrens 
890fa9e4066Sahrens static nvlist_t *
891e14bb325SJeff Bonwick make_vdev_root(char *path, char *aux, size_t size, uint64_t ashift,
892e14bb325SJeff Bonwick 	int log, int r, int m, int t)
893fa9e4066Sahrens {
894fa9e4066Sahrens 	nvlist_t *root, **child;
895fa9e4066Sahrens 	int c;
896fa9e4066Sahrens 
897fa9e4066Sahrens 	ASSERT(t > 0);
898fa9e4066Sahrens 
899fa9e4066Sahrens 	child = umem_alloc(t * sizeof (nvlist_t *), UMEM_NOFAIL);
900fa9e4066Sahrens 
90131157203SJeff Bonwick 	for (c = 0; c < t; c++) {
90231157203SJeff Bonwick 		child[c] = make_vdev_mirror(path, aux, size, ashift, r, m);
90331157203SJeff Bonwick 		VERIFY(nvlist_add_uint64(child[c], ZPOOL_CONFIG_IS_LOG,
90431157203SJeff Bonwick 		    log) == 0);
90531157203SJeff Bonwick 	}
906fa9e4066Sahrens 
907fa9e4066Sahrens 	VERIFY(nvlist_alloc(&root, NV_UNIQUE_NAME, 0) == 0);
908fa9e4066Sahrens 	VERIFY(nvlist_add_string(root, ZPOOL_CONFIG_TYPE, VDEV_TYPE_ROOT) == 0);
909e14bb325SJeff Bonwick 	VERIFY(nvlist_add_nvlist_array(root, aux ? aux : ZPOOL_CONFIG_CHILDREN,
910fa9e4066Sahrens 	    child, t) == 0);
911fa9e4066Sahrens 
912fa9e4066Sahrens 	for (c = 0; c < t; c++)
913fa9e4066Sahrens 		nvlist_free(child[c]);
914fa9e4066Sahrens 
915fa9e4066Sahrens 	umem_free(child, t * sizeof (nvlist_t *));
916fa9e4066Sahrens 
917fa9e4066Sahrens 	return (root);
918fa9e4066Sahrens }
919fa9e4066Sahrens 
920b24ab676SJeff Bonwick static int
921b24ab676SJeff Bonwick ztest_random_blocksize(void)
922fa9e4066Sahrens {
923b24ab676SJeff Bonwick 	return (1 << (SPA_MINBLOCKSHIFT +
924b24ab676SJeff Bonwick 	    ztest_random(SPA_MAXBLOCKSHIFT - SPA_MINBLOCKSHIFT + 1)));
925b24ab676SJeff Bonwick }
926fa9e4066Sahrens 
927b24ab676SJeff Bonwick static int
928b24ab676SJeff Bonwick ztest_random_ibshift(void)
929b24ab676SJeff Bonwick {
930b24ab676SJeff Bonwick 	return (DN_MIN_INDBLKSHIFT +
931b24ab676SJeff Bonwick 	    ztest_random(DN_MAX_INDBLKSHIFT - DN_MIN_INDBLKSHIFT + 1));
932fa9e4066Sahrens }
933fa9e4066Sahrens 
934b24ab676SJeff Bonwick static uint64_t
935b24ab676SJeff Bonwick ztest_random_vdev_top(spa_t *spa, boolean_t log_ok)
936fa9e4066Sahrens {
937b24ab676SJeff Bonwick 	uint64_t top;
938b24ab676SJeff Bonwick 	vdev_t *rvd = spa->spa_root_vdev;
939b24ab676SJeff Bonwick 	vdev_t *tvd;
940fa9e4066Sahrens 
941b24ab676SJeff Bonwick 	ASSERT(spa_config_held(spa, SCL_ALL, RW_READER) != 0);
942fa9e4066Sahrens 
943b24ab676SJeff Bonwick 	do {
944b24ab676SJeff Bonwick 		top = ztest_random(rvd->vdev_children);
945b24ab676SJeff Bonwick 		tvd = rvd->vdev_child[top];
946b24ab676SJeff Bonwick 	} while (tvd->vdev_ishole || (tvd->vdev_islog && !log_ok) ||
947b24ab676SJeff Bonwick 	    tvd->vdev_mg == NULL || tvd->vdev_mg->mg_class == NULL);
948fa9e4066Sahrens 
949b24ab676SJeff Bonwick 	return (top);
950fa9e4066Sahrens }
951fa9e4066Sahrens 
952b24ab676SJeff Bonwick static uint64_t
953b24ab676SJeff Bonwick ztest_random_dsl_prop(zfs_prop_t prop)
954fa9e4066Sahrens {
955b24ab676SJeff Bonwick 	uint64_t value;
956b24ab676SJeff Bonwick 
957b24ab676SJeff Bonwick 	do {
958b24ab676SJeff Bonwick 		value = zfs_prop_random_value(prop, ztest_random(-1ULL));
959b24ab676SJeff Bonwick 	} while (prop == ZFS_PROP_CHECKSUM && value == ZIO_CHECKSUM_OFF);
960b24ab676SJeff Bonwick 
961b24ab676SJeff Bonwick 	return (value);
962fa9e4066Sahrens }
963fa9e4066Sahrens 
964fa9e4066Sahrens static int
965b24ab676SJeff Bonwick ztest_dsl_prop_set_uint64(char *osname, zfs_prop_t prop, uint64_t value,
966b24ab676SJeff Bonwick     boolean_t inherit)
967fa9e4066Sahrens {
968b24ab676SJeff Bonwick 	const char *propname = zfs_prop_to_name(prop);
969b24ab676SJeff Bonwick 	const char *valname;
970b24ab676SJeff Bonwick 	char setpoint[MAXPATHLEN];
971b24ab676SJeff Bonwick 	uint64_t curval;
972fa9e4066Sahrens 	int error;
973fa9e4066Sahrens 
97492241e0bSTom Erickson 	error = dsl_prop_set(osname, propname,
97592241e0bSTom Erickson 	    (inherit ? ZPROP_SRC_NONE : ZPROP_SRC_LOCAL),
97692241e0bSTom Erickson 	    sizeof (value), 1, &value);
977fa9e4066Sahrens 
978b24ab676SJeff Bonwick 	if (error == ENOSPC) {
979b24ab676SJeff Bonwick 		ztest_record_enospc(FTAG);
980fa9e4066Sahrens 		return (error);
981fa9e4066Sahrens 	}
982*fb09f5aaSMadhav Suresh 	ASSERT0(error);
983fa9e4066Sahrens 
984b24ab676SJeff Bonwick 	VERIFY3U(dsl_prop_get(osname, propname, sizeof (curval),
985b24ab676SJeff Bonwick 	    1, &curval, setpoint), ==, 0);
986b24ab676SJeff Bonwick 
987420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 6) {
988b24ab676SJeff Bonwick 		VERIFY(zfs_prop_index_to_string(prop, curval, &valname) == 0);
989b24ab676SJeff Bonwick 		(void) printf("%s %s = %s at '%s'\n",
990b24ab676SJeff Bonwick 		    osname, propname, valname, setpoint);
991fa9e4066Sahrens 	}
992fa9e4066Sahrens 
993fa9e4066Sahrens 	return (error);
994fa9e4066Sahrens }
995fa9e4066Sahrens 
996fa9e4066Sahrens static int
997420dfc95SChris Siden ztest_spa_prop_set_uint64(zpool_prop_t prop, uint64_t value)
998fa9e4066Sahrens {
999420dfc95SChris Siden 	spa_t *spa = ztest_spa;
1000b24ab676SJeff Bonwick 	nvlist_t *props = NULL;
1001fa9e4066Sahrens 	int error;
1002fa9e4066Sahrens 
1003b24ab676SJeff Bonwick 	VERIFY(nvlist_alloc(&props, NV_UNIQUE_NAME, 0) == 0);
1004b24ab676SJeff Bonwick 	VERIFY(nvlist_add_uint64(props, zpool_prop_to_name(prop), value) == 0);
1005fa9e4066Sahrens 
1006b24ab676SJeff Bonwick 	error = spa_prop_set(spa, props);
1007b24ab676SJeff Bonwick 
1008b24ab676SJeff Bonwick 	nvlist_free(props);
1009b24ab676SJeff Bonwick 
1010b24ab676SJeff Bonwick 	if (error == ENOSPC) {
1011b24ab676SJeff Bonwick 		ztest_record_enospc(FTAG);
1012fa9e4066Sahrens 		return (error);
1013fa9e4066Sahrens 	}
1014*fb09f5aaSMadhav Suresh 	ASSERT0(error);
1015fa9e4066Sahrens 
1016fa9e4066Sahrens 	return (error);
1017fa9e4066Sahrens }
1018fa9e4066Sahrens 
1019b24ab676SJeff Bonwick static void
1020b24ab676SJeff Bonwick ztest_rll_init(rll_t *rll)
1021b24ab676SJeff Bonwick {
1022b24ab676SJeff Bonwick 	rll->rll_writer = NULL;
1023b24ab676SJeff Bonwick 	rll->rll_readers = 0;
1024b24ab676SJeff Bonwick 	VERIFY(_mutex_init(&rll->rll_lock, USYNC_THREAD, NULL) == 0);
1025b24ab676SJeff Bonwick 	VERIFY(cond_init(&rll->rll_cv, USYNC_THREAD, NULL) == 0);
1026b24ab676SJeff Bonwick }
1027fa9e4066Sahrens 
1028b24ab676SJeff Bonwick static void
1029b24ab676SJeff Bonwick ztest_rll_destroy(rll_t *rll)
1030fa9e4066Sahrens {
1031b24ab676SJeff Bonwick 	ASSERT(rll->rll_writer == NULL);
1032b24ab676SJeff Bonwick 	ASSERT(rll->rll_readers == 0);
1033b24ab676SJeff Bonwick 	VERIFY(_mutex_destroy(&rll->rll_lock) == 0);
1034b24ab676SJeff Bonwick 	VERIFY(cond_destroy(&rll->rll_cv) == 0);
1035b24ab676SJeff Bonwick }
1036fa9e4066Sahrens 
1037b24ab676SJeff Bonwick static void
1038b24ab676SJeff Bonwick ztest_rll_lock(rll_t *rll, rl_type_t type)
1039b24ab676SJeff Bonwick {
1040b24ab676SJeff Bonwick 	VERIFY(mutex_lock(&rll->rll_lock) == 0);
1041fa9e4066Sahrens 
1042b24ab676SJeff Bonwick 	if (type == RL_READER) {
1043b24ab676SJeff Bonwick 		while (rll->rll_writer != NULL)
1044b24ab676SJeff Bonwick 			(void) cond_wait(&rll->rll_cv, &rll->rll_lock);
1045b24ab676SJeff Bonwick 		rll->rll_readers++;
1046b24ab676SJeff Bonwick 	} else {
1047b24ab676SJeff Bonwick 		while (rll->rll_writer != NULL || rll->rll_readers)
1048b24ab676SJeff Bonwick 			(void) cond_wait(&rll->rll_cv, &rll->rll_lock);
1049b24ab676SJeff Bonwick 		rll->rll_writer = curthread;
1050b24ab676SJeff Bonwick 	}
1051fa9e4066Sahrens 
1052b24ab676SJeff Bonwick 	VERIFY(mutex_unlock(&rll->rll_lock) == 0);
1053b24ab676SJeff Bonwick }
1054fa9e4066Sahrens 
1055b24ab676SJeff Bonwick static void
1056b24ab676SJeff Bonwick ztest_rll_unlock(rll_t *rll)
1057b24ab676SJeff Bonwick {
1058b24ab676SJeff Bonwick 	VERIFY(mutex_lock(&rll->rll_lock) == 0);
1059fa9e4066Sahrens 
1060b24ab676SJeff Bonwick 	if (rll->rll_writer) {
1061b24ab676SJeff Bonwick 		ASSERT(rll->rll_readers == 0);
1062b24ab676SJeff Bonwick 		rll->rll_writer = NULL;
1063b24ab676SJeff Bonwick 	} else {
1064b24ab676SJeff Bonwick 		ASSERT(rll->rll_readers != 0);
1065b24ab676SJeff Bonwick 		ASSERT(rll->rll_writer == NULL);
1066b24ab676SJeff Bonwick 		rll->rll_readers--;
1067b24ab676SJeff Bonwick 	}
1068fa9e4066Sahrens 
1069b24ab676SJeff Bonwick 	if (rll->rll_writer == NULL && rll->rll_readers == 0)
1070b24ab676SJeff Bonwick 		VERIFY(cond_broadcast(&rll->rll_cv) == 0);
1071b24ab676SJeff Bonwick 
1072b24ab676SJeff Bonwick 	VERIFY(mutex_unlock(&rll->rll_lock) == 0);
1073fa9e4066Sahrens }
1074fa9e4066Sahrens 
1075b24ab676SJeff Bonwick static void
1076b24ab676SJeff Bonwick ztest_object_lock(ztest_ds_t *zd, uint64_t object, rl_type_t type)
107731157203SJeff Bonwick {
1078b24ab676SJeff Bonwick 	rll_t *rll = &zd->zd_object_lock[object & (ZTEST_OBJECT_LOCKS - 1)];
107931157203SJeff Bonwick 
1080b24ab676SJeff Bonwick 	ztest_rll_lock(rll, type);
1081b24ab676SJeff Bonwick }
108231157203SJeff Bonwick 
1083b24ab676SJeff Bonwick static void
1084b24ab676SJeff Bonwick ztest_object_unlock(ztest_ds_t *zd, uint64_t object)
1085b24ab676SJeff Bonwick {
1086b24ab676SJeff Bonwick 	rll_t *rll = &zd->zd_object_lock[object & (ZTEST_OBJECT_LOCKS - 1)];
108731157203SJeff Bonwick 
1088b24ab676SJeff Bonwick 	ztest_rll_unlock(rll);
108931157203SJeff Bonwick }
109031157203SJeff Bonwick 
1091b24ab676SJeff Bonwick static rl_t *
1092b24ab676SJeff Bonwick ztest_range_lock(ztest_ds_t *zd, uint64_t object, uint64_t offset,
1093b24ab676SJeff Bonwick     uint64_t size, rl_type_t type)
109488ecc943SGeorge Wilson {
1095b24ab676SJeff Bonwick 	uint64_t hash = object ^ (offset % (ZTEST_RANGE_LOCKS + 1));
1096b24ab676SJeff Bonwick 	rll_t *rll = &zd->zd_range_lock[hash & (ZTEST_RANGE_LOCKS - 1)];
1097b24ab676SJeff Bonwick 	rl_t *rl;
109888ecc943SGeorge Wilson 
1099b24ab676SJeff Bonwick 	rl = umem_alloc(sizeof (*rl), UMEM_NOFAIL);
1100b24ab676SJeff Bonwick 	rl->rl_object = object;
1101b24ab676SJeff Bonwick 	rl->rl_offset = offset;
1102b24ab676SJeff Bonwick 	rl->rl_size = size;
1103b24ab676SJeff Bonwick 	rl->rl_lock = rll;
110488ecc943SGeorge Wilson 
1105b24ab676SJeff Bonwick 	ztest_rll_lock(rll, type);
110688ecc943SGeorge Wilson 
1107b24ab676SJeff Bonwick 	return (rl);
110888ecc943SGeorge Wilson }
110988ecc943SGeorge Wilson 
1110b24ab676SJeff Bonwick static void
1111b24ab676SJeff Bonwick ztest_range_unlock(rl_t *rl)
1112fa9e4066Sahrens {
1113b24ab676SJeff Bonwick 	rll_t *rll = rl->rl_lock;
1114fa9e4066Sahrens 
1115b24ab676SJeff Bonwick 	ztest_rll_unlock(rll);
1116fa9e4066Sahrens 
1117b24ab676SJeff Bonwick 	umem_free(rl, sizeof (*rl));
1118b24ab676SJeff Bonwick }
1119fa9e4066Sahrens 
1120b24ab676SJeff Bonwick static void
1121420dfc95SChris Siden ztest_zd_init(ztest_ds_t *zd, ztest_shared_ds_t *szd, objset_t *os)
1122b24ab676SJeff Bonwick {
1123b24ab676SJeff Bonwick 	zd->zd_os = os;
1124b24ab676SJeff Bonwick 	zd->zd_zilog = dmu_objset_zil(os);
1125420dfc95SChris Siden 	zd->zd_shared = szd;
1126b24ab676SJeff Bonwick 	dmu_objset_name(os, zd->zd_name);
1127fa9e4066Sahrens 
1128420dfc95SChris Siden 	if (zd->zd_shared != NULL)
1129420dfc95SChris Siden 		zd->zd_shared->zd_seq = 0;
1130420dfc95SChris Siden 
1131c9ba2a43SEric Schrock 	VERIFY(rwlock_init(&zd->zd_zilog_lock, USYNC_THREAD, NULL) == 0);
1132b24ab676SJeff Bonwick 	VERIFY(_mutex_init(&zd->zd_dirobj_lock, USYNC_THREAD, NULL) == 0);
1133b24ab676SJeff Bonwick 
1134b24ab676SJeff Bonwick 	for (int l = 0; l < ZTEST_OBJECT_LOCKS; l++)
1135b24ab676SJeff Bonwick 		ztest_rll_init(&zd->zd_object_lock[l]);
1136b24ab676SJeff Bonwick 
1137b24ab676SJeff Bonwick 	for (int l = 0; l < ZTEST_RANGE_LOCKS; l++)
1138b24ab676SJeff Bonwick 		ztest_rll_init(&zd->zd_range_lock[l]);
1139b24ab676SJeff Bonwick }
1140b24ab676SJeff Bonwick 
1141b24ab676SJeff Bonwick static void
1142b24ab676SJeff Bonwick ztest_zd_fini(ztest_ds_t *zd)
1143b24ab676SJeff Bonwick {
1144b24ab676SJeff Bonwick 	VERIFY(_mutex_destroy(&zd->zd_dirobj_lock) == 0);
1145b24ab676SJeff Bonwick 
1146b24ab676SJeff Bonwick 	for (int l = 0; l < ZTEST_OBJECT_LOCKS; l++)
1147b24ab676SJeff Bonwick 		ztest_rll_destroy(&zd->zd_object_lock[l]);
1148b24ab676SJeff Bonwick 
1149b24ab676SJeff Bonwick 	for (int l = 0; l < ZTEST_RANGE_LOCKS; l++)
1150b24ab676SJeff Bonwick 		ztest_rll_destroy(&zd->zd_range_lock[l]);
1151b24ab676SJeff Bonwick }
1152b24ab676SJeff Bonwick 
1153b24ab676SJeff Bonwick #define	TXG_MIGHTWAIT	(ztest_random(10) == 0 ? TXG_NOWAIT : TXG_WAIT)
1154b24ab676SJeff Bonwick 
1155b24ab676SJeff Bonwick static uint64_t
1156b24ab676SJeff Bonwick ztest_tx_assign(dmu_tx_t *tx, uint64_t txg_how, const char *tag)
1157b24ab676SJeff Bonwick {
1158b24ab676SJeff Bonwick 	uint64_t txg;
1159b24ab676SJeff Bonwick 	int error;
1160b24ab676SJeff Bonwick 
1161b24ab676SJeff Bonwick 	/*
1162b24ab676SJeff Bonwick 	 * Attempt to assign tx to some transaction group.
1163b24ab676SJeff Bonwick 	 */
1164b24ab676SJeff Bonwick 	error = dmu_tx_assign(tx, txg_how);
1165b24ab676SJeff Bonwick 	if (error) {
1166b24ab676SJeff Bonwick 		if (error == ERESTART) {
1167b24ab676SJeff Bonwick 			ASSERT(txg_how == TXG_NOWAIT);
1168b24ab676SJeff Bonwick 			dmu_tx_wait(tx);
1169b24ab676SJeff Bonwick 		} else {
1170b24ab676SJeff Bonwick 			ASSERT3U(error, ==, ENOSPC);
1171b24ab676SJeff Bonwick 			ztest_record_enospc(tag);
1172b24ab676SJeff Bonwick 		}
1173b24ab676SJeff Bonwick 		dmu_tx_abort(tx);
1174b24ab676SJeff Bonwick 		return (0);
1175b24ab676SJeff Bonwick 	}
1176b24ab676SJeff Bonwick 	txg = dmu_tx_get_txg(tx);
1177b24ab676SJeff Bonwick 	ASSERT(txg != 0);
1178b24ab676SJeff Bonwick 	return (txg);
1179b24ab676SJeff Bonwick }
1180b24ab676SJeff Bonwick 
1181b24ab676SJeff Bonwick static void
1182b24ab676SJeff Bonwick ztest_pattern_set(void *buf, uint64_t size, uint64_t value)
1183b24ab676SJeff Bonwick {
1184b24ab676SJeff Bonwick 	uint64_t *ip = buf;
1185b24ab676SJeff Bonwick 	uint64_t *ip_end = (uint64_t *)((uintptr_t)buf + (uintptr_t)size);
1186b24ab676SJeff Bonwick 
1187b24ab676SJeff Bonwick 	while (ip < ip_end)
1188b24ab676SJeff Bonwick 		*ip++ = value;
1189b24ab676SJeff Bonwick }
1190b24ab676SJeff Bonwick 
1191b24ab676SJeff Bonwick static boolean_t
1192b24ab676SJeff Bonwick ztest_pattern_match(void *buf, uint64_t size, uint64_t value)
1193b24ab676SJeff Bonwick {
1194b24ab676SJeff Bonwick 	uint64_t *ip = buf;
1195b24ab676SJeff Bonwick 	uint64_t *ip_end = (uint64_t *)((uintptr_t)buf + (uintptr_t)size);
1196b24ab676SJeff Bonwick 	uint64_t diff = 0;
1197b24ab676SJeff Bonwick 
1198b24ab676SJeff Bonwick 	while (ip < ip_end)
1199b24ab676SJeff Bonwick 		diff |= (value - *ip++);
1200b24ab676SJeff Bonwick 
1201b24ab676SJeff Bonwick 	return (diff == 0);
1202b24ab676SJeff Bonwick }
1203b24ab676SJeff Bonwick 
1204b24ab676SJeff Bonwick static void
1205b24ab676SJeff Bonwick ztest_bt_generate(ztest_block_tag_t *bt, objset_t *os, uint64_t object,
1206b24ab676SJeff Bonwick     uint64_t offset, uint64_t gen, uint64_t txg, uint64_t crtxg)
1207b24ab676SJeff Bonwick {
1208b24ab676SJeff Bonwick 	bt->bt_magic = BT_MAGIC;
1209b24ab676SJeff Bonwick 	bt->bt_objset = dmu_objset_id(os);
1210b24ab676SJeff Bonwick 	bt->bt_object = object;
1211b24ab676SJeff Bonwick 	bt->bt_offset = offset;
1212b24ab676SJeff Bonwick 	bt->bt_gen = gen;
1213b24ab676SJeff Bonwick 	bt->bt_txg = txg;
1214b24ab676SJeff Bonwick 	bt->bt_crtxg = crtxg;
1215b24ab676SJeff Bonwick }
1216b24ab676SJeff Bonwick 
1217b24ab676SJeff Bonwick static void
1218b24ab676SJeff Bonwick ztest_bt_verify(ztest_block_tag_t *bt, objset_t *os, uint64_t object,
1219b24ab676SJeff Bonwick     uint64_t offset, uint64_t gen, uint64_t txg, uint64_t crtxg)
1220b24ab676SJeff Bonwick {
1221b24ab676SJeff Bonwick 	ASSERT(bt->bt_magic == BT_MAGIC);
1222b24ab676SJeff Bonwick 	ASSERT(bt->bt_objset == dmu_objset_id(os));
1223b24ab676SJeff Bonwick 	ASSERT(bt->bt_object == object);
1224b24ab676SJeff Bonwick 	ASSERT(bt->bt_offset == offset);
1225b24ab676SJeff Bonwick 	ASSERT(bt->bt_gen <= gen);
1226b24ab676SJeff Bonwick 	ASSERT(bt->bt_txg <= txg);
1227b24ab676SJeff Bonwick 	ASSERT(bt->bt_crtxg == crtxg);
1228b24ab676SJeff Bonwick }
1229b24ab676SJeff Bonwick 
1230b24ab676SJeff Bonwick static ztest_block_tag_t *
1231b24ab676SJeff Bonwick ztest_bt_bonus(dmu_buf_t *db)
1232b24ab676SJeff Bonwick {
1233b24ab676SJeff Bonwick 	dmu_object_info_t doi;
1234b24ab676SJeff Bonwick 	ztest_block_tag_t *bt;
1235b24ab676SJeff Bonwick 
1236b24ab676SJeff Bonwick 	dmu_object_info_from_db(db, &doi);
1237b24ab676SJeff Bonwick 	ASSERT3U(doi.doi_bonus_size, <=, db->db_size);
1238b24ab676SJeff Bonwick 	ASSERT3U(doi.doi_bonus_size, >=, sizeof (*bt));
1239b24ab676SJeff Bonwick 	bt = (void *)((char *)db->db_data + doi.doi_bonus_size - sizeof (*bt));
1240b24ab676SJeff Bonwick 
1241b24ab676SJeff Bonwick 	return (bt);
1242b24ab676SJeff Bonwick }
1243b24ab676SJeff Bonwick 
1244b24ab676SJeff Bonwick /*
1245b24ab676SJeff Bonwick  * ZIL logging ops
1246b24ab676SJeff Bonwick  */
1247b24ab676SJeff Bonwick 
1248b24ab676SJeff Bonwick #define	lrz_type	lr_mode
1249b24ab676SJeff Bonwick #define	lrz_blocksize	lr_uid
1250b24ab676SJeff Bonwick #define	lrz_ibshift	lr_gid
1251b24ab676SJeff Bonwick #define	lrz_bonustype	lr_rdev
1252b24ab676SJeff Bonwick #define	lrz_bonuslen	lr_crtime[1]
1253b24ab676SJeff Bonwick 
12545002558fSNeil Perrin static void
1255b24ab676SJeff Bonwick ztest_log_create(ztest_ds_t *zd, dmu_tx_t *tx, lr_create_t *lr)
1256b24ab676SJeff Bonwick {
1257b24ab676SJeff Bonwick 	char *name = (void *)(lr + 1);		/* name follows lr */
1258b24ab676SJeff Bonwick 	size_t namesize = strlen(name) + 1;
1259b24ab676SJeff Bonwick 	itx_t *itx;
1260b24ab676SJeff Bonwick 
1261b24ab676SJeff Bonwick 	if (zil_replaying(zd->zd_zilog, tx))
12625002558fSNeil Perrin 		return;
1263b24ab676SJeff Bonwick 
1264b24ab676SJeff Bonwick 	itx = zil_itx_create(TX_CREATE, sizeof (*lr) + namesize);
1265b24ab676SJeff Bonwick 	bcopy(&lr->lr_common + 1, &itx->itx_lr + 1,
1266b24ab676SJeff Bonwick 	    sizeof (*lr) + namesize - sizeof (lr_t));
1267b24ab676SJeff Bonwick 
12685002558fSNeil Perrin 	zil_itx_assign(zd->zd_zilog, itx, tx);
1269b24ab676SJeff Bonwick }
1270b24ab676SJeff Bonwick 
12715002558fSNeil Perrin static void
12725002558fSNeil Perrin ztest_log_remove(ztest_ds_t *zd, dmu_tx_t *tx, lr_remove_t *lr, uint64_t object)
1273b24ab676SJeff Bonwick {
1274b24ab676SJeff Bonwick 	char *name = (void *)(lr + 1);		/* name follows lr */
1275b24ab676SJeff Bonwick 	size_t namesize = strlen(name) + 1;
1276b24ab676SJeff Bonwick 	itx_t *itx;
1277b24ab676SJeff Bonwick 
1278b24ab676SJeff Bonwick 	if (zil_replaying(zd->zd_zilog, tx))
12795002558fSNeil Perrin 		return;
1280b24ab676SJeff Bonwick 
1281b24ab676SJeff Bonwick 	itx = zil_itx_create(TX_REMOVE, sizeof (*lr) + namesize);
1282b24ab676SJeff Bonwick 	bcopy(&lr->lr_common + 1, &itx->itx_lr + 1,
1283b24ab676SJeff Bonwick 	    sizeof (*lr) + namesize - sizeof (lr_t));
1284b24ab676SJeff Bonwick 
128551bd2f97SNeil Perrin 	itx->itx_oid = object;
12865002558fSNeil Perrin 	zil_itx_assign(zd->zd_zilog, itx, tx);
1287b24ab676SJeff Bonwick }
1288b24ab676SJeff Bonwick 
12895002558fSNeil Perrin static void
1290b24ab676SJeff Bonwick ztest_log_write(ztest_ds_t *zd, dmu_tx_t *tx, lr_write_t *lr)
1291b24ab676SJeff Bonwick {
1292b24ab676SJeff Bonwick 	itx_t *itx;
1293b24ab676SJeff Bonwick 	itx_wr_state_t write_state = ztest_random(WR_NUM_STATES);
1294b24ab676SJeff Bonwick 
1295b24ab676SJeff Bonwick 	if (zil_replaying(zd->zd_zilog, tx))
12965002558fSNeil Perrin 		return;
1297b24ab676SJeff Bonwick 
1298b24ab676SJeff Bonwick 	if (lr->lr_length > ZIL_MAX_LOG_DATA)
1299b24ab676SJeff Bonwick 		write_state = WR_INDIRECT;
1300b24ab676SJeff Bonwick 
1301b24ab676SJeff Bonwick 	itx = zil_itx_create(TX_WRITE,
1302b24ab676SJeff Bonwick 	    sizeof (*lr) + (write_state == WR_COPIED ? lr->lr_length : 0));
1303b24ab676SJeff Bonwick 
1304b24ab676SJeff Bonwick 	if (write_state == WR_COPIED &&
1305b24ab676SJeff Bonwick 	    dmu_read(zd->zd_os, lr->lr_foid, lr->lr_offset, lr->lr_length,
1306b24ab676SJeff Bonwick 	    ((lr_write_t *)&itx->itx_lr) + 1, DMU_READ_NO_PREFETCH) != 0) {
1307b24ab676SJeff Bonwick 		zil_itx_destroy(itx);
1308b24ab676SJeff Bonwick 		itx = zil_itx_create(TX_WRITE, sizeof (*lr));
1309b24ab676SJeff Bonwick 		write_state = WR_NEED_COPY;
1310b24ab676SJeff Bonwick 	}
1311b24ab676SJeff Bonwick 	itx->itx_private = zd;
1312b24ab676SJeff Bonwick 	itx->itx_wr_state = write_state;
1313b24ab676SJeff Bonwick 	itx->itx_sync = (ztest_random(8) == 0);
1314b24ab676SJeff Bonwick 	itx->itx_sod += (write_state == WR_NEED_COPY ? lr->lr_length : 0);
1315b24ab676SJeff Bonwick 
1316b24ab676SJeff Bonwick 	bcopy(&lr->lr_common + 1, &itx->itx_lr + 1,
1317b24ab676SJeff Bonwick 	    sizeof (*lr) - sizeof (lr_t));
1318b24ab676SJeff Bonwick 
13195002558fSNeil Perrin 	zil_itx_assign(zd->zd_zilog, itx, tx);
1320b24ab676SJeff Bonwick }
1321b24ab676SJeff Bonwick 
13225002558fSNeil Perrin static void
1323b24ab676SJeff Bonwick ztest_log_truncate(ztest_ds_t *zd, dmu_tx_t *tx, lr_truncate_t *lr)
1324b24ab676SJeff Bonwick {
1325b24ab676SJeff Bonwick 	itx_t *itx;
1326b24ab676SJeff Bonwick 
1327b24ab676SJeff Bonwick 	if (zil_replaying(zd->zd_zilog, tx))
13285002558fSNeil Perrin 		return;
1329b24ab676SJeff Bonwick 
1330b24ab676SJeff Bonwick 	itx = zil_itx_create(TX_TRUNCATE, sizeof (*lr));
1331b24ab676SJeff Bonwick 	bcopy(&lr->lr_common + 1, &itx->itx_lr + 1,
1332b24ab676SJeff Bonwick 	    sizeof (*lr) - sizeof (lr_t));
1333b24ab676SJeff Bonwick 
13345002558fSNeil Perrin 	itx->itx_sync = B_FALSE;
13355002558fSNeil Perrin 	zil_itx_assign(zd->zd_zilog, itx, tx);
1336b24ab676SJeff Bonwick }
1337b24ab676SJeff Bonwick 
13385002558fSNeil Perrin static void
1339b24ab676SJeff Bonwick ztest_log_setattr(ztest_ds_t *zd, dmu_tx_t *tx, lr_setattr_t *lr)
1340b24ab676SJeff Bonwick {
1341b24ab676SJeff Bonwick 	itx_t *itx;
1342b24ab676SJeff Bonwick 
1343b24ab676SJeff Bonwick 	if (zil_replaying(zd->zd_zilog, tx))
13445002558fSNeil Perrin 		return;
1345b24ab676SJeff Bonwick 
1346b24ab676SJeff Bonwick 	itx = zil_itx_create(TX_SETATTR, sizeof (*lr));
1347b24ab676SJeff Bonwick 	bcopy(&lr->lr_common + 1, &itx->itx_lr + 1,
1348b24ab676SJeff Bonwick 	    sizeof (*lr) - sizeof (lr_t));
1349b24ab676SJeff Bonwick 
13505002558fSNeil Perrin 	itx->itx_sync = B_FALSE;
13515002558fSNeil Perrin 	zil_itx_assign(zd->zd_zilog, itx, tx);
1352b24ab676SJeff Bonwick }
1353b24ab676SJeff Bonwick 
1354b24ab676SJeff Bonwick /*
1355b24ab676SJeff Bonwick  * ZIL replay ops
1356b24ab676SJeff Bonwick  */
1357b24ab676SJeff Bonwick static int
1358b24ab676SJeff Bonwick ztest_replay_create(ztest_ds_t *zd, lr_create_t *lr, boolean_t byteswap)
1359b24ab676SJeff Bonwick {
1360b24ab676SJeff Bonwick 	char *name = (void *)(lr + 1);		/* name follows lr */
1361b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
1362b24ab676SJeff Bonwick 	ztest_block_tag_t *bbt;
1363b24ab676SJeff Bonwick 	dmu_buf_t *db;
1364b24ab676SJeff Bonwick 	dmu_tx_t *tx;
1365b24ab676SJeff Bonwick 	uint64_t txg;
1366b24ab676SJeff Bonwick 	int error = 0;
1367b24ab676SJeff Bonwick 
1368b24ab676SJeff Bonwick 	if (byteswap)
1369b24ab676SJeff Bonwick 		byteswap_uint64_array(lr, sizeof (*lr));
1370b24ab676SJeff Bonwick 
1371b24ab676SJeff Bonwick 	ASSERT(lr->lr_doid == ZTEST_DIROBJ);
1372b24ab676SJeff Bonwick 	ASSERT(name[0] != '\0');
1373b24ab676SJeff Bonwick 
1374b24ab676SJeff Bonwick 	tx = dmu_tx_create(os);
1375b24ab676SJeff Bonwick 
1376b24ab676SJeff Bonwick 	dmu_tx_hold_zap(tx, lr->lr_doid, B_TRUE, name);
1377b24ab676SJeff Bonwick 
1378b24ab676SJeff Bonwick 	if (lr->lrz_type == DMU_OT_ZAP_OTHER) {
1379b24ab676SJeff Bonwick 		dmu_tx_hold_zap(tx, DMU_NEW_OBJECT, B_TRUE, NULL);
1380b24ab676SJeff Bonwick 	} else {
1381b24ab676SJeff Bonwick 		dmu_tx_hold_bonus(tx, DMU_NEW_OBJECT);
1382b24ab676SJeff Bonwick 	}
1383b24ab676SJeff Bonwick 
1384b24ab676SJeff Bonwick 	txg = ztest_tx_assign(tx, TXG_WAIT, FTAG);
1385b24ab676SJeff Bonwick 	if (txg == 0)
1386b24ab676SJeff Bonwick 		return (ENOSPC);
1387b24ab676SJeff Bonwick 
1388b24ab676SJeff Bonwick 	ASSERT(dmu_objset_zil(os)->zl_replay == !!lr->lr_foid);
1389b24ab676SJeff Bonwick 
1390b24ab676SJeff Bonwick 	if (lr->lrz_type == DMU_OT_ZAP_OTHER) {
1391b24ab676SJeff Bonwick 		if (lr->lr_foid == 0) {
1392b24ab676SJeff Bonwick 			lr->lr_foid = zap_create(os,
1393b24ab676SJeff Bonwick 			    lr->lrz_type, lr->lrz_bonustype,
1394b24ab676SJeff Bonwick 			    lr->lrz_bonuslen, tx);
1395b24ab676SJeff Bonwick 		} else {
1396b24ab676SJeff Bonwick 			error = zap_create_claim(os, lr->lr_foid,
1397b24ab676SJeff Bonwick 			    lr->lrz_type, lr->lrz_bonustype,
1398b24ab676SJeff Bonwick 			    lr->lrz_bonuslen, tx);
1399b24ab676SJeff Bonwick 		}
1400b24ab676SJeff Bonwick 	} else {
1401b24ab676SJeff Bonwick 		if (lr->lr_foid == 0) {
1402b24ab676SJeff Bonwick 			lr->lr_foid = dmu_object_alloc(os,
1403b24ab676SJeff Bonwick 			    lr->lrz_type, 0, lr->lrz_bonustype,
1404b24ab676SJeff Bonwick 			    lr->lrz_bonuslen, tx);
1405b24ab676SJeff Bonwick 		} else {
1406b24ab676SJeff Bonwick 			error = dmu_object_claim(os, lr->lr_foid,
1407b24ab676SJeff Bonwick 			    lr->lrz_type, 0, lr->lrz_bonustype,
1408b24ab676SJeff Bonwick 			    lr->lrz_bonuslen, tx);
1409b24ab676SJeff Bonwick 		}
1410b24ab676SJeff Bonwick 	}
1411b24ab676SJeff Bonwick 
1412b24ab676SJeff Bonwick 	if (error) {
1413b24ab676SJeff Bonwick 		ASSERT3U(error, ==, EEXIST);
1414b24ab676SJeff Bonwick 		ASSERT(zd->zd_zilog->zl_replay);
1415b24ab676SJeff Bonwick 		dmu_tx_commit(tx);
1416b24ab676SJeff Bonwick 		return (error);
1417b24ab676SJeff Bonwick 	}
1418b24ab676SJeff Bonwick 
1419b24ab676SJeff Bonwick 	ASSERT(lr->lr_foid != 0);
1420b24ab676SJeff Bonwick 
1421b24ab676SJeff Bonwick 	if (lr->lrz_type != DMU_OT_ZAP_OTHER)
1422b420f3adSRichard Lowe 		VERIFY3U(0, ==, dmu_object_set_blocksize(os, lr->lr_foid,
1423b24ab676SJeff Bonwick 		    lr->lrz_blocksize, lr->lrz_ibshift, tx));
1424b24ab676SJeff Bonwick 
1425b420f3adSRichard Lowe 	VERIFY3U(0, ==, dmu_bonus_hold(os, lr->lr_foid, FTAG, &db));
1426b24ab676SJeff Bonwick 	bbt = ztest_bt_bonus(db);
1427b24ab676SJeff Bonwick 	dmu_buf_will_dirty(db, tx);
1428b24ab676SJeff Bonwick 	ztest_bt_generate(bbt, os, lr->lr_foid, -1ULL, lr->lr_gen, txg, txg);
1429b24ab676SJeff Bonwick 	dmu_buf_rele(db, FTAG);
1430b24ab676SJeff Bonwick 
1431b420f3adSRichard Lowe 	VERIFY3U(0, ==, zap_add(os, lr->lr_doid, name, sizeof (uint64_t), 1,
1432b24ab676SJeff Bonwick 	    &lr->lr_foid, tx));
1433b24ab676SJeff Bonwick 
1434b24ab676SJeff Bonwick 	(void) ztest_log_create(zd, tx, lr);
1435b24ab676SJeff Bonwick 
1436b24ab676SJeff Bonwick 	dmu_tx_commit(tx);
1437b24ab676SJeff Bonwick 
1438b24ab676SJeff Bonwick 	return (0);
1439b24ab676SJeff Bonwick }
1440b24ab676SJeff Bonwick 
1441b24ab676SJeff Bonwick static int
1442b24ab676SJeff Bonwick ztest_replay_remove(ztest_ds_t *zd, lr_remove_t *lr, boolean_t byteswap)
1443b24ab676SJeff Bonwick {
1444b24ab676SJeff Bonwick 	char *name = (void *)(lr + 1);		/* name follows lr */
1445b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
1446b24ab676SJeff Bonwick 	dmu_object_info_t doi;
1447b24ab676SJeff Bonwick 	dmu_tx_t *tx;
1448b24ab676SJeff Bonwick 	uint64_t object, txg;
1449b24ab676SJeff Bonwick 
1450b24ab676SJeff Bonwick 	if (byteswap)
1451b24ab676SJeff Bonwick 		byteswap_uint64_array(lr, sizeof (*lr));
1452b24ab676SJeff Bonwick 
1453b24ab676SJeff Bonwick 	ASSERT(lr->lr_doid == ZTEST_DIROBJ);
1454b24ab676SJeff Bonwick 	ASSERT(name[0] != '\0');
1455b24ab676SJeff Bonwick 
1456b420f3adSRichard Lowe 	VERIFY3U(0, ==,
1457b24ab676SJeff Bonwick 	    zap_lookup(os, lr->lr_doid, name, sizeof (object), 1, &object));
1458b24ab676SJeff Bonwick 	ASSERT(object != 0);
1459b24ab676SJeff Bonwick 
1460b24ab676SJeff Bonwick 	ztest_object_lock(zd, object, RL_WRITER);
1461b24ab676SJeff Bonwick 
1462b420f3adSRichard Lowe 	VERIFY3U(0, ==, dmu_object_info(os, object, &doi));
1463b24ab676SJeff Bonwick 
1464b24ab676SJeff Bonwick 	tx = dmu_tx_create(os);
1465b24ab676SJeff Bonwick 
1466b24ab676SJeff Bonwick 	dmu_tx_hold_zap(tx, lr->lr_doid, B_FALSE, name);
1467b24ab676SJeff Bonwick 	dmu_tx_hold_free(tx, object, 0, DMU_OBJECT_END);
1468b24ab676SJeff Bonwick 
1469b24ab676SJeff Bonwick 	txg = ztest_tx_assign(tx, TXG_WAIT, FTAG);
1470b24ab676SJeff Bonwick 	if (txg == 0) {
1471b24ab676SJeff Bonwick 		ztest_object_unlock(zd, object);
1472b24ab676SJeff Bonwick 		return (ENOSPC);
1473b24ab676SJeff Bonwick 	}
1474b24ab676SJeff Bonwick 
1475b24ab676SJeff Bonwick 	if (doi.doi_type == DMU_OT_ZAP_OTHER) {
1476b420f3adSRichard Lowe 		VERIFY3U(0, ==, zap_destroy(os, object, tx));
1477b24ab676SJeff Bonwick 	} else {
1478b420f3adSRichard Lowe 		VERIFY3U(0, ==, dmu_object_free(os, object, tx));
1479b24ab676SJeff Bonwick 	}
1480b24ab676SJeff Bonwick 
1481b420f3adSRichard Lowe 	VERIFY3U(0, ==, zap_remove(os, lr->lr_doid, name, tx));
1482b24ab676SJeff Bonwick 
14835002558fSNeil Perrin 	(void) ztest_log_remove(zd, tx, lr, object);
1484b24ab676SJeff Bonwick 
1485b24ab676SJeff Bonwick 	dmu_tx_commit(tx);
1486b24ab676SJeff Bonwick 
1487b24ab676SJeff Bonwick 	ztest_object_unlock(zd, object);
1488b24ab676SJeff Bonwick 
1489b24ab676SJeff Bonwick 	return (0);
1490b24ab676SJeff Bonwick }
1491b24ab676SJeff Bonwick 
1492b24ab676SJeff Bonwick static int
1493b24ab676SJeff Bonwick ztest_replay_write(ztest_ds_t *zd, lr_write_t *lr, boolean_t byteswap)
1494b24ab676SJeff Bonwick {
1495b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
1496b24ab676SJeff Bonwick 	void *data = lr + 1;			/* data follows lr */
1497b24ab676SJeff Bonwick 	uint64_t offset, length;
1498b24ab676SJeff Bonwick 	ztest_block_tag_t *bt = data;
1499b24ab676SJeff Bonwick 	ztest_block_tag_t *bbt;
1500b24ab676SJeff Bonwick 	uint64_t gen, txg, lrtxg, crtxg;
1501b24ab676SJeff Bonwick 	dmu_object_info_t doi;
1502b24ab676SJeff Bonwick 	dmu_tx_t *tx;
1503b24ab676SJeff Bonwick 	dmu_buf_t *db;
1504b24ab676SJeff Bonwick 	arc_buf_t *abuf = NULL;
1505b24ab676SJeff Bonwick 	rl_t *rl;
1506b24ab676SJeff Bonwick 
1507b24ab676SJeff Bonwick 	if (byteswap)
1508b24ab676SJeff Bonwick 		byteswap_uint64_array(lr, sizeof (*lr));
1509b24ab676SJeff Bonwick 
1510b24ab676SJeff Bonwick 	offset = lr->lr_offset;
1511b24ab676SJeff Bonwick 	length = lr->lr_length;
1512b24ab676SJeff Bonwick 
1513b24ab676SJeff Bonwick 	/* If it's a dmu_sync() block, write the whole block */
1514b24ab676SJeff Bonwick 	if (lr->lr_common.lrc_reclen == sizeof (lr_write_t)) {
1515b24ab676SJeff Bonwick 		uint64_t blocksize = BP_GET_LSIZE(&lr->lr_blkptr);
1516b24ab676SJeff Bonwick 		if (length < blocksize) {
1517b24ab676SJeff Bonwick 			offset -= offset % blocksize;
1518b24ab676SJeff Bonwick 			length = blocksize;
1519b24ab676SJeff Bonwick 		}
1520b24ab676SJeff Bonwick 	}
1521b24ab676SJeff Bonwick 
1522b24ab676SJeff Bonwick 	if (bt->bt_magic == BSWAP_64(BT_MAGIC))
1523b24ab676SJeff Bonwick 		byteswap_uint64_array(bt, sizeof (*bt));
1524b24ab676SJeff Bonwick 
1525b24ab676SJeff Bonwick 	if (bt->bt_magic != BT_MAGIC)
1526b24ab676SJeff Bonwick 		bt = NULL;
1527b24ab676SJeff Bonwick 
1528b24ab676SJeff Bonwick 	ztest_object_lock(zd, lr->lr_foid, RL_READER);
1529b24ab676SJeff Bonwick 	rl = ztest_range_lock(zd, lr->lr_foid, offset, length, RL_WRITER);
1530b24ab676SJeff Bonwick 
1531b420f3adSRichard Lowe 	VERIFY3U(0, ==, dmu_bonus_hold(os, lr->lr_foid, FTAG, &db));
1532b24ab676SJeff Bonwick 
1533b24ab676SJeff Bonwick 	dmu_object_info_from_db(db, &doi);
1534b24ab676SJeff Bonwick 
1535b24ab676SJeff Bonwick 	bbt = ztest_bt_bonus(db);
1536b24ab676SJeff Bonwick 	ASSERT3U(bbt->bt_magic, ==, BT_MAGIC);
1537b24ab676SJeff Bonwick 	gen = bbt->bt_gen;
1538b24ab676SJeff Bonwick 	crtxg = bbt->bt_crtxg;
1539b24ab676SJeff Bonwick 	lrtxg = lr->lr_common.lrc_txg;
1540b24ab676SJeff Bonwick 
1541b24ab676SJeff Bonwick 	tx = dmu_tx_create(os);
1542b24ab676SJeff Bonwick 
1543b24ab676SJeff Bonwick 	dmu_tx_hold_write(tx, lr->lr_foid, offset, length);
1544b24ab676SJeff Bonwick 
1545b24ab676SJeff Bonwick 	if (ztest_random(8) == 0 && length == doi.doi_data_block_size &&
1546b24ab676SJeff Bonwick 	    P2PHASE(offset, length) == 0)
1547b24ab676SJeff Bonwick 		abuf = dmu_request_arcbuf(db, length);
1548b24ab676SJeff Bonwick 
1549b24ab676SJeff Bonwick 	txg = ztest_tx_assign(tx, TXG_WAIT, FTAG);
1550b24ab676SJeff Bonwick 	if (txg == 0) {
1551b24ab676SJeff Bonwick 		if (abuf != NULL)
1552b24ab676SJeff Bonwick 			dmu_return_arcbuf(abuf);
1553b24ab676SJeff Bonwick 		dmu_buf_rele(db, FTAG);
1554b24ab676SJeff Bonwick 		ztest_range_unlock(rl);
1555b24ab676SJeff Bonwick 		ztest_object_unlock(zd, lr->lr_foid);
1556b24ab676SJeff Bonwick 		return (ENOSPC);
1557b24ab676SJeff Bonwick 	}
1558b24ab676SJeff Bonwick 
1559b24ab676SJeff Bonwick 	if (bt != NULL) {
1560b24ab676SJeff Bonwick 		/*
1561b24ab676SJeff Bonwick 		 * Usually, verify the old data before writing new data --
1562b24ab676SJeff Bonwick 		 * but not always, because we also want to verify correct
1563b24ab676SJeff Bonwick 		 * behavior when the data was not recently read into cache.
1564b24ab676SJeff Bonwick 		 */
1565b24ab676SJeff Bonwick 		ASSERT(offset % doi.doi_data_block_size == 0);
1566b24ab676SJeff Bonwick 		if (ztest_random(4) != 0) {
1567b24ab676SJeff Bonwick 			int prefetch = ztest_random(2) ?
1568b24ab676SJeff Bonwick 			    DMU_READ_PREFETCH : DMU_READ_NO_PREFETCH;
1569b24ab676SJeff Bonwick 			ztest_block_tag_t rbt;
1570b24ab676SJeff Bonwick 
1571b24ab676SJeff Bonwick 			VERIFY(dmu_read(os, lr->lr_foid, offset,
1572b24ab676SJeff Bonwick 			    sizeof (rbt), &rbt, prefetch) == 0);
1573b24ab676SJeff Bonwick 			if (rbt.bt_magic == BT_MAGIC) {
1574b24ab676SJeff Bonwick 				ztest_bt_verify(&rbt, os, lr->lr_foid,
1575b24ab676SJeff Bonwick 				    offset, gen, txg, crtxg);
1576b24ab676SJeff Bonwick 			}
1577b24ab676SJeff Bonwick 		}
1578b24ab676SJeff Bonwick 
1579b24ab676SJeff Bonwick 		/*
1580b24ab676SJeff Bonwick 		 * Writes can appear to be newer than the bonus buffer because
1581b24ab676SJeff Bonwick 		 * the ztest_get_data() callback does a dmu_read() of the
1582b24ab676SJeff Bonwick 		 * open-context data, which may be different than the data
1583b24ab676SJeff Bonwick 		 * as it was when the write was generated.
1584b24ab676SJeff Bonwick 		 */
1585b24ab676SJeff Bonwick 		if (zd->zd_zilog->zl_replay) {
1586b24ab676SJeff Bonwick 			ztest_bt_verify(bt, os, lr->lr_foid, offset,
1587b24ab676SJeff Bonwick 			    MAX(gen, bt->bt_gen), MAX(txg, lrtxg),
1588b24ab676SJeff Bonwick 			    bt->bt_crtxg);
1589b24ab676SJeff Bonwick 		}
1590b24ab676SJeff Bonwick 
1591b24ab676SJeff Bonwick 		/*
1592b24ab676SJeff Bonwick 		 * Set the bt's gen/txg to the bonus buffer's gen/txg
1593b24ab676SJeff Bonwick 		 * so that all of the usual ASSERTs will work.
1594b24ab676SJeff Bonwick 		 */
1595b24ab676SJeff Bonwick 		ztest_bt_generate(bt, os, lr->lr_foid, offset, gen, txg, crtxg);
1596b24ab676SJeff Bonwick 	}
1597b24ab676SJeff Bonwick 
1598b24ab676SJeff Bonwick 	if (abuf == NULL) {
1599b24ab676SJeff Bonwick 		dmu_write(os, lr->lr_foid, offset, length, data, tx);
1600b24ab676SJeff Bonwick 	} else {
1601b24ab676SJeff Bonwick 		bcopy(data, abuf->b_data, length);
1602b24ab676SJeff Bonwick 		dmu_assign_arcbuf(db, offset, abuf, tx);
1603b24ab676SJeff Bonwick 	}
1604b24ab676SJeff Bonwick 
1605b24ab676SJeff Bonwick 	(void) ztest_log_write(zd, tx, lr);
1606b24ab676SJeff Bonwick 
1607b24ab676SJeff Bonwick 	dmu_buf_rele(db, FTAG);
1608b24ab676SJeff Bonwick 
1609b24ab676SJeff Bonwick 	dmu_tx_commit(tx);
1610b24ab676SJeff Bonwick 
1611b24ab676SJeff Bonwick 	ztest_range_unlock(rl);
1612b24ab676SJeff Bonwick 	ztest_object_unlock(zd, lr->lr_foid);
1613b24ab676SJeff Bonwick 
1614b24ab676SJeff Bonwick 	return (0);
1615b24ab676SJeff Bonwick }
1616b24ab676SJeff Bonwick 
1617b24ab676SJeff Bonwick static int
1618b24ab676SJeff Bonwick ztest_replay_truncate(ztest_ds_t *zd, lr_truncate_t *lr, boolean_t byteswap)
1619b24ab676SJeff Bonwick {
1620b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
1621b24ab676SJeff Bonwick 	dmu_tx_t *tx;
1622b24ab676SJeff Bonwick 	uint64_t txg;
1623b24ab676SJeff Bonwick 	rl_t *rl;
1624b24ab676SJeff Bonwick 
1625b24ab676SJeff Bonwick 	if (byteswap)
1626b24ab676SJeff Bonwick 		byteswap_uint64_array(lr, sizeof (*lr));
1627b24ab676SJeff Bonwick 
1628b24ab676SJeff Bonwick 	ztest_object_lock(zd, lr->lr_foid, RL_READER);
1629b24ab676SJeff Bonwick 	rl = ztest_range_lock(zd, lr->lr_foid, lr->lr_offset, lr->lr_length,
1630b24ab676SJeff Bonwick 	    RL_WRITER);
1631b24ab676SJeff Bonwick 
1632b24ab676SJeff Bonwick 	tx = dmu_tx_create(os);
1633b24ab676SJeff Bonwick 
1634b24ab676SJeff Bonwick 	dmu_tx_hold_free(tx, lr->lr_foid, lr->lr_offset, lr->lr_length);
1635b24ab676SJeff Bonwick 
1636b24ab676SJeff Bonwick 	txg = ztest_tx_assign(tx, TXG_WAIT, FTAG);
1637b24ab676SJeff Bonwick 	if (txg == 0) {
1638b24ab676SJeff Bonwick 		ztest_range_unlock(rl);
1639b24ab676SJeff Bonwick 		ztest_object_unlock(zd, lr->lr_foid);
1640b24ab676SJeff Bonwick 		return (ENOSPC);
1641b24ab676SJeff Bonwick 	}
1642b24ab676SJeff Bonwick 
1643b24ab676SJeff Bonwick 	VERIFY(dmu_free_range(os, lr->lr_foid, lr->lr_offset,
1644b24ab676SJeff Bonwick 	    lr->lr_length, tx) == 0);
1645b24ab676SJeff Bonwick 
1646b24ab676SJeff Bonwick 	(void) ztest_log_truncate(zd, tx, lr);
1647b24ab676SJeff Bonwick 
1648b24ab676SJeff Bonwick 	dmu_tx_commit(tx);
1649b24ab676SJeff Bonwick 
1650b24ab676SJeff Bonwick 	ztest_range_unlock(rl);
1651b24ab676SJeff Bonwick 	ztest_object_unlock(zd, lr->lr_foid);
1652b24ab676SJeff Bonwick 
1653b24ab676SJeff Bonwick 	return (0);
1654b24ab676SJeff Bonwick }
1655b24ab676SJeff Bonwick 
1656b24ab676SJeff Bonwick static int
1657b24ab676SJeff Bonwick ztest_replay_setattr(ztest_ds_t *zd, lr_setattr_t *lr, boolean_t byteswap)
1658b24ab676SJeff Bonwick {
1659b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
1660b24ab676SJeff Bonwick 	dmu_tx_t *tx;
1661b24ab676SJeff Bonwick 	dmu_buf_t *db;
1662b24ab676SJeff Bonwick 	ztest_block_tag_t *bbt;
1663b24ab676SJeff Bonwick 	uint64_t txg, lrtxg, crtxg;
1664b24ab676SJeff Bonwick 
1665b24ab676SJeff Bonwick 	if (byteswap)
1666b24ab676SJeff Bonwick 		byteswap_uint64_array(lr, sizeof (*lr));
1667b24ab676SJeff Bonwick 
1668b24ab676SJeff Bonwick 	ztest_object_lock(zd, lr->lr_foid, RL_WRITER);
1669b24ab676SJeff Bonwick 
1670b420f3adSRichard Lowe 	VERIFY3U(0, ==, dmu_bonus_hold(os, lr->lr_foid, FTAG, &db));
1671b24ab676SJeff Bonwick 
1672b24ab676SJeff Bonwick 	tx = dmu_tx_create(os);
1673b24ab676SJeff Bonwick 	dmu_tx_hold_bonus(tx, lr->lr_foid);
1674b24ab676SJeff Bonwick 
1675b24ab676SJeff Bonwick 	txg = ztest_tx_assign(tx, TXG_WAIT, FTAG);
1676b24ab676SJeff Bonwick 	if (txg == 0) {
1677b24ab676SJeff Bonwick 		dmu_buf_rele(db, FTAG);
1678b24ab676SJeff Bonwick 		ztest_object_unlock(zd, lr->lr_foid);
1679b24ab676SJeff Bonwick 		return (ENOSPC);
1680b24ab676SJeff Bonwick 	}
1681b24ab676SJeff Bonwick 
1682b24ab676SJeff Bonwick 	bbt = ztest_bt_bonus(db);
1683b24ab676SJeff Bonwick 	ASSERT3U(bbt->bt_magic, ==, BT_MAGIC);
1684b24ab676SJeff Bonwick 	crtxg = bbt->bt_crtxg;
1685b24ab676SJeff Bonwick 	lrtxg = lr->lr_common.lrc_txg;
1686b24ab676SJeff Bonwick 
1687b24ab676SJeff Bonwick 	if (zd->zd_zilog->zl_replay) {
1688b24ab676SJeff Bonwick 		ASSERT(lr->lr_size != 0);
1689b24ab676SJeff Bonwick 		ASSERT(lr->lr_mode != 0);
1690b24ab676SJeff Bonwick 		ASSERT(lrtxg != 0);
1691b24ab676SJeff Bonwick 	} else {
1692b24ab676SJeff Bonwick 		/*
1693b24ab676SJeff Bonwick 		 * Randomly change the size and increment the generation.
1694b24ab676SJeff Bonwick 		 */
1695b24ab676SJeff Bonwick 		lr->lr_size = (ztest_random(db->db_size / sizeof (*bbt)) + 1) *
1696b24ab676SJeff Bonwick 		    sizeof (*bbt);
1697b24ab676SJeff Bonwick 		lr->lr_mode = bbt->bt_gen + 1;
1698b24ab676SJeff Bonwick 		ASSERT(lrtxg == 0);
1699b24ab676SJeff Bonwick 	}
1700b24ab676SJeff Bonwick 
1701b24ab676SJeff Bonwick 	/*
1702b24ab676SJeff Bonwick 	 * Verify that the current bonus buffer is not newer than our txg.
1703b24ab676SJeff Bonwick 	 */
1704b24ab676SJeff Bonwick 	ztest_bt_verify(bbt, os, lr->lr_foid, -1ULL, lr->lr_mode,
1705b24ab676SJeff Bonwick 	    MAX(txg, lrtxg), crtxg);
1706b24ab676SJeff Bonwick 
1707b24ab676SJeff Bonwick 	dmu_buf_will_dirty(db, tx);
1708b24ab676SJeff Bonwick 
1709b24ab676SJeff Bonwick 	ASSERT3U(lr->lr_size, >=, sizeof (*bbt));
1710b24ab676SJeff Bonwick 	ASSERT3U(lr->lr_size, <=, db->db_size);
1711*fb09f5aaSMadhav Suresh 	VERIFY0(dmu_set_bonus(db, lr->lr_size, tx));
1712b24ab676SJeff Bonwick 	bbt = ztest_bt_bonus(db);
1713b24ab676SJeff Bonwick 
1714b24ab676SJeff Bonwick 	ztest_bt_generate(bbt, os, lr->lr_foid, -1ULL, lr->lr_mode, txg, crtxg);
1715b24ab676SJeff Bonwick 
1716b24ab676SJeff Bonwick 	dmu_buf_rele(db, FTAG);
1717b24ab676SJeff Bonwick 
1718b24ab676SJeff Bonwick 	(void) ztest_log_setattr(zd, tx, lr);
1719b24ab676SJeff Bonwick 
1720b24ab676SJeff Bonwick 	dmu_tx_commit(tx);
1721b24ab676SJeff Bonwick 
1722b24ab676SJeff Bonwick 	ztest_object_unlock(zd, lr->lr_foid);
1723b24ab676SJeff Bonwick 
1724b24ab676SJeff Bonwick 	return (0);
1725b24ab676SJeff Bonwick }
1726b24ab676SJeff Bonwick 
1727b24ab676SJeff Bonwick zil_replay_func_t *ztest_replay_vector[TX_MAX_TYPE] = {
1728b24ab676SJeff Bonwick 	NULL,			/* 0 no such transaction type */
1729b24ab676SJeff Bonwick 	ztest_replay_create,	/* TX_CREATE */
1730b24ab676SJeff Bonwick 	NULL,			/* TX_MKDIR */
1731b24ab676SJeff Bonwick 	NULL,			/* TX_MKXATTR */
1732b24ab676SJeff Bonwick 	NULL,			/* TX_SYMLINK */
1733b24ab676SJeff Bonwick 	ztest_replay_remove,	/* TX_REMOVE */
1734b24ab676SJeff Bonwick 	NULL,			/* TX_RMDIR */
1735b24ab676SJeff Bonwick 	NULL,			/* TX_LINK */
1736b24ab676SJeff Bonwick 	NULL,			/* TX_RENAME */
1737b24ab676SJeff Bonwick 	ztest_replay_write,	/* TX_WRITE */
1738b24ab676SJeff Bonwick 	ztest_replay_truncate,	/* TX_TRUNCATE */
1739b24ab676SJeff Bonwick 	ztest_replay_setattr,	/* TX_SETATTR */
1740b24ab676SJeff Bonwick 	NULL,			/* TX_ACL */
1741b24ab676SJeff Bonwick 	NULL,			/* TX_CREATE_ACL */
1742b24ab676SJeff Bonwick 	NULL,			/* TX_CREATE_ATTR */
1743b24ab676SJeff Bonwick 	NULL,			/* TX_CREATE_ACL_ATTR */
1744b24ab676SJeff Bonwick 	NULL,			/* TX_MKDIR_ACL */
1745b24ab676SJeff Bonwick 	NULL,			/* TX_MKDIR_ATTR */
1746b24ab676SJeff Bonwick 	NULL,			/* TX_MKDIR_ACL_ATTR */
1747b24ab676SJeff Bonwick 	NULL,			/* TX_WRITE2 */
1748b24ab676SJeff Bonwick };
1749b24ab676SJeff Bonwick 
1750b24ab676SJeff Bonwick /*
1751b24ab676SJeff Bonwick  * ZIL get_data callbacks
1752b24ab676SJeff Bonwick  */
1753b24ab676SJeff Bonwick 
1754b24ab676SJeff Bonwick static void
1755b24ab676SJeff Bonwick ztest_get_done(zgd_t *zgd, int error)
1756b24ab676SJeff Bonwick {
1757b24ab676SJeff Bonwick 	ztest_ds_t *zd = zgd->zgd_private;
1758b24ab676SJeff Bonwick 	uint64_t object = zgd->zgd_rl->rl_object;
1759b24ab676SJeff Bonwick 
1760b24ab676SJeff Bonwick 	if (zgd->zgd_db)
1761b24ab676SJeff Bonwick 		dmu_buf_rele(zgd->zgd_db, zgd);
1762b24ab676SJeff Bonwick 
1763b24ab676SJeff Bonwick 	ztest_range_unlock(zgd->zgd_rl);
1764b24ab676SJeff Bonwick 	ztest_object_unlock(zd, object);
1765b24ab676SJeff Bonwick 
1766b24ab676SJeff Bonwick 	if (error == 0 && zgd->zgd_bp)
1767b24ab676SJeff Bonwick 		zil_add_block(zgd->zgd_zilog, zgd->zgd_bp);
1768b24ab676SJeff Bonwick 
1769b24ab676SJeff Bonwick 	umem_free(zgd, sizeof (*zgd));
1770b24ab676SJeff Bonwick }
1771b24ab676SJeff Bonwick 
1772b24ab676SJeff Bonwick static int
1773b24ab676SJeff Bonwick ztest_get_data(void *arg, lr_write_t *lr, char *buf, zio_t *zio)
1774b24ab676SJeff Bonwick {
1775b24ab676SJeff Bonwick 	ztest_ds_t *zd = arg;
1776b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
1777b24ab676SJeff Bonwick 	uint64_t object = lr->lr_foid;
1778b24ab676SJeff Bonwick 	uint64_t offset = lr->lr_offset;
1779b24ab676SJeff Bonwick 	uint64_t size = lr->lr_length;
1780b24ab676SJeff Bonwick 	blkptr_t *bp = &lr->lr_blkptr;
1781b24ab676SJeff Bonwick 	uint64_t txg = lr->lr_common.lrc_txg;
1782b24ab676SJeff Bonwick 	uint64_t crtxg;
1783b24ab676SJeff Bonwick 	dmu_object_info_t doi;
1784b24ab676SJeff Bonwick 	dmu_buf_t *db;
1785b24ab676SJeff Bonwick 	zgd_t *zgd;
1786b24ab676SJeff Bonwick 	int error;
1787b24ab676SJeff Bonwick 
1788b24ab676SJeff Bonwick 	ztest_object_lock(zd, object, RL_READER);
1789b24ab676SJeff Bonwick 	error = dmu_bonus_hold(os, object, FTAG, &db);
1790b24ab676SJeff Bonwick 	if (error) {
1791b24ab676SJeff Bonwick 		ztest_object_unlock(zd, object);
1792b24ab676SJeff Bonwick 		return (error);
1793b24ab676SJeff Bonwick 	}
1794b24ab676SJeff Bonwick 
1795b24ab676SJeff Bonwick 	crtxg = ztest_bt_bonus(db)->bt_crtxg;
1796b24ab676SJeff Bonwick 
1797b24ab676SJeff Bonwick 	if (crtxg == 0 || crtxg > txg) {
1798b24ab676SJeff Bonwick 		dmu_buf_rele(db, FTAG);
1799b24ab676SJeff Bonwick 		ztest_object_unlock(zd, object);
1800b24ab676SJeff Bonwick 		return (ENOENT);
1801b24ab676SJeff Bonwick 	}
1802b24ab676SJeff Bonwick 
1803b24ab676SJeff Bonwick 	dmu_object_info_from_db(db, &doi);
1804b24ab676SJeff Bonwick 	dmu_buf_rele(db, FTAG);
1805b24ab676SJeff Bonwick 	db = NULL;
1806b24ab676SJeff Bonwick 
1807b24ab676SJeff Bonwick 	zgd = umem_zalloc(sizeof (*zgd), UMEM_NOFAIL);
1808b24ab676SJeff Bonwick 	zgd->zgd_zilog = zd->zd_zilog;
1809b24ab676SJeff Bonwick 	zgd->zgd_private = zd;
1810b24ab676SJeff Bonwick 
1811b24ab676SJeff Bonwick 	if (buf != NULL) {	/* immediate write */
1812b24ab676SJeff Bonwick 		zgd->zgd_rl = ztest_range_lock(zd, object, offset, size,
1813b24ab676SJeff Bonwick 		    RL_READER);
1814b24ab676SJeff Bonwick 
1815b24ab676SJeff Bonwick 		error = dmu_read(os, object, offset, size, buf,
1816b24ab676SJeff Bonwick 		    DMU_READ_NO_PREFETCH);
1817b24ab676SJeff Bonwick 		ASSERT(error == 0);
1818b24ab676SJeff Bonwick 	} else {
1819b24ab676SJeff Bonwick 		size = doi.doi_data_block_size;
1820dfe73b3dSJeff Bonwick 		if (ISP2(size)) {
1821b24ab676SJeff Bonwick 			offset = P2ALIGN(offset, size);
1822dfe73b3dSJeff Bonwick 		} else {
1823dfe73b3dSJeff Bonwick 			ASSERT(offset < size);
1824dfe73b3dSJeff Bonwick 			offset = 0;
1825dfe73b3dSJeff Bonwick 		}
1826b24ab676SJeff Bonwick 
1827b24ab676SJeff Bonwick 		zgd->zgd_rl = ztest_range_lock(zd, object, offset, size,
1828b24ab676SJeff Bonwick 		    RL_READER);
1829b24ab676SJeff Bonwick 
183047cb52daSJeff Bonwick 		error = dmu_buf_hold(os, object, offset, zgd, &db,
183147cb52daSJeff Bonwick 		    DMU_READ_NO_PREFETCH);
1832b24ab676SJeff Bonwick 
1833b24ab676SJeff Bonwick 		if (error == 0) {
1834b24ab676SJeff Bonwick 			zgd->zgd_db = db;
1835b24ab676SJeff Bonwick 			zgd->zgd_bp = bp;
1836b24ab676SJeff Bonwick 
1837b24ab676SJeff Bonwick 			ASSERT(db->db_offset == offset);
1838b24ab676SJeff Bonwick 			ASSERT(db->db_size == size);
1839b24ab676SJeff Bonwick 
1840b24ab676SJeff Bonwick 			error = dmu_sync(zio, lr->lr_common.lrc_txg,
1841b24ab676SJeff Bonwick 			    ztest_get_done, zgd);
1842b24ab676SJeff Bonwick 
1843b24ab676SJeff Bonwick 			if (error == 0)
1844b24ab676SJeff Bonwick 				return (0);
1845b24ab676SJeff Bonwick 		}
1846b24ab676SJeff Bonwick 	}
1847b24ab676SJeff Bonwick 
1848b24ab676SJeff Bonwick 	ztest_get_done(zgd, error);
1849b24ab676SJeff Bonwick 
1850b24ab676SJeff Bonwick 	return (error);
1851b24ab676SJeff Bonwick }
1852b24ab676SJeff Bonwick 
1853b24ab676SJeff Bonwick static void *
1854b24ab676SJeff Bonwick ztest_lr_alloc(size_t lrsize, char *name)
1855b24ab676SJeff Bonwick {
1856b24ab676SJeff Bonwick 	char *lr;
1857b24ab676SJeff Bonwick 	size_t namesize = name ? strlen(name) + 1 : 0;
1858b24ab676SJeff Bonwick 
1859b24ab676SJeff Bonwick 	lr = umem_zalloc(lrsize + namesize, UMEM_NOFAIL);
1860b24ab676SJeff Bonwick 
1861b24ab676SJeff Bonwick 	if (name)
1862b24ab676SJeff Bonwick 		bcopy(name, lr + lrsize, namesize);
1863b24ab676SJeff Bonwick 
1864b24ab676SJeff Bonwick 	return (lr);
1865b24ab676SJeff Bonwick }
1866b24ab676SJeff Bonwick 
1867b24ab676SJeff Bonwick void
1868b24ab676SJeff Bonwick ztest_lr_free(void *lr, size_t lrsize, char *name)
1869b24ab676SJeff Bonwick {
1870b24ab676SJeff Bonwick 	size_t namesize = name ? strlen(name) + 1 : 0;
1871b24ab676SJeff Bonwick 
1872b24ab676SJeff Bonwick 	umem_free(lr, lrsize + namesize);
1873b24ab676SJeff Bonwick }
1874b24ab676SJeff Bonwick 
1875b24ab676SJeff Bonwick /*
1876b24ab676SJeff Bonwick  * Lookup a bunch of objects.  Returns the number of objects not found.
1877b24ab676SJeff Bonwick  */
1878b24ab676SJeff Bonwick static int
1879b24ab676SJeff Bonwick ztest_lookup(ztest_ds_t *zd, ztest_od_t *od, int count)
1880b24ab676SJeff Bonwick {
1881b24ab676SJeff Bonwick 	int missing = 0;
1882b24ab676SJeff Bonwick 	int error;
1883b24ab676SJeff Bonwick 
1884b24ab676SJeff Bonwick 	ASSERT(_mutex_held(&zd->zd_dirobj_lock));
1885b24ab676SJeff Bonwick 
1886b24ab676SJeff Bonwick 	for (int i = 0; i < count; i++, od++) {
1887b24ab676SJeff Bonwick 		od->od_object = 0;
1888b24ab676SJeff Bonwick 		error = zap_lookup(zd->zd_os, od->od_dir, od->od_name,
1889b24ab676SJeff Bonwick 		    sizeof (uint64_t), 1, &od->od_object);
1890b24ab676SJeff Bonwick 		if (error) {
1891b24ab676SJeff Bonwick 			ASSERT(error == ENOENT);
1892b24ab676SJeff Bonwick 			ASSERT(od->od_object == 0);
1893b24ab676SJeff Bonwick 			missing++;
1894b24ab676SJeff Bonwick 		} else {
1895b24ab676SJeff Bonwick 			dmu_buf_t *db;
1896b24ab676SJeff Bonwick 			ztest_block_tag_t *bbt;
1897b24ab676SJeff Bonwick 			dmu_object_info_t doi;
1898b24ab676SJeff Bonwick 
1899b24ab676SJeff Bonwick 			ASSERT(od->od_object != 0);
1900b24ab676SJeff Bonwick 			ASSERT(missing == 0);	/* there should be no gaps */
1901b24ab676SJeff Bonwick 
1902b24ab676SJeff Bonwick 			ztest_object_lock(zd, od->od_object, RL_READER);
1903b420f3adSRichard Lowe 			VERIFY3U(0, ==, dmu_bonus_hold(zd->zd_os,
1904b24ab676SJeff Bonwick 			    od->od_object, FTAG, &db));
1905b24ab676SJeff Bonwick 			dmu_object_info_from_db(db, &doi);
1906b24ab676SJeff Bonwick 			bbt = ztest_bt_bonus(db);
1907b24ab676SJeff Bonwick 			ASSERT3U(bbt->bt_magic, ==, BT_MAGIC);
1908b24ab676SJeff Bonwick 			od->od_type = doi.doi_type;
1909b24ab676SJeff Bonwick 			od->od_blocksize = doi.doi_data_block_size;
1910b24ab676SJeff Bonwick 			od->od_gen = bbt->bt_gen;
1911b24ab676SJeff Bonwick 			dmu_buf_rele(db, FTAG);
1912b24ab676SJeff Bonwick 			ztest_object_unlock(zd, od->od_object);
1913b24ab676SJeff Bonwick 		}
1914b24ab676SJeff Bonwick 	}
1915b24ab676SJeff Bonwick 
1916b24ab676SJeff Bonwick 	return (missing);
1917b24ab676SJeff Bonwick }
1918b24ab676SJeff Bonwick 
1919b24ab676SJeff Bonwick static int
1920b24ab676SJeff Bonwick ztest_create(ztest_ds_t *zd, ztest_od_t *od, int count)
1921b24ab676SJeff Bonwick {
1922b24ab676SJeff Bonwick 	int missing = 0;
1923b24ab676SJeff Bonwick 
1924b24ab676SJeff Bonwick 	ASSERT(_mutex_held(&zd->zd_dirobj_lock));
1925b24ab676SJeff Bonwick 
1926b24ab676SJeff Bonwick 	for (int i = 0; i < count; i++, od++) {
1927b24ab676SJeff Bonwick 		if (missing) {
1928b24ab676SJeff Bonwick 			od->od_object = 0;
1929b24ab676SJeff Bonwick 			missing++;
1930b24ab676SJeff Bonwick 			continue;
1931b24ab676SJeff Bonwick 		}
1932b24ab676SJeff Bonwick 
1933b24ab676SJeff Bonwick 		lr_create_t *lr = ztest_lr_alloc(sizeof (*lr), od->od_name);
1934b24ab676SJeff Bonwick 
1935b24ab676SJeff Bonwick 		lr->lr_doid = od->od_dir;
1936b24ab676SJeff Bonwick 		lr->lr_foid = 0;	/* 0 to allocate, > 0 to claim */
1937b24ab676SJeff Bonwick 		lr->lrz_type = od->od_crtype;
1938b24ab676SJeff Bonwick 		lr->lrz_blocksize = od->od_crblocksize;
1939b24ab676SJeff Bonwick 		lr->lrz_ibshift = ztest_random_ibshift();
1940b24ab676SJeff Bonwick 		lr->lrz_bonustype = DMU_OT_UINT64_OTHER;
1941b24ab676SJeff Bonwick 		lr->lrz_bonuslen = dmu_bonus_max();
1942b24ab676SJeff Bonwick 		lr->lr_gen = od->od_crgen;
1943b24ab676SJeff Bonwick 		lr->lr_crtime[0] = time(NULL);
1944b24ab676SJeff Bonwick 
1945b24ab676SJeff Bonwick 		if (ztest_replay_create(zd, lr, B_FALSE) != 0) {
1946b24ab676SJeff Bonwick 			ASSERT(missing == 0);
1947b24ab676SJeff Bonwick 			od->od_object = 0;
1948b24ab676SJeff Bonwick 			missing++;
1949b24ab676SJeff Bonwick 		} else {
1950b24ab676SJeff Bonwick 			od->od_object = lr->lr_foid;
1951b24ab676SJeff Bonwick 			od->od_type = od->od_crtype;
1952b24ab676SJeff Bonwick 			od->od_blocksize = od->od_crblocksize;
1953b24ab676SJeff Bonwick 			od->od_gen = od->od_crgen;
1954b24ab676SJeff Bonwick 			ASSERT(od->od_object != 0);
1955b24ab676SJeff Bonwick 		}
1956b24ab676SJeff Bonwick 
1957b24ab676SJeff Bonwick 		ztest_lr_free(lr, sizeof (*lr), od->od_name);
1958b24ab676SJeff Bonwick 	}
1959b24ab676SJeff Bonwick 
1960b24ab676SJeff Bonwick 	return (missing);
1961b24ab676SJeff Bonwick }
1962b24ab676SJeff Bonwick 
1963b24ab676SJeff Bonwick static int
1964b24ab676SJeff Bonwick ztest_remove(ztest_ds_t *zd, ztest_od_t *od, int count)
1965b24ab676SJeff Bonwick {
1966b24ab676SJeff Bonwick 	int missing = 0;
1967b24ab676SJeff Bonwick 	int error;
1968b24ab676SJeff Bonwick 
1969b24ab676SJeff Bonwick 	ASSERT(_mutex_held(&zd->zd_dirobj_lock));
1970b24ab676SJeff Bonwick 
1971b24ab676SJeff Bonwick 	od += count - 1;
1972b24ab676SJeff Bonwick 
1973b24ab676SJeff Bonwick 	for (int i = count - 1; i >= 0; i--, od--) {
1974b24ab676SJeff Bonwick 		if (missing) {
1975b24ab676SJeff Bonwick 			missing++;
1976b24ab676SJeff Bonwick 			continue;
1977b24ab676SJeff Bonwick 		}
1978b24ab676SJeff Bonwick 
1979b24ab676SJeff Bonwick 		if (od->od_object == 0)
1980b24ab676SJeff Bonwick 			continue;
1981b24ab676SJeff Bonwick 
1982b24ab676SJeff Bonwick 		lr_remove_t *lr = ztest_lr_alloc(sizeof (*lr), od->od_name);
1983b24ab676SJeff Bonwick 
1984b24ab676SJeff Bonwick 		lr->lr_doid = od->od_dir;
1985b24ab676SJeff Bonwick 
1986b24ab676SJeff Bonwick 		if ((error = ztest_replay_remove(zd, lr, B_FALSE)) != 0) {
1987b24ab676SJeff Bonwick 			ASSERT3U(error, ==, ENOSPC);
1988b24ab676SJeff Bonwick 			missing++;
1989b24ab676SJeff Bonwick 		} else {
1990b24ab676SJeff Bonwick 			od->od_object = 0;
1991b24ab676SJeff Bonwick 		}
1992b24ab676SJeff Bonwick 		ztest_lr_free(lr, sizeof (*lr), od->od_name);
1993b24ab676SJeff Bonwick 	}
1994b24ab676SJeff Bonwick 
1995b24ab676SJeff Bonwick 	return (missing);
1996b24ab676SJeff Bonwick }
1997b24ab676SJeff Bonwick 
1998b24ab676SJeff Bonwick static int
1999b24ab676SJeff Bonwick ztest_write(ztest_ds_t *zd, uint64_t object, uint64_t offset, uint64_t size,
2000b24ab676SJeff Bonwick     void *data)
2001b24ab676SJeff Bonwick {
2002b24ab676SJeff Bonwick 	lr_write_t *lr;
2003b24ab676SJeff Bonwick 	int error;
2004b24ab676SJeff Bonwick 
2005b24ab676SJeff Bonwick 	lr = ztest_lr_alloc(sizeof (*lr) + size, NULL);
2006b24ab676SJeff Bonwick 
2007b24ab676SJeff Bonwick 	lr->lr_foid = object;
2008b24ab676SJeff Bonwick 	lr->lr_offset = offset;
2009b24ab676SJeff Bonwick 	lr->lr_length = size;
2010b24ab676SJeff Bonwick 	lr->lr_blkoff = 0;
2011b24ab676SJeff Bonwick 	BP_ZERO(&lr->lr_blkptr);
2012b24ab676SJeff Bonwick 
2013b24ab676SJeff Bonwick 	bcopy(data, lr + 1, size);
2014b24ab676SJeff Bonwick 
2015b24ab676SJeff Bonwick 	error = ztest_replay_write(zd, lr, B_FALSE);
2016b24ab676SJeff Bonwick 
2017b24ab676SJeff Bonwick 	ztest_lr_free(lr, sizeof (*lr) + size, NULL);
2018b24ab676SJeff Bonwick 
2019b24ab676SJeff Bonwick 	return (error);
2020b24ab676SJeff Bonwick }
2021b24ab676SJeff Bonwick 
2022b24ab676SJeff Bonwick static int
2023b24ab676SJeff Bonwick ztest_truncate(ztest_ds_t *zd, uint64_t object, uint64_t offset, uint64_t size)
2024b24ab676SJeff Bonwick {
2025b24ab676SJeff Bonwick 	lr_truncate_t *lr;
2026b24ab676SJeff Bonwick 	int error;
2027b24ab676SJeff Bonwick 
2028b24ab676SJeff Bonwick 	lr = ztest_lr_alloc(sizeof (*lr), NULL);
2029b24ab676SJeff Bonwick 
2030b24ab676SJeff Bonwick 	lr->lr_foid = object;
2031b24ab676SJeff Bonwick 	lr->lr_offset = offset;
2032b24ab676SJeff Bonwick 	lr->lr_length = size;
2033b24ab676SJeff Bonwick 
2034b24ab676SJeff Bonwick 	error = ztest_replay_truncate(zd, lr, B_FALSE);
2035b24ab676SJeff Bonwick 
2036b24ab676SJeff Bonwick 	ztest_lr_free(lr, sizeof (*lr), NULL);
2037b24ab676SJeff Bonwick 
2038b24ab676SJeff Bonwick 	return (error);
2039b24ab676SJeff Bonwick }
2040b24ab676SJeff Bonwick 
2041b24ab676SJeff Bonwick static int
2042b24ab676SJeff Bonwick ztest_setattr(ztest_ds_t *zd, uint64_t object)
2043b24ab676SJeff Bonwick {
2044b24ab676SJeff Bonwick 	lr_setattr_t *lr;
2045b24ab676SJeff Bonwick 	int error;
2046b24ab676SJeff Bonwick 
2047b24ab676SJeff Bonwick 	lr = ztest_lr_alloc(sizeof (*lr), NULL);
2048b24ab676SJeff Bonwick 
2049b24ab676SJeff Bonwick 	lr->lr_foid = object;
2050b24ab676SJeff Bonwick 	lr->lr_size = 0;
2051b24ab676SJeff Bonwick 	lr->lr_mode = 0;
2052b24ab676SJeff Bonwick 
2053b24ab676SJeff Bonwick 	error = ztest_replay_setattr(zd, lr, B_FALSE);
2054b24ab676SJeff Bonwick 
2055b24ab676SJeff Bonwick 	ztest_lr_free(lr, sizeof (*lr), NULL);
2056b24ab676SJeff Bonwick 
2057b24ab676SJeff Bonwick 	return (error);
2058b24ab676SJeff Bonwick }
2059b24ab676SJeff Bonwick 
2060b24ab676SJeff Bonwick static void
2061b24ab676SJeff Bonwick ztest_prealloc(ztest_ds_t *zd, uint64_t object, uint64_t offset, uint64_t size)
2062b24ab676SJeff Bonwick {
2063b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
2064b24ab676SJeff Bonwick 	dmu_tx_t *tx;
2065b24ab676SJeff Bonwick 	uint64_t txg;
2066b24ab676SJeff Bonwick 	rl_t *rl;
2067b24ab676SJeff Bonwick 
2068b24ab676SJeff Bonwick 	txg_wait_synced(dmu_objset_pool(os), 0);
2069b24ab676SJeff Bonwick 
2070b24ab676SJeff Bonwick 	ztest_object_lock(zd, object, RL_READER);
2071b24ab676SJeff Bonwick 	rl = ztest_range_lock(zd, object, offset, size, RL_WRITER);
2072b24ab676SJeff Bonwick 
2073b24ab676SJeff Bonwick 	tx = dmu_tx_create(os);
2074b24ab676SJeff Bonwick 
2075b24ab676SJeff Bonwick 	dmu_tx_hold_write(tx, object, offset, size);
2076b24ab676SJeff Bonwick 
2077b24ab676SJeff Bonwick 	txg = ztest_tx_assign(tx, TXG_WAIT, FTAG);
2078b24ab676SJeff Bonwick 
2079b24ab676SJeff Bonwick 	if (txg != 0) {
2080b24ab676SJeff Bonwick 		dmu_prealloc(os, object, offset, size, tx);
2081b24ab676SJeff Bonwick 		dmu_tx_commit(tx);
2082b24ab676SJeff Bonwick 		txg_wait_synced(dmu_objset_pool(os), txg);
2083b24ab676SJeff Bonwick 	} else {
2084b24ab676SJeff Bonwick 		(void) dmu_free_long_range(os, object, offset, size);
2085b24ab676SJeff Bonwick 	}
2086b24ab676SJeff Bonwick 
2087b24ab676SJeff Bonwick 	ztest_range_unlock(rl);
2088b24ab676SJeff Bonwick 	ztest_object_unlock(zd, object);
2089b24ab676SJeff Bonwick }
2090b24ab676SJeff Bonwick 
2091b24ab676SJeff Bonwick static void
2092b24ab676SJeff Bonwick ztest_io(ztest_ds_t *zd, uint64_t object, uint64_t offset)
2093b24ab676SJeff Bonwick {
2094b24ab676SJeff Bonwick 	ztest_block_tag_t wbt;
2095b24ab676SJeff Bonwick 	dmu_object_info_t doi;
2096b24ab676SJeff Bonwick 	enum ztest_io_type io_type;
2097b24ab676SJeff Bonwick 	uint64_t blocksize;
2098b24ab676SJeff Bonwick 	void *data;
2099b24ab676SJeff Bonwick 
2100b24ab676SJeff Bonwick 	VERIFY(dmu_object_info(zd->zd_os, object, &doi) == 0);
2101b24ab676SJeff Bonwick 	blocksize = doi.doi_data_block_size;
2102b24ab676SJeff Bonwick 	data = umem_alloc(blocksize, UMEM_NOFAIL);
2103b24ab676SJeff Bonwick 
2104b24ab676SJeff Bonwick 	/*
2105b24ab676SJeff Bonwick 	 * Pick an i/o type at random, biased toward writing block tags.
2106b24ab676SJeff Bonwick 	 */
2107b24ab676SJeff Bonwick 	io_type = ztest_random(ZTEST_IO_TYPES);
2108b24ab676SJeff Bonwick 	if (ztest_random(2) == 0)
2109b24ab676SJeff Bonwick 		io_type = ZTEST_IO_WRITE_TAG;
2110b24ab676SJeff Bonwick 
2111c9ba2a43SEric Schrock 	(void) rw_rdlock(&zd->zd_zilog_lock);
2112c9ba2a43SEric Schrock 
2113b24ab676SJeff Bonwick 	switch (io_type) {
2114b24ab676SJeff Bonwick 
2115b24ab676SJeff Bonwick 	case ZTEST_IO_WRITE_TAG:
2116b24ab676SJeff Bonwick 		ztest_bt_generate(&wbt, zd->zd_os, object, offset, 0, 0, 0);
2117b24ab676SJeff Bonwick 		(void) ztest_write(zd, object, offset, sizeof (wbt), &wbt);
2118b24ab676SJeff Bonwick 		break;
2119b24ab676SJeff Bonwick 
2120b24ab676SJeff Bonwick 	case ZTEST_IO_WRITE_PATTERN:
2121b24ab676SJeff Bonwick 		(void) memset(data, 'a' + (object + offset) % 5, blocksize);
2122b24ab676SJeff Bonwick 		if (ztest_random(2) == 0) {
2123b24ab676SJeff Bonwick 			/*
2124b24ab676SJeff Bonwick 			 * Induce fletcher2 collisions to ensure that
2125b24ab676SJeff Bonwick 			 * zio_ddt_collision() detects and resolves them
2126b24ab676SJeff Bonwick 			 * when using fletcher2-verify for deduplication.
2127b24ab676SJeff Bonwick 			 */
2128b24ab676SJeff Bonwick 			((uint64_t *)data)[0] ^= 1ULL << 63;
2129b24ab676SJeff Bonwick 			((uint64_t *)data)[4] ^= 1ULL << 63;
2130b24ab676SJeff Bonwick 		}
2131b24ab676SJeff Bonwick 		(void) ztest_write(zd, object, offset, blocksize, data);
2132b24ab676SJeff Bonwick 		break;
2133b24ab676SJeff Bonwick 
2134b24ab676SJeff Bonwick 	case ZTEST_IO_WRITE_ZEROES:
2135b24ab676SJeff Bonwick 		bzero(data, blocksize);
2136b24ab676SJeff Bonwick 		(void) ztest_write(zd, object, offset, blocksize, data);
2137b24ab676SJeff Bonwick 		break;
2138b24ab676SJeff Bonwick 
2139b24ab676SJeff Bonwick 	case ZTEST_IO_TRUNCATE:
2140b24ab676SJeff Bonwick 		(void) ztest_truncate(zd, object, offset, blocksize);
2141b24ab676SJeff Bonwick 		break;
2142b24ab676SJeff Bonwick 
2143b24ab676SJeff Bonwick 	case ZTEST_IO_SETATTR:
2144b24ab676SJeff Bonwick 		(void) ztest_setattr(zd, object);
2145b24ab676SJeff Bonwick 		break;
2146b24ab676SJeff Bonwick 	}
2147b24ab676SJeff Bonwick 
2148c9ba2a43SEric Schrock 	(void) rw_unlock(&zd->zd_zilog_lock);
2149c9ba2a43SEric Schrock 
2150b24ab676SJeff Bonwick 	umem_free(data, blocksize);
2151b24ab676SJeff Bonwick }
2152b24ab676SJeff Bonwick 
2153b24ab676SJeff Bonwick /*
2154b24ab676SJeff Bonwick  * Initialize an object description template.
2155b24ab676SJeff Bonwick  */
2156b24ab676SJeff Bonwick static void
2157b24ab676SJeff Bonwick ztest_od_init(ztest_od_t *od, uint64_t id, char *tag, uint64_t index,
2158b24ab676SJeff Bonwick     dmu_object_type_t type, uint64_t blocksize, uint64_t gen)
2159b24ab676SJeff Bonwick {
2160b24ab676SJeff Bonwick 	od->od_dir = ZTEST_DIROBJ;
2161b24ab676SJeff Bonwick 	od->od_object = 0;
2162b24ab676SJeff Bonwick 
2163b24ab676SJeff Bonwick 	od->od_crtype = type;
2164b24ab676SJeff Bonwick 	od->od_crblocksize = blocksize ? blocksize : ztest_random_blocksize();
2165b24ab676SJeff Bonwick 	od->od_crgen = gen;
2166b24ab676SJeff Bonwick 
2167b24ab676SJeff Bonwick 	od->od_type = DMU_OT_NONE;
2168b24ab676SJeff Bonwick 	od->od_blocksize = 0;
2169b24ab676SJeff Bonwick 	od->od_gen = 0;
2170b24ab676SJeff Bonwick 
2171b24ab676SJeff Bonwick 	(void) snprintf(od->od_name, sizeof (od->od_name), "%s(%lld)[%llu]",
2172b24ab676SJeff Bonwick 	    tag, (int64_t)id, index);
2173b24ab676SJeff Bonwick }
2174b24ab676SJeff Bonwick 
2175b24ab676SJeff Bonwick /*
2176b24ab676SJeff Bonwick  * Lookup or create the objects for a test using the od template.
2177b24ab676SJeff Bonwick  * If the objects do not all exist, or if 'remove' is specified,
2178b24ab676SJeff Bonwick  * remove any existing objects and create new ones.  Otherwise,
2179b24ab676SJeff Bonwick  * use the existing objects.
2180b24ab676SJeff Bonwick  */
2181b24ab676SJeff Bonwick static int
2182b24ab676SJeff Bonwick ztest_object_init(ztest_ds_t *zd, ztest_od_t *od, size_t size, boolean_t remove)
2183b24ab676SJeff Bonwick {
2184b24ab676SJeff Bonwick 	int count = size / sizeof (*od);
2185b24ab676SJeff Bonwick 	int rv = 0;
2186b24ab676SJeff Bonwick 
2187b24ab676SJeff Bonwick 	VERIFY(mutex_lock(&zd->zd_dirobj_lock) == 0);
2188b24ab676SJeff Bonwick 	if ((ztest_lookup(zd, od, count) != 0 || remove) &&
2189b24ab676SJeff Bonwick 	    (ztest_remove(zd, od, count) != 0 ||
2190b24ab676SJeff Bonwick 	    ztest_create(zd, od, count) != 0))
2191b24ab676SJeff Bonwick 		rv = -1;
2192b24ab676SJeff Bonwick 	zd->zd_od = od;
2193b24ab676SJeff Bonwick 	VERIFY(mutex_unlock(&zd->zd_dirobj_lock) == 0);
2194b24ab676SJeff Bonwick 
2195b24ab676SJeff Bonwick 	return (rv);
2196b24ab676SJeff Bonwick }
2197b24ab676SJeff Bonwick 
2198b24ab676SJeff Bonwick /* ARGSUSED */
2199b24ab676SJeff Bonwick void
2200b24ab676SJeff Bonwick ztest_zil_commit(ztest_ds_t *zd, uint64_t id)
2201b24ab676SJeff Bonwick {
2202b24ab676SJeff Bonwick 	zilog_t *zilog = zd->zd_zilog;
2203b24ab676SJeff Bonwick 
2204c9ba2a43SEric Schrock 	(void) rw_rdlock(&zd->zd_zilog_lock);
2205c9ba2a43SEric Schrock 
22065002558fSNeil Perrin 	zil_commit(zilog, ztest_random(ZTEST_OBJECTS));
2207b24ab676SJeff Bonwick 
2208b24ab676SJeff Bonwick 	/*
2209b24ab676SJeff Bonwick 	 * Remember the committed values in zd, which is in parent/child
2210b24ab676SJeff Bonwick 	 * shared memory.  If we die, the next iteration of ztest_run()
2211b24ab676SJeff Bonwick 	 * will verify that the log really does contain this record.
2212b24ab676SJeff Bonwick 	 */
2213b24ab676SJeff Bonwick 	mutex_enter(&zilog->zl_lock);
2214420dfc95SChris Siden 	ASSERT(zd->zd_shared != NULL);
2215420dfc95SChris Siden 	ASSERT3U(zd->zd_shared->zd_seq, <=, zilog->zl_commit_lr_seq);
2216420dfc95SChris Siden 	zd->zd_shared->zd_seq = zilog->zl_commit_lr_seq;
2217b24ab676SJeff Bonwick 	mutex_exit(&zilog->zl_lock);
2218c9ba2a43SEric Schrock 
2219c9ba2a43SEric Schrock 	(void) rw_unlock(&zd->zd_zilog_lock);
2220c9ba2a43SEric Schrock }
2221c9ba2a43SEric Schrock 
2222c9ba2a43SEric Schrock /*
2223c9ba2a43SEric Schrock  * This function is designed to simulate the operations that occur during a
2224c9ba2a43SEric Schrock  * mount/unmount operation.  We hold the dataset across these operations in an
2225c9ba2a43SEric Schrock  * attempt to expose any implicit assumptions about ZIL management.
2226c9ba2a43SEric Schrock  */
2227c9ba2a43SEric Schrock /* ARGSUSED */
2228c9ba2a43SEric Schrock void
2229c9ba2a43SEric Schrock ztest_zil_remount(ztest_ds_t *zd, uint64_t id)
2230c9ba2a43SEric Schrock {
2231c9ba2a43SEric Schrock 	objset_t *os = zd->zd_os;
2232c9ba2a43SEric Schrock 
2233ce636f8bSMatthew Ahrens 	VERIFY(mutex_lock(&zd->zd_dirobj_lock) == 0);
2234c9ba2a43SEric Schrock 	(void) rw_wrlock(&zd->zd_zilog_lock);
2235c9ba2a43SEric Schrock 
2236c9ba2a43SEric Schrock 	/* zfsvfs_teardown() */
2237c9ba2a43SEric Schrock 	zil_close(zd->zd_zilog);
2238c9ba2a43SEric Schrock 
2239c9ba2a43SEric Schrock 	/* zfsvfs_setup() */
2240c9ba2a43SEric Schrock 	VERIFY(zil_open(os, ztest_get_data) == zd->zd_zilog);
2241c9ba2a43SEric Schrock 	zil_replay(os, zd, ztest_replay_vector);
2242c9ba2a43SEric Schrock 
2243c9ba2a43SEric Schrock 	(void) rw_unlock(&zd->zd_zilog_lock);
2244ce636f8bSMatthew Ahrens 	VERIFY(mutex_unlock(&zd->zd_dirobj_lock) == 0);
2245b24ab676SJeff Bonwick }
2246b24ab676SJeff Bonwick 
2247b24ab676SJeff Bonwick /*
2248b24ab676SJeff Bonwick  * Verify that we can't destroy an active pool, create an existing pool,
2249b24ab676SJeff Bonwick  * or create a pool with a bad vdev spec.
2250b24ab676SJeff Bonwick  */
2251b24ab676SJeff Bonwick /* ARGSUSED */
2252b24ab676SJeff Bonwick void
2253b24ab676SJeff Bonwick ztest_spa_create_destroy(ztest_ds_t *zd, uint64_t id)
2254b24ab676SJeff Bonwick {
2255420dfc95SChris Siden 	ztest_shared_opts_t *zo = &ztest_opts;
2256b24ab676SJeff Bonwick 	spa_t *spa;
2257b24ab676SJeff Bonwick 	nvlist_t *nvroot;
2258b24ab676SJeff Bonwick 
2259b24ab676SJeff Bonwick 	/*
2260b24ab676SJeff Bonwick 	 * Attempt to create using a bad file.
2261b24ab676SJeff Bonwick 	 */
2262b24ab676SJeff Bonwick 	nvroot = make_vdev_root("/dev/bogus", NULL, 0, 0, 0, 0, 0, 1);
2263b24ab676SJeff Bonwick 	VERIFY3U(ENOENT, ==,
22644445fffbSMatthew Ahrens 	    spa_create("ztest_bad_file", nvroot, NULL, NULL));
2265b24ab676SJeff Bonwick 	nvlist_free(nvroot);
2266b24ab676SJeff Bonwick 
2267b24ab676SJeff Bonwick 	/*
2268b24ab676SJeff Bonwick 	 * Attempt to create using a bad mirror.
2269b24ab676SJeff Bonwick 	 */
2270b24ab676SJeff Bonwick 	nvroot = make_vdev_root("/dev/bogus", NULL, 0, 0, 0, 0, 2, 1);
2271b24ab676SJeff Bonwick 	VERIFY3U(ENOENT, ==,
22724445fffbSMatthew Ahrens 	    spa_create("ztest_bad_mirror", nvroot, NULL, NULL));
2273b24ab676SJeff Bonwick 	nvlist_free(nvroot);
2274b24ab676SJeff Bonwick 
2275b24ab676SJeff Bonwick 	/*
2276b24ab676SJeff Bonwick 	 * Attempt to create an existing pool.  It shouldn't matter
2277b24ab676SJeff Bonwick 	 * what's in the nvroot; we should fail with EEXIST.
2278b24ab676SJeff Bonwick 	 */
2279420dfc95SChris Siden 	(void) rw_rdlock(&ztest_name_lock);
2280b24ab676SJeff Bonwick 	nvroot = make_vdev_root("/dev/bogus", NULL, 0, 0, 0, 0, 0, 1);
22814445fffbSMatthew Ahrens 	VERIFY3U(EEXIST, ==, spa_create(zo->zo_pool, nvroot, NULL, NULL));
2282b24ab676SJeff Bonwick 	nvlist_free(nvroot);
2283b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_open(zo->zo_pool, &spa, FTAG));
2284420dfc95SChris Siden 	VERIFY3U(EBUSY, ==, spa_destroy(zo->zo_pool));
2285b24ab676SJeff Bonwick 	spa_close(spa, FTAG);
2286b24ab676SJeff Bonwick 
2287420dfc95SChris Siden 	(void) rw_unlock(&ztest_name_lock);
2288b24ab676SJeff Bonwick }
2289b24ab676SJeff Bonwick 
2290b24ab676SJeff Bonwick static vdev_t *
2291b24ab676SJeff Bonwick vdev_lookup_by_path(vdev_t *vd, const char *path)
2292b24ab676SJeff Bonwick {
2293b24ab676SJeff Bonwick 	vdev_t *mvd;
2294b24ab676SJeff Bonwick 
2295b24ab676SJeff Bonwick 	if (vd->vdev_path != NULL && strcmp(path, vd->vdev_path) == 0)
2296b24ab676SJeff Bonwick 		return (vd);
2297b24ab676SJeff Bonwick 
2298b24ab676SJeff Bonwick 	for (int c = 0; c < vd->vdev_children; c++)
2299b24ab676SJeff Bonwick 		if ((mvd = vdev_lookup_by_path(vd->vdev_child[c], path)) !=
2300b24ab676SJeff Bonwick 		    NULL)
2301b24ab676SJeff Bonwick 			return (mvd);
2302b24ab676SJeff Bonwick 
2303b24ab676SJeff Bonwick 	return (NULL);
2304b24ab676SJeff Bonwick }
2305b24ab676SJeff Bonwick 
2306b24ab676SJeff Bonwick /*
2307b24ab676SJeff Bonwick  * Find the first available hole which can be used as a top-level.
2308b24ab676SJeff Bonwick  */
2309b24ab676SJeff Bonwick int
2310b24ab676SJeff Bonwick find_vdev_hole(spa_t *spa)
2311b24ab676SJeff Bonwick {
2312b24ab676SJeff Bonwick 	vdev_t *rvd = spa->spa_root_vdev;
2313b24ab676SJeff Bonwick 	int c;
2314b24ab676SJeff Bonwick 
2315b24ab676SJeff Bonwick 	ASSERT(spa_config_held(spa, SCL_VDEV, RW_READER) == SCL_VDEV);
2316b24ab676SJeff Bonwick 
2317b24ab676SJeff Bonwick 	for (c = 0; c < rvd->vdev_children; c++) {
2318b24ab676SJeff Bonwick 		vdev_t *cvd = rvd->vdev_child[c];
2319b24ab676SJeff Bonwick 
2320b24ab676SJeff Bonwick 		if (cvd->vdev_ishole)
2321b24ab676SJeff Bonwick 			break;
2322b24ab676SJeff Bonwick 	}
2323b24ab676SJeff Bonwick 	return (c);
2324b24ab676SJeff Bonwick }
2325b24ab676SJeff Bonwick 
2326b24ab676SJeff Bonwick /*
2327b24ab676SJeff Bonwick  * Verify that vdev_add() works as expected.
2328b24ab676SJeff Bonwick  */
2329b24ab676SJeff Bonwick /* ARGSUSED */
2330b24ab676SJeff Bonwick void
2331b24ab676SJeff Bonwick ztest_vdev_add_remove(ztest_ds_t *zd, uint64_t id)
2332b24ab676SJeff Bonwick {
2333b24ab676SJeff Bonwick 	ztest_shared_t *zs = ztest_shared;
2334420dfc95SChris Siden 	spa_t *spa = ztest_spa;
23351195e687SMark J Musante 	uint64_t leaves;
2336b24ab676SJeff Bonwick 	uint64_t guid;
2337b24ab676SJeff Bonwick 	nvlist_t *nvroot;
2338b24ab676SJeff Bonwick 	int error;
2339b24ab676SJeff Bonwick 
2340420dfc95SChris Siden 	VERIFY(mutex_lock(&ztest_vdev_lock) == 0);
2341420dfc95SChris Siden 	leaves =
2342420dfc95SChris Siden 	    MAX(zs->zs_mirrors + zs->zs_splits, 1) * ztest_opts.zo_raidz;
2343b24ab676SJeff Bonwick 
2344b24ab676SJeff Bonwick 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
2345b24ab676SJeff Bonwick 
2346b24ab676SJeff Bonwick 	ztest_shared->zs_vdev_next_leaf = find_vdev_hole(spa) * leaves;
2347b24ab676SJeff Bonwick 
2348b24ab676SJeff Bonwick 	/*
2349b24ab676SJeff Bonwick 	 * If we have slogs then remove them 1/4 of the time.
2350b24ab676SJeff Bonwick 	 */
235188ecc943SGeorge Wilson 	if (spa_has_slogs(spa) && ztest_random(4) == 0) {
235288ecc943SGeorge Wilson 		/*
235388ecc943SGeorge Wilson 		 * Grab the guid from the head of the log class rotor.
235488ecc943SGeorge Wilson 		 */
2355b24ab676SJeff Bonwick 		guid = spa_log_class(spa)->mc_rotor->mg_vd->vdev_guid;
23568654d025Sperrin 
235788ecc943SGeorge Wilson 		spa_config_exit(spa, SCL_VDEV, FTAG);
2358fa9e4066Sahrens 
235988ecc943SGeorge Wilson 		/*
236088ecc943SGeorge Wilson 		 * We have to grab the zs_name_lock as writer to
236188ecc943SGeorge Wilson 		 * prevent a race between removing a slog (dmu_objset_find)
236288ecc943SGeorge Wilson 		 * and destroying a dataset. Removing the slog will
236388ecc943SGeorge Wilson 		 * grab a reference on the dataset which may cause
236488ecc943SGeorge Wilson 		 * dmu_objset_destroy() to fail with EBUSY thus
236588ecc943SGeorge Wilson 		 * leaving the dataset in an inconsistent state.
236688ecc943SGeorge Wilson 		 */
2367420dfc95SChris Siden 		VERIFY(rw_wrlock(&ztest_name_lock) == 0);
236888ecc943SGeorge Wilson 		error = spa_vdev_remove(spa, guid, B_FALSE);
2369420dfc95SChris Siden 		VERIFY(rw_unlock(&ztest_name_lock) == 0);
2370fa9e4066Sahrens 
237188ecc943SGeorge Wilson 		if (error && error != EEXIST)
237288ecc943SGeorge Wilson 			fatal(0, "spa_vdev_remove() = %d", error);
237388ecc943SGeorge Wilson 	} else {
237488ecc943SGeorge Wilson 		spa_config_exit(spa, SCL_VDEV, FTAG);
237588ecc943SGeorge Wilson 
237688ecc943SGeorge Wilson 		/*
237788ecc943SGeorge Wilson 		 * Make 1/4 of the devices be log devices.
237888ecc943SGeorge Wilson 		 */
2379420dfc95SChris Siden 		nvroot = make_vdev_root(NULL, NULL,
2380420dfc95SChris Siden 		    ztest_opts.zo_vdev_size, 0,
2381420dfc95SChris Siden 		    ztest_random(4) == 0, ztest_opts.zo_raidz,
2382420dfc95SChris Siden 		    zs->zs_mirrors, 1);
238388ecc943SGeorge Wilson 
238488ecc943SGeorge Wilson 		error = spa_vdev_add(spa, nvroot);
238588ecc943SGeorge Wilson 		nvlist_free(nvroot);
238688ecc943SGeorge Wilson 
238788ecc943SGeorge Wilson 		if (error == ENOSPC)
238888ecc943SGeorge Wilson 			ztest_record_enospc("spa_vdev_add");
238988ecc943SGeorge Wilson 		else if (error != 0)
239088ecc943SGeorge Wilson 			fatal(0, "spa_vdev_add() = %d", error);
239188ecc943SGeorge Wilson 	}
239288ecc943SGeorge Wilson 
2393420dfc95SChris Siden 	VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
2394e14bb325SJeff Bonwick }
2395fa9e4066Sahrens 
2396e14bb325SJeff Bonwick /*
2397e14bb325SJeff Bonwick  * Verify that adding/removing aux devices (l2arc, hot spare) works as expected.
2398e14bb325SJeff Bonwick  */
2399b24ab676SJeff Bonwick /* ARGSUSED */
2400e14bb325SJeff Bonwick void
2401b24ab676SJeff Bonwick ztest_vdev_aux_add_remove(ztest_ds_t *zd, uint64_t id)
2402e14bb325SJeff Bonwick {
2403b24ab676SJeff Bonwick 	ztest_shared_t *zs = ztest_shared;
2404420dfc95SChris Siden 	spa_t *spa = ztest_spa;
240531157203SJeff Bonwick 	vdev_t *rvd = spa->spa_root_vdev;
2406e14bb325SJeff Bonwick 	spa_aux_vdev_t *sav;
2407e14bb325SJeff Bonwick 	char *aux;
2408e14bb325SJeff Bonwick 	uint64_t guid = 0;
2409e14bb325SJeff Bonwick 	int error;
2410e14bb325SJeff Bonwick 
241131157203SJeff Bonwick 	if (ztest_random(2) == 0) {
2412e14bb325SJeff Bonwick 		sav = &spa->spa_spares;
2413e14bb325SJeff Bonwick 		aux = ZPOOL_CONFIG_SPARES;
2414e14bb325SJeff Bonwick 	} else {
2415e14bb325SJeff Bonwick 		sav = &spa->spa_l2cache;
2416e14bb325SJeff Bonwick 		aux = ZPOOL_CONFIG_L2CACHE;
2417e14bb325SJeff Bonwick 	}
2418e14bb325SJeff Bonwick 
2419420dfc95SChris Siden 	VERIFY(mutex_lock(&ztest_vdev_lock) == 0);
2420e14bb325SJeff Bonwick 
2421e14bb325SJeff Bonwick 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
2422e14bb325SJeff Bonwick 
2423e14bb325SJeff Bonwick 	if (sav->sav_count != 0 && ztest_random(4) == 0) {
2424e14bb325SJeff Bonwick 		/*
2425e14bb325SJeff Bonwick 		 * Pick a random device to remove.
2426e14bb325SJeff Bonwick 		 */
2427e14bb325SJeff Bonwick 		guid = sav->sav_vdevs[ztest_random(sav->sav_count)]->vdev_guid;
2428e14bb325SJeff Bonwick 	} else {
2429e14bb325SJeff Bonwick 		/*
2430e14bb325SJeff Bonwick 		 * Find an unused device we can add.
2431e14bb325SJeff Bonwick 		 */
2432b24ab676SJeff Bonwick 		zs->zs_vdev_aux = 0;
2433e14bb325SJeff Bonwick 		for (;;) {
2434e14bb325SJeff Bonwick 			char path[MAXPATHLEN];
2435e14bb325SJeff Bonwick 			int c;
2436420dfc95SChris Siden 			(void) snprintf(path, sizeof (path), ztest_aux_template,
2437420dfc95SChris Siden 			    ztest_opts.zo_dir, ztest_opts.zo_pool, aux,
2438420dfc95SChris Siden 			    zs->zs_vdev_aux);
2439e14bb325SJeff Bonwick 			for (c = 0; c < sav->sav_count; c++)
2440e14bb325SJeff Bonwick 				if (strcmp(sav->sav_vdevs[c]->vdev_path,
2441e14bb325SJeff Bonwick 				    path) == 0)
2442e14bb325SJeff Bonwick 					break;
244331157203SJeff Bonwick 			if (c == sav->sav_count &&
244431157203SJeff Bonwick 			    vdev_lookup_by_path(rvd, path) == NULL)
2445e14bb325SJeff Bonwick 				break;
2446b24ab676SJeff Bonwick 			zs->zs_vdev_aux++;
2447e14bb325SJeff Bonwick 		}
2448e14bb325SJeff Bonwick 	}
2449e14bb325SJeff Bonwick 
2450e14bb325SJeff Bonwick 	spa_config_exit(spa, SCL_VDEV, FTAG);
2451e14bb325SJeff Bonwick 
2452e14bb325SJeff Bonwick 	if (guid == 0) {
2453e14bb325SJeff Bonwick 		/*
2454e14bb325SJeff Bonwick 		 * Add a new device.
2455e14bb325SJeff Bonwick 		 */
245631157203SJeff Bonwick 		nvlist_t *nvroot = make_vdev_root(NULL, aux,
2457420dfc95SChris Siden 		    (ztest_opts.zo_vdev_size * 5) / 4, 0, 0, 0, 0, 1);
2458e14bb325SJeff Bonwick 		error = spa_vdev_add(spa, nvroot);
2459e14bb325SJeff Bonwick 		if (error != 0)
2460e14bb325SJeff Bonwick 			fatal(0, "spa_vdev_add(%p) = %d", nvroot, error);
2461e14bb325SJeff Bonwick 		nvlist_free(nvroot);
2462e14bb325SJeff Bonwick 	} else {
2463e14bb325SJeff Bonwick 		/*
2464e14bb325SJeff Bonwick 		 * Remove an existing device.  Sometimes, dirty its
2465e14bb325SJeff Bonwick 		 * vdev state first to make sure we handle removal
2466e14bb325SJeff Bonwick 		 * of devices that have pending state changes.
2467e14bb325SJeff Bonwick 		 */
2468e14bb325SJeff Bonwick 		if (ztest_random(2) == 0)
2469573ca77eSGeorge Wilson 			(void) vdev_online(spa, guid, 0, NULL);
2470e14bb325SJeff Bonwick 
2471e14bb325SJeff Bonwick 		error = spa_vdev_remove(spa, guid, B_FALSE);
2472e14bb325SJeff Bonwick 		if (error != 0 && error != EBUSY)
2473e14bb325SJeff Bonwick 			fatal(0, "spa_vdev_remove(%llu) = %d", guid, error);
2474e14bb325SJeff Bonwick 	}
2475e14bb325SJeff Bonwick 
2476420dfc95SChris Siden 	VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
2477fa9e4066Sahrens }
2478fa9e4066Sahrens 
24791195e687SMark J Musante /*
24801195e687SMark J Musante  * split a pool if it has mirror tlvdevs
24811195e687SMark J Musante  */
24821195e687SMark J Musante /* ARGSUSED */
24831195e687SMark J Musante void
24841195e687SMark J Musante ztest_split_pool(ztest_ds_t *zd, uint64_t id)
24851195e687SMark J Musante {
24861195e687SMark J Musante 	ztest_shared_t *zs = ztest_shared;
2487420dfc95SChris Siden 	spa_t *spa = ztest_spa;
24881195e687SMark J Musante 	vdev_t *rvd = spa->spa_root_vdev;
24891195e687SMark J Musante 	nvlist_t *tree, **child, *config, *split, **schild;
24901195e687SMark J Musante 	uint_t c, children, schildren = 0, lastlogid = 0;
24911195e687SMark J Musante 	int error = 0;
24921195e687SMark J Musante 
2493420dfc95SChris Siden 	VERIFY(mutex_lock(&ztest_vdev_lock) == 0);
24941195e687SMark J Musante 
24951195e687SMark J Musante 	/* ensure we have a useable config; mirrors of raidz aren't supported */
2496420dfc95SChris Siden 	if (zs->zs_mirrors < 3 || ztest_opts.zo_raidz > 1) {
2497420dfc95SChris Siden 		VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
24981195e687SMark J Musante 		return;
24991195e687SMark J Musante 	}
25001195e687SMark J Musante 
25011195e687SMark J Musante 	/* clean up the old pool, if any */
25021195e687SMark J Musante 	(void) spa_destroy("splitp");
25031195e687SMark J Musante 
25041195e687SMark J Musante 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
25051195e687SMark J Musante 
25061195e687SMark J Musante 	/* generate a config from the existing config */
250798295d61SMark J Musante 	mutex_enter(&spa->spa_props_lock);
25081195e687SMark J Musante 	VERIFY(nvlist_lookup_nvlist(spa->spa_config, ZPOOL_CONFIG_VDEV_TREE,
25091195e687SMark J Musante 	    &tree) == 0);
251098295d61SMark J Musante 	mutex_exit(&spa->spa_props_lock);
251198295d61SMark J Musante 
25121195e687SMark J Musante 	VERIFY(nvlist_lookup_nvlist_array(tree, ZPOOL_CONFIG_CHILDREN, &child,
25131195e687SMark J Musante 	    &children) == 0);
25141195e687SMark J Musante 
25151195e687SMark J Musante 	schild = malloc(rvd->vdev_children * sizeof (nvlist_t *));
25161195e687SMark J Musante 	for (c = 0; c < children; c++) {
25171195e687SMark J Musante 		vdev_t *tvd = rvd->vdev_child[c];
25181195e687SMark J Musante 		nvlist_t **mchild;
25191195e687SMark J Musante 		uint_t mchildren;
25201195e687SMark J Musante 
25211195e687SMark J Musante 		if (tvd->vdev_islog || tvd->vdev_ops == &vdev_hole_ops) {
25221195e687SMark J Musante 			VERIFY(nvlist_alloc(&schild[schildren], NV_UNIQUE_NAME,
25231195e687SMark J Musante 			    0) == 0);
25241195e687SMark J Musante 			VERIFY(nvlist_add_string(schild[schildren],
25251195e687SMark J Musante 			    ZPOOL_CONFIG_TYPE, VDEV_TYPE_HOLE) == 0);
25261195e687SMark J Musante 			VERIFY(nvlist_add_uint64(schild[schildren],
25271195e687SMark J Musante 			    ZPOOL_CONFIG_IS_HOLE, 1) == 0);
25281195e687SMark J Musante 			if (lastlogid == 0)
25291195e687SMark J Musante 				lastlogid = schildren;
25301195e687SMark J Musante 			++schildren;
25311195e687SMark J Musante 			continue;
25321195e687SMark J Musante 		}
25331195e687SMark J Musante 		lastlogid = 0;
25341195e687SMark J Musante 		VERIFY(nvlist_lookup_nvlist_array(child[c],
25351195e687SMark J Musante 		    ZPOOL_CONFIG_CHILDREN, &mchild, &mchildren) == 0);
25361195e687SMark J Musante 		VERIFY(nvlist_dup(mchild[0], &schild[schildren++], 0) == 0);
25371195e687SMark J Musante 	}
25381195e687SMark J Musante 
25391195e687SMark J Musante 	/* OK, create a config that can be used to split */
25401195e687SMark J Musante 	VERIFY(nvlist_alloc(&split, NV_UNIQUE_NAME, 0) == 0);
25411195e687SMark J Musante 	VERIFY(nvlist_add_string(split, ZPOOL_CONFIG_TYPE,
25421195e687SMark J Musante 	    VDEV_TYPE_ROOT) == 0);
25431195e687SMark J Musante 	VERIFY(nvlist_add_nvlist_array(split, ZPOOL_CONFIG_CHILDREN, schild,
25441195e687SMark J Musante 	    lastlogid != 0 ? lastlogid : schildren) == 0);
25451195e687SMark J Musante 
25461195e687SMark J Musante 	VERIFY(nvlist_alloc(&config, NV_UNIQUE_NAME, 0) == 0);
25471195e687SMark J Musante 	VERIFY(nvlist_add_nvlist(config, ZPOOL_CONFIG_VDEV_TREE, split) == 0);
25481195e687SMark J Musante 
25491195e687SMark J Musante 	for (c = 0; c < schildren; c++)
25501195e687SMark J Musante 		nvlist_free(schild[c]);
25511195e687SMark J Musante 	free(schild);
25521195e687SMark J Musante 	nvlist_free(split);
25531195e687SMark J Musante 
25541195e687SMark J Musante 	spa_config_exit(spa, SCL_VDEV, FTAG);
25551195e687SMark J Musante 
2556420dfc95SChris Siden 	(void) rw_wrlock(&ztest_name_lock);
25571195e687SMark J Musante 	error = spa_vdev_split_mirror(spa, "splitp", config, NULL, B_FALSE);
2558420dfc95SChris Siden 	(void) rw_unlock(&ztest_name_lock);
25591195e687SMark J Musante 
25601195e687SMark J Musante 	nvlist_free(config);
25611195e687SMark J Musante 
25621195e687SMark J Musante 	if (error == 0) {
25631195e687SMark J Musante 		(void) printf("successful split - results:\n");
25641195e687SMark J Musante 		mutex_enter(&spa_namespace_lock);
25651195e687SMark J Musante 		show_pool_stats(spa);
25661195e687SMark J Musante 		show_pool_stats(spa_lookup("splitp"));
25671195e687SMark J Musante 		mutex_exit(&spa_namespace_lock);
25681195e687SMark J Musante 		++zs->zs_splits;
25691195e687SMark J Musante 		--zs->zs_mirrors;
25701195e687SMark J Musante 	}
2571420dfc95SChris Siden 	VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
25721195e687SMark J Musante 
25731195e687SMark J Musante }
25741195e687SMark J Musante 
2575fa9e4066Sahrens /*
2576fa9e4066Sahrens  * Verify that we can attach and detach devices.
2577fa9e4066Sahrens  */
2578b24ab676SJeff Bonwick /* ARGSUSED */
2579fa9e4066Sahrens void
2580b24ab676SJeff Bonwick ztest_vdev_attach_detach(ztest_ds_t *zd, uint64_t id)
2581fa9e4066Sahrens {
2582b24ab676SJeff Bonwick 	ztest_shared_t *zs = ztest_shared;
2583420dfc95SChris Siden 	spa_t *spa = ztest_spa;
258431157203SJeff Bonwick 	spa_aux_vdev_t *sav = &spa->spa_spares;
2585fa9e4066Sahrens 	vdev_t *rvd = spa->spa_root_vdev;
2586ea8dc4b6Seschrock 	vdev_t *oldvd, *newvd, *pvd;
2587e14bb325SJeff Bonwick 	nvlist_t *root;
25881195e687SMark J Musante 	uint64_t leaves;
2589fa9e4066Sahrens 	uint64_t leaf, top;
2590ecc2d604Sbonwick 	uint64_t ashift = ztest_get_ashift();
25918ad4d6ddSJeff Bonwick 	uint64_t oldguid, pguid;
2592ea8dc4b6Seschrock 	size_t oldsize, newsize;
2593ea8dc4b6Seschrock 	char oldpath[MAXPATHLEN], newpath[MAXPATHLEN];
2594fa9e4066Sahrens 	int replacing;
25959af0a4dfSJeff Bonwick 	int oldvd_has_siblings = B_FALSE;
259631157203SJeff Bonwick 	int newvd_is_spare = B_FALSE;
259731157203SJeff Bonwick 	int oldvd_is_log;
2598fa9e4066Sahrens 	int error, expected_error;
2599fa9e4066Sahrens 
2600420dfc95SChris Siden 	VERIFY(mutex_lock(&ztest_vdev_lock) == 0);
2601420dfc95SChris Siden 	leaves = MAX(zs->zs_mirrors, 1) * ztest_opts.zo_raidz;
2602fa9e4066Sahrens 
2603e14bb325SJeff Bonwick 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
2604fa9e4066Sahrens 
2605fa9e4066Sahrens 	/*
2606fa9e4066Sahrens 	 * Decide whether to do an attach or a replace.
2607fa9e4066Sahrens 	 */
2608fa9e4066Sahrens 	replacing = ztest_random(2);
2609fa9e4066Sahrens 
2610fa9e4066Sahrens 	/*
2611fa9e4066Sahrens 	 * Pick a random top-level vdev.
2612fa9e4066Sahrens 	 */
2613b24ab676SJeff Bonwick 	top = ztest_random_vdev_top(spa, B_TRUE);
2614fa9e4066Sahrens 
2615fa9e4066Sahrens 	/*
2616fa9e4066Sahrens 	 * Pick a random leaf within it.
2617fa9e4066Sahrens 	 */
2618d41c4376SMark J Musante 	leaf = ztest_random(leaves);
2619fa9e4066Sahrens 
2620fa9e4066Sahrens 	/*
262131157203SJeff Bonwick 	 * Locate this vdev.
2622fa9e4066Sahrens 	 */
262331157203SJeff Bonwick 	oldvd = rvd->vdev_child[top];
26241195e687SMark J Musante 	if (zs->zs_mirrors >= 1) {
26258ad4d6ddSJeff Bonwick 		ASSERT(oldvd->vdev_ops == &vdev_mirror_ops);
26261195e687SMark J Musante 		ASSERT(oldvd->vdev_children >= zs->zs_mirrors);
2627420dfc95SChris Siden 		oldvd = oldvd->vdev_child[leaf / ztest_opts.zo_raidz];
26288ad4d6ddSJeff Bonwick 	}
2629420dfc95SChris Siden 	if (ztest_opts.zo_raidz > 1) {
26308ad4d6ddSJeff Bonwick 		ASSERT(oldvd->vdev_ops == &vdev_raidz_ops);
2631420dfc95SChris Siden 		ASSERT(oldvd->vdev_children == ztest_opts.zo_raidz);
2632420dfc95SChris Siden 		oldvd = oldvd->vdev_child[leaf % ztest_opts.zo_raidz];
26338ad4d6ddSJeff Bonwick 	}
2634fa9e4066Sahrens 
2635fa9e4066Sahrens 	/*
263631157203SJeff Bonwick 	 * If we're already doing an attach or replace, oldvd may be a
263731157203SJeff Bonwick 	 * mirror vdev -- in which case, pick a random child.
2638fa9e4066Sahrens 	 */
263931157203SJeff Bonwick 	while (oldvd->vdev_children != 0) {
26409af0a4dfSJeff Bonwick 		oldvd_has_siblings = B_TRUE;
26418ad4d6ddSJeff Bonwick 		ASSERT(oldvd->vdev_children >= 2);
26428ad4d6ddSJeff Bonwick 		oldvd = oldvd->vdev_child[ztest_random(oldvd->vdev_children)];
264331157203SJeff Bonwick 	}
2644fa9e4066Sahrens 
264531157203SJeff Bonwick 	oldguid = oldvd->vdev_guid;
2646573ca77eSGeorge Wilson 	oldsize = vdev_get_min_asize(oldvd);
264731157203SJeff Bonwick 	oldvd_is_log = oldvd->vdev_top->vdev_islog;
264831157203SJeff Bonwick 	(void) strcpy(oldpath, oldvd->vdev_path);
2649ea8dc4b6Seschrock 	pvd = oldvd->vdev_parent;
26508ad4d6ddSJeff Bonwick 	pguid = pvd->vdev_guid;
2651fa9e4066Sahrens 
26529af0a4dfSJeff Bonwick 	/*
26539af0a4dfSJeff Bonwick 	 * If oldvd has siblings, then half of the time, detach it.
26549af0a4dfSJeff Bonwick 	 */
26559af0a4dfSJeff Bonwick 	if (oldvd_has_siblings && ztest_random(2) == 0) {
26569af0a4dfSJeff Bonwick 		spa_config_exit(spa, SCL_VDEV, FTAG);
26578ad4d6ddSJeff Bonwick 		error = spa_vdev_detach(spa, oldguid, pguid, B_FALSE);
26588ad4d6ddSJeff Bonwick 		if (error != 0 && error != ENODEV && error != EBUSY &&
26598ad4d6ddSJeff Bonwick 		    error != ENOTSUP)
26608ad4d6ddSJeff Bonwick 			fatal(0, "detach (%s) returned %d", oldpath, error);
2661420dfc95SChris Siden 		VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
26629af0a4dfSJeff Bonwick 		return;
26639af0a4dfSJeff Bonwick 	}
26649af0a4dfSJeff Bonwick 
2665fa9e4066Sahrens 	/*
266631157203SJeff Bonwick 	 * For the new vdev, choose with equal probability between the two
266731157203SJeff Bonwick 	 * standard paths (ending in either 'a' or 'b') or a random hot spare.
2668fa9e4066Sahrens 	 */
266931157203SJeff Bonwick 	if (sav->sav_count != 0 && ztest_random(3) == 0) {
267031157203SJeff Bonwick 		newvd = sav->sav_vdevs[ztest_random(sav->sav_count)];
267131157203SJeff Bonwick 		newvd_is_spare = B_TRUE;
267231157203SJeff Bonwick 		(void) strcpy(newpath, newvd->vdev_path);
267331157203SJeff Bonwick 	} else {
267431157203SJeff Bonwick 		(void) snprintf(newpath, sizeof (newpath), ztest_dev_template,
2675420dfc95SChris Siden 		    ztest_opts.zo_dir, ztest_opts.zo_pool,
2676420dfc95SChris Siden 		    top * leaves + leaf);
267731157203SJeff Bonwick 		if (ztest_random(2) == 0)
267831157203SJeff Bonwick 			newpath[strlen(newpath) - 1] = 'b';
267931157203SJeff Bonwick 		newvd = vdev_lookup_by_path(rvd, newpath);
268031157203SJeff Bonwick 	}
268131157203SJeff Bonwick 
268231157203SJeff Bonwick 	if (newvd) {
2683573ca77eSGeorge Wilson 		newsize = vdev_get_min_asize(newvd);
268431157203SJeff Bonwick 	} else {
268531157203SJeff Bonwick 		/*
268631157203SJeff Bonwick 		 * Make newsize a little bigger or smaller than oldsize.
268731157203SJeff Bonwick 		 * If it's smaller, the attach should fail.
268831157203SJeff Bonwick 		 * If it's larger, and we're doing a replace,
268931157203SJeff Bonwick 		 * we should get dynamic LUN growth when we're done.
269031157203SJeff Bonwick 		 */
269131157203SJeff Bonwick 		newsize = 10 * oldsize / (9 + ztest_random(3));
269231157203SJeff Bonwick 	}
2693fa9e4066Sahrens 
2694fa9e4066Sahrens 	/*
2695fa9e4066Sahrens 	 * If pvd is not a mirror or root, the attach should fail with ENOTSUP,
2696fa9e4066Sahrens 	 * unless it's a replace; in that case any non-replacing parent is OK.
2697fa9e4066Sahrens 	 *
2698ea8dc4b6Seschrock 	 * If newvd is already part of the pool, it should fail with EBUSY.
2699fa9e4066Sahrens 	 *
2700ea8dc4b6Seschrock 	 * If newvd is too small, it should fail with EOVERFLOW.
2701fa9e4066Sahrens 	 */
270231157203SJeff Bonwick 	if (pvd->vdev_ops != &vdev_mirror_ops &&
270331157203SJeff Bonwick 	    pvd->vdev_ops != &vdev_root_ops && (!replacing ||
270431157203SJeff Bonwick 	    pvd->vdev_ops == &vdev_replacing_ops ||
270531157203SJeff Bonwick 	    pvd->vdev_ops == &vdev_spare_ops))
270631157203SJeff Bonwick 		expected_error = ENOTSUP;
270731157203SJeff Bonwick 	else if (newvd_is_spare && (!replacing || oldvd_is_log))
2708fa9e4066Sahrens 		expected_error = ENOTSUP;
270931157203SJeff Bonwick 	else if (newvd == oldvd)
271031157203SJeff Bonwick 		expected_error = replacing ? 0 : EBUSY;
271131157203SJeff Bonwick 	else if (vdev_lookup_by_path(rvd, newpath) != NULL)
271231157203SJeff Bonwick 		expected_error = EBUSY;
2713ea8dc4b6Seschrock 	else if (newsize < oldsize)
2714fa9e4066Sahrens 		expected_error = EOVERFLOW;
2715ecc2d604Sbonwick 	else if (ashift > oldvd->vdev_top->vdev_ashift)
2716ecc2d604Sbonwick 		expected_error = EDOM;
2717fa9e4066Sahrens 	else
2718fa9e4066Sahrens 		expected_error = 0;
2719fa9e4066Sahrens 
2720e14bb325SJeff Bonwick 	spa_config_exit(spa, SCL_VDEV, FTAG);
2721fa9e4066Sahrens 
2722fa9e4066Sahrens 	/*
2723ea8dc4b6Seschrock 	 * Build the nvlist describing newpath.
2724fa9e4066Sahrens 	 */
2725e14bb325SJeff Bonwick 	root = make_vdev_root(newpath, NULL, newvd == NULL ? newsize : 0,
2726e14bb325SJeff Bonwick 	    ashift, 0, 0, 0, 1);
2727fa9e4066Sahrens 
272831157203SJeff Bonwick 	error = spa_vdev_attach(spa, oldguid, root, replacing);
2729fa9e4066Sahrens 
2730fa9e4066Sahrens 	nvlist_free(root);
2731fa9e4066Sahrens 
2732fa9e4066Sahrens 	/*
2733fa9e4066Sahrens 	 * If our parent was the replacing vdev, but the replace completed,
2734fa9e4066Sahrens 	 * then instead of failing with ENOTSUP we may either succeed,
2735fa9e4066Sahrens 	 * fail with ENODEV, or fail with EOVERFLOW.
2736fa9e4066Sahrens 	 */
2737fa9e4066Sahrens 	if (expected_error == ENOTSUP &&
2738fa9e4066Sahrens 	    (error == 0 || error == ENODEV || error == EOVERFLOW))
2739fa9e4066Sahrens 		expected_error = error;
2740fa9e4066Sahrens 
2741f0aa80d4Sbonwick 	/*
2742f0aa80d4Sbonwick 	 * If someone grew the LUN, the replacement may be too small.
2743f0aa80d4Sbonwick 	 */
2744088f3894Sahrens 	if (error == EOVERFLOW || error == EBUSY)
2745f0aa80d4Sbonwick 		expected_error = error;
2746f0aa80d4Sbonwick 
2747088f3894Sahrens 	/* XXX workaround 6690467 */
2748088f3894Sahrens 	if (error != expected_error && expected_error != EBUSY) {
2749088f3894Sahrens 		fatal(0, "attach (%s %llu, %s %llu, %d) "
2750088f3894Sahrens 		    "returned %d, expected %d",
2751088f3894Sahrens 		    oldpath, (longlong_t)oldsize, newpath,
2752088f3894Sahrens 		    (longlong_t)newsize, replacing, error, expected_error);
2753fa9e4066Sahrens 	}
2754fa9e4066Sahrens 
2755420dfc95SChris Siden 	VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
2756fa9e4066Sahrens }
2757fa9e4066Sahrens 
2758573ca77eSGeorge Wilson /*
2759573ca77eSGeorge Wilson  * Callback function which expands the physical size of the vdev.
2760573ca77eSGeorge Wilson  */
2761573ca77eSGeorge Wilson vdev_t *
2762573ca77eSGeorge Wilson grow_vdev(vdev_t *vd, void *arg)
2763573ca77eSGeorge Wilson {
2764573ca77eSGeorge Wilson 	spa_t *spa = vd->vdev_spa;
2765573ca77eSGeorge Wilson 	size_t *newsize = arg;
2766573ca77eSGeorge Wilson 	size_t fsize;
2767573ca77eSGeorge Wilson 	int fd;
2768573ca77eSGeorge Wilson 
2769573ca77eSGeorge Wilson 	ASSERT(spa_config_held(spa, SCL_STATE, RW_READER) == SCL_STATE);
2770573ca77eSGeorge Wilson 	ASSERT(vd->vdev_ops->vdev_op_leaf);
2771573ca77eSGeorge Wilson 
2772573ca77eSGeorge Wilson 	if ((fd = open(vd->vdev_path, O_RDWR)) == -1)
2773573ca77eSGeorge Wilson 		return (vd);
2774573ca77eSGeorge Wilson 
2775573ca77eSGeorge Wilson 	fsize = lseek(fd, 0, SEEK_END);
2776573ca77eSGeorge Wilson 	(void) ftruncate(fd, *newsize);
2777573ca77eSGeorge Wilson 
2778420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 6) {
2779573ca77eSGeorge Wilson 		(void) printf("%s grew from %lu to %lu bytes\n",
2780573ca77eSGeorge Wilson 		    vd->vdev_path, (ulong_t)fsize, (ulong_t)*newsize);
2781573ca77eSGeorge Wilson 	}
2782573ca77eSGeorge Wilson 	(void) close(fd);
2783573ca77eSGeorge Wilson 	return (NULL);
2784573ca77eSGeorge Wilson }
2785573ca77eSGeorge Wilson 
2786573ca77eSGeorge Wilson /*
2787573ca77eSGeorge Wilson  * Callback function which expands a given vdev by calling vdev_online().
2788573ca77eSGeorge Wilson  */
2789573ca77eSGeorge Wilson /* ARGSUSED */
2790573ca77eSGeorge Wilson vdev_t *
2791573ca77eSGeorge Wilson online_vdev(vdev_t *vd, void *arg)
2792573ca77eSGeorge Wilson {
2793573ca77eSGeorge Wilson 	spa_t *spa = vd->vdev_spa;
2794573ca77eSGeorge Wilson 	vdev_t *tvd = vd->vdev_top;
2795573ca77eSGeorge Wilson 	uint64_t guid = vd->vdev_guid;
27968f18d1faSGeorge Wilson 	uint64_t generation = spa->spa_config_generation + 1;
2797095bcd66SGeorge Wilson 	vdev_state_t newstate = VDEV_STATE_UNKNOWN;
2798095bcd66SGeorge Wilson 	int error;
2799573ca77eSGeorge Wilson 
2800573ca77eSGeorge Wilson 	ASSERT(spa_config_held(spa, SCL_STATE, RW_READER) == SCL_STATE);
2801573ca77eSGeorge Wilson 	ASSERT(vd->vdev_ops->vdev_op_leaf);
2802573ca77eSGeorge Wilson 
2803573ca77eSGeorge Wilson 	/* Calling vdev_online will initialize the new metaslabs */
2804573ca77eSGeorge Wilson 	spa_config_exit(spa, SCL_STATE, spa);
2805095bcd66SGeorge Wilson 	error = vdev_online(spa, guid, ZFS_ONLINE_EXPAND, &newstate);
2806573ca77eSGeorge Wilson 	spa_config_enter(spa, SCL_STATE, spa, RW_READER);
2807573ca77eSGeorge Wilson 
2808095bcd66SGeorge Wilson 	/*
2809095bcd66SGeorge Wilson 	 * If vdev_online returned an error or the underlying vdev_open
2810095bcd66SGeorge Wilson 	 * failed then we abort the expand. The only way to know that
2811095bcd66SGeorge Wilson 	 * vdev_open fails is by checking the returned newstate.
2812095bcd66SGeorge Wilson 	 */
2813095bcd66SGeorge Wilson 	if (error || newstate != VDEV_STATE_HEALTHY) {
2814420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 5) {
2815095bcd66SGeorge Wilson 			(void) printf("Unable to expand vdev, state %llu, "
2816095bcd66SGeorge Wilson 			    "error %d\n", (u_longlong_t)newstate, error);
2817095bcd66SGeorge Wilson 		}
2818095bcd66SGeorge Wilson 		return (vd);
2819095bcd66SGeorge Wilson 	}
2820095bcd66SGeorge Wilson 	ASSERT3U(newstate, ==, VDEV_STATE_HEALTHY);
2821095bcd66SGeorge Wilson 
2822573ca77eSGeorge Wilson 	/*
2823573ca77eSGeorge Wilson 	 * Since we dropped the lock we need to ensure that we're
2824573ca77eSGeorge Wilson 	 * still talking to the original vdev. It's possible this
2825573ca77eSGeorge Wilson 	 * vdev may have been detached/replaced while we were
2826573ca77eSGeorge Wilson 	 * trying to online it.
2827573ca77eSGeorge Wilson 	 */
28288f18d1faSGeorge Wilson 	if (generation != spa->spa_config_generation) {
2829420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 5) {
28308f18d1faSGeorge Wilson 			(void) printf("vdev configuration has changed, "
28318f18d1faSGeorge Wilson 			    "guid %llu, state %llu, expected gen %llu, "
2832b24ab676SJeff Bonwick 			    "got gen %llu\n",
2833b24ab676SJeff Bonwick 			    (u_longlong_t)guid,
28348f18d1faSGeorge Wilson 			    (u_longlong_t)tvd->vdev_state,
28358f18d1faSGeorge Wilson 			    (u_longlong_t)generation,
28368f18d1faSGeorge Wilson 			    (u_longlong_t)spa->spa_config_generation);
2837573ca77eSGeorge Wilson 		}
2838573ca77eSGeorge Wilson 		return (vd);
2839573ca77eSGeorge Wilson 	}
2840573ca77eSGeorge Wilson 	return (NULL);
2841573ca77eSGeorge Wilson }
2842573ca77eSGeorge Wilson 
2843573ca77eSGeorge Wilson /*
2844573ca77eSGeorge Wilson  * Traverse the vdev tree calling the supplied function.
2845573ca77eSGeorge Wilson  * We continue to walk the tree until we either have walked all
2846573ca77eSGeorge Wilson  * children or we receive a non-NULL return from the callback.
2847573ca77eSGeorge Wilson  * If a NULL callback is passed, then we just return back the first
2848573ca77eSGeorge Wilson  * leaf vdev we encounter.
2849573ca77eSGeorge Wilson  */
2850573ca77eSGeorge Wilson vdev_t *
2851573ca77eSGeorge Wilson vdev_walk_tree(vdev_t *vd, vdev_t *(*func)(vdev_t *, void *), void *arg)
2852573ca77eSGeorge Wilson {
2853573ca77eSGeorge Wilson 	if (vd->vdev_ops->vdev_op_leaf) {
2854573ca77eSGeorge Wilson 		if (func == NULL)
2855573ca77eSGeorge Wilson 			return (vd);
2856573ca77eSGeorge Wilson 		else
2857573ca77eSGeorge Wilson 			return (func(vd, arg));
2858573ca77eSGeorge Wilson 	}
2859573ca77eSGeorge Wilson 
2860573ca77eSGeorge Wilson 	for (uint_t c = 0; c < vd->vdev_children; c++) {
2861573ca77eSGeorge Wilson 		vdev_t *cvd = vd->vdev_child[c];
2862573ca77eSGeorge Wilson 		if ((cvd = vdev_walk_tree(cvd, func, arg)) != NULL)
2863573ca77eSGeorge Wilson 			return (cvd);
2864573ca77eSGeorge Wilson 	}
2865573ca77eSGeorge Wilson 	return (NULL);
2866573ca77eSGeorge Wilson }
2867573ca77eSGeorge Wilson 
2868fa9e4066Sahrens /*
2869fa9e4066Sahrens  * Verify that dynamic LUN growth works as expected.
2870fa9e4066Sahrens  */
2871b24ab676SJeff Bonwick /* ARGSUSED */
2872fa9e4066Sahrens void
2873b24ab676SJeff Bonwick ztest_vdev_LUN_growth(ztest_ds_t *zd, uint64_t id)
2874fa9e4066Sahrens {
2875420dfc95SChris Siden 	spa_t *spa = ztest_spa;
2876b24ab676SJeff Bonwick 	vdev_t *vd, *tvd;
2877b24ab676SJeff Bonwick 	metaslab_class_t *mc;
2878b24ab676SJeff Bonwick 	metaslab_group_t *mg;
2879573ca77eSGeorge Wilson 	size_t psize, newsize;
2880b24ab676SJeff Bonwick 	uint64_t top;
2881b24ab676SJeff Bonwick 	uint64_t old_class_space, new_class_space, old_ms_count, new_ms_count;
2882fa9e4066Sahrens 
2883420dfc95SChris Siden 	VERIFY(mutex_lock(&ztest_vdev_lock) == 0);
2884573ca77eSGeorge Wilson 	spa_config_enter(spa, SCL_STATE, spa, RW_READER);
2885573ca77eSGeorge Wilson 
2886b24ab676SJeff Bonwick 	top = ztest_random_vdev_top(spa, B_TRUE);
2887573ca77eSGeorge Wilson 
2888b24ab676SJeff Bonwick 	tvd = spa->spa_root_vdev->vdev_child[top];
2889b24ab676SJeff Bonwick 	mg = tvd->vdev_mg;
2890b24ab676SJeff Bonwick 	mc = mg->mg_class;
2891b24ab676SJeff Bonwick 	old_ms_count = tvd->vdev_ms_count;
2892b24ab676SJeff Bonwick 	old_class_space = metaslab_class_get_space(mc);
2893fa9e4066Sahrens 
2894fa9e4066Sahrens 	/*
2895573ca77eSGeorge Wilson 	 * Determine the size of the first leaf vdev associated with
2896573ca77eSGeorge Wilson 	 * our top-level device.
2897fa9e4066Sahrens 	 */
2898573ca77eSGeorge Wilson 	vd = vdev_walk_tree(tvd, NULL, NULL);
2899573ca77eSGeorge Wilson 	ASSERT3P(vd, !=, NULL);
2900573ca77eSGeorge Wilson 	ASSERT(vd->vdev_ops->vdev_op_leaf);
2901fa9e4066Sahrens 
2902573ca77eSGeorge Wilson 	psize = vd->vdev_psize;
2903fa9e4066Sahrens 
2904573ca77eSGeorge Wilson 	/*
29058f18d1faSGeorge Wilson 	 * We only try to expand the vdev if it's healthy, less than 4x its
29068f18d1faSGeorge Wilson 	 * original size, and it has a valid psize.
2907573ca77eSGeorge Wilson 	 */
29088f18d1faSGeorge Wilson 	if (tvd->vdev_state != VDEV_STATE_HEALTHY ||
2909420dfc95SChris Siden 	    psize == 0 || psize >= 4 * ztest_opts.zo_vdev_size) {
2910573ca77eSGeorge Wilson 		spa_config_exit(spa, SCL_STATE, spa);
2911420dfc95SChris Siden 		VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
2912573ca77eSGeorge Wilson 		return;
2913573ca77eSGeorge Wilson 	}
2914573ca77eSGeorge Wilson 	ASSERT(psize > 0);
2915573ca77eSGeorge Wilson 	newsize = psize + psize / 8;
2916573ca77eSGeorge Wilson 	ASSERT3U(newsize, >, psize);
2917fa9e4066Sahrens 
2918420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 6) {
2919b24ab676SJeff Bonwick 		(void) printf("Expanding LUN %s from %lu to %lu\n",
2920573ca77eSGeorge Wilson 		    vd->vdev_path, (ulong_t)psize, (ulong_t)newsize);
2921573ca77eSGeorge Wilson 	}
2922573ca77eSGeorge Wilson 
2923573ca77eSGeorge Wilson 	/*
2924573ca77eSGeorge Wilson 	 * Growing the vdev is a two step process:
2925573ca77eSGeorge Wilson 	 *	1). expand the physical size (i.e. relabel)
2926573ca77eSGeorge Wilson 	 *	2). online the vdev to create the new metaslabs
2927573ca77eSGeorge Wilson 	 */
2928573ca77eSGeorge Wilson 	if (vdev_walk_tree(tvd, grow_vdev, &newsize) != NULL ||
2929573ca77eSGeorge Wilson 	    vdev_walk_tree(tvd, online_vdev, NULL) != NULL ||
2930573ca77eSGeorge Wilson 	    tvd->vdev_state != VDEV_STATE_HEALTHY) {
2931420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 5) {
2932b24ab676SJeff Bonwick 			(void) printf("Could not expand LUN because "
2933b24ab676SJeff Bonwick 			    "the vdev configuration changed.\n");
2934b24ab676SJeff Bonwick 		}
2935b24ab676SJeff Bonwick 		spa_config_exit(spa, SCL_STATE, spa);
2936420dfc95SChris Siden 		VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
2937b24ab676SJeff Bonwick 		return;
2938b24ab676SJeff Bonwick 	}
2939b24ab676SJeff Bonwick 
2940b24ab676SJeff Bonwick 	spa_config_exit(spa, SCL_STATE, spa);
2941b24ab676SJeff Bonwick 
2942b24ab676SJeff Bonwick 	/*
2943b24ab676SJeff Bonwick 	 * Expanding the LUN will update the config asynchronously,
2944b24ab676SJeff Bonwick 	 * thus we must wait for the async thread to complete any
2945b24ab676SJeff Bonwick 	 * pending tasks before proceeding.
2946b24ab676SJeff Bonwick 	 */
2947b24ab676SJeff Bonwick 	for (;;) {
2948b24ab676SJeff Bonwick 		boolean_t done;
2949b24ab676SJeff Bonwick 		mutex_enter(&spa->spa_async_lock);
2950b24ab676SJeff Bonwick 		done = (spa->spa_async_thread == NULL && !spa->spa_async_tasks);
2951b24ab676SJeff Bonwick 		mutex_exit(&spa->spa_async_lock);
2952b24ab676SJeff Bonwick 		if (done)
2953b24ab676SJeff Bonwick 			break;
2954b24ab676SJeff Bonwick 		txg_wait_synced(spa_get_dsl(spa), 0);
2955b24ab676SJeff Bonwick 		(void) poll(NULL, 0, 100);
2956b24ab676SJeff Bonwick 	}
2957b24ab676SJeff Bonwick 
2958b24ab676SJeff Bonwick 	spa_config_enter(spa, SCL_STATE, spa, RW_READER);
2959b24ab676SJeff Bonwick 
2960b24ab676SJeff Bonwick 	tvd = spa->spa_root_vdev->vdev_child[top];
2961b24ab676SJeff Bonwick 	new_ms_count = tvd->vdev_ms_count;
2962b24ab676SJeff Bonwick 	new_class_space = metaslab_class_get_space(mc);
2963b24ab676SJeff Bonwick 
2964b24ab676SJeff Bonwick 	if (tvd->vdev_mg != mg || mg->mg_class != mc) {
2965420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 5) {
2966b24ab676SJeff Bonwick 			(void) printf("Could not verify LUN expansion due to "
2967b24ab676SJeff Bonwick 			    "intervening vdev offline or remove.\n");
2968fa9e4066Sahrens 		}
2969b24ab676SJeff Bonwick 		spa_config_exit(spa, SCL_STATE, spa);
2970420dfc95SChris Siden 		VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
2971573ca77eSGeorge Wilson 		return;
2972fa9e4066Sahrens 	}
2973fa9e4066Sahrens 
2974573ca77eSGeorge Wilson 	/*
2975b24ab676SJeff Bonwick 	 * Make sure we were able to grow the vdev.
2976573ca77eSGeorge Wilson 	 */
2977b24ab676SJeff Bonwick 	if (new_ms_count <= old_ms_count)
2978b24ab676SJeff Bonwick 		fatal(0, "LUN expansion failed: ms_count %llu <= %llu\n",
2979b24ab676SJeff Bonwick 		    old_ms_count, new_ms_count);
2980573ca77eSGeorge Wilson 
2981573ca77eSGeorge Wilson 	/*
2982573ca77eSGeorge Wilson 	 * Make sure we were able to grow the pool.
2983573ca77eSGeorge Wilson 	 */
2984b24ab676SJeff Bonwick 	if (new_class_space <= old_class_space)
2985b24ab676SJeff Bonwick 		fatal(0, "LUN expansion failed: class_space %llu <= %llu\n",
2986b24ab676SJeff Bonwick 		    old_class_space, new_class_space);
2987b24ab676SJeff Bonwick 
2988420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 5) {
2989573ca77eSGeorge Wilson 		char oldnumbuf[6], newnumbuf[6];
2990573ca77eSGeorge Wilson 
2991b24ab676SJeff Bonwick 		nicenum(old_class_space, oldnumbuf);
2992b24ab676SJeff Bonwick 		nicenum(new_class_space, newnumbuf);
2993573ca77eSGeorge Wilson 		(void) printf("%s grew from %s to %s\n",
2994573ca77eSGeorge Wilson 		    spa->spa_name, oldnumbuf, newnumbuf);
2995573ca77eSGeorge Wilson 	}
2996b24ab676SJeff Bonwick 
2997573ca77eSGeorge Wilson 	spa_config_exit(spa, SCL_STATE, spa);
2998420dfc95SChris Siden 	VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
2999fa9e4066Sahrens }
3000fa9e4066Sahrens 
3001b24ab676SJeff Bonwick /*
3002b24ab676SJeff Bonwick  * Verify that dmu_objset_{create,destroy,open,close} work as expected.
3003b24ab676SJeff Bonwick  */
3004fa9e4066Sahrens /* ARGSUSED */
3005fa9e4066Sahrens static void
3006b24ab676SJeff Bonwick ztest_objset_create_cb(objset_t *os, void *arg, cred_t *cr, dmu_tx_t *tx)
3007fa9e4066Sahrens {
3008fa9e4066Sahrens 	/*
3009b24ab676SJeff Bonwick 	 * Create the objects common to all ztest datasets.
3010fa9e4066Sahrens 	 */
3011b24ab676SJeff Bonwick 	VERIFY(zap_create_claim(os, ZTEST_DIROBJ,
3012fa9e4066Sahrens 	    DMU_OT_ZAP_OTHER, DMU_OT_NONE, 0, tx) == 0);
3013fa9e4066Sahrens }
3014fa9e4066Sahrens 
301555da60b9SMark J Musante static int
301655da60b9SMark J Musante ztest_dataset_create(char *dsname)
301755da60b9SMark J Musante {
301855da60b9SMark J Musante 	uint64_t zilset = ztest_random(100);
301955da60b9SMark J Musante 	int err = dmu_objset_create(dsname, DMU_OST_OTHER, 0,
302055da60b9SMark J Musante 	    ztest_objset_create_cb, NULL);
302155da60b9SMark J Musante 
302255da60b9SMark J Musante 	if (err || zilset < 80)
302355da60b9SMark J Musante 		return (err);
302455da60b9SMark J Musante 
3025420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 6)
3026420dfc95SChris Siden 		(void) printf("Setting dataset %s to sync always\n", dsname);
302755da60b9SMark J Musante 	return (ztest_dsl_prop_set_uint64(dsname, ZFS_PROP_SYNC,
302855da60b9SMark J Musante 	    ZFS_SYNC_ALWAYS, B_FALSE));
302955da60b9SMark J Musante }
303055da60b9SMark J Musante 
3031b24ab676SJeff Bonwick /* ARGSUSED */
30321d452cf5Sahrens static int
3033fd136879SMatthew Ahrens ztest_objset_destroy_cb(const char *name, void *arg)
3034fa9e4066Sahrens {
3035fa9e4066Sahrens 	objset_t *os;
3036b24ab676SJeff Bonwick 	dmu_object_info_t doi;
3037fa9e4066Sahrens 	int error;
3038fa9e4066Sahrens 
3039fa9e4066Sahrens 	/*
3040fa9e4066Sahrens 	 * Verify that the dataset contains a directory object.
3041fa9e4066Sahrens 	 */
3042b420f3adSRichard Lowe 	VERIFY3U(0, ==, dmu_objset_hold(name, FTAG, &os));
3043b24ab676SJeff Bonwick 	error = dmu_object_info(os, ZTEST_DIROBJ, &doi);
3044e1930233Sbonwick 	if (error != ENOENT) {
3045e1930233Sbonwick 		/* We could have crashed in the middle of destroying it */
3046*fb09f5aaSMadhav Suresh 		ASSERT0(error);
3047b24ab676SJeff Bonwick 		ASSERT3U(doi.doi_type, ==, DMU_OT_ZAP_OTHER);
3048b24ab676SJeff Bonwick 		ASSERT3S(doi.doi_physical_blocks_512, >=, 0);
3049e1930233Sbonwick 	}
3050503ad85cSMatthew Ahrens 	dmu_objset_rele(os, FTAG);
3051fa9e4066Sahrens 
3052fa9e4066Sahrens 	/*
3053fa9e4066Sahrens 	 * Destroy the dataset.
3054fa9e4066Sahrens 	 */
3055b420f3adSRichard Lowe 	VERIFY3U(0, ==, dmu_objset_destroy(name, B_FALSE));
30561d452cf5Sahrens 	return (0);
3057fa9e4066Sahrens }
3058fa9e4066Sahrens 
3059b24ab676SJeff Bonwick static boolean_t
3060b24ab676SJeff Bonwick ztest_snapshot_create(char *osname, uint64_t id)
3061fa9e4066Sahrens {
3062b24ab676SJeff Bonwick 	char snapname[MAXNAMELEN];
3063b24ab676SJeff Bonwick 	int error;
3064b24ab676SJeff Bonwick 
3065b24ab676SJeff Bonwick 	(void) snprintf(snapname, MAXNAMELEN, "%s@%llu", osname,
3066b24ab676SJeff Bonwick 	    (u_longlong_t)id);
3067b24ab676SJeff Bonwick 
30684445fffbSMatthew Ahrens 	error = dmu_objset_snapshot_one(osname, strchr(snapname, '@') + 1);
3069b24ab676SJeff Bonwick 	if (error == ENOSPC) {
3070b24ab676SJeff Bonwick 		ztest_record_enospc(FTAG);
3071b24ab676SJeff Bonwick 		return (B_FALSE);
3072b24ab676SJeff Bonwick 	}
3073b24ab676SJeff Bonwick 	if (error != 0 && error != EEXIST)
3074b24ab676SJeff Bonwick 		fatal(0, "ztest_snapshot_create(%s) = %d", snapname, error);
3075b24ab676SJeff Bonwick 	return (B_TRUE);
3076b24ab676SJeff Bonwick }
3077b24ab676SJeff Bonwick 
3078b24ab676SJeff Bonwick static boolean_t
3079b24ab676SJeff Bonwick ztest_snapshot_destroy(char *osname, uint64_t id)
3080b24ab676SJeff Bonwick {
3081b24ab676SJeff Bonwick 	char snapname[MAXNAMELEN];
3082b24ab676SJeff Bonwick 	int error;
3083b24ab676SJeff Bonwick 
3084b24ab676SJeff Bonwick 	(void) snprintf(snapname, MAXNAMELEN, "%s@%llu", osname,
3085b24ab676SJeff Bonwick 	    (u_longlong_t)id);
3086b24ab676SJeff Bonwick 
3087b24ab676SJeff Bonwick 	error = dmu_objset_destroy(snapname, B_FALSE);
3088b24ab676SJeff Bonwick 	if (error != 0 && error != ENOENT)
3089b24ab676SJeff Bonwick 		fatal(0, "ztest_snapshot_destroy(%s) = %d", snapname, error);
3090b24ab676SJeff Bonwick 	return (B_TRUE);
3091fa9e4066Sahrens }
3092fa9e4066Sahrens 
3093b24ab676SJeff Bonwick /* ARGSUSED */
3094fa9e4066Sahrens void
3095b24ab676SJeff Bonwick ztest_dmu_objset_create_destroy(ztest_ds_t *zd, uint64_t id)
3096fa9e4066Sahrens {
3097b24ab676SJeff Bonwick 	ztest_ds_t zdtmp;
3098b24ab676SJeff Bonwick 	int iters;
3099fa9e4066Sahrens 	int error;
3100745cd3c5Smaybee 	objset_t *os, *os2;
3101b24ab676SJeff Bonwick 	char name[MAXNAMELEN];
3102fa9e4066Sahrens 	zilog_t *zilog;
3103fa9e4066Sahrens 
3104420dfc95SChris Siden 	(void) rw_rdlock(&ztest_name_lock);
3105b24ab676SJeff Bonwick 
3106b24ab676SJeff Bonwick 	(void) snprintf(name, MAXNAMELEN, "%s/temp_%llu",
3107420dfc95SChris Siden 	    ztest_opts.zo_pool, (u_longlong_t)id);
3108fa9e4066Sahrens 
3109fa9e4066Sahrens 	/*
3110fa9e4066Sahrens 	 * If this dataset exists from a previous run, process its replay log
3111fa9e4066Sahrens 	 * half of the time.  If we don't replay it, then dmu_objset_destroy()
3112b24ab676SJeff Bonwick 	 * (invoked from ztest_objset_destroy_cb()) should just throw it away.
3113fa9e4066Sahrens 	 */
3114fa9e4066Sahrens 	if (ztest_random(2) == 0 &&
3115503ad85cSMatthew Ahrens 	    dmu_objset_own(name, DMU_OST_OTHER, B_FALSE, FTAG, &os) == 0) {
3116420dfc95SChris Siden 		ztest_zd_init(&zdtmp, NULL, os);
3117b24ab676SJeff Bonwick 		zil_replay(os, &zdtmp, ztest_replay_vector);
3118b24ab676SJeff Bonwick 		ztest_zd_fini(&zdtmp);
3119503ad85cSMatthew Ahrens 		dmu_objset_disown(os, FTAG);
3120fa9e4066Sahrens 	}
3121fa9e4066Sahrens 
3122fa9e4066Sahrens 	/*
3123fa9e4066Sahrens 	 * There may be an old instance of the dataset we're about to
3124fa9e4066Sahrens 	 * create lying around from a previous run.  If so, destroy it
3125fa9e4066Sahrens 	 * and all of its snapshots.
3126fa9e4066Sahrens 	 */
3127b24ab676SJeff Bonwick 	(void) dmu_objset_find(name, ztest_objset_destroy_cb, NULL,
31280b69c2f0Sahrens 	    DS_FIND_CHILDREN | DS_FIND_SNAPSHOTS);
3129fa9e4066Sahrens 
3130fa9e4066Sahrens 	/*
3131fa9e4066Sahrens 	 * Verify that the destroyed dataset is no longer in the namespace.
3132fa9e4066Sahrens 	 */
3133b24ab676SJeff Bonwick 	VERIFY3U(ENOENT, ==, dmu_objset_hold(name, FTAG, &os));
3134fa9e4066Sahrens 
3135fa9e4066Sahrens 	/*
3136fa9e4066Sahrens 	 * Verify that we can create a new dataset.
3137fa9e4066Sahrens 	 */
313855da60b9SMark J Musante 	error = ztest_dataset_create(name);
3139fa9e4066Sahrens 	if (error) {
3140fa9e4066Sahrens 		if (error == ENOSPC) {
3141b24ab676SJeff Bonwick 			ztest_record_enospc(FTAG);
3142420dfc95SChris Siden 			(void) rw_unlock(&ztest_name_lock);
3143fa9e4066Sahrens 			return;
3144fa9e4066Sahrens 		}
3145fa9e4066Sahrens 		fatal(0, "dmu_objset_create(%s) = %d", name, error);
3146fa9e4066Sahrens 	}
3147fa9e4066Sahrens 
3148b420f3adSRichard Lowe 	VERIFY3U(0, ==,
3149b420f3adSRichard Lowe 	    dmu_objset_own(name, DMU_OST_OTHER, B_FALSE, FTAG, &os));
3150b24ab676SJeff Bonwick 
3151420dfc95SChris Siden 	ztest_zd_init(&zdtmp, NULL, os);
3152fa9e4066Sahrens 
3153fa9e4066Sahrens 	/*
3154fa9e4066Sahrens 	 * Open the intent log for it.
3155fa9e4066Sahrens 	 */
3156b24ab676SJeff Bonwick 	zilog = zil_open(os, ztest_get_data);
3157fa9e4066Sahrens 
3158fa9e4066Sahrens 	/*
3159b24ab676SJeff Bonwick 	 * Put some objects in there, do a little I/O to them,
3160b24ab676SJeff Bonwick 	 * and randomly take a couple of snapshots along the way.
3161fa9e4066Sahrens 	 */
3162b24ab676SJeff Bonwick 	iters = ztest_random(5);
3163b24ab676SJeff Bonwick 	for (int i = 0; i < iters; i++) {
3164b24ab676SJeff Bonwick 		ztest_dmu_object_alloc_free(&zdtmp, id);
3165b24ab676SJeff Bonwick 		if (ztest_random(iters) == 0)
3166b24ab676SJeff Bonwick 			(void) ztest_snapshot_create(name, i);
3167fa9e4066Sahrens 	}
3168fa9e4066Sahrens 
3169fa9e4066Sahrens 	/*
3170fa9e4066Sahrens 	 * Verify that we cannot create an existing dataset.
3171fa9e4066Sahrens 	 */
3172b24ab676SJeff Bonwick 	VERIFY3U(EEXIST, ==,
3173b24ab676SJeff Bonwick 	    dmu_objset_create(name, DMU_OST_OTHER, 0, NULL, NULL));
3174fa9e4066Sahrens 
3175fa9e4066Sahrens 	/*
3176503ad85cSMatthew Ahrens 	 * Verify that we can hold an objset that is also owned.
3177745cd3c5Smaybee 	 */
3178b420f3adSRichard Lowe 	VERIFY3U(0, ==, dmu_objset_hold(name, FTAG, &os2));
3179503ad85cSMatthew Ahrens 	dmu_objset_rele(os2, FTAG);
3180503ad85cSMatthew Ahrens 
3181503ad85cSMatthew Ahrens 	/*
3182b24ab676SJeff Bonwick 	 * Verify that we cannot own an objset that is already owned.
3183503ad85cSMatthew Ahrens 	 */
3184b24ab676SJeff Bonwick 	VERIFY3U(EBUSY, ==,
3185b24ab676SJeff Bonwick 	    dmu_objset_own(name, DMU_OST_OTHER, B_FALSE, FTAG, &os2));
3186fa9e4066Sahrens 
3187fa9e4066Sahrens 	zil_close(zilog);
3188503ad85cSMatthew Ahrens 	dmu_objset_disown(os, FTAG);
3189b24ab676SJeff Bonwick 	ztest_zd_fini(&zdtmp);
3190fa9e4066Sahrens 
3191420dfc95SChris Siden 	(void) rw_unlock(&ztest_name_lock);
3192fa9e4066Sahrens }
3193fa9e4066Sahrens 
3194fa9e4066Sahrens /*
3195fa9e4066Sahrens  * Verify that dmu_snapshot_{create,destroy,open,close} work as expected.
3196fa9e4066Sahrens  */
3197fa9e4066Sahrens void
3198b24ab676SJeff Bonwick ztest_dmu_snapshot_create_destroy(ztest_ds_t *zd, uint64_t id)
3199fa9e4066Sahrens {
3200420dfc95SChris Siden 	(void) rw_rdlock(&ztest_name_lock);
3201b24ab676SJeff Bonwick 	(void) ztest_snapshot_destroy(zd->zd_name, id);
3202b24ab676SJeff Bonwick 	(void) ztest_snapshot_create(zd->zd_name, id);
3203420dfc95SChris Siden 	(void) rw_unlock(&ztest_name_lock);
32044f5064b7SMark J Musante }
32054f5064b7SMark J Musante 
3206718d718aSGeorge Wilson /*
3207718d718aSGeorge Wilson  * Cleanup non-standard snapshots and clones.
3208718d718aSGeorge Wilson  */
3209718d718aSGeorge Wilson void
3210b24ab676SJeff Bonwick ztest_dsl_dataset_cleanup(char *osname, uint64_t id)
3211718d718aSGeorge Wilson {
3212b24ab676SJeff Bonwick 	char snap1name[MAXNAMELEN];
3213b24ab676SJeff Bonwick 	char clone1name[MAXNAMELEN];
3214b24ab676SJeff Bonwick 	char snap2name[MAXNAMELEN];
3215b24ab676SJeff Bonwick 	char clone2name[MAXNAMELEN];
3216b24ab676SJeff Bonwick 	char snap3name[MAXNAMELEN];
3217718d718aSGeorge Wilson 	int error;
3218718d718aSGeorge Wilson 
3219b24ab676SJeff Bonwick 	(void) snprintf(snap1name, MAXNAMELEN, "%s@s1_%llu", osname, id);
3220b24ab676SJeff Bonwick 	(void) snprintf(clone1name, MAXNAMELEN, "%s/c1_%llu", osname, id);
3221b24ab676SJeff Bonwick 	(void) snprintf(snap2name, MAXNAMELEN, "%s@s2_%llu", clone1name, id);
3222b24ab676SJeff Bonwick 	(void) snprintf(clone2name, MAXNAMELEN, "%s/c2_%llu", osname, id);
3223b24ab676SJeff Bonwick 	(void) snprintf(snap3name, MAXNAMELEN, "%s@s3_%llu", clone1name, id);
3224718d718aSGeorge Wilson 
3225842727c2SChris Kirby 	error = dmu_objset_destroy(clone2name, B_FALSE);
3226718d718aSGeorge Wilson 	if (error && error != ENOENT)
3227718d718aSGeorge Wilson 		fatal(0, "dmu_objset_destroy(%s) = %d", clone2name, error);
3228842727c2SChris Kirby 	error = dmu_objset_destroy(snap3name, B_FALSE);
3229718d718aSGeorge Wilson 	if (error && error != ENOENT)
3230718d718aSGeorge Wilson 		fatal(0, "dmu_objset_destroy(%s) = %d", snap3name, error);
3231842727c2SChris Kirby 	error = dmu_objset_destroy(snap2name, B_FALSE);
3232718d718aSGeorge Wilson 	if (error && error != ENOENT)
3233718d718aSGeorge Wilson 		fatal(0, "dmu_objset_destroy(%s) = %d", snap2name, error);
3234842727c2SChris Kirby 	error = dmu_objset_destroy(clone1name, B_FALSE);
3235718d718aSGeorge Wilson 	if (error && error != ENOENT)
3236718d718aSGeorge Wilson 		fatal(0, "dmu_objset_destroy(%s) = %d", clone1name, error);
3237842727c2SChris Kirby 	error = dmu_objset_destroy(snap1name, B_FALSE);
3238718d718aSGeorge Wilson 	if (error && error != ENOENT)
3239718d718aSGeorge Wilson 		fatal(0, "dmu_objset_destroy(%s) = %d", snap1name, error);
3240718d718aSGeorge Wilson }
3241718d718aSGeorge Wilson 
32424f5064b7SMark J Musante /*
32434f5064b7SMark J Musante  * Verify dsl_dataset_promote handles EBUSY
32444f5064b7SMark J Musante  */
32454f5064b7SMark J Musante void
3246b24ab676SJeff Bonwick ztest_dsl_dataset_promote_busy(ztest_ds_t *zd, uint64_t id)
32474f5064b7SMark J Musante {
32484f5064b7SMark J Musante 	objset_t *clone;
32494f5064b7SMark J Musante 	dsl_dataset_t *ds;
3250b24ab676SJeff Bonwick 	char snap1name[MAXNAMELEN];
3251b24ab676SJeff Bonwick 	char clone1name[MAXNAMELEN];
3252b24ab676SJeff Bonwick 	char snap2name[MAXNAMELEN];
3253b24ab676SJeff Bonwick 	char clone2name[MAXNAMELEN];
3254b24ab676SJeff Bonwick 	char snap3name[MAXNAMELEN];
3255b24ab676SJeff Bonwick 	char *osname = zd->zd_name;
3256b24ab676SJeff Bonwick 	int error;
32574f5064b7SMark J Musante 
3258420dfc95SChris Siden 	(void) rw_rdlock(&ztest_name_lock);
32594f5064b7SMark J Musante 
3260b24ab676SJeff Bonwick 	ztest_dsl_dataset_cleanup(osname, id);
3261718d718aSGeorge Wilson 
3262b24ab676SJeff Bonwick 	(void) snprintf(snap1name, MAXNAMELEN, "%s@s1_%llu", osname, id);
3263b24ab676SJeff Bonwick 	(void) snprintf(clone1name, MAXNAMELEN, "%s/c1_%llu", osname, id);
3264b24ab676SJeff Bonwick 	(void) snprintf(snap2name, MAXNAMELEN, "%s@s2_%llu", clone1name, id);
3265b24ab676SJeff Bonwick 	(void) snprintf(clone2name, MAXNAMELEN, "%s/c2_%llu", osname, id);
3266b24ab676SJeff Bonwick 	(void) snprintf(snap3name, MAXNAMELEN, "%s@s3_%llu", clone1name, id);
326705312e2cSMark J Musante 
32684445fffbSMatthew Ahrens 	error = dmu_objset_snapshot_one(osname, strchr(snap1name, '@') + 1);
3269b2d22b66SGeorge Wilson 	if (error && error != EEXIST) {
3270b2d22b66SGeorge Wilson 		if (error == ENOSPC) {
3271b24ab676SJeff Bonwick 			ztest_record_enospc(FTAG);
3272b2d22b66SGeorge Wilson 			goto out;
3273b2d22b66SGeorge Wilson 		}
3274b2d22b66SGeorge Wilson 		fatal(0, "dmu_take_snapshot(%s) = %d", snap1name, error);
3275b2d22b66SGeorge Wilson 	}
32764f5064b7SMark J Musante 
3277503ad85cSMatthew Ahrens 	error = dmu_objset_hold(snap1name, FTAG, &clone);
32784f5064b7SMark J Musante 	if (error)
32794f5064b7SMark J Musante 		fatal(0, "dmu_open_snapshot(%s) = %d", snap1name, error);
32804f5064b7SMark J Musante 
3281ae46e4c7SMatthew Ahrens 	error = dmu_objset_clone(clone1name, dmu_objset_ds(clone), 0);
3282503ad85cSMatthew Ahrens 	dmu_objset_rele(clone, FTAG);
3283b2d22b66SGeorge Wilson 	if (error) {
3284b2d22b66SGeorge Wilson 		if (error == ENOSPC) {
3285b24ab676SJeff Bonwick 			ztest_record_enospc(FTAG);
3286b2d22b66SGeorge Wilson 			goto out;
3287b2d22b66SGeorge Wilson 		}
3288b2d22b66SGeorge Wilson 		fatal(0, "dmu_objset_create(%s) = %d", clone1name, error);
3289b2d22b66SGeorge Wilson 	}
32904f5064b7SMark J Musante 
32914445fffbSMatthew Ahrens 	error = dmu_objset_snapshot_one(clone1name, strchr(snap2name, '@') + 1);
3292b2d22b66SGeorge Wilson 	if (error && error != EEXIST) {
3293b2d22b66SGeorge Wilson 		if (error == ENOSPC) {
3294b24ab676SJeff Bonwick 			ztest_record_enospc(FTAG);
3295b2d22b66SGeorge Wilson 			goto out;
3296b2d22b66SGeorge Wilson 		}
3297b2d22b66SGeorge Wilson 		fatal(0, "dmu_open_snapshot(%s) = %d", snap2name, error);
3298b2d22b66SGeorge Wilson 	}
32994f5064b7SMark J Musante 
33004445fffbSMatthew Ahrens 	error = dmu_objset_snapshot_one(clone1name, strchr(snap3name, '@') + 1);
3301b2d22b66SGeorge Wilson 	if (error && error != EEXIST) {
3302b2d22b66SGeorge Wilson 		if (error == ENOSPC) {
3303b24ab676SJeff Bonwick 			ztest_record_enospc(FTAG);
3304b2d22b66SGeorge Wilson 			goto out;
3305b2d22b66SGeorge Wilson 		}
3306b2d22b66SGeorge Wilson 		fatal(0, "dmu_open_snapshot(%s) = %d", snap3name, error);
3307b2d22b66SGeorge Wilson 	}
33084f5064b7SMark J Musante 
3309503ad85cSMatthew Ahrens 	error = dmu_objset_hold(snap3name, FTAG, &clone);
33104f5064b7SMark J Musante 	if (error)
33114f5064b7SMark J Musante 		fatal(0, "dmu_open_snapshot(%s) = %d", snap3name, error);
33124f5064b7SMark J Musante 
3313ae46e4c7SMatthew Ahrens 	error = dmu_objset_clone(clone2name, dmu_objset_ds(clone), 0);
3314503ad85cSMatthew Ahrens 	dmu_objset_rele(clone, FTAG);
3315b2d22b66SGeorge Wilson 	if (error) {
3316b2d22b66SGeorge Wilson 		if (error == ENOSPC) {
3317b24ab676SJeff Bonwick 			ztest_record_enospc(FTAG);
3318b2d22b66SGeorge Wilson 			goto out;
3319b2d22b66SGeorge Wilson 		}
3320b2d22b66SGeorge Wilson 		fatal(0, "dmu_objset_create(%s) = %d", clone2name, error);
3321b2d22b66SGeorge Wilson 	}
33224f5064b7SMark J Musante 
3323cde58dbcSMatthew Ahrens 	error = dsl_dataset_own(snap2name, B_FALSE, FTAG, &ds);
33244f5064b7SMark J Musante 	if (error)
3325cde58dbcSMatthew Ahrens 		fatal(0, "dsl_dataset_own(%s) = %d", snap2name, error);
3326681d9761SEric Taylor 	error = dsl_dataset_promote(clone2name, NULL);
33274f5064b7SMark J Musante 	if (error != EBUSY)
33284f5064b7SMark J Musante 		fatal(0, "dsl_dataset_promote(%s), %d, not EBUSY", clone2name,
33294f5064b7SMark J Musante 		    error);
33304f5064b7SMark J Musante 	dsl_dataset_disown(ds, FTAG);
33314f5064b7SMark J Musante 
3332b2d22b66SGeorge Wilson out:
3333b24ab676SJeff Bonwick 	ztest_dsl_dataset_cleanup(osname, id);
33344f5064b7SMark J Musante 
3335420dfc95SChris Siden 	(void) rw_unlock(&ztest_name_lock);
3336fa9e4066Sahrens }
3337fa9e4066Sahrens 
3338fa9e4066Sahrens /*
3339fa9e4066Sahrens  * Verify that dmu_object_{alloc,free} work as expected.
3340fa9e4066Sahrens  */
3341fa9e4066Sahrens void
3342b24ab676SJeff Bonwick ztest_dmu_object_alloc_free(ztest_ds_t *zd, uint64_t id)
3343fa9e4066Sahrens {
3344b24ab676SJeff Bonwick 	ztest_od_t od[4];
3345b24ab676SJeff Bonwick 	int batchsize = sizeof (od) / sizeof (od[0]);
3346fa9e4066Sahrens 
3347b24ab676SJeff Bonwick 	for (int b = 0; b < batchsize; b++)
3348b24ab676SJeff Bonwick 		ztest_od_init(&od[b], id, FTAG, b, DMU_OT_UINT64_OTHER, 0, 0);
3349fa9e4066Sahrens 
3350fa9e4066Sahrens 	/*
3351b24ab676SJeff Bonwick 	 * Destroy the previous batch of objects, create a new batch,
3352b24ab676SJeff Bonwick 	 * and do some I/O on the new objects.
3353fa9e4066Sahrens 	 */
3354b24ab676SJeff Bonwick 	if (ztest_object_init(zd, od, sizeof (od), B_TRUE) != 0)
3355b24ab676SJeff Bonwick 		return;
3356fa9e4066Sahrens 
3357b24ab676SJeff Bonwick 	while (ztest_random(4 * batchsize) != 0)
3358b24ab676SJeff Bonwick 		ztest_io(zd, od[ztest_random(batchsize)].od_object,
3359b24ab676SJeff Bonwick 		    ztest_random(ZTEST_RANGE_LOCKS) << SPA_MAXBLOCKSHIFT);
3360fa9e4066Sahrens }
3361fa9e4066Sahrens 
3362fa9e4066Sahrens /*
3363fa9e4066Sahrens  * Verify that dmu_{read,write} work as expected.
3364fa9e4066Sahrens  */
3365fa9e4066Sahrens void
3366b24ab676SJeff Bonwick ztest_dmu_read_write(ztest_ds_t *zd, uint64_t id)
3367fa9e4066Sahrens {
3368b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
3369b24ab676SJeff Bonwick 	ztest_od_t od[2];
3370fa9e4066Sahrens 	dmu_tx_t *tx;
3371fa9e4066Sahrens 	int i, freeit, error;
3372fa9e4066Sahrens 	uint64_t n, s, txg;
3373fa9e4066Sahrens 	bufwad_t *packbuf, *bigbuf, *pack, *bigH, *bigT;
3374b24ab676SJeff Bonwick 	uint64_t packobj, packoff, packsize, bigobj, bigoff, bigsize;
3375b24ab676SJeff Bonwick 	uint64_t chunksize = (1000 + ztest_random(1000)) * sizeof (uint64_t);
3376fa9e4066Sahrens 	uint64_t regions = 997;
3377fa9e4066Sahrens 	uint64_t stride = 123456789ULL;
3378fa9e4066Sahrens 	uint64_t width = 40;
3379fa9e4066Sahrens 	int free_percent = 5;
3380fa9e4066Sahrens 
3381fa9e4066Sahrens 	/*
3382fa9e4066Sahrens 	 * This test uses two objects, packobj and bigobj, that are always
3383fa9e4066Sahrens 	 * updated together (i.e. in the same tx) so that their contents are
3384fa9e4066Sahrens 	 * in sync and can be compared.  Their contents relate to each other
3385fa9e4066Sahrens 	 * in a simple way: packobj is a dense array of 'bufwad' structures,
3386fa9e4066Sahrens 	 * while bigobj is a sparse array of the same bufwads.  Specifically,
3387fa9e4066Sahrens 	 * for any index n, there are three bufwads that should be identical:
3388fa9e4066Sahrens 	 *
3389fa9e4066Sahrens 	 *	packobj, at offset n * sizeof (bufwad_t)
3390fa9e4066Sahrens 	 *	bigobj, at the head of the nth chunk
3391fa9e4066Sahrens 	 *	bigobj, at the tail of the nth chunk
3392fa9e4066Sahrens 	 *
3393fa9e4066Sahrens 	 * The chunk size is arbitrary. It doesn't have to be a power of two,
3394fa9e4066Sahrens 	 * and it doesn't have any relation to the object blocksize.
3395fa9e4066Sahrens 	 * The only requirement is that it can hold at least two bufwads.
3396fa9e4066Sahrens 	 *
3397fa9e4066Sahrens 	 * Normally, we write the bufwad to each of these locations.
3398fa9e4066Sahrens 	 * However, free_percent of the time we instead write zeroes to
3399fa9e4066Sahrens 	 * packobj and perform a dmu_free_range() on bigobj.  By comparing
3400fa9e4066Sahrens 	 * bigobj to packobj, we can verify that the DMU is correctly
3401fa9e4066Sahrens 	 * tracking which parts of an object are allocated and free,
3402fa9e4066Sahrens 	 * and that the contents of the allocated blocks are correct.
3403fa9e4066Sahrens 	 */
3404fa9e4066Sahrens 
3405fa9e4066Sahrens 	/*
3406fa9e4066Sahrens 	 * Read the directory info.  If it's the first time, set things up.
3407fa9e4066Sahrens 	 */
3408b24ab676SJeff Bonwick 	ztest_od_init(&od[0], id, FTAG, 0, DMU_OT_UINT64_OTHER, 0, chunksize);
3409b24ab676SJeff Bonwick 	ztest_od_init(&od[1], id, FTAG, 1, DMU_OT_UINT64_OTHER, 0, chunksize);
3410fa9e4066Sahrens 
3411b24ab676SJeff Bonwick 	if (ztest_object_init(zd, od, sizeof (od), B_FALSE) != 0)
3412b24ab676SJeff Bonwick 		return;
3413fa9e4066Sahrens 
3414b24ab676SJeff Bonwick 	bigobj = od[0].od_object;
3415b24ab676SJeff Bonwick 	packobj = od[1].od_object;
3416b24ab676SJeff Bonwick 	chunksize = od[0].od_gen;
3417b24ab676SJeff Bonwick 	ASSERT(chunksize == od[1].od_gen);
3418fa9e4066Sahrens 
3419fa9e4066Sahrens 	/*
3420fa9e4066Sahrens 	 * Prefetch a random chunk of the big object.
3421fa9e4066Sahrens 	 * Our aim here is to get some async reads in flight
3422fa9e4066Sahrens 	 * for blocks that we may free below; the DMU should
3423fa9e4066Sahrens 	 * handle this race correctly.
3424fa9e4066Sahrens 	 */
3425fa9e4066Sahrens 	n = ztest_random(regions) * stride + ztest_random(width);
3426fa9e4066Sahrens 	s = 1 + ztest_random(2 * width - 1);
3427b24ab676SJeff Bonwick 	dmu_prefetch(os, bigobj, n * chunksize, s * chunksize);
3428fa9e4066Sahrens 
3429fa9e4066Sahrens 	/*
3430fa9e4066Sahrens 	 * Pick a random index and compute the offsets into packobj and bigobj.
3431fa9e4066Sahrens 	 */
3432fa9e4066Sahrens 	n = ztest_random(regions) * stride + ztest_random(width);
3433fa9e4066Sahrens 	s = 1 + ztest_random(width - 1);
3434fa9e4066Sahrens 
3435fa9e4066Sahrens 	packoff = n * sizeof (bufwad_t);
3436fa9e4066Sahrens 	packsize = s * sizeof (bufwad_t);
3437fa9e4066Sahrens 
3438b24ab676SJeff Bonwick 	bigoff = n * chunksize;
3439b24ab676SJeff Bonwick 	bigsize = s * chunksize;
3440fa9e4066Sahrens 
3441fa9e4066Sahrens 	packbuf = umem_alloc(packsize, UMEM_NOFAIL);
3442fa9e4066Sahrens 	bigbuf = umem_alloc(bigsize, UMEM_NOFAIL);
3443fa9e4066Sahrens 
3444fa9e4066Sahrens 	/*
3445fa9e4066Sahrens 	 * free_percent of the time, free a range of bigobj rather than
3446fa9e4066Sahrens 	 * overwriting it.
3447fa9e4066Sahrens 	 */
3448fa9e4066Sahrens 	freeit = (ztest_random(100) < free_percent);
3449fa9e4066Sahrens 
3450fa9e4066Sahrens 	/*
3451fa9e4066Sahrens 	 * Read the current contents of our objects.
3452fa9e4066Sahrens 	 */
3453b24ab676SJeff Bonwick 	error = dmu_read(os, packobj, packoff, packsize, packbuf,
34547bfdf011SNeil Perrin 	    DMU_READ_PREFETCH);
3455*fb09f5aaSMadhav Suresh 	ASSERT0(error);
3456b24ab676SJeff Bonwick 	error = dmu_read(os, bigobj, bigoff, bigsize, bigbuf,
34577bfdf011SNeil Perrin 	    DMU_READ_PREFETCH);
3458*fb09f5aaSMadhav Suresh 	ASSERT0(error);
3459fa9e4066Sahrens 
3460fa9e4066Sahrens 	/*
3461fa9e4066Sahrens 	 * Get a tx for the mods to both packobj and bigobj.
3462fa9e4066Sahrens 	 */
3463fa9e4066Sahrens 	tx = dmu_tx_create(os);
3464fa9e4066Sahrens 
3465b24ab676SJeff Bonwick 	dmu_tx_hold_write(tx, packobj, packoff, packsize);
3466fa9e4066Sahrens 
3467fa9e4066Sahrens 	if (freeit)
3468b24ab676SJeff Bonwick 		dmu_tx_hold_free(tx, bigobj, bigoff, bigsize);
3469fa9e4066Sahrens 	else
3470b24ab676SJeff Bonwick 		dmu_tx_hold_write(tx, bigobj, bigoff, bigsize);
3471fa9e4066Sahrens 
3472b24ab676SJeff Bonwick 	txg = ztest_tx_assign(tx, TXG_MIGHTWAIT, FTAG);
3473b24ab676SJeff Bonwick 	if (txg == 0) {
3474fa9e4066Sahrens 		umem_free(packbuf, packsize);
3475fa9e4066Sahrens 		umem_free(bigbuf, bigsize);
3476fa9e4066Sahrens 		return;
3477fa9e4066Sahrens 	}
3478fa9e4066Sahrens 
3479b24ab676SJeff Bonwick 	dmu_object_set_checksum(os, bigobj,
3480b24ab676SJeff Bonwick 	    (enum zio_checksum)ztest_random_dsl_prop(ZFS_PROP_CHECKSUM), tx);
3481b24ab676SJeff Bonwick 
3482b24ab676SJeff Bonwick 	dmu_object_set_compress(os, bigobj,
3483b24ab676SJeff Bonwick 	    (enum zio_compress)ztest_random_dsl_prop(ZFS_PROP_COMPRESSION), tx);
3484fa9e4066Sahrens 
3485fa9e4066Sahrens 	/*
3486fa9e4066Sahrens 	 * For each index from n to n + s, verify that the existing bufwad
3487fa9e4066Sahrens 	 * in packobj matches the bufwads at the head and tail of the
3488fa9e4066Sahrens 	 * corresponding chunk in bigobj.  Then update all three bufwads
3489fa9e4066Sahrens 	 * with the new values we want to write out.
3490fa9e4066Sahrens 	 */
3491fa9e4066Sahrens 	for (i = 0; i < s; i++) {
3492fa9e4066Sahrens 		/* LINTED */
3493fa9e4066Sahrens 		pack = (bufwad_t *)((char *)packbuf + i * sizeof (bufwad_t));
3494fa9e4066Sahrens 		/* LINTED */
3495b24ab676SJeff Bonwick 		bigH = (bufwad_t *)((char *)bigbuf + i * chunksize);
3496fa9e4066Sahrens 		/* LINTED */
3497b24ab676SJeff Bonwick 		bigT = (bufwad_t *)((char *)bigH + chunksize) - 1;
3498fa9e4066Sahrens 
3499fa9e4066Sahrens 		ASSERT((uintptr_t)bigH - (uintptr_t)bigbuf < bigsize);
3500fa9e4066Sahrens 		ASSERT((uintptr_t)bigT - (uintptr_t)bigbuf < bigsize);
3501fa9e4066Sahrens 
3502fa9e4066Sahrens 		if (pack->bw_txg > txg)
3503fa9e4066Sahrens 			fatal(0, "future leak: got %llx, open txg is %llx",
3504fa9e4066Sahrens 			    pack->bw_txg, txg);
3505fa9e4066Sahrens 
3506fa9e4066Sahrens 		if (pack->bw_data != 0 && pack->bw_index != n + i)
3507fa9e4066Sahrens 			fatal(0, "wrong index: got %llx, wanted %llx+%llx",
3508fa9e4066Sahrens 			    pack->bw_index, n, i);
3509fa9e4066Sahrens 
3510fa9e4066Sahrens 		if (bcmp(pack, bigH, sizeof (bufwad_t)) != 0)
3511fa9e4066Sahrens 			fatal(0, "pack/bigH mismatch in %p/%p", pack, bigH);
3512fa9e4066Sahrens 
3513fa9e4066Sahrens 		if (bcmp(pack, bigT, sizeof (bufwad_t)) != 0)
3514fa9e4066Sahrens 			fatal(0, "pack/bigT mismatch in %p/%p", pack, bigT);
3515fa9e4066Sahrens 
3516fa9e4066Sahrens 		if (freeit) {
3517fa9e4066Sahrens 			bzero(pack, sizeof (bufwad_t));
3518fa9e4066Sahrens 		} else {
3519fa9e4066Sahrens 			pack->bw_index = n + i;
3520fa9e4066Sahrens 			pack->bw_txg = txg;
3521fa9e4066Sahrens 			pack->bw_data = 1 + ztest_random(-2ULL);
3522fa9e4066Sahrens 		}
3523fa9e4066Sahrens 		*bigH = *pack;
3524fa9e4066Sahrens 		*bigT = *pack;
3525fa9e4066Sahrens 	}
3526fa9e4066Sahrens 
3527fa9e4066Sahrens 	/*
3528fa9e4066Sahrens 	 * We've verified all the old bufwads, and made new ones.
3529fa9e4066Sahrens 	 * Now write them out.
3530fa9e4066Sahrens 	 */
3531b24ab676SJeff Bonwick 	dmu_write(os, packobj, packoff, packsize, packbuf, tx);
3532fa9e4066Sahrens 
3533fa9e4066Sahrens 	if (freeit) {
3534420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 7) {
3535fa9e4066Sahrens 			(void) printf("freeing offset %llx size %llx"
3536fa9e4066Sahrens 			    " txg %llx\n",
3537fa9e4066Sahrens 			    (u_longlong_t)bigoff,
3538fa9e4066Sahrens 			    (u_longlong_t)bigsize,
3539fa9e4066Sahrens 			    (u_longlong_t)txg);
3540fa9e4066Sahrens 		}
3541b24ab676SJeff Bonwick 		VERIFY(0 == dmu_free_range(os, bigobj, bigoff, bigsize, tx));
3542fa9e4066Sahrens 	} else {
3543420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 7) {
3544fa9e4066Sahrens 			(void) printf("writing offset %llx size %llx"
3545fa9e4066Sahrens 			    " txg %llx\n",
3546fa9e4066Sahrens 			    (u_longlong_t)bigoff,
3547fa9e4066Sahrens 			    (u_longlong_t)bigsize,
3548fa9e4066Sahrens 			    (u_longlong_t)txg);
3549fa9e4066Sahrens 		}
3550b24ab676SJeff Bonwick 		dmu_write(os, bigobj, bigoff, bigsize, bigbuf, tx);
3551fa9e4066Sahrens 	}
3552fa9e4066Sahrens 
3553fa9e4066Sahrens 	dmu_tx_commit(tx);
3554fa9e4066Sahrens 
3555fa9e4066Sahrens 	/*
3556fa9e4066Sahrens 	 * Sanity check the stuff we just wrote.
3557fa9e4066Sahrens 	 */
3558fa9e4066Sahrens 	{
3559fa9e4066Sahrens 		void *packcheck = umem_alloc(packsize, UMEM_NOFAIL);
3560fa9e4066Sahrens 		void *bigcheck = umem_alloc(bigsize, UMEM_NOFAIL);
3561fa9e4066Sahrens 
3562b24ab676SJeff Bonwick 		VERIFY(0 == dmu_read(os, packobj, packoff,
35637bfdf011SNeil Perrin 		    packsize, packcheck, DMU_READ_PREFETCH));
3564b24ab676SJeff Bonwick 		VERIFY(0 == dmu_read(os, bigobj, bigoff,
35657bfdf011SNeil Perrin 		    bigsize, bigcheck, DMU_READ_PREFETCH));
3566fa9e4066Sahrens 
3567fa9e4066Sahrens 		ASSERT(bcmp(packbuf, packcheck, packsize) == 0);
3568fa9e4066Sahrens 		ASSERT(bcmp(bigbuf, bigcheck, bigsize) == 0);
3569fa9e4066Sahrens 
3570fa9e4066Sahrens 		umem_free(packcheck, packsize);
3571fa9e4066Sahrens 		umem_free(bigcheck, bigsize);
3572fa9e4066Sahrens 	}
3573fa9e4066Sahrens 
3574fa9e4066Sahrens 	umem_free(packbuf, packsize);
3575fa9e4066Sahrens 	umem_free(bigbuf, bigsize);
3576fa9e4066Sahrens }
3577fa9e4066Sahrens 
35782fdbea25SAleksandr Guzovskiy void
35792fdbea25SAleksandr Guzovskiy compare_and_update_pbbufs(uint64_t s, bufwad_t *packbuf, bufwad_t *bigbuf,
3580b24ab676SJeff Bonwick     uint64_t bigsize, uint64_t n, uint64_t chunksize, uint64_t txg)
35812fdbea25SAleksandr Guzovskiy {
35822fdbea25SAleksandr Guzovskiy 	uint64_t i;
35832fdbea25SAleksandr Guzovskiy 	bufwad_t *pack;
35842fdbea25SAleksandr Guzovskiy 	bufwad_t *bigH;
35852fdbea25SAleksandr Guzovskiy 	bufwad_t *bigT;
35862fdbea25SAleksandr Guzovskiy 
35872fdbea25SAleksandr Guzovskiy 	/*
35882fdbea25SAleksandr Guzovskiy 	 * For each index from n to n + s, verify that the existing bufwad
35892fdbea25SAleksandr Guzovskiy 	 * in packobj matches the bufwads at the head and tail of the
35902fdbea25SAleksandr Guzovskiy 	 * corresponding chunk in bigobj.  Then update all three bufwads
35912fdbea25SAleksandr Guzovskiy 	 * with the new values we want to write out.
35922fdbea25SAleksandr Guzovskiy 	 */
35932fdbea25SAleksandr Guzovskiy 	for (i = 0; i < s; i++) {
35942fdbea25SAleksandr Guzovskiy 		/* LINTED */
35952fdbea25SAleksandr Guzovskiy 		pack = (bufwad_t *)((char *)packbuf + i * sizeof (bufwad_t));
35962fdbea25SAleksandr Guzovskiy 		/* LINTED */
3597b24ab676SJeff Bonwick 		bigH = (bufwad_t *)((char *)bigbuf + i * chunksize);
35982fdbea25SAleksandr Guzovskiy 		/* LINTED */
3599b24ab676SJeff Bonwick 		bigT = (bufwad_t *)((char *)bigH + chunksize) - 1;
36002fdbea25SAleksandr Guzovskiy 
36012fdbea25SAleksandr Guzovskiy 		ASSERT((uintptr_t)bigH - (uintptr_t)bigbuf < bigsize);
36022fdbea25SAleksandr Guzovskiy 		ASSERT((uintptr_t)bigT - (uintptr_t)bigbuf < bigsize);
36032fdbea25SAleksandr Guzovskiy 
36042fdbea25SAleksandr Guzovskiy 		if (pack->bw_txg > txg)
36052fdbea25SAleksandr Guzovskiy 			fatal(0, "future leak: got %llx, open txg is %llx",
36062fdbea25SAleksandr Guzovskiy 			    pack->bw_txg, txg);
36072fdbea25SAleksandr Guzovskiy 
36082fdbea25SAleksandr Guzovskiy 		if (pack->bw_data != 0 && pack->bw_index != n + i)
36092fdbea25SAleksandr Guzovskiy 			fatal(0, "wrong index: got %llx, wanted %llx+%llx",
36102fdbea25SAleksandr Guzovskiy 			    pack->bw_index, n, i);
36112fdbea25SAleksandr Guzovskiy 
36122fdbea25SAleksandr Guzovskiy 		if (bcmp(pack, bigH, sizeof (bufwad_t)) != 0)
36132fdbea25SAleksandr Guzovskiy 			fatal(0, "pack/bigH mismatch in %p/%p", pack, bigH);
36142fdbea25SAleksandr Guzovskiy 
36152fdbea25SAleksandr Guzovskiy 		if (bcmp(pack, bigT, sizeof (bufwad_t)) != 0)
36162fdbea25SAleksandr Guzovskiy 			fatal(0, "pack/bigT mismatch in %p/%p", pack, bigT);
36172fdbea25SAleksandr Guzovskiy 
36182fdbea25SAleksandr Guzovskiy 		pack->bw_index = n + i;
36192fdbea25SAleksandr Guzovskiy 		pack->bw_txg = txg;
36202fdbea25SAleksandr Guzovskiy 		pack->bw_data = 1 + ztest_random(-2ULL);
36212fdbea25SAleksandr Guzovskiy 
36222fdbea25SAleksandr Guzovskiy 		*bigH = *pack;
36232fdbea25SAleksandr Guzovskiy 		*bigT = *pack;
36242fdbea25SAleksandr Guzovskiy 	}
36252fdbea25SAleksandr Guzovskiy }
36262fdbea25SAleksandr Guzovskiy 
36272fdbea25SAleksandr Guzovskiy void
3628b24ab676SJeff Bonwick ztest_dmu_read_write_zcopy(ztest_ds_t *zd, uint64_t id)
36292fdbea25SAleksandr Guzovskiy {
3630b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
3631b24ab676SJeff Bonwick 	ztest_od_t od[2];
36322fdbea25SAleksandr Guzovskiy 	dmu_tx_t *tx;
36332fdbea25SAleksandr Guzovskiy 	uint64_t i;
36342fdbea25SAleksandr Guzovskiy 	int error;
36352fdbea25SAleksandr Guzovskiy 	uint64_t n, s, txg;
36362fdbea25SAleksandr Guzovskiy 	bufwad_t *packbuf, *bigbuf;
3637b24ab676SJeff Bonwick 	uint64_t packobj, packoff, packsize, bigobj, bigoff, bigsize;
3638b24ab676SJeff Bonwick 	uint64_t blocksize = ztest_random_blocksize();
3639b24ab676SJeff Bonwick 	uint64_t chunksize = blocksize;
36402fdbea25SAleksandr Guzovskiy 	uint64_t regions = 997;
36412fdbea25SAleksandr Guzovskiy 	uint64_t stride = 123456789ULL;
36422fdbea25SAleksandr Guzovskiy 	uint64_t width = 9;
36432fdbea25SAleksandr Guzovskiy 	dmu_buf_t *bonus_db;
36442fdbea25SAleksandr Guzovskiy 	arc_buf_t **bigbuf_arcbufs;
3645b24ab676SJeff Bonwick 	dmu_object_info_t doi;
36462fdbea25SAleksandr Guzovskiy 
36472fdbea25SAleksandr Guzovskiy 	/*
36482fdbea25SAleksandr Guzovskiy 	 * This test uses two objects, packobj and bigobj, that are always
36492fdbea25SAleksandr Guzovskiy 	 * updated together (i.e. in the same tx) so that their contents are
36502fdbea25SAleksandr Guzovskiy 	 * in sync and can be compared.  Their contents relate to each other
36512fdbea25SAleksandr Guzovskiy 	 * in a simple way: packobj is a dense array of 'bufwad' structures,
36522fdbea25SAleksandr Guzovskiy 	 * while bigobj is a sparse array of the same bufwads.  Specifically,
36532fdbea25SAleksandr Guzovskiy 	 * for any index n, there are three bufwads that should be identical:
36542fdbea25SAleksandr Guzovskiy 	 *
36552fdbea25SAleksandr Guzovskiy 	 *	packobj, at offset n * sizeof (bufwad_t)
36562fdbea25SAleksandr Guzovskiy 	 *	bigobj, at the head of the nth chunk
36572fdbea25SAleksandr Guzovskiy 	 *	bigobj, at the tail of the nth chunk
36582fdbea25SAleksandr Guzovskiy 	 *
36592fdbea25SAleksandr Guzovskiy 	 * The chunk size is set equal to bigobj block size so that
36602fdbea25SAleksandr Guzovskiy 	 * dmu_assign_arcbuf() can be tested for object updates.
36612fdbea25SAleksandr Guzovskiy 	 */
36622fdbea25SAleksandr Guzovskiy 
36632fdbea25SAleksandr Guzovskiy 	/*
36642fdbea25SAleksandr Guzovskiy 	 * Read the directory info.  If it's the first time, set things up.
36652fdbea25SAleksandr Guzovskiy 	 */
3666b24ab676SJeff Bonwick 	ztest_od_init(&od[0], id, FTAG, 0, DMU_OT_UINT64_OTHER, blocksize, 0);
3667b24ab676SJeff Bonwick 	ztest_od_init(&od[1], id, FTAG, 1, DMU_OT_UINT64_OTHER, 0, chunksize);
36682fdbea25SAleksandr Guzovskiy 
3669b24ab676SJeff Bonwick 	if (ztest_object_init(zd, od, sizeof (od), B_FALSE) != 0)
3670b24ab676SJeff Bonwick 		return;
36712fdbea25SAleksandr Guzovskiy 
3672b24ab676SJeff Bonwick 	bigobj = od[0].od_object;
3673b24ab676SJeff Bonwick 	packobj = od[1].od_object;
3674b24ab676SJeff Bonwick 	blocksize = od[0].od_blocksize;
3675b24ab676SJeff Bonwick 	chunksize = blocksize;
3676b24ab676SJeff Bonwick 	ASSERT(chunksize == od[1].od_gen);
36772fdbea25SAleksandr Guzovskiy 
3678b24ab676SJeff Bonwick 	VERIFY(dmu_object_info(os, bigobj, &doi) == 0);
3679b24ab676SJeff Bonwick 	VERIFY(ISP2(doi.doi_data_block_size));
3680b24ab676SJeff Bonwick 	VERIFY(chunksize == doi.doi_data_block_size);
3681b24ab676SJeff Bonwick 	VERIFY(chunksize >= 2 * sizeof (bufwad_t));
36822fdbea25SAleksandr Guzovskiy 
36832fdbea25SAleksandr Guzovskiy 	/*
36842fdbea25SAleksandr Guzovskiy 	 * Pick a random index and compute the offsets into packobj and bigobj.
36852fdbea25SAleksandr Guzovskiy 	 */
36862fdbea25SAleksandr Guzovskiy 	n = ztest_random(regions) * stride + ztest_random(width);
36872fdbea25SAleksandr Guzovskiy 	s = 1 + ztest_random(width - 1);
36882fdbea25SAleksandr Guzovskiy 
36892fdbea25SAleksandr Guzovskiy 	packoff = n * sizeof (bufwad_t);
36902fdbea25SAleksandr Guzovskiy 	packsize = s * sizeof (bufwad_t);
36912fdbea25SAleksandr Guzovskiy 
3692b24ab676SJeff Bonwick 	bigoff = n * chunksize;
3693b24ab676SJeff Bonwick 	bigsize = s * chunksize;
36942fdbea25SAleksandr Guzovskiy 
36952fdbea25SAleksandr Guzovskiy 	packbuf = umem_zalloc(packsize, UMEM_NOFAIL);
36962fdbea25SAleksandr Guzovskiy 	bigbuf = umem_zalloc(bigsize, UMEM_NOFAIL);
36972fdbea25SAleksandr Guzovskiy 
3698b420f3adSRichard Lowe 	VERIFY3U(0, ==, dmu_bonus_hold(os, bigobj, FTAG, &bonus_db));
36992fdbea25SAleksandr Guzovskiy 
37002fdbea25SAleksandr Guzovskiy 	bigbuf_arcbufs = umem_zalloc(2 * s * sizeof (arc_buf_t *), UMEM_NOFAIL);
37012fdbea25SAleksandr Guzovskiy 
37022fdbea25SAleksandr Guzovskiy 	/*
37032fdbea25SAleksandr Guzovskiy 	 * Iteration 0 test zcopy for DB_UNCACHED dbufs.
37042fdbea25SAleksandr Guzovskiy 	 * Iteration 1 test zcopy to already referenced dbufs.
37052fdbea25SAleksandr Guzovskiy 	 * Iteration 2 test zcopy to dirty dbuf in the same txg.
37062fdbea25SAleksandr Guzovskiy 	 * Iteration 3 test zcopy to dbuf dirty in previous txg.
37072fdbea25SAleksandr Guzovskiy 	 * Iteration 4 test zcopy when dbuf is no longer dirty.
37082fdbea25SAleksandr Guzovskiy 	 * Iteration 5 test zcopy when it can't be done.
37092fdbea25SAleksandr Guzovskiy 	 * Iteration 6 one more zcopy write.
37102fdbea25SAleksandr Guzovskiy 	 */
37112fdbea25SAleksandr Guzovskiy 	for (i = 0; i < 7; i++) {
37122fdbea25SAleksandr Guzovskiy 		uint64_t j;
37132fdbea25SAleksandr Guzovskiy 		uint64_t off;
37142fdbea25SAleksandr Guzovskiy 
37152fdbea25SAleksandr Guzovskiy 		/*
37162fdbea25SAleksandr Guzovskiy 		 * In iteration 5 (i == 5) use arcbufs
37172fdbea25SAleksandr Guzovskiy 		 * that don't match bigobj blksz to test
37182fdbea25SAleksandr Guzovskiy 		 * dmu_assign_arcbuf() when it can't directly
37192fdbea25SAleksandr Guzovskiy 		 * assign an arcbuf to a dbuf.
37202fdbea25SAleksandr Guzovskiy 		 */
37212fdbea25SAleksandr Guzovskiy 		for (j = 0; j < s; j++) {
37222fdbea25SAleksandr Guzovskiy 			if (i != 5) {
37232fdbea25SAleksandr Guzovskiy 				bigbuf_arcbufs[j] =
3724b24ab676SJeff Bonwick 				    dmu_request_arcbuf(bonus_db, chunksize);
37252fdbea25SAleksandr Guzovskiy 			} else {
37262fdbea25SAleksandr Guzovskiy 				bigbuf_arcbufs[2 * j] =
3727b24ab676SJeff Bonwick 				    dmu_request_arcbuf(bonus_db, chunksize / 2);
37282fdbea25SAleksandr Guzovskiy 				bigbuf_arcbufs[2 * j + 1] =
3729b24ab676SJeff Bonwick 				    dmu_request_arcbuf(bonus_db, chunksize / 2);
37302fdbea25SAleksandr Guzovskiy 			}
37312fdbea25SAleksandr Guzovskiy 		}
37322fdbea25SAleksandr Guzovskiy 
37332fdbea25SAleksandr Guzovskiy 		/*
37342fdbea25SAleksandr Guzovskiy 		 * Get a tx for the mods to both packobj and bigobj.
37352fdbea25SAleksandr Guzovskiy 		 */
37362fdbea25SAleksandr Guzovskiy 		tx = dmu_tx_create(os);
37372fdbea25SAleksandr Guzovskiy 
3738b24ab676SJeff Bonwick 		dmu_tx_hold_write(tx, packobj, packoff, packsize);
3739b24ab676SJeff Bonwick 		dmu_tx_hold_write(tx, bigobj, bigoff, bigsize);
37402fdbea25SAleksandr Guzovskiy 
3741b24ab676SJeff Bonwick 		txg = ztest_tx_assign(tx, TXG_MIGHTWAIT, FTAG);
3742b24ab676SJeff Bonwick 		if (txg == 0) {
37432fdbea25SAleksandr Guzovskiy 			umem_free(packbuf, packsize);
37442fdbea25SAleksandr Guzovskiy 			umem_free(bigbuf, bigsize);
37452fdbea25SAleksandr Guzovskiy 			for (j = 0; j < s; j++) {
37462fdbea25SAleksandr Guzovskiy 				if (i != 5) {
37472fdbea25SAleksandr Guzovskiy 					dmu_return_arcbuf(bigbuf_arcbufs[j]);
37482fdbea25SAleksandr Guzovskiy 				} else {
37492fdbea25SAleksandr Guzovskiy 					dmu_return_arcbuf(
37502fdbea25SAleksandr Guzovskiy 					    bigbuf_arcbufs[2 * j]);
37512fdbea25SAleksandr Guzovskiy 					dmu_return_arcbuf(
37522fdbea25SAleksandr Guzovskiy 					    bigbuf_arcbufs[2 * j + 1]);
37532fdbea25SAleksandr Guzovskiy 				}
37542fdbea25SAleksandr Guzovskiy 			}
37552fdbea25SAleksandr Guzovskiy 			umem_free(bigbuf_arcbufs, 2 * s * sizeof (arc_buf_t *));
37562fdbea25SAleksandr Guzovskiy 			dmu_buf_rele(bonus_db, FTAG);
37572fdbea25SAleksandr Guzovskiy 			return;
37582fdbea25SAleksandr Guzovskiy 		}
37592fdbea25SAleksandr Guzovskiy 
37602fdbea25SAleksandr Guzovskiy 		/*
37612fdbea25SAleksandr Guzovskiy 		 * 50% of the time don't read objects in the 1st iteration to
37622fdbea25SAleksandr Guzovskiy 		 * test dmu_assign_arcbuf() for the case when there're no
37632fdbea25SAleksandr Guzovskiy 		 * existing dbufs for the specified offsets.
37642fdbea25SAleksandr Guzovskiy 		 */
37652fdbea25SAleksandr Guzovskiy 		if (i != 0 || ztest_random(2) != 0) {
3766b24ab676SJeff Bonwick 			error = dmu_read(os, packobj, packoff,
37677bfdf011SNeil Perrin 			    packsize, packbuf, DMU_READ_PREFETCH);
3768*fb09f5aaSMadhav Suresh 			ASSERT0(error);
3769b24ab676SJeff Bonwick 			error = dmu_read(os, bigobj, bigoff, bigsize,
37707bfdf011SNeil Perrin 			    bigbuf, DMU_READ_PREFETCH);
3771*fb09f5aaSMadhav Suresh 			ASSERT0(error);
37722fdbea25SAleksandr Guzovskiy 		}
37732fdbea25SAleksandr Guzovskiy 		compare_and_update_pbbufs(s, packbuf, bigbuf, bigsize,
3774b24ab676SJeff Bonwick 		    n, chunksize, txg);
37752fdbea25SAleksandr Guzovskiy 
37762fdbea25SAleksandr Guzovskiy 		/*
37772fdbea25SAleksandr Guzovskiy 		 * We've verified all the old bufwads, and made new ones.
37782fdbea25SAleksandr Guzovskiy 		 * Now write them out.
37792fdbea25SAleksandr Guzovskiy 		 */
3780b24ab676SJeff Bonwick 		dmu_write(os, packobj, packoff, packsize, packbuf, tx);
3781420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 7) {
37822fdbea25SAleksandr Guzovskiy 			(void) printf("writing offset %llx size %llx"
37832fdbea25SAleksandr Guzovskiy 			    " txg %llx\n",
37842fdbea25SAleksandr Guzovskiy 			    (u_longlong_t)bigoff,
37852fdbea25SAleksandr Guzovskiy 			    (u_longlong_t)bigsize,
37862fdbea25SAleksandr Guzovskiy 			    (u_longlong_t)txg);
37872fdbea25SAleksandr Guzovskiy 		}
3788b24ab676SJeff Bonwick 		for (off = bigoff, j = 0; j < s; j++, off += chunksize) {
37892fdbea25SAleksandr Guzovskiy 			dmu_buf_t *dbt;
37902fdbea25SAleksandr Guzovskiy 			if (i != 5) {
37912fdbea25SAleksandr Guzovskiy 				bcopy((caddr_t)bigbuf + (off - bigoff),
3792b24ab676SJeff Bonwick 				    bigbuf_arcbufs[j]->b_data, chunksize);
37932fdbea25SAleksandr Guzovskiy 			} else {
37942fdbea25SAleksandr Guzovskiy 				bcopy((caddr_t)bigbuf + (off - bigoff),
37952fdbea25SAleksandr Guzovskiy 				    bigbuf_arcbufs[2 * j]->b_data,
3796b24ab676SJeff Bonwick 				    chunksize / 2);
37972fdbea25SAleksandr Guzovskiy 				bcopy((caddr_t)bigbuf + (off - bigoff) +
3798b24ab676SJeff Bonwick 				    chunksize / 2,
37992fdbea25SAleksandr Guzovskiy 				    bigbuf_arcbufs[2 * j + 1]->b_data,
3800b24ab676SJeff Bonwick 				    chunksize / 2);
38012fdbea25SAleksandr Guzovskiy 			}
38022fdbea25SAleksandr Guzovskiy 
38032fdbea25SAleksandr Guzovskiy 			if (i == 1) {
3804b24ab676SJeff Bonwick 				VERIFY(dmu_buf_hold(os, bigobj, off,
380547cb52daSJeff Bonwick 				    FTAG, &dbt, DMU_READ_NO_PREFETCH) == 0);
38062fdbea25SAleksandr Guzovskiy 			}
38072fdbea25SAleksandr Guzovskiy 			if (i != 5) {
38082fdbea25SAleksandr Guzovskiy 				dmu_assign_arcbuf(bonus_db, off,
38092fdbea25SAleksandr Guzovskiy 				    bigbuf_arcbufs[j], tx);
38102fdbea25SAleksandr Guzovskiy 			} else {
38112fdbea25SAleksandr Guzovskiy 				dmu_assign_arcbuf(bonus_db, off,
38122fdbea25SAleksandr Guzovskiy 				    bigbuf_arcbufs[2 * j], tx);
38132fdbea25SAleksandr Guzovskiy 				dmu_assign_arcbuf(bonus_db,
3814b24ab676SJeff Bonwick 				    off + chunksize / 2,
38152fdbea25SAleksandr Guzovskiy 				    bigbuf_arcbufs[2 * j + 1], tx);
38162fdbea25SAleksandr Guzovskiy 			}
38172fdbea25SAleksandr Guzovskiy 			if (i == 1) {
38182fdbea25SAleksandr Guzovskiy 				dmu_buf_rele(dbt, FTAG);
38192fdbea25SAleksandr Guzovskiy 			}
38202fdbea25SAleksandr Guzovskiy 		}
38212fdbea25SAleksandr Guzovskiy 		dmu_tx_commit(tx);
38222fdbea25SAleksandr Guzovskiy 
38232fdbea25SAleksandr Guzovskiy 		/*
38242fdbea25SAleksandr Guzovskiy 		 * Sanity check the stuff we just wrote.
38252fdbea25SAleksandr Guzovskiy 		 */
38262fdbea25SAleksandr Guzovskiy 		{
38272fdbea25SAleksandr Guzovskiy 			void *packcheck = umem_alloc(packsize, UMEM_NOFAIL);
38282fdbea25SAleksandr Guzovskiy 			void *bigcheck = umem_alloc(bigsize, UMEM_NOFAIL);
38292fdbea25SAleksandr Guzovskiy 
3830b24ab676SJeff Bonwick 			VERIFY(0 == dmu_read(os, packobj, packoff,
38317bfdf011SNeil Perrin 			    packsize, packcheck, DMU_READ_PREFETCH));
3832b24ab676SJeff Bonwick 			VERIFY(0 == dmu_read(os, bigobj, bigoff,
38337bfdf011SNeil Perrin 			    bigsize, bigcheck, DMU_READ_PREFETCH));
38342fdbea25SAleksandr Guzovskiy 
38352fdbea25SAleksandr Guzovskiy 			ASSERT(bcmp(packbuf, packcheck, packsize) == 0);
38362fdbea25SAleksandr Guzovskiy 			ASSERT(bcmp(bigbuf, bigcheck, bigsize) == 0);
38372fdbea25SAleksandr Guzovskiy 
38382fdbea25SAleksandr Guzovskiy 			umem_free(packcheck, packsize);
38392fdbea25SAleksandr Guzovskiy 			umem_free(bigcheck, bigsize);
38402fdbea25SAleksandr Guzovskiy 		}
38412fdbea25SAleksandr Guzovskiy 		if (i == 2) {
3842b24ab676SJeff Bonwick 			txg_wait_open(dmu_objset_pool(os), 0);
3843b24ab676SJeff Bonwick 		} else if (i == 3) {
3844b24ab676SJeff Bonwick 			txg_wait_synced(dmu_objset_pool(os), 0);
3845fa9e4066Sahrens 		}
3846e05725b1Sbonwick 	}
3847e05725b1Sbonwick 
3848b24ab676SJeff Bonwick 	dmu_buf_rele(bonus_db, FTAG);
3849b24ab676SJeff Bonwick 	umem_free(packbuf, packsize);
3850b24ab676SJeff Bonwick 	umem_free(bigbuf, bigsize);
3851b24ab676SJeff Bonwick 	umem_free(bigbuf_arcbufs, 2 * s * sizeof (arc_buf_t *));
3852b24ab676SJeff Bonwick }
3853fa9e4066Sahrens 
3854b24ab676SJeff Bonwick /* ARGSUSED */
3855b24ab676SJeff Bonwick void
3856b24ab676SJeff Bonwick ztest_dmu_write_parallel(ztest_ds_t *zd, uint64_t id)
3857b24ab676SJeff Bonwick {
3858b24ab676SJeff Bonwick 	ztest_od_t od[1];
3859b24ab676SJeff Bonwick 	uint64_t offset = (1ULL << (ztest_random(20) + 43)) +
3860b24ab676SJeff Bonwick 	    (ztest_random(ZTEST_RANGE_LOCKS) << SPA_MAXBLOCKSHIFT);
3861fa9e4066Sahrens 
3862e05725b1Sbonwick 	/*
3863b24ab676SJeff Bonwick 	 * Have multiple threads write to large offsets in an object
3864b24ab676SJeff Bonwick 	 * to verify that parallel writes to an object -- even to the
3865b24ab676SJeff Bonwick 	 * same blocks within the object -- doesn't cause any trouble.
3866e05725b1Sbonwick 	 */
3867b24ab676SJeff Bonwick 	ztest_od_init(&od[0], ID_PARALLEL, FTAG, 0, DMU_OT_UINT64_OTHER, 0, 0);
3868fa9e4066Sahrens 
3869b24ab676SJeff Bonwick 	if (ztest_object_init(zd, od, sizeof (od), B_FALSE) != 0)
3870e05725b1Sbonwick 		return;
3871fa9e4066Sahrens 
3872b24ab676SJeff Bonwick 	while (ztest_random(10) != 0)
3873b24ab676SJeff Bonwick 		ztest_io(zd, od[0].od_object, offset);
3874b24ab676SJeff Bonwick }
3875fa9e4066Sahrens 
3876b24ab676SJeff Bonwick void
3877b24ab676SJeff Bonwick ztest_dmu_prealloc(ztest_ds_t *zd, uint64_t id)
3878b24ab676SJeff Bonwick {
3879b24ab676SJeff Bonwick 	ztest_od_t od[1];
3880b24ab676SJeff Bonwick 	uint64_t offset = (1ULL << (ztest_random(4) + SPA_MAXBLOCKSHIFT)) +
3881b24ab676SJeff Bonwick 	    (ztest_random(ZTEST_RANGE_LOCKS) << SPA_MAXBLOCKSHIFT);
3882b24ab676SJeff Bonwick 	uint64_t count = ztest_random(20) + 1;
3883b24ab676SJeff Bonwick 	uint64_t blocksize = ztest_random_blocksize();
3884b24ab676SJeff Bonwick 	void *data;
3885fa9e4066Sahrens 
3886b24ab676SJeff Bonwick 	ztest_od_init(&od[0], id, FTAG, 0, DMU_OT_UINT64_OTHER, blocksize, 0);
3887fa9e4066Sahrens 
3888b24ab676SJeff Bonwick 	if (ztest_object_init(zd, od, sizeof (od), !ztest_random(2)) != 0)
3889e05725b1Sbonwick 		return;
38900e34b6a7Sbonwick 
3891b24ab676SJeff Bonwick 	if (ztest_truncate(zd, od[0].od_object, offset, count * blocksize) != 0)
3892f4a72450SJeff Bonwick 		return;
3893f4a72450SJeff Bonwick 
3894b24ab676SJeff Bonwick 	ztest_prealloc(zd, od[0].od_object, offset, count * blocksize);
3895fa9e4066Sahrens 
3896b24ab676SJeff Bonwick 	data = umem_zalloc(blocksize, UMEM_NOFAIL);
3897b24ab676SJeff Bonwick 
3898b24ab676SJeff Bonwick 	while (ztest_random(count) != 0) {
3899b24ab676SJeff Bonwick 		uint64_t randoff = offset + (ztest_random(count) * blocksize);
3900b24ab676SJeff Bonwick 		if (ztest_write(zd, od[0].od_object, randoff, blocksize,
3901b24ab676SJeff Bonwick 		    data) != 0)
3902b24ab676SJeff Bonwick 			break;
3903b24ab676SJeff Bonwick 		while (ztest_random(4) != 0)
3904b24ab676SJeff Bonwick 			ztest_io(zd, od[0].od_object, randoff);
3905b24ab676SJeff Bonwick 	}
3906b24ab676SJeff Bonwick 
3907b24ab676SJeff Bonwick 	umem_free(data, blocksize);
3908fa9e4066Sahrens }
3909fa9e4066Sahrens 
3910fa9e4066Sahrens /*
3911fa9e4066Sahrens  * Verify that zap_{create,destroy,add,remove,update} work as expected.
3912fa9e4066Sahrens  */
3913fa9e4066Sahrens #define	ZTEST_ZAP_MIN_INTS	1
3914fa9e4066Sahrens #define	ZTEST_ZAP_MAX_INTS	4
3915fa9e4066Sahrens #define	ZTEST_ZAP_MAX_PROPS	1000
3916fa9e4066Sahrens 
3917fa9e4066Sahrens void
3918b24ab676SJeff Bonwick ztest_zap(ztest_ds_t *zd, uint64_t id)
3919fa9e4066Sahrens {
3920b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
3921b24ab676SJeff Bonwick 	ztest_od_t od[1];
3922fa9e4066Sahrens 	uint64_t object;
3923fa9e4066Sahrens 	uint64_t txg, last_txg;
3924fa9e4066Sahrens 	uint64_t value[ZTEST_ZAP_MAX_INTS];
3925fa9e4066Sahrens 	uint64_t zl_ints, zl_intsize, prop;
3926fa9e4066Sahrens 	int i, ints;
3927fa9e4066Sahrens 	dmu_tx_t *tx;
3928fa9e4066Sahrens 	char propname[100], txgname[100];
3929fa9e4066Sahrens 	int error;
3930fa9e4066Sahrens 	char *hc[2] = { "s.acl.h", ".s.open.h.hyLZlg" };
3931fa9e4066Sahrens 
3932b24ab676SJeff Bonwick 	ztest_od_init(&od[0], id, FTAG, 0, DMU_OT_ZAP_OTHER, 0, 0);
3933fa9e4066Sahrens 
3934b24ab676SJeff Bonwick 	if (ztest_object_init(zd, od, sizeof (od), !ztest_random(2)) != 0)
3935b24ab676SJeff Bonwick 		return;
3936fa9e4066Sahrens 
3937b24ab676SJeff Bonwick 	object = od[0].od_object;
3938fa9e4066Sahrens 
3939b24ab676SJeff Bonwick 	/*
3940b24ab676SJeff Bonwick 	 * Generate a known hash collision, and verify that
3941b24ab676SJeff Bonwick 	 * we can lookup and remove both entries.
3942b24ab676SJeff Bonwick 	 */
3943b24ab676SJeff Bonwick 	tx = dmu_tx_create(os);
3944b24ab676SJeff Bonwick 	dmu_tx_hold_zap(tx, object, B_TRUE, NULL);
3945b24ab676SJeff Bonwick 	txg = ztest_tx_assign(tx, TXG_MIGHTWAIT, FTAG);
3946b24ab676SJeff Bonwick 	if (txg == 0)
3947b24ab676SJeff Bonwick 		return;
3948b24ab676SJeff Bonwick 	for (i = 0; i < 2; i++) {
3949b24ab676SJeff Bonwick 		value[i] = i;
3950b420f3adSRichard Lowe 		VERIFY3U(0, ==, zap_add(os, object, hc[i], sizeof (uint64_t),
3951b24ab676SJeff Bonwick 		    1, &value[i], tx));
3952b24ab676SJeff Bonwick 	}
3953b24ab676SJeff Bonwick 	for (i = 0; i < 2; i++) {
3954b24ab676SJeff Bonwick 		VERIFY3U(EEXIST, ==, zap_add(os, object, hc[i],
3955b24ab676SJeff Bonwick 		    sizeof (uint64_t), 1, &value[i], tx));
3956b420f3adSRichard Lowe 		VERIFY3U(0, ==,
3957b420f3adSRichard Lowe 		    zap_length(os, object, hc[i], &zl_intsize, &zl_ints));
3958b24ab676SJeff Bonwick 		ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
3959b24ab676SJeff Bonwick 		ASSERT3U(zl_ints, ==, 1);
3960fa9e4066Sahrens 	}
3961b24ab676SJeff Bonwick 	for (i = 0; i < 2; i++) {
3962b420f3adSRichard Lowe 		VERIFY3U(0, ==, zap_remove(os, object, hc[i], tx));
3963b24ab676SJeff Bonwick 	}
3964b24ab676SJeff Bonwick 	dmu_tx_commit(tx);
3965fa9e4066Sahrens 
3966b24ab676SJeff Bonwick 	/*
3967b24ab676SJeff Bonwick 	 * Generate a buch of random entries.
3968b24ab676SJeff Bonwick 	 */
3969fa9e4066Sahrens 	ints = MAX(ZTEST_ZAP_MIN_INTS, object % ZTEST_ZAP_MAX_INTS);
3970fa9e4066Sahrens 
3971e05725b1Sbonwick 	prop = ztest_random(ZTEST_ZAP_MAX_PROPS);
3972e05725b1Sbonwick 	(void) sprintf(propname, "prop_%llu", (u_longlong_t)prop);
3973e05725b1Sbonwick 	(void) sprintf(txgname, "txg_%llu", (u_longlong_t)prop);
3974e05725b1Sbonwick 	bzero(value, sizeof (value));
3975e05725b1Sbonwick 	last_txg = 0;
3976fa9e4066Sahrens 
3977e05725b1Sbonwick 	/*
3978e05725b1Sbonwick 	 * If these zap entries already exist, validate their contents.
3979e05725b1Sbonwick 	 */
3980e05725b1Sbonwick 	error = zap_length(os, object, txgname, &zl_intsize, &zl_ints);
3981e05725b1Sbonwick 	if (error == 0) {
3982e05725b1Sbonwick 		ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
3983e05725b1Sbonwick 		ASSERT3U(zl_ints, ==, 1);
3984fa9e4066Sahrens 
3985e05725b1Sbonwick 		VERIFY(zap_lookup(os, object, txgname, zl_intsize,
3986e05725b1Sbonwick 		    zl_ints, &last_txg) == 0);
3987fa9e4066Sahrens 
3988e05725b1Sbonwick 		VERIFY(zap_length(os, object, propname, &zl_intsize,
3989e05725b1Sbonwick 		    &zl_ints) == 0);
3990fa9e4066Sahrens 
3991e05725b1Sbonwick 		ASSERT3U(zl_intsize, ==, sizeof (uint64_t));
3992e05725b1Sbonwick 		ASSERT3U(zl_ints, ==, ints);
3993fa9e4066Sahrens 
3994e05725b1Sbonwick 		VERIFY(zap_lookup(os, object, propname, zl_intsize,
3995e05725b1Sbonwick 		    zl_ints, value) == 0);
3996fa9e4066Sahrens 
3997e05725b1Sbonwick 		for (i = 0; i < ints; i++) {
3998e05725b1Sbonwick 			ASSERT3U(value[i], ==, last_txg + object + i);
3999fa9e4066Sahrens 		}
4000e05725b1Sbonwick 	} else {
4001e05725b1Sbonwick 		ASSERT3U(error, ==, ENOENT);
4002e05725b1Sbonwick 	}
4003fa9e4066Sahrens 
4004e05725b1Sbonwick 	/*
4005e05725b1Sbonwick 	 * Atomically update two entries in our zap object.
4006e05725b1Sbonwick 	 * The first is named txg_%llu, and contains the txg
4007e05725b1Sbonwick 	 * in which the property was last updated.  The second
4008e05725b1Sbonwick 	 * is named prop_%llu, and the nth element of its value
4009e05725b1Sbonwick 	 * should be txg + object + n.
4010e05725b1Sbonwick 	 */
4011e05725b1Sbonwick 	tx = dmu_tx_create(os);
4012b24ab676SJeff Bonwick 	dmu_tx_hold_zap(tx, object, B_TRUE, NULL);
4013b24ab676SJeff Bonwick 	txg = ztest_tx_assign(tx, TXG_MIGHTWAIT, FTAG);
4014b24ab676SJeff Bonwick 	if (txg == 0)
4015e05725b1Sbonwick 		return;
4016fa9e4066Sahrens 
4017e05725b1Sbonwick 	if (last_txg > txg)
4018e05725b1Sbonwick 		fatal(0, "zap future leak: old %llu new %llu", last_txg, txg);
4019fa9e4066Sahrens 
4020e05725b1Sbonwick 	for (i = 0; i < ints; i++)
4021e05725b1Sbonwick 		value[i] = txg + object + i;
4022fa9e4066Sahrens 
4023b420f3adSRichard Lowe 	VERIFY3U(0, ==, zap_update(os, object, txgname, sizeof (uint64_t),
4024b24ab676SJeff Bonwick 	    1, &txg, tx));
4025b420f3adSRichard Lowe 	VERIFY3U(0, ==, zap_update(os, object, propname, sizeof (uint64_t),
4026b24ab676SJeff Bonwick 	    ints, value, tx));
4027fa9e4066Sahrens 
4028e05725b1Sbonwick 	dmu_tx_commit(tx);
4029fa9e4066Sahrens 
4030e05725b1Sbonwick 	/*
4031e05725b1Sbonwick 	 * Remove a random pair of entries.
4032e05725b1Sbonwick 	 */
4033e05725b1Sbonwick 	prop = ztest_random(ZTEST_ZAP_MAX_PROPS);
4034e05725b1Sbonwick 	(void) sprintf(propname, "prop_%llu", (u_longlong_t)prop);
4035e05725b1Sbonwick 	(void) sprintf(txgname, "txg_%llu", (u_longlong_t)prop);
4036fa9e4066Sahrens 
4037e05725b1Sbonwick 	error = zap_length(os, object, txgname, &zl_intsize, &zl_ints);
4038fa9e4066Sahrens 
4039e05725b1Sbonwick 	if (error == ENOENT)
4040e05725b1Sbonwick 		return;
4041fa9e4066Sahrens 
4042*fb09f5aaSMadhav Suresh 	ASSERT0(error);
4043fa9e4066Sahrens 
4044e05725b1Sbonwick 	tx = dmu_tx_create(os);
4045b24ab676SJeff Bonwick 	dmu_tx_hold_zap(tx, object, B_TRUE, NULL);
4046b24ab676SJeff Bonwick 	txg = ztest_tx_assign(tx, TXG_MIGHTWAIT, FTAG);
4047b24ab676SJeff Bonwick 	if (txg == 0)
404812a2833aSSanjeev Bagewadi 		return;
4049b420f3adSRichard Lowe 	VERIFY3U(0, ==, zap_remove(os, object, txgname, tx));
4050b420f3adSRichard Lowe 	VERIFY3U(0, ==, zap_remove(os, object, propname, tx));
405112a2833aSSanjeev Bagewadi 	dmu_tx_commit(tx);
405212a2833aSSanjeev Bagewadi }
405312a2833aSSanjeev Bagewadi 
405412a2833aSSanjeev Bagewadi /*
405512a2833aSSanjeev Bagewadi  * Testcase to test the upgrading of a microzap to fatzap.
405612a2833aSSanjeev Bagewadi  */
405712a2833aSSanjeev Bagewadi void
4058b24ab676SJeff Bonwick ztest_fzap(ztest_ds_t *zd, uint64_t id)
405912a2833aSSanjeev Bagewadi {
4060b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
4061b24ab676SJeff Bonwick 	ztest_od_t od[1];
4062b24ab676SJeff Bonwick 	uint64_t object, txg;
406312a2833aSSanjeev Bagewadi 
4064b24ab676SJeff Bonwick 	ztest_od_init(&od[0], id, FTAG, 0, DMU_OT_ZAP_OTHER, 0, 0);
406512a2833aSSanjeev Bagewadi 
4066b24ab676SJeff Bonwick 	if (ztest_object_init(zd, od, sizeof (od), !ztest_random(2)) != 0)
4067b24ab676SJeff Bonwick 		return;
406812a2833aSSanjeev Bagewadi 
4069b24ab676SJeff Bonwick 	object = od[0].od_object;
407012a2833aSSanjeev Bagewadi 
407112a2833aSSanjeev Bagewadi 	/*
4072b24ab676SJeff Bonwick 	 * Add entries to this ZAP and make sure it spills over
407312a2833aSSanjeev Bagewadi 	 * and gets upgraded to a fatzap. Also, since we are adding
4074b24ab676SJeff Bonwick 	 * 2050 entries we should see ptrtbl growth and leaf-block split.
407512a2833aSSanjeev Bagewadi 	 */
4076b24ab676SJeff Bonwick 	for (int i = 0; i < 2050; i++) {
4077b24ab676SJeff Bonwick 		char name[MAXNAMELEN];
4078b24ab676SJeff Bonwick 		uint64_t value = i;
4079b24ab676SJeff Bonwick 		dmu_tx_t *tx;
4080b24ab676SJeff Bonwick 		int error;
408112a2833aSSanjeev Bagewadi 
4082b24ab676SJeff Bonwick 		(void) snprintf(name, sizeof (name), "fzap-%llu-%llu",
4083b24ab676SJeff Bonwick 		    id, value);
408412a2833aSSanjeev Bagewadi 
4085b24ab676SJeff Bonwick 		tx = dmu_tx_create(os);
4086b24ab676SJeff Bonwick 		dmu_tx_hold_zap(tx, object, B_TRUE, name);
4087b24ab676SJeff Bonwick 		txg = ztest_tx_assign(tx, TXG_MIGHTWAIT, FTAG);
4088b24ab676SJeff Bonwick 		if (txg == 0)
408912a2833aSSanjeev Bagewadi 			return;
4090b24ab676SJeff Bonwick 		error = zap_add(os, object, name, sizeof (uint64_t), 1,
4091b24ab676SJeff Bonwick 		    &value, tx);
409212a2833aSSanjeev Bagewadi 		ASSERT(error == 0 || error == EEXIST);
409312a2833aSSanjeev Bagewadi 		dmu_tx_commit(tx);
409412a2833aSSanjeev Bagewadi 	}
4095fa9e4066Sahrens }
4096fa9e4066Sahrens 
4097b24ab676SJeff Bonwick /* ARGSUSED */
4098fa9e4066Sahrens void
4099b24ab676SJeff Bonwick ztest_zap_parallel(ztest_ds_t *zd, uint64_t id)
4100fa9e4066Sahrens {
4101b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
4102b24ab676SJeff Bonwick 	ztest_od_t od[1];
4103fa9e4066Sahrens 	uint64_t txg, object, count, wsize, wc, zl_wsize, zl_wc;
4104fa9e4066Sahrens 	dmu_tx_t *tx;
4105fa9e4066Sahrens 	int i, namelen, error;
4106b24ab676SJeff Bonwick 	int micro = ztest_random(2);
4107fa9e4066Sahrens 	char name[20], string_value[20];
4108fa9e4066Sahrens 	void *data;
4109fa9e4066Sahrens 
4110b24ab676SJeff Bonwick 	ztest_od_init(&od[0], ID_PARALLEL, FTAG, micro, DMU_OT_ZAP_OTHER, 0, 0);
4111b24ab676SJeff Bonwick 
4112b24ab676SJeff Bonwick 	if (ztest_object_init(zd, od, sizeof (od), B_FALSE) != 0)
4113b24ab676SJeff Bonwick 		return;
4114b24ab676SJeff Bonwick 
4115b24ab676SJeff Bonwick 	object = od[0].od_object;
4116b24ab676SJeff Bonwick 
4117e05725b1Sbonwick 	/*
4118e05725b1Sbonwick 	 * Generate a random name of the form 'xxx.....' where each
4119e05725b1Sbonwick 	 * x is a random printable character and the dots are dots.
4120e05725b1Sbonwick 	 * There are 94 such characters, and the name length goes from
4121e05725b1Sbonwick 	 * 6 to 20, so there are 94^3 * 15 = 12,458,760 possible names.
4122e05725b1Sbonwick 	 */
4123e05725b1Sbonwick 	namelen = ztest_random(sizeof (name) - 5) + 5 + 1;
4124fa9e4066Sahrens 
4125e05725b1Sbonwick 	for (i = 0; i < 3; i++)
4126e05725b1Sbonwick 		name[i] = '!' + ztest_random('~' - '!' + 1);
4127e05725b1Sbonwick 	for (; i < namelen - 1; i++)
4128e05725b1Sbonwick 		name[i] = '.';
4129e05725b1Sbonwick 	name[i] = '\0';
4130fa9e4066Sahrens 
4131b24ab676SJeff Bonwick 	if ((namelen & 1) || micro) {
4132e05725b1Sbonwick 		wsize = sizeof (txg);
4133e05725b1Sbonwick 		wc = 1;
4134e05725b1Sbonwick 		data = &txg;
4135e05725b1Sbonwick 	} else {
4136e05725b1Sbonwick 		wsize = 1;
4137e05725b1Sbonwick 		wc = namelen;
4138e05725b1Sbonwick 		data = string_value;
4139e05725b1Sbonwick 	}
4140fa9e4066Sahrens 
4141e05725b1Sbonwick 	count = -1ULL;
4142e05725b1Sbonwick 	VERIFY(zap_count(os, object, &count) == 0);
4143e05725b1Sbonwick 	ASSERT(count != -1ULL);
4144fa9e4066Sahrens 
4145e05725b1Sbonwick 	/*
4146e05725b1Sbonwick 	 * Select an operation: length, lookup, add, update, remove.
4147e05725b1Sbonwick 	 */
4148e05725b1Sbonwick 	i = ztest_random(5);
4149e05725b1Sbonwick 
4150e05725b1Sbonwick 	if (i >= 2) {
4151e05725b1Sbonwick 		tx = dmu_tx_create(os);
4152b24ab676SJeff Bonwick 		dmu_tx_hold_zap(tx, object, B_TRUE, NULL);
4153b24ab676SJeff Bonwick 		txg = ztest_tx_assign(tx, TXG_MIGHTWAIT, FTAG);
4154b24ab676SJeff Bonwick 		if (txg == 0)
4155e05725b1Sbonwick 			return;
4156e05725b1Sbonwick 		bcopy(name, string_value, namelen);
4157e05725b1Sbonwick 	} else {
4158e05725b1Sbonwick 		tx = NULL;
4159e05725b1Sbonwick 		txg = 0;
4160e05725b1Sbonwick 		bzero(string_value, namelen);
4161e05725b1Sbonwick 	}
4162fa9e4066Sahrens 
4163e05725b1Sbonwick 	switch (i) {
4164fa9e4066Sahrens 
4165e05725b1Sbonwick 	case 0:
4166e05725b1Sbonwick 		error = zap_length(os, object, name, &zl_wsize, &zl_wc);
4167e05725b1Sbonwick 		if (error == 0) {
4168e05725b1Sbonwick 			ASSERT3U(wsize, ==, zl_wsize);
4169e05725b1Sbonwick 			ASSERT3U(wc, ==, zl_wc);
4170e05725b1Sbonwick 		} else {
4171e05725b1Sbonwick 			ASSERT3U(error, ==, ENOENT);
4172e05725b1Sbonwick 		}
4173e05725b1Sbonwick 		break;
4174fa9e4066Sahrens 
4175e05725b1Sbonwick 	case 1:
4176e05725b1Sbonwick 		error = zap_lookup(os, object, name, wsize, wc, data);
4177e05725b1Sbonwick 		if (error == 0) {
4178e05725b1Sbonwick 			if (data == string_value &&
4179e05725b1Sbonwick 			    bcmp(name, data, namelen) != 0)
4180e05725b1Sbonwick 				fatal(0, "name '%s' != val '%s' len %d",
4181e05725b1Sbonwick 				    name, data, namelen);
4182e05725b1Sbonwick 		} else {
4183e05725b1Sbonwick 			ASSERT3U(error, ==, ENOENT);
4184e05725b1Sbonwick 		}
4185e05725b1Sbonwick 		break;
4186fa9e4066Sahrens 
4187e05725b1Sbonwick 	case 2:
4188e05725b1Sbonwick 		error = zap_add(os, object, name, wsize, wc, data, tx);
4189e05725b1Sbonwick 		ASSERT(error == 0 || error == EEXIST);
4190e05725b1Sbonwick 		break;
4191fa9e4066Sahrens 
4192e05725b1Sbonwick 	case 3:
4193e05725b1Sbonwick 		VERIFY(zap_update(os, object, name, wsize, wc, data, tx) == 0);
4194e05725b1Sbonwick 		break;
4195fa9e4066Sahrens 
4196e05725b1Sbonwick 	case 4:
4197e05725b1Sbonwick 		error = zap_remove(os, object, name, tx);
4198e05725b1Sbonwick 		ASSERT(error == 0 || error == ENOENT);
4199e05725b1Sbonwick 		break;
4200fa9e4066Sahrens 	}
4201e05725b1Sbonwick 
4202e05725b1Sbonwick 	if (tx != NULL)
4203e05725b1Sbonwick 		dmu_tx_commit(tx);
4204fa9e4066Sahrens }
4205fa9e4066Sahrens 
4206d20e665cSRicardo M. Correia /*
4207d20e665cSRicardo M. Correia  * Commit callback data.
4208d20e665cSRicardo M. Correia  */
4209d20e665cSRicardo M. Correia typedef struct ztest_cb_data {
4210d20e665cSRicardo M. Correia 	list_node_t		zcd_node;
4211d20e665cSRicardo M. Correia 	uint64_t		zcd_txg;
4212d20e665cSRicardo M. Correia 	int			zcd_expected_err;
4213d20e665cSRicardo M. Correia 	boolean_t		zcd_added;
4214d20e665cSRicardo M. Correia 	boolean_t		zcd_called;
4215d20e665cSRicardo M. Correia 	spa_t			*zcd_spa;
4216d20e665cSRicardo M. Correia } ztest_cb_data_t;
4217d20e665cSRicardo M. Correia 
4218d20e665cSRicardo M. Correia /* This is the actual commit callback function */
4219d20e665cSRicardo M. Correia static void
4220d20e665cSRicardo M. Correia ztest_commit_callback(void *arg, int error)
4221d20e665cSRicardo M. Correia {
4222d20e665cSRicardo M. Correia 	ztest_cb_data_t *data = arg;
4223d20e665cSRicardo M. Correia 	uint64_t synced_txg;
4224d20e665cSRicardo M. Correia 
4225d20e665cSRicardo M. Correia 	VERIFY(data != NULL);
4226d20e665cSRicardo M. Correia 	VERIFY3S(data->zcd_expected_err, ==, error);
4227d20e665cSRicardo M. Correia 	VERIFY(!data->zcd_called);
4228d20e665cSRicardo M. Correia 
4229d20e665cSRicardo M. Correia 	synced_txg = spa_last_synced_txg(data->zcd_spa);
4230d20e665cSRicardo M. Correia 	if (data->zcd_txg > synced_txg)
4231d20e665cSRicardo M. Correia 		fatal(0, "commit callback of txg %" PRIu64 " called prematurely"
4232d20e665cSRicardo M. Correia 		    ", last synced txg = %" PRIu64 "\n", data->zcd_txg,
4233d20e665cSRicardo M. Correia 		    synced_txg);
4234d20e665cSRicardo M. Correia 
4235d20e665cSRicardo M. Correia 	data->zcd_called = B_TRUE;
4236d20e665cSRicardo M. Correia 
4237d20e665cSRicardo M. Correia 	if (error == ECANCELED) {
4238*fb09f5aaSMadhav Suresh 		ASSERT0(data->zcd_txg);
4239d20e665cSRicardo M. Correia 		ASSERT(!data->zcd_added);
4240d20e665cSRicardo M. Correia 
4241d20e665cSRicardo M. Correia 		/*
4242d20e665cSRicardo M. Correia 		 * The private callback data should be destroyed here, but
4243d20e665cSRicardo M. Correia 		 * since we are going to check the zcd_called field after
4244d20e665cSRicardo M. Correia 		 * dmu_tx_abort(), we will destroy it there.
4245d20e665cSRicardo M. Correia 		 */
4246d20e665cSRicardo M. Correia 		return;
4247d20e665cSRicardo M. Correia 	}
4248d20e665cSRicardo M. Correia 
4249d20e665cSRicardo M. Correia 	/* Was this callback added to the global callback list? */
4250d20e665cSRicardo M. Correia 	if (!data->zcd_added)
4251d20e665cSRicardo M. Correia 		goto out;
4252d20e665cSRicardo M. Correia 
4253d20e665cSRicardo M. Correia 	ASSERT3U(data->zcd_txg, !=, 0);
4254d20e665cSRicardo M. Correia 
4255d20e665cSRicardo M. Correia 	/* Remove our callback from the list */
4256d20e665cSRicardo M. Correia 	(void) mutex_lock(&zcl.zcl_callbacks_lock);
4257d20e665cSRicardo M. Correia 	list_remove(&zcl.zcl_callbacks, data);
4258d20e665cSRicardo M. Correia 	(void) mutex_unlock(&zcl.zcl_callbacks_lock);
4259d20e665cSRicardo M. Correia 
4260d20e665cSRicardo M. Correia out:
4261d20e665cSRicardo M. Correia 	umem_free(data, sizeof (ztest_cb_data_t));
4262d20e665cSRicardo M. Correia }
4263d20e665cSRicardo M. Correia 
4264d20e665cSRicardo M. Correia /* Allocate and initialize callback data structure */
4265d20e665cSRicardo M. Correia static ztest_cb_data_t *
4266d20e665cSRicardo M. Correia ztest_create_cb_data(objset_t *os, uint64_t txg)
4267d20e665cSRicardo M. Correia {
4268d20e665cSRicardo M. Correia 	ztest_cb_data_t *cb_data;
4269d20e665cSRicardo M. Correia 
4270d20e665cSRicardo M. Correia 	cb_data = umem_zalloc(sizeof (ztest_cb_data_t), UMEM_NOFAIL);
4271d20e665cSRicardo M. Correia 
4272d20e665cSRicardo M. Correia 	cb_data->zcd_txg = txg;
4273d20e665cSRicardo M. Correia 	cb_data->zcd_spa = dmu_objset_spa(os);
4274d20e665cSRicardo M. Correia 
4275d20e665cSRicardo M. Correia 	return (cb_data);
4276d20e665cSRicardo M. Correia }
4277d20e665cSRicardo M. Correia 
4278d20e665cSRicardo M. Correia /*
4279d20e665cSRicardo M. Correia  * If a number of txgs equal to this threshold have been created after a commit
4280d20e665cSRicardo M. Correia  * callback has been registered but not called, then we assume there is an
4281d20e665cSRicardo M. Correia  * implementation bug.
4282d20e665cSRicardo M. Correia  */
4283d20e665cSRicardo M. Correia #define	ZTEST_COMMIT_CALLBACK_THRESH	(TXG_CONCURRENT_STATES + 2)
4284d20e665cSRicardo M. Correia 
4285d20e665cSRicardo M. Correia /*
4286d20e665cSRicardo M. Correia  * Commit callback test.
4287d20e665cSRicardo M. Correia  */
4288d20e665cSRicardo M. Correia void
4289b24ab676SJeff Bonwick ztest_dmu_commit_callbacks(ztest_ds_t *zd, uint64_t id)
4290d20e665cSRicardo M. Correia {
4291b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
4292b24ab676SJeff Bonwick 	ztest_od_t od[1];
4293d20e665cSRicardo M. Correia 	dmu_tx_t *tx;
4294d20e665cSRicardo M. Correia 	ztest_cb_data_t *cb_data[3], *tmp_cb;
4295d20e665cSRicardo M. Correia 	uint64_t old_txg, txg;
4296d20e665cSRicardo M. Correia 	int i, error;
4297d20e665cSRicardo M. Correia 
4298b24ab676SJeff Bonwick 	ztest_od_init(&od[0], id, FTAG, 0, DMU_OT_UINT64_OTHER, 0, 0);
4299b24ab676SJeff Bonwick 
4300b24ab676SJeff Bonwick 	if (ztest_object_init(zd, od, sizeof (od), B_FALSE) != 0)
4301b24ab676SJeff Bonwick 		return;
4302b24ab676SJeff Bonwick 
4303d20e665cSRicardo M. Correia 	tx = dmu_tx_create(os);
4304d20e665cSRicardo M. Correia 
4305d20e665cSRicardo M. Correia 	cb_data[0] = ztest_create_cb_data(os, 0);
4306d20e665cSRicardo M. Correia 	dmu_tx_callback_register(tx, ztest_commit_callback, cb_data[0]);
4307d20e665cSRicardo M. Correia 
4308b24ab676SJeff Bonwick 	dmu_tx_hold_write(tx, od[0].od_object, 0, sizeof (uint64_t));
4309d20e665cSRicardo M. Correia 
4310d20e665cSRicardo M. Correia 	/* Every once in a while, abort the transaction on purpose */
4311d20e665cSRicardo M. Correia 	if (ztest_random(100) == 0)
4312d20e665cSRicardo M. Correia 		error = -1;
4313d20e665cSRicardo M. Correia 
4314d20e665cSRicardo M. Correia 	if (!error)
4315d20e665cSRicardo M. Correia 		error = dmu_tx_assign(tx, TXG_NOWAIT);
4316d20e665cSRicardo M. Correia 
4317d20e665cSRicardo M. Correia 	txg = error ? 0 : dmu_tx_get_txg(tx);
4318d20e665cSRicardo M. Correia 
4319d20e665cSRicardo M. Correia 	cb_data[0]->zcd_txg = txg;
4320d20e665cSRicardo M. Correia 	cb_data[1] = ztest_create_cb_data(os, txg);
4321d20e665cSRicardo M. Correia 	dmu_tx_callback_register(tx, ztest_commit_callback, cb_data[1]);
4322d20e665cSRicardo M. Correia 
4323d20e665cSRicardo M. Correia 	if (error) {
4324d20e665cSRicardo M. Correia 		/*
4325d20e665cSRicardo M. Correia 		 * It's not a strict requirement to call the registered
4326d20e665cSRicardo M. Correia 		 * callbacks from inside dmu_tx_abort(), but that's what
4327d20e665cSRicardo M. Correia 		 * it's supposed to happen in the current implementation
4328d20e665cSRicardo M. Correia 		 * so we will check for that.
4329d20e665cSRicardo M. Correia 		 */
4330d20e665cSRicardo M. Correia 		for (i = 0; i < 2; i++) {
4331d20e665cSRicardo M. Correia 			cb_data[i]->zcd_expected_err = ECANCELED;
4332d20e665cSRicardo M. Correia 			VERIFY(!cb_data[i]->zcd_called);
4333d20e665cSRicardo M. Correia 		}
4334d20e665cSRicardo M. Correia 
4335d20e665cSRicardo M. Correia 		dmu_tx_abort(tx);
4336d20e665cSRicardo M. Correia 
4337d20e665cSRicardo M. Correia 		for (i = 0; i < 2; i++) {
4338d20e665cSRicardo M. Correia 			VERIFY(cb_data[i]->zcd_called);
4339d20e665cSRicardo M. Correia 			umem_free(cb_data[i], sizeof (ztest_cb_data_t));
4340d20e665cSRicardo M. Correia 		}
4341d20e665cSRicardo M. Correia 
4342d20e665cSRicardo M. Correia 		return;
4343d20e665cSRicardo M. Correia 	}
4344d20e665cSRicardo M. Correia 
4345d20e665cSRicardo M. Correia 	cb_data[2] = ztest_create_cb_data(os, txg);
4346d20e665cSRicardo M. Correia 	dmu_tx_callback_register(tx, ztest_commit_callback, cb_data[2]);
4347d20e665cSRicardo M. Correia 
4348d20e665cSRicardo M. Correia 	/*
4349d20e665cSRicardo M. Correia 	 * Read existing data to make sure there isn't a future leak.
4350d20e665cSRicardo M. Correia 	 */
4351b24ab676SJeff Bonwick 	VERIFY(0 == dmu_read(os, od[0].od_object, 0, sizeof (uint64_t),
4352d20e665cSRicardo M. Correia 	    &old_txg, DMU_READ_PREFETCH));
4353d20e665cSRicardo M. Correia 
4354d20e665cSRicardo M. Correia 	if (old_txg > txg)
4355d20e665cSRicardo M. Correia 		fatal(0, "future leak: got %" PRIu64 ", open txg is %" PRIu64,
4356d20e665cSRicardo M. Correia 		    old_txg, txg);
4357d20e665cSRicardo M. Correia 
4358b24ab676SJeff Bonwick 	dmu_write(os, od[0].od_object, 0, sizeof (uint64_t), &txg, tx);
4359d20e665cSRicardo M. Correia 
4360d20e665cSRicardo M. Correia 	(void) mutex_lock(&zcl.zcl_callbacks_lock);
4361d20e665cSRicardo M. Correia 
4362d20e665cSRicardo M. Correia 	/*
4363d20e665cSRicardo M. Correia 	 * Since commit callbacks don't have any ordering requirement and since
4364d20e665cSRicardo M. Correia 	 * it is theoretically possible for a commit callback to be called
4365d20e665cSRicardo M. Correia 	 * after an arbitrary amount of time has elapsed since its txg has been
4366d20e665cSRicardo M. Correia 	 * synced, it is difficult to reliably determine whether a commit
4367d20e665cSRicardo M. Correia 	 * callback hasn't been called due to high load or due to a flawed
4368d20e665cSRicardo M. Correia 	 * implementation.
4369d20e665cSRicardo M. Correia 	 *
4370d20e665cSRicardo M. Correia 	 * In practice, we will assume that if after a certain number of txgs a
4371d20e665cSRicardo M. Correia 	 * commit callback hasn't been called, then most likely there's an
4372d20e665cSRicardo M. Correia 	 * implementation bug..
4373d20e665cSRicardo M. Correia 	 */
4374d20e665cSRicardo M. Correia 	tmp_cb = list_head(&zcl.zcl_callbacks);
4375d20e665cSRicardo M. Correia 	if (tmp_cb != NULL &&
4376d20e665cSRicardo M. Correia 	    tmp_cb->zcd_txg > txg - ZTEST_COMMIT_CALLBACK_THRESH) {
4377d20e665cSRicardo M. Correia 		fatal(0, "Commit callback threshold exceeded, oldest txg: %"
4378d20e665cSRicardo M. Correia 		    PRIu64 ", open txg: %" PRIu64 "\n", tmp_cb->zcd_txg, txg);
4379d20e665cSRicardo M. Correia 	}
4380d20e665cSRicardo M. Correia 
4381d20e665cSRicardo M. Correia 	/*
4382d20e665cSRicardo M. Correia 	 * Let's find the place to insert our callbacks.
4383d20e665cSRicardo M. Correia 	 *
4384d20e665cSRicardo M. Correia 	 * Even though the list is ordered by txg, it is possible for the
4385d20e665cSRicardo M. Correia 	 * insertion point to not be the end because our txg may already be
4386d20e665cSRicardo M. Correia 	 * quiescing at this point and other callbacks in the open txg
4387d20e665cSRicardo M. Correia 	 * (from other objsets) may have sneaked in.
4388d20e665cSRicardo M. Correia 	 */
4389d20e665cSRicardo M. Correia 	tmp_cb = list_tail(&zcl.zcl_callbacks);
4390d20e665cSRicardo M. Correia 	while (tmp_cb != NULL && tmp_cb->zcd_txg > txg)
4391d20e665cSRicardo M. Correia 		tmp_cb = list_prev(&zcl.zcl_callbacks, tmp_cb);
4392d20e665cSRicardo M. Correia 
4393d20e665cSRicardo M. Correia 	/* Add the 3 callbacks to the list */
4394d20e665cSRicardo M. Correia 	for (i = 0; i < 3; i++) {
4395d20e665cSRicardo M. Correia 		if (tmp_cb == NULL)
4396d20e665cSRicardo M. Correia 			list_insert_head(&zcl.zcl_callbacks, cb_data[i]);
4397d20e665cSRicardo M. Correia 		else
4398d20e665cSRicardo M. Correia 			list_insert_after(&zcl.zcl_callbacks, tmp_cb,
4399d20e665cSRicardo M. Correia 			    cb_data[i]);
4400d20e665cSRicardo M. Correia 
4401d20e665cSRicardo M. Correia 		cb_data[i]->zcd_added = B_TRUE;
4402d20e665cSRicardo M. Correia 		VERIFY(!cb_data[i]->zcd_called);
4403d20e665cSRicardo M. Correia 
4404d20e665cSRicardo M. Correia 		tmp_cb = cb_data[i];
4405d20e665cSRicardo M. Correia 	}
4406d20e665cSRicardo M. Correia 
4407d20e665cSRicardo M. Correia 	(void) mutex_unlock(&zcl.zcl_callbacks_lock);
4408d20e665cSRicardo M. Correia 
4409d20e665cSRicardo M. Correia 	dmu_tx_commit(tx);
4410d20e665cSRicardo M. Correia }
4411d20e665cSRicardo M. Correia 
4412b24ab676SJeff Bonwick /* ARGSUSED */
4413fa9e4066Sahrens void
4414b24ab676SJeff Bonwick ztest_dsl_prop_get_set(ztest_ds_t *zd, uint64_t id)
4415fa9e4066Sahrens {
4416b24ab676SJeff Bonwick 	zfs_prop_t proplist[] = {
4417b24ab676SJeff Bonwick 		ZFS_PROP_CHECKSUM,
4418b24ab676SJeff Bonwick 		ZFS_PROP_COMPRESSION,
4419b24ab676SJeff Bonwick 		ZFS_PROP_COPIES,
4420b24ab676SJeff Bonwick 		ZFS_PROP_DEDUP
4421b24ab676SJeff Bonwick 	};
4422fa9e4066Sahrens 
4423420dfc95SChris Siden 	(void) rw_rdlock(&ztest_name_lock);
4424fa9e4066Sahrens 
4425b24ab676SJeff Bonwick 	for (int p = 0; p < sizeof (proplist) / sizeof (proplist[0]); p++)
4426b24ab676SJeff Bonwick 		(void) ztest_dsl_prop_set_uint64(zd->zd_name, proplist[p],
4427b24ab676SJeff Bonwick 		    ztest_random_dsl_prop(proplist[p]), (int)ztest_random(2));
4428fa9e4066Sahrens 
4429420dfc95SChris Siden 	(void) rw_unlock(&ztest_name_lock);
4430b24ab676SJeff Bonwick }
4431fa9e4066Sahrens 
4432b24ab676SJeff Bonwick /* ARGSUSED */
4433b24ab676SJeff Bonwick void
4434b24ab676SJeff Bonwick ztest_spa_prop_get_set(ztest_ds_t *zd, uint64_t id)
4435b24ab676SJeff Bonwick {
4436b24ab676SJeff Bonwick 	nvlist_t *props = NULL;
4437ea8dc4b6Seschrock 
4438420dfc95SChris Siden 	(void) rw_rdlock(&ztest_name_lock);
4439ea8dc4b6Seschrock 
4440420dfc95SChris Siden 	(void) ztest_spa_prop_set_uint64(ZPOOL_PROP_DEDUPDITTO,
4441b24ab676SJeff Bonwick 	    ZIO_DEDUPDITTO_MIN + ztest_random(ZIO_DEDUPDITTO_MIN));
4442fa9e4066Sahrens 
4443*fb09f5aaSMadhav Suresh 	VERIFY0(spa_prop_get(ztest_spa, &props));
4444fa9e4066Sahrens 
4445420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 6)
4446b24ab676SJeff Bonwick 		dump_nvlist(props, 4);
4447fa9e4066Sahrens 
4448b24ab676SJeff Bonwick 	nvlist_free(props);
4449fa9e4066Sahrens 
4450420dfc95SChris Siden 	(void) rw_unlock(&ztest_name_lock);
4451fa9e4066Sahrens }
4452fa9e4066Sahrens 
44535c987a37SChris Kirby /*
44545c987a37SChris Kirby  * Test snapshot hold/release and deferred destroy.
44555c987a37SChris Kirby  */
44565c987a37SChris Kirby void
4457b24ab676SJeff Bonwick ztest_dmu_snapshot_hold(ztest_ds_t *zd, uint64_t id)
44585c987a37SChris Kirby {
44595c987a37SChris Kirby 	int error;
4460b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
44615c987a37SChris Kirby 	objset_t *origin;
44625c987a37SChris Kirby 	char snapname[100];
44635c987a37SChris Kirby 	char fullname[100];
44645c987a37SChris Kirby 	char clonename[100];
44655c987a37SChris Kirby 	char tag[100];
44665c987a37SChris Kirby 	char osname[MAXNAMELEN];
44675c987a37SChris Kirby 
4468420dfc95SChris Siden 	(void) rw_rdlock(&ztest_name_lock);
44695c987a37SChris Kirby 
44705c987a37SChris Kirby 	dmu_objset_name(os, osname);
44715c987a37SChris Kirby 
4472b24ab676SJeff Bonwick 	(void) snprintf(snapname, 100, "sh1_%llu", id);
44735c987a37SChris Kirby 	(void) snprintf(fullname, 100, "%s@%s", osname, snapname);
4474b24ab676SJeff Bonwick 	(void) snprintf(clonename, 100, "%s/ch1_%llu", osname, id);
4475b24ab676SJeff Bonwick 	(void) snprintf(tag, 100, "%tag_%llu", id);
44765c987a37SChris Kirby 
44775c987a37SChris Kirby 	/*
44785c987a37SChris Kirby 	 * Clean up from any previous run.
44795c987a37SChris Kirby 	 */
44805c987a37SChris Kirby 	(void) dmu_objset_destroy(clonename, B_FALSE);
44815c987a37SChris Kirby 	(void) dsl_dataset_user_release(osname, snapname, tag, B_FALSE);
44825c987a37SChris Kirby 	(void) dmu_objset_destroy(fullname, B_FALSE);
44835c987a37SChris Kirby 
44845c987a37SChris Kirby 	/*
44855c987a37SChris Kirby 	 * Create snapshot, clone it, mark snap for deferred destroy,
44865c987a37SChris Kirby 	 * destroy clone, verify snap was also destroyed.
44875c987a37SChris Kirby 	 */
44884445fffbSMatthew Ahrens 	error = dmu_objset_snapshot_one(osname, snapname);
44895c987a37SChris Kirby 	if (error) {
44905c987a37SChris Kirby 		if (error == ENOSPC) {
44915c987a37SChris Kirby 			ztest_record_enospc("dmu_objset_snapshot");
44925c987a37SChris Kirby 			goto out;
44935c987a37SChris Kirby 		}
44945c987a37SChris Kirby 		fatal(0, "dmu_objset_snapshot(%s) = %d", fullname, error);
44955c987a37SChris Kirby 	}
44965c987a37SChris Kirby 
44975c987a37SChris Kirby 	error = dmu_objset_hold(fullname, FTAG, &origin);
44985c987a37SChris Kirby 	if (error)
44995c987a37SChris Kirby 		fatal(0, "dmu_objset_hold(%s) = %d", fullname, error);
45005c987a37SChris Kirby 
45015c987a37SChris Kirby 	error = dmu_objset_clone(clonename, dmu_objset_ds(origin), 0);
45025c987a37SChris Kirby 	dmu_objset_rele(origin, FTAG);
45035c987a37SChris Kirby 	if (error) {
45045c987a37SChris Kirby 		if (error == ENOSPC) {
45055c987a37SChris Kirby 			ztest_record_enospc("dmu_objset_clone");
45065c987a37SChris Kirby 			goto out;
45075c987a37SChris Kirby 		}
45085c987a37SChris Kirby 		fatal(0, "dmu_objset_clone(%s) = %d", clonename, error);
45095c987a37SChris Kirby 	}
45105c987a37SChris Kirby 
45115c987a37SChris Kirby 	error = dmu_objset_destroy(fullname, B_TRUE);
45125c987a37SChris Kirby 	if (error) {
45135c987a37SChris Kirby 		fatal(0, "dmu_objset_destroy(%s, B_TRUE) = %d",
45145c987a37SChris Kirby 		    fullname, error);
45155c987a37SChris Kirby 	}
45165c987a37SChris Kirby 
45175c987a37SChris Kirby 	error = dmu_objset_destroy(clonename, B_FALSE);
45185c987a37SChris Kirby 	if (error)
45195c987a37SChris Kirby 		fatal(0, "dmu_objset_destroy(%s) = %d", clonename, error);
45205c987a37SChris Kirby 
45215c987a37SChris Kirby 	error = dmu_objset_hold(fullname, FTAG, &origin);
45225c987a37SChris Kirby 	if (error != ENOENT)
45235c987a37SChris Kirby 		fatal(0, "dmu_objset_hold(%s) = %d", fullname, error);
45245c987a37SChris Kirby 
45255c987a37SChris Kirby 	/*
45265c987a37SChris Kirby 	 * Create snapshot, add temporary hold, verify that we can't
45275c987a37SChris Kirby 	 * destroy a held snapshot, mark for deferred destroy,
45285c987a37SChris Kirby 	 * release hold, verify snapshot was destroyed.
45295c987a37SChris Kirby 	 */
45304445fffbSMatthew Ahrens 	error = dmu_objset_snapshot_one(osname, snapname);
45315c987a37SChris Kirby 	if (error) {
45325c987a37SChris Kirby 		if (error == ENOSPC) {
45335c987a37SChris Kirby 			ztest_record_enospc("dmu_objset_snapshot");
45345c987a37SChris Kirby 			goto out;
45355c987a37SChris Kirby 		}
45365c987a37SChris Kirby 		fatal(0, "dmu_objset_snapshot(%s) = %d", fullname, error);
45375c987a37SChris Kirby 	}
45385c987a37SChris Kirby 
4539c99e4bdcSChris Kirby 	error = dsl_dataset_user_hold(osname, snapname, tag, B_FALSE,
4540c99e4bdcSChris Kirby 	    B_TRUE, -1);
45415c987a37SChris Kirby 	if (error)
45425c987a37SChris Kirby 		fatal(0, "dsl_dataset_user_hold(%s)", fullname, tag);
45435c987a37SChris Kirby 
45445c987a37SChris Kirby 	error = dmu_objset_destroy(fullname, B_FALSE);
45455c987a37SChris Kirby 	if (error != EBUSY) {
45465c987a37SChris Kirby 		fatal(0, "dmu_objset_destroy(%s, B_FALSE) = %d",
45475c987a37SChris Kirby 		    fullname, error);
45485c987a37SChris Kirby 	}
45495c987a37SChris Kirby 
45505c987a37SChris Kirby 	error = dmu_objset_destroy(fullname, B_TRUE);
45515c987a37SChris Kirby 	if (error) {
45525c987a37SChris Kirby 		fatal(0, "dmu_objset_destroy(%s, B_TRUE) = %d",
45535c987a37SChris Kirby 		    fullname, error);
45545c987a37SChris Kirby 	}
45555c987a37SChris Kirby 
45565c987a37SChris Kirby 	error = dsl_dataset_user_release(osname, snapname, tag, B_FALSE);
45575c987a37SChris Kirby 	if (error)
45585c987a37SChris Kirby 		fatal(0, "dsl_dataset_user_release(%s)", fullname, tag);
45595c987a37SChris Kirby 
45605c987a37SChris Kirby 	VERIFY(dmu_objset_hold(fullname, FTAG, &origin) == ENOENT);
45615c987a37SChris Kirby 
45625c987a37SChris Kirby out:
4563420dfc95SChris Siden 	(void) rw_unlock(&ztest_name_lock);
45645c987a37SChris Kirby }
45655c987a37SChris Kirby 
4566fa9e4066Sahrens /*
4567fa9e4066Sahrens  * Inject random faults into the on-disk data.
4568fa9e4066Sahrens  */
4569b24ab676SJeff Bonwick /* ARGSUSED */
4570fa9e4066Sahrens void
4571b24ab676SJeff Bonwick ztest_fault_inject(ztest_ds_t *zd, uint64_t id)
4572fa9e4066Sahrens {
4573b24ab676SJeff Bonwick 	ztest_shared_t *zs = ztest_shared;
4574420dfc95SChris Siden 	spa_t *spa = ztest_spa;
4575fa9e4066Sahrens 	int fd;
4576fa9e4066Sahrens 	uint64_t offset;
45771195e687SMark J Musante 	uint64_t leaves;
4578fa9e4066Sahrens 	uint64_t bad = 0x1990c0ffeedecade;
4579fa9e4066Sahrens 	uint64_t top, leaf;
4580fa9e4066Sahrens 	char path0[MAXPATHLEN];
4581fa9e4066Sahrens 	char pathrand[MAXPATHLEN];
4582fa9e4066Sahrens 	size_t fsize;
4583fa9e4066Sahrens 	int bshift = SPA_MAXBLOCKSHIFT + 2;	/* don't scrog all labels */
4584fa9e4066Sahrens 	int iters = 1000;
45851195e687SMark J Musante 	int maxfaults;
45861195e687SMark J Musante 	int mirror_save;
4587e14bb325SJeff Bonwick 	vdev_t *vd0 = NULL;
4588ea8dc4b6Seschrock 	uint64_t guid0 = 0;
45898f18d1faSGeorge Wilson 	boolean_t islog = B_FALSE;
4590ea8dc4b6Seschrock 
4591420dfc95SChris Siden 	VERIFY(mutex_lock(&ztest_vdev_lock) == 0);
45921195e687SMark J Musante 	maxfaults = MAXFAULTS();
4593420dfc95SChris Siden 	leaves = MAX(zs->zs_mirrors, 1) * ztest_opts.zo_raidz;
45941195e687SMark J Musante 	mirror_save = zs->zs_mirrors;
4595420dfc95SChris Siden 	VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
45961195e687SMark J Musante 
4597e14bb325SJeff Bonwick 	ASSERT(leaves >= 1);
4598fa9e4066Sahrens 
4599fa9e4066Sahrens 	/*
4600e14bb325SJeff Bonwick 	 * We need SCL_STATE here because we're going to look at vd0->vdev_tsd.
4601fa9e4066Sahrens 	 */
4602e14bb325SJeff Bonwick 	spa_config_enter(spa, SCL_STATE, FTAG, RW_READER);
4603fa9e4066Sahrens 
4604e14bb325SJeff Bonwick 	if (ztest_random(2) == 0) {
4605e14bb325SJeff Bonwick 		/*
4606b24ab676SJeff Bonwick 		 * Inject errors on a normal data device or slog device.
4607e14bb325SJeff Bonwick 		 */
4608b24ab676SJeff Bonwick 		top = ztest_random_vdev_top(spa, B_TRUE);
46091195e687SMark J Musante 		leaf = ztest_random(leaves) + zs->zs_splits;
4610fa9e4066Sahrens 
4611e14bb325SJeff Bonwick 		/*
4612e14bb325SJeff Bonwick 		 * Generate paths to the first leaf in this top-level vdev,
4613e14bb325SJeff Bonwick 		 * and to the random leaf we selected.  We'll induce transient
4614e14bb325SJeff Bonwick 		 * write failures and random online/offline activity on leaf 0,
4615e14bb325SJeff Bonwick 		 * and we'll write random garbage to the randomly chosen leaf.
4616e14bb325SJeff Bonwick 		 */
4617e14bb325SJeff Bonwick 		(void) snprintf(path0, sizeof (path0), ztest_dev_template,
4618420dfc95SChris Siden 		    ztest_opts.zo_dir, ztest_opts.zo_pool,
4619420dfc95SChris Siden 		    top * leaves + zs->zs_splits);
4620e14bb325SJeff Bonwick 		(void) snprintf(pathrand, sizeof (pathrand), ztest_dev_template,
4621420dfc95SChris Siden 		    ztest_opts.zo_dir, ztest_opts.zo_pool,
4622420dfc95SChris Siden 		    top * leaves + leaf);
4623fa9e4066Sahrens 
4624e14bb325SJeff Bonwick 		vd0 = vdev_lookup_by_path(spa->spa_root_vdev, path0);
46258f18d1faSGeorge Wilson 		if (vd0 != NULL && vd0->vdev_top->vdev_islog)
46268f18d1faSGeorge Wilson 			islog = B_TRUE;
46278f18d1faSGeorge Wilson 
4628e14bb325SJeff Bonwick 		if (vd0 != NULL && maxfaults != 1) {
4629e14bb325SJeff Bonwick 			/*
4630e14bb325SJeff Bonwick 			 * Make vd0 explicitly claim to be unreadable,
4631e14bb325SJeff Bonwick 			 * or unwriteable, or reach behind its back
4632e14bb325SJeff Bonwick 			 * and close the underlying fd.  We can do this if
4633e14bb325SJeff Bonwick 			 * maxfaults == 0 because we'll fail and reexecute,
4634e14bb325SJeff Bonwick 			 * and we can do it if maxfaults >= 2 because we'll
4635e14bb325SJeff Bonwick 			 * have enough redundancy.  If maxfaults == 1, the
4636e14bb325SJeff Bonwick 			 * combination of this with injection of random data
4637e14bb325SJeff Bonwick 			 * corruption below exceeds the pool's fault tolerance.
4638e14bb325SJeff Bonwick 			 */
4639e14bb325SJeff Bonwick 			vdev_file_t *vf = vd0->vdev_tsd;
4640e14bb325SJeff Bonwick 
4641e14bb325SJeff Bonwick 			if (vf != NULL && ztest_random(3) == 0) {
4642e14bb325SJeff Bonwick 				(void) close(vf->vf_vnode->v_fd);
4643e14bb325SJeff Bonwick 				vf->vf_vnode->v_fd = -1;
4644e14bb325SJeff Bonwick 			} else if (ztest_random(2) == 0) {
4645e14bb325SJeff Bonwick 				vd0->vdev_cant_read = B_TRUE;
4646e14bb325SJeff Bonwick 			} else {
4647e14bb325SJeff Bonwick 				vd0->vdev_cant_write = B_TRUE;
4648e14bb325SJeff Bonwick 			}
4649e14bb325SJeff Bonwick 			guid0 = vd0->vdev_guid;
4650e14bb325SJeff Bonwick 		}
4651e14bb325SJeff Bonwick 	} else {
4652e14bb325SJeff Bonwick 		/*
4653e14bb325SJeff Bonwick 		 * Inject errors on an l2cache device.
4654e14bb325SJeff Bonwick 		 */
4655e14bb325SJeff Bonwick 		spa_aux_vdev_t *sav = &spa->spa_l2cache;
4656fa9e4066Sahrens 
4657e14bb325SJeff Bonwick 		if (sav->sav_count == 0) {
4658e14bb325SJeff Bonwick 			spa_config_exit(spa, SCL_STATE, FTAG);
4659e14bb325SJeff Bonwick 			return;
4660e14bb325SJeff Bonwick 		}
4661e14bb325SJeff Bonwick 		vd0 = sav->sav_vdevs[ztest_random(sav->sav_count)];
4662ea8dc4b6Seschrock 		guid0 = vd0->vdev_guid;
4663e14bb325SJeff Bonwick 		(void) strcpy(path0, vd0->vdev_path);
4664e14bb325SJeff Bonwick 		(void) strcpy(pathrand, vd0->vdev_path);
4665e14bb325SJeff Bonwick 
4666e14bb325SJeff Bonwick 		leaf = 0;
4667e14bb325SJeff Bonwick 		leaves = 1;
4668e14bb325SJeff Bonwick 		maxfaults = INT_MAX;	/* no limit on cache devices */
4669fa9e4066Sahrens 	}
4670fa9e4066Sahrens 
4671e14bb325SJeff Bonwick 	spa_config_exit(spa, SCL_STATE, FTAG);
4672e14bb325SJeff Bonwick 
4673fa9e4066Sahrens 	/*
46748f18d1faSGeorge Wilson 	 * If we can tolerate two or more faults, or we're dealing
46758f18d1faSGeorge Wilson 	 * with a slog, randomly online/offline vd0.
4676fa9e4066Sahrens 	 */
46778f18d1faSGeorge Wilson 	if ((maxfaults >= 2 || islog) && guid0 != 0) {
46788ad4d6ddSJeff Bonwick 		if (ztest_random(10) < 6) {
46798ad4d6ddSJeff Bonwick 			int flags = (ztest_random(2) == 0 ?
46808ad4d6ddSJeff Bonwick 			    ZFS_OFFLINE_TEMPORARY : 0);
46818f18d1faSGeorge Wilson 
46828f18d1faSGeorge Wilson 			/*
46838f18d1faSGeorge Wilson 			 * We have to grab the zs_name_lock as writer to
46848f18d1faSGeorge Wilson 			 * prevent a race between offlining a slog and
46858f18d1faSGeorge Wilson 			 * destroying a dataset. Offlining the slog will
46868f18d1faSGeorge Wilson 			 * grab a reference on the dataset which may cause
46878f18d1faSGeorge Wilson 			 * dmu_objset_destroy() to fail with EBUSY thus
46888f18d1faSGeorge Wilson 			 * leaving the dataset in an inconsistent state.
46898f18d1faSGeorge Wilson 			 */
46908f18d1faSGeorge Wilson 			if (islog)
4691420dfc95SChris Siden 				(void) rw_wrlock(&ztest_name_lock);
46928f18d1faSGeorge Wilson 
46938ad4d6ddSJeff Bonwick 			VERIFY(vdev_offline(spa, guid0, flags) != EBUSY);
46948f18d1faSGeorge Wilson 
46958f18d1faSGeorge Wilson 			if (islog)
4696420dfc95SChris Siden 				(void) rw_unlock(&ztest_name_lock);
46978ad4d6ddSJeff Bonwick 		} else {
46988ad4d6ddSJeff Bonwick 			(void) vdev_online(spa, guid0, 0, NULL);
46998ad4d6ddSJeff Bonwick 		}
4700fa9e4066Sahrens 	}
4701fa9e4066Sahrens 
47028f18d1faSGeorge Wilson 	if (maxfaults == 0)
47038f18d1faSGeorge Wilson 		return;
47048f18d1faSGeorge Wilson 
4705fa9e4066Sahrens 	/*
4706ea8dc4b6Seschrock 	 * We have at least single-fault tolerance, so inject data corruption.
4707fa9e4066Sahrens 	 */
4708fa9e4066Sahrens 	fd = open(pathrand, O_RDWR);
4709fa9e4066Sahrens 
4710fa9e4066Sahrens 	if (fd == -1)	/* we hit a gap in the device namespace */
4711fa9e4066Sahrens 		return;
4712fa9e4066Sahrens 
4713fa9e4066Sahrens 	fsize = lseek(fd, 0, SEEK_END);
4714fa9e4066Sahrens 
4715fa9e4066Sahrens 	while (--iters != 0) {
4716fa9e4066Sahrens 		offset = ztest_random(fsize / (leaves << bshift)) *
4717fa9e4066Sahrens 		    (leaves << bshift) + (leaf << bshift) +
4718fa9e4066Sahrens 		    (ztest_random(1ULL << (bshift - 1)) & -8ULL);
4719fa9e4066Sahrens 
4720fa9e4066Sahrens 		if (offset >= fsize)
4721fa9e4066Sahrens 			continue;
4722fa9e4066Sahrens 
4723420dfc95SChris Siden 		VERIFY(mutex_lock(&ztest_vdev_lock) == 0);
47241195e687SMark J Musante 		if (mirror_save != zs->zs_mirrors) {
4725420dfc95SChris Siden 			VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
47261195e687SMark J Musante 			(void) close(fd);
47271195e687SMark J Musante 			return;
47281195e687SMark J Musante 		}
4729fa9e4066Sahrens 
4730fa9e4066Sahrens 		if (pwrite(fd, &bad, sizeof (bad), offset) != sizeof (bad))
4731fa9e4066Sahrens 			fatal(1, "can't inject bad word at 0x%llx in %s",
4732fa9e4066Sahrens 			    offset, pathrand);
47331195e687SMark J Musante 
4734420dfc95SChris Siden 		VERIFY(mutex_unlock(&ztest_vdev_lock) == 0);
47351195e687SMark J Musante 
4736420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 7)
47371195e687SMark J Musante 			(void) printf("injected bad word into %s,"
47381195e687SMark J Musante 			    " offset 0x%llx\n", pathrand, (u_longlong_t)offset);
4739fa9e4066Sahrens 	}
4740fa9e4066Sahrens 
4741fa9e4066Sahrens 	(void) close(fd);
4742fa9e4066Sahrens }
4743fa9e4066Sahrens 
4744fa9e4066Sahrens /*
4745b24ab676SJeff Bonwick  * Verify that DDT repair works as expected.
4746fa9e4066Sahrens  */
4747fa9e4066Sahrens void
4748b24ab676SJeff Bonwick ztest_ddt_repair(ztest_ds_t *zd, uint64_t id)
4749fa9e4066Sahrens {
4750b24ab676SJeff Bonwick 	ztest_shared_t *zs = ztest_shared;
4751420dfc95SChris Siden 	spa_t *spa = ztest_spa;
4752b24ab676SJeff Bonwick 	objset_t *os = zd->zd_os;
4753b24ab676SJeff Bonwick 	ztest_od_t od[1];
4754b24ab676SJeff Bonwick 	uint64_t object, blocksize, txg, pattern, psize;
4755b24ab676SJeff Bonwick 	enum zio_checksum checksum = spa_dedup_checksum(spa);
4756b24ab676SJeff Bonwick 	dmu_buf_t *db;
4757b24ab676SJeff Bonwick 	dmu_tx_t *tx;
4758b24ab676SJeff Bonwick 	void *buf;
4759b24ab676SJeff Bonwick 	blkptr_t blk;
4760b24ab676SJeff Bonwick 	int copies = 2 * ZIO_DEDUPDITTO_MIN;
4761fa9e4066Sahrens 
4762b24ab676SJeff Bonwick 	blocksize = ztest_random_blocksize();
4763b24ab676SJeff Bonwick 	blocksize = MIN(blocksize, 2048);	/* because we write so many */
4764fa9e4066Sahrens 
4765b24ab676SJeff Bonwick 	ztest_od_init(&od[0], id, FTAG, 0, DMU_OT_UINT64_OTHER, blocksize, 0);
4766fa9e4066Sahrens 
4767b24ab676SJeff Bonwick 	if (ztest_object_init(zd, od, sizeof (od), B_FALSE) != 0)
4768b24ab676SJeff Bonwick 		return;
4769fa9e4066Sahrens 
4770fa9e4066Sahrens 	/*
4771b24ab676SJeff Bonwick 	 * Take the name lock as writer to prevent anyone else from changing
4772b24ab676SJeff Bonwick 	 * the pool and dataset properies we need to maintain during this test.
4773fa9e4066Sahrens 	 */
4774420dfc95SChris Siden 	(void) rw_wrlock(&ztest_name_lock);
4775fa9e4066Sahrens 
4776b24ab676SJeff Bonwick 	if (ztest_dsl_prop_set_uint64(zd->zd_name, ZFS_PROP_DEDUP, checksum,
4777b24ab676SJeff Bonwick 	    B_FALSE) != 0 ||
4778b24ab676SJeff Bonwick 	    ztest_dsl_prop_set_uint64(zd->zd_name, ZFS_PROP_COPIES, 1,
4779b24ab676SJeff Bonwick 	    B_FALSE) != 0) {
4780420dfc95SChris Siden 		(void) rw_unlock(&ztest_name_lock);
4781b24ab676SJeff Bonwick 		return;
4782b24ab676SJeff Bonwick 	}
4783b24ab676SJeff Bonwick 
4784b24ab676SJeff Bonwick 	object = od[0].od_object;
4785b24ab676SJeff Bonwick 	blocksize = od[0].od_blocksize;
4786e9103aaeSGarrett D'Amore 	pattern = zs->zs_guid ^ dmu_objset_fsid_guid(os);
4787b24ab676SJeff Bonwick 
4788b24ab676SJeff Bonwick 	ASSERT(object != 0);
4789b24ab676SJeff Bonwick 
4790b24ab676SJeff Bonwick 	tx = dmu_tx_create(os);
4791b24ab676SJeff Bonwick 	dmu_tx_hold_write(tx, object, 0, copies * blocksize);
4792b24ab676SJeff Bonwick 	txg = ztest_tx_assign(tx, TXG_WAIT, FTAG);
4793b24ab676SJeff Bonwick 	if (txg == 0) {
4794420dfc95SChris Siden 		(void) rw_unlock(&ztest_name_lock);
4795b24ab676SJeff Bonwick 		return;
4796b24ab676SJeff Bonwick 	}
4797fa9e4066Sahrens 
4798fa9e4066Sahrens 	/*
4799b24ab676SJeff Bonwick 	 * Write all the copies of our block.
4800fa9e4066Sahrens 	 */
4801b24ab676SJeff Bonwick 	for (int i = 0; i < copies; i++) {
4802b24ab676SJeff Bonwick 		uint64_t offset = i * blocksize;
480347cb52daSJeff Bonwick 		VERIFY(dmu_buf_hold(os, object, offset, FTAG, &db,
480447cb52daSJeff Bonwick 		    DMU_READ_NO_PREFETCH) == 0);
4805b24ab676SJeff Bonwick 		ASSERT(db->db_offset == offset);
4806b24ab676SJeff Bonwick 		ASSERT(db->db_size == blocksize);
4807b24ab676SJeff Bonwick 		ASSERT(ztest_pattern_match(db->db_data, db->db_size, pattern) ||
4808b24ab676SJeff Bonwick 		    ztest_pattern_match(db->db_data, db->db_size, 0ULL));
4809b24ab676SJeff Bonwick 		dmu_buf_will_fill(db, tx);
4810b24ab676SJeff Bonwick 		ztest_pattern_set(db->db_data, db->db_size, pattern);
4811b24ab676SJeff Bonwick 		dmu_buf_rele(db, FTAG);
4812b24ab676SJeff Bonwick 	}
4813fa9e4066Sahrens 
4814b24ab676SJeff Bonwick 	dmu_tx_commit(tx);
4815b24ab676SJeff Bonwick 	txg_wait_synced(spa_get_dsl(spa), txg);
4816fa9e4066Sahrens 
4817fa9e4066Sahrens 	/*
4818b24ab676SJeff Bonwick 	 * Find out what block we got.
4819fa9e4066Sahrens 	 */
482047cb52daSJeff Bonwick 	VERIFY(dmu_buf_hold(os, object, 0, FTAG, &db,
482147cb52daSJeff Bonwick 	    DMU_READ_NO_PREFETCH) == 0);
4822b24ab676SJeff Bonwick 	blk = *((dmu_buf_impl_t *)db)->db_blkptr;
4823b24ab676SJeff Bonwick 	dmu_buf_rele(db, FTAG);
4824fa9e4066Sahrens 
4825fa9e4066Sahrens 	/*
4826b24ab676SJeff Bonwick 	 * Damage the block.  Dedup-ditto will save us when we read it later.
4827fa9e4066Sahrens 	 */
4828b24ab676SJeff Bonwick 	psize = BP_GET_PSIZE(&blk);
4829b24ab676SJeff Bonwick 	buf = zio_buf_alloc(psize);
4830b24ab676SJeff Bonwick 	ztest_pattern_set(buf, psize, ~pattern);
4831fa9e4066Sahrens 
4832b24ab676SJeff Bonwick 	(void) zio_wait(zio_rewrite(NULL, spa, 0, &blk,
4833b24ab676SJeff Bonwick 	    buf, psize, NULL, NULL, ZIO_PRIORITY_SYNC_WRITE,
4834b24ab676SJeff Bonwick 	    ZIO_FLAG_CANFAIL | ZIO_FLAG_INDUCE_DAMAGE, NULL));
4835fa9e4066Sahrens 
4836b24ab676SJeff Bonwick 	zio_buf_free(buf, psize);
4837fa9e4066Sahrens 
4838420dfc95SChris Siden 	(void) rw_unlock(&ztest_name_lock);
4839fa9e4066Sahrens }
4840fa9e4066Sahrens 
4841fa9e4066Sahrens /*
4842b24ab676SJeff Bonwick  * Scrub the pool.
4843fa9e4066Sahrens  */
4844b24ab676SJeff Bonwick /* ARGSUSED */
4845b24ab676SJeff Bonwick void
4846b24ab676SJeff Bonwick ztest_scrub(ztest_ds_t *zd, uint64_t id)
4847fa9e4066Sahrens {
4848420dfc95SChris Siden 	spa_t *spa = ztest_spa;
4849fa9e4066Sahrens 
48503f9d6ad7SLin Ling 	(void) spa_scan(spa, POOL_SCAN_SCRUB);
4851b24ab676SJeff Bonwick 	(void) poll(NULL, 0, 100); /* wait a moment, then force a restart */
48523f9d6ad7SLin Ling 	(void) spa_scan(spa, POOL_SCAN_SCRUB);
4853b24ab676SJeff Bonwick }
4854fa9e4066Sahrens 
4855e9103aaeSGarrett D'Amore /*
4856e9103aaeSGarrett D'Amore  * Change the guid for the pool.
4857e9103aaeSGarrett D'Amore  */
4858e9103aaeSGarrett D'Amore /* ARGSUSED */
4859e9103aaeSGarrett D'Amore void
4860e9103aaeSGarrett D'Amore ztest_reguid(ztest_ds_t *zd, uint64_t id)
4861e9103aaeSGarrett D'Amore {
4862420dfc95SChris Siden 	spa_t *spa = ztest_spa;
4863e9103aaeSGarrett D'Amore 	uint64_t orig, load;
4864dfbb9432SGeorge Wilson 	int error;
4865e9103aaeSGarrett D'Amore 
4866e9103aaeSGarrett D'Amore 	orig = spa_guid(spa);
4867e9103aaeSGarrett D'Amore 	load = spa_load_guid(spa);
4868dfbb9432SGeorge Wilson 
4869dfbb9432SGeorge Wilson 	(void) rw_wrlock(&ztest_name_lock);
4870dfbb9432SGeorge Wilson 	error = spa_change_guid(spa);
4871dfbb9432SGeorge Wilson 	(void) rw_unlock(&ztest_name_lock);
4872dfbb9432SGeorge Wilson 
4873dfbb9432SGeorge Wilson 	if (error != 0)
4874e9103aaeSGarrett D'Amore 		return;
4875e9103aaeSGarrett D'Amore 
4876420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 3) {
4877e9103aaeSGarrett D'Amore 		(void) printf("Changed guid old %llu -> %llu\n",
4878e9103aaeSGarrett D'Amore 		    (u_longlong_t)orig, (u_longlong_t)spa_guid(spa));
4879e9103aaeSGarrett D'Amore 	}
4880e9103aaeSGarrett D'Amore 
4881e9103aaeSGarrett D'Amore 	VERIFY3U(orig, !=, spa_guid(spa));
4882e9103aaeSGarrett D'Amore 	VERIFY3U(load, ==, spa_load_guid(spa));
4883e9103aaeSGarrett D'Amore }
4884e9103aaeSGarrett D'Amore 
4885b24ab676SJeff Bonwick /*
4886b24ab676SJeff Bonwick  * Rename the pool to a different name and then rename it back.
4887b24ab676SJeff Bonwick  */
4888b24ab676SJeff Bonwick /* ARGSUSED */
4889b24ab676SJeff Bonwick void
4890b24ab676SJeff Bonwick ztest_spa_rename(ztest_ds_t *zd, uint64_t id)
4891b24ab676SJeff Bonwick {
4892b24ab676SJeff Bonwick 	char *oldname, *newname;
4893b24ab676SJeff Bonwick 	spa_t *spa;
4894fa9e4066Sahrens 
4895420dfc95SChris Siden 	(void) rw_wrlock(&ztest_name_lock);
4896fa9e4066Sahrens 
4897420dfc95SChris Siden 	oldname = ztest_opts.zo_pool;
4898b24ab676SJeff Bonwick 	newname = umem_alloc(strlen(oldname) + 5, UMEM_NOFAIL);
4899b24ab676SJeff Bonwick 	(void) strcpy(newname, oldname);
4900b24ab676SJeff Bonwick 	(void) strcat(newname, "_tmp");
4901fa9e4066Sahrens 
4902fa9e4066Sahrens 	/*
4903b24ab676SJeff Bonwick 	 * Do the rename
4904fa9e4066Sahrens 	 */
4905b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_rename(oldname, newname));
4906fa9e4066Sahrens 
4907fa9e4066Sahrens 	/*
4908b24ab676SJeff Bonwick 	 * Try to open it under the old name, which shouldn't exist
4909fa9e4066Sahrens 	 */
4910b24ab676SJeff Bonwick 	VERIFY3U(ENOENT, ==, spa_open(oldname, &spa, FTAG));
4911fa9e4066Sahrens 
4912fa9e4066Sahrens 	/*
4913b24ab676SJeff Bonwick 	 * Open it under the new name and make sure it's still the same spa_t.
4914fa9e4066Sahrens 	 */
4915b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_open(newname, &spa, FTAG));
4916fa9e4066Sahrens 
4917420dfc95SChris Siden 	ASSERT(spa == ztest_spa);
4918b24ab676SJeff Bonwick 	spa_close(spa, FTAG);
4919fa9e4066Sahrens 
4920b24ab676SJeff Bonwick 	/*
4921b24ab676SJeff Bonwick 	 * Rename it back to the original
4922b24ab676SJeff Bonwick 	 */
4923b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_rename(newname, oldname));
4924fa9e4066Sahrens 
4925fa9e4066Sahrens 	/*
4926b24ab676SJeff Bonwick 	 * Make sure it can still be opened
4927fa9e4066Sahrens 	 */
4928b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_open(oldname, &spa, FTAG));
4929fa9e4066Sahrens 
4930420dfc95SChris Siden 	ASSERT(spa == ztest_spa);
4931b24ab676SJeff Bonwick 	spa_close(spa, FTAG);
4932fa9e4066Sahrens 
4933b24ab676SJeff Bonwick 	umem_free(newname, strlen(newname) + 1);
4934b24ab676SJeff Bonwick 
4935420dfc95SChris Siden 	(void) rw_unlock(&ztest_name_lock);
4936fa9e4066Sahrens }
4937fa9e4066Sahrens 
4938b24ab676SJeff Bonwick /*
4939b24ab676SJeff Bonwick  * Verify pool integrity by running zdb.
4940b24ab676SJeff Bonwick  */
4941fa9e4066Sahrens static void
4942b24ab676SJeff Bonwick ztest_run_zdb(char *pool)
4943fa9e4066Sahrens {
4944fa9e4066Sahrens 	int status;
4945fa9e4066Sahrens 	char zdb[MAXPATHLEN + MAXNAMELEN + 20];
4946fa9e4066Sahrens 	char zbuf[1024];
4947fa9e4066Sahrens 	char *bin;
49488654d025Sperrin 	char *ztest;
49498654d025Sperrin 	char *isa;
49508654d025Sperrin 	int isalen;
4951fa9e4066Sahrens 	FILE *fp;
4952fa9e4066Sahrens 
4953fa9e4066Sahrens 	(void) realpath(getexecname(), zdb);
4954fa9e4066Sahrens 
4955fa9e4066Sahrens 	/* zdb lives in /usr/sbin, while ztest lives in /usr/bin */
4956fa9e4066Sahrens 	bin = strstr(zdb, "/usr/bin/");
49578654d025Sperrin 	ztest = strstr(bin, "/ztest");
49588654d025Sperrin 	isa = bin + 8;
49598654d025Sperrin 	isalen = ztest - isa;
49608654d025Sperrin 	isa = strdup(isa);
4961fa9e4066Sahrens 	/* LINTED */
49623a57275aSck 	(void) sprintf(bin,
4963f0ba89beSJeff Bonwick 	    "/usr/sbin%.*s/zdb -bcc%s%s -U %s %s",
49648654d025Sperrin 	    isalen,
49658654d025Sperrin 	    isa,
4966420dfc95SChris Siden 	    ztest_opts.zo_verbose >= 3 ? "s" : "",
4967420dfc95SChris Siden 	    ztest_opts.zo_verbose >= 4 ? "v" : "",
4968f0ba89beSJeff Bonwick 	    spa_config_path,
496988b7b0f2SMatthew Ahrens 	    pool);
49708654d025Sperrin 	free(isa);
4971fa9e4066Sahrens 
4972420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 5)
4973fa9e4066Sahrens 		(void) printf("Executing %s\n", strstr(zdb, "zdb "));
4974fa9e4066Sahrens 
4975fa9e4066Sahrens 	fp = popen(zdb, "r");
4976fa9e4066Sahrens 
4977fa9e4066Sahrens 	while (fgets(zbuf, sizeof (zbuf), fp) != NULL)
4978420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 3)
4979fa9e4066Sahrens 			(void) printf("%s", zbuf);
4980fa9e4066Sahrens 
4981fa9e4066Sahrens 	status = pclose(fp);
4982fa9e4066Sahrens 
4983fa9e4066Sahrens 	if (status == 0)
4984fa9e4066Sahrens 		return;
4985fa9e4066Sahrens 
4986fa9e4066Sahrens 	ztest_dump_core = 0;
4987fa9e4066Sahrens 	if (WIFEXITED(status))
4988fa9e4066Sahrens 		fatal(0, "'%s' exit code %d", zdb, WEXITSTATUS(status));
4989fa9e4066Sahrens 	else
4990fa9e4066Sahrens 		fatal(0, "'%s' died with signal %d", zdb, WTERMSIG(status));
4991fa9e4066Sahrens }
4992fa9e4066Sahrens 
4993fa9e4066Sahrens static void
4994fa9e4066Sahrens ztest_walk_pool_directory(char *header)
4995fa9e4066Sahrens {
4996fa9e4066Sahrens 	spa_t *spa = NULL;
4997fa9e4066Sahrens 
4998420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 6)
4999fa9e4066Sahrens 		(void) printf("%s\n", header);
5000fa9e4066Sahrens 
5001fa9e4066Sahrens 	mutex_enter(&spa_namespace_lock);
5002fa9e4066Sahrens 	while ((spa = spa_next(spa)) != NULL)
5003420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 6)
5004fa9e4066Sahrens 			(void) printf("\t%s\n", spa_name(spa));
5005fa9e4066Sahrens 	mutex_exit(&spa_namespace_lock);
5006fa9e4066Sahrens }
5007fa9e4066Sahrens 
5008fa9e4066Sahrens static void
5009fa9e4066Sahrens ztest_spa_import_export(char *oldname, char *newname)
5010fa9e4066Sahrens {
50118ad4d6ddSJeff Bonwick 	nvlist_t *config, *newconfig;
5012fa9e4066Sahrens 	uint64_t pool_guid;
5013fa9e4066Sahrens 	spa_t *spa;
5014fa9e4066Sahrens 
5015420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 4) {
5016fa9e4066Sahrens 		(void) printf("import/export: old = %s, new = %s\n",
5017fa9e4066Sahrens 		    oldname, newname);
5018fa9e4066Sahrens 	}
5019fa9e4066Sahrens 
5020fa9e4066Sahrens 	/*
5021fa9e4066Sahrens 	 * Clean up from previous runs.
5022fa9e4066Sahrens 	 */
5023fa9e4066Sahrens 	(void) spa_destroy(newname);
5024fa9e4066Sahrens 
5025fa9e4066Sahrens 	/*
5026fa9e4066Sahrens 	 * Get the pool's configuration and guid.
5027fa9e4066Sahrens 	 */
5028b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_open(oldname, &spa, FTAG));
5029fa9e4066Sahrens 
50308ad4d6ddSJeff Bonwick 	/*
50318ad4d6ddSJeff Bonwick 	 * Kick off a scrub to tickle scrub/export races.
50328ad4d6ddSJeff Bonwick 	 */
50338ad4d6ddSJeff Bonwick 	if (ztest_random(2) == 0)
50343f9d6ad7SLin Ling 		(void) spa_scan(spa, POOL_SCAN_SCRUB);
50358ad4d6ddSJeff Bonwick 
5036fa9e4066Sahrens 	pool_guid = spa_guid(spa);
5037fa9e4066Sahrens 	spa_close(spa, FTAG);
5038fa9e4066Sahrens 
5039fa9e4066Sahrens 	ztest_walk_pool_directory("pools before export");
5040fa9e4066Sahrens 
5041fa9e4066Sahrens 	/*
5042fa9e4066Sahrens 	 * Export it.
5043fa9e4066Sahrens 	 */
5044b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_export(oldname, &config, B_FALSE, B_FALSE));
5045fa9e4066Sahrens 
5046fa9e4066Sahrens 	ztest_walk_pool_directory("pools after export");
5047fa9e4066Sahrens 
50488ad4d6ddSJeff Bonwick 	/*
50498ad4d6ddSJeff Bonwick 	 * Try to import it.
50508ad4d6ddSJeff Bonwick 	 */
50518ad4d6ddSJeff Bonwick 	newconfig = spa_tryimport(config);
50528ad4d6ddSJeff Bonwick 	ASSERT(newconfig != NULL);
50538ad4d6ddSJeff Bonwick 	nvlist_free(newconfig);
50548ad4d6ddSJeff Bonwick 
5055fa9e4066Sahrens 	/*
5056fa9e4066Sahrens 	 * Import it under the new name.
5057fa9e4066Sahrens 	 */
5058b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_import(newname, config, NULL, 0));
5059fa9e4066Sahrens 
5060fa9e4066Sahrens 	ztest_walk_pool_directory("pools after import");
5061fa9e4066Sahrens 
5062fa9e4066Sahrens 	/*
5063fa9e4066Sahrens 	 * Try to import it again -- should fail with EEXIST.
5064fa9e4066Sahrens 	 */
50654b964adaSGeorge Wilson 	VERIFY3U(EEXIST, ==, spa_import(newname, config, NULL, 0));
5066fa9e4066Sahrens 
5067fa9e4066Sahrens 	/*
5068fa9e4066Sahrens 	 * Try to import it under a different name -- should fail with EEXIST.
5069fa9e4066Sahrens 	 */
50704b964adaSGeorge Wilson 	VERIFY3U(EEXIST, ==, spa_import(oldname, config, NULL, 0));
5071fa9e4066Sahrens 
5072fa9e4066Sahrens 	/*
5073fa9e4066Sahrens 	 * Verify that the pool is no longer visible under the old name.
5074fa9e4066Sahrens 	 */
5075b24ab676SJeff Bonwick 	VERIFY3U(ENOENT, ==, spa_open(oldname, &spa, FTAG));
5076fa9e4066Sahrens 
5077fa9e4066Sahrens 	/*
5078fa9e4066Sahrens 	 * Verify that we can open and close the pool using the new name.
5079fa9e4066Sahrens 	 */
5080b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_open(newname, &spa, FTAG));
5081fa9e4066Sahrens 	ASSERT(pool_guid == spa_guid(spa));
5082fa9e4066Sahrens 	spa_close(spa, FTAG);
5083fa9e4066Sahrens 
5084fa9e4066Sahrens 	nvlist_free(config);
5085fa9e4066Sahrens }
5086fa9e4066Sahrens 
50878ad4d6ddSJeff Bonwick static void
50888ad4d6ddSJeff Bonwick ztest_resume(spa_t *spa)
50898ad4d6ddSJeff Bonwick {
5090420dfc95SChris Siden 	if (spa_suspended(spa) && ztest_opts.zo_verbose >= 6)
5091b24ab676SJeff Bonwick 		(void) printf("resuming from suspended state\n");
5092b24ab676SJeff Bonwick 	spa_vdev_state_enter(spa, SCL_NONE);
5093b24ab676SJeff Bonwick 	vdev_clear(spa, NULL);
5094b24ab676SJeff Bonwick 	(void) spa_vdev_state_exit(spa, NULL, 0);
5095b24ab676SJeff Bonwick 	(void) zio_resume(spa);
50968ad4d6ddSJeff Bonwick }
50978ad4d6ddSJeff Bonwick 
50980a4e9518Sgw static void *
50998ad4d6ddSJeff Bonwick ztest_resume_thread(void *arg)
51000a4e9518Sgw {
5101e14bb325SJeff Bonwick 	spa_t *spa = arg;
51020a4e9518Sgw 
51030a4e9518Sgw 	while (!ztest_exiting) {
5104b24ab676SJeff Bonwick 		if (spa_suspended(spa))
5105b24ab676SJeff Bonwick 			ztest_resume(spa);
5106b24ab676SJeff Bonwick 		(void) poll(NULL, 0, 100);
51070a4e9518Sgw 	}
51080a4e9518Sgw 	return (NULL);
51090a4e9518Sgw }
51100a4e9518Sgw 
5111b24ab676SJeff Bonwick static void *
5112b24ab676SJeff Bonwick ztest_deadman_thread(void *arg)
5113b24ab676SJeff Bonwick {
5114b24ab676SJeff Bonwick 	ztest_shared_t *zs = arg;
5115b24ab676SJeff Bonwick 	int grace = 300;
5116b24ab676SJeff Bonwick 	hrtime_t delta;
5117b24ab676SJeff Bonwick 
5118b24ab676SJeff Bonwick 	delta = (zs->zs_thread_stop - zs->zs_thread_start) / NANOSEC + grace;
5119b24ab676SJeff Bonwick 
5120b24ab676SJeff Bonwick 	(void) poll(NULL, 0, (int)(1000 * delta));
5121b24ab676SJeff Bonwick 
5122b24ab676SJeff Bonwick 	fatal(0, "failed to complete within %d seconds of deadline", grace);
5123b24ab676SJeff Bonwick 
5124b24ab676SJeff Bonwick 	return (NULL);
5125b24ab676SJeff Bonwick }
5126b24ab676SJeff Bonwick 
5127b24ab676SJeff Bonwick static void
5128420dfc95SChris Siden ztest_execute(int test, ztest_info_t *zi, uint64_t id)
5129b24ab676SJeff Bonwick {
5130420dfc95SChris Siden 	ztest_ds_t *zd = &ztest_ds[id % ztest_opts.zo_datasets];
5131420dfc95SChris Siden 	ztest_shared_callstate_t *zc = ZTEST_GET_SHARED_CALLSTATE(test);
5132b24ab676SJeff Bonwick 	hrtime_t functime = gethrtime();
5133b24ab676SJeff Bonwick 
5134b24ab676SJeff Bonwick 	for (int i = 0; i < zi->zi_iters; i++)
5135b24ab676SJeff Bonwick 		zi->zi_func(zd, id);
5136b24ab676SJeff Bonwick 
5137b24ab676SJeff Bonwick 	functime = gethrtime() - functime;
5138b24ab676SJeff Bonwick 
5139420dfc95SChris Siden 	atomic_add_64(&zc->zc_count, 1);
5140420dfc95SChris Siden 	atomic_add_64(&zc->zc_time, functime);
5141b24ab676SJeff Bonwick 
5142420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 4) {
5143b24ab676SJeff Bonwick 		Dl_info dli;
5144b24ab676SJeff Bonwick 		(void) dladdr((void *)zi->zi_func, &dli);
5145b24ab676SJeff Bonwick 		(void) printf("%6.2f sec in %s\n",
5146b24ab676SJeff Bonwick 		    (double)functime / NANOSEC, dli.dli_sname);
5147b24ab676SJeff Bonwick 	}
5148b24ab676SJeff Bonwick }
5149b24ab676SJeff Bonwick 
5150fa9e4066Sahrens static void *
5151fa9e4066Sahrens ztest_thread(void *arg)
5152fa9e4066Sahrens {
5153420dfc95SChris Siden 	int rand;
5154b24ab676SJeff Bonwick 	uint64_t id = (uintptr_t)arg;
5155fa9e4066Sahrens 	ztest_shared_t *zs = ztest_shared;
5156b24ab676SJeff Bonwick 	uint64_t call_next;
5157b24ab676SJeff Bonwick 	hrtime_t now;
5158fa9e4066Sahrens 	ztest_info_t *zi;
5159420dfc95SChris Siden 	ztest_shared_callstate_t *zc;
5160fa9e4066Sahrens 
5161b24ab676SJeff Bonwick 	while ((now = gethrtime()) < zs->zs_thread_stop) {
5162fa9e4066Sahrens 		/*
5163fa9e4066Sahrens 		 * See if it's time to force a crash.
5164fa9e4066Sahrens 		 */
5165b24ab676SJeff Bonwick 		if (now > zs->zs_thread_kill)
5166b24ab676SJeff Bonwick 			ztest_kill(zs);
5167fa9e4066Sahrens 
5168fa9e4066Sahrens 		/*
5169b24ab676SJeff Bonwick 		 * If we're getting ENOSPC with some regularity, stop.
5170fa9e4066Sahrens 		 */
5171b24ab676SJeff Bonwick 		if (zs->zs_enospc_count > 10)
5172b24ab676SJeff Bonwick 			break;
5173fa9e4066Sahrens 
5174fa9e4066Sahrens 		/*
5175b24ab676SJeff Bonwick 		 * Pick a random function to execute.
5176fa9e4066Sahrens 		 */
5177420dfc95SChris Siden 		rand = ztest_random(ZTEST_FUNCS);
5178420dfc95SChris Siden 		zi = &ztest_info[rand];
5179420dfc95SChris Siden 		zc = ZTEST_GET_SHARED_CALLSTATE(rand);
5180420dfc95SChris Siden 		call_next = zc->zc_next;
5181b24ab676SJeff Bonwick 
5182b24ab676SJeff Bonwick 		if (now >= call_next &&
5183420dfc95SChris Siden 		    atomic_cas_64(&zc->zc_next, call_next, call_next +
5184420dfc95SChris Siden 		    ztest_random(2 * zi->zi_interval[0] + 1)) == call_next) {
5185420dfc95SChris Siden 			ztest_execute(rand, zi, id);
5186420dfc95SChris Siden 		}
5187b24ab676SJeff Bonwick 	}
5188fa9e4066Sahrens 
5189b24ab676SJeff Bonwick 	return (NULL);
5190b24ab676SJeff Bonwick }
5191fa9e4066Sahrens 
5192b24ab676SJeff Bonwick static void
5193b24ab676SJeff Bonwick ztest_dataset_name(char *dsname, char *pool, int d)
5194b24ab676SJeff Bonwick {
5195b24ab676SJeff Bonwick 	(void) snprintf(dsname, MAXNAMELEN, "%s/ds_%d", pool, d);
5196b24ab676SJeff Bonwick }
5197fa9e4066Sahrens 
5198b24ab676SJeff Bonwick static void
5199420dfc95SChris Siden ztest_dataset_destroy(int d)
5200b24ab676SJeff Bonwick {
5201b24ab676SJeff Bonwick 	char name[MAXNAMELEN];
5202fa9e4066Sahrens 
5203420dfc95SChris Siden 	ztest_dataset_name(name, ztest_opts.zo_pool, d);
5204fa9e4066Sahrens 
5205420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 3)
5206b24ab676SJeff Bonwick 		(void) printf("Destroying %s to free up space\n", name);
5207fa9e4066Sahrens 
5208b24ab676SJeff Bonwick 	/*
5209b24ab676SJeff Bonwick 	 * Cleanup any non-standard clones and snapshots.  In general,
5210b24ab676SJeff Bonwick 	 * ztest thread t operates on dataset (t % zopt_datasets),
5211b24ab676SJeff Bonwick 	 * so there may be more than one thing to clean up.
5212b24ab676SJeff Bonwick 	 */
5213420dfc95SChris Siden 	for (int t = d; t < ztest_opts.zo_threads;
5214420dfc95SChris Siden 	    t += ztest_opts.zo_datasets) {
5215b24ab676SJeff Bonwick 		ztest_dsl_dataset_cleanup(name, t);
5216420dfc95SChris Siden 	}
5217fa9e4066Sahrens 
5218b24ab676SJeff Bonwick 	(void) dmu_objset_find(name, ztest_objset_destroy_cb, NULL,
5219b24ab676SJeff Bonwick 	    DS_FIND_SNAPSHOTS | DS_FIND_CHILDREN);
5220b24ab676SJeff Bonwick }
5221b24ab676SJeff Bonwick 
5222b24ab676SJeff Bonwick static void
5223b24ab676SJeff Bonwick ztest_dataset_dirobj_verify(ztest_ds_t *zd)
5224b24ab676SJeff Bonwick {
5225b24ab676SJeff Bonwick 	uint64_t usedobjs, dirobjs, scratch;
5226b24ab676SJeff Bonwick 
5227b24ab676SJeff Bonwick 	/*
5228b24ab676SJeff Bonwick 	 * ZTEST_DIROBJ is the object directory for the entire dataset.
5229b24ab676SJeff Bonwick 	 * Therefore, the number of objects in use should equal the
5230b24ab676SJeff Bonwick 	 * number of ZTEST_DIROBJ entries, +1 for ZTEST_DIROBJ itself.
5231b24ab676SJeff Bonwick 	 * If not, we have an object leak.
5232b24ab676SJeff Bonwick 	 *
5233b24ab676SJeff Bonwick 	 * Note that we can only check this in ztest_dataset_open(),
5234b24ab676SJeff Bonwick 	 * when the open-context and syncing-context values agree.
5235b24ab676SJeff Bonwick 	 * That's because zap_count() returns the open-context value,
5236b24ab676SJeff Bonwick 	 * while dmu_objset_space() returns the rootbp fill count.
5237b24ab676SJeff Bonwick 	 */
5238b420f3adSRichard Lowe 	VERIFY3U(0, ==, zap_count(zd->zd_os, ZTEST_DIROBJ, &dirobjs));
5239b24ab676SJeff Bonwick 	dmu_objset_space(zd->zd_os, &scratch, &scratch, &usedobjs, &scratch);
5240b24ab676SJeff Bonwick 	ASSERT3U(dirobjs + 1, ==, usedobjs);
5241b24ab676SJeff Bonwick }
5242b24ab676SJeff Bonwick 
5243b24ab676SJeff Bonwick static int
5244420dfc95SChris Siden ztest_dataset_open(int d)
5245b24ab676SJeff Bonwick {
5246420dfc95SChris Siden 	ztest_ds_t *zd = &ztest_ds[d];
5247420dfc95SChris Siden 	uint64_t committed_seq = ZTEST_GET_SHARED_DS(d)->zd_seq;
5248b24ab676SJeff Bonwick 	objset_t *os;
5249b24ab676SJeff Bonwick 	zilog_t *zilog;
5250b24ab676SJeff Bonwick 	char name[MAXNAMELEN];
5251b24ab676SJeff Bonwick 	int error;
5252b24ab676SJeff Bonwick 
5253420dfc95SChris Siden 	ztest_dataset_name(name, ztest_opts.zo_pool, d);
5254b24ab676SJeff Bonwick 
5255420dfc95SChris Siden 	(void) rw_rdlock(&ztest_name_lock);
5256b24ab676SJeff Bonwick 
525755da60b9SMark J Musante 	error = ztest_dataset_create(name);
5258b24ab676SJeff Bonwick 	if (error == ENOSPC) {
5259420dfc95SChris Siden 		(void) rw_unlock(&ztest_name_lock);
5260b24ab676SJeff Bonwick 		ztest_record_enospc(FTAG);
5261b24ab676SJeff Bonwick 		return (error);
5262fa9e4066Sahrens 	}
5263b24ab676SJeff Bonwick 	ASSERT(error == 0 || error == EEXIST);
5264fa9e4066Sahrens 
5265*fb09f5aaSMadhav Suresh 	VERIFY0(dmu_objset_hold(name, zd, &os));
5266420dfc95SChris Siden 	(void) rw_unlock(&ztest_name_lock);
5267b24ab676SJeff Bonwick 
5268420dfc95SChris Siden 	ztest_zd_init(zd, ZTEST_GET_SHARED_DS(d), os);
5269b24ab676SJeff Bonwick 
5270b24ab676SJeff Bonwick 	zilog = zd->zd_zilog;
5271b24ab676SJeff Bonwick 
5272b24ab676SJeff Bonwick 	if (zilog->zl_header->zh_claim_lr_seq != 0 &&
5273b24ab676SJeff Bonwick 	    zilog->zl_header->zh_claim_lr_seq < committed_seq)
5274b24ab676SJeff Bonwick 		fatal(0, "missing log records: claimed %llu < committed %llu",
5275b24ab676SJeff Bonwick 		    zilog->zl_header->zh_claim_lr_seq, committed_seq);
5276b24ab676SJeff Bonwick 
5277b24ab676SJeff Bonwick 	ztest_dataset_dirobj_verify(zd);
5278b24ab676SJeff Bonwick 
5279b24ab676SJeff Bonwick 	zil_replay(os, zd, ztest_replay_vector);
5280b24ab676SJeff Bonwick 
5281b24ab676SJeff Bonwick 	ztest_dataset_dirobj_verify(zd);
5282b24ab676SJeff Bonwick 
5283420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 6)
5284b24ab676SJeff Bonwick 		(void) printf("%s replay %llu blocks, %llu records, seq %llu\n",
5285b24ab676SJeff Bonwick 		    zd->zd_name,
5286b24ab676SJeff Bonwick 		    (u_longlong_t)zilog->zl_parse_blk_count,
5287b24ab676SJeff Bonwick 		    (u_longlong_t)zilog->zl_parse_lr_count,
5288b24ab676SJeff Bonwick 		    (u_longlong_t)zilog->zl_replaying_seq);
5289b24ab676SJeff Bonwick 
5290b24ab676SJeff Bonwick 	zilog = zil_open(os, ztest_get_data);
5291b24ab676SJeff Bonwick 
5292b24ab676SJeff Bonwick 	if (zilog->zl_replaying_seq != 0 &&
5293b24ab676SJeff Bonwick 	    zilog->zl_replaying_seq < committed_seq)
5294b24ab676SJeff Bonwick 		fatal(0, "missing log records: replayed %llu < committed %llu",
5295b24ab676SJeff Bonwick 		    zilog->zl_replaying_seq, committed_seq);
5296b24ab676SJeff Bonwick 
5297b24ab676SJeff Bonwick 	return (0);
5298b24ab676SJeff Bonwick }
5299b24ab676SJeff Bonwick 
5300b24ab676SJeff Bonwick static void
5301420dfc95SChris Siden ztest_dataset_close(int d)
5302b24ab676SJeff Bonwick {
5303420dfc95SChris Siden 	ztest_ds_t *zd = &ztest_ds[d];
5304b24ab676SJeff Bonwick 
5305b24ab676SJeff Bonwick 	zil_close(zd->zd_zilog);
5306b24ab676SJeff Bonwick 	dmu_objset_rele(zd->zd_os, zd);
5307b24ab676SJeff Bonwick 
5308b24ab676SJeff Bonwick 	ztest_zd_fini(zd);
5309fa9e4066Sahrens }
5310fa9e4066Sahrens 
5311fa9e4066Sahrens /*
5312fa9e4066Sahrens  * Kick off threads to run tests on all datasets in parallel.
5313fa9e4066Sahrens  */
5314fa9e4066Sahrens static void
5315b24ab676SJeff Bonwick ztest_run(ztest_shared_t *zs)
5316fa9e4066Sahrens {
5317b24ab676SJeff Bonwick 	thread_t *tid;
5318fa9e4066Sahrens 	spa_t *spa;
5319e9103aaeSGarrett D'Amore 	objset_t *os;
5320e14bb325SJeff Bonwick 	thread_t resume_tid;
5321b24ab676SJeff Bonwick 	int error;
5322e14bb325SJeff Bonwick 
5323e14bb325SJeff Bonwick 	ztest_exiting = B_FALSE;
5324fa9e4066Sahrens 
5325fa9e4066Sahrens 	/*
5326b24ab676SJeff Bonwick 	 * Initialize parent/child shared state.
5327fa9e4066Sahrens 	 */
5328420dfc95SChris Siden 	VERIFY(_mutex_init(&ztest_vdev_lock, USYNC_THREAD, NULL) == 0);
5329420dfc95SChris Siden 	VERIFY(rwlock_init(&ztest_name_lock, USYNC_THREAD, NULL) == 0);
5330fa9e4066Sahrens 
5331b24ab676SJeff Bonwick 	zs->zs_thread_start = gethrtime();
5332420dfc95SChris Siden 	zs->zs_thread_stop =
5333420dfc95SChris Siden 	    zs->zs_thread_start + ztest_opts.zo_passtime * NANOSEC;
5334b24ab676SJeff Bonwick 	zs->zs_thread_stop = MIN(zs->zs_thread_stop, zs->zs_proc_stop);
5335b24ab676SJeff Bonwick 	zs->zs_thread_kill = zs->zs_thread_stop;
5336420dfc95SChris Siden 	if (ztest_random(100) < ztest_opts.zo_killrate) {
5337420dfc95SChris Siden 		zs->zs_thread_kill -=
5338420dfc95SChris Siden 		    ztest_random(ztest_opts.zo_passtime * NANOSEC);
5339420dfc95SChris Siden 	}
5340fa9e4066Sahrens 
5341b24ab676SJeff Bonwick 	(void) _mutex_init(&zcl.zcl_callbacks_lock, USYNC_THREAD, NULL);
5342fa9e4066Sahrens 
5343b24ab676SJeff Bonwick 	list_create(&zcl.zcl_callbacks, sizeof (ztest_cb_data_t),
5344b24ab676SJeff Bonwick 	    offsetof(ztest_cb_data_t, zcd_node));
5345fa9e4066Sahrens 
53460a4e9518Sgw 	/*
5347e14bb325SJeff Bonwick 	 * Open our pool.
53480a4e9518Sgw 	 */
5349b24ab676SJeff Bonwick 	kernel_init(FREAD | FWRITE);
5350420dfc95SChris Siden 	VERIFY(spa_open(ztest_opts.zo_pool, &spa, FTAG) == 0);
535109c9d376SGeorge Wilson 	spa->spa_debug = B_TRUE;
5352420dfc95SChris Siden 	ztest_spa = spa;
5353b24ab676SJeff Bonwick 
5354b420f3adSRichard Lowe 	VERIFY3U(0, ==, dmu_objset_hold(ztest_opts.zo_pool, FTAG, &os));
5355e9103aaeSGarrett D'Amore 	zs->zs_guid = dmu_objset_fsid_guid(os);
5356e9103aaeSGarrett D'Amore 	dmu_objset_rele(os, FTAG);
5357e9103aaeSGarrett D'Amore 
5358b24ab676SJeff Bonwick 	spa->spa_dedup_ditto = 2 * ZIO_DEDUPDITTO_MIN;
53590a4e9518Sgw 
53608ad4d6ddSJeff Bonwick 	/*
53618ad4d6ddSJeff Bonwick 	 * We don't expect the pool to suspend unless maxfaults == 0,
53628ad4d6ddSJeff Bonwick 	 * in which case ztest_fault_inject() temporarily takes away
53638ad4d6ddSJeff Bonwick 	 * the only valid replica.
53648ad4d6ddSJeff Bonwick 	 */
53651195e687SMark J Musante 	if (MAXFAULTS() == 0)
53668ad4d6ddSJeff Bonwick 		spa->spa_failmode = ZIO_FAILURE_MODE_WAIT;
53678ad4d6ddSJeff Bonwick 	else
53688ad4d6ddSJeff Bonwick 		spa->spa_failmode = ZIO_FAILURE_MODE_PANIC;
53698ad4d6ddSJeff Bonwick 
5370fa9e4066Sahrens 	/*
5371e14bb325SJeff Bonwick 	 * Create a thread to periodically resume suspended I/O.
5372fa9e4066Sahrens 	 */
53738ad4d6ddSJeff Bonwick 	VERIFY(thr_create(0, 0, ztest_resume_thread, spa, THR_BOUND,
5374e14bb325SJeff Bonwick 	    &resume_tid) == 0);
5375fa9e4066Sahrens 
5376b24ab676SJeff Bonwick 	/*
5377b24ab676SJeff Bonwick 	 * Create a deadman thread to abort() if we hang.
5378b24ab676SJeff Bonwick 	 */
5379b24ab676SJeff Bonwick 	VERIFY(thr_create(0, 0, ztest_deadman_thread, zs, THR_BOUND,
5380b24ab676SJeff Bonwick 	    NULL) == 0);
5381b24ab676SJeff Bonwick 
5382fa9e4066Sahrens 	/*
5383fa9e4066Sahrens 	 * Verify that we can safely inquire about about any object,
5384fa9e4066Sahrens 	 * whether it's allocated or not.  To make it interesting,
5385fa9e4066Sahrens 	 * we probe a 5-wide window around each power of two.
5386fa9e4066Sahrens 	 * This hits all edge cases, including zero and the max.
5387fa9e4066Sahrens 	 */
5388b24ab676SJeff Bonwick 	for (int t = 0; t < 64; t++) {
5389b24ab676SJeff Bonwick 		for (int d = -5; d <= 5; d++) {
5390fa9e4066Sahrens 			error = dmu_object_info(spa->spa_meta_objset,
5391fa9e4066Sahrens 			    (1ULL << t) + d, NULL);
5392ea8dc4b6Seschrock 			ASSERT(error == 0 || error == ENOENT ||
5393ea8dc4b6Seschrock 			    error == EINVAL);
5394fa9e4066Sahrens 		}
5395fa9e4066Sahrens 	}
5396fa9e4066Sahrens 
5397fa9e4066Sahrens 	/*
5398b24ab676SJeff Bonwick 	 * If we got any ENOSPC errors on the previous run, destroy something.
5399fa9e4066Sahrens 	 */
5400b24ab676SJeff Bonwick 	if (zs->zs_enospc_count != 0) {
5401420dfc95SChris Siden 		int d = ztest_random(ztest_opts.zo_datasets);
5402420dfc95SChris Siden 		ztest_dataset_destroy(d);
5403b24ab676SJeff Bonwick 	}
5404fa9e4066Sahrens 	zs->zs_enospc_count = 0;
5405fa9e4066Sahrens 
5406420dfc95SChris Siden 	tid = umem_zalloc(ztest_opts.zo_threads * sizeof (thread_t),
5407420dfc95SChris Siden 	    UMEM_NOFAIL);
5408fa9e4066Sahrens 
5409420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 4)
5410fa9e4066Sahrens 		(void) printf("starting main threads...\n");
5411fa9e4066Sahrens 
5412b24ab676SJeff Bonwick 	/*
5413b24ab676SJeff Bonwick 	 * Kick off all the tests that run in parallel.
5414b24ab676SJeff Bonwick 	 */
5415420dfc95SChris Siden 	for (int t = 0; t < ztest_opts.zo_threads; t++) {
5416420dfc95SChris Siden 		if (t < ztest_opts.zo_datasets &&
5417420dfc95SChris Siden 		    ztest_dataset_open(t) != 0)
5418b24ab676SJeff Bonwick 			return;
5419b24ab676SJeff Bonwick 		VERIFY(thr_create(0, 0, ztest_thread, (void *)(uintptr_t)t,
5420b24ab676SJeff Bonwick 		    THR_BOUND, &tid[t]) == 0);
5421fa9e4066Sahrens 	}
5422fa9e4066Sahrens 
5423b24ab676SJeff Bonwick 	/*
5424b24ab676SJeff Bonwick 	 * Wait for all of the tests to complete.  We go in reverse order
5425b24ab676SJeff Bonwick 	 * so we don't close datasets while threads are still using them.
5426b24ab676SJeff Bonwick 	 */
5427420dfc95SChris Siden 	for (int t = ztest_opts.zo_threads - 1; t >= 0; t--) {
5428b24ab676SJeff Bonwick 		VERIFY(thr_join(tid[t], NULL, NULL) == 0);
5429420dfc95SChris Siden 		if (t < ztest_opts.zo_datasets)
5430420dfc95SChris Siden 			ztest_dataset_close(t);
5431fa9e4066Sahrens 	}
5432fa9e4066Sahrens 
5433fa9e4066Sahrens 	txg_wait_synced(spa_get_dsl(spa), 0);
5434fa9e4066Sahrens 
5435b24ab676SJeff Bonwick 	zs->zs_alloc = metaslab_class_get_alloc(spa_normal_class(spa));
5436b24ab676SJeff Bonwick 	zs->zs_space = metaslab_class_get_space(spa_normal_class(spa));
5437b24ab676SJeff Bonwick 
5438420dfc95SChris Siden 	umem_free(tid, ztest_opts.zo_threads * sizeof (thread_t));
5439b24ab676SJeff Bonwick 
5440b24ab676SJeff Bonwick 	/* Kill the resume thread */
5441b24ab676SJeff Bonwick 	ztest_exiting = B_TRUE;
5442b24ab676SJeff Bonwick 	VERIFY(thr_join(resume_tid, NULL, NULL) == 0);
5443b24ab676SJeff Bonwick 	ztest_resume(spa);
5444b24ab676SJeff Bonwick 
5445b24ab676SJeff Bonwick 	/*
5446b24ab676SJeff Bonwick 	 * Right before closing the pool, kick off a bunch of async I/O;
5447b24ab676SJeff Bonwick 	 * spa_close() should wait for it to complete.
5448b24ab676SJeff Bonwick 	 */
5449b24ab676SJeff Bonwick 	for (uint64_t object = 1; object < 50; object++)
5450b24ab676SJeff Bonwick 		dmu_prefetch(spa->spa_meta_objset, object, 0, 1ULL << 20);
5451b24ab676SJeff Bonwick 
5452b24ab676SJeff Bonwick 	spa_close(spa, FTAG);
5453fa9e4066Sahrens 
5454fa9e4066Sahrens 	/*
5455b24ab676SJeff Bonwick 	 * Verify that we can loop over all pools.
5456fa9e4066Sahrens 	 */
5457b24ab676SJeff Bonwick 	mutex_enter(&spa_namespace_lock);
5458b24ab676SJeff Bonwick 	for (spa = spa_next(NULL); spa != NULL; spa = spa_next(spa))
5459420dfc95SChris Siden 		if (ztest_opts.zo_verbose > 3)
5460b24ab676SJeff Bonwick 			(void) printf("spa_next: found %s\n", spa_name(spa));
5461b24ab676SJeff Bonwick 	mutex_exit(&spa_namespace_lock);
5462b24ab676SJeff Bonwick 
5463b24ab676SJeff Bonwick 	/*
5464b24ab676SJeff Bonwick 	 * Verify that we can export the pool and reimport it under a
5465b24ab676SJeff Bonwick 	 * different name.
5466b24ab676SJeff Bonwick 	 */
5467b24ab676SJeff Bonwick 	if (ztest_random(2) == 0) {
5468b24ab676SJeff Bonwick 		char name[MAXNAMELEN];
5469420dfc95SChris Siden 		(void) snprintf(name, MAXNAMELEN, "%s_import",
5470420dfc95SChris Siden 		    ztest_opts.zo_pool);
5471420dfc95SChris Siden 		ztest_spa_import_export(ztest_opts.zo_pool, name);
5472420dfc95SChris Siden 		ztest_spa_import_export(name, ztest_opts.zo_pool);
5473b24ab676SJeff Bonwick 	}
5474b24ab676SJeff Bonwick 
5475b24ab676SJeff Bonwick 	kernel_fini();
547626cb54a9SGeorge Wilson 
547726cb54a9SGeorge Wilson 	list_destroy(&zcl.zcl_callbacks);
547826cb54a9SGeorge Wilson 
547926cb54a9SGeorge Wilson 	(void) _mutex_destroy(&zcl.zcl_callbacks_lock);
548026cb54a9SGeorge Wilson 
5481420dfc95SChris Siden 	(void) rwlock_destroy(&ztest_name_lock);
5482420dfc95SChris Siden 	(void) _mutex_destroy(&ztest_vdev_lock);
5483b24ab676SJeff Bonwick }
5484b24ab676SJeff Bonwick 
5485b24ab676SJeff Bonwick static void
5486420dfc95SChris Siden ztest_freeze(void)
5487b24ab676SJeff Bonwick {
5488420dfc95SChris Siden 	ztest_ds_t *zd = &ztest_ds[0];
5489b24ab676SJeff Bonwick 	spa_t *spa;
54902215e990SMark J Musante 	int numloops = 0;
5491b24ab676SJeff Bonwick 
5492420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 3)
5493b24ab676SJeff Bonwick 		(void) printf("testing spa_freeze()...\n");
5494718d718aSGeorge Wilson 
5495b24ab676SJeff Bonwick 	kernel_init(FREAD | FWRITE);
5496b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_open(ztest_opts.zo_pool, &spa, FTAG));
5497b420f3adSRichard Lowe 	VERIFY3U(0, ==, ztest_dataset_open(0));
5498718d718aSGeorge Wilson 
5499b24ab676SJeff Bonwick 	/*
5500b24ab676SJeff Bonwick 	 * Force the first log block to be transactionally allocated.
5501b24ab676SJeff Bonwick 	 * We have to do this before we freeze the pool -- otherwise
5502b24ab676SJeff Bonwick 	 * the log chain won't be anchored.
5503b24ab676SJeff Bonwick 	 */
5504b24ab676SJeff Bonwick 	while (BP_IS_HOLE(&zd->zd_zilog->zl_header->zh_log)) {
5505b24ab676SJeff Bonwick 		ztest_dmu_object_alloc_free(zd, 0);
55065002558fSNeil Perrin 		zil_commit(zd->zd_zilog, 0);
5507fa9e4066Sahrens 	}
5508fa9e4066Sahrens 
5509ea8dc4b6Seschrock 	txg_wait_synced(spa_get_dsl(spa), 0);
5510fa9e4066Sahrens 
5511b24ab676SJeff Bonwick 	/*
5512b24ab676SJeff Bonwick 	 * Freeze the pool.  This stops spa_sync() from doing anything,
5513b24ab676SJeff Bonwick 	 * so that the only way to record changes from now on is the ZIL.
5514b24ab676SJeff Bonwick 	 */
5515b24ab676SJeff Bonwick 	spa_freeze(spa);
5516e14bb325SJeff Bonwick 
5517b24ab676SJeff Bonwick 	/*
5518b24ab676SJeff Bonwick 	 * Run tests that generate log records but don't alter the pool config
5519b24ab676SJeff Bonwick 	 * or depend on DSL sync tasks (snapshots, objset create/destroy, etc).
5520b24ab676SJeff Bonwick 	 * We do a txg_wait_synced() after each iteration to force the txg
5521b24ab676SJeff Bonwick 	 * to increase well beyond the last synced value in the uberblock.
5522b24ab676SJeff Bonwick 	 * The ZIL should be OK with that.
5523b24ab676SJeff Bonwick 	 */
5524420dfc95SChris Siden 	while (ztest_random(10) != 0 &&
5525420dfc95SChris Siden 	    numloops++ < ztest_opts.zo_maxloops) {
5526b24ab676SJeff Bonwick 		ztest_dmu_write_parallel(zd, 0);
5527b24ab676SJeff Bonwick 		ztest_dmu_object_alloc_free(zd, 0);
5528b24ab676SJeff Bonwick 		txg_wait_synced(spa_get_dsl(spa), 0);
5529b24ab676SJeff Bonwick 	}
5530e14bb325SJeff Bonwick 
5531fa9e4066Sahrens 	/*
5532b24ab676SJeff Bonwick 	 * Commit all of the changes we just generated.
5533fa9e4066Sahrens 	 */
55345002558fSNeil Perrin 	zil_commit(zd->zd_zilog, 0);
5535b24ab676SJeff Bonwick 	txg_wait_synced(spa_get_dsl(spa), 0);
5536fa9e4066Sahrens 
5537b24ab676SJeff Bonwick 	/*
5538b24ab676SJeff Bonwick 	 * Close our dataset and close the pool.
5539b24ab676SJeff Bonwick 	 */
5540420dfc95SChris Siden 	ztest_dataset_close(0);
5541e05725b1Sbonwick 	spa_close(spa, FTAG);
5542b24ab676SJeff Bonwick 	kernel_fini();
5543e05725b1Sbonwick 
5544b24ab676SJeff Bonwick 	/*
5545b24ab676SJeff Bonwick 	 * Open and close the pool and dataset to induce log replay.
5546b24ab676SJeff Bonwick 	 */
5547b24ab676SJeff Bonwick 	kernel_init(FREAD | FWRITE);
5548b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_open(ztest_opts.zo_pool, &spa, FTAG));
5549ce636f8bSMatthew Ahrens 	ASSERT(spa_freeze_txg(spa) == UINT64_MAX);
5550b420f3adSRichard Lowe 	VERIFY3U(0, ==, ztest_dataset_open(0));
5551420dfc95SChris Siden 	ztest_dataset_close(0);
5552dfbb9432SGeorge Wilson 
5553dfbb9432SGeorge Wilson 	spa->spa_debug = B_TRUE;
5554dfbb9432SGeorge Wilson 	ztest_spa = spa;
5555dfbb9432SGeorge Wilson 	txg_wait_synced(spa_get_dsl(spa), 0);
5556dfbb9432SGeorge Wilson 	ztest_reguid(NULL, 0);
5557dfbb9432SGeorge Wilson 
5558b24ab676SJeff Bonwick 	spa_close(spa, FTAG);
5559fa9e4066Sahrens 	kernel_fini();
5560fa9e4066Sahrens }
5561fa9e4066Sahrens 
5562fa9e4066Sahrens void
5563fa9e4066Sahrens print_time(hrtime_t t, char *timebuf)
5564fa9e4066Sahrens {
5565fa9e4066Sahrens 	hrtime_t s = t / NANOSEC;
5566fa9e4066Sahrens 	hrtime_t m = s / 60;
5567fa9e4066Sahrens 	hrtime_t h = m / 60;
5568fa9e4066Sahrens 	hrtime_t d = h / 24;
5569fa9e4066Sahrens 
5570fa9e4066Sahrens 	s -= m * 60;
5571fa9e4066Sahrens 	m -= h * 60;
5572fa9e4066Sahrens 	h -= d * 24;
5573fa9e4066Sahrens 
5574fa9e4066Sahrens 	timebuf[0] = '\0';
5575fa9e4066Sahrens 
5576fa9e4066Sahrens 	if (d)
5577fa9e4066Sahrens 		(void) sprintf(timebuf,
5578fa9e4066Sahrens 		    "%llud%02lluh%02llum%02llus", d, h, m, s);
5579fa9e4066Sahrens 	else if (h)
5580fa9e4066Sahrens 		(void) sprintf(timebuf, "%lluh%02llum%02llus", h, m, s);
5581fa9e4066Sahrens 	else if (m)
5582fa9e4066Sahrens 		(void) sprintf(timebuf, "%llum%02llus", m, s);
5583fa9e4066Sahrens 	else
5584fa9e4066Sahrens 		(void) sprintf(timebuf, "%llus", s);
5585fa9e4066Sahrens }
5586fa9e4066Sahrens 
55871195e687SMark J Musante static nvlist_t *
55881195e687SMark J Musante make_random_props()
55891195e687SMark J Musante {
55901195e687SMark J Musante 	nvlist_t *props;
55911195e687SMark J Musante 
55921195e687SMark J Musante 	VERIFY(nvlist_alloc(&props, NV_UNIQUE_NAME, 0) == 0);
5593ad135b5dSChristopher Siden 	if (ztest_random(2) == 0)
5594ad135b5dSChristopher Siden 		return (props);
55951195e687SMark J Musante 	VERIFY(nvlist_add_uint64(props, "autoreplace", 1) == 0);
55961195e687SMark J Musante 
55971195e687SMark J Musante 	return (props);
55981195e687SMark J Musante }
55991195e687SMark J Musante 
5600fa9e4066Sahrens /*
5601fa9e4066Sahrens  * Create a storage pool with the given name and initial vdev size.
5602b24ab676SJeff Bonwick  * Then test spa_freeze() functionality.
5603fa9e4066Sahrens  */
5604fa9e4066Sahrens static void
5605b24ab676SJeff Bonwick ztest_init(ztest_shared_t *zs)
5606fa9e4066Sahrens {
5607fa9e4066Sahrens 	spa_t *spa;
56081195e687SMark J Musante 	nvlist_t *nvroot, *props;
5609fa9e4066Sahrens 
5610420dfc95SChris Siden 	VERIFY(_mutex_init(&ztest_vdev_lock, USYNC_THREAD, NULL) == 0);
5611420dfc95SChris Siden 	VERIFY(rwlock_init(&ztest_name_lock, USYNC_THREAD, NULL) == 0);
5612b24ab676SJeff Bonwick 
5613fa9e4066Sahrens 	kernel_init(FREAD | FWRITE);
5614fa9e4066Sahrens 
5615fa9e4066Sahrens 	/*
5616fa9e4066Sahrens 	 * Create the storage pool.
5617fa9e4066Sahrens 	 */
5618420dfc95SChris Siden 	(void) spa_destroy(ztest_opts.zo_pool);
561988ecc943SGeorge Wilson 	ztest_shared->zs_vdev_next_leaf = 0;
56201195e687SMark J Musante 	zs->zs_splits = 0;
5621420dfc95SChris Siden 	zs->zs_mirrors = ztest_opts.zo_mirrors;
5622420dfc95SChris Siden 	nvroot = make_vdev_root(NULL, NULL, ztest_opts.zo_vdev_size, 0,
5623420dfc95SChris Siden 	    0, ztest_opts.zo_raidz, zs->zs_mirrors, 1);
56241195e687SMark J Musante 	props = make_random_props();
5625ad135b5dSChristopher Siden 	for (int i = 0; i < SPA_FEATURES; i++) {
5626ad135b5dSChristopher Siden 		char buf[1024];
5627ad135b5dSChristopher Siden 		(void) snprintf(buf, sizeof (buf), "feature@%s",
5628ad135b5dSChristopher Siden 		    spa_feature_table[i].fi_uname);
5629b420f3adSRichard Lowe 		VERIFY3U(0, ==, nvlist_add_uint64(props, buf, 0));
5630ad135b5dSChristopher Siden 	}
5631b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_create(ztest_opts.zo_pool, nvroot, props, NULL));
5632fa9e4066Sahrens 	nvlist_free(nvroot);
5633fa9e4066Sahrens 
5634b420f3adSRichard Lowe 	VERIFY3U(0, ==, spa_open(ztest_opts.zo_pool, &spa, FTAG));
5635420dfc95SChris Siden 	zs->zs_metaslab_sz =
5636420dfc95SChris Siden 	    1ULL << spa->spa_root_vdev->vdev_child[0]->vdev_ms_shift;
5637ad135b5dSChristopher Siden 
5638fa9e4066Sahrens 	spa_close(spa, FTAG);
5639fa9e4066Sahrens 
5640fa9e4066Sahrens 	kernel_fini();
5641b24ab676SJeff Bonwick 
5642420dfc95SChris Siden 	ztest_run_zdb(ztest_opts.zo_pool);
5643420dfc95SChris Siden 
5644420dfc95SChris Siden 	ztest_freeze();
5645420dfc95SChris Siden 
5646420dfc95SChris Siden 	ztest_run_zdb(ztest_opts.zo_pool);
5647420dfc95SChris Siden 
5648420dfc95SChris Siden 	(void) rwlock_destroy(&ztest_name_lock);
5649420dfc95SChris Siden 	(void) _mutex_destroy(&ztest_vdev_lock);
5650420dfc95SChris Siden }
5651420dfc95SChris Siden 
5652420dfc95SChris Siden static void
5653420dfc95SChris Siden setup_fds(void)
5654420dfc95SChris Siden {
5655420dfc95SChris Siden 	int fd;
5656420dfc95SChris Siden 
5657420dfc95SChris Siden 	char *tmp = tempnam(NULL, NULL);
5658420dfc95SChris Siden 	fd = open(tmp, O_RDWR | O_CREAT, 0700);
5659420dfc95SChris Siden 	ASSERT3U(fd, ==, ZTEST_FD_DATA);
5660420dfc95SChris Siden 	(void) unlink(tmp);
5661420dfc95SChris Siden 	free(tmp);
5662420dfc95SChris Siden 
5663420dfc95SChris Siden 	fd = open("/dev/urandom", O_RDONLY);
5664420dfc95SChris Siden 	ASSERT3U(fd, ==, ZTEST_FD_RAND);
5665420dfc95SChris Siden }
5666420dfc95SChris Siden 
5667ad135b5dSChristopher Siden static int
5668ad135b5dSChristopher Siden shared_data_size(ztest_shared_hdr_t *hdr)
5669ad135b5dSChristopher Siden {
5670ad135b5dSChristopher Siden 	int size;
5671ad135b5dSChristopher Siden 
5672ad135b5dSChristopher Siden 	size = hdr->zh_hdr_size;
5673ad135b5dSChristopher Siden 	size += hdr->zh_opts_size;
5674ad135b5dSChristopher Siden 	size += hdr->zh_size;
5675ad135b5dSChristopher Siden 	size += hdr->zh_stats_size * hdr->zh_stats_count;
5676ad135b5dSChristopher Siden 	size += hdr->zh_ds_size * hdr->zh_ds_count;
5677ad135b5dSChristopher Siden 
5678ad135b5dSChristopher Siden 	return (size);
5679ad135b5dSChristopher Siden }
5680ad135b5dSChristopher Siden 
5681420dfc95SChris Siden static void
5682420dfc95SChris Siden setup_hdr(void)
5683420dfc95SChris Siden {
5684ad135b5dSChristopher Siden 	int size;
5685420dfc95SChris Siden 	ztest_shared_hdr_t *hdr;
5686420dfc95SChris Siden 
5687420dfc95SChris Siden 	hdr = (void *)mmap(0, P2ROUNDUP(sizeof (*hdr), getpagesize()),
5688420dfc95SChris Siden 	    PROT_READ | PROT_WRITE, MAP_SHARED, ZTEST_FD_DATA, 0);
5689420dfc95SChris Siden 	ASSERT(hdr != MAP_FAILED);
5690420dfc95SChris Siden 
5691b420f3adSRichard Lowe 	VERIFY3U(0, ==, ftruncate(ZTEST_FD_DATA, sizeof (ztest_shared_hdr_t)));
5692ad135b5dSChristopher Siden 
5693420dfc95SChris Siden 	hdr->zh_hdr_size = sizeof (ztest_shared_hdr_t);
5694420dfc95SChris Siden 	hdr->zh_opts_size = sizeof (ztest_shared_opts_t);
5695420dfc95SChris Siden 	hdr->zh_size = sizeof (ztest_shared_t);
5696420dfc95SChris Siden 	hdr->zh_stats_size = sizeof (ztest_shared_callstate_t);
5697420dfc95SChris Siden 	hdr->zh_stats_count = ZTEST_FUNCS;
5698420dfc95SChris Siden 	hdr->zh_ds_size = sizeof (ztest_shared_ds_t);
5699420dfc95SChris Siden 	hdr->zh_ds_count = ztest_opts.zo_datasets;
5700420dfc95SChris Siden 
5701ad135b5dSChristopher Siden 	size = shared_data_size(hdr);
5702b420f3adSRichard Lowe 	VERIFY3U(0, ==, ftruncate(ZTEST_FD_DATA, size));
5703ad135b5dSChristopher Siden 
5704420dfc95SChris Siden 	(void) munmap((caddr_t)hdr, P2ROUNDUP(sizeof (*hdr), getpagesize()));
5705420dfc95SChris Siden }
5706420dfc95SChris Siden 
5707420dfc95SChris Siden static void
5708420dfc95SChris Siden setup_data(void)
5709420dfc95SChris Siden {
5710420dfc95SChris Siden 	int size, offset;
5711420dfc95SChris Siden 	ztest_shared_hdr_t *hdr;
5712420dfc95SChris Siden 	uint8_t *buf;
5713420dfc95SChris Siden 
5714420dfc95SChris Siden 	hdr = (void *)mmap(0, P2ROUNDUP(sizeof (*hdr), getpagesize()),
5715420dfc95SChris Siden 	    PROT_READ, MAP_SHARED, ZTEST_FD_DATA, 0);
5716420dfc95SChris Siden 	ASSERT(hdr != MAP_FAILED);
5717420dfc95SChris Siden 
5718ad135b5dSChristopher Siden 	size = shared_data_size(hdr);
5719420dfc95SChris Siden 
5720420dfc95SChris Siden 	(void) munmap((caddr_t)hdr, P2ROUNDUP(sizeof (*hdr), getpagesize()));
5721420dfc95SChris Siden 	hdr = ztest_shared_hdr = (void *)mmap(0, P2ROUNDUP(size, getpagesize()),
5722420dfc95SChris Siden 	    PROT_READ | PROT_WRITE, MAP_SHARED, ZTEST_FD_DATA, 0);
5723420dfc95SChris Siden 	ASSERT(hdr != MAP_FAILED);
5724420dfc95SChris Siden 	buf = (uint8_t *)hdr;
5725420dfc95SChris Siden 
5726420dfc95SChris Siden 	offset = hdr->zh_hdr_size;
5727420dfc95SChris Siden 	ztest_shared_opts = (void *)&buf[offset];
5728420dfc95SChris Siden 	offset += hdr->zh_opts_size;
5729420dfc95SChris Siden 	ztest_shared = (void *)&buf[offset];
5730420dfc95SChris Siden 	offset += hdr->zh_size;
5731420dfc95SChris Siden 	ztest_shared_callstate = (void *)&buf[offset];
5732420dfc95SChris Siden 	offset += hdr->zh_stats_size * hdr->zh_stats_count;
5733420dfc95SChris Siden 	ztest_shared_ds = (void *)&buf[offset];
5734420dfc95SChris Siden }
5735420dfc95SChris Siden 
5736420dfc95SChris Siden static boolean_t
5737420dfc95SChris Siden exec_child(char *cmd, char *libpath, boolean_t ignorekill, int *statusp)
5738420dfc95SChris Siden {
5739420dfc95SChris Siden 	pid_t pid;
5740420dfc95SChris Siden 	int status;
5741420dfc95SChris Siden 	char cmdbuf[MAXPATHLEN];
5742420dfc95SChris Siden 
5743420dfc95SChris Siden 	pid = fork();
5744420dfc95SChris Siden 
5745420dfc95SChris Siden 	if (cmd == NULL) {
5746420dfc95SChris Siden 		(void) strlcpy(cmdbuf, getexecname(), sizeof (cmdbuf));
5747420dfc95SChris Siden 		cmd = cmdbuf;
5748420dfc95SChris Siden 	}
5749b24ab676SJeff Bonwick 
5750420dfc95SChris Siden 	if (pid == -1)
5751420dfc95SChris Siden 		fatal(1, "fork failed");
5752b24ab676SJeff Bonwick 
5753420dfc95SChris Siden 	if (pid == 0) {	/* child */
5754420dfc95SChris Siden 		char *emptyargv[2] = { cmd, NULL };
575526cb54a9SGeorge Wilson 
5756420dfc95SChris Siden 		struct rlimit rl = { 1024, 1024 };
5757420dfc95SChris Siden 		(void) setrlimit(RLIMIT_NOFILE, &rl);
5758420dfc95SChris Siden 		(void) enable_extended_FILE_stdio(-1, -1);
5759420dfc95SChris Siden 		if (libpath != NULL)
5760420dfc95SChris Siden 			VERIFY(0 == setenv("LD_LIBRARY_PATH", libpath, 1));
5761420dfc95SChris Siden 		(void) execv(cmd, emptyargv);
5762420dfc95SChris Siden 		ztest_dump_core = B_FALSE;
5763420dfc95SChris Siden 		fatal(B_TRUE, "exec failed: %s", cmd);
5764420dfc95SChris Siden 	}
5765420dfc95SChris Siden 
5766420dfc95SChris Siden 	while (waitpid(pid, &status, 0) != pid)
5767420dfc95SChris Siden 		continue;
5768420dfc95SChris Siden 	if (statusp != NULL)
5769420dfc95SChris Siden 		*statusp = status;
5770420dfc95SChris Siden 
5771420dfc95SChris Siden 	if (WIFEXITED(status)) {
5772420dfc95SChris Siden 		if (WEXITSTATUS(status) != 0) {
5773420dfc95SChris Siden 			(void) fprintf(stderr, "child exited with code %d\n",
5774420dfc95SChris Siden 			    WEXITSTATUS(status));
5775420dfc95SChris Siden 			exit(2);
5776420dfc95SChris Siden 		}
5777420dfc95SChris Siden 		return (B_FALSE);
5778420dfc95SChris Siden 	} else if (WIFSIGNALED(status)) {
5779420dfc95SChris Siden 		if (!ignorekill || WTERMSIG(status) != SIGKILL) {
5780420dfc95SChris Siden 			(void) fprintf(stderr, "child died with signal %d\n",
5781420dfc95SChris Siden 			    WTERMSIG(status));
5782420dfc95SChris Siden 			exit(3);
5783420dfc95SChris Siden 		}
5784420dfc95SChris Siden 		return (B_TRUE);
5785420dfc95SChris Siden 	} else {
5786420dfc95SChris Siden 		(void) fprintf(stderr, "something strange happened to child\n");
5787420dfc95SChris Siden 		exit(4);
5788420dfc95SChris Siden 		/* NOTREACHED */
5789420dfc95SChris Siden 	}
5790420dfc95SChris Siden }
5791420dfc95SChris Siden 
5792420dfc95SChris Siden static void
5793420dfc95SChris Siden ztest_run_init(void)
5794420dfc95SChris Siden {
5795420dfc95SChris Siden 	ztest_shared_t *zs = ztest_shared;
5796420dfc95SChris Siden 
5797420dfc95SChris Siden 	ASSERT(ztest_opts.zo_init != 0);
5798420dfc95SChris Siden 
5799420dfc95SChris Siden 	/*
5800420dfc95SChris Siden 	 * Blow away any existing copy of zpool.cache
5801420dfc95SChris Siden 	 */
5802420dfc95SChris Siden 	(void) remove(spa_config_path);
5803420dfc95SChris Siden 
5804420dfc95SChris Siden 	/*
5805420dfc95SChris Siden 	 * Create and initialize our storage pool.
5806420dfc95SChris Siden 	 */
5807420dfc95SChris Siden 	for (int i = 1; i <= ztest_opts.zo_init; i++) {
5808420dfc95SChris Siden 		bzero(zs, sizeof (ztest_shared_t));
5809420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 3 &&
5810420dfc95SChris Siden 		    ztest_opts.zo_init != 1) {
5811420dfc95SChris Siden 			(void) printf("ztest_init(), pass %d\n", i);
5812420dfc95SChris Siden 		}
5813420dfc95SChris Siden 		ztest_init(zs);
5814420dfc95SChris Siden 	}
5815fa9e4066Sahrens }
5816fa9e4066Sahrens 
5817fa9e4066Sahrens int
5818fa9e4066Sahrens main(int argc, char **argv)
5819fa9e4066Sahrens {
5820fa9e4066Sahrens 	int kills = 0;
5821fa9e4066Sahrens 	int iters = 0;
5822420dfc95SChris Siden 	int older = 0;
5823420dfc95SChris Siden 	int newer = 0;
5824fa9e4066Sahrens 	ztest_shared_t *zs;
5825fa9e4066Sahrens 	ztest_info_t *zi;
5826420dfc95SChris Siden 	ztest_shared_callstate_t *zc;
5827fa9e4066Sahrens 	char timebuf[100];
5828fa9e4066Sahrens 	char numbuf[6];
5829b24ab676SJeff Bonwick 	spa_t *spa;
5830420dfc95SChris Siden 	char cmd[MAXNAMELEN];
5831420dfc95SChris Siden 	boolean_t hasalt;
5832420dfc95SChris Siden 
5833420dfc95SChris Siden 	boolean_t ischild = (0 == lseek(ZTEST_FD_DATA, 0, SEEK_CUR));
5834420dfc95SChris Siden 	ASSERT(ischild || errno == EBADF);
5835fa9e4066Sahrens 
5836fa9e4066Sahrens 	(void) setvbuf(stdout, NULL, _IOLBF, 0);
5837fa9e4066Sahrens 
5838cd1c8b85SMatthew Ahrens 	dprintf_setup(&argc, argv);
5839cd1c8b85SMatthew Ahrens 
5840420dfc95SChris Siden 	if (!ischild) {
5841420dfc95SChris Siden 		process_options(argc, argv);
5842fa9e4066Sahrens 
5843420dfc95SChris Siden 		setup_fds();
5844420dfc95SChris Siden 		setup_hdr();
5845420dfc95SChris Siden 		setup_data();
5846420dfc95SChris Siden 		bcopy(&ztest_opts, ztest_shared_opts,
5847420dfc95SChris Siden 		    sizeof (*ztest_shared_opts));
5848420dfc95SChris Siden 	} else {
5849420dfc95SChris Siden 		setup_data();
5850420dfc95SChris Siden 		bcopy(ztest_shared_opts, &ztest_opts, sizeof (ztest_opts));
5851420dfc95SChris Siden 	}
5852420dfc95SChris Siden 	ASSERT3U(ztest_opts.zo_datasets, ==, ztest_shared_hdr->zh_ds_count);
5853fa9e4066Sahrens 
5854f0ba89beSJeff Bonwick 	/* Override location of zpool.cache */
5855420dfc95SChris Siden 	(void) asprintf((char **)&spa_config_path, "%s/zpool.cache",
5856420dfc95SChris Siden 	    ztest_opts.zo_dir);
5857f0ba89beSJeff Bonwick 
5858420dfc95SChris Siden 	ztest_ds = umem_alloc(ztest_opts.zo_datasets * sizeof (ztest_ds_t),
5859420dfc95SChris Siden 	    UMEM_NOFAIL);
5860420dfc95SChris Siden 	zs = ztest_shared;
5861ea8dc4b6Seschrock 
5862420dfc95SChris Siden 	if (ischild) {
5863420dfc95SChris Siden 		metaslab_gang_bang = ztest_opts.zo_metaslab_gang_bang;
5864420dfc95SChris Siden 		metaslab_df_alloc_threshold =
5865420dfc95SChris Siden 		    zs->zs_metaslab_df_alloc_threshold;
5866b24ab676SJeff Bonwick 
5867420dfc95SChris Siden 		if (zs->zs_do_init)
5868420dfc95SChris Siden 			ztest_run_init();
5869420dfc95SChris Siden 		else
5870420dfc95SChris Siden 			ztest_run(zs);
5871420dfc95SChris Siden 		exit(0);
5872420dfc95SChris Siden 	}
5873fa9e4066Sahrens 
5874420dfc95SChris Siden 	hasalt = (strlen(ztest_opts.zo_alt_ztest) != 0);
5875420dfc95SChris Siden 
5876420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 1) {
5877fa9e4066Sahrens 		(void) printf("%llu vdevs, %d datasets, %d threads,"
5878fa9e4066Sahrens 		    " %llu seconds...\n",
5879420dfc95SChris Siden 		    (u_longlong_t)ztest_opts.zo_vdevs,
5880420dfc95SChris Siden 		    ztest_opts.zo_datasets,
5881420dfc95SChris Siden 		    ztest_opts.zo_threads,
5882420dfc95SChris Siden 		    (u_longlong_t)ztest_opts.zo_time);
5883fa9e4066Sahrens 	}
5884fa9e4066Sahrens 
5885420dfc95SChris Siden 	(void) strlcpy(cmd, getexecname(), sizeof (cmd));
5886420dfc95SChris Siden 
5887420dfc95SChris Siden 	zs->zs_do_init = B_TRUE;
5888420dfc95SChris Siden 	if (strlen(ztest_opts.zo_alt_ztest) != 0) {
5889420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 1) {
5890420dfc95SChris Siden 			(void) printf("Executing older ztest for "
5891420dfc95SChris Siden 			    "initialization: %s\n", ztest_opts.zo_alt_ztest);
5892420dfc95SChris Siden 		}
5893420dfc95SChris Siden 		VERIFY(!exec_child(ztest_opts.zo_alt_ztest,
5894420dfc95SChris Siden 		    ztest_opts.zo_alt_libpath, B_FALSE, NULL));
5895420dfc95SChris Siden 	} else {
5896420dfc95SChris Siden 		VERIFY(!exec_child(NULL, NULL, B_FALSE, NULL));
5897fa9e4066Sahrens 	}
5898420dfc95SChris Siden 	zs->zs_do_init = B_FALSE;
5899fa9e4066Sahrens 
5900b24ab676SJeff Bonwick 	zs->zs_proc_start = gethrtime();
5901420dfc95SChris Siden 	zs->zs_proc_stop = zs->zs_proc_start + ztest_opts.zo_time * NANOSEC;
5902fa9e4066Sahrens 
5903b24ab676SJeff Bonwick 	for (int f = 0; f < ZTEST_FUNCS; f++) {
5904420dfc95SChris Siden 		zi = &ztest_info[f];
5905420dfc95SChris Siden 		zc = ZTEST_GET_SHARED_CALLSTATE(f);
5906b24ab676SJeff Bonwick 		if (zs->zs_proc_start + zi->zi_interval[0] > zs->zs_proc_stop)
5907420dfc95SChris Siden 			zc->zc_next = UINT64_MAX;
5908fa9e4066Sahrens 		else
5909420dfc95SChris Siden 			zc->zc_next = zs->zs_proc_start +
5910b24ab676SJeff Bonwick 			    ztest_random(2 * zi->zi_interval[0] + 1);
5911fa9e4066Sahrens 	}
5912fa9e4066Sahrens 
5913fa9e4066Sahrens 	/*
5914fa9e4066Sahrens 	 * Run the tests in a loop.  These tests include fault injection
5915fa9e4066Sahrens 	 * to verify that self-healing data works, and forced crashes
5916fa9e4066Sahrens 	 * to verify that we never lose on-disk consistency.
5917fa9e4066Sahrens 	 */
5918b24ab676SJeff Bonwick 	while (gethrtime() < zs->zs_proc_stop) {
5919fa9e4066Sahrens 		int status;
5920420dfc95SChris Siden 		boolean_t killed;
5921fa9e4066Sahrens 
5922fa9e4066Sahrens 		/*
5923fa9e4066Sahrens 		 * Initialize the workload counters for each function.
5924fa9e4066Sahrens 		 */
5925b24ab676SJeff Bonwick 		for (int f = 0; f < ZTEST_FUNCS; f++) {
5926420dfc95SChris Siden 			zc = ZTEST_GET_SHARED_CALLSTATE(f);
5927420dfc95SChris Siden 			zc->zc_count = 0;
5928420dfc95SChris Siden 			zc->zc_time = 0;
5929fa9e4066Sahrens 		}
5930fa9e4066Sahrens 
5931d6e555bdSGeorge Wilson 		/* Set the allocation switch size */
5932420dfc95SChris Siden 		zs->zs_metaslab_df_alloc_threshold =
5933420dfc95SChris Siden 		    ztest_random(zs->zs_metaslab_sz / 4) + 1;
5934fa9e4066Sahrens 
5935420dfc95SChris Siden 		if (!hasalt || ztest_random(2) == 0) {
5936420dfc95SChris Siden 			if (hasalt && ztest_opts.zo_verbose >= 1) {
5937420dfc95SChris Siden 				(void) printf("Executing newer ztest: %s\n",
5938420dfc95SChris Siden 				    cmd);
5939fa9e4066Sahrens 			}
5940420dfc95SChris Siden 			newer++;
5941420dfc95SChris Siden 			killed = exec_child(cmd, NULL, B_TRUE, &status);
59425ad82045Snd 		} else {
5943420dfc95SChris Siden 			if (hasalt && ztest_opts.zo_verbose >= 1) {
5944420dfc95SChris Siden 				(void) printf("Executing older ztest: %s\n",
5945420dfc95SChris Siden 				    ztest_opts.zo_alt_ztest);
5946420dfc95SChris Siden 			}
5947420dfc95SChris Siden 			older++;
5948420dfc95SChris Siden 			killed = exec_child(ztest_opts.zo_alt_ztest,
5949420dfc95SChris Siden 			    ztest_opts.zo_alt_libpath, B_TRUE, &status);
5950fa9e4066Sahrens 		}
5951fa9e4066Sahrens 
5952420dfc95SChris Siden 		if (killed)
5953420dfc95SChris Siden 			kills++;
5954fa9e4066Sahrens 		iters++;
5955fa9e4066Sahrens 
5956420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 1) {
5957fa9e4066Sahrens 			hrtime_t now = gethrtime();
5958fa9e4066Sahrens 
5959b24ab676SJeff Bonwick 			now = MIN(now, zs->zs_proc_stop);
5960b24ab676SJeff Bonwick 			print_time(zs->zs_proc_stop - now, timebuf);
5961fa9e4066Sahrens 			nicenum(zs->zs_space, numbuf);
5962fa9e4066Sahrens 
5963fa9e4066Sahrens 			(void) printf("Pass %3d, %8s, %3llu ENOSPC, "
5964fa9e4066Sahrens 			    "%4.1f%% of %5s used, %3.0f%% done, %8s to go\n",
5965fa9e4066Sahrens 			    iters,
5966fa9e4066Sahrens 			    WIFEXITED(status) ? "Complete" : "SIGKILL",
5967fa9e4066Sahrens 			    (u_longlong_t)zs->zs_enospc_count,
5968fa9e4066Sahrens 			    100.0 * zs->zs_alloc / zs->zs_space,
5969fa9e4066Sahrens 			    numbuf,
5970b24ab676SJeff Bonwick 			    100.0 * (now - zs->zs_proc_start) /
5971420dfc95SChris Siden 			    (ztest_opts.zo_time * NANOSEC), timebuf);
5972fa9e4066Sahrens 		}
5973fa9e4066Sahrens 
5974420dfc95SChris Siden 		if (ztest_opts.zo_verbose >= 2) {
5975fa9e4066Sahrens 			(void) printf("\nWorkload summary:\n\n");
5976fa9e4066Sahrens 			(void) printf("%7s %9s   %s\n",
5977fa9e4066Sahrens 			    "Calls", "Time", "Function");
5978fa9e4066Sahrens 			(void) printf("%7s %9s   %s\n",
5979fa9e4066Sahrens 			    "-----", "----", "--------");
5980b24ab676SJeff Bonwick 			for (int f = 0; f < ZTEST_FUNCS; f++) {
5981fa9e4066Sahrens 				Dl_info dli;
5982fa9e4066Sahrens 
5983420dfc95SChris Siden 				zi = &ztest_info[f];
5984420dfc95SChris Siden 				zc = ZTEST_GET_SHARED_CALLSTATE(f);
5985420dfc95SChris Siden 				print_time(zc->zc_time, timebuf);
5986fa9e4066Sahrens 				(void) dladdr((void *)zi->zi_func, &dli);
5987fa9e4066Sahrens 				(void) printf("%7llu %9s   %s\n",
5988420dfc95SChris Siden 				    (u_longlong_t)zc->zc_count, timebuf,
5989fa9e4066Sahrens 				    dli.dli_sname);
5990fa9e4066Sahrens 			}
5991fa9e4066Sahrens 			(void) printf("\n");
5992fa9e4066Sahrens 		}
5993fa9e4066Sahrens 
5994fa9e4066Sahrens 		/*
5995b24ab676SJeff Bonwick 		 * It's possible that we killed a child during a rename test,
5996b24ab676SJeff Bonwick 		 * in which case we'll have a 'ztest_tmp' pool lying around
5997b24ab676SJeff Bonwick 		 * instead of 'ztest'.  Do a blind rename in case this happened.
5998fa9e4066Sahrens 		 */
5999b24ab676SJeff Bonwick 		kernel_init(FREAD);
6000420dfc95SChris Siden 		if (spa_open(ztest_opts.zo_pool, &spa, FTAG) == 0) {
6001b24ab676SJeff Bonwick 			spa_close(spa, FTAG);
6002b24ab676SJeff Bonwick 		} else {
6003b24ab676SJeff Bonwick 			char tmpname[MAXNAMELEN];
6004b24ab676SJeff Bonwick 			kernel_fini();
6005b24ab676SJeff Bonwick 			kernel_init(FREAD | FWRITE);
6006b24ab676SJeff Bonwick 			(void) snprintf(tmpname, sizeof (tmpname), "%s_tmp",
6007420dfc95SChris Siden 			    ztest_opts.zo_pool);
6008420dfc95SChris Siden 			(void) spa_rename(tmpname, ztest_opts.zo_pool);
6009b24ab676SJeff Bonwick 		}
6010fa9e4066Sahrens 		kernel_fini();
6011fa9e4066Sahrens 
6012420dfc95SChris Siden 		ztest_run_zdb(ztest_opts.zo_pool);
6013b24ab676SJeff Bonwick 	}
6014fa9e4066Sahrens 
6015420dfc95SChris Siden 	if (ztest_opts.zo_verbose >= 1) {
6016420dfc95SChris Siden 		if (hasalt) {
6017420dfc95SChris Siden 			(void) printf("%d runs of older ztest: %s\n", older,
6018420dfc95SChris Siden 			    ztest_opts.zo_alt_ztest);
6019420dfc95SChris Siden 			(void) printf("%d runs of newer ztest: %s\n", newer,
6020420dfc95SChris Siden 			    cmd);
6021420dfc95SChris Siden 		}
6022fa9e4066Sahrens 		(void) printf("%d killed, %d completed, %.0f%% kill rate\n",
6023fa9e4066Sahrens 		    kills, iters - kills, (100.0 * kills) / MAX(1, iters));
6024fa9e4066Sahrens 	}
6025fa9e4066Sahrens 
6026fa9e4066Sahrens 	return (0);
6027fa9e4066Sahrens }
6028