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 /* 238584SLori.Alt@Sun.COM * Copyright 2009 Sun Microsystems, Inc. All rights reserved. 245367Sahrens * Use is subject to license terms. 255367Sahrens */ 265367Sahrens 275367Sahrens #include <assert.h> 285367Sahrens #include <ctype.h> 295367Sahrens #include <errno.h> 305367Sahrens #include <libintl.h> 315367Sahrens #include <stdio.h> 325367Sahrens #include <stdlib.h> 335367Sahrens #include <strings.h> 345367Sahrens #include <unistd.h> 355367Sahrens #include <stddef.h> 365367Sahrens #include <fcntl.h> 375367Sahrens #include <sys/mount.h> 385367Sahrens 395367Sahrens #include <libzfs.h> 405367Sahrens 415367Sahrens #include "zfs_namecheck.h" 425367Sahrens #include "zfs_prop.h" 4310602SLori.Alt@Sun.COM #include "zfs_fletcher.h" 445367Sahrens #include "libzfs_impl.h" 455367Sahrens 467366STim.Haley@Sun.COM static int zfs_receive_impl(libzfs_handle_t *, const char *, recvflags_t, 477366STim.Haley@Sun.COM int, avl_tree_t *, char **); 487366STim.Haley@Sun.COM 495367Sahrens /* 505367Sahrens * Routines for dealing with the AVL tree of fs-nvlists 515367Sahrens */ 525367Sahrens typedef struct fsavl_node { 535367Sahrens avl_node_t fn_node; 545367Sahrens nvlist_t *fn_nvfs; 555367Sahrens char *fn_snapname; 565367Sahrens uint64_t fn_guid; 575367Sahrens } fsavl_node_t; 585367Sahrens 595367Sahrens static int 605367Sahrens fsavl_compare(const void *arg1, const void *arg2) 615367Sahrens { 625367Sahrens const fsavl_node_t *fn1 = arg1; 635367Sahrens const fsavl_node_t *fn2 = arg2; 645367Sahrens 655367Sahrens if (fn1->fn_guid > fn2->fn_guid) 665367Sahrens return (+1); 675367Sahrens else if (fn1->fn_guid < fn2->fn_guid) 685367Sahrens return (-1); 695367Sahrens else 705367Sahrens return (0); 715367Sahrens } 725367Sahrens 735367Sahrens /* 745367Sahrens * Given the GUID of a snapshot, find its containing filesystem and 755367Sahrens * (optionally) name. 765367Sahrens */ 775367Sahrens static nvlist_t * 785367Sahrens fsavl_find(avl_tree_t *avl, uint64_t snapguid, char **snapname) 795367Sahrens { 805367Sahrens fsavl_node_t fn_find; 815367Sahrens fsavl_node_t *fn; 825367Sahrens 835367Sahrens fn_find.fn_guid = snapguid; 845367Sahrens 855367Sahrens fn = avl_find(avl, &fn_find, NULL); 865367Sahrens if (fn) { 875367Sahrens if (snapname) 885367Sahrens *snapname = fn->fn_snapname; 895367Sahrens return (fn->fn_nvfs); 905367Sahrens } 915367Sahrens return (NULL); 925367Sahrens } 935367Sahrens 945896Sahl static void 955896Sahl fsavl_destroy(avl_tree_t *avl) 965896Sahl { 975896Sahl fsavl_node_t *fn; 985896Sahl void *cookie; 995896Sahl 1005896Sahl if (avl == NULL) 1015896Sahl return; 1025896Sahl 1035896Sahl cookie = NULL; 1045896Sahl while ((fn = avl_destroy_nodes(avl, &cookie)) != NULL) 1055896Sahl free(fn); 1065896Sahl avl_destroy(avl); 1075896Sahl free(avl); 1085896Sahl } 1095896Sahl 11010200SLori.Alt@Sun.COM /* 11110200SLori.Alt@Sun.COM * Given an nvlist, produce an avl tree of snapshots, ordered by guid 11210200SLori.Alt@Sun.COM */ 1135367Sahrens static avl_tree_t * 1145367Sahrens fsavl_create(nvlist_t *fss) 1155367Sahrens { 1165367Sahrens avl_tree_t *fsavl; 1175367Sahrens nvpair_t *fselem = NULL; 1185367Sahrens 1195896Sahl if ((fsavl = malloc(sizeof (avl_tree_t))) == NULL) 1205896Sahl return (NULL); 1215896Sahl 1225367Sahrens avl_create(fsavl, fsavl_compare, sizeof (fsavl_node_t), 1235367Sahrens offsetof(fsavl_node_t, fn_node)); 1245367Sahrens 1255367Sahrens while ((fselem = nvlist_next_nvpair(fss, fselem)) != NULL) { 1265367Sahrens nvlist_t *nvfs, *snaps; 1275367Sahrens nvpair_t *snapelem = NULL; 1285367Sahrens 1295367Sahrens VERIFY(0 == nvpair_value_nvlist(fselem, &nvfs)); 1305367Sahrens VERIFY(0 == nvlist_lookup_nvlist(nvfs, "snaps", &snaps)); 1315367Sahrens 1325367Sahrens while ((snapelem = 1335367Sahrens nvlist_next_nvpair(snaps, snapelem)) != NULL) { 1345367Sahrens fsavl_node_t *fn; 1355367Sahrens uint64_t guid; 1365367Sahrens 1375367Sahrens VERIFY(0 == nvpair_value_uint64(snapelem, &guid)); 1385896Sahl if ((fn = malloc(sizeof (fsavl_node_t))) == NULL) { 1395896Sahl fsavl_destroy(fsavl); 1405896Sahl return (NULL); 1415896Sahl } 1425367Sahrens fn->fn_nvfs = nvfs; 1435367Sahrens fn->fn_snapname = nvpair_name(snapelem); 1445367Sahrens fn->fn_guid = guid; 1455367Sahrens 1465367Sahrens /* 1475367Sahrens * Note: if there are multiple snaps with the 1485367Sahrens * same GUID, we ignore all but one. 1495367Sahrens */ 1505367Sahrens if (avl_find(fsavl, fn, NULL) == NULL) 1515367Sahrens avl_add(fsavl, fn); 1525367Sahrens else 1535367Sahrens free(fn); 1545367Sahrens } 1555367Sahrens } 1565367Sahrens 1575367Sahrens return (fsavl); 1585367Sahrens } 1595367Sahrens 1605367Sahrens /* 1615367Sahrens * Routines for dealing with the giant nvlist of fs-nvlists, etc. 1625367Sahrens */ 1635367Sahrens typedef struct send_data { 1645367Sahrens uint64_t parent_fromsnap_guid; 1655367Sahrens nvlist_t *parent_snaps; 1665367Sahrens nvlist_t *fss; 1677265Sahrens nvlist_t *snapprops; 1685367Sahrens const char *fromsnap; 1695367Sahrens const char *tosnap; 1705367Sahrens 1715367Sahrens /* 1725367Sahrens * The header nvlist is of the following format: 1735367Sahrens * { 1745367Sahrens * "tosnap" -> string 1755367Sahrens * "fromsnap" -> string (if incremental) 1765367Sahrens * "fss" -> { 1775367Sahrens * id -> { 1785367Sahrens * 1795367Sahrens * "name" -> string (full name; for debugging) 1805367Sahrens * "parentfromsnap" -> number (guid of fromsnap in parent) 1815367Sahrens * 1825367Sahrens * "props" -> { name -> value (only if set here) } 1835367Sahrens * "snaps" -> { name (lastname) -> number (guid) } 1847265Sahrens * "snapprops" -> { name (lastname) -> { name -> value } } 1855367Sahrens * 1865367Sahrens * "origin" -> number (guid) (if clone) 1875367Sahrens * "sent" -> boolean (not on-disk) 1885367Sahrens * } 1895367Sahrens * } 1905367Sahrens * } 1915367Sahrens * 1925367Sahrens */ 1935367Sahrens } send_data_t; 1945367Sahrens 1957265Sahrens static void send_iterate_prop(zfs_handle_t *zhp, nvlist_t *nv); 1967265Sahrens 1975367Sahrens static int 1985367Sahrens send_iterate_snap(zfs_handle_t *zhp, void *arg) 1995367Sahrens { 2005367Sahrens send_data_t *sd = arg; 2015367Sahrens uint64_t guid = zhp->zfs_dmustats.dds_guid; 2025367Sahrens char *snapname; 2037265Sahrens nvlist_t *nv; 2045367Sahrens 2055367Sahrens snapname = strrchr(zhp->zfs_name, '@')+1; 2065367Sahrens 2075367Sahrens VERIFY(0 == nvlist_add_uint64(sd->parent_snaps, snapname, guid)); 2085367Sahrens /* 2095367Sahrens * NB: if there is no fromsnap here (it's a newly created fs in 2105367Sahrens * an incremental replication), we will substitute the tosnap. 2115367Sahrens */ 2125367Sahrens if ((sd->fromsnap && strcmp(snapname, sd->fromsnap) == 0) || 2135367Sahrens (sd->parent_fromsnap_guid == 0 && sd->tosnap && 2145367Sahrens strcmp(snapname, sd->tosnap) == 0)) { 2155367Sahrens sd->parent_fromsnap_guid = guid; 2165367Sahrens } 2175367Sahrens 2187265Sahrens VERIFY(0 == nvlist_alloc(&nv, NV_UNIQUE_NAME, 0)); 2197265Sahrens send_iterate_prop(zhp, nv); 2207265Sahrens VERIFY(0 == nvlist_add_nvlist(sd->snapprops, snapname, nv)); 2217265Sahrens nvlist_free(nv); 2227265Sahrens 2235367Sahrens zfs_close(zhp); 2245367Sahrens return (0); 2255367Sahrens } 2265367Sahrens 2275367Sahrens static void 2285367Sahrens send_iterate_prop(zfs_handle_t *zhp, nvlist_t *nv) 2295367Sahrens { 2305367Sahrens nvpair_t *elem = NULL; 2315367Sahrens 2325367Sahrens while ((elem = nvlist_next_nvpair(zhp->zfs_props, elem)) != NULL) { 2335367Sahrens char *propname = nvpair_name(elem); 2345367Sahrens zfs_prop_t prop = zfs_name_to_prop(propname); 2355367Sahrens nvlist_t *propnv; 2365367Sahrens 237*10960SEric.Schrock@Sun.COM if (!zfs_prop_user(propname)) { 238*10960SEric.Schrock@Sun.COM /* 239*10960SEric.Schrock@Sun.COM * Realistically, this should never happen. However, 240*10960SEric.Schrock@Sun.COM * we want the ability to add DSL properties without 241*10960SEric.Schrock@Sun.COM * needing to make incompatible version changes. We 242*10960SEric.Schrock@Sun.COM * need to ignore unknown properties to allow older 243*10960SEric.Schrock@Sun.COM * software to still send datasets containing these 244*10960SEric.Schrock@Sun.COM * properties, with the unknown properties elided. 245*10960SEric.Schrock@Sun.COM */ 246*10960SEric.Schrock@Sun.COM if (prop == ZPROP_INVAL) 247*10960SEric.Schrock@Sun.COM continue; 2489396SMatthew.Ahrens@Sun.COM 249*10960SEric.Schrock@Sun.COM if (zfs_prop_readonly(prop)) 250*10960SEric.Schrock@Sun.COM continue; 251*10960SEric.Schrock@Sun.COM } 2525367Sahrens 2535367Sahrens verify(nvpair_value_nvlist(elem, &propnv) == 0); 25410195SSanjeev.Bagewadi@Sun.COM if (prop == ZFS_PROP_QUOTA || prop == ZFS_PROP_RESERVATION || 25510195SSanjeev.Bagewadi@Sun.COM prop == ZFS_PROP_REFQUOTA || 25610195SSanjeev.Bagewadi@Sun.COM prop == ZFS_PROP_REFRESERVATION) { 2575367Sahrens /* these guys are modifyable, but have no source */ 2585367Sahrens uint64_t value; 2595367Sahrens verify(nvlist_lookup_uint64(propnv, 2605367Sahrens ZPROP_VALUE, &value) == 0); 2617265Sahrens if (zhp->zfs_type == ZFS_TYPE_SNAPSHOT) 2627265Sahrens continue; 2635367Sahrens } else { 2645367Sahrens char *source; 2655367Sahrens if (nvlist_lookup_string(propnv, 2665367Sahrens ZPROP_SOURCE, &source) != 0) 2675367Sahrens continue; 2685367Sahrens if (strcmp(source, zhp->zfs_name) != 0) 2695367Sahrens continue; 2705367Sahrens } 2715367Sahrens 2725367Sahrens if (zfs_prop_user(propname) || 2735367Sahrens zfs_prop_get_type(prop) == PROP_TYPE_STRING) { 2745367Sahrens char *value; 2755367Sahrens verify(nvlist_lookup_string(propnv, 2765367Sahrens ZPROP_VALUE, &value) == 0); 2775367Sahrens VERIFY(0 == nvlist_add_string(nv, propname, value)); 2785367Sahrens } else { 2795367Sahrens uint64_t value; 2805367Sahrens verify(nvlist_lookup_uint64(propnv, 2815367Sahrens ZPROP_VALUE, &value) == 0); 2825367Sahrens VERIFY(0 == nvlist_add_uint64(nv, propname, value)); 2835367Sahrens } 2845367Sahrens } 2855367Sahrens } 2865367Sahrens 28710200SLori.Alt@Sun.COM /* 28810200SLori.Alt@Sun.COM * recursively generate nvlists describing datasets. See comment 28910200SLori.Alt@Sun.COM * for the data structure send_data_t above for description of contents 29010200SLori.Alt@Sun.COM * of the nvlist. 29110200SLori.Alt@Sun.COM */ 2925367Sahrens static int 2935367Sahrens send_iterate_fs(zfs_handle_t *zhp, void *arg) 2945367Sahrens { 2955367Sahrens send_data_t *sd = arg; 2965367Sahrens nvlist_t *nvfs, *nv; 2975367Sahrens int rv; 2985367Sahrens uint64_t parent_fromsnap_guid_save = sd->parent_fromsnap_guid; 2995367Sahrens uint64_t guid = zhp->zfs_dmustats.dds_guid; 3005367Sahrens char guidstring[64]; 3015367Sahrens 3025367Sahrens VERIFY(0 == nvlist_alloc(&nvfs, NV_UNIQUE_NAME, 0)); 3035367Sahrens VERIFY(0 == nvlist_add_string(nvfs, "name", zhp->zfs_name)); 3045367Sahrens VERIFY(0 == nvlist_add_uint64(nvfs, "parentfromsnap", 3055367Sahrens sd->parent_fromsnap_guid)); 3065367Sahrens 3075367Sahrens if (zhp->zfs_dmustats.dds_origin[0]) { 3085367Sahrens zfs_handle_t *origin = zfs_open(zhp->zfs_hdl, 3095367Sahrens zhp->zfs_dmustats.dds_origin, ZFS_TYPE_SNAPSHOT); 3105367Sahrens if (origin == NULL) 3115367Sahrens return (-1); 3125367Sahrens VERIFY(0 == nvlist_add_uint64(nvfs, "origin", 3135367Sahrens origin->zfs_dmustats.dds_guid)); 3145367Sahrens } 3155367Sahrens 3165367Sahrens /* iterate over props */ 3175367Sahrens VERIFY(0 == nvlist_alloc(&nv, NV_UNIQUE_NAME, 0)); 3185367Sahrens send_iterate_prop(zhp, nv); 3195367Sahrens VERIFY(0 == nvlist_add_nvlist(nvfs, "props", nv)); 3205367Sahrens nvlist_free(nv); 3215367Sahrens 3225367Sahrens /* iterate over snaps, and set sd->parent_fromsnap_guid */ 3235367Sahrens sd->parent_fromsnap_guid = 0; 3245367Sahrens VERIFY(0 == nvlist_alloc(&sd->parent_snaps, NV_UNIQUE_NAME, 0)); 3257265Sahrens VERIFY(0 == nvlist_alloc(&sd->snapprops, NV_UNIQUE_NAME, 0)); 3265367Sahrens (void) zfs_iter_snapshots(zhp, send_iterate_snap, sd); 3275367Sahrens VERIFY(0 == nvlist_add_nvlist(nvfs, "snaps", sd->parent_snaps)); 3287265Sahrens VERIFY(0 == nvlist_add_nvlist(nvfs, "snapprops", sd->snapprops)); 3295367Sahrens nvlist_free(sd->parent_snaps); 3307265Sahrens nvlist_free(sd->snapprops); 3315367Sahrens 3325367Sahrens /* add this fs to nvlist */ 3335367Sahrens (void) snprintf(guidstring, sizeof (guidstring), 3345367Sahrens "0x%llx", (longlong_t)guid); 3355367Sahrens VERIFY(0 == nvlist_add_nvlist(sd->fss, guidstring, nvfs)); 3365367Sahrens nvlist_free(nvfs); 3375367Sahrens 3385367Sahrens /* iterate over children */ 3395367Sahrens rv = zfs_iter_filesystems(zhp, send_iterate_fs, sd); 3405367Sahrens 3415367Sahrens sd->parent_fromsnap_guid = parent_fromsnap_guid_save; 3425367Sahrens 3435367Sahrens zfs_close(zhp); 3445367Sahrens return (rv); 3455367Sahrens } 3465367Sahrens 3475367Sahrens static int 3485367Sahrens gather_nvlist(libzfs_handle_t *hdl, const char *fsname, const char *fromsnap, 3495367Sahrens const char *tosnap, nvlist_t **nvlp, avl_tree_t **avlp) 3505367Sahrens { 3515367Sahrens zfs_handle_t *zhp; 3525367Sahrens send_data_t sd = { 0 }; 3535367Sahrens int error; 3545367Sahrens 3555367Sahrens zhp = zfs_open(hdl, fsname, ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME); 3565367Sahrens if (zhp == NULL) 3575367Sahrens return (EZFS_BADTYPE); 3585367Sahrens 3595367Sahrens VERIFY(0 == nvlist_alloc(&sd.fss, NV_UNIQUE_NAME, 0)); 3605367Sahrens sd.fromsnap = fromsnap; 3615367Sahrens sd.tosnap = tosnap; 3625896Sahl 3635896Sahl if ((error = send_iterate_fs(zhp, &sd)) != 0) { 3645896Sahl nvlist_free(sd.fss); 3655896Sahl if (avlp != NULL) 3665896Sahl *avlp = NULL; 3675896Sahl *nvlp = NULL; 3685896Sahl return (error); 3695896Sahl } 3705896Sahl 3715896Sahl if (avlp != NULL && (*avlp = fsavl_create(sd.fss)) == NULL) { 3725896Sahl nvlist_free(sd.fss); 3735896Sahl *nvlp = NULL; 3745896Sahl return (EZFS_NOMEM); 3755896Sahl } 3765367Sahrens 3775367Sahrens *nvlp = sd.fss; 3785896Sahl return (0); 3795367Sahrens } 3805367Sahrens 3815367Sahrens /* 3825367Sahrens * Routines for dealing with the sorted snapshot functionality 3835367Sahrens */ 3845367Sahrens typedef struct zfs_node { 3855367Sahrens zfs_handle_t *zn_handle; 3865367Sahrens avl_node_t zn_avlnode; 3875367Sahrens } zfs_node_t; 3885367Sahrens 3895367Sahrens static int 3905367Sahrens zfs_sort_snaps(zfs_handle_t *zhp, void *data) 3915367Sahrens { 3925367Sahrens avl_tree_t *avl = data; 3935367Sahrens zfs_node_t *node = zfs_alloc(zhp->zfs_hdl, sizeof (zfs_node_t)); 3945367Sahrens 3955367Sahrens node->zn_handle = zhp; 3965367Sahrens avl_add(avl, node); 3975367Sahrens return (0); 3985367Sahrens } 3995367Sahrens 4005367Sahrens /* ARGSUSED */ 4015367Sahrens static int 4025367Sahrens zfs_snapshot_compare(const void *larg, const void *rarg) 4035367Sahrens { 4045367Sahrens zfs_handle_t *l = ((zfs_node_t *)larg)->zn_handle; 4055367Sahrens zfs_handle_t *r = ((zfs_node_t *)rarg)->zn_handle; 4065367Sahrens uint64_t lcreate, rcreate; 4075367Sahrens 4085367Sahrens /* 4095367Sahrens * Sort them according to creation time. We use the hidden 4105367Sahrens * CREATETXG property to get an absolute ordering of snapshots. 4115367Sahrens */ 4125367Sahrens lcreate = zfs_prop_get_int(l, ZFS_PROP_CREATETXG); 4135367Sahrens rcreate = zfs_prop_get_int(r, ZFS_PROP_CREATETXG); 4145367Sahrens 4155367Sahrens if (lcreate < rcreate) 4165367Sahrens return (-1); 4175367Sahrens else if (lcreate > rcreate) 4185367Sahrens return (+1); 4195367Sahrens else 4205367Sahrens return (0); 4215367Sahrens } 4225367Sahrens 42310342Schris.kirby@sun.com int 4245367Sahrens zfs_iter_snapshots_sorted(zfs_handle_t *zhp, zfs_iter_f callback, void *data) 4255367Sahrens { 4265367Sahrens int ret = 0; 4275367Sahrens zfs_node_t *node; 4285367Sahrens avl_tree_t avl; 4295367Sahrens void *cookie = NULL; 4305367Sahrens 4315367Sahrens avl_create(&avl, zfs_snapshot_compare, 4325367Sahrens sizeof (zfs_node_t), offsetof(zfs_node_t, zn_avlnode)); 4335367Sahrens 4345367Sahrens ret = zfs_iter_snapshots(zhp, zfs_sort_snaps, &avl); 4355367Sahrens 4365367Sahrens for (node = avl_first(&avl); node != NULL; node = AVL_NEXT(&avl, node)) 4375367Sahrens ret |= callback(node->zn_handle, data); 4385367Sahrens 4395367Sahrens while ((node = avl_destroy_nodes(&avl, &cookie)) != NULL) 4405367Sahrens free(node); 4415367Sahrens 4425367Sahrens avl_destroy(&avl); 4435367Sahrens 4445367Sahrens return (ret); 4455367Sahrens } 4465367Sahrens 4475367Sahrens /* 4485367Sahrens * Routines specific to "zfs send" 4495367Sahrens */ 4505367Sahrens typedef struct send_dump_data { 4515367Sahrens /* these are all just the short snapname (the part after the @) */ 4525367Sahrens const char *fromsnap; 4535367Sahrens const char *tosnap; 4545367Sahrens char lastsnap[ZFS_MAXNAMELEN]; 4555367Sahrens boolean_t seenfrom, seento, replicate, doall, fromorigin; 4565367Sahrens boolean_t verbose; 4575367Sahrens int outfd; 4585367Sahrens boolean_t err; 4595367Sahrens nvlist_t *fss; 4605367Sahrens avl_tree_t *fsavl; 4615367Sahrens } send_dump_data_t; 4625367Sahrens 4635367Sahrens /* 4645367Sahrens * Dumps a backup of the given snapshot (incremental from fromsnap if it's not 4655367Sahrens * NULL) to the file descriptor specified by outfd. 4665367Sahrens */ 4675367Sahrens static int 4685367Sahrens dump_ioctl(zfs_handle_t *zhp, const char *fromsnap, boolean_t fromorigin, 4695367Sahrens int outfd) 4705367Sahrens { 4715367Sahrens zfs_cmd_t zc = { 0 }; 4725367Sahrens libzfs_handle_t *hdl = zhp->zfs_hdl; 4735367Sahrens 4745367Sahrens assert(zhp->zfs_type == ZFS_TYPE_SNAPSHOT); 4755367Sahrens assert(fromsnap == NULL || fromsnap[0] == '\0' || !fromorigin); 4765367Sahrens 4775367Sahrens (void) strlcpy(zc.zc_name, zhp->zfs_name, sizeof (zc.zc_name)); 4785367Sahrens if (fromsnap) 4795765Sek110237 (void) strlcpy(zc.zc_value, fromsnap, sizeof (zc.zc_value)); 4805367Sahrens zc.zc_cookie = outfd; 4815367Sahrens zc.zc_obj = fromorigin; 4825367Sahrens 4835367Sahrens if (ioctl(zhp->zfs_hdl->libzfs_fd, ZFS_IOC_SEND, &zc) != 0) { 4845367Sahrens char errbuf[1024]; 4855367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 4865367Sahrens "warning: cannot send '%s'"), zhp->zfs_name); 4875367Sahrens 4885367Sahrens switch (errno) { 4895367Sahrens 4905367Sahrens case EXDEV: 4915367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 4925367Sahrens "not an earlier snapshot from the same fs")); 4935367Sahrens return (zfs_error(hdl, EZFS_CROSSTARGET, errbuf)); 4945367Sahrens 4955367Sahrens case ENOENT: 4965367Sahrens if (zfs_dataset_exists(hdl, zc.zc_name, 4975367Sahrens ZFS_TYPE_SNAPSHOT)) { 4985367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 4995367Sahrens "incremental source (@%s) does not exist"), 5005367Sahrens zc.zc_value); 5015367Sahrens } 5025367Sahrens return (zfs_error(hdl, EZFS_NOENT, errbuf)); 5035367Sahrens 5045367Sahrens case EDQUOT: 5055367Sahrens case EFBIG: 5065367Sahrens case EIO: 5075367Sahrens case ENOLINK: 5085367Sahrens case ENOSPC: 5095367Sahrens case ENOSTR: 5105367Sahrens case ENXIO: 5115367Sahrens case EPIPE: 5125367Sahrens case ERANGE: 5135367Sahrens case EFAULT: 5145367Sahrens case EROFS: 5155367Sahrens zfs_error_aux(hdl, strerror(errno)); 5165367Sahrens return (zfs_error(hdl, EZFS_BADBACKUP, errbuf)); 5175367Sahrens 5185367Sahrens default: 5195367Sahrens return (zfs_standard_error(hdl, errno, errbuf)); 5205367Sahrens } 5215367Sahrens } 5225367Sahrens 5235367Sahrens return (0); 5245367Sahrens } 5255367Sahrens 5265367Sahrens static int 5275367Sahrens dump_snapshot(zfs_handle_t *zhp, void *arg) 5285367Sahrens { 5295367Sahrens send_dump_data_t *sdd = arg; 5305367Sahrens const char *thissnap; 5315367Sahrens int err; 5325367Sahrens 5335367Sahrens thissnap = strchr(zhp->zfs_name, '@') + 1; 5345367Sahrens 5355367Sahrens if (sdd->fromsnap && !sdd->seenfrom && 5365367Sahrens strcmp(sdd->fromsnap, thissnap) == 0) { 5375367Sahrens sdd->seenfrom = B_TRUE; 5385367Sahrens (void) strcpy(sdd->lastsnap, thissnap); 5395367Sahrens zfs_close(zhp); 5405367Sahrens return (0); 5415367Sahrens } 5425367Sahrens 5435367Sahrens if (sdd->seento || !sdd->seenfrom) { 5445367Sahrens zfs_close(zhp); 5455367Sahrens return (0); 5465367Sahrens } 5475367Sahrens 5485367Sahrens /* send it */ 5495367Sahrens if (sdd->verbose) { 5505367Sahrens (void) fprintf(stderr, "sending from @%s to %s\n", 5515367Sahrens sdd->lastsnap, zhp->zfs_name); 5525367Sahrens } 5535367Sahrens 5545367Sahrens err = dump_ioctl(zhp, sdd->lastsnap, 5555367Sahrens sdd->lastsnap[0] == '\0' && (sdd->fromorigin || sdd->replicate), 5565367Sahrens sdd->outfd); 5575367Sahrens 5585367Sahrens if (!sdd->seento && strcmp(sdd->tosnap, thissnap) == 0) 5595367Sahrens sdd->seento = B_TRUE; 5605367Sahrens 5615367Sahrens (void) strcpy(sdd->lastsnap, thissnap); 5625367Sahrens zfs_close(zhp); 5635367Sahrens return (err); 5645367Sahrens } 5655367Sahrens 5665367Sahrens static int 5675367Sahrens dump_filesystem(zfs_handle_t *zhp, void *arg) 5685367Sahrens { 5695367Sahrens int rv = 0; 5705367Sahrens send_dump_data_t *sdd = arg; 5715367Sahrens boolean_t missingfrom = B_FALSE; 5725367Sahrens zfs_cmd_t zc = { 0 }; 5735367Sahrens 5745367Sahrens (void) snprintf(zc.zc_name, sizeof (zc.zc_name), "%s@%s", 5755367Sahrens zhp->zfs_name, sdd->tosnap); 5765367Sahrens if (ioctl(zhp->zfs_hdl->libzfs_fd, ZFS_IOC_OBJSET_STATS, &zc) != 0) { 5775367Sahrens (void) fprintf(stderr, "WARNING: " 5785367Sahrens "could not send %s@%s: does not exist\n", 5795367Sahrens zhp->zfs_name, sdd->tosnap); 5805367Sahrens sdd->err = B_TRUE; 5815367Sahrens return (0); 5825367Sahrens } 5835367Sahrens 5845367Sahrens if (sdd->replicate && sdd->fromsnap) { 5855367Sahrens /* 5865367Sahrens * If this fs does not have fromsnap, and we're doing 5875367Sahrens * recursive, we need to send a full stream from the 5885367Sahrens * beginning (or an incremental from the origin if this 5895367Sahrens * is a clone). If we're doing non-recursive, then let 5905367Sahrens * them get the error. 5915367Sahrens */ 5925367Sahrens (void) snprintf(zc.zc_name, sizeof (zc.zc_name), "%s@%s", 5935367Sahrens zhp->zfs_name, sdd->fromsnap); 5945367Sahrens if (ioctl(zhp->zfs_hdl->libzfs_fd, 5955367Sahrens ZFS_IOC_OBJSET_STATS, &zc) != 0) { 5965367Sahrens missingfrom = B_TRUE; 5975367Sahrens } 5985367Sahrens } 5995367Sahrens 6005367Sahrens if (sdd->doall) { 6015367Sahrens sdd->seenfrom = sdd->seento = sdd->lastsnap[0] = 0; 6025367Sahrens if (sdd->fromsnap == NULL || missingfrom) 6035367Sahrens sdd->seenfrom = B_TRUE; 6045367Sahrens 6055367Sahrens rv = zfs_iter_snapshots_sorted(zhp, dump_snapshot, arg); 6065367Sahrens if (!sdd->seenfrom) { 6075367Sahrens (void) fprintf(stderr, 6085367Sahrens "WARNING: could not send %s@%s:\n" 6095367Sahrens "incremental source (%s@%s) does not exist\n", 6105367Sahrens zhp->zfs_name, sdd->tosnap, 6115367Sahrens zhp->zfs_name, sdd->fromsnap); 6125367Sahrens sdd->err = B_TRUE; 6135367Sahrens } else if (!sdd->seento) { 6149396SMatthew.Ahrens@Sun.COM if (sdd->fromsnap) { 6159396SMatthew.Ahrens@Sun.COM (void) fprintf(stderr, 6169396SMatthew.Ahrens@Sun.COM "WARNING: could not send %s@%s:\n" 6179396SMatthew.Ahrens@Sun.COM "incremental source (%s@%s) " 6189396SMatthew.Ahrens@Sun.COM "is not earlier than it\n", 6199396SMatthew.Ahrens@Sun.COM zhp->zfs_name, sdd->tosnap, 6209396SMatthew.Ahrens@Sun.COM zhp->zfs_name, sdd->fromsnap); 6219396SMatthew.Ahrens@Sun.COM } else { 6229396SMatthew.Ahrens@Sun.COM (void) fprintf(stderr, "WARNING: " 6239396SMatthew.Ahrens@Sun.COM "could not send %s@%s: does not exist\n", 6249396SMatthew.Ahrens@Sun.COM zhp->zfs_name, sdd->tosnap); 6259396SMatthew.Ahrens@Sun.COM } 6265367Sahrens sdd->err = B_TRUE; 6275367Sahrens } 6285367Sahrens } else { 6295367Sahrens zfs_handle_t *snapzhp; 6305367Sahrens char snapname[ZFS_MAXNAMELEN]; 6315367Sahrens 6325367Sahrens (void) snprintf(snapname, sizeof (snapname), "%s@%s", 6335367Sahrens zfs_get_name(zhp), sdd->tosnap); 6345367Sahrens snapzhp = zfs_open(zhp->zfs_hdl, snapname, ZFS_TYPE_SNAPSHOT); 6356476Sahl if (snapzhp == NULL) { 6366476Sahl rv = -1; 6376476Sahl } else { 6386476Sahl rv = dump_ioctl(snapzhp, 6396476Sahl missingfrom ? NULL : sdd->fromsnap, 6406476Sahl sdd->fromorigin || missingfrom, 6416476Sahl sdd->outfd); 6426476Sahl sdd->seento = B_TRUE; 6436476Sahl zfs_close(snapzhp); 6446476Sahl } 6455367Sahrens } 6465367Sahrens 6475367Sahrens return (rv); 6485367Sahrens } 6495367Sahrens 6505367Sahrens static int 6515367Sahrens dump_filesystems(zfs_handle_t *rzhp, void *arg) 6525367Sahrens { 6535367Sahrens send_dump_data_t *sdd = arg; 6545367Sahrens nvpair_t *fspair; 6555367Sahrens boolean_t needagain, progress; 6565367Sahrens 6575367Sahrens if (!sdd->replicate) 6585367Sahrens return (dump_filesystem(rzhp, sdd)); 6595367Sahrens 6605367Sahrens again: 6615367Sahrens needagain = progress = B_FALSE; 6625367Sahrens for (fspair = nvlist_next_nvpair(sdd->fss, NULL); fspair; 6635367Sahrens fspair = nvlist_next_nvpair(sdd->fss, fspair)) { 6645367Sahrens nvlist_t *fslist; 6655367Sahrens char *fsname; 6665367Sahrens zfs_handle_t *zhp; 6675367Sahrens int err; 6685367Sahrens uint64_t origin_guid = 0; 6695367Sahrens nvlist_t *origin_nv; 6705367Sahrens 6715367Sahrens VERIFY(nvpair_value_nvlist(fspair, &fslist) == 0); 6725367Sahrens if (nvlist_lookup_boolean(fslist, "sent") == 0) 6735367Sahrens continue; 6745367Sahrens 6755367Sahrens VERIFY(nvlist_lookup_string(fslist, "name", &fsname) == 0); 6765367Sahrens (void) nvlist_lookup_uint64(fslist, "origin", &origin_guid); 6775367Sahrens 6785367Sahrens origin_nv = fsavl_find(sdd->fsavl, origin_guid, NULL); 6795367Sahrens if (origin_nv && 6805367Sahrens nvlist_lookup_boolean(origin_nv, "sent") == ENOENT) { 6815367Sahrens /* 6825367Sahrens * origin has not been sent yet; 6835367Sahrens * skip this clone. 6845367Sahrens */ 6855367Sahrens needagain = B_TRUE; 6865367Sahrens continue; 6875367Sahrens } 6885367Sahrens 6895367Sahrens zhp = zfs_open(rzhp->zfs_hdl, fsname, ZFS_TYPE_DATASET); 6906476Sahl if (zhp == NULL) 6916476Sahl return (-1); 6925367Sahrens err = dump_filesystem(zhp, sdd); 6935367Sahrens VERIFY(nvlist_add_boolean(fslist, "sent") == 0); 6945367Sahrens progress = B_TRUE; 6955367Sahrens zfs_close(zhp); 6965367Sahrens if (err) 6975367Sahrens return (err); 6985367Sahrens } 6995367Sahrens if (needagain) { 7005367Sahrens assert(progress); 7015367Sahrens goto again; 7025367Sahrens } 7035367Sahrens return (0); 7045367Sahrens } 7055367Sahrens 7065367Sahrens /* 70710200SLori.Alt@Sun.COM * Generate a send stream for the dataset identified by the argument zhp. 70810200SLori.Alt@Sun.COM * 70910200SLori.Alt@Sun.COM * The content of the send stream is the snapshot identified by 71010200SLori.Alt@Sun.COM * 'tosnap'. Incremental streams are requested in two ways: 71110200SLori.Alt@Sun.COM * - from the snapshot identified by "fromsnap" (if non-null) or 71210200SLori.Alt@Sun.COM * - from the origin of the dataset identified by zhp, which must 71310200SLori.Alt@Sun.COM * be a clone. In this case, "fromsnap" is null and "fromorigin" 71410200SLori.Alt@Sun.COM * is TRUE. 71510200SLori.Alt@Sun.COM * 71610200SLori.Alt@Sun.COM * The send stream is recursive (i.e. dumps a hierarchy of snapshots) and 71710200SLori.Alt@Sun.COM * uses a special header (with a version field of DMU_BACKUP_HEADER_VERSION) 71810200SLori.Alt@Sun.COM * if "replicate" is set. If "doall" is set, dump all the intermediate 71910200SLori.Alt@Sun.COM * snapshots. The DMU_BACKUP_HEADER_VERSION header is used in the "doall" 72010200SLori.Alt@Sun.COM * case too. 7215367Sahrens */ 7225367Sahrens int 7235367Sahrens zfs_send(zfs_handle_t *zhp, const char *fromsnap, const char *tosnap, 7245367Sahrens boolean_t replicate, boolean_t doall, boolean_t fromorigin, 7255367Sahrens boolean_t verbose, int outfd) 7265367Sahrens { 7275367Sahrens char errbuf[1024]; 7285367Sahrens send_dump_data_t sdd = { 0 }; 7295367Sahrens int err; 7305367Sahrens nvlist_t *fss = NULL; 7315367Sahrens avl_tree_t *fsavl = NULL; 73210342Schris.kirby@sun.com char holdtag[128]; 73310342Schris.kirby@sun.com static uint64_t holdseq; 73410735Schris.kirby@sun.com int spa_version; 73510735Schris.kirby@sun.com boolean_t holdsnaps = B_FALSE; 7365367Sahrens 7375367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 7385367Sahrens "cannot send '%s'"), zhp->zfs_name); 7395367Sahrens 7405367Sahrens if (fromsnap && fromsnap[0] == '\0') { 7415367Sahrens zfs_error_aux(zhp->zfs_hdl, dgettext(TEXT_DOMAIN, 7425367Sahrens "zero-length incremental source")); 7435367Sahrens return (zfs_error(zhp->zfs_hdl, EZFS_NOENT, errbuf)); 7445367Sahrens } 7455367Sahrens 74610735Schris.kirby@sun.com if (zfs_spa_version(zhp, &spa_version) == 0 && 74710735Schris.kirby@sun.com spa_version >= SPA_VERSION_USERREFS) 74810735Schris.kirby@sun.com holdsnaps = B_TRUE; 74910735Schris.kirby@sun.com 7505367Sahrens if (replicate || doall) { 7515367Sahrens dmu_replay_record_t drr = { 0 }; 7525367Sahrens char *packbuf = NULL; 7535367Sahrens size_t buflen = 0; 7545367Sahrens zio_cksum_t zc = { 0 }; 7555367Sahrens 7565367Sahrens assert(fromsnap || doall); 7575367Sahrens 75810735Schris.kirby@sun.com if (holdsnaps) { 75910735Schris.kirby@sun.com (void) snprintf(holdtag, sizeof (holdtag), 76010735Schris.kirby@sun.com ".send-%d-%llu", getpid(), (u_longlong_t)holdseq); 76110735Schris.kirby@sun.com ++holdseq; 76210735Schris.kirby@sun.com err = zfs_hold_range(zhp, fromsnap, tosnap, 76310735Schris.kirby@sun.com holdtag, B_TRUE); 76410735Schris.kirby@sun.com if (err) 76510735Schris.kirby@sun.com return (err); 76610735Schris.kirby@sun.com } 7675367Sahrens if (replicate) { 7685367Sahrens nvlist_t *hdrnv; 7695367Sahrens 7705367Sahrens VERIFY(0 == nvlist_alloc(&hdrnv, NV_UNIQUE_NAME, 0)); 7715367Sahrens if (fromsnap) { 7725367Sahrens VERIFY(0 == nvlist_add_string(hdrnv, 7735367Sahrens "fromsnap", fromsnap)); 7745367Sahrens } 7755367Sahrens VERIFY(0 == nvlist_add_string(hdrnv, "tosnap", tosnap)); 7765367Sahrens 7775367Sahrens err = gather_nvlist(zhp->zfs_hdl, zhp->zfs_name, 7785367Sahrens fromsnap, tosnap, &fss, &fsavl); 77910342Schris.kirby@sun.com if (err) { 78010735Schris.kirby@sun.com if (holdsnaps) { 78110735Schris.kirby@sun.com (void) zfs_release_range(zhp, fromsnap, 78210735Schris.kirby@sun.com tosnap, holdtag); 78310735Schris.kirby@sun.com } 7845367Sahrens return (err); 78510342Schris.kirby@sun.com } 7865367Sahrens VERIFY(0 == nvlist_add_nvlist(hdrnv, "fss", fss)); 7875367Sahrens err = nvlist_pack(hdrnv, &packbuf, &buflen, 7885367Sahrens NV_ENCODE_XDR, 0); 7895367Sahrens nvlist_free(hdrnv); 7905367Sahrens if (err) { 7915367Sahrens fsavl_destroy(fsavl); 7925367Sahrens nvlist_free(fss); 79310735Schris.kirby@sun.com if (holdsnaps) { 79410735Schris.kirby@sun.com (void) zfs_release_range(zhp, fromsnap, 79510735Schris.kirby@sun.com tosnap, holdtag); 79610735Schris.kirby@sun.com } 7975367Sahrens return (zfs_standard_error(zhp->zfs_hdl, 7985367Sahrens err, errbuf)); 7995367Sahrens } 8005367Sahrens } 8015367Sahrens 8025367Sahrens /* write first begin record */ 8035367Sahrens drr.drr_type = DRR_BEGIN; 8045367Sahrens drr.drr_u.drr_begin.drr_magic = DMU_BACKUP_MAGIC; 8055367Sahrens drr.drr_u.drr_begin.drr_version = DMU_BACKUP_HEADER_VERSION; 8065367Sahrens (void) snprintf(drr.drr_u.drr_begin.drr_toname, 8075367Sahrens sizeof (drr.drr_u.drr_begin.drr_toname), 8085367Sahrens "%s@%s", zhp->zfs_name, tosnap); 8095367Sahrens drr.drr_payloadlen = buflen; 8105367Sahrens fletcher_4_incremental_native(&drr, sizeof (drr), &zc); 8115367Sahrens err = write(outfd, &drr, sizeof (drr)); 8125367Sahrens 8135367Sahrens /* write header nvlist */ 8145367Sahrens if (err != -1) { 8155367Sahrens fletcher_4_incremental_native(packbuf, buflen, &zc); 8165367Sahrens err = write(outfd, packbuf, buflen); 8175367Sahrens } 8185367Sahrens free(packbuf); 8195367Sahrens if (err == -1) { 8205367Sahrens fsavl_destroy(fsavl); 8215367Sahrens nvlist_free(fss); 82210735Schris.kirby@sun.com if (holdsnaps) { 82310735Schris.kirby@sun.com (void) zfs_release_range(zhp, fromsnap, tosnap, 82410735Schris.kirby@sun.com holdtag); 82510735Schris.kirby@sun.com } 8265367Sahrens return (zfs_standard_error(zhp->zfs_hdl, 8275367Sahrens errno, errbuf)); 8285367Sahrens } 8295367Sahrens 8305367Sahrens /* write end record */ 8315367Sahrens if (err != -1) { 8325367Sahrens bzero(&drr, sizeof (drr)); 8335367Sahrens drr.drr_type = DRR_END; 8345367Sahrens drr.drr_u.drr_end.drr_checksum = zc; 8355367Sahrens err = write(outfd, &drr, sizeof (drr)); 8365367Sahrens if (err == -1) { 8375367Sahrens fsavl_destroy(fsavl); 8385367Sahrens nvlist_free(fss); 83910735Schris.kirby@sun.com if (holdsnaps) { 84010735Schris.kirby@sun.com (void) zfs_release_range(zhp, fromsnap, 84110735Schris.kirby@sun.com tosnap, holdtag); 84210735Schris.kirby@sun.com } 8435367Sahrens return (zfs_standard_error(zhp->zfs_hdl, 8445367Sahrens errno, errbuf)); 8455367Sahrens } 8465367Sahrens } 8475367Sahrens } 8485367Sahrens 8495367Sahrens /* dump each stream */ 8505367Sahrens sdd.fromsnap = fromsnap; 8515367Sahrens sdd.tosnap = tosnap; 8525367Sahrens sdd.outfd = outfd; 8535367Sahrens sdd.replicate = replicate; 8545367Sahrens sdd.doall = doall; 8555367Sahrens sdd.fromorigin = fromorigin; 8565367Sahrens sdd.fss = fss; 8575367Sahrens sdd.fsavl = fsavl; 8585367Sahrens sdd.verbose = verbose; 8595367Sahrens err = dump_filesystems(zhp, &sdd); 8605367Sahrens fsavl_destroy(fsavl); 8615367Sahrens nvlist_free(fss); 8625367Sahrens 8635367Sahrens if (replicate || doall) { 8645367Sahrens /* 8655367Sahrens * write final end record. NB: want to do this even if 8665367Sahrens * there was some error, because it might not be totally 8675367Sahrens * failed. 8685367Sahrens */ 8695367Sahrens dmu_replay_record_t drr = { 0 }; 8705367Sahrens drr.drr_type = DRR_END; 87110735Schris.kirby@sun.com if (holdsnaps) { 87210735Schris.kirby@sun.com (void) zfs_release_range(zhp, fromsnap, tosnap, 87310735Schris.kirby@sun.com holdtag); 87410735Schris.kirby@sun.com } 8755367Sahrens if (write(outfd, &drr, sizeof (drr)) == -1) { 8765367Sahrens return (zfs_standard_error(zhp->zfs_hdl, 8775367Sahrens errno, errbuf)); 8785367Sahrens } 8795367Sahrens } 8805367Sahrens 8815367Sahrens return (err || sdd.err); 8825367Sahrens } 8835367Sahrens 8845367Sahrens /* 8855367Sahrens * Routines specific to "zfs recv" 8865367Sahrens */ 8875367Sahrens 8885367Sahrens static int 8895367Sahrens recv_read(libzfs_handle_t *hdl, int fd, void *buf, int ilen, 8905367Sahrens boolean_t byteswap, zio_cksum_t *zc) 8915367Sahrens { 8925367Sahrens char *cp = buf; 8935367Sahrens int rv; 8945367Sahrens int len = ilen; 8955367Sahrens 8965367Sahrens do { 8975367Sahrens rv = read(fd, cp, len); 8985367Sahrens cp += rv; 8995367Sahrens len -= rv; 9005367Sahrens } while (rv > 0); 9015367Sahrens 9025367Sahrens if (rv < 0 || len != 0) { 9035367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 9045367Sahrens "failed to read from stream")); 9055367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, dgettext(TEXT_DOMAIN, 9065367Sahrens "cannot receive"))); 9075367Sahrens } 9085367Sahrens 9095367Sahrens if (zc) { 9105367Sahrens if (byteswap) 9115367Sahrens fletcher_4_incremental_byteswap(buf, ilen, zc); 9125367Sahrens else 9135367Sahrens fletcher_4_incremental_native(buf, ilen, zc); 9145367Sahrens } 9155367Sahrens return (0); 9165367Sahrens } 9175367Sahrens 9185367Sahrens static int 9195367Sahrens recv_read_nvlist(libzfs_handle_t *hdl, int fd, int len, nvlist_t **nvp, 9205367Sahrens boolean_t byteswap, zio_cksum_t *zc) 9215367Sahrens { 9225367Sahrens char *buf; 9235367Sahrens int err; 9245367Sahrens 9255367Sahrens buf = zfs_alloc(hdl, len); 9265367Sahrens if (buf == NULL) 9275367Sahrens return (ENOMEM); 9285367Sahrens 9295367Sahrens err = recv_read(hdl, fd, buf, len, byteswap, zc); 9305367Sahrens if (err != 0) { 9315367Sahrens free(buf); 9325367Sahrens return (err); 9335367Sahrens } 9345367Sahrens 9355367Sahrens err = nvlist_unpack(buf, len, nvp, 0); 9365367Sahrens free(buf); 9375367Sahrens if (err != 0) { 9385367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid " 9395367Sahrens "stream (malformed nvlist)")); 9405367Sahrens return (EINVAL); 9415367Sahrens } 9425367Sahrens return (0); 9435367Sahrens } 9445367Sahrens 9455367Sahrens static int 9465367Sahrens recv_rename(libzfs_handle_t *hdl, const char *name, const char *tryname, 9475367Sahrens int baselen, char *newname, recvflags_t flags) 9485367Sahrens { 9495367Sahrens static int seq; 9505367Sahrens zfs_cmd_t zc = { 0 }; 9515367Sahrens int err; 9526047Sahrens prop_changelist_t *clp; 9536047Sahrens zfs_handle_t *zhp; 9545367Sahrens 9556047Sahrens zhp = zfs_open(hdl, name, ZFS_TYPE_DATASET); 9566047Sahrens if (zhp == NULL) 9576047Sahrens return (-1); 9587366STim.Haley@Sun.COM clp = changelist_gather(zhp, ZFS_PROP_NAME, 0, 9597366STim.Haley@Sun.COM flags.force ? MS_FORCE : 0); 9606047Sahrens zfs_close(zhp); 9616047Sahrens if (clp == NULL) 9626047Sahrens return (-1); 9636047Sahrens err = changelist_prefix(clp); 9646047Sahrens if (err) 9656047Sahrens return (err); 9665367Sahrens 96710242Schris.kirby@sun.com zc.zc_objset_type = DMU_OST_ZFS; 96810242Schris.kirby@sun.com (void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name)); 96910242Schris.kirby@sun.com 9705367Sahrens if (tryname) { 9715367Sahrens (void) strcpy(newname, tryname); 9725367Sahrens 9735367Sahrens (void) strlcpy(zc.zc_value, tryname, sizeof (zc.zc_value)); 9745367Sahrens 9755367Sahrens if (flags.verbose) { 9765367Sahrens (void) printf("attempting rename %s to %s\n", 9775367Sahrens zc.zc_name, zc.zc_value); 9785367Sahrens } 9795367Sahrens err = ioctl(hdl->libzfs_fd, ZFS_IOC_RENAME, &zc); 9806047Sahrens if (err == 0) 9815367Sahrens changelist_rename(clp, name, tryname); 9825367Sahrens } else { 9835367Sahrens err = ENOENT; 9845367Sahrens } 9855367Sahrens 9865367Sahrens if (err != 0 && strncmp(name+baselen, "recv-", 5) != 0) { 9875367Sahrens seq++; 9885367Sahrens 9895367Sahrens (void) strncpy(newname, name, baselen); 9905367Sahrens (void) snprintf(newname+baselen, ZFS_MAXNAMELEN-baselen, 9915367Sahrens "recv-%u-%u", getpid(), seq); 9925367Sahrens (void) strlcpy(zc.zc_value, newname, sizeof (zc.zc_value)); 9935367Sahrens 9945367Sahrens if (flags.verbose) { 9955367Sahrens (void) printf("failed - trying rename %s to %s\n", 9965367Sahrens zc.zc_name, zc.zc_value); 9975367Sahrens } 9985367Sahrens err = ioctl(hdl->libzfs_fd, ZFS_IOC_RENAME, &zc); 9996047Sahrens if (err == 0) 10005367Sahrens changelist_rename(clp, name, newname); 10015367Sahrens if (err && flags.verbose) { 10025367Sahrens (void) printf("failed (%u) - " 10035367Sahrens "will try again on next pass\n", errno); 10045367Sahrens } 10055367Sahrens err = EAGAIN; 10065367Sahrens } else if (flags.verbose) { 10075367Sahrens if (err == 0) 10085367Sahrens (void) printf("success\n"); 10095367Sahrens else 10105367Sahrens (void) printf("failed (%u)\n", errno); 10115367Sahrens } 10125367Sahrens 10136047Sahrens (void) changelist_postfix(clp); 10146047Sahrens changelist_free(clp); 10155367Sahrens 10165367Sahrens return (err); 10175367Sahrens } 10185367Sahrens 10195367Sahrens static int 10205367Sahrens recv_destroy(libzfs_handle_t *hdl, const char *name, int baselen, 10215367Sahrens char *newname, recvflags_t flags) 10225367Sahrens { 10235367Sahrens zfs_cmd_t zc = { 0 }; 10246047Sahrens int err = 0; 10256047Sahrens prop_changelist_t *clp; 10266047Sahrens zfs_handle_t *zhp; 102710242Schris.kirby@sun.com boolean_t defer = B_FALSE; 102810242Schris.kirby@sun.com int spa_version; 10296047Sahrens 10306047Sahrens zhp = zfs_open(hdl, name, ZFS_TYPE_DATASET); 10316047Sahrens if (zhp == NULL) 10326047Sahrens return (-1); 10337366STim.Haley@Sun.COM clp = changelist_gather(zhp, ZFS_PROP_NAME, 0, 10347366STim.Haley@Sun.COM flags.force ? MS_FORCE : 0); 103510242Schris.kirby@sun.com if (zfs_get_type(zhp) == ZFS_TYPE_SNAPSHOT && 103610242Schris.kirby@sun.com zfs_spa_version(zhp, &spa_version) == 0 && 103710242Schris.kirby@sun.com spa_version >= SPA_VERSION_USERREFS) 103810242Schris.kirby@sun.com defer = B_TRUE; 10396047Sahrens zfs_close(zhp); 10406047Sahrens if (clp == NULL) 10416047Sahrens return (-1); 10426047Sahrens err = changelist_prefix(clp); 10436047Sahrens if (err) 10446047Sahrens return (err); 10455367Sahrens 10465367Sahrens zc.zc_objset_type = DMU_OST_ZFS; 104710242Schris.kirby@sun.com zc.zc_defer_destroy = defer; 10485367Sahrens (void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name)); 10495367Sahrens 10505367Sahrens if (flags.verbose) 10515367Sahrens (void) printf("attempting destroy %s\n", zc.zc_name); 10525367Sahrens err = ioctl(hdl->libzfs_fd, ZFS_IOC_DESTROY, &zc); 10536047Sahrens if (err == 0) { 10546047Sahrens if (flags.verbose) 10556047Sahrens (void) printf("success\n"); 10566047Sahrens changelist_remove(clp, zc.zc_name); 10575367Sahrens } 10585367Sahrens 10596047Sahrens (void) changelist_postfix(clp); 10606047Sahrens changelist_free(clp); 10616047Sahrens 106210242Schris.kirby@sun.com /* 106310242Schris.kirby@sun.com * Deferred destroy should always succeed. Since we can't tell 106410242Schris.kirby@sun.com * if it destroyed the dataset or just marked it for deferred 106510242Schris.kirby@sun.com * destroy, always do the rename just in case. 106610242Schris.kirby@sun.com */ 106710242Schris.kirby@sun.com if (err != 0 || defer) 10686047Sahrens err = recv_rename(hdl, name, NULL, baselen, newname, flags); 10695367Sahrens 10705367Sahrens return (err); 10715367Sahrens } 10725367Sahrens 10735367Sahrens typedef struct guid_to_name_data { 10745367Sahrens uint64_t guid; 10755367Sahrens char *name; 10765367Sahrens } guid_to_name_data_t; 10775367Sahrens 10785367Sahrens static int 10795367Sahrens guid_to_name_cb(zfs_handle_t *zhp, void *arg) 10805367Sahrens { 10815367Sahrens guid_to_name_data_t *gtnd = arg; 10825367Sahrens int err; 10835367Sahrens 10845367Sahrens if (zhp->zfs_dmustats.dds_guid == gtnd->guid) { 10855367Sahrens (void) strcpy(gtnd->name, zhp->zfs_name); 10865367Sahrens return (EEXIST); 10875367Sahrens } 10885367Sahrens err = zfs_iter_children(zhp, guid_to_name_cb, gtnd); 10895367Sahrens zfs_close(zhp); 10905367Sahrens return (err); 10915367Sahrens } 10925367Sahrens 10935367Sahrens static int 10945367Sahrens guid_to_name(libzfs_handle_t *hdl, const char *parent, uint64_t guid, 10955367Sahrens char *name) 10965367Sahrens { 10975367Sahrens /* exhaustive search all local snapshots */ 10985367Sahrens guid_to_name_data_t gtnd; 10995367Sahrens int err = 0; 11005367Sahrens zfs_handle_t *zhp; 11015367Sahrens char *cp; 11025367Sahrens 11035367Sahrens gtnd.guid = guid; 11045367Sahrens gtnd.name = name; 11055367Sahrens 11065367Sahrens if (strchr(parent, '@') == NULL) { 11075367Sahrens zhp = make_dataset_handle(hdl, parent); 11085367Sahrens if (zhp != NULL) { 11095367Sahrens err = zfs_iter_children(zhp, guid_to_name_cb, >nd); 11105367Sahrens zfs_close(zhp); 11115367Sahrens if (err == EEXIST) 11125367Sahrens return (0); 11135367Sahrens } 11145367Sahrens } 11155367Sahrens 11165367Sahrens cp = strchr(parent, '/'); 11175367Sahrens if (cp) 11185367Sahrens *cp = '\0'; 11195367Sahrens zhp = make_dataset_handle(hdl, parent); 11205367Sahrens if (cp) 11215367Sahrens *cp = '/'; 11225367Sahrens 11235367Sahrens if (zhp) { 11245367Sahrens err = zfs_iter_children(zhp, guid_to_name_cb, >nd); 11255367Sahrens zfs_close(zhp); 11265367Sahrens } 11275367Sahrens 11285367Sahrens return (err == EEXIST ? 0 : ENOENT); 11295367Sahrens 11305367Sahrens } 11315367Sahrens 11325367Sahrens /* 11335367Sahrens * Return true if dataset guid1 is created before guid2. 11345367Sahrens */ 11356476Sahl static int 11365367Sahrens created_before(libzfs_handle_t *hdl, avl_tree_t *avl, 11375367Sahrens uint64_t guid1, uint64_t guid2) 11385367Sahrens { 11395367Sahrens nvlist_t *nvfs; 11405367Sahrens char *fsname, *snapname; 11415367Sahrens char buf[ZFS_MAXNAMELEN]; 11426476Sahl int rv; 11435367Sahrens zfs_node_t zn1, zn2; 11445367Sahrens 11455367Sahrens if (guid2 == 0) 11466476Sahl return (0); 11475367Sahrens if (guid1 == 0) 11486476Sahl return (1); 11495367Sahrens 11505367Sahrens nvfs = fsavl_find(avl, guid1, &snapname); 11515367Sahrens VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname)); 11525367Sahrens (void) snprintf(buf, sizeof (buf), "%s@%s", fsname, snapname); 11535367Sahrens zn1.zn_handle = zfs_open(hdl, buf, ZFS_TYPE_SNAPSHOT); 11546476Sahl if (zn1.zn_handle == NULL) 11556476Sahl return (-1); 11565367Sahrens 11575367Sahrens nvfs = fsavl_find(avl, guid2, &snapname); 11585367Sahrens VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname)); 11595367Sahrens (void) snprintf(buf, sizeof (buf), "%s@%s", fsname, snapname); 11605367Sahrens zn2.zn_handle = zfs_open(hdl, buf, ZFS_TYPE_SNAPSHOT); 11616476Sahl if (zn2.zn_handle == NULL) { 11626476Sahl zfs_close(zn2.zn_handle); 11636476Sahl return (-1); 11646476Sahl } 11655367Sahrens 11665367Sahrens rv = (zfs_snapshot_compare(&zn1, &zn2) == -1); 11675367Sahrens 11685367Sahrens zfs_close(zn1.zn_handle); 11695367Sahrens zfs_close(zn2.zn_handle); 11705367Sahrens 11715367Sahrens return (rv); 11725367Sahrens } 11735367Sahrens 11745367Sahrens static int 11755367Sahrens recv_incremental_replication(libzfs_handle_t *hdl, const char *tofs, 11765367Sahrens recvflags_t flags, nvlist_t *stream_nv, avl_tree_t *stream_avl) 11775367Sahrens { 11785367Sahrens nvlist_t *local_nv; 11795367Sahrens avl_tree_t *local_avl; 11805367Sahrens nvpair_t *fselem, *nextfselem; 11815367Sahrens char *tosnap, *fromsnap; 11825367Sahrens char newname[ZFS_MAXNAMELEN]; 11835367Sahrens int error; 11845367Sahrens boolean_t needagain, progress; 118510299SPrabahar.Jeyaram@Sun.COM char *s1, *s2; 11865367Sahrens 11875367Sahrens VERIFY(0 == nvlist_lookup_string(stream_nv, "fromsnap", &fromsnap)); 11885367Sahrens VERIFY(0 == nvlist_lookup_string(stream_nv, "tosnap", &tosnap)); 11895367Sahrens 11905367Sahrens if (flags.dryrun) 11915367Sahrens return (0); 11925367Sahrens 11935367Sahrens again: 11945367Sahrens needagain = progress = B_FALSE; 11955367Sahrens 11965367Sahrens if ((error = gather_nvlist(hdl, tofs, fromsnap, NULL, 11975367Sahrens &local_nv, &local_avl)) != 0) 11985367Sahrens return (error); 11995367Sahrens 12005367Sahrens /* 12015367Sahrens * Process deletes and renames 12025367Sahrens */ 12035367Sahrens for (fselem = nvlist_next_nvpair(local_nv, NULL); 12045367Sahrens fselem; fselem = nextfselem) { 12055367Sahrens nvlist_t *nvfs, *snaps; 12065367Sahrens nvlist_t *stream_nvfs = NULL; 12075367Sahrens nvpair_t *snapelem, *nextsnapelem; 12085367Sahrens uint64_t fromguid = 0; 12095367Sahrens uint64_t originguid = 0; 12105367Sahrens uint64_t stream_originguid = 0; 12115367Sahrens uint64_t parent_fromsnap_guid, stream_parent_fromsnap_guid; 12125367Sahrens char *fsname, *stream_fsname; 12135367Sahrens 12145367Sahrens nextfselem = nvlist_next_nvpair(local_nv, fselem); 12155367Sahrens 12165367Sahrens VERIFY(0 == nvpair_value_nvlist(fselem, &nvfs)); 12175367Sahrens VERIFY(0 == nvlist_lookup_nvlist(nvfs, "snaps", &snaps)); 12185367Sahrens VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname)); 12195367Sahrens VERIFY(0 == nvlist_lookup_uint64(nvfs, "parentfromsnap", 12205367Sahrens &parent_fromsnap_guid)); 12215367Sahrens (void) nvlist_lookup_uint64(nvfs, "origin", &originguid); 12225367Sahrens 12235367Sahrens /* 12245367Sahrens * First find the stream's fs, so we can check for 12255367Sahrens * a different origin (due to "zfs promote") 12265367Sahrens */ 12275367Sahrens for (snapelem = nvlist_next_nvpair(snaps, NULL); 12285367Sahrens snapelem; snapelem = nvlist_next_nvpair(snaps, snapelem)) { 12295367Sahrens uint64_t thisguid; 12305367Sahrens 12315367Sahrens VERIFY(0 == nvpair_value_uint64(snapelem, &thisguid)); 12325367Sahrens stream_nvfs = fsavl_find(stream_avl, thisguid, NULL); 12335367Sahrens 12345367Sahrens if (stream_nvfs != NULL) 12355367Sahrens break; 12365367Sahrens } 12375367Sahrens 12385367Sahrens /* check for promote */ 12395367Sahrens (void) nvlist_lookup_uint64(stream_nvfs, "origin", 12405367Sahrens &stream_originguid); 12415367Sahrens if (stream_nvfs && originguid != stream_originguid) { 12426476Sahl switch (created_before(hdl, local_avl, 12436476Sahl stream_originguid, originguid)) { 12446476Sahl case 1: { 12455367Sahrens /* promote it! */ 12465367Sahrens zfs_cmd_t zc = { 0 }; 12475367Sahrens nvlist_t *origin_nvfs; 12485367Sahrens char *origin_fsname; 12495367Sahrens 12505367Sahrens if (flags.verbose) 12515367Sahrens (void) printf("promoting %s\n", fsname); 12525367Sahrens 12535367Sahrens origin_nvfs = fsavl_find(local_avl, originguid, 12545367Sahrens NULL); 12555367Sahrens VERIFY(0 == nvlist_lookup_string(origin_nvfs, 12565367Sahrens "name", &origin_fsname)); 12575367Sahrens (void) strlcpy(zc.zc_value, origin_fsname, 12585367Sahrens sizeof (zc.zc_value)); 12595367Sahrens (void) strlcpy(zc.zc_name, fsname, 12605367Sahrens sizeof (zc.zc_name)); 12615367Sahrens error = zfs_ioctl(hdl, ZFS_IOC_PROMOTE, &zc); 12625367Sahrens if (error == 0) 12635367Sahrens progress = B_TRUE; 12646476Sahl break; 12656476Sahl } 12666476Sahl default: 12676476Sahl break; 12686476Sahl case -1: 12696476Sahl fsavl_destroy(local_avl); 12706476Sahl nvlist_free(local_nv); 12716476Sahl return (-1); 12725367Sahrens } 12735367Sahrens /* 12745367Sahrens * We had/have the wrong origin, therefore our 12755367Sahrens * list of snapshots is wrong. Need to handle 12765367Sahrens * them on the next pass. 12775367Sahrens */ 12785367Sahrens needagain = B_TRUE; 12795367Sahrens continue; 12805367Sahrens } 12815367Sahrens 12825367Sahrens for (snapelem = nvlist_next_nvpair(snaps, NULL); 12835367Sahrens snapelem; snapelem = nextsnapelem) { 12845367Sahrens uint64_t thisguid; 12855367Sahrens char *stream_snapname; 12867265Sahrens nvlist_t *found, *props; 12875367Sahrens 12885367Sahrens nextsnapelem = nvlist_next_nvpair(snaps, snapelem); 12895367Sahrens 12905367Sahrens VERIFY(0 == nvpair_value_uint64(snapelem, &thisguid)); 12915367Sahrens found = fsavl_find(stream_avl, thisguid, 12925367Sahrens &stream_snapname); 12935367Sahrens 12945367Sahrens /* check for delete */ 12955367Sahrens if (found == NULL) { 12965367Sahrens char name[ZFS_MAXNAMELEN]; 12975367Sahrens 12985367Sahrens if (!flags.force) 12995367Sahrens continue; 13005367Sahrens 13015367Sahrens (void) snprintf(name, sizeof (name), "%s@%s", 13025367Sahrens fsname, nvpair_name(snapelem)); 13035367Sahrens 13045367Sahrens error = recv_destroy(hdl, name, 13055367Sahrens strlen(fsname)+1, newname, flags); 13065367Sahrens if (error) 13075367Sahrens needagain = B_TRUE; 13085367Sahrens else 13095367Sahrens progress = B_TRUE; 13105367Sahrens continue; 13115367Sahrens } 13125367Sahrens 13135367Sahrens stream_nvfs = found; 13145367Sahrens 13157265Sahrens if (0 == nvlist_lookup_nvlist(stream_nvfs, "snapprops", 13167265Sahrens &props) && 0 == nvlist_lookup_nvlist(props, 13177265Sahrens stream_snapname, &props)) { 13187265Sahrens zfs_cmd_t zc = { 0 }; 13197265Sahrens 13207265Sahrens zc.zc_cookie = B_TRUE; /* clear current props */ 13217273Sahrens (void) snprintf(zc.zc_name, sizeof (zc.zc_name), 13227265Sahrens "%s@%s", fsname, nvpair_name(snapelem)); 13237265Sahrens if (zcmd_write_src_nvlist(hdl, &zc, 13247265Sahrens props) == 0) { 13257265Sahrens (void) zfs_ioctl(hdl, 13267265Sahrens ZFS_IOC_SET_PROP, &zc); 13277265Sahrens zcmd_free_nvlists(&zc); 13287265Sahrens } 13297265Sahrens } 13307265Sahrens 13315367Sahrens /* check for different snapname */ 13325367Sahrens if (strcmp(nvpair_name(snapelem), 13335367Sahrens stream_snapname) != 0) { 13345367Sahrens char name[ZFS_MAXNAMELEN]; 13355367Sahrens char tryname[ZFS_MAXNAMELEN]; 13365367Sahrens 13375367Sahrens (void) snprintf(name, sizeof (name), "%s@%s", 13385367Sahrens fsname, nvpair_name(snapelem)); 13395367Sahrens (void) snprintf(tryname, sizeof (name), "%s@%s", 13405367Sahrens fsname, stream_snapname); 13415367Sahrens 13425367Sahrens error = recv_rename(hdl, name, tryname, 13435367Sahrens strlen(fsname)+1, newname, flags); 13445367Sahrens if (error) 13455367Sahrens needagain = B_TRUE; 13465367Sahrens else 13475367Sahrens progress = B_TRUE; 13485367Sahrens } 13495367Sahrens 13505367Sahrens if (strcmp(stream_snapname, fromsnap) == 0) 13515367Sahrens fromguid = thisguid; 13525367Sahrens } 13535367Sahrens 13545367Sahrens /* check for delete */ 13555367Sahrens if (stream_nvfs == NULL) { 13565367Sahrens if (!flags.force) 13575367Sahrens continue; 13585367Sahrens 13595367Sahrens error = recv_destroy(hdl, fsname, strlen(tofs)+1, 13605367Sahrens newname, flags); 13615367Sahrens if (error) 13625367Sahrens needagain = B_TRUE; 13635367Sahrens else 13645367Sahrens progress = B_TRUE; 13655367Sahrens continue; 13665367Sahrens } 13675367Sahrens 13685367Sahrens if (fromguid == 0 && flags.verbose) { 13695367Sahrens (void) printf("local fs %s does not have fromsnap " 13705367Sahrens "(%s in stream); must have been deleted locally; " 13715367Sahrens "ignoring\n", fsname, fromsnap); 13725367Sahrens continue; 13735367Sahrens } 13745367Sahrens 13755367Sahrens VERIFY(0 == nvlist_lookup_string(stream_nvfs, 13765367Sahrens "name", &stream_fsname)); 13775367Sahrens VERIFY(0 == nvlist_lookup_uint64(stream_nvfs, 13785367Sahrens "parentfromsnap", &stream_parent_fromsnap_guid)); 13795367Sahrens 138010302SPrabahar.Jeyaram@Sun.COM s1 = strrchr(fsname, '/'); 138110302SPrabahar.Jeyaram@Sun.COM s2 = strrchr(stream_fsname, '/'); 138210302SPrabahar.Jeyaram@Sun.COM 13835367Sahrens /* check for rename */ 13845367Sahrens if ((stream_parent_fromsnap_guid != 0 && 13855367Sahrens stream_parent_fromsnap_guid != parent_fromsnap_guid) || 138610302SPrabahar.Jeyaram@Sun.COM ((s1 != NULL) && (s2 != NULL) && strcmp(s1, s2) != 0)) { 13875367Sahrens nvlist_t *parent; 13885367Sahrens char tryname[ZFS_MAXNAMELEN]; 13895367Sahrens 13905367Sahrens parent = fsavl_find(local_avl, 13915367Sahrens stream_parent_fromsnap_guid, NULL); 13925367Sahrens /* 13935367Sahrens * NB: parent might not be found if we used the 13945367Sahrens * tosnap for stream_parent_fromsnap_guid, 13955367Sahrens * because the parent is a newly-created fs; 13965367Sahrens * we'll be able to rename it after we recv the 13975367Sahrens * new fs. 13985367Sahrens */ 13995367Sahrens if (parent != NULL) { 14005367Sahrens char *pname; 14015367Sahrens 14025367Sahrens VERIFY(0 == nvlist_lookup_string(parent, "name", 14035367Sahrens &pname)); 14045367Sahrens (void) snprintf(tryname, sizeof (tryname), 14055367Sahrens "%s%s", pname, strrchr(stream_fsname, '/')); 14065367Sahrens } else { 14075367Sahrens tryname[0] = '\0'; 14085367Sahrens if (flags.verbose) { 14095367Sahrens (void) printf("local fs %s new parent " 14105367Sahrens "not found\n", fsname); 14115367Sahrens } 14125367Sahrens } 14135367Sahrens 14145367Sahrens error = recv_rename(hdl, fsname, tryname, 14155367Sahrens strlen(tofs)+1, newname, flags); 14165367Sahrens if (error) 14175367Sahrens needagain = B_TRUE; 14185367Sahrens else 14195367Sahrens progress = B_TRUE; 14205367Sahrens } 14215367Sahrens } 14225367Sahrens 14235367Sahrens fsavl_destroy(local_avl); 14245367Sahrens nvlist_free(local_nv); 14255367Sahrens 14265367Sahrens if (needagain && progress) { 14275367Sahrens /* do another pass to fix up temporary names */ 14285367Sahrens if (flags.verbose) 14295367Sahrens (void) printf("another pass:\n"); 14305367Sahrens goto again; 14315367Sahrens } 14325367Sahrens 14335367Sahrens return (needagain); 14345367Sahrens } 14355367Sahrens 14365367Sahrens static int 14375367Sahrens zfs_receive_package(libzfs_handle_t *hdl, int fd, const char *destname, 14387366STim.Haley@Sun.COM recvflags_t flags, dmu_replay_record_t *drr, zio_cksum_t *zc, 14397366STim.Haley@Sun.COM char **top_zfs) 14405367Sahrens { 14415367Sahrens nvlist_t *stream_nv = NULL; 14425367Sahrens avl_tree_t *stream_avl = NULL; 14435367Sahrens char *fromsnap = NULL; 14445367Sahrens char tofs[ZFS_MAXNAMELEN]; 14455367Sahrens char errbuf[1024]; 14465367Sahrens dmu_replay_record_t drre; 14475367Sahrens int error; 14485367Sahrens boolean_t anyerr = B_FALSE; 14496479Sahrens boolean_t softerr = B_FALSE; 14505367Sahrens 14515367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 14525367Sahrens "cannot receive")); 14535367Sahrens 14545367Sahrens if (strchr(destname, '@')) { 14555367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 14565367Sahrens "can not specify snapshot name for multi-snapshot stream")); 14575367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, errbuf)); 14585367Sahrens } 14595367Sahrens 14605367Sahrens assert(drr->drr_type == DRR_BEGIN); 14615367Sahrens assert(drr->drr_u.drr_begin.drr_magic == DMU_BACKUP_MAGIC); 14625367Sahrens assert(drr->drr_u.drr_begin.drr_version == DMU_BACKUP_HEADER_VERSION); 14635367Sahrens 14645367Sahrens /* 14655367Sahrens * Read in the nvlist from the stream. 14665367Sahrens */ 14675367Sahrens if (drr->drr_payloadlen != 0) { 14685367Sahrens if (!flags.isprefix) { 14695367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 14705367Sahrens "must use -d to receive replication " 14715367Sahrens "(send -R) stream")); 14725367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, errbuf)); 14735367Sahrens } 14745367Sahrens 14755367Sahrens error = recv_read_nvlist(hdl, fd, drr->drr_payloadlen, 14765367Sahrens &stream_nv, flags.byteswap, zc); 14775367Sahrens if (error) { 14785367Sahrens error = zfs_error(hdl, EZFS_BADSTREAM, errbuf); 14795367Sahrens goto out; 14805367Sahrens } 14815367Sahrens } 14825367Sahrens 14835367Sahrens /* 14845367Sahrens * Read in the end record and verify checksum. 14855367Sahrens */ 14865367Sahrens if (0 != (error = recv_read(hdl, fd, &drre, sizeof (drre), 14875367Sahrens flags.byteswap, NULL))) 14885367Sahrens goto out; 14895367Sahrens if (flags.byteswap) { 14905367Sahrens drre.drr_type = BSWAP_32(drre.drr_type); 14915367Sahrens drre.drr_u.drr_end.drr_checksum.zc_word[0] = 14925367Sahrens BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[0]); 14935367Sahrens drre.drr_u.drr_end.drr_checksum.zc_word[1] = 14945367Sahrens BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[1]); 14955367Sahrens drre.drr_u.drr_end.drr_checksum.zc_word[2] = 14965367Sahrens BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[2]); 14975367Sahrens drre.drr_u.drr_end.drr_checksum.zc_word[3] = 14985367Sahrens BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[3]); 14995367Sahrens } 15005367Sahrens if (drre.drr_type != DRR_END) { 15015367Sahrens error = zfs_error(hdl, EZFS_BADSTREAM, errbuf); 15025367Sahrens goto out; 15035367Sahrens } 15045367Sahrens if (!ZIO_CHECKSUM_EQUAL(drre.drr_u.drr_end.drr_checksum, *zc)) { 15055367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 15065367Sahrens "incorrect header checksum")); 15075367Sahrens error = zfs_error(hdl, EZFS_BADSTREAM, errbuf); 15085367Sahrens goto out; 15095367Sahrens } 15105367Sahrens 15115367Sahrens (void) nvlist_lookup_string(stream_nv, "fromsnap", &fromsnap); 15125367Sahrens 15135367Sahrens if (drr->drr_payloadlen != 0) { 15145367Sahrens nvlist_t *stream_fss; 15155367Sahrens 15165367Sahrens VERIFY(0 == nvlist_lookup_nvlist(stream_nv, "fss", 15175367Sahrens &stream_fss)); 15185896Sahl if ((stream_avl = fsavl_create(stream_fss)) == NULL) { 15195896Sahl zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 15205896Sahl "couldn't allocate avl tree")); 15215896Sahl error = zfs_error(hdl, EZFS_NOMEM, errbuf); 15225896Sahl goto out; 15235896Sahl } 15245367Sahrens 15255367Sahrens if (fromsnap != NULL) { 15265367Sahrens (void) strlcpy(tofs, destname, ZFS_MAXNAMELEN); 15275367Sahrens if (flags.isprefix) { 15285367Sahrens int i = strcspn(drr->drr_u.drr_begin.drr_toname, 15295367Sahrens "/@"); 15305367Sahrens /* zfs_receive_one() will create_parents() */ 15315367Sahrens (void) strlcat(tofs, 15325367Sahrens &drr->drr_u.drr_begin.drr_toname[i], 15335367Sahrens ZFS_MAXNAMELEN); 15345367Sahrens *strchr(tofs, '@') = '\0'; 15355367Sahrens } 15366479Sahrens softerr = recv_incremental_replication(hdl, tofs, 15375367Sahrens flags, stream_nv, stream_avl); 15385367Sahrens } 15395367Sahrens } 15405367Sahrens 15415367Sahrens 15425367Sahrens /* Finally, receive each contained stream */ 15435367Sahrens do { 15445367Sahrens /* 15455367Sahrens * we should figure out if it has a recoverable 15465367Sahrens * error, in which case do a recv_skip() and drive on. 15475367Sahrens * Note, if we fail due to already having this guid, 15485367Sahrens * zfs_receive_one() will take care of it (ie, 15495367Sahrens * recv_skip() and return 0). 15505367Sahrens */ 15517366STim.Haley@Sun.COM error = zfs_receive_impl(hdl, destname, flags, fd, 15527366STim.Haley@Sun.COM stream_avl, top_zfs); 15535367Sahrens if (error == ENODATA) { 15545367Sahrens error = 0; 15555367Sahrens break; 15565367Sahrens } 15575367Sahrens anyerr |= error; 15585367Sahrens } while (error == 0); 15595367Sahrens 15605367Sahrens if (drr->drr_payloadlen != 0 && fromsnap != NULL) { 15615367Sahrens /* 15625367Sahrens * Now that we have the fs's they sent us, try the 15635367Sahrens * renames again. 15645367Sahrens */ 15656479Sahrens softerr = recv_incremental_replication(hdl, tofs, flags, 15665367Sahrens stream_nv, stream_avl); 15675367Sahrens } 15685367Sahrens 15695367Sahrens out: 15705367Sahrens fsavl_destroy(stream_avl); 15715367Sahrens if (stream_nv) 15725367Sahrens nvlist_free(stream_nv); 15736479Sahrens if (softerr) 15746479Sahrens error = -2; 15755367Sahrens if (anyerr) 15765367Sahrens error = -1; 15775367Sahrens return (error); 15785367Sahrens } 15795367Sahrens 15805367Sahrens static int 15815367Sahrens recv_skip(libzfs_handle_t *hdl, int fd, boolean_t byteswap) 15825367Sahrens { 15835367Sahrens dmu_replay_record_t *drr; 15845367Sahrens void *buf = malloc(1<<20); 15855367Sahrens 15865367Sahrens /* XXX would be great to use lseek if possible... */ 15875367Sahrens drr = buf; 15885367Sahrens 15895367Sahrens while (recv_read(hdl, fd, drr, sizeof (dmu_replay_record_t), 15905367Sahrens byteswap, NULL) == 0) { 15915367Sahrens if (byteswap) 15925367Sahrens drr->drr_type = BSWAP_32(drr->drr_type); 15935367Sahrens 15945367Sahrens switch (drr->drr_type) { 15955367Sahrens case DRR_BEGIN: 15965367Sahrens /* NB: not to be used on v2 stream packages */ 15975367Sahrens assert(drr->drr_payloadlen == 0); 15985367Sahrens break; 15995367Sahrens 16005367Sahrens case DRR_END: 16015367Sahrens free(buf); 16025367Sahrens return (0); 16035367Sahrens 16045367Sahrens case DRR_OBJECT: 16055367Sahrens if (byteswap) { 16065367Sahrens drr->drr_u.drr_object.drr_bonuslen = 16075367Sahrens BSWAP_32(drr->drr_u.drr_object. 16085367Sahrens drr_bonuslen); 16095367Sahrens } 16105367Sahrens (void) recv_read(hdl, fd, buf, 16115367Sahrens P2ROUNDUP(drr->drr_u.drr_object.drr_bonuslen, 8), 16125367Sahrens B_FALSE, NULL); 16135367Sahrens break; 16145367Sahrens 16155367Sahrens case DRR_WRITE: 16165367Sahrens if (byteswap) { 16175367Sahrens drr->drr_u.drr_write.drr_length = 16187366STim.Haley@Sun.COM BSWAP_64(drr->drr_u.drr_write.drr_length); 16195367Sahrens } 16205367Sahrens (void) recv_read(hdl, fd, buf, 16215367Sahrens drr->drr_u.drr_write.drr_length, B_FALSE, NULL); 16225367Sahrens break; 16235367Sahrens 16245367Sahrens case DRR_FREEOBJECTS: 16255367Sahrens case DRR_FREE: 16265367Sahrens break; 16275367Sahrens 16285367Sahrens default: 16295367Sahrens assert(!"invalid record type"); 16305367Sahrens } 16315367Sahrens } 16325367Sahrens 16335367Sahrens free(buf); 16345367Sahrens return (-1); 16355367Sahrens } 16365367Sahrens 16375367Sahrens /* 16385367Sahrens * Restores a backup of tosnap from the file descriptor specified by infd. 16395367Sahrens */ 16405367Sahrens static int 16415367Sahrens zfs_receive_one(libzfs_handle_t *hdl, int infd, const char *tosnap, 16425367Sahrens recvflags_t flags, dmu_replay_record_t *drr, 16437366STim.Haley@Sun.COM dmu_replay_record_t *drr_noswap, avl_tree_t *stream_avl, 16447366STim.Haley@Sun.COM char **top_zfs) 16455367Sahrens { 16465367Sahrens zfs_cmd_t zc = { 0 }; 16475367Sahrens time_t begin_time; 16485367Sahrens int ioctl_err, ioctl_errno, err, choplen; 16495367Sahrens char *cp; 16505367Sahrens struct drr_begin *drrb = &drr->drr_u.drr_begin; 16515367Sahrens char errbuf[1024]; 16525367Sahrens char chopprefix[ZFS_MAXNAMELEN]; 16535367Sahrens boolean_t newfs = B_FALSE; 16545367Sahrens boolean_t stream_wantsnewfs; 16555367Sahrens uint64_t parent_snapguid = 0; 16565367Sahrens prop_changelist_t *clp = NULL; 16577265Sahrens nvlist_t *snapprops_nvlist = NULL; 16585367Sahrens 16595367Sahrens begin_time = time(NULL); 16605367Sahrens 16615367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 16625367Sahrens "cannot receive")); 16635367Sahrens 16645367Sahrens if (stream_avl != NULL) { 16657265Sahrens char *snapname; 16667265Sahrens nvlist_t *fs = fsavl_find(stream_avl, drrb->drr_toguid, 16677265Sahrens &snapname); 16685367Sahrens nvlist_t *props; 16695896Sahl int ret; 16705367Sahrens 16715367Sahrens (void) nvlist_lookup_uint64(fs, "parentfromsnap", 16725367Sahrens &parent_snapguid); 16735367Sahrens err = nvlist_lookup_nvlist(fs, "props", &props); 16745367Sahrens if (err) 16755367Sahrens VERIFY(0 == nvlist_alloc(&props, NV_UNIQUE_NAME, 0)); 16765896Sahl 16775367Sahrens if (flags.canmountoff) { 16785367Sahrens VERIFY(0 == nvlist_add_uint64(props, 16795367Sahrens zfs_prop_to_name(ZFS_PROP_CANMOUNT), 0)); 16805367Sahrens } 16815896Sahl ret = zcmd_write_src_nvlist(hdl, &zc, props); 16825367Sahrens if (err) 16835367Sahrens nvlist_free(props); 16845896Sahl 16857265Sahrens if (0 == nvlist_lookup_nvlist(fs, "snapprops", &props)) { 16867265Sahrens VERIFY(0 == nvlist_lookup_nvlist(props, 16877265Sahrens snapname, &snapprops_nvlist)); 16887265Sahrens } 16897265Sahrens 16905896Sahl if (ret != 0) 16915896Sahl return (-1); 16925367Sahrens } 16935367Sahrens 16945367Sahrens /* 16955367Sahrens * Determine how much of the snapshot name stored in the stream 16965367Sahrens * we are going to tack on to the name they specified on the 16975367Sahrens * command line, and how much we are going to chop off. 16985367Sahrens * 16995367Sahrens * If they specified a snapshot, chop the entire name stored in 17005367Sahrens * the stream. 17015367Sahrens */ 17025367Sahrens (void) strcpy(chopprefix, drrb->drr_toname); 17035367Sahrens if (flags.isprefix) { 17045367Sahrens /* 17055367Sahrens * They specified a fs with -d, we want to tack on 17065367Sahrens * everything but the pool name stored in the stream 17075367Sahrens */ 17085367Sahrens if (strchr(tosnap, '@')) { 17095367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid " 17105367Sahrens "argument - snapshot not allowed with -d")); 17115367Sahrens return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf)); 17125367Sahrens } 17135367Sahrens cp = strchr(chopprefix, '/'); 17145367Sahrens if (cp == NULL) 17155367Sahrens cp = strchr(chopprefix, '@'); 17165367Sahrens *cp = '\0'; 17175367Sahrens } else if (strchr(tosnap, '@') == NULL) { 17185367Sahrens /* 17195367Sahrens * If they specified a filesystem without -d, we want to 17205367Sahrens * tack on everything after the fs specified in the 17215367Sahrens * first name from the stream. 17225367Sahrens */ 17235367Sahrens cp = strchr(chopprefix, '@'); 17245367Sahrens *cp = '\0'; 17255367Sahrens } 17265367Sahrens choplen = strlen(chopprefix); 17275367Sahrens 17285367Sahrens /* 17295367Sahrens * Determine name of destination snapshot, store in zc_value. 17305367Sahrens */ 17315367Sahrens (void) strcpy(zc.zc_value, tosnap); 17325367Sahrens (void) strncat(zc.zc_value, drrb->drr_toname+choplen, 17335367Sahrens sizeof (zc.zc_value)); 17345896Sahl if (!zfs_name_valid(zc.zc_value, ZFS_TYPE_SNAPSHOT)) { 17355896Sahl zcmd_free_nvlists(&zc); 17365367Sahrens return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf)); 17375896Sahl } 17385367Sahrens 17395367Sahrens /* 17405367Sahrens * Determine the name of the origin snapshot, store in zc_string. 17415367Sahrens */ 17425367Sahrens if (drrb->drr_flags & DRR_FLAG_CLONE) { 17435367Sahrens if (guid_to_name(hdl, tosnap, 17445367Sahrens drrb->drr_fromguid, zc.zc_string) != 0) { 17455896Sahl zcmd_free_nvlists(&zc); 17465367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 17475367Sahrens "local origin for clone %s does not exist"), 17485367Sahrens zc.zc_value); 17495367Sahrens return (zfs_error(hdl, EZFS_NOENT, errbuf)); 17505367Sahrens } 17515367Sahrens if (flags.verbose) 17525367Sahrens (void) printf("found clone origin %s\n", zc.zc_string); 17535367Sahrens } 17545367Sahrens 17555367Sahrens stream_wantsnewfs = (drrb->drr_fromguid == NULL || 17565367Sahrens (drrb->drr_flags & DRR_FLAG_CLONE)); 17575367Sahrens 17585367Sahrens if (stream_wantsnewfs) { 17595367Sahrens /* 17605367Sahrens * if the parent fs does not exist, look for it based on 17615367Sahrens * the parent snap GUID 17625367Sahrens */ 17635367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 17645367Sahrens "cannot receive new filesystem stream")); 17655367Sahrens 17665367Sahrens (void) strcpy(zc.zc_name, zc.zc_value); 17675367Sahrens cp = strrchr(zc.zc_name, '/'); 17685367Sahrens if (cp) 17695367Sahrens *cp = '\0'; 17705367Sahrens if (cp && 17715367Sahrens !zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) { 17725367Sahrens char suffix[ZFS_MAXNAMELEN]; 17735367Sahrens (void) strcpy(suffix, strrchr(zc.zc_value, '/')); 17745367Sahrens if (guid_to_name(hdl, tosnap, parent_snapguid, 17755367Sahrens zc.zc_value) == 0) { 17765367Sahrens *strchr(zc.zc_value, '@') = '\0'; 17775367Sahrens (void) strcat(zc.zc_value, suffix); 17785367Sahrens } 17795367Sahrens } 17805367Sahrens } else { 17815367Sahrens /* 17825367Sahrens * if the fs does not exist, look for it based on the 17835367Sahrens * fromsnap GUID 17845367Sahrens */ 17855367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 17865367Sahrens "cannot receive incremental stream")); 17875367Sahrens 17885367Sahrens (void) strcpy(zc.zc_name, zc.zc_value); 17895367Sahrens *strchr(zc.zc_name, '@') = '\0'; 17905367Sahrens 17915367Sahrens if (!zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) { 17925367Sahrens char snap[ZFS_MAXNAMELEN]; 17935367Sahrens (void) strcpy(snap, strchr(zc.zc_value, '@')); 17945367Sahrens if (guid_to_name(hdl, tosnap, drrb->drr_fromguid, 17955367Sahrens zc.zc_value) == 0) { 17965367Sahrens *strchr(zc.zc_value, '@') = '\0'; 17975367Sahrens (void) strcat(zc.zc_value, snap); 17985367Sahrens } 17995367Sahrens } 18005367Sahrens } 18015367Sahrens 18025367Sahrens (void) strcpy(zc.zc_name, zc.zc_value); 18035367Sahrens *strchr(zc.zc_name, '@') = '\0'; 18045367Sahrens 18055367Sahrens if (zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) { 18065367Sahrens zfs_handle_t *zhp; 18075367Sahrens /* 18085367Sahrens * Destination fs exists. Therefore this should either 18095367Sahrens * be an incremental, or the stream specifies a new fs 18105367Sahrens * (full stream or clone) and they want us to blow it 18115367Sahrens * away (and have therefore specified -F and removed any 18125367Sahrens * snapshots). 18135367Sahrens */ 18145367Sahrens 18155367Sahrens if (stream_wantsnewfs) { 18165367Sahrens if (!flags.force) { 18175896Sahl zcmd_free_nvlists(&zc); 18185367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 18195367Sahrens "destination '%s' exists\n" 18205367Sahrens "must specify -F to overwrite it"), 18215367Sahrens zc.zc_name); 18225367Sahrens return (zfs_error(hdl, EZFS_EXISTS, errbuf)); 18235367Sahrens } 18245367Sahrens if (ioctl(hdl->libzfs_fd, ZFS_IOC_SNAPSHOT_LIST_NEXT, 18255367Sahrens &zc) == 0) { 18265896Sahl zcmd_free_nvlists(&zc); 18275367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 18285367Sahrens "destination has snapshots (eg. %s)\n" 18295367Sahrens "must destroy them to overwrite it"), 18305367Sahrens zc.zc_name); 18315367Sahrens return (zfs_error(hdl, EZFS_EXISTS, errbuf)); 18325367Sahrens } 18335367Sahrens } 18345367Sahrens 18355896Sahl if ((zhp = zfs_open(hdl, zc.zc_name, 18365896Sahl ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME)) == NULL) { 18375896Sahl zcmd_free_nvlists(&zc); 18385367Sahrens return (-1); 18395896Sahl } 18405896Sahl 18415367Sahrens if (stream_wantsnewfs && 18425367Sahrens zhp->zfs_dmustats.dds_origin[0]) { 18435896Sahl zcmd_free_nvlists(&zc); 18445367Sahrens zfs_close(zhp); 18455367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 18465367Sahrens "destination '%s' is a clone\n" 18475367Sahrens "must destroy it to overwrite it"), 18485367Sahrens zc.zc_name); 18495367Sahrens return (zfs_error(hdl, EZFS_EXISTS, errbuf)); 18505367Sahrens } 18515367Sahrens 18525367Sahrens if (!flags.dryrun && zhp->zfs_type == ZFS_TYPE_FILESYSTEM && 18535367Sahrens stream_wantsnewfs) { 18545367Sahrens /* We can't do online recv in this case */ 18557366STim.Haley@Sun.COM clp = changelist_gather(zhp, ZFS_PROP_NAME, 0, 0); 18565896Sahl if (clp == NULL) { 185710200SLori.Alt@Sun.COM zfs_close(zhp); 18585896Sahl zcmd_free_nvlists(&zc); 18595367Sahrens return (-1); 18605896Sahl } 18615367Sahrens if (changelist_prefix(clp) != 0) { 18625367Sahrens changelist_free(clp); 186310200SLori.Alt@Sun.COM zfs_close(zhp); 18645896Sahl zcmd_free_nvlists(&zc); 18655367Sahrens return (-1); 18665367Sahrens } 18675367Sahrens } 18685367Sahrens zfs_close(zhp); 18695367Sahrens } else { 18705367Sahrens /* 18715896Sahl * Destination filesystem does not exist. Therefore we better 18725896Sahl * be creating a new filesystem (either from a full backup, or 18735896Sahl * a clone). It would therefore be invalid if the user 18745896Sahl * specified only the pool name (i.e. if the destination name 18755896Sahl * contained no slash character). 18765367Sahrens */ 18775896Sahl if (!stream_wantsnewfs || 18785896Sahl (cp = strrchr(zc.zc_name, '/')) == NULL) { 18795896Sahl zcmd_free_nvlists(&zc); 18805367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 18815367Sahrens "destination '%s' does not exist"), zc.zc_name); 18825367Sahrens return (zfs_error(hdl, EZFS_NOENT, errbuf)); 18835367Sahrens } 18845367Sahrens 18855896Sahl /* 18865896Sahl * Trim off the final dataset component so we perform the 18875896Sahl * recvbackup ioctl to the filesystems's parent. 18885896Sahl */ 18895896Sahl *cp = '\0'; 18905367Sahrens 18915896Sahl if (flags.isprefix && !flags.dryrun && 18925896Sahl create_parents(hdl, zc.zc_value, strlen(tosnap)) != 0) { 18935896Sahl zcmd_free_nvlists(&zc); 18945896Sahl return (zfs_error(hdl, EZFS_BADRESTORE, errbuf)); 18955367Sahrens } 18965367Sahrens 18975367Sahrens newfs = B_TRUE; 18985367Sahrens } 18995367Sahrens 19005367Sahrens zc.zc_begin_record = drr_noswap->drr_u.drr_begin; 19015367Sahrens zc.zc_cookie = infd; 19025367Sahrens zc.zc_guid = flags.force; 19035367Sahrens if (flags.verbose) { 19045367Sahrens (void) printf("%s %s stream of %s into %s\n", 19055367Sahrens flags.dryrun ? "would receive" : "receiving", 19065367Sahrens drrb->drr_fromguid ? "incremental" : "full", 19075367Sahrens drrb->drr_toname, zc.zc_value); 19085367Sahrens (void) fflush(stdout); 19095367Sahrens } 19105367Sahrens 19115896Sahl if (flags.dryrun) { 19125896Sahl zcmd_free_nvlists(&zc); 19135367Sahrens return (recv_skip(hdl, infd, flags.byteswap)); 19145896Sahl } 19155367Sahrens 19165367Sahrens err = ioctl_err = zfs_ioctl(hdl, ZFS_IOC_RECV, &zc); 19175367Sahrens ioctl_errno = errno; 19187265Sahrens zcmd_free_nvlists(&zc); 19197265Sahrens 19207265Sahrens if (err == 0 && snapprops_nvlist) { 19217265Sahrens zfs_cmd_t zc2 = { 0 }; 19227265Sahrens 19237273Sahrens (void) strcpy(zc2.zc_name, zc.zc_value); 19247265Sahrens if (zcmd_write_src_nvlist(hdl, &zc2, snapprops_nvlist) == 0) { 19257265Sahrens (void) zfs_ioctl(hdl, ZFS_IOC_SET_PROP, &zc2); 19267265Sahrens zcmd_free_nvlists(&zc2); 19277265Sahrens } 19287265Sahrens } 19297265Sahrens 19305367Sahrens if (err && (ioctl_errno == ENOENT || ioctl_errno == ENODEV)) { 19315367Sahrens /* 19325367Sahrens * It may be that this snapshot already exists, 19335367Sahrens * in which case we want to consume & ignore it 19345367Sahrens * rather than failing. 19355367Sahrens */ 19365367Sahrens avl_tree_t *local_avl; 19375367Sahrens nvlist_t *local_nv, *fs; 19385367Sahrens char *cp = strchr(zc.zc_value, '@'); 19395367Sahrens 19405367Sahrens /* 19415367Sahrens * XXX Do this faster by just iterating over snaps in 19425367Sahrens * this fs. Also if zc_value does not exist, we will 19435367Sahrens * get a strange "does not exist" error message. 19445367Sahrens */ 19455367Sahrens *cp = '\0'; 19465367Sahrens if (gather_nvlist(hdl, zc.zc_value, NULL, NULL, 19475367Sahrens &local_nv, &local_avl) == 0) { 19485367Sahrens *cp = '@'; 19495367Sahrens fs = fsavl_find(local_avl, drrb->drr_toguid, NULL); 19505367Sahrens fsavl_destroy(local_avl); 19515367Sahrens nvlist_free(local_nv); 19525367Sahrens 19535367Sahrens if (fs != NULL) { 19545367Sahrens if (flags.verbose) { 19555367Sahrens (void) printf("snap %s already exists; " 19565367Sahrens "ignoring\n", zc.zc_value); 19575367Sahrens } 19585367Sahrens ioctl_err = recv_skip(hdl, infd, 19595367Sahrens flags.byteswap); 19605367Sahrens } 19615367Sahrens } 19625367Sahrens *cp = '@'; 19635367Sahrens } 19645367Sahrens 19655896Sahl 19665367Sahrens if (ioctl_err != 0) { 19675367Sahrens switch (ioctl_errno) { 19685367Sahrens case ENODEV: 19695367Sahrens cp = strchr(zc.zc_value, '@'); 19705367Sahrens *cp = '\0'; 19715367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 19725367Sahrens "most recent snapshot of %s does not\n" 19735367Sahrens "match incremental source"), zc.zc_value); 19745367Sahrens (void) zfs_error(hdl, EZFS_BADRESTORE, errbuf); 19755367Sahrens *cp = '@'; 19765367Sahrens break; 19775367Sahrens case ETXTBSY: 19785367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 19795367Sahrens "destination %s has been modified\n" 19805367Sahrens "since most recent snapshot"), zc.zc_name); 19815367Sahrens (void) zfs_error(hdl, EZFS_BADRESTORE, errbuf); 19825367Sahrens break; 19835367Sahrens case EEXIST: 19845367Sahrens cp = strchr(zc.zc_value, '@'); 19855367Sahrens if (newfs) { 19865367Sahrens /* it's the containing fs that exists */ 19875367Sahrens *cp = '\0'; 19885367Sahrens } 19895367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 19905367Sahrens "destination already exists")); 19915367Sahrens (void) zfs_error_fmt(hdl, EZFS_EXISTS, 19925367Sahrens dgettext(TEXT_DOMAIN, "cannot restore to %s"), 19935367Sahrens zc.zc_value); 19945367Sahrens *cp = '@'; 19955367Sahrens break; 19965367Sahrens case EINVAL: 19975367Sahrens (void) zfs_error(hdl, EZFS_BADSTREAM, errbuf); 19985367Sahrens break; 19995367Sahrens case ECKSUM: 20005367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 20015367Sahrens "invalid stream (checksum mismatch)")); 20025367Sahrens (void) zfs_error(hdl, EZFS_BADSTREAM, errbuf); 20035367Sahrens break; 20045367Sahrens default: 20055367Sahrens (void) zfs_standard_error(hdl, ioctl_errno, errbuf); 20065367Sahrens } 20075367Sahrens } 20085367Sahrens 20095367Sahrens /* 201010588SEric.Taylor@Sun.COM * Mount the target filesystem (if created). Also mount any 201110588SEric.Taylor@Sun.COM * children of the target filesystem if we did a replication 201210588SEric.Taylor@Sun.COM * receive (indicated by stream_avl being non-NULL). 20135367Sahrens */ 20145367Sahrens cp = strchr(zc.zc_value, '@'); 20155367Sahrens if (cp && (ioctl_err == 0 || !newfs)) { 20165367Sahrens zfs_handle_t *h; 20175367Sahrens 20185367Sahrens *cp = '\0'; 20195367Sahrens h = zfs_open(hdl, zc.zc_value, 20205367Sahrens ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME); 20216476Sahl if (h != NULL) { 20225367Sahrens if (h->zfs_type == ZFS_TYPE_VOLUME) { 20237366STim.Haley@Sun.COM *cp = '@'; 202410200SLori.Alt@Sun.COM } else if (newfs || stream_avl) { 20257366STim.Haley@Sun.COM /* 20267366STim.Haley@Sun.COM * Track the first/top of hierarchy fs, 20277366STim.Haley@Sun.COM * for mounting and sharing later. 20287366STim.Haley@Sun.COM */ 20297366STim.Haley@Sun.COM if (top_zfs && *top_zfs == NULL) 20307366STim.Haley@Sun.COM *top_zfs = zfs_strdup(hdl, zc.zc_value); 20315367Sahrens } 20326476Sahl zfs_close(h); 20335367Sahrens } 20347366STim.Haley@Sun.COM *cp = '@'; 20355367Sahrens } 20365367Sahrens 20375367Sahrens if (clp) { 20385367Sahrens err |= changelist_postfix(clp); 20395367Sahrens changelist_free(clp); 20405367Sahrens } 20415367Sahrens 20425367Sahrens if (err || ioctl_err) 20435367Sahrens return (-1); 20445367Sahrens 20455367Sahrens if (flags.verbose) { 20465367Sahrens char buf1[64]; 20475367Sahrens char buf2[64]; 20485367Sahrens uint64_t bytes = zc.zc_cookie; 20495367Sahrens time_t delta = time(NULL) - begin_time; 20505367Sahrens if (delta == 0) 20515367Sahrens delta = 1; 20525367Sahrens zfs_nicenum(bytes, buf1, sizeof (buf1)); 20535367Sahrens zfs_nicenum(bytes/delta, buf2, sizeof (buf1)); 20545367Sahrens 20555367Sahrens (void) printf("received %sB stream in %lu seconds (%sB/sec)\n", 20565367Sahrens buf1, delta, buf2); 20575367Sahrens } 20585367Sahrens 20595367Sahrens return (0); 20605367Sahrens } 20615367Sahrens 20627366STim.Haley@Sun.COM static int 20637366STim.Haley@Sun.COM zfs_receive_impl(libzfs_handle_t *hdl, const char *tosnap, recvflags_t flags, 20647366STim.Haley@Sun.COM int infd, avl_tree_t *stream_avl, char **top_zfs) 20655367Sahrens { 20665367Sahrens int err; 20675367Sahrens dmu_replay_record_t drr, drr_noswap; 20685367Sahrens struct drr_begin *drrb = &drr.drr_u.drr_begin; 20695367Sahrens char errbuf[1024]; 20705367Sahrens zio_cksum_t zcksum = { 0 }; 20715367Sahrens 20725367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 20735367Sahrens "cannot receive")); 20745367Sahrens 20755367Sahrens if (flags.isprefix && 20765367Sahrens !zfs_dataset_exists(hdl, tosnap, ZFS_TYPE_DATASET)) { 20775367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "specified fs " 20785367Sahrens "(%s) does not exist"), tosnap); 20795367Sahrens return (zfs_error(hdl, EZFS_NOENT, errbuf)); 20805367Sahrens } 20815367Sahrens 20825367Sahrens /* read in the BEGIN record */ 20835367Sahrens if (0 != (err = recv_read(hdl, infd, &drr, sizeof (drr), B_FALSE, 20845367Sahrens &zcksum))) 20855367Sahrens return (err); 20865367Sahrens 20875367Sahrens if (drr.drr_type == DRR_END || drr.drr_type == BSWAP_32(DRR_END)) { 20885367Sahrens /* It's the double end record at the end of a package */ 20895367Sahrens return (ENODATA); 20905367Sahrens } 20915367Sahrens 20925367Sahrens /* the kernel needs the non-byteswapped begin record */ 20935367Sahrens drr_noswap = drr; 20945367Sahrens 20955446Sahrens flags.byteswap = B_FALSE; 20965367Sahrens if (drrb->drr_magic == BSWAP_64(DMU_BACKUP_MAGIC)) { 20975367Sahrens /* 20985367Sahrens * We computed the checksum in the wrong byteorder in 20995367Sahrens * recv_read() above; do it again correctly. 21005367Sahrens */ 21015367Sahrens bzero(&zcksum, sizeof (zio_cksum_t)); 21025367Sahrens fletcher_4_incremental_byteswap(&drr, sizeof (drr), &zcksum); 21035367Sahrens flags.byteswap = B_TRUE; 21045367Sahrens 21055367Sahrens drr.drr_type = BSWAP_32(drr.drr_type); 21065367Sahrens drr.drr_payloadlen = BSWAP_32(drr.drr_payloadlen); 21075367Sahrens drrb->drr_magic = BSWAP_64(drrb->drr_magic); 21085367Sahrens drrb->drr_version = BSWAP_64(drrb->drr_version); 21095367Sahrens drrb->drr_creation_time = BSWAP_64(drrb->drr_creation_time); 21105367Sahrens drrb->drr_type = BSWAP_32(drrb->drr_type); 21115367Sahrens drrb->drr_flags = BSWAP_32(drrb->drr_flags); 21125367Sahrens drrb->drr_toguid = BSWAP_64(drrb->drr_toguid); 21135367Sahrens drrb->drr_fromguid = BSWAP_64(drrb->drr_fromguid); 21145367Sahrens } 21155367Sahrens 21165367Sahrens if (drrb->drr_magic != DMU_BACKUP_MAGIC || drr.drr_type != DRR_BEGIN) { 21175367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid " 21185367Sahrens "stream (bad magic number)")); 21195367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, errbuf)); 21205367Sahrens } 21215367Sahrens 21225367Sahrens if (strchr(drrb->drr_toname, '@') == NULL) { 21235367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid " 21245367Sahrens "stream (bad snapshot name)")); 21255367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, errbuf)); 21265367Sahrens } 21275367Sahrens 21285367Sahrens if (drrb->drr_version == DMU_BACKUP_STREAM_VERSION) { 21295367Sahrens return (zfs_receive_one(hdl, infd, tosnap, flags, 21307366STim.Haley@Sun.COM &drr, &drr_noswap, stream_avl, top_zfs)); 21315367Sahrens } else if (drrb->drr_version == DMU_BACKUP_HEADER_VERSION) { 21325367Sahrens return (zfs_receive_package(hdl, infd, tosnap, flags, 21337366STim.Haley@Sun.COM &drr, &zcksum, top_zfs)); 21345367Sahrens } else { 21355367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 21365367Sahrens "stream is unsupported version %llu"), 21375367Sahrens drrb->drr_version); 21385367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, errbuf)); 21395367Sahrens } 21405367Sahrens } 21417366STim.Haley@Sun.COM 21427366STim.Haley@Sun.COM /* 21437366STim.Haley@Sun.COM * Restores a backup of tosnap from the file descriptor specified by infd. 21447366STim.Haley@Sun.COM * Return 0 on total success, -2 if some things couldn't be 21457366STim.Haley@Sun.COM * destroyed/renamed/promoted, -1 if some things couldn't be received. 21467366STim.Haley@Sun.COM * (-1 will override -2). 21477366STim.Haley@Sun.COM */ 21487366STim.Haley@Sun.COM int 21497366STim.Haley@Sun.COM zfs_receive(libzfs_handle_t *hdl, const char *tosnap, recvflags_t flags, 21507366STim.Haley@Sun.COM int infd, avl_tree_t *stream_avl) 21517366STim.Haley@Sun.COM { 21527366STim.Haley@Sun.COM char *top_zfs = NULL; 21537366STim.Haley@Sun.COM int err; 21547366STim.Haley@Sun.COM 21557366STim.Haley@Sun.COM err = zfs_receive_impl(hdl, tosnap, flags, infd, stream_avl, &top_zfs); 21567366STim.Haley@Sun.COM 21578584SLori.Alt@Sun.COM if (err == 0 && !flags.nomount && top_zfs) { 21587366STim.Haley@Sun.COM zfs_handle_t *zhp; 21597366STim.Haley@Sun.COM prop_changelist_t *clp; 21607366STim.Haley@Sun.COM 21617366STim.Haley@Sun.COM zhp = zfs_open(hdl, top_zfs, ZFS_TYPE_FILESYSTEM); 21627366STim.Haley@Sun.COM if (zhp != NULL) { 21637366STim.Haley@Sun.COM clp = changelist_gather(zhp, ZFS_PROP_MOUNTPOINT, 21647366STim.Haley@Sun.COM CL_GATHER_MOUNT_ALWAYS, 0); 21657366STim.Haley@Sun.COM zfs_close(zhp); 21667366STim.Haley@Sun.COM if (clp != NULL) { 21677366STim.Haley@Sun.COM /* mount and share received datasets */ 21687366STim.Haley@Sun.COM err = changelist_postfix(clp); 21697366STim.Haley@Sun.COM changelist_free(clp); 21707366STim.Haley@Sun.COM } 21717366STim.Haley@Sun.COM } 21727366STim.Haley@Sun.COM if (zhp == NULL || clp == NULL || err) 21737366STim.Haley@Sun.COM err = -1; 21747366STim.Haley@Sun.COM } 21757366STim.Haley@Sun.COM if (top_zfs) 21767366STim.Haley@Sun.COM free(top_zfs); 21777366STim.Haley@Sun.COM 21787366STim.Haley@Sun.COM return (err); 21797366STim.Haley@Sun.COM } 2180