xref: /freebsd/sys/contrib/openzfs/module/zfs/dmu_recv.c (revision 071ab5a1f3cbfd29c8fbec27f7e619418adaf074)
161145dc2SMartin Matuska // SPDX-License-Identifier: CDDL-1.0
2eda14cbcSMatt Macy /*
3eda14cbcSMatt Macy  * CDDL HEADER START
4eda14cbcSMatt Macy  *
5eda14cbcSMatt Macy  * The contents of this file are subject to the terms of the
6eda14cbcSMatt Macy  * Common Development and Distribution License (the "License").
7eda14cbcSMatt Macy  * You may not use this file except in compliance with the License.
8eda14cbcSMatt Macy  *
9eda14cbcSMatt Macy  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10271171e0SMartin Matuska  * or https://opensource.org/licenses/CDDL-1.0.
11eda14cbcSMatt Macy  * See the License for the specific language governing permissions
12eda14cbcSMatt Macy  * and limitations under the License.
13eda14cbcSMatt Macy  *
14eda14cbcSMatt Macy  * When distributing Covered Code, include this CDDL HEADER in each
15eda14cbcSMatt Macy  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16eda14cbcSMatt Macy  * If applicable, add the following below this CDDL HEADER, with the
17eda14cbcSMatt Macy  * fields enclosed by brackets "[]" replaced with your own identifying
18eda14cbcSMatt Macy  * information: Portions Copyright [yyyy] [name of copyright owner]
19eda14cbcSMatt Macy  *
20eda14cbcSMatt Macy  * CDDL HEADER END
21eda14cbcSMatt Macy  */
22eda14cbcSMatt Macy /*
23eda14cbcSMatt Macy  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
24eda14cbcSMatt Macy  * Copyright 2011 Nexenta Systems, Inc. All rights reserved.
25eda14cbcSMatt Macy  * Copyright (c) 2011, 2020 by Delphix. All rights reserved.
26eda14cbcSMatt Macy  * Copyright (c) 2014, Joyent, Inc. All rights reserved.
27eda14cbcSMatt Macy  * Copyright 2014 HybridCluster. All rights reserved.
28eda14cbcSMatt Macy  * Copyright (c) 2018, loli10K <ezomori.nozomu@gmail.com>. All rights reserved.
297a7741afSMartin Matuska  * Copyright (c) 2019, 2024, Klara, Inc.
30eda14cbcSMatt Macy  * Copyright (c) 2019, Allan Jude
31271171e0SMartin Matuska  * Copyright (c) 2019 Datto Inc.
32271171e0SMartin Matuska  * Copyright (c) 2022 Axcient.
33*071ab5a1SMartin Matuska  * Copyright (c) 2025, Rob Norris <robn@despairlabs.com>
34eda14cbcSMatt Macy  */
35eda14cbcSMatt Macy 
3615f0b8c3SMartin Matuska #include <sys/arc.h>
37271171e0SMartin Matuska #include <sys/spa_impl.h>
38eda14cbcSMatt Macy #include <sys/dmu.h>
39eda14cbcSMatt Macy #include <sys/dmu_impl.h>
40eda14cbcSMatt Macy #include <sys/dmu_send.h>
41eda14cbcSMatt Macy #include <sys/dmu_recv.h>
42eda14cbcSMatt Macy #include <sys/dmu_tx.h>
43eda14cbcSMatt Macy #include <sys/dbuf.h>
44eda14cbcSMatt Macy #include <sys/dnode.h>
45eda14cbcSMatt Macy #include <sys/zfs_context.h>
46eda14cbcSMatt Macy #include <sys/dmu_objset.h>
47eda14cbcSMatt Macy #include <sys/dmu_traverse.h>
48eda14cbcSMatt Macy #include <sys/dsl_dataset.h>
49eda14cbcSMatt Macy #include <sys/dsl_dir.h>
50eda14cbcSMatt Macy #include <sys/dsl_prop.h>
51eda14cbcSMatt Macy #include <sys/dsl_pool.h>
52eda14cbcSMatt Macy #include <sys/dsl_synctask.h>
53eda14cbcSMatt Macy #include <sys/zfs_ioctl.h>
54eda14cbcSMatt Macy #include <sys/zap.h>
55eda14cbcSMatt Macy #include <sys/zvol.h>
56eda14cbcSMatt Macy #include <sys/zio_checksum.h>
57eda14cbcSMatt Macy #include <sys/zfs_znode.h>
58eda14cbcSMatt Macy #include <zfs_fletcher.h>
59eda14cbcSMatt Macy #include <sys/avl.h>
60eda14cbcSMatt Macy #include <sys/ddt.h>
61eda14cbcSMatt Macy #include <sys/zfs_onexit.h>
62eda14cbcSMatt Macy #include <sys/dsl_destroy.h>
63eda14cbcSMatt Macy #include <sys/blkptr.h>
64eda14cbcSMatt Macy #include <sys/dsl_bookmark.h>
65eda14cbcSMatt Macy #include <sys/zfeature.h>
66eda14cbcSMatt Macy #include <sys/bqueue.h>
67eda14cbcSMatt Macy #include <sys/objlist.h>
68eda14cbcSMatt Macy #ifdef _KERNEL
69eda14cbcSMatt Macy #include <sys/zfs_vfsops.h>
70eda14cbcSMatt Macy #endif
71eda14cbcSMatt Macy #include <sys/zfs_file.h>
72*071ab5a1SMartin Matuska #include <sys/cred.h>
73eda14cbcSMatt Macy 
74be181ee2SMartin Matuska static uint_t zfs_recv_queue_length = SPA_MAXBLOCKSIZE;
75be181ee2SMartin Matuska static uint_t zfs_recv_queue_ff = 20;
76be181ee2SMartin Matuska static uint_t zfs_recv_write_batch_size = 1024 * 1024;
77271171e0SMartin Matuska static int zfs_recv_best_effort_corrective = 0;
78eda14cbcSMatt Macy 
79a0b956f5SMartin Matuska static const void *const dmu_recv_tag = "dmu_recv_tag";
80e92ffd9bSMartin Matuska const char *const recv_clone_name = "%recv";
81eda14cbcSMatt Macy 
8215f0b8c3SMartin Matuska typedef enum {
8315f0b8c3SMartin Matuska 	ORNS_NO,
8415f0b8c3SMartin Matuska 	ORNS_YES,
8515f0b8c3SMartin Matuska 	ORNS_MAYBE
8615f0b8c3SMartin Matuska } or_need_sync_t;
8715f0b8c3SMartin Matuska 
88eda14cbcSMatt Macy static int receive_read_payload_and_next_header(dmu_recv_cookie_t *ra, int len,
89eda14cbcSMatt Macy     void *buf);
90eda14cbcSMatt Macy 
91eda14cbcSMatt Macy struct receive_record_arg {
92eda14cbcSMatt Macy 	dmu_replay_record_t header;
93eda14cbcSMatt Macy 	void *payload; /* Pointer to a buffer containing the payload */
94eda14cbcSMatt Macy 	/*
957877fdebSMatt Macy 	 * If the record is a WRITE or SPILL, pointer to the abd containing the
96eda14cbcSMatt Macy 	 * payload.
97eda14cbcSMatt Macy 	 */
987877fdebSMatt Macy 	abd_t *abd;
99eda14cbcSMatt Macy 	int payload_size;
100eda14cbcSMatt Macy 	uint64_t bytes_read; /* bytes read from stream when record created */
101eda14cbcSMatt Macy 	boolean_t eos_marker; /* Marks the end of the stream */
102eda14cbcSMatt Macy 	bqueue_node_t node;
103eda14cbcSMatt Macy };
104eda14cbcSMatt Macy 
105eda14cbcSMatt Macy struct receive_writer_arg {
106eda14cbcSMatt Macy 	objset_t *os;
107eda14cbcSMatt Macy 	boolean_t byteswap;
108eda14cbcSMatt Macy 	bqueue_t q;
109eda14cbcSMatt Macy 
110eda14cbcSMatt Macy 	/*
1117877fdebSMatt Macy 	 * These three members are used to signal to the main thread when
1127877fdebSMatt Macy 	 * we're done.
113eda14cbcSMatt Macy 	 */
114eda14cbcSMatt Macy 	kmutex_t mutex;
115eda14cbcSMatt Macy 	kcondvar_t cv;
116eda14cbcSMatt Macy 	boolean_t done;
117eda14cbcSMatt Macy 
118eda14cbcSMatt Macy 	int err;
119271171e0SMartin Matuska 	const char *tofs;
120271171e0SMartin Matuska 	boolean_t heal;
121eda14cbcSMatt Macy 	boolean_t resumable;
122eda14cbcSMatt Macy 	boolean_t raw;   /* DMU_BACKUP_FEATURE_RAW set */
123eda14cbcSMatt Macy 	boolean_t spill; /* DRR_FLAG_SPILL_BLOCK set */
124eda14cbcSMatt Macy 	boolean_t full;  /* this is a full send stream */
125eda14cbcSMatt Macy 	uint64_t last_object;
126eda14cbcSMatt Macy 	uint64_t last_offset;
127eda14cbcSMatt Macy 	uint64_t max_object; /* highest object ID referenced in stream */
128eda14cbcSMatt Macy 	uint64_t bytes_read; /* bytes read when current record created */
129eda14cbcSMatt Macy 
130eda14cbcSMatt Macy 	list_t write_batch;
131eda14cbcSMatt Macy 
132eda14cbcSMatt Macy 	/* Encryption parameters for the last received DRR_OBJECT_RANGE */
133eda14cbcSMatt Macy 	boolean_t or_crypt_params_present;
134eda14cbcSMatt Macy 	uint64_t or_firstobj;
135eda14cbcSMatt Macy 	uint64_t or_numslots;
136eda14cbcSMatt Macy 	uint8_t or_salt[ZIO_DATA_SALT_LEN];
137eda14cbcSMatt Macy 	uint8_t or_iv[ZIO_DATA_IV_LEN];
138eda14cbcSMatt Macy 	uint8_t or_mac[ZIO_DATA_MAC_LEN];
139eda14cbcSMatt Macy 	boolean_t or_byteorder;
140271171e0SMartin Matuska 	zio_t *heal_pio;
14115f0b8c3SMartin Matuska 
14215f0b8c3SMartin Matuska 	/* Keep track of DRR_FREEOBJECTS right after DRR_OBJECT_RANGE */
14315f0b8c3SMartin Matuska 	or_need_sync_t or_need_sync;
144eda14cbcSMatt Macy };
145eda14cbcSMatt Macy 
146eda14cbcSMatt Macy typedef struct dmu_recv_begin_arg {
147eda14cbcSMatt Macy 	const char *drba_origin;
148eda14cbcSMatt Macy 	dmu_recv_cookie_t *drba_cookie;
149eda14cbcSMatt Macy 	cred_t *drba_cred;
150eda14cbcSMatt Macy 	dsl_crypto_params_t *drba_dcp;
151eda14cbcSMatt Macy } dmu_recv_begin_arg_t;
152eda14cbcSMatt Macy 
153eda14cbcSMatt Macy static void
byteswap_record(dmu_replay_record_t * drr)154eda14cbcSMatt Macy byteswap_record(dmu_replay_record_t *drr)
155eda14cbcSMatt Macy {
156eda14cbcSMatt Macy #define	DO64(X) (drr->drr_u.X = BSWAP_64(drr->drr_u.X))
157eda14cbcSMatt Macy #define	DO32(X) (drr->drr_u.X = BSWAP_32(drr->drr_u.X))
158eda14cbcSMatt Macy 	drr->drr_type = BSWAP_32(drr->drr_type);
159eda14cbcSMatt Macy 	drr->drr_payloadlen = BSWAP_32(drr->drr_payloadlen);
160eda14cbcSMatt Macy 
161eda14cbcSMatt Macy 	switch (drr->drr_type) {
162eda14cbcSMatt Macy 	case DRR_BEGIN:
163eda14cbcSMatt Macy 		DO64(drr_begin.drr_magic);
164eda14cbcSMatt Macy 		DO64(drr_begin.drr_versioninfo);
165eda14cbcSMatt Macy 		DO64(drr_begin.drr_creation_time);
166eda14cbcSMatt Macy 		DO32(drr_begin.drr_type);
167eda14cbcSMatt Macy 		DO32(drr_begin.drr_flags);
168eda14cbcSMatt Macy 		DO64(drr_begin.drr_toguid);
169eda14cbcSMatt Macy 		DO64(drr_begin.drr_fromguid);
170eda14cbcSMatt Macy 		break;
171eda14cbcSMatt Macy 	case DRR_OBJECT:
172eda14cbcSMatt Macy 		DO64(drr_object.drr_object);
173eda14cbcSMatt Macy 		DO32(drr_object.drr_type);
174eda14cbcSMatt Macy 		DO32(drr_object.drr_bonustype);
175eda14cbcSMatt Macy 		DO32(drr_object.drr_blksz);
176eda14cbcSMatt Macy 		DO32(drr_object.drr_bonuslen);
177eda14cbcSMatt Macy 		DO32(drr_object.drr_raw_bonuslen);
178eda14cbcSMatt Macy 		DO64(drr_object.drr_toguid);
179eda14cbcSMatt Macy 		DO64(drr_object.drr_maxblkid);
180eda14cbcSMatt Macy 		break;
181eda14cbcSMatt Macy 	case DRR_FREEOBJECTS:
182eda14cbcSMatt Macy 		DO64(drr_freeobjects.drr_firstobj);
183eda14cbcSMatt Macy 		DO64(drr_freeobjects.drr_numobjs);
184eda14cbcSMatt Macy 		DO64(drr_freeobjects.drr_toguid);
185eda14cbcSMatt Macy 		break;
186eda14cbcSMatt Macy 	case DRR_WRITE:
187eda14cbcSMatt Macy 		DO64(drr_write.drr_object);
188eda14cbcSMatt Macy 		DO32(drr_write.drr_type);
189eda14cbcSMatt Macy 		DO64(drr_write.drr_offset);
190eda14cbcSMatt Macy 		DO64(drr_write.drr_logical_size);
191eda14cbcSMatt Macy 		DO64(drr_write.drr_toguid);
192eda14cbcSMatt Macy 		ZIO_CHECKSUM_BSWAP(&drr->drr_u.drr_write.drr_key.ddk_cksum);
193eda14cbcSMatt Macy 		DO64(drr_write.drr_key.ddk_prop);
194eda14cbcSMatt Macy 		DO64(drr_write.drr_compressed_size);
195eda14cbcSMatt Macy 		break;
196eda14cbcSMatt Macy 	case DRR_WRITE_EMBEDDED:
197eda14cbcSMatt Macy 		DO64(drr_write_embedded.drr_object);
198eda14cbcSMatt Macy 		DO64(drr_write_embedded.drr_offset);
199eda14cbcSMatt Macy 		DO64(drr_write_embedded.drr_length);
200eda14cbcSMatt Macy 		DO64(drr_write_embedded.drr_toguid);
201eda14cbcSMatt Macy 		DO32(drr_write_embedded.drr_lsize);
202eda14cbcSMatt Macy 		DO32(drr_write_embedded.drr_psize);
203eda14cbcSMatt Macy 		break;
204eda14cbcSMatt Macy 	case DRR_FREE:
205eda14cbcSMatt Macy 		DO64(drr_free.drr_object);
206eda14cbcSMatt Macy 		DO64(drr_free.drr_offset);
207eda14cbcSMatt Macy 		DO64(drr_free.drr_length);
208eda14cbcSMatt Macy 		DO64(drr_free.drr_toguid);
209eda14cbcSMatt Macy 		break;
210eda14cbcSMatt Macy 	case DRR_SPILL:
211eda14cbcSMatt Macy 		DO64(drr_spill.drr_object);
212eda14cbcSMatt Macy 		DO64(drr_spill.drr_length);
213eda14cbcSMatt Macy 		DO64(drr_spill.drr_toguid);
214eda14cbcSMatt Macy 		DO64(drr_spill.drr_compressed_size);
215eda14cbcSMatt Macy 		DO32(drr_spill.drr_type);
216eda14cbcSMatt Macy 		break;
217eda14cbcSMatt Macy 	case DRR_OBJECT_RANGE:
218eda14cbcSMatt Macy 		DO64(drr_object_range.drr_firstobj);
219eda14cbcSMatt Macy 		DO64(drr_object_range.drr_numslots);
220eda14cbcSMatt Macy 		DO64(drr_object_range.drr_toguid);
221eda14cbcSMatt Macy 		break;
222eda14cbcSMatt Macy 	case DRR_REDACT:
223eda14cbcSMatt Macy 		DO64(drr_redact.drr_object);
224eda14cbcSMatt Macy 		DO64(drr_redact.drr_offset);
225eda14cbcSMatt Macy 		DO64(drr_redact.drr_length);
226eda14cbcSMatt Macy 		DO64(drr_redact.drr_toguid);
227eda14cbcSMatt Macy 		break;
228eda14cbcSMatt Macy 	case DRR_END:
229eda14cbcSMatt Macy 		DO64(drr_end.drr_toguid);
230eda14cbcSMatt Macy 		ZIO_CHECKSUM_BSWAP(&drr->drr_u.drr_end.drr_checksum);
231eda14cbcSMatt Macy 		break;
232eda14cbcSMatt Macy 	default:
233eda14cbcSMatt Macy 		break;
234eda14cbcSMatt Macy 	}
235eda14cbcSMatt Macy 
236eda14cbcSMatt Macy 	if (drr->drr_type != DRR_BEGIN) {
237eda14cbcSMatt Macy 		ZIO_CHECKSUM_BSWAP(&drr->drr_u.drr_checksum.drr_checksum);
238eda14cbcSMatt Macy 	}
239eda14cbcSMatt Macy 
240eda14cbcSMatt Macy #undef DO64
241eda14cbcSMatt Macy #undef DO32
242eda14cbcSMatt Macy }
243eda14cbcSMatt Macy 
244eda14cbcSMatt Macy static boolean_t
redact_snaps_contains(uint64_t * snaps,uint64_t num_snaps,uint64_t guid)245eda14cbcSMatt Macy redact_snaps_contains(uint64_t *snaps, uint64_t num_snaps, uint64_t guid)
246eda14cbcSMatt Macy {
247eda14cbcSMatt Macy 	for (int i = 0; i < num_snaps; i++) {
248eda14cbcSMatt Macy 		if (snaps[i] == guid)
249eda14cbcSMatt Macy 			return (B_TRUE);
250eda14cbcSMatt Macy 	}
251eda14cbcSMatt Macy 	return (B_FALSE);
252eda14cbcSMatt Macy }
253eda14cbcSMatt Macy 
254eda14cbcSMatt Macy /*
255eda14cbcSMatt Macy  * Check that the new stream we're trying to receive is redacted with respect to
256eda14cbcSMatt Macy  * a subset of the snapshots that the origin was redacted with respect to.  For
257eda14cbcSMatt Macy  * the reasons behind this, see the man page on redacted zfs sends and receives.
258eda14cbcSMatt Macy  */
259eda14cbcSMatt Macy static boolean_t
compatible_redact_snaps(uint64_t * origin_snaps,uint64_t origin_num_snaps,uint64_t * redact_snaps,uint64_t num_redact_snaps)260eda14cbcSMatt Macy compatible_redact_snaps(uint64_t *origin_snaps, uint64_t origin_num_snaps,
261eda14cbcSMatt Macy     uint64_t *redact_snaps, uint64_t num_redact_snaps)
262eda14cbcSMatt Macy {
263eda14cbcSMatt Macy 	/*
264eda14cbcSMatt Macy 	 * Short circuit the comparison; if we are redacted with respect to
265eda14cbcSMatt Macy 	 * more snapshots than the origin, we can't be redacted with respect
266eda14cbcSMatt Macy 	 * to a subset.
267eda14cbcSMatt Macy 	 */
268eda14cbcSMatt Macy 	if (num_redact_snaps > origin_num_snaps) {
269eda14cbcSMatt Macy 		return (B_FALSE);
270eda14cbcSMatt Macy 	}
271eda14cbcSMatt Macy 
272eda14cbcSMatt Macy 	for (int i = 0; i < num_redact_snaps; i++) {
273eda14cbcSMatt Macy 		if (!redact_snaps_contains(origin_snaps, origin_num_snaps,
274eda14cbcSMatt Macy 		    redact_snaps[i])) {
275eda14cbcSMatt Macy 			return (B_FALSE);
276eda14cbcSMatt Macy 		}
277eda14cbcSMatt Macy 	}
278eda14cbcSMatt Macy 	return (B_TRUE);
279eda14cbcSMatt Macy }
280eda14cbcSMatt Macy 
281eda14cbcSMatt Macy static boolean_t
redact_check(dmu_recv_begin_arg_t * drba,dsl_dataset_t * origin)282eda14cbcSMatt Macy redact_check(dmu_recv_begin_arg_t *drba, dsl_dataset_t *origin)
283eda14cbcSMatt Macy {
284eda14cbcSMatt Macy 	uint64_t *origin_snaps;
285eda14cbcSMatt Macy 	uint64_t origin_num_snaps;
286eda14cbcSMatt Macy 	dmu_recv_cookie_t *drc = drba->drba_cookie;
287eda14cbcSMatt Macy 	struct drr_begin *drrb = drc->drc_drrb;
288eda14cbcSMatt Macy 	int featureflags = DMU_GET_FEATUREFLAGS(drrb->drr_versioninfo);
289eda14cbcSMatt Macy 	int err = 0;
290eda14cbcSMatt Macy 	boolean_t ret = B_TRUE;
291eda14cbcSMatt Macy 	uint64_t *redact_snaps;
292eda14cbcSMatt Macy 	uint_t numredactsnaps;
293eda14cbcSMatt Macy 
294eda14cbcSMatt Macy 	/*
295eda14cbcSMatt Macy 	 * If this is a full send stream, we're safe no matter what.
296eda14cbcSMatt Macy 	 */
297eda14cbcSMatt Macy 	if (drrb->drr_fromguid == 0)
298eda14cbcSMatt Macy 		return (ret);
299eda14cbcSMatt Macy 
300eda14cbcSMatt Macy 	VERIFY(dsl_dataset_get_uint64_array_feature(origin,
301eda14cbcSMatt Macy 	    SPA_FEATURE_REDACTED_DATASETS, &origin_num_snaps, &origin_snaps));
302eda14cbcSMatt Macy 
303eda14cbcSMatt Macy 	if (nvlist_lookup_uint64_array(drc->drc_begin_nvl,
304eda14cbcSMatt Macy 	    BEGINNV_REDACT_FROM_SNAPS, &redact_snaps, &numredactsnaps) ==
305eda14cbcSMatt Macy 	    0) {
306eda14cbcSMatt Macy 		/*
307eda14cbcSMatt Macy 		 * If the send stream was sent from the redaction bookmark or
308eda14cbcSMatt Macy 		 * the redacted version of the dataset, then we're safe.  Verify
309eda14cbcSMatt Macy 		 * that this is from the a compatible redaction bookmark or
310eda14cbcSMatt Macy 		 * redacted dataset.
311eda14cbcSMatt Macy 		 */
312eda14cbcSMatt Macy 		if (!compatible_redact_snaps(origin_snaps, origin_num_snaps,
313eda14cbcSMatt Macy 		    redact_snaps, numredactsnaps)) {
314eda14cbcSMatt Macy 			err = EINVAL;
315eda14cbcSMatt Macy 		}
316eda14cbcSMatt Macy 	} else if (featureflags & DMU_BACKUP_FEATURE_REDACTED) {
317eda14cbcSMatt Macy 		/*
318eda14cbcSMatt Macy 		 * If the stream is redacted, it must be redacted with respect
319eda14cbcSMatt Macy 		 * to a subset of what the origin is redacted with respect to.
320eda14cbcSMatt Macy 		 * See case number 2 in the zfs man page section on redacted zfs
321eda14cbcSMatt Macy 		 * send.
322eda14cbcSMatt Macy 		 */
323eda14cbcSMatt Macy 		err = nvlist_lookup_uint64_array(drc->drc_begin_nvl,
324eda14cbcSMatt Macy 		    BEGINNV_REDACT_SNAPS, &redact_snaps, &numredactsnaps);
325eda14cbcSMatt Macy 
326eda14cbcSMatt Macy 		if (err != 0 || !compatible_redact_snaps(origin_snaps,
327eda14cbcSMatt Macy 		    origin_num_snaps, redact_snaps, numredactsnaps)) {
328eda14cbcSMatt Macy 			err = EINVAL;
329eda14cbcSMatt Macy 		}
330eda14cbcSMatt Macy 	} else if (!redact_snaps_contains(origin_snaps, origin_num_snaps,
331eda14cbcSMatt Macy 	    drrb->drr_toguid)) {
332eda14cbcSMatt Macy 		/*
333eda14cbcSMatt Macy 		 * If the stream isn't redacted but the origin is, this must be
334eda14cbcSMatt Macy 		 * one of the snapshots the origin is redacted with respect to.
335eda14cbcSMatt Macy 		 * See case number 1 in the zfs man page section on redacted zfs
336eda14cbcSMatt Macy 		 * send.
337eda14cbcSMatt Macy 		 */
338eda14cbcSMatt Macy 		err = EINVAL;
339eda14cbcSMatt Macy 	}
340eda14cbcSMatt Macy 
341eda14cbcSMatt Macy 	if (err != 0)
342eda14cbcSMatt Macy 		ret = B_FALSE;
343eda14cbcSMatt Macy 	return (ret);
344eda14cbcSMatt Macy }
345eda14cbcSMatt Macy 
346eda14cbcSMatt Macy /*
347eda14cbcSMatt Macy  * If we previously received a stream with --large-block, we don't support
348eda14cbcSMatt Macy  * receiving an incremental on top of it without --large-block.  This avoids
349eda14cbcSMatt Macy  * forcing a read-modify-write or trying to re-aggregate a string of WRITE
350eda14cbcSMatt Macy  * records.
351eda14cbcSMatt Macy  */
352eda14cbcSMatt Macy static int
recv_check_large_blocks(dsl_dataset_t * ds,uint64_t featureflags)353eda14cbcSMatt Macy recv_check_large_blocks(dsl_dataset_t *ds, uint64_t featureflags)
354eda14cbcSMatt Macy {
355eda14cbcSMatt Macy 	if (dsl_dataset_feature_is_active(ds, SPA_FEATURE_LARGE_BLOCKS) &&
356eda14cbcSMatt Macy 	    !(featureflags & DMU_BACKUP_FEATURE_LARGE_BLOCKS))
357eda14cbcSMatt Macy 		return (SET_ERROR(ZFS_ERR_STREAM_LARGE_BLOCK_MISMATCH));
358eda14cbcSMatt Macy 	return (0);
359eda14cbcSMatt Macy }
360eda14cbcSMatt Macy 
361eda14cbcSMatt Macy static int
recv_begin_check_existing_impl(dmu_recv_begin_arg_t * drba,dsl_dataset_t * ds,uint64_t fromguid,uint64_t featureflags)362eda14cbcSMatt Macy recv_begin_check_existing_impl(dmu_recv_begin_arg_t *drba, dsl_dataset_t *ds,
363eda14cbcSMatt Macy     uint64_t fromguid, uint64_t featureflags)
364eda14cbcSMatt Macy {
365271171e0SMartin Matuska 	uint64_t obj;
366eda14cbcSMatt Macy 	uint64_t children;
367eda14cbcSMatt Macy 	int error;
368271171e0SMartin Matuska 	dsl_dataset_t *snap;
369eda14cbcSMatt Macy 	dsl_pool_t *dp = ds->ds_dir->dd_pool;
370eda14cbcSMatt Macy 	boolean_t encrypted = ds->ds_dir->dd_crypto_obj != 0;
371eda14cbcSMatt Macy 	boolean_t raw = (featureflags & DMU_BACKUP_FEATURE_RAW) != 0;
372eda14cbcSMatt Macy 	boolean_t embed = (featureflags & DMU_BACKUP_FEATURE_EMBED_DATA) != 0;
373eda14cbcSMatt Macy 
374eda14cbcSMatt Macy 	/* Temporary clone name must not exist. */
375eda14cbcSMatt Macy 	error = zap_lookup(dp->dp_meta_objset,
376eda14cbcSMatt Macy 	    dsl_dir_phys(ds->ds_dir)->dd_child_dir_zapobj, recv_clone_name,
377271171e0SMartin Matuska 	    8, 1, &obj);
378eda14cbcSMatt Macy 	if (error != ENOENT)
379eda14cbcSMatt Macy 		return (error == 0 ? SET_ERROR(EBUSY) : error);
380eda14cbcSMatt Macy 
381eda14cbcSMatt Macy 	/* Resume state must not be set. */
382eda14cbcSMatt Macy 	if (dsl_dataset_has_resume_receive_state(ds))
383eda14cbcSMatt Macy 		return (SET_ERROR(EBUSY));
384eda14cbcSMatt Macy 
385271171e0SMartin Matuska 	/* New snapshot name must not exist if we're not healing it. */
386eda14cbcSMatt Macy 	error = zap_lookup(dp->dp_meta_objset,
387eda14cbcSMatt Macy 	    dsl_dataset_phys(ds)->ds_snapnames_zapobj,
388271171e0SMartin Matuska 	    drba->drba_cookie->drc_tosnap, 8, 1, &obj);
389271171e0SMartin Matuska 	if (drba->drba_cookie->drc_heal) {
390271171e0SMartin Matuska 		if (error != 0)
391271171e0SMartin Matuska 			return (error);
392271171e0SMartin Matuska 	} else if (error != ENOENT) {
393eda14cbcSMatt Macy 		return (error == 0 ? SET_ERROR(EEXIST) : error);
394271171e0SMartin Matuska 	}
395eda14cbcSMatt Macy 
396eda14cbcSMatt Macy 	/* Must not have children if receiving a ZVOL. */
397eda14cbcSMatt Macy 	error = zap_count(dp->dp_meta_objset,
398eda14cbcSMatt Macy 	    dsl_dir_phys(ds->ds_dir)->dd_child_dir_zapobj, &children);
399eda14cbcSMatt Macy 	if (error != 0)
400eda14cbcSMatt Macy 		return (error);
401eda14cbcSMatt Macy 	if (drba->drba_cookie->drc_drrb->drr_type != DMU_OST_ZFS &&
402eda14cbcSMatt Macy 	    children > 0)
403eda14cbcSMatt Macy 		return (SET_ERROR(ZFS_ERR_WRONG_PARENT));
404eda14cbcSMatt Macy 
405eda14cbcSMatt Macy 	/*
406eda14cbcSMatt Macy 	 * Check snapshot limit before receiving. We'll recheck again at the
407eda14cbcSMatt Macy 	 * end, but might as well abort before receiving if we're already over
408eda14cbcSMatt Macy 	 * the limit.
409eda14cbcSMatt Macy 	 *
410eda14cbcSMatt Macy 	 * Note that we do not check the file system limit with
411eda14cbcSMatt Macy 	 * dsl_dir_fscount_check because the temporary %clones don't count
412eda14cbcSMatt Macy 	 * against that limit.
413eda14cbcSMatt Macy 	 */
414eda14cbcSMatt Macy 	error = dsl_fs_ss_limit_check(ds->ds_dir, 1, ZFS_PROP_SNAPSHOT_LIMIT,
415*071ab5a1SMartin Matuska 	    NULL, drba->drba_cred);
416eda14cbcSMatt Macy 	if (error != 0)
417eda14cbcSMatt Macy 		return (error);
418eda14cbcSMatt Macy 
419271171e0SMartin Matuska 	if (drba->drba_cookie->drc_heal) {
420271171e0SMartin Matuska 		/* Encryption is incompatible with embedded data. */
421271171e0SMartin Matuska 		if (encrypted && embed)
422271171e0SMartin Matuska 			return (SET_ERROR(EINVAL));
423271171e0SMartin Matuska 
424271171e0SMartin Matuska 		/* Healing is not supported when in 'force' mode. */
425271171e0SMartin Matuska 		if (drba->drba_cookie->drc_force)
426271171e0SMartin Matuska 			return (SET_ERROR(EINVAL));
427271171e0SMartin Matuska 
428271171e0SMartin Matuska 		/* Must have keys loaded if doing encrypted non-raw recv. */
429271171e0SMartin Matuska 		if (encrypted && !raw) {
430271171e0SMartin Matuska 			if (spa_keystore_lookup_key(dp->dp_spa, ds->ds_object,
431271171e0SMartin Matuska 			    NULL, NULL) != 0)
432271171e0SMartin Matuska 				return (SET_ERROR(EACCES));
433271171e0SMartin Matuska 		}
434271171e0SMartin Matuska 
435271171e0SMartin Matuska 		error = dsl_dataset_hold_obj(dp, obj, FTAG, &snap);
436271171e0SMartin Matuska 		if (error != 0)
437271171e0SMartin Matuska 			return (error);
438271171e0SMartin Matuska 
439271171e0SMartin Matuska 		/*
440271171e0SMartin Matuska 		 * When not doing best effort corrective recv healing can only
441271171e0SMartin Matuska 		 * be done if the send stream is for the same snapshot as the
442271171e0SMartin Matuska 		 * one we are trying to heal.
443271171e0SMartin Matuska 		 */
444271171e0SMartin Matuska 		if (zfs_recv_best_effort_corrective == 0 &&
445271171e0SMartin Matuska 		    drba->drba_cookie->drc_drrb->drr_toguid !=
446271171e0SMartin Matuska 		    dsl_dataset_phys(snap)->ds_guid) {
447271171e0SMartin Matuska 			dsl_dataset_rele(snap, FTAG);
448271171e0SMartin Matuska 			return (SET_ERROR(ENOTSUP));
449271171e0SMartin Matuska 		}
450271171e0SMartin Matuska 		dsl_dataset_rele(snap, FTAG);
451271171e0SMartin Matuska 	} else if (fromguid != 0) {
452271171e0SMartin Matuska 		/* Sanity check the incremental recv */
453eda14cbcSMatt Macy 		uint64_t obj = dsl_dataset_phys(ds)->ds_prev_snap_obj;
454eda14cbcSMatt Macy 
455eda14cbcSMatt Macy 		/* Can't perform a raw receive on top of a non-raw receive */
456eda14cbcSMatt Macy 		if (!encrypted && raw)
457eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
458eda14cbcSMatt Macy 
459eda14cbcSMatt Macy 		/* Encryption is incompatible with embedded data */
460eda14cbcSMatt Macy 		if (encrypted && embed)
461eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
462eda14cbcSMatt Macy 
463eda14cbcSMatt Macy 		/* Find snapshot in this dir that matches fromguid. */
464eda14cbcSMatt Macy 		while (obj != 0) {
465eda14cbcSMatt Macy 			error = dsl_dataset_hold_obj(dp, obj, FTAG,
466eda14cbcSMatt Macy 			    &snap);
467eda14cbcSMatt Macy 			if (error != 0)
468eda14cbcSMatt Macy 				return (SET_ERROR(ENODEV));
469eda14cbcSMatt Macy 			if (snap->ds_dir != ds->ds_dir) {
470eda14cbcSMatt Macy 				dsl_dataset_rele(snap, FTAG);
471eda14cbcSMatt Macy 				return (SET_ERROR(ENODEV));
472eda14cbcSMatt Macy 			}
473eda14cbcSMatt Macy 			if (dsl_dataset_phys(snap)->ds_guid == fromguid)
474eda14cbcSMatt Macy 				break;
475eda14cbcSMatt Macy 			obj = dsl_dataset_phys(snap)->ds_prev_snap_obj;
476eda14cbcSMatt Macy 			dsl_dataset_rele(snap, FTAG);
477eda14cbcSMatt Macy 		}
478eda14cbcSMatt Macy 		if (obj == 0)
479eda14cbcSMatt Macy 			return (SET_ERROR(ENODEV));
480eda14cbcSMatt Macy 
481eda14cbcSMatt Macy 		if (drba->drba_cookie->drc_force) {
482eda14cbcSMatt Macy 			drba->drba_cookie->drc_fromsnapobj = obj;
483eda14cbcSMatt Macy 		} else {
484eda14cbcSMatt Macy 			/*
485eda14cbcSMatt Macy 			 * If we are not forcing, there must be no
486eda14cbcSMatt Macy 			 * changes since fromsnap. Raw sends have an
487eda14cbcSMatt Macy 			 * additional constraint that requires that
488eda14cbcSMatt Macy 			 * no "noop" snapshots exist between fromsnap
489eda14cbcSMatt Macy 			 * and tosnap for the IVset checking code to
490eda14cbcSMatt Macy 			 * work properly.
491eda14cbcSMatt Macy 			 */
492eda14cbcSMatt Macy 			if (dsl_dataset_modified_since_snap(ds, snap) ||
493eda14cbcSMatt Macy 			    (raw &&
494eda14cbcSMatt Macy 			    dsl_dataset_phys(ds)->ds_prev_snap_obj !=
495eda14cbcSMatt Macy 			    snap->ds_object)) {
496eda14cbcSMatt Macy 				dsl_dataset_rele(snap, FTAG);
497eda14cbcSMatt Macy 				return (SET_ERROR(ETXTBSY));
498eda14cbcSMatt Macy 			}
499eda14cbcSMatt Macy 			drba->drba_cookie->drc_fromsnapobj =
500eda14cbcSMatt Macy 			    ds->ds_prev->ds_object;
501eda14cbcSMatt Macy 		}
502eda14cbcSMatt Macy 
503eda14cbcSMatt Macy 		if (dsl_dataset_feature_is_active(snap,
504eda14cbcSMatt Macy 		    SPA_FEATURE_REDACTED_DATASETS) && !redact_check(drba,
505eda14cbcSMatt Macy 		    snap)) {
506eda14cbcSMatt Macy 			dsl_dataset_rele(snap, FTAG);
507eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
508eda14cbcSMatt Macy 		}
509eda14cbcSMatt Macy 
510eda14cbcSMatt Macy 		error = recv_check_large_blocks(snap, featureflags);
511eda14cbcSMatt Macy 		if (error != 0) {
512eda14cbcSMatt Macy 			dsl_dataset_rele(snap, FTAG);
513eda14cbcSMatt Macy 			return (error);
514eda14cbcSMatt Macy 		}
515eda14cbcSMatt Macy 
516eda14cbcSMatt Macy 		dsl_dataset_rele(snap, FTAG);
517eda14cbcSMatt Macy 	} else {
518271171e0SMartin Matuska 		/* If full and not healing then must be forced. */
519eda14cbcSMatt Macy 		if (!drba->drba_cookie->drc_force)
520eda14cbcSMatt Macy 			return (SET_ERROR(EEXIST));
521eda14cbcSMatt Macy 
522eda14cbcSMatt Macy 		/*
523eda14cbcSMatt Macy 		 * We don't support using zfs recv -F to blow away
524eda14cbcSMatt Macy 		 * encrypted filesystems. This would require the
525eda14cbcSMatt Macy 		 * dsl dir to point to the old encryption key and
526eda14cbcSMatt Macy 		 * the new one at the same time during the receive.
527eda14cbcSMatt Macy 		 */
528eda14cbcSMatt Macy 		if ((!encrypted && raw) || encrypted)
529eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
530eda14cbcSMatt Macy 
531eda14cbcSMatt Macy 		/*
532eda14cbcSMatt Macy 		 * Perform the same encryption checks we would if
533eda14cbcSMatt Macy 		 * we were creating a new dataset from scratch.
534eda14cbcSMatt Macy 		 */
535eda14cbcSMatt Macy 		if (!raw) {
536eda14cbcSMatt Macy 			boolean_t will_encrypt;
537eda14cbcSMatt Macy 
538eda14cbcSMatt Macy 			error = dmu_objset_create_crypt_check(
539eda14cbcSMatt Macy 			    ds->ds_dir->dd_parent, drba->drba_dcp,
540eda14cbcSMatt Macy 			    &will_encrypt);
541eda14cbcSMatt Macy 			if (error != 0)
542eda14cbcSMatt Macy 				return (error);
543eda14cbcSMatt Macy 
544eda14cbcSMatt Macy 			if (will_encrypt && embed)
545eda14cbcSMatt Macy 				return (SET_ERROR(EINVAL));
546eda14cbcSMatt Macy 		}
547eda14cbcSMatt Macy 	}
548eda14cbcSMatt Macy 
549eda14cbcSMatt Macy 	return (0);
550eda14cbcSMatt Macy }
551eda14cbcSMatt Macy 
552eda14cbcSMatt Macy /*
553eda14cbcSMatt Macy  * Check that any feature flags used in the data stream we're receiving are
554eda14cbcSMatt Macy  * supported by the pool we are receiving into.
555eda14cbcSMatt Macy  *
556eda14cbcSMatt Macy  * Note that some of the features we explicitly check here have additional
557eda14cbcSMatt Macy  * (implicit) features they depend on, but those dependencies are enforced
558eda14cbcSMatt Macy  * through the zfeature_register() calls declaring the features that we
559eda14cbcSMatt Macy  * explicitly check.
560eda14cbcSMatt Macy  */
561eda14cbcSMatt Macy static int
recv_begin_check_feature_flags_impl(uint64_t featureflags,spa_t * spa)562eda14cbcSMatt Macy recv_begin_check_feature_flags_impl(uint64_t featureflags, spa_t *spa)
563eda14cbcSMatt Macy {
564eda14cbcSMatt Macy 	/*
565eda14cbcSMatt Macy 	 * Check if there are any unsupported feature flags.
566eda14cbcSMatt Macy 	 */
567eda14cbcSMatt Macy 	if (!DMU_STREAM_SUPPORTED(featureflags)) {
568eda14cbcSMatt Macy 		return (SET_ERROR(ZFS_ERR_UNKNOWN_SEND_STREAM_FEATURE));
569eda14cbcSMatt Macy 	}
570eda14cbcSMatt Macy 
571eda14cbcSMatt Macy 	/* Verify pool version supports SA if SA_SPILL feature set */
572eda14cbcSMatt Macy 	if ((featureflags & DMU_BACKUP_FEATURE_SA_SPILL) &&
573eda14cbcSMatt Macy 	    spa_version(spa) < SPA_VERSION_SA)
574eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
575eda14cbcSMatt Macy 
576eda14cbcSMatt Macy 	/*
577eda14cbcSMatt Macy 	 * LZ4 compressed, ZSTD compressed, embedded, mooched, large blocks,
578eda14cbcSMatt Macy 	 * and large_dnodes in the stream can only be used if those pool
579eda14cbcSMatt Macy 	 * features are enabled because we don't attempt to decompress /
580eda14cbcSMatt Macy 	 * un-embed / un-mooch / split up the blocks / dnodes during the
581eda14cbcSMatt Macy 	 * receive process.
582eda14cbcSMatt Macy 	 */
583eda14cbcSMatt Macy 	if ((featureflags & DMU_BACKUP_FEATURE_LZ4) &&
584eda14cbcSMatt Macy 	    !spa_feature_is_enabled(spa, SPA_FEATURE_LZ4_COMPRESS))
585eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
586eda14cbcSMatt Macy 	if ((featureflags & DMU_BACKUP_FEATURE_ZSTD) &&
587eda14cbcSMatt Macy 	    !spa_feature_is_enabled(spa, SPA_FEATURE_ZSTD_COMPRESS))
588eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
589eda14cbcSMatt Macy 	if ((featureflags & DMU_BACKUP_FEATURE_EMBED_DATA) &&
590eda14cbcSMatt Macy 	    !spa_feature_is_enabled(spa, SPA_FEATURE_EMBEDDED_DATA))
591eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
592eda14cbcSMatt Macy 	if ((featureflags & DMU_BACKUP_FEATURE_LARGE_BLOCKS) &&
593eda14cbcSMatt Macy 	    !spa_feature_is_enabled(spa, SPA_FEATURE_LARGE_BLOCKS))
594eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
595eda14cbcSMatt Macy 	if ((featureflags & DMU_BACKUP_FEATURE_LARGE_DNODE) &&
596eda14cbcSMatt Macy 	    !spa_feature_is_enabled(spa, SPA_FEATURE_LARGE_DNODE))
597eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
5987a7741afSMartin Matuska 	if ((featureflags & DMU_BACKUP_FEATURE_LARGE_MICROZAP) &&
5997a7741afSMartin Matuska 	    !spa_feature_is_enabled(spa, SPA_FEATURE_LARGE_MICROZAP))
6007a7741afSMartin Matuska 		return (SET_ERROR(ENOTSUP));
601eda14cbcSMatt Macy 
602eda14cbcSMatt Macy 	/*
603eda14cbcSMatt Macy 	 * Receiving redacted streams requires that redacted datasets are
604eda14cbcSMatt Macy 	 * enabled.
605eda14cbcSMatt Macy 	 */
606eda14cbcSMatt Macy 	if ((featureflags & DMU_BACKUP_FEATURE_REDACTED) &&
607eda14cbcSMatt Macy 	    !spa_feature_is_enabled(spa, SPA_FEATURE_REDACTED_DATASETS))
608eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
609eda14cbcSMatt Macy 
6107a7741afSMartin Matuska 	/*
6117a7741afSMartin Matuska 	 * If the LONGNAME is not enabled on the target, fail that request.
6127a7741afSMartin Matuska 	 */
6137a7741afSMartin Matuska 	if ((featureflags & DMU_BACKUP_FEATURE_LONGNAME) &&
6147a7741afSMartin Matuska 	    !spa_feature_is_enabled(spa, SPA_FEATURE_LONGNAME))
6157a7741afSMartin Matuska 		return (SET_ERROR(ENOTSUP));
6167a7741afSMartin Matuska 
617eda14cbcSMatt Macy 	return (0);
618eda14cbcSMatt Macy }
619eda14cbcSMatt Macy 
620eda14cbcSMatt Macy static int
dmu_recv_begin_check(void * arg,dmu_tx_t * tx)621eda14cbcSMatt Macy dmu_recv_begin_check(void *arg, dmu_tx_t *tx)
622eda14cbcSMatt Macy {
623eda14cbcSMatt Macy 	dmu_recv_begin_arg_t *drba = arg;
624eda14cbcSMatt Macy 	dsl_pool_t *dp = dmu_tx_pool(tx);
625eda14cbcSMatt Macy 	struct drr_begin *drrb = drba->drba_cookie->drc_drrb;
626eda14cbcSMatt Macy 	uint64_t fromguid = drrb->drr_fromguid;
627eda14cbcSMatt Macy 	int flags = drrb->drr_flags;
6287877fdebSMatt Macy 	ds_hold_flags_t dsflags = DS_HOLD_FLAG_NONE;
629eda14cbcSMatt Macy 	int error;
630eda14cbcSMatt Macy 	uint64_t featureflags = drba->drba_cookie->drc_featureflags;
631eda14cbcSMatt Macy 	dsl_dataset_t *ds;
632eda14cbcSMatt Macy 	const char *tofs = drba->drba_cookie->drc_tofs;
633eda14cbcSMatt Macy 
634eda14cbcSMatt Macy 	/* already checked */
635eda14cbcSMatt Macy 	ASSERT3U(drrb->drr_magic, ==, DMU_BACKUP_MAGIC);
636eda14cbcSMatt Macy 	ASSERT(!(featureflags & DMU_BACKUP_FEATURE_RESUMING));
637eda14cbcSMatt Macy 
638eda14cbcSMatt Macy 	if (DMU_GET_STREAM_HDRTYPE(drrb->drr_versioninfo) ==
639eda14cbcSMatt Macy 	    DMU_COMPOUNDSTREAM ||
640eda14cbcSMatt Macy 	    drrb->drr_type >= DMU_OST_NUMTYPES ||
641eda14cbcSMatt Macy 	    ((flags & DRR_FLAG_CLONE) && drba->drba_origin == NULL))
642eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
643eda14cbcSMatt Macy 
644eda14cbcSMatt Macy 	error = recv_begin_check_feature_flags_impl(featureflags, dp->dp_spa);
645eda14cbcSMatt Macy 	if (error != 0)
646eda14cbcSMatt Macy 		return (error);
647eda14cbcSMatt Macy 
648eda14cbcSMatt Macy 	/* Resumable receives require extensible datasets */
649eda14cbcSMatt Macy 	if (drba->drba_cookie->drc_resumable &&
650eda14cbcSMatt Macy 	    !spa_feature_is_enabled(dp->dp_spa, SPA_FEATURE_EXTENSIBLE_DATASET))
651eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
652eda14cbcSMatt Macy 
653eda14cbcSMatt Macy 	if (featureflags & DMU_BACKUP_FEATURE_RAW) {
654eda14cbcSMatt Macy 		/* raw receives require the encryption feature */
655eda14cbcSMatt Macy 		if (!spa_feature_is_enabled(dp->dp_spa, SPA_FEATURE_ENCRYPTION))
656eda14cbcSMatt Macy 			return (SET_ERROR(ENOTSUP));
657eda14cbcSMatt Macy 
658eda14cbcSMatt Macy 		/* embedded data is incompatible with encryption and raw recv */
659eda14cbcSMatt Macy 		if (featureflags & DMU_BACKUP_FEATURE_EMBED_DATA)
660eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
661eda14cbcSMatt Macy 
662eda14cbcSMatt Macy 		/* raw receives require spill block allocation flag */
663eda14cbcSMatt Macy 		if (!(flags & DRR_FLAG_SPILL_BLOCK))
664eda14cbcSMatt Macy 			return (SET_ERROR(ZFS_ERR_SPILL_BLOCK_FLAG_MISSING));
665eda14cbcSMatt Macy 	} else {
666c03c5b1cSMartin Matuska 		/*
667c03c5b1cSMartin Matuska 		 * We support unencrypted datasets below encrypted ones now,
668c03c5b1cSMartin Matuska 		 * so add the DS_HOLD_FLAG_DECRYPT flag only if we are dealing
669c03c5b1cSMartin Matuska 		 * with a dataset we may encrypt.
670c03c5b1cSMartin Matuska 		 */
671dbd5678dSMartin Matuska 		if (drba->drba_dcp == NULL ||
672c03c5b1cSMartin Matuska 		    drba->drba_dcp->cp_crypt != ZIO_CRYPT_OFF) {
673eda14cbcSMatt Macy 			dsflags |= DS_HOLD_FLAG_DECRYPT;
674eda14cbcSMatt Macy 		}
675c03c5b1cSMartin Matuska 	}
676eda14cbcSMatt Macy 
677eda14cbcSMatt Macy 	error = dsl_dataset_hold_flags(dp, tofs, dsflags, FTAG, &ds);
678eda14cbcSMatt Macy 	if (error == 0) {
679eda14cbcSMatt Macy 		/* target fs already exists; recv into temp clone */
680eda14cbcSMatt Macy 
681eda14cbcSMatt Macy 		/* Can't recv a clone into an existing fs */
682eda14cbcSMatt Macy 		if (flags & DRR_FLAG_CLONE || drba->drba_origin) {
683eda14cbcSMatt Macy 			dsl_dataset_rele_flags(ds, dsflags, FTAG);
684eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
685eda14cbcSMatt Macy 		}
686eda14cbcSMatt Macy 
687eda14cbcSMatt Macy 		error = recv_begin_check_existing_impl(drba, ds, fromguid,
688eda14cbcSMatt Macy 		    featureflags);
689eda14cbcSMatt Macy 		dsl_dataset_rele_flags(ds, dsflags, FTAG);
690eda14cbcSMatt Macy 	} else if (error == ENOENT) {
691eda14cbcSMatt Macy 		/* target fs does not exist; must be a full backup or clone */
692eda14cbcSMatt Macy 		char buf[ZFS_MAX_DATASET_NAME_LEN];
693eda14cbcSMatt Macy 		objset_t *os;
694eda14cbcSMatt Macy 
695271171e0SMartin Matuska 		/* healing recv must be done "into" an existing snapshot */
696271171e0SMartin Matuska 		if (drba->drba_cookie->drc_heal == B_TRUE)
697271171e0SMartin Matuska 			return (SET_ERROR(ENOTSUP));
698271171e0SMartin Matuska 
699eda14cbcSMatt Macy 		/*
700eda14cbcSMatt Macy 		 * If it's a non-clone incremental, we are missing the
701eda14cbcSMatt Macy 		 * target fs, so fail the recv.
702eda14cbcSMatt Macy 		 */
703eda14cbcSMatt Macy 		if (fromguid != 0 && !((flags & DRR_FLAG_CLONE) ||
704eda14cbcSMatt Macy 		    drba->drba_origin))
705eda14cbcSMatt Macy 			return (SET_ERROR(ENOENT));
706eda14cbcSMatt Macy 
707eda14cbcSMatt Macy 		/*
708eda14cbcSMatt Macy 		 * If we're receiving a full send as a clone, and it doesn't
709eda14cbcSMatt Macy 		 * contain all the necessary free records and freeobject
710eda14cbcSMatt Macy 		 * records, reject it.
711eda14cbcSMatt Macy 		 */
712eda14cbcSMatt Macy 		if (fromguid == 0 && drba->drba_origin != NULL &&
713eda14cbcSMatt Macy 		    !(flags & DRR_FLAG_FREERECORDS))
714eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
715eda14cbcSMatt Macy 
716eda14cbcSMatt Macy 		/* Open the parent of tofs */
717eda14cbcSMatt Macy 		ASSERT3U(strlen(tofs), <, sizeof (buf));
718eda14cbcSMatt Macy 		(void) strlcpy(buf, tofs, strrchr(tofs, '/') - tofs + 1);
719eda14cbcSMatt Macy 		error = dsl_dataset_hold(dp, buf, FTAG, &ds);
720eda14cbcSMatt Macy 		if (error != 0)
721eda14cbcSMatt Macy 			return (error);
722eda14cbcSMatt Macy 
723eda14cbcSMatt Macy 		if ((featureflags & DMU_BACKUP_FEATURE_RAW) == 0 &&
724eda14cbcSMatt Macy 		    drba->drba_origin == NULL) {
725eda14cbcSMatt Macy 			boolean_t will_encrypt;
726eda14cbcSMatt Macy 
727eda14cbcSMatt Macy 			/*
728eda14cbcSMatt Macy 			 * Check that we aren't breaking any encryption rules
729eda14cbcSMatt Macy 			 * and that we have all the parameters we need to
730eda14cbcSMatt Macy 			 * create an encrypted dataset if necessary. If we are
731eda14cbcSMatt Macy 			 * making an encrypted dataset the stream can't have
732eda14cbcSMatt Macy 			 * embedded data.
733eda14cbcSMatt Macy 			 */
734eda14cbcSMatt Macy 			error = dmu_objset_create_crypt_check(ds->ds_dir,
735eda14cbcSMatt Macy 			    drba->drba_dcp, &will_encrypt);
736eda14cbcSMatt Macy 			if (error != 0) {
737eda14cbcSMatt Macy 				dsl_dataset_rele(ds, FTAG);
738eda14cbcSMatt Macy 				return (error);
739eda14cbcSMatt Macy 			}
740eda14cbcSMatt Macy 
741eda14cbcSMatt Macy 			if (will_encrypt &&
742eda14cbcSMatt Macy 			    (featureflags & DMU_BACKUP_FEATURE_EMBED_DATA)) {
743eda14cbcSMatt Macy 				dsl_dataset_rele(ds, FTAG);
744eda14cbcSMatt Macy 				return (SET_ERROR(EINVAL));
745eda14cbcSMatt Macy 			}
746eda14cbcSMatt Macy 		}
747eda14cbcSMatt Macy 
748eda14cbcSMatt Macy 		/*
749eda14cbcSMatt Macy 		 * Check filesystem and snapshot limits before receiving. We'll
750eda14cbcSMatt Macy 		 * recheck snapshot limits again at the end (we create the
751eda14cbcSMatt Macy 		 * filesystems and increment those counts during begin_sync).
752eda14cbcSMatt Macy 		 */
753eda14cbcSMatt Macy 		error = dsl_fs_ss_limit_check(ds->ds_dir, 1,
754*071ab5a1SMartin Matuska 		    ZFS_PROP_FILESYSTEM_LIMIT, NULL, drba->drba_cred);
755eda14cbcSMatt Macy 		if (error != 0) {
756eda14cbcSMatt Macy 			dsl_dataset_rele(ds, FTAG);
757eda14cbcSMatt Macy 			return (error);
758eda14cbcSMatt Macy 		}
759eda14cbcSMatt Macy 
760eda14cbcSMatt Macy 		error = dsl_fs_ss_limit_check(ds->ds_dir, 1,
761*071ab5a1SMartin Matuska 		    ZFS_PROP_SNAPSHOT_LIMIT, NULL, drba->drba_cred);
762eda14cbcSMatt Macy 		if (error != 0) {
763eda14cbcSMatt Macy 			dsl_dataset_rele(ds, FTAG);
764eda14cbcSMatt Macy 			return (error);
765eda14cbcSMatt Macy 		}
766eda14cbcSMatt Macy 
767eda14cbcSMatt Macy 		/* can't recv below anything but filesystems (eg. no ZVOLs) */
768eda14cbcSMatt Macy 		error = dmu_objset_from_ds(ds, &os);
769eda14cbcSMatt Macy 		if (error != 0) {
770eda14cbcSMatt Macy 			dsl_dataset_rele(ds, FTAG);
771eda14cbcSMatt Macy 			return (error);
772eda14cbcSMatt Macy 		}
773eda14cbcSMatt Macy 		if (dmu_objset_type(os) != DMU_OST_ZFS) {
774eda14cbcSMatt Macy 			dsl_dataset_rele(ds, FTAG);
775eda14cbcSMatt Macy 			return (SET_ERROR(ZFS_ERR_WRONG_PARENT));
776eda14cbcSMatt Macy 		}
777eda14cbcSMatt Macy 
778eda14cbcSMatt Macy 		if (drba->drba_origin != NULL) {
779eda14cbcSMatt Macy 			dsl_dataset_t *origin;
780eda14cbcSMatt Macy 			error = dsl_dataset_hold_flags(dp, drba->drba_origin,
781eda14cbcSMatt Macy 			    dsflags, FTAG, &origin);
782eda14cbcSMatt Macy 			if (error != 0) {
783eda14cbcSMatt Macy 				dsl_dataset_rele(ds, FTAG);
784eda14cbcSMatt Macy 				return (error);
785eda14cbcSMatt Macy 			}
786eda14cbcSMatt Macy 			if (!origin->ds_is_snapshot) {
787eda14cbcSMatt Macy 				dsl_dataset_rele_flags(origin, dsflags, FTAG);
788eda14cbcSMatt Macy 				dsl_dataset_rele(ds, FTAG);
789eda14cbcSMatt Macy 				return (SET_ERROR(EINVAL));
790eda14cbcSMatt Macy 			}
791eda14cbcSMatt Macy 			if (dsl_dataset_phys(origin)->ds_guid != fromguid &&
792eda14cbcSMatt Macy 			    fromguid != 0) {
793eda14cbcSMatt Macy 				dsl_dataset_rele_flags(origin, dsflags, FTAG);
794eda14cbcSMatt Macy 				dsl_dataset_rele(ds, FTAG);
795eda14cbcSMatt Macy 				return (SET_ERROR(ENODEV));
796eda14cbcSMatt Macy 			}
797eda14cbcSMatt Macy 
798eda14cbcSMatt Macy 			if (origin->ds_dir->dd_crypto_obj != 0 &&
799eda14cbcSMatt Macy 			    (featureflags & DMU_BACKUP_FEATURE_EMBED_DATA)) {
800eda14cbcSMatt Macy 				dsl_dataset_rele_flags(origin, dsflags, FTAG);
801eda14cbcSMatt Macy 				dsl_dataset_rele(ds, FTAG);
802eda14cbcSMatt Macy 				return (SET_ERROR(EINVAL));
803eda14cbcSMatt Macy 			}
804eda14cbcSMatt Macy 
805eda14cbcSMatt Macy 			/*
806eda14cbcSMatt Macy 			 * If the origin is redacted we need to verify that this
807eda14cbcSMatt Macy 			 * send stream can safely be received on top of the
808eda14cbcSMatt Macy 			 * origin.
809eda14cbcSMatt Macy 			 */
810eda14cbcSMatt Macy 			if (dsl_dataset_feature_is_active(origin,
811eda14cbcSMatt Macy 			    SPA_FEATURE_REDACTED_DATASETS)) {
812eda14cbcSMatt Macy 				if (!redact_check(drba, origin)) {
813eda14cbcSMatt Macy 					dsl_dataset_rele_flags(origin, dsflags,
814eda14cbcSMatt Macy 					    FTAG);
815eda14cbcSMatt Macy 					dsl_dataset_rele_flags(ds, dsflags,
816eda14cbcSMatt Macy 					    FTAG);
817eda14cbcSMatt Macy 					return (SET_ERROR(EINVAL));
818eda14cbcSMatt Macy 				}
819eda14cbcSMatt Macy 			}
820eda14cbcSMatt Macy 
821eda14cbcSMatt Macy 			error = recv_check_large_blocks(ds, featureflags);
822eda14cbcSMatt Macy 			if (error != 0) {
823eda14cbcSMatt Macy 				dsl_dataset_rele_flags(origin, dsflags, FTAG);
824eda14cbcSMatt Macy 				dsl_dataset_rele_flags(ds, dsflags, FTAG);
825eda14cbcSMatt Macy 				return (error);
826eda14cbcSMatt Macy 			}
827eda14cbcSMatt Macy 
828eda14cbcSMatt Macy 			dsl_dataset_rele_flags(origin, dsflags, FTAG);
829eda14cbcSMatt Macy 		}
830eda14cbcSMatt Macy 
831eda14cbcSMatt Macy 		dsl_dataset_rele(ds, FTAG);
832eda14cbcSMatt Macy 		error = 0;
833eda14cbcSMatt Macy 	}
834eda14cbcSMatt Macy 	return (error);
835eda14cbcSMatt Macy }
836eda14cbcSMatt Macy 
837eda14cbcSMatt Macy static void
dmu_recv_begin_sync(void * arg,dmu_tx_t * tx)838eda14cbcSMatt Macy dmu_recv_begin_sync(void *arg, dmu_tx_t *tx)
839eda14cbcSMatt Macy {
840eda14cbcSMatt Macy 	dmu_recv_begin_arg_t *drba = arg;
841eda14cbcSMatt Macy 	dsl_pool_t *dp = dmu_tx_pool(tx);
842eda14cbcSMatt Macy 	objset_t *mos = dp->dp_meta_objset;
843eda14cbcSMatt Macy 	dmu_recv_cookie_t *drc = drba->drba_cookie;
844eda14cbcSMatt Macy 	struct drr_begin *drrb = drc->drc_drrb;
845eda14cbcSMatt Macy 	const char *tofs = drc->drc_tofs;
846eda14cbcSMatt Macy 	uint64_t featureflags = drc->drc_featureflags;
847eda14cbcSMatt Macy 	dsl_dataset_t *ds, *newds;
848eda14cbcSMatt Macy 	objset_t *os;
849eda14cbcSMatt Macy 	uint64_t dsobj;
8507877fdebSMatt Macy 	ds_hold_flags_t dsflags = DS_HOLD_FLAG_NONE;
851eda14cbcSMatt Macy 	int error;
852eda14cbcSMatt Macy 	uint64_t crflags = 0;
853eda14cbcSMatt Macy 	dsl_crypto_params_t dummy_dcp = { 0 };
854eda14cbcSMatt Macy 	dsl_crypto_params_t *dcp = drba->drba_dcp;
855eda14cbcSMatt Macy 
856eda14cbcSMatt Macy 	if (drrb->drr_flags & DRR_FLAG_CI_DATA)
857eda14cbcSMatt Macy 		crflags |= DS_FLAG_CI_DATASET;
858eda14cbcSMatt Macy 
859eda14cbcSMatt Macy 	if ((featureflags & DMU_BACKUP_FEATURE_RAW) == 0)
860eda14cbcSMatt Macy 		dsflags |= DS_HOLD_FLAG_DECRYPT;
861eda14cbcSMatt Macy 
862eda14cbcSMatt Macy 	/*
863eda14cbcSMatt Macy 	 * Raw, non-incremental recvs always use a dummy dcp with
864eda14cbcSMatt Macy 	 * the raw cmd set. Raw incremental recvs do not use a dcp
865eda14cbcSMatt Macy 	 * since the encryption parameters are already set in stone.
866eda14cbcSMatt Macy 	 */
867eda14cbcSMatt Macy 	if (dcp == NULL && drrb->drr_fromguid == 0 &&
868eda14cbcSMatt Macy 	    drba->drba_origin == NULL) {
869eda14cbcSMatt Macy 		ASSERT3P(dcp, ==, NULL);
870eda14cbcSMatt Macy 		dcp = &dummy_dcp;
871eda14cbcSMatt Macy 
872eda14cbcSMatt Macy 		if (featureflags & DMU_BACKUP_FEATURE_RAW)
873eda14cbcSMatt Macy 			dcp->cp_cmd = DCP_CMD_RAW_RECV;
874eda14cbcSMatt Macy 	}
875eda14cbcSMatt Macy 
876eda14cbcSMatt Macy 	error = dsl_dataset_hold_flags(dp, tofs, dsflags, FTAG, &ds);
877eda14cbcSMatt Macy 	if (error == 0) {
878271171e0SMartin Matuska 		/* Create temporary clone unless we're doing corrective recv */
879eda14cbcSMatt Macy 		dsl_dataset_t *snap = NULL;
880eda14cbcSMatt Macy 
881eda14cbcSMatt Macy 		if (drba->drba_cookie->drc_fromsnapobj != 0) {
882eda14cbcSMatt Macy 			VERIFY0(dsl_dataset_hold_obj(dp,
883eda14cbcSMatt Macy 			    drba->drba_cookie->drc_fromsnapobj, FTAG, &snap));
884eda14cbcSMatt Macy 			ASSERT3P(dcp, ==, NULL);
885eda14cbcSMatt Macy 		}
886271171e0SMartin Matuska 		if (drc->drc_heal) {
887271171e0SMartin Matuska 			/* When healing we want to use the provided snapshot */
888271171e0SMartin Matuska 			VERIFY0(dsl_dataset_snap_lookup(ds, drc->drc_tosnap,
889271171e0SMartin Matuska 			    &dsobj));
890271171e0SMartin Matuska 		} else {
891271171e0SMartin Matuska 			dsobj = dsl_dataset_create_sync(ds->ds_dir,
892271171e0SMartin Matuska 			    recv_clone_name, snap, crflags, drba->drba_cred,
893271171e0SMartin Matuska 			    dcp, tx);
894271171e0SMartin Matuska 		}
895eda14cbcSMatt Macy 		if (drba->drba_cookie->drc_fromsnapobj != 0)
896eda14cbcSMatt Macy 			dsl_dataset_rele(snap, FTAG);
897eda14cbcSMatt Macy 		dsl_dataset_rele_flags(ds, dsflags, FTAG);
898eda14cbcSMatt Macy 	} else {
899eda14cbcSMatt Macy 		dsl_dir_t *dd;
900eda14cbcSMatt Macy 		const char *tail;
901eda14cbcSMatt Macy 		dsl_dataset_t *origin = NULL;
902eda14cbcSMatt Macy 
903eda14cbcSMatt Macy 		VERIFY0(dsl_dir_hold(dp, tofs, FTAG, &dd, &tail));
904eda14cbcSMatt Macy 
905eda14cbcSMatt Macy 		if (drba->drba_origin != NULL) {
906eda14cbcSMatt Macy 			VERIFY0(dsl_dataset_hold(dp, drba->drba_origin,
907eda14cbcSMatt Macy 			    FTAG, &origin));
908eda14cbcSMatt Macy 			ASSERT3P(dcp, ==, NULL);
909eda14cbcSMatt Macy 		}
910eda14cbcSMatt Macy 
911eda14cbcSMatt Macy 		/* Create new dataset. */
912eda14cbcSMatt Macy 		dsobj = dsl_dataset_create_sync(dd, strrchr(tofs, '/') + 1,
913eda14cbcSMatt Macy 		    origin, crflags, drba->drba_cred, dcp, tx);
914eda14cbcSMatt Macy 		if (origin != NULL)
915eda14cbcSMatt Macy 			dsl_dataset_rele(origin, FTAG);
916eda14cbcSMatt Macy 		dsl_dir_rele(dd, FTAG);
917eda14cbcSMatt Macy 		drc->drc_newfs = B_TRUE;
918eda14cbcSMatt Macy 	}
919eda14cbcSMatt Macy 	VERIFY0(dsl_dataset_own_obj_force(dp, dsobj, dsflags, dmu_recv_tag,
920eda14cbcSMatt Macy 	    &newds));
921eda14cbcSMatt Macy 	if (dsl_dataset_feature_is_active(newds,
922eda14cbcSMatt Macy 	    SPA_FEATURE_REDACTED_DATASETS)) {
923eda14cbcSMatt Macy 		/*
924eda14cbcSMatt Macy 		 * If the origin dataset is redacted, the child will be redacted
925eda14cbcSMatt Macy 		 * when we create it.  We clear the new dataset's
926eda14cbcSMatt Macy 		 * redaction info; if it should be redacted, we'll fill
927eda14cbcSMatt Macy 		 * in its information later.
928eda14cbcSMatt Macy 		 */
929eda14cbcSMatt Macy 		dsl_dataset_deactivate_feature(newds,
930eda14cbcSMatt Macy 		    SPA_FEATURE_REDACTED_DATASETS, tx);
931eda14cbcSMatt Macy 	}
932eda14cbcSMatt Macy 	VERIFY0(dmu_objset_from_ds(newds, &os));
933eda14cbcSMatt Macy 
934eda14cbcSMatt Macy 	if (drc->drc_resumable) {
935eda14cbcSMatt Macy 		dsl_dataset_zapify(newds, tx);
936eda14cbcSMatt Macy 		if (drrb->drr_fromguid != 0) {
937eda14cbcSMatt Macy 			VERIFY0(zap_add(mos, dsobj, DS_FIELD_RESUME_FROMGUID,
938eda14cbcSMatt Macy 			    8, 1, &drrb->drr_fromguid, tx));
939eda14cbcSMatt Macy 		}
940eda14cbcSMatt Macy 		VERIFY0(zap_add(mos, dsobj, DS_FIELD_RESUME_TOGUID,
941eda14cbcSMatt Macy 		    8, 1, &drrb->drr_toguid, tx));
942eda14cbcSMatt Macy 		VERIFY0(zap_add(mos, dsobj, DS_FIELD_RESUME_TONAME,
943eda14cbcSMatt Macy 		    1, strlen(drrb->drr_toname) + 1, drrb->drr_toname, tx));
944eda14cbcSMatt Macy 		uint64_t one = 1;
945eda14cbcSMatt Macy 		uint64_t zero = 0;
946eda14cbcSMatt Macy 		VERIFY0(zap_add(mos, dsobj, DS_FIELD_RESUME_OBJECT,
947eda14cbcSMatt Macy 		    8, 1, &one, tx));
948eda14cbcSMatt Macy 		VERIFY0(zap_add(mos, dsobj, DS_FIELD_RESUME_OFFSET,
949eda14cbcSMatt Macy 		    8, 1, &zero, tx));
950eda14cbcSMatt Macy 		VERIFY0(zap_add(mos, dsobj, DS_FIELD_RESUME_BYTES,
951eda14cbcSMatt Macy 		    8, 1, &zero, tx));
952eda14cbcSMatt Macy 		if (featureflags & DMU_BACKUP_FEATURE_LARGE_BLOCKS) {
953eda14cbcSMatt Macy 			VERIFY0(zap_add(mos, dsobj, DS_FIELD_RESUME_LARGEBLOCK,
954eda14cbcSMatt Macy 			    8, 1, &one, tx));
955eda14cbcSMatt Macy 		}
956eda14cbcSMatt Macy 		if (featureflags & DMU_BACKUP_FEATURE_EMBED_DATA) {
957eda14cbcSMatt Macy 			VERIFY0(zap_add(mos, dsobj, DS_FIELD_RESUME_EMBEDOK,
958eda14cbcSMatt Macy 			    8, 1, &one, tx));
959eda14cbcSMatt Macy 		}
960eda14cbcSMatt Macy 		if (featureflags & DMU_BACKUP_FEATURE_COMPRESSED) {
961eda14cbcSMatt Macy 			VERIFY0(zap_add(mos, dsobj, DS_FIELD_RESUME_COMPRESSOK,
962eda14cbcSMatt Macy 			    8, 1, &one, tx));
963eda14cbcSMatt Macy 		}
964eda14cbcSMatt Macy 		if (featureflags & DMU_BACKUP_FEATURE_RAW) {
965eda14cbcSMatt Macy 			VERIFY0(zap_add(mos, dsobj, DS_FIELD_RESUME_RAWOK,
966eda14cbcSMatt Macy 			    8, 1, &one, tx));
967eda14cbcSMatt Macy 		}
968eda14cbcSMatt Macy 
969eda14cbcSMatt Macy 		uint64_t *redact_snaps;
970eda14cbcSMatt Macy 		uint_t numredactsnaps;
971eda14cbcSMatt Macy 		if (nvlist_lookup_uint64_array(drc->drc_begin_nvl,
972eda14cbcSMatt Macy 		    BEGINNV_REDACT_FROM_SNAPS, &redact_snaps,
973eda14cbcSMatt Macy 		    &numredactsnaps) == 0) {
974eda14cbcSMatt Macy 			VERIFY0(zap_add(mos, dsobj,
975eda14cbcSMatt Macy 			    DS_FIELD_RESUME_REDACT_BOOKMARK_SNAPS,
976eda14cbcSMatt Macy 			    sizeof (*redact_snaps), numredactsnaps,
977eda14cbcSMatt Macy 			    redact_snaps, tx));
978eda14cbcSMatt Macy 		}
979eda14cbcSMatt Macy 	}
980eda14cbcSMatt Macy 
981eda14cbcSMatt Macy 	/*
982eda14cbcSMatt Macy 	 * Usually the os->os_encrypted value is tied to the presence of a
983eda14cbcSMatt Macy 	 * DSL Crypto Key object in the dd. However, that will not be received
984eda14cbcSMatt Macy 	 * until dmu_recv_stream(), so we set the value manually for now.
985eda14cbcSMatt Macy 	 */
986eda14cbcSMatt Macy 	if (featureflags & DMU_BACKUP_FEATURE_RAW) {
987eda14cbcSMatt Macy 		os->os_encrypted = B_TRUE;
988eda14cbcSMatt Macy 		drba->drba_cookie->drc_raw = B_TRUE;
989eda14cbcSMatt Macy 	}
990eda14cbcSMatt Macy 
991eda14cbcSMatt Macy 	if (featureflags & DMU_BACKUP_FEATURE_REDACTED) {
992eda14cbcSMatt Macy 		uint64_t *redact_snaps;
993eda14cbcSMatt Macy 		uint_t numredactsnaps;
994eda14cbcSMatt Macy 		VERIFY0(nvlist_lookup_uint64_array(drc->drc_begin_nvl,
995eda14cbcSMatt Macy 		    BEGINNV_REDACT_SNAPS, &redact_snaps, &numredactsnaps));
996eda14cbcSMatt Macy 		dsl_dataset_activate_redaction(newds, redact_snaps,
997eda14cbcSMatt Macy 		    numredactsnaps, tx);
998eda14cbcSMatt Macy 	}
999eda14cbcSMatt Macy 
10007a7741afSMartin Matuska 	if (featureflags & DMU_BACKUP_FEATURE_LARGE_MICROZAP) {
10017a7741afSMartin Matuska 		/*
10027a7741afSMartin Matuska 		 * The source has seen a large microzap at least once in its
10037a7741afSMartin Matuska 		 * life, so we activate the feature here to match. It's not
10047a7741afSMartin Matuska 		 * strictly necessary since a large microzap is usable without
10057a7741afSMartin Matuska 		 * the feature active, but if that object is sent on from here,
10067a7741afSMartin Matuska 		 * we need this info to know to add the stream feature.
10077a7741afSMartin Matuska 		 *
10087a7741afSMartin Matuska 		 * There may be no large microzap in the incoming stream, or
10097a7741afSMartin Matuska 		 * ever again, but this is a very niche feature and its very
10107a7741afSMartin Matuska 		 * difficult to spot a large microzap in the stream, so its
10117a7741afSMartin Matuska 		 * not worth the effort of trying harder to activate the
10127a7741afSMartin Matuska 		 * feature at first use.
10137a7741afSMartin Matuska 		 */
10147a7741afSMartin Matuska 		dsl_dataset_activate_feature(dsobj, SPA_FEATURE_LARGE_MICROZAP,
10157a7741afSMartin Matuska 		    (void *)B_TRUE, tx);
10167a7741afSMartin Matuska 	}
10177a7741afSMartin Matuska 
1018eda14cbcSMatt Macy 	dmu_buf_will_dirty(newds->ds_dbuf, tx);
1019eda14cbcSMatt Macy 	dsl_dataset_phys(newds)->ds_flags |= DS_FLAG_INCONSISTENT;
1020eda14cbcSMatt Macy 
1021eda14cbcSMatt Macy 	/*
10227a7741afSMartin Matuska 	 * Activate longname feature if received
10237a7741afSMartin Matuska 	 */
10247a7741afSMartin Matuska 	if (featureflags & DMU_BACKUP_FEATURE_LONGNAME &&
10257a7741afSMartin Matuska 	    !dsl_dataset_feature_is_active(newds, SPA_FEATURE_LONGNAME)) {
10267a7741afSMartin Matuska 		dsl_dataset_activate_feature(newds->ds_object,
10277a7741afSMartin Matuska 		    SPA_FEATURE_LONGNAME, (void *)B_TRUE, tx);
10287a7741afSMartin Matuska 		newds->ds_feature[SPA_FEATURE_LONGNAME] = (void *)B_TRUE;
10297a7741afSMartin Matuska 	}
10307a7741afSMartin Matuska 
10317a7741afSMartin Matuska 	/*
1032eda14cbcSMatt Macy 	 * If we actually created a non-clone, we need to create the objset
1033eda14cbcSMatt Macy 	 * in our new dataset. If this is a raw send we postpone this until
1034eda14cbcSMatt Macy 	 * dmu_recv_stream() so that we can allocate the metadnode with the
1035eda14cbcSMatt Macy 	 * properties from the DRR_BEGIN payload.
1036eda14cbcSMatt Macy 	 */
1037eda14cbcSMatt Macy 	rrw_enter(&newds->ds_bp_rwlock, RW_READER, FTAG);
1038eda14cbcSMatt Macy 	if (BP_IS_HOLE(dsl_dataset_get_blkptr(newds)) &&
1039271171e0SMartin Matuska 	    (featureflags & DMU_BACKUP_FEATURE_RAW) == 0 &&
1040271171e0SMartin Matuska 	    !drc->drc_heal) {
1041eda14cbcSMatt Macy 		(void) dmu_objset_create_impl(dp->dp_spa,
1042eda14cbcSMatt Macy 		    newds, dsl_dataset_get_blkptr(newds), drrb->drr_type, tx);
1043eda14cbcSMatt Macy 	}
1044eda14cbcSMatt Macy 	rrw_exit(&newds->ds_bp_rwlock, FTAG);
1045eda14cbcSMatt Macy 
1046eda14cbcSMatt Macy 	drba->drba_cookie->drc_ds = newds;
1047eda14cbcSMatt Macy 	drba->drba_cookie->drc_os = os;
1048eda14cbcSMatt Macy 
1049eda14cbcSMatt Macy 	spa_history_log_internal_ds(newds, "receive", tx, " ");
1050eda14cbcSMatt Macy }
1051eda14cbcSMatt Macy 
1052eda14cbcSMatt Macy static int
dmu_recv_resume_begin_check(void * arg,dmu_tx_t * tx)1053eda14cbcSMatt Macy dmu_recv_resume_begin_check(void *arg, dmu_tx_t *tx)
1054eda14cbcSMatt Macy {
1055eda14cbcSMatt Macy 	dmu_recv_begin_arg_t *drba = arg;
1056eda14cbcSMatt Macy 	dmu_recv_cookie_t *drc = drba->drba_cookie;
1057eda14cbcSMatt Macy 	dsl_pool_t *dp = dmu_tx_pool(tx);
1058eda14cbcSMatt Macy 	struct drr_begin *drrb = drc->drc_drrb;
1059eda14cbcSMatt Macy 	int error;
10607877fdebSMatt Macy 	ds_hold_flags_t dsflags = DS_HOLD_FLAG_NONE;
1061eda14cbcSMatt Macy 	dsl_dataset_t *ds;
1062eda14cbcSMatt Macy 	const char *tofs = drc->drc_tofs;
1063eda14cbcSMatt Macy 
1064eda14cbcSMatt Macy 	/* already checked */
1065eda14cbcSMatt Macy 	ASSERT3U(drrb->drr_magic, ==, DMU_BACKUP_MAGIC);
1066eda14cbcSMatt Macy 	ASSERT(drc->drc_featureflags & DMU_BACKUP_FEATURE_RESUMING);
1067eda14cbcSMatt Macy 
1068eda14cbcSMatt Macy 	if (DMU_GET_STREAM_HDRTYPE(drrb->drr_versioninfo) ==
1069eda14cbcSMatt Macy 	    DMU_COMPOUNDSTREAM ||
1070eda14cbcSMatt Macy 	    drrb->drr_type >= DMU_OST_NUMTYPES)
1071eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
1072eda14cbcSMatt Macy 
1073eda14cbcSMatt Macy 	/*
1074eda14cbcSMatt Macy 	 * This is mostly a sanity check since we should have already done these
1075eda14cbcSMatt Macy 	 * checks during a previous attempt to receive the data.
1076eda14cbcSMatt Macy 	 */
1077eda14cbcSMatt Macy 	error = recv_begin_check_feature_flags_impl(drc->drc_featureflags,
1078eda14cbcSMatt Macy 	    dp->dp_spa);
1079eda14cbcSMatt Macy 	if (error != 0)
1080eda14cbcSMatt Macy 		return (error);
1081eda14cbcSMatt Macy 
1082eda14cbcSMatt Macy 	/* 6 extra bytes for /%recv */
1083eda14cbcSMatt Macy 	char recvname[ZFS_MAX_DATASET_NAME_LEN + 6];
1084eda14cbcSMatt Macy 
1085eda14cbcSMatt Macy 	(void) snprintf(recvname, sizeof (recvname), "%s/%s",
1086eda14cbcSMatt Macy 	    tofs, recv_clone_name);
1087eda14cbcSMatt Macy 
1088eda14cbcSMatt Macy 	if (drc->drc_featureflags & DMU_BACKUP_FEATURE_RAW) {
1089eda14cbcSMatt Macy 		/* raw receives require spill block allocation flag */
1090eda14cbcSMatt Macy 		if (!(drrb->drr_flags & DRR_FLAG_SPILL_BLOCK))
1091eda14cbcSMatt Macy 			return (SET_ERROR(ZFS_ERR_SPILL_BLOCK_FLAG_MISSING));
1092eda14cbcSMatt Macy 	} else {
1093eda14cbcSMatt Macy 		dsflags |= DS_HOLD_FLAG_DECRYPT;
1094eda14cbcSMatt Macy 	}
1095eda14cbcSMatt Macy 
1096be181ee2SMartin Matuska 	boolean_t recvexist = B_TRUE;
1097eda14cbcSMatt Macy 	if (dsl_dataset_hold_flags(dp, recvname, dsflags, FTAG, &ds) != 0) {
1098eda14cbcSMatt Macy 		/* %recv does not exist; continue in tofs */
1099be181ee2SMartin Matuska 		recvexist = B_FALSE;
1100eda14cbcSMatt Macy 		error = dsl_dataset_hold_flags(dp, tofs, dsflags, FTAG, &ds);
1101eda14cbcSMatt Macy 		if (error != 0)
1102eda14cbcSMatt Macy 			return (error);
1103eda14cbcSMatt Macy 	}
1104eda14cbcSMatt Macy 
1105be181ee2SMartin Matuska 	/*
1106be181ee2SMartin Matuska 	 * Resume of full/newfs recv on existing dataset should be done with
1107be181ee2SMartin Matuska 	 * force flag
1108be181ee2SMartin Matuska 	 */
1109be181ee2SMartin Matuska 	if (recvexist && drrb->drr_fromguid == 0 && !drc->drc_force) {
1110be181ee2SMartin Matuska 		dsl_dataset_rele_flags(ds, dsflags, FTAG);
1111be181ee2SMartin Matuska 		return (SET_ERROR(ZFS_ERR_RESUME_EXISTS));
1112be181ee2SMartin Matuska 	}
1113be181ee2SMartin Matuska 
1114eda14cbcSMatt Macy 	/* check that ds is marked inconsistent */
1115eda14cbcSMatt Macy 	if (!DS_IS_INCONSISTENT(ds)) {
1116eda14cbcSMatt Macy 		dsl_dataset_rele_flags(ds, dsflags, FTAG);
1117eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
1118eda14cbcSMatt Macy 	}
1119eda14cbcSMatt Macy 
1120eda14cbcSMatt Macy 	/* check that there is resuming data, and that the toguid matches */
1121eda14cbcSMatt Macy 	if (!dsl_dataset_is_zapified(ds)) {
1122eda14cbcSMatt Macy 		dsl_dataset_rele_flags(ds, dsflags, FTAG);
1123eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
1124eda14cbcSMatt Macy 	}
1125eda14cbcSMatt Macy 	uint64_t val;
1126eda14cbcSMatt Macy 	error = zap_lookup(dp->dp_meta_objset, ds->ds_object,
1127eda14cbcSMatt Macy 	    DS_FIELD_RESUME_TOGUID, sizeof (val), 1, &val);
1128eda14cbcSMatt Macy 	if (error != 0 || drrb->drr_toguid != val) {
1129eda14cbcSMatt Macy 		dsl_dataset_rele_flags(ds, dsflags, FTAG);
1130eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
1131eda14cbcSMatt Macy 	}
1132eda14cbcSMatt Macy 
1133eda14cbcSMatt Macy 	/*
1134eda14cbcSMatt Macy 	 * Check if the receive is still running.  If so, it will be owned.
1135eda14cbcSMatt Macy 	 * Note that nothing else can own the dataset (e.g. after the receive
1136eda14cbcSMatt Macy 	 * fails) because it will be marked inconsistent.
1137eda14cbcSMatt Macy 	 */
1138eda14cbcSMatt Macy 	if (dsl_dataset_has_owner(ds)) {
1139eda14cbcSMatt Macy 		dsl_dataset_rele_flags(ds, dsflags, FTAG);
1140eda14cbcSMatt Macy 		return (SET_ERROR(EBUSY));
1141eda14cbcSMatt Macy 	}
1142eda14cbcSMatt Macy 
1143eda14cbcSMatt Macy 	/* There should not be any snapshots of this fs yet. */
1144eda14cbcSMatt Macy 	if (ds->ds_prev != NULL && ds->ds_prev->ds_dir == ds->ds_dir) {
1145eda14cbcSMatt Macy 		dsl_dataset_rele_flags(ds, dsflags, FTAG);
1146eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
1147eda14cbcSMatt Macy 	}
1148eda14cbcSMatt Macy 
1149eda14cbcSMatt Macy 	/*
1150eda14cbcSMatt Macy 	 * Note: resume point will be checked when we process the first WRITE
1151eda14cbcSMatt Macy 	 * record.
1152eda14cbcSMatt Macy 	 */
1153eda14cbcSMatt Macy 
1154eda14cbcSMatt Macy 	/* check that the origin matches */
1155eda14cbcSMatt Macy 	val = 0;
1156eda14cbcSMatt Macy 	(void) zap_lookup(dp->dp_meta_objset, ds->ds_object,
1157eda14cbcSMatt Macy 	    DS_FIELD_RESUME_FROMGUID, sizeof (val), 1, &val);
1158eda14cbcSMatt Macy 	if (drrb->drr_fromguid != val) {
1159eda14cbcSMatt Macy 		dsl_dataset_rele_flags(ds, dsflags, FTAG);
1160eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
1161eda14cbcSMatt Macy 	}
1162eda14cbcSMatt Macy 
1163eda14cbcSMatt Macy 	if (ds->ds_prev != NULL && drrb->drr_fromguid != 0)
1164eda14cbcSMatt Macy 		drc->drc_fromsnapobj = ds->ds_prev->ds_object;
1165eda14cbcSMatt Macy 
1166eda14cbcSMatt Macy 	/*
1167eda14cbcSMatt Macy 	 * If we're resuming, and the send is redacted, then the original send
1168eda14cbcSMatt Macy 	 * must have been redacted, and must have been redacted with respect to
1169eda14cbcSMatt Macy 	 * the same snapshots.
1170eda14cbcSMatt Macy 	 */
1171eda14cbcSMatt Macy 	if (drc->drc_featureflags & DMU_BACKUP_FEATURE_REDACTED) {
1172eda14cbcSMatt Macy 		uint64_t num_ds_redact_snaps;
1173eda14cbcSMatt Macy 		uint64_t *ds_redact_snaps;
1174eda14cbcSMatt Macy 
1175eda14cbcSMatt Macy 		uint_t num_stream_redact_snaps;
1176eda14cbcSMatt Macy 		uint64_t *stream_redact_snaps;
1177eda14cbcSMatt Macy 
1178eda14cbcSMatt Macy 		if (nvlist_lookup_uint64_array(drc->drc_begin_nvl,
1179eda14cbcSMatt Macy 		    BEGINNV_REDACT_SNAPS, &stream_redact_snaps,
1180eda14cbcSMatt Macy 		    &num_stream_redact_snaps) != 0) {
1181eda14cbcSMatt Macy 			dsl_dataset_rele_flags(ds, dsflags, FTAG);
1182eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
1183eda14cbcSMatt Macy 		}
1184eda14cbcSMatt Macy 
1185eda14cbcSMatt Macy 		if (!dsl_dataset_get_uint64_array_feature(ds,
1186eda14cbcSMatt Macy 		    SPA_FEATURE_REDACTED_DATASETS, &num_ds_redact_snaps,
1187eda14cbcSMatt Macy 		    &ds_redact_snaps)) {
1188eda14cbcSMatt Macy 			dsl_dataset_rele_flags(ds, dsflags, FTAG);
1189eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
1190eda14cbcSMatt Macy 		}
1191eda14cbcSMatt Macy 
1192eda14cbcSMatt Macy 		for (int i = 0; i < num_ds_redact_snaps; i++) {
1193eda14cbcSMatt Macy 			if (!redact_snaps_contains(ds_redact_snaps,
1194eda14cbcSMatt Macy 			    num_ds_redact_snaps, stream_redact_snaps[i])) {
1195eda14cbcSMatt Macy 				dsl_dataset_rele_flags(ds, dsflags, FTAG);
1196eda14cbcSMatt Macy 				return (SET_ERROR(EINVAL));
1197eda14cbcSMatt Macy 			}
1198eda14cbcSMatt Macy 		}
1199eda14cbcSMatt Macy 	}
1200eda14cbcSMatt Macy 
1201eda14cbcSMatt Macy 	error = recv_check_large_blocks(ds, drc->drc_featureflags);
1202eda14cbcSMatt Macy 	if (error != 0) {
1203eda14cbcSMatt Macy 		dsl_dataset_rele_flags(ds, dsflags, FTAG);
1204eda14cbcSMatt Macy 		return (error);
1205eda14cbcSMatt Macy 	}
1206eda14cbcSMatt Macy 
1207eda14cbcSMatt Macy 	dsl_dataset_rele_flags(ds, dsflags, FTAG);
1208eda14cbcSMatt Macy 	return (0);
1209eda14cbcSMatt Macy }
1210eda14cbcSMatt Macy 
1211eda14cbcSMatt Macy static void
dmu_recv_resume_begin_sync(void * arg,dmu_tx_t * tx)1212eda14cbcSMatt Macy dmu_recv_resume_begin_sync(void *arg, dmu_tx_t *tx)
1213eda14cbcSMatt Macy {
1214eda14cbcSMatt Macy 	dmu_recv_begin_arg_t *drba = arg;
1215eda14cbcSMatt Macy 	dsl_pool_t *dp = dmu_tx_pool(tx);
1216eda14cbcSMatt Macy 	const char *tofs = drba->drba_cookie->drc_tofs;
1217eda14cbcSMatt Macy 	uint64_t featureflags = drba->drba_cookie->drc_featureflags;
1218eda14cbcSMatt Macy 	dsl_dataset_t *ds;
12197877fdebSMatt Macy 	ds_hold_flags_t dsflags = DS_HOLD_FLAG_NONE;
1220eda14cbcSMatt Macy 	/* 6 extra bytes for /%recv */
1221eda14cbcSMatt Macy 	char recvname[ZFS_MAX_DATASET_NAME_LEN + 6];
1222eda14cbcSMatt Macy 
1223eda14cbcSMatt Macy 	(void) snprintf(recvname, sizeof (recvname), "%s/%s", tofs,
1224eda14cbcSMatt Macy 	    recv_clone_name);
1225eda14cbcSMatt Macy 
1226eda14cbcSMatt Macy 	if (featureflags & DMU_BACKUP_FEATURE_RAW) {
1227eda14cbcSMatt Macy 		drba->drba_cookie->drc_raw = B_TRUE;
1228eda14cbcSMatt Macy 	} else {
1229eda14cbcSMatt Macy 		dsflags |= DS_HOLD_FLAG_DECRYPT;
1230eda14cbcSMatt Macy 	}
1231eda14cbcSMatt Macy 
1232eda14cbcSMatt Macy 	if (dsl_dataset_own_force(dp, recvname, dsflags, dmu_recv_tag, &ds)
1233eda14cbcSMatt Macy 	    != 0) {
1234eda14cbcSMatt Macy 		/* %recv does not exist; continue in tofs */
1235eda14cbcSMatt Macy 		VERIFY0(dsl_dataset_own_force(dp, tofs, dsflags, dmu_recv_tag,
1236eda14cbcSMatt Macy 		    &ds));
1237eda14cbcSMatt Macy 		drba->drba_cookie->drc_newfs = B_TRUE;
1238eda14cbcSMatt Macy 	}
1239eda14cbcSMatt Macy 
1240eda14cbcSMatt Macy 	ASSERT(DS_IS_INCONSISTENT(ds));
1241eda14cbcSMatt Macy 	rrw_enter(&ds->ds_bp_rwlock, RW_READER, FTAG);
1242eda14cbcSMatt Macy 	ASSERT(!BP_IS_HOLE(dsl_dataset_get_blkptr(ds)) ||
1243eda14cbcSMatt Macy 	    drba->drba_cookie->drc_raw);
1244eda14cbcSMatt Macy 	rrw_exit(&ds->ds_bp_rwlock, FTAG);
1245eda14cbcSMatt Macy 
1246eda14cbcSMatt Macy 	drba->drba_cookie->drc_ds = ds;
1247eda14cbcSMatt Macy 	VERIFY0(dmu_objset_from_ds(ds, &drba->drba_cookie->drc_os));
1248eda14cbcSMatt Macy 	drba->drba_cookie->drc_should_save = B_TRUE;
1249eda14cbcSMatt Macy 
1250eda14cbcSMatt Macy 	spa_history_log_internal_ds(ds, "resume receive", tx, " ");
1251eda14cbcSMatt Macy }
1252eda14cbcSMatt Macy 
1253eda14cbcSMatt Macy /*
1254eda14cbcSMatt Macy  * NB: callers *MUST* call dmu_recv_stream() if dmu_recv_begin()
1255eda14cbcSMatt Macy  * succeeds; otherwise we will leak the holds on the datasets.
1256eda14cbcSMatt Macy  */
1257eda14cbcSMatt Macy int
dmu_recv_begin(const char * tofs,const char * tosnap,dmu_replay_record_t * drr_begin,boolean_t force,boolean_t heal,boolean_t resumable,nvlist_t * localprops,nvlist_t * hidden_args,const char * origin,dmu_recv_cookie_t * drc,zfs_file_t * fp,offset_t * voffp)12582a58b312SMartin Matuska dmu_recv_begin(const char *tofs, const char *tosnap,
12592a58b312SMartin Matuska     dmu_replay_record_t *drr_begin, boolean_t force, boolean_t heal,
12602a58b312SMartin Matuska     boolean_t resumable, nvlist_t *localprops, nvlist_t *hidden_args,
12612a58b312SMartin Matuska     const char *origin, dmu_recv_cookie_t *drc, zfs_file_t *fp,
12622a58b312SMartin Matuska     offset_t *voffp)
1263eda14cbcSMatt Macy {
1264eda14cbcSMatt Macy 	dmu_recv_begin_arg_t drba = { 0 };
12652a58b312SMartin Matuska 	int err = 0;
1266eda14cbcSMatt Macy 
1267*071ab5a1SMartin Matuska 	cred_t *cr = CRED();
1268*071ab5a1SMartin Matuska 	crhold(cr);
1269*071ab5a1SMartin Matuska 
1270da5137abSMartin Matuska 	memset(drc, 0, sizeof (dmu_recv_cookie_t));
1271eda14cbcSMatt Macy 	drc->drc_drr_begin = drr_begin;
1272eda14cbcSMatt Macy 	drc->drc_drrb = &drr_begin->drr_u.drr_begin;
1273eda14cbcSMatt Macy 	drc->drc_tosnap = tosnap;
1274eda14cbcSMatt Macy 	drc->drc_tofs = tofs;
1275eda14cbcSMatt Macy 	drc->drc_force = force;
1276271171e0SMartin Matuska 	drc->drc_heal = heal;
1277eda14cbcSMatt Macy 	drc->drc_resumable = resumable;
1278*071ab5a1SMartin Matuska 	drc->drc_cred = cr;
1279eda14cbcSMatt Macy 	drc->drc_clone = (origin != NULL);
1280eda14cbcSMatt Macy 
1281eda14cbcSMatt Macy 	if (drc->drc_drrb->drr_magic == BSWAP_64(DMU_BACKUP_MAGIC)) {
1282eda14cbcSMatt Macy 		drc->drc_byteswap = B_TRUE;
1283eda14cbcSMatt Macy 		(void) fletcher_4_incremental_byteswap(drr_begin,
1284eda14cbcSMatt Macy 		    sizeof (dmu_replay_record_t), &drc->drc_cksum);
1285eda14cbcSMatt Macy 		byteswap_record(drr_begin);
1286eda14cbcSMatt Macy 	} else if (drc->drc_drrb->drr_magic == DMU_BACKUP_MAGIC) {
1287eda14cbcSMatt Macy 		(void) fletcher_4_incremental_native(drr_begin,
1288eda14cbcSMatt Macy 		    sizeof (dmu_replay_record_t), &drc->drc_cksum);
1289eda14cbcSMatt Macy 	} else {
1290*071ab5a1SMartin Matuska 		crfree(cr);
1291*071ab5a1SMartin Matuska 		drc->drc_cred = NULL;
1292eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
1293eda14cbcSMatt Macy 	}
1294eda14cbcSMatt Macy 
1295eda14cbcSMatt Macy 	drc->drc_fp = fp;
1296eda14cbcSMatt Macy 	drc->drc_voff = *voffp;
1297eda14cbcSMatt Macy 	drc->drc_featureflags =
1298eda14cbcSMatt Macy 	    DMU_GET_FEATUREFLAGS(drc->drc_drrb->drr_versioninfo);
1299eda14cbcSMatt Macy 
1300eda14cbcSMatt Macy 	uint32_t payloadlen = drc->drc_drr_begin->drr_payloadlen;
130115f0b8c3SMartin Matuska 
130215f0b8c3SMartin Matuska 	/*
130315f0b8c3SMartin Matuska 	 * Since OpenZFS 2.0.0, we have enforced a 64MB limit in userspace
130415f0b8c3SMartin Matuska 	 * configurable via ZFS_SENDRECV_MAX_NVLIST. We enforce 256MB as a hard
130515f0b8c3SMartin Matuska 	 * upper limit. Systems with less than 1GB of RAM will see a lower
130615f0b8c3SMartin Matuska 	 * limit from `arc_all_memory() / 4`.
130715f0b8c3SMartin Matuska 	 */
1308*071ab5a1SMartin Matuska 	if (payloadlen > (MIN((1U << 28), arc_all_memory() / 4))) {
1309*071ab5a1SMartin Matuska 		crfree(cr);
1310*071ab5a1SMartin Matuska 		drc->drc_cred = NULL;
1311*071ab5a1SMartin Matuska 		return (SET_ERROR(E2BIG));
1312*071ab5a1SMartin Matuska 	}
13132a58b312SMartin Matuska 
13142a58b312SMartin Matuska 	if (payloadlen != 0) {
13152a58b312SMartin Matuska 		void *payload = vmem_alloc(payloadlen, KM_SLEEP);
13162a58b312SMartin Matuska 		/*
13172a58b312SMartin Matuska 		 * For compatibility with recursive send streams, we don't do
13182a58b312SMartin Matuska 		 * this here if the stream could be part of a package. Instead,
13192a58b312SMartin Matuska 		 * we'll do it in dmu_recv_stream. If we pull the next header
13202a58b312SMartin Matuska 		 * too early, and it's the END record, we break the `recv_skip`
13212a58b312SMartin Matuska 		 * logic.
13222a58b312SMartin Matuska 		 */
1323eda14cbcSMatt Macy 
1324eda14cbcSMatt Macy 		err = receive_read_payload_and_next_header(drc, payloadlen,
1325eda14cbcSMatt Macy 		    payload);
1326eda14cbcSMatt Macy 		if (err != 0) {
132715f0b8c3SMartin Matuska 			vmem_free(payload, payloadlen);
1328*071ab5a1SMartin Matuska 			crfree(cr);
1329*071ab5a1SMartin Matuska 			drc->drc_cred = NULL;
1330eda14cbcSMatt Macy 			return (err);
1331eda14cbcSMatt Macy 		}
1332eda14cbcSMatt Macy 		err = nvlist_unpack(payload, payloadlen, &drc->drc_begin_nvl,
1333eda14cbcSMatt Macy 		    KM_SLEEP);
133415f0b8c3SMartin Matuska 		vmem_free(payload, payloadlen);
1335eda14cbcSMatt Macy 		if (err != 0) {
1336eda14cbcSMatt Macy 			kmem_free(drc->drc_next_rrd,
1337eda14cbcSMatt Macy 			    sizeof (*drc->drc_next_rrd));
1338*071ab5a1SMartin Matuska 			crfree(cr);
1339*071ab5a1SMartin Matuska 			drc->drc_cred = NULL;
1340eda14cbcSMatt Macy 			return (err);
1341eda14cbcSMatt Macy 		}
1342eda14cbcSMatt Macy 	}
1343eda14cbcSMatt Macy 
1344eda14cbcSMatt Macy 	if (drc->drc_drrb->drr_flags & DRR_FLAG_SPILL_BLOCK)
1345eda14cbcSMatt Macy 		drc->drc_spill = B_TRUE;
1346eda14cbcSMatt Macy 
1347eda14cbcSMatt Macy 	drba.drba_origin = origin;
1348eda14cbcSMatt Macy 	drba.drba_cookie = drc;
1349*071ab5a1SMartin Matuska 	drba.drba_cred = drc->drc_cred;
1350eda14cbcSMatt Macy 
1351eda14cbcSMatt Macy 	if (drc->drc_featureflags & DMU_BACKUP_FEATURE_RESUMING) {
1352eda14cbcSMatt Macy 		err = dsl_sync_task(tofs,
1353eda14cbcSMatt Macy 		    dmu_recv_resume_begin_check, dmu_recv_resume_begin_sync,
1354eda14cbcSMatt Macy 		    &drba, 5, ZFS_SPACE_CHECK_NORMAL);
1355eda14cbcSMatt Macy 	} else {
1356eda14cbcSMatt Macy 		/*
1357eda14cbcSMatt Macy 		 * For non-raw, non-incremental, non-resuming receives the
1358eda14cbcSMatt Macy 		 * user can specify encryption parameters on the command line
1359eda14cbcSMatt Macy 		 * with "zfs recv -o". For these receives we create a dcp and
1360eda14cbcSMatt Macy 		 * pass it to the sync task. Creating the dcp will implicitly
1361eda14cbcSMatt Macy 		 * remove the encryption params from the localprops nvlist,
1362eda14cbcSMatt Macy 		 * which avoids errors when trying to set these normally
1363eda14cbcSMatt Macy 		 * read-only properties. Any other kind of receive that
1364eda14cbcSMatt Macy 		 * attempts to set these properties will fail as a result.
1365eda14cbcSMatt Macy 		 */
1366eda14cbcSMatt Macy 		if ((DMU_GET_FEATUREFLAGS(drc->drc_drrb->drr_versioninfo) &
1367eda14cbcSMatt Macy 		    DMU_BACKUP_FEATURE_RAW) == 0 &&
1368eda14cbcSMatt Macy 		    origin == NULL && drc->drc_drrb->drr_fromguid == 0) {
1369eda14cbcSMatt Macy 			err = dsl_crypto_params_create_nvlist(DCP_CMD_NONE,
1370eda14cbcSMatt Macy 			    localprops, hidden_args, &drba.drba_dcp);
1371eda14cbcSMatt Macy 		}
1372eda14cbcSMatt Macy 
1373eda14cbcSMatt Macy 		if (err == 0) {
1374eda14cbcSMatt Macy 			err = dsl_sync_task(tofs,
1375eda14cbcSMatt Macy 			    dmu_recv_begin_check, dmu_recv_begin_sync,
1376eda14cbcSMatt Macy 			    &drba, 5, ZFS_SPACE_CHECK_NORMAL);
1377eda14cbcSMatt Macy 			dsl_crypto_params_free(drba.drba_dcp, !!err);
1378eda14cbcSMatt Macy 		}
1379eda14cbcSMatt Macy 	}
1380eda14cbcSMatt Macy 
1381eda14cbcSMatt Macy 	if (err != 0) {
1382eda14cbcSMatt Macy 		kmem_free(drc->drc_next_rrd, sizeof (*drc->drc_next_rrd));
1383eda14cbcSMatt Macy 		nvlist_free(drc->drc_begin_nvl);
1384*071ab5a1SMartin Matuska 		crfree(cr);
1385*071ab5a1SMartin Matuska 		drc->drc_cred = NULL;
1386eda14cbcSMatt Macy 	}
1387eda14cbcSMatt Macy 	return (err);
1388eda14cbcSMatt Macy }
1389eda14cbcSMatt Macy 
1390271171e0SMartin Matuska /*
1391271171e0SMartin Matuska  * Holds data need for corrective recv callback
1392271171e0SMartin Matuska  */
1393271171e0SMartin Matuska typedef struct cr_cb_data {
1394271171e0SMartin Matuska 	uint64_t size;
1395271171e0SMartin Matuska 	zbookmark_phys_t zb;
1396271171e0SMartin Matuska 	spa_t *spa;
1397271171e0SMartin Matuska } cr_cb_data_t;
1398271171e0SMartin Matuska 
1399271171e0SMartin Matuska static void
corrective_read_done(zio_t * zio)1400271171e0SMartin Matuska corrective_read_done(zio_t *zio)
1401271171e0SMartin Matuska {
1402271171e0SMartin Matuska 	cr_cb_data_t *data = zio->io_private;
1403271171e0SMartin Matuska 	/* Corruption corrected; update error log if needed */
1404783d3ff6SMartin Matuska 	if (zio->io_error == 0) {
1405783d3ff6SMartin Matuska 		spa_remove_error(data->spa, &data->zb,
1406783d3ff6SMartin Matuska 		    BP_GET_LOGICAL_BIRTH(zio->io_bp));
1407783d3ff6SMartin Matuska 	}
1408271171e0SMartin Matuska 	kmem_free(data, sizeof (cr_cb_data_t));
1409271171e0SMartin Matuska 	abd_free(zio->io_abd);
1410271171e0SMartin Matuska }
1411271171e0SMartin Matuska 
1412271171e0SMartin Matuska /*
1413271171e0SMartin Matuska  * zio_rewrite the data pointed to by bp with the data from the rrd's abd.
1414271171e0SMartin Matuska  */
1415271171e0SMartin Matuska static int
do_corrective_recv(struct receive_writer_arg * rwa,struct drr_write * drrw,struct receive_record_arg * rrd,blkptr_t * bp)1416271171e0SMartin Matuska do_corrective_recv(struct receive_writer_arg *rwa, struct drr_write *drrw,
1417271171e0SMartin Matuska     struct receive_record_arg *rrd, blkptr_t *bp)
1418271171e0SMartin Matuska {
1419271171e0SMartin Matuska 	int err;
1420271171e0SMartin Matuska 	zio_t *io;
1421271171e0SMartin Matuska 	zbookmark_phys_t zb;
1422271171e0SMartin Matuska 	dnode_t *dn;
1423271171e0SMartin Matuska 	abd_t *abd = rrd->abd;
1424271171e0SMartin Matuska 	zio_cksum_t bp_cksum = bp->blk_cksum;
14254e8d558cSMartin Matuska 	zio_flag_t flags = ZIO_FLAG_SPECULATIVE | ZIO_FLAG_DONT_RETRY |
14264e8d558cSMartin Matuska 	    ZIO_FLAG_CANFAIL;
1427271171e0SMartin Matuska 
1428271171e0SMartin Matuska 	if (rwa->raw)
1429271171e0SMartin Matuska 		flags |= ZIO_FLAG_RAW;
1430271171e0SMartin Matuska 
1431271171e0SMartin Matuska 	err = dnode_hold(rwa->os, drrw->drr_object, FTAG, &dn);
1432271171e0SMartin Matuska 	if (err != 0)
1433271171e0SMartin Matuska 		return (err);
1434271171e0SMartin Matuska 	SET_BOOKMARK(&zb, dmu_objset_id(rwa->os), drrw->drr_object, 0,
1435271171e0SMartin Matuska 	    dbuf_whichblock(dn, 0, drrw->drr_offset));
1436271171e0SMartin Matuska 	dnode_rele(dn, FTAG);
1437271171e0SMartin Matuska 
1438271171e0SMartin Matuska 	if (!rwa->raw && DRR_WRITE_COMPRESSED(drrw)) {
1439271171e0SMartin Matuska 		/* Decompress the stream data */
1440271171e0SMartin Matuska 		abd_t *dabd = abd_alloc_linear(
1441271171e0SMartin Matuska 		    drrw->drr_logical_size, B_FALSE);
1442271171e0SMartin Matuska 		err = zio_decompress_data(drrw->drr_compressiontype,
1443e2df9bb4SMartin Matuska 		    abd, dabd, abd_get_size(abd),
1444271171e0SMartin Matuska 		    abd_get_size(dabd), NULL);
1445271171e0SMartin Matuska 
1446271171e0SMartin Matuska 		if (err != 0) {
1447271171e0SMartin Matuska 			abd_free(dabd);
1448271171e0SMartin Matuska 			return (err);
1449271171e0SMartin Matuska 		}
1450271171e0SMartin Matuska 		/* Swap in the newly decompressed data into the abd */
1451271171e0SMartin Matuska 		abd_free(abd);
1452271171e0SMartin Matuska 		abd = dabd;
1453271171e0SMartin Matuska 	}
1454271171e0SMartin Matuska 
1455271171e0SMartin Matuska 	if (!rwa->raw && BP_GET_COMPRESS(bp) != ZIO_COMPRESS_OFF) {
1456271171e0SMartin Matuska 		/* Recompress the data */
1457271171e0SMartin Matuska 		abd_t *cabd = abd_alloc_linear(BP_GET_PSIZE(bp),
1458271171e0SMartin Matuska 		    B_FALSE);
1459271171e0SMartin Matuska 		uint64_t csize = zio_compress_data(BP_GET_COMPRESS(bp),
14607a7741afSMartin Matuska 		    abd, &cabd, abd_get_size(abd), BP_GET_PSIZE(bp),
1461271171e0SMartin Matuska 		    rwa->os->os_complevel);
1462271171e0SMartin Matuska 		abd_zero_off(cabd, csize, BP_GET_PSIZE(bp) - csize);
1463271171e0SMartin Matuska 		/* Swap in newly compressed data into the abd */
1464271171e0SMartin Matuska 		abd_free(abd);
1465271171e0SMartin Matuska 		abd = cabd;
1466271171e0SMartin Matuska 		flags |= ZIO_FLAG_RAW_COMPRESS;
1467271171e0SMartin Matuska 	}
1468271171e0SMartin Matuska 
1469271171e0SMartin Matuska 	/*
1470271171e0SMartin Matuska 	 * The stream is not encrypted but the data on-disk is.
1471271171e0SMartin Matuska 	 * We need to re-encrypt the buf using the same
1472271171e0SMartin Matuska 	 * encryption type, salt, iv, and mac that was used to encrypt
1473271171e0SMartin Matuska 	 * the block previosly.
1474271171e0SMartin Matuska 	 */
1475271171e0SMartin Matuska 	if (!rwa->raw && BP_USES_CRYPT(bp)) {
1476271171e0SMartin Matuska 		dsl_dataset_t *ds;
1477271171e0SMartin Matuska 		dsl_crypto_key_t *dck = NULL;
1478271171e0SMartin Matuska 		uint8_t salt[ZIO_DATA_SALT_LEN];
1479271171e0SMartin Matuska 		uint8_t iv[ZIO_DATA_IV_LEN];
1480271171e0SMartin Matuska 		uint8_t mac[ZIO_DATA_MAC_LEN];
1481271171e0SMartin Matuska 		boolean_t no_crypt = B_FALSE;
1482271171e0SMartin Matuska 		dsl_pool_t *dp = dmu_objset_pool(rwa->os);
1483271171e0SMartin Matuska 		abd_t *eabd = abd_alloc_linear(BP_GET_PSIZE(bp), B_FALSE);
1484271171e0SMartin Matuska 
1485271171e0SMartin Matuska 		zio_crypt_decode_params_bp(bp, salt, iv);
1486271171e0SMartin Matuska 		zio_crypt_decode_mac_bp(bp, mac);
1487271171e0SMartin Matuska 
1488271171e0SMartin Matuska 		dsl_pool_config_enter(dp, FTAG);
1489271171e0SMartin Matuska 		err = dsl_dataset_hold_flags(dp, rwa->tofs,
1490271171e0SMartin Matuska 		    DS_HOLD_FLAG_DECRYPT, FTAG, &ds);
1491271171e0SMartin Matuska 		if (err != 0) {
1492271171e0SMartin Matuska 			dsl_pool_config_exit(dp, FTAG);
1493271171e0SMartin Matuska 			abd_free(eabd);
1494271171e0SMartin Matuska 			return (SET_ERROR(EACCES));
1495271171e0SMartin Matuska 		}
1496271171e0SMartin Matuska 
1497271171e0SMartin Matuska 		/* Look up the key from the spa's keystore */
1498271171e0SMartin Matuska 		err = spa_keystore_lookup_key(rwa->os->os_spa,
1499271171e0SMartin Matuska 		    zb.zb_objset, FTAG, &dck);
1500271171e0SMartin Matuska 		if (err != 0) {
1501271171e0SMartin Matuska 			dsl_dataset_rele_flags(ds, DS_HOLD_FLAG_DECRYPT,
1502271171e0SMartin Matuska 			    FTAG);
1503271171e0SMartin Matuska 			dsl_pool_config_exit(dp, FTAG);
1504271171e0SMartin Matuska 			abd_free(eabd);
1505271171e0SMartin Matuska 			return (SET_ERROR(EACCES));
1506271171e0SMartin Matuska 		}
1507271171e0SMartin Matuska 
1508271171e0SMartin Matuska 		err = zio_do_crypt_abd(B_TRUE, &dck->dck_key,
1509271171e0SMartin Matuska 		    BP_GET_TYPE(bp), BP_SHOULD_BYTESWAP(bp), salt, iv,
1510271171e0SMartin Matuska 		    mac, abd_get_size(abd), abd, eabd, &no_crypt);
1511271171e0SMartin Matuska 
1512271171e0SMartin Matuska 		spa_keystore_dsl_key_rele(rwa->os->os_spa, dck, FTAG);
1513271171e0SMartin Matuska 		dsl_dataset_rele_flags(ds, DS_HOLD_FLAG_DECRYPT, FTAG);
1514271171e0SMartin Matuska 		dsl_pool_config_exit(dp, FTAG);
1515271171e0SMartin Matuska 
1516271171e0SMartin Matuska 		ASSERT0(no_crypt);
1517271171e0SMartin Matuska 		if (err != 0) {
1518271171e0SMartin Matuska 			abd_free(eabd);
1519271171e0SMartin Matuska 			return (err);
1520271171e0SMartin Matuska 		}
1521271171e0SMartin Matuska 		/* Swap in the newly encrypted data into the abd */
1522271171e0SMartin Matuska 		abd_free(abd);
1523271171e0SMartin Matuska 		abd = eabd;
1524271171e0SMartin Matuska 
1525271171e0SMartin Matuska 		/*
1526271171e0SMartin Matuska 		 * We want to prevent zio_rewrite() from trying to
1527271171e0SMartin Matuska 		 * encrypt the data again
1528271171e0SMartin Matuska 		 */
1529271171e0SMartin Matuska 		flags |= ZIO_FLAG_RAW_ENCRYPT;
1530271171e0SMartin Matuska 	}
1531271171e0SMartin Matuska 	rrd->abd = abd;
1532271171e0SMartin Matuska 
1533783d3ff6SMartin Matuska 	io = zio_rewrite(NULL, rwa->os->os_spa, BP_GET_LOGICAL_BIRTH(bp), bp,
1534783d3ff6SMartin Matuska 	    abd, BP_GET_PSIZE(bp), NULL, NULL, ZIO_PRIORITY_SYNC_WRITE, flags,
1535783d3ff6SMartin Matuska 	    &zb);
1536271171e0SMartin Matuska 
1537271171e0SMartin Matuska 	ASSERT(abd_get_size(abd) == BP_GET_LSIZE(bp) ||
1538271171e0SMartin Matuska 	    abd_get_size(abd) == BP_GET_PSIZE(bp));
1539271171e0SMartin Matuska 
1540271171e0SMartin Matuska 	/* compute new bp checksum value and make sure it matches the old one */
1541271171e0SMartin Matuska 	zio_checksum_compute(io, BP_GET_CHECKSUM(bp), abd, abd_get_size(abd));
1542271171e0SMartin Matuska 	if (!ZIO_CHECKSUM_EQUAL(bp_cksum, io->io_bp->blk_cksum)) {
1543271171e0SMartin Matuska 		zio_destroy(io);
1544271171e0SMartin Matuska 		if (zfs_recv_best_effort_corrective != 0)
1545271171e0SMartin Matuska 			return (0);
1546271171e0SMartin Matuska 		return (SET_ERROR(ECKSUM));
1547271171e0SMartin Matuska 	}
1548271171e0SMartin Matuska 
1549271171e0SMartin Matuska 	/* Correct the corruption in place */
1550271171e0SMartin Matuska 	err = zio_wait(io);
1551271171e0SMartin Matuska 	if (err == 0) {
1552271171e0SMartin Matuska 		cr_cb_data_t *cb_data =
1553271171e0SMartin Matuska 		    kmem_alloc(sizeof (cr_cb_data_t), KM_SLEEP);
1554271171e0SMartin Matuska 		cb_data->spa = rwa->os->os_spa;
1555271171e0SMartin Matuska 		cb_data->size = drrw->drr_logical_size;
1556271171e0SMartin Matuska 		cb_data->zb = zb;
1557271171e0SMartin Matuska 		/* Test if healing worked by re-reading the bp */
1558271171e0SMartin Matuska 		err = zio_wait(zio_read(rwa->heal_pio, rwa->os->os_spa, bp,
1559271171e0SMartin Matuska 		    abd_alloc_for_io(drrw->drr_logical_size, B_FALSE),
1560271171e0SMartin Matuska 		    drrw->drr_logical_size, corrective_read_done,
1561271171e0SMartin Matuska 		    cb_data, ZIO_PRIORITY_ASYNC_READ, flags, NULL));
1562271171e0SMartin Matuska 	}
1563271171e0SMartin Matuska 	if (err != 0 && zfs_recv_best_effort_corrective != 0)
1564271171e0SMartin Matuska 		err = 0;
1565271171e0SMartin Matuska 
1566271171e0SMartin Matuska 	return (err);
1567271171e0SMartin Matuska }
1568271171e0SMartin Matuska 
1569eda14cbcSMatt Macy static int
receive_read(dmu_recv_cookie_t * drc,int len,void * buf)1570eda14cbcSMatt Macy receive_read(dmu_recv_cookie_t *drc, int len, void *buf)
1571eda14cbcSMatt Macy {
1572eda14cbcSMatt Macy 	int done = 0;
1573eda14cbcSMatt Macy 
1574eda14cbcSMatt Macy 	/*
1575eda14cbcSMatt Macy 	 * The code doesn't rely on this (lengths being multiples of 8).  See
1576eda14cbcSMatt Macy 	 * comment in dump_bytes.
1577eda14cbcSMatt Macy 	 */
1578eda14cbcSMatt Macy 	ASSERT(len % 8 == 0 ||
1579eda14cbcSMatt Macy 	    (drc->drc_featureflags & DMU_BACKUP_FEATURE_RAW) != 0);
1580eda14cbcSMatt Macy 
1581eda14cbcSMatt Macy 	while (done < len) {
158215f0b8c3SMartin Matuska 		ssize_t resid = len - done;
1583eda14cbcSMatt Macy 		zfs_file_t *fp = drc->drc_fp;
1584eda14cbcSMatt Macy 		int err = zfs_file_read(fp, (char *)buf + done,
1585eda14cbcSMatt Macy 		    len - done, &resid);
158615f0b8c3SMartin Matuska 		if (err == 0 && resid == len - done) {
1587eda14cbcSMatt Macy 			/*
1588eda14cbcSMatt Macy 			 * Note: ECKSUM or ZFS_ERR_STREAM_TRUNCATED indicates
1589eda14cbcSMatt Macy 			 * that the receive was interrupted and can
1590eda14cbcSMatt Macy 			 * potentially be resumed.
1591eda14cbcSMatt Macy 			 */
1592eda14cbcSMatt Macy 			err = SET_ERROR(ZFS_ERR_STREAM_TRUNCATED);
1593eda14cbcSMatt Macy 		}
1594eda14cbcSMatt Macy 		drc->drc_voff += len - done - resid;
1595eda14cbcSMatt Macy 		done = len - resid;
1596eda14cbcSMatt Macy 		if (err != 0)
1597eda14cbcSMatt Macy 			return (err);
1598eda14cbcSMatt Macy 	}
1599eda14cbcSMatt Macy 
1600eda14cbcSMatt Macy 	drc->drc_bytes_read += len;
1601eda14cbcSMatt Macy 
1602eda14cbcSMatt Macy 	ASSERT3U(done, ==, len);
1603eda14cbcSMatt Macy 	return (0);
1604eda14cbcSMatt Macy }
1605eda14cbcSMatt Macy 
1606eda14cbcSMatt Macy static inline uint8_t
deduce_nblkptr(dmu_object_type_t bonus_type,uint64_t bonus_size)1607eda14cbcSMatt Macy deduce_nblkptr(dmu_object_type_t bonus_type, uint64_t bonus_size)
1608eda14cbcSMatt Macy {
1609eda14cbcSMatt Macy 	if (bonus_type == DMU_OT_SA) {
1610eda14cbcSMatt Macy 		return (1);
1611eda14cbcSMatt Macy 	} else {
1612eda14cbcSMatt Macy 		return (1 +
1613eda14cbcSMatt Macy 		    ((DN_OLD_MAX_BONUSLEN -
1614eda14cbcSMatt Macy 		    MIN(DN_OLD_MAX_BONUSLEN, bonus_size)) >> SPA_BLKPTRSHIFT));
1615eda14cbcSMatt Macy 	}
1616eda14cbcSMatt Macy }
1617eda14cbcSMatt Macy 
1618eda14cbcSMatt Macy static void
save_resume_state(struct receive_writer_arg * rwa,uint64_t object,uint64_t offset,dmu_tx_t * tx)1619eda14cbcSMatt Macy save_resume_state(struct receive_writer_arg *rwa,
1620eda14cbcSMatt Macy     uint64_t object, uint64_t offset, dmu_tx_t *tx)
1621eda14cbcSMatt Macy {
1622eda14cbcSMatt Macy 	int txgoff = dmu_tx_get_txg(tx) & TXG_MASK;
1623eda14cbcSMatt Macy 
1624eda14cbcSMatt Macy 	if (!rwa->resumable)
1625eda14cbcSMatt Macy 		return;
1626eda14cbcSMatt Macy 
1627eda14cbcSMatt Macy 	/*
1628eda14cbcSMatt Macy 	 * We use ds_resume_bytes[] != 0 to indicate that we need to
1629eda14cbcSMatt Macy 	 * update this on disk, so it must not be 0.
1630eda14cbcSMatt Macy 	 */
1631eda14cbcSMatt Macy 	ASSERT(rwa->bytes_read != 0);
1632eda14cbcSMatt Macy 
1633eda14cbcSMatt Macy 	/*
1634eda14cbcSMatt Macy 	 * We only resume from write records, which have a valid
1635eda14cbcSMatt Macy 	 * (non-meta-dnode) object number.
1636eda14cbcSMatt Macy 	 */
1637eda14cbcSMatt Macy 	ASSERT(object != 0);
1638eda14cbcSMatt Macy 
1639eda14cbcSMatt Macy 	/*
1640eda14cbcSMatt Macy 	 * For resuming to work correctly, we must receive records in order,
1641eda14cbcSMatt Macy 	 * sorted by object,offset.  This is checked by the callers, but
1642eda14cbcSMatt Macy 	 * assert it here for good measure.
1643eda14cbcSMatt Macy 	 */
1644eda14cbcSMatt Macy 	ASSERT3U(object, >=, rwa->os->os_dsl_dataset->ds_resume_object[txgoff]);
1645eda14cbcSMatt Macy 	ASSERT(object != rwa->os->os_dsl_dataset->ds_resume_object[txgoff] ||
1646eda14cbcSMatt Macy 	    offset >= rwa->os->os_dsl_dataset->ds_resume_offset[txgoff]);
1647eda14cbcSMatt Macy 	ASSERT3U(rwa->bytes_read, >=,
1648eda14cbcSMatt Macy 	    rwa->os->os_dsl_dataset->ds_resume_bytes[txgoff]);
1649eda14cbcSMatt Macy 
1650eda14cbcSMatt Macy 	rwa->os->os_dsl_dataset->ds_resume_object[txgoff] = object;
1651eda14cbcSMatt Macy 	rwa->os->os_dsl_dataset->ds_resume_offset[txgoff] = offset;
1652eda14cbcSMatt Macy 	rwa->os->os_dsl_dataset->ds_resume_bytes[txgoff] = rwa->bytes_read;
1653eda14cbcSMatt Macy }
1654eda14cbcSMatt Macy 
1655eda14cbcSMatt Macy static int
receive_object_is_same_generation(objset_t * os,uint64_t object,dmu_object_type_t old_bonus_type,dmu_object_type_t new_bonus_type,const void * new_bonus,boolean_t * samegenp)1656eda14cbcSMatt Macy receive_object_is_same_generation(objset_t *os, uint64_t object,
1657eda14cbcSMatt Macy     dmu_object_type_t old_bonus_type, dmu_object_type_t new_bonus_type,
1658eda14cbcSMatt Macy     const void *new_bonus, boolean_t *samegenp)
1659eda14cbcSMatt Macy {
1660eda14cbcSMatt Macy 	zfs_file_info_t zoi;
1661eda14cbcSMatt Macy 	int err;
1662eda14cbcSMatt Macy 
1663eda14cbcSMatt Macy 	dmu_buf_t *old_bonus_dbuf;
1664eda14cbcSMatt Macy 	err = dmu_bonus_hold(os, object, FTAG, &old_bonus_dbuf);
1665eda14cbcSMatt Macy 	if (err != 0)
1666eda14cbcSMatt Macy 		return (err);
1667eda14cbcSMatt Macy 	err = dmu_get_file_info(os, old_bonus_type, old_bonus_dbuf->db_data,
1668eda14cbcSMatt Macy 	    &zoi);
1669eda14cbcSMatt Macy 	dmu_buf_rele(old_bonus_dbuf, FTAG);
1670eda14cbcSMatt Macy 	if (err != 0)
1671eda14cbcSMatt Macy 		return (err);
1672eda14cbcSMatt Macy 	uint64_t old_gen = zoi.zfi_generation;
1673eda14cbcSMatt Macy 
1674eda14cbcSMatt Macy 	err = dmu_get_file_info(os, new_bonus_type, new_bonus, &zoi);
1675eda14cbcSMatt Macy 	if (err != 0)
1676eda14cbcSMatt Macy 		return (err);
1677eda14cbcSMatt Macy 	uint64_t new_gen = zoi.zfi_generation;
1678eda14cbcSMatt Macy 
1679eda14cbcSMatt Macy 	*samegenp = (old_gen == new_gen);
1680eda14cbcSMatt Macy 	return (0);
1681eda14cbcSMatt Macy }
1682eda14cbcSMatt Macy 
1683eda14cbcSMatt Macy static int
receive_handle_existing_object(const struct receive_writer_arg * rwa,const struct drr_object * drro,const dmu_object_info_t * doi,const void * bonus_data,uint64_t * object_to_hold,uint32_t * new_blksz)1684eda14cbcSMatt Macy receive_handle_existing_object(const struct receive_writer_arg *rwa,
1685eda14cbcSMatt Macy     const struct drr_object *drro, const dmu_object_info_t *doi,
1686eda14cbcSMatt Macy     const void *bonus_data,
1687eda14cbcSMatt Macy     uint64_t *object_to_hold, uint32_t *new_blksz)
1688eda14cbcSMatt Macy {
1689eda14cbcSMatt Macy 	uint32_t indblksz = drro->drr_indblkshift ?
1690eda14cbcSMatt Macy 	    1ULL << drro->drr_indblkshift : 0;
1691eda14cbcSMatt Macy 	int nblkptr = deduce_nblkptr(drro->drr_bonustype,
1692eda14cbcSMatt Macy 	    drro->drr_bonuslen);
1693eda14cbcSMatt Macy 	uint8_t dn_slots = drro->drr_dn_slots != 0 ?
1694eda14cbcSMatt Macy 	    drro->drr_dn_slots : DNODE_MIN_SLOTS;
1695eda14cbcSMatt Macy 	boolean_t do_free_range = B_FALSE;
1696eda14cbcSMatt Macy 	int err;
1697eda14cbcSMatt Macy 
1698eda14cbcSMatt Macy 	*object_to_hold = drro->drr_object;
1699eda14cbcSMatt Macy 
1700eda14cbcSMatt Macy 	/* nblkptr should be bounded by the bonus size and type */
1701eda14cbcSMatt Macy 	if (rwa->raw && nblkptr != drro->drr_nblkptr)
1702eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
1703eda14cbcSMatt Macy 
1704eda14cbcSMatt Macy 	/*
1705eda14cbcSMatt Macy 	 * After the previous send stream, the sending system may
1706eda14cbcSMatt Macy 	 * have freed this object, and then happened to re-allocate
1707eda14cbcSMatt Macy 	 * this object number in a later txg. In this case, we are
1708eda14cbcSMatt Macy 	 * receiving a different logical file, and the block size may
1709eda14cbcSMatt Macy 	 * appear to be different.  i.e. we may have a different
1710eda14cbcSMatt Macy 	 * block size for this object than what the send stream says.
1711eda14cbcSMatt Macy 	 * In this case we need to remove the object's contents,
1712eda14cbcSMatt Macy 	 * so that its structure can be changed and then its contents
1713eda14cbcSMatt Macy 	 * entirely replaced by subsequent WRITE records.
1714eda14cbcSMatt Macy 	 *
1715eda14cbcSMatt Macy 	 * If this is a -L (--large-block) incremental stream, and
1716eda14cbcSMatt Macy 	 * the previous stream was not -L, the block size may appear
1717eda14cbcSMatt Macy 	 * to increase.  i.e. we may have a smaller block size for
1718eda14cbcSMatt Macy 	 * this object than what the send stream says.  In this case
1719eda14cbcSMatt Macy 	 * we need to keep the object's contents and block size
1720eda14cbcSMatt Macy 	 * intact, so that we don't lose parts of the object's
1721eda14cbcSMatt Macy 	 * contents that are not changed by this incremental send
1722eda14cbcSMatt Macy 	 * stream.
1723eda14cbcSMatt Macy 	 *
1724eda14cbcSMatt Macy 	 * We can distinguish between the two above cases by using
1725eda14cbcSMatt Macy 	 * the ZPL's generation number (see
1726eda14cbcSMatt Macy 	 * receive_object_is_same_generation()).  However, we only
1727eda14cbcSMatt Macy 	 * want to rely on the generation number when absolutely
1728eda14cbcSMatt Macy 	 * necessary, because with raw receives, the generation is
1729eda14cbcSMatt Macy 	 * encrypted.  We also want to minimize dependence on the
1730eda14cbcSMatt Macy 	 * ZPL, so that other types of datasets can also be received
1731eda14cbcSMatt Macy 	 * (e.g. ZVOLs, although note that ZVOLS currently do not
1732eda14cbcSMatt Macy 	 * reallocate their objects or change their structure).
1733eda14cbcSMatt Macy 	 * Therefore, we check a number of different cases where we
1734eda14cbcSMatt Macy 	 * know it is safe to discard the object's contents, before
1735eda14cbcSMatt Macy 	 * using the ZPL's generation number to make the above
1736eda14cbcSMatt Macy 	 * distinction.
1737eda14cbcSMatt Macy 	 */
1738eda14cbcSMatt Macy 	if (drro->drr_blksz != doi->doi_data_block_size) {
1739eda14cbcSMatt Macy 		if (rwa->raw) {
1740eda14cbcSMatt Macy 			/*
1741eda14cbcSMatt Macy 			 * RAW streams always have large blocks, so
1742eda14cbcSMatt Macy 			 * we are sure that the data is not needed
1743eda14cbcSMatt Macy 			 * due to changing --large-block to be on.
1744eda14cbcSMatt Macy 			 * Which is fortunate since the bonus buffer
1745eda14cbcSMatt Macy 			 * (which contains the ZPL generation) is
1746eda14cbcSMatt Macy 			 * encrypted, and the key might not be
1747eda14cbcSMatt Macy 			 * loaded.
1748eda14cbcSMatt Macy 			 */
1749eda14cbcSMatt Macy 			do_free_range = B_TRUE;
1750eda14cbcSMatt Macy 		} else if (rwa->full) {
1751eda14cbcSMatt Macy 			/*
1752eda14cbcSMatt Macy 			 * This is a full send stream, so it always
1753eda14cbcSMatt Macy 			 * replaces what we have.  Even if the
1754eda14cbcSMatt Macy 			 * generation numbers happen to match, this
1755eda14cbcSMatt Macy 			 * can not actually be the same logical file.
1756eda14cbcSMatt Macy 			 * This is relevant when receiving a full
1757eda14cbcSMatt Macy 			 * send as a clone.
1758eda14cbcSMatt Macy 			 */
1759eda14cbcSMatt Macy 			do_free_range = B_TRUE;
1760eda14cbcSMatt Macy 		} else if (drro->drr_type !=
1761eda14cbcSMatt Macy 		    DMU_OT_PLAIN_FILE_CONTENTS ||
1762eda14cbcSMatt Macy 		    doi->doi_type != DMU_OT_PLAIN_FILE_CONTENTS) {
1763eda14cbcSMatt Macy 			/*
1764eda14cbcSMatt Macy 			 * PLAIN_FILE_CONTENTS are the only type of
1765eda14cbcSMatt Macy 			 * objects that have ever been stored with
1766eda14cbcSMatt Macy 			 * large blocks, so we don't need the special
1767eda14cbcSMatt Macy 			 * logic below.  ZAP blocks can shrink (when
1768eda14cbcSMatt Macy 			 * there's only one block), so we don't want
1769eda14cbcSMatt Macy 			 * to hit the error below about block size
1770eda14cbcSMatt Macy 			 * only increasing.
1771eda14cbcSMatt Macy 			 */
1772eda14cbcSMatt Macy 			do_free_range = B_TRUE;
1773eda14cbcSMatt Macy 		} else if (doi->doi_max_offset <=
1774eda14cbcSMatt Macy 		    doi->doi_data_block_size) {
1775eda14cbcSMatt Macy 			/*
1776eda14cbcSMatt Macy 			 * There is only one block.  We can free it,
1777eda14cbcSMatt Macy 			 * because its contents will be replaced by a
1778eda14cbcSMatt Macy 			 * WRITE record.  This can not be the no-L ->
1779eda14cbcSMatt Macy 			 * -L case, because the no-L case would have
1780eda14cbcSMatt Macy 			 * resulted in multiple blocks.  If we
1781eda14cbcSMatt Macy 			 * supported -L -> no-L, it would not be safe
1782eda14cbcSMatt Macy 			 * to free the file's contents.  Fortunately,
1783eda14cbcSMatt Macy 			 * that is not allowed (see
1784eda14cbcSMatt Macy 			 * recv_check_large_blocks()).
1785eda14cbcSMatt Macy 			 */
1786eda14cbcSMatt Macy 			do_free_range = B_TRUE;
1787eda14cbcSMatt Macy 		} else {
1788eda14cbcSMatt Macy 			boolean_t is_same_gen;
1789eda14cbcSMatt Macy 			err = receive_object_is_same_generation(rwa->os,
1790eda14cbcSMatt Macy 			    drro->drr_object, doi->doi_bonus_type,
1791eda14cbcSMatt Macy 			    drro->drr_bonustype, bonus_data, &is_same_gen);
1792eda14cbcSMatt Macy 			if (err != 0)
1793eda14cbcSMatt Macy 				return (SET_ERROR(EINVAL));
1794eda14cbcSMatt Macy 
1795eda14cbcSMatt Macy 			if (is_same_gen) {
1796eda14cbcSMatt Macy 				/*
1797eda14cbcSMatt Macy 				 * This is the same logical file, and
1798eda14cbcSMatt Macy 				 * the block size must be increasing.
1799eda14cbcSMatt Macy 				 * It could only decrease if
1800eda14cbcSMatt Macy 				 * --large-block was changed to be
1801eda14cbcSMatt Macy 				 * off, which is checked in
1802eda14cbcSMatt Macy 				 * recv_check_large_blocks().
1803eda14cbcSMatt Macy 				 */
1804eda14cbcSMatt Macy 				if (drro->drr_blksz <=
1805eda14cbcSMatt Macy 				    doi->doi_data_block_size)
1806eda14cbcSMatt Macy 					return (SET_ERROR(EINVAL));
1807eda14cbcSMatt Macy 				/*
1808eda14cbcSMatt Macy 				 * We keep the existing blocksize and
1809eda14cbcSMatt Macy 				 * contents.
1810eda14cbcSMatt Macy 				 */
1811eda14cbcSMatt Macy 				*new_blksz =
1812eda14cbcSMatt Macy 				    doi->doi_data_block_size;
1813eda14cbcSMatt Macy 			} else {
1814eda14cbcSMatt Macy 				do_free_range = B_TRUE;
1815eda14cbcSMatt Macy 			}
1816eda14cbcSMatt Macy 		}
1817eda14cbcSMatt Macy 	}
1818eda14cbcSMatt Macy 
1819eda14cbcSMatt Macy 	/* nblkptr can only decrease if the object was reallocated */
1820eda14cbcSMatt Macy 	if (nblkptr < doi->doi_nblkptr)
1821eda14cbcSMatt Macy 		do_free_range = B_TRUE;
1822eda14cbcSMatt Macy 
1823eda14cbcSMatt Macy 	/* number of slots can only change on reallocation */
1824eda14cbcSMatt Macy 	if (dn_slots != doi->doi_dnodesize >> DNODE_SHIFT)
1825eda14cbcSMatt Macy 		do_free_range = B_TRUE;
1826eda14cbcSMatt Macy 
1827eda14cbcSMatt Macy 	/*
1828eda14cbcSMatt Macy 	 * For raw sends we also check a few other fields to
1829eda14cbcSMatt Macy 	 * ensure we are preserving the objset structure exactly
1830eda14cbcSMatt Macy 	 * as it was on the receive side:
1831eda14cbcSMatt Macy 	 *     - A changed indirect block size
1832eda14cbcSMatt Macy 	 *     - A smaller nlevels
1833eda14cbcSMatt Macy 	 */
1834eda14cbcSMatt Macy 	if (rwa->raw) {
1835eda14cbcSMatt Macy 		if (indblksz != doi->doi_metadata_block_size)
1836eda14cbcSMatt Macy 			do_free_range = B_TRUE;
1837eda14cbcSMatt Macy 		if (drro->drr_nlevels < doi->doi_indirection)
1838eda14cbcSMatt Macy 			do_free_range = B_TRUE;
1839eda14cbcSMatt Macy 	}
1840eda14cbcSMatt Macy 
1841eda14cbcSMatt Macy 	if (do_free_range) {
1842eda14cbcSMatt Macy 		err = dmu_free_long_range(rwa->os, drro->drr_object,
1843eda14cbcSMatt Macy 		    0, DMU_OBJECT_END);
1844eda14cbcSMatt Macy 		if (err != 0)
1845eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
1846eda14cbcSMatt Macy 	}
1847eda14cbcSMatt Macy 
1848eda14cbcSMatt Macy 	/*
1849315ee00fSMartin Matuska 	 * The dmu does not currently support decreasing nlevels or changing
1850315ee00fSMartin Matuska 	 * indirect block size if there is already one, same as changing the
1851315ee00fSMartin Matuska 	 * number of of dnode slots on an object.  For non-raw sends this
1852315ee00fSMartin Matuska 	 * does not matter and the new object can just use the previous one's
1853315ee00fSMartin Matuska 	 * parameters.  For raw sends, however, the structure of the received
1854315ee00fSMartin Matuska 	 * dnode (including indirects and dnode slots) must match that of the
1855315ee00fSMartin Matuska 	 * send side.  Therefore, instead of using dmu_object_reclaim(), we
1856315ee00fSMartin Matuska 	 * must free the object completely and call dmu_object_claim_dnsize()
1857315ee00fSMartin Matuska 	 * instead.
1858eda14cbcSMatt Macy 	 */
1859315ee00fSMartin Matuska 	if ((rwa->raw && ((doi->doi_indirection > 1 &&
1860315ee00fSMartin Matuska 	    indblksz != doi->doi_metadata_block_size) ||
1861315ee00fSMartin Matuska 	    drro->drr_nlevels < doi->doi_indirection)) ||
1862eda14cbcSMatt Macy 	    dn_slots != doi->doi_dnodesize >> DNODE_SHIFT) {
1863eda14cbcSMatt Macy 		err = dmu_free_long_object(rwa->os, drro->drr_object);
1864eda14cbcSMatt Macy 		if (err != 0)
1865eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
1866eda14cbcSMatt Macy 
1867eda14cbcSMatt Macy 		txg_wait_synced(dmu_objset_pool(rwa->os), 0);
1868eda14cbcSMatt Macy 		*object_to_hold = DMU_NEW_OBJECT;
1869eda14cbcSMatt Macy 	}
1870eda14cbcSMatt Macy 
1871eda14cbcSMatt Macy 	/*
1872eda14cbcSMatt Macy 	 * For raw receives, free everything beyond the new incoming
1873eda14cbcSMatt Macy 	 * maxblkid. Normally this would be done with a DRR_FREE
1874eda14cbcSMatt Macy 	 * record that would come after this DRR_OBJECT record is
1875eda14cbcSMatt Macy 	 * processed. However, for raw receives we manually set the
1876eda14cbcSMatt Macy 	 * maxblkid from the drr_maxblkid and so we must first free
1877eda14cbcSMatt Macy 	 * everything above that blkid to ensure the DMU is always
1878eda14cbcSMatt Macy 	 * consistent with itself. We will never free the first block
1879eda14cbcSMatt Macy 	 * of the object here because a maxblkid of 0 could indicate
1880eda14cbcSMatt Macy 	 * an object with a single block or one with no blocks. This
1881eda14cbcSMatt Macy 	 * free may be skipped when dmu_free_long_range() was called
1882eda14cbcSMatt Macy 	 * above since it covers the entire object's contents.
1883eda14cbcSMatt Macy 	 */
1884eda14cbcSMatt Macy 	if (rwa->raw && *object_to_hold != DMU_NEW_OBJECT && !do_free_range) {
1885eda14cbcSMatt Macy 		err = dmu_free_long_range(rwa->os, drro->drr_object,
1886eda14cbcSMatt Macy 		    (drro->drr_maxblkid + 1) * doi->doi_data_block_size,
1887eda14cbcSMatt Macy 		    DMU_OBJECT_END);
1888eda14cbcSMatt Macy 		if (err != 0)
1889eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
1890eda14cbcSMatt Macy 	}
1891eda14cbcSMatt Macy 	return (0);
1892eda14cbcSMatt Macy }
1893eda14cbcSMatt Macy 
1894eda14cbcSMatt Macy noinline static int
receive_object(struct receive_writer_arg * rwa,struct drr_object * drro,void * data)1895eda14cbcSMatt Macy receive_object(struct receive_writer_arg *rwa, struct drr_object *drro,
1896eda14cbcSMatt Macy     void *data)
1897eda14cbcSMatt Macy {
1898eda14cbcSMatt Macy 	dmu_object_info_t doi;
1899eda14cbcSMatt Macy 	dmu_tx_t *tx;
1900eda14cbcSMatt Macy 	int err;
1901eda14cbcSMatt Macy 	uint32_t new_blksz = drro->drr_blksz;
1902eda14cbcSMatt Macy 	uint8_t dn_slots = drro->drr_dn_slots != 0 ?
1903eda14cbcSMatt Macy 	    drro->drr_dn_slots : DNODE_MIN_SLOTS;
1904eda14cbcSMatt Macy 
1905eda14cbcSMatt Macy 	if (drro->drr_type == DMU_OT_NONE ||
1906eda14cbcSMatt Macy 	    !DMU_OT_IS_VALID(drro->drr_type) ||
1907eda14cbcSMatt Macy 	    !DMU_OT_IS_VALID(drro->drr_bonustype) ||
1908eda14cbcSMatt Macy 	    drro->drr_checksumtype >= ZIO_CHECKSUM_FUNCTIONS ||
1909eda14cbcSMatt Macy 	    drro->drr_compress >= ZIO_COMPRESS_FUNCTIONS ||
1910eda14cbcSMatt Macy 	    P2PHASE(drro->drr_blksz, SPA_MINBLOCKSIZE) ||
1911eda14cbcSMatt Macy 	    drro->drr_blksz < SPA_MINBLOCKSIZE ||
1912eda14cbcSMatt Macy 	    drro->drr_blksz > spa_maxblocksize(dmu_objset_spa(rwa->os)) ||
1913eda14cbcSMatt Macy 	    drro->drr_bonuslen >
1914eda14cbcSMatt Macy 	    DN_BONUS_SIZE(spa_maxdnodesize(dmu_objset_spa(rwa->os))) ||
1915eda14cbcSMatt Macy 	    dn_slots >
1916eda14cbcSMatt Macy 	    (spa_maxdnodesize(dmu_objset_spa(rwa->os)) >> DNODE_SHIFT)) {
1917eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
1918eda14cbcSMatt Macy 	}
1919eda14cbcSMatt Macy 
1920eda14cbcSMatt Macy 	if (rwa->raw) {
1921eda14cbcSMatt Macy 		/*
1922eda14cbcSMatt Macy 		 * We should have received a DRR_OBJECT_RANGE record
1923eda14cbcSMatt Macy 		 * containing this block and stored it in rwa.
1924eda14cbcSMatt Macy 		 */
1925eda14cbcSMatt Macy 		if (drro->drr_object < rwa->or_firstobj ||
1926eda14cbcSMatt Macy 		    drro->drr_object >= rwa->or_firstobj + rwa->or_numslots ||
1927eda14cbcSMatt Macy 		    drro->drr_raw_bonuslen < drro->drr_bonuslen ||
1928eda14cbcSMatt Macy 		    drro->drr_indblkshift > SPA_MAXBLOCKSHIFT ||
1929eda14cbcSMatt Macy 		    drro->drr_nlevels > DN_MAX_LEVELS ||
1930eda14cbcSMatt Macy 		    drro->drr_nblkptr > DN_MAX_NBLKPTR ||
1931eda14cbcSMatt Macy 		    DN_SLOTS_TO_BONUSLEN(dn_slots) <
1932eda14cbcSMatt Macy 		    drro->drr_raw_bonuslen)
1933eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
1934eda14cbcSMatt Macy 	} else {
1935eda14cbcSMatt Macy 		/*
1936eda14cbcSMatt Macy 		 * The DRR_OBJECT_SPILL flag is valid when the DRR_BEGIN
1937eda14cbcSMatt Macy 		 * record indicates this by setting DRR_FLAG_SPILL_BLOCK.
1938eda14cbcSMatt Macy 		 */
1939eda14cbcSMatt Macy 		if (((drro->drr_flags & ~(DRR_OBJECT_SPILL))) ||
1940eda14cbcSMatt Macy 		    (!rwa->spill && DRR_OBJECT_HAS_SPILL(drro->drr_flags))) {
1941eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
1942eda14cbcSMatt Macy 		}
1943eda14cbcSMatt Macy 
1944eda14cbcSMatt Macy 		if (drro->drr_raw_bonuslen != 0 || drro->drr_nblkptr != 0 ||
1945eda14cbcSMatt Macy 		    drro->drr_indblkshift != 0 || drro->drr_nlevels != 0) {
1946eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
1947eda14cbcSMatt Macy 		}
1948eda14cbcSMatt Macy 	}
1949eda14cbcSMatt Macy 
1950eda14cbcSMatt Macy 	err = dmu_object_info(rwa->os, drro->drr_object, &doi);
1951eda14cbcSMatt Macy 
1952eda14cbcSMatt Macy 	if (err != 0 && err != ENOENT && err != EEXIST)
1953eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
1954eda14cbcSMatt Macy 
1955eda14cbcSMatt Macy 	if (drro->drr_object > rwa->max_object)
1956eda14cbcSMatt Macy 		rwa->max_object = drro->drr_object;
1957eda14cbcSMatt Macy 
1958eda14cbcSMatt Macy 	/*
1959eda14cbcSMatt Macy 	 * If we are losing blkptrs or changing the block size this must
1960eda14cbcSMatt Macy 	 * be a new file instance.  We must clear out the previous file
1961eda14cbcSMatt Macy 	 * contents before we can change this type of metadata in the dnode.
1962eda14cbcSMatt Macy 	 * Raw receives will also check that the indirect structure of the
1963eda14cbcSMatt Macy 	 * dnode hasn't changed.
1964eda14cbcSMatt Macy 	 */
1965eda14cbcSMatt Macy 	uint64_t object_to_hold;
1966eda14cbcSMatt Macy 	if (err == 0) {
1967eda14cbcSMatt Macy 		err = receive_handle_existing_object(rwa, drro, &doi, data,
1968eda14cbcSMatt Macy 		    &object_to_hold, &new_blksz);
1969be181ee2SMartin Matuska 		if (err != 0)
1970be181ee2SMartin Matuska 			return (err);
1971eda14cbcSMatt Macy 	} else if (err == EEXIST) {
1972eda14cbcSMatt Macy 		/*
1973eda14cbcSMatt Macy 		 * The object requested is currently an interior slot of a
1974eda14cbcSMatt Macy 		 * multi-slot dnode. This will be resolved when the next txg
1975eda14cbcSMatt Macy 		 * is synced out, since the send stream will have told us
1976eda14cbcSMatt Macy 		 * to free this slot when we freed the associated dnode
1977eda14cbcSMatt Macy 		 * earlier in the stream.
1978eda14cbcSMatt Macy 		 */
1979eda14cbcSMatt Macy 		txg_wait_synced(dmu_objset_pool(rwa->os), 0);
1980eda14cbcSMatt Macy 
1981eda14cbcSMatt Macy 		if (dmu_object_info(rwa->os, drro->drr_object, NULL) != ENOENT)
1982eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
1983eda14cbcSMatt Macy 
1984eda14cbcSMatt Macy 		/* object was freed and we are about to allocate a new one */
1985eda14cbcSMatt Macy 		object_to_hold = DMU_NEW_OBJECT;
1986eda14cbcSMatt Macy 	} else {
198715f0b8c3SMartin Matuska 		/*
198815f0b8c3SMartin Matuska 		 * If the only record in this range so far was DRR_FREEOBJECTS
198915f0b8c3SMartin Matuska 		 * with at least one actually freed object, it's possible that
199015f0b8c3SMartin Matuska 		 * the block will now be converted to a hole. We need to wait
199115f0b8c3SMartin Matuska 		 * for the txg to sync to prevent races.
199215f0b8c3SMartin Matuska 		 */
199315f0b8c3SMartin Matuska 		if (rwa->or_need_sync == ORNS_YES)
199415f0b8c3SMartin Matuska 			txg_wait_synced(dmu_objset_pool(rwa->os), 0);
199515f0b8c3SMartin Matuska 
1996eda14cbcSMatt Macy 		/* object is free and we are about to allocate a new one */
1997eda14cbcSMatt Macy 		object_to_hold = DMU_NEW_OBJECT;
1998eda14cbcSMatt Macy 	}
1999eda14cbcSMatt Macy 
200015f0b8c3SMartin Matuska 	/* Only relevant for the first object in the range */
200115f0b8c3SMartin Matuska 	rwa->or_need_sync = ORNS_NO;
200215f0b8c3SMartin Matuska 
2003eda14cbcSMatt Macy 	/*
2004eda14cbcSMatt Macy 	 * If this is a multi-slot dnode there is a chance that this
2005eda14cbcSMatt Macy 	 * object will expand into a slot that is already used by
2006eda14cbcSMatt Macy 	 * another object from the previous snapshot. We must free
2007eda14cbcSMatt Macy 	 * these objects before we attempt to allocate the new dnode.
2008eda14cbcSMatt Macy 	 */
2009eda14cbcSMatt Macy 	if (dn_slots > 1) {
2010eda14cbcSMatt Macy 		boolean_t need_sync = B_FALSE;
2011eda14cbcSMatt Macy 
2012eda14cbcSMatt Macy 		for (uint64_t slot = drro->drr_object + 1;
2013eda14cbcSMatt Macy 		    slot < drro->drr_object + dn_slots;
2014eda14cbcSMatt Macy 		    slot++) {
2015eda14cbcSMatt Macy 			dmu_object_info_t slot_doi;
2016eda14cbcSMatt Macy 
2017eda14cbcSMatt Macy 			err = dmu_object_info(rwa->os, slot, &slot_doi);
2018eda14cbcSMatt Macy 			if (err == ENOENT || err == EEXIST)
2019eda14cbcSMatt Macy 				continue;
2020eda14cbcSMatt Macy 			else if (err != 0)
2021eda14cbcSMatt Macy 				return (err);
2022eda14cbcSMatt Macy 
2023eda14cbcSMatt Macy 			err = dmu_free_long_object(rwa->os, slot);
2024eda14cbcSMatt Macy 			if (err != 0)
2025eda14cbcSMatt Macy 				return (err);
2026eda14cbcSMatt Macy 
2027eda14cbcSMatt Macy 			need_sync = B_TRUE;
2028eda14cbcSMatt Macy 		}
2029eda14cbcSMatt Macy 
2030eda14cbcSMatt Macy 		if (need_sync)
2031eda14cbcSMatt Macy 			txg_wait_synced(dmu_objset_pool(rwa->os), 0);
2032eda14cbcSMatt Macy 	}
2033eda14cbcSMatt Macy 
2034eda14cbcSMatt Macy 	tx = dmu_tx_create(rwa->os);
2035eda14cbcSMatt Macy 	dmu_tx_hold_bonus(tx, object_to_hold);
2036eda14cbcSMatt Macy 	dmu_tx_hold_write(tx, object_to_hold, 0, 0);
203761145dc2SMartin Matuska 	err = dmu_tx_assign(tx, DMU_TX_WAIT);
2038eda14cbcSMatt Macy 	if (err != 0) {
2039eda14cbcSMatt Macy 		dmu_tx_abort(tx);
2040eda14cbcSMatt Macy 		return (err);
2041eda14cbcSMatt Macy 	}
2042eda14cbcSMatt Macy 
2043eda14cbcSMatt Macy 	if (object_to_hold == DMU_NEW_OBJECT) {
2044eda14cbcSMatt Macy 		/* Currently free, wants to be allocated */
2045eda14cbcSMatt Macy 		err = dmu_object_claim_dnsize(rwa->os, drro->drr_object,
2046eda14cbcSMatt Macy 		    drro->drr_type, new_blksz,
2047eda14cbcSMatt Macy 		    drro->drr_bonustype, drro->drr_bonuslen,
2048eda14cbcSMatt Macy 		    dn_slots << DNODE_SHIFT, tx);
2049eda14cbcSMatt Macy 	} else if (drro->drr_type != doi.doi_type ||
2050eda14cbcSMatt Macy 	    new_blksz != doi.doi_data_block_size ||
2051eda14cbcSMatt Macy 	    drro->drr_bonustype != doi.doi_bonus_type ||
2052eda14cbcSMatt Macy 	    drro->drr_bonuslen != doi.doi_bonus_size) {
2053eda14cbcSMatt Macy 		/* Currently allocated, but with different properties */
2054eda14cbcSMatt Macy 		err = dmu_object_reclaim_dnsize(rwa->os, drro->drr_object,
2055eda14cbcSMatt Macy 		    drro->drr_type, new_blksz,
2056eda14cbcSMatt Macy 		    drro->drr_bonustype, drro->drr_bonuslen,
2057eda14cbcSMatt Macy 		    dn_slots << DNODE_SHIFT, rwa->spill ?
2058eda14cbcSMatt Macy 		    DRR_OBJECT_HAS_SPILL(drro->drr_flags) : B_FALSE, tx);
2059eda14cbcSMatt Macy 	} else if (rwa->spill && !DRR_OBJECT_HAS_SPILL(drro->drr_flags)) {
2060eda14cbcSMatt Macy 		/*
2061eda14cbcSMatt Macy 		 * Currently allocated, the existing version of this object
2062eda14cbcSMatt Macy 		 * may reference a spill block that is no longer allocated
2063eda14cbcSMatt Macy 		 * at the source and needs to be freed.
2064eda14cbcSMatt Macy 		 */
2065eda14cbcSMatt Macy 		err = dmu_object_rm_spill(rwa->os, drro->drr_object, tx);
2066eda14cbcSMatt Macy 	}
2067eda14cbcSMatt Macy 
2068eda14cbcSMatt Macy 	if (err != 0) {
2069eda14cbcSMatt Macy 		dmu_tx_commit(tx);
2070eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2071eda14cbcSMatt Macy 	}
2072eda14cbcSMatt Macy 
2073eda14cbcSMatt Macy 	if (rwa->or_crypt_params_present) {
2074eda14cbcSMatt Macy 		/*
2075eda14cbcSMatt Macy 		 * Set the crypt params for the buffer associated with this
2076eda14cbcSMatt Macy 		 * range of dnodes.  This causes the blkptr_t to have the
2077eda14cbcSMatt Macy 		 * same crypt params (byteorder, salt, iv, mac) as on the
2078eda14cbcSMatt Macy 		 * sending side.
2079eda14cbcSMatt Macy 		 *
2080eda14cbcSMatt Macy 		 * Since we are committing this tx now, it is possible for
2081eda14cbcSMatt Macy 		 * the dnode block to end up on-disk with the incorrect MAC,
2082eda14cbcSMatt Macy 		 * if subsequent objects in this block are received in a
2083eda14cbcSMatt Macy 		 * different txg.  However, since the dataset is marked as
2084eda14cbcSMatt Macy 		 * inconsistent, no code paths will do a non-raw read (or
2085eda14cbcSMatt Macy 		 * decrypt the block / verify the MAC). The receive code and
2086eda14cbcSMatt Macy 		 * scrub code can safely do raw reads and verify the
2087eda14cbcSMatt Macy 		 * checksum.  They don't need to verify the MAC.
2088eda14cbcSMatt Macy 		 */
2089eda14cbcSMatt Macy 		dmu_buf_t *db = NULL;
2090eda14cbcSMatt Macy 		uint64_t offset = rwa->or_firstobj * DNODE_MIN_SIZE;
2091eda14cbcSMatt Macy 
2092eda14cbcSMatt Macy 		err = dmu_buf_hold_by_dnode(DMU_META_DNODE(rwa->os),
2093eda14cbcSMatt Macy 		    offset, FTAG, &db, DMU_READ_PREFETCH | DMU_READ_NO_DECRYPT);
2094eda14cbcSMatt Macy 		if (err != 0) {
2095eda14cbcSMatt Macy 			dmu_tx_commit(tx);
2096eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
2097eda14cbcSMatt Macy 		}
2098eda14cbcSMatt Macy 
2099eda14cbcSMatt Macy 		dmu_buf_set_crypt_params(db, rwa->or_byteorder,
2100eda14cbcSMatt Macy 		    rwa->or_salt, rwa->or_iv, rwa->or_mac, tx);
2101eda14cbcSMatt Macy 
2102eda14cbcSMatt Macy 		dmu_buf_rele(db, FTAG);
2103eda14cbcSMatt Macy 
2104eda14cbcSMatt Macy 		rwa->or_crypt_params_present = B_FALSE;
2105eda14cbcSMatt Macy 	}
2106eda14cbcSMatt Macy 
2107eda14cbcSMatt Macy 	dmu_object_set_checksum(rwa->os, drro->drr_object,
2108eda14cbcSMatt Macy 	    drro->drr_checksumtype, tx);
2109eda14cbcSMatt Macy 	dmu_object_set_compress(rwa->os, drro->drr_object,
2110eda14cbcSMatt Macy 	    drro->drr_compress, tx);
2111eda14cbcSMatt Macy 
2112eda14cbcSMatt Macy 	/* handle more restrictive dnode structuring for raw recvs */
2113eda14cbcSMatt Macy 	if (rwa->raw) {
2114eda14cbcSMatt Macy 		/*
2115eda14cbcSMatt Macy 		 * Set the indirect block size, block shift, nlevels.
2116eda14cbcSMatt Macy 		 * This will not fail because we ensured all of the
2117eda14cbcSMatt Macy 		 * blocks were freed earlier if this is a new object.
2118eda14cbcSMatt Macy 		 * For non-new objects block size and indirect block
2119eda14cbcSMatt Macy 		 * shift cannot change and nlevels can only increase.
2120eda14cbcSMatt Macy 		 */
2121eda14cbcSMatt Macy 		ASSERT3U(new_blksz, ==, drro->drr_blksz);
2122eda14cbcSMatt Macy 		VERIFY0(dmu_object_set_blocksize(rwa->os, drro->drr_object,
2123eda14cbcSMatt Macy 		    drro->drr_blksz, drro->drr_indblkshift, tx));
2124eda14cbcSMatt Macy 		VERIFY0(dmu_object_set_nlevels(rwa->os, drro->drr_object,
2125eda14cbcSMatt Macy 		    drro->drr_nlevels, tx));
2126eda14cbcSMatt Macy 
2127eda14cbcSMatt Macy 		/*
2128eda14cbcSMatt Macy 		 * Set the maxblkid. This will always succeed because
2129eda14cbcSMatt Macy 		 * we freed all blocks beyond the new maxblkid above.
2130eda14cbcSMatt Macy 		 */
2131eda14cbcSMatt Macy 		VERIFY0(dmu_object_set_maxblkid(rwa->os, drro->drr_object,
2132eda14cbcSMatt Macy 		    drro->drr_maxblkid, tx));
2133eda14cbcSMatt Macy 	}
2134eda14cbcSMatt Macy 
2135eda14cbcSMatt Macy 	if (data != NULL) {
2136eda14cbcSMatt Macy 		dmu_buf_t *db;
2137eda14cbcSMatt Macy 		dnode_t *dn;
2138eda14cbcSMatt Macy 		uint32_t flags = DMU_READ_NO_PREFETCH;
2139eda14cbcSMatt Macy 
2140eda14cbcSMatt Macy 		if (rwa->raw)
2141eda14cbcSMatt Macy 			flags |= DMU_READ_NO_DECRYPT;
2142eda14cbcSMatt Macy 
2143eda14cbcSMatt Macy 		VERIFY0(dnode_hold(rwa->os, drro->drr_object, FTAG, &dn));
2144eda14cbcSMatt Macy 		VERIFY0(dmu_bonus_hold_by_dnode(dn, FTAG, &db, flags));
2145eda14cbcSMatt Macy 
2146eda14cbcSMatt Macy 		dmu_buf_will_dirty(db, tx);
2147eda14cbcSMatt Macy 
2148eda14cbcSMatt Macy 		ASSERT3U(db->db_size, >=, drro->drr_bonuslen);
2149da5137abSMartin Matuska 		memcpy(db->db_data, data, DRR_OBJECT_PAYLOAD_SIZE(drro));
2150eda14cbcSMatt Macy 
2151eda14cbcSMatt Macy 		/*
2152eda14cbcSMatt Macy 		 * Raw bonus buffers have their byteorder determined by the
2153eda14cbcSMatt Macy 		 * DRR_OBJECT_RANGE record.
2154eda14cbcSMatt Macy 		 */
2155eda14cbcSMatt Macy 		if (rwa->byteswap && !rwa->raw) {
2156eda14cbcSMatt Macy 			dmu_object_byteswap_t byteswap =
2157eda14cbcSMatt Macy 			    DMU_OT_BYTESWAP(drro->drr_bonustype);
2158eda14cbcSMatt Macy 			dmu_ot_byteswap[byteswap].ob_func(db->db_data,
2159eda14cbcSMatt Macy 			    DRR_OBJECT_PAYLOAD_SIZE(drro));
2160eda14cbcSMatt Macy 		}
2161eda14cbcSMatt Macy 		dmu_buf_rele(db, FTAG);
2162eda14cbcSMatt Macy 		dnode_rele(dn, FTAG);
2163eda14cbcSMatt Macy 	}
2164783d3ff6SMartin Matuska 
2165783d3ff6SMartin Matuska 	/*
2166783d3ff6SMartin Matuska 	 * If the receive fails, we want the resume stream to start with the
2167783d3ff6SMartin Matuska 	 * same record that we last successfully received. There is no way to
2168783d3ff6SMartin Matuska 	 * request resume from the object record, but we can benefit from the
2169783d3ff6SMartin Matuska 	 * fact that sender always sends object record before anything else,
2170783d3ff6SMartin Matuska 	 * after which it will "resend" data at offset 0 and resume normally.
2171783d3ff6SMartin Matuska 	 */
2172783d3ff6SMartin Matuska 	save_resume_state(rwa, drro->drr_object, 0, tx);
2173783d3ff6SMartin Matuska 
2174eda14cbcSMatt Macy 	dmu_tx_commit(tx);
2175eda14cbcSMatt Macy 
2176eda14cbcSMatt Macy 	return (0);
2177eda14cbcSMatt Macy }
2178eda14cbcSMatt Macy 
2179eda14cbcSMatt Macy noinline static int
receive_freeobjects(struct receive_writer_arg * rwa,struct drr_freeobjects * drrfo)2180eda14cbcSMatt Macy receive_freeobjects(struct receive_writer_arg *rwa,
2181eda14cbcSMatt Macy     struct drr_freeobjects *drrfo)
2182eda14cbcSMatt Macy {
2183eda14cbcSMatt Macy 	uint64_t obj;
2184eda14cbcSMatt Macy 	int next_err = 0;
2185eda14cbcSMatt Macy 
2186eda14cbcSMatt Macy 	if (drrfo->drr_firstobj + drrfo->drr_numobjs < drrfo->drr_firstobj)
2187eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2188eda14cbcSMatt Macy 
2189eda14cbcSMatt Macy 	for (obj = drrfo->drr_firstobj == 0 ? 1 : drrfo->drr_firstobj;
2190eda14cbcSMatt Macy 	    obj < drrfo->drr_firstobj + drrfo->drr_numobjs &&
2191eda14cbcSMatt Macy 	    obj < DN_MAX_OBJECT && next_err == 0;
2192eda14cbcSMatt Macy 	    next_err = dmu_object_next(rwa->os, &obj, FALSE, 0)) {
2193eda14cbcSMatt Macy 		dmu_object_info_t doi;
2194eda14cbcSMatt Macy 		int err;
2195eda14cbcSMatt Macy 
2196eda14cbcSMatt Macy 		err = dmu_object_info(rwa->os, obj, &doi);
2197eda14cbcSMatt Macy 		if (err == ENOENT)
2198eda14cbcSMatt Macy 			continue;
2199eda14cbcSMatt Macy 		else if (err != 0)
2200eda14cbcSMatt Macy 			return (err);
2201eda14cbcSMatt Macy 
2202eda14cbcSMatt Macy 		err = dmu_free_long_object(rwa->os, obj);
2203eda14cbcSMatt Macy 
2204eda14cbcSMatt Macy 		if (err != 0)
2205eda14cbcSMatt Macy 			return (err);
220615f0b8c3SMartin Matuska 
220715f0b8c3SMartin Matuska 		if (rwa->or_need_sync == ORNS_MAYBE)
220815f0b8c3SMartin Matuska 			rwa->or_need_sync = ORNS_YES;
2209eda14cbcSMatt Macy 	}
2210eda14cbcSMatt Macy 	if (next_err != ESRCH)
2211eda14cbcSMatt Macy 		return (next_err);
2212eda14cbcSMatt Macy 	return (0);
2213eda14cbcSMatt Macy }
2214eda14cbcSMatt Macy 
2215eda14cbcSMatt Macy /*
2216eda14cbcSMatt Macy  * Note: if this fails, the caller will clean up any records left on the
2217eda14cbcSMatt Macy  * rwa->write_batch list.
2218eda14cbcSMatt Macy  */
2219eda14cbcSMatt Macy static int
flush_write_batch_impl(struct receive_writer_arg * rwa)2220eda14cbcSMatt Macy flush_write_batch_impl(struct receive_writer_arg *rwa)
2221eda14cbcSMatt Macy {
2222eda14cbcSMatt Macy 	dnode_t *dn;
2223eda14cbcSMatt Macy 	int err;
2224eda14cbcSMatt Macy 
2225eda14cbcSMatt Macy 	if (dnode_hold(rwa->os, rwa->last_object, FTAG, &dn) != 0)
2226eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2227eda14cbcSMatt Macy 
2228eda14cbcSMatt Macy 	struct receive_record_arg *last_rrd = list_tail(&rwa->write_batch);
2229eda14cbcSMatt Macy 	struct drr_write *last_drrw = &last_rrd->header.drr_u.drr_write;
2230eda14cbcSMatt Macy 
2231eda14cbcSMatt Macy 	struct receive_record_arg *first_rrd = list_head(&rwa->write_batch);
2232eda14cbcSMatt Macy 	struct drr_write *first_drrw = &first_rrd->header.drr_u.drr_write;
2233eda14cbcSMatt Macy 
2234eda14cbcSMatt Macy 	ASSERT3U(rwa->last_object, ==, last_drrw->drr_object);
2235eda14cbcSMatt Macy 	ASSERT3U(rwa->last_offset, ==, last_drrw->drr_offset);
2236eda14cbcSMatt Macy 
2237eda14cbcSMatt Macy 	dmu_tx_t *tx = dmu_tx_create(rwa->os);
2238eda14cbcSMatt Macy 	dmu_tx_hold_write_by_dnode(tx, dn, first_drrw->drr_offset,
2239eda14cbcSMatt Macy 	    last_drrw->drr_offset - first_drrw->drr_offset +
2240eda14cbcSMatt Macy 	    last_drrw->drr_logical_size);
224161145dc2SMartin Matuska 	err = dmu_tx_assign(tx, DMU_TX_WAIT);
2242eda14cbcSMatt Macy 	if (err != 0) {
2243eda14cbcSMatt Macy 		dmu_tx_abort(tx);
2244eda14cbcSMatt Macy 		dnode_rele(dn, FTAG);
2245eda14cbcSMatt Macy 		return (err);
2246eda14cbcSMatt Macy 	}
2247eda14cbcSMatt Macy 
2248eda14cbcSMatt Macy 	struct receive_record_arg *rrd;
2249eda14cbcSMatt Macy 	while ((rrd = list_head(&rwa->write_batch)) != NULL) {
2250eda14cbcSMatt Macy 		struct drr_write *drrw = &rrd->header.drr_u.drr_write;
22517877fdebSMatt Macy 		abd_t *abd = rrd->abd;
2252eda14cbcSMatt Macy 
2253eda14cbcSMatt Macy 		ASSERT3U(drrw->drr_object, ==, rwa->last_object);
2254eda14cbcSMatt Macy 
22557877fdebSMatt Macy 		if (drrw->drr_logical_size != dn->dn_datablksz) {
22567877fdebSMatt Macy 			/*
22577877fdebSMatt Macy 			 * The WRITE record is larger than the object's block
22587877fdebSMatt Macy 			 * size.  We must be receiving an incremental
22597877fdebSMatt Macy 			 * large-block stream into a dataset that previously did
22607877fdebSMatt Macy 			 * a non-large-block receive.  Lightweight writes must
22617877fdebSMatt Macy 			 * be exactly one block, so we need to decompress the
22627877fdebSMatt Macy 			 * data (if compressed) and do a normal dmu_write().
22637877fdebSMatt Macy 			 */
22647877fdebSMatt Macy 			ASSERT3U(drrw->drr_logical_size, >, dn->dn_datablksz);
22657877fdebSMatt Macy 			if (DRR_WRITE_COMPRESSED(drrw)) {
22667877fdebSMatt Macy 				abd_t *decomp_abd =
22677877fdebSMatt Macy 				    abd_alloc_linear(drrw->drr_logical_size,
22687877fdebSMatt Macy 				    B_FALSE);
22697877fdebSMatt Macy 
22707877fdebSMatt Macy 				err = zio_decompress_data(
22717877fdebSMatt Macy 				    drrw->drr_compressiontype,
2272e2df9bb4SMartin Matuska 				    abd, decomp_abd,
22737877fdebSMatt Macy 				    abd_get_size(abd),
22747877fdebSMatt Macy 				    abd_get_size(decomp_abd), NULL);
22757877fdebSMatt Macy 
22767877fdebSMatt Macy 				if (err == 0) {
22777877fdebSMatt Macy 					dmu_write_by_dnode(dn,
22787877fdebSMatt Macy 					    drrw->drr_offset,
22797877fdebSMatt Macy 					    drrw->drr_logical_size,
22807877fdebSMatt Macy 					    abd_to_buf(decomp_abd), tx);
22817877fdebSMatt Macy 				}
22827877fdebSMatt Macy 				abd_free(decomp_abd);
22837877fdebSMatt Macy 			} else {
22847877fdebSMatt Macy 				dmu_write_by_dnode(dn,
22857877fdebSMatt Macy 				    drrw->drr_offset,
22867877fdebSMatt Macy 				    drrw->drr_logical_size,
22877877fdebSMatt Macy 				    abd_to_buf(abd), tx);
22887877fdebSMatt Macy 			}
22897877fdebSMatt Macy 			if (err == 0)
22907877fdebSMatt Macy 				abd_free(abd);
22917877fdebSMatt Macy 		} else {
22922a58b312SMartin Matuska 			zio_prop_t zp = {0};
22937877fdebSMatt Macy 			dmu_write_policy(rwa->os, dn, 0, 0, &zp);
22947877fdebSMatt Macy 
2295dbd5678dSMartin Matuska 			zio_flag_t zio_flags = 0;
22967877fdebSMatt Macy 
22977877fdebSMatt Macy 			if (rwa->raw) {
22987877fdebSMatt Macy 				zp.zp_encrypt = B_TRUE;
22997877fdebSMatt Macy 				zp.zp_compress = drrw->drr_compressiontype;
23007877fdebSMatt Macy 				zp.zp_byteorder = ZFS_HOST_BYTEORDER ^
23017877fdebSMatt Macy 				    !!DRR_IS_RAW_BYTESWAPPED(drrw->drr_flags) ^
23027877fdebSMatt Macy 				    rwa->byteswap;
2303da5137abSMartin Matuska 				memcpy(zp.zp_salt, drrw->drr_salt,
23047877fdebSMatt Macy 				    ZIO_DATA_SALT_LEN);
2305da5137abSMartin Matuska 				memcpy(zp.zp_iv, drrw->drr_iv,
23067877fdebSMatt Macy 				    ZIO_DATA_IV_LEN);
2307da5137abSMartin Matuska 				memcpy(zp.zp_mac, drrw->drr_mac,
23087877fdebSMatt Macy 				    ZIO_DATA_MAC_LEN);
23097877fdebSMatt Macy 				if (DMU_OT_IS_ENCRYPTED(zp.zp_type)) {
23107877fdebSMatt Macy 					zp.zp_nopwrite = B_FALSE;
23117877fdebSMatt Macy 					zp.zp_copies = MIN(zp.zp_copies,
23127877fdebSMatt Macy 					    SPA_DVAS_PER_BP - 1);
231361145dc2SMartin Matuska 					zp.zp_gang_copies =
231461145dc2SMartin Matuska 					    MIN(zp.zp_gang_copies,
231561145dc2SMartin Matuska 					    SPA_DVAS_PER_BP - 1);
23167877fdebSMatt Macy 				}
23177877fdebSMatt Macy 				zio_flags |= ZIO_FLAG_RAW;
23187877fdebSMatt Macy 			} else if (DRR_WRITE_COMPRESSED(drrw)) {
23197877fdebSMatt Macy 				ASSERT3U(drrw->drr_compressed_size, >, 0);
23207877fdebSMatt Macy 				ASSERT3U(drrw->drr_logical_size, >=,
23217877fdebSMatt Macy 				    drrw->drr_compressed_size);
23227877fdebSMatt Macy 				zp.zp_compress = drrw->drr_compressiontype;
23237877fdebSMatt Macy 				zio_flags |= ZIO_FLAG_RAW_COMPRESS;
23247877fdebSMatt Macy 			} else if (rwa->byteswap) {
23257877fdebSMatt Macy 				/*
23267877fdebSMatt Macy 				 * Note: compressed blocks never need to be
23277877fdebSMatt Macy 				 * byteswapped, because WRITE records for
23287877fdebSMatt Macy 				 * metadata blocks are never compressed. The
23297877fdebSMatt Macy 				 * exception is raw streams, which are written
23307877fdebSMatt Macy 				 * in the original byteorder, and the byteorder
23317877fdebSMatt Macy 				 * bit is preserved in the BP by setting
23327877fdebSMatt Macy 				 * zp_byteorder above.
23337877fdebSMatt Macy 				 */
2334eda14cbcSMatt Macy 				dmu_object_byteswap_t byteswap =
2335eda14cbcSMatt Macy 				    DMU_OT_BYTESWAP(drrw->drr_type);
23367877fdebSMatt Macy 				dmu_ot_byteswap[byteswap].ob_func(
23377877fdebSMatt Macy 				    abd_to_buf(abd),
2338eda14cbcSMatt Macy 				    DRR_WRITE_PAYLOAD_SIZE(drrw));
2339eda14cbcSMatt Macy 			}
2340eda14cbcSMatt Macy 
2341eda14cbcSMatt Macy 			/*
23427877fdebSMatt Macy 			 * Since this data can't be read until the receive
23437877fdebSMatt Macy 			 * completes, we can do a "lightweight" write for
23447877fdebSMatt Macy 			 * improved performance.
2345eda14cbcSMatt Macy 			 */
23467877fdebSMatt Macy 			err = dmu_lightweight_write_by_dnode(dn,
23477877fdebSMatt Macy 			    drrw->drr_offset, abd, &zp, zio_flags, tx);
2348eda14cbcSMatt Macy 		}
2349eda14cbcSMatt Macy 
2350eda14cbcSMatt Macy 		if (err != 0) {
2351eda14cbcSMatt Macy 			/*
2352eda14cbcSMatt Macy 			 * This rrd is left on the list, so the caller will
23537877fdebSMatt Macy 			 * free it (and the abd).
2354eda14cbcSMatt Macy 			 */
2355eda14cbcSMatt Macy 			break;
2356eda14cbcSMatt Macy 		}
2357eda14cbcSMatt Macy 
2358eda14cbcSMatt Macy 		/*
2359eda14cbcSMatt Macy 		 * Note: If the receive fails, we want the resume stream to
2360eda14cbcSMatt Macy 		 * start with the same record that we last successfully
2361eda14cbcSMatt Macy 		 * received (as opposed to the next record), so that we can
2362eda14cbcSMatt Macy 		 * verify that we are resuming from the correct location.
2363eda14cbcSMatt Macy 		 */
2364eda14cbcSMatt Macy 		save_resume_state(rwa, drrw->drr_object, drrw->drr_offset, tx);
2365eda14cbcSMatt Macy 
2366eda14cbcSMatt Macy 		list_remove(&rwa->write_batch, rrd);
2367eda14cbcSMatt Macy 		kmem_free(rrd, sizeof (*rrd));
2368eda14cbcSMatt Macy 	}
2369eda14cbcSMatt Macy 
2370eda14cbcSMatt Macy 	dmu_tx_commit(tx);
2371eda14cbcSMatt Macy 	dnode_rele(dn, FTAG);
2372eda14cbcSMatt Macy 	return (err);
2373eda14cbcSMatt Macy }
2374eda14cbcSMatt Macy 
2375eda14cbcSMatt Macy noinline static int
flush_write_batch(struct receive_writer_arg * rwa)2376eda14cbcSMatt Macy flush_write_batch(struct receive_writer_arg *rwa)
2377eda14cbcSMatt Macy {
2378eda14cbcSMatt Macy 	if (list_is_empty(&rwa->write_batch))
2379eda14cbcSMatt Macy 		return (0);
2380eda14cbcSMatt Macy 	int err = rwa->err;
2381eda14cbcSMatt Macy 	if (err == 0)
2382eda14cbcSMatt Macy 		err = flush_write_batch_impl(rwa);
2383eda14cbcSMatt Macy 	if (err != 0) {
2384eda14cbcSMatt Macy 		struct receive_record_arg *rrd;
2385eda14cbcSMatt Macy 		while ((rrd = list_remove_head(&rwa->write_batch)) != NULL) {
23867877fdebSMatt Macy 			abd_free(rrd->abd);
2387eda14cbcSMatt Macy 			kmem_free(rrd, sizeof (*rrd));
2388eda14cbcSMatt Macy 		}
2389eda14cbcSMatt Macy 	}
2390eda14cbcSMatt Macy 	ASSERT(list_is_empty(&rwa->write_batch));
2391eda14cbcSMatt Macy 	return (err);
2392eda14cbcSMatt Macy }
2393eda14cbcSMatt Macy 
2394eda14cbcSMatt Macy noinline static int
receive_process_write_record(struct receive_writer_arg * rwa,struct receive_record_arg * rrd)2395eda14cbcSMatt Macy receive_process_write_record(struct receive_writer_arg *rwa,
2396eda14cbcSMatt Macy     struct receive_record_arg *rrd)
2397eda14cbcSMatt Macy {
2398eda14cbcSMatt Macy 	int err = 0;
2399eda14cbcSMatt Macy 
2400eda14cbcSMatt Macy 	ASSERT3U(rrd->header.drr_type, ==, DRR_WRITE);
2401eda14cbcSMatt Macy 	struct drr_write *drrw = &rrd->header.drr_u.drr_write;
2402eda14cbcSMatt Macy 
2403eda14cbcSMatt Macy 	if (drrw->drr_offset + drrw->drr_logical_size < drrw->drr_offset ||
2404eda14cbcSMatt Macy 	    !DMU_OT_IS_VALID(drrw->drr_type))
2405eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2406eda14cbcSMatt Macy 
2407271171e0SMartin Matuska 	if (rwa->heal) {
2408271171e0SMartin Matuska 		blkptr_t *bp;
2409271171e0SMartin Matuska 		dmu_buf_t *dbp;
2410271171e0SMartin Matuska 		int flags = DB_RF_CANFAIL;
2411271171e0SMartin Matuska 
2412271171e0SMartin Matuska 		if (rwa->raw)
2413271171e0SMartin Matuska 			flags |= DB_RF_NO_DECRYPT;
2414271171e0SMartin Matuska 
2415271171e0SMartin Matuska 		if (rwa->byteswap) {
2416271171e0SMartin Matuska 			dmu_object_byteswap_t byteswap =
2417271171e0SMartin Matuska 			    DMU_OT_BYTESWAP(drrw->drr_type);
2418271171e0SMartin Matuska 			dmu_ot_byteswap[byteswap].ob_func(abd_to_buf(rrd->abd),
2419271171e0SMartin Matuska 			    DRR_WRITE_PAYLOAD_SIZE(drrw));
2420271171e0SMartin Matuska 		}
2421271171e0SMartin Matuska 
2422271171e0SMartin Matuska 		err = dmu_buf_hold_noread(rwa->os, drrw->drr_object,
2423271171e0SMartin Matuska 		    drrw->drr_offset, FTAG, &dbp);
2424271171e0SMartin Matuska 		if (err != 0)
2425271171e0SMartin Matuska 			return (err);
2426271171e0SMartin Matuska 
2427271171e0SMartin Matuska 		/* Try to read the object to see if it needs healing */
2428271171e0SMartin Matuska 		err = dbuf_read((dmu_buf_impl_t *)dbp, NULL, flags);
2429271171e0SMartin Matuska 		/*
2430271171e0SMartin Matuska 		 * We only try to heal when dbuf_read() returns a ECKSUMs.
2431271171e0SMartin Matuska 		 * Other errors (even EIO) get returned to caller.
2432271171e0SMartin Matuska 		 * EIO indicates that the device is not present/accessible,
2433271171e0SMartin Matuska 		 * so writing to it will likely fail.
2434271171e0SMartin Matuska 		 * If the block is healthy, we don't want to overwrite it
2435271171e0SMartin Matuska 		 * unnecessarily.
2436271171e0SMartin Matuska 		 */
2437271171e0SMartin Matuska 		if (err != ECKSUM) {
2438271171e0SMartin Matuska 			dmu_buf_rele(dbp, FTAG);
2439271171e0SMartin Matuska 			return (err);
2440271171e0SMartin Matuska 		}
2441271171e0SMartin Matuska 		/* Make sure the on-disk block and recv record sizes match */
2442783d3ff6SMartin Matuska 		if (drrw->drr_logical_size != dbp->db_size) {
2443271171e0SMartin Matuska 			err = ENOTSUP;
2444271171e0SMartin Matuska 			dmu_buf_rele(dbp, FTAG);
2445271171e0SMartin Matuska 			return (err);
2446271171e0SMartin Matuska 		}
2447271171e0SMartin Matuska 		/* Get the block pointer for the corrupted block */
2448271171e0SMartin Matuska 		bp = dmu_buf_get_blkptr(dbp);
2449271171e0SMartin Matuska 		err = do_corrective_recv(rwa, drrw, rrd, bp);
2450271171e0SMartin Matuska 		dmu_buf_rele(dbp, FTAG);
2451271171e0SMartin Matuska 		return (err);
2452271171e0SMartin Matuska 	}
2453271171e0SMartin Matuska 
2454eda14cbcSMatt Macy 	/*
2455eda14cbcSMatt Macy 	 * For resuming to work, records must be in increasing order
2456eda14cbcSMatt Macy 	 * by (object, offset).
2457eda14cbcSMatt Macy 	 */
2458eda14cbcSMatt Macy 	if (drrw->drr_object < rwa->last_object ||
2459eda14cbcSMatt Macy 	    (drrw->drr_object == rwa->last_object &&
2460eda14cbcSMatt Macy 	    drrw->drr_offset < rwa->last_offset)) {
2461eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2462eda14cbcSMatt Macy 	}
2463eda14cbcSMatt Macy 
2464eda14cbcSMatt Macy 	struct receive_record_arg *first_rrd = list_head(&rwa->write_batch);
2465eda14cbcSMatt Macy 	struct drr_write *first_drrw = &first_rrd->header.drr_u.drr_write;
2466eda14cbcSMatt Macy 	uint64_t batch_size =
2467eda14cbcSMatt Macy 	    MIN(zfs_recv_write_batch_size, DMU_MAX_ACCESS / 2);
2468eda14cbcSMatt Macy 	if (first_rrd != NULL &&
2469eda14cbcSMatt Macy 	    (drrw->drr_object != first_drrw->drr_object ||
2470eda14cbcSMatt Macy 	    drrw->drr_offset >= first_drrw->drr_offset + batch_size)) {
2471eda14cbcSMatt Macy 		err = flush_write_batch(rwa);
2472eda14cbcSMatt Macy 		if (err != 0)
2473eda14cbcSMatt Macy 			return (err);
2474eda14cbcSMatt Macy 	}
2475eda14cbcSMatt Macy 
2476eda14cbcSMatt Macy 	rwa->last_object = drrw->drr_object;
2477eda14cbcSMatt Macy 	rwa->last_offset = drrw->drr_offset;
2478eda14cbcSMatt Macy 
2479eda14cbcSMatt Macy 	if (rwa->last_object > rwa->max_object)
2480eda14cbcSMatt Macy 		rwa->max_object = rwa->last_object;
2481eda14cbcSMatt Macy 
2482eda14cbcSMatt Macy 	list_insert_tail(&rwa->write_batch, rrd);
2483eda14cbcSMatt Macy 	/*
2484eda14cbcSMatt Macy 	 * Return EAGAIN to indicate that we will use this rrd again,
2485eda14cbcSMatt Macy 	 * so the caller should not free it
2486eda14cbcSMatt Macy 	 */
2487eda14cbcSMatt Macy 	return (EAGAIN);
2488eda14cbcSMatt Macy }
2489eda14cbcSMatt Macy 
2490eda14cbcSMatt Macy static int
receive_write_embedded(struct receive_writer_arg * rwa,struct drr_write_embedded * drrwe,void * data)2491eda14cbcSMatt Macy receive_write_embedded(struct receive_writer_arg *rwa,
2492eda14cbcSMatt Macy     struct drr_write_embedded *drrwe, void *data)
2493eda14cbcSMatt Macy {
2494eda14cbcSMatt Macy 	dmu_tx_t *tx;
2495eda14cbcSMatt Macy 	int err;
2496eda14cbcSMatt Macy 
2497eda14cbcSMatt Macy 	if (drrwe->drr_offset + drrwe->drr_length < drrwe->drr_offset)
2498eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2499eda14cbcSMatt Macy 
2500eda14cbcSMatt Macy 	if (drrwe->drr_psize > BPE_PAYLOAD_SIZE)
2501eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2502eda14cbcSMatt Macy 
2503eda14cbcSMatt Macy 	if (drrwe->drr_etype >= NUM_BP_EMBEDDED_TYPES)
2504eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2505eda14cbcSMatt Macy 	if (drrwe->drr_compression >= ZIO_COMPRESS_FUNCTIONS)
2506eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2507eda14cbcSMatt Macy 	if (rwa->raw)
2508eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2509eda14cbcSMatt Macy 
2510eda14cbcSMatt Macy 	if (drrwe->drr_object > rwa->max_object)
2511eda14cbcSMatt Macy 		rwa->max_object = drrwe->drr_object;
2512eda14cbcSMatt Macy 
2513eda14cbcSMatt Macy 	tx = dmu_tx_create(rwa->os);
2514eda14cbcSMatt Macy 
2515eda14cbcSMatt Macy 	dmu_tx_hold_write(tx, drrwe->drr_object,
2516eda14cbcSMatt Macy 	    drrwe->drr_offset, drrwe->drr_length);
251761145dc2SMartin Matuska 	err = dmu_tx_assign(tx, DMU_TX_WAIT);
2518eda14cbcSMatt Macy 	if (err != 0) {
2519eda14cbcSMatt Macy 		dmu_tx_abort(tx);
2520eda14cbcSMatt Macy 		return (err);
2521eda14cbcSMatt Macy 	}
2522eda14cbcSMatt Macy 
2523eda14cbcSMatt Macy 	dmu_write_embedded(rwa->os, drrwe->drr_object,
2524eda14cbcSMatt Macy 	    drrwe->drr_offset, data, drrwe->drr_etype,
2525eda14cbcSMatt Macy 	    drrwe->drr_compression, drrwe->drr_lsize, drrwe->drr_psize,
2526eda14cbcSMatt Macy 	    rwa->byteswap ^ ZFS_HOST_BYTEORDER, tx);
2527eda14cbcSMatt Macy 
2528eda14cbcSMatt Macy 	/* See comment in restore_write. */
2529eda14cbcSMatt Macy 	save_resume_state(rwa, drrwe->drr_object, drrwe->drr_offset, tx);
2530eda14cbcSMatt Macy 	dmu_tx_commit(tx);
2531eda14cbcSMatt Macy 	return (0);
2532eda14cbcSMatt Macy }
2533eda14cbcSMatt Macy 
2534eda14cbcSMatt Macy static int
receive_spill(struct receive_writer_arg * rwa,struct drr_spill * drrs,abd_t * abd)2535eda14cbcSMatt Macy receive_spill(struct receive_writer_arg *rwa, struct drr_spill *drrs,
25367877fdebSMatt Macy     abd_t *abd)
2537eda14cbcSMatt Macy {
2538eda14cbcSMatt Macy 	dmu_buf_t *db, *db_spill;
2539eda14cbcSMatt Macy 	int err;
2540eda14cbcSMatt Macy 
2541eda14cbcSMatt Macy 	if (drrs->drr_length < SPA_MINBLOCKSIZE ||
2542eda14cbcSMatt Macy 	    drrs->drr_length > spa_maxblocksize(dmu_objset_spa(rwa->os)))
2543eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2544eda14cbcSMatt Macy 
2545eda14cbcSMatt Macy 	/*
2546eda14cbcSMatt Macy 	 * This is an unmodified spill block which was added to the stream
2547eda14cbcSMatt Macy 	 * to resolve an issue with incorrectly removing spill blocks.  It
2548eda14cbcSMatt Macy 	 * should be ignored by current versions of the code which support
2549eda14cbcSMatt Macy 	 * the DRR_FLAG_SPILL_BLOCK flag.
2550eda14cbcSMatt Macy 	 */
2551eda14cbcSMatt Macy 	if (rwa->spill && DRR_SPILL_IS_UNMODIFIED(drrs->drr_flags)) {
25527877fdebSMatt Macy 		abd_free(abd);
2553eda14cbcSMatt Macy 		return (0);
2554eda14cbcSMatt Macy 	}
2555eda14cbcSMatt Macy 
2556eda14cbcSMatt Macy 	if (rwa->raw) {
2557eda14cbcSMatt Macy 		if (!DMU_OT_IS_VALID(drrs->drr_type) ||
2558eda14cbcSMatt Macy 		    drrs->drr_compressiontype >= ZIO_COMPRESS_FUNCTIONS ||
2559eda14cbcSMatt Macy 		    drrs->drr_compressed_size == 0)
2560eda14cbcSMatt Macy 			return (SET_ERROR(EINVAL));
2561eda14cbcSMatt Macy 	}
2562eda14cbcSMatt Macy 
2563eda14cbcSMatt Macy 	if (dmu_object_info(rwa->os, drrs->drr_object, NULL) != 0)
2564eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2565eda14cbcSMatt Macy 
2566eda14cbcSMatt Macy 	if (drrs->drr_object > rwa->max_object)
2567eda14cbcSMatt Macy 		rwa->max_object = drrs->drr_object;
2568eda14cbcSMatt Macy 
2569eda14cbcSMatt Macy 	VERIFY0(dmu_bonus_hold(rwa->os, drrs->drr_object, FTAG, &db));
2570eda14cbcSMatt Macy 	if ((err = dmu_spill_hold_by_bonus(db, DMU_READ_NO_DECRYPT, FTAG,
2571eda14cbcSMatt Macy 	    &db_spill)) != 0) {
2572eda14cbcSMatt Macy 		dmu_buf_rele(db, FTAG);
2573eda14cbcSMatt Macy 		return (err);
2574eda14cbcSMatt Macy 	}
2575eda14cbcSMatt Macy 
25767877fdebSMatt Macy 	dmu_tx_t *tx = dmu_tx_create(rwa->os);
2577eda14cbcSMatt Macy 
2578eda14cbcSMatt Macy 	dmu_tx_hold_spill(tx, db->db_object);
2579eda14cbcSMatt Macy 
258061145dc2SMartin Matuska 	err = dmu_tx_assign(tx, DMU_TX_WAIT);
2581eda14cbcSMatt Macy 	if (err != 0) {
2582eda14cbcSMatt Macy 		dmu_buf_rele(db, FTAG);
2583eda14cbcSMatt Macy 		dmu_buf_rele(db_spill, FTAG);
2584eda14cbcSMatt Macy 		dmu_tx_abort(tx);
2585eda14cbcSMatt Macy 		return (err);
2586eda14cbcSMatt Macy 	}
2587eda14cbcSMatt Macy 
2588eda14cbcSMatt Macy 	/*
2589eda14cbcSMatt Macy 	 * Spill blocks may both grow and shrink.  When a change in size
2590eda14cbcSMatt Macy 	 * occurs any existing dbuf must be updated to match the logical
2591eda14cbcSMatt Macy 	 * size of the provided arc_buf_t.
2592eda14cbcSMatt Macy 	 */
2593eda14cbcSMatt Macy 	if (db_spill->db_size != drrs->drr_length) {
2594188408daSMartin Matuska 		dmu_buf_will_fill(db_spill, tx, B_FALSE);
25957877fdebSMatt Macy 		VERIFY0(dbuf_spill_set_blksz(db_spill,
2596eda14cbcSMatt Macy 		    drrs->drr_length, tx));
2597eda14cbcSMatt Macy 	}
2598eda14cbcSMatt Macy 
25997877fdebSMatt Macy 	arc_buf_t *abuf;
26007877fdebSMatt Macy 	if (rwa->raw) {
26017877fdebSMatt Macy 		boolean_t byteorder = ZFS_HOST_BYTEORDER ^
26027877fdebSMatt Macy 		    !!DRR_IS_RAW_BYTESWAPPED(drrs->drr_flags) ^
26037877fdebSMatt Macy 		    rwa->byteswap;
26047877fdebSMatt Macy 
26057877fdebSMatt Macy 		abuf = arc_loan_raw_buf(dmu_objset_spa(rwa->os),
26067877fdebSMatt Macy 		    drrs->drr_object, byteorder, drrs->drr_salt,
26077877fdebSMatt Macy 		    drrs->drr_iv, drrs->drr_mac, drrs->drr_type,
26087877fdebSMatt Macy 		    drrs->drr_compressed_size, drrs->drr_length,
26097877fdebSMatt Macy 		    drrs->drr_compressiontype, 0);
26107877fdebSMatt Macy 	} else {
26117877fdebSMatt Macy 		abuf = arc_loan_buf(dmu_objset_spa(rwa->os),
26127877fdebSMatt Macy 		    DMU_OT_IS_METADATA(drrs->drr_type),
26137877fdebSMatt Macy 		    drrs->drr_length);
26147877fdebSMatt Macy 		if (rwa->byteswap) {
2615eda14cbcSMatt Macy 			dmu_object_byteswap_t byteswap =
2616eda14cbcSMatt Macy 			    DMU_OT_BYTESWAP(drrs->drr_type);
26177877fdebSMatt Macy 			dmu_ot_byteswap[byteswap].ob_func(abd_to_buf(abd),
2618eda14cbcSMatt Macy 			    DRR_SPILL_PAYLOAD_SIZE(drrs));
2619eda14cbcSMatt Macy 		}
26207877fdebSMatt Macy 	}
2621eda14cbcSMatt Macy 
2622da5137abSMartin Matuska 	memcpy(abuf->b_data, abd_to_buf(abd), DRR_SPILL_PAYLOAD_SIZE(drrs));
26237877fdebSMatt Macy 	abd_free(abd);
2624eda14cbcSMatt Macy 	dbuf_assign_arcbuf((dmu_buf_impl_t *)db_spill, abuf, tx);
2625eda14cbcSMatt Macy 
2626eda14cbcSMatt Macy 	dmu_buf_rele(db, FTAG);
2627eda14cbcSMatt Macy 	dmu_buf_rele(db_spill, FTAG);
2628eda14cbcSMatt Macy 
2629eda14cbcSMatt Macy 	dmu_tx_commit(tx);
2630eda14cbcSMatt Macy 	return (0);
2631eda14cbcSMatt Macy }
2632eda14cbcSMatt Macy 
2633eda14cbcSMatt Macy noinline static int
receive_free(struct receive_writer_arg * rwa,struct drr_free * drrf)2634eda14cbcSMatt Macy receive_free(struct receive_writer_arg *rwa, struct drr_free *drrf)
2635eda14cbcSMatt Macy {
2636eda14cbcSMatt Macy 	int err;
2637eda14cbcSMatt Macy 
2638eda14cbcSMatt Macy 	if (drrf->drr_length != -1ULL &&
2639eda14cbcSMatt Macy 	    drrf->drr_offset + drrf->drr_length < drrf->drr_offset)
2640eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2641eda14cbcSMatt Macy 
2642eda14cbcSMatt Macy 	if (dmu_object_info(rwa->os, drrf->drr_object, NULL) != 0)
2643eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2644eda14cbcSMatt Macy 
2645eda14cbcSMatt Macy 	if (drrf->drr_object > rwa->max_object)
2646eda14cbcSMatt Macy 		rwa->max_object = drrf->drr_object;
2647eda14cbcSMatt Macy 
2648eda14cbcSMatt Macy 	err = dmu_free_long_range(rwa->os, drrf->drr_object,
2649eda14cbcSMatt Macy 	    drrf->drr_offset, drrf->drr_length);
2650eda14cbcSMatt Macy 
2651eda14cbcSMatt Macy 	return (err);
2652eda14cbcSMatt Macy }
2653eda14cbcSMatt Macy 
2654eda14cbcSMatt Macy static int
receive_object_range(struct receive_writer_arg * rwa,struct drr_object_range * drror)2655eda14cbcSMatt Macy receive_object_range(struct receive_writer_arg *rwa,
2656eda14cbcSMatt Macy     struct drr_object_range *drror)
2657eda14cbcSMatt Macy {
2658eda14cbcSMatt Macy 	/*
2659eda14cbcSMatt Macy 	 * By default, we assume this block is in our native format
2660eda14cbcSMatt Macy 	 * (ZFS_HOST_BYTEORDER). We then take into account whether
2661eda14cbcSMatt Macy 	 * the send stream is byteswapped (rwa->byteswap). Finally,
2662eda14cbcSMatt Macy 	 * we need to byteswap again if this particular block was
2663eda14cbcSMatt Macy 	 * in non-native format on the send side.
2664eda14cbcSMatt Macy 	 */
2665eda14cbcSMatt Macy 	boolean_t byteorder = ZFS_HOST_BYTEORDER ^ rwa->byteswap ^
2666eda14cbcSMatt Macy 	    !!DRR_IS_RAW_BYTESWAPPED(drror->drr_flags);
2667eda14cbcSMatt Macy 
2668eda14cbcSMatt Macy 	/*
2669eda14cbcSMatt Macy 	 * Since dnode block sizes are constant, we should not need to worry
2670eda14cbcSMatt Macy 	 * about making sure that the dnode block size is the same on the
2671eda14cbcSMatt Macy 	 * sending and receiving sides for the time being. For non-raw sends,
2672eda14cbcSMatt Macy 	 * this does not matter (and in fact we do not send a DRR_OBJECT_RANGE
2673eda14cbcSMatt Macy 	 * record at all). Raw sends require this record type because the
2674eda14cbcSMatt Macy 	 * encryption parameters are used to protect an entire block of bonus
2675eda14cbcSMatt Macy 	 * buffers. If the size of dnode blocks ever becomes variable,
2676eda14cbcSMatt Macy 	 * handling will need to be added to ensure that dnode block sizes
2677eda14cbcSMatt Macy 	 * match on the sending and receiving side.
2678eda14cbcSMatt Macy 	 */
2679eda14cbcSMatt Macy 	if (drror->drr_numslots != DNODES_PER_BLOCK ||
2680eda14cbcSMatt Macy 	    P2PHASE(drror->drr_firstobj, DNODES_PER_BLOCK) != 0 ||
2681eda14cbcSMatt Macy 	    !rwa->raw)
2682eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2683eda14cbcSMatt Macy 
2684eda14cbcSMatt Macy 	if (drror->drr_firstobj > rwa->max_object)
2685eda14cbcSMatt Macy 		rwa->max_object = drror->drr_firstobj;
2686eda14cbcSMatt Macy 
2687eda14cbcSMatt Macy 	/*
2688eda14cbcSMatt Macy 	 * The DRR_OBJECT_RANGE handling must be deferred to receive_object()
2689eda14cbcSMatt Macy 	 * so that the block of dnodes is not written out when it's empty,
2690eda14cbcSMatt Macy 	 * and converted to a HOLE BP.
2691eda14cbcSMatt Macy 	 */
2692eda14cbcSMatt Macy 	rwa->or_crypt_params_present = B_TRUE;
2693eda14cbcSMatt Macy 	rwa->or_firstobj = drror->drr_firstobj;
2694eda14cbcSMatt Macy 	rwa->or_numslots = drror->drr_numslots;
2695da5137abSMartin Matuska 	memcpy(rwa->or_salt, drror->drr_salt, ZIO_DATA_SALT_LEN);
2696da5137abSMartin Matuska 	memcpy(rwa->or_iv, drror->drr_iv, ZIO_DATA_IV_LEN);
2697da5137abSMartin Matuska 	memcpy(rwa->or_mac, drror->drr_mac, ZIO_DATA_MAC_LEN);
2698eda14cbcSMatt Macy 	rwa->or_byteorder = byteorder;
2699eda14cbcSMatt Macy 
270015f0b8c3SMartin Matuska 	rwa->or_need_sync = ORNS_MAYBE;
270115f0b8c3SMartin Matuska 
2702eda14cbcSMatt Macy 	return (0);
2703eda14cbcSMatt Macy }
2704eda14cbcSMatt Macy 
2705eda14cbcSMatt Macy /*
2706eda14cbcSMatt Macy  * Until we have the ability to redact large ranges of data efficiently, we
2707eda14cbcSMatt Macy  * process these records as frees.
2708eda14cbcSMatt Macy  */
2709eda14cbcSMatt Macy noinline static int
receive_redact(struct receive_writer_arg * rwa,struct drr_redact * drrr)2710eda14cbcSMatt Macy receive_redact(struct receive_writer_arg *rwa, struct drr_redact *drrr)
2711eda14cbcSMatt Macy {
2712eda14cbcSMatt Macy 	struct drr_free drrf = {0};
2713eda14cbcSMatt Macy 	drrf.drr_length = drrr->drr_length;
2714eda14cbcSMatt Macy 	drrf.drr_object = drrr->drr_object;
2715eda14cbcSMatt Macy 	drrf.drr_offset = drrr->drr_offset;
2716eda14cbcSMatt Macy 	drrf.drr_toguid = drrr->drr_toguid;
2717eda14cbcSMatt Macy 	return (receive_free(rwa, &drrf));
2718eda14cbcSMatt Macy }
2719eda14cbcSMatt Macy 
2720eda14cbcSMatt Macy /* used to destroy the drc_ds on error */
2721eda14cbcSMatt Macy static void
dmu_recv_cleanup_ds(dmu_recv_cookie_t * drc)2722eda14cbcSMatt Macy dmu_recv_cleanup_ds(dmu_recv_cookie_t *drc)
2723eda14cbcSMatt Macy {
2724eda14cbcSMatt Macy 	dsl_dataset_t *ds = drc->drc_ds;
27257877fdebSMatt Macy 	ds_hold_flags_t dsflags;
2726eda14cbcSMatt Macy 
27277877fdebSMatt Macy 	dsflags = (drc->drc_raw) ? DS_HOLD_FLAG_NONE : DS_HOLD_FLAG_DECRYPT;
2728eda14cbcSMatt Macy 	/*
2729eda14cbcSMatt Macy 	 * Wait for the txg sync before cleaning up the receive. For
2730eda14cbcSMatt Macy 	 * resumable receives, this ensures that our resume state has
2731eda14cbcSMatt Macy 	 * been written out to disk. For raw receives, this ensures
2732eda14cbcSMatt Macy 	 * that the user accounting code will not attempt to do anything
2733eda14cbcSMatt Macy 	 * after we stopped receiving the dataset.
2734eda14cbcSMatt Macy 	 */
2735eda14cbcSMatt Macy 	txg_wait_synced(ds->ds_dir->dd_pool, 0);
2736eda14cbcSMatt Macy 	ds->ds_objset->os_raw_receive = B_FALSE;
2737eda14cbcSMatt Macy 
2738eda14cbcSMatt Macy 	rrw_enter(&ds->ds_bp_rwlock, RW_READER, FTAG);
2739eda14cbcSMatt Macy 	if (drc->drc_resumable && drc->drc_should_save &&
2740eda14cbcSMatt Macy 	    !BP_IS_HOLE(dsl_dataset_get_blkptr(ds))) {
2741eda14cbcSMatt Macy 		rrw_exit(&ds->ds_bp_rwlock, FTAG);
2742eda14cbcSMatt Macy 		dsl_dataset_disown(ds, dsflags, dmu_recv_tag);
2743eda14cbcSMatt Macy 	} else {
2744eda14cbcSMatt Macy 		char name[ZFS_MAX_DATASET_NAME_LEN];
2745eda14cbcSMatt Macy 		rrw_exit(&ds->ds_bp_rwlock, FTAG);
2746eda14cbcSMatt Macy 		dsl_dataset_name(ds, name);
2747eda14cbcSMatt Macy 		dsl_dataset_disown(ds, dsflags, dmu_recv_tag);
2748271171e0SMartin Matuska 		if (!drc->drc_heal)
2749eda14cbcSMatt Macy 			(void) dsl_destroy_head(name);
2750eda14cbcSMatt Macy 	}
2751eda14cbcSMatt Macy }
2752eda14cbcSMatt Macy 
2753eda14cbcSMatt Macy static void
receive_cksum(dmu_recv_cookie_t * drc,int len,void * buf)2754eda14cbcSMatt Macy receive_cksum(dmu_recv_cookie_t *drc, int len, void *buf)
2755eda14cbcSMatt Macy {
2756eda14cbcSMatt Macy 	if (drc->drc_byteswap) {
2757eda14cbcSMatt Macy 		(void) fletcher_4_incremental_byteswap(buf, len,
2758eda14cbcSMatt Macy 		    &drc->drc_cksum);
2759eda14cbcSMatt Macy 	} else {
2760eda14cbcSMatt Macy 		(void) fletcher_4_incremental_native(buf, len, &drc->drc_cksum);
2761eda14cbcSMatt Macy 	}
2762eda14cbcSMatt Macy }
2763eda14cbcSMatt Macy 
2764eda14cbcSMatt Macy /*
2765eda14cbcSMatt Macy  * Read the payload into a buffer of size len, and update the current record's
2766eda14cbcSMatt Macy  * payload field.
2767eda14cbcSMatt Macy  * Allocate drc->drc_next_rrd and read the next record's header into
2768eda14cbcSMatt Macy  * drc->drc_next_rrd->header.
2769eda14cbcSMatt Macy  * Verify checksum of payload and next record.
2770eda14cbcSMatt Macy  */
2771eda14cbcSMatt Macy static int
receive_read_payload_and_next_header(dmu_recv_cookie_t * drc,int len,void * buf)2772eda14cbcSMatt Macy receive_read_payload_and_next_header(dmu_recv_cookie_t *drc, int len, void *buf)
2773eda14cbcSMatt Macy {
2774eda14cbcSMatt Macy 	int err;
2775eda14cbcSMatt Macy 
2776eda14cbcSMatt Macy 	if (len != 0) {
2777eda14cbcSMatt Macy 		ASSERT3U(len, <=, SPA_MAXBLOCKSIZE);
2778eda14cbcSMatt Macy 		err = receive_read(drc, len, buf);
2779eda14cbcSMatt Macy 		if (err != 0)
2780eda14cbcSMatt Macy 			return (err);
2781eda14cbcSMatt Macy 		receive_cksum(drc, len, buf);
2782eda14cbcSMatt Macy 
2783eda14cbcSMatt Macy 		/* note: rrd is NULL when reading the begin record's payload */
2784eda14cbcSMatt Macy 		if (drc->drc_rrd != NULL) {
2785eda14cbcSMatt Macy 			drc->drc_rrd->payload = buf;
2786eda14cbcSMatt Macy 			drc->drc_rrd->payload_size = len;
2787eda14cbcSMatt Macy 			drc->drc_rrd->bytes_read = drc->drc_bytes_read;
2788eda14cbcSMatt Macy 		}
2789eda14cbcSMatt Macy 	} else {
2790eda14cbcSMatt Macy 		ASSERT3P(buf, ==, NULL);
2791eda14cbcSMatt Macy 	}
2792eda14cbcSMatt Macy 
2793eda14cbcSMatt Macy 	drc->drc_prev_cksum = drc->drc_cksum;
2794eda14cbcSMatt Macy 
2795eda14cbcSMatt Macy 	drc->drc_next_rrd = kmem_zalloc(sizeof (*drc->drc_next_rrd), KM_SLEEP);
2796eda14cbcSMatt Macy 	err = receive_read(drc, sizeof (drc->drc_next_rrd->header),
2797eda14cbcSMatt Macy 	    &drc->drc_next_rrd->header);
2798eda14cbcSMatt Macy 	drc->drc_next_rrd->bytes_read = drc->drc_bytes_read;
2799eda14cbcSMatt Macy 
2800eda14cbcSMatt Macy 	if (err != 0) {
2801eda14cbcSMatt Macy 		kmem_free(drc->drc_next_rrd, sizeof (*drc->drc_next_rrd));
2802eda14cbcSMatt Macy 		drc->drc_next_rrd = NULL;
2803eda14cbcSMatt Macy 		return (err);
2804eda14cbcSMatt Macy 	}
2805eda14cbcSMatt Macy 	if (drc->drc_next_rrd->header.drr_type == DRR_BEGIN) {
2806eda14cbcSMatt Macy 		kmem_free(drc->drc_next_rrd, sizeof (*drc->drc_next_rrd));
2807eda14cbcSMatt Macy 		drc->drc_next_rrd = NULL;
2808eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2809eda14cbcSMatt Macy 	}
2810eda14cbcSMatt Macy 
2811eda14cbcSMatt Macy 	/*
2812eda14cbcSMatt Macy 	 * Note: checksum is of everything up to but not including the
2813eda14cbcSMatt Macy 	 * checksum itself.
2814eda14cbcSMatt Macy 	 */
2815eda14cbcSMatt Macy 	ASSERT3U(offsetof(dmu_replay_record_t, drr_u.drr_checksum.drr_checksum),
2816eda14cbcSMatt Macy 	    ==, sizeof (dmu_replay_record_t) - sizeof (zio_cksum_t));
2817eda14cbcSMatt Macy 	receive_cksum(drc,
2818eda14cbcSMatt Macy 	    offsetof(dmu_replay_record_t, drr_u.drr_checksum.drr_checksum),
2819eda14cbcSMatt Macy 	    &drc->drc_next_rrd->header);
2820eda14cbcSMatt Macy 
2821eda14cbcSMatt Macy 	zio_cksum_t cksum_orig =
2822eda14cbcSMatt Macy 	    drc->drc_next_rrd->header.drr_u.drr_checksum.drr_checksum;
2823eda14cbcSMatt Macy 	zio_cksum_t *cksump =
2824eda14cbcSMatt Macy 	    &drc->drc_next_rrd->header.drr_u.drr_checksum.drr_checksum;
2825eda14cbcSMatt Macy 
2826eda14cbcSMatt Macy 	if (drc->drc_byteswap)
2827eda14cbcSMatt Macy 		byteswap_record(&drc->drc_next_rrd->header);
2828eda14cbcSMatt Macy 
2829eda14cbcSMatt Macy 	if ((!ZIO_CHECKSUM_IS_ZERO(cksump)) &&
2830eda14cbcSMatt Macy 	    !ZIO_CHECKSUM_EQUAL(drc->drc_cksum, *cksump)) {
2831eda14cbcSMatt Macy 		kmem_free(drc->drc_next_rrd, sizeof (*drc->drc_next_rrd));
2832eda14cbcSMatt Macy 		drc->drc_next_rrd = NULL;
2833eda14cbcSMatt Macy 		return (SET_ERROR(ECKSUM));
2834eda14cbcSMatt Macy 	}
2835eda14cbcSMatt Macy 
2836eda14cbcSMatt Macy 	receive_cksum(drc, sizeof (cksum_orig), &cksum_orig);
2837eda14cbcSMatt Macy 
2838eda14cbcSMatt Macy 	return (0);
2839eda14cbcSMatt Macy }
2840eda14cbcSMatt Macy 
2841eda14cbcSMatt Macy /*
2842eda14cbcSMatt Macy  * Issue the prefetch reads for any necessary indirect blocks.
2843eda14cbcSMatt Macy  *
2844eda14cbcSMatt Macy  * We use the object ignore list to tell us whether or not to issue prefetches
2845eda14cbcSMatt Macy  * for a given object.  We do this for both correctness (in case the blocksize
2846eda14cbcSMatt Macy  * of an object has changed) and performance (if the object doesn't exist, don't
2847eda14cbcSMatt Macy  * needlessly try to issue prefetches).  We also trim the list as we go through
2848eda14cbcSMatt Macy  * the stream to prevent it from growing to an unbounded size.
2849eda14cbcSMatt Macy  *
2850eda14cbcSMatt Macy  * The object numbers within will always be in sorted order, and any write
2851eda14cbcSMatt Macy  * records we see will also be in sorted order, but they're not sorted with
2852eda14cbcSMatt Macy  * respect to each other (i.e. we can get several object records before
2853eda14cbcSMatt Macy  * receiving each object's write records).  As a result, once we've reached a
2854eda14cbcSMatt Macy  * given object number, we can safely remove any reference to lower object
2855eda14cbcSMatt Macy  * numbers in the ignore list. In practice, we receive up to 32 object records
2856eda14cbcSMatt Macy  * before receiving write records, so the list can have up to 32 nodes in it.
2857eda14cbcSMatt Macy  */
2858eda14cbcSMatt Macy static void
receive_read_prefetch(dmu_recv_cookie_t * drc,uint64_t object,uint64_t offset,uint64_t length)2859eda14cbcSMatt Macy receive_read_prefetch(dmu_recv_cookie_t *drc, uint64_t object, uint64_t offset,
2860eda14cbcSMatt Macy     uint64_t length)
2861eda14cbcSMatt Macy {
2862eda14cbcSMatt Macy 	if (!objlist_exists(drc->drc_ignore_objlist, object)) {
2863eda14cbcSMatt Macy 		dmu_prefetch(drc->drc_os, object, 1, offset, length,
2864eda14cbcSMatt Macy 		    ZIO_PRIORITY_SYNC_READ);
2865eda14cbcSMatt Macy 	}
2866eda14cbcSMatt Macy }
2867eda14cbcSMatt Macy 
2868eda14cbcSMatt Macy /*
2869eda14cbcSMatt Macy  * Read records off the stream, issuing any necessary prefetches.
2870eda14cbcSMatt Macy  */
2871eda14cbcSMatt Macy static int
receive_read_record(dmu_recv_cookie_t * drc)2872eda14cbcSMatt Macy receive_read_record(dmu_recv_cookie_t *drc)
2873eda14cbcSMatt Macy {
2874eda14cbcSMatt Macy 	int err;
2875eda14cbcSMatt Macy 
2876eda14cbcSMatt Macy 	switch (drc->drc_rrd->header.drr_type) {
2877eda14cbcSMatt Macy 	case DRR_OBJECT:
2878eda14cbcSMatt Macy 	{
2879eda14cbcSMatt Macy 		struct drr_object *drro =
2880eda14cbcSMatt Macy 		    &drc->drc_rrd->header.drr_u.drr_object;
2881eda14cbcSMatt Macy 		uint32_t size = DRR_OBJECT_PAYLOAD_SIZE(drro);
2882eda14cbcSMatt Macy 		void *buf = NULL;
2883eda14cbcSMatt Macy 		dmu_object_info_t doi;
2884eda14cbcSMatt Macy 
2885eda14cbcSMatt Macy 		if (size != 0)
2886eda14cbcSMatt Macy 			buf = kmem_zalloc(size, KM_SLEEP);
2887eda14cbcSMatt Macy 
2888eda14cbcSMatt Macy 		err = receive_read_payload_and_next_header(drc, size, buf);
2889eda14cbcSMatt Macy 		if (err != 0) {
2890eda14cbcSMatt Macy 			kmem_free(buf, size);
2891eda14cbcSMatt Macy 			return (err);
2892eda14cbcSMatt Macy 		}
2893eda14cbcSMatt Macy 		err = dmu_object_info(drc->drc_os, drro->drr_object, &doi);
2894eda14cbcSMatt Macy 		/*
2895eda14cbcSMatt Macy 		 * See receive_read_prefetch for an explanation why we're
2896eda14cbcSMatt Macy 		 * storing this object in the ignore_obj_list.
2897eda14cbcSMatt Macy 		 */
2898eda14cbcSMatt Macy 		if (err == ENOENT || err == EEXIST ||
2899eda14cbcSMatt Macy 		    (err == 0 && doi.doi_data_block_size != drro->drr_blksz)) {
2900eda14cbcSMatt Macy 			objlist_insert(drc->drc_ignore_objlist,
2901eda14cbcSMatt Macy 			    drro->drr_object);
2902eda14cbcSMatt Macy 			err = 0;
2903eda14cbcSMatt Macy 		}
2904eda14cbcSMatt Macy 		return (err);
2905eda14cbcSMatt Macy 	}
2906eda14cbcSMatt Macy 	case DRR_FREEOBJECTS:
2907eda14cbcSMatt Macy 	{
2908eda14cbcSMatt Macy 		err = receive_read_payload_and_next_header(drc, 0, NULL);
2909eda14cbcSMatt Macy 		return (err);
2910eda14cbcSMatt Macy 	}
2911eda14cbcSMatt Macy 	case DRR_WRITE:
2912eda14cbcSMatt Macy 	{
2913eda14cbcSMatt Macy 		struct drr_write *drrw = &drc->drc_rrd->header.drr_u.drr_write;
29147877fdebSMatt Macy 		int size = DRR_WRITE_PAYLOAD_SIZE(drrw);
29157877fdebSMatt Macy 		abd_t *abd = abd_alloc_linear(size, B_FALSE);
29167877fdebSMatt Macy 		err = receive_read_payload_and_next_header(drc, size,
29177877fdebSMatt Macy 		    abd_to_buf(abd));
2918eda14cbcSMatt Macy 		if (err != 0) {
29197877fdebSMatt Macy 			abd_free(abd);
2920eda14cbcSMatt Macy 			return (err);
2921eda14cbcSMatt Macy 		}
29227877fdebSMatt Macy 		drc->drc_rrd->abd = abd;
2923eda14cbcSMatt Macy 		receive_read_prefetch(drc, drrw->drr_object, drrw->drr_offset,
2924eda14cbcSMatt Macy 		    drrw->drr_logical_size);
2925eda14cbcSMatt Macy 		return (err);
2926eda14cbcSMatt Macy 	}
2927eda14cbcSMatt Macy 	case DRR_WRITE_EMBEDDED:
2928eda14cbcSMatt Macy 	{
2929eda14cbcSMatt Macy 		struct drr_write_embedded *drrwe =
2930eda14cbcSMatt Macy 		    &drc->drc_rrd->header.drr_u.drr_write_embedded;
2931eda14cbcSMatt Macy 		uint32_t size = P2ROUNDUP(drrwe->drr_psize, 8);
2932eda14cbcSMatt Macy 		void *buf = kmem_zalloc(size, KM_SLEEP);
2933eda14cbcSMatt Macy 
2934eda14cbcSMatt Macy 		err = receive_read_payload_and_next_header(drc, size, buf);
2935eda14cbcSMatt Macy 		if (err != 0) {
2936eda14cbcSMatt Macy 			kmem_free(buf, size);
2937eda14cbcSMatt Macy 			return (err);
2938eda14cbcSMatt Macy 		}
2939eda14cbcSMatt Macy 
2940eda14cbcSMatt Macy 		receive_read_prefetch(drc, drrwe->drr_object, drrwe->drr_offset,
2941eda14cbcSMatt Macy 		    drrwe->drr_length);
2942eda14cbcSMatt Macy 		return (err);
2943eda14cbcSMatt Macy 	}
2944eda14cbcSMatt Macy 	case DRR_FREE:
2945eda14cbcSMatt Macy 	case DRR_REDACT:
2946eda14cbcSMatt Macy 	{
2947eda14cbcSMatt Macy 		/*
2948eda14cbcSMatt Macy 		 * It might be beneficial to prefetch indirect blocks here, but
2949eda14cbcSMatt Macy 		 * we don't really have the data to decide for sure.
2950eda14cbcSMatt Macy 		 */
2951eda14cbcSMatt Macy 		err = receive_read_payload_and_next_header(drc, 0, NULL);
2952eda14cbcSMatt Macy 		return (err);
2953eda14cbcSMatt Macy 	}
2954eda14cbcSMatt Macy 	case DRR_END:
2955eda14cbcSMatt Macy 	{
2956eda14cbcSMatt Macy 		struct drr_end *drre = &drc->drc_rrd->header.drr_u.drr_end;
2957eda14cbcSMatt Macy 		if (!ZIO_CHECKSUM_EQUAL(drc->drc_prev_cksum,
2958eda14cbcSMatt Macy 		    drre->drr_checksum))
2959eda14cbcSMatt Macy 			return (SET_ERROR(ECKSUM));
2960eda14cbcSMatt Macy 		return (0);
2961eda14cbcSMatt Macy 	}
2962eda14cbcSMatt Macy 	case DRR_SPILL:
2963eda14cbcSMatt Macy 	{
2964eda14cbcSMatt Macy 		struct drr_spill *drrs = &drc->drc_rrd->header.drr_u.drr_spill;
29657877fdebSMatt Macy 		int size = DRR_SPILL_PAYLOAD_SIZE(drrs);
29667877fdebSMatt Macy 		abd_t *abd = abd_alloc_linear(size, B_FALSE);
29677877fdebSMatt Macy 		err = receive_read_payload_and_next_header(drc, size,
29687877fdebSMatt Macy 		    abd_to_buf(abd));
2969eda14cbcSMatt Macy 		if (err != 0)
29707877fdebSMatt Macy 			abd_free(abd);
2971eda14cbcSMatt Macy 		else
29727877fdebSMatt Macy 			drc->drc_rrd->abd = abd;
2973eda14cbcSMatt Macy 		return (err);
2974eda14cbcSMatt Macy 	}
2975eda14cbcSMatt Macy 	case DRR_OBJECT_RANGE:
2976eda14cbcSMatt Macy 	{
2977eda14cbcSMatt Macy 		err = receive_read_payload_and_next_header(drc, 0, NULL);
2978eda14cbcSMatt Macy 		return (err);
2979eda14cbcSMatt Macy 
2980eda14cbcSMatt Macy 	}
2981eda14cbcSMatt Macy 	default:
2982eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
2983eda14cbcSMatt Macy 	}
2984eda14cbcSMatt Macy }
2985eda14cbcSMatt Macy 
2986eda14cbcSMatt Macy 
2987eda14cbcSMatt Macy 
2988eda14cbcSMatt Macy static void
dprintf_drr(struct receive_record_arg * rrd,int err)2989eda14cbcSMatt Macy dprintf_drr(struct receive_record_arg *rrd, int err)
2990eda14cbcSMatt Macy {
2991eda14cbcSMatt Macy #ifdef ZFS_DEBUG
2992eda14cbcSMatt Macy 	switch (rrd->header.drr_type) {
2993eda14cbcSMatt Macy 	case DRR_OBJECT:
2994eda14cbcSMatt Macy 	{
2995eda14cbcSMatt Macy 		struct drr_object *drro = &rrd->header.drr_u.drr_object;
2996eda14cbcSMatt Macy 		dprintf("drr_type = OBJECT obj = %llu type = %u "
2997eda14cbcSMatt Macy 		    "bonustype = %u blksz = %u bonuslen = %u cksumtype = %u "
2998eda14cbcSMatt Macy 		    "compress = %u dn_slots = %u err = %d\n",
299933b8c039SMartin Matuska 		    (u_longlong_t)drro->drr_object, drro->drr_type,
300033b8c039SMartin Matuska 		    drro->drr_bonustype, drro->drr_blksz, drro->drr_bonuslen,
3001eda14cbcSMatt Macy 		    drro->drr_checksumtype, drro->drr_compress,
3002eda14cbcSMatt Macy 		    drro->drr_dn_slots, err);
3003eda14cbcSMatt Macy 		break;
3004eda14cbcSMatt Macy 	}
3005eda14cbcSMatt Macy 	case DRR_FREEOBJECTS:
3006eda14cbcSMatt Macy 	{
3007eda14cbcSMatt Macy 		struct drr_freeobjects *drrfo =
3008eda14cbcSMatt Macy 		    &rrd->header.drr_u.drr_freeobjects;
3009eda14cbcSMatt Macy 		dprintf("drr_type = FREEOBJECTS firstobj = %llu "
3010eda14cbcSMatt Macy 		    "numobjs = %llu err = %d\n",
301133b8c039SMartin Matuska 		    (u_longlong_t)drrfo->drr_firstobj,
301233b8c039SMartin Matuska 		    (u_longlong_t)drrfo->drr_numobjs, err);
3013eda14cbcSMatt Macy 		break;
3014eda14cbcSMatt Macy 	}
3015eda14cbcSMatt Macy 	case DRR_WRITE:
3016eda14cbcSMatt Macy 	{
3017eda14cbcSMatt Macy 		struct drr_write *drrw = &rrd->header.drr_u.drr_write;
3018eda14cbcSMatt Macy 		dprintf("drr_type = WRITE obj = %llu type = %u offset = %llu "
3019eda14cbcSMatt Macy 		    "lsize = %llu cksumtype = %u flags = %u "
3020eda14cbcSMatt Macy 		    "compress = %u psize = %llu err = %d\n",
302133b8c039SMartin Matuska 		    (u_longlong_t)drrw->drr_object, drrw->drr_type,
302233b8c039SMartin Matuska 		    (u_longlong_t)drrw->drr_offset,
302333b8c039SMartin Matuska 		    (u_longlong_t)drrw->drr_logical_size,
302433b8c039SMartin Matuska 		    drrw->drr_checksumtype, drrw->drr_flags,
302533b8c039SMartin Matuska 		    drrw->drr_compressiontype,
302633b8c039SMartin Matuska 		    (u_longlong_t)drrw->drr_compressed_size, err);
3027eda14cbcSMatt Macy 		break;
3028eda14cbcSMatt Macy 	}
3029eda14cbcSMatt Macy 	case DRR_WRITE_BYREF:
3030eda14cbcSMatt Macy 	{
3031eda14cbcSMatt Macy 		struct drr_write_byref *drrwbr =
3032eda14cbcSMatt Macy 		    &rrd->header.drr_u.drr_write_byref;
3033eda14cbcSMatt Macy 		dprintf("drr_type = WRITE_BYREF obj = %llu offset = %llu "
3034eda14cbcSMatt Macy 		    "length = %llu toguid = %llx refguid = %llx "
3035eda14cbcSMatt Macy 		    "refobject = %llu refoffset = %llu cksumtype = %u "
3036eda14cbcSMatt Macy 		    "flags = %u err = %d\n",
303733b8c039SMartin Matuska 		    (u_longlong_t)drrwbr->drr_object,
303833b8c039SMartin Matuska 		    (u_longlong_t)drrwbr->drr_offset,
303933b8c039SMartin Matuska 		    (u_longlong_t)drrwbr->drr_length,
304033b8c039SMartin Matuska 		    (u_longlong_t)drrwbr->drr_toguid,
304133b8c039SMartin Matuska 		    (u_longlong_t)drrwbr->drr_refguid,
304233b8c039SMartin Matuska 		    (u_longlong_t)drrwbr->drr_refobject,
304333b8c039SMartin Matuska 		    (u_longlong_t)drrwbr->drr_refoffset,
304433b8c039SMartin Matuska 		    drrwbr->drr_checksumtype, drrwbr->drr_flags, err);
3045eda14cbcSMatt Macy 		break;
3046eda14cbcSMatt Macy 	}
3047eda14cbcSMatt Macy 	case DRR_WRITE_EMBEDDED:
3048eda14cbcSMatt Macy 	{
3049eda14cbcSMatt Macy 		struct drr_write_embedded *drrwe =
3050eda14cbcSMatt Macy 		    &rrd->header.drr_u.drr_write_embedded;
3051eda14cbcSMatt Macy 		dprintf("drr_type = WRITE_EMBEDDED obj = %llu offset = %llu "
3052eda14cbcSMatt Macy 		    "length = %llu compress = %u etype = %u lsize = %u "
3053eda14cbcSMatt Macy 		    "psize = %u err = %d\n",
305433b8c039SMartin Matuska 		    (u_longlong_t)drrwe->drr_object,
305533b8c039SMartin Matuska 		    (u_longlong_t)drrwe->drr_offset,
305633b8c039SMartin Matuska 		    (u_longlong_t)drrwe->drr_length,
3057eda14cbcSMatt Macy 		    drrwe->drr_compression, drrwe->drr_etype,
3058eda14cbcSMatt Macy 		    drrwe->drr_lsize, drrwe->drr_psize, err);
3059eda14cbcSMatt Macy 		break;
3060eda14cbcSMatt Macy 	}
3061eda14cbcSMatt Macy 	case DRR_FREE:
3062eda14cbcSMatt Macy 	{
3063eda14cbcSMatt Macy 		struct drr_free *drrf = &rrd->header.drr_u.drr_free;
3064eda14cbcSMatt Macy 		dprintf("drr_type = FREE obj = %llu offset = %llu "
3065eda14cbcSMatt Macy 		    "length = %lld err = %d\n",
306633b8c039SMartin Matuska 		    (u_longlong_t)drrf->drr_object,
306733b8c039SMartin Matuska 		    (u_longlong_t)drrf->drr_offset,
306833b8c039SMartin Matuska 		    (longlong_t)drrf->drr_length,
3069eda14cbcSMatt Macy 		    err);
3070eda14cbcSMatt Macy 		break;
3071eda14cbcSMatt Macy 	}
3072eda14cbcSMatt Macy 	case DRR_SPILL:
3073eda14cbcSMatt Macy 	{
3074eda14cbcSMatt Macy 		struct drr_spill *drrs = &rrd->header.drr_u.drr_spill;
3075eda14cbcSMatt Macy 		dprintf("drr_type = SPILL obj = %llu length = %llu "
307633b8c039SMartin Matuska 		    "err = %d\n", (u_longlong_t)drrs->drr_object,
307733b8c039SMartin Matuska 		    (u_longlong_t)drrs->drr_length, err);
3078eda14cbcSMatt Macy 		break;
3079eda14cbcSMatt Macy 	}
3080eda14cbcSMatt Macy 	case DRR_OBJECT_RANGE:
3081eda14cbcSMatt Macy 	{
3082eda14cbcSMatt Macy 		struct drr_object_range *drror =
3083eda14cbcSMatt Macy 		    &rrd->header.drr_u.drr_object_range;
3084eda14cbcSMatt Macy 		dprintf("drr_type = OBJECT_RANGE firstobj = %llu "
3085eda14cbcSMatt Macy 		    "numslots = %llu flags = %u err = %d\n",
308633b8c039SMartin Matuska 		    (u_longlong_t)drror->drr_firstobj,
308733b8c039SMartin Matuska 		    (u_longlong_t)drror->drr_numslots,
3088eda14cbcSMatt Macy 		    drror->drr_flags, err);
3089eda14cbcSMatt Macy 		break;
3090eda14cbcSMatt Macy 	}
3091eda14cbcSMatt Macy 	default:
3092eda14cbcSMatt Macy 		return;
3093eda14cbcSMatt Macy 	}
3094eda14cbcSMatt Macy #endif
3095eda14cbcSMatt Macy }
3096eda14cbcSMatt Macy 
3097eda14cbcSMatt Macy /*
3098eda14cbcSMatt Macy  * Commit the records to the pool.
3099eda14cbcSMatt Macy  */
3100eda14cbcSMatt Macy static int
receive_process_record(struct receive_writer_arg * rwa,struct receive_record_arg * rrd)3101eda14cbcSMatt Macy receive_process_record(struct receive_writer_arg *rwa,
3102eda14cbcSMatt Macy     struct receive_record_arg *rrd)
3103eda14cbcSMatt Macy {
3104eda14cbcSMatt Macy 	int err;
3105eda14cbcSMatt Macy 
3106eda14cbcSMatt Macy 	/* Processing in order, therefore bytes_read should be increasing. */
3107eda14cbcSMatt Macy 	ASSERT3U(rrd->bytes_read, >=, rwa->bytes_read);
3108eda14cbcSMatt Macy 	rwa->bytes_read = rrd->bytes_read;
3109eda14cbcSMatt Macy 
3110271171e0SMartin Matuska 	/* We can only heal write records; other ones get ignored */
3111271171e0SMartin Matuska 	if (rwa->heal && rrd->header.drr_type != DRR_WRITE) {
3112271171e0SMartin Matuska 		if (rrd->abd != NULL) {
3113271171e0SMartin Matuska 			abd_free(rrd->abd);
3114271171e0SMartin Matuska 			rrd->abd = NULL;
3115271171e0SMartin Matuska 		} else if (rrd->payload != NULL) {
3116271171e0SMartin Matuska 			kmem_free(rrd->payload, rrd->payload_size);
3117271171e0SMartin Matuska 			rrd->payload = NULL;
3118271171e0SMartin Matuska 		}
3119271171e0SMartin Matuska 		return (0);
3120271171e0SMartin Matuska 	}
3121271171e0SMartin Matuska 
3122271171e0SMartin Matuska 	if (!rwa->heal && rrd->header.drr_type != DRR_WRITE) {
3123eda14cbcSMatt Macy 		err = flush_write_batch(rwa);
3124eda14cbcSMatt Macy 		if (err != 0) {
31257877fdebSMatt Macy 			if (rrd->abd != NULL) {
31267877fdebSMatt Macy 				abd_free(rrd->abd);
31277877fdebSMatt Macy 				rrd->abd = NULL;
3128eda14cbcSMatt Macy 				rrd->payload = NULL;
3129eda14cbcSMatt Macy 			} else if (rrd->payload != NULL) {
3130eda14cbcSMatt Macy 				kmem_free(rrd->payload, rrd->payload_size);
3131eda14cbcSMatt Macy 				rrd->payload = NULL;
3132eda14cbcSMatt Macy 			}
3133eda14cbcSMatt Macy 
3134eda14cbcSMatt Macy 			return (err);
3135eda14cbcSMatt Macy 		}
3136eda14cbcSMatt Macy 	}
3137eda14cbcSMatt Macy 
3138eda14cbcSMatt Macy 	switch (rrd->header.drr_type) {
3139eda14cbcSMatt Macy 	case DRR_OBJECT:
3140eda14cbcSMatt Macy 	{
3141eda14cbcSMatt Macy 		struct drr_object *drro = &rrd->header.drr_u.drr_object;
3142eda14cbcSMatt Macy 		err = receive_object(rwa, drro, rrd->payload);
3143eda14cbcSMatt Macy 		kmem_free(rrd->payload, rrd->payload_size);
3144eda14cbcSMatt Macy 		rrd->payload = NULL;
3145eda14cbcSMatt Macy 		break;
3146eda14cbcSMatt Macy 	}
3147eda14cbcSMatt Macy 	case DRR_FREEOBJECTS:
3148eda14cbcSMatt Macy 	{
3149eda14cbcSMatt Macy 		struct drr_freeobjects *drrfo =
3150eda14cbcSMatt Macy 		    &rrd->header.drr_u.drr_freeobjects;
3151eda14cbcSMatt Macy 		err = receive_freeobjects(rwa, drrfo);
3152eda14cbcSMatt Macy 		break;
3153eda14cbcSMatt Macy 	}
3154eda14cbcSMatt Macy 	case DRR_WRITE:
3155eda14cbcSMatt Macy 	{
3156eda14cbcSMatt Macy 		err = receive_process_write_record(rwa, rrd);
3157271171e0SMartin Matuska 		if (rwa->heal) {
3158eda14cbcSMatt Macy 			/*
3159271171e0SMartin Matuska 			 * If healing - always free the abd after processing
3160271171e0SMartin Matuska 			 */
3161271171e0SMartin Matuska 			abd_free(rrd->abd);
3162271171e0SMartin Matuska 			rrd->abd = NULL;
3163271171e0SMartin Matuska 		} else if (err != EAGAIN) {
3164271171e0SMartin Matuska 			/*
3165271171e0SMartin Matuska 			 * On success, a non-healing
3166271171e0SMartin Matuska 			 * receive_process_write_record() returns
3167eda14cbcSMatt Macy 			 * EAGAIN to indicate that we do not want to free
3168eda14cbcSMatt Macy 			 * the rrd or arc_buf.
3169eda14cbcSMatt Macy 			 */
3170eda14cbcSMatt Macy 			ASSERT(err != 0);
31717877fdebSMatt Macy 			abd_free(rrd->abd);
31727877fdebSMatt Macy 			rrd->abd = NULL;
3173eda14cbcSMatt Macy 		}
3174eda14cbcSMatt Macy 		break;
3175eda14cbcSMatt Macy 	}
3176eda14cbcSMatt Macy 	case DRR_WRITE_EMBEDDED:
3177eda14cbcSMatt Macy 	{
3178eda14cbcSMatt Macy 		struct drr_write_embedded *drrwe =
3179eda14cbcSMatt Macy 		    &rrd->header.drr_u.drr_write_embedded;
3180eda14cbcSMatt Macy 		err = receive_write_embedded(rwa, drrwe, rrd->payload);
3181eda14cbcSMatt Macy 		kmem_free(rrd->payload, rrd->payload_size);
3182eda14cbcSMatt Macy 		rrd->payload = NULL;
3183eda14cbcSMatt Macy 		break;
3184eda14cbcSMatt Macy 	}
3185eda14cbcSMatt Macy 	case DRR_FREE:
3186eda14cbcSMatt Macy 	{
3187eda14cbcSMatt Macy 		struct drr_free *drrf = &rrd->header.drr_u.drr_free;
3188eda14cbcSMatt Macy 		err = receive_free(rwa, drrf);
3189eda14cbcSMatt Macy 		break;
3190eda14cbcSMatt Macy 	}
3191eda14cbcSMatt Macy 	case DRR_SPILL:
3192eda14cbcSMatt Macy 	{
3193eda14cbcSMatt Macy 		struct drr_spill *drrs = &rrd->header.drr_u.drr_spill;
31947877fdebSMatt Macy 		err = receive_spill(rwa, drrs, rrd->abd);
3195eda14cbcSMatt Macy 		if (err != 0)
31967877fdebSMatt Macy 			abd_free(rrd->abd);
31977877fdebSMatt Macy 		rrd->abd = NULL;
3198eda14cbcSMatt Macy 		rrd->payload = NULL;
3199eda14cbcSMatt Macy 		break;
3200eda14cbcSMatt Macy 	}
3201eda14cbcSMatt Macy 	case DRR_OBJECT_RANGE:
3202eda14cbcSMatt Macy 	{
3203eda14cbcSMatt Macy 		struct drr_object_range *drror =
3204eda14cbcSMatt Macy 		    &rrd->header.drr_u.drr_object_range;
3205eda14cbcSMatt Macy 		err = receive_object_range(rwa, drror);
3206eda14cbcSMatt Macy 		break;
3207eda14cbcSMatt Macy 	}
3208eda14cbcSMatt Macy 	case DRR_REDACT:
3209eda14cbcSMatt Macy 	{
3210eda14cbcSMatt Macy 		struct drr_redact *drrr = &rrd->header.drr_u.drr_redact;
3211eda14cbcSMatt Macy 		err = receive_redact(rwa, drrr);
3212eda14cbcSMatt Macy 		break;
3213eda14cbcSMatt Macy 	}
3214eda14cbcSMatt Macy 	default:
3215eda14cbcSMatt Macy 		err = (SET_ERROR(EINVAL));
3216eda14cbcSMatt Macy 	}
3217eda14cbcSMatt Macy 
3218eda14cbcSMatt Macy 	if (err != 0)
3219eda14cbcSMatt Macy 		dprintf_drr(rrd, err);
3220eda14cbcSMatt Macy 
3221eda14cbcSMatt Macy 	return (err);
3222eda14cbcSMatt Macy }
3223eda14cbcSMatt Macy 
3224eda14cbcSMatt Macy /*
3225eda14cbcSMatt Macy  * dmu_recv_stream's worker thread; pull records off the queue, and then call
3226eda14cbcSMatt Macy  * receive_process_record  When we're done, signal the main thread and exit.
3227eda14cbcSMatt Macy  */
3228da5137abSMartin Matuska static __attribute__((noreturn)) void
receive_writer_thread(void * arg)3229eda14cbcSMatt Macy receive_writer_thread(void *arg)
3230eda14cbcSMatt Macy {
3231eda14cbcSMatt Macy 	struct receive_writer_arg *rwa = arg;
3232eda14cbcSMatt Macy 	struct receive_record_arg *rrd;
3233eda14cbcSMatt Macy 	fstrans_cookie_t cookie = spl_fstrans_mark();
3234eda14cbcSMatt Macy 
3235eda14cbcSMatt Macy 	for (rrd = bqueue_dequeue(&rwa->q); !rrd->eos_marker;
3236eda14cbcSMatt Macy 	    rrd = bqueue_dequeue(&rwa->q)) {
3237eda14cbcSMatt Macy 		/*
3238eda14cbcSMatt Macy 		 * If there's an error, the main thread will stop putting things
3239eda14cbcSMatt Macy 		 * on the queue, but we need to clear everything in it before we
3240eda14cbcSMatt Macy 		 * can exit.
3241eda14cbcSMatt Macy 		 */
3242eda14cbcSMatt Macy 		int err = 0;
3243eda14cbcSMatt Macy 		if (rwa->err == 0) {
3244eda14cbcSMatt Macy 			err = receive_process_record(rwa, rrd);
32457877fdebSMatt Macy 		} else if (rrd->abd != NULL) {
32467877fdebSMatt Macy 			abd_free(rrd->abd);
32477877fdebSMatt Macy 			rrd->abd = NULL;
3248eda14cbcSMatt Macy 			rrd->payload = NULL;
3249eda14cbcSMatt Macy 		} else if (rrd->payload != NULL) {
3250eda14cbcSMatt Macy 			kmem_free(rrd->payload, rrd->payload_size);
3251eda14cbcSMatt Macy 			rrd->payload = NULL;
3252eda14cbcSMatt Macy 		}
3253eda14cbcSMatt Macy 		/*
3254eda14cbcSMatt Macy 		 * EAGAIN indicates that this record has been saved (on
3255eda14cbcSMatt Macy 		 * raw->write_batch), and will be used again, so we don't
3256eda14cbcSMatt Macy 		 * free it.
3257271171e0SMartin Matuska 		 * When healing data we always need to free the record.
3258eda14cbcSMatt Macy 		 */
3259271171e0SMartin Matuska 		if (err != EAGAIN || rwa->heal) {
3260eda14cbcSMatt Macy 			if (rwa->err == 0)
3261eda14cbcSMatt Macy 				rwa->err = err;
3262eda14cbcSMatt Macy 			kmem_free(rrd, sizeof (*rrd));
3263eda14cbcSMatt Macy 		}
3264eda14cbcSMatt Macy 	}
3265eda14cbcSMatt Macy 	kmem_free(rrd, sizeof (*rrd));
3266eda14cbcSMatt Macy 
3267271171e0SMartin Matuska 	if (rwa->heal) {
3268271171e0SMartin Matuska 		zio_wait(rwa->heal_pio);
3269271171e0SMartin Matuska 	} else {
3270eda14cbcSMatt Macy 		int err = flush_write_batch(rwa);
3271eda14cbcSMatt Macy 		if (rwa->err == 0)
3272eda14cbcSMatt Macy 			rwa->err = err;
3273271171e0SMartin Matuska 	}
3274eda14cbcSMatt Macy 	mutex_enter(&rwa->mutex);
3275eda14cbcSMatt Macy 	rwa->done = B_TRUE;
3276eda14cbcSMatt Macy 	cv_signal(&rwa->cv);
3277eda14cbcSMatt Macy 	mutex_exit(&rwa->mutex);
3278eda14cbcSMatt Macy 	spl_fstrans_unmark(cookie);
3279eda14cbcSMatt Macy 	thread_exit();
3280eda14cbcSMatt Macy }
3281eda14cbcSMatt Macy 
3282eda14cbcSMatt Macy static int
resume_check(dmu_recv_cookie_t * drc,nvlist_t * begin_nvl)3283eda14cbcSMatt Macy resume_check(dmu_recv_cookie_t *drc, nvlist_t *begin_nvl)
3284eda14cbcSMatt Macy {
3285eda14cbcSMatt Macy 	uint64_t val;
3286eda14cbcSMatt Macy 	objset_t *mos = dmu_objset_pool(drc->drc_os)->dp_meta_objset;
3287eda14cbcSMatt Macy 	uint64_t dsobj = dmu_objset_id(drc->drc_os);
3288eda14cbcSMatt Macy 	uint64_t resume_obj, resume_off;
3289eda14cbcSMatt Macy 
3290eda14cbcSMatt Macy 	if (nvlist_lookup_uint64(begin_nvl,
3291eda14cbcSMatt Macy 	    "resume_object", &resume_obj) != 0 ||
3292eda14cbcSMatt Macy 	    nvlist_lookup_uint64(begin_nvl,
3293eda14cbcSMatt Macy 	    "resume_offset", &resume_off) != 0) {
3294eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
3295eda14cbcSMatt Macy 	}
3296eda14cbcSMatt Macy 	VERIFY0(zap_lookup(mos, dsobj,
3297eda14cbcSMatt Macy 	    DS_FIELD_RESUME_OBJECT, sizeof (val), 1, &val));
3298eda14cbcSMatt Macy 	if (resume_obj != val)
3299eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
3300eda14cbcSMatt Macy 	VERIFY0(zap_lookup(mos, dsobj,
3301eda14cbcSMatt Macy 	    DS_FIELD_RESUME_OFFSET, sizeof (val), 1, &val));
3302eda14cbcSMatt Macy 	if (resume_off != val)
3303eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
3304eda14cbcSMatt Macy 
3305eda14cbcSMatt Macy 	return (0);
3306eda14cbcSMatt Macy }
3307eda14cbcSMatt Macy 
3308eda14cbcSMatt Macy /*
3309eda14cbcSMatt Macy  * Read in the stream's records, one by one, and apply them to the pool.  There
3310eda14cbcSMatt Macy  * are two threads involved; the thread that calls this function will spin up a
3311eda14cbcSMatt Macy  * worker thread, read the records off the stream one by one, and issue
3312eda14cbcSMatt Macy  * prefetches for any necessary indirect blocks.  It will then push the records
3313eda14cbcSMatt Macy  * onto an internal blocking queue.  The worker thread will pull the records off
3314eda14cbcSMatt Macy  * the queue, and actually write the data into the DMU.  This way, the worker
3315eda14cbcSMatt Macy  * thread doesn't have to wait for reads to complete, since everything it needs
3316eda14cbcSMatt Macy  * (the indirect blocks) will be prefetched.
3317eda14cbcSMatt Macy  *
3318eda14cbcSMatt Macy  * NB: callers *must* call dmu_recv_end() if this succeeds.
3319eda14cbcSMatt Macy  */
3320eda14cbcSMatt Macy int
dmu_recv_stream(dmu_recv_cookie_t * drc,offset_t * voffp)3321eda14cbcSMatt Macy dmu_recv_stream(dmu_recv_cookie_t *drc, offset_t *voffp)
3322eda14cbcSMatt Macy {
3323eda14cbcSMatt Macy 	int err = 0;
3324eda14cbcSMatt Macy 	struct receive_writer_arg *rwa = kmem_zalloc(sizeof (*rwa), KM_SLEEP);
3325eda14cbcSMatt Macy 
332616038816SMartin Matuska 	if (dsl_dataset_has_resume_receive_state(drc->drc_ds)) {
332716038816SMartin Matuska 		uint64_t bytes = 0;
3328eda14cbcSMatt Macy 		(void) zap_lookup(drc->drc_ds->ds_dir->dd_pool->dp_meta_objset,
3329eda14cbcSMatt Macy 		    drc->drc_ds->ds_object, DS_FIELD_RESUME_BYTES,
3330eda14cbcSMatt Macy 		    sizeof (bytes), 1, &bytes);
3331eda14cbcSMatt Macy 		drc->drc_bytes_read += bytes;
3332eda14cbcSMatt Macy 	}
3333eda14cbcSMatt Macy 
3334eda14cbcSMatt Macy 	drc->drc_ignore_objlist = objlist_create();
3335eda14cbcSMatt Macy 
3336eda14cbcSMatt Macy 	/* these were verified in dmu_recv_begin */
3337eda14cbcSMatt Macy 	ASSERT3U(DMU_GET_STREAM_HDRTYPE(drc->drc_drrb->drr_versioninfo), ==,
3338eda14cbcSMatt Macy 	    DMU_SUBSTREAM);
3339eda14cbcSMatt Macy 	ASSERT3U(drc->drc_drrb->drr_type, <, DMU_OST_NUMTYPES);
3340eda14cbcSMatt Macy 
3341eda14cbcSMatt Macy 	ASSERT(dsl_dataset_phys(drc->drc_ds)->ds_flags & DS_FLAG_INCONSISTENT);
3342eda14cbcSMatt Macy 	ASSERT0(drc->drc_os->os_encrypted &&
3343eda14cbcSMatt Macy 	    (drc->drc_featureflags & DMU_BACKUP_FEATURE_EMBED_DATA));
3344eda14cbcSMatt Macy 
3345eda14cbcSMatt Macy 	/* handle DSL encryption key payload */
3346eda14cbcSMatt Macy 	if (drc->drc_featureflags & DMU_BACKUP_FEATURE_RAW) {
3347eda14cbcSMatt Macy 		nvlist_t *keynvl = NULL;
3348eda14cbcSMatt Macy 
3349eda14cbcSMatt Macy 		ASSERT(drc->drc_os->os_encrypted);
3350eda14cbcSMatt Macy 		ASSERT(drc->drc_raw);
3351eda14cbcSMatt Macy 
3352eda14cbcSMatt Macy 		err = nvlist_lookup_nvlist(drc->drc_begin_nvl, "crypt_keydata",
3353eda14cbcSMatt Macy 		    &keynvl);
3354eda14cbcSMatt Macy 		if (err != 0)
3355eda14cbcSMatt Macy 			goto out;
3356eda14cbcSMatt Macy 
3357271171e0SMartin Matuska 		if (!drc->drc_heal) {
3358eda14cbcSMatt Macy 			/*
3359eda14cbcSMatt Macy 			 * If this is a new dataset we set the key immediately.
3360eda14cbcSMatt Macy 			 * Otherwise we don't want to change the key until we
3361271171e0SMartin Matuska 			 * are sure the rest of the receive succeeded so we
3362271171e0SMartin Matuska 			 * stash the keynvl away until then.
3363eda14cbcSMatt Macy 			 */
3364eda14cbcSMatt Macy 			err = dsl_crypto_recv_raw(spa_name(drc->drc_os->os_spa),
3365eda14cbcSMatt Macy 			    drc->drc_ds->ds_object, drc->drc_fromsnapobj,
3366eda14cbcSMatt Macy 			    drc->drc_drrb->drr_type, keynvl, drc->drc_newfs);
3367eda14cbcSMatt Macy 			if (err != 0)
3368eda14cbcSMatt Macy 				goto out;
3369271171e0SMartin Matuska 		}
3370eda14cbcSMatt Macy 
3371eda14cbcSMatt Macy 		/* see comment in dmu_recv_end_sync() */
3372eda14cbcSMatt Macy 		drc->drc_ivset_guid = 0;
3373eda14cbcSMatt Macy 		(void) nvlist_lookup_uint64(keynvl, "to_ivset_guid",
3374eda14cbcSMatt Macy 		    &drc->drc_ivset_guid);
3375eda14cbcSMatt Macy 
3376eda14cbcSMatt Macy 		if (!drc->drc_newfs)
3377eda14cbcSMatt Macy 			drc->drc_keynvl = fnvlist_dup(keynvl);
3378eda14cbcSMatt Macy 	}
3379eda14cbcSMatt Macy 
3380eda14cbcSMatt Macy 	if (drc->drc_featureflags & DMU_BACKUP_FEATURE_RESUMING) {
3381eda14cbcSMatt Macy 		err = resume_check(drc, drc->drc_begin_nvl);
3382eda14cbcSMatt Macy 		if (err != 0)
3383eda14cbcSMatt Macy 			goto out;
3384eda14cbcSMatt Macy 	}
3385eda14cbcSMatt Macy 
3386eda14cbcSMatt Macy 	/*
33872a58b312SMartin Matuska 	 * For compatibility with recursive send streams, we do this here,
33882a58b312SMartin Matuska 	 * rather than in dmu_recv_begin. If we pull the next header too
33892a58b312SMartin Matuska 	 * early, and it's the END record, we break the `recv_skip` logic.
33902a58b312SMartin Matuska 	 */
33912a58b312SMartin Matuska 	if (drc->drc_drr_begin->drr_payloadlen == 0) {
33922a58b312SMartin Matuska 		err = receive_read_payload_and_next_header(drc, 0, NULL);
33932a58b312SMartin Matuska 		if (err != 0)
33942a58b312SMartin Matuska 			goto out;
33952a58b312SMartin Matuska 	}
33962a58b312SMartin Matuska 
33972a58b312SMartin Matuska 	/*
3398eda14cbcSMatt Macy 	 * If we failed before this point we will clean up any new resume
3399eda14cbcSMatt Macy 	 * state that was created. Now that we've gotten past the initial
3400eda14cbcSMatt Macy 	 * checks we are ok to retain that resume state.
3401eda14cbcSMatt Macy 	 */
3402eda14cbcSMatt Macy 	drc->drc_should_save = B_TRUE;
3403eda14cbcSMatt Macy 
3404eda14cbcSMatt Macy 	(void) bqueue_init(&rwa->q, zfs_recv_queue_ff,
3405eda14cbcSMatt Macy 	    MAX(zfs_recv_queue_length, 2 * zfs_max_recordsize),
3406eda14cbcSMatt Macy 	    offsetof(struct receive_record_arg, node));
3407eda14cbcSMatt Macy 	cv_init(&rwa->cv, NULL, CV_DEFAULT, NULL);
3408eda14cbcSMatt Macy 	mutex_init(&rwa->mutex, NULL, MUTEX_DEFAULT, NULL);
3409eda14cbcSMatt Macy 	rwa->os = drc->drc_os;
3410eda14cbcSMatt Macy 	rwa->byteswap = drc->drc_byteswap;
3411271171e0SMartin Matuska 	rwa->heal = drc->drc_heal;
3412271171e0SMartin Matuska 	rwa->tofs = drc->drc_tofs;
3413eda14cbcSMatt Macy 	rwa->resumable = drc->drc_resumable;
3414eda14cbcSMatt Macy 	rwa->raw = drc->drc_raw;
3415eda14cbcSMatt Macy 	rwa->spill = drc->drc_spill;
3416eda14cbcSMatt Macy 	rwa->full = (drc->drc_drr_begin->drr_u.drr_begin.drr_fromguid == 0);
3417eda14cbcSMatt Macy 	rwa->os->os_raw_receive = drc->drc_raw;
3418271171e0SMartin Matuska 	if (drc->drc_heal) {
3419271171e0SMartin Matuska 		rwa->heal_pio = zio_root(drc->drc_os->os_spa, NULL, NULL,
3420271171e0SMartin Matuska 		    ZIO_FLAG_GODFATHER);
3421271171e0SMartin Matuska 	}
3422eda14cbcSMatt Macy 	list_create(&rwa->write_batch, sizeof (struct receive_record_arg),
3423eda14cbcSMatt Macy 	    offsetof(struct receive_record_arg, node.bqn_node));
3424eda14cbcSMatt Macy 
3425eda14cbcSMatt Macy 	(void) thread_create(NULL, 0, receive_writer_thread, rwa, 0, curproc,
3426eda14cbcSMatt Macy 	    TS_RUN, minclsyspri);
3427eda14cbcSMatt Macy 	/*
3428eda14cbcSMatt Macy 	 * We're reading rwa->err without locks, which is safe since we are the
3429eda14cbcSMatt Macy 	 * only reader, and the worker thread is the only writer.  It's ok if we
3430eda14cbcSMatt Macy 	 * miss a write for an iteration or two of the loop, since the writer
3431eda14cbcSMatt Macy 	 * thread will keep freeing records we send it until we send it an eos
3432eda14cbcSMatt Macy 	 * marker.
3433eda14cbcSMatt Macy 	 *
3434eda14cbcSMatt Macy 	 * We can leave this loop in 3 ways:  First, if rwa->err is
3435eda14cbcSMatt Macy 	 * non-zero.  In that case, the writer thread will free the rrd we just
3436eda14cbcSMatt Macy 	 * pushed.  Second, if  we're interrupted; in that case, either it's the
3437eda14cbcSMatt Macy 	 * first loop and drc->drc_rrd was never allocated, or it's later, and
3438eda14cbcSMatt Macy 	 * drc->drc_rrd has been handed off to the writer thread who will free
3439eda14cbcSMatt Macy 	 * it.  Finally, if receive_read_record fails or we're at the end of the
3440eda14cbcSMatt Macy 	 * stream, then we free drc->drc_rrd and exit.
3441eda14cbcSMatt Macy 	 */
3442eda14cbcSMatt Macy 	while (rwa->err == 0) {
3443aca928a5SMartin Matuska 		if (issig()) {
3444eda14cbcSMatt Macy 			err = SET_ERROR(EINTR);
3445eda14cbcSMatt Macy 			break;
3446eda14cbcSMatt Macy 		}
3447eda14cbcSMatt Macy 
3448eda14cbcSMatt Macy 		ASSERT3P(drc->drc_rrd, ==, NULL);
3449eda14cbcSMatt Macy 		drc->drc_rrd = drc->drc_next_rrd;
3450eda14cbcSMatt Macy 		drc->drc_next_rrd = NULL;
3451eda14cbcSMatt Macy 		/* Allocates and loads header into drc->drc_next_rrd */
3452eda14cbcSMatt Macy 		err = receive_read_record(drc);
3453eda14cbcSMatt Macy 
3454eda14cbcSMatt Macy 		if (drc->drc_rrd->header.drr_type == DRR_END || err != 0) {
3455eda14cbcSMatt Macy 			kmem_free(drc->drc_rrd, sizeof (*drc->drc_rrd));
3456eda14cbcSMatt Macy 			drc->drc_rrd = NULL;
3457eda14cbcSMatt Macy 			break;
3458eda14cbcSMatt Macy 		}
3459eda14cbcSMatt Macy 
3460eda14cbcSMatt Macy 		bqueue_enqueue(&rwa->q, drc->drc_rrd,
3461eda14cbcSMatt Macy 		    sizeof (struct receive_record_arg) +
3462eda14cbcSMatt Macy 		    drc->drc_rrd->payload_size);
3463eda14cbcSMatt Macy 		drc->drc_rrd = NULL;
3464eda14cbcSMatt Macy 	}
3465eda14cbcSMatt Macy 
3466eda14cbcSMatt Macy 	ASSERT3P(drc->drc_rrd, ==, NULL);
3467eda14cbcSMatt Macy 	drc->drc_rrd = kmem_zalloc(sizeof (*drc->drc_rrd), KM_SLEEP);
3468eda14cbcSMatt Macy 	drc->drc_rrd->eos_marker = B_TRUE;
3469eda14cbcSMatt Macy 	bqueue_enqueue_flush(&rwa->q, drc->drc_rrd, 1);
3470eda14cbcSMatt Macy 
3471eda14cbcSMatt Macy 	mutex_enter(&rwa->mutex);
3472eda14cbcSMatt Macy 	while (!rwa->done) {
3473eda14cbcSMatt Macy 		/*
3474eda14cbcSMatt Macy 		 * We need to use cv_wait_sig() so that any process that may
3475eda14cbcSMatt Macy 		 * be sleeping here can still fork.
3476eda14cbcSMatt Macy 		 */
3477eda14cbcSMatt Macy 		(void) cv_wait_sig(&rwa->cv, &rwa->mutex);
3478eda14cbcSMatt Macy 	}
3479eda14cbcSMatt Macy 	mutex_exit(&rwa->mutex);
3480eda14cbcSMatt Macy 
3481eda14cbcSMatt Macy 	/*
3482eda14cbcSMatt Macy 	 * If we are receiving a full stream as a clone, all object IDs which
3483eda14cbcSMatt Macy 	 * are greater than the maximum ID referenced in the stream are
3484eda14cbcSMatt Macy 	 * by definition unused and must be freed.
3485eda14cbcSMatt Macy 	 */
3486eda14cbcSMatt Macy 	if (drc->drc_clone && drc->drc_drrb->drr_fromguid == 0) {
3487eda14cbcSMatt Macy 		uint64_t obj = rwa->max_object + 1;
3488eda14cbcSMatt Macy 		int free_err = 0;
3489eda14cbcSMatt Macy 		int next_err = 0;
3490eda14cbcSMatt Macy 
3491eda14cbcSMatt Macy 		while (next_err == 0) {
3492eda14cbcSMatt Macy 			free_err = dmu_free_long_object(rwa->os, obj);
3493eda14cbcSMatt Macy 			if (free_err != 0 && free_err != ENOENT)
3494eda14cbcSMatt Macy 				break;
3495eda14cbcSMatt Macy 
3496eda14cbcSMatt Macy 			next_err = dmu_object_next(rwa->os, &obj, FALSE, 0);
3497eda14cbcSMatt Macy 		}
3498eda14cbcSMatt Macy 
3499eda14cbcSMatt Macy 		if (err == 0) {
3500eda14cbcSMatt Macy 			if (free_err != 0 && free_err != ENOENT)
3501eda14cbcSMatt Macy 				err = free_err;
3502eda14cbcSMatt Macy 			else if (next_err != ESRCH)
3503eda14cbcSMatt Macy 				err = next_err;
3504eda14cbcSMatt Macy 		}
3505eda14cbcSMatt Macy 	}
3506eda14cbcSMatt Macy 
3507eda14cbcSMatt Macy 	cv_destroy(&rwa->cv);
3508eda14cbcSMatt Macy 	mutex_destroy(&rwa->mutex);
3509eda14cbcSMatt Macy 	bqueue_destroy(&rwa->q);
3510eda14cbcSMatt Macy 	list_destroy(&rwa->write_batch);
3511eda14cbcSMatt Macy 	if (err == 0)
3512eda14cbcSMatt Macy 		err = rwa->err;
3513eda14cbcSMatt Macy 
3514eda14cbcSMatt Macy out:
3515eda14cbcSMatt Macy 	/*
3516eda14cbcSMatt Macy 	 * If we hit an error before we started the receive_writer_thread
3517eda14cbcSMatt Macy 	 * we need to clean up the next_rrd we create by processing the
3518eda14cbcSMatt Macy 	 * DRR_BEGIN record.
3519eda14cbcSMatt Macy 	 */
3520eda14cbcSMatt Macy 	if (drc->drc_next_rrd != NULL)
3521eda14cbcSMatt Macy 		kmem_free(drc->drc_next_rrd, sizeof (*drc->drc_next_rrd));
3522eda14cbcSMatt Macy 
3523eda14cbcSMatt Macy 	/*
3524eda14cbcSMatt Macy 	 * The objset will be invalidated by dmu_recv_end() when we do
3525eda14cbcSMatt Macy 	 * dsl_dataset_clone_swap_sync_impl().
3526eda14cbcSMatt Macy 	 */
3527eda14cbcSMatt Macy 	drc->drc_os = NULL;
3528eda14cbcSMatt Macy 
3529eda14cbcSMatt Macy 	kmem_free(rwa, sizeof (*rwa));
3530eda14cbcSMatt Macy 	nvlist_free(drc->drc_begin_nvl);
3531eda14cbcSMatt Macy 
3532eda14cbcSMatt Macy 	if (err != 0) {
3533eda14cbcSMatt Macy 		/*
3534eda14cbcSMatt Macy 		 * Clean up references. If receive is not resumable,
3535eda14cbcSMatt Macy 		 * destroy what we created, so we don't leave it in
3536eda14cbcSMatt Macy 		 * the inconsistent state.
3537eda14cbcSMatt Macy 		 */
3538eda14cbcSMatt Macy 		dmu_recv_cleanup_ds(drc);
3539eda14cbcSMatt Macy 		nvlist_free(drc->drc_keynvl);
3540*071ab5a1SMartin Matuska 		crfree(drc->drc_cred);
3541*071ab5a1SMartin Matuska 		drc->drc_cred = NULL;
3542eda14cbcSMatt Macy 	}
3543eda14cbcSMatt Macy 
3544eda14cbcSMatt Macy 	objlist_destroy(drc->drc_ignore_objlist);
3545eda14cbcSMatt Macy 	drc->drc_ignore_objlist = NULL;
3546eda14cbcSMatt Macy 	*voffp = drc->drc_voff;
3547eda14cbcSMatt Macy 	return (err);
3548eda14cbcSMatt Macy }
3549eda14cbcSMatt Macy 
3550eda14cbcSMatt Macy static int
dmu_recv_end_check(void * arg,dmu_tx_t * tx)3551eda14cbcSMatt Macy dmu_recv_end_check(void *arg, dmu_tx_t *tx)
3552eda14cbcSMatt Macy {
3553eda14cbcSMatt Macy 	dmu_recv_cookie_t *drc = arg;
3554eda14cbcSMatt Macy 	dsl_pool_t *dp = dmu_tx_pool(tx);
3555eda14cbcSMatt Macy 	int error;
3556eda14cbcSMatt Macy 
3557eda14cbcSMatt Macy 	ASSERT3P(drc->drc_ds->ds_owner, ==, dmu_recv_tag);
3558eda14cbcSMatt Macy 
3559271171e0SMartin Matuska 	if (drc->drc_heal) {
3560271171e0SMartin Matuska 		error = 0;
3561271171e0SMartin Matuska 	} else if (!drc->drc_newfs) {
3562eda14cbcSMatt Macy 		dsl_dataset_t *origin_head;
3563eda14cbcSMatt Macy 
3564eda14cbcSMatt Macy 		error = dsl_dataset_hold(dp, drc->drc_tofs, FTAG, &origin_head);
3565eda14cbcSMatt Macy 		if (error != 0)
3566eda14cbcSMatt Macy 			return (error);
3567eda14cbcSMatt Macy 		if (drc->drc_force) {
3568eda14cbcSMatt Macy 			/*
3569eda14cbcSMatt Macy 			 * We will destroy any snapshots in tofs (i.e. before
3570eda14cbcSMatt Macy 			 * origin_head) that are after the origin (which is
3571eda14cbcSMatt Macy 			 * the snap before drc_ds, because drc_ds can not
3572eda14cbcSMatt Macy 			 * have any snaps of its own).
3573eda14cbcSMatt Macy 			 */
3574eda14cbcSMatt Macy 			uint64_t obj;
3575eda14cbcSMatt Macy 
3576eda14cbcSMatt Macy 			obj = dsl_dataset_phys(origin_head)->ds_prev_snap_obj;
3577eda14cbcSMatt Macy 			while (obj !=
3578eda14cbcSMatt Macy 			    dsl_dataset_phys(drc->drc_ds)->ds_prev_snap_obj) {
3579eda14cbcSMatt Macy 				dsl_dataset_t *snap;
3580eda14cbcSMatt Macy 				error = dsl_dataset_hold_obj(dp, obj, FTAG,
3581eda14cbcSMatt Macy 				    &snap);
3582eda14cbcSMatt Macy 				if (error != 0)
3583eda14cbcSMatt Macy 					break;
3584eda14cbcSMatt Macy 				if (snap->ds_dir != origin_head->ds_dir)
3585eda14cbcSMatt Macy 					error = SET_ERROR(EINVAL);
3586eda14cbcSMatt Macy 				if (error == 0)  {
3587eda14cbcSMatt Macy 					error = dsl_destroy_snapshot_check_impl(
3588eda14cbcSMatt Macy 					    snap, B_FALSE);
3589eda14cbcSMatt Macy 				}
3590eda14cbcSMatt Macy 				obj = dsl_dataset_phys(snap)->ds_prev_snap_obj;
3591eda14cbcSMatt Macy 				dsl_dataset_rele(snap, FTAG);
3592eda14cbcSMatt Macy 				if (error != 0)
3593eda14cbcSMatt Macy 					break;
3594eda14cbcSMatt Macy 			}
3595eda14cbcSMatt Macy 			if (error != 0) {
3596eda14cbcSMatt Macy 				dsl_dataset_rele(origin_head, FTAG);
3597eda14cbcSMatt Macy 				return (error);
3598eda14cbcSMatt Macy 			}
3599eda14cbcSMatt Macy 		}
3600eda14cbcSMatt Macy 		if (drc->drc_keynvl != NULL) {
3601eda14cbcSMatt Macy 			error = dsl_crypto_recv_raw_key_check(drc->drc_ds,
3602eda14cbcSMatt Macy 			    drc->drc_keynvl, tx);
3603eda14cbcSMatt Macy 			if (error != 0) {
3604eda14cbcSMatt Macy 				dsl_dataset_rele(origin_head, FTAG);
3605eda14cbcSMatt Macy 				return (error);
3606eda14cbcSMatt Macy 			}
3607eda14cbcSMatt Macy 		}
3608eda14cbcSMatt Macy 
3609eda14cbcSMatt Macy 		error = dsl_dataset_clone_swap_check_impl(drc->drc_ds,
3610eda14cbcSMatt Macy 		    origin_head, drc->drc_force, drc->drc_owner, tx);
3611eda14cbcSMatt Macy 		if (error != 0) {
3612eda14cbcSMatt Macy 			dsl_dataset_rele(origin_head, FTAG);
3613eda14cbcSMatt Macy 			return (error);
3614eda14cbcSMatt Macy 		}
3615eda14cbcSMatt Macy 		error = dsl_dataset_snapshot_check_impl(origin_head,
3616*071ab5a1SMartin Matuska 		    drc->drc_tosnap, tx, B_TRUE, 1, drc->drc_cred);
3617eda14cbcSMatt Macy 		dsl_dataset_rele(origin_head, FTAG);
3618eda14cbcSMatt Macy 		if (error != 0)
3619eda14cbcSMatt Macy 			return (error);
3620eda14cbcSMatt Macy 
3621eda14cbcSMatt Macy 		error = dsl_destroy_head_check_impl(drc->drc_ds, 1);
3622eda14cbcSMatt Macy 	} else {
3623eda14cbcSMatt Macy 		error = dsl_dataset_snapshot_check_impl(drc->drc_ds,
3624*071ab5a1SMartin Matuska 		    drc->drc_tosnap, tx, B_TRUE, 1, drc->drc_cred);
3625eda14cbcSMatt Macy 	}
3626eda14cbcSMatt Macy 	return (error);
3627eda14cbcSMatt Macy }
3628eda14cbcSMatt Macy 
3629eda14cbcSMatt Macy static void
dmu_recv_end_sync(void * arg,dmu_tx_t * tx)3630eda14cbcSMatt Macy dmu_recv_end_sync(void *arg, dmu_tx_t *tx)
3631eda14cbcSMatt Macy {
3632eda14cbcSMatt Macy 	dmu_recv_cookie_t *drc = arg;
3633eda14cbcSMatt Macy 	dsl_pool_t *dp = dmu_tx_pool(tx);
3634eda14cbcSMatt Macy 	boolean_t encrypted = drc->drc_ds->ds_dir->dd_crypto_obj != 0;
3635271171e0SMartin Matuska 	uint64_t newsnapobj = 0;
3636eda14cbcSMatt Macy 
3637eda14cbcSMatt Macy 	spa_history_log_internal_ds(drc->drc_ds, "finish receiving",
3638eda14cbcSMatt Macy 	    tx, "snap=%s", drc->drc_tosnap);
3639eda14cbcSMatt Macy 	drc->drc_ds->ds_objset->os_raw_receive = B_FALSE;
3640eda14cbcSMatt Macy 
3641271171e0SMartin Matuska 	if (drc->drc_heal) {
3642271171e0SMartin Matuska 		if (drc->drc_keynvl != NULL) {
3643271171e0SMartin Matuska 			nvlist_free(drc->drc_keynvl);
3644271171e0SMartin Matuska 			drc->drc_keynvl = NULL;
3645271171e0SMartin Matuska 		}
3646271171e0SMartin Matuska 	} else if (!drc->drc_newfs) {
3647eda14cbcSMatt Macy 		dsl_dataset_t *origin_head;
3648eda14cbcSMatt Macy 
3649eda14cbcSMatt Macy 		VERIFY0(dsl_dataset_hold(dp, drc->drc_tofs, FTAG,
3650eda14cbcSMatt Macy 		    &origin_head));
3651eda14cbcSMatt Macy 
3652eda14cbcSMatt Macy 		if (drc->drc_force) {
3653eda14cbcSMatt Macy 			/*
3654eda14cbcSMatt Macy 			 * Destroy any snapshots of drc_tofs (origin_head)
3655eda14cbcSMatt Macy 			 * after the origin (the snap before drc_ds).
3656eda14cbcSMatt Macy 			 */
3657eda14cbcSMatt Macy 			uint64_t obj;
3658eda14cbcSMatt Macy 
3659eda14cbcSMatt Macy 			obj = dsl_dataset_phys(origin_head)->ds_prev_snap_obj;
3660eda14cbcSMatt Macy 			while (obj !=
3661eda14cbcSMatt Macy 			    dsl_dataset_phys(drc->drc_ds)->ds_prev_snap_obj) {
3662eda14cbcSMatt Macy 				dsl_dataset_t *snap;
3663eda14cbcSMatt Macy 				VERIFY0(dsl_dataset_hold_obj(dp, obj, FTAG,
3664eda14cbcSMatt Macy 				    &snap));
3665eda14cbcSMatt Macy 				ASSERT3P(snap->ds_dir, ==, origin_head->ds_dir);
3666eda14cbcSMatt Macy 				obj = dsl_dataset_phys(snap)->ds_prev_snap_obj;
3667eda14cbcSMatt Macy 				dsl_destroy_snapshot_sync_impl(snap,
3668eda14cbcSMatt Macy 				    B_FALSE, tx);
3669eda14cbcSMatt Macy 				dsl_dataset_rele(snap, FTAG);
3670eda14cbcSMatt Macy 			}
3671eda14cbcSMatt Macy 		}
3672eda14cbcSMatt Macy 		if (drc->drc_keynvl != NULL) {
3673eda14cbcSMatt Macy 			dsl_crypto_recv_raw_key_sync(drc->drc_ds,
3674eda14cbcSMatt Macy 			    drc->drc_keynvl, tx);
3675eda14cbcSMatt Macy 			nvlist_free(drc->drc_keynvl);
3676eda14cbcSMatt Macy 			drc->drc_keynvl = NULL;
3677eda14cbcSMatt Macy 		}
3678eda14cbcSMatt Macy 
3679eda14cbcSMatt Macy 		VERIFY3P(drc->drc_ds->ds_prev, ==,
3680eda14cbcSMatt Macy 		    origin_head->ds_prev);
3681eda14cbcSMatt Macy 
3682eda14cbcSMatt Macy 		dsl_dataset_clone_swap_sync_impl(drc->drc_ds,
3683eda14cbcSMatt Macy 		    origin_head, tx);
3684eda14cbcSMatt Macy 		/*
3685eda14cbcSMatt Macy 		 * The objset was evicted by dsl_dataset_clone_swap_sync_impl,
3686eda14cbcSMatt Macy 		 * so drc_os is no longer valid.
3687eda14cbcSMatt Macy 		 */
3688eda14cbcSMatt Macy 		drc->drc_os = NULL;
3689eda14cbcSMatt Macy 
3690eda14cbcSMatt Macy 		dsl_dataset_snapshot_sync_impl(origin_head,
3691eda14cbcSMatt Macy 		    drc->drc_tosnap, tx);
3692eda14cbcSMatt Macy 
3693eda14cbcSMatt Macy 		/* set snapshot's creation time and guid */
3694eda14cbcSMatt Macy 		dmu_buf_will_dirty(origin_head->ds_prev->ds_dbuf, tx);
3695eda14cbcSMatt Macy 		dsl_dataset_phys(origin_head->ds_prev)->ds_creation_time =
3696eda14cbcSMatt Macy 		    drc->drc_drrb->drr_creation_time;
3697eda14cbcSMatt Macy 		dsl_dataset_phys(origin_head->ds_prev)->ds_guid =
3698eda14cbcSMatt Macy 		    drc->drc_drrb->drr_toguid;
3699eda14cbcSMatt Macy 		dsl_dataset_phys(origin_head->ds_prev)->ds_flags &=
3700eda14cbcSMatt Macy 		    ~DS_FLAG_INCONSISTENT;
3701eda14cbcSMatt Macy 
3702eda14cbcSMatt Macy 		dmu_buf_will_dirty(origin_head->ds_dbuf, tx);
3703eda14cbcSMatt Macy 		dsl_dataset_phys(origin_head)->ds_flags &=
3704eda14cbcSMatt Macy 		    ~DS_FLAG_INCONSISTENT;
3705eda14cbcSMatt Macy 
3706eda14cbcSMatt Macy 		newsnapobj =
3707eda14cbcSMatt Macy 		    dsl_dataset_phys(origin_head)->ds_prev_snap_obj;
3708eda14cbcSMatt Macy 
3709eda14cbcSMatt Macy 		dsl_dataset_rele(origin_head, FTAG);
3710eda14cbcSMatt Macy 		dsl_destroy_head_sync_impl(drc->drc_ds, tx);
3711eda14cbcSMatt Macy 
3712eda14cbcSMatt Macy 		if (drc->drc_owner != NULL)
3713eda14cbcSMatt Macy 			VERIFY3P(origin_head->ds_owner, ==, drc->drc_owner);
3714eda14cbcSMatt Macy 	} else {
3715eda14cbcSMatt Macy 		dsl_dataset_t *ds = drc->drc_ds;
3716eda14cbcSMatt Macy 
3717eda14cbcSMatt Macy 		dsl_dataset_snapshot_sync_impl(ds, drc->drc_tosnap, tx);
3718eda14cbcSMatt Macy 
3719eda14cbcSMatt Macy 		/* set snapshot's creation time and guid */
3720eda14cbcSMatt Macy 		dmu_buf_will_dirty(ds->ds_prev->ds_dbuf, tx);
3721eda14cbcSMatt Macy 		dsl_dataset_phys(ds->ds_prev)->ds_creation_time =
3722eda14cbcSMatt Macy 		    drc->drc_drrb->drr_creation_time;
3723eda14cbcSMatt Macy 		dsl_dataset_phys(ds->ds_prev)->ds_guid =
3724eda14cbcSMatt Macy 		    drc->drc_drrb->drr_toguid;
3725eda14cbcSMatt Macy 		dsl_dataset_phys(ds->ds_prev)->ds_flags &=
3726eda14cbcSMatt Macy 		    ~DS_FLAG_INCONSISTENT;
3727eda14cbcSMatt Macy 
3728eda14cbcSMatt Macy 		dmu_buf_will_dirty(ds->ds_dbuf, tx);
3729eda14cbcSMatt Macy 		dsl_dataset_phys(ds)->ds_flags &= ~DS_FLAG_INCONSISTENT;
3730eda14cbcSMatt Macy 		if (dsl_dataset_has_resume_receive_state(ds)) {
3731eda14cbcSMatt Macy 			(void) zap_remove(dp->dp_meta_objset, ds->ds_object,
3732eda14cbcSMatt Macy 			    DS_FIELD_RESUME_FROMGUID, tx);
3733eda14cbcSMatt Macy 			(void) zap_remove(dp->dp_meta_objset, ds->ds_object,
3734eda14cbcSMatt Macy 			    DS_FIELD_RESUME_OBJECT, tx);
3735eda14cbcSMatt Macy 			(void) zap_remove(dp->dp_meta_objset, ds->ds_object,
3736eda14cbcSMatt Macy 			    DS_FIELD_RESUME_OFFSET, tx);
3737eda14cbcSMatt Macy 			(void) zap_remove(dp->dp_meta_objset, ds->ds_object,
3738eda14cbcSMatt Macy 			    DS_FIELD_RESUME_BYTES, tx);
3739eda14cbcSMatt Macy 			(void) zap_remove(dp->dp_meta_objset, ds->ds_object,
3740eda14cbcSMatt Macy 			    DS_FIELD_RESUME_TOGUID, tx);
3741eda14cbcSMatt Macy 			(void) zap_remove(dp->dp_meta_objset, ds->ds_object,
3742eda14cbcSMatt Macy 			    DS_FIELD_RESUME_TONAME, tx);
3743eda14cbcSMatt Macy 			(void) zap_remove(dp->dp_meta_objset, ds->ds_object,
3744eda14cbcSMatt Macy 			    DS_FIELD_RESUME_REDACT_BOOKMARK_SNAPS, tx);
3745eda14cbcSMatt Macy 		}
3746eda14cbcSMatt Macy 		newsnapobj =
3747eda14cbcSMatt Macy 		    dsl_dataset_phys(drc->drc_ds)->ds_prev_snap_obj;
3748eda14cbcSMatt Macy 	}
3749eda14cbcSMatt Macy 
3750eda14cbcSMatt Macy 	/*
3751eda14cbcSMatt Macy 	 * If this is a raw receive, the crypt_keydata nvlist will include
3752eda14cbcSMatt Macy 	 * a to_ivset_guid for us to set on the new snapshot. This value
3753eda14cbcSMatt Macy 	 * will override the value generated by the snapshot code. However,
3754eda14cbcSMatt Macy 	 * this value may not be present, because older implementations of
3755eda14cbcSMatt Macy 	 * the raw send code did not include this value, and we are still
3756eda14cbcSMatt Macy 	 * allowed to receive them if the zfs_disable_ivset_guid_check
3757eda14cbcSMatt Macy 	 * tunable is set, in which case we will leave the newly-generated
3758eda14cbcSMatt Macy 	 * value.
3759eda14cbcSMatt Macy 	 */
3760271171e0SMartin Matuska 	if (!drc->drc_heal && drc->drc_raw && drc->drc_ivset_guid != 0) {
3761eda14cbcSMatt Macy 		dmu_object_zapify(dp->dp_meta_objset, newsnapobj,
3762eda14cbcSMatt Macy 		    DMU_OT_DSL_DATASET, tx);
3763eda14cbcSMatt Macy 		VERIFY0(zap_update(dp->dp_meta_objset, newsnapobj,
3764eda14cbcSMatt Macy 		    DS_FIELD_IVSET_GUID, sizeof (uint64_t), 1,
3765eda14cbcSMatt Macy 		    &drc->drc_ivset_guid, tx));
3766eda14cbcSMatt Macy 	}
3767eda14cbcSMatt Macy 
3768eda14cbcSMatt Macy 	/*
3769eda14cbcSMatt Macy 	 * Release the hold from dmu_recv_begin.  This must be done before
3770eda14cbcSMatt Macy 	 * we return to open context, so that when we free the dataset's dnode
3771eda14cbcSMatt Macy 	 * we can evict its bonus buffer. Since the dataset may be destroyed
3772eda14cbcSMatt Macy 	 * at this point (and therefore won't have a valid pointer to the spa)
3773eda14cbcSMatt Macy 	 * we release the key mapping manually here while we do have a valid
3774eda14cbcSMatt Macy 	 * pointer, if it exists.
3775eda14cbcSMatt Macy 	 */
3776eda14cbcSMatt Macy 	if (!drc->drc_raw && encrypted) {
3777eda14cbcSMatt Macy 		(void) spa_keystore_remove_mapping(dmu_tx_pool(tx)->dp_spa,
3778eda14cbcSMatt Macy 		    drc->drc_ds->ds_object, drc->drc_ds);
3779eda14cbcSMatt Macy 	}
3780eda14cbcSMatt Macy 	dsl_dataset_disown(drc->drc_ds, 0, dmu_recv_tag);
3781eda14cbcSMatt Macy 	drc->drc_ds = NULL;
3782eda14cbcSMatt Macy }
3783eda14cbcSMatt Macy 
3784eda14cbcSMatt Macy static int dmu_recv_end_modified_blocks = 3;
3785eda14cbcSMatt Macy 
3786eda14cbcSMatt Macy static int
dmu_recv_existing_end(dmu_recv_cookie_t * drc)3787eda14cbcSMatt Macy dmu_recv_existing_end(dmu_recv_cookie_t *drc)
3788eda14cbcSMatt Macy {
3789eda14cbcSMatt Macy #ifdef _KERNEL
3790eda14cbcSMatt Macy 	/*
3791eda14cbcSMatt Macy 	 * We will be destroying the ds; make sure its origin is unmounted if
3792eda14cbcSMatt Macy 	 * necessary.
3793eda14cbcSMatt Macy 	 */
3794eda14cbcSMatt Macy 	char name[ZFS_MAX_DATASET_NAME_LEN];
3795eda14cbcSMatt Macy 	dsl_dataset_name(drc->drc_ds, name);
3796eda14cbcSMatt Macy 	zfs_destroy_unmount_origin(name);
3797eda14cbcSMatt Macy #endif
3798eda14cbcSMatt Macy 
3799eda14cbcSMatt Macy 	return (dsl_sync_task(drc->drc_tofs,
3800eda14cbcSMatt Macy 	    dmu_recv_end_check, dmu_recv_end_sync, drc,
3801eda14cbcSMatt Macy 	    dmu_recv_end_modified_blocks, ZFS_SPACE_CHECK_NORMAL));
3802eda14cbcSMatt Macy }
3803eda14cbcSMatt Macy 
3804eda14cbcSMatt Macy static int
dmu_recv_new_end(dmu_recv_cookie_t * drc)3805eda14cbcSMatt Macy dmu_recv_new_end(dmu_recv_cookie_t *drc)
3806eda14cbcSMatt Macy {
3807eda14cbcSMatt Macy 	return (dsl_sync_task(drc->drc_tofs,
3808eda14cbcSMatt Macy 	    dmu_recv_end_check, dmu_recv_end_sync, drc,
3809eda14cbcSMatt Macy 	    dmu_recv_end_modified_blocks, ZFS_SPACE_CHECK_NORMAL));
3810eda14cbcSMatt Macy }
3811eda14cbcSMatt Macy 
3812eda14cbcSMatt Macy int
dmu_recv_end(dmu_recv_cookie_t * drc,void * owner)3813eda14cbcSMatt Macy dmu_recv_end(dmu_recv_cookie_t *drc, void *owner)
3814eda14cbcSMatt Macy {
3815eda14cbcSMatt Macy 	int error;
3816eda14cbcSMatt Macy 
3817eda14cbcSMatt Macy 	drc->drc_owner = owner;
3818eda14cbcSMatt Macy 
3819eda14cbcSMatt Macy 	if (drc->drc_newfs)
3820eda14cbcSMatt Macy 		error = dmu_recv_new_end(drc);
3821eda14cbcSMatt Macy 	else
3822eda14cbcSMatt Macy 		error = dmu_recv_existing_end(drc);
3823eda14cbcSMatt Macy 
3824eda14cbcSMatt Macy 	if (error != 0) {
3825eda14cbcSMatt Macy 		dmu_recv_cleanup_ds(drc);
3826eda14cbcSMatt Macy 		nvlist_free(drc->drc_keynvl);
3827271171e0SMartin Matuska 	} else if (!drc->drc_heal) {
3828eda14cbcSMatt Macy 		if (drc->drc_newfs) {
3829eda14cbcSMatt Macy 			zvol_create_minor(drc->drc_tofs);
3830eda14cbcSMatt Macy 		}
3831eda14cbcSMatt Macy 		char *snapname = kmem_asprintf("%s@%s",
3832eda14cbcSMatt Macy 		    drc->drc_tofs, drc->drc_tosnap);
3833eda14cbcSMatt Macy 		zvol_create_minor(snapname);
3834eda14cbcSMatt Macy 		kmem_strfree(snapname);
3835eda14cbcSMatt Macy 	}
3836*071ab5a1SMartin Matuska 
3837*071ab5a1SMartin Matuska 	crfree(drc->drc_cred);
3838*071ab5a1SMartin Matuska 	drc->drc_cred = NULL;
3839*071ab5a1SMartin Matuska 
3840eda14cbcSMatt Macy 	return (error);
3841eda14cbcSMatt Macy }
3842eda14cbcSMatt Macy 
3843eda14cbcSMatt Macy /*
3844eda14cbcSMatt Macy  * Return TRUE if this objset is currently being received into.
3845eda14cbcSMatt Macy  */
3846eda14cbcSMatt Macy boolean_t
dmu_objset_is_receiving(objset_t * os)3847eda14cbcSMatt Macy dmu_objset_is_receiving(objset_t *os)
3848eda14cbcSMatt Macy {
3849eda14cbcSMatt Macy 	return (os->os_dsl_dataset != NULL &&
3850eda14cbcSMatt Macy 	    os->os_dsl_dataset->ds_owner == dmu_recv_tag);
3851eda14cbcSMatt Macy }
3852eda14cbcSMatt Macy 
3853be181ee2SMartin Matuska ZFS_MODULE_PARAM(zfs_recv, zfs_recv_, queue_length, UINT, ZMOD_RW,
3854eda14cbcSMatt Macy 	"Maximum receive queue length");
3855eda14cbcSMatt Macy 
3856be181ee2SMartin Matuska ZFS_MODULE_PARAM(zfs_recv, zfs_recv_, queue_ff, UINT, ZMOD_RW,
3857eda14cbcSMatt Macy 	"Receive queue fill fraction");
3858eda14cbcSMatt Macy 
3859be181ee2SMartin Matuska ZFS_MODULE_PARAM(zfs_recv, zfs_recv_, write_batch_size, UINT, ZMOD_RW,
3860eda14cbcSMatt Macy 	"Maximum amount of writes to batch into one transaction");
3861271171e0SMartin Matuska 
3862271171e0SMartin Matuska ZFS_MODULE_PARAM(zfs_recv, zfs_recv_, best_effort_corrective, INT, ZMOD_RW,
3863271171e0SMartin Matuska 	"Ignore errors during corrective receive");
3864