15367Sahrens /*
25367Sahrens * CDDL HEADER START
35367Sahrens *
45367Sahrens * The contents of this file are subject to the terms of the
55367Sahrens * Common Development and Distribution License (the "License").
65367Sahrens * You may not use this file except in compliance with the License.
75367Sahrens *
85367Sahrens * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
95367Sahrens * or http://www.opensolaris.org/os/licensing.
105367Sahrens * See the License for the specific language governing permissions
115367Sahrens * and limitations under the License.
125367Sahrens *
135367Sahrens * When distributing Covered Code, include this CDDL HEADER in each
145367Sahrens * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
155367Sahrens * If applicable, add the following below this CDDL HEADER, with the
165367Sahrens * fields enclosed by brackets "[]" replaced with your own identifying
175367Sahrens * information: Portions Copyright [yyyy] [name of copyright owner]
185367Sahrens *
195367Sahrens * CDDL HEADER END
205367Sahrens */
215367Sahrens
225367Sahrens /*
2312070SMark.Shellenbaum@Sun.COM * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
245367Sahrens */
255367Sahrens
265367Sahrens #include <assert.h>
275367Sahrens #include <ctype.h>
285367Sahrens #include <errno.h>
295367Sahrens #include <libintl.h>
305367Sahrens #include <stdio.h>
315367Sahrens #include <stdlib.h>
325367Sahrens #include <strings.h>
335367Sahrens #include <unistd.h>
345367Sahrens #include <stddef.h>
355367Sahrens #include <fcntl.h>
365367Sahrens #include <sys/mount.h>
3711007SLori.Alt@Sun.COM #include <pthread.h>
3811007SLori.Alt@Sun.COM #include <umem.h>
395367Sahrens
405367Sahrens #include <libzfs.h>
415367Sahrens
425367Sahrens #include "zfs_namecheck.h"
435367Sahrens #include "zfs_prop.h"
4410602SLori.Alt@Sun.COM #include "zfs_fletcher.h"
455367Sahrens #include "libzfs_impl.h"
4611007SLori.Alt@Sun.COM #include <sha2.h>
4711381SLori.Alt@Sun.COM #include <sys/zio_checksum.h>
4811381SLori.Alt@Sun.COM #include <sys/ddt.h>
495367Sahrens
5011022STom.Erickson@Sun.COM /* in libzfs_dataset.c */
5111022STom.Erickson@Sun.COM extern void zfs_setprop_error(libzfs_handle_t *, zfs_prop_t, int, char *);
5211022STom.Erickson@Sun.COM
537366STim.Haley@Sun.COM static int zfs_receive_impl(libzfs_handle_t *, const char *, recvflags_t,
5412527SChris.Kirby@oracle.com int, const char *, nvlist_t *, avl_tree_t *, char **, int, uint64_t *);
557366STim.Haley@Sun.COM
5611007SLori.Alt@Sun.COM static const zio_cksum_t zero_cksum = { 0 };
5711007SLori.Alt@Sun.COM
5811007SLori.Alt@Sun.COM typedef struct dedup_arg {
5911007SLori.Alt@Sun.COM int inputfd;
6011007SLori.Alt@Sun.COM int outputfd;
6111007SLori.Alt@Sun.COM libzfs_handle_t *dedup_hdl;
6211007SLori.Alt@Sun.COM } dedup_arg_t;
6311007SLori.Alt@Sun.COM
6411007SLori.Alt@Sun.COM typedef struct dataref {
6511007SLori.Alt@Sun.COM uint64_t ref_guid;
6611007SLori.Alt@Sun.COM uint64_t ref_object;
6711007SLori.Alt@Sun.COM uint64_t ref_offset;
6811007SLori.Alt@Sun.COM } dataref_t;
6911007SLori.Alt@Sun.COM
7011007SLori.Alt@Sun.COM typedef struct dedup_entry {
7111007SLori.Alt@Sun.COM struct dedup_entry *dde_next;
7211007SLori.Alt@Sun.COM zio_cksum_t dde_chksum;
7311381SLori.Alt@Sun.COM uint64_t dde_prop;
7411007SLori.Alt@Sun.COM dataref_t dde_ref;
7511007SLori.Alt@Sun.COM } dedup_entry_t;
7611007SLori.Alt@Sun.COM
7711007SLori.Alt@Sun.COM #define MAX_DDT_PHYSMEM_PERCENT 20
7811007SLori.Alt@Sun.COM #define SMALLEST_POSSIBLE_MAX_DDT_MB 128
7911007SLori.Alt@Sun.COM
8011007SLori.Alt@Sun.COM typedef struct dedup_table {
8111007SLori.Alt@Sun.COM dedup_entry_t **dedup_hash_array;
8211007SLori.Alt@Sun.COM umem_cache_t *ddecache;
8311007SLori.Alt@Sun.COM uint64_t max_ddt_size; /* max dedup table size in bytes */
8411007SLori.Alt@Sun.COM uint64_t cur_ddt_size; /* current dedup table size in bytes */
8511007SLori.Alt@Sun.COM uint64_t ddt_count;
8611007SLori.Alt@Sun.COM int numhashbits;
8711007SLori.Alt@Sun.COM boolean_t ddt_full;
8811007SLori.Alt@Sun.COM } dedup_table_t;
8911007SLori.Alt@Sun.COM
9011007SLori.Alt@Sun.COM static int
high_order_bit(uint64_t n)9111007SLori.Alt@Sun.COM high_order_bit(uint64_t n)
9211007SLori.Alt@Sun.COM {
9311007SLori.Alt@Sun.COM int count;
9411007SLori.Alt@Sun.COM
9511007SLori.Alt@Sun.COM for (count = 0; n != 0; count++)
9611007SLori.Alt@Sun.COM n >>= 1;
9711007SLori.Alt@Sun.COM return (count);
9811007SLori.Alt@Sun.COM }
9911007SLori.Alt@Sun.COM
10011007SLori.Alt@Sun.COM static size_t
ssread(void * buf,size_t len,FILE * stream)10111007SLori.Alt@Sun.COM ssread(void *buf, size_t len, FILE *stream)
10211007SLori.Alt@Sun.COM {
10311007SLori.Alt@Sun.COM size_t outlen;
10411007SLori.Alt@Sun.COM
10511007SLori.Alt@Sun.COM if ((outlen = fread(buf, len, 1, stream)) == 0)
10611007SLori.Alt@Sun.COM return (0);
10711007SLori.Alt@Sun.COM
10811007SLori.Alt@Sun.COM return (outlen);
10911007SLori.Alt@Sun.COM }
11011007SLori.Alt@Sun.COM
11111007SLori.Alt@Sun.COM static void
ddt_hash_append(libzfs_handle_t * hdl,dedup_table_t * ddt,dedup_entry_t ** ddepp,zio_cksum_t * cs,uint64_t prop,dataref_t * dr)11211007SLori.Alt@Sun.COM ddt_hash_append(libzfs_handle_t *hdl, dedup_table_t *ddt, dedup_entry_t **ddepp,
11311381SLori.Alt@Sun.COM zio_cksum_t *cs, uint64_t prop, dataref_t *dr)
11411007SLori.Alt@Sun.COM {
11511007SLori.Alt@Sun.COM dedup_entry_t *dde;
11611007SLori.Alt@Sun.COM
11711007SLori.Alt@Sun.COM if (ddt->cur_ddt_size >= ddt->max_ddt_size) {
11811007SLori.Alt@Sun.COM if (ddt->ddt_full == B_FALSE) {
11911007SLori.Alt@Sun.COM zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
12011007SLori.Alt@Sun.COM "Dedup table full. Deduplication will continue "
12111007SLori.Alt@Sun.COM "with existing table entries"));
12211007SLori.Alt@Sun.COM ddt->ddt_full = B_TRUE;
12311007SLori.Alt@Sun.COM }
12411007SLori.Alt@Sun.COM return;
12511007SLori.Alt@Sun.COM }
12611007SLori.Alt@Sun.COM
12711007SLori.Alt@Sun.COM if ((dde = umem_cache_alloc(ddt->ddecache, UMEM_DEFAULT))
12811007SLori.Alt@Sun.COM != NULL) {
12911007SLori.Alt@Sun.COM assert(*ddepp == NULL);
13011007SLori.Alt@Sun.COM dde->dde_next = NULL;
13111007SLori.Alt@Sun.COM dde->dde_chksum = *cs;
13211381SLori.Alt@Sun.COM dde->dde_prop = prop;
13311007SLori.Alt@Sun.COM dde->dde_ref = *dr;
13411007SLori.Alt@Sun.COM *ddepp = dde;
13511007SLori.Alt@Sun.COM ddt->cur_ddt_size += sizeof (dedup_entry_t);
13611007SLori.Alt@Sun.COM ddt->ddt_count++;
13711007SLori.Alt@Sun.COM }
13811007SLori.Alt@Sun.COM }
13911007SLori.Alt@Sun.COM
14011007SLori.Alt@Sun.COM /*
14111007SLori.Alt@Sun.COM * Using the specified dedup table, do a lookup for an entry with
14211007SLori.Alt@Sun.COM * the checksum cs. If found, return the block's reference info
14311007SLori.Alt@Sun.COM * in *dr. Otherwise, insert a new entry in the dedup table, using
14411007SLori.Alt@Sun.COM * the reference information specified by *dr.
14511007SLori.Alt@Sun.COM *
14611007SLori.Alt@Sun.COM * return value: true - entry was found
14711007SLori.Alt@Sun.COM * false - entry was not found
14811007SLori.Alt@Sun.COM */
14911007SLori.Alt@Sun.COM static boolean_t
ddt_update(libzfs_handle_t * hdl,dedup_table_t * ddt,zio_cksum_t * cs,uint64_t prop,dataref_t * dr)15011007SLori.Alt@Sun.COM ddt_update(libzfs_handle_t *hdl, dedup_table_t *ddt, zio_cksum_t *cs,
15111381SLori.Alt@Sun.COM uint64_t prop, dataref_t *dr)
15211007SLori.Alt@Sun.COM {
15311007SLori.Alt@Sun.COM uint32_t hashcode;
15411007SLori.Alt@Sun.COM dedup_entry_t **ddepp;
15511007SLori.Alt@Sun.COM
15611007SLori.Alt@Sun.COM hashcode = BF64_GET(cs->zc_word[0], 0, ddt->numhashbits);
15711007SLori.Alt@Sun.COM
15811007SLori.Alt@Sun.COM for (ddepp = &(ddt->dedup_hash_array[hashcode]); *ddepp != NULL;
15911007SLori.Alt@Sun.COM ddepp = &((*ddepp)->dde_next)) {
16011381SLori.Alt@Sun.COM if (ZIO_CHECKSUM_EQUAL(((*ddepp)->dde_chksum), *cs) &&
16111381SLori.Alt@Sun.COM (*ddepp)->dde_prop == prop) {
16211007SLori.Alt@Sun.COM *dr = (*ddepp)->dde_ref;
16311007SLori.Alt@Sun.COM return (B_TRUE);
16411007SLori.Alt@Sun.COM }
16511007SLori.Alt@Sun.COM }
16611381SLori.Alt@Sun.COM ddt_hash_append(hdl, ddt, ddepp, cs, prop, dr);
16711007SLori.Alt@Sun.COM return (B_FALSE);
16811007SLori.Alt@Sun.COM }
16911007SLori.Alt@Sun.COM
17011007SLori.Alt@Sun.COM static int
cksum_and_write(const void * buf,uint64_t len,zio_cksum_t * zc,int outfd)17111007SLori.Alt@Sun.COM cksum_and_write(const void *buf, uint64_t len, zio_cksum_t *zc, int outfd)
17211007SLori.Alt@Sun.COM {
17311007SLori.Alt@Sun.COM fletcher_4_incremental_native(buf, len, zc);
17411007SLori.Alt@Sun.COM return (write(outfd, buf, len));
17511007SLori.Alt@Sun.COM }
17611007SLori.Alt@Sun.COM
17711007SLori.Alt@Sun.COM /*
17811007SLori.Alt@Sun.COM * This function is started in a separate thread when the dedup option
17911007SLori.Alt@Sun.COM * has been requested. The main send thread determines the list of
18011007SLori.Alt@Sun.COM * snapshots to be included in the send stream and makes the ioctl calls
18111007SLori.Alt@Sun.COM * for each one. But instead of having the ioctl send the output to the
18211007SLori.Alt@Sun.COM * the output fd specified by the caller of zfs_send()), the
18311007SLori.Alt@Sun.COM * ioctl is told to direct the output to a pipe, which is read by the
18411007SLori.Alt@Sun.COM * alternate thread running THIS function. This function does the
18511007SLori.Alt@Sun.COM * dedup'ing by:
18611007SLori.Alt@Sun.COM * 1. building a dedup table (the DDT)
18711007SLori.Alt@Sun.COM * 2. doing checksums on each data block and inserting a record in the DDT
18811007SLori.Alt@Sun.COM * 3. looking for matching checksums, and
18911007SLori.Alt@Sun.COM * 4. sending a DRR_WRITE_BYREF record instead of a write record whenever
19011007SLori.Alt@Sun.COM * a duplicate block is found.
19111007SLori.Alt@Sun.COM * The output of this function then goes to the output fd requested
19211007SLori.Alt@Sun.COM * by the caller of zfs_send().
19311007SLori.Alt@Sun.COM */
19411007SLori.Alt@Sun.COM static void *
cksummer(void * arg)19511007SLori.Alt@Sun.COM cksummer(void *arg)
19611007SLori.Alt@Sun.COM {
19711007SLori.Alt@Sun.COM dedup_arg_t *dda = arg;
19811007SLori.Alt@Sun.COM char *buf = malloc(1<<20);
19911007SLori.Alt@Sun.COM dmu_replay_record_t thedrr;
20011007SLori.Alt@Sun.COM dmu_replay_record_t *drr = &thedrr;
20111007SLori.Alt@Sun.COM struct drr_begin *drrb = &thedrr.drr_u.drr_begin;
20211007SLori.Alt@Sun.COM struct drr_end *drre = &thedrr.drr_u.drr_end;
20311007SLori.Alt@Sun.COM struct drr_object *drro = &thedrr.drr_u.drr_object;
20411007SLori.Alt@Sun.COM struct drr_write *drrw = &thedrr.drr_u.drr_write;
20511935SMark.Shellenbaum@Sun.COM struct drr_spill *drrs = &thedrr.drr_u.drr_spill;
20611007SLori.Alt@Sun.COM FILE *ofp;
20711007SLori.Alt@Sun.COM int outfd;
20811381SLori.Alt@Sun.COM dmu_replay_record_t wbr_drr = {0};
20911007SLori.Alt@Sun.COM struct drr_write_byref *wbr_drrr = &wbr_drr.drr_u.drr_write_byref;
21011007SLori.Alt@Sun.COM dedup_table_t ddt;
21111007SLori.Alt@Sun.COM zio_cksum_t stream_cksum;
21211007SLori.Alt@Sun.COM uint64_t physmem = sysconf(_SC_PHYS_PAGES) * sysconf(_SC_PAGESIZE);
21311007SLori.Alt@Sun.COM uint64_t numbuckets;
21411007SLori.Alt@Sun.COM
21511007SLori.Alt@Sun.COM ddt.max_ddt_size =
21611007SLori.Alt@Sun.COM MAX((physmem * MAX_DDT_PHYSMEM_PERCENT)/100,
21711007SLori.Alt@Sun.COM SMALLEST_POSSIBLE_MAX_DDT_MB<<20);
21811007SLori.Alt@Sun.COM
21911007SLori.Alt@Sun.COM numbuckets = ddt.max_ddt_size/(sizeof (dedup_entry_t));
22011007SLori.Alt@Sun.COM
22111007SLori.Alt@Sun.COM /*
22211007SLori.Alt@Sun.COM * numbuckets must be a power of 2. Increase number to
22311007SLori.Alt@Sun.COM * a power of 2 if necessary.
22411007SLori.Alt@Sun.COM */
22511007SLori.Alt@Sun.COM if (!ISP2(numbuckets))
22611007SLori.Alt@Sun.COM numbuckets = 1 << high_order_bit(numbuckets);
22711007SLori.Alt@Sun.COM
22811007SLori.Alt@Sun.COM ddt.dedup_hash_array = calloc(numbuckets, sizeof (dedup_entry_t *));
22911007SLori.Alt@Sun.COM ddt.ddecache = umem_cache_create("dde", sizeof (dedup_entry_t), 0,
23011007SLori.Alt@Sun.COM NULL, NULL, NULL, NULL, NULL, 0);
23111007SLori.Alt@Sun.COM ddt.cur_ddt_size = numbuckets * sizeof (dedup_entry_t *);
23211007SLori.Alt@Sun.COM ddt.numhashbits = high_order_bit(numbuckets) - 1;
23311007SLori.Alt@Sun.COM ddt.ddt_full = B_FALSE;
23411007SLori.Alt@Sun.COM
23511007SLori.Alt@Sun.COM /* Initialize the write-by-reference block. */
23611007SLori.Alt@Sun.COM wbr_drr.drr_type = DRR_WRITE_BYREF;
23711007SLori.Alt@Sun.COM wbr_drr.drr_payloadlen = 0;
23811007SLori.Alt@Sun.COM
23911007SLori.Alt@Sun.COM outfd = dda->outputfd;
24011007SLori.Alt@Sun.COM ofp = fdopen(dda->inputfd, "r");
24111007SLori.Alt@Sun.COM while (ssread(drr, sizeof (dmu_replay_record_t), ofp) != 0) {
24211007SLori.Alt@Sun.COM
24311007SLori.Alt@Sun.COM switch (drr->drr_type) {
24411007SLori.Alt@Sun.COM case DRR_BEGIN:
24511007SLori.Alt@Sun.COM {
24611007SLori.Alt@Sun.COM int fflags;
24711007SLori.Alt@Sun.COM ZIO_SET_CHECKSUM(&stream_cksum, 0, 0, 0, 0);
24811007SLori.Alt@Sun.COM
24911007SLori.Alt@Sun.COM /* set the DEDUP feature flag for this stream */
25011007SLori.Alt@Sun.COM fflags = DMU_GET_FEATUREFLAGS(drrb->drr_versioninfo);
25111381SLori.Alt@Sun.COM fflags |= (DMU_BACKUP_FEATURE_DEDUP |
25211381SLori.Alt@Sun.COM DMU_BACKUP_FEATURE_DEDUPPROPS);
25311007SLori.Alt@Sun.COM DMU_SET_FEATUREFLAGS(drrb->drr_versioninfo, fflags);
25411007SLori.Alt@Sun.COM
25511007SLori.Alt@Sun.COM if (cksum_and_write(drr, sizeof (dmu_replay_record_t),
25611007SLori.Alt@Sun.COM &stream_cksum, outfd) == -1)
25711007SLori.Alt@Sun.COM goto out;
25811007SLori.Alt@Sun.COM if (DMU_GET_STREAM_HDRTYPE(drrb->drr_versioninfo) ==
25911007SLori.Alt@Sun.COM DMU_COMPOUNDSTREAM && drr->drr_payloadlen != 0) {
26011007SLori.Alt@Sun.COM int sz = drr->drr_payloadlen;
26111007SLori.Alt@Sun.COM
26211007SLori.Alt@Sun.COM if (sz > 1<<20) {
26311007SLori.Alt@Sun.COM free(buf);
26411007SLori.Alt@Sun.COM buf = malloc(sz);
26511007SLori.Alt@Sun.COM }
26611007SLori.Alt@Sun.COM (void) ssread(buf, sz, ofp);
26711007SLori.Alt@Sun.COM if (ferror(stdin))
26811007SLori.Alt@Sun.COM perror("fread");
26911007SLori.Alt@Sun.COM if (cksum_and_write(buf, sz, &stream_cksum,
27011007SLori.Alt@Sun.COM outfd) == -1)
27111007SLori.Alt@Sun.COM goto out;
27211007SLori.Alt@Sun.COM }
27311007SLori.Alt@Sun.COM break;
27411007SLori.Alt@Sun.COM }
27511007SLori.Alt@Sun.COM
27611007SLori.Alt@Sun.COM case DRR_END:
27711007SLori.Alt@Sun.COM {
27811007SLori.Alt@Sun.COM /* use the recalculated checksum */
27911007SLori.Alt@Sun.COM ZIO_SET_CHECKSUM(&drre->drr_checksum,
28011007SLori.Alt@Sun.COM stream_cksum.zc_word[0], stream_cksum.zc_word[1],
28111007SLori.Alt@Sun.COM stream_cksum.zc_word[2], stream_cksum.zc_word[3]);
28211007SLori.Alt@Sun.COM if ((write(outfd, drr,
28311007SLori.Alt@Sun.COM sizeof (dmu_replay_record_t))) == -1)
28411007SLori.Alt@Sun.COM goto out;
28511007SLori.Alt@Sun.COM break;
28611007SLori.Alt@Sun.COM }
28711007SLori.Alt@Sun.COM
28811007SLori.Alt@Sun.COM case DRR_OBJECT:
28911007SLori.Alt@Sun.COM {
29011007SLori.Alt@Sun.COM if (cksum_and_write(drr, sizeof (dmu_replay_record_t),
29111007SLori.Alt@Sun.COM &stream_cksum, outfd) == -1)
29211007SLori.Alt@Sun.COM goto out;
29311007SLori.Alt@Sun.COM if (drro->drr_bonuslen > 0) {
29411007SLori.Alt@Sun.COM (void) ssread(buf,
29511007SLori.Alt@Sun.COM P2ROUNDUP((uint64_t)drro->drr_bonuslen, 8),
29611007SLori.Alt@Sun.COM ofp);
29711007SLori.Alt@Sun.COM if (cksum_and_write(buf,
29811007SLori.Alt@Sun.COM P2ROUNDUP((uint64_t)drro->drr_bonuslen, 8),
29911007SLori.Alt@Sun.COM &stream_cksum, outfd) == -1)
30011007SLori.Alt@Sun.COM goto out;
30111007SLori.Alt@Sun.COM }
30211007SLori.Alt@Sun.COM break;
30311007SLori.Alt@Sun.COM }
30411007SLori.Alt@Sun.COM
30511935SMark.Shellenbaum@Sun.COM case DRR_SPILL:
30611935SMark.Shellenbaum@Sun.COM {
30711935SMark.Shellenbaum@Sun.COM if (cksum_and_write(drr, sizeof (dmu_replay_record_t),
30811935SMark.Shellenbaum@Sun.COM &stream_cksum, outfd) == -1)
30911935SMark.Shellenbaum@Sun.COM goto out;
31011935SMark.Shellenbaum@Sun.COM (void) ssread(buf, drrs->drr_length, ofp);
31111935SMark.Shellenbaum@Sun.COM if (cksum_and_write(buf, drrs->drr_length,
31211935SMark.Shellenbaum@Sun.COM &stream_cksum, outfd) == -1)
31311935SMark.Shellenbaum@Sun.COM goto out;
31411935SMark.Shellenbaum@Sun.COM break;
31511935SMark.Shellenbaum@Sun.COM }
31611935SMark.Shellenbaum@Sun.COM
31711007SLori.Alt@Sun.COM case DRR_FREEOBJECTS:
31811007SLori.Alt@Sun.COM {
31911007SLori.Alt@Sun.COM if (cksum_and_write(drr, sizeof (dmu_replay_record_t),
32011007SLori.Alt@Sun.COM &stream_cksum, outfd) == -1)
32111007SLori.Alt@Sun.COM goto out;
32211007SLori.Alt@Sun.COM break;
32311007SLori.Alt@Sun.COM }
32411007SLori.Alt@Sun.COM
32511007SLori.Alt@Sun.COM case DRR_WRITE:
32611007SLori.Alt@Sun.COM {
32711007SLori.Alt@Sun.COM dataref_t dataref;
32811007SLori.Alt@Sun.COM
32911007SLori.Alt@Sun.COM (void) ssread(buf, drrw->drr_length, ofp);
33011381SLori.Alt@Sun.COM
33111007SLori.Alt@Sun.COM /*
33211381SLori.Alt@Sun.COM * Use the existing checksum if it's dedup-capable,
33311381SLori.Alt@Sun.COM * else calculate a SHA256 checksum for it.
33411007SLori.Alt@Sun.COM */
33511381SLori.Alt@Sun.COM
33611381SLori.Alt@Sun.COM if (ZIO_CHECKSUM_EQUAL(drrw->drr_key.ddk_cksum,
33711381SLori.Alt@Sun.COM zero_cksum) ||
33811381SLori.Alt@Sun.COM !DRR_IS_DEDUP_CAPABLE(drrw->drr_checksumflags)) {
33911007SLori.Alt@Sun.COM SHA256_CTX ctx;
34011007SLori.Alt@Sun.COM zio_cksum_t tmpsha256;
34111007SLori.Alt@Sun.COM
34211007SLori.Alt@Sun.COM SHA256Init(&ctx);
34311007SLori.Alt@Sun.COM SHA256Update(&ctx, buf, drrw->drr_length);
34411007SLori.Alt@Sun.COM SHA256Final(&tmpsha256, &ctx);
34511381SLori.Alt@Sun.COM drrw->drr_key.ddk_cksum.zc_word[0] =
34611007SLori.Alt@Sun.COM BE_64(tmpsha256.zc_word[0]);
34711381SLori.Alt@Sun.COM drrw->drr_key.ddk_cksum.zc_word[1] =
34811007SLori.Alt@Sun.COM BE_64(tmpsha256.zc_word[1]);
34911381SLori.Alt@Sun.COM drrw->drr_key.ddk_cksum.zc_word[2] =
35011007SLori.Alt@Sun.COM BE_64(tmpsha256.zc_word[2]);
35111381SLori.Alt@Sun.COM drrw->drr_key.ddk_cksum.zc_word[3] =
35211007SLori.Alt@Sun.COM BE_64(tmpsha256.zc_word[3]);
35311381SLori.Alt@Sun.COM drrw->drr_checksumtype = ZIO_CHECKSUM_SHA256;
35411381SLori.Alt@Sun.COM drrw->drr_checksumflags = DRR_CHECKSUM_DEDUP;
35511007SLori.Alt@Sun.COM }
35611007SLori.Alt@Sun.COM
35711007SLori.Alt@Sun.COM dataref.ref_guid = drrw->drr_toguid;
35811007SLori.Alt@Sun.COM dataref.ref_object = drrw->drr_object;
35911007SLori.Alt@Sun.COM dataref.ref_offset = drrw->drr_offset;
36011007SLori.Alt@Sun.COM
36111007SLori.Alt@Sun.COM if (ddt_update(dda->dedup_hdl, &ddt,
36211381SLori.Alt@Sun.COM &drrw->drr_key.ddk_cksum, drrw->drr_key.ddk_prop,
36311381SLori.Alt@Sun.COM &dataref)) {
36411007SLori.Alt@Sun.COM /* block already present in stream */
36511007SLori.Alt@Sun.COM wbr_drrr->drr_object = drrw->drr_object;
36611007SLori.Alt@Sun.COM wbr_drrr->drr_offset = drrw->drr_offset;
36711007SLori.Alt@Sun.COM wbr_drrr->drr_length = drrw->drr_length;
36811007SLori.Alt@Sun.COM wbr_drrr->drr_toguid = drrw->drr_toguid;
36911007SLori.Alt@Sun.COM wbr_drrr->drr_refguid = dataref.ref_guid;
37011007SLori.Alt@Sun.COM wbr_drrr->drr_refobject =
37111007SLori.Alt@Sun.COM dataref.ref_object;
37211007SLori.Alt@Sun.COM wbr_drrr->drr_refoffset =
37311007SLori.Alt@Sun.COM dataref.ref_offset;
37411007SLori.Alt@Sun.COM
37511381SLori.Alt@Sun.COM wbr_drrr->drr_checksumtype =
37611381SLori.Alt@Sun.COM drrw->drr_checksumtype;
37711381SLori.Alt@Sun.COM wbr_drrr->drr_checksumflags =
37811381SLori.Alt@Sun.COM drrw->drr_checksumtype;
37911381SLori.Alt@Sun.COM wbr_drrr->drr_key.ddk_cksum =
38011381SLori.Alt@Sun.COM drrw->drr_key.ddk_cksum;
38111381SLori.Alt@Sun.COM wbr_drrr->drr_key.ddk_prop =
38211381SLori.Alt@Sun.COM drrw->drr_key.ddk_prop;
38311007SLori.Alt@Sun.COM
38411007SLori.Alt@Sun.COM if (cksum_and_write(&wbr_drr,
38511007SLori.Alt@Sun.COM sizeof (dmu_replay_record_t), &stream_cksum,
38611007SLori.Alt@Sun.COM outfd) == -1)
38711007SLori.Alt@Sun.COM goto out;
38811007SLori.Alt@Sun.COM } else {
38911007SLori.Alt@Sun.COM /* block not previously seen */
39011007SLori.Alt@Sun.COM if (cksum_and_write(drr,
39111007SLori.Alt@Sun.COM sizeof (dmu_replay_record_t), &stream_cksum,
39211007SLori.Alt@Sun.COM outfd) == -1)
39311007SLori.Alt@Sun.COM goto out;
39411007SLori.Alt@Sun.COM if (cksum_and_write(buf,
39511007SLori.Alt@Sun.COM drrw->drr_length,
39611007SLori.Alt@Sun.COM &stream_cksum, outfd) == -1)
39711007SLori.Alt@Sun.COM goto out;
39811007SLori.Alt@Sun.COM }
39911007SLori.Alt@Sun.COM break;
40011007SLori.Alt@Sun.COM }
40111007SLori.Alt@Sun.COM
40211007SLori.Alt@Sun.COM case DRR_FREE:
40311007SLori.Alt@Sun.COM {
40411007SLori.Alt@Sun.COM if (cksum_and_write(drr, sizeof (dmu_replay_record_t),
40511007SLori.Alt@Sun.COM &stream_cksum, outfd) == -1)
40611007SLori.Alt@Sun.COM goto out;
40711007SLori.Alt@Sun.COM break;
40811007SLori.Alt@Sun.COM }
40911007SLori.Alt@Sun.COM
41011007SLori.Alt@Sun.COM default:
41111007SLori.Alt@Sun.COM (void) printf("INVALID record type 0x%x\n",
41211007SLori.Alt@Sun.COM drr->drr_type);
41311007SLori.Alt@Sun.COM /* should never happen, so assert */
41411007SLori.Alt@Sun.COM assert(B_FALSE);
41511007SLori.Alt@Sun.COM }
41611007SLori.Alt@Sun.COM }
41711007SLori.Alt@Sun.COM out:
41811007SLori.Alt@Sun.COM umem_cache_destroy(ddt.ddecache);
41911007SLori.Alt@Sun.COM free(ddt.dedup_hash_array);
42011007SLori.Alt@Sun.COM free(buf);
42111007SLori.Alt@Sun.COM (void) fclose(ofp);
42211007SLori.Alt@Sun.COM
42311007SLori.Alt@Sun.COM return (NULL);
42411007SLori.Alt@Sun.COM }
42511007SLori.Alt@Sun.COM
4265367Sahrens /*
4275367Sahrens * Routines for dealing with the AVL tree of fs-nvlists
4285367Sahrens */
4295367Sahrens typedef struct fsavl_node {
4305367Sahrens avl_node_t fn_node;
4315367Sahrens nvlist_t *fn_nvfs;
4325367Sahrens char *fn_snapname;
4335367Sahrens uint64_t fn_guid;
4345367Sahrens } fsavl_node_t;
4355367Sahrens
4365367Sahrens static int
fsavl_compare(const void * arg1,const void * arg2)4375367Sahrens fsavl_compare(const void *arg1, const void *arg2)
4385367Sahrens {
4395367Sahrens const fsavl_node_t *fn1 = arg1;
4405367Sahrens const fsavl_node_t *fn2 = arg2;
4415367Sahrens
4425367Sahrens if (fn1->fn_guid > fn2->fn_guid)
4435367Sahrens return (+1);
4445367Sahrens else if (fn1->fn_guid < fn2->fn_guid)
4455367Sahrens return (-1);
4465367Sahrens else
4475367Sahrens return (0);
4485367Sahrens }
4495367Sahrens
4505367Sahrens /*
4515367Sahrens * Given the GUID of a snapshot, find its containing filesystem and
4525367Sahrens * (optionally) name.
4535367Sahrens */
4545367Sahrens static nvlist_t *
fsavl_find(avl_tree_t * avl,uint64_t snapguid,char ** snapname)4555367Sahrens fsavl_find(avl_tree_t *avl, uint64_t snapguid, char **snapname)
4565367Sahrens {
4575367Sahrens fsavl_node_t fn_find;
4585367Sahrens fsavl_node_t *fn;
4595367Sahrens
4605367Sahrens fn_find.fn_guid = snapguid;
4615367Sahrens
4625367Sahrens fn = avl_find(avl, &fn_find, NULL);
4635367Sahrens if (fn) {
4645367Sahrens if (snapname)
4655367Sahrens *snapname = fn->fn_snapname;
4665367Sahrens return (fn->fn_nvfs);
4675367Sahrens }
4685367Sahrens return (NULL);
4695367Sahrens }
4705367Sahrens
4715896Sahl static void
fsavl_destroy(avl_tree_t * avl)4725896Sahl fsavl_destroy(avl_tree_t *avl)
4735896Sahl {
4745896Sahl fsavl_node_t *fn;
4755896Sahl void *cookie;
4765896Sahl
4775896Sahl if (avl == NULL)
4785896Sahl return;
4795896Sahl
4805896Sahl cookie = NULL;
4815896Sahl while ((fn = avl_destroy_nodes(avl, &cookie)) != NULL)
4825896Sahl free(fn);
4835896Sahl avl_destroy(avl);
4845896Sahl free(avl);
4855896Sahl }
4865896Sahl
48710200SLori.Alt@Sun.COM /*
48810200SLori.Alt@Sun.COM * Given an nvlist, produce an avl tree of snapshots, ordered by guid
48910200SLori.Alt@Sun.COM */
4905367Sahrens static avl_tree_t *
fsavl_create(nvlist_t * fss)4915367Sahrens fsavl_create(nvlist_t *fss)
4925367Sahrens {
4935367Sahrens avl_tree_t *fsavl;
4945367Sahrens nvpair_t *fselem = NULL;
4955367Sahrens
4965896Sahl if ((fsavl = malloc(sizeof (avl_tree_t))) == NULL)
4975896Sahl return (NULL);
4985896Sahl
4995367Sahrens avl_create(fsavl, fsavl_compare, sizeof (fsavl_node_t),
5005367Sahrens offsetof(fsavl_node_t, fn_node));
5015367Sahrens
5025367Sahrens while ((fselem = nvlist_next_nvpair(fss, fselem)) != NULL) {
5035367Sahrens nvlist_t *nvfs, *snaps;
5045367Sahrens nvpair_t *snapelem = NULL;
5055367Sahrens
5065367Sahrens VERIFY(0 == nvpair_value_nvlist(fselem, &nvfs));
5075367Sahrens VERIFY(0 == nvlist_lookup_nvlist(nvfs, "snaps", &snaps));
5085367Sahrens
5095367Sahrens while ((snapelem =
5105367Sahrens nvlist_next_nvpair(snaps, snapelem)) != NULL) {
5115367Sahrens fsavl_node_t *fn;
5125367Sahrens uint64_t guid;
5135367Sahrens
5145367Sahrens VERIFY(0 == nvpair_value_uint64(snapelem, &guid));
5155896Sahl if ((fn = malloc(sizeof (fsavl_node_t))) == NULL) {
5165896Sahl fsavl_destroy(fsavl);
5175896Sahl return (NULL);
5185896Sahl }
5195367Sahrens fn->fn_nvfs = nvfs;
5205367Sahrens fn->fn_snapname = nvpair_name(snapelem);
5215367Sahrens fn->fn_guid = guid;
5225367Sahrens
5235367Sahrens /*
5245367Sahrens * Note: if there are multiple snaps with the
5255367Sahrens * same GUID, we ignore all but one.
5265367Sahrens */
5275367Sahrens if (avl_find(fsavl, fn, NULL) == NULL)
5285367Sahrens avl_add(fsavl, fn);
5295367Sahrens else
5305367Sahrens free(fn);
5315367Sahrens }
5325367Sahrens }
5335367Sahrens
5345367Sahrens return (fsavl);
5355367Sahrens }
5365367Sahrens
5375367Sahrens /*
5385367Sahrens * Routines for dealing with the giant nvlist of fs-nvlists, etc.
5395367Sahrens */
5405367Sahrens typedef struct send_data {
5415367Sahrens uint64_t parent_fromsnap_guid;
5425367Sahrens nvlist_t *parent_snaps;
5435367Sahrens nvlist_t *fss;
5447265Sahrens nvlist_t *snapprops;
5455367Sahrens const char *fromsnap;
5465367Sahrens const char *tosnap;
54711022STom.Erickson@Sun.COM boolean_t recursive;
5485367Sahrens
5495367Sahrens /*
5505367Sahrens * The header nvlist is of the following format:
5515367Sahrens * {
5525367Sahrens * "tosnap" -> string
5535367Sahrens * "fromsnap" -> string (if incremental)
5545367Sahrens * "fss" -> {
5555367Sahrens * id -> {
5565367Sahrens *
5575367Sahrens * "name" -> string (full name; for debugging)
5585367Sahrens * "parentfromsnap" -> number (guid of fromsnap in parent)
5595367Sahrens *
5605367Sahrens * "props" -> { name -> value (only if set here) }
5615367Sahrens * "snaps" -> { name (lastname) -> number (guid) }
5627265Sahrens * "snapprops" -> { name (lastname) -> { name -> value } }
5635367Sahrens *
5645367Sahrens * "origin" -> number (guid) (if clone)
5655367Sahrens * "sent" -> boolean (not on-disk)
5665367Sahrens * }
5675367Sahrens * }
5685367Sahrens * }
5695367Sahrens *
5705367Sahrens */
5715367Sahrens } send_data_t;
5725367Sahrens
5737265Sahrens static void send_iterate_prop(zfs_handle_t *zhp, nvlist_t *nv);
5747265Sahrens
5755367Sahrens static int
send_iterate_snap(zfs_handle_t * zhp,void * arg)5765367Sahrens send_iterate_snap(zfs_handle_t *zhp, void *arg)
5775367Sahrens {
5785367Sahrens send_data_t *sd = arg;
5795367Sahrens uint64_t guid = zhp->zfs_dmustats.dds_guid;
5805367Sahrens char *snapname;
5817265Sahrens nvlist_t *nv;
5825367Sahrens
5835367Sahrens snapname = strrchr(zhp->zfs_name, '@')+1;
5845367Sahrens
5855367Sahrens VERIFY(0 == nvlist_add_uint64(sd->parent_snaps, snapname, guid));
5865367Sahrens /*
5875367Sahrens * NB: if there is no fromsnap here (it's a newly created fs in
5885367Sahrens * an incremental replication), we will substitute the tosnap.
5895367Sahrens */
5905367Sahrens if ((sd->fromsnap && strcmp(snapname, sd->fromsnap) == 0) ||
5915367Sahrens (sd->parent_fromsnap_guid == 0 && sd->tosnap &&
5925367Sahrens strcmp(snapname, sd->tosnap) == 0)) {
5935367Sahrens sd->parent_fromsnap_guid = guid;
5945367Sahrens }
5955367Sahrens
5967265Sahrens VERIFY(0 == nvlist_alloc(&nv, NV_UNIQUE_NAME, 0));
5977265Sahrens send_iterate_prop(zhp, nv);
5987265Sahrens VERIFY(0 == nvlist_add_nvlist(sd->snapprops, snapname, nv));
5997265Sahrens nvlist_free(nv);
6007265Sahrens
6015367Sahrens zfs_close(zhp);
6025367Sahrens return (0);
6035367Sahrens }
6045367Sahrens
6055367Sahrens static void
send_iterate_prop(zfs_handle_t * zhp,nvlist_t * nv)6065367Sahrens send_iterate_prop(zfs_handle_t *zhp, nvlist_t *nv)
6075367Sahrens {
6085367Sahrens nvpair_t *elem = NULL;
6095367Sahrens
6105367Sahrens while ((elem = nvlist_next_nvpair(zhp->zfs_props, elem)) != NULL) {
6115367Sahrens char *propname = nvpair_name(elem);
6125367Sahrens zfs_prop_t prop = zfs_name_to_prop(propname);
6135367Sahrens nvlist_t *propnv;
6145367Sahrens
61510960SEric.Schrock@Sun.COM if (!zfs_prop_user(propname)) {
61610960SEric.Schrock@Sun.COM /*
61710960SEric.Schrock@Sun.COM * Realistically, this should never happen. However,
61810960SEric.Schrock@Sun.COM * we want the ability to add DSL properties without
61910960SEric.Schrock@Sun.COM * needing to make incompatible version changes. We
62010960SEric.Schrock@Sun.COM * need to ignore unknown properties to allow older
62110960SEric.Schrock@Sun.COM * software to still send datasets containing these
62210960SEric.Schrock@Sun.COM * properties, with the unknown properties elided.
62310960SEric.Schrock@Sun.COM */
62410960SEric.Schrock@Sun.COM if (prop == ZPROP_INVAL)
62510960SEric.Schrock@Sun.COM continue;
6269396SMatthew.Ahrens@Sun.COM
62710960SEric.Schrock@Sun.COM if (zfs_prop_readonly(prop))
62810960SEric.Schrock@Sun.COM continue;
62910960SEric.Schrock@Sun.COM }
6305367Sahrens
6315367Sahrens verify(nvpair_value_nvlist(elem, &propnv) == 0);
63210195SSanjeev.Bagewadi@Sun.COM if (prop == ZFS_PROP_QUOTA || prop == ZFS_PROP_RESERVATION ||
63310195SSanjeev.Bagewadi@Sun.COM prop == ZFS_PROP_REFQUOTA ||
63410195SSanjeev.Bagewadi@Sun.COM prop == ZFS_PROP_REFRESERVATION) {
63511022STom.Erickson@Sun.COM char *source;
6365367Sahrens uint64_t value;
6375367Sahrens verify(nvlist_lookup_uint64(propnv,
6385367Sahrens ZPROP_VALUE, &value) == 0);
6397265Sahrens if (zhp->zfs_type == ZFS_TYPE_SNAPSHOT)
6407265Sahrens continue;
64111022STom.Erickson@Sun.COM /*
64211022STom.Erickson@Sun.COM * May have no source before SPA_VERSION_RECVD_PROPS,
64311022STom.Erickson@Sun.COM * but is still modifiable.
64411022STom.Erickson@Sun.COM */
64511022STom.Erickson@Sun.COM if (nvlist_lookup_string(propnv,
64611022STom.Erickson@Sun.COM ZPROP_SOURCE, &source) == 0) {
64711022STom.Erickson@Sun.COM if ((strcmp(source, zhp->zfs_name) != 0) &&
64811022STom.Erickson@Sun.COM (strcmp(source,
64911022STom.Erickson@Sun.COM ZPROP_SOURCE_VAL_RECVD) != 0))
65011022STom.Erickson@Sun.COM continue;
65111022STom.Erickson@Sun.COM }
6525367Sahrens } else {
6535367Sahrens char *source;
6545367Sahrens if (nvlist_lookup_string(propnv,
6555367Sahrens ZPROP_SOURCE, &source) != 0)
6565367Sahrens continue;
65711022STom.Erickson@Sun.COM if ((strcmp(source, zhp->zfs_name) != 0) &&
65811022STom.Erickson@Sun.COM (strcmp(source, ZPROP_SOURCE_VAL_RECVD) != 0))
6595367Sahrens continue;
6605367Sahrens }
6615367Sahrens
6625367Sahrens if (zfs_prop_user(propname) ||
6635367Sahrens zfs_prop_get_type(prop) == PROP_TYPE_STRING) {
6645367Sahrens char *value;
6655367Sahrens verify(nvlist_lookup_string(propnv,
6665367Sahrens ZPROP_VALUE, &value) == 0);
6675367Sahrens VERIFY(0 == nvlist_add_string(nv, propname, value));
6685367Sahrens } else {
6695367Sahrens uint64_t value;
6705367Sahrens verify(nvlist_lookup_uint64(propnv,
6715367Sahrens ZPROP_VALUE, &value) == 0);
6725367Sahrens VERIFY(0 == nvlist_add_uint64(nv, propname, value));
6735367Sahrens }
6745367Sahrens }
6755367Sahrens }
6765367Sahrens
67710200SLori.Alt@Sun.COM /*
67810200SLori.Alt@Sun.COM * recursively generate nvlists describing datasets. See comment
67910200SLori.Alt@Sun.COM * for the data structure send_data_t above for description of contents
68010200SLori.Alt@Sun.COM * of the nvlist.
68110200SLori.Alt@Sun.COM */
6825367Sahrens static int
send_iterate_fs(zfs_handle_t * zhp,void * arg)6835367Sahrens send_iterate_fs(zfs_handle_t *zhp, void *arg)
6845367Sahrens {
6855367Sahrens send_data_t *sd = arg;
6865367Sahrens nvlist_t *nvfs, *nv;
68711022STom.Erickson@Sun.COM int rv = 0;
6885367Sahrens uint64_t parent_fromsnap_guid_save = sd->parent_fromsnap_guid;
6895367Sahrens uint64_t guid = zhp->zfs_dmustats.dds_guid;
6905367Sahrens char guidstring[64];
6915367Sahrens
6925367Sahrens VERIFY(0 == nvlist_alloc(&nvfs, NV_UNIQUE_NAME, 0));
6935367Sahrens VERIFY(0 == nvlist_add_string(nvfs, "name", zhp->zfs_name));
6945367Sahrens VERIFY(0 == nvlist_add_uint64(nvfs, "parentfromsnap",
6955367Sahrens sd->parent_fromsnap_guid));
6965367Sahrens
6975367Sahrens if (zhp->zfs_dmustats.dds_origin[0]) {
6985367Sahrens zfs_handle_t *origin = zfs_open(zhp->zfs_hdl,
6995367Sahrens zhp->zfs_dmustats.dds_origin, ZFS_TYPE_SNAPSHOT);
7005367Sahrens if (origin == NULL)
7015367Sahrens return (-1);
7025367Sahrens VERIFY(0 == nvlist_add_uint64(nvfs, "origin",
7035367Sahrens origin->zfs_dmustats.dds_guid));
7045367Sahrens }
7055367Sahrens
7065367Sahrens /* iterate over props */
7075367Sahrens VERIFY(0 == nvlist_alloc(&nv, NV_UNIQUE_NAME, 0));
7085367Sahrens send_iterate_prop(zhp, nv);
7095367Sahrens VERIFY(0 == nvlist_add_nvlist(nvfs, "props", nv));
7105367Sahrens nvlist_free(nv);
7115367Sahrens
7125367Sahrens /* iterate over snaps, and set sd->parent_fromsnap_guid */
7135367Sahrens sd->parent_fromsnap_guid = 0;
7145367Sahrens VERIFY(0 == nvlist_alloc(&sd->parent_snaps, NV_UNIQUE_NAME, 0));
7157265Sahrens VERIFY(0 == nvlist_alloc(&sd->snapprops, NV_UNIQUE_NAME, 0));
7165367Sahrens (void) zfs_iter_snapshots(zhp, send_iterate_snap, sd);
7175367Sahrens VERIFY(0 == nvlist_add_nvlist(nvfs, "snaps", sd->parent_snaps));
7187265Sahrens VERIFY(0 == nvlist_add_nvlist(nvfs, "snapprops", sd->snapprops));
7195367Sahrens nvlist_free(sd->parent_snaps);
7207265Sahrens nvlist_free(sd->snapprops);
7215367Sahrens
7225367Sahrens /* add this fs to nvlist */
7235367Sahrens (void) snprintf(guidstring, sizeof (guidstring),
7245367Sahrens "0x%llx", (longlong_t)guid);
7255367Sahrens VERIFY(0 == nvlist_add_nvlist(sd->fss, guidstring, nvfs));
7265367Sahrens nvlist_free(nvfs);
7275367Sahrens
7285367Sahrens /* iterate over children */
72911022STom.Erickson@Sun.COM if (sd->recursive)
73011022STom.Erickson@Sun.COM rv = zfs_iter_filesystems(zhp, send_iterate_fs, sd);
7315367Sahrens
7325367Sahrens sd->parent_fromsnap_guid = parent_fromsnap_guid_save;
7335367Sahrens
7345367Sahrens zfs_close(zhp);
7355367Sahrens return (rv);
7365367Sahrens }
7375367Sahrens
7385367Sahrens static int
gather_nvlist(libzfs_handle_t * hdl,const char * fsname,const char * fromsnap,const char * tosnap,boolean_t recursive,nvlist_t ** nvlp,avl_tree_t ** avlp)7395367Sahrens gather_nvlist(libzfs_handle_t *hdl, const char *fsname, const char *fromsnap,
74011022STom.Erickson@Sun.COM const char *tosnap, boolean_t recursive, nvlist_t **nvlp, avl_tree_t **avlp)
7415367Sahrens {
7425367Sahrens zfs_handle_t *zhp;
7435367Sahrens send_data_t sd = { 0 };
7445367Sahrens int error;
7455367Sahrens
7465367Sahrens zhp = zfs_open(hdl, fsname, ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME);
7475367Sahrens if (zhp == NULL)
7485367Sahrens return (EZFS_BADTYPE);
7495367Sahrens
7505367Sahrens VERIFY(0 == nvlist_alloc(&sd.fss, NV_UNIQUE_NAME, 0));
7515367Sahrens sd.fromsnap = fromsnap;
7525367Sahrens sd.tosnap = tosnap;
75311022STom.Erickson@Sun.COM sd.recursive = recursive;
7545896Sahl
7555896Sahl if ((error = send_iterate_fs(zhp, &sd)) != 0) {
7565896Sahl nvlist_free(sd.fss);
7575896Sahl if (avlp != NULL)
7585896Sahl *avlp = NULL;
7595896Sahl *nvlp = NULL;
7605896Sahl return (error);
7615896Sahl }
7625896Sahl
7635896Sahl if (avlp != NULL && (*avlp = fsavl_create(sd.fss)) == NULL) {
7645896Sahl nvlist_free(sd.fss);
7655896Sahl *nvlp = NULL;
7665896Sahl return (EZFS_NOMEM);
7675896Sahl }
7685367Sahrens
7695367Sahrens *nvlp = sd.fss;
7705896Sahl return (0);
7715367Sahrens }
7725367Sahrens
7735367Sahrens /*
7745367Sahrens * Routines for dealing with the sorted snapshot functionality
7755367Sahrens */
7765367Sahrens typedef struct zfs_node {
7775367Sahrens zfs_handle_t *zn_handle;
7785367Sahrens avl_node_t zn_avlnode;
7795367Sahrens } zfs_node_t;
7805367Sahrens
7815367Sahrens static int
zfs_sort_snaps(zfs_handle_t * zhp,void * data)7825367Sahrens zfs_sort_snaps(zfs_handle_t *zhp, void *data)
7835367Sahrens {
7845367Sahrens avl_tree_t *avl = data;
785*12786SChris.Kirby@oracle.com zfs_node_t *node;
786*12786SChris.Kirby@oracle.com zfs_node_t search;
787*12786SChris.Kirby@oracle.com
788*12786SChris.Kirby@oracle.com search.zn_handle = zhp;
789*12786SChris.Kirby@oracle.com node = avl_find(avl, &search, NULL);
790*12786SChris.Kirby@oracle.com if (node) {
791*12786SChris.Kirby@oracle.com /*
792*12786SChris.Kirby@oracle.com * If this snapshot was renamed while we were creating the
793*12786SChris.Kirby@oracle.com * AVL tree, it's possible that we already inserted it under
794*12786SChris.Kirby@oracle.com * its old name. Remove the old handle before adding the new
795*12786SChris.Kirby@oracle.com * one.
796*12786SChris.Kirby@oracle.com */
797*12786SChris.Kirby@oracle.com zfs_close(node->zn_handle);
798*12786SChris.Kirby@oracle.com avl_remove(avl, node);
799*12786SChris.Kirby@oracle.com free(node);
800*12786SChris.Kirby@oracle.com }
801*12786SChris.Kirby@oracle.com
802*12786SChris.Kirby@oracle.com node = zfs_alloc(zhp->zfs_hdl, sizeof (zfs_node_t));
8035367Sahrens node->zn_handle = zhp;
8045367Sahrens avl_add(avl, node);
805*12786SChris.Kirby@oracle.com
8065367Sahrens return (0);
8075367Sahrens }
8085367Sahrens
8095367Sahrens static int
zfs_snapshot_compare(const void * larg,const void * rarg)8105367Sahrens zfs_snapshot_compare(const void *larg, const void *rarg)
8115367Sahrens {
8125367Sahrens zfs_handle_t *l = ((zfs_node_t *)larg)->zn_handle;
8135367Sahrens zfs_handle_t *r = ((zfs_node_t *)rarg)->zn_handle;
8145367Sahrens uint64_t lcreate, rcreate;
8155367Sahrens
8165367Sahrens /*
8175367Sahrens * Sort them according to creation time. We use the hidden
8185367Sahrens * CREATETXG property to get an absolute ordering of snapshots.
8195367Sahrens */
8205367Sahrens lcreate = zfs_prop_get_int(l, ZFS_PROP_CREATETXG);
8215367Sahrens rcreate = zfs_prop_get_int(r, ZFS_PROP_CREATETXG);
8225367Sahrens
8235367Sahrens if (lcreate < rcreate)
8245367Sahrens return (-1);
8255367Sahrens else if (lcreate > rcreate)
8265367Sahrens return (+1);
8275367Sahrens else
8285367Sahrens return (0);
8295367Sahrens }
8305367Sahrens
83110342Schris.kirby@sun.com int
zfs_iter_snapshots_sorted(zfs_handle_t * zhp,zfs_iter_f callback,void * data)8325367Sahrens zfs_iter_snapshots_sorted(zfs_handle_t *zhp, zfs_iter_f callback, void *data)
8335367Sahrens {
8345367Sahrens int ret = 0;
8355367Sahrens zfs_node_t *node;
8365367Sahrens avl_tree_t avl;
8375367Sahrens void *cookie = NULL;
8385367Sahrens
8395367Sahrens avl_create(&avl, zfs_snapshot_compare,
8405367Sahrens sizeof (zfs_node_t), offsetof(zfs_node_t, zn_avlnode));
8415367Sahrens
8425367Sahrens ret = zfs_iter_snapshots(zhp, zfs_sort_snaps, &avl);
8435367Sahrens
8445367Sahrens for (node = avl_first(&avl); node != NULL; node = AVL_NEXT(&avl, node))
8455367Sahrens ret |= callback(node->zn_handle, data);
8465367Sahrens
8475367Sahrens while ((node = avl_destroy_nodes(&avl, &cookie)) != NULL)
8485367Sahrens free(node);
8495367Sahrens
8505367Sahrens avl_destroy(&avl);
8515367Sahrens
8525367Sahrens return (ret);
8535367Sahrens }
8545367Sahrens
8555367Sahrens /*
8565367Sahrens * Routines specific to "zfs send"
8575367Sahrens */
8585367Sahrens typedef struct send_dump_data {
8595367Sahrens /* these are all just the short snapname (the part after the @) */
8605367Sahrens const char *fromsnap;
8615367Sahrens const char *tosnap;
86211007SLori.Alt@Sun.COM char prevsnap[ZFS_MAXNAMELEN];
863*12786SChris.Kirby@oracle.com uint64_t prevsnap_obj;
8645367Sahrens boolean_t seenfrom, seento, replicate, doall, fromorigin;
8655367Sahrens boolean_t verbose;
8665367Sahrens int outfd;
8675367Sahrens boolean_t err;
8685367Sahrens nvlist_t *fss;
8695367Sahrens avl_tree_t *fsavl;
87011007SLori.Alt@Sun.COM snapfilter_cb_t *filter_cb;
87111007SLori.Alt@Sun.COM void *filter_cb_arg;
87212296SLin.Ling@Sun.COM nvlist_t *debugnv;
873*12786SChris.Kirby@oracle.com char holdtag[ZFS_MAXNAMELEN];
874*12786SChris.Kirby@oracle.com int cleanup_fd;
8755367Sahrens } send_dump_data_t;
8765367Sahrens
8775367Sahrens /*
8785367Sahrens * Dumps a backup of the given snapshot (incremental from fromsnap if it's not
8795367Sahrens * NULL) to the file descriptor specified by outfd.
8805367Sahrens */
8815367Sahrens static int
dump_ioctl(zfs_handle_t * zhp,const char * fromsnap,uint64_t fromsnap_obj,boolean_t fromorigin,int outfd,nvlist_t * debugnv)882*12786SChris.Kirby@oracle.com dump_ioctl(zfs_handle_t *zhp, const char *fromsnap, uint64_t fromsnap_obj,
883*12786SChris.Kirby@oracle.com boolean_t fromorigin, int outfd, nvlist_t *debugnv)
8845367Sahrens {
8855367Sahrens zfs_cmd_t zc = { 0 };
8865367Sahrens libzfs_handle_t *hdl = zhp->zfs_hdl;
88712296SLin.Ling@Sun.COM nvlist_t *thisdbg;
8885367Sahrens
8895367Sahrens assert(zhp->zfs_type == ZFS_TYPE_SNAPSHOT);
890*12786SChris.Kirby@oracle.com assert(fromsnap_obj == 0 || !fromorigin);
8915367Sahrens
8925367Sahrens (void) strlcpy(zc.zc_name, zhp->zfs_name, sizeof (zc.zc_name));
8935367Sahrens zc.zc_cookie = outfd;
8945367Sahrens zc.zc_obj = fromorigin;
895*12786SChris.Kirby@oracle.com zc.zc_sendobj = zfs_prop_get_int(zhp, ZFS_PROP_OBJSETID);
896*12786SChris.Kirby@oracle.com zc.zc_fromobj = fromsnap_obj;
89711668SChris.Kirby@sun.com
89812296SLin.Ling@Sun.COM VERIFY(0 == nvlist_alloc(&thisdbg, NV_UNIQUE_NAME, 0));
89912296SLin.Ling@Sun.COM if (fromsnap && fromsnap[0] != '\0') {
90012296SLin.Ling@Sun.COM VERIFY(0 == nvlist_add_string(thisdbg,
90112296SLin.Ling@Sun.COM "fromsnap", fromsnap));
90212296SLin.Ling@Sun.COM }
90312296SLin.Ling@Sun.COM
9045367Sahrens if (ioctl(zhp->zfs_hdl->libzfs_fd, ZFS_IOC_SEND, &zc) != 0) {
9055367Sahrens char errbuf[1024];
9065367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
9075367Sahrens "warning: cannot send '%s'"), zhp->zfs_name);
9085367Sahrens
90912296SLin.Ling@Sun.COM VERIFY(0 == nvlist_add_uint64(thisdbg, "error", errno));
91012296SLin.Ling@Sun.COM if (debugnv) {
91112296SLin.Ling@Sun.COM VERIFY(0 == nvlist_add_nvlist(debugnv,
91212296SLin.Ling@Sun.COM zhp->zfs_name, thisdbg));
91312296SLin.Ling@Sun.COM }
91412296SLin.Ling@Sun.COM nvlist_free(thisdbg);
91512296SLin.Ling@Sun.COM
9165367Sahrens switch (errno) {
9175367Sahrens
9185367Sahrens case EXDEV:
9195367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
9205367Sahrens "not an earlier snapshot from the same fs"));
9215367Sahrens return (zfs_error(hdl, EZFS_CROSSTARGET, errbuf));
9225367Sahrens
9235367Sahrens case ENOENT:
9245367Sahrens if (zfs_dataset_exists(hdl, zc.zc_name,
9255367Sahrens ZFS_TYPE_SNAPSHOT)) {
9265367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
9275367Sahrens "incremental source (@%s) does not exist"),
9285367Sahrens zc.zc_value);
9295367Sahrens }
9305367Sahrens return (zfs_error(hdl, EZFS_NOENT, errbuf));
9315367Sahrens
9325367Sahrens case EDQUOT:
9335367Sahrens case EFBIG:
9345367Sahrens case EIO:
9355367Sahrens case ENOLINK:
9365367Sahrens case ENOSPC:
9375367Sahrens case ENOSTR:
9385367Sahrens case ENXIO:
9395367Sahrens case EPIPE:
9405367Sahrens case ERANGE:
9415367Sahrens case EFAULT:
9425367Sahrens case EROFS:
9435367Sahrens zfs_error_aux(hdl, strerror(errno));
9445367Sahrens return (zfs_error(hdl, EZFS_BADBACKUP, errbuf));
9455367Sahrens
9465367Sahrens default:
9475367Sahrens return (zfs_standard_error(hdl, errno, errbuf));
9485367Sahrens }
9495367Sahrens }
9505367Sahrens
95112296SLin.Ling@Sun.COM if (debugnv)
95212296SLin.Ling@Sun.COM VERIFY(0 == nvlist_add_nvlist(debugnv, zhp->zfs_name, thisdbg));
95312296SLin.Ling@Sun.COM nvlist_free(thisdbg);
95412296SLin.Ling@Sun.COM
9555367Sahrens return (0);
9565367Sahrens }
9575367Sahrens
9585367Sahrens static int
hold_for_send(zfs_handle_t * zhp,send_dump_data_t * sdd)959*12786SChris.Kirby@oracle.com hold_for_send(zfs_handle_t *zhp, send_dump_data_t *sdd)
960*12786SChris.Kirby@oracle.com {
961*12786SChris.Kirby@oracle.com zfs_handle_t *pzhp;
962*12786SChris.Kirby@oracle.com int error = 0;
963*12786SChris.Kirby@oracle.com char *thissnap;
964*12786SChris.Kirby@oracle.com
965*12786SChris.Kirby@oracle.com assert(zhp->zfs_type == ZFS_TYPE_SNAPSHOT);
966*12786SChris.Kirby@oracle.com
967*12786SChris.Kirby@oracle.com /*
968*12786SChris.Kirby@oracle.com * zfs_send() only opens a cleanup_fd for sends that need it,
969*12786SChris.Kirby@oracle.com * e.g. replication and doall.
970*12786SChris.Kirby@oracle.com */
971*12786SChris.Kirby@oracle.com if (sdd->cleanup_fd == -1)
972*12786SChris.Kirby@oracle.com return (0);
973*12786SChris.Kirby@oracle.com
974*12786SChris.Kirby@oracle.com thissnap = strchr(zhp->zfs_name, '@') + 1;
975*12786SChris.Kirby@oracle.com *(thissnap - 1) = '\0';
976*12786SChris.Kirby@oracle.com pzhp = zfs_open(zhp->zfs_hdl, zhp->zfs_name, ZFS_TYPE_DATASET);
977*12786SChris.Kirby@oracle.com *(thissnap - 1) = '@';
978*12786SChris.Kirby@oracle.com
979*12786SChris.Kirby@oracle.com /*
980*12786SChris.Kirby@oracle.com * It's OK if the parent no longer exists. The send code will
981*12786SChris.Kirby@oracle.com * handle that error.
982*12786SChris.Kirby@oracle.com */
983*12786SChris.Kirby@oracle.com if (pzhp) {
984*12786SChris.Kirby@oracle.com error = zfs_hold(pzhp, thissnap, sdd->holdtag,
985*12786SChris.Kirby@oracle.com B_FALSE, B_TRUE, B_TRUE, sdd->cleanup_fd,
986*12786SChris.Kirby@oracle.com zfs_prop_get_int(zhp, ZFS_PROP_OBJSETID),
987*12786SChris.Kirby@oracle.com zfs_prop_get_int(zhp, ZFS_PROP_CREATETXG));
988*12786SChris.Kirby@oracle.com zfs_close(pzhp);
989*12786SChris.Kirby@oracle.com }
990*12786SChris.Kirby@oracle.com
991*12786SChris.Kirby@oracle.com return (error);
992*12786SChris.Kirby@oracle.com }
993*12786SChris.Kirby@oracle.com
994*12786SChris.Kirby@oracle.com static int
dump_snapshot(zfs_handle_t * zhp,void * arg)9955367Sahrens dump_snapshot(zfs_handle_t *zhp, void *arg)
9965367Sahrens {
9975367Sahrens send_dump_data_t *sdd = arg;
998*12786SChris.Kirby@oracle.com char *thissnap;
9995367Sahrens int err;
100012282STom.Erickson@Sun.COM boolean_t isfromsnap, istosnap;
100112282STom.Erickson@Sun.COM boolean_t exclude = B_FALSE;
10025367Sahrens
10035367Sahrens thissnap = strchr(zhp->zfs_name, '@') + 1;
100412282STom.Erickson@Sun.COM isfromsnap = (sdd->fromsnap != NULL &&
100512282STom.Erickson@Sun.COM strcmp(sdd->fromsnap, thissnap) == 0);
100612282STom.Erickson@Sun.COM
100712282STom.Erickson@Sun.COM if (!sdd->seenfrom && isfromsnap) {
1008*12786SChris.Kirby@oracle.com err = hold_for_send(zhp, sdd);
1009*12786SChris.Kirby@oracle.com if (err == 0) {
1010*12786SChris.Kirby@oracle.com sdd->seenfrom = B_TRUE;
1011*12786SChris.Kirby@oracle.com (void) strcpy(sdd->prevsnap, thissnap);
1012*12786SChris.Kirby@oracle.com sdd->prevsnap_obj = zfs_prop_get_int(zhp,
1013*12786SChris.Kirby@oracle.com ZFS_PROP_OBJSETID);
1014*12786SChris.Kirby@oracle.com } else if (err == ENOENT) {
1015*12786SChris.Kirby@oracle.com err = 0;
1016*12786SChris.Kirby@oracle.com }
10175367Sahrens zfs_close(zhp);
1018*12786SChris.Kirby@oracle.com return (err);
10195367Sahrens }
10205367Sahrens
10215367Sahrens if (sdd->seento || !sdd->seenfrom) {
10225367Sahrens zfs_close(zhp);
10235367Sahrens return (0);
10245367Sahrens }
10255367Sahrens
102612282STom.Erickson@Sun.COM istosnap = (strcmp(sdd->tosnap, thissnap) == 0);
102712282STom.Erickson@Sun.COM if (istosnap)
102811007SLori.Alt@Sun.COM sdd->seento = B_TRUE;
102911007SLori.Alt@Sun.COM
103012282STom.Erickson@Sun.COM if (!sdd->doall && !isfromsnap && !istosnap) {
103112282STom.Erickson@Sun.COM if (sdd->replicate) {
103212282STom.Erickson@Sun.COM char *snapname;
103312282STom.Erickson@Sun.COM nvlist_t *snapprops;
103412282STom.Erickson@Sun.COM /*
103512282STom.Erickson@Sun.COM * Filter out all intermediate snapshots except origin
103612282STom.Erickson@Sun.COM * snapshots needed to replicate clones.
103712282STom.Erickson@Sun.COM */
103812282STom.Erickson@Sun.COM nvlist_t *nvfs = fsavl_find(sdd->fsavl,
103912282STom.Erickson@Sun.COM zhp->zfs_dmustats.dds_guid, &snapname);
104012282STom.Erickson@Sun.COM
104112282STom.Erickson@Sun.COM VERIFY(0 == nvlist_lookup_nvlist(nvfs,
104212282STom.Erickson@Sun.COM "snapprops", &snapprops));
104312282STom.Erickson@Sun.COM VERIFY(0 == nvlist_lookup_nvlist(snapprops,
104412282STom.Erickson@Sun.COM thissnap, &snapprops));
104512282STom.Erickson@Sun.COM exclude = !nvlist_exists(snapprops, "is_clone_origin");
104612282STom.Erickson@Sun.COM } else {
104712282STom.Erickson@Sun.COM exclude = B_TRUE;
104812282STom.Erickson@Sun.COM }
104912282STom.Erickson@Sun.COM }
105012282STom.Erickson@Sun.COM
105111007SLori.Alt@Sun.COM /*
105211007SLori.Alt@Sun.COM * If a filter function exists, call it to determine whether
105311007SLori.Alt@Sun.COM * this snapshot will be sent.
105411007SLori.Alt@Sun.COM */
105512282STom.Erickson@Sun.COM if (exclude || (sdd->filter_cb != NULL &&
105612282STom.Erickson@Sun.COM sdd->filter_cb(zhp, sdd->filter_cb_arg) == B_FALSE)) {
105711007SLori.Alt@Sun.COM /*
105811007SLori.Alt@Sun.COM * This snapshot is filtered out. Don't send it, and don't
1059*12786SChris.Kirby@oracle.com * set prevsnap_obj, so it will be as if this snapshot didn't
106011007SLori.Alt@Sun.COM * exist, and the next accepted snapshot will be sent as
106111007SLori.Alt@Sun.COM * an incremental from the last accepted one, or as the
106211007SLori.Alt@Sun.COM * first (and full) snapshot in the case of a replication,
106311007SLori.Alt@Sun.COM * non-incremental send.
106411007SLori.Alt@Sun.COM */
106511007SLori.Alt@Sun.COM zfs_close(zhp);
106611007SLori.Alt@Sun.COM return (0);
106711007SLori.Alt@Sun.COM }
106811007SLori.Alt@Sun.COM
1069*12786SChris.Kirby@oracle.com err = hold_for_send(zhp, sdd);
1070*12786SChris.Kirby@oracle.com if (err) {
1071*12786SChris.Kirby@oracle.com if (err == ENOENT)
1072*12786SChris.Kirby@oracle.com err = 0;
1073*12786SChris.Kirby@oracle.com zfs_close(zhp);
1074*12786SChris.Kirby@oracle.com return (err);
1075*12786SChris.Kirby@oracle.com }
1076*12786SChris.Kirby@oracle.com
10775367Sahrens /* send it */
10785367Sahrens if (sdd->verbose) {
10795367Sahrens (void) fprintf(stderr, "sending from @%s to %s\n",
108011007SLori.Alt@Sun.COM sdd->prevsnap, zhp->zfs_name);
10815367Sahrens }
10825367Sahrens
1083*12786SChris.Kirby@oracle.com err = dump_ioctl(zhp, sdd->prevsnap, sdd->prevsnap_obj,
108411007SLori.Alt@Sun.COM sdd->prevsnap[0] == '\0' && (sdd->fromorigin || sdd->replicate),
1085*12786SChris.Kirby@oracle.com sdd->outfd, sdd->debugnv);
1086*12786SChris.Kirby@oracle.com
1087*12786SChris.Kirby@oracle.com (void) strcpy(sdd->prevsnap, thissnap);
1088*12786SChris.Kirby@oracle.com sdd->prevsnap_obj = zfs_prop_get_int(zhp, ZFS_PROP_OBJSETID);
10895367Sahrens zfs_close(zhp);
10905367Sahrens return (err);
10915367Sahrens }
10925367Sahrens
10935367Sahrens static int
dump_filesystem(zfs_handle_t * zhp,void * arg)10945367Sahrens dump_filesystem(zfs_handle_t *zhp, void *arg)
10955367Sahrens {
10965367Sahrens int rv = 0;
10975367Sahrens send_dump_data_t *sdd = arg;
10985367Sahrens boolean_t missingfrom = B_FALSE;
10995367Sahrens zfs_cmd_t zc = { 0 };
11005367Sahrens
11015367Sahrens (void) snprintf(zc.zc_name, sizeof (zc.zc_name), "%s@%s",
11025367Sahrens zhp->zfs_name, sdd->tosnap);
11035367Sahrens if (ioctl(zhp->zfs_hdl->libzfs_fd, ZFS_IOC_OBJSET_STATS, &zc) != 0) {
11045367Sahrens (void) fprintf(stderr, "WARNING: "
11055367Sahrens "could not send %s@%s: does not exist\n",
11065367Sahrens zhp->zfs_name, sdd->tosnap);
11075367Sahrens sdd->err = B_TRUE;
11085367Sahrens return (0);
11095367Sahrens }
11105367Sahrens
11115367Sahrens if (sdd->replicate && sdd->fromsnap) {
11125367Sahrens /*
11135367Sahrens * If this fs does not have fromsnap, and we're doing
11145367Sahrens * recursive, we need to send a full stream from the
11155367Sahrens * beginning (or an incremental from the origin if this
11165367Sahrens * is a clone). If we're doing non-recursive, then let
11175367Sahrens * them get the error.
11185367Sahrens */
11195367Sahrens (void) snprintf(zc.zc_name, sizeof (zc.zc_name), "%s@%s",
11205367Sahrens zhp->zfs_name, sdd->fromsnap);
11215367Sahrens if (ioctl(zhp->zfs_hdl->libzfs_fd,
11225367Sahrens ZFS_IOC_OBJSET_STATS, &zc) != 0) {
11235367Sahrens missingfrom = B_TRUE;
11245367Sahrens }
11255367Sahrens }
11265367Sahrens
112712282STom.Erickson@Sun.COM sdd->seenfrom = sdd->seento = sdd->prevsnap[0] = 0;
1128*12786SChris.Kirby@oracle.com sdd->prevsnap_obj = 0;
112912282STom.Erickson@Sun.COM if (sdd->fromsnap == NULL || missingfrom)
113012282STom.Erickson@Sun.COM sdd->seenfrom = B_TRUE;
113112282STom.Erickson@Sun.COM
113212282STom.Erickson@Sun.COM rv = zfs_iter_snapshots_sorted(zhp, dump_snapshot, arg);
113312282STom.Erickson@Sun.COM if (!sdd->seenfrom) {
113412282STom.Erickson@Sun.COM (void) fprintf(stderr,
113512282STom.Erickson@Sun.COM "WARNING: could not send %s@%s:\n"
113612282STom.Erickson@Sun.COM "incremental source (%s@%s) does not exist\n",
113712282STom.Erickson@Sun.COM zhp->zfs_name, sdd->tosnap,
113812282STom.Erickson@Sun.COM zhp->zfs_name, sdd->fromsnap);
113912282STom.Erickson@Sun.COM sdd->err = B_TRUE;
114012282STom.Erickson@Sun.COM } else if (!sdd->seento) {
114112282STom.Erickson@Sun.COM if (sdd->fromsnap) {
11425367Sahrens (void) fprintf(stderr,
11435367Sahrens "WARNING: could not send %s@%s:\n"
114412282STom.Erickson@Sun.COM "incremental source (%s@%s) "
114512282STom.Erickson@Sun.COM "is not earlier than it\n",
11465367Sahrens zhp->zfs_name, sdd->tosnap,
11475367Sahrens zhp->zfs_name, sdd->fromsnap);
114812282STom.Erickson@Sun.COM } else {
114912282STom.Erickson@Sun.COM (void) fprintf(stderr, "WARNING: "
115012282STom.Erickson@Sun.COM "could not send %s@%s: does not exist\n",
115112282STom.Erickson@Sun.COM zhp->zfs_name, sdd->tosnap);
11525367Sahrens }
115312282STom.Erickson@Sun.COM sdd->err = B_TRUE;
11545367Sahrens }
11555367Sahrens
11565367Sahrens return (rv);
11575367Sahrens }
11585367Sahrens
11595367Sahrens static int
dump_filesystems(zfs_handle_t * rzhp,void * arg)11605367Sahrens dump_filesystems(zfs_handle_t *rzhp, void *arg)
11615367Sahrens {
11625367Sahrens send_dump_data_t *sdd = arg;
11635367Sahrens nvpair_t *fspair;
11645367Sahrens boolean_t needagain, progress;
11655367Sahrens
11665367Sahrens if (!sdd->replicate)
11675367Sahrens return (dump_filesystem(rzhp, sdd));
11685367Sahrens
116912282STom.Erickson@Sun.COM /* Mark the clone origin snapshots. */
117012282STom.Erickson@Sun.COM for (fspair = nvlist_next_nvpair(sdd->fss, NULL); fspair;
117112282STom.Erickson@Sun.COM fspair = nvlist_next_nvpair(sdd->fss, fspair)) {
117212282STom.Erickson@Sun.COM nvlist_t *nvfs;
117312282STom.Erickson@Sun.COM uint64_t origin_guid = 0;
117412282STom.Erickson@Sun.COM
117512282STom.Erickson@Sun.COM VERIFY(0 == nvpair_value_nvlist(fspair, &nvfs));
117612282STom.Erickson@Sun.COM (void) nvlist_lookup_uint64(nvfs, "origin", &origin_guid);
117712282STom.Erickson@Sun.COM if (origin_guid != 0) {
117812282STom.Erickson@Sun.COM char *snapname;
117912282STom.Erickson@Sun.COM nvlist_t *origin_nv = fsavl_find(sdd->fsavl,
118012282STom.Erickson@Sun.COM origin_guid, &snapname);
118112282STom.Erickson@Sun.COM if (origin_nv != NULL) {
118212282STom.Erickson@Sun.COM nvlist_t *snapprops;
118312282STom.Erickson@Sun.COM VERIFY(0 == nvlist_lookup_nvlist(origin_nv,
118412282STom.Erickson@Sun.COM "snapprops", &snapprops));
118512282STom.Erickson@Sun.COM VERIFY(0 == nvlist_lookup_nvlist(snapprops,
118612282STom.Erickson@Sun.COM snapname, &snapprops));
118712282STom.Erickson@Sun.COM VERIFY(0 == nvlist_add_boolean(
118812282STom.Erickson@Sun.COM snapprops, "is_clone_origin"));
118912282STom.Erickson@Sun.COM }
119012282STom.Erickson@Sun.COM }
119112282STom.Erickson@Sun.COM }
11925367Sahrens again:
11935367Sahrens needagain = progress = B_FALSE;
11945367Sahrens for (fspair = nvlist_next_nvpair(sdd->fss, NULL); fspair;
11955367Sahrens fspair = nvlist_next_nvpair(sdd->fss, fspair)) {
11965367Sahrens nvlist_t *fslist;
11975367Sahrens char *fsname;
11985367Sahrens zfs_handle_t *zhp;
11995367Sahrens int err;
12005367Sahrens uint64_t origin_guid = 0;
12015367Sahrens
12025367Sahrens VERIFY(nvpair_value_nvlist(fspair, &fslist) == 0);
12035367Sahrens if (nvlist_lookup_boolean(fslist, "sent") == 0)
12045367Sahrens continue;
12055367Sahrens
12065367Sahrens VERIFY(nvlist_lookup_string(fslist, "name", &fsname) == 0);
12075367Sahrens (void) nvlist_lookup_uint64(fslist, "origin", &origin_guid);
12085367Sahrens
120912282STom.Erickson@Sun.COM if (origin_guid != 0) {
121012282STom.Erickson@Sun.COM nvlist_t *origin_nv = fsavl_find(sdd->fsavl,
121112282STom.Erickson@Sun.COM origin_guid, NULL);
121212282STom.Erickson@Sun.COM if (origin_nv != NULL &&
121312282STom.Erickson@Sun.COM nvlist_lookup_boolean(origin_nv,
121412282STom.Erickson@Sun.COM "sent") == ENOENT) {
121512282STom.Erickson@Sun.COM /*
121612282STom.Erickson@Sun.COM * origin has not been sent yet;
121712282STom.Erickson@Sun.COM * skip this clone.
121812282STom.Erickson@Sun.COM */
121912282STom.Erickson@Sun.COM needagain = B_TRUE;
122012282STom.Erickson@Sun.COM continue;
122112282STom.Erickson@Sun.COM }
12225367Sahrens }
12235367Sahrens
12245367Sahrens zhp = zfs_open(rzhp->zfs_hdl, fsname, ZFS_TYPE_DATASET);
12256476Sahl if (zhp == NULL)
12266476Sahl return (-1);
12275367Sahrens err = dump_filesystem(zhp, sdd);
12285367Sahrens VERIFY(nvlist_add_boolean(fslist, "sent") == 0);
12295367Sahrens progress = B_TRUE;
12305367Sahrens zfs_close(zhp);
12315367Sahrens if (err)
12325367Sahrens return (err);
12335367Sahrens }
12345367Sahrens if (needagain) {
12355367Sahrens assert(progress);
12365367Sahrens goto again;
12375367Sahrens }
12385367Sahrens return (0);
12395367Sahrens }
12405367Sahrens
12415367Sahrens /*
124210200SLori.Alt@Sun.COM * Generate a send stream for the dataset identified by the argument zhp.
124310200SLori.Alt@Sun.COM *
124410200SLori.Alt@Sun.COM * The content of the send stream is the snapshot identified by
124510200SLori.Alt@Sun.COM * 'tosnap'. Incremental streams are requested in two ways:
124610200SLori.Alt@Sun.COM * - from the snapshot identified by "fromsnap" (if non-null) or
124710200SLori.Alt@Sun.COM * - from the origin of the dataset identified by zhp, which must
124810200SLori.Alt@Sun.COM * be a clone. In this case, "fromsnap" is null and "fromorigin"
124910200SLori.Alt@Sun.COM * is TRUE.
125010200SLori.Alt@Sun.COM *
125110200SLori.Alt@Sun.COM * The send stream is recursive (i.e. dumps a hierarchy of snapshots) and
125211007SLori.Alt@Sun.COM * uses a special header (with a hdrtype field of DMU_COMPOUNDSTREAM)
125310200SLori.Alt@Sun.COM * if "replicate" is set. If "doall" is set, dump all the intermediate
125411007SLori.Alt@Sun.COM * snapshots. The DMU_COMPOUNDSTREAM header is used in the "doall"
125511022STom.Erickson@Sun.COM * case too. If "props" is set, send properties.
12565367Sahrens */
12575367Sahrens int
zfs_send(zfs_handle_t * zhp,const char * fromsnap,const char * tosnap,sendflags_t flags,int outfd,snapfilter_cb_t filter_func,void * cb_arg,nvlist_t ** debugnvp)12585367Sahrens zfs_send(zfs_handle_t *zhp, const char *fromsnap, const char *tosnap,
125911007SLori.Alt@Sun.COM sendflags_t flags, int outfd, snapfilter_cb_t filter_func,
126012296SLin.Ling@Sun.COM void *cb_arg, nvlist_t **debugnvp)
12615367Sahrens {
12625367Sahrens char errbuf[1024];
12635367Sahrens send_dump_data_t sdd = { 0 };
12645367Sahrens int err;
12655367Sahrens nvlist_t *fss = NULL;
12665367Sahrens avl_tree_t *fsavl = NULL;
126710342Schris.kirby@sun.com static uint64_t holdseq;
126810735Schris.kirby@sun.com int spa_version;
126910735Schris.kirby@sun.com boolean_t holdsnaps = B_FALSE;
127011007SLori.Alt@Sun.COM pthread_t tid;
127111007SLori.Alt@Sun.COM int pipefd[2];
127211007SLori.Alt@Sun.COM dedup_arg_t dda = { 0 };
127311007SLori.Alt@Sun.COM int featureflags = 0;
127411935SMark.Shellenbaum@Sun.COM
12755367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
12765367Sahrens "cannot send '%s'"), zhp->zfs_name);
12775367Sahrens
12785367Sahrens if (fromsnap && fromsnap[0] == '\0') {
12795367Sahrens zfs_error_aux(zhp->zfs_hdl, dgettext(TEXT_DOMAIN,
12805367Sahrens "zero-length incremental source"));
12815367Sahrens return (zfs_error(zhp->zfs_hdl, EZFS_NOENT, errbuf));
12825367Sahrens }
12835367Sahrens
1284*12786SChris.Kirby@oracle.com if (zhp->zfs_type == ZFS_TYPE_FILESYSTEM) {
1285*12786SChris.Kirby@oracle.com uint64_t version;
1286*12786SChris.Kirby@oracle.com version = zfs_prop_get_int(zhp, ZFS_PROP_VERSION);
1287*12786SChris.Kirby@oracle.com if (version >= ZPL_VERSION_SA) {
1288*12786SChris.Kirby@oracle.com featureflags |= DMU_BACKUP_FEATURE_SA_SPILL;
1289*12786SChris.Kirby@oracle.com }
1290*12786SChris.Kirby@oracle.com }
1291*12786SChris.Kirby@oracle.com
129210735Schris.kirby@sun.com if (zfs_spa_version(zhp, &spa_version) == 0 &&
1293*12786SChris.Kirby@oracle.com spa_version >= SPA_VERSION_USERREFS &&
1294*12786SChris.Kirby@oracle.com (flags.doall || flags.replicate))
129510735Schris.kirby@sun.com holdsnaps = B_TRUE;
129610735Schris.kirby@sun.com
129711007SLori.Alt@Sun.COM if (flags.dedup) {
129811381SLori.Alt@Sun.COM featureflags |= (DMU_BACKUP_FEATURE_DEDUP |
129911381SLori.Alt@Sun.COM DMU_BACKUP_FEATURE_DEDUPPROPS);
130011007SLori.Alt@Sun.COM if (err = pipe(pipefd)) {
130111007SLori.Alt@Sun.COM zfs_error_aux(zhp->zfs_hdl, strerror(errno));
130211007SLori.Alt@Sun.COM return (zfs_error(zhp->zfs_hdl, EZFS_PIPEFAILED,
130311007SLori.Alt@Sun.COM errbuf));
130411007SLori.Alt@Sun.COM }
130511007SLori.Alt@Sun.COM dda.outputfd = outfd;
130611007SLori.Alt@Sun.COM dda.inputfd = pipefd[1];
130711007SLori.Alt@Sun.COM dda.dedup_hdl = zhp->zfs_hdl;
130811007SLori.Alt@Sun.COM if (err = pthread_create(&tid, NULL, cksummer, &dda)) {
130911007SLori.Alt@Sun.COM (void) close(pipefd[0]);
131011007SLori.Alt@Sun.COM (void) close(pipefd[1]);
131111007SLori.Alt@Sun.COM zfs_error_aux(zhp->zfs_hdl, strerror(errno));
131211007SLori.Alt@Sun.COM return (zfs_error(zhp->zfs_hdl,
131311007SLori.Alt@Sun.COM EZFS_THREADCREATEFAILED, errbuf));
131411007SLori.Alt@Sun.COM }
131511007SLori.Alt@Sun.COM }
131611007SLori.Alt@Sun.COM
131711022STom.Erickson@Sun.COM if (flags.replicate || flags.doall || flags.props) {
13185367Sahrens dmu_replay_record_t drr = { 0 };
13195367Sahrens char *packbuf = NULL;
13205367Sahrens size_t buflen = 0;
13215367Sahrens zio_cksum_t zc = { 0 };
13225367Sahrens
132311022STom.Erickson@Sun.COM if (flags.replicate || flags.props) {
13245367Sahrens nvlist_t *hdrnv;
13255367Sahrens
13265367Sahrens VERIFY(0 == nvlist_alloc(&hdrnv, NV_UNIQUE_NAME, 0));
13275367Sahrens if (fromsnap) {
13285367Sahrens VERIFY(0 == nvlist_add_string(hdrnv,
13295367Sahrens "fromsnap", fromsnap));
13305367Sahrens }
13315367Sahrens VERIFY(0 == nvlist_add_string(hdrnv, "tosnap", tosnap));
133211022STom.Erickson@Sun.COM if (!flags.replicate) {
133311022STom.Erickson@Sun.COM VERIFY(0 == nvlist_add_boolean(hdrnv,
133411022STom.Erickson@Sun.COM "not_recursive"));
133511022STom.Erickson@Sun.COM }
13365367Sahrens
13375367Sahrens err = gather_nvlist(zhp->zfs_hdl, zhp->zfs_name,
133811022STom.Erickson@Sun.COM fromsnap, tosnap, flags.replicate, &fss, &fsavl);
133912527SChris.Kirby@oracle.com if (err)
134011007SLori.Alt@Sun.COM goto err_out;
13415367Sahrens VERIFY(0 == nvlist_add_nvlist(hdrnv, "fss", fss));
13425367Sahrens err = nvlist_pack(hdrnv, &packbuf, &buflen,
13435367Sahrens NV_ENCODE_XDR, 0);
134412296SLin.Ling@Sun.COM if (debugnvp)
134512296SLin.Ling@Sun.COM *debugnvp = hdrnv;
134612296SLin.Ling@Sun.COM else
134712296SLin.Ling@Sun.COM nvlist_free(hdrnv);
13485367Sahrens if (err) {
13495367Sahrens fsavl_destroy(fsavl);
13505367Sahrens nvlist_free(fss);
135111007SLori.Alt@Sun.COM goto stderr_out;
13525367Sahrens }
13535367Sahrens }
13545367Sahrens
13555367Sahrens /* write first begin record */
13565367Sahrens drr.drr_type = DRR_BEGIN;
13575367Sahrens drr.drr_u.drr_begin.drr_magic = DMU_BACKUP_MAGIC;
135811007SLori.Alt@Sun.COM DMU_SET_STREAM_HDRTYPE(drr.drr_u.drr_begin.drr_versioninfo,
135911007SLori.Alt@Sun.COM DMU_COMPOUNDSTREAM);
136011007SLori.Alt@Sun.COM DMU_SET_FEATUREFLAGS(drr.drr_u.drr_begin.drr_versioninfo,
136111007SLori.Alt@Sun.COM featureflags);
13625367Sahrens (void) snprintf(drr.drr_u.drr_begin.drr_toname,
13635367Sahrens sizeof (drr.drr_u.drr_begin.drr_toname),
13645367Sahrens "%s@%s", zhp->zfs_name, tosnap);
13655367Sahrens drr.drr_payloadlen = buflen;
136611007SLori.Alt@Sun.COM err = cksum_and_write(&drr, sizeof (drr), &zc, outfd);
13675367Sahrens
13685367Sahrens /* write header nvlist */
136911022STom.Erickson@Sun.COM if (err != -1 && packbuf != NULL) {
137011007SLori.Alt@Sun.COM err = cksum_and_write(packbuf, buflen, &zc, outfd);
13715367Sahrens }
13725367Sahrens free(packbuf);
13735367Sahrens if (err == -1) {
13745367Sahrens fsavl_destroy(fsavl);
13755367Sahrens nvlist_free(fss);
137611007SLori.Alt@Sun.COM err = errno;
137711007SLori.Alt@Sun.COM goto stderr_out;
13785367Sahrens }
13795367Sahrens
13805367Sahrens /* write end record */
13815367Sahrens if (err != -1) {
13825367Sahrens bzero(&drr, sizeof (drr));
13835367Sahrens drr.drr_type = DRR_END;
13845367Sahrens drr.drr_u.drr_end.drr_checksum = zc;
13855367Sahrens err = write(outfd, &drr, sizeof (drr));
13865367Sahrens if (err == -1) {
13875367Sahrens fsavl_destroy(fsavl);
13885367Sahrens nvlist_free(fss);
138911668SChris.Kirby@sun.com err = errno;
139011007SLori.Alt@Sun.COM goto stderr_out;
13915367Sahrens }
13925367Sahrens }
13935367Sahrens }
13945367Sahrens
13955367Sahrens /* dump each stream */
13965367Sahrens sdd.fromsnap = fromsnap;
13975367Sahrens sdd.tosnap = tosnap;
139811007SLori.Alt@Sun.COM if (flags.dedup)
139911007SLori.Alt@Sun.COM sdd.outfd = pipefd[0];
140011007SLori.Alt@Sun.COM else
140111007SLori.Alt@Sun.COM sdd.outfd = outfd;
140211007SLori.Alt@Sun.COM sdd.replicate = flags.replicate;
140311007SLori.Alt@Sun.COM sdd.doall = flags.doall;
140411007SLori.Alt@Sun.COM sdd.fromorigin = flags.fromorigin;
14055367Sahrens sdd.fss = fss;
14065367Sahrens sdd.fsavl = fsavl;
140711007SLori.Alt@Sun.COM sdd.verbose = flags.verbose;
140811007SLori.Alt@Sun.COM sdd.filter_cb = filter_func;
140911007SLori.Alt@Sun.COM sdd.filter_cb_arg = cb_arg;
141012296SLin.Ling@Sun.COM if (debugnvp)
141112296SLin.Ling@Sun.COM sdd.debugnv = *debugnvp;
1412*12786SChris.Kirby@oracle.com if (holdsnaps) {
1413*12786SChris.Kirby@oracle.com ++holdseq;
1414*12786SChris.Kirby@oracle.com (void) snprintf(sdd.holdtag, sizeof (sdd.holdtag),
1415*12786SChris.Kirby@oracle.com ".send-%d-%llu", getpid(), (u_longlong_t)holdseq);
1416*12786SChris.Kirby@oracle.com sdd.cleanup_fd = open(ZFS_DEV, O_RDWR|O_EXCL);
1417*12786SChris.Kirby@oracle.com if (sdd.cleanup_fd < 0) {
1418*12786SChris.Kirby@oracle.com err = errno;
1419*12786SChris.Kirby@oracle.com goto stderr_out;
1420*12786SChris.Kirby@oracle.com }
1421*12786SChris.Kirby@oracle.com } else {
1422*12786SChris.Kirby@oracle.com sdd.cleanup_fd = -1;
1423*12786SChris.Kirby@oracle.com }
14245367Sahrens err = dump_filesystems(zhp, &sdd);
14255367Sahrens fsavl_destroy(fsavl);
14265367Sahrens nvlist_free(fss);
14275367Sahrens
142811007SLori.Alt@Sun.COM if (flags.dedup) {
142911007SLori.Alt@Sun.COM (void) close(pipefd[0]);
143011007SLori.Alt@Sun.COM (void) pthread_join(tid, NULL);
143111007SLori.Alt@Sun.COM }
143211022STom.Erickson@Sun.COM
1433*12786SChris.Kirby@oracle.com if (sdd.cleanup_fd != -1) {
1434*12786SChris.Kirby@oracle.com VERIFY(0 == close(sdd.cleanup_fd));
1435*12786SChris.Kirby@oracle.com sdd.cleanup_fd = -1;
143612527SChris.Kirby@oracle.com }
143712527SChris.Kirby@oracle.com
143811022STom.Erickson@Sun.COM if (flags.replicate || flags.doall || flags.props) {
14395367Sahrens /*
14405367Sahrens * write final end record. NB: want to do this even if
14415367Sahrens * there was some error, because it might not be totally
14425367Sahrens * failed.
14435367Sahrens */
14445367Sahrens dmu_replay_record_t drr = { 0 };
14455367Sahrens drr.drr_type = DRR_END;
14465367Sahrens if (write(outfd, &drr, sizeof (drr)) == -1) {
14475367Sahrens return (zfs_standard_error(zhp->zfs_hdl,
14485367Sahrens errno, errbuf));
14495367Sahrens }
14505367Sahrens }
14515367Sahrens
14525367Sahrens return (err || sdd.err);
145311007SLori.Alt@Sun.COM
145411007SLori.Alt@Sun.COM stderr_out:
145511007SLori.Alt@Sun.COM err = zfs_standard_error(zhp->zfs_hdl, err, errbuf);
145611007SLori.Alt@Sun.COM err_out:
1457*12786SChris.Kirby@oracle.com if (sdd.cleanup_fd != -1)
1458*12786SChris.Kirby@oracle.com VERIFY(0 == close(sdd.cleanup_fd));
145911007SLori.Alt@Sun.COM if (flags.dedup) {
146011007SLori.Alt@Sun.COM (void) pthread_cancel(tid);
146111007SLori.Alt@Sun.COM (void) pthread_join(tid, NULL);
146211007SLori.Alt@Sun.COM (void) close(pipefd[0]);
146311007SLori.Alt@Sun.COM }
146411007SLori.Alt@Sun.COM return (err);
14655367Sahrens }
14665367Sahrens
14675367Sahrens /*
14685367Sahrens * Routines specific to "zfs recv"
14695367Sahrens */
14705367Sahrens
14715367Sahrens static int
recv_read(libzfs_handle_t * hdl,int fd,void * buf,int ilen,boolean_t byteswap,zio_cksum_t * zc)14725367Sahrens recv_read(libzfs_handle_t *hdl, int fd, void *buf, int ilen,
14735367Sahrens boolean_t byteswap, zio_cksum_t *zc)
14745367Sahrens {
14755367Sahrens char *cp = buf;
14765367Sahrens int rv;
14775367Sahrens int len = ilen;
14785367Sahrens
14795367Sahrens do {
14805367Sahrens rv = read(fd, cp, len);
14815367Sahrens cp += rv;
14825367Sahrens len -= rv;
14835367Sahrens } while (rv > 0);
14845367Sahrens
14855367Sahrens if (rv < 0 || len != 0) {
14865367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
14875367Sahrens "failed to read from stream"));
14885367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, dgettext(TEXT_DOMAIN,
14895367Sahrens "cannot receive")));
14905367Sahrens }
14915367Sahrens
14925367Sahrens if (zc) {
14935367Sahrens if (byteswap)
14945367Sahrens fletcher_4_incremental_byteswap(buf, ilen, zc);
14955367Sahrens else
14965367Sahrens fletcher_4_incremental_native(buf, ilen, zc);
14975367Sahrens }
14985367Sahrens return (0);
14995367Sahrens }
15005367Sahrens
15015367Sahrens static int
recv_read_nvlist(libzfs_handle_t * hdl,int fd,int len,nvlist_t ** nvp,boolean_t byteswap,zio_cksum_t * zc)15025367Sahrens recv_read_nvlist(libzfs_handle_t *hdl, int fd, int len, nvlist_t **nvp,
15035367Sahrens boolean_t byteswap, zio_cksum_t *zc)
15045367Sahrens {
15055367Sahrens char *buf;
15065367Sahrens int err;
15075367Sahrens
15085367Sahrens buf = zfs_alloc(hdl, len);
15095367Sahrens if (buf == NULL)
15105367Sahrens return (ENOMEM);
15115367Sahrens
15125367Sahrens err = recv_read(hdl, fd, buf, len, byteswap, zc);
15135367Sahrens if (err != 0) {
15145367Sahrens free(buf);
15155367Sahrens return (err);
15165367Sahrens }
15175367Sahrens
15185367Sahrens err = nvlist_unpack(buf, len, nvp, 0);
15195367Sahrens free(buf);
15205367Sahrens if (err != 0) {
15215367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
15225367Sahrens "stream (malformed nvlist)"));
15235367Sahrens return (EINVAL);
15245367Sahrens }
15255367Sahrens return (0);
15265367Sahrens }
15275367Sahrens
15285367Sahrens static int
recv_rename(libzfs_handle_t * hdl,const char * name,const char * tryname,int baselen,char * newname,recvflags_t flags)15295367Sahrens recv_rename(libzfs_handle_t *hdl, const char *name, const char *tryname,
15305367Sahrens int baselen, char *newname, recvflags_t flags)
15315367Sahrens {
15325367Sahrens static int seq;
15335367Sahrens zfs_cmd_t zc = { 0 };
15345367Sahrens int err;
15356047Sahrens prop_changelist_t *clp;
15366047Sahrens zfs_handle_t *zhp;
15375367Sahrens
15386047Sahrens zhp = zfs_open(hdl, name, ZFS_TYPE_DATASET);
15396047Sahrens if (zhp == NULL)
15406047Sahrens return (-1);
15417366STim.Haley@Sun.COM clp = changelist_gather(zhp, ZFS_PROP_NAME, 0,
15427366STim.Haley@Sun.COM flags.force ? MS_FORCE : 0);
15436047Sahrens zfs_close(zhp);
15446047Sahrens if (clp == NULL)
15456047Sahrens return (-1);
15466047Sahrens err = changelist_prefix(clp);
15476047Sahrens if (err)
15486047Sahrens return (err);
15495367Sahrens
155010242Schris.kirby@sun.com zc.zc_objset_type = DMU_OST_ZFS;
155110242Schris.kirby@sun.com (void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name));
155210242Schris.kirby@sun.com
15535367Sahrens if (tryname) {
15545367Sahrens (void) strcpy(newname, tryname);
15555367Sahrens
15565367Sahrens (void) strlcpy(zc.zc_value, tryname, sizeof (zc.zc_value));
15575367Sahrens
15585367Sahrens if (flags.verbose) {
15595367Sahrens (void) printf("attempting rename %s to %s\n",
15605367Sahrens zc.zc_name, zc.zc_value);
15615367Sahrens }
15625367Sahrens err = ioctl(hdl->libzfs_fd, ZFS_IOC_RENAME, &zc);
15636047Sahrens if (err == 0)
15645367Sahrens changelist_rename(clp, name, tryname);
15655367Sahrens } else {
15665367Sahrens err = ENOENT;
15675367Sahrens }
15685367Sahrens
15695367Sahrens if (err != 0 && strncmp(name+baselen, "recv-", 5) != 0) {
15705367Sahrens seq++;
15715367Sahrens
15725367Sahrens (void) strncpy(newname, name, baselen);
15735367Sahrens (void) snprintf(newname+baselen, ZFS_MAXNAMELEN-baselen,
15745367Sahrens "recv-%u-%u", getpid(), seq);
15755367Sahrens (void) strlcpy(zc.zc_value, newname, sizeof (zc.zc_value));
15765367Sahrens
15775367Sahrens if (flags.verbose) {
15785367Sahrens (void) printf("failed - trying rename %s to %s\n",
15795367Sahrens zc.zc_name, zc.zc_value);
15805367Sahrens }
15815367Sahrens err = ioctl(hdl->libzfs_fd, ZFS_IOC_RENAME, &zc);
15826047Sahrens if (err == 0)
15835367Sahrens changelist_rename(clp, name, newname);
15845367Sahrens if (err && flags.verbose) {
15855367Sahrens (void) printf("failed (%u) - "
15865367Sahrens "will try again on next pass\n", errno);
15875367Sahrens }
15885367Sahrens err = EAGAIN;
15895367Sahrens } else if (flags.verbose) {
15905367Sahrens if (err == 0)
15915367Sahrens (void) printf("success\n");
15925367Sahrens else
15935367Sahrens (void) printf("failed (%u)\n", errno);
15945367Sahrens }
15955367Sahrens
15966047Sahrens (void) changelist_postfix(clp);
15976047Sahrens changelist_free(clp);
15985367Sahrens
15995367Sahrens return (err);
16005367Sahrens }
16015367Sahrens
16025367Sahrens static int
recv_destroy(libzfs_handle_t * hdl,const char * name,int baselen,char * newname,recvflags_t flags)16035367Sahrens recv_destroy(libzfs_handle_t *hdl, const char *name, int baselen,
16045367Sahrens char *newname, recvflags_t flags)
16055367Sahrens {
16065367Sahrens zfs_cmd_t zc = { 0 };
16076047Sahrens int err = 0;
16086047Sahrens prop_changelist_t *clp;
16096047Sahrens zfs_handle_t *zhp;
161010242Schris.kirby@sun.com boolean_t defer = B_FALSE;
161110242Schris.kirby@sun.com int spa_version;
16126047Sahrens
16136047Sahrens zhp = zfs_open(hdl, name, ZFS_TYPE_DATASET);
16146047Sahrens if (zhp == NULL)
16156047Sahrens return (-1);
16167366STim.Haley@Sun.COM clp = changelist_gather(zhp, ZFS_PROP_NAME, 0,
16177366STim.Haley@Sun.COM flags.force ? MS_FORCE : 0);
161810242Schris.kirby@sun.com if (zfs_get_type(zhp) == ZFS_TYPE_SNAPSHOT &&
161910242Schris.kirby@sun.com zfs_spa_version(zhp, &spa_version) == 0 &&
162010242Schris.kirby@sun.com spa_version >= SPA_VERSION_USERREFS)
162110242Schris.kirby@sun.com defer = B_TRUE;
16226047Sahrens zfs_close(zhp);
16236047Sahrens if (clp == NULL)
16246047Sahrens return (-1);
16256047Sahrens err = changelist_prefix(clp);
16266047Sahrens if (err)
16276047Sahrens return (err);
16285367Sahrens
16295367Sahrens zc.zc_objset_type = DMU_OST_ZFS;
163010242Schris.kirby@sun.com zc.zc_defer_destroy = defer;
16315367Sahrens (void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name));
16325367Sahrens
16335367Sahrens if (flags.verbose)
16345367Sahrens (void) printf("attempting destroy %s\n", zc.zc_name);
16355367Sahrens err = ioctl(hdl->libzfs_fd, ZFS_IOC_DESTROY, &zc);
16366047Sahrens if (err == 0) {
16376047Sahrens if (flags.verbose)
16386047Sahrens (void) printf("success\n");
16396047Sahrens changelist_remove(clp, zc.zc_name);
16405367Sahrens }
16415367Sahrens
16426047Sahrens (void) changelist_postfix(clp);
16436047Sahrens changelist_free(clp);
16446047Sahrens
164510242Schris.kirby@sun.com /*
164611022STom.Erickson@Sun.COM * Deferred destroy might destroy the snapshot or only mark it to be
164711022STom.Erickson@Sun.COM * destroyed later, and it returns success in either case.
164810242Schris.kirby@sun.com */
164911022STom.Erickson@Sun.COM if (err != 0 || (defer && zfs_dataset_exists(hdl, name,
165011022STom.Erickson@Sun.COM ZFS_TYPE_SNAPSHOT))) {
16516047Sahrens err = recv_rename(hdl, name, NULL, baselen, newname, flags);
165211022STom.Erickson@Sun.COM }
16535367Sahrens
16545367Sahrens return (err);
16555367Sahrens }
16565367Sahrens
16575367Sahrens typedef struct guid_to_name_data {
16585367Sahrens uint64_t guid;
16595367Sahrens char *name;
16605367Sahrens } guid_to_name_data_t;
16615367Sahrens
16625367Sahrens static int
guid_to_name_cb(zfs_handle_t * zhp,void * arg)16635367Sahrens guid_to_name_cb(zfs_handle_t *zhp, void *arg)
16645367Sahrens {
16655367Sahrens guid_to_name_data_t *gtnd = arg;
16665367Sahrens int err;
16675367Sahrens
16685367Sahrens if (zhp->zfs_dmustats.dds_guid == gtnd->guid) {
16695367Sahrens (void) strcpy(gtnd->name, zhp->zfs_name);
167011515STom.Erickson@Sun.COM zfs_close(zhp);
16715367Sahrens return (EEXIST);
16725367Sahrens }
16735367Sahrens err = zfs_iter_children(zhp, guid_to_name_cb, gtnd);
16745367Sahrens zfs_close(zhp);
16755367Sahrens return (err);
16765367Sahrens }
16775367Sahrens
16785367Sahrens static int
guid_to_name(libzfs_handle_t * hdl,const char * parent,uint64_t guid,char * name)16795367Sahrens guid_to_name(libzfs_handle_t *hdl, const char *parent, uint64_t guid,
16805367Sahrens char *name)
16815367Sahrens {
16825367Sahrens /* exhaustive search all local snapshots */
16835367Sahrens guid_to_name_data_t gtnd;
16845367Sahrens int err = 0;
16855367Sahrens zfs_handle_t *zhp;
16865367Sahrens char *cp;
16875367Sahrens
16885367Sahrens gtnd.guid = guid;
16895367Sahrens gtnd.name = name;
16905367Sahrens
16915367Sahrens if (strchr(parent, '@') == NULL) {
16925367Sahrens zhp = make_dataset_handle(hdl, parent);
16935367Sahrens if (zhp != NULL) {
16945367Sahrens err = zfs_iter_children(zhp, guid_to_name_cb, >nd);
16955367Sahrens zfs_close(zhp);
16965367Sahrens if (err == EEXIST)
16975367Sahrens return (0);
16985367Sahrens }
16995367Sahrens }
17005367Sahrens
17015367Sahrens cp = strchr(parent, '/');
17025367Sahrens if (cp)
17035367Sahrens *cp = '\0';
17045367Sahrens zhp = make_dataset_handle(hdl, parent);
17055367Sahrens if (cp)
17065367Sahrens *cp = '/';
17075367Sahrens
17085367Sahrens if (zhp) {
17095367Sahrens err = zfs_iter_children(zhp, guid_to_name_cb, >nd);
17105367Sahrens zfs_close(zhp);
17115367Sahrens }
17125367Sahrens
17135367Sahrens return (err == EEXIST ? 0 : ENOENT);
17145367Sahrens
17155367Sahrens }
17165367Sahrens
17175367Sahrens /*
17185367Sahrens * Return true if dataset guid1 is created before guid2.
17195367Sahrens */
17206476Sahl static int
created_before(libzfs_handle_t * hdl,avl_tree_t * avl,uint64_t guid1,uint64_t guid2)17215367Sahrens created_before(libzfs_handle_t *hdl, avl_tree_t *avl,
17225367Sahrens uint64_t guid1, uint64_t guid2)
17235367Sahrens {
17245367Sahrens nvlist_t *nvfs;
17255367Sahrens char *fsname, *snapname;
17265367Sahrens char buf[ZFS_MAXNAMELEN];
17276476Sahl int rv;
17285367Sahrens zfs_node_t zn1, zn2;
17295367Sahrens
17305367Sahrens if (guid2 == 0)
17316476Sahl return (0);
17325367Sahrens if (guid1 == 0)
17336476Sahl return (1);
17345367Sahrens
17355367Sahrens nvfs = fsavl_find(avl, guid1, &snapname);
17365367Sahrens VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname));
17375367Sahrens (void) snprintf(buf, sizeof (buf), "%s@%s", fsname, snapname);
17385367Sahrens zn1.zn_handle = zfs_open(hdl, buf, ZFS_TYPE_SNAPSHOT);
17396476Sahl if (zn1.zn_handle == NULL)
17406476Sahl return (-1);
17415367Sahrens
17425367Sahrens nvfs = fsavl_find(avl, guid2, &snapname);
17435367Sahrens VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname));
17445367Sahrens (void) snprintf(buf, sizeof (buf), "%s@%s", fsname, snapname);
17455367Sahrens zn2.zn_handle = zfs_open(hdl, buf, ZFS_TYPE_SNAPSHOT);
17466476Sahl if (zn2.zn_handle == NULL) {
17476476Sahl zfs_close(zn2.zn_handle);
17486476Sahl return (-1);
17496476Sahl }
17505367Sahrens
17515367Sahrens rv = (zfs_snapshot_compare(&zn1, &zn2) == -1);
17525367Sahrens
17535367Sahrens zfs_close(zn1.zn_handle);
17545367Sahrens zfs_close(zn2.zn_handle);
17555367Sahrens
17565367Sahrens return (rv);
17575367Sahrens }
17585367Sahrens
17595367Sahrens static int
recv_incremental_replication(libzfs_handle_t * hdl,const char * tofs,recvflags_t flags,nvlist_t * stream_nv,avl_tree_t * stream_avl,nvlist_t * renamed)17605367Sahrens recv_incremental_replication(libzfs_handle_t *hdl, const char *tofs,
176111834STom.Erickson@Sun.COM recvflags_t flags, nvlist_t *stream_nv, avl_tree_t *stream_avl,
176211834STom.Erickson@Sun.COM nvlist_t *renamed)
17635367Sahrens {
17645367Sahrens nvlist_t *local_nv;
17655367Sahrens avl_tree_t *local_avl;
17665367Sahrens nvpair_t *fselem, *nextfselem;
176711834STom.Erickson@Sun.COM char *fromsnap;
17685367Sahrens char newname[ZFS_MAXNAMELEN];
17695367Sahrens int error;
177011022STom.Erickson@Sun.COM boolean_t needagain, progress, recursive;
177110299SPrabahar.Jeyaram@Sun.COM char *s1, *s2;
17725367Sahrens
17735367Sahrens VERIFY(0 == nvlist_lookup_string(stream_nv, "fromsnap", &fromsnap));
17745367Sahrens
177511022STom.Erickson@Sun.COM recursive = (nvlist_lookup_boolean(stream_nv, "not_recursive") ==
177611022STom.Erickson@Sun.COM ENOENT);
177711022STom.Erickson@Sun.COM
17785367Sahrens if (flags.dryrun)
17795367Sahrens return (0);
17805367Sahrens
17815367Sahrens again:
17825367Sahrens needagain = progress = B_FALSE;
17835367Sahrens
17845367Sahrens if ((error = gather_nvlist(hdl, tofs, fromsnap, NULL,
178511022STom.Erickson@Sun.COM recursive, &local_nv, &local_avl)) != 0)
17865367Sahrens return (error);
17875367Sahrens
17885367Sahrens /*
17895367Sahrens * Process deletes and renames
17905367Sahrens */
17915367Sahrens for (fselem = nvlist_next_nvpair(local_nv, NULL);
17925367Sahrens fselem; fselem = nextfselem) {
17935367Sahrens nvlist_t *nvfs, *snaps;
17945367Sahrens nvlist_t *stream_nvfs = NULL;
17955367Sahrens nvpair_t *snapelem, *nextsnapelem;
17965367Sahrens uint64_t fromguid = 0;
17975367Sahrens uint64_t originguid = 0;
17985367Sahrens uint64_t stream_originguid = 0;
17995367Sahrens uint64_t parent_fromsnap_guid, stream_parent_fromsnap_guid;
18005367Sahrens char *fsname, *stream_fsname;
18015367Sahrens
18025367Sahrens nextfselem = nvlist_next_nvpair(local_nv, fselem);
18035367Sahrens
18045367Sahrens VERIFY(0 == nvpair_value_nvlist(fselem, &nvfs));
18055367Sahrens VERIFY(0 == nvlist_lookup_nvlist(nvfs, "snaps", &snaps));
18065367Sahrens VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname));
18075367Sahrens VERIFY(0 == nvlist_lookup_uint64(nvfs, "parentfromsnap",
18085367Sahrens &parent_fromsnap_guid));
18095367Sahrens (void) nvlist_lookup_uint64(nvfs, "origin", &originguid);
18105367Sahrens
18115367Sahrens /*
18125367Sahrens * First find the stream's fs, so we can check for
18135367Sahrens * a different origin (due to "zfs promote")
18145367Sahrens */
18155367Sahrens for (snapelem = nvlist_next_nvpair(snaps, NULL);
18165367Sahrens snapelem; snapelem = nvlist_next_nvpair(snaps, snapelem)) {
18175367Sahrens uint64_t thisguid;
18185367Sahrens
18195367Sahrens VERIFY(0 == nvpair_value_uint64(snapelem, &thisguid));
18205367Sahrens stream_nvfs = fsavl_find(stream_avl, thisguid, NULL);
18215367Sahrens
18225367Sahrens if (stream_nvfs != NULL)
18235367Sahrens break;
18245367Sahrens }
18255367Sahrens
18265367Sahrens /* check for promote */
18275367Sahrens (void) nvlist_lookup_uint64(stream_nvfs, "origin",
18285367Sahrens &stream_originguid);
18295367Sahrens if (stream_nvfs && originguid != stream_originguid) {
18306476Sahl switch (created_before(hdl, local_avl,
18316476Sahl stream_originguid, originguid)) {
18326476Sahl case 1: {
18335367Sahrens /* promote it! */
18345367Sahrens zfs_cmd_t zc = { 0 };
18355367Sahrens nvlist_t *origin_nvfs;
18365367Sahrens char *origin_fsname;
18375367Sahrens
18385367Sahrens if (flags.verbose)
18395367Sahrens (void) printf("promoting %s\n", fsname);
18405367Sahrens
18415367Sahrens origin_nvfs = fsavl_find(local_avl, originguid,
18425367Sahrens NULL);
18435367Sahrens VERIFY(0 == nvlist_lookup_string(origin_nvfs,
18445367Sahrens "name", &origin_fsname));
18455367Sahrens (void) strlcpy(zc.zc_value, origin_fsname,
18465367Sahrens sizeof (zc.zc_value));
18475367Sahrens (void) strlcpy(zc.zc_name, fsname,
18485367Sahrens sizeof (zc.zc_name));
18495367Sahrens error = zfs_ioctl(hdl, ZFS_IOC_PROMOTE, &zc);
18505367Sahrens if (error == 0)
18515367Sahrens progress = B_TRUE;
18526476Sahl break;
18536476Sahl }
18546476Sahl default:
18556476Sahl break;
18566476Sahl case -1:
18576476Sahl fsavl_destroy(local_avl);
18586476Sahl nvlist_free(local_nv);
18596476Sahl return (-1);
18605367Sahrens }
18615367Sahrens /*
18625367Sahrens * We had/have the wrong origin, therefore our
18635367Sahrens * list of snapshots is wrong. Need to handle
18645367Sahrens * them on the next pass.
18655367Sahrens */
18665367Sahrens needagain = B_TRUE;
18675367Sahrens continue;
18685367Sahrens }
18695367Sahrens
18705367Sahrens for (snapelem = nvlist_next_nvpair(snaps, NULL);
18715367Sahrens snapelem; snapelem = nextsnapelem) {
18725367Sahrens uint64_t thisguid;
18735367Sahrens char *stream_snapname;
18747265Sahrens nvlist_t *found, *props;
18755367Sahrens
18765367Sahrens nextsnapelem = nvlist_next_nvpair(snaps, snapelem);
18775367Sahrens
18785367Sahrens VERIFY(0 == nvpair_value_uint64(snapelem, &thisguid));
18795367Sahrens found = fsavl_find(stream_avl, thisguid,
18805367Sahrens &stream_snapname);
18815367Sahrens
18825367Sahrens /* check for delete */
18835367Sahrens if (found == NULL) {
18845367Sahrens char name[ZFS_MAXNAMELEN];
18855367Sahrens
18865367Sahrens if (!flags.force)
18875367Sahrens continue;
18885367Sahrens
18895367Sahrens (void) snprintf(name, sizeof (name), "%s@%s",
18905367Sahrens fsname, nvpair_name(snapelem));
18915367Sahrens
18925367Sahrens error = recv_destroy(hdl, name,
18935367Sahrens strlen(fsname)+1, newname, flags);
18945367Sahrens if (error)
18955367Sahrens needagain = B_TRUE;
18965367Sahrens else
18975367Sahrens progress = B_TRUE;
18985367Sahrens continue;
18995367Sahrens }
19005367Sahrens
19015367Sahrens stream_nvfs = found;
19025367Sahrens
19037265Sahrens if (0 == nvlist_lookup_nvlist(stream_nvfs, "snapprops",
19047265Sahrens &props) && 0 == nvlist_lookup_nvlist(props,
19057265Sahrens stream_snapname, &props)) {
19067265Sahrens zfs_cmd_t zc = { 0 };
19077265Sahrens
190811022STom.Erickson@Sun.COM zc.zc_cookie = B_TRUE; /* received */
19097273Sahrens (void) snprintf(zc.zc_name, sizeof (zc.zc_name),
19107265Sahrens "%s@%s", fsname, nvpair_name(snapelem));
19117265Sahrens if (zcmd_write_src_nvlist(hdl, &zc,
19127265Sahrens props) == 0) {
19137265Sahrens (void) zfs_ioctl(hdl,
19147265Sahrens ZFS_IOC_SET_PROP, &zc);
19157265Sahrens zcmd_free_nvlists(&zc);
19167265Sahrens }
19177265Sahrens }
19187265Sahrens
19195367Sahrens /* check for different snapname */
19205367Sahrens if (strcmp(nvpair_name(snapelem),
19215367Sahrens stream_snapname) != 0) {
19225367Sahrens char name[ZFS_MAXNAMELEN];
19235367Sahrens char tryname[ZFS_MAXNAMELEN];
19245367Sahrens
19255367Sahrens (void) snprintf(name, sizeof (name), "%s@%s",
19265367Sahrens fsname, nvpair_name(snapelem));
19275367Sahrens (void) snprintf(tryname, sizeof (name), "%s@%s",
19285367Sahrens fsname, stream_snapname);
19295367Sahrens
19305367Sahrens error = recv_rename(hdl, name, tryname,
19315367Sahrens strlen(fsname)+1, newname, flags);
19325367Sahrens if (error)
19335367Sahrens needagain = B_TRUE;
19345367Sahrens else
19355367Sahrens progress = B_TRUE;
19365367Sahrens }
19375367Sahrens
19385367Sahrens if (strcmp(stream_snapname, fromsnap) == 0)
19395367Sahrens fromguid = thisguid;
19405367Sahrens }
19415367Sahrens
19425367Sahrens /* check for delete */
19435367Sahrens if (stream_nvfs == NULL) {
19445367Sahrens if (!flags.force)
19455367Sahrens continue;
19465367Sahrens
19475367Sahrens error = recv_destroy(hdl, fsname, strlen(tofs)+1,
19485367Sahrens newname, flags);
19495367Sahrens if (error)
19505367Sahrens needagain = B_TRUE;
19515367Sahrens else
19525367Sahrens progress = B_TRUE;
19535367Sahrens continue;
19545367Sahrens }
19555367Sahrens
195611834STom.Erickson@Sun.COM if (fromguid == 0) {
195711834STom.Erickson@Sun.COM if (flags.verbose) {
195811834STom.Erickson@Sun.COM (void) printf("local fs %s does not have "
195911834STom.Erickson@Sun.COM "fromsnap (%s in stream); must have "
196011834STom.Erickson@Sun.COM "been deleted locally; ignoring\n",
196111834STom.Erickson@Sun.COM fsname, fromsnap);
196211834STom.Erickson@Sun.COM }
19635367Sahrens continue;
19645367Sahrens }
19655367Sahrens
19665367Sahrens VERIFY(0 == nvlist_lookup_string(stream_nvfs,
19675367Sahrens "name", &stream_fsname));
19685367Sahrens VERIFY(0 == nvlist_lookup_uint64(stream_nvfs,
19695367Sahrens "parentfromsnap", &stream_parent_fromsnap_guid));
19705367Sahrens
197110302SPrabahar.Jeyaram@Sun.COM s1 = strrchr(fsname, '/');
197210302SPrabahar.Jeyaram@Sun.COM s2 = strrchr(stream_fsname, '/');
197310302SPrabahar.Jeyaram@Sun.COM
197411834STom.Erickson@Sun.COM /*
197511834STom.Erickson@Sun.COM * Check for rename. If the exact receive path is specified, it
197611834STom.Erickson@Sun.COM * does not count as a rename, but we still need to check the
197711834STom.Erickson@Sun.COM * datasets beneath it.
197811834STom.Erickson@Sun.COM */
19795367Sahrens if ((stream_parent_fromsnap_guid != 0 &&
198011834STom.Erickson@Sun.COM parent_fromsnap_guid != 0 &&
19815367Sahrens stream_parent_fromsnap_guid != parent_fromsnap_guid) ||
198211834STom.Erickson@Sun.COM ((flags.isprefix || strcmp(tofs, fsname) != 0) &&
198311834STom.Erickson@Sun.COM (s1 != NULL) && (s2 != NULL) && strcmp(s1, s2) != 0)) {
19845367Sahrens nvlist_t *parent;
19855367Sahrens char tryname[ZFS_MAXNAMELEN];
19865367Sahrens
19875367Sahrens parent = fsavl_find(local_avl,
19885367Sahrens stream_parent_fromsnap_guid, NULL);
19895367Sahrens /*
19905367Sahrens * NB: parent might not be found if we used the
19915367Sahrens * tosnap for stream_parent_fromsnap_guid,
19925367Sahrens * because the parent is a newly-created fs;
19935367Sahrens * we'll be able to rename it after we recv the
19945367Sahrens * new fs.
19955367Sahrens */
19965367Sahrens if (parent != NULL) {
19975367Sahrens char *pname;
19985367Sahrens
19995367Sahrens VERIFY(0 == nvlist_lookup_string(parent, "name",
20005367Sahrens &pname));
20015367Sahrens (void) snprintf(tryname, sizeof (tryname),
20025367Sahrens "%s%s", pname, strrchr(stream_fsname, '/'));
20035367Sahrens } else {
20045367Sahrens tryname[0] = '\0';
20055367Sahrens if (flags.verbose) {
20065367Sahrens (void) printf("local fs %s new parent "
20075367Sahrens "not found\n", fsname);
20085367Sahrens }
20095367Sahrens }
20105367Sahrens
201111834STom.Erickson@Sun.COM newname[0] = '\0';
201211834STom.Erickson@Sun.COM
20135367Sahrens error = recv_rename(hdl, fsname, tryname,
20145367Sahrens strlen(tofs)+1, newname, flags);
201511834STom.Erickson@Sun.COM
201611834STom.Erickson@Sun.COM if (renamed != NULL && newname[0] != '\0') {
201711834STom.Erickson@Sun.COM VERIFY(0 == nvlist_add_boolean(renamed,
201811834STom.Erickson@Sun.COM newname));
201911834STom.Erickson@Sun.COM }
202011834STom.Erickson@Sun.COM
20215367Sahrens if (error)
20225367Sahrens needagain = B_TRUE;
20235367Sahrens else
20245367Sahrens progress = B_TRUE;
20255367Sahrens }
20265367Sahrens }
20275367Sahrens
20285367Sahrens fsavl_destroy(local_avl);
20295367Sahrens nvlist_free(local_nv);
20305367Sahrens
20315367Sahrens if (needagain && progress) {
20325367Sahrens /* do another pass to fix up temporary names */
20335367Sahrens if (flags.verbose)
20345367Sahrens (void) printf("another pass:\n");
20355367Sahrens goto again;
20365367Sahrens }
20375367Sahrens
20385367Sahrens return (needagain);
20395367Sahrens }
20405367Sahrens
20415367Sahrens static int
zfs_receive_package(libzfs_handle_t * hdl,int fd,const char * destname,recvflags_t flags,dmu_replay_record_t * drr,zio_cksum_t * zc,char ** top_zfs,int cleanup_fd,uint64_t * action_handlep)20425367Sahrens zfs_receive_package(libzfs_handle_t *hdl, int fd, const char *destname,
20437366STim.Haley@Sun.COM recvflags_t flags, dmu_replay_record_t *drr, zio_cksum_t *zc,
204412527SChris.Kirby@oracle.com char **top_zfs, int cleanup_fd, uint64_t *action_handlep)
20455367Sahrens {
20465367Sahrens nvlist_t *stream_nv = NULL;
20475367Sahrens avl_tree_t *stream_avl = NULL;
20485367Sahrens char *fromsnap = NULL;
204911834STom.Erickson@Sun.COM char *cp;
20505367Sahrens char tofs[ZFS_MAXNAMELEN];
205111834STom.Erickson@Sun.COM char sendfs[ZFS_MAXNAMELEN];
20525367Sahrens char errbuf[1024];
20535367Sahrens dmu_replay_record_t drre;
20545367Sahrens int error;
20555367Sahrens boolean_t anyerr = B_FALSE;
20566479Sahrens boolean_t softerr = B_FALSE;
205711834STom.Erickson@Sun.COM boolean_t recursive;
20585367Sahrens
20595367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
20605367Sahrens "cannot receive"));
20615367Sahrens
20625367Sahrens assert(drr->drr_type == DRR_BEGIN);
20635367Sahrens assert(drr->drr_u.drr_begin.drr_magic == DMU_BACKUP_MAGIC);
206411007SLori.Alt@Sun.COM assert(DMU_GET_STREAM_HDRTYPE(drr->drr_u.drr_begin.drr_versioninfo) ==
206511007SLori.Alt@Sun.COM DMU_COMPOUNDSTREAM);
20665367Sahrens
20675367Sahrens /*
20685367Sahrens * Read in the nvlist from the stream.
20695367Sahrens */
20705367Sahrens if (drr->drr_payloadlen != 0) {
20715367Sahrens error = recv_read_nvlist(hdl, fd, drr->drr_payloadlen,
20725367Sahrens &stream_nv, flags.byteswap, zc);
20735367Sahrens if (error) {
20745367Sahrens error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
20755367Sahrens goto out;
20765367Sahrens }
20775367Sahrens }
20785367Sahrens
207911834STom.Erickson@Sun.COM recursive = (nvlist_lookup_boolean(stream_nv, "not_recursive") ==
208011834STom.Erickson@Sun.COM ENOENT);
208111834STom.Erickson@Sun.COM
208211834STom.Erickson@Sun.COM if (recursive && strchr(destname, '@')) {
208311834STom.Erickson@Sun.COM zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
208411834STom.Erickson@Sun.COM "cannot specify snapshot name for multi-snapshot stream"));
208511834STom.Erickson@Sun.COM error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
208611834STom.Erickson@Sun.COM goto out;
208711834STom.Erickson@Sun.COM }
208811834STom.Erickson@Sun.COM
20895367Sahrens /*
20905367Sahrens * Read in the end record and verify checksum.
20915367Sahrens */
20925367Sahrens if (0 != (error = recv_read(hdl, fd, &drre, sizeof (drre),
20935367Sahrens flags.byteswap, NULL)))
20945367Sahrens goto out;
20955367Sahrens if (flags.byteswap) {
20965367Sahrens drre.drr_type = BSWAP_32(drre.drr_type);
20975367Sahrens drre.drr_u.drr_end.drr_checksum.zc_word[0] =
20985367Sahrens BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[0]);
20995367Sahrens drre.drr_u.drr_end.drr_checksum.zc_word[1] =
21005367Sahrens BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[1]);
21015367Sahrens drre.drr_u.drr_end.drr_checksum.zc_word[2] =
21025367Sahrens BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[2]);
21035367Sahrens drre.drr_u.drr_end.drr_checksum.zc_word[3] =
21045367Sahrens BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[3]);
21055367Sahrens }
21065367Sahrens if (drre.drr_type != DRR_END) {
21075367Sahrens error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
21085367Sahrens goto out;
21095367Sahrens }
21105367Sahrens if (!ZIO_CHECKSUM_EQUAL(drre.drr_u.drr_end.drr_checksum, *zc)) {
21115367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
21125367Sahrens "incorrect header checksum"));
21135367Sahrens error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
21145367Sahrens goto out;
21155367Sahrens }
21165367Sahrens
21175367Sahrens (void) nvlist_lookup_string(stream_nv, "fromsnap", &fromsnap);
21185367Sahrens
21195367Sahrens if (drr->drr_payloadlen != 0) {
21205367Sahrens nvlist_t *stream_fss;
21215367Sahrens
21225367Sahrens VERIFY(0 == nvlist_lookup_nvlist(stream_nv, "fss",
21235367Sahrens &stream_fss));
21245896Sahl if ((stream_avl = fsavl_create(stream_fss)) == NULL) {
21255896Sahl zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
21265896Sahl "couldn't allocate avl tree"));
21275896Sahl error = zfs_error(hdl, EZFS_NOMEM, errbuf);
21285896Sahl goto out;
21295896Sahl }
21305367Sahrens
21315367Sahrens if (fromsnap != NULL) {
213211834STom.Erickson@Sun.COM nvlist_t *renamed = NULL;
213311834STom.Erickson@Sun.COM nvpair_t *pair = NULL;
213411834STom.Erickson@Sun.COM
21355367Sahrens (void) strlcpy(tofs, destname, ZFS_MAXNAMELEN);
21365367Sahrens if (flags.isprefix) {
213711834STom.Erickson@Sun.COM struct drr_begin *drrb = &drr->drr_u.drr_begin;
213811834STom.Erickson@Sun.COM int i;
213911834STom.Erickson@Sun.COM
214011834STom.Erickson@Sun.COM if (flags.istail) {
214111834STom.Erickson@Sun.COM cp = strrchr(drrb->drr_toname, '/');
214211834STom.Erickson@Sun.COM if (cp == NULL) {
214311834STom.Erickson@Sun.COM (void) strlcat(tofs, "/",
214411834STom.Erickson@Sun.COM ZFS_MAXNAMELEN);
214511834STom.Erickson@Sun.COM i = 0;
214611834STom.Erickson@Sun.COM } else {
214711834STom.Erickson@Sun.COM i = (cp - drrb->drr_toname);
214811834STom.Erickson@Sun.COM }
214911834STom.Erickson@Sun.COM } else {
215011834STom.Erickson@Sun.COM i = strcspn(drrb->drr_toname, "/@");
215111834STom.Erickson@Sun.COM }
21525367Sahrens /* zfs_receive_one() will create_parents() */
215311834STom.Erickson@Sun.COM (void) strlcat(tofs, &drrb->drr_toname[i],
21545367Sahrens ZFS_MAXNAMELEN);
21555367Sahrens *strchr(tofs, '@') = '\0';
21565367Sahrens }
215711834STom.Erickson@Sun.COM
215811834STom.Erickson@Sun.COM if (recursive && !flags.dryrun && !flags.nomount) {
215911834STom.Erickson@Sun.COM VERIFY(0 == nvlist_alloc(&renamed,
216011834STom.Erickson@Sun.COM NV_UNIQUE_NAME, 0));
216111834STom.Erickson@Sun.COM }
216211834STom.Erickson@Sun.COM
216311834STom.Erickson@Sun.COM softerr = recv_incremental_replication(hdl, tofs, flags,
216411834STom.Erickson@Sun.COM stream_nv, stream_avl, renamed);
216511834STom.Erickson@Sun.COM
216611834STom.Erickson@Sun.COM /* Unmount renamed filesystems before receiving. */
216711834STom.Erickson@Sun.COM while ((pair = nvlist_next_nvpair(renamed,
216811834STom.Erickson@Sun.COM pair)) != NULL) {
216911834STom.Erickson@Sun.COM zfs_handle_t *zhp;
217011834STom.Erickson@Sun.COM prop_changelist_t *clp = NULL;
217111834STom.Erickson@Sun.COM
217211834STom.Erickson@Sun.COM zhp = zfs_open(hdl, nvpair_name(pair),
217311834STom.Erickson@Sun.COM ZFS_TYPE_FILESYSTEM);
217411834STom.Erickson@Sun.COM if (zhp != NULL) {
217511834STom.Erickson@Sun.COM clp = changelist_gather(zhp,
217611834STom.Erickson@Sun.COM ZFS_PROP_MOUNTPOINT, 0, 0);
217711834STom.Erickson@Sun.COM zfs_close(zhp);
217811834STom.Erickson@Sun.COM if (clp != NULL) {
217911834STom.Erickson@Sun.COM softerr |=
218011834STom.Erickson@Sun.COM changelist_prefix(clp);
218111834STom.Erickson@Sun.COM changelist_free(clp);
218211834STom.Erickson@Sun.COM }
218311834STom.Erickson@Sun.COM }
218411834STom.Erickson@Sun.COM }
218511834STom.Erickson@Sun.COM
218611834STom.Erickson@Sun.COM nvlist_free(renamed);
21875367Sahrens }
21885367Sahrens }
21895367Sahrens
219011834STom.Erickson@Sun.COM /*
219111834STom.Erickson@Sun.COM * Get the fs specified by the first path in the stream (the top level
219211834STom.Erickson@Sun.COM * specified by 'zfs send') and pass it to each invocation of
219311834STom.Erickson@Sun.COM * zfs_receive_one().
219411834STom.Erickson@Sun.COM */
219511834STom.Erickson@Sun.COM (void) strlcpy(sendfs, drr->drr_u.drr_begin.drr_toname,
219611834STom.Erickson@Sun.COM ZFS_MAXNAMELEN);
219711834STom.Erickson@Sun.COM if ((cp = strchr(sendfs, '@')) != NULL)
219811834STom.Erickson@Sun.COM *cp = '\0';
21995367Sahrens
22005367Sahrens /* Finally, receive each contained stream */
22015367Sahrens do {
22025367Sahrens /*
22035367Sahrens * we should figure out if it has a recoverable
22045367Sahrens * error, in which case do a recv_skip() and drive on.
22055367Sahrens * Note, if we fail due to already having this guid,
22065367Sahrens * zfs_receive_one() will take care of it (ie,
22075367Sahrens * recv_skip() and return 0).
22085367Sahrens */
22097366STim.Haley@Sun.COM error = zfs_receive_impl(hdl, destname, flags, fd,
221012527SChris.Kirby@oracle.com sendfs, stream_nv, stream_avl, top_zfs, cleanup_fd,
221112527SChris.Kirby@oracle.com action_handlep);
22125367Sahrens if (error == ENODATA) {
22135367Sahrens error = 0;
22145367Sahrens break;
22155367Sahrens }
22165367Sahrens anyerr |= error;
22175367Sahrens } while (error == 0);
22185367Sahrens
22195367Sahrens if (drr->drr_payloadlen != 0 && fromsnap != NULL) {
22205367Sahrens /*
22215367Sahrens * Now that we have the fs's they sent us, try the
22225367Sahrens * renames again.
22235367Sahrens */
22246479Sahrens softerr = recv_incremental_replication(hdl, tofs, flags,
222511834STom.Erickson@Sun.COM stream_nv, stream_avl, NULL);
22265367Sahrens }
22275367Sahrens
22285367Sahrens out:
22295367Sahrens fsavl_destroy(stream_avl);
22305367Sahrens if (stream_nv)
22315367Sahrens nvlist_free(stream_nv);
22326479Sahrens if (softerr)
22336479Sahrens error = -2;
22345367Sahrens if (anyerr)
22355367Sahrens error = -1;
22365367Sahrens return (error);
22375367Sahrens }
22385367Sahrens
223911022STom.Erickson@Sun.COM static void
trunc_prop_errs(int truncated)224011022STom.Erickson@Sun.COM trunc_prop_errs(int truncated)
224111022STom.Erickson@Sun.COM {
224211022STom.Erickson@Sun.COM ASSERT(truncated != 0);
224311022STom.Erickson@Sun.COM
224411022STom.Erickson@Sun.COM if (truncated == 1)
224511022STom.Erickson@Sun.COM (void) fprintf(stderr, dgettext(TEXT_DOMAIN,
224611022STom.Erickson@Sun.COM "1 more property could not be set\n"));
224711022STom.Erickson@Sun.COM else
224811022STom.Erickson@Sun.COM (void) fprintf(stderr, dgettext(TEXT_DOMAIN,
224911022STom.Erickson@Sun.COM "%d more properties could not be set\n"), truncated);
225011022STom.Erickson@Sun.COM }
225111022STom.Erickson@Sun.COM
22525367Sahrens static int
recv_skip(libzfs_handle_t * hdl,int fd,boolean_t byteswap)22535367Sahrens recv_skip(libzfs_handle_t *hdl, int fd, boolean_t byteswap)
22545367Sahrens {
22555367Sahrens dmu_replay_record_t *drr;
22565367Sahrens void *buf = malloc(1<<20);
225711007SLori.Alt@Sun.COM char errbuf[1024];
225811007SLori.Alt@Sun.COM
225911007SLori.Alt@Sun.COM (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
226011007SLori.Alt@Sun.COM "cannot receive:"));
22615367Sahrens
22625367Sahrens /* XXX would be great to use lseek if possible... */
22635367Sahrens drr = buf;
22645367Sahrens
22655367Sahrens while (recv_read(hdl, fd, drr, sizeof (dmu_replay_record_t),
22665367Sahrens byteswap, NULL) == 0) {
22675367Sahrens if (byteswap)
22685367Sahrens drr->drr_type = BSWAP_32(drr->drr_type);
22695367Sahrens
22705367Sahrens switch (drr->drr_type) {
22715367Sahrens case DRR_BEGIN:
22725367Sahrens /* NB: not to be used on v2 stream packages */
227311007SLori.Alt@Sun.COM if (drr->drr_payloadlen != 0) {
227411007SLori.Alt@Sun.COM zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
227511007SLori.Alt@Sun.COM "invalid substream header"));
227611007SLori.Alt@Sun.COM return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
227711007SLori.Alt@Sun.COM }
22785367Sahrens break;
22795367Sahrens
22805367Sahrens case DRR_END:
22815367Sahrens free(buf);
22825367Sahrens return (0);
22835367Sahrens
22845367Sahrens case DRR_OBJECT:
22855367Sahrens if (byteswap) {
22865367Sahrens drr->drr_u.drr_object.drr_bonuslen =
22875367Sahrens BSWAP_32(drr->drr_u.drr_object.
22885367Sahrens drr_bonuslen);
22895367Sahrens }
22905367Sahrens (void) recv_read(hdl, fd, buf,
22915367Sahrens P2ROUNDUP(drr->drr_u.drr_object.drr_bonuslen, 8),
22925367Sahrens B_FALSE, NULL);
22935367Sahrens break;
22945367Sahrens
22955367Sahrens case DRR_WRITE:
22965367Sahrens if (byteswap) {
22975367Sahrens drr->drr_u.drr_write.drr_length =
22987366STim.Haley@Sun.COM BSWAP_64(drr->drr_u.drr_write.drr_length);
22995367Sahrens }
23005367Sahrens (void) recv_read(hdl, fd, buf,
23015367Sahrens drr->drr_u.drr_write.drr_length, B_FALSE, NULL);
23025367Sahrens break;
230311935SMark.Shellenbaum@Sun.COM case DRR_SPILL:
230411935SMark.Shellenbaum@Sun.COM if (byteswap) {
230511935SMark.Shellenbaum@Sun.COM drr->drr_u.drr_write.drr_length =
230611935SMark.Shellenbaum@Sun.COM BSWAP_64(drr->drr_u.drr_spill.drr_length);
230711935SMark.Shellenbaum@Sun.COM }
230811935SMark.Shellenbaum@Sun.COM (void) recv_read(hdl, fd, buf,
230911935SMark.Shellenbaum@Sun.COM drr->drr_u.drr_spill.drr_length, B_FALSE, NULL);
231011935SMark.Shellenbaum@Sun.COM break;
231111007SLori.Alt@Sun.COM case DRR_WRITE_BYREF:
23125367Sahrens case DRR_FREEOBJECTS:
23135367Sahrens case DRR_FREE:
23145367Sahrens break;
23155367Sahrens
23165367Sahrens default:
231711007SLori.Alt@Sun.COM zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
231811007SLori.Alt@Sun.COM "invalid record type"));
231911007SLori.Alt@Sun.COM return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
23205367Sahrens }
23215367Sahrens }
23225367Sahrens
23235367Sahrens free(buf);
23245367Sahrens return (-1);
23255367Sahrens }
23265367Sahrens
23275367Sahrens /*
23285367Sahrens * Restores a backup of tosnap from the file descriptor specified by infd.
23295367Sahrens */
23305367Sahrens static int
zfs_receive_one(libzfs_handle_t * hdl,int infd,const char * tosnap,recvflags_t flags,dmu_replay_record_t * drr,dmu_replay_record_t * drr_noswap,const char * sendfs,nvlist_t * stream_nv,avl_tree_t * stream_avl,char ** top_zfs,int cleanup_fd,uint64_t * action_handlep)23315367Sahrens zfs_receive_one(libzfs_handle_t *hdl, int infd, const char *tosnap,
23325367Sahrens recvflags_t flags, dmu_replay_record_t *drr,
233311834STom.Erickson@Sun.COM dmu_replay_record_t *drr_noswap, const char *sendfs,
233412527SChris.Kirby@oracle.com nvlist_t *stream_nv, avl_tree_t *stream_avl, char **top_zfs, int cleanup_fd,
233512527SChris.Kirby@oracle.com uint64_t *action_handlep)
23365367Sahrens {
23375367Sahrens zfs_cmd_t zc = { 0 };
23385367Sahrens time_t begin_time;
233911834STom.Erickson@Sun.COM int ioctl_err, ioctl_errno, err;
23405367Sahrens char *cp;
23415367Sahrens struct drr_begin *drrb = &drr->drr_u.drr_begin;
23425367Sahrens char errbuf[1024];
234311022STom.Erickson@Sun.COM char prop_errbuf[1024];
234411834STom.Erickson@Sun.COM const char *chopprefix;
23455367Sahrens boolean_t newfs = B_FALSE;
23465367Sahrens boolean_t stream_wantsnewfs;
23475367Sahrens uint64_t parent_snapguid = 0;
23485367Sahrens prop_changelist_t *clp = NULL;
23497265Sahrens nvlist_t *snapprops_nvlist = NULL;
235011022STom.Erickson@Sun.COM zprop_errflags_t prop_errflags;
235111834STom.Erickson@Sun.COM boolean_t recursive;
23525367Sahrens
23535367Sahrens begin_time = time(NULL);
23545367Sahrens
23555367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
23565367Sahrens "cannot receive"));
23575367Sahrens
235811834STom.Erickson@Sun.COM recursive = (nvlist_lookup_boolean(stream_nv, "not_recursive") ==
235911834STom.Erickson@Sun.COM ENOENT);
236011834STom.Erickson@Sun.COM
23615367Sahrens if (stream_avl != NULL) {
23627265Sahrens char *snapname;
23637265Sahrens nvlist_t *fs = fsavl_find(stream_avl, drrb->drr_toguid,
23647265Sahrens &snapname);
23655367Sahrens nvlist_t *props;
23665896Sahl int ret;
23675367Sahrens
23685367Sahrens (void) nvlist_lookup_uint64(fs, "parentfromsnap",
23695367Sahrens &parent_snapguid);
23705367Sahrens err = nvlist_lookup_nvlist(fs, "props", &props);
23715367Sahrens if (err)
23725367Sahrens VERIFY(0 == nvlist_alloc(&props, NV_UNIQUE_NAME, 0));
23735896Sahl
23745367Sahrens if (flags.canmountoff) {
23755367Sahrens VERIFY(0 == nvlist_add_uint64(props,
23765367Sahrens zfs_prop_to_name(ZFS_PROP_CANMOUNT), 0));
23775367Sahrens }
23785896Sahl ret = zcmd_write_src_nvlist(hdl, &zc, props);
23795367Sahrens if (err)
23805367Sahrens nvlist_free(props);
23815896Sahl
23827265Sahrens if (0 == nvlist_lookup_nvlist(fs, "snapprops", &props)) {
23837265Sahrens VERIFY(0 == nvlist_lookup_nvlist(props,
23847265Sahrens snapname, &snapprops_nvlist));
23857265Sahrens }
23867265Sahrens
23875896Sahl if (ret != 0)
23885896Sahl return (-1);
23895367Sahrens }
23905367Sahrens
239111834STom.Erickson@Sun.COM cp = NULL;
239211834STom.Erickson@Sun.COM
23935367Sahrens /*
23945367Sahrens * Determine how much of the snapshot name stored in the stream
23955367Sahrens * we are going to tack on to the name they specified on the
23965367Sahrens * command line, and how much we are going to chop off.
23975367Sahrens *
23985367Sahrens * If they specified a snapshot, chop the entire name stored in
23995367Sahrens * the stream.
24005367Sahrens */
240111834STom.Erickson@Sun.COM if (flags.istail) {
24025367Sahrens /*
240311834STom.Erickson@Sun.COM * A filesystem was specified with -e. We want to tack on only
240411834STom.Erickson@Sun.COM * the tail of the sent snapshot path.
24055367Sahrens */
24065367Sahrens if (strchr(tosnap, '@')) {
24075367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
240811834STom.Erickson@Sun.COM "argument - snapshot not allowed with -e"));
24095367Sahrens return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf));
24105367Sahrens }
241111834STom.Erickson@Sun.COM
241211834STom.Erickson@Sun.COM chopprefix = strrchr(sendfs, '/');
241311834STom.Erickson@Sun.COM
241411834STom.Erickson@Sun.COM if (chopprefix == NULL) {
241511834STom.Erickson@Sun.COM /*
241611834STom.Erickson@Sun.COM * The tail is the poolname, so we need to
241711834STom.Erickson@Sun.COM * prepend a path separator.
241811834STom.Erickson@Sun.COM */
241911834STom.Erickson@Sun.COM int len = strlen(drrb->drr_toname);
242011834STom.Erickson@Sun.COM cp = malloc(len + 2);
242111834STom.Erickson@Sun.COM cp[0] = '/';
242211834STom.Erickson@Sun.COM (void) strcpy(&cp[1], drrb->drr_toname);
242311834STom.Erickson@Sun.COM chopprefix = cp;
242411834STom.Erickson@Sun.COM } else {
242511834STom.Erickson@Sun.COM chopprefix = drrb->drr_toname + (chopprefix - sendfs);
242611834STom.Erickson@Sun.COM }
242711834STom.Erickson@Sun.COM } else if (flags.isprefix) {
242811834STom.Erickson@Sun.COM /*
242911834STom.Erickson@Sun.COM * A filesystem was specified with -d. We want to tack on
243011834STom.Erickson@Sun.COM * everything but the first element of the sent snapshot path
243111834STom.Erickson@Sun.COM * (all but the pool name).
243211834STom.Erickson@Sun.COM */
243311834STom.Erickson@Sun.COM if (strchr(tosnap, '@')) {
243411834STom.Erickson@Sun.COM zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
243511834STom.Erickson@Sun.COM "argument - snapshot not allowed with -d"));
243611834STom.Erickson@Sun.COM return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf));
243711834STom.Erickson@Sun.COM }
243811834STom.Erickson@Sun.COM
243911834STom.Erickson@Sun.COM chopprefix = strchr(drrb->drr_toname, '/');
244011834STom.Erickson@Sun.COM if (chopprefix == NULL)
244111834STom.Erickson@Sun.COM chopprefix = strchr(drrb->drr_toname, '@');
24425367Sahrens } else if (strchr(tosnap, '@') == NULL) {
24435367Sahrens /*
244411834STom.Erickson@Sun.COM * If a filesystem was specified without -d or -e, we want to
244511834STom.Erickson@Sun.COM * tack on everything after the fs specified by 'zfs send'.
24465367Sahrens */
244711834STom.Erickson@Sun.COM chopprefix = drrb->drr_toname + strlen(sendfs);
244811834STom.Erickson@Sun.COM } else {
244911834STom.Erickson@Sun.COM /* A snapshot was specified as an exact path (no -d or -e). */
245011834STom.Erickson@Sun.COM if (recursive) {
245111834STom.Erickson@Sun.COM zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
245211834STom.Erickson@Sun.COM "cannot specify snapshot name for multi-snapshot "
245311834STom.Erickson@Sun.COM "stream"));
245411834STom.Erickson@Sun.COM return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
245511834STom.Erickson@Sun.COM }
245611834STom.Erickson@Sun.COM chopprefix = drrb->drr_toname + strlen(drrb->drr_toname);
24575367Sahrens }
245811834STom.Erickson@Sun.COM
245911834STom.Erickson@Sun.COM ASSERT(strstr(drrb->drr_toname, sendfs) == drrb->drr_toname);
246011834STom.Erickson@Sun.COM ASSERT(chopprefix > drrb->drr_toname);
246111834STom.Erickson@Sun.COM ASSERT(chopprefix <= drrb->drr_toname + strlen(drrb->drr_toname));
246211834STom.Erickson@Sun.COM ASSERT(chopprefix[0] == '/' || chopprefix[0] == '@' ||
246311834STom.Erickson@Sun.COM chopprefix[0] == '\0');
24645367Sahrens
24655367Sahrens /*
24665367Sahrens * Determine name of destination snapshot, store in zc_value.
24675367Sahrens */
246811007SLori.Alt@Sun.COM (void) strcpy(zc.zc_top_ds, tosnap);
24695367Sahrens (void) strcpy(zc.zc_value, tosnap);
247011834STom.Erickson@Sun.COM (void) strncat(zc.zc_value, chopprefix, sizeof (zc.zc_value));
247111834STom.Erickson@Sun.COM free(cp);
24725896Sahl if (!zfs_name_valid(zc.zc_value, ZFS_TYPE_SNAPSHOT)) {
24735896Sahl zcmd_free_nvlists(&zc);
24745367Sahrens return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf));
24755896Sahl }
24765367Sahrens
24775367Sahrens /*
24785367Sahrens * Determine the name of the origin snapshot, store in zc_string.
24795367Sahrens */
24805367Sahrens if (drrb->drr_flags & DRR_FLAG_CLONE) {
24815367Sahrens if (guid_to_name(hdl, tosnap,
24825367Sahrens drrb->drr_fromguid, zc.zc_string) != 0) {
24835896Sahl zcmd_free_nvlists(&zc);
24845367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
24855367Sahrens "local origin for clone %s does not exist"),
24865367Sahrens zc.zc_value);
24875367Sahrens return (zfs_error(hdl, EZFS_NOENT, errbuf));
24885367Sahrens }
24895367Sahrens if (flags.verbose)
24905367Sahrens (void) printf("found clone origin %s\n", zc.zc_string);
24915367Sahrens }
24925367Sahrens
24935367Sahrens stream_wantsnewfs = (drrb->drr_fromguid == NULL ||
24945367Sahrens (drrb->drr_flags & DRR_FLAG_CLONE));
24955367Sahrens
24965367Sahrens if (stream_wantsnewfs) {
24975367Sahrens /*
24985367Sahrens * if the parent fs does not exist, look for it based on
24995367Sahrens * the parent snap GUID
25005367Sahrens */
25015367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
25025367Sahrens "cannot receive new filesystem stream"));
25035367Sahrens
25045367Sahrens (void) strcpy(zc.zc_name, zc.zc_value);
25055367Sahrens cp = strrchr(zc.zc_name, '/');
25065367Sahrens if (cp)
25075367Sahrens *cp = '\0';
25085367Sahrens if (cp &&
25095367Sahrens !zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) {
25105367Sahrens char suffix[ZFS_MAXNAMELEN];
25115367Sahrens (void) strcpy(suffix, strrchr(zc.zc_value, '/'));
25125367Sahrens if (guid_to_name(hdl, tosnap, parent_snapguid,
25135367Sahrens zc.zc_value) == 0) {
25145367Sahrens *strchr(zc.zc_value, '@') = '\0';
25155367Sahrens (void) strcat(zc.zc_value, suffix);
25165367Sahrens }
25175367Sahrens }
25185367Sahrens } else {
25195367Sahrens /*
25205367Sahrens * if the fs does not exist, look for it based on the
25215367Sahrens * fromsnap GUID
25225367Sahrens */
25235367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
25245367Sahrens "cannot receive incremental stream"));
25255367Sahrens
25265367Sahrens (void) strcpy(zc.zc_name, zc.zc_value);
25275367Sahrens *strchr(zc.zc_name, '@') = '\0';
25285367Sahrens
252911976STom.Erickson@Sun.COM /*
253011976STom.Erickson@Sun.COM * If the exact receive path was specified and this is the
253111976STom.Erickson@Sun.COM * topmost path in the stream, then if the fs does not exist we
253211976STom.Erickson@Sun.COM * should look no further.
253311976STom.Erickson@Sun.COM */
253411976STom.Erickson@Sun.COM if ((flags.isprefix || (*(chopprefix = drrb->drr_toname +
253511976STom.Erickson@Sun.COM strlen(sendfs)) != '\0' && *chopprefix != '@')) &&
253611976STom.Erickson@Sun.COM !zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) {
25375367Sahrens char snap[ZFS_MAXNAMELEN];
25385367Sahrens (void) strcpy(snap, strchr(zc.zc_value, '@'));
25395367Sahrens if (guid_to_name(hdl, tosnap, drrb->drr_fromguid,
25405367Sahrens zc.zc_value) == 0) {
25415367Sahrens *strchr(zc.zc_value, '@') = '\0';
25425367Sahrens (void) strcat(zc.zc_value, snap);
25435367Sahrens }
25445367Sahrens }
25455367Sahrens }
25465367Sahrens
25475367Sahrens (void) strcpy(zc.zc_name, zc.zc_value);
25485367Sahrens *strchr(zc.zc_name, '@') = '\0';
25495367Sahrens
25505367Sahrens if (zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) {
25515367Sahrens zfs_handle_t *zhp;
255211834STom.Erickson@Sun.COM
25535367Sahrens /*
25545367Sahrens * Destination fs exists. Therefore this should either
25555367Sahrens * be an incremental, or the stream specifies a new fs
25565367Sahrens * (full stream or clone) and they want us to blow it
25575367Sahrens * away (and have therefore specified -F and removed any
25585367Sahrens * snapshots).
25595367Sahrens */
25605367Sahrens if (stream_wantsnewfs) {
25615367Sahrens if (!flags.force) {
25625896Sahl zcmd_free_nvlists(&zc);
25635367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
25645367Sahrens "destination '%s' exists\n"
25655367Sahrens "must specify -F to overwrite it"),
25665367Sahrens zc.zc_name);
25675367Sahrens return (zfs_error(hdl, EZFS_EXISTS, errbuf));
25685367Sahrens }
25695367Sahrens if (ioctl(hdl->libzfs_fd, ZFS_IOC_SNAPSHOT_LIST_NEXT,
25705367Sahrens &zc) == 0) {
25715896Sahl zcmd_free_nvlists(&zc);
25725367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
25735367Sahrens "destination has snapshots (eg. %s)\n"
25745367Sahrens "must destroy them to overwrite it"),
25755367Sahrens zc.zc_name);
25765367Sahrens return (zfs_error(hdl, EZFS_EXISTS, errbuf));
25775367Sahrens }
25785367Sahrens }
25795367Sahrens
25805896Sahl if ((zhp = zfs_open(hdl, zc.zc_name,
25815896Sahl ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME)) == NULL) {
25825896Sahl zcmd_free_nvlists(&zc);
25835367Sahrens return (-1);
25845896Sahl }
25855896Sahl
25865367Sahrens if (stream_wantsnewfs &&
25875367Sahrens zhp->zfs_dmustats.dds_origin[0]) {
25885896Sahl zcmd_free_nvlists(&zc);
25895367Sahrens zfs_close(zhp);
25905367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
25915367Sahrens "destination '%s' is a clone\n"
25925367Sahrens "must destroy it to overwrite it"),
25935367Sahrens zc.zc_name);
25945367Sahrens return (zfs_error(hdl, EZFS_EXISTS, errbuf));
25955367Sahrens }
25965367Sahrens
25975367Sahrens if (!flags.dryrun && zhp->zfs_type == ZFS_TYPE_FILESYSTEM &&
25985367Sahrens stream_wantsnewfs) {
25995367Sahrens /* We can't do online recv in this case */
26007366STim.Haley@Sun.COM clp = changelist_gather(zhp, ZFS_PROP_NAME, 0, 0);
26015896Sahl if (clp == NULL) {
260210200SLori.Alt@Sun.COM zfs_close(zhp);
26035896Sahl zcmd_free_nvlists(&zc);
26045367Sahrens return (-1);
26055896Sahl }
26065367Sahrens if (changelist_prefix(clp) != 0) {
26075367Sahrens changelist_free(clp);
260810200SLori.Alt@Sun.COM zfs_close(zhp);
26095896Sahl zcmd_free_nvlists(&zc);
26105367Sahrens return (-1);
26115367Sahrens }
26125367Sahrens }
26135367Sahrens zfs_close(zhp);
26145367Sahrens } else {
26155367Sahrens /*
26165896Sahl * Destination filesystem does not exist. Therefore we better
26175896Sahl * be creating a new filesystem (either from a full backup, or
26185896Sahl * a clone). It would therefore be invalid if the user
26195896Sahl * specified only the pool name (i.e. if the destination name
26205896Sahl * contained no slash character).
26215367Sahrens */
26225896Sahl if (!stream_wantsnewfs ||
26235896Sahl (cp = strrchr(zc.zc_name, '/')) == NULL) {
26245896Sahl zcmd_free_nvlists(&zc);
26255367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
26265367Sahrens "destination '%s' does not exist"), zc.zc_name);
26275367Sahrens return (zfs_error(hdl, EZFS_NOENT, errbuf));
26285367Sahrens }
26295367Sahrens
26305896Sahl /*
26315896Sahl * Trim off the final dataset component so we perform the
26325896Sahl * recvbackup ioctl to the filesystems's parent.
26335896Sahl */
26345896Sahl *cp = '\0';
26355367Sahrens
263611834STom.Erickson@Sun.COM if (flags.isprefix && !flags.istail && !flags.dryrun &&
26375896Sahl create_parents(hdl, zc.zc_value, strlen(tosnap)) != 0) {
26385896Sahl zcmd_free_nvlists(&zc);
26395896Sahl return (zfs_error(hdl, EZFS_BADRESTORE, errbuf));
26405367Sahrens }
26415367Sahrens
26425367Sahrens newfs = B_TRUE;
26435367Sahrens }
26445367Sahrens
26455367Sahrens zc.zc_begin_record = drr_noswap->drr_u.drr_begin;
26465367Sahrens zc.zc_cookie = infd;
26475367Sahrens zc.zc_guid = flags.force;
26485367Sahrens if (flags.verbose) {
26495367Sahrens (void) printf("%s %s stream of %s into %s\n",
26505367Sahrens flags.dryrun ? "would receive" : "receiving",
26515367Sahrens drrb->drr_fromguid ? "incremental" : "full",
26525367Sahrens drrb->drr_toname, zc.zc_value);
26535367Sahrens (void) fflush(stdout);
26545367Sahrens }
26555367Sahrens
26565896Sahl if (flags.dryrun) {
26575896Sahl zcmd_free_nvlists(&zc);
26585367Sahrens return (recv_skip(hdl, infd, flags.byteswap));
26595896Sahl }
26605367Sahrens
266111022STom.Erickson@Sun.COM zc.zc_nvlist_dst = (uint64_t)(uintptr_t)prop_errbuf;
266211022STom.Erickson@Sun.COM zc.zc_nvlist_dst_size = sizeof (prop_errbuf);
266312527SChris.Kirby@oracle.com zc.zc_cleanup_fd = cleanup_fd;
266412527SChris.Kirby@oracle.com zc.zc_action_handle = *action_handlep;
266511022STom.Erickson@Sun.COM
26665367Sahrens err = ioctl_err = zfs_ioctl(hdl, ZFS_IOC_RECV, &zc);
26675367Sahrens ioctl_errno = errno;
266811022STom.Erickson@Sun.COM prop_errflags = (zprop_errflags_t)zc.zc_obj;
266911022STom.Erickson@Sun.COM
267011022STom.Erickson@Sun.COM if (err == 0) {
267111022STom.Erickson@Sun.COM nvlist_t *prop_errors;
267211022STom.Erickson@Sun.COM VERIFY(0 == nvlist_unpack((void *)(uintptr_t)zc.zc_nvlist_dst,
267311022STom.Erickson@Sun.COM zc.zc_nvlist_dst_size, &prop_errors, 0));
267411022STom.Erickson@Sun.COM
267511022STom.Erickson@Sun.COM nvpair_t *prop_err = NULL;
267611022STom.Erickson@Sun.COM
267711022STom.Erickson@Sun.COM while ((prop_err = nvlist_next_nvpair(prop_errors,
267811022STom.Erickson@Sun.COM prop_err)) != NULL) {
267911022STom.Erickson@Sun.COM char tbuf[1024];
268011022STom.Erickson@Sun.COM zfs_prop_t prop;
268111022STom.Erickson@Sun.COM int intval;
268211022STom.Erickson@Sun.COM
268311022STom.Erickson@Sun.COM prop = zfs_name_to_prop(nvpair_name(prop_err));
268411022STom.Erickson@Sun.COM (void) nvpair_value_int32(prop_err, &intval);
268511022STom.Erickson@Sun.COM if (strcmp(nvpair_name(prop_err),
268611022STom.Erickson@Sun.COM ZPROP_N_MORE_ERRORS) == 0) {
268711022STom.Erickson@Sun.COM trunc_prop_errs(intval);
268811022STom.Erickson@Sun.COM break;
268911022STom.Erickson@Sun.COM } else {
269011022STom.Erickson@Sun.COM (void) snprintf(tbuf, sizeof (tbuf),
269111022STom.Erickson@Sun.COM dgettext(TEXT_DOMAIN,
269211022STom.Erickson@Sun.COM "cannot receive %s property on %s"),
269311022STom.Erickson@Sun.COM nvpair_name(prop_err), zc.zc_name);
269411022STom.Erickson@Sun.COM zfs_setprop_error(hdl, prop, intval, tbuf);
269511022STom.Erickson@Sun.COM }
269611022STom.Erickson@Sun.COM }
269711022STom.Erickson@Sun.COM nvlist_free(prop_errors);
269811022STom.Erickson@Sun.COM }
269911022STom.Erickson@Sun.COM
270011022STom.Erickson@Sun.COM zc.zc_nvlist_dst = 0;
270111022STom.Erickson@Sun.COM zc.zc_nvlist_dst_size = 0;
27027265Sahrens zcmd_free_nvlists(&zc);
27037265Sahrens
27047265Sahrens if (err == 0 && snapprops_nvlist) {
27057265Sahrens zfs_cmd_t zc2 = { 0 };
27067265Sahrens
27077273Sahrens (void) strcpy(zc2.zc_name, zc.zc_value);
270811022STom.Erickson@Sun.COM zc2.zc_cookie = B_TRUE; /* received */
27097265Sahrens if (zcmd_write_src_nvlist(hdl, &zc2, snapprops_nvlist) == 0) {
27107265Sahrens (void) zfs_ioctl(hdl, ZFS_IOC_SET_PROP, &zc2);
27117265Sahrens zcmd_free_nvlists(&zc2);
27127265Sahrens }
27137265Sahrens }
27147265Sahrens
271511834STom.Erickson@Sun.COM if (err && (ioctl_errno == ENOENT || ioctl_errno == EEXIST)) {
27165367Sahrens /*
27175367Sahrens * It may be that this snapshot already exists,
27185367Sahrens * in which case we want to consume & ignore it
27195367Sahrens * rather than failing.
27205367Sahrens */
27215367Sahrens avl_tree_t *local_avl;
27225367Sahrens nvlist_t *local_nv, *fs;
272311834STom.Erickson@Sun.COM cp = strchr(zc.zc_value, '@');
27245367Sahrens
27255367Sahrens /*
27265367Sahrens * XXX Do this faster by just iterating over snaps in
27275367Sahrens * this fs. Also if zc_value does not exist, we will
27285367Sahrens * get a strange "does not exist" error message.
27295367Sahrens */
27305367Sahrens *cp = '\0';
273111022STom.Erickson@Sun.COM if (gather_nvlist(hdl, zc.zc_value, NULL, NULL, B_FALSE,
27325367Sahrens &local_nv, &local_avl) == 0) {
27335367Sahrens *cp = '@';
27345367Sahrens fs = fsavl_find(local_avl, drrb->drr_toguid, NULL);
27355367Sahrens fsavl_destroy(local_avl);
27365367Sahrens nvlist_free(local_nv);
27375367Sahrens
27385367Sahrens if (fs != NULL) {
27395367Sahrens if (flags.verbose) {
27405367Sahrens (void) printf("snap %s already exists; "
27415367Sahrens "ignoring\n", zc.zc_value);
27425367Sahrens }
274311022STom.Erickson@Sun.COM err = ioctl_err = recv_skip(hdl, infd,
27445367Sahrens flags.byteswap);
27455367Sahrens }
27465367Sahrens }
27475367Sahrens *cp = '@';
27485367Sahrens }
27495367Sahrens
27505367Sahrens if (ioctl_err != 0) {
27515367Sahrens switch (ioctl_errno) {
27525367Sahrens case ENODEV:
27535367Sahrens cp = strchr(zc.zc_value, '@');
27545367Sahrens *cp = '\0';
27555367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
27565367Sahrens "most recent snapshot of %s does not\n"
27575367Sahrens "match incremental source"), zc.zc_value);
27585367Sahrens (void) zfs_error(hdl, EZFS_BADRESTORE, errbuf);
27595367Sahrens *cp = '@';
27605367Sahrens break;
27615367Sahrens case ETXTBSY:
27625367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
27635367Sahrens "destination %s has been modified\n"
27645367Sahrens "since most recent snapshot"), zc.zc_name);
27655367Sahrens (void) zfs_error(hdl, EZFS_BADRESTORE, errbuf);
27665367Sahrens break;
27675367Sahrens case EEXIST:
27685367Sahrens cp = strchr(zc.zc_value, '@');
27695367Sahrens if (newfs) {
27705367Sahrens /* it's the containing fs that exists */
27715367Sahrens *cp = '\0';
27725367Sahrens }
27735367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
27745367Sahrens "destination already exists"));
27755367Sahrens (void) zfs_error_fmt(hdl, EZFS_EXISTS,
27765367Sahrens dgettext(TEXT_DOMAIN, "cannot restore to %s"),
27775367Sahrens zc.zc_value);
27785367Sahrens *cp = '@';
27795367Sahrens break;
27805367Sahrens case EINVAL:
27815367Sahrens (void) zfs_error(hdl, EZFS_BADSTREAM, errbuf);
27825367Sahrens break;
27835367Sahrens case ECKSUM:
27845367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
27855367Sahrens "invalid stream (checksum mismatch)"));
27865367Sahrens (void) zfs_error(hdl, EZFS_BADSTREAM, errbuf);
27875367Sahrens break;
278812070SMark.Shellenbaum@Sun.COM case ENOTSUP:
278912070SMark.Shellenbaum@Sun.COM zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
279012070SMark.Shellenbaum@Sun.COM "pool must be upgraded to receive this stream."));
279112070SMark.Shellenbaum@Sun.COM (void) zfs_error(hdl, EZFS_BADVERSION, errbuf);
279212070SMark.Shellenbaum@Sun.COM break;
279312141SJohn.Harres@Sun.COM case EDQUOT:
279412141SJohn.Harres@Sun.COM zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
279512141SJohn.Harres@Sun.COM "destination %s space quota exceeded"), zc.zc_name);
279612141SJohn.Harres@Sun.COM (void) zfs_error(hdl, EZFS_BADRESTORE, errbuf);
279712141SJohn.Harres@Sun.COM break;
27985367Sahrens default:
27995367Sahrens (void) zfs_standard_error(hdl, ioctl_errno, errbuf);
28005367Sahrens }
28015367Sahrens }
28025367Sahrens
28035367Sahrens /*
280410588SEric.Taylor@Sun.COM * Mount the target filesystem (if created). Also mount any
280510588SEric.Taylor@Sun.COM * children of the target filesystem if we did a replication
280610588SEric.Taylor@Sun.COM * receive (indicated by stream_avl being non-NULL).
28075367Sahrens */
28085367Sahrens cp = strchr(zc.zc_value, '@');
28095367Sahrens if (cp && (ioctl_err == 0 || !newfs)) {
28105367Sahrens zfs_handle_t *h;
28115367Sahrens
28125367Sahrens *cp = '\0';
28135367Sahrens h = zfs_open(hdl, zc.zc_value,
28145367Sahrens ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME);
28156476Sahl if (h != NULL) {
28165367Sahrens if (h->zfs_type == ZFS_TYPE_VOLUME) {
28177366STim.Haley@Sun.COM *cp = '@';
281810200SLori.Alt@Sun.COM } else if (newfs || stream_avl) {
28197366STim.Haley@Sun.COM /*
28207366STim.Haley@Sun.COM * Track the first/top of hierarchy fs,
28217366STim.Haley@Sun.COM * for mounting and sharing later.
28227366STim.Haley@Sun.COM */
28237366STim.Haley@Sun.COM if (top_zfs && *top_zfs == NULL)
28247366STim.Haley@Sun.COM *top_zfs = zfs_strdup(hdl, zc.zc_value);
28255367Sahrens }
28266476Sahl zfs_close(h);
28275367Sahrens }
28287366STim.Haley@Sun.COM *cp = '@';
28295367Sahrens }
28305367Sahrens
28315367Sahrens if (clp) {
28325367Sahrens err |= changelist_postfix(clp);
28335367Sahrens changelist_free(clp);
28345367Sahrens }
28355367Sahrens
283611022STom.Erickson@Sun.COM if (prop_errflags & ZPROP_ERR_NOCLEAR) {
283711022STom.Erickson@Sun.COM (void) fprintf(stderr, dgettext(TEXT_DOMAIN, "Warning: "
283811022STom.Erickson@Sun.COM "failed to clear unreceived properties on %s"),
283911022STom.Erickson@Sun.COM zc.zc_name);
284011022STom.Erickson@Sun.COM (void) fprintf(stderr, "\n");
284111022STom.Erickson@Sun.COM }
284211022STom.Erickson@Sun.COM if (prop_errflags & ZPROP_ERR_NORESTORE) {
284311022STom.Erickson@Sun.COM (void) fprintf(stderr, dgettext(TEXT_DOMAIN, "Warning: "
284411022STom.Erickson@Sun.COM "failed to restore original properties on %s"),
284511022STom.Erickson@Sun.COM zc.zc_name);
284611022STom.Erickson@Sun.COM (void) fprintf(stderr, "\n");
284711022STom.Erickson@Sun.COM }
284811022STom.Erickson@Sun.COM
28495367Sahrens if (err || ioctl_err)
28505367Sahrens return (-1);
28515367Sahrens
285212527SChris.Kirby@oracle.com *action_handlep = zc.zc_action_handle;
285312527SChris.Kirby@oracle.com
28545367Sahrens if (flags.verbose) {
28555367Sahrens char buf1[64];
28565367Sahrens char buf2[64];
28575367Sahrens uint64_t bytes = zc.zc_cookie;
28585367Sahrens time_t delta = time(NULL) - begin_time;
28595367Sahrens if (delta == 0)
28605367Sahrens delta = 1;
28615367Sahrens zfs_nicenum(bytes, buf1, sizeof (buf1));
28625367Sahrens zfs_nicenum(bytes/delta, buf2, sizeof (buf1));
28635367Sahrens
28645367Sahrens (void) printf("received %sB stream in %lu seconds (%sB/sec)\n",
28655367Sahrens buf1, delta, buf2);
28665367Sahrens }
28675367Sahrens
28685367Sahrens return (0);
28695367Sahrens }
28705367Sahrens
28717366STim.Haley@Sun.COM static int
zfs_receive_impl(libzfs_handle_t * hdl,const char * tosnap,recvflags_t flags,int infd,const char * sendfs,nvlist_t * stream_nv,avl_tree_t * stream_avl,char ** top_zfs,int cleanup_fd,uint64_t * action_handlep)28727366STim.Haley@Sun.COM zfs_receive_impl(libzfs_handle_t *hdl, const char *tosnap, recvflags_t flags,
287311834STom.Erickson@Sun.COM int infd, const char *sendfs, nvlist_t *stream_nv, avl_tree_t *stream_avl,
287412527SChris.Kirby@oracle.com char **top_zfs, int cleanup_fd, uint64_t *action_handlep)
28755367Sahrens {
28765367Sahrens int err;
28775367Sahrens dmu_replay_record_t drr, drr_noswap;
28785367Sahrens struct drr_begin *drrb = &drr.drr_u.drr_begin;
28795367Sahrens char errbuf[1024];
28805367Sahrens zio_cksum_t zcksum = { 0 };
288111007SLori.Alt@Sun.COM uint64_t featureflags;
288211007SLori.Alt@Sun.COM int hdrtype;
28835367Sahrens
28845367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
28855367Sahrens "cannot receive"));
28865367Sahrens
28875367Sahrens if (flags.isprefix &&
28885367Sahrens !zfs_dataset_exists(hdl, tosnap, ZFS_TYPE_DATASET)) {
28895367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "specified fs "
28905367Sahrens "(%s) does not exist"), tosnap);
28915367Sahrens return (zfs_error(hdl, EZFS_NOENT, errbuf));
28925367Sahrens }
28935367Sahrens
28945367Sahrens /* read in the BEGIN record */
28955367Sahrens if (0 != (err = recv_read(hdl, infd, &drr, sizeof (drr), B_FALSE,
28965367Sahrens &zcksum)))
28975367Sahrens return (err);
28985367Sahrens
28995367Sahrens if (drr.drr_type == DRR_END || drr.drr_type == BSWAP_32(DRR_END)) {
29005367Sahrens /* It's the double end record at the end of a package */
29015367Sahrens return (ENODATA);
29025367Sahrens }
29035367Sahrens
29045367Sahrens /* the kernel needs the non-byteswapped begin record */
29055367Sahrens drr_noswap = drr;
29065367Sahrens
29075446Sahrens flags.byteswap = B_FALSE;
29085367Sahrens if (drrb->drr_magic == BSWAP_64(DMU_BACKUP_MAGIC)) {
29095367Sahrens /*
29105367Sahrens * We computed the checksum in the wrong byteorder in
29115367Sahrens * recv_read() above; do it again correctly.
29125367Sahrens */
29135367Sahrens bzero(&zcksum, sizeof (zio_cksum_t));
29145367Sahrens fletcher_4_incremental_byteswap(&drr, sizeof (drr), &zcksum);
29155367Sahrens flags.byteswap = B_TRUE;
29165367Sahrens
29175367Sahrens drr.drr_type = BSWAP_32(drr.drr_type);
29185367Sahrens drr.drr_payloadlen = BSWAP_32(drr.drr_payloadlen);
29195367Sahrens drrb->drr_magic = BSWAP_64(drrb->drr_magic);
292011007SLori.Alt@Sun.COM drrb->drr_versioninfo = BSWAP_64(drrb->drr_versioninfo);
29215367Sahrens drrb->drr_creation_time = BSWAP_64(drrb->drr_creation_time);
29225367Sahrens drrb->drr_type = BSWAP_32(drrb->drr_type);
29235367Sahrens drrb->drr_flags = BSWAP_32(drrb->drr_flags);
29245367Sahrens drrb->drr_toguid = BSWAP_64(drrb->drr_toguid);
29255367Sahrens drrb->drr_fromguid = BSWAP_64(drrb->drr_fromguid);
29265367Sahrens }
29275367Sahrens
29285367Sahrens if (drrb->drr_magic != DMU_BACKUP_MAGIC || drr.drr_type != DRR_BEGIN) {
29295367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
29305367Sahrens "stream (bad magic number)"));
29315367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
29325367Sahrens }
29335367Sahrens
293411007SLori.Alt@Sun.COM featureflags = DMU_GET_FEATUREFLAGS(drrb->drr_versioninfo);
293511007SLori.Alt@Sun.COM hdrtype = DMU_GET_STREAM_HDRTYPE(drrb->drr_versioninfo);
293611007SLori.Alt@Sun.COM
293711007SLori.Alt@Sun.COM if (!DMU_STREAM_SUPPORTED(featureflags) ||
293811007SLori.Alt@Sun.COM (hdrtype != DMU_SUBSTREAM && hdrtype != DMU_COMPOUNDSTREAM)) {
293911007SLori.Alt@Sun.COM zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
294011007SLori.Alt@Sun.COM "stream has unsupported feature, feature flags = %lx"),
294111007SLori.Alt@Sun.COM featureflags);
294211007SLori.Alt@Sun.COM return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
294311007SLori.Alt@Sun.COM }
294411007SLori.Alt@Sun.COM
29455367Sahrens if (strchr(drrb->drr_toname, '@') == NULL) {
29465367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
29475367Sahrens "stream (bad snapshot name)"));
29485367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
29495367Sahrens }
29505367Sahrens
295111007SLori.Alt@Sun.COM if (DMU_GET_STREAM_HDRTYPE(drrb->drr_versioninfo) == DMU_SUBSTREAM) {
295211834STom.Erickson@Sun.COM char nonpackage_sendfs[ZFS_MAXNAMELEN];
295311834STom.Erickson@Sun.COM if (sendfs == NULL) {
295411834STom.Erickson@Sun.COM /*
295511834STom.Erickson@Sun.COM * We were not called from zfs_receive_package(). Get
295611834STom.Erickson@Sun.COM * the fs specified by 'zfs send'.
295711834STom.Erickson@Sun.COM */
295811834STom.Erickson@Sun.COM char *cp;
295911834STom.Erickson@Sun.COM (void) strlcpy(nonpackage_sendfs,
296011834STom.Erickson@Sun.COM drr.drr_u.drr_begin.drr_toname, ZFS_MAXNAMELEN);
296111834STom.Erickson@Sun.COM if ((cp = strchr(nonpackage_sendfs, '@')) != NULL)
296211834STom.Erickson@Sun.COM *cp = '\0';
296311834STom.Erickson@Sun.COM sendfs = nonpackage_sendfs;
296411834STom.Erickson@Sun.COM }
29655367Sahrens return (zfs_receive_one(hdl, infd, tosnap, flags,
296611834STom.Erickson@Sun.COM &drr, &drr_noswap, sendfs, stream_nv, stream_avl,
296712527SChris.Kirby@oracle.com top_zfs, cleanup_fd, action_handlep));
296811976STom.Erickson@Sun.COM } else {
296911007SLori.Alt@Sun.COM assert(DMU_GET_STREAM_HDRTYPE(drrb->drr_versioninfo) ==
297011007SLori.Alt@Sun.COM DMU_COMPOUNDSTREAM);
29715367Sahrens return (zfs_receive_package(hdl, infd, tosnap, flags,
297212527SChris.Kirby@oracle.com &drr, &zcksum, top_zfs, cleanup_fd, action_handlep));
29735367Sahrens }
29745367Sahrens }
29757366STim.Haley@Sun.COM
29767366STim.Haley@Sun.COM /*
29777366STim.Haley@Sun.COM * Restores a backup of tosnap from the file descriptor specified by infd.
29787366STim.Haley@Sun.COM * Return 0 on total success, -2 if some things couldn't be
29797366STim.Haley@Sun.COM * destroyed/renamed/promoted, -1 if some things couldn't be received.
29807366STim.Haley@Sun.COM * (-1 will override -2).
29817366STim.Haley@Sun.COM */
29827366STim.Haley@Sun.COM int
zfs_receive(libzfs_handle_t * hdl,const char * tosnap,recvflags_t flags,int infd,avl_tree_t * stream_avl)29837366STim.Haley@Sun.COM zfs_receive(libzfs_handle_t *hdl, const char *tosnap, recvflags_t flags,
29847366STim.Haley@Sun.COM int infd, avl_tree_t *stream_avl)
29857366STim.Haley@Sun.COM {
29867366STim.Haley@Sun.COM char *top_zfs = NULL;
29877366STim.Haley@Sun.COM int err;
298812527SChris.Kirby@oracle.com int cleanup_fd;
298912527SChris.Kirby@oracle.com uint64_t action_handle = 0;
299012527SChris.Kirby@oracle.com
299112527SChris.Kirby@oracle.com cleanup_fd = open(ZFS_DEV, O_RDWR|O_EXCL);
299212527SChris.Kirby@oracle.com VERIFY(cleanup_fd >= 0);
29937366STim.Haley@Sun.COM
299411834STom.Erickson@Sun.COM err = zfs_receive_impl(hdl, tosnap, flags, infd, NULL, NULL,
299512527SChris.Kirby@oracle.com stream_avl, &top_zfs, cleanup_fd, &action_handle);
299612527SChris.Kirby@oracle.com
299712527SChris.Kirby@oracle.com VERIFY(0 == close(cleanup_fd));
29987366STim.Haley@Sun.COM
29998584SLori.Alt@Sun.COM if (err == 0 && !flags.nomount && top_zfs) {
30007366STim.Haley@Sun.COM zfs_handle_t *zhp;
30017366STim.Haley@Sun.COM prop_changelist_t *clp;
30027366STim.Haley@Sun.COM
30037366STim.Haley@Sun.COM zhp = zfs_open(hdl, top_zfs, ZFS_TYPE_FILESYSTEM);
30047366STim.Haley@Sun.COM if (zhp != NULL) {
30057366STim.Haley@Sun.COM clp = changelist_gather(zhp, ZFS_PROP_MOUNTPOINT,
30067366STim.Haley@Sun.COM CL_GATHER_MOUNT_ALWAYS, 0);
30077366STim.Haley@Sun.COM zfs_close(zhp);
30087366STim.Haley@Sun.COM if (clp != NULL) {
30097366STim.Haley@Sun.COM /* mount and share received datasets */
30107366STim.Haley@Sun.COM err = changelist_postfix(clp);
30117366STim.Haley@Sun.COM changelist_free(clp);
30127366STim.Haley@Sun.COM }
30137366STim.Haley@Sun.COM }
30147366STim.Haley@Sun.COM if (zhp == NULL || clp == NULL || err)
30157366STim.Haley@Sun.COM err = -1;
30167366STim.Haley@Sun.COM }
30177366STim.Haley@Sun.COM if (top_zfs)
30187366STim.Haley@Sun.COM free(top_zfs);
30197366STim.Haley@Sun.COM
30207366STim.Haley@Sun.COM return (err);
30217366STim.Haley@Sun.COM }
3022