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